1 /*
2 *
3 * honggfuzz - utilities
4 * -----------------------------------------
5 *
6 * Author: Robert Swiecki <swiecki@google.com>
7 *
8 * Copyright 2010-2018 by Google Inc. All Rights Reserved.
9 *
10 * Licensed under the Apache License, Version 2.0 (the "License"); you may
11 * not use this file except in compliance with the License. You may obtain
12 * a copy of the License at
13 *
14 * http://www.apache.org/licenses/LICENSE-2.0
15 *
16 * Unless required by applicable law or agreed to in writing, software
17 * distributed under the License is distributed on an "AS IS" BASIS,
18 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or
19 * implied. See the License for the specific language governing
20 * permissions and limitations under the License.
21 *
22 */
23
24 #ifndef _HF_COMMON_UTIL_H_
25 #define _HF_COMMON_UTIL_H_
26
27 #include <pthread.h>
28 #include <stdarg.h>
29 #ifdef __clang__
30 #include <stdatomic.h>
31 #endif
32 #include <stdbool.h>
33 #include <stdint.h>
34 #include <time.h>
35
36 #define MX_LOCK(m) util_mutexLock(m, __func__, __LINE__)
37 #define MX_UNLOCK(m) util_mutexUnlock(m, __func__, __LINE__)
38 #define MX_RWLOCK_READ(m) util_mutexRWLockRead(m, __func__, __LINE__)
39 #define MX_RWLOCK_WRITE(m) util_mutexRWLockWrite(m, __func__, __LINE__)
40 #define MX_RWLOCK_UNLOCK(m) util_mutexRWUnlock(m, __func__, __LINE__)
41
42 /* Atomics */
43 #define ATOMIC_GET(x) __atomic_load_n(&(x), __ATOMIC_SEQ_CST)
44 #define ATOMIC_SET(x, y) __atomic_store_n(&(x), y, __ATOMIC_SEQ_CST)
45 #define ATOMIC_CLEAR(x) __atomic_store_n(&(x), 0, __ATOMIC_SEQ_CST)
46 #define ATOMIC_XCHG(x, y) __atomic_exchange_n(&(x), y, __ATOMIC_SEQ_CST)
47
48 #define ATOMIC_PRE_INC(x) __atomic_add_fetch(&(x), 1, __ATOMIC_SEQ_CST)
49 #define ATOMIC_POST_INC(x) __atomic_fetch_add(&(x), 1, __ATOMIC_SEQ_CST)
50
51 #define ATOMIC_PRE_DEC(x) __atomic_sub_fetch(&(x), 1, __ATOMIC_SEQ_CST)
52 #define ATOMIC_POST_DEC(x) __atomic_fetch_sub(&(x), 1, __ATOMIC_SEQ_CST)
53
54 #define ATOMIC_PRE_ADD(x, y) __atomic_add_fetch(&(x), y, __ATOMIC_SEQ_CST)
55 #define ATOMIC_POST_ADD(x, y) __atomic_fetch_add(&(x), y, __ATOMIC_SEQ_CST)
56
57 #define ATOMIC_PRE_SUB(x, y) __atomic_sub_fetch(&(x), y, __ATOMIC_SEQ_CST)
58 #define ATOMIC_POST_SUB(x, y) __atomic_fetch_sub(&(x), y, __ATOMIC_SEQ_CST)
59
60 #define ATOMIC_PRE_AND(x, y) __atomic_and_fetch(&(x), y, __ATOMIC_SEQ_CST)
61 #define ATOMIC_POST_AND(x, y) __atomic_fetch_and(&(x), y, __ATOMIC_SEQ_CST)
62
63 #define ATOMIC_PRE_OR(x, y) __atomic_or_fetch(&(x), y, __ATOMIC_SEQ_CST)
64 #define ATOMIC_POST_OR(x, y) __atomic_fetch_or(&(x), y, __ATOMIC_SEQ_CST)
65
66 #define ATOMIC_PRE_INC_RELAXED(x) __atomic_add_fetch(&(x), 1, __ATOMIC_RELAXED)
67 #define ATOMIC_POST_OR_RELAXED(x, y) __atomic_fetch_or(&(x), y, __ATOMIC_RELAXED)
68
ATOMIC_BTS(uint8_t * addr,size_t offset)69 __attribute__((always_inline)) static inline uint8_t ATOMIC_BTS(uint8_t* addr, size_t offset) {
70 uint8_t oldbit;
71 addr += (offset / 8);
72 oldbit = ATOMIC_POST_OR_RELAXED(*addr, ((uint8_t)1U << (offset % 8)));
73 return oldbit;
74 }
75
76 extern void* util_Malloc(size_t sz);
77
78 extern void* util_Calloc(size_t sz);
79
80 extern void* util_MMap(size_t sz);
81
82 extern void* util_Realloc(void* ptr, size_t sz);
83
84 extern char* util_StrDup(const char* s);
85
86 extern uint64_t util_rndGet(uint64_t min, uint64_t max);
87
88 extern void util_rndBuf(uint8_t* buf, size_t sz);
89
90 extern void util_rndBufPrintable(uint8_t* buf, size_t sz);
91
92 extern uint64_t util_rnd64(void);
93
94 extern uint8_t util_rndPrintable(void);
95
96 extern void util_turnToPrintable(uint8_t* buf, size_t sz);
97
98 extern int util_ssnprintf(char* str, size_t size, const char* format, ...)
99 __attribute__((format(printf, 3, 4)));
100
101 extern int util_vssnprintf(char* str, size_t size, const char* format, va_list ap);
102
103 extern bool util_strStartsWith(const char* str, const char* tofind);
104
105 extern void util_getLocalTime(const char* fmt, char* buf, size_t len, time_t tm);
106
107 extern void util_closeStdio(bool close_stdin, bool close_stdout, bool close_stderr);
108
109 extern uint64_t util_hash(const char* buf, size_t len);
110
111 extern int64_t util_timeNowMillis(void);
112 extern void util_sleepForMSec(uint64_t msec);
113
114 extern uint64_t util_getUINT32(const uint8_t* buf);
115 extern uint64_t util_getUINT64(const uint8_t* buf);
116
117 extern void util_mutexLock(pthread_mutex_t* mutex, const char* func, int line);
118 extern void util_mutexUnlock(pthread_mutex_t* mutex, const char* func, int line);
119
120 extern void util_mutexRWLockRead(pthread_rwlock_t* mutex, const char* func, int line);
121 extern void util_mutexRWLockWrite(pthread_rwlock_t* mutex, const char* func, int line);
122 extern void util_mutexRWUnlock(pthread_rwlock_t* mutex, const char* func, int line);
123
124 extern int64_t fastArray64Search(uint64_t* array, size_t arraySz, uint64_t key);
125
126 extern bool util_isANumber(const char* s);
127
128 extern size_t util_decodeCString(char* s);
129
130 extern uint64_t util_CRC64(const uint8_t* buf, size_t len);
131 extern uint64_t util_CRC64Rev(const uint8_t* buf, size_t len);
132
133 #endif /* ifndef _HF_COMMON_UTIL_H_ */
134