144 lines
3.1 KiB
Go
144 lines
3.1 KiB
Go
package bootcommand
|
|
|
|
import (
|
|
"fmt"
|
|
"log"
|
|
"os"
|
|
"strings"
|
|
"time"
|
|
"unicode"
|
|
|
|
"github.com/hashicorp/packer/common"
|
|
vnc "github.com/mitchellh/go-vnc"
|
|
)
|
|
|
|
const KeyLeftShift uint32 = 0xFFE1
|
|
|
|
type vncDriver struct {
|
|
c *vnc.ClientConn
|
|
interval time.Duration
|
|
specialMap map[string]uint32
|
|
// keyEvent can set this error which will prevent it from continuing
|
|
err error
|
|
}
|
|
|
|
func NewVNCDriver(c *vnc.ClientConn) *vncDriver {
|
|
// We delay (default 100ms) between each key event to allow for CPU or
|
|
// network latency. See PackerKeyEnv for tuning.
|
|
keyInterval := common.PackerKeyDefault
|
|
if delay, err := time.ParseDuration(os.Getenv(common.PackerKeyEnv)); err == nil {
|
|
keyInterval = delay
|
|
}
|
|
|
|
// Scancodes reference: https://github.com/qemu/qemu/blob/master/ui/vnc_keysym.h
|
|
sMap := make(map[string]uint32)
|
|
sMap["bs"] = 0xFF08
|
|
sMap["del"] = 0xFFFF
|
|
sMap["enter"] = 0xFF0D
|
|
sMap["esc"] = 0xFF1B
|
|
sMap["f1"] = 0xFFBE
|
|
sMap["f2"] = 0xFFBF
|
|
sMap["f3"] = 0xFFC0
|
|
sMap["f4"] = 0xFFC1
|
|
sMap["f5"] = 0xFFC2
|
|
sMap["f6"] = 0xFFC3
|
|
sMap["f7"] = 0xFFC4
|
|
sMap["f8"] = 0xFFC5
|
|
sMap["f9"] = 0xFFC6
|
|
sMap["f10"] = 0xFFC7
|
|
sMap["f11"] = 0xFFC8
|
|
sMap["f12"] = 0xFFC9
|
|
sMap["return"] = 0xFF0D
|
|
sMap["tab"] = 0xFF09
|
|
sMap["up"] = 0xFF52
|
|
sMap["down"] = 0xFF54
|
|
sMap["left"] = 0xFF51
|
|
sMap["right"] = 0xFF53
|
|
sMap["spacebar"] = 0x020
|
|
sMap["insert"] = 0xFF63
|
|
sMap["home"] = 0xFF50
|
|
sMap["end"] = 0xFF57
|
|
sMap["pageUp"] = 0xFF55
|
|
sMap["pageDown"] = 0xFF56
|
|
sMap["leftAlt"] = 0xFFE9
|
|
sMap["leftCtrl"] = 0xFFE3
|
|
sMap["leftShift"] = 0xFFE1
|
|
sMap["rightAlt"] = 0xFFEA
|
|
sMap["rightCtrl"] = 0xFFE4
|
|
sMap["rightShift"] = 0xFFE2
|
|
sMap["leftSuper"] = 0xFFEB
|
|
sMap["rightSuper"] = 0xFFEC
|
|
|
|
return &vncDriver{
|
|
c: c,
|
|
interval: keyInterval,
|
|
specialMap: sMap,
|
|
}
|
|
}
|
|
|
|
func (d *vncDriver) keyEvent(k uint32, down bool) error {
|
|
if d.err != nil {
|
|
return nil
|
|
}
|
|
if err := d.c.KeyEvent(k, down); err != nil {
|
|
d.err = err
|
|
return err
|
|
}
|
|
time.Sleep(d.interval)
|
|
return nil
|
|
}
|
|
|
|
// Finalize does nothing here
|
|
func (d *vncDriver) Finalize() error {
|
|
return nil
|
|
}
|
|
|
|
func (d *vncDriver) SendKey(key rune, action KeyAction) error {
|
|
keyShift := unicode.IsUpper(key) || strings.ContainsRune(shiftedChars, key)
|
|
keyCode := uint32(key)
|
|
log.Printf("Sending char '%c', code 0x%X, shift %v", key, keyCode, keyShift)
|
|
|
|
switch action {
|
|
case KeyOn:
|
|
if keyShift {
|
|
d.keyEvent(KeyLeftShift, true)
|
|
}
|
|
d.keyEvent(keyCode, true)
|
|
case KeyOff:
|
|
if keyShift {
|
|
d.keyEvent(KeyLeftShift, false)
|
|
}
|
|
d.keyEvent(keyCode, false)
|
|
case KeyPress:
|
|
if keyShift {
|
|
d.keyEvent(KeyLeftShift, true)
|
|
}
|
|
d.keyEvent(keyCode, true)
|
|
d.keyEvent(keyCode, false)
|
|
if keyShift {
|
|
d.keyEvent(KeyLeftShift, false)
|
|
}
|
|
}
|
|
return d.err
|
|
}
|
|
|
|
func (d *vncDriver) SendSpecial(special string, action KeyAction) error {
|
|
keyCode, ok := d.specialMap[special]
|
|
if !ok {
|
|
return fmt.Errorf("special %s not found.", special)
|
|
}
|
|
log.Printf("Special code '<%s>' found, replacing with: 0x%X", special, keyCode)
|
|
|
|
switch action {
|
|
case KeyOn:
|
|
d.keyEvent(keyCode, true)
|
|
case KeyOff:
|
|
d.keyEvent(keyCode, false)
|
|
case KeyPress:
|
|
d.keyEvent(keyCode, true)
|
|
d.keyEvent(keyCode, false)
|
|
}
|
|
|
|
return d.err
|
|
}
|