116 lines
3.8 KiB
Go
116 lines
3.8 KiB
Go
package vnc
|
|
|
|
import (
|
|
"bytes"
|
|
"image"
|
|
"sync"
|
|
)
|
|
|
|
// EncodingType represents a known VNC encoding type.
|
|
type EncodingType int32
|
|
|
|
//go:generate stringer -type=EncodingType
|
|
|
|
const (
|
|
// EncRaw raw encoding
|
|
EncRaw EncodingType = 0
|
|
// EncCopyRect copyrect encoding
|
|
EncCopyRect EncodingType = 1
|
|
|
|
EncRRE EncodingType = 2
|
|
EncCoRRE EncodingType = 4
|
|
EncHextile EncodingType = 5
|
|
EncZlib EncodingType = 6
|
|
EncTight EncodingType = 7
|
|
EncZlibHex EncodingType = 8
|
|
EncUltra1 EncodingType = 9
|
|
EncUltra2 EncodingType = 10
|
|
EncJPEG EncodingType = 21
|
|
EncJRLE EncodingType = 22
|
|
EncTRLE EncodingType = 15
|
|
EncZRLE EncodingType = 16
|
|
EncAtenAST2100 EncodingType = 0x57
|
|
EncAtenASTJPEG EncodingType = 0x58
|
|
EncAtenHermon EncodingType = 0x59
|
|
EncAtenYarkon EncodingType = 0x60
|
|
EncAtenPilot3 EncodingType = 0x61
|
|
EncJPEGQualityLevelPseudo10 EncodingType = -23
|
|
EncJPEGQualityLevelPseudo9 EncodingType = -24
|
|
EncJPEGQualityLevelPseudo8 EncodingType = -25
|
|
EncJPEGQualityLevelPseudo7 EncodingType = -26
|
|
EncJPEGQualityLevelPseudo6 EncodingType = -27
|
|
EncJPEGQualityLevelPseudo5 EncodingType = -28
|
|
EncJPEGQualityLevelPseudo4 EncodingType = -29
|
|
EncJPEGQualityLevelPseudo3 EncodingType = -30
|
|
EncJPEGQualityLevelPseudo2 EncodingType = -31
|
|
EncJPEGQualityLevelPseudo1 EncodingType = -32
|
|
EncCursorPseudo EncodingType = -239
|
|
EncXCursorPseudo EncodingType = -240
|
|
EncDesktopSizePseudo EncodingType = -223
|
|
EncLastRectPseudo EncodingType = -224
|
|
EncCompressionLevel10 EncodingType = -247
|
|
EncCompressionLevel9 EncodingType = -248
|
|
EncCompressionLevel8 EncodingType = -249
|
|
EncCompressionLevel7 EncodingType = -250
|
|
EncCompressionLevel6 EncodingType = -251
|
|
EncCompressionLevel5 EncodingType = -252
|
|
EncCompressionLevel4 EncodingType = -253
|
|
EncCompressionLevel3 EncodingType = -254
|
|
EncCompressionLevel2 EncodingType = -255
|
|
EncCompressionLevel1 EncodingType = -256
|
|
EncQEMUPointerMotionChangePseudo EncodingType = -257
|
|
EncQEMUExtendedKeyEventPseudo EncodingType = -258
|
|
EncTightPng EncodingType = -260
|
|
EncDesktopNamePseudo EncodingType = -307
|
|
EncExtendedDesktopSizePseudo EncodingType = -308
|
|
EncXvpPseudo EncodingType = -309
|
|
EncClientRedirect EncodingType = -311
|
|
EncFencePseudo EncodingType = -312
|
|
EncContinuousUpdatesPseudo EncodingType = -313
|
|
)
|
|
|
|
var bPool = sync.Pool{
|
|
New: func() interface{} {
|
|
// The Pool's New function should generally only return pointer
|
|
// types, since a pointer can be put into the return interface
|
|
// value without an allocation:
|
|
return new(bytes.Buffer)
|
|
},
|
|
}
|
|
|
|
// Encoding represents interface for vnc encoding
|
|
type Encoding interface {
|
|
Type() EncodingType
|
|
Read(Conn, *Rectangle) error
|
|
Write(Conn, *Rectangle) error
|
|
Supported(Conn) bool
|
|
}
|
|
|
|
func setBit(n uint8, pos uint8) uint8 {
|
|
n |= (1 << pos)
|
|
return n
|
|
}
|
|
|
|
func clrBit(n uint8, pos uint8) uint8 {
|
|
n = n &^ (1 << pos)
|
|
return n
|
|
}
|
|
|
|
func hasBit(n uint8, pos uint8) bool {
|
|
v := n & (1 << pos)
|
|
return (v > 0)
|
|
}
|
|
|
|
func getBit(n uint8, pos uint8) uint8 {
|
|
n = n & (1 << pos)
|
|
return n
|
|
}
|
|
|
|
func newRGBAImage(rgba []byte, rect *Rectangle) image.Image {
|
|
img := &image.RGBA{Stride: 4 * int(rect.Width)}
|
|
img.Pix = rgba
|
|
img.Rect.Max.X = int(rect.Width)
|
|
img.Rect.Max.Y = int(rect.Height)
|
|
return img
|
|
}
|