ing
This commit is contained in:
crusader
142
matcher/ftp/ftp.go
Normal file
142
matcher/ftp/ftp.go
Normal file
@@ -0,0 +1,142 @@
|
||||
package ftp
|
||||
|
||||
import (
|
||||
"git.loafle.net/overflow/overflow_discovery/match/packet"
|
||||
"git.loafle.net/overflow/overflow_discovery/model/scaninfo"
|
||||
log "github.com/cihub/seelog"
|
||||
)
|
||||
|
||||
// FTP Status codes, defined in RFC 959
|
||||
const (
|
||||
statusReadyServer = "120"
|
||||
statusOK = "200"
|
||||
statusNewConnectOK = "220"
|
||||
statusSystemNameOK = "215"
|
||||
statusCloseConnect = "221"
|
||||
statusUnkownCMD = "202"
|
||||
statusTlsUseOK = "234"
|
||||
statusCloseControlConnect = "421"
|
||||
statusSyntaxErr = "500"
|
||||
statusParamSyntaxErr = "501"
|
||||
statusNotUseCMD = "502"
|
||||
statusIncorrectCMD = "503"
|
||||
statusTlsNotUse = "534"
|
||||
statusNeedUserId = "332"
|
||||
)
|
||||
|
||||
type FTPMatcher struct {
|
||||
sendPackets []*packet.Packet
|
||||
isFtps bool
|
||||
}
|
||||
|
||||
func (ftp *FTPMatcher) Match(index int, packet *packet.Packet, info scaninfo.ServiceScanInfo) bool {
|
||||
result := false
|
||||
|
||||
if packet == nil || packet.Buffer == nil || packet.Len == 0 {
|
||||
log.Error("Packet nil")
|
||||
return result
|
||||
}
|
||||
|
||||
str := string(packet.Buffer)
|
||||
//fmt.Println(str)
|
||||
code := str[:3]
|
||||
|
||||
if index == 0 {
|
||||
switch code {
|
||||
case statusNewConnectOK, statusReadyServer:
|
||||
//fmt.Println(code)
|
||||
result = true
|
||||
break
|
||||
}
|
||||
} else if index == 1 {
|
||||
switch code {
|
||||
case statusSystemNameOK, statusSyntaxErr, statusParamSyntaxErr, statusNotUseCMD:
|
||||
//fmt.Println(code)
|
||||
result = true
|
||||
break
|
||||
}
|
||||
} else if index == 2 {
|
||||
switch code {
|
||||
case statusIncorrectCMD, statusParamSyntaxErr, statusNotUseCMD, statusNeedUserId:
|
||||
//fmt.Println(code)
|
||||
result = true
|
||||
break
|
||||
}
|
||||
} else if index == 3 {
|
||||
switch code {
|
||||
case statusCloseConnect, statusSyntaxErr:
|
||||
//fmt.Println(code)
|
||||
result = true
|
||||
break
|
||||
}
|
||||
}
|
||||
|
||||
if index == 3 && result == true {
|
||||
var err error
|
||||
var isfs bool
|
||||
|
||||
//fmt.Println(info.Port.Host.Ip, info.Port.Port)
|
||||
|
||||
isfs, err = StartCheckFTPS(info.GetIP(), info.GetPort())
|
||||
|
||||
if isfs && err == nil {
|
||||
ftp.isFtps = isfs
|
||||
} else if err != nil {
|
||||
log.Warn("FTPS Check Error : ", err.Error())
|
||||
}
|
||||
}
|
||||
|
||||
return result
|
||||
}
|
||||
|
||||
func (ftp *FTPMatcher) PacketCount() int {
|
||||
return len(ftp.sendPackets)
|
||||
}
|
||||
|
||||
func (ftp *FTPMatcher) Packet(index int) *packet.Packet {
|
||||
return ftp.sendPackets[index]
|
||||
}
|
||||
|
||||
func (ftp *FTPMatcher) ServiceName() string {
|
||||
re := ""
|
||||
if ftp.isFtps {
|
||||
re = "FTPS"
|
||||
} else {
|
||||
re = "FTP"
|
||||
}
|
||||
return re
|
||||
}
|
||||
|
||||
func (ftp *FTPMatcher) IsError(index int, packet *packet.Packet, info scaninfo.ServiceScanInfo) bool {
|
||||
return false
|
||||
}
|
||||
|
||||
func (ftp *FTPMatcher) HasResponse(index int) bool {
|
||||
return false
|
||||
}
|
||||
|
||||
func (ftp *FTPMatcher) IsPrePacket() bool {
|
||||
return true
|
||||
}
|
||||
|
||||
func NewFTPMatcher() *FTPMatcher {
|
||||
|
||||
ftm := FTPMatcher{}
|
||||
|
||||
sysStr := "SYST\r\n"
|
||||
systByte := make([]byte, len(sysStr))
|
||||
copy(systByte[:], sysStr)
|
||||
ftm.sendPackets = append(ftm.sendPackets, packet.NewPacket(systByte, len(sysStr)))
|
||||
|
||||
passStr := "PASS \r\n"
|
||||
passByte := make([]byte, len(passStr))
|
||||
copy(passByte[:], passStr)
|
||||
ftm.sendPackets = append(ftm.sendPackets, packet.NewPacket(passByte, len(passStr)))
|
||||
|
||||
quitStr := "QUIT\r\n"
|
||||
quitByte := make([]byte, len(quitStr))
|
||||
copy(quitByte[:], quitStr)
|
||||
ftm.sendPackets = append(ftm.sendPackets, packet.NewPacket(quitByte, len(quitStr)))
|
||||
|
||||
return &ftm
|
||||
}
|
||||
234
matcher/ftp/ftp_test.go
Normal file
234
matcher/ftp/ftp_test.go
Normal file
@@ -0,0 +1,234 @@
|
||||
package ftp
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"testing"
|
||||
"git.loafle.net/overflow/overflow_discovery/model/scaninfo"
|
||||
"git.loafle.net/overflow/overflow_discovery/match/packet"
|
||||
|
||||
"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 := NewFTPMatcher()
|
||||
//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 := scaninfo.NewScanInfoImpl("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(0, packet.NewPacket(bytes, le), nil)
|
||||
|
||||
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(ii+1, packet.NewPacket(bytes, l), info)
|
||||
|
||||
fmt.Println(b)
|
||||
|
||||
}
|
||||
|
||||
fmt.Println("Service Name : ", ftm.ServiceName())
|
||||
}
|
||||
Reference in New Issue
Block a user