1 /**
2 * Copyright (c) 2016-present, Yann Collet, Facebook, Inc.
3 * All rights reserved.
4 *
5 * This source code is licensed under the BSD-style license found in the
6 * LICENSE file in the root directory of https://github.com/facebook/zstd.
7 * An additional grant of patent rights can be found in the PATENTS file in the
8 * same directory.
9 *
10 * This program is free software; you can redistribute it and/or modify it under
11 * the terms of the GNU General Public License version 2 as published by the
12 * Free Software Foundation. This program is dual-licensed; you may select
13 * either version 2 of the GNU General Public License ("GPL") or BSD license
14 * ("BSD").
15 */
16
17 #ifndef MEM_H_MODULE
18 #define MEM_H_MODULE
19
20 /*-****************************************
21 * Dependencies
22 ******************************************/
23 #ifdef __XEN__
24 #include <xen/string.h> /* memcpy */
25 #include <xen/types.h> /* size_t, ptrdiff_t */
26 #include <xen/unaligned.h>
27 #endif
28
29 /*-****************************************
30 * Compiler specifics
31 ******************************************/
32 #define ZSTD_STATIC static inline
33
34 /*-**************************************************************
35 * Basic Types
36 *****************************************************************/
37 typedef uint8_t BYTE;
38 typedef uint16_t U16;
39 typedef int16_t S16;
40 typedef uint32_t U32;
41 typedef int32_t S32;
42 typedef uint64_t U64;
43 typedef int64_t S64;
44 typedef ptrdiff_t iPtrDiff;
45 typedef uintptr_t uPtrDiff;
46
47 /*-**************************************************************
48 * Memory I/O
49 *****************************************************************/
ZSTD_32bits(void)50 ZSTD_STATIC unsigned ZSTD_32bits(void) { return sizeof(size_t) == 4; }
ZSTD_64bits(void)51 ZSTD_STATIC unsigned ZSTD_64bits(void) { return sizeof(size_t) == 8; }
52
53 #if defined(__LITTLE_ENDIAN)
54 #define ZSTD_LITTLE_ENDIAN 1
55 #else
56 #define ZSTD_LITTLE_ENDIAN 0
57 #endif
58
ZSTD_isLittleEndian(void)59 ZSTD_STATIC unsigned ZSTD_isLittleEndian(void) { return ZSTD_LITTLE_ENDIAN; }
60
ZSTD_read16(const void * memPtr)61 ZSTD_STATIC U16 ZSTD_read16(const void *memPtr) { return get_unaligned((const U16 *)memPtr); }
62
ZSTD_read32(const void * memPtr)63 ZSTD_STATIC U32 ZSTD_read32(const void *memPtr) { return get_unaligned((const U32 *)memPtr); }
64
ZSTD_read64(const void * memPtr)65 ZSTD_STATIC U64 ZSTD_read64(const void *memPtr) { return get_unaligned((const U64 *)memPtr); }
66
ZSTD_readST(const void * memPtr)67 ZSTD_STATIC size_t ZSTD_readST(const void *memPtr) { return get_unaligned((const size_t *)memPtr); }
68
ZSTD_write16(void * memPtr,U16 value)69 ZSTD_STATIC void ZSTD_write16(void *memPtr, U16 value) { put_unaligned(value, (U16 *)memPtr); }
70
ZSTD_write32(void * memPtr,U32 value)71 ZSTD_STATIC void ZSTD_write32(void *memPtr, U32 value) { put_unaligned(value, (U32 *)memPtr); }
72
ZSTD_write64(void * memPtr,U64 value)73 ZSTD_STATIC void ZSTD_write64(void *memPtr, U64 value) { put_unaligned(value, (U64 *)memPtr); }
74
75 /*=== Little endian r/w ===*/
76
ZSTD_readLE16(const void * memPtr)77 ZSTD_STATIC U16 ZSTD_readLE16(const void *memPtr) { return get_unaligned_le16(memPtr); }
78
ZSTD_writeLE16(void * memPtr,U16 val)79 ZSTD_STATIC void ZSTD_writeLE16(void *memPtr, U16 val) { put_unaligned_le16(val, memPtr); }
80
ZSTD_readLE24(const void * memPtr)81 ZSTD_STATIC U32 ZSTD_readLE24(const void *memPtr) { return ZSTD_readLE16(memPtr) + (((const BYTE *)memPtr)[2] << 16); }
82
ZSTD_writeLE24(void * memPtr,U32 val)83 ZSTD_STATIC void ZSTD_writeLE24(void *memPtr, U32 val)
84 {
85 ZSTD_writeLE16(memPtr, (U16)val);
86 ((BYTE *)memPtr)[2] = (BYTE)(val >> 16);
87 }
88
ZSTD_readLE32(const void * memPtr)89 ZSTD_STATIC U32 ZSTD_readLE32(const void *memPtr) { return get_unaligned_le32(memPtr); }
90
ZSTD_writeLE32(void * memPtr,U32 val32)91 ZSTD_STATIC void ZSTD_writeLE32(void *memPtr, U32 val32) { put_unaligned_le32(val32, memPtr); }
92
ZSTD_readLE64(const void * memPtr)93 ZSTD_STATIC U64 ZSTD_readLE64(const void *memPtr) { return get_unaligned_le64(memPtr); }
94
ZSTD_writeLE64(void * memPtr,U64 val64)95 ZSTD_STATIC void ZSTD_writeLE64(void *memPtr, U64 val64) { put_unaligned_le64(val64, memPtr); }
96
ZSTD_readLEST(const void * memPtr)97 ZSTD_STATIC size_t ZSTD_readLEST(const void *memPtr)
98 {
99 if (ZSTD_32bits())
100 return (size_t)ZSTD_readLE32(memPtr);
101 else
102 return (size_t)ZSTD_readLE64(memPtr);
103 }
104
ZSTD_writeLEST(void * memPtr,size_t val)105 ZSTD_STATIC void ZSTD_writeLEST(void *memPtr, size_t val)
106 {
107 if (ZSTD_32bits())
108 ZSTD_writeLE32(memPtr, (U32)val);
109 else
110 ZSTD_writeLE64(memPtr, (U64)val);
111 }
112
113 /*=== Big endian r/w ===*/
114
ZSTD_readBE32(const void * memPtr)115 ZSTD_STATIC U32 ZSTD_readBE32(const void *memPtr) { return get_unaligned_be32(memPtr); }
116
ZSTD_writeBE32(void * memPtr,U32 val32)117 ZSTD_STATIC void ZSTD_writeBE32(void *memPtr, U32 val32) { put_unaligned_be32(val32, memPtr); }
118
ZSTD_readBE64(const void * memPtr)119 ZSTD_STATIC U64 ZSTD_readBE64(const void *memPtr) { return get_unaligned_be64(memPtr); }
120
ZSTD_writeBE64(void * memPtr,U64 val64)121 ZSTD_STATIC void ZSTD_writeBE64(void *memPtr, U64 val64) { put_unaligned_be64(val64, memPtr); }
122
ZSTD_readBEST(const void * memPtr)123 ZSTD_STATIC size_t ZSTD_readBEST(const void *memPtr)
124 {
125 if (ZSTD_32bits())
126 return (size_t)ZSTD_readBE32(memPtr);
127 else
128 return (size_t)ZSTD_readBE64(memPtr);
129 }
130
ZSTD_writeBEST(void * memPtr,size_t val)131 ZSTD_STATIC void ZSTD_writeBEST(void *memPtr, size_t val)
132 {
133 if (ZSTD_32bits())
134 ZSTD_writeBE32(memPtr, (U32)val);
135 else
136 ZSTD_writeBE64(memPtr, (U64)val);
137 }
138
139 /* function safe only for comparisons */
ZSTD_readMINMATCH(const void * memPtr,U32 length)140 ZSTD_STATIC U32 ZSTD_readMINMATCH(const void *memPtr, U32 length)
141 {
142 switch (length) {
143 default:
144 case 4: return ZSTD_read32(memPtr);
145 case 3:
146 if (ZSTD_isLittleEndian())
147 return ZSTD_read32(memPtr) << 8;
148 else
149 return ZSTD_read32(memPtr) >> 8;
150 }
151 }
152
153 #endif /* MEM_H_MODULE */
154