overflow_discovery/service/matcher/ftp/ftp_test.go

package ftp

import (
	"fmt"
	"testing"

	"git.loafle.net/overflow/overflow_discovery/service/matcher"

	"net"
)

//type FTP struct {
//    conn net.Conn
//    addr string
//
//    reader *bufio.Reader
//    writer *bufio.Writer
//}
//
//func (ftp *FTP) Close()  {
//    ftp.conn.Close()
//}
//
//func Connect(addr string) (*FTP, error)  {
//    var err error
//    var conn net.Conn
//
//    if conn, err = net.Dial("tcp", addr); err != nil {
//        return nil, err
//    }
//
//    writer := bufio.NewWriter(conn)
//    reader := bufio.NewReader(conn)
//
//    obj := &FTP{
//        conn:conn,
//        addr:addr,
//        reader:reader,
//        writer:writer,
//    }
//    recv, _ := obj.receive()
//
//    fmt.Println(recv)
//
//    return obj, nil
//
//}
//
//func (ftp *FTP) receive() (string, error) {
//    line, err := ftp.receiveLine()
//
//    if err != nil {
//        return line, err
//    }
//
//    fmt.Println("len : ", len(line))
//    fmt.Println("line[3] :", line[3])
//    //
//    //if (len(line) >= 4) && (line[3] == '-') {
//    //    closingCode := line[:3] + " "
//    //
//    //    for {
//    //        str, err := ftp.receiveLine()
//    //        fmt.Println("str pre: ", str)
//    //        line = line + str
//    //        fmt.Println("str after: ", line)
//    //        if err != nil {
//    //            return line, err
//    //        }
//    //
//    //        if len(str) < 4 {
//    //            fmt.Println("Uncorrectly terminated response")
//    //        }else {
//    //            if str[:4] == closingCode {
//    //                break
//    //            }
//    //        }
//    //    }
//    //}
//
//    ftp.ReadAndDiscard()
//
//    fmt.Println("receive line: ", line)
//    return line, err
//}
//
//func (ftp *FTP) ReadAndDiscard() (int, error) {
//    var i int
//    bufferSize := ftp.reader.Buffered()
//
//    for i = 0; i < bufferSize ; i++  {
//        if _, err := ftp.reader.ReadByte(); err != nil {
//            return i, err
//        }
//    }
//
//    return i, nil
//}
//
//func (ftp *FTP) send(command string, arguments ...interface{}) error {
//
//    command = fmt.Sprintf(command)
//    command += "\r\n"
//
//    if _, err := ftp.writer.WriteString(command); err != nil {
//        return err
//    }
//
//    if err := ftp.writer.Flush(); err != nil {
//        return err
//    }
//
//    return nil
//}
//
//func (ftp *FTP) cmd(expects string, command string, args ...interface{}) (line string, err error)  {
//
//    if err = ftp.send(command, args); err != nil {
//        return
//    }
//
//    if line, err = ftp.receive(); err != nil {
//        return
//    }
//
//
//    if !strings.HasPrefix(line, expects) {
//        err = errors.New(line)
//        return
//    }
//
//    return
//}
//
//func (ftp *FTP) receiveLine() (string, error)  {
//    line, err := ftp.reader.ReadString('\n')
//
//    log.Printf("< %s", line)
//
//    return line, err
//}
//
//func (ftp *FTP) Syst() (line string, err error) {
//    if err := ftp.send("SYST"); err != nil {
//        return "", err
//    }
//
//    if line, err = ftp.receive(); err != nil {
//        return
//    }
//
//    if !strings.HasPrefix(line, "215") {
//        err = errors.New(line)
//        return
//    }
//
//    return strings.SplitN(strings.TrimSpace(line), " ", 2)[1], nil
//}

//func TestFtp(t *testing.T)  {
//    var err error
//    var ftp *FTP
//    //var f *FTPMatcher
//
//    if ftp, err = Connect("192.168.1.202:21"); err != nil {
//        panic(err)
//    }
//
//    //f.Match(0, nil,nil)
//    ftp.Syst()
//    ftp.cmd("503","PASS ")
//    ftp.cmd("221","QUIT")
//    defer ftp.Close()
//}

func TestMatchFTP(t *testing.T) {
	ftm := NewMatcher()
	//fmt.Println(ftm)
	//fmt.Println(ftm.sendPackets[0])

	//log.LoadLogConfig("../../../../../../../../bin/log.xml")
	//defer log.Flush()

	//port := types.NewPort("21", types.NewHost("192.168.1.202"), types.TYPE_TCP)

	info := matcher.NewMatchInfo("192.168.1.15", 21)
	//
	//var ipport string
	//ipport = port.Host.Ip + ":" + string(port.Port)
	//
	//log.Debug(ipport)

	client, _ := net.Dial("tcp", "192.168.1.15:21")

	defer client.Close()

	//reader := bufio.NewReader(client)
	//writer := bufio.NewWriter(client)

	fmt.Println(ftm.PacketCount())
	//fmt.Println(reader.ReadString('\n'))

	bytes := make([]byte, 512)

	le, _ := client.Read(bytes)

	fmt.Println(bytes)

	b := ftm.Match(info, 0, matcher.NewPacket(bytes, le))

	fmt.Println(b)

	for ii := 0; ii < ftm.PacketCount(); ii++ {
		pack := ftm.Packet(ii)

		fmt.Println(pack)

		//writer.WriteString(pack)
		client.Write(pack.Buffer)
		//fmt.Println(reader.ReadString('\n'))
		bytes := make([]byte, 512)

		l, _ := client.Read(bytes)

		//fmt.Println(bytes)

		b := ftm.Match(info, ii+1, matcher.NewPacket(bytes, l))

		fmt.Println(b)

	}

	fmt.Println("Service Name : ", ftm.ServiceName())
}