mirror of
https://github.com/vdsm/virtual-dsm.git
synced 2025-02-24 13:30:02 +08:00
Delete main.go
This commit is contained in:
Kroese
GitHub
parent
1f38322ae1
commit
7134a3c0e8
263
serial/main.go
263
serial/main.go
@ -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
|
||||
|
||||
}
|
||||
Loading…
x
Reference in New Issue
Block a user