123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243 |
- /* Copyright (C) 2002-2005 RealVNC Ltd. All Rights Reserved.
- * Copyright 2009-2014 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.
- */
- #include <stdio.h>
- #include <rdr/InStream.h>
- #include <rfb/msgTypes.h>
- #include <rfb/qemuTypes.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_)
- {
- }
-
- SMsgReader::~SMsgReader()
- {
- }
-
- void SMsgReader::readClientInit()
- {
- bool shared = is->readU8();
- handler->clientInit(shared);
- }
-
- void SMsgReader::readMsg()
- {
- int msgType = is->readU8();
- switch (msgType) {
- case msgTypeSetPixelFormat:
- readSetPixelFormat();
- break;
- case msgTypeSetEncodings:
- readSetEncodings();
- break;
- case msgTypeSetDesktopSize:
- readSetDesktopSize();
- break;
- case msgTypeFramebufferUpdateRequest:
- readFramebufferUpdateRequest();
- break;
- case msgTypeEnableContinuousUpdates:
- readEnableContinuousUpdates();
- break;
- case msgTypeClientFence:
- readFence();
- break;
- case msgTypeKeyEvent:
- readKeyEvent();
- break;
- case msgTypePointerEvent:
- readPointerEvent();
- break;
- case msgTypeClientCutText:
- readClientCutText();
- break;
- case msgTypeQEMUClientMessage:
- readQEMUMessage();
- break;
- default:
- vlog.error("unknown message type %d", msgType);
- throw Exception("unknown message type");
- }
- }
-
- void SMsgReader::readSetPixelFormat()
- {
- is->skip(3);
- PixelFormat pf;
- pf.read(is);
- handler->setPixelFormat(pf);
- }
-
- void SMsgReader::readSetEncodings()
- {
- is->skip(1);
- int nEncodings = is->readU16();
- rdr::S32Array encodings(nEncodings);
- for (int i = 0; i < nEncodings; i++)
- encodings.buf[i] = is->readU32();
- handler->setEncodings(nEncodings, encodings.buf);
- }
-
- void SMsgReader::readSetDesktopSize()
- {
- int width, height;
- int screens, i;
- rdr::U32 id, flags;
- int sx, sy, sw, sh;
- ScreenSet layout;
-
- is->skip(1);
-
- width = is->readU16();
- height = is->readU16();
-
- screens = is->readU8();
- is->skip(1);
-
- 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);
- }
-
- void SMsgReader::readFramebufferUpdateRequest()
- {
- 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);
- }
-
- void SMsgReader::readEnableContinuousUpdates()
- {
- bool enable;
- int x, y, w, h;
-
- enable = is->readU8();
-
- x = is->readU16();
- y = is->readU16();
- w = is->readU16();
- h = is->readU16();
-
- handler->enableContinuousUpdates(enable, x, y, w, h);
- }
-
- void SMsgReader::readFence()
- {
- rdr::U32 flags;
- rdr::U8 len;
- char data[64];
-
- is->skip(3);
-
- flags = is->readU32();
-
- len = is->readU8();
- if (len > sizeof(data)) {
- vlog.error("Ignoring fence with too large payload");
- is->skip(len);
- return;
- }
-
- is->readBytes(data, len);
-
- handler->fence(flags, len, data);
- }
-
- void SMsgReader::readKeyEvent()
- {
- bool down = is->readU8();
- is->skip(2);
- rdr::U32 key = is->readU32();
- handler->keyEvent(key, 0, down);
- }
-
- void SMsgReader::readPointerEvent()
- {
- int mask = is->readU8();
- int x = is->readU16();
- int y = is->readU16();
- handler->pointerEvent(Point(x, y), mask);
- }
-
-
- void SMsgReader::readClientCutText()
- {
- is->skip(3);
- int len = is->readU32();
- if (len < 0) {
- throw Exception("Cut text too long.");
- }
- if (len > maxCutText) {
- is->skip(len);
- vlog.error("Cut text too long (%d bytes) - ignoring", len);
- return;
- }
- CharArray ca(len);
- is->readBytes(ca.buf, len);
- CharArray filtered(convertLF(ca.buf, len));
- handler->clientCutText(filtered.buf);
- }
-
- void SMsgReader::readQEMUMessage()
- {
- int subType = is->readU8();
- switch (subType) {
- case qemuExtendedKeyEvent:
- readQEMUKeyEvent();
- break;
- default:
- throw Exception("unknown QEMU submessage type %d", subType);
- }
- }
-
- void SMsgReader::readQEMUKeyEvent()
- {
- bool down = is->readU16();
- rdr::U32 keysym = is->readU32();
- rdr::U32 keycode = is->readU32();
- if (!keycode) {
- vlog.error("Key event without keycode - ignoring");
- return;
- }
- handler->keyEvent(keysym, keycode, down);
- }
|