Initial

clamd.go

@@ -0,0 +1,275 @@
+/*
+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 (
+	"errors"
+	"fmt"
+	"io"
+	"strings"
+)
+
+type Clamd struct {
+	address string
+}
+
+type Stats struct {
+	Pools    string
+	State    string
+	Threads  string
+	Memstats string
+	Queue    string
+}
+
+var EICAR = []byte(`X5O!P%@AP[4\PZX54(P^)7CC)7}$EICAR-STANDARD-ANTIVIRUS-TEST-FILE!$H+H*`)
+
+func (c *Clamd) newConnection() (*CLAMDConn, error) {
+	conn, err := newCLAMDUnixConn(c.address)
+	return conn, err
+}
+
+func (c *Clamd) simpleCommand(command string) (chan string, error) {
+	conn, err := newCLAMDUnixConn(c.address)
+	if err != nil {
+		return nil, err
+	}
+
+	// defer conn.Close()
+
+	err = conn.sendCommand(command)
+	if err != nil {
+		return nil, err
+	}
+
+	ch, wg, err := conn.readResponse()
+
+	go func() {
+		// wait for waitgroup
+		wg.Wait()
+
+		// close connection
+		conn.Close()
+	}()
+
+	return ch, err
+}
+
+/*
+Check the daemon's state (should reply with PONG).
+*/
+func (c *Clamd) Ping() error {
+	ch, err := c.simpleCommand("PING")
+	if err != nil {
+		return err
+	}
+
+	select {
+	case s := (<-ch):
+		switch s {
+		case "PONG":
+			return nil
+		default:
+			return errors.New(fmt.Sprintf("Invalid response, got %s.", s))
+		}
+	}
+
+	return nil
+}
+
+/*
+Print program and database versions.
+*/
+func (c *Clamd) Version() (chan string, error) {
+	dataArrays, err := c.simpleCommand("VERSION")
+	return dataArrays, err
+}
+
+/*
+On this command clamd provides statistics about the scan queue, contents of scan
+queue, and memory usage. The exact reply format is subject to changes in future
+releases.
+*/
+func (c *Clamd) Stats() (*Stats, error) {
+	ch, err := c.simpleCommand("STATS")
+	if err != nil {
+		return nil, err
+	}
+
+	stats := &Stats{}
+
+	for s := range ch {
+		if strings.HasPrefix(s, "POOLS") {
+			stats.Pools = strings.Trim(s[6:], " ")
+		} else if strings.HasPrefix(s, "STATE") {
+			stats.State = s
+		} else if strings.HasPrefix(s, "THREADS") {
+			stats.Threads = s
+		} else if strings.HasPrefix(s, "QUEUE") {
+			stats.Queue = s
+		} else if strings.HasPrefix(s, "MEMSTATS") {
+			stats.Memstats = s
+		} else if strings.HasPrefix(s, "END") {
+		} else {
+			//	return nil, errors.New(fmt.Sprintf("Unknown response, got %s.", s))
+		}
+	}
+
+	return stats, nil
+}
+
+/*
+Reload the databases.
+*/
+func (c *Clamd) Reload() error {
+	ch, err := c.simpleCommand("RELOAD")
+	if err != nil {
+		return err
+	}
+
+	select {
+	case s := (<-ch):
+		switch s {
+		case "RELOADING":
+			return nil
+		default:
+			return errors.New(fmt.Sprintf("Invalid response, got %s.", s))
+		}
+	}
+
+	return nil
+}
+
+func (c *Clamd) Shutdown() error {
+	_, err := c.simpleCommand("SHUTDOWN")
+	if err != nil {
+		return err
+	}
+
+	return err
+}
+
+/*
+Scan file or directory (recursively) with archive support enabled (a full path is
+required).
+*/
+func (c *Clamd) ScanFile(path string) (chan string, error) {
+	command := fmt.Sprintf("SCAN %s", path)
+	ch, err := c.simpleCommand(command)
+	return ch, err
+}
+
+/*
+Scan file or directory (recursively) with archive and special file support disabled
+(a full path is required).
+*/
+func (c *Clamd) RawScanFile(path string) (chan string, error) {
+	command := fmt.Sprintf("RAWSCAN %s", path)
+	ch, err := c.simpleCommand(command)
+	return ch, err
+}
+
+/*
+Scan file in a standard way or scan directory (recursively) using multiple threads
+(to make the scanning faster on SMP machines).
+*/
+func (c *Clamd) MultiScanFile(path string) (chan string, error) {
+	command := fmt.Sprintf("MULTISCAN %s", path)
+	ch, err := c.simpleCommand(command)
+	return ch, err
+}
+
+/*
+Scan file or directory (recursively) with archive support enabled and don’t stop
+the scanning when a virus is found.
+*/
+func (c *Clamd) ContScanFile(path string) (chan string, error) {
+	command := fmt.Sprintf("CONTSCAN %s", path)
+	ch, err := c.simpleCommand(command)
+	return ch, err
+}
+
+/*
+Scan file or directory (recursively) with archive support enabled and don’t stop
+the scanning when a virus is found.
+*/
+func (c *Clamd) AllMatchScanFile(path string) (chan string, error) {
+	command := fmt.Sprintf("ALLMATCHSCAN %s", path)
+	ch, err := c.simpleCommand(command)
+	return ch, err
+}
+
+/*
+Scan a stream of data. The stream is sent to clamd in chunks, after INSTREAM,
+on the same socket on which the command was sent. This avoids the overhead
+of establishing new TCP connections and problems with NAT. The format of the
+chunk is: <length><data> where <length> is the size of the following data in
+bytes expressed as a 4 byte unsigned integer in network byte order and <data> is
+the actual chunk. Streaming is terminated by sending a zero-length chunk. Note:
+do not exceed StreamMaxLength as defined in clamd.conf, otherwise clamd will
+reply with INSTREAM size limit exceeded and close the connection
+*/
+func (c *Clamd) ScanStream(r io.Reader) (chan string, error) {
+	conn, err := c.newConnection()
+	if err != nil {
+		return nil, err
+	}
+
+	conn.sendCommand("INSTREAM")
+
+	for {
+		buf := make([]byte, CHUNK_SIZE)
+
+		nr, err := r.Read(buf)
+		if err != nil {
+			break
+		}
+
+		if nr == 0 {
+			break
+		}
+
+		conn.sendChunk(buf[:nr])
+	}
+
+	err = conn.sendEOF()
+	if err != nil {
+		return nil, err
+	}
+
+	ch, wg, err := conn.readResponse()
+
+	go func() {
+		wg.Wait()
+		conn.Close()
+	}()
+
+	return ch, nil
+}
+
+func NewClamd(address string) *Clamd {
+	clamd := &Clamd{address: address}
+	return clamd
+}

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
+}

examples/main.go

@@ -0,0 +1,72 @@
+/*
+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 main
+
+import (
+	_ "bytes"
+	"fmt"
+	"github.com/dutchcoders/go-clamd"
+)
+
+func main() {
+	fmt.Println("Made with <3 DutchCoders")
+
+	c := clamd.NewClamd("/tmp/clamd.socket")
+	_ = c
+
+	/*
+		reader := bytes.NewReader(clamd.EICAR)
+		response, err := c.ScanStream(reader)
+
+		for s := range response {
+			fmt.Printf("%v %v\n", s, err)
+		}
+
+		response, err = c.ScanFile(".")
+
+		for s := range response {
+			fmt.Printf("%v %v\n", s, err)
+		}
+
+		response, err = c.Version()
+
+		for s := range response {
+			fmt.Printf("%v %v\n", s, err)
+		}
+	*/
+
+	err := c.Ping()
+	fmt.Printf("Ping: %v\n", err)
+
+	stats, err := c.Stats()
+	fmt.Printf("%v %v\n", stats, err)
+
+	err = c.Reload()
+	fmt.Printf("Reload: %v\n", err)
+
+	// response, err = c.Shutdown()
+	// fmt.Println(response)
+}