robert.swiecki | 3bb518c | 2010-10-14 00:48:24 +0000 | [diff] [blame] | 1 | /* |
robert.swiecki@gmail.com | 3b630b4 | 2015-02-16 10:53:53 +0000 | [diff] [blame] | 2 | * |
robert.swiecki@gmail.com | 97c7733 | 2015-02-14 23:06:58 +0000 | [diff] [blame] | 3 | * honggfuzz - utilities |
| 4 | * ----------------------------------------- |
robert.swiecki@gmail.com | 3b630b4 | 2015-02-16 10:53:53 +0000 | [diff] [blame] | 5 | * |
robert.swiecki@gmail.com | 772b33d | 2015-02-14 20:35:00 +0000 | [diff] [blame] | 6 | * Author: Robert Swiecki <swiecki@google.com> |
robert.swiecki@gmail.com | 3b630b4 | 2015-02-16 10:53:53 +0000 | [diff] [blame] | 7 | * |
robert.swiecki@gmail.com | 772b33d | 2015-02-14 20:35:00 +0000 | [diff] [blame] | 8 | * Copyright 2010-2015 by Google Inc. All Rights Reserved. |
robert.swiecki@gmail.com | 3b630b4 | 2015-02-16 10:53:53 +0000 | [diff] [blame] | 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 |
robert.swiecki@gmail.com | 772b33d | 2015-02-14 20:35:00 +0000 | [diff] [blame] | 12 | * a copy of the License at |
robert.swiecki@gmail.com | 3b630b4 | 2015-02-16 10:53:53 +0000 | [diff] [blame] | 13 | * |
robert.swiecki@gmail.com | 772b33d | 2015-02-14 20:35:00 +0000 | [diff] [blame] | 14 | * http://www.apache.org/licenses/LICENSE-2.0 |
robert.swiecki@gmail.com | 3b630b4 | 2015-02-16 10:53:53 +0000 | [diff] [blame] | 15 | * |
robert.swiecki@gmail.com | 772b33d | 2015-02-14 20:35:00 +0000 | [diff] [blame] | 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. |
robert.swiecki@gmail.com | 3b630b4 | 2015-02-16 10:53:53 +0000 | [diff] [blame] | 21 | * |
robert.swiecki@gmail.com | 772b33d | 2015-02-14 20:35:00 +0000 | [diff] [blame] | 22 | */ |
robert.swiecki | 3bb518c | 2010-10-14 00:48:24 +0000 | [diff] [blame] | 23 | |
Jagger | 876a74c | 2016-02-09 22:09:11 +0100 | [diff] [blame] | 24 | #ifndef _HF_UTIL_H_ |
| 25 | #define _HF_UTIL_H_ |
robert.swiecki | 3bb518c | 2010-10-14 00:48:24 +0000 | [diff] [blame] | 26 | |
robert.swiecki@gmail.com | 36700b5 | 2015-02-22 05:03:16 +0000 | [diff] [blame] | 27 | #include <stdarg.h> |
Jagger | c8b6eda | 2016-09-01 03:52:35 +0200 | [diff] [blame] | 28 | #ifdef __clang__ |
| 29 | #include <stdatomic.h> |
| 30 | #endif |
robert.swiecki | 3bb518c | 2010-10-14 00:48:24 +0000 | [diff] [blame] | 31 | #include <stdint.h> |
| 32 | |
Jagger | 6caced3 | 2016-03-10 21:42:23 +0100 | [diff] [blame] | 33 | #define MX_LOCK(m) util_mutexLock(m, __func__, __LINE__) |
| 34 | #define MX_UNLOCK(m) util_mutexUnlock(m, __func__, __LINE__) |
Jagger | 4fe1869 | 2016-04-22 23:15:07 +0200 | [diff] [blame] | 35 | #define MX_SCOPED_LOCK(m) MX_LOCK(m); defer { MX_UNLOCK(m); } |
Jagger | 6caced3 | 2016-03-10 21:42:23 +0100 | [diff] [blame] | 36 | |
Jagger | 7d4f3fd | 2016-09-01 03:48:17 +0200 | [diff] [blame] | 37 | /* Atomics */ |
| 38 | #define ATOMIC_GET(x) __atomic_load_n(&(x), __ATOMIC_SEQ_CST) |
| 39 | #define ATOMIC_SET(x, y) __atomic_store_n(&(x), y, __ATOMIC_SEQ_CST) |
| 40 | #define ATOMIC_CLEAR(x) __atomic_store_n(&(x), 0, __ATOMIC_SEQ_CST) |
| 41 | #define ATOMIC_XCHG(x, y) __atomic_exchange_n(&(x), y, __ATOMIC_SEQ_CST) |
| 42 | |
| 43 | #define ATOMIC_PRE_INC(x) __atomic_add_fetch(&(x), 1, __ATOMIC_SEQ_CST) |
| 44 | #define ATOMIC_POST_INC(x) __atomic_fetch_add(&(x), 1, __ATOMIC_SEQ_CST) |
| 45 | |
| 46 | #define ATOMIC_PRE_DEC(x) __atomic_sub_fetch(&(x), 1, __ATOMIC_SEQ_CST) |
| 47 | #define ATOMIC_POST_DEC(x) __atomic_fetch_sub(&(x), 1, __ATOMIC_SEQ_CST) |
| 48 | |
| 49 | #define ATOMIC_PRE_ADD(x, y) __atomic_add_fetch(&(x), y, __ATOMIC_SEQ_CST) |
| 50 | #define ATOMIC_POST_ADD(x, y) __atomic_fetch_add(&(x), y, __ATOMIC_SEQ_CST) |
| 51 | |
| 52 | #define ATOMIC_PRE_SUB(x, y) __atomic_sub_fetch(&(x), y, __ATOMIC_SEQ_CST) |
| 53 | #define ATOMIC_POST_SUB(x, y) __atomic_fetch_sub(&(x), y, __ATOMIC_SEQ_CST) |
| 54 | |
| 55 | #define ATOMIC_PRE_AND(x, y) __atomic_and_fetch(&(x), y, __ATOMIC_SEQ_CST) |
| 56 | #define ATOMIC_POST_AND(x, y) __atomic_fetch_and(&(x), y, __ATOMIC_SEQ_CST) |
| 57 | |
| 58 | #define ATOMIC_PRE_OR(x, y) __atomic_or_fetch(&(x), y, __ATOMIC_SEQ_CST) |
| 59 | #define ATOMIC_POST_OR(x, y) __atomic_fetch_or(&(x), y, __ATOMIC_SEQ_CST) |
| 60 | |
| 61 | #define ATOMIC_PRE_INC_RELAXED(x) __atomic_add_fetch(&(x), 1, __ATOMIC_RELAXED) |
| 62 | #define ATOMIC_POST_OR_RELAXED(x, y) __atomic_fetch_or(&(x), y, __ATOMIC_RELAXED) |
| 63 | |
Jagger | 2ed5b73 | 2016-09-04 17:45:59 +0200 | [diff] [blame] | 64 | __attribute__ ((always_inline)) |
Jagger | 7d4f3fd | 2016-09-01 03:48:17 +0200 | [diff] [blame] | 65 | static inline uint8_t ATOMIC_BTS(uint8_t * addr, size_t offset) |
| 66 | { |
| 67 | uint8_t oldbit; |
| 68 | addr += (offset / 8); |
| 69 | #if defined(__x86_64__) |
| 70 | /* *INDENT-OFF* */ |
| 71 | __asm__("lock btsq %2, %1\n" |
| 72 | "sbb %0, %0\n" |
| 73 | :"=r"(oldbit), "+m"(*addr) |
| 74 | :"Ir"(offset % 8) |
| 75 | ); |
| 76 | /* *INDENT-ON* */ |
| 77 | #else |
| 78 | oldbit = ATOMIC_POST_OR(*addr, ((uint8_t) 1U << (offset % 8))); |
| 79 | #endif |
| 80 | return oldbit; |
| 81 | } |
| 82 | |
Robert Swiecki | 3bfc33c | 2016-03-14 18:12:41 +0100 | [diff] [blame] | 83 | extern void *util_Malloc(size_t sz); |
| 84 | |
Robert Swiecki | ac56a50 | 2016-03-17 18:59:46 +0100 | [diff] [blame] | 85 | extern void *util_Calloc(size_t sz); |
| 86 | |
| 87 | extern void *util_MMap(size_t sz); |
| 88 | |
Jagger | f49962d | 2016-07-21 22:49:54 +0200 | [diff] [blame] | 89 | extern void *util_Realloc(void *ptr, size_t sz); |
| 90 | |
Jagger | 6539b3a | 2016-04-06 02:32:50 +0200 | [diff] [blame] | 91 | extern char *util_StrDup(const char *s); |
| 92 | |
robert.swiecki@gmail.com | 90e9911 | 2015-02-15 02:05:14 +0000 | [diff] [blame] | 93 | extern uint64_t util_rndGet(uint64_t min, uint64_t max); |
robert.swiecki | 3bb518c | 2010-10-14 00:48:24 +0000 | [diff] [blame] | 94 | |
robert.swiecki@gmail.com | 89cc38c | 2015-02-23 02:52:08 +0000 | [diff] [blame] | 95 | extern void util_rndBuf(uint8_t * buf, size_t sz); |
| 96 | |
Jagger | c44a89a | 2016-09-25 16:42:44 +0200 | [diff] [blame] | 97 | extern uint64_t util_rnd64(void); |
| 98 | |
robert.swiecki@gmail.com | 9e2d43d | 2015-02-16 12:17:39 +0000 | [diff] [blame] | 99 | extern int util_ssnprintf(char *str, size_t size, const char *format, ...); |
| 100 | |
| 101 | extern int util_vssnprintf(char *str, size_t size, const char *format, va_list ap); |
robert.swiecki@gmail.com | e7190b9 | 2015-02-14 23:05:42 +0000 | [diff] [blame] | 102 | |
Robert Swiecki | 81c6a0d | 2015-09-08 15:43:20 +0200 | [diff] [blame] | 103 | extern void util_getLocalTime(const char *fmt, char *buf, size_t len, time_t tm); |
robert.swiecki | 3bb518c | 2010-10-14 00:48:24 +0000 | [diff] [blame] | 104 | |
robert.swiecki | aac72e0 | 2010-12-13 19:49:15 +0000 | [diff] [blame] | 105 | extern void util_nullifyStdio(void); |
robert.swiecki | 3bb518c | 2010-10-14 00:48:24 +0000 | [diff] [blame] | 106 | |
Robert Swiecki | 1e1b353 | 2016-03-10 18:43:28 +0100 | [diff] [blame] | 107 | extern bool util_redirectStdin(const char *inputFile); |
robert.swiecki | 3bb518c | 2010-10-14 00:48:24 +0000 | [diff] [blame] | 108 | |
groebert@google.com | 1bd4c21 | 2013-06-19 11:13:56 +0000 | [diff] [blame] | 109 | extern uint64_t util_hash(const char *buf, size_t len); |
| 110 | |
robert.swiecki@gmail.com | 3213a11 | 2015-03-12 01:42:02 +0000 | [diff] [blame] | 111 | extern int64_t util_timeNowMillis(void); |
| 112 | |
Anestis Bechtsoudis | a16f70f | 2016-01-03 13:03:21 +0200 | [diff] [blame] | 113 | extern uint64_t util_getUINT32(const uint8_t * buf); |
Anestis Bechtsoudis | be0ac7b | 2015-12-26 15:38:47 +0200 | [diff] [blame] | 114 | extern uint64_t util_getUINT64(const uint8_t * buf); |
robert.swiecki@gmail.com | 549ff18 | 2015-02-28 22:38:00 +0000 | [diff] [blame] | 115 | |
Jagger | 6caced3 | 2016-03-10 21:42:23 +0100 | [diff] [blame] | 116 | extern void util_mutexLock(pthread_mutex_t * mutex, const char *func, int line); |
| 117 | extern void util_mutexUnlock(pthread_mutex_t * mutex, const char *func, int line); |
Jagger | 421aca8 | 2015-09-05 14:09:47 +0200 | [diff] [blame] | 118 | |
Anestis Bechtsoudis | d59af69 | 2015-09-21 15:15:05 +0300 | [diff] [blame] | 119 | extern int64_t fastArray64Search(uint64_t * array, size_t arraySz, uint64_t key); |
| 120 | |
Anestis Bechtsoudis | 413cb13 | 2016-02-07 12:59:00 +0200 | [diff] [blame] | 121 | extern bool util_isANumber(const char *s); |
| 122 | |
Jagger | d619ed1 | 2016-09-22 04:08:49 +0200 | [diff] [blame] | 123 | extern size_t util_decodeCString(char *s); |
Robert Swiecki | fe8d289 | 2016-09-13 17:57:27 +0200 | [diff] [blame] | 124 | |
robert.swiecki | 3bb518c | 2010-10-14 00:48:24 +0000 | [diff] [blame] | 125 | #endif |