mirror of
https://github.com/amitbet/vncproxy.git
synced 2025-10-27 23:20:57 +00:00
Compare commits
1 Commits
v1.1
...
keyframes_
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
1bce301125 |
@@ -11,7 +11,7 @@ An RFB proxy, written in go that can save and replay FBS files
|
||||
- Tested on tight encoding with:
|
||||
- Tightvnc (client + java client + server)
|
||||
- FBS player (tightVnc Java player)
|
||||
- NoVnc(web client) => use -wsPort to open a websocket
|
||||
- NoVnc(web client)
|
||||
- ChickenOfTheVnc(client)
|
||||
- VineVnc(server)
|
||||
- TigerVnc(client)
|
||||
@@ -25,7 +25,7 @@ An RFB proxy, written in go that can save and replay FBS files
|
||||
## Usage:
|
||||
recorder -recDir=./recording.rbs -targHost=192.168.0.100 -targPort=5903 -targPass=@@@@@
|
||||
player -fbsFile=./myrec.fbs -tcpPort=5905
|
||||
proxy -recDir=./recordings/ -targHost=192.168.0.100 -targPort=5903 -targPass=@@@@@ -tcpPort=5903 -wsPort=5905 -vncPass=@!@!@!
|
||||
proxy -recDir=./recordings/ -targHost=192.168.0.100 -targPort=5903 -targPass=@@@@@ -tcpPort=5903 -vncPass=@!@!@!
|
||||
|
||||
### Code usage examples
|
||||
* player/main.go (fbs recording vnc client)
|
||||
|
||||
@@ -506,7 +506,7 @@ func (c *ClientConn) mainLoop() {
|
||||
// Unsupported message type! Bad!
|
||||
break
|
||||
}
|
||||
logger.Debugf("ClientConn.MainLoop: got ServerMessage:%s", common.ServerMessageType(messageType))
|
||||
logger.Infof("ClientConn.MainLoop: got ServerMessage:%s", common.ServerMessageType(messageType))
|
||||
reader.SendMessageStart(common.ServerMessageType(messageType))
|
||||
reader.PublishBytes([]byte{byte(messageType)})
|
||||
|
||||
|
||||
@@ -63,7 +63,7 @@ func (fbm *MsgFramebufferUpdate) Read(c common.IClientConn, r *common.RfbReadHel
|
||||
// We must always support the raw encoding
|
||||
rawEnc := new(encodings.RawEncoding)
|
||||
encMap[rawEnc.Type()] = rawEnc
|
||||
logger.Debugf("MsgFramebufferUpdate.Read: numrects= %d", numRects)
|
||||
logger.Infof("MsgFramebufferUpdate.Read: numrects= %d", numRects)
|
||||
|
||||
rects := make([]common.Rectangle, numRects)
|
||||
for i := uint16(0); i < numRects; i++ {
|
||||
@@ -90,7 +90,7 @@ func (fbm *MsgFramebufferUpdate) Read(c common.IClientConn, r *common.RfbReadHel
|
||||
|
||||
encType := common.EncodingType(encodingTypeInt)
|
||||
|
||||
logger.Debugf("MsgFramebufferUpdate.Read: rect# %d, rect hdr data: enctype=%s, data: %s", i, encType, string(jBytes))
|
||||
logger.Infof("MsgFramebufferUpdate.Read: rect# %d, rect hdr data: enctype=%s, data: %s", i, encType, string(jBytes))
|
||||
enc, supported := encMap[encodingTypeInt]
|
||||
if supported {
|
||||
var err error
|
||||
|
||||
@@ -73,8 +73,6 @@ func (enct EncodingType) String() string {
|
||||
return "EncJPEGQualityLevelPseudo1"
|
||||
case EncCursorPseudo:
|
||||
return "EncCursorPseudo"
|
||||
case EncLedStatePseudo:
|
||||
return "EncLedStatePseudo"
|
||||
case EncDesktopSizePseudo:
|
||||
return "EncDesktopSizePseudo"
|
||||
case EncLastRectPseudo:
|
||||
@@ -185,7 +183,6 @@ const (
|
||||
EncQEMUPointerMotionChangePseudo EncodingType = -257
|
||||
EncQEMUExtendedKeyEventPseudo EncodingType = -258
|
||||
EncTightPng EncodingType = -260
|
||||
EncLedStatePseudo EncodingType = -261
|
||||
EncExtendedDesktopSizePseudo EncodingType = -308
|
||||
EncXvpPseudo EncodingType = -309
|
||||
EncFencePseudo EncodingType = -312
|
||||
|
||||
@@ -1,31 +0,0 @@
|
||||
package encodings
|
||||
|
||||
import (
|
||||
"io"
|
||||
"vncproxy/common"
|
||||
"vncproxy/logger"
|
||||
)
|
||||
|
||||
type EncLedStatePseudo struct {
|
||||
LedState uint8
|
||||
}
|
||||
|
||||
func (pe *EncLedStatePseudo) Type() int32 {
|
||||
return int32(common.EncLedStatePseudo)
|
||||
}
|
||||
func (pe *EncLedStatePseudo) WriteTo(w io.Writer) (n int, err error) {
|
||||
w.Write([]byte{pe.LedState})
|
||||
return 1, nil
|
||||
}
|
||||
func (pe *EncLedStatePseudo) Read(pf *common.PixelFormat, rect *common.Rectangle, r *common.RfbReadHelper) (common.IEncoding, error) {
|
||||
if rect.Width*rect.Height == 0 {
|
||||
return pe, nil
|
||||
}
|
||||
u8, err := r.ReadUint8()
|
||||
pe.LedState = u8
|
||||
if err != nil {
|
||||
logger.Error("error while reading led state: ", err)
|
||||
return pe, err
|
||||
}
|
||||
return pe, nil
|
||||
}
|
||||
@@ -142,7 +142,7 @@ func handleTightFilters(subencoding uint8, pixelFmt *common.PixelFormat, rect *c
|
||||
return
|
||||
}
|
||||
|
||||
paletteSize := int(colorCount) + 1 // add one more
|
||||
paletteSize := colorCount + 1 // add one more
|
||||
logger.Debugf("handleTightFilters: ----PALETTE_FILTER: paletteSize=%d bytesPixel=%d\n", paletteSize, bytesPixel)
|
||||
//complete palette
|
||||
_, err = r.ReadBytes(int(paletteSize) * bytesPixel)
|
||||
|
||||
@@ -4,30 +4,6 @@ import "fmt"
|
||||
|
||||
var simpleLogger = SimpleLogger{LogLevelInfo}
|
||||
|
||||
func SetLogLevel(logLevel string) {
|
||||
level := GetLogLevel(logLevel)
|
||||
fmt.Println("Log level set to: ", logLevel)
|
||||
simpleLogger = SimpleLogger{level}
|
||||
}
|
||||
|
||||
func GetLogLevel(logLevel string) LogLevel {
|
||||
switch logLevel {
|
||||
case "trace":
|
||||
return LogLevelTrace
|
||||
case "debug":
|
||||
return LogLevelDebug
|
||||
case "info":
|
||||
return LogLevelInfo
|
||||
case "warn":
|
||||
return LogLevelWarn
|
||||
case "error":
|
||||
return LogLevelError
|
||||
case "fatal":
|
||||
return LogLevelFatal
|
||||
}
|
||||
return LogLevelInfo
|
||||
}
|
||||
|
||||
type Logger interface {
|
||||
Debug(v ...interface{})
|
||||
Debugf(format string, v ...interface{})
|
||||
@@ -151,13 +127,6 @@ func Debugf(format string, v ...interface{}) {
|
||||
simpleLogger.Debugf(format, v...)
|
||||
}
|
||||
|
||||
func Trace(v ...interface{}) {
|
||||
simpleLogger.Trace(v...)
|
||||
}
|
||||
func Tracef(format string, v ...interface{}) {
|
||||
simpleLogger.Tracef(format, v...)
|
||||
}
|
||||
|
||||
func Info(v ...interface{}) {
|
||||
simpleLogger.Info(v...)
|
||||
}
|
||||
|
||||
@@ -15,10 +15,8 @@ func main() {
|
||||
wsPort := flag.String("wsPort", "", "websocket port for player to listen to client connections")
|
||||
tcpPort := flag.String("tcpPort", "", "tcp port for player to listen to client connections")
|
||||
fbsFile := flag.String("fbsFile", "", "fbs file to serve to all connecting clients")
|
||||
logLevel := flag.String("logLevel", "info", "change logging level")
|
||||
|
||||
flag.Parse()
|
||||
logger.SetLogLevel(*logLevel)
|
||||
|
||||
fmt.Println("**************************************************************************")
|
||||
fmt.Println("*** This is a toy server that replays a single FBS file to all clients ***")
|
||||
|
||||
@@ -58,7 +58,6 @@ func NewFbsReader(fbsFile string) (*FbsReader, error) {
|
||||
&encodings.TightEncoding{},
|
||||
&encodings.TightPngEncoding{},
|
||||
&encodings.EncCursorPseudo{},
|
||||
&encodings.EncLedStatePseudo{},
|
||||
&encodings.RawEncoding{},
|
||||
&encodings.RREEncoding{},
|
||||
},
|
||||
|
||||
@@ -18,7 +18,6 @@ func TestServer(t *testing.T) {
|
||||
&encodings.RawEncoding{},
|
||||
&encodings.TightEncoding{},
|
||||
&encodings.EncCursorPseudo{},
|
||||
&encodings.EncLedStatePseudo{},
|
||||
//encodings.TightPngEncoding{},
|
||||
&encodings.RREEncoding{},
|
||||
&encodings.ZLibEncoding{},
|
||||
|
||||
199
player/rfb-reader.go
Normal file
199
player/rfb-reader.go
Normal file
@@ -0,0 +1,199 @@
|
||||
package player
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"encoding/binary"
|
||||
"io"
|
||||
"os"
|
||||
"vncproxy/common"
|
||||
"vncproxy/encodings"
|
||||
"vncproxy/logger"
|
||||
)
|
||||
|
||||
type RfbReader struct {
|
||||
reader io.Reader
|
||||
buffer bytes.Buffer
|
||||
currentTimestamp int
|
||||
pixelFormat *common.PixelFormat
|
||||
encodings []common.IEncoding
|
||||
}
|
||||
|
||||
/**************************************************************
|
||||
** RFB File documentation:
|
||||
** Sections:
|
||||
** 0. header:
|
||||
* index seek position
|
||||
*
|
||||
** 1. init message
|
||||
** 2. content
|
||||
* frame message:
|
||||
* size, timestamp, type, content
|
||||
** 3. index:
|
||||
* each frame message start position, full/incremental, timestamp
|
||||
*
|
||||
***************************************************************/
|
||||
|
||||
func (rfb *RfbReader) CurrentTimestamp() int {
|
||||
return rfb.currentTimestamp
|
||||
}
|
||||
|
||||
func (rfb *RfbReader) Read(p []byte) (n int, err error) {
|
||||
if rfb.buffer.Len() < len(p) {
|
||||
seg, err := rfb.ReadSegment()
|
||||
|
||||
if err != nil {
|
||||
logger.Error("rfbReader.Read: error reading rfbsegment: ", err)
|
||||
return 0, err
|
||||
}
|
||||
rfb.buffer.Write(seg.bytes)
|
||||
rfb.currentTimestamp = int(seg.timestamp)
|
||||
}
|
||||
return rfb.buffer.Read(p)
|
||||
}
|
||||
|
||||
func (rfb *RfbReader) CurrentPixelFormat() *common.PixelFormat { return rfb.pixelFormat }
|
||||
|
||||
//func (rfb *rfbReader) CurrentColorMap() *common.ColorMap { return &common.ColorMap{} }
|
||||
func (rfb *RfbReader) Encodings() []common.IEncoding { return rfb.encodings }
|
||||
|
||||
func NewRfbReader(rfbFile string) (*RfbReader, error) {
|
||||
|
||||
reader, err := os.OpenFile(rfbFile, os.O_RDONLY, 0644)
|
||||
if err != nil {
|
||||
logger.Error("NewrfbReader: can't open rfb file: ", rfbFile)
|
||||
return nil, err
|
||||
}
|
||||
return &RfbReader{reader: reader,
|
||||
encodings: []common.IEncoding{
|
||||
&encodings.CopyRectEncoding{},
|
||||
&encodings.ZLibEncoding{},
|
||||
&encodings.ZRLEEncoding{},
|
||||
&encodings.CoRREEncoding{},
|
||||
&encodings.HextileEncoding{},
|
||||
&encodings.TightEncoding{},
|
||||
&encodings.TightPngEncoding{},
|
||||
&encodings.EncCursorPseudo{},
|
||||
&encodings.RawEncoding{},
|
||||
&encodings.RREEncoding{},
|
||||
},
|
||||
}, nil
|
||||
|
||||
}
|
||||
|
||||
func (rfb *RfbReader) ReadStartSession() (*common.ServerInit, error) {
|
||||
|
||||
initMsg := common.ServerInit{}
|
||||
reader := rfb.reader
|
||||
|
||||
var framebufferWidth uint16
|
||||
var framebufferHeight uint16
|
||||
var SecTypeNone uint32
|
||||
//read rfb header information (the only part done without the [size|data|timestamp] block wrapper)
|
||||
//.("rfb 001.000\n")
|
||||
bytes := make([]byte, 12)
|
||||
_, err := reader.Read(bytes)
|
||||
if err != nil {
|
||||
logger.Error("rfbReader.ReadStartSession: error reading rbs init message - rfb file Version:", err)
|
||||
return nil, err
|
||||
}
|
||||
|
||||
//read the version message into the buffer so it will be written in the first rbs block
|
||||
//RFB 003.008\n
|
||||
bytes = make([]byte, 12)
|
||||
_, err = rfb.Read(bytes)
|
||||
if err != nil {
|
||||
logger.Error("rfbReader.ReadStartSession: error reading rbs init - RFB Version: ", err)
|
||||
return nil, err
|
||||
}
|
||||
|
||||
//push sec type and fb dimensions
|
||||
binary.Read(rfb, binary.BigEndian, &SecTypeNone)
|
||||
if err != nil {
|
||||
logger.Error("rfbReader.ReadStartSession: error reading rbs init - SecType: ", err)
|
||||
}
|
||||
|
||||
//read frame buffer width, height
|
||||
binary.Read(rfb, binary.BigEndian, &framebufferWidth)
|
||||
if err != nil {
|
||||
logger.Error("rfbReader.ReadStartSession: error reading rbs init - FBWidth: ", err)
|
||||
return nil, err
|
||||
}
|
||||
initMsg.FBWidth = framebufferWidth
|
||||
|
||||
binary.Read(rfb, binary.BigEndian, &framebufferHeight)
|
||||
if err != nil {
|
||||
logger.Error("rfbReader.ReadStartSession: error reading rbs init - FBHeight: ", err)
|
||||
return nil, err
|
||||
}
|
||||
initMsg.FBHeight = framebufferHeight
|
||||
|
||||
//read pixel format
|
||||
pixelFormat := &common.PixelFormat{}
|
||||
binary.Read(rfb, binary.BigEndian, pixelFormat)
|
||||
if err != nil {
|
||||
logger.Error("rfbReader.ReadStartSession: error reading rbs init - Pixelformat: ", err)
|
||||
return nil, err
|
||||
}
|
||||
initMsg.PixelFormat = *pixelFormat
|
||||
//read padding
|
||||
bytes = make([]byte, 3)
|
||||
rfb.Read(bytes)
|
||||
rfb.pixelFormat = pixelFormat
|
||||
|
||||
//read desktop name
|
||||
var desknameLen uint32
|
||||
binary.Read(rfb, binary.BigEndian, &desknameLen)
|
||||
if err != nil {
|
||||
logger.Error("rfbReader.ReadStartSession: error reading rbs init - deskname Len: ", err)
|
||||
return nil, err
|
||||
}
|
||||
initMsg.NameLength = desknameLen
|
||||
|
||||
bytes = make([]byte, desknameLen)
|
||||
rfb.Read(bytes)
|
||||
if err != nil {
|
||||
logger.Error("rfbReader.ReadStartSession: error reading rbs init - desktopName: ", err)
|
||||
return nil, err
|
||||
}
|
||||
|
||||
initMsg.NameText = bytes
|
||||
|
||||
return &initMsg, nil
|
||||
}
|
||||
|
||||
func (rfb *RfbReader) ReadSegment() (*FbsSegment, error) {
|
||||
reader := rfb.reader
|
||||
var bytesLen uint32
|
||||
|
||||
//read length
|
||||
err := binary.Read(reader, binary.BigEndian, &bytesLen)
|
||||
if err != nil {
|
||||
logger.Error("rfbReader.ReadStartSession: read len, error reading rbs file: ", err)
|
||||
return nil, err
|
||||
}
|
||||
|
||||
paddedSize := (bytesLen + 3) & 0x7FFFFFFC
|
||||
|
||||
//read bytes
|
||||
bytes := make([]byte, paddedSize)
|
||||
_, err = reader.Read(bytes)
|
||||
if err != nil {
|
||||
logger.Error("rfbReader.ReadSegment: read bytes, error reading rbs file: ", err)
|
||||
return nil, err
|
||||
}
|
||||
|
||||
//remove padding
|
||||
actualBytes := bytes[:bytesLen]
|
||||
|
||||
//read timestamp
|
||||
var timeSinceStart uint32
|
||||
binary.Read(reader, binary.BigEndian, &timeSinceStart)
|
||||
if err != nil {
|
||||
logger.Error("rfbReader.ReadSegment: read timestamp, error reading rbs file: ", err)
|
||||
return nil, err
|
||||
}
|
||||
|
||||
//timeStamp := time.Unix(timeSinceStart, 0)
|
||||
seg := &FbsSegment{bytes: actualBytes, timestamp: timeSinceStart}
|
||||
return seg, nil
|
||||
}
|
||||
@@ -14,10 +14,8 @@ func main() {
|
||||
var targetVncPort = flag.String("targPort", "", "target vnc server port")
|
||||
var targetVncHost = flag.String("targHost", "", "target vnc server host")
|
||||
var targetVncPass = flag.String("targPass", "", "target vnc password")
|
||||
var logLevel = flag.String("logLevel", "info", "change logging level")
|
||||
|
||||
flag.Parse()
|
||||
logger.SetLogLevel(*logLevel)
|
||||
|
||||
if *tcpPort == "" && *wsPort == "" {
|
||||
logger.Error("no listening port defined")
|
||||
|
||||
@@ -13,7 +13,7 @@ type ClientUpdater struct {
|
||||
|
||||
// Consume recieves vnc-server-bound messages (Client messages) and updates the server part of the proxy
|
||||
func (cc *ClientUpdater) Consume(seg *common.RfbSegment) error {
|
||||
logger.Tracef("ClientUpdater.Consume (vnc-server-bound): got segment type=%s bytes: %v", seg.SegmentType, seg.Bytes)
|
||||
//logger.Debugf("ClientUpdater.Consume (vnc-server-bound): got segment type=%s bytes: %v", seg.SegmentType, seg.Bytes)
|
||||
switch seg.SegmentType {
|
||||
|
||||
case common.SegmentFullyParsedClientMessage:
|
||||
|
||||
@@ -118,7 +118,6 @@ func (vp *VncProxy) newServerConnHandler(cfg *server.ServerConfig, sconn *server
|
||||
&encodings.RawEncoding{},
|
||||
&encodings.TightEncoding{},
|
||||
&encodings.EncCursorPseudo{},
|
||||
&encodings.EncLedStatePseudo{},
|
||||
&encodings.TightPngEncoding{},
|
||||
&encodings.RREEncoding{},
|
||||
&encodings.ZLibEncoding{},
|
||||
|
||||
@@ -16,14 +16,12 @@ 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("recFile", "", "FBS file to create, recordings WILL NOT RECORD IF EMPTY.")
|
||||
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")
|
||||
var targetVncHost = flag.String("targHost", "localhost", "target vnc hostname")
|
||||
var logLevel = flag.String("logLevel", "info", "change logging level")
|
||||
|
||||
flag.Parse()
|
||||
logger.SetLogLevel(*logLevel)
|
||||
|
||||
if *targetVncHost == "" {
|
||||
logger.Error("no target vnc server host defined")
|
||||
@@ -42,8 +40,6 @@ func main() {
|
||||
}
|
||||
if *recordDir == "" {
|
||||
logger.Warn("FBS recording is turned off")
|
||||
} else {
|
||||
logger.Infof("Recording rfb stream into file: '%s'", *recordDir)
|
||||
}
|
||||
|
||||
//nc, err := net.Dial("tcp", "192.168.1.101:5903")
|
||||
@@ -71,7 +67,11 @@ func main() {
|
||||
})
|
||||
|
||||
clientConn.Listeners.AddListener(rec)
|
||||
clientConn.Listeners.AddListener(&recorder.RfbRequester{Conn: clientConn, Name: "Rfb Requester"})
|
||||
clientConn.Listeners.AddListener(&recorder.RfbRequester{
|
||||
Conn: clientConn,
|
||||
Name: "Rfb Requester",
|
||||
FullScreenRefreshInSec: 30, //create a full refresh key frame every 30sec for seeking
|
||||
})
|
||||
clientConn.Connect()
|
||||
|
||||
if err != nil {
|
||||
|
||||
@@ -189,6 +189,11 @@ func (r *Recorder) writeToDisk() error {
|
||||
paddedSize := (bytesLen + 3) & 0x7FFFFFFC
|
||||
paddingSize := paddedSize - bytesLen
|
||||
|
||||
/// KeyFramePos, _ := r.writer.Seek(0, os.SEEK_CUR)
|
||||
/// fi, err := r.writer.Stat()
|
||||
/// KeyFramePos := fi.Size() + KeyFramePosInBuffer
|
||||
// now save the KF pos in some file
|
||||
|
||||
//logger.Debugf("paddedSize=%d paddingSize=%d bytesLen=%d", paddedSize, paddingSize, bytesLen)
|
||||
//write buffer padded to 32bit
|
||||
_, err := r.buffer.WriteTo(r.writer)
|
||||
|
||||
@@ -8,11 +8,13 @@ import (
|
||||
)
|
||||
|
||||
type RfbRequester struct {
|
||||
Conn *client.ClientConn
|
||||
Name string
|
||||
Width uint16
|
||||
Height uint16
|
||||
lastRequestTime time.Time
|
||||
Conn *client.ClientConn
|
||||
Name string
|
||||
Width uint16
|
||||
Height uint16
|
||||
lastRequestTime time.Time
|
||||
nextFullScreenRefresh time.Time
|
||||
FullScreenRefreshInSec int // refresh interval (creates keyframes) if 0, disables keyframe creation
|
||||
}
|
||||
|
||||
func (p *RfbRequester) Consume(seg *common.RfbSegment) error {
|
||||
@@ -29,6 +31,7 @@ func (p *RfbRequester) Consume(seg *common.RfbSegment) error {
|
||||
p.Height = serverInitMessage.FBHeight
|
||||
p.lastRequestTime = time.Now()
|
||||
p.Conn.FramebufferUpdateRequest(false, 0, 0, p.Width, p.Height)
|
||||
p.nextFullScreenRefresh = time.Now().Add(time.Duration(p.FullScreenRefreshInSec) * time.Second)
|
||||
|
||||
case common.SegmentMessageStart:
|
||||
case common.SegmentRectSeparator:
|
||||
@@ -39,7 +42,20 @@ func (p *RfbRequester) Consume(seg *common.RfbSegment) error {
|
||||
// timeForNextReq := p.lastRequestTime.Unix() + minTimeBetweenReq.Nanoseconds()/1000
|
||||
// if seg.UpcomingObjectType == int(common.FramebufferUpdate) && time.Now().Unix() > timeForNextReq {
|
||||
//time.Sleep(300 * time.Millisecond)
|
||||
p.Conn.FramebufferUpdateRequest(true, 0, 0, p.Width, p.Height)
|
||||
p.lastRequestTime = time.Now()
|
||||
incremental := true
|
||||
|
||||
if p.FullScreenRefreshInSec > 0 {
|
||||
// if p.nextFullScreenRefresh.IsZero() {
|
||||
// p.nextFullScreenRefresh = time.Now().Add(time.Duration(p.FullScreenRefreshInSec) * time.Second)
|
||||
// }
|
||||
if time.Now().Sub(p.nextFullScreenRefresh) <= 0 {
|
||||
logger.Warn(">>Creating keyframe")
|
||||
p.nextFullScreenRefresh = time.Now().Add(time.Duration(p.FullScreenRefreshInSec) * time.Second)
|
||||
incremental = false
|
||||
}
|
||||
}
|
||||
p.Conn.FramebufferUpdateRequest(incremental, 0, 0, p.Width, p.Height)
|
||||
//}
|
||||
default:
|
||||
}
|
||||
|
||||
@@ -310,34 +310,3 @@ func (msg *MsgClientCutText) Write(c io.Writer) error {
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
// MsgClientQemuExtendedKey holds the wire format message, for qemu keys
|
||||
type MsgClientQemuExtendedKey struct {
|
||||
SubType uint8 // sub type
|
||||
IsDown uint16 // button down indicator
|
||||
KeySym uint32 // key symbol
|
||||
KeyCode uint32 // key code
|
||||
}
|
||||
|
||||
func (*MsgClientQemuExtendedKey) Type() common.ClientMessageType {
|
||||
return common.QEMUExtendedKeyEventMsgType
|
||||
}
|
||||
|
||||
func (*MsgClientQemuExtendedKey) Read(c io.Reader) (common.ClientMessage, error) {
|
||||
msg := MsgClientQemuExtendedKey{}
|
||||
|
||||
if err := binary.Read(c, binary.BigEndian, &msg); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return &msg, nil
|
||||
}
|
||||
|
||||
func (msg *MsgClientQemuExtendedKey) Write(c io.Writer) error {
|
||||
if err := binary.Write(c, binary.BigEndian, msg.Type()); err != nil {
|
||||
return err
|
||||
}
|
||||
if err := binary.Write(c, binary.BigEndian, msg); err != nil {
|
||||
return err
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
@@ -95,7 +95,7 @@ func ServerSecurityHandler(cfg *ServerConfig, c *ServerConn) error {
|
||||
|
||||
sType, ok := secTypes[secType]
|
||||
if !ok {
|
||||
return fmt.Errorf("server type %d not implemented", secType)
|
||||
return fmt.Errorf("server type %d not implemented")
|
||||
}
|
||||
|
||||
var authCode uint32
|
||||
@@ -135,7 +135,7 @@ func ServerServerInitHandler(cfg *ServerConfig, c *ServerConn) error {
|
||||
NameLength: uint32(len(cfg.DesktopName)),
|
||||
NameText: []byte(cfg.DesktopName),
|
||||
}
|
||||
logger.Debugf("Server.ServerServerInitHandler initMessage: %v", srvInit)
|
||||
logger.Infof("Server.ServerServerInitHandler initMessage: %v", srvInit)
|
||||
if err := binary.Write(c, binary.BigEndian, srvInit.FBWidth); err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
@@ -168,17 +168,16 @@ func (c *ServerConn) handle() error {
|
||||
default:
|
||||
var messageType common.ClientMessageType
|
||||
if err := binary.Read(c, binary.BigEndian, &messageType); err != nil {
|
||||
logger.Errorf("ServerConn.handle error: %v", err)
|
||||
logger.Errorf("IServerConn.handle error: %v", err)
|
||||
return err
|
||||
}
|
||||
logger.Debugf("ServerConn.handle: got messagetype, %d", messageType)
|
||||
msg, ok := clientMessages[messageType]
|
||||
logger.Debugf("ServerConn.handle: found message type, %v", ok)
|
||||
if !ok {
|
||||
logger.Errorf("ServerConn.handle: unsupported message-type: %v", messageType)
|
||||
return fmt.Errorf("IServerConn.Handle: unsupported message-type: %v", messageType)
|
||||
}
|
||||
|
||||
parsedMsg, err := msg.Read(c)
|
||||
logger.Debugf("ServerConn.handle: got parsed messagetype, %v", parsedMsg)
|
||||
|
||||
//update connection for pixel format / color map changes
|
||||
switch parsedMsg.Type() {
|
||||
case common.SetPixelFormatMsgType:
|
||||
@@ -197,7 +196,7 @@ func (c *ServerConn) handle() error {
|
||||
return err
|
||||
}
|
||||
|
||||
logger.Debugf("IServerConn.Handle got ClientMessage: %s, %v", parsedMsg.Type(), parsedMsg)
|
||||
logger.Infof("IServerConn.Handle got ClientMessage: %s, %v", parsedMsg.Type(), parsedMsg)
|
||||
//TODO: treat set encodings by allowing only supported encoding in proxy configurations
|
||||
//// if parsedMsg.Type() == common.SetEncodingsMsgType{
|
||||
//// c.cfg.Encodings
|
||||
|
||||
@@ -15,7 +15,6 @@ var DefaultClientMessages = []common.ClientMessage{
|
||||
&MsgKeyEvent{},
|
||||
&MsgPointerEvent{},
|
||||
&MsgClientCutText{},
|
||||
&MsgClientQemuExtendedKey{},
|
||||
}
|
||||
|
||||
// FramebufferUpdate holds a FramebufferUpdate wire format message.
|
||||
|
||||
Reference in New Issue
Block a user