client.go

// Implementation of §7.5 Client-to-Server Messages.

package vnc

import (
	"fmt"
	"strings"
	"unicode"

	"github.com/golang/glog"
	"github.com/kward/go-vnc/buttons"
	"github.com/kward/go-vnc/encodings"
	"github.com/kward/go-vnc/keys"
	"github.com/kward/go-vnc/logging"
	"github.com/kward/go-vnc/messages"
	"github.com/kward/go-vnc/rfbflags"
)

// SetPixelFormatMessage holds the wire format message.
type SetPixelFormatMessage struct {
	Msg messages.ClientMessage // message-type
	_   [3]byte                // padding
	PF  PixelFormat            // pixel-format
}

// SetPixelFormat sets the format in which pixel values should be sent
// in FramebufferUpdate messages from the server.
//
// See RFC 6143 Section 7.5.1
func (c *ClientConn) SetPixelFormat(pf PixelFormat) error {
	if logging.V(logging.FnDeclLevel) {
		glog.Infof("ClientConn.%s", logging.FnNameWithArgs("%s", pf))
	}

	msg := SetPixelFormatMessage{
		Msg: messages.SetPixelFormat,
		PF:  pf,
	}
	if err := c.send(msg); err != nil {
		return err
	}

	// Invalidate the color map.
	if !rfbflags.IsTrueColor(pf.TrueColor) {
		c.colorMap = [256]Color{}
	}

	c.pixelFormat = pf
	return nil
}

// SetEncodingsMessage holds the wire format message, sans encoding-type field.
type SetEncodingsMessage struct {
	Msg     messages.ClientMessage // message-type
	_       [1]byte                // padding
	NumEncs uint16                 // number-of-encodings
}

// SetEncodings sets the encoding types in which the pixel data can be sent
// from the server. After calling this method, the encs slice given should not
// be modified.
//
// TODO(kward:20170306) Fix bad practice of mixing of protocol and internal
// state here.
//
// See RFC 6143 Section 7.5.2
func (c *ClientConn) SetEncodings(encs Encodings) error {
	if logging.V(logging.FnDeclLevel) {
		glog.Infof("ClientConn.%s", logging.FnNameWithArgs("%s", encs))
	}

	// Make sure RawEncoding is supported.
	haveRaw := false
	for _, v := range encs {
		if v.Type() == encodings.Raw {
			haveRaw = true
			break
		}
	}
	if !haveRaw {
		encs = append(encs, &RawEncoding{})
	}

	buf := NewBuffer(nil)

	// Prepare message.
	msg := SetEncodingsMessage{
		Msg:     messages.SetEncodings,
		NumEncs: uint16(len(encs)),
	}
	if err := buf.Write(msg); err != nil {
		return err
	}
	bytes, err := encs.Marshal()
	if err != nil {
		return err
	}
	if err := buf.Write(bytes); err != nil {
		return err
	}

	// Send message.
	if err := c.send(buf.Bytes()); err != nil {
		return err
	}

	c.encodings = encs
	return nil
}

// FramebufferUpdateRequestMessage holds the wire format message.
type FramebufferUpdateRequestMessage struct {
	Msg           messages.ClientMessage // message-type
	Inc           rfbflags.RFBFlag       // incremental
	X, Y          uint16                 // x-, y-position
	Width, Height uint16                 // width, height
}

// Requests a framebuffer update from the server. There may be an indefinite
// time between the request and the actual framebuffer update being received.
//
// See RFC 6143 Section 7.5.3
func (c *ClientConn) FramebufferUpdateRequest(inc rfbflags.RFBFlag, x, y, w, h uint16) error {
	msg := FramebufferUpdateRequestMessage{messages.FramebufferUpdateRequest, inc, x, y, w, h}
	return c.send(&msg)
}

// KeyEventMessage holds the wire format message.
type KeyEventMessage struct {
	Msg      messages.ClientMessage // message-type
	DownFlag rfbflags.RFBFlag       // down-flag
	_        [2]byte                // padding
	Key      keys.Key               // key
}

const (
	PressKey   = true
	ReleaseKey = false
)

// KeyEvent indicates a key press or release and sends it to the server.
// The key is indicated using the X Window System "keysym" value. Constants are
// provided in `keys/keys.go`. To simulate a key press, you must send a key with
// both a true and false down event.
//
// See RFC 6143 Section 7.5.4.
func (c *ClientConn) KeyEvent(key keys.Key, down bool) error {
	if logging.V(logging.FnDeclLevel) {
		glog.Infof("ClientConnt.%s", logging.FnNameWithArgs("%s, %t", key, down))
	}

	msg := KeyEventMessage{messages.KeyEvent, rfbflags.BoolToRFBFlag(down), [2]byte{}, key}
	if err := c.send(msg); err != nil {
		return err
	}

	settleUI()
	return nil
}

// PointerEventMessage holds the wire format message.
type PointerEventMessage struct {
	Msg  messages.ClientMessage // message-type
	Mask uint8                  // button-mask
	X, Y uint16                 // x-, y-position
}

// PointerEvent indicates that pointer movement or a pointer button
// press or release.
//
// The `button` is a bitwise mask of various Button values. When a button
// is set, it is pressed, when it is unset, it is released.
//
// See RFC 6143 Section 7.5.5
func (c *ClientConn) PointerEvent(button buttons.Button, x, y uint16) error {
	if logging.V(logging.FnDeclLevel) {
		glog.Info(logging.FnNameWithArgs("%s, %d, %d", button, x, y))
	}

	msg := PointerEventMessage{messages.PointerEvent, uint8(button), x, y}
	if err := c.send(msg); err != nil {
		return err
	}

	settleUI()
	return nil
}

// ClientCutTextMessage holds the wire format message, sans the text field.
type ClientCutTextMessage struct {
	Msg    messages.ClientMessage // message-type
	_      [3]byte                // padding
	Length uint32                 // length
}

// ClientCutText tells the server that the client has new text in its cut buffer.
// The text string MUST only contain Latin-1 characters. This encoding
// is compatible with Go's native string format, but can only use up to
// unicode.MaxLatin1 values.
//
// See RFC 6143 Section 7.5.6
func (c *ClientConn) ClientCutText(text string) error {
	if logging.V(logging.FnDeclLevel) {
		glog.Info(logging.FnNameWithArgs("%s", text))
	}

	for _, char := range text {
		if char > unicode.MaxLatin1 {
			return NewVNCError(fmt.Sprintf("Character %q is not valid Latin-1", char))
		}
	}

	// Strip carriage-return (0x0d) chars.
	// From RFC: "Ends of lines are represented by the newline character (0x0a)
	// alone. No carriage-return (0x0d) is used."
	text = strings.Join(strings.Split(text, "\r"), "")

	msg := ClientCutTextMessage{
		Msg:    messages.ClientCutText,
		Length: uint32(len(text)),
	}
	if err := c.send(msg); err != nil {
		return err
	}
	if err := c.send([]byte(text)); err != nil {
		return err
	}

	settleUI()
	return nil
}