Skip to content

Commit

Permalink
Make this thing build outside AFL source code tree
Browse files Browse the repository at this point in the history
  • Loading branch information
Shnatsel committed Aug 23, 2018
1 parent c3d69d9 commit d4a415f
Show file tree
Hide file tree
Showing 3 changed files with 91 additions and 5 deletions.
4 changes: 1 addition & 3 deletions Makefile
Original file line number Diff line number Diff line change
Expand Up @@ -16,14 +16,12 @@
PREFIX ?= /usr/local
HELPER_PATH = $(PREFIX)/lib/afl

VERSION = $(shell grep '^\#define VERSION ' ../config.h | cut -d '"' -f2)

CFLAGS ?= -O3 -funroll-loops
CFLAGS += -Wall -D_FORTIFY_SOURCE=2 -g -Wno-pointer-sign

all: libdislocator.so

libdislocator.so: libdislocator.so.c ../config.h
libdislocator.so: libdislocator.so.c
$(CC) $(CFLAGS) -shared -fPIC $< -o $@ $(LDFLAGS)

.NOTPARALLEL: clean
Expand Down
6 changes: 4 additions & 2 deletions libdislocator.so.c
Original file line number Diff line number Diff line change
Expand Up @@ -25,8 +25,7 @@
#include <limits.h>
#include <sys/mman.h>

#include "../config.h"
#include "../types.h"
#include "./types.h"

#ifndef PAGE_SIZE
# define PAGE_SIZE 4096
Expand All @@ -36,6 +35,9 @@
# define MAP_ANONYMOUS MAP_ANON
#endif /* !MAP_ANONYMOUS */

/* Maximum allocator request size (keep well under INT_MAX): */
#define MAX_ALLOC 0x40000000

/* Error / message handling: */

#define DEBUGF(_x...) do { \
Expand Down
86 changes: 86 additions & 0 deletions types.h
Original file line number Diff line number Diff line change
@@ -0,0 +1,86 @@
/*
american fuzzy lop - type definitions and minor macros
------------------------------------------------------
Written and maintained by Michal Zalewski <lcamtuf@google.com>
Copyright 2013, 2014, 2015 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
*/

#ifndef _HAVE_TYPES_H
#define _HAVE_TYPES_H

#include <stdint.h>
#include <stdlib.h>

typedef uint8_t u8;
typedef uint16_t u16;
typedef uint32_t u32;

/*
Ugh. There is an unintended compiler / glibc #include glitch caused by
combining the u64 type an %llu in format strings, necessitating a workaround.
In essence, the compiler is always looking for 'unsigned long long' for %llu.
On 32-bit systems, the u64 type (aliased to uint64_t) is expanded to
'unsigned long long' in <bits/types.h>, so everything checks out.
But on 64-bit systems, it is #ifdef'ed in the same file as 'unsigned long'.
Now, it only happens in circumstances where the type happens to have the
expected bit width, *but* the compiler does not know that... and complains
about 'unsigned long' being unsafe to pass to %llu.
*/

#ifdef __x86_64__
typedef unsigned long long u64;
#else
typedef uint64_t u64;
#endif /* ^__x86_64__ */

typedef int8_t s8;
typedef int16_t s16;
typedef int32_t s32;
typedef int64_t s64;

#ifndef MIN
# define MIN(_a,_b) ((_a) > (_b) ? (_b) : (_a))
# define MAX(_a,_b) ((_a) > (_b) ? (_a) : (_b))
#endif /* !MIN */

#define SWAP16(_x) ({ \
u16 _ret = (_x); \
(u16)((_ret << 8) | (_ret >> 8)); \
})

#define SWAP32(_x) ({ \
u32 _ret = (_x); \
(u32)((_ret << 24) | (_ret >> 24) | \
((_ret << 8) & 0x00FF0000) | \
((_ret >> 8) & 0x0000FF00)); \
})

#ifdef AFL_LLVM_PASS
# define AFL_R(x) (random() % (x))
#else
# define R(x) (random() % (x))
#endif /* ^AFL_LLVM_PASS */

#define STRINGIFY_INTERNAL(x) #x
#define STRINGIFY(x) STRINGIFY_INTERNAL(x)

#define MEM_BARRIER() \
asm volatile("" ::: "memory")

#define likely(_x) __builtin_expect(!!(_x), 1)
#define unlikely(_x) __builtin_expect(!!(_x), 0)

#endif /* ! _HAVE_TYPES_H */

0 comments on commit d4a415f

Please sign in to comment.