56 lines
1.5 KiB
Go
56 lines
1.5 KiB
Go
package socket
|
|
|
|
import (
|
|
"Watchdog_Linux-systemd/command"
|
|
"Watchdog_Linux-systemd/postLog"
|
|
"fmt"
|
|
"net"
|
|
)
|
|
|
|
func BootSocket(networkType, listenAddr string, listenPort int) error {
|
|
listen, err := net.Listen(networkType, fmt.Sprintf("%s:%d", listenAddr, listenPort))
|
|
if err != nil {
|
|
postLog.Fatal(fmt.Sprintf("[Socket] Failed to listen: %v, err: %v, %v", listenAddr, listenPort, err))
|
|
return fmt.Errorf("failed to listen: %v, err: %v, %v", listenAddr, listenPort, err)
|
|
}
|
|
defer listen.Close()
|
|
|
|
postLog.Info(fmt.Sprintf("Server is running on %s:%d", listenAddr, listenPort))
|
|
|
|
for {
|
|
conn, err := listen.Accept()
|
|
if err != nil {
|
|
postLog.Error(fmt.Sprintf("Failed to accept: %v, err: %v", conn, err))
|
|
}
|
|
go handleRequest(conn)
|
|
}
|
|
}
|
|
|
|
func handleRequest(conn net.Conn) {
|
|
defer conn.Close()
|
|
|
|
buffer := make([]byte, 1024)
|
|
n, err := conn.Read(buffer)
|
|
if err != nil {
|
|
postLog.Error(fmt.Sprintf("Failed to read: %v, err: %v", conn, err))
|
|
return
|
|
}
|
|
|
|
recvMsg := string(buffer[:n])
|
|
postLog.Debug(fmt.Sprintf("Received: %v", recvMsg))
|
|
responseMsg := ""
|
|
if recvMsg == "watchdogAgentConnectionTest" {
|
|
responseMsg = "success"
|
|
} else {
|
|
err := command.ExecuteCommand(recvMsg)
|
|
if err != nil {
|
|
responseMsg = fmt.Sprintf("error: %v", err)
|
|
} else {
|
|
responseMsg = "success"
|
|
}
|
|
}
|
|
if _, err := conn.Write([]byte(responseMsg + "\n")); err != nil {
|
|
postLog.Error(fmt.Sprintf("Failed to write: %v, err: %v", conn, err))
|
|
}
|
|
}
|