package ftp import ( "fmt" "testing" cnsm "git.loafle.net/commons_go/network_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 := cnsm.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, cnsm.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, cnsm.NewPacket(bytes, l)) fmt.Println(b) } fmt.Println("Service Name : ", ftm.ServiceName()) }