qowevisa/tricrypt
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Some stuff to cgui

Browse Source
This commit is contained in:
qowevisa 2024-06-07 21:57:28 +03:00
parent 9d0e49c461
commit 5a2a368b9e
1 changed files with 22 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
cmd/cgui/main.go
View File

@ -69,6 +69,8 @@ func main() {
select {} select {}
} }
var r com.RegisteredUser
func readFromServer(conn net.Conn, ws *websocket.Conn) { func readFromServer(conn net.Conn, ws *websocket.Conn) {
buf := make([]byte, 70000) buf := make([]byte, 70000)
for { for {
@ -91,6 +93,9 @@ func readFromServer(conn net.Conn, ws *websocket.Conn) {
newID := binary.BigEndian.Uint16(msg.Data) newID := binary.BigEndian.Uint16(msg.Data)
msg.FromID = newID msg.FromID = newID
msg.Data = []byte{} msg.Data = []byte{}
r.ID = newID
r.IsRegistered = true
break
} }
log.Printf("client: readServer: sending message to websocket: %v", *msg) log.Printf("client: readServer: sending message to websocket: %v", *msg)
ws.WriteJSON(*msg) ws.WriteJSON(*msg)
@ -105,8 +110,24 @@ func readFromWebSocket(conn net.Conn, ws *websocket.Conn) {
log.Printf("WebSocket read error: %s", err) log.Printf("WebSocket read error: %s", err)
return return
} }
msg.Version = com.V1
log.Printf("client: readWS: received message from Electron: %v", msg) log.Printf("client: readWS: received message from Electron: %v", msg)
msg.Version = com.V1
switch msg.ID {
case com.ID_CLIENT_ASK_SERVER_LINK:
l, err := r.GenerateLink(msg.ToID)
if err != nil {
log.Printf("Error: link: %v", err)
continue
}
answ, err := com.ClientSendServerLink(l)
if err != nil {
log.Printf("Error: com: %v", err)
continue
}
log.Printf("client: readWS: sending data to server: %v", answ)
conn.Write(answ)
continue
}
encodedMsg, err := msg.Bytes() encodedMsg, err := msg.Bytes()
if err != nil { if err != nil {
log.Printf("Encoding error: %s", err) log.Printf("Encoding error: %s", err)