sendnrw/rdpgw
2
0
Fork 0
mirror of https://github.com/bolkedebruin/rdpgw.git synced 2026-03-29 15:06:36 +00:00
Code Issues Packages Projects Releases 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()