12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196 |
- /*
- * Copyright (c)2013-2020 ZeroTier, Inc.
- *
- * Use of this software is governed by the Business Source License included
- * in the LICENSE.TXT file in the project's root directory.
- *
- * Change Date: 2026-01-01
- *
- * On the date above, in accordance with the Business Source License, use
- * of this software will be governed by version 2.0 of the Apache License.
- */
- /****/
- #include <stdint.h>
- #include <stddef.h>
- #include <string.h>
- #include <stdlib.h>
- #include <stdio.h>
- #include "Packet.hpp"
- #if defined(ZT_USE_X64_ASM_SALSA2012) && defined(ZT_ARCH_X64)
- #include "../ext/x64-salsa2012-asm/salsa2012.h"
- #endif
- #ifdef ZT_USE_ARM32_NEON_ASM_SALSA2012
- #include "../ext/arm32-neon-salsa2012-asm/salsa2012.h"
- #endif
- #ifdef _MSC_VER
- #define FORCE_INLINE static __forceinline
- #include <intrin.h>
- #pragma warning(disable : 4127) /* disable: C4127: conditional expression is constant */
- #pragma warning(disable : 4293) /* disable: C4293: too large shift (32-bits) */
- #else
- #define FORCE_INLINE static inline
- #endif
- namespace ZeroTier {
- /************************************************************************** */
- /* Set up macros for fast single-pass ASM Salsa20/12 crypto, if we have it */
- // x64 SSE crypto
- #if defined(ZT_USE_X64_ASM_SALSA2012) && defined(ZT_ARCH_X64)
- #define ZT_HAS_FAST_CRYPTO() (true)
- #define ZT_FAST_SINGLE_PASS_SALSA2012(b,l,n,k) zt_salsa2012_amd64_xmm6(reinterpret_cast<unsigned char *>(b),(l),reinterpret_cast<const unsigned char *>(n),reinterpret_cast<const unsigned char *>(k))
- #endif
- // ARM (32-bit) NEON crypto (must be detected)
- #ifdef ZT_USE_ARM32_NEON_ASM_SALSA2012
- class _FastCryptoChecker
- {
- public:
- _FastCryptoChecker() : canHas(zt_arm_has_neon()) {}
- bool canHas;
- };
- static const _FastCryptoChecker _ZT_FAST_CRYPTO_CHECK;
- #define ZT_HAS_FAST_CRYPTO() (_ZT_FAST_CRYPTO_CHECK.canHas)
- #define ZT_FAST_SINGLE_PASS_SALSA2012(b,l,n,k) zt_salsa2012_armneon3_xor(reinterpret_cast<unsigned char *>(b),(const unsigned char *)0,(l),reinterpret_cast<const unsigned char *>(n),reinterpret_cast<const unsigned char *>(k))
- #endif
- // No fast crypto available
- #ifndef ZT_HAS_FAST_CRYPTO
- #define ZT_HAS_FAST_CRYPTO() (false)
- #define ZT_FAST_SINGLE_PASS_SALSA2012(b,l,n,k) {}
- #endif
- /************************************************************************** */
- /* LZ4 is shipped encapsulated into Packet in an anonymous namespace.
- *
- * We're doing this as a deliberate workaround for various Linux distribution
- * policies that forbid static linking of support libraries.
- *
- * The reason is that relying on distribution versions of LZ4 has been too
- * big a source of bugs and compatibility issues. The LZ4 API is not stable
- * enough across versions, and dependency hell ensues. So fark it. */
- /* Needless to say the code in this anonymous namespace should be considered
- * BSD 2-clause licensed. */
- namespace {
- /* lz4.h ------------------------------------------------------------------ */
- /*
- * LZ4 - Fast LZ compression algorithm
- * Header File
- * Copyright (C) 2011-2016, Yann Collet.
- BSD 2-Clause License (http://www.opensource.org/licenses/bsd-license.php)
- 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
- OWNER 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.
- You can contact the author at :
- - LZ4 homepage : http://www.lz4.org
- - LZ4 source repository : https://github.com/lz4/lz4
- */
- /**
- Introduction
- LZ4 is lossless compression algorithm, providing compression speed at 400 MB/s per core,
- scalable with multi-cores CPU. It features an extremely fast decoder, with speed in
- multiple GB/s per core, typically reaching RAM speed limits on multi-core systems.
- The LZ4 compression library provides in-memory compression and decompression functions.
- Compression can be done in:
- - a single step (described as Simple Functions)
- - a single step, reusing a context (described in Advanced Functions)
- - unbounded multiple steps (described as Streaming compression)
- lz4.h provides block compression functions. It gives full buffer control to user.
- Decompressing an lz4-compressed block also requires metadata (such as compressed size).
- Each application is free to encode such metadata in whichever way it wants.
- An additional format, called LZ4 frame specification (doc/lz4_Frame_format.md),
- take care of encoding standard metadata alongside LZ4-compressed blocks.
- If your application requires interoperability, it's recommended to use it.
- A library is provided to take care of it, see lz4frame.h.
- */
- #define LZ4_VERSION_MAJOR 1 /* for breaking interface changes */
- #define LZ4_VERSION_MINOR 7 /* for new (non-breaking) interface capabilities */
- #define LZ4_VERSION_RELEASE 5 /* for tweaks, bug-fixes, or development */
- #define LZ4_VERSION_NUMBER (LZ4_VERSION_MAJOR *100*100 + LZ4_VERSION_MINOR *100 + LZ4_VERSION_RELEASE)
- #define LZ4_LIB_VERSION LZ4_VERSION_MAJOR.LZ4_VERSION_MINOR.LZ4_VERSION_RELEASE
- #define LZ4_QUOTE(str) #str
- #define LZ4_EXPAND_AND_QUOTE(str) LZ4_QUOTE(str)
- #define LZ4_VERSION_STRING LZ4_EXPAND_AND_QUOTE(LZ4_LIB_VERSION)
- #define LZ4_MEMORY_USAGE 14
- #define LZ4_MAX_INPUT_SIZE 0x7E000000 /* 2 113 929 216 bytes */
- #define LZ4_COMPRESSBOUND(isize) ((unsigned)(isize) > (unsigned)LZ4_MAX_INPUT_SIZE ? 0 : (isize) + ((isize)/255) + 16)
- typedef union LZ4_stream_u LZ4_stream_t; /* incomplete type (defined later) */
- static inline void LZ4_resetStream (LZ4_stream_t* streamPtr);
- #define LZ4_HASHLOG (LZ4_MEMORY_USAGE-2)
- #define LZ4_HASHTABLESIZE (1 << LZ4_MEMORY_USAGE)
- #define LZ4_HASH_SIZE_U32 (1 << LZ4_HASHLOG) /* required as macro for static allocation */
- typedef struct {
- uint32_t hashTable[LZ4_HASH_SIZE_U32];
- uint32_t currentOffset;
- uint32_t initCheck;
- const uint8_t* dictionary;
- uint8_t* bufferStart; /* obsolete, used for slideInputBuffer */
- uint32_t dictSize;
- } LZ4_stream_t_internal;
- typedef struct {
- const uint8_t* externalDict;
- size_t extDictSize;
- const uint8_t* prefixEnd;
- size_t prefixSize;
- } LZ4_streamDecode_t_internal;
- #define LZ4_STREAMSIZE_U64 ((1 << (LZ4_MEMORY_USAGE-3)) + 4)
- #define LZ4_STREAMSIZE (LZ4_STREAMSIZE_U64 * sizeof(unsigned long long))
- union LZ4_stream_u {
- unsigned long long table[LZ4_STREAMSIZE_U64];
- LZ4_stream_t_internal internal_donotuse;
- } ; /* previously typedef'd to LZ4_stream_t */
- #define LZ4_STREAMDECODESIZE_U64 4
- #define LZ4_STREAMDECODESIZE (LZ4_STREAMDECODESIZE_U64 * sizeof(unsigned long long))
- union LZ4_streamDecode_u {
- unsigned long long table[LZ4_STREAMDECODESIZE_U64];
- LZ4_streamDecode_t_internal internal_donotuse;
- } ; /* previously typedef'd to LZ4_streamDecode_t */
- #ifndef HEAPMODE
- #define HEAPMODE 0
- #endif
- #ifdef ZT_NO_TYPE_PUNNING
- #define LZ4_FORCE_MEMORY_ACCESS 0
- #else
- #define LZ4_FORCE_MEMORY_ACCESS 2
- #endif
- #if defined(_MSC_VER) && defined(_WIN32_WCE) /* Visual Studio for Windows CE does not support Hardware bit count */
- #define LZ4_FORCE_SW_BITCOUNT
- #endif
- #ifndef FORCE_INLINE
- #define FORCE_INLINE static inline
- #endif
- #define ALLOCATOR(n,s) calloc(n,s)
- #define FREEMEM free
- #define MEM_INIT memset
- typedef uint8_t BYTE;
- typedef uint16_t U16;
- typedef uint32_t U32;
- typedef int32_t S32;
- typedef uint64_t U64;
- typedef uintptr_t uptrval;
- typedef uintptr_t reg_t;
- static inline unsigned LZ4_isLittleEndian(void)
- {
- const union { U32 u; BYTE c[4]; } one = { 1 }; /* don't use static : performance detrimental */
- return one.c[0];
- }
- #if defined(LZ4_FORCE_MEMORY_ACCESS) && (LZ4_FORCE_MEMORY_ACCESS==2)
- static U16 LZ4_read16(const void* memPtr) { return *(const U16*) memPtr; }
- static U32 LZ4_read32(const void* memPtr) { return *(const U32*) memPtr; }
- static reg_t LZ4_read_ARCH(const void* memPtr) { return *(const reg_t*) memPtr; }
- static void LZ4_write16(void* memPtr, U16 value) { *(U16*)memPtr = value; }
- static void LZ4_write32(void* memPtr, U32 value) { *(U32*)memPtr = value; }
- #elif defined(LZ4_FORCE_MEMORY_ACCESS) && (LZ4_FORCE_MEMORY_ACCESS==1)
- typedef union { U16 u16; U32 u32; reg_t uArch; } __attribute__((packed)) unalign;
- static U16 LZ4_read16(const void* ptr) { return ((const unalign*)ptr)->u16; }
- static U32 LZ4_read32(const void* ptr) { return ((const unalign*)ptr)->u32; }
- static reg_t LZ4_read_ARCH(const void* ptr) { return ((const unalign*)ptr)->uArch; }
- static void LZ4_write16(void* memPtr, U16 value) { ((unalign*)memPtr)->u16 = value; }
- static void LZ4_write32(void* memPtr, U32 value) { ((unalign*)memPtr)->u32 = value; }
- #else /* safe and portable access through memcpy() */
- static inline U16 LZ4_read16(const void* memPtr)
- {
- U16 val; memcpy(&val, memPtr, sizeof(val)); return val;
- }
- static inline U32 LZ4_read32(const void* memPtr)
- {
- U32 val; memcpy(&val, memPtr, sizeof(val)); return val;
- }
- static inline reg_t LZ4_read_ARCH(const void* memPtr)
- {
- reg_t val; memcpy(&val, memPtr, sizeof(val)); return val;
- }
- static inline void LZ4_write16(void* memPtr, U16 value)
- {
- memcpy(memPtr, &value, sizeof(value));
- }
- static inline void LZ4_write32(void* memPtr, U32 value)
- {
- memcpy(memPtr, &value, sizeof(value));
- }
- #endif /* LZ4_FORCE_MEMORY_ACCESS */
- static inline U16 LZ4_readLE16(const void* memPtr)
- {
- if (LZ4_isLittleEndian()) {
- return LZ4_read16(memPtr);
- } else {
- const BYTE* p = (const BYTE*)memPtr;
- return (U16)((U16)p[0] + (p[1]<<8));
- }
- }
- static inline void LZ4_writeLE16(void* memPtr, U16 value)
- {
- if (LZ4_isLittleEndian()) {
- LZ4_write16(memPtr, value);
- } else {
- BYTE* p = (BYTE*)memPtr;
- p[0] = (BYTE) value;
- p[1] = (BYTE)(value>>8);
- }
- }
- static inline void LZ4_copy8(void* dst, const void* src)
- {
- memcpy(dst,src,8);
- }
- static inline void LZ4_wildCopy(void* dstPtr, const void* srcPtr, void* dstEnd)
- {
- BYTE* d = (BYTE*)dstPtr;
- const BYTE* s = (const BYTE*)srcPtr;
- BYTE* const e = (BYTE*)dstEnd;
- do {
- LZ4_copy8(d,s);
- d+=8;
- s+=8;
- } while (d<e);
- }
- #define MINMATCH 4
- #define WILDCOPYLENGTH 8
- #define LASTLITERALS 5
- #define MFLIMIT (WILDCOPYLENGTH+MINMATCH)
- static const int LZ4_minLength = (MFLIMIT+1);
- #define KB *(1 <<10)
- #define MB *(1 <<20)
- #define GB *(1U<<30)
- #define MAXD_LOG 16
- #define MAX_DISTANCE ((1 << MAXD_LOG) - 1)
- #define ML_BITS 4
- #define ML_MASK ((1U<<ML_BITS)-1)
- #define RUN_BITS (8-ML_BITS)
- #define RUN_MASK ((1U<<RUN_BITS)-1)
- #define LZ4_STATIC_ASSERT(c) { enum { LZ4_static_assert = 1/(int)(!!(c)) }; } /* use only *after* variable declarations */
- static inline unsigned LZ4_NbCommonBytes (reg_t val)
- {
- if (LZ4_isLittleEndian()) {
- if (sizeof(val)==8) {
- # if defined(_MSC_VER) && defined(_WIN64) && !defined(LZ4_FORCE_SW_BITCOUNT)
- unsigned long r = 0;
- _BitScanForward64( &r, (U64)val );
- return (int)(r>>3);
- # elif (defined(__clang__) || (defined(__GNUC__) && (__GNUC__>=3))) && !defined(LZ4_FORCE_SW_BITCOUNT)
- return (__builtin_ctzll((U64)val) >> 3);
- # else
- static const int DeBruijnBytePos[64] = { 0, 0, 0, 0, 0, 1, 1, 2, 0, 3, 1, 3, 1, 4, 2, 7, 0, 2, 3, 6, 1, 5, 3, 5, 1, 3, 4, 4, 2, 5, 6, 7, 7, 0, 1, 2, 3, 3, 4, 6, 2, 6, 5, 5, 3, 4, 5, 6, 7, 1, 2, 4, 6, 4, 4, 5, 7, 2, 6, 5, 7, 6, 7, 7 };
- return DeBruijnBytePos[((U64)((val & -(long long)val) * 0x0218A392CDABBD3FULL)) >> 58];
- # endif
- } else /* 32 bits */ {
- # if defined(_MSC_VER) && !defined(LZ4_FORCE_SW_BITCOUNT)
- unsigned long r;
- _BitScanForward( &r, (U32)val );
- return (int)(r>>3);
- # elif (defined(__clang__) || (defined(__GNUC__) && (__GNUC__>=3))) && !defined(LZ4_FORCE_SW_BITCOUNT)
- return (__builtin_ctz((U32)val) >> 3);
- # else
- static const int DeBruijnBytePos[32] = { 0, 0, 3, 0, 3, 1, 3, 0, 3, 2, 2, 1, 3, 2, 0, 1, 3, 3, 1, 2, 2, 2, 2, 0, 3, 1, 2, 0, 1, 0, 1, 1 };
- return DeBruijnBytePos[((U32)((val & -(S32)val) * 0x077CB531U)) >> 27];
- # endif
- }
- } else /* Big Endian CPU */ {
- if (sizeof(val)==8) {
- # if defined(_MSC_VER) && defined(_WIN64) && !defined(LZ4_FORCE_SW_BITCOUNT)
- unsigned long r = 0;
- _BitScanReverse64( &r, val );
- return (unsigned)(r>>3);
- # elif (defined(__clang__) || (defined(__GNUC__) && (__GNUC__>=3))) && !defined(LZ4_FORCE_SW_BITCOUNT)
- return (__builtin_clzll((U64)val) >> 3);
- # else
- unsigned r;
- if (!(val>>32)) {
- r=4;
- } else {
- r=0;
- val>>=32;
- }
- if (!(val>>16)) {
- r+=2;
- val>>=8;
- } else {
- val>>=24;
- }
- r += (!val);
- return r;
- # endif
- } else /* 32 bits */ {
- # if defined(_MSC_VER) && !defined(LZ4_FORCE_SW_BITCOUNT)
- unsigned long r = 0;
- _BitScanReverse( &r, (unsigned long)val );
- return (unsigned)(r>>3);
- # elif (defined(__clang__) || (defined(__GNUC__) && (__GNUC__>=3))) && !defined(LZ4_FORCE_SW_BITCOUNT)
- return (__builtin_clz((U32)val) >> 3);
- # else
- unsigned r;
- if (!(val>>16)) {
- r=2;
- val>>=8;
- } else {
- r=0;
- val>>=24;
- }
- r += (!val);
- return r;
- # endif
- }
- }
- }
- #define STEPSIZE sizeof(reg_t)
- static inline unsigned LZ4_count(const BYTE* pIn, const BYTE* pMatch, const BYTE* pInLimit)
- {
- const BYTE* const pStart = pIn;
- while (likely(pIn<pInLimit-(STEPSIZE-1))) {
- reg_t const diff = LZ4_read_ARCH(pMatch) ^ LZ4_read_ARCH(pIn);
- if (!diff) {
- pIn+=STEPSIZE;
- pMatch+=STEPSIZE; continue;
- }
- pIn += LZ4_NbCommonBytes(diff);
- return (unsigned)(pIn - pStart);
- }
- if ((STEPSIZE==8) && (pIn<(pInLimit-3)) && (LZ4_read32(pMatch) == LZ4_read32(pIn))) {
- pIn+=4; pMatch+=4;
- }
- if ((pIn<(pInLimit-1)) && (LZ4_read16(pMatch) == LZ4_read16(pIn))) {
- pIn+=2; pMatch+=2;
- }
- if ((pIn<pInLimit) && (*pMatch == *pIn)) {
- pIn++;
- }
- return (unsigned)(pIn - pStart);
- }
- static const int LZ4_64Klimit = ((64 KB) + (MFLIMIT-1));
- static const U32 LZ4_skipTrigger = 6; /* Increase this value ==> compression run slower on incompressible data */
- typedef enum { notLimited = 0, limitedOutput = 1 } limitedOutput_directive;
- typedef enum { byPtr, byU32, byU16 } tableType_t;
- typedef enum { noDict = 0, withPrefix64k, usingExtDict } dict_directive;
- typedef enum { noDictIssue = 0, dictSmall } dictIssue_directive;
- typedef enum { endOnOutputSize = 0, endOnInputSize = 1 } endCondition_directive;
- typedef enum { full = 0, partial = 1 } earlyEnd_directive;
- static inline int LZ4_compressBound(int isize) { return LZ4_COMPRESSBOUND(isize); }
- static inline U32 LZ4_hash4(U32 sequence, tableType_t const tableType)
- {
- if (tableType == byU16) {
- return ((sequence * 2654435761U) >> ((MINMATCH*8)-(LZ4_HASHLOG+1)));
- } else {
- return ((sequence * 2654435761U) >> ((MINMATCH*8)-LZ4_HASHLOG));
- }
- }
- static inline U32 LZ4_hash5(U64 sequence, tableType_t const tableType)
- {
- static const U64 prime5bytes = 889523592379ULL;
- static const U64 prime8bytes = 11400714785074694791ULL;
- const U32 hashLog = (tableType == byU16) ? LZ4_HASHLOG+1 : LZ4_HASHLOG;
- if (LZ4_isLittleEndian()) {
- return (U32)(((sequence << 24) * prime5bytes) >> (64 - hashLog));
- } else {
- return (U32)(((sequence >> 24) * prime8bytes) >> (64 - hashLog));
- }
- }
- FORCE_INLINE U32 LZ4_hashPosition(const void* const p, tableType_t const tableType)
- {
- if ((sizeof(reg_t)==8) && (tableType != byU16)) {
- return LZ4_hash5(LZ4_read_ARCH(p), tableType);
- }
- return LZ4_hash4(LZ4_read32(p), tableType);
- }
- static inline void LZ4_putPositionOnHash(const BYTE* p, U32 h, void* tableBase, tableType_t const tableType, const BYTE* srcBase)
- {
- switch (tableType) {
- case byPtr: {
- const BYTE** hashTable = (const BYTE**)tableBase; hashTable[h] = p;
- return;
- }
- case byU32: {
- U32* hashTable = (U32*) tableBase; hashTable[h] = (U32)(p-srcBase);
- return;
- }
- case byU16: {
- U16* hashTable = (U16*) tableBase; hashTable[h] = (U16)(p-srcBase);
- return;
- }
- }
- }
- FORCE_INLINE void LZ4_putPosition(const BYTE* p, void* tableBase, tableType_t tableType, const BYTE* srcBase)
- {
- U32 const h = LZ4_hashPosition(p, tableType);
- LZ4_putPositionOnHash(p, h, tableBase, tableType, srcBase);
- }
- static inline const BYTE* LZ4_getPositionOnHash(U32 h, void* tableBase, tableType_t tableType, const BYTE* srcBase)
- {
- if (tableType == byPtr) {
- const BYTE** hashTable = (const BYTE**) tableBase;
- return hashTable[h];
- }
- if (tableType == byU32) {
- const U32* const hashTable = (U32*) tableBase;
- return hashTable[h] + srcBase;
- }
- { /* default, to ensure a return */
- const U16* const hashTable = (U16*) tableBase;
- return hashTable[h] + srcBase;
- }
- }
- FORCE_INLINE const BYTE* LZ4_getPosition(const BYTE* p, void* tableBase, tableType_t tableType, const BYTE* srcBase)
- {
- U32 const h = LZ4_hashPosition(p, tableType);
- return LZ4_getPositionOnHash(h, tableBase, tableType, srcBase);
- }
- FORCE_INLINE int LZ4_compress_generic(
- LZ4_stream_t_internal* const cctx,
- const char* const source,
- char* const dest,
- const int inputSize,
- const int maxOutputSize,
- const limitedOutput_directive outputLimited,
- const tableType_t tableType,
- const dict_directive dict,
- const dictIssue_directive dictIssue,
- const U32 acceleration)
- {
- const BYTE* ip = (const BYTE*) source;
- const BYTE* base;
- const BYTE* lowLimit;
- const BYTE* const lowRefLimit = ip - cctx->dictSize;
- const BYTE* const dictionary = cctx->dictionary;
- const BYTE* const dictEnd = dictionary + cctx->dictSize;
- const ptrdiff_t dictDelta = dictEnd - (const BYTE*)source;
- const BYTE* anchor = (const BYTE*) source;
- const BYTE* const iend = ip + inputSize;
- const BYTE* const mflimit = iend - MFLIMIT;
- const BYTE* const matchlimit = iend - LASTLITERALS;
- BYTE* op = (BYTE*) dest;
- BYTE* const olimit = op + maxOutputSize;
- U32 forwardH;
- /* Init conditions */
- if ((U32)inputSize > (U32)LZ4_MAX_INPUT_SIZE) {
- return 0; /* Unsupported inputSize, too large (or negative) */
- }
- switch(dict) {
- case noDict:
- default:
- base = (const BYTE*)source;
- lowLimit = (const BYTE*)source;
- break;
- case withPrefix64k:
- base = (const BYTE*)source - cctx->currentOffset;
- lowLimit = (const BYTE*)source - cctx->dictSize;
- break;
- case usingExtDict:
- base = (const BYTE*)source - cctx->currentOffset;
- lowLimit = (const BYTE*)source;
- break;
- }
- if ((tableType == byU16) && (inputSize>=LZ4_64Klimit)) {
- return 0; /* Size too large (not within 64K limit) */
- }
- if (inputSize<LZ4_minLength) {
- goto _last_literals; /* Input too small, no compression (all literals) */
- }
- /* First Byte */
- LZ4_putPosition(ip, cctx->hashTable, tableType, base);
- ip++;
- forwardH = LZ4_hashPosition(ip, tableType);
- /* Main Loop */
- for ( ; ; ) {
- ptrdiff_t refDelta = 0;
- const BYTE* match;
- BYTE* token;
- /* Find a match */
- {
- const BYTE* forwardIp = ip;
- unsigned step = 1;
- unsigned searchMatchNb = acceleration << LZ4_skipTrigger;
- do {
- U32 const h = forwardH;
- ip = forwardIp;
- forwardIp += step;
- step = (searchMatchNb++ >> LZ4_skipTrigger);
- if (unlikely(forwardIp > mflimit)) {
- goto _last_literals;
- }
- match = LZ4_getPositionOnHash(h, cctx->hashTable, tableType, base);
- if (dict==usingExtDict) {
- if (match < (const BYTE*)source) {
- refDelta = dictDelta;
- lowLimit = dictionary;
- } else {
- refDelta = 0;
- lowLimit = (const BYTE*)source;
- }
- }
- forwardH = LZ4_hashPosition(forwardIp, tableType);
- LZ4_putPositionOnHash(ip, h, cctx->hashTable, tableType, base);
- } while ( ((dictIssue==dictSmall) ? (match < lowRefLimit) : 0)
- || ((tableType==byU16) ? 0 : (match + MAX_DISTANCE < ip))
- || (LZ4_read32(match+refDelta) != LZ4_read32(ip)) );
- }
- /* Catch up */
- while (((ip>anchor) & (match+refDelta > lowLimit)) && (unlikely(ip[-1]==match[refDelta-1]))) {
- ip--;
- match--;
- }
- /* Encode Literals */
- {
- unsigned const litLength = (unsigned)(ip - anchor);
- token = op++;
- if ((outputLimited) && /* Check output buffer overflow */
- (unlikely(op + litLength + (2 + 1 + LASTLITERALS) + (litLength/255) > olimit))) {
- return 0;
- }
- if (litLength >= RUN_MASK) {
- int len = (int)litLength-RUN_MASK;
- *token = (RUN_MASK<<ML_BITS);
- for(; len >= 255 ; len-=255) {
- *op++ = 255;
- }
- *op++ = (BYTE)len;
- } else {
- *token = (BYTE)(litLength<<ML_BITS);
- }
- /* Copy Literals */
- LZ4_wildCopy(op, anchor, op+litLength);
- op+=litLength;
- }
- _next_match:
- /* Encode Offset */
- LZ4_writeLE16(op, (U16)(ip-match));
- op+=2;
- /* Encode MatchLength */
- {
- unsigned matchCode;
- if ((dict==usingExtDict) && (lowLimit==dictionary)) {
- const BYTE* limit;
- match += refDelta;
- limit = ip + (dictEnd-match);
- if (limit > matchlimit) {
- limit = matchlimit;
- }
- matchCode = LZ4_count(ip+MINMATCH, match+MINMATCH, limit);
- ip += MINMATCH + matchCode;
- if (ip==limit) {
- unsigned const more = LZ4_count(ip, (const BYTE*)source, matchlimit);
- matchCode += more;
- ip += more;
- }
- } else {
- matchCode = LZ4_count(ip+MINMATCH, match+MINMATCH, matchlimit);
- ip += MINMATCH + matchCode;
- }
- if ( outputLimited && /* Check output buffer overflow */
- (unlikely(op + (1 + LASTLITERALS) + (matchCode>>8) > olimit)) ) {
- return 0;
- }
- if (matchCode >= ML_MASK) {
- *token += ML_MASK;
- matchCode -= ML_MASK;
- LZ4_write32(op, 0xFFFFFFFF);
- while (matchCode >= 4*255) {
- op+=4;
- LZ4_write32(op, 0xFFFFFFFF);
- matchCode -= 4*255;
- }
- op += matchCode / 255;
- *op++ = (BYTE)(matchCode % 255);
- } else {
- *token += (BYTE)(matchCode);
- }
- }
- anchor = ip;
- /* Test end of chunk */
- if (ip > mflimit) {
- break;
- }
- /* Fill table */
- LZ4_putPosition(ip-2, cctx->hashTable, tableType, base);
- /* Test next position */
- match = LZ4_getPosition(ip, cctx->hashTable, tableType, base);
- if (dict==usingExtDict) {
- if (match < (const BYTE*)source) {
- refDelta = dictDelta;
- lowLimit = dictionary;
- } else {
- refDelta = 0;
- lowLimit = (const BYTE*)source;
- }
- }
- LZ4_putPosition(ip, cctx->hashTable, tableType, base);
- if ( ((dictIssue==dictSmall) ? (match>=lowRefLimit) : 1)
- && (match+MAX_DISTANCE>=ip)
- && (LZ4_read32(match+refDelta)==LZ4_read32(ip)) ) {
- token=op++;
- *token=0;
- goto _next_match;
- }
- /* Prepare next loop */
- forwardH = LZ4_hashPosition(++ip, tableType);
- }
- _last_literals:
- /* Encode Last Literals */
- {
- size_t const lastRun = (size_t)(iend - anchor);
- if ( (outputLimited) && /* Check output buffer overflow */
- ((op - (BYTE*)dest) + lastRun + 1 + ((lastRun+255-RUN_MASK)/255) > (U32)maxOutputSize) ) {
- return 0;
- }
- if (lastRun >= RUN_MASK) {
- size_t accumulator = lastRun - RUN_MASK;
- *op++ = RUN_MASK << ML_BITS;
- for(; accumulator >= 255 ; accumulator-=255) {
- *op++ = 255;
- }
- *op++ = (BYTE) accumulator;
- } else {
- *op++ = (BYTE)(lastRun<<ML_BITS);
- }
- memcpy(op, anchor, lastRun);
- op += lastRun;
- }
- /* End */
- return (int) (((char*)op)-dest);
- }
- static inline int LZ4_compress_fast_extState(void* state, const char* source, char* dest, int inputSize, int maxOutputSize, int acceleration)
- {
- LZ4_stream_t_internal* ctx = &((LZ4_stream_t*)state)->internal_donotuse;
- LZ4_resetStream((LZ4_stream_t*)state);
- //if (acceleration < 1) acceleration = ACCELERATION_DEFAULT;
- if (maxOutputSize >= LZ4_compressBound(inputSize)) {
- if (inputSize < LZ4_64Klimit) {
- return LZ4_compress_generic(ctx, source, dest, inputSize, 0, notLimited, byU16, noDict, noDictIssue, acceleration);
- } else {
- return LZ4_compress_generic(ctx, source, dest, inputSize, 0, notLimited, (sizeof(void*)==8) ? byU32 : byPtr, noDict, noDictIssue, acceleration);
- }
- } else {
- if (inputSize < LZ4_64Klimit) {
- return LZ4_compress_generic(ctx, source, dest, inputSize, maxOutputSize, limitedOutput, byU16, noDict, noDictIssue, acceleration);
- } else {
- return LZ4_compress_generic(ctx, source, dest, inputSize, maxOutputSize, limitedOutput, (sizeof(void*)==8) ? byU32 : byPtr, noDict, noDictIssue, acceleration);
- }
- }
- }
- static inline int LZ4_compress_fast(const char* source, char* dest, int inputSize, int maxOutputSize, int acceleration)
- {
- #if (HEAPMODE)
- void* ctxPtr = ALLOCATOR(1, sizeof(LZ4_stream_t)); /* malloc-calloc always properly aligned */
- #else
- LZ4_stream_t ctx;
- void* const ctxPtr = &ctx;
- #endif
- int const result = LZ4_compress_fast_extState(ctxPtr, source, dest, inputSize, maxOutputSize, acceleration);
- #if (HEAPMODE)
- FREEMEM(ctxPtr);
- #endif
- return result;
- }
- static inline void LZ4_resetStream (LZ4_stream_t* LZ4_stream)
- {
- MEM_INIT(LZ4_stream, 0, sizeof(LZ4_stream_t));
- }
- FORCE_INLINE int LZ4_decompress_generic(
- const char* const source,
- char* const dest,
- int inputSize,
- int outputSize, /* If endOnInput==endOnInputSize, this value is the max size of Output Buffer. */
- int endOnInput, /* endOnOutputSize, endOnInputSize */
- int partialDecoding, /* full, partial */
- int targetOutputSize, /* only used if partialDecoding==partial */
- int dict, /* noDict, withPrefix64k, usingExtDict */
- const BYTE* const lowPrefix, /* == dest when no prefix */
- const BYTE* const dictStart, /* only if dict==usingExtDict */
- const size_t dictSize /* note : = 0 if noDict */
- )
- {
- /* Local Variables */
- const BYTE* ip = (const BYTE*) source;
- const BYTE* const iend = ip + inputSize;
- BYTE* op = (BYTE*) dest;
- BYTE* const oend = op + outputSize;
- BYTE* cpy;
- BYTE* oexit = op + targetOutputSize;
- const BYTE* const lowLimit = lowPrefix - dictSize;
- const BYTE* const dictEnd = (const BYTE*)dictStart + dictSize;
- const unsigned dec32table[] = {0, 1, 2, 1, 4, 4, 4, 4};
- const int dec64table[] = {0, 0, 0, -1, 0, 1, 2, 3};
- const int safeDecode = (endOnInput==endOnInputSize);
- const int checkOffset = ((safeDecode) && (dictSize < (int)(64 KB)));
- /* Special cases */
- if ((partialDecoding) && (oexit > oend-MFLIMIT)) {
- oexit = oend-MFLIMIT; /* targetOutputSize too high => decode everything */
- }
- if ((endOnInput) && (unlikely(outputSize==0))) {
- return ((inputSize==1) && (*ip==0)) ? 0 : -1; /* Empty output buffer */
- }
- if ((!endOnInput) && (unlikely(outputSize==0))) {
- return (*ip==0?1:-1);
- }
- /* Main Loop : decode sequences */
- while (1) {
- size_t length;
- const BYTE* match;
- size_t offset;
- /* get literal length */
- unsigned const token = *ip++;
- if ((length=(token>>ML_BITS)) == RUN_MASK) {
- unsigned s;
- do {
- s = *ip++;
- length += s;
- } while ( likely(endOnInput ? ip<iend-RUN_MASK : 1) & (s==255) );
- if ((safeDecode) && unlikely((uptrval)(op)+length<(uptrval)(op))) {
- goto _output_error; /* overflow detection */
- }
- if ((safeDecode) && unlikely((uptrval)(ip)+length<(uptrval)(ip))) {
- goto _output_error; /* overflow detection */
- }
- }
- /* copy literals */
- cpy = op+length;
- if ( ((endOnInput) && ((cpy>(partialDecoding?oexit:oend-MFLIMIT)) || (ip+length>iend-(2+1+LASTLITERALS))) )
- || ((!endOnInput) && (cpy>oend-WILDCOPYLENGTH)) ) {
- if (partialDecoding) {
- if (cpy > oend) {
- goto _output_error; /* Error : write attempt beyond end of output buffer */
- }
- if ((endOnInput) && (ip+length > iend)) {
- goto _output_error; /* Error : read attempt beyond end of input buffer */
- }
- } else {
- if ((!endOnInput) && (cpy != oend)) {
- goto _output_error; /* Error : block decoding must stop exactly there */
- }
- if ((endOnInput) && ((ip+length != iend) || (cpy > oend))) {
- goto _output_error; /* Error : input must be consumed */
- }
- }
- memcpy(op, ip, length);
- ip += length;
- op += length;
- break; /* Necessarily EOF, due to parsing restrictions */
- }
- LZ4_wildCopy(op, ip, cpy);
- ip += length;
- op = cpy;
- /* get offset */
- offset = LZ4_readLE16(ip);
- ip += 2;
- match = op - offset;
- if ((checkOffset) && (unlikely(match < lowLimit))) {
- goto _output_error; /* Error : offset outside buffers */
- }
- LZ4_write32(op, (U32)offset); /* costs ~1%; silence an msan warning when offset==0 */
- /* get matchlength */
- length = token & ML_MASK;
- if (length == ML_MASK) {
- unsigned s;
- do {
- s = *ip++;
- if ((endOnInput) && (ip > iend-LASTLITERALS)) {
- goto _output_error;
- }
- length += s;
- } while (s==255);
- if ((safeDecode) && unlikely((uptrval)(op)+length<(uptrval)op)) {
- goto _output_error; /* overflow detection */
- }
- }
- length += MINMATCH;
- /* check external dictionary */
- if ((dict==usingExtDict) && (match < lowPrefix)) {
- if (unlikely(op+length > oend-LASTLITERALS)) {
- goto _output_error; /* doesn't respect parsing restriction */
- }
- if (length <= (size_t)(lowPrefix-match)) {
- /* match can be copied as a single segment from external dictionary */
- memmove(op, dictEnd - (lowPrefix-match), length);
- op += length;
- } else {
- /* match encompass external dictionary and current block */
- size_t const copySize = (size_t)(lowPrefix-match);
- size_t const restSize = length - copySize;
- memcpy(op, dictEnd - copySize, copySize);
- op += copySize;
- if (restSize > (size_t)(op-lowPrefix)) { /* overlap copy */
- BYTE* const endOfMatch = op + restSize;
- const BYTE* copyFrom = lowPrefix;
- while (op < endOfMatch) {
- *op++ = *copyFrom++;
- }
- } else {
- memcpy(op, lowPrefix, restSize);
- op += restSize;
- }
- }
- continue;
- }
- /* copy match within block */
- cpy = op + length;
- if (unlikely(offset<8)) {
- const int dec64 = dec64table[offset];
- op[0] = match[0];
- op[1] = match[1];
- op[2] = match[2];
- op[3] = match[3];
- match += dec32table[offset];
- memcpy(op+4, match, 4);
- match -= dec64;
- } else {
- LZ4_copy8(op, match);
- match+=8;
- }
- op += 8;
- if (unlikely(cpy>oend-12)) {
- BYTE* const oCopyLimit = oend-(WILDCOPYLENGTH-1);
- if (cpy > oend-LASTLITERALS) {
- goto _output_error; /* Error : last LASTLITERALS bytes must be literals (uncompressed) */
- }
- if (op < oCopyLimit) {
- LZ4_wildCopy(op, match, oCopyLimit);
- match += oCopyLimit - op;
- op = oCopyLimit;
- }
- while (op<cpy) {
- *op++ = *match++;
- }
- } else {
- LZ4_copy8(op, match);
- if (length>16) {
- LZ4_wildCopy(op+8, match+8, cpy);
- }
- }
- op=cpy; /* correction */
- }
- /* end of decoding */
- if (endOnInput) {
- return (int) (((char*)op)-dest); /* Nb of output bytes decoded */
- } else {
- return (int) (((const char*)ip)-source); /* Nb of input bytes read */
- }
- /* Overflow error detected */
- _output_error:
- return (int) (-(((const char*)ip)-source))-1;
- }
- static inline int LZ4_decompress_safe(const char* source, char* dest, int compressedSize, int maxDecompressedSize)
- {
- return LZ4_decompress_generic(source, dest, compressedSize, maxDecompressedSize, endOnInputSize, full, 0, noDict, (BYTE*)dest, NULL, 0);
- }
- } // anonymous namespace
- /************************************************************************** */
- /************************************************************************** */
- const unsigned char Packet::ZERO_KEY[32] = { 0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0 };
- void Packet::armor(const void *key,bool encryptPayload,const AES aesKeys[2])
- {
- uint8_t *const data = reinterpret_cast<uint8_t *>(unsafeData());
- if ((aesKeys) && (encryptPayload)) {
- //char tmp0[16],tmp1[16];
- setCipher(ZT_PROTO_CIPHER_SUITE__AES_GMAC_SIV);
- uint8_t *const payload = data + ZT_PACKET_IDX_VERB;
- const unsigned int payloadLen = size() - ZT_PACKET_IDX_VERB;
- AES::GMACSIVEncryptor enc(aesKeys[0],aesKeys[1]);
- enc.init(Utils::loadMachineEndian<uint64_t>(data + ZT_PACKET_IDX_IV),payload);
- enc.aad(data + ZT_PACKET_IDX_DEST,11);
- enc.update1(payload,payloadLen);
- enc.finish1();
- enc.update2(payload,payloadLen);
- const uint64_t *const tag = enc.finish2();
- #ifdef ZT_NO_UNALIGNED_ACCESS
- Utils::copy<8>(data,tag);
- Utils::copy<8>(data + ZT_PACKET_IDX_MAC,tag + 1);
- #else
- *reinterpret_cast<uint64_t *>(data + ZT_PACKET_IDX_IV) = tag[0];
- *reinterpret_cast<uint64_t *>(data + ZT_PACKET_IDX_MAC) = tag[1];
- #endif
- } else {
- setCipher(encryptPayload ? ZT_PROTO_CIPHER_SUITE__C25519_POLY1305_SALSA2012 : ZT_PROTO_CIPHER_SUITE__C25519_POLY1305_NONE);
- uint8_t mangledKey[32];
- _salsa20MangleKey((const unsigned char *)key,mangledKey);
- if (ZT_HAS_FAST_CRYPTO()) {
- const unsigned int payloadLen = (encryptPayload) ? (size() - ZT_PACKET_IDX_VERB) : 0;
- uint64_t keyStream[(ZT_PROTO_MAX_PACKET_LENGTH + 64 + 8) / 8];
- ZT_FAST_SINGLE_PASS_SALSA2012(keyStream,payloadLen + 64,(data + ZT_PACKET_IDX_IV),mangledKey);
- Salsa20::memxor(data + ZT_PACKET_IDX_VERB,reinterpret_cast<const uint8_t *>(keyStream + 8),payloadLen);
- uint64_t mac[2];
- Poly1305::compute(mac,data + ZT_PACKET_IDX_VERB,size() - ZT_PACKET_IDX_VERB,keyStream);
- #ifdef ZT_NO_TYPE_PUNNING
- memcpy(data + ZT_PACKET_IDX_MAC,mac,8);
- #else
- (*reinterpret_cast<uint64_t *>(data + ZT_PACKET_IDX_MAC)) = mac[0];
- #endif
- } else {
- Salsa20 s20(mangledKey,data + ZT_PACKET_IDX_IV);
- uint64_t macKey[4];
- s20.crypt12(ZERO_KEY,macKey,sizeof(macKey));
- uint8_t *const payload = data + ZT_PACKET_IDX_VERB;
- const unsigned int payloadLen = size() - ZT_PACKET_IDX_VERB;
- if (encryptPayload) {
- s20.crypt12(payload,payload,payloadLen);
- }
- uint64_t mac[2];
- Poly1305::compute(mac,payload,payloadLen,macKey);
- memcpy(data + ZT_PACKET_IDX_MAC,mac,8);
- }
- }
- }
- bool Packet::dearmor(const void *key,const AES aesKeys[2])
- {
- uint8_t *const data = reinterpret_cast<uint8_t *>(unsafeData());
- const unsigned int payloadLen = size() - ZT_PACKET_IDX_VERB;
- unsigned char *const payload = data + ZT_PACKET_IDX_VERB;
- const unsigned int cs = cipher();
- if (cs == ZT_PROTO_CIPHER_SUITE__AES_GMAC_SIV) {
- if (aesKeys) {
- uint64_t tag[2];
- #ifdef ZT_NO_UNALIGNED_ACCESS
- Utils::copy<8>(tag, data);
- Utils::copy<8>(tag + 1, data + ZT_PACKET_IDX_MAC);
- #else
- tag[0] = *reinterpret_cast<uint64_t *>(data + ZT_PACKET_IDX_IV);
- tag[1] = *reinterpret_cast<uint64_t *>(data + ZT_PACKET_IDX_MAC);
- #endif
- AES::GMACSIVDecryptor dec(aesKeys[0],aesKeys[1]);
- dec.init(tag, payload);
- const uint8_t oldFlags = data[ZT_PACKET_IDX_FLAGS];
- data[ZT_PACKET_IDX_FLAGS] &= 0xf8;
- dec.aad(data + ZT_PACKET_IDX_DEST,11);
- data[ZT_PACKET_IDX_FLAGS] = oldFlags;
- dec.update(payload, payloadLen);
- return dec.finish();
- }
- } else if ((cs == ZT_PROTO_CIPHER_SUITE__C25519_POLY1305_NONE)||(cs == ZT_PROTO_CIPHER_SUITE__C25519_POLY1305_SALSA2012)) {
- uint8_t mangledKey[32];
- _salsa20MangleKey((const unsigned char *)key,mangledKey);
- if (ZT_HAS_FAST_CRYPTO()) {
- uint64_t keyStream[(ZT_PROTO_MAX_PACKET_LENGTH + 64 + 8) / 8];
- ZT_FAST_SINGLE_PASS_SALSA2012(keyStream,((cs == ZT_PROTO_CIPHER_SUITE__C25519_POLY1305_SALSA2012) ? (payloadLen + 64) : 64),(data + ZT_PACKET_IDX_IV),mangledKey);
- uint64_t mac[2];
- Poly1305::compute(mac,payload,payloadLen,keyStream);
- #ifdef ZT_NO_TYPE_PUNNING
- if (!Utils::secureEq(mac,data + ZT_PACKET_IDX_MAC,8)) {
- return false;
- }
- #else
- if ((*reinterpret_cast<const uint64_t *>(data + ZT_PACKET_IDX_MAC)) != mac[0]) { // also secure, constant time
- return false;
- }
- #endif
- if (cs == ZT_PROTO_CIPHER_SUITE__C25519_POLY1305_SALSA2012) {
- Salsa20::memxor(data + ZT_PACKET_IDX_VERB,reinterpret_cast<const uint8_t *>(keyStream + 8),payloadLen);
- }
- } else {
- Salsa20 s20(mangledKey,data + ZT_PACKET_IDX_IV);
- uint64_t macKey[4];
- s20.crypt12(ZERO_KEY,macKey,sizeof(macKey));
- uint64_t mac[2];
- Poly1305::compute(mac,payload,payloadLen,macKey);
- #ifdef ZT_NO_TYPE_PUNNING
- if (!Utils::secureEq(mac,data + ZT_PACKET_IDX_MAC,8)) {
- return false;
- }
- #else
- if ((*reinterpret_cast<const uint64_t *>(data + ZT_PACKET_IDX_MAC)) != mac[0]) { // also secure, constant time
- return false;
- }
- #endif
- if (cs == ZT_PROTO_CIPHER_SUITE__C25519_POLY1305_SALSA2012) {
- s20.crypt12(payload,payload,payloadLen);
- }
- }
- return true;
- }
- return false;
- }
- void Packet::cryptField(const void *key,unsigned int start,unsigned int len)
- {
- uint8_t *const data = reinterpret_cast<uint8_t *>(unsafeData());
- uint8_t iv[8];
- for(int i=0;i<8;++i) {
- iv[i] = data[i];
- }
- iv[7] &= 0xf8; // mask off least significant 3 bits of packet ID / IV since this is unset when this function gets called
- Salsa20 s20(key,iv);
- s20.crypt12(data + start,data + start,len);
- }
- bool Packet::compress()
- {
- char *const data = reinterpret_cast<char *>(unsafeData());
- char buf[ZT_PROTO_MAX_PACKET_LENGTH * 2];
- if ((!compressed())&&(size() > (ZT_PACKET_IDX_PAYLOAD + 64))) { // don't bother compressing tiny packets
- int pl = (int)(size() - ZT_PACKET_IDX_PAYLOAD);
- int cl = LZ4_compress_fast(data + ZT_PACKET_IDX_PAYLOAD,buf,pl,ZT_PROTO_MAX_PACKET_LENGTH * 2,1);
- if ((cl > 0)&&(cl < pl)) {
- data[ZT_PACKET_IDX_VERB] |= (char)ZT_PROTO_VERB_FLAG_COMPRESSED;
- setSize((unsigned int)cl + ZT_PACKET_IDX_PAYLOAD);
- memcpy(data + ZT_PACKET_IDX_PAYLOAD,buf,cl);
- return true;
- }
- }
- data[ZT_PACKET_IDX_VERB] &= (char)(~ZT_PROTO_VERB_FLAG_COMPRESSED);
- return false;
- }
- bool Packet::uncompress()
- {
- char *const data = reinterpret_cast<char *>(unsafeData());
- char buf[ZT_PROTO_MAX_PACKET_LENGTH];
- if ((compressed())&&(size() >= ZT_PROTO_MIN_PACKET_LENGTH)) {
- if (size() > ZT_PACKET_IDX_PAYLOAD) {
- unsigned int compLen = size() - ZT_PACKET_IDX_PAYLOAD;
- int ucl = LZ4_decompress_safe((const char *)data + ZT_PACKET_IDX_PAYLOAD,buf,compLen,sizeof(buf));
- if ((ucl > 0)&&(ucl <= (int)(capacity() - ZT_PACKET_IDX_PAYLOAD))) {
- setSize((unsigned int)ucl + ZT_PACKET_IDX_PAYLOAD);
- memcpy(data + ZT_PACKET_IDX_PAYLOAD,buf,ucl);
- } else {
- return false;
- }
- }
- data[ZT_PACKET_IDX_VERB] &= (char)(~ZT_PROTO_VERB_FLAG_COMPRESSED);
- }
- return true;
- }
- } // namespace ZeroTier
|