123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471 |
- /* Copyright (C) 2002-2005 RealVNC Ltd. All Rights Reserved.
- * Copyright 2009-2019 Pierre Ossman for Cendio AB
- *
- * This is free software; you can redistribute it and/or modify
- * it under the terms of the GNU General Public License as published by
- * the Free Software Foundation; either version 2 of the License, or
- * (at your option) any later version.
- *
- * This software is distributed in the hope that it will be useful,
- * but WITHOUT ANY WARRANTY; without even the implied warranty of
- * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
- * GNU General Public License for more details.
- *
- * You should have received a copy of the GNU General Public License
- * along with this software; if not, write to the Free Software
- * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307,
- * USA.
- */
-
- #ifdef HAVE_CONFIG_H
- #include <config.h>
- #endif
-
- #include <stdio.h>
-
- #include <rdr/InStream.h>
- #include <rdr/ZlibInStream.h>
-
- #include <rfb/msgTypes.h>
- #include <rfb/qemuTypes.h>
- #include <rfb/clipboardTypes.h>
- #include <rfb/Exception.h>
- #include <rfb/util.h>
- #include <rfb/SMsgHandler.h>
- #include <rfb/SMsgReader.h>
- #include <rfb/Configuration.h>
- #include <rfb/LogWriter.h>
-
- using namespace rfb;
-
- static LogWriter vlog("SMsgReader");
-
- static IntParameter maxCutText("MaxCutText", "Maximum permitted length of an incoming clipboard update", 256*1024);
-
- SMsgReader::SMsgReader(SMsgHandler* handler_, rdr::InStream* is_)
- : handler(handler_), is(is_), state(MSGSTATE_IDLE)
- {
- }
-
- SMsgReader::~SMsgReader()
- {
- }
-
- bool SMsgReader::readClientInit()
- {
- if (!is->hasData(1))
- return false;
- bool shared = is->readU8();
- handler->clientInit(shared);
- return true;
- }
-
- bool SMsgReader::readMsg()
- {
- bool ret;
-
- if (state == MSGSTATE_IDLE) {
- if (!is->hasData(1))
- return false;
-
- currentMsgType = is->readU8();
- state = MSGSTATE_MESSAGE;
- }
-
- switch (currentMsgType) {
- case msgTypeSetPixelFormat:
- ret = readSetPixelFormat();
- break;
- case msgTypeSetEncodings:
- ret = readSetEncodings();
- break;
- case msgTypeSetDesktopSize:
- ret = readSetDesktopSize();
- break;
- case msgTypeFramebufferUpdateRequest:
- ret = readFramebufferUpdateRequest();
- break;
- case msgTypeEnableContinuousUpdates:
- ret = readEnableContinuousUpdates();
- break;
- case msgTypeClientFence:
- ret = readFence();
- break;
- case msgTypeKeyEvent:
- ret = readKeyEvent();
- break;
- case msgTypePointerEvent:
- ret = readPointerEvent();
- break;
- case msgTypeClientCutText:
- ret = readClientCutText();
- break;
- case msgTypeQEMUClientMessage:
- ret = readQEMUMessage();
- break;
- default:
- vlog.error("unknown message type %d", currentMsgType);
- throw Exception("unknown message type");
- }
-
- if (ret)
- state = MSGSTATE_IDLE;
-
- return ret;
- }
-
- bool SMsgReader::readSetPixelFormat()
- {
- if (!is->hasData(3 + 16))
- return false;
- is->skip(3);
- PixelFormat pf;
- pf.read(is);
- handler->setPixelFormat(pf);
- return true;
- }
-
- bool SMsgReader::readSetEncodings()
- {
- if (!is->hasData(1 + 2))
- return false;
-
- is->setRestorePoint();
-
- is->skip(1);
-
- int nEncodings = is->readU16();
-
- if (!is->hasDataOrRestore(nEncodings * 4))
- return false;
- is->clearRestorePoint();
-
- rdr::S32Array encodings(nEncodings);
- for (int i = 0; i < nEncodings; i++)
- encodings.buf[i] = is->readU32();
-
- handler->setEncodings(nEncodings, encodings.buf);
-
- return true;
- }
-
- bool SMsgReader::readSetDesktopSize()
- {
- int width, height;
- int screens, i;
- rdr::U32 id, flags;
- int sx, sy, sw, sh;
- ScreenSet layout;
-
- if (!is->hasData(1 + 2 + 2 + 1 + 1))
- return true;
-
- is->setRestorePoint();
-
- is->skip(1);
-
- width = is->readU16();
- height = is->readU16();
-
- screens = is->readU8();
- is->skip(1);
-
- if (!is->hasDataOrRestore(screens * (4 + 2 + 2 + 2 + 2 + 4)))
- return false;
- is->clearRestorePoint();
-
- for (i = 0;i < screens;i++) {
- id = is->readU32();
- sx = is->readU16();
- sy = is->readU16();
- sw = is->readU16();
- sh = is->readU16();
- flags = is->readU32();
-
- layout.add_screen(Screen(id, sx, sy, sw, sh, flags));
- }
-
- handler->setDesktopSize(width, height, layout);
-
- return true;
- }
-
- bool SMsgReader::readFramebufferUpdateRequest()
- {
- if (!is->hasData(1 + 2 + 2 + 2 + 2))
- return false;
- bool inc = is->readU8();
- int x = is->readU16();
- int y = is->readU16();
- int w = is->readU16();
- int h = is->readU16();
- handler->framebufferUpdateRequest(Rect(x, y, x+w, y+h), inc);
- return true;
- }
-
- bool SMsgReader::readEnableContinuousUpdates()
- {
- bool enable;
- int x, y, w, h;
-
- if (!is->hasData(1 + 2 + 2 + 2 + 2))
- return false;
-
- enable = is->readU8();
-
- x = is->readU16();
- y = is->readU16();
- w = is->readU16();
- h = is->readU16();
-
- handler->enableContinuousUpdates(enable, x, y, w, h);
-
- return true;
- }
-
- bool SMsgReader::readFence()
- {
- rdr::U32 flags;
- rdr::U8 len;
- char data[64];
-
- if (!is->hasData(3 + 4 + 1))
- return false;
-
- is->setRestorePoint();
-
- is->skip(3);
-
- flags = is->readU32();
-
- len = is->readU8();
-
- if (!is->hasDataOrRestore(len))
- return false;
- is->clearRestorePoint();
-
- if (len > sizeof(data)) {
- vlog.error("Ignoring fence with too large payload");
- is->skip(len);
- return true;
- }
-
- is->readBytes(data, len);
-
- handler->fence(flags, len, data);
-
- return true;
- }
-
- bool SMsgReader::readKeyEvent()
- {
- if (!is->hasData(1 + 2 + 4))
- return false;
- bool down = is->readU8();
- is->skip(2);
- rdr::U32 key = is->readU32();
- handler->keyEvent(key, 0, down);
- return true;
- }
-
- bool SMsgReader::readPointerEvent()
- {
- if (!is->hasData(1 + 2 + 2))
- return false;
- int mask = is->readU8();
- int x = is->readU16();
- int y = is->readU16();
- handler->pointerEvent(Point(x, y), mask);
- return true;
- }
-
-
- bool SMsgReader::readClientCutText()
- {
- if (!is->hasData(3 + 4))
- return false;
-
- is->setRestorePoint();
-
- is->skip(3);
- rdr::U32 len = is->readU32();
-
- if (len & 0x80000000) {
- rdr::S32 slen = len;
- slen = -slen;
- if (readExtendedClipboard(slen)) {
- is->clearRestorePoint();
- return true;
- } else {
- is->gotoRestorePoint();
- return false;
- }
- }
-
- if (!is->hasDataOrRestore(len))
- return false;
- is->clearRestorePoint();
-
- if (len > (size_t)maxCutText) {
- is->skip(len);
- vlog.error("Cut text too long (%d bytes) - ignoring", len);
- return true;
- }
-
- CharArray ca(len);
- is->readBytes(ca.buf, len);
- CharArray filtered(convertLF(ca.buf, len));
- handler->clientCutText(filtered.buf);
-
- return true;
- }
-
- bool SMsgReader::readExtendedClipboard(rdr::S32 len)
- {
- rdr::U32 flags;
- rdr::U32 action;
-
- if (!is->hasData(len))
- return false;
-
- if (len < 4)
- throw Exception("Invalid extended clipboard message");
- if (len > maxCutText) {
- vlog.error("Extended clipboard message too long (%d bytes) - ignoring", len);
- is->skip(len);
- return true;
- }
-
- flags = is->readU32();
- action = flags & clipboardActionMask;
-
- if (action & clipboardCaps) {
- int i;
- size_t num;
- rdr::U32 lengths[16];
-
- num = 0;
- for (i = 0;i < 16;i++) {
- if (flags & (1 << i))
- num++;
- }
-
- if (len < (rdr::S32)(4 + 4*num))
- throw Exception("Invalid extended clipboard message");
-
- num = 0;
- for (i = 0;i < 16;i++) {
- if (flags & (1 << i))
- lengths[num++] = is->readU32();
- }
-
- handler->handleClipboardCaps(flags, lengths);
- } else if (action == clipboardProvide) {
- rdr::ZlibInStream zis;
-
- int i;
- size_t num;
- size_t lengths[16];
- rdr::U8* buffers[16];
-
- zis.setUnderlying(is, len - 4);
-
- num = 0;
- for (i = 0;i < 16;i++) {
- if (!(flags & 1 << i))
- continue;
-
- if (!zis.hasData(4))
- throw Exception("Extended clipboard decode error");
-
- lengths[num] = zis.readU32();
-
- if (!zis.hasData(lengths[num]))
- throw Exception("Extended clipboard decode error");
-
- if (lengths[num] > (size_t)maxCutText) {
- vlog.error("Extended clipboard data too long (%d bytes) - ignoring",
- (unsigned)lengths[num]);
- zis.skip(lengths[num]);
- flags &= ~(1 << i);
- continue;
- }
-
- buffers[num] = new rdr::U8[lengths[num]];
- zis.readBytes(buffers[num], lengths[num]);
- num++;
- }
-
- zis.flushUnderlying();
- zis.setUnderlying(NULL, 0);
-
- handler->handleClipboardProvide(flags, lengths, buffers);
-
- num = 0;
- for (i = 0;i < 16;i++) {
- if (!(flags & 1 << i))
- continue;
- delete [] buffers[num++];
- }
- } else {
- switch (action) {
- case clipboardRequest:
- handler->handleClipboardRequest(flags);
- break;
- case clipboardPeek:
- handler->handleClipboardPeek(flags);
- break;
- case clipboardNotify:
- handler->handleClipboardNotify(flags);
- break;
- default:
- throw Exception("Invalid extended clipboard action");
- }
- }
-
- return true;
- }
-
- bool SMsgReader::readQEMUMessage()
- {
- int subType;
- bool ret;
-
- if (!is->hasData(1))
- return false;
-
- is->setRestorePoint();
-
- subType = is->readU8();
-
- switch (subType) {
- case qemuExtendedKeyEvent:
- ret = readQEMUKeyEvent();
- break;
- default:
- throw Exception("unknown QEMU submessage type %d", subType);
- }
-
- if (!ret) {
- is->gotoRestorePoint();
- return false;
- } else {
- is->clearRestorePoint();
- return true;
- }
- }
-
- bool SMsgReader::readQEMUKeyEvent()
- {
- if (!is->hasData(2 + 4 + 4))
- return false;
- bool down = is->readU16();
- rdr::U32 keysym = is->readU32();
- rdr::U32 keycode = is->readU32();
- if (!keycode) {
- vlog.error("Key event without keycode - ignoring");
- return true;
- }
- handler->keyEvent(keysym, keycode, down);
- return true;
- }
|