/**************************************************************************/ /* */ /* Copyright (c) 2001, 2010 NoMachine, http://www.nomachine.com/. */ /* */ /* NXCOMP, NX protocol compression and NX extensions to this software */ /* are copyright of NoMachine. Redistribution and use of the present */ /* software is allowed according to terms specified in the file LICENSE */ /* which comes in the source distribution. */ /* */ /* Check http://www.nomachine.com/licensing.html for applicability. */ /* */ /* NX and NoMachine are trademarks of Medialogic S.p.A. */ /* */ /* All rights reserved. */ /* */ /**************************************************************************/ #ifndef DecodeBuffer_H #define DecodeBuffer_H #include <string.h> #include "IntCache.h" #include "CharCache.h" #include "XidCache.h" #include "FreeCache.h" #include "OpcodeCache.h" #include "ActionCache.h" #define DECODE_BUFFER_OVERFLOW_SIZE 4194304 #define DECODE_BUFFER_POSTFIX_SIZE 1 class DecodeBuffer { public: DecodeBuffer(const unsigned char *data, unsigned int length); ~DecodeBuffer() { } int decodeValue(unsigned int &value, unsigned int numBits, unsigned int blockSize = 0, int endOkay = 0); int decodeCachedValue(unsigned int &value, unsigned int numBits, IntCache &cache, unsigned int blockSize = 0, int endOkay = 0); int decodeCachedValue(unsigned char &value, unsigned int numBits, CharCache &cache, unsigned int blockSize = 0, int endOkay = 0); void decodeDiffCachedValue(unsigned int &value, unsigned int &previous, unsigned int numBits, IntCache &cache, unsigned int blockSize = 0) { decodeCachedValue(value, numBits, cache, blockSize); previous += (value + 1); previous &= (0xffffffff >> (32 - numBits)); value = previous; } void decodeBoolValue(unsigned int &value) { decodeValue(value, 1); } int decodeOpcodeValue(unsigned char &value, OpcodeCache &cache, int endOkay = 0) { int result = decodeCachedValue(value, 8, cache.base_[cache.slot_], 8, endOkay); if (result == 1) { cache.slot_ = value; } return result; } void decodeActionValue(unsigned char &value, unsigned short &position, ActionCache &cache); void decodeNewXidValue(unsigned int &value, unsigned int &lastId, IntCache &lastIdCache, IntCache &cache, FreeCache &freeCache); void decodeNewXidValue(unsigned int &value, unsigned int &lastId, IntCache &lastIdCache, XidCache &cache, FreeCache &freeCache); void decodeXidValue(unsigned int &value, XidCache &cache); void decodeFreeXidValue(unsigned int &value, FreeCache &cache); void decodeTextData(unsigned char *buffer, unsigned int numBytes) { decodeMemory(buffer, numBytes); } void decodeIntData(unsigned char *buffer, unsigned int numBytes) { decodeMemory(buffer, numBytes); } void decodeLongData(unsigned char *buffer, unsigned int numBytes) { decodeMemory(buffer, numBytes); } const unsigned char *decodeMemory(unsigned int numBytes); void decodeMemory(unsigned char *buffer, unsigned int numBytes) { memcpy(buffer, decodeMemory(numBytes), numBytes); } private: const unsigned char *buffer_; const unsigned char *end_; const unsigned char *nextSrc_; unsigned char srcMask_; }; #endif /* DecodeBuffer_H */