123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206 |
- /* Copyright (C) 2002-2005 RealVNC Ltd. All Rights Reserved.
- * Copyright 2014-2022 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 <rdr/InStream.h>
- #include <rdr/MemInStream.h>
- #include <rdr/OutStream.h>
-
- #include <rfb/Exception.h>
- #include <rfb/ServerParams.h>
- #include <rfb/PixelBuffer.h>
- #include <rfb/HextileDecoder.h>
- #include <rfb/hextileConstants.h>
-
- using namespace rfb;
-
- HextileDecoder::HextileDecoder() : Decoder(DecoderPlain)
- {
- }
-
- HextileDecoder::~HextileDecoder()
- {
- }
-
- bool HextileDecoder::readRect(const Rect& r, rdr::InStream* is,
- const ServerParams& server, rdr::OutStream* os)
- {
- Rect t;
- size_t bytesPerPixel;
-
- is->setRestorePoint();
-
- bytesPerPixel = server.pf().bpp/8;
-
- for (t.tl.y = r.tl.y; t.tl.y < r.br.y; t.tl.y += 16) {
-
- t.br.y = __rfbmin(r.br.y, t.tl.y + 16);
-
- for (t.tl.x = r.tl.x; t.tl.x < r.br.x; t.tl.x += 16) {
- uint8_t tileType;
-
- t.br.x = __rfbmin(r.br.x, t.tl.x + 16);
-
- if (!is->hasDataOrRestore(1))
- return false;
-
- tileType = is->readU8();
- os->writeU8(tileType);
-
- if (tileType & hextileRaw) {
- if (!is->hasDataOrRestore(t.area() * bytesPerPixel))
- return false;
- os->copyBytes(is, t.area() * bytesPerPixel);
- continue;
- }
-
-
- if (tileType & hextileBgSpecified) {
- if (!is->hasDataOrRestore(bytesPerPixel))
- return false;
- os->copyBytes(is, bytesPerPixel);
- }
-
- if (tileType & hextileFgSpecified) {
- if (!is->hasDataOrRestore(bytesPerPixel))
- return false;
- os->copyBytes(is, bytesPerPixel);
- }
-
- if (tileType & hextileAnySubrects) {
- uint8_t nSubrects;
-
- if (!is->hasDataOrRestore(1))
- return false;
-
- nSubrects = is->readU8();
- os->writeU8(nSubrects);
-
- if (tileType & hextileSubrectsColoured) {
- if (!is->hasDataOrRestore(nSubrects * (bytesPerPixel + 2)))
- return false;
- os->copyBytes(is, nSubrects * (bytesPerPixel + 2));
- } else {
- if (!is->hasDataOrRestore(nSubrects * 2))
- return false;
- os->copyBytes(is, nSubrects * 2);
- }
- }
- }
- }
-
- is->clearRestorePoint();
-
- return true;
- }
-
- void HextileDecoder::decodeRect(const Rect& r, const uint8_t* buffer,
- size_t buflen, const ServerParams& server,
- ModifiablePixelBuffer* pb)
- {
- rdr::MemInStream is(buffer, buflen);
- const PixelFormat& pf = server.pf();
- switch (pf.bpp) {
- case 8: hextileDecode<uint8_t >(r, &is, pf, pb); break;
- case 16: hextileDecode<uint16_t>(r, &is, pf, pb); break;
- case 32: hextileDecode<uint32_t>(r, &is, pf, pb); break;
- }
- }
-
- template<class T>
- inline T HextileDecoder::readPixel(rdr::InStream* is)
- {
- if (sizeof(T) == 1)
- return is->readOpaque8();
- if (sizeof(T) == 2)
- return is->readOpaque16();
- if (sizeof(T) == 4)
- return is->readOpaque32();
- }
-
- template<class T>
- void HextileDecoder::hextileDecode(const Rect& r, rdr::InStream* is,
- const PixelFormat& pf,
- ModifiablePixelBuffer* pb)
- {
- Rect t;
- T bg = 0;
- T fg = 0;
- T buf[16 * 16];
-
- for (t.tl.y = r.tl.y; t.tl.y < r.br.y; t.tl.y += 16) {
-
- t.br.y = __rfbmin(r.br.y, t.tl.y + 16);
-
- for (t.tl.x = r.tl.x; t.tl.x < r.br.x; t.tl.x += 16) {
-
- t.br.x = __rfbmin(r.br.x, t.tl.x + 16);
-
- int tileType = is->readU8();
-
- if (tileType & hextileRaw) {
- is->readBytes((uint8_t*)buf, t.area() * sizeof(T));
- pb->imageRect(pf, t, buf);
- continue;
- }
-
- if (tileType & hextileBgSpecified)
- bg = readPixel<T>(is);
-
- int len = t.area();
- T* ptr = buf;
- while (len-- > 0) *ptr++ = bg;
-
- if (tileType & hextileFgSpecified)
- fg = readPixel<T>(is);
-
- if (tileType & hextileAnySubrects) {
- int nSubrects = is->readU8();
-
- for (int i = 0; i < nSubrects; i++) {
-
- if (tileType & hextileSubrectsColoured)
- fg = readPixel<T>(is);
-
- int xy = is->readU8();
- int wh = is->readU8();
-
- int x = ((xy >> 4) & 15);
- int y = (xy & 15);
- int w = ((wh >> 4) & 15) + 1;
- int h = (wh & 15) + 1;
- if (x + w > 16 || y + h > 16) {
- throw rfb::Exception("HEXTILE_DECODE: Hextile out of bounds");
- }
- T* ptr = buf + y * t.width() + x;
- int rowAdd = t.width() - w;
- while (h-- > 0) {
- int len = w;
- while (len-- > 0) *ptr++ = fg;
- ptr += rowAdd;
- }
- }
- }
- pb->imageRect(pf, t, buf);
- }
- }
- }
|