package smb_protocol_crawler

import (
	"encoding/json"
	config "git.loafle.net/overflow/overflow_probe/agent_api/config_manager"
	"git.loafle.net/overflow/overflow_probe/crawler"
	"git.loafle.net/overflow/overflow_probe/matcher/smb"
)

type SMBHeahthCrawler struct {
	crawler.SocketHeahthCrawler
}

func (r *SMBHeahthCrawler) Internal(params config.Config) ([]byte, error) {

	b, err := r.CheckHeahth(params)
	if err != nil {
		return nil, err
	}
	return json.Marshal(b)
}

func NewSMBHeahthCrawler() *SMBHeahthCrawler {
	r := &SMBHeahthCrawler{}
	r.SetInternal(r)
	r.SetMatcher(smb.NewSMBMatcher())
	return r
}

//package smb_protocol_crawler
//
//
//import (
//	"git.loafle.net/overflow/of_rpc/models/param"
//	"encoding/json"
//	"git.loafle.net/overflow/of_rpc/models/protocol"
//	"strconv"
//
//	"net"
//
//	"git.loafle.net/overflow/collector/core/scan/service/matcher/smb"
//	"git.loafle.net/overflow/collector/core/scan/service/matcher/packet"
//	rpc "git.loafle.net/overflow/of_rpc/json"
//
//	"log"
//	"crypto/tls"
//)
//
//////////////// Delegate.Get implement /////////////////
//type SMBProtocol struct{}
//
//func (r *SMBProtocol) Get(input *param.Input, output *param.Output) error {
//
//	rr := &protocol.HealthRequest{}
//	json.Unmarshal(input.Data, &rr)
//
//	match, err := check(input.Ip, input.Port, rr.IsSSL)
//
//	if err != nil {
//		return err
//	}
//
//
//	retVal := &protocol.HealthResponse{}
//	retVal.IsMatch = match
//
//	b, err := json.Marshal(retVal)
//
//	if err != nil {
//		return err
//	}
//
//	*output = param.Output{Data: b}
//	return nil
//}
//
//
//func connect(ip string, port uint16, ssl bool) (net.Conn, error) {
//
//	var addr string = ip
//	addr += ":"
//	addr += strconv.Itoa(int(port))
//
//	if ssl == false {
//		conn, err := net.Dial("tcp", addr)
//		if err != nil {
//			return nil, err
//		}
//		return conn, nil
//	} else {
//		conn, err := tls.Dial(
//			"tcp",
//			addr,
//			&tls.Config{
//				InsecureSkipVerify: true,
//				ServerName:         ip,
//			},
//		)
//		if err != nil {
//			return nil, err
//		}
//		return conn, nil
//	}
//}
//
//
//
//func check(ip string, port uint16, ssl bool) (bool, error) {
//
//	conn, err := connect(ip, port, ssl)
//	if err != nil {
//		return false, err
//	}
//	defer conn.Close()
//
//	m := smb.NewSMBMatcher()
//
//	for i := 0; i < m.PacketCount(); i++ {
//
//		pack := m.Packet(i)
//		conn.Write(pack.Buffer)
//		bytes := make([]byte, 1024)
//		n, _ := conn.Read(bytes)
//		p := packet.NewPacket(bytes, n)
//
//		if m.IsNoResponse(i) == true { // empty last response
//			break
//		}
//
//		if m.Match(i, p, nil) == false {
//			return false, nil
//		}
//	}
//
//	return true, nil
//}
//
//func start() {
//
//	rpc.SetDelegate(new(SMBProtocol))
//	err := rpc.StartJSONRPC()
//	if err != nil {
//		log.Fatal(err)
//	}
//}
//
//func main() {
//	start()
//}
