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 | |
robert.swiecki | 3bb518c | 2010-10-14 00:48:24 +0000 | [diff] [blame] | 24 | #include <fcntl.h> |
robert.swiecki | 3bb518c | 2010-10-14 00:48:24 +0000 | [diff] [blame] | 25 | #include <math.h> |
robert.swiecki@gmail.com | e7190b9 | 2015-02-14 23:05:42 +0000 | [diff] [blame] | 26 | #include <stdarg.h> |
robert.swiecki@gmail.com | ba85c3e | 2015-02-02 14:55:16 +0000 | [diff] [blame] | 27 | #include <stdint.h> |
robert.swiecki@gmail.com | e7190b9 | 2015-02-14 23:05:42 +0000 | [diff] [blame] | 28 | #include <stdio.h> |
robert.swiecki@gmail.com | ba85c3e | 2015-02-02 14:55:16 +0000 | [diff] [blame] | 29 | #include <stdlib.h> |
robert.swiecki@gmail.com | e7190b9 | 2015-02-14 23:05:42 +0000 | [diff] [blame] | 30 | #include <string.h> |
robert.swiecki@gmail.com | ba85c3e | 2015-02-02 14:55:16 +0000 | [diff] [blame] | 31 | #include <sys/stat.h> |
| 32 | #include <sys/time.h> |
| 33 | #include <sys/types.h> |
robert.swiecki | 3bb518c | 2010-10-14 00:48:24 +0000 | [diff] [blame] | 34 | #include <time.h> |
robert.swiecki@gmail.com | ba85c3e | 2015-02-02 14:55:16 +0000 | [diff] [blame] | 35 | #include <unistd.h> |
robert.swiecki | 3bb518c | 2010-10-14 00:48:24 +0000 | [diff] [blame] | 36 | |
| 37 | #include "common.h" |
robert.swiecki@gmail.com | 26a9ab7 | 2015-02-22 13:41:18 +0000 | [diff] [blame] | 38 | #include "files.h" |
robert.swiecki | 3bb518c | 2010-10-14 00:48:24 +0000 | [diff] [blame] | 39 | #include "log.h" |
| 40 | |
Anestis Bechtsoudis | 4f9bfd0 | 2015-09-10 17:31:58 +0300 | [diff] [blame] | 41 | #if defined(__builtin_bswap16) |
| 42 | #define SWAP16(x) __builtin_bswap16(x) |
| 43 | #else |
| 44 | #define SWAP16(x) ((x & 0xff) << 8) | ((x & 0xff00) >> 8) |
| 45 | #endif |
| 46 | |
| 47 | #if defined(__builtin_bswap32) |
| 48 | #define SWAP32(x) __builtin_bswap32(x) |
| 49 | #else |
| 50 | #define SWAP32(x) ((x & 0xff) << 24) | ((x & 0xff00) << 8) | ((x & 0xff0000) >> 8) | ((x & 0xff000000) >> 24) |
| 51 | #endif |
| 52 | |
robert.swiecki@gmail.com | 26a9ab7 | 2015-02-22 13:41:18 +0000 | [diff] [blame] | 53 | static int util_urandomFd = -1; |
robert.swiecki@gmail.com | 180263f | 2015-02-13 13:57:15 +0000 | [diff] [blame] | 54 | |
robert.swiecki@gmail.com | 90e9911 | 2015-02-15 02:05:14 +0000 | [diff] [blame] | 55 | uint64_t util_rndGet(uint64_t min, uint64_t max) |
robert.swiecki | 3bb518c | 2010-10-14 00:48:24 +0000 | [diff] [blame] | 56 | { |
robert.swiecki@gmail.com | 26a9ab7 | 2015-02-22 13:41:18 +0000 | [diff] [blame] | 57 | if (util_urandomFd == -1) { |
| 58 | if ((util_urandomFd = open("/dev/urandom", O_RDONLY)) == -1) { |
| 59 | LOGMSG_P(l_FATAL, "Couldn't open /dev/urandom for writing"); |
| 60 | } |
robert.swiecki@gmail.com | 180263f | 2015-02-13 13:57:15 +0000 | [diff] [blame] | 61 | } |
robert.swiecki | 3bb518c | 2010-10-14 00:48:24 +0000 | [diff] [blame] | 62 | |
robert.swiecki@gmail.com | 90e9911 | 2015-02-15 02:05:14 +0000 | [diff] [blame] | 63 | uint64_t rnd; |
robert.swiecki@gmail.com | 26a9ab7 | 2015-02-22 13:41:18 +0000 | [diff] [blame] | 64 | if (files_readFromFd(util_urandomFd, (uint8_t *) & rnd, sizeof(rnd)) == false) { |
| 65 | LOGMSG_P(l_FATAL, "Failed reading from /dev/urandom"); |
robert.swiecki@gmail.com | 4be12dc | 2015-02-13 14:03:53 +0000 | [diff] [blame] | 66 | } |
robert.swiecki@gmail.com | 180263f | 2015-02-13 13:57:15 +0000 | [diff] [blame] | 67 | |
robert.swiecki@gmail.com | bce825a | 2015-02-13 23:26:07 +0000 | [diff] [blame] | 68 | if (min > max) { |
| 69 | LOGMSG(l_FATAL, "min:%d > max:%d", min, max); |
| 70 | } |
| 71 | |
robert.swiecki@gmail.com | b038337 | 2015-02-13 14:17:27 +0000 | [diff] [blame] | 72 | return ((rnd % (max - min + 1)) + min); |
robert.swiecki | 3bb518c | 2010-10-14 00:48:24 +0000 | [diff] [blame] | 73 | } |
| 74 | |
robert.swiecki@gmail.com | 89cc38c | 2015-02-23 02:52:08 +0000 | [diff] [blame] | 75 | void util_rndBuf(uint8_t * buf, size_t sz) |
| 76 | { |
robert.swiecki@gmail.com | ace4086 | 2015-03-08 07:09:56 +0000 | [diff] [blame] | 77 | /* MMIX LCG PRNG */ |
| 78 | uint64_t a = 6364136223846793005ULL; |
| 79 | uint64_t c = 1442695040888963407ULL; |
robert.swiecki@gmail.com | e6b1552 | 2015-03-27 00:33:42 +0000 | [diff] [blame] | 80 | uint64_t x = util_rndGet(0, 1ULL << 60); |
robert.swiecki@gmail.com | 89cc38c | 2015-02-23 02:52:08 +0000 | [diff] [blame] | 81 | |
robert.swiecki@gmail.com | 08174c7 | 2015-02-28 14:38:06 +0000 | [diff] [blame] | 82 | for (size_t i = 0; i < sz; i++) { |
robert.swiecki@gmail.com | ace4086 | 2015-03-08 07:09:56 +0000 | [diff] [blame] | 83 | x = (a * x + c); |
| 84 | buf[i] = (uint8_t) ((x >> 32) & 0xFF); |
robert.swiecki@gmail.com | 89cc38c | 2015-02-23 02:52:08 +0000 | [diff] [blame] | 85 | } |
| 86 | |
| 87 | return; |
| 88 | } |
| 89 | |
robert.swiecki@gmail.com | 9e2d43d | 2015-02-16 12:17:39 +0000 | [diff] [blame] | 90 | int util_vssnprintf(char *str, size_t size, const char *format, va_list ap) |
| 91 | { |
| 92 | char buf1[size]; |
| 93 | char buf2[size]; |
| 94 | |
| 95 | strncpy(buf1, str, size); |
| 96 | |
| 97 | vsnprintf(buf2, size, format, ap); |
| 98 | |
| 99 | return snprintf(str, size, "%s%s", buf1, buf2); |
| 100 | } |
| 101 | |
| 102 | int util_ssnprintf(char *str, size_t size, const char *format, ...) |
robert.swiecki@gmail.com | e7190b9 | 2015-02-14 23:05:42 +0000 | [diff] [blame] | 103 | { |
| 104 | char buf1[size]; |
| 105 | char buf2[size]; |
| 106 | |
| 107 | strncpy(buf1, str, size); |
| 108 | |
| 109 | va_list args; |
| 110 | va_start(args, format); |
| 111 | vsnprintf(buf2, size, format, args); |
| 112 | va_end(args); |
| 113 | |
robert.swiecki@gmail.com | 9e2d43d | 2015-02-16 12:17:39 +0000 | [diff] [blame] | 114 | return snprintf(str, size, "%s%s", buf1, buf2); |
robert.swiecki@gmail.com | e7190b9 | 2015-02-14 23:05:42 +0000 | [diff] [blame] | 115 | } |
| 116 | |
Robert Swiecki | 81c6a0d | 2015-09-08 15:43:20 +0200 | [diff] [blame] | 117 | 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] | 118 | { |
| 119 | struct tm ltime; |
Robert Swiecki | 81c6a0d | 2015-09-08 15:43:20 +0200 | [diff] [blame] | 120 | localtime_r(&tm, <ime); |
| 121 | if (strftime(buf, len, fmt, <ime) < 1) { |
| 122 | snprintf(buf, len, "[date fetch error]"); |
| 123 | } |
robert.swiecki | 3bb518c | 2010-10-14 00:48:24 +0000 | [diff] [blame] | 124 | } |
| 125 | |
robert.swiecki | 40499ff | 2010-12-13 19:47:08 +0000 | [diff] [blame] | 126 | void util_nullifyStdio(void) |
robert.swiecki | 3bb518c | 2010-10-14 00:48:24 +0000 | [diff] [blame] | 127 | { |
| 128 | int fd = open("/dev/null", O_RDWR); |
| 129 | |
| 130 | if (fd == -1) { |
| 131 | LOGMSG_P(l_ERROR, "Couldn't open '/dev/null'"); |
| 132 | return; |
| 133 | } |
| 134 | |
| 135 | dup2(fd, 0); |
| 136 | dup2(fd, 1); |
| 137 | dup2(fd, 2); |
| 138 | |
| 139 | if (fd > 2) { |
| 140 | close(fd); |
| 141 | } |
| 142 | |
| 143 | return; |
| 144 | } |
| 145 | |
| 146 | bool util_redirectStdin(char *inputFile) |
| 147 | { |
| 148 | int fd = open(inputFile, O_RDONLY); |
| 149 | |
| 150 | if (fd == -1) { |
| 151 | LOGMSG_P(l_ERROR, "Couldn't open '%s'", inputFile); |
| 152 | return false; |
| 153 | } |
| 154 | |
| 155 | dup2(fd, 0); |
| 156 | if (fd != 0) { |
| 157 | close(fd); |
| 158 | } |
| 159 | |
| 160 | return true; |
| 161 | } |
| 162 | |
robert.swiecki | 40499ff | 2010-12-13 19:47:08 +0000 | [diff] [blame] | 163 | void util_recoverStdio(void) |
robert.swiecki | 3bb518c | 2010-10-14 00:48:24 +0000 | [diff] [blame] | 164 | { |
| 165 | int fd = open("/dev/tty", O_RDWR); |
| 166 | |
| 167 | if (fd == -1) { |
| 168 | LOGMSG_P(l_ERROR, "Couldn't open '/dev/tty'"); |
| 169 | return; |
| 170 | } |
| 171 | |
| 172 | dup2(fd, 0); |
| 173 | dup2(fd, 1); |
| 174 | dup2(fd, 2); |
| 175 | |
robert.swiecki@gmail.com | df28694 | 2015-04-15 21:45:25 +0000 | [diff] [blame] | 176 | if (tcsetpgrp(fd, getpid()) == -1) { |
| 177 | LOGMSG_P(l_WARN, "tcsetpgrp(%d) failed", getpid()); |
| 178 | } |
| 179 | |
robert.swiecki | 3bb518c | 2010-10-14 00:48:24 +0000 | [diff] [blame] | 180 | if (fd > 2) { |
| 181 | close(fd); |
| 182 | } |
robert.swiecki | 3bb518c | 2010-10-14 00:48:24 +0000 | [diff] [blame] | 183 | return; |
| 184 | } |
groebert@google.com | 1bd4c21 | 2013-06-19 11:13:56 +0000 | [diff] [blame] | 185 | |
robert.swiecki@gmail.com | 772b33d | 2015-02-14 20:35:00 +0000 | [diff] [blame] | 186 | /* |
robert.swiecki@gmail.com | 3b630b4 | 2015-02-16 10:53:53 +0000 | [diff] [blame] | 187 | * This is not a cryptographically secure hash |
robert.swiecki@gmail.com | 772b33d | 2015-02-14 20:35:00 +0000 | [diff] [blame] | 188 | */ |
groebert@google.com | 1bd4c21 | 2013-06-19 11:13:56 +0000 | [diff] [blame] | 189 | extern uint64_t util_hash(const char *buf, size_t len) |
| 190 | { |
| 191 | uint64_t ret = 0; |
| 192 | |
| 193 | for (size_t i = 0; i < len; i++) { |
| 194 | ret += buf[i]; |
| 195 | ret += (ret << 10); |
| 196 | ret ^= (ret >> 6); |
| 197 | } |
| 198 | |
| 199 | return ret; |
| 200 | } |
robert.swiecki@gmail.com | 549ff18 | 2015-02-28 22:38:00 +0000 | [diff] [blame] | 201 | |
robert.swiecki@gmail.com | 3213a11 | 2015-03-12 01:42:02 +0000 | [diff] [blame] | 202 | extern int64_t util_timeNowMillis(void) |
| 203 | { |
| 204 | struct timeval tv; |
| 205 | if (gettimeofday(&tv, NULL) == -1) { |
| 206 | LOGMSG_P(l_FATAL, "gettimeofday()"); |
| 207 | } |
| 208 | |
| 209 | return (((int64_t) tv.tv_sec * 1000LL) + ((int64_t) tv.tv_usec / 1000LL)); |
| 210 | } |
| 211 | |
robert.swiecki@gmail.com | 549ff18 | 2015-02-28 22:38:00 +0000 | [diff] [blame] | 212 | extern uint16_t util_ToFromBE16(uint16_t val) |
| 213 | { |
| 214 | #if __BYTE_ORDER == __BIG_ENDIAN |
| 215 | return val; |
| 216 | #elif __BYTE_ORDER == __LITTLE_ENDIAN |
Anestis Bechtsoudis | 4f9bfd0 | 2015-09-10 17:31:58 +0300 | [diff] [blame] | 217 | return SWAP16(val); |
robert.swiecki@gmail.com | 549ff18 | 2015-02-28 22:38:00 +0000 | [diff] [blame] | 218 | #else |
| 219 | #error "Unknown ENDIANESS" |
| 220 | #endif |
| 221 | } |
| 222 | |
| 223 | extern uint16_t util_ToFromLE16(uint16_t val) |
| 224 | { |
| 225 | #if __BYTE_ORDER == __BIG_ENDIAN |
Anestis Bechtsoudis | 4f9bfd0 | 2015-09-10 17:31:58 +0300 | [diff] [blame] | 226 | return SWAP16(val); |
robert.swiecki@gmail.com | 549ff18 | 2015-02-28 22:38:00 +0000 | [diff] [blame] | 227 | #elif __BYTE_ORDER == __LITTLE_ENDIAN |
| 228 | return val; |
| 229 | #else |
| 230 | #error "Unknown ENDIANESS" |
| 231 | #endif |
| 232 | } |
| 233 | |
| 234 | extern uint32_t util_ToFromBE32(uint32_t val) |
| 235 | { |
| 236 | #if __BYTE_ORDER == __BIG_ENDIAN |
| 237 | return val; |
| 238 | #elif __BYTE_ORDER == __LITTLE_ENDIAN |
Anestis Bechtsoudis | 4f9bfd0 | 2015-09-10 17:31:58 +0300 | [diff] [blame] | 239 | return SWAP32(val); |
robert.swiecki@gmail.com | 549ff18 | 2015-02-28 22:38:00 +0000 | [diff] [blame] | 240 | #else |
| 241 | #error "Unknown ENDIANESS" |
| 242 | #endif |
| 243 | } |
| 244 | |
| 245 | extern uint32_t util_ToFromLE32(uint32_t val) |
| 246 | { |
| 247 | #if __BYTE_ORDER == __BIG_ENDIAN |
Anestis Bechtsoudis | 4f9bfd0 | 2015-09-10 17:31:58 +0300 | [diff] [blame] | 248 | return SWAP32(val); |
robert.swiecki@gmail.com | 549ff18 | 2015-02-28 22:38:00 +0000 | [diff] [blame] | 249 | #elif __BYTE_ORDER == __LITTLE_ENDIAN |
| 250 | return val; |
| 251 | #else |
| 252 | #error "Unknown ENDIANESS" |
| 253 | #endif |
| 254 | } |
Jagger | 421aca8 | 2015-09-05 14:09:47 +0200 | [diff] [blame] | 255 | |
| 256 | extern void MX_LOCK(pthread_mutex_t * mutex) |
| 257 | { |
| 258 | if (pthread_mutex_lock(mutex)) { |
| 259 | LOGMSG_P(l_FATAL, "pthread_mutex_lock(%p)", mutex); |
| 260 | } |
| 261 | } |
| 262 | |
| 263 | extern void MX_UNLOCK(pthread_mutex_t * mutex) |
| 264 | { |
| 265 | if (pthread_mutex_unlock(mutex)) { |
| 266 | LOGMSG_P(l_FATAL, "pthread_mutex_unlock(%p)", mutex); |
| 267 | } |
| 268 | } |
Anestis Bechtsoudis | d59af69 | 2015-09-21 15:15:05 +0300 | [diff] [blame^] | 269 | |
| 270 | extern int64_t fastArray64Search(uint64_t * array, size_t arraySz, uint64_t key) |
| 271 | { |
| 272 | size_t low = 0; |
| 273 | size_t high = arraySz - 1; |
| 274 | size_t mid; |
| 275 | |
| 276 | while (array[high] != array[low] && key >= array[low] && key <= array[high]) { |
| 277 | mid = low + (key - array[low]) * ((high - low) / (array[high] - array[low])); |
| 278 | |
| 279 | if (array[mid] < key) { |
| 280 | low = mid + 1; |
| 281 | } else if (key < array[mid]) { |
| 282 | high = mid - 1; |
| 283 | } else { |
| 284 | return mid; |
| 285 | } |
| 286 | } |
| 287 | |
| 288 | if (key == array[low]) { |
| 289 | return low; |
| 290 | } else { |
| 291 | return -1; |
| 292 | } |
| 293 | } |