Initial

conn.go

@@ -0,0 +1,122 @@
+/*
+Open Source Initiative OSI - The MIT License (MIT):Licensing
+
+The MIT License (MIT)
+Copyright (c) 2013 DutchCoders <http://github.com/dutchcoders/>
+
+Permission is hereby granted, free of charge, to any person obtaining a copy of
+this software and associated documentation files (the "Software"), to deal in
+the Software without restriction, including without limitation the rights to
+use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies
+of the Software, and to permit persons to whom the Software is furnished to do
+so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.
+*/
+
+package clamd
+
+import (
+	"bufio"
+	"fmt"
+	"io"
+	"net"
+	"strings"
+	"sync"
+)
+
+const CHUNK_SIZE = 1024
+
+type CLAMDConn struct {
+	net.Conn
+}
+
+func (conn *CLAMDConn) sendCommand(command string) error {
+	commandBytes := []byte(fmt.Sprintf("n%s\n", command))
+
+	_, err := conn.Write(commandBytes)
+	return err
+}
+
+func (conn *CLAMDConn) sendEOF() error {
+	_, err := conn.Write([]byte{0, 0, 0, 0})
+	return err
+}
+
+func (conn *CLAMDConn) sendChunk(data []byte) error {
+	var buf [4]byte
+	lenData := len(data)
+	buf[0] = byte(lenData >> 24)
+	buf[1] = byte(lenData >> 16)
+	buf[2] = byte(lenData >> 8)
+	buf[3] = byte(lenData >> 0)
+
+	a := buf
+
+	b := make([]byte, len(a))
+	for i := range a {
+		b[i] = a[i]
+	}
+
+	conn.Write(b)
+
+	_, err := conn.Write(data)
+	return err
+}
+
+func (c *CLAMDConn) readResponse() (chan string, sync.WaitGroup, error) {
+	var wg sync.WaitGroup
+
+	wg.Add(1)
+
+	// read data
+	reader := bufio.NewReader(c)
+
+	// reading
+	ch := make(chan string)
+
+	// var dataArrays []string
+	go func() {
+		fmt.Println("Start")
+		defer func() {
+			fmt.Println("Cleaning up")
+			close(ch)
+			wg.Done()
+		}()
+
+		for {
+			line, err := reader.ReadString('\n')
+			if err == io.EOF {
+				return
+			}
+
+			if err != nil {
+				return
+			}
+
+			line = strings.TrimRight(line, " \t\r\n")
+
+			ch <- line
+		}
+	}()
+
+	return ch, wg, nil
+}
+
+func newCLAMDUnixConn(address string) (*CLAMDConn, error) {
+	conn, err := net.Dial("unix", address)
+	if err != nil {
+		return nil, err
+	}
+
+	return &CLAMDConn{Conn: conn}, err
+}