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 | |
robert.swiecki@gmail.com | 26a9ab7 | 2015-02-22 13:41:18 +0000 | [diff] [blame] | 41 | static int util_urandomFd = -1; |
robert.swiecki@gmail.com | 180263f | 2015-02-13 13:57:15 +0000 | [diff] [blame] | 42 | |
robert.swiecki@gmail.com | 90e9911 | 2015-02-15 02:05:14 +0000 | [diff] [blame] | 43 | uint64_t util_rndGet(uint64_t min, uint64_t max) |
robert.swiecki | 3bb518c | 2010-10-14 00:48:24 +0000 | [diff] [blame] | 44 | { |
robert.swiecki@gmail.com | 26a9ab7 | 2015-02-22 13:41:18 +0000 | [diff] [blame] | 45 | if (util_urandomFd == -1) { |
| 46 | if ((util_urandomFd = open("/dev/urandom", O_RDONLY)) == -1) { |
| 47 | LOGMSG_P(l_FATAL, "Couldn't open /dev/urandom for writing"); |
| 48 | } |
robert.swiecki@gmail.com | 180263f | 2015-02-13 13:57:15 +0000 | [diff] [blame] | 49 | } |
robert.swiecki | 3bb518c | 2010-10-14 00:48:24 +0000 | [diff] [blame] | 50 | |
robert.swiecki@gmail.com | 90e9911 | 2015-02-15 02:05:14 +0000 | [diff] [blame] | 51 | uint64_t rnd; |
robert.swiecki@gmail.com | 26a9ab7 | 2015-02-22 13:41:18 +0000 | [diff] [blame] | 52 | if (files_readFromFd(util_urandomFd, (uint8_t *) & rnd, sizeof(rnd)) == false) { |
| 53 | LOGMSG_P(l_FATAL, "Failed reading from /dev/urandom"); |
robert.swiecki@gmail.com | 4be12dc | 2015-02-13 14:03:53 +0000 | [diff] [blame] | 54 | } |
robert.swiecki@gmail.com | 180263f | 2015-02-13 13:57:15 +0000 | [diff] [blame] | 55 | |
robert.swiecki@gmail.com | bce825a | 2015-02-13 23:26:07 +0000 | [diff] [blame] | 56 | if (min > max) { |
| 57 | LOGMSG(l_FATAL, "min:%d > max:%d", min, max); |
| 58 | } |
| 59 | |
robert.swiecki@gmail.com | b038337 | 2015-02-13 14:17:27 +0000 | [diff] [blame] | 60 | return ((rnd % (max - min + 1)) + min); |
robert.swiecki | 3bb518c | 2010-10-14 00:48:24 +0000 | [diff] [blame] | 61 | } |
| 62 | |
robert.swiecki@gmail.com | 89cc38c | 2015-02-23 02:52:08 +0000 | [diff] [blame] | 63 | void util_rndBuf(uint8_t * buf, size_t sz) |
| 64 | { |
robert.swiecki@gmail.com | ace4086 | 2015-03-08 07:09:56 +0000 | [diff] [blame] | 65 | /* MMIX LCG PRNG */ |
| 66 | uint64_t a = 6364136223846793005ULL; |
| 67 | uint64_t c = 1442695040888963407ULL; |
robert.swiecki@gmail.com | e6b1552 | 2015-03-27 00:33:42 +0000 | [diff] [blame] | 68 | uint64_t x = util_rndGet(0, 1ULL << 60); |
robert.swiecki@gmail.com | 89cc38c | 2015-02-23 02:52:08 +0000 | [diff] [blame] | 69 | |
robert.swiecki@gmail.com | 08174c7 | 2015-02-28 14:38:06 +0000 | [diff] [blame] | 70 | for (size_t i = 0; i < sz; i++) { |
robert.swiecki@gmail.com | ace4086 | 2015-03-08 07:09:56 +0000 | [diff] [blame] | 71 | x = (a * x + c); |
| 72 | buf[i] = (uint8_t) ((x >> 32) & 0xFF); |
robert.swiecki@gmail.com | 89cc38c | 2015-02-23 02:52:08 +0000 | [diff] [blame] | 73 | } |
| 74 | |
| 75 | return; |
| 76 | } |
| 77 | |
robert.swiecki@gmail.com | 9e2d43d | 2015-02-16 12:17:39 +0000 | [diff] [blame] | 78 | int util_vssnprintf(char *str, size_t size, const char *format, va_list ap) |
| 79 | { |
| 80 | char buf1[size]; |
| 81 | char buf2[size]; |
| 82 | |
| 83 | strncpy(buf1, str, size); |
| 84 | |
| 85 | vsnprintf(buf2, size, format, ap); |
| 86 | |
| 87 | return snprintf(str, size, "%s%s", buf1, buf2); |
| 88 | } |
| 89 | |
| 90 | 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] | 91 | { |
| 92 | char buf1[size]; |
| 93 | char buf2[size]; |
| 94 | |
| 95 | strncpy(buf1, str, size); |
| 96 | |
| 97 | va_list args; |
| 98 | va_start(args, format); |
| 99 | vsnprintf(buf2, size, format, args); |
| 100 | va_end(args); |
| 101 | |
robert.swiecki@gmail.com | 9e2d43d | 2015-02-16 12:17:39 +0000 | [diff] [blame] | 102 | return snprintf(str, size, "%s%s", buf1, buf2); |
robert.swiecki@gmail.com | e7190b9 | 2015-02-14 23:05:42 +0000 | [diff] [blame] | 103 | } |
| 104 | |
robert.swiecki | 3bb518c | 2010-10-14 00:48:24 +0000 | [diff] [blame] | 105 | void util_getLocalTime(const char *fmt, char *buf, size_t len) |
| 106 | { |
| 107 | struct tm ltime; |
| 108 | |
| 109 | time_t t = time(NULL); |
| 110 | |
| 111 | localtime_r(&t, <ime); |
| 112 | strftime(buf, len, fmt, <ime); |
| 113 | } |
| 114 | |
robert.swiecki | 40499ff | 2010-12-13 19:47:08 +0000 | [diff] [blame] | 115 | void util_nullifyStdio(void) |
robert.swiecki | 3bb518c | 2010-10-14 00:48:24 +0000 | [diff] [blame] | 116 | { |
| 117 | int fd = open("/dev/null", O_RDWR); |
| 118 | |
| 119 | if (fd == -1) { |
| 120 | LOGMSG_P(l_ERROR, "Couldn't open '/dev/null'"); |
| 121 | return; |
| 122 | } |
| 123 | |
| 124 | dup2(fd, 0); |
| 125 | dup2(fd, 1); |
| 126 | dup2(fd, 2); |
| 127 | |
| 128 | if (fd > 2) { |
| 129 | close(fd); |
| 130 | } |
| 131 | |
| 132 | return; |
| 133 | } |
| 134 | |
| 135 | bool util_redirectStdin(char *inputFile) |
| 136 | { |
| 137 | int fd = open(inputFile, O_RDONLY); |
| 138 | |
| 139 | if (fd == -1) { |
| 140 | LOGMSG_P(l_ERROR, "Couldn't open '%s'", inputFile); |
| 141 | return false; |
| 142 | } |
| 143 | |
| 144 | dup2(fd, 0); |
| 145 | if (fd != 0) { |
| 146 | close(fd); |
| 147 | } |
| 148 | |
| 149 | return true; |
| 150 | } |
| 151 | |
robert.swiecki | 40499ff | 2010-12-13 19:47:08 +0000 | [diff] [blame] | 152 | void util_recoverStdio(void) |
robert.swiecki | 3bb518c | 2010-10-14 00:48:24 +0000 | [diff] [blame] | 153 | { |
| 154 | int fd = open("/dev/tty", O_RDWR); |
| 155 | |
| 156 | if (fd == -1) { |
| 157 | LOGMSG_P(l_ERROR, "Couldn't open '/dev/tty'"); |
| 158 | return; |
| 159 | } |
| 160 | |
| 161 | dup2(fd, 0); |
| 162 | dup2(fd, 1); |
| 163 | dup2(fd, 2); |
| 164 | |
robert.swiecki@gmail.com | df28694 | 2015-04-15 21:45:25 +0000 | [diff] [blame] | 165 | if (tcsetpgrp(fd, getpid()) == -1) { |
| 166 | LOGMSG_P(l_WARN, "tcsetpgrp(%d) failed", getpid()); |
| 167 | } |
| 168 | |
robert.swiecki | 3bb518c | 2010-10-14 00:48:24 +0000 | [diff] [blame] | 169 | if (fd > 2) { |
| 170 | close(fd); |
| 171 | } |
robert.swiecki | 3bb518c | 2010-10-14 00:48:24 +0000 | [diff] [blame] | 172 | return; |
| 173 | } |
groebert@google.com | 1bd4c21 | 2013-06-19 11:13:56 +0000 | [diff] [blame] | 174 | |
robert.swiecki@gmail.com | 772b33d | 2015-02-14 20:35:00 +0000 | [diff] [blame] | 175 | /* |
robert.swiecki@gmail.com | 3b630b4 | 2015-02-16 10:53:53 +0000 | [diff] [blame] | 176 | * This is not a cryptographically secure hash |
robert.swiecki@gmail.com | 772b33d | 2015-02-14 20:35:00 +0000 | [diff] [blame] | 177 | */ |
groebert@google.com | 1bd4c21 | 2013-06-19 11:13:56 +0000 | [diff] [blame] | 178 | extern uint64_t util_hash(const char *buf, size_t len) |
| 179 | { |
| 180 | uint64_t ret = 0; |
| 181 | |
| 182 | for (size_t i = 0; i < len; i++) { |
| 183 | ret += buf[i]; |
| 184 | ret += (ret << 10); |
| 185 | ret ^= (ret >> 6); |
| 186 | } |
| 187 | |
| 188 | return ret; |
| 189 | } |
robert.swiecki@gmail.com | 549ff18 | 2015-02-28 22:38:00 +0000 | [diff] [blame] | 190 | |
robert.swiecki@gmail.com | 3213a11 | 2015-03-12 01:42:02 +0000 | [diff] [blame] | 191 | extern int64_t util_timeNowMillis(void) |
| 192 | { |
| 193 | struct timeval tv; |
| 194 | if (gettimeofday(&tv, NULL) == -1) { |
| 195 | LOGMSG_P(l_FATAL, "gettimeofday()"); |
| 196 | } |
| 197 | |
| 198 | return (((int64_t) tv.tv_sec * 1000LL) + ((int64_t) tv.tv_usec / 1000LL)); |
| 199 | } |
| 200 | |
robert.swiecki@gmail.com | 549ff18 | 2015-02-28 22:38:00 +0000 | [diff] [blame] | 201 | extern uint16_t util_ToFromBE16(uint16_t val) |
| 202 | { |
| 203 | #if __BYTE_ORDER == __BIG_ENDIAN |
| 204 | return val; |
| 205 | #elif __BYTE_ORDER == __LITTLE_ENDIAN |
robert.swiecki@gmail.com | f36c6cc | 2015-03-02 03:38:53 +0000 | [diff] [blame] | 206 | return __builtin_bswap16(val); |
robert.swiecki@gmail.com | 549ff18 | 2015-02-28 22:38:00 +0000 | [diff] [blame] | 207 | #else |
| 208 | #error "Unknown ENDIANESS" |
| 209 | #endif |
| 210 | } |
| 211 | |
| 212 | extern uint16_t util_ToFromLE16(uint16_t val) |
| 213 | { |
| 214 | #if __BYTE_ORDER == __BIG_ENDIAN |
robert.swiecki@gmail.com | f36c6cc | 2015-03-02 03:38:53 +0000 | [diff] [blame] | 215 | return __builtin_bswap16(val); |
robert.swiecki@gmail.com | 549ff18 | 2015-02-28 22:38:00 +0000 | [diff] [blame] | 216 | #elif __BYTE_ORDER == __LITTLE_ENDIAN |
| 217 | return val; |
| 218 | #else |
| 219 | #error "Unknown ENDIANESS" |
| 220 | #endif |
| 221 | } |
| 222 | |
| 223 | extern uint32_t util_ToFromBE32(uint32_t val) |
| 224 | { |
| 225 | #if __BYTE_ORDER == __BIG_ENDIAN |
| 226 | return val; |
| 227 | #elif __BYTE_ORDER == __LITTLE_ENDIAN |
robert.swiecki@gmail.com | f36c6cc | 2015-03-02 03:38:53 +0000 | [diff] [blame] | 228 | return __builtin_bswap32(val); |
robert.swiecki@gmail.com | 549ff18 | 2015-02-28 22:38:00 +0000 | [diff] [blame] | 229 | #else |
| 230 | #error "Unknown ENDIANESS" |
| 231 | #endif |
| 232 | } |
| 233 | |
| 234 | extern uint32_t util_ToFromLE32(uint32_t val) |
| 235 | { |
| 236 | #if __BYTE_ORDER == __BIG_ENDIAN |
robert.swiecki@gmail.com | f36c6cc | 2015-03-02 03:38:53 +0000 | [diff] [blame] | 237 | return __builtin_bswap32(val); |
robert.swiecki@gmail.com | 549ff18 | 2015-02-28 22:38:00 +0000 | [diff] [blame] | 238 | #elif __BYTE_ORDER == __LITTLE_ENDIAN |
| 239 | return val; |
| 240 | #else |
| 241 | #error "Unknown ENDIANESS" |
| 242 | #endif |
| 243 | } |