Example #1
0
File: server.go Project: y8y/OWL-v3
func NewTCPServe() *tcpserver.Server {
	tcpAddr, err := net.ResolveTCPAddr("tcp4", cfg.TCPBIND)
	listener, err := net.ListenTCP("tcp4", tcpAddr)
	if err != nil {
		panic(err)
	}
	return tcpserver.NewServer(listener, &handle{})
}
Example #2
0
File: server.go Project: y8y/OWL-v3
func NewProxy(cfg *Config) *Proxy {
	tcpAddr, err := net.ResolveTCPAddr("tcp4", cfg.TCPBIND)
	listener, err := net.ListenTCP("tcp4", tcpAddr)
	if err != nil {
		panic(err)
	}
	return &Proxy{tcpserver.NewServer(listener, &handle{}), ""}
}
Example #3
0
File: server.go Project: y8y/OWL-v3
func StartTCPServe(cfg *Config) *tcpserver.Server {
	addr, err := net.ResolveTCPAddr("tcp", cfg.TCPBIND)
	if err != nil {
		panic(err)
	}
	listener, err := net.ListenTCP("tcp", addr)
	if err != nil {
		panic(err)
	}
	s := tcpserver.NewServer(listener, &handle{})
	s.SetPacketLimitSize(uint32(cfg.MAX_PACKET_SIZE))
	go s.Start()
	return s
}
Example #4
0
File: main.go Project: y8y/OWL-v3
func main() {
	//set workpath
	file, _ := exec.LookPath(os.Args[0])
	pth, _ := path.Split(file)
	os.Chdir(pth)

	cfg, err := load_config("./conf/client.conf")
	if err != nil {
		fmt.Printf("load config file error , %v\n", err)
		os.Exit(1)
	}

	for _, dir := range []string{cfg.LOG_DIR, "var", "update", "plugins"} {
		if err := os.MkdirAll(dir, 0755); err != nil {
			fmt.Println("create dir(%s) error %v", dir, err)
			os.Exit(1)
		}
	}

	if err := init_log(cfg); err != nil {
		fmt.Println("main.init_log error %s", err)
		os.Exit(1)
	}

	fd, err := os.Create("var/guard.pid")
	if err != nil {
		log.Error("create pid file error <%v> ", err)
		return
	}
	fd.WriteString(strconv.Itoa(os.Getpid()))
	fd.Close()
	tcpAddr, err := net.ResolveTCPAddr("tcp", cfg.GUARDBIND)
	if err != nil {
		panic(err)
	}
	listener, err := net.ListenTCP("tcp", tcpAddr)
	if err != nil {
		panic(err)
	}
	server := tcpserver.NewServer(listener, &handle{})
	go server.Start()
START:

	conn, err := net.DialTimeout("tcp", cfg.TCPBIND, time.Second*5)
	if err != nil {
		log.Warn("connect to guard error %s", err.Error())
		StartClient()
		time.Sleep(time.Second * 10)
		goto START
	}
	for {
		var buf bytes.Buffer
		head := make([]byte, 4)
		binary.BigEndian.PutUint32(head, uint32(1))
		binary.Write(&buf, binary.BigEndian, head)
		binary.Write(&buf, binary.BigEndian, byte(GUARDHB))
		lenght, err := conn.Write(buf.Bytes())
		if err != nil {
			log.Warn("send packet to guard error %s", err.Error())
			StartClient()
			time.Sleep(time.Second * 10)
			goto START
		}
		log.Info("send hb packet to client doen. %d bytes", lenght)
		time.Sleep(time.Second * 1)
	}

}