FreeRDP/libfreerdp/codec/ncrush.c

/**
 * FreeRDP: A Remote Desktop Protocol Implementation
 * NCrush (RDP6) Bulk Data Compression
 *
 * Copyright 2014 Marc-Andre Moreau <marcandre.moreau@gmail.com>
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

#ifdef HAVE_CONFIG_H
#include "config.h"
#endif

#include <winpr/crt.h>
#include <winpr/print.h>
#include <winpr/bitstream.h>

#include <freerdp/codec/ncrush.h>

const BYTE NCrushMaskTable[39] =
{
	0x11, /* 0 */
	0x9E, /* 1 */
	0xA1, /* 2 */
	0x00, /* 3 */
	0x00, /* 4 */
	0x01, /* 5 */
	0x00, /* 6 */
	0x03, /* 7 */
	0x00, /* 8 */
	0x07, /* 9 */
	0x00, /* 10 */
	0x0F, /* 11 */
	0x00, /* 12 */
	0x1F, /* 13 */
	0x00, /* 14 */
	0x3F, /* 15 */
	0x00, /* 16 */
	0x7F, /* 17 */
	0x00, /* 18 */
	0xFF, /* 19 */
	0x00, /* 20 */
	0xFF, /* 21 */
	0x01, /* 22 */
	0xFF, /* 23 */
	0x03, /* 24 */
	0xFF, /* 25 */
	0x07, /* 26 */
	0xFF, /* 27 */
	0x0F, /* 28 */
	0xFF, /* 29 */
	0x1F, /* 30 */
	0xFF, /* 31 */
	0x3F, /* 32 */
	0xFF, /* 33 */
	0x7F, /* 34 */
	0xFF, /* 35 */
	0xFF, /* 36 */
	0x00, /* 37 */
	0x00 /* 38 */
};

const BYTE HuffLengthLEC[294] =
{
	6, /* 0 */
	6, /* 1 */
	6, /* 2 */
	7, /* 3 */
	7, /* 4 */
	7, /* 5 */
	7, /* 6 */
	7, /* 7 */
	7, /* 8 */
	7, /* 9 */
	7, /* 10 */
	8, /* 11 */
	8, /* 12 */
	8, /* 13 */
	8, /* 14 */
	8, /* 15 */
	8, /* 16 */
	8, /* 17 */
	9, /* 18 */
	8, /* 19 */
	9, /* 20 */
	9, /* 21 */
	9, /* 22 */
	9, /* 23 */
	8, /* 24 */
	8, /* 25 */
	9, /* 26 */
	9, /* 27 */
	9, /* 28 */
	9, /* 29 */
	9, /* 30 */
	9, /* 31 */
	8, /* 32 */
	9, /* 33 */
	9, /* 34 */
	10, /* 35 */
	9, /* 36 */
	9, /* 37 */
	9, /* 38 */
	9, /* 39 */
	9, /* 40 */
	9, /* 41 */
	9, /* 42 */
	10, /* 43 */
	9, /* 44 */
	10, /* 45 */
	10, /* 46 */
	10, /* 47 */
	9, /* 48 */
	9, /* 49 */
	10, /* 50 */
	9, /* 51 */
	10, /* 52 */
	9, /* 53 */
	10, /* 54 */
	9, /* 55 */
	9, /* 56 */
	9, /* 57 */
	10, /* 58 */
	10, /* 59 */
	9, /* 60 */
	10, /* 61 */
	9, /* 62 */
	9, /* 63 */
	8, /* 64 */
	9, /* 65 */
	9, /* 66 */
	9, /* 67 */
	9, /* 68 */
	10, /* 69 */
	10, /* 70 */
	10, /* 71 */
	9, /* 72 */
	9, /* 73 */
	10, /* 74 */
	10, /* 75 */
	10, /* 76 */
	10, /* 77 */
	10, /* 78 */
	10, /* 79 */
	9, /* 80 */
	9, /* 81 */
	10, /* 82 */
	10, /* 83 */
	10, /* 84 */
	10, /* 85 */
	10, /* 86 */
	10, /* 87 */
	10, /* 88 */
	9, /* 89 */
	10, /* 90 */
	10, /* 91 */
	10, /* 92 */
	10, /* 93 */
	10, /* 94 */
	10, /* 95 */
	8, /* 96 */
	10, /* 97 */
	10, /* 98 */
	10, /* 99 */
	10, /* 100 */
	10, /* 101 */
	10, /* 102 */
	10, /* 103 */
	10, /* 104 */
	10, /* 105 */
	10, /* 106 */
	10, /* 107 */
	10, /* 108 */
	10, /* 109 */
	10, /* 110 */
	10, /* 111 */
	9, /* 112 */
	10, /* 113 */
	10, /* 114 */
	10, /* 115 */
	10, /* 116 */
	10, /* 117 */
	10, /* 118 */
	10, /* 119 */
	9, /* 120 */
	10, /* 121 */
	10, /* 122 */
	10, /* 123 */
	10, /* 124 */
	10, /* 125 */
	10, /* 126 */
	9, /* 127 */
	7, /* 128 */
	9, /* 129 */
	9, /* 130 */
	10, /* 131 */
	9, /* 132 */
	10, /* 133 */
	10, /* 134 */
	10, /* 135 */
	9, /* 136 */
	10, /* 137 */
	10, /* 138 */
	10, /* 139 */
	10, /* 140 */
	10, /* 141 */
	10, /* 142 */
	10, /* 143 */
	9, /* 144 */
	10, /* 145 */
	10, /* 146 */
	10, /* 147 */
	10, /* 148 */
	10, /* 149 */
	10, /* 150 */
	10, /* 151 */
	10, /* 152 */
	10, /* 153 */
	10, /* 154 */
	10, /* 155 */
	10, /* 156 */
	10, /* 157 */
	10, /* 158 */
	10, /* 159 */
	10, /* 160 */
	10, /* 161 */
	10, /* 162 */
	10, /* 163 */
	10, /* 164 */
	10, /* 165 */
	10, /* 166 */
	10, /* 167 */
	10, /* 168 */
	10, /* 169 */
	10, /* 170 */
	13, /* 171 */
	10, /* 172 */
	10, /* 173 */
	10, /* 174 */
	10, /* 175 */
	10, /* 176 */
	10, /* 177 */
	11, /* 178 */
	10, /* 179 */
	10, /* 180 */
	10, /* 181 */
	10, /* 182 */
	10, /* 183 */
	10, /* 184 */
	10, /* 185 */
	10, /* 186 */
	10, /* 187 */
	10, /* 188 */
	10, /* 189 */
	10, /* 190 */
	10, /* 191 */
	9, /* 192 */
	10, /* 193 */
	10, /* 194 */
	10, /* 195 */
	10, /* 196 */
	10, /* 197 */
	9, /* 198 */
	10, /* 199 */
	10, /* 200 */
	10, /* 201 */
	10, /* 202 */
	10, /* 203 */
	9, /* 204 */
	10, /* 205 */
	10, /* 206 */
	10, /* 207 */
	9, /* 208 */
	10, /* 209 */
	10, /* 210 */
	10, /* 211 */
	10, /* 212 */
	10, /* 213 */
	10, /* 214 */
	10, /* 215 */
	10, /* 216 */
	10, /* 217 */
	10, /* 218 */
	10, /* 219 */
	10, /* 220 */
	10, /* 221 */
	10, /* 222 */
	10, /* 223 */
	9, /* 224 */
	10, /* 225 */
	10, /* 226 */
	10, /* 227 */
	10, /* 228 */
	10, /* 229 */
	10, /* 230 */
	10, /* 231 */
	10, /* 232 */
	10, /* 233 */
	10, /* 234 */
	10, /* 235 */
	10, /* 236 */
	10, /* 237 */
	9, /* 238 */
	10, /* 239 */
	8, /* 240 */
	9, /* 241 */
	9, /* 242 */
	10, /* 243 */
	9, /* 244 */
	10, /* 245 */
	10, /* 246 */
	10, /* 247 */
	9, /* 248 */
	10, /* 249 */
	10, /* 250 */
	10, /* 251 */
	9, /* 252 */
	9, /* 253 */
	8, /* 254 */
	7, /* 255 */
	13, /* 256 */
	13, /* 257 */
	7, /* 258 */
	7, /* 259 */
	10, /* 260 */
	7, /* 261 */
	7, /* 262 */
	6, /* 263 */
	6, /* 264 */
	6, /* 265 */
	6, /* 266 */
	5, /* 267 */
	6, /* 268 */
	6, /* 269 */
	6, /* 270 */
	5, /* 271 */
	6, /* 272 */
	5, /* 273 */
	6, /* 274 */
	6, /* 275 */
	6, /* 276 */
	6, /* 277 */
	6, /* 278 */
	6, /* 279 */
	6, /* 280 */
	6, /* 281 */
	6, /* 282 */
	6, /* 283 */
	6, /* 284 */
	6, /* 285 */
	6, /* 286 */
	6, /* 287 */
	8, /* 288 */
	5, /* 289 */
	6, /* 290 */
	7, /* 291 */
	7, /* 292 */
	13 /* 293 */
};

const BYTE HuffLengthL[32] =
{
	4, /* 0 */
	2, /* 1 */
	3, /* 2 */
	4, /* 3 */
	3, /* 4 */
	4, /* 5 */
	4, /* 6 */
	5, /* 7 */
	4, /* 8 */
	5, /* 9 */
	5, /* 10 */
	6, /* 11 */
	6, /* 12 */
	7, /* 13 */
	7, /* 14 */
	8, /* 15 */
	7, /* 16 */
	8, /* 17 */
	8, /* 18 */
	9, /* 19 */
	9, /* 20 */
	8, /* 21 */
	9, /* 22 */
	9, /* 23 */
	9, /* 24 */
	9, /* 25 */
	9, /* 26 */
	9, /* 27 */
	9, /* 28 */
	9, /* 29 */
	9, /* 30 */
	9 /* 31 */
};

int ncrush_decompress(NCRUSH_CONTEXT* ncrush, BYTE* pSrcData, UINT32 SrcSize, BYTE** ppDstData, UINT32* pDstSize, UINT32 flags)
{
	UINT32 bits;
	UINT32 nbits;
	BYTE* SrcPtr;
	UINT32 CopyOffset;
	UINT32 LengthOfMatch;
	UINT32 accumulator;
	BYTE* HistoryPtr;
	UINT32 HistoryOffset;
	BYTE* HistoryBuffer;
	BYTE* HistoryBufferEnd;
	UINT32 HistoryBufferSize;
	wBitStream* bs = ncrush->bs;

	HistoryBuffer = ncrush->HistoryBuffer;
	HistoryBufferSize = ncrush->HistoryBufferSize;
	HistoryBufferEnd = &HistoryBuffer[HistoryBufferSize - 1];

	BitStream_Attach(bs, pSrcData, SrcSize);
	BitStream_Fetch(bs);

	if (flags & PACKET_AT_FRONT)
	{
		if ((ncrush->HistoryPtr - 32768) <= HistoryBuffer)
			return -1;

		CopyMemory(HistoryBuffer, (ncrush->HistoryPtr - 32768), 32768);
		ncrush->HistoryOffset = 32768;
		ncrush->HistoryPtr = &(HistoryBuffer[ncrush->HistoryOffset]);
	}

	if (flags & PACKET_FLUSHED)
	{
		ncrush->HistoryOffset = 0;
		ncrush->HistoryPtr = HistoryBuffer;
		ZeroMemory(HistoryBuffer, ncrush->HistoryBufferSize);
		ZeroMemory(&(ncrush->OffsetCache), sizeof(ncrush->OffsetCache));
	}

	HistoryPtr = ncrush->HistoryPtr;
	HistoryOffset = ncrush->HistoryOffset;

	if (!(flags & PACKET_COMPRESSED))
	{
		CopyMemory(HistoryPtr, pSrcData, SrcSize);
		HistoryPtr += SrcSize;
		HistoryOffset += SrcSize;
		ncrush->HistoryPtr = HistoryPtr;
		ncrush->HistoryOffset = HistoryOffset;
		*ppDstData = HistoryPtr;
		*pDstSize = SrcSize;
		return 1;
	}

	nbits = 32;
	bits = *((UINT32*) pSrcData);
	SrcPtr = pSrcData + 4;

	accumulator = bits & *((UINT16*) &NCrushMaskTable[29]);

	printf("MaskedBits: 0x%04X Mask: 0x%04X\n", accumulator, *((UINT16*) &NCrushMaskTable[29]));

	return 1;
}

int ncrush_compress(NCRUSH_CONTEXT* ncrush, BYTE* pSrcData, UINT32 SrcSize, BYTE* pDstData, UINT32* pDstSize, UINT32* pFlags)
{
	return 1;
}

NCRUSH_CONTEXT* ncrush_context_new(BOOL Compressor)
{
	NCRUSH_CONTEXT* ncrush;

	ncrush = (NCRUSH_CONTEXT*) calloc(1, sizeof(NCRUSH_CONTEXT));

	if (ncrush)
	{
		ncrush->Compressor = Compressor;

		ncrush->bs = BitStream_New();

		ncrush->HistoryBufferSize = 65536;
		ZeroMemory(&(ncrush->HistoryBuffer), sizeof(ncrush->HistoryBuffer));
		ZeroMemory(&(ncrush->OffsetCache), sizeof(ncrush->OffsetCache));

		ncrush->HistoryOffset = 0;
		ncrush->HistoryPtr = &(ncrush->HistoryBuffer[ncrush->HistoryOffset]);
	}

	return ncrush;
}

void ncrush_context_free(NCRUSH_CONTEXT* ncrush)
{
	if (ncrush)
	{
		BitStream_Free(ncrush->bs);

		free(ncrush);
	}
}