1 /* 2 * 3 * honggfuzz - utilities 4 * ----------------------------------------- 5 * 6 * Author: Robert Swiecki <swiecki (at) google.com> 7 * 8 * Copyright 2010-2015 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_UTIL_H_ 25 #define _HF_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 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 uint64_t util_rnd64(void); 91 92 extern int util_ssnprintf(char* str, size_t size, const char* format, ...); 93 94 extern int util_vssnprintf(char* str, size_t size, const char* format, va_list ap); 95 96 extern void util_getLocalTime(const char* fmt, char* buf, size_t len, time_t tm); 97 98 extern void util_nullifyStdio(void); 99 100 extern bool util_redirectStdin(const char* inputFile); 101 102 extern uint64_t util_hash(const char* buf, size_t len); 103 104 extern int64_t util_timeNowMillis(void); 105 106 extern uint64_t util_getUINT32(const uint8_t* buf); 107 extern uint64_t util_getUINT64(const uint8_t* buf); 108 109 extern void util_mutexLock(pthread_mutex_t* mutex, const char* func, int line); 110 extern void util_mutexUnlock(pthread_mutex_t* mutex, const char* func, int line); 111 112 extern void util_mutexRWLockRead(pthread_rwlock_t* mutex, const char* func, int line); 113 extern void util_mutexRWLockWrite(pthread_rwlock_t* mutex, const char* func, int line); 114 extern void util_mutexRWUnlock(pthread_rwlock_t* mutex, const char* func, int line); 115 116 extern int64_t fastArray64Search(uint64_t* array, size_t arraySz, uint64_t key); 117 118 extern bool util_isANumber(const char* s); 119 120 extern size_t util_decodeCString(char* s); 121 122 extern uint64_t util_CRC64(const uint8_t* buf, size_t len); 123 extern uint64_t util_CRC64Rev(const uint8_t* buf, size_t len); 124 125 #endif 126
