github/vncproxy
1
0
Fork 0
mirror of https://github.com/amitbet/vncproxy.git synced 2025-06-30 08:02:12 +00:00
Code Issues Projects Releases Wiki Activity

Fix recording bug in proxy

Browse Source 
This change fixes a bug in the proxy session recording management, where
the proxy would record sessions even if the user didn't provide a value
for the `-recDir` CLI flag.
This commit is contained in:
Marc Falzon 2018-11-01 10:02:35 +01:00
parent 3e882d6140
commit 06e555775f
1 changed files with 17 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

29
proxy/cmd/main.go
View File

@ -1,9 +1,12 @@
package main
import "vncproxy/proxy"
import "flag"
import "vncproxy/logger"
import "os"
import (
"flag"
"os"
"vncproxy/logger"
vncproxy "vncproxy/proxy"
)
func main() {
//create default session if required
@ -34,30 +37,32 @@ func main() {
if *vncPass == "" {
logger.Warn("proxy will have no password")
}
if *recordDir == "" {
logger.Warn("FBS recording is turned off")
}
tcpUrl := ""
if *tcpPort != "" {
tcpUrl = ":" + string(*tcpPort)
}
proxy := &proxy.VncProxy{
proxy := &vncproxy.VncProxy{
WsListeningUrl: "http://0.0.0.0:" + string(*wsPort) + "/", // empty = not listening on ws
RecordingDir: *recordDir, //"/Users/amitbet/vncRec", // empty = no recording
TcpListeningUrl: tcpUrl,
ProxyVncPassword: *vncPass, //empty = no auth
SingleSession: &proxy.VncSession{
SingleSession: &vncproxy.VncSession{
TargetHostname: *targetVncHost,
TargetPort: *targetVncPort,
TargetPassword: *targetVncPass, //"vncPass",
ID: "dummySession",
Status: proxy.SessionStatusInit,
Type: proxy.SessionTypeRecordingProxy,
Status: vncproxy.SessionStatusInit,
Type: vncproxy.SessionTypeProxyPass,
}, // to be used when not using sessions
UsingSessions: false, //false = single session - defined in the var above
}
if *recordDir != "" {
logger.Warn("FBS recording is turned on")
proxy.RecordingDir = *recordDir
proxy.SingleSession.Type = vncproxy.SessionTypeRecordingProxy
}
proxy.StartListening()
}