110 lines
3.2 KiB
C
110 lines
3.2 KiB
C
/*
|
|
* Copyright (c) 2019 TAOS Data, Inc. <jhtao@taosdata.com>
|
|
*
|
|
* This program is free software: you can use, redistribute, and/or modify
|
|
* it under the terms of the GNU Affero General Public License, version 3
|
|
* or later ("AGPL"), as published by the Free Software Foundation.
|
|
*
|
|
* This program is distributed in the hope that it will be useful, but WITHOUT
|
|
* ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
|
|
* FITNESS FOR A PARTICULAR PURPOSE.
|
|
*
|
|
* You should have received a copy of the GNU Affero General Public License
|
|
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
*/
|
|
|
|
#ifndef _TDB_UTIL_H_
|
|
#define _TDB_UTIL_H_
|
|
|
|
#ifdef __cplusplus
|
|
extern "C" {
|
|
#endif
|
|
|
|
#if __STDC_VERSION__ >= 201112L
|
|
#define TDB_STATIC_ASSERT(op, info) static_assert(op, info)
|
|
#else
|
|
#define TDB_STATIC_ASSERT(op, info)
|
|
#endif
|
|
|
|
#define TDB_ROUND8(x) (((x) + 7) & ~7)
|
|
|
|
int tdbGnrtFileID(const char *fname, uint8_t *fileid, bool unique);
|
|
|
|
int tdbGetFileSize(const char *fname, int pgSize, SPgno *pSize);
|
|
|
|
#define TDB_REALLOC(PTR, SIZE) \
|
|
({ \
|
|
void *nPtr; \
|
|
if ((PTR) == NULL || ((int *)(PTR))[-1] < (SIZE)) { \
|
|
nPtr = tdbOsRealloc((PTR) ? (char *)(PTR) - sizeof(int) : NULL, (SIZE) + sizeof(int)); \
|
|
if (nPtr) { \
|
|
((int *)nPtr)[0] = (SIZE); \
|
|
nPtr = (char *)nPtr + sizeof(int); \
|
|
} \
|
|
} else { \
|
|
nPtr = (PTR); \
|
|
} \
|
|
nPtr; \
|
|
})
|
|
|
|
#define TDB_FREE(PTR) \
|
|
do { \
|
|
if (PTR) { \
|
|
tdbOsFree((char *)(PTR) - sizeof(int)); \
|
|
} \
|
|
} while (0)
|
|
|
|
static inline void *tdbDefaultMalloc(void *arg, size_t size) {
|
|
void *ptr;
|
|
|
|
ptr = tdbOsMalloc(size);
|
|
|
|
return ptr;
|
|
}
|
|
|
|
static inline void tdbDefaultFree(void *arg, void *ptr) { tdbOsFree(ptr); }
|
|
|
|
static inline int tdbPutVarInt(u8 *p, int v) {
|
|
int n = 0;
|
|
|
|
for (;;) {
|
|
if (v <= 0x7f) {
|
|
p[n++] = v;
|
|
break;
|
|
}
|
|
|
|
p[n++] = (v & 0x7f) | 0x80;
|
|
v >>= 7;
|
|
}
|
|
|
|
ASSERT(n < 6);
|
|
|
|
return n;
|
|
}
|
|
|
|
static inline int tdbGetVarInt(const u8 *p, int *v) {
|
|
int n = 0;
|
|
int tv = 0;
|
|
|
|
for (;;) {
|
|
if (p[n] <= 0x7f) {
|
|
tv = (tv << 7) | p[n];
|
|
n++;
|
|
break;
|
|
}
|
|
|
|
tv = (tv << 7) | (p[n] & 0x7f);
|
|
n++;
|
|
}
|
|
|
|
ASSERT(n < 6);
|
|
|
|
*v = tv;
|
|
return n;
|
|
}
|
|
|
|
#ifdef __cplusplus
|
|
}
|
|
#endif
|
|
|
|
#endif /*_TDB_UTIL_H_*/ |