/**************************************************************************/ /* */ /* Copyright (c) 2001, 2011 NoMachine (http://www.nomachine.com) */ /* Copyright (c) 2008-2014 Oleksandr Shneyder <o.shneyder@phoca-gmbh.de> */ /* Copyright (c) 2014-2016 Ulrich Sibiller <uli42@gmx.de> */ /* Copyright (c) 2014-2016 Mihai Moldovan <ionic@ionic.de> */ /* Copyright (c) 2011-2016 Mike Gabriel <mike.gabriel@das-netzwerkteam.de>*/ /* Copyright (c) 2015-2016 Qindel Group (http://www.qindel.com) */ /* */ /* NXCOMP, NX protocol compression and NX extensions to this software */ /* are copyright of the aforementioned persons and companies. */ /* */ /* Redistribution and use of the present software is allowed according */ /* to terms specified in the file LICENSE.nxcomp which comes in the */ /* source distribution. */ /* */ /* All rights reserved. */ /* */ /* NOTE: This software has received contributions from various other */ /* contributors, only the core maintainers and supporters are listed as */ /* copyright holders. Please contact us, if you feel you should be listed */ /* as copyright holder, as well. */ /* */ /**************************************************************************/ #include "SetUnpackAlpha.h" #include "ClientCache.h" #include "EncodeBuffer.h" #include "DecodeBuffer.h" #include "WriteBuffer.h" // // Set the verbosity level. // #define PANIC #define WARNING #undef TEST #undef DEBUG #undef DUMP // // Constructors and destructors. // SetUnpackAlphaStore::SetUnpackAlphaStore(StaticCompressor *compressor) : MessageStore(compressor) { enableCache = SETUNPACKALPHA_ENABLE_CACHE; enableData = SETUNPACKALPHA_ENABLE_DATA; enableCompress = SETUNPACKALPHA_ENABLE_COMPRESS_IF_PROTO_STEP_7; dataLimit = SETUNPACKALPHA_DATA_LIMIT; dataOffset = SETUNPACKALPHA_DATA_OFFSET_IF_PROTO_STEP_7; cacheSlots = SETUNPACKALPHA_CACHE_SLOTS; cacheThreshold = SETUNPACKALPHA_CACHE_THRESHOLD; cacheLowerThreshold = SETUNPACKALPHA_CACHE_LOWER_THRESHOLD; // Since ProtoStep8 (#issue 108) enableSplit = SETUNPACKALPHA_ENABLE_SPLIT_IF_PROTO_STEP_8; messages_ -> resize(cacheSlots); for (T_messages::iterator i = messages_ -> begin(); i < messages_ -> end(); i++) { *i = NULL; } temporary_ = NULL; } SetUnpackAlphaStore::~SetUnpackAlphaStore() { for (T_messages::iterator i = messages_ -> begin(); i < messages_ -> end(); i++) { destroy(*i); } destroy(temporary_); } // // Here are the methods to handle messages' content. // int SetUnpackAlphaStore::encodeIdentity(EncodeBuffer &encodeBuffer, const unsigned char *buffer, const unsigned int size, int bigEndian, ChannelCache *channelCache) const { ClientCache *clientCache = (ClientCache *) channelCache; #ifdef DEBUG *logofs << name() << ": Encoding full message identity.\n" << logofs_flush; #endif // // Encode the source length first because // we need it to determine the size of // the output buffer. // // SrcLength. encodeBuffer.encodeValue(GetULONG(buffer + 8, bigEndian), 32, 9); // Client. encodeBuffer.encodeCachedValue(*(buffer + 1), 8, clientCache -> resourceCache); // Method. encodeBuffer.encodeCachedValue(*(buffer + 4), 8, clientCache -> methodCache); // DstLength. encodeBuffer.encodeValue(GetULONG(buffer + 12, bigEndian), 32, 9); #ifdef DEBUG *logofs << name() << ": Encoded full message identity.\n" << logofs_flush; #endif return 1; } int SetUnpackAlphaStore::decodeIdentity(DecodeBuffer &decodeBuffer, unsigned char *&buffer, unsigned int &size, int bigEndian, WriteBuffer *writeBuffer, ChannelCache *channelCache) const { ClientCache *clientCache = (ClientCache *) channelCache; #ifdef DEBUG *logofs << name() << ": Decoding full message identity.\n" << logofs_flush; #endif unsigned int value; unsigned char cValue; // SrcLength. decodeBuffer.decodeValue(value, 32, 9); size = RoundUp4(value) + 16; buffer = writeBuffer -> addMessage(size); PutULONG(value, buffer + 8, bigEndian); // Client. decodeBuffer.decodeCachedValue(cValue, 8, clientCache -> resourceCache); *(buffer + 1) = cValue; // Method. decodeBuffer.decodeCachedValue(cValue, 8, clientCache -> methodCache); *(buffer + 4) = cValue; // DstLength. decodeBuffer.decodeValue(value, 32, 9); PutULONG(value, buffer + 12, bigEndian); #ifdef DEBUG *logofs << name() << ": Decoded full message identity.\n" << logofs_flush; #endif return 1; } int SetUnpackAlphaStore::parseIdentity(Message *message, const unsigned char *buffer, unsigned int size, int bigEndian) const { SetUnpackAlphaMessage *setUnpackAlpha = (SetUnpackAlphaMessage *) message; setUnpackAlpha -> client = *(buffer + 1); setUnpackAlpha -> method = *(buffer + 4); setUnpackAlpha -> src_length = GetULONG(buffer + 8, bigEndian); setUnpackAlpha -> dst_length = GetULONG(buffer + 12, bigEndian); #ifdef DEBUG *logofs << name() << ": Parsed identity for message at " << message << ".\n" << logofs_flush; #endif return 1; } int SetUnpackAlphaStore::unparseIdentity(const Message *message, unsigned char *buffer, unsigned int size, int bigEndian) const { SetUnpackAlphaMessage *setUnpackAlpha = (SetUnpackAlphaMessage *) message; *(buffer + 1) = setUnpackAlpha -> client; *(buffer + 4) = setUnpackAlpha -> method; PutULONG(setUnpackAlpha -> src_length, buffer + 8, bigEndian); PutULONG(setUnpackAlpha -> dst_length, buffer + 12, bigEndian); #ifdef DEBUG *logofs << name() << ": Unparsed identity for message at " << message << ".\n" << logofs_flush; #endif return 1; } void SetUnpackAlphaStore::dumpIdentity(const Message *message) const { #ifdef DUMP SetUnpackAlphaMessage *setUnpackAlpha = (SetUnpackAlphaMessage *) message; *logofs << name() << ": Identity client " << (unsigned int) setUnpackAlpha -> client << " method " << (unsigned int) setUnpackAlpha -> method << " source length " << setUnpackAlpha -> src_length << " destination length " << setUnpackAlpha -> dst_length << " size " << setUnpackAlpha -> size_ << ".\n"; #endif } void SetUnpackAlphaStore::identityChecksum(const Message *message, const unsigned char *buffer, unsigned int size, int bigEndian) const { // // Include the pack method and the source // and destination length. // md5_append(md5_state_, buffer + 4, 1); md5_append(md5_state_, buffer + 8, 8); } void SetUnpackAlphaStore::updateIdentity(EncodeBuffer &encodeBuffer, const Message *message, const Message *cachedMessage, ChannelCache *channelCache) const { SetUnpackAlphaMessage *setUnpackAlpha = (SetUnpackAlphaMessage *) message; SetUnpackAlphaMessage *cachedSetUnpackAlpha = (SetUnpackAlphaMessage *) cachedMessage; ClientCache *clientCache = (ClientCache *) channelCache; encodeBuffer.encodeCachedValue(setUnpackAlpha -> client, 8, clientCache -> resourceCache); cachedSetUnpackAlpha -> client = setUnpackAlpha -> client; } void SetUnpackAlphaStore::updateIdentity(DecodeBuffer &decodeBuffer, const Message *message, ChannelCache *channelCache) const { SetUnpackAlphaMessage *setUnpackAlpha = (SetUnpackAlphaMessage *) message; ClientCache *clientCache = (ClientCache *) channelCache; decodeBuffer.decodeCachedValue(setUnpackAlpha -> client, 8, clientCache -> resourceCache); }