From 7134a3c0e88d1b2afe3ee3cddc4ac3fedbb4decf Mon Sep 17 00:00:00 2001 From: Kroese Date: Wed, 26 Apr 2023 05:59:06 +0200 Subject: [PATCH] Delete main.go --- serial/main.go | 263 ------------------------------------------------- 1 file changed, 263 deletions(-) delete mode 100644 serial/main.go diff --git a/serial/main.go b/serial/main.go deleted file mode 100644 index 8c1311c..0000000 --- a/serial/main.go +++ /dev/null @@ -1,263 +0,0 @@ -package main - -import ( - "fmt" - "log" - "net" - "flag" - "bytes" - "strconv" - "net/http" - "math/rand" - "encoding/binary" - "github.com/gorilla/mux" -) - -type REQ struct { - RandID int64 - GuestUUID [16]byte - GuestID int64 - IsReq int32 - IsResp int32 - NeedResponse int32 - ReqLength int32 - RespLength int32 - CommandID int32 - SubCommand int32 - Reserve int32 -} - -var GuestCPUs = flag.Int("cpu", 1, "Num of Guest cpu") -var Cluster_UUID = "3bdea92b-68f4-4fe9-aa4b-d645c3c63864" -var HostDSMfixNumber = flag.Int("fixNumber", 0, "Fix Number of Host") -var VMMTimestamp = flag.Int("vmmts", 1679863686, "VMM Timestamp") -var VMMVersion = flag.String("vmmversion", "2.6.1-12139", "VMM version") -var HostSN = flag.String("hostsn", "0000000000000", "Host SN, 13 bytes") -var GuestSN = flag.String("guestsn", "0000000000000", "Guest SN, 13 bytes") -var HostDSMBuildNumber = flag.Int("buildnumber", 42962, "Build Number of Host") -var GuestCPU_ARCH = flag.String("cpu_arch", "QEMU, Virtual CPU, X86_64", "CPU arch") -var GuestUUID = flag.String("guestuuid", "ba13a19a-c0c1-4fef-9346-915ed3b98341", "Guest UUID") - -var ApiPort = flag.String("api", ":2210", "API port") -var ListenAddr = flag.String("addr", "0.0.0.0:12345", "Listen address") - -var LastConnection net.Conn - -func main() { - - flag.Parse() - - r := mux.NewRouter() - r.HandleFunc("/", home) - r.HandleFunc("/write", write) - go http.ListenAndServe(*ApiPort, r) - - listener, err := net.Listen("tcp", *ListenAddr) - - if err != nil { - log.Fatalln("Error listening", err.Error()) - return - } - - log.Println("Start listen on " + *ListenAddr) - - for { - conn, err := listener.Accept() - if err != nil { - log.Fatalln("Error on accept", err.Error()) - return - } - log.Printf("New connection from %s\n", conn.RemoteAddr().String()) - - go incoming_conn(conn) - } -} - -func incoming_conn(conn net.Conn) { - - LastConnection = conn - - for { - buf := make([]byte, 4096) - len, err := conn.Read(buf) - if err != nil { - log.Println("Error on read", err.Error()) - return - } - if len != 4096 { - log.Printf("Read %d Bytes, not 4096\n", len) - // something wrong, close and wait for reconnect - conn.Close() - return - } - go process_req(buf, conn) - //log.Printf("Read %d Bytes\n%#v\n", len, string(buf[:len])) - } -} - -var commandsName = map[int]string{ - 3: "Guest Power info", - 4: "Host DSM version", - 5: "Guest SN", - 7: "Guest CPU info", - 9: "Host DSM version", - 8: "VMM version", - 10: "Get Guest Info", - 11: "Guest UUID", - 12: "Cluster UUID", - 13: "Host SN", - 16: "Update Deadline", - 17: "Guest Timestamp", -} - -func process_req(buf []byte, conn net.Conn) { - var req REQ - var data string - err := binary.Read(bytes.NewReader(buf), binary.LittleEndian, &req) - if err != nil { - log.Printf("Error on decode %s\n", err) - return - } - - if req.IsReq == 1 { - data = string(buf[64 : 64+req.ReqLength]) - } else if req.IsResp == 1 { - data = string(buf[64 : 64+req.RespLength]) - } - - // log.Printf("%#v\n", req) - log.Printf("Command: %s from Guest:%d \n", commandsName[int(req.CommandID)], req.GuestID) - if data != "" { - log.Printf("Info: %s\n", data) - } - // Hard code of command - switch req.CommandID { - case 3: - // Guest start/reboot - case 4: - // Host DSM version - data = fmt.Sprintf(`{"buildnumber":%d,"smallfixnumber":%d}`, *HostDSMBuildNumber, *HostDSMfixNumber) - case 5: - // Guest SN - data = *GuestSN - case 7: - // CPU info - // {"cpuinfo":"QEMU, Virtual CPU, X86_64, 1" "vcpu_num":1} - data = fmt.Sprintf(`{"cpuinfo":"%s","vcpu_num":%d}`, - *GuestCPU_ARCH+", "+strconv.Itoa(*GuestCPUs), *GuestCPUs) - case 8: - data = fmt.Sprintf(`{"id":"Virtualization","name":"Virtual Machine Manager","timestamp":%d,"version":"%s"}`, - *VMMTimestamp, *VMMVersion) - case 9: - // Version Info - case 10: - // Guest Info - case 11: - // Guest UUID - data = *GuestUUID - case 12: - // cluster UUID - data = Cluster_UUID - case 13: - // Host SN - data = *HostSN - case 16: - // Update Dead line time, always 0x7fffffffffffffff - data = "9223372036854775807" - case 17: - // TimeStamp - default: - log.Printf("No handler for this command %d\n", req.CommandID) - return - } - - // if it's a req and need response - if req.IsReq == 1 && req.NeedResponse == 1 { - buf = make([]byte, 0, 4096) - writer := bytes.NewBuffer(buf) - req.IsResp = 1 - req.IsReq = 0 - req.ReqLength = 0 - req.RespLength = int32(len([]byte(data)) + 1) - log.Printf("Response data: %s\n", data) - - // write to buf - binary.Write(writer, binary.LittleEndian, &req) - writer.Write([]byte(data)) - res := writer.Bytes() - // full fill 4096 - buf = make([]byte, 4096, 4096) - copy(buf, res) - conn.Write(buf) - } -} - -func home(w http.ResponseWriter, r *http.Request) { - - w.Header().Set("Content-Type", "application/json") - w.WriteHeader(http.StatusInternalServerError) - w.Write([]byte(`{"status": "error", "data": null, "message": "No command specified"}`)) -} - - -func write(w http.ResponseWriter, r *http.Request) { - - w.Header().Set("Content-Type", "application/json") - - var err error - var commandID int - - query := r.URL.Query() - commandID, err = strconv.Atoi(query.Get("command")) - - if (err != nil || commandID < 1) { - w.WriteHeader(http.StatusInternalServerError) - w.Write([]byte(`{"status": "error", "data": null, "message": "Invalid command ID"}`)) - return - } - - if (send_command((int32)(commandID), 1) == false) { - w.WriteHeader(http.StatusInternalServerError) - w.Write([]byte(`{"status": "error", "data": null, "message": "Failed to send command"}`)) - return - } - - w.WriteHeader(http.StatusOK) - w.Write([]byte(`{"status": "success", "data": null, "message": null}`)) - return -} - -func send_command(CommandID int32, SubCommand int32) bool { - - var req REQ - - req.CommandID = CommandID - req.SubCommand = SubCommand - - req.IsReq = 1 - req.IsResp = 0 - req.ReqLength = 0 - req.RespLength = 0 - req.NeedResponse = 0 - req.GuestID = 10000000 - req.RandID = rand.Int63() - - var buf = make([]byte, 0, 4096) - var writer = bytes.NewBuffer(buf) - - // write to buf - binary.Write(writer, binary.LittleEndian, &req) - res := writer.Bytes() - - // full fill 4096 - buf = make([]byte, 4096, 4096) - copy(buf, res) - - //log.Printf("Writing command %d\n", CommandID) - - if (LastConnection == nil) { return false } - - LastConnection.Write(buf) - return true - -}