255 lines
8.1 KiB
Go
255 lines
8.1 KiB
Go
package ipv4
|
|
|
|
import (
|
|
"fmt"
|
|
"net"
|
|
"time"
|
|
|
|
"git.loafle.net/commons/logging-go"
|
|
csm "git.loafle.net/commons/service_matcher-go"
|
|
cuej "git.loafle.net/commons/util-go/encoding/json"
|
|
occc "git.loafle.net/overflow/commons-go/core/constants"
|
|
occu "git.loafle.net/overflow/commons-go/core/util"
|
|
ocmd "git.loafle.net/overflow/commons-go/model/discovery"
|
|
"git.loafle.net/overflow/container_discovery/internal/matcher"
|
|
)
|
|
|
|
func scanServiceTCP(port *ocmd.Port, ds *ocmd.DiscoverService, resultChan chan interface{}, errChan chan error, stopChan chan struct{}) bool {
|
|
hostIP := port.Host.IPV4
|
|
portNumber, err := cuej.NumberToInt(port.PortNumber)
|
|
if err != nil {
|
|
errChan <- fmt.Errorf("Discovery: Service scan on %s:%s error has occurred %v ", hostIP, port.PortNumber, err)
|
|
return false
|
|
}
|
|
|
|
info := csm.NewMatchInfo(hostIP, portNumber)
|
|
var s *ocmd.Service
|
|
|
|
scs := []serviceConnector{
|
|
&normalServiceConn{
|
|
cryptoType: occc.CryptoTypeNONE.String(),
|
|
},
|
|
&tlsServiceConn{
|
|
cryptoType: occc.CryptoTypeTLS.String(),
|
|
},
|
|
}
|
|
|
|
LOOP:
|
|
for i := 0; i < len(scs); i++ {
|
|
sc := scs[i]
|
|
|
|
conn, err := sc.Dial(hostIP, portNumber)
|
|
if err != nil {
|
|
errChan <- fmt.Errorf("Discovery: Service scan[%s] on %s:%d error has occurred %v ", sc.CryptoType(), hostIP, portNumber, err)
|
|
return false
|
|
}
|
|
logging.Logger().Debugf("Discovery: Service scan connected[%s:%d] %s", hostIP, portNumber, sc.CryptoType())
|
|
|
|
buf := make([]byte, 1024)
|
|
|
|
if err := conn.SetReadDeadline(time.Now().Add(1 * time.Second)); nil != err {
|
|
logging.Logger().Debugf("Discovery: cannot set readdeadline connected[%s:%d] %s", hostIP, portNumber, sc.CryptoType())
|
|
return false
|
|
}
|
|
rn, err := conn.Read(buf)
|
|
if err != nil {
|
|
rn = 0
|
|
}
|
|
if rn != 0 {
|
|
s = hadlePrePacket(info, sc, conn, csm.NewPacket(buf, rn))
|
|
} else {
|
|
conn.Close()
|
|
s = hadlePostPacket(info, sc)
|
|
}
|
|
|
|
if nil != s {
|
|
break LOOP
|
|
}
|
|
|
|
select {
|
|
case <-stopChan:
|
|
return false
|
|
default:
|
|
}
|
|
}
|
|
|
|
if nil != s {
|
|
s.Port = port
|
|
resultChan <- s
|
|
return true
|
|
}
|
|
|
|
return false
|
|
}
|
|
|
|
func hadlePrePacket(info csm.MatchInfo, sc serviceConnector, conn net.Conn, packet *csm.Packet) *ocmd.Service {
|
|
defer func() {
|
|
conn.Close()
|
|
}()
|
|
|
|
// logging.Logger().Debugf("Discovery: Service scan pre packet length[%d], buf[%v]", packet.Len, packet.Buffer)
|
|
logging.Logger().Debugf("Discovery: Service scan[%s] on %s:%d pre packet length[%d]", sc.CryptoType(), info.IP(), info.Port(), packet.Len)
|
|
|
|
ms := matcher.GetTCPMatchers(true)
|
|
buf := make([]byte, 1024)
|
|
var s *ocmd.Service
|
|
|
|
LOOP:
|
|
for i := 0; i < len(ms); i++ {
|
|
m := ms[i]
|
|
|
|
if m.Match(info, 0, packet) {
|
|
packetCount := m.PacketCount()
|
|
|
|
if 0 == packetCount {
|
|
s = &ocmd.Service{
|
|
ServiceName: m.Name(),
|
|
CryptoType: occc.ToCryptoType(sc.CryptoType()),
|
|
DiscoveredDate: occu.NowPtr(),
|
|
}
|
|
break LOOP
|
|
}
|
|
|
|
found := false
|
|
|
|
for j := 0; j < packetCount; j++ {
|
|
tPacket := m.Packet(j)
|
|
// logging.Logger().Debugf("Discovery: Service scan send packet length[%d], buf[%v]", tPacket.Len, tPacket.Buffer)
|
|
logging.Logger().Debugf("Discovery: Service scan[%s] on %s:%d send packet length[%d]", sc.CryptoType(), info.IP(), info.Port(), tPacket.Len)
|
|
|
|
if err := conn.SetWriteDeadline(time.Now().Add(1 * time.Second)); nil != err {
|
|
logging.Logger().Debugf("Discovery: cannot set writeDeadLine Service scan[%s] on %s:%d send packet length[%d]", sc.CryptoType(), info.IP(), info.Port(), tPacket.Len)
|
|
break LOOP
|
|
}
|
|
wn, err := conn.Write(tPacket.Buffer)
|
|
if nil != err {
|
|
logging.Logger().Debugf("Discovery: Service scan[%s] on %s:%d send packet error %v", sc.CryptoType(), info.IP(), info.Port(), err)
|
|
continue LOOP
|
|
}
|
|
if wn != tPacket.Len {
|
|
logging.Logger().Debugf("Discovery: Service scan[%s] on %s:%d send packet length[%d] not same with %d", sc.CryptoType(), info.IP(), info.Port(), wn, tPacket.Len)
|
|
continue LOOP
|
|
}
|
|
|
|
if err := conn.SetReadDeadline(time.Now().Add(1 * time.Second)); nil != err {
|
|
logging.Logger().Debugf("Discovery: cannot set readDeadLine Service scan[%s] on %s:%d send packet length[%d]", sc.CryptoType(), info.IP(), info.Port(), tPacket.Len)
|
|
break LOOP
|
|
}
|
|
rn, err := conn.Read(buf)
|
|
if nil != err {
|
|
logging.Logger().Debugf("Discovery: Service scan[%s] on %s:%d receive packet error %v", sc.CryptoType(), info.IP(), info.Port(), err)
|
|
break LOOP
|
|
}
|
|
|
|
if m.Match(info, j+1, csm.NewPacket(buf, rn)) {
|
|
// logging.Logger().Debugf("Discovery: Service scan receive match length[%d], buf[%v]", rn, buf)
|
|
logging.Logger().Debugf("Discovery: Service scan[%s] on %s:%d receive match length[%d]", sc.CryptoType(), info.IP(), info.Port(), rn)
|
|
found = true
|
|
} else {
|
|
// logging.Logger().Debugf("Discovery: Service scan receive not match length[%d], buf[%v]", rn, buf)
|
|
logging.Logger().Debugf("Discovery: Service scan[%s] on %s:%d receive not match length[%d]", sc.CryptoType(), info.IP(), info.Port(), rn)
|
|
found = false
|
|
continue LOOP
|
|
}
|
|
}
|
|
|
|
if found {
|
|
s = &ocmd.Service{
|
|
ServiceName: m.Name(),
|
|
CryptoType: occc.ToCryptoType(sc.CryptoType()),
|
|
DiscoveredDate: occu.NowPtr(),
|
|
}
|
|
break LOOP
|
|
}
|
|
}
|
|
}
|
|
|
|
return s
|
|
}
|
|
|
|
func hadlePostPacket(info csm.MatchInfo, sc serviceConnector) *ocmd.Service {
|
|
defer func() {
|
|
}()
|
|
|
|
ms := matcher.GetTCPMatchers(false)
|
|
buf := make([]byte, 1024)
|
|
var s *ocmd.Service
|
|
|
|
LOOP:
|
|
for i := 0; i < len(ms); i++ {
|
|
m := ms[i]
|
|
|
|
conn, err := sc.Dial(info.IP(), info.Port())
|
|
if err != nil {
|
|
logging.Logger().Debugf("Discovery: Service scan[%s] on %s:%d socket dial error %v", sc.CryptoType(), info.IP(), info.Port(), err)
|
|
break LOOP
|
|
}
|
|
|
|
packetCount := m.PacketCount()
|
|
for j := 0; j < packetCount; j++ {
|
|
tPacket := m.Packet(j)
|
|
// logging.Logger().Debugf("Discovery: Service scan send packet length[%d], buf[%v]", tPacket.Len, tPacket.Buffer)
|
|
logging.Logger().Debugf("Discovery: Service scan[%s] on %s:%d send packet length[%d]", sc.CryptoType(), info.IP(), info.Port(), tPacket.Len)
|
|
|
|
if err := conn.SetWriteDeadline(time.Now().Add(1 * time.Second)); nil != err {
|
|
logging.Logger().Debugf("Discovery: cannot set writeDeadLine Service scan[%s] on %s:%d send packet length[%d]", sc.CryptoType(), info.IP(), info.Port(), tPacket.Len)
|
|
break
|
|
}
|
|
wn, err := conn.Write(tPacket.Buffer)
|
|
if nil != err {
|
|
logging.Logger().Debugf("Discovery: Service scan[%s] on %s:%d send packet error %v", sc.CryptoType(), info.IP(), info.Port(), err)
|
|
break
|
|
}
|
|
if wn != tPacket.Len {
|
|
logging.Logger().Debugf("Discovery: Service scan[%s] on %s:%d send packet length[%d] not same with %d", sc.CryptoType(), info.IP(), info.Port(), wn, tPacket.Len)
|
|
break
|
|
}
|
|
|
|
if err := conn.SetReadDeadline(time.Now().Add(1 * time.Second)); nil != err {
|
|
logging.Logger().Debugf("Discovery: cannot set readDeadLine Service scan[%s] on %s:%d send packet length[%d]", sc.CryptoType(), info.IP(), info.Port(), tPacket.Len)
|
|
break
|
|
}
|
|
rn, err := conn.Read(buf)
|
|
if nil != err {
|
|
if !m.HasResponse(j) {
|
|
s = &ocmd.Service{
|
|
ServiceName: m.Name(),
|
|
CryptoType: occc.ToCryptoType(sc.CryptoType()),
|
|
DiscoveredDate: occu.NowPtr(),
|
|
}
|
|
break
|
|
}
|
|
|
|
logging.Logger().Debugf("Discovery: Service scan[%s] on %s:%d receive packet error %v", sc.CryptoType(), info.IP(), info.Port(), err)
|
|
break
|
|
}
|
|
|
|
if m.Match(info, j, csm.NewPacket(buf, rn)) {
|
|
if packetCount-1 == j {
|
|
s = &ocmd.Service{
|
|
ServiceName: m.Name(),
|
|
CryptoType: occc.ToCryptoType(sc.CryptoType()),
|
|
DiscoveredDate: occu.NowPtr(),
|
|
}
|
|
break
|
|
}
|
|
|
|
// logging.Logger().Debugf("Discovery: Service scan receive match length[%d], buf[%v]", rn, buf)
|
|
logging.Logger().Debugf("Discovery: Service scan[%s] on %s:%d receive match length[%d]", sc.CryptoType(), info.IP(), info.Port(), rn)
|
|
continue
|
|
} else {
|
|
// logging.Logger().Debugf("Discovery: Service scan receive not match length[%d], buf[%v]", rn, buf)
|
|
logging.Logger().Debugf("Discovery: Service scan[%s] on %s:%d receive not match length[%d]", sc.CryptoType(), info.IP(), info.Port(), rn)
|
|
break
|
|
}
|
|
}
|
|
conn.Close()
|
|
|
|
if nil != s {
|
|
break LOOP
|
|
}
|
|
}
|
|
|
|
return s
|
|
}
|