moved and parametrized the recorder

2025-09-26 03:56:19 +00:00 · 2017-08-03 14:30:39 +03:00
parent 8a510da111
commit 39d74c9762
4 changed files with 15 additions and 8 deletions
--- a/recorder/cmd/main.go
+++ b/recorder/cmd/main.go
@@ -0,0 +1,113 @@
+package main
+
+import (
+	"flag"
+	"net"
+	"time"
+	"vncproxy/client"
+	"vncproxy/common"
+	"vncproxy/encodings"
+	"vncproxy/logger"
+	"vncproxy/recorder"
+)
+
+func main() {
+	// var tcpPort = flag.String("tcpPort", "", "tcp port")
+	// var wsPort = flag.String("wsPort", "", "websocket port")
+	// var vncPass = flag.String("vncPass", "", "password on incoming vnc connections to the proxy, defaults to no password")
+	var recordDir = flag.String("recDir", "", "path to save FBS recordings WILL NOT RECORD IF EMPTY.")
+	var targetVncPort = flag.String("targPort", "", "target vnc server port")
+	var targetVncPass = flag.String("targPass", "", "target vnc password")
+
+	flag.Parse()
+
+	//nc, err := net.Dial("tcp", "192.168.1.101:5903")
+	nc, err := net.Dial("tcp", "localhost:"+*targetVncPort)
+
+	if err != nil {
+		logger.Errorf("error connecting to vnc server: %s", err)
+	}
+	var noauth client.ClientAuthNone
+	authArr := []client.ClientAuth{&client.PasswordAuth{Password: *targetVncPass}, &noauth}
+
+	//vncSrvMessagesChan := make(chan common.ServerMessage)
+
+	//rec, err := recorder.NewRecorder("c:/Users/betzalel/recording.rbs")
+	rec, err := recorder.NewRecorder(*recordDir) //"/Users/amitbet/vncRec/recording.rbs")
+	if err != nil {
+		logger.Errorf("error creating recorder: %s", err)
+		return
+	}
+
+	clientConn, err := client.NewClientConn(nc,
+		&client.ClientConfig{
+			Auth:      authArr,
+			Exclusive: true,
+		})
+
+	clientConn.Listeners.AddListener(rec)
+	clientConn.Listeners.AddListener(&recorder.RfbRequester{Conn: clientConn, Name: "Rfb Requester"})
+	clientConn.Connect()
+
+	if err != nil {
+		logger.Errorf("error creating client: %s", err)
+		return
+	}
+	// err = clientConn.FramebufferUpdateRequest(false, 0, 0, 1024, 768)
+	// if err != nil {
+	// 	logger.Errorf("error requesting fb update: %s", err)
+	// }
+	encs := []common.IEncoding{
+		&encodings.TightEncoding{},
+		//&encodings.TightPngEncoding{},
+		//rre := encodings.RREEncoding{},
+		//zlib := encodings.ZLibEncoding{},
+		//zrle := encodings.ZRLEEncoding{},
+		//&encodings.CopyRectEncoding{},
+		//coRRE := encodings.CoRREEncoding{},
+		//hextile := encodings.HextileEncoding{},
+		&encodings.PseudoEncoding{int32(common.EncJPEGQualityLevelPseudo8)},
+	}
+
+	clientConn.SetEncodings(encs)
+	//width := uint16(1280)
+	//height := uint16(800)
+
+	//clientConn.FramebufferUpdateRequest(false, 0, 0, width, height)
+
+	// // clientConn.SetPixelFormat(&common.PixelFormat{
+	// // 	BPP:        32,
+	// // 	Depth:      24,
+	// // 	BigEndian:  0,
+	// // 	TrueColor:  1,
+	// // 	RedMax:     255,
+	// // 	GreenMax:   255,
+	// // 	BlueMax:    255,
+	// // 	RedShift:   16,
+	// // 	GreenShift: 8,
+	// // 	BlueShift:  0,
+	// // })
+
+	// start := getNowMillisec()
+	// go func() {
+	// 	for {
+	// 		if getNowMillisec()-start >= 10000 {
+	// 			break
+	// 		}
+
+	// 		err = clientConn.FramebufferUpdateRequest(true, 0, 0, 1280, 800)
+	// 		if err != nil {
+	// 			logger.Errorf("error requesting fb update: %s", err)
+	// 		}
+	// 		time.Sleep(250 * time.Millisecond)
+	// 	}
+	// 	clientConn.Close()
+	// }()
+
+	for {
+		time.Sleep(time.Minute)
+	}
+}
+func getNowMillisec() int {
+	return int(time.Now().UnixNano() / int64(time.Millisecond))
+}