You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
222 lines
8.4 KiB
222 lines
8.4 KiB
4 years ago
|
#if ENABLE_UNSAFE_MSGPACK
|
||
|
|
|
||
|
|
#region license
|
||
|
|
|
||
|
|
/*
|
||
|
|
Copyright (c) 2013, Milosz Krajewski
|
||
|
|
All rights reserved.
|
||
|
|
|
||
|
|
Redistribution and use in source and binary forms, with or without modification, are permitted provided
|
||
|
|
that the following conditions are met:
|
||
|
|
|
||
|
|
* Redistributions of source code must retain the above copyright notice, this list of conditions
|
||
|
|
and the following disclaimer.
|
||
|
|
|
||
|
|
* Redistributions in binary form must reproduce the above copyright notice, this list of conditions
|
||
|
|
and the following disclaimer in the documentation and/or other materials provided with the distribution.
|
||
|
|
|
||
|
|
THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND ANY EXPRESS OR IMPLIED
|
||
|
|
WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
|
||
|
|
A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE
|
||
|
|
FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
|
||
|
|
LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
|
||
|
|
INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
|
||
|
|
OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN
|
||
|
|
IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
||
|
|
*/
|
||
|
|
|
||
|
|
#endregion
|
||
|
|
|
||
|
|
using System;
|
||
|
|
|
||
|
|
namespace MessagePack.LZ4
|
||
|
|
{
|
||
|
|
/// <summary>Unsafe LZ4 codec.</summary>
|
||
|
|
public static partial class LZ4Codec
|
||
|
|
{
|
||
|
|
/// <summary>Copies block of memory.</summary>
|
||
|
|
/// <param name="src">The source.</param>
|
||
|
|
/// <param name="dst">The destination.</param>
|
||
|
|
/// <param name="len">The length (in bytes).</param>
|
||
|
|
private static unsafe void BlockCopy(byte* src, byte* dst, int len)
|
||
|
|
{
|
||
|
|
while (len >= 8)
|
||
|
|
{
|
||
|
|
*(ulong*)dst = *(ulong*)src;
|
||
|
|
dst += 8;
|
||
|
|
src += 8;
|
||
|
|
len -= 8;
|
||
|
|
}
|
||
|
|
if (len >= 4)
|
||
|
|
{
|
||
|
|
*(uint*)dst = *(uint*)src;
|
||
|
|
dst += 4;
|
||
|
|
src += 4;
|
||
|
|
len -= 4;
|
||
|
|
}
|
||
|
|
if (len >= 2)
|
||
|
|
{
|
||
|
|
*(ushort*)dst = *(ushort*)src;
|
||
|
|
dst += 2;
|
||
|
|
src += 2;
|
||
|
|
len -= 2;
|
||
|
|
}
|
||
|
|
if (len >= 1)
|
||
|
|
{
|
||
|
|
*dst = *src; /* d++; s++; l--; */
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
/// <summary>Encodes the specified input.</summary>
|
||
|
|
/// <param name="input">The input.</param>
|
||
|
|
/// <param name="inputOffset">The input offset.</param>
|
||
|
|
/// <param name="inputLength">Length of the input.</param>
|
||
|
|
/// <param name="output">The output.</param>
|
||
|
|
/// <param name="outputOffset">The output offset.</param>
|
||
|
|
/// <param name="outputLength">Length of the output.</param>
|
||
|
|
/// <returns>Number of bytes written.</returns>
|
||
|
|
public static unsafe int Encode32Unsafe(
|
||
|
|
byte[] input,
|
||
|
|
int inputOffset,
|
||
|
|
int inputLength,
|
||
|
|
byte[] output,
|
||
|
|
int outputOffset,
|
||
|
|
int outputLength)
|
||
|
|
{
|
||
|
|
CheckArguments(input, inputOffset, inputLength, output, outputOffset, outputLength);
|
||
|
|
|
||
|
|
if (outputLength == 0) return 0;
|
||
|
|
|
||
|
|
fixed (byte* inputPtr = &input[inputOffset])
|
||
|
|
fixed (byte* outputPtr = &output[outputOffset])
|
||
|
|
{
|
||
|
|
if (inputLength < LZ4_64KLIMIT)
|
||
|
|
{
|
||
|
|
var uHashTable = HashTablePool.GetUShortHashTablePool();
|
||
|
|
fixed (ushort* hash1 = &uHashTable[0])
|
||
|
|
{
|
||
|
|
return LZ4_compress64kCtx_32(hash1, inputPtr, outputPtr, inputLength, outputLength);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
var bHashTable = HashTablePool.GetUIntHashTablePool();
|
||
|
|
fixed (uint* hash2 = &bHashTable[0])
|
||
|
|
{
|
||
|
|
return LZ4_compressCtx_32(hash2, inputPtr, outputPtr, inputLength, outputLength);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
/// <summary>Decodes the specified input.</summary>
|
||
|
|
/// <param name="input">The input.</param>
|
||
|
|
/// <param name="inputOffset">The input offset.</param>
|
||
|
|
/// <param name="inputLength">Length of the input.</param>
|
||
|
|
/// <param name="output">The output.</param>
|
||
|
|
/// <param name="outputOffset">The output offset.</param>
|
||
|
|
/// <param name="outputLength">Length of the output.</param>
|
||
|
|
/// <returns>Number of bytes written.</returns>
|
||
|
|
public static unsafe int Decode32Unsafe(
|
||
|
|
byte[] input,
|
||
|
|
int inputOffset,
|
||
|
|
int inputLength,
|
||
|
|
byte[] output,
|
||
|
|
int outputOffset,
|
||
|
|
int outputLength)
|
||
|
|
{
|
||
|
|
CheckArguments(
|
||
|
|
input, inputOffset, inputLength,
|
||
|
|
output, outputOffset, outputLength);
|
||
|
|
|
||
|
|
if (outputLength == 0) return 0;
|
||
|
|
|
||
|
|
fixed (byte* inputPtr = &input[inputOffset])
|
||
|
|
fixed (byte* outputPtr = &output[outputOffset])
|
||
|
|
{
|
||
|
|
var length = LZ4_uncompress_32(inputPtr, outputPtr, outputLength);
|
||
|
|
if (length != inputLength)
|
||
|
|
throw new ArgumentException("LZ4 block is corrupted, or invalid length has been given.");
|
||
|
|
return outputLength;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
/// <summary>Encodes the specified input.</summary>
|
||
|
|
/// <param name="input">The input.</param>
|
||
|
|
/// <param name="inputOffset">The input offset.</param>
|
||
|
|
/// <param name="inputLength">Length of the input.</param>
|
||
|
|
/// <param name="output">The output.</param>
|
||
|
|
/// <param name="outputOffset">The output offset.</param>
|
||
|
|
/// <param name="outputLength">Length of the output.</param>
|
||
|
|
/// <returns>Number of bytes written.</returns>
|
||
|
|
public static unsafe int Encode64Unsafe(
|
||
|
|
byte[] input,
|
||
|
|
int inputOffset,
|
||
|
|
int inputLength,
|
||
|
|
byte[] output,
|
||
|
|
int outputOffset,
|
||
|
|
int outputLength)
|
||
|
|
{
|
||
|
|
CheckArguments(
|
||
|
|
input, inputOffset, inputLength,
|
||
|
|
output, outputOffset, outputLength);
|
||
|
|
|
||
|
|
if (outputLength == 0) return 0;
|
||
|
|
|
||
|
|
fixed (byte* inputPtr = &input[inputOffset])
|
||
|
|
fixed (byte* outputPtr = &output[outputOffset])
|
||
|
|
{
|
||
|
|
if (inputLength < LZ4_64KLIMIT)
|
||
|
|
{
|
||
|
|
var uHashTable = HashTablePool.GetUShortHashTablePool();
|
||
|
|
fixed (ushort* h1 = &uHashTable[0])
|
||
|
|
{
|
||
|
|
return LZ4_compress64kCtx_64(h1, inputPtr, outputPtr, inputLength, outputLength);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
var uiHashTable = HashTablePool.GetUIntHashTablePool();
|
||
|
|
fixed (uint* h2 = &uiHashTable[0])
|
||
|
|
{
|
||
|
|
return LZ4_compressCtx_64(h2, inputPtr, outputPtr, inputLength, outputLength);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
/// <summary>Decode64s the specified input.</summary>
|
||
|
|
/// <param name="input">The input.</param>
|
||
|
|
/// <param name="inputOffset">The input offset.</param>
|
||
|
|
/// <param name="inputLength">Length of the input.</param>
|
||
|
|
/// <param name="output">The output.</param>
|
||
|
|
/// <param name="outputOffset">The output offset.</param>
|
||
|
|
/// <param name="outputLength">Length of the output.</param>
|
||
|
|
/// <returns>Number of bytes written.</returns>
|
||
|
|
public static unsafe int Decode64Unsafe(
|
||
|
|
byte[] input,
|
||
|
|
int inputOffset,
|
||
|
|
int inputLength,
|
||
|
|
byte[] output,
|
||
|
|
int outputOffset,
|
||
|
|
int outputLength)
|
||
|
|
{
|
||
|
|
CheckArguments(
|
||
|
|
input, inputOffset, inputLength,
|
||
|
|
output, outputOffset, outputLength);
|
||
|
|
|
||
|
|
if (outputLength == 0) return 0;
|
||
|
|
|
||
|
|
fixed (byte* inputPtr = &input[inputOffset])
|
||
|
|
fixed (byte* outputPtr = &output[outputOffset])
|
||
|
|
{
|
||
|
|
var length = LZ4_uncompress_64(inputPtr, outputPtr, outputLength);
|
||
|
|
if (length != inputLength)
|
||
|
|
throw new ArgumentException("LZ4 block is corrupted, or invalid length has been given.");
|
||
|
|
return outputLength;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
#endif
|