go-rfb/example/proxy/main.go
Vasiliy Tolstov 49c704bc67 complete proxy support
Signed-off-by: Vasiliy Tolstov <v.tolstov@selfip.ru>
2017-06-13 16:20:35 +03:00

106 lines
2.5 KiB
Go

package main
import (
"context"
"log"
"net"
vnc "github.com/vtolstov/go-vnc"
)
func main() {
ln, err := net.Listen("tcp", ":5900")
if err != nil {
log.Fatalf("Error listen. %v", err)
}
schClient := make(chan vnc.ClientMessage)
schServer := make(chan vnc.ServerMessage)
scfg := &vnc.ServerConfig{
Width: 800,
Height: 600,
VersionHandler: vnc.ServerVersionHandler,
SecurityHandler: vnc.ServerSecurityHandler,
SecurityHandlers: []vnc.SecurityHandler{&vnc.ClientAuthNone{}},
ClientInitHandler: vnc.ServerClientInitHandler,
ServerInitHandler: vnc.ServerServerInitHandler,
Encodings: []vnc.Encoding{&vnc.RawEncoding{}},
PixelFormat: vnc.PixelFormat32bit,
ClientMessageCh: schClient,
ServerMessageCh: schServer,
ClientMessages: vnc.DefaultClientMessages,
DesktopName: []byte("vnc proxy"),
}
c, err := net.Dial("tcp", "127.0.0.1:5995")
if err != nil {
log.Fatalf("Error dial. %v", err)
}
cchServer := make(chan vnc.ServerMessage)
cchClient := make(chan vnc.ClientMessage)
ccfg := &vnc.ClientConfig{
VersionHandler: vnc.ClientVersionHandler,
SecurityHandler: vnc.ClientSecurityHandler,
SecurityHandlers: []vnc.SecurityHandler{&vnc.ClientAuthNone{}},
ClientInitHandler: vnc.ClientClientInitHandler,
ServerInitHandler: vnc.ClientServerInitHandler,
PixelFormat: vnc.PixelFormat32bit,
ClientMessageCh: cchClient,
ServerMessageCh: cchServer,
ServerMessages: vnc.DefaultServerMessages,
Encodings: []vnc.Encoding{&vnc.RawEncoding{}},
}
cc, err := vnc.Connect(context.Background(), c, ccfg)
if err != nil {
log.Fatalf("Error dial. %v", err)
}
scfg.Width = cc.Width()
scfg.Height = cc.Height()
scfg.PixelFormat = cc.PixelFormat()
go vnc.Serve(context.Background(), ln, scfg)
defer cc.Close()
go cc.Handle()
for {
select {
case msg := <-cchServer:
switch msg.Type() {
default:
schServer <- msg
}
case msg := <-schClient:
switch msg.Type() {
case vnc.SetEncodingsMsgType:
msg0 := &vnc.SetPixelFormat{
PF: *vnc.PixelFormat32bit,
}
cchClient <- msg0
encRaw := &vnc.RawEncoding{}
msg1 := &vnc.SetEncodings{
Encodings: []vnc.EncodingType{encRaw.Type()},
}
cchClient <- msg1
/*
msg2 := &vnc.FramebufferUpdateRequest{
Inc: 0,
X: 0,
Y: 0,
Width: cc.Width(),
Height: cc.Height(),
}
cchClient <- msg2
*/
case vnc.SetPixelFormatMsgType:
cchClient <- msg
default:
cchClient <- msg
}
}
}
}