zydronium/rdpgw
1
0
Fork 0
mirror of https://github.com/bolkedebruin/rdpgw.git synced 2025-07-21 18:15:54 +02:00
Code Issues Projects Releases Packages Wiki Activity

Avoid concurrency issues

Browse source
This commit is contained in:
Bolke de Bruin 2020-07-13 11:01:39 +02:00
parent 27d449d438
commit 728fc76e8b
1 changed files with 4 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

6
rdg.go
View file

@ -220,7 +220,8 @@ func handleWebsocketProtocol(conn *websocket.Conn) {
case PKT_TYPE_DATA: case PKT_TYPE_DATA:
forwardDataPacket(remote, pkt) forwardDataPacket(remote, pkt)
case PKT_TYPE_KEEPALIVE: case PKT_TYPE_KEEPALIVE:
conn.WriteMessage(mt, createPacket(PKT_TYPE_KEEPALIVE, []byte{})) // do not write to make sure we do not create concurrency issues
// conn.WriteMessage(mt, createPacket(PKT_TYPE_KEEPALIVE, []byte{}))
case PKT_TYPE_CLOSE_CHANNEL: case PKT_TYPE_CLOSE_CHANNEL:
remote.Close() remote.Close()
return return
@ -314,7 +315,8 @@ func handleLegacyProtocol(w http.ResponseWriter, r *http.Request) {
case PKT_TYPE_DATA: case PKT_TYPE_DATA:
forwardDataPacket(remote, packet) forwardDataPacket(remote, packet)
case PKT_TYPE_KEEPALIVE: case PKT_TYPE_KEEPALIVE:
s.ConnOut.Write(createPacket(PKT_TYPE_KEEPALIVE, []byte{})) // avoid concurrency issues
// s.ConnOut.Write(createPacket(PKT_TYPE_KEEPALIVE, []byte{}))
case PKT_TYPE_CLOSE_CHANNEL: case PKT_TYPE_CLOSE_CHANNEL:
s.ConnIn.Close() s.ConnIn.Close()
s.ConnOut.Close() s.ConnOut.Close()