util.c - platform/external/honggfuzz - Gitiles

 /*
  *
  * honggfuzz - utilities
  * -----------------------------------------
  *
  * Author: Robert Swiecki <swiecki@google.com>
  *
  * Copyright 2010-2015 by Google Inc. All Rights Reserved.
  *
  * Licensed under the Apache License, Version 2.0 (the "License"); you may
  * not use this file except in compliance with the License. You may obtain
  * a copy of the License at
  *
  * http://www.apache.org/licenses/LICENSE-2.0
  *
  * Unless required by applicable law or agreed to in writing, software
  * distributed under the License is distributed on an "AS IS" BASIS,
  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or
  * implied. See the License for the specific language governing
  * permissions and limitations under the License.
  *
  */

 #include <fcntl.h>
 #include <inttypes.h>
 #include <math.h>
 #include <stdarg.h>
 #include <stdint.h>
 #include <stdio.h>
 #include <stdlib.h>
 #include <string.h>
 #include <sys/mman.h>
 #include <sys/stat.h>
 #include <sys/time.h>
 #include <sys/types.h>
 #include <time.h>
 #include <unistd.h>
 #include <ctype.h>

 #include "common.h"
 #include "files.h"
 #include "log.h"

 void *util_Malloc(size_t sz)
 {
     void *p = malloc(sz);
     if (p == NULL) {
         LOG_F("malloc(size='%zu')", sz);
     }
     return p;
 }

 void *util_Calloc(size_t sz)
 {
     void *p = util_Malloc(sz);
     memset(p, '\0', sz);
     return p;
 }

 void *util_MMap(size_t sz)
 {
     void *p = mmap(NULL, sz, PROT_READ | PROT_WRITE, MAP_ANON | MAP_PRIVATE | MAP_NORESERVE, -1, 0);
     if (p == MAP_FAILED) {
         LOG_F("mmap(size='%zu')", sz);
     }
     return p;
 }

 char *util_StrDup(const char *s)
 {
     char *ret = strdup(s);
     if (ret == NULL) {
         LOG_F("strdup(size=%zu)", strlen(s));
     }
     return ret;
 }

 static int util_urandomFd = -1;
 static __thread uint64_t rndX;
 static __thread uint64_t rndIni = false;

 uint64_t util_rndGet(uint64_t min, uint64_t max)
 {
     if (min > max) {
         LOG_F("min:%" PRIu64 " > max:%" PRIu64, min, max);
     }

     if (util_urandomFd == -1) {
         if ((util_urandomFd = open("/dev/urandom", O_RDONLY | O_CLOEXEC)) == -1) {
             PLOG_F("Couldn't open /dev/urandom for writing");
         }
     }

     if (rndIni == false) {
         if (files_readFromFd(util_urandomFd, (uint8_t *) & rndX, sizeof(rndX)) != sizeof(rndX)) {
             PLOG_F("Couldn't read '%zu' bytes from /dev/urandom", sizeof(rndX));
         }
         rndIni = true;
     }

     /* MMIX LCG PRNG */
     static const uint64_t a = 6364136223846793005ULL;
     static const uint64_t c = 1442695040888963407ULL;

     rndX = (a * rndX + c);

     return ((rndX % (max - min + 1)) + min);
 }

 void util_rndBuf(uint8_t * buf, size_t sz)
 {
     /* MMIX LCG PRNG */
     static const uint64_t a = 6364136223846793005ULL;
     static const uint64_t c = 1442695040888963407ULL;
     uint64_t x = util_rndGet(0, 1ULL << 62);

     for (size_t i = 0; i < sz; i++) {
         x = (a * x + c);
         buf[i] = (uint8_t) ((x & 0xFF0000) > 16);
     }
 }

 /*
  * Function has variable length stack size, although already we know it's invoked
  * with relatively small sizes (max is _HF_REPORT_SIZE), thus safe to silent warning.
  */
 #pragma GCC diagnostic push
 #pragma GCC diagnostic ignored "-Wframe-larger-than="
 int util_vssnprintf(char *str, size_t size, const char *format, va_list ap)
 {
     char buf1[size];
     char buf2[size];

     strncpy(buf1, str, size);

     vsnprintf(buf2, size, format, ap);

     return snprintf(str, size, "%s%s", buf1, buf2);
 }

 int util_ssnprintf(char *str, size_t size, const char *format, ...)
 {
     char buf1[size];
     char buf2[size];

     strncpy(buf1, str, size);

     va_list args;
     va_start(args, format);
     vsnprintf(buf2, size, format, args);
     va_end(args);

     return snprintf(str, size, "%s%s", buf1, buf2);
 }

 #pragma GCC diagnostic pop      /* EOF diagnostic ignored "-Wstack-usage=" */

 void util_getLocalTime(const char *fmt, char *buf, size_t len, time_t tm)
 {
     struct tm ltime;
     localtime_r(&tm, &ltime);
     if (strftime(buf, len, fmt, &ltime) < 1) {
         snprintf(buf, len, "[date fetch error]");
     }
 }

 void util_nullifyStdio(void)
 {
     int fd = open("/dev/null", O_RDWR);

     if (fd == -1) {
         PLOG_E("Couldn't open '/dev/null'");
         return;
     }

     dup2(fd, 0);
     dup2(fd, 1);
     dup2(fd, 2);

     if (fd > 2) {
         close(fd);
     }
 }

 bool util_redirectStdin(const char *inputFile)
 {
     int fd = open(inputFile, O_RDONLY);

     if (fd == -1) {
         PLOG_E("Couldn't open '%s'", inputFile);
         return false;
     }

     dup2(fd, 0);
     if (fd != 0) {
         close(fd);
     }

     return true;
 }

 /*
  * This is not a cryptographically secure hash
  */
 uint64_t util_hash(const char *buf, size_t len)
 {
     uint64_t ret = 0;

     for (size_t i = 0; i < len; i++) {
         ret += buf[i];
         ret += (ret << 10);
         ret ^= (ret >> 6);
     }

     return ret;
 }

 int64_t util_timeNowMillis(void)
 {
     struct timeval tv;
     if (gettimeofday(&tv, NULL) == -1) {
         PLOG_F("gettimeofday()");
     }

     return (((int64_t) tv.tv_sec * 1000LL) + ((int64_t) tv.tv_usec / 1000LL));
 }

 uint64_t util_getUINT32(const uint8_t * buf)
 {
     uint32_t r;
     memcpy(&r, buf, sizeof(r));
     return (uint64_t) r;
 }

 uint64_t util_getUINT64(const uint8_t * buf)
 {
     uint64_t r;
     memcpy(&r, buf, sizeof(r));
     return r;
 }

 void util_mutexLock(pthread_mutex_t * mutex, const char *func, int line)
 {
     if (pthread_mutex_lock(mutex)) {
         PLOG_F("%s():%d pthread_mutex_lock(%p)", func, line, (void *)mutex);
     }
 }

 void util_mutexUnlock(pthread_mutex_t * mutex, const char *func, int line)
 {
     if (pthread_mutex_unlock(mutex)) {
         PLOG_F("%s():%d pthread_mutex_unlock(%p)", func, line, (void *)mutex);
     }
 }

 int64_t fastArray64Search(uint64_t * array, size_t arraySz, uint64_t key)
 {
     size_t low = 0;
     size_t high = arraySz - 1;
     size_t mid;

     while (array[high] != array[low] && key >= array[low] && key <= array[high]) {
         mid = low + (key - array[low]) * ((high - low) / (array[high] - array[low]));

         if (array[mid] < key) {
             low = mid + 1;
         } else if (key < array[mid]) {
             high = mid - 1;
         } else {
             return mid;
         }
     }

     if (key == array[low]) {
         return low;
     } else {
         return -1;
     }
 }

 bool util_isANumber(const char *s)
 {
     if (!isdigit(s[0])) {
         return false;
     }
     for (int i = 0; s[i]; s++) {
         if (!isdigit(s[i]) && s[i] != 'x') {
             return false;
         }
     }
     return true;
 }
	/*
	*
	* honggfuzz - utilities
	* -----------------------------------------
	*
	* Author: Robert Swiecki <swiecki@google.com>
	*
	* Copyright 2010-2015 by Google Inc. All Rights Reserved.
	*
	* Licensed under the Apache License, Version 2.0 (the "License"); you may
	* not use this file except in compliance with the License. You may obtain
	* a copy of the License at
	*
	* http://www.apache.org/licenses/LICENSE-2.0
	*
	* Unless required by applicable law or agreed to in writing, software
	* distributed under the License is distributed on an "AS IS" BASIS,
	* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or
	* implied. See the License for the specific language governing
	* permissions and limitations under the License.
	*
	*/

	#include <fcntl.h>
	#include <inttypes.h>
	#include <math.h>
	#include <stdarg.h>
	#include <stdint.h>
	#include <stdio.h>
	#include <stdlib.h>
	#include <string.h>
	#include <sys/mman.h>
	#include <sys/stat.h>
	#include <sys/time.h>
	#include <sys/types.h>
	#include <time.h>
	#include <unistd.h>
	#include <ctype.h>

	#include "common.h"
	#include "files.h"
	#include "log.h"

	void *util_Malloc(size_t sz)
	{
	void *p = malloc(sz);
	if (p == NULL) {
	LOG_F("malloc(size='%zu')", sz);
	}
	return p;
	}

	void *util_Calloc(size_t sz)
	{
	void *p = util_Malloc(sz);
	memset(p, '\0', sz);
	return p;
	}

	void *util_MMap(size_t sz)
	{
	void *p = mmap(NULL, sz, PROT_READ \| PROT_WRITE, MAP_ANON \| MAP_PRIVATE \| MAP_NORESERVE, -1, 0);
	if (p == MAP_FAILED) {
	LOG_F("mmap(size='%zu')", sz);
	}
	return p;
	}

	char util_StrDup(const char s)
	{
	char *ret = strdup(s);
	if (ret == NULL) {
	LOG_F("strdup(size=%zu)", strlen(s));
	}
	return ret;
	}

	static int util_urandomFd = -1;
	static __thread uint64_t rndX;
	static __thread uint64_t rndIni = false;

	uint64_t util_rndGet(uint64_t min, uint64_t max)
	{
	if (min > max) {
	LOG_F("min:%" PRIu64 " > max:%" PRIu64, min, max);
	}

	if (util_urandomFd == -1) {
	if ((util_urandomFd = open("/dev/urandom", O_RDONLY \| O_CLOEXEC)) == -1) {
	PLOG_F("Couldn't open /dev/urandom for writing");
	}
	}

	if (rndIni == false) {
	if (files_readFromFd(util_urandomFd, (uint8_t *) & rndX, sizeof(rndX)) != sizeof(rndX)) {
	PLOG_F("Couldn't read '%zu' bytes from /dev/urandom", sizeof(rndX));
	}
	rndIni = true;
	}

	/* MMIX LCG PRNG */
	static const uint64_t a = 6364136223846793005ULL;
	static const uint64_t c = 1442695040888963407ULL;

	rndX = (a * rndX + c);

	return ((rndX % (max - min + 1)) + min);
	}

	void util_rndBuf(uint8_t * buf, size_t sz)
	{
	/* MMIX LCG PRNG */
	static const uint64_t a = 6364136223846793005ULL;
	static const uint64_t c = 1442695040888963407ULL;
	uint64_t x = util_rndGet(0, 1ULL << 62);

	for (size_t i = 0; i < sz; i++) {
	x = (a * x + c);
	buf[i] = (uint8_t) ((x & 0xFF0000) > 16);
	}
	}

	/*
	* Function has variable length stack size, although already we know it's invoked
	* with relatively small sizes (max is _HF_REPORT_SIZE), thus safe to silent warning.
	*/
	#pragma GCC diagnostic push
	#pragma GCC diagnostic ignored "-Wframe-larger-than="
	int util_vssnprintf(char str, size_t size, const char format, va_list ap)
	{
	char buf1[size];
	char buf2[size];

	strncpy(buf1, str, size);

	vsnprintf(buf2, size, format, ap);

	return snprintf(str, size, "%s%s", buf1, buf2);
	}

	int util_ssnprintf(char str, size_t size, const char format, ...)
	{
	char buf1[size];
	char buf2[size];

	strncpy(buf1, str, size);

	va_list args;
	va_start(args, format);
	vsnprintf(buf2, size, format, args);
	va_end(args);

	return snprintf(str, size, "%s%s", buf1, buf2);
	}

	#pragma GCC diagnostic pop /* EOF diagnostic ignored "-Wstack-usage=" */

	void util_getLocalTime(const char fmt, char buf, size_t len, time_t tm)
	{
	struct tm ltime;
	localtime_r(&tm, &ltime);
	if (strftime(buf, len, fmt, &ltime) < 1) {
	snprintf(buf, len, "[date fetch error]");
	}
	}

	void util_nullifyStdio(void)
	{
	int fd = open("/dev/null", O_RDWR);

	if (fd == -1) {
	PLOG_E("Couldn't open '/dev/null'");
	return;
	}

	dup2(fd, 0);
	dup2(fd, 1);
	dup2(fd, 2);

	if (fd > 2) {
	close(fd);
	}
	}

	bool util_redirectStdin(const char *inputFile)
	{
	int fd = open(inputFile, O_RDONLY);

	if (fd == -1) {
	PLOG_E("Couldn't open '%s'", inputFile);
	return false;
	}

	dup2(fd, 0);
	if (fd != 0) {
	close(fd);
	}

	return true;
	}

	/*
	* This is not a cryptographically secure hash
	*/
	uint64_t util_hash(const char *buf, size_t len)
	{
	uint64_t ret = 0;

	for (size_t i = 0; i < len; i++) {
	ret += buf[i];
	ret += (ret << 10);
	ret ^= (ret >> 6);
	}

	return ret;
	}

	int64_t util_timeNowMillis(void)
	{
	struct timeval tv;
	if (gettimeofday(&tv, NULL) == -1) {
	PLOG_F("gettimeofday()");
	}

	return (((int64_t) tv.tv_sec * 1000LL) + ((int64_t) tv.tv_usec / 1000LL));
	}

	uint64_t util_getUINT32(const uint8_t * buf)
	{
	uint32_t r;
	memcpy(&r, buf, sizeof(r));
	return (uint64_t) r;
	}

	uint64_t util_getUINT64(const uint8_t * buf)
	{
	uint64_t r;
	memcpy(&r, buf, sizeof(r));
	return r;
	}

	void util_mutexLock(pthread_mutex_t * mutex, const char *func, int line)
	{
	if (pthread_mutex_lock(mutex)) {
	PLOG_F("%s():%d pthread_mutex_lock(%p)", func, line, (void *)mutex);
	}
	}

	void util_mutexUnlock(pthread_mutex_t * mutex, const char *func, int line)
	{
	if (pthread_mutex_unlock(mutex)) {
	PLOG_F("%s():%d pthread_mutex_unlock(%p)", func, line, (void *)mutex);
	}
	}

	int64_t fastArray64Search(uint64_t * array, size_t arraySz, uint64_t key)
	{
	size_t low = 0;
	size_t high = arraySz - 1;
	size_t mid;

	while (array[high] != array[low] && key >= array[low] && key <= array[high]) {
	mid = low + (key - array[low]) * ((high - low) / (array[high] - array[low]));

	if (array[mid] < key) {
	low = mid + 1;
	} else if (key < array[mid]) {
	high = mid - 1;
	} else {
	return mid;
	}
	}

	if (key == array[low]) {
	return low;
	} else {
	return -1;
	}
	}

	bool util_isANumber(const char *s)
	{
	if (!isdigit(s[0])) {
	return false;
	}
	for (int i = 0; s[i]; s++) {
	if (!isdigit(s[i]) && s[i] != 'x') {
	return false;
	}
	}
	return true;
	}