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