1#!/bin/sh
2# Copyright (c) Facebook, Inc. and its affiliates. All Rights Reserved.
3#
4# Set environment variables so that we can compile rocksdb using
5# fbcode settings.  It uses the latest g++ and clang compilers and also
6# uses jemalloc
7# Environment variables that change the behavior of this script:
8# PIC_BUILD -- if true, it will only take pic versions of libraries from fbcode. libraries that don't have pic variant will not be included
9
10
11BASEDIR=`dirname $BASH_SOURCE`
12source "$BASEDIR/dependencies.sh"
13
14CFLAGS=""
15
16# libgcc
17LIBGCC_INCLUDE="$LIBGCC_BASE/include"
18LIBGCC_LIBS=" -L $LIBGCC_BASE/lib"
19
20# glibc
21GLIBC_INCLUDE="$GLIBC_BASE/include"
22GLIBC_LIBS=" -L $GLIBC_BASE/lib"
23
24# snappy
25SNAPPY_INCLUDE=" -I $SNAPPY_BASE/include/"
26if test -z $PIC_BUILD; then
27  SNAPPY_LIBS=" $SNAPPY_BASE/lib/libsnappy.a"
28else
29  SNAPPY_LIBS=" $SNAPPY_BASE/lib/libsnappy_pic.a"
30fi
31CFLAGS+=" -DSNAPPY"
32
33if test -z $PIC_BUILD; then
34  # location of zlib headers and libraries
35  ZLIB_INCLUDE=" -I $ZLIB_BASE/include/"
36  ZLIB_LIBS=" $ZLIB_BASE/lib/libz.a"
37  CFLAGS+=" -DZLIB"
38
39  # location of bzip headers and libraries
40  BZIP_INCLUDE=" -I $BZIP2_BASE/include/"
41  BZIP_LIBS=" $BZIP2_BASE/lib/libbz2.a"
42  CFLAGS+=" -DBZIP2"
43
44  LZ4_INCLUDE=" -I $LZ4_BASE/include/"
45  LZ4_LIBS=" $LZ4_BASE/lib/liblz4.a"
46  CFLAGS+=" -DLZ4"
47fi
48
49ZSTD_INCLUDE=" -I $ZSTD_BASE/include/"
50if test -z $PIC_BUILD; then
51  ZSTD_LIBS=" $ZSTD_BASE/lib/libzstd.a"
52else
53  ZSTD_LIBS=" $ZSTD_BASE/lib/libzstd_pic.a"
54fi
55CFLAGS+=" -DZSTD -DZSTD_STATIC_LINKING_ONLY"
56
57# location of gflags headers and libraries
58GFLAGS_INCLUDE=" -I $GFLAGS_BASE/include/"
59if test -z $PIC_BUILD; then
60  GFLAGS_LIBS=" $GFLAGS_BASE/lib/libgflags.a"
61else
62  GFLAGS_LIBS=" $GFLAGS_BASE/lib/libgflags_pic.a"
63fi
64CFLAGS+=" -DGFLAGS=gflags"
65
66# location of jemalloc
67JEMALLOC_INCLUDE=" -I $JEMALLOC_BASE/include/"
68JEMALLOC_LIB=" $JEMALLOC_BASE/lib/libjemalloc.a"
69
70if test -z $PIC_BUILD; then
71  # location of numa
72  NUMA_INCLUDE=" -I $NUMA_BASE/include/"
73  NUMA_LIB=" $NUMA_BASE/lib/libnuma.a"
74  CFLAGS+=" -DNUMA"
75
76  # location of libunwind
77  LIBUNWIND="$LIBUNWIND_BASE/lib/libunwind.a"
78fi
79
80# location of TBB
81TBB_INCLUDE=" -isystem $TBB_BASE/include/"
82if test -z $PIC_BUILD; then
83  TBB_LIBS="$TBB_BASE/lib/libtbb.a"
84else
85  TBB_LIBS="$TBB_BASE/lib/libtbb_pic.a"
86fi
87CFLAGS+=" -DTBB"
88
89test "$USE_SSE" || USE_SSE=1
90export USE_SSE
91test "$PORTABLE" || PORTABLE=1
92export PORTABLE
93
94BINUTILS="$BINUTILS_BASE/bin"
95AR="$BINUTILS/ar"
96
97DEPS_INCLUDE="$SNAPPY_INCLUDE $ZLIB_INCLUDE $BZIP_INCLUDE $LZ4_INCLUDE $ZSTD_INCLUDE $GFLAGS_INCLUDE $NUMA_INCLUDE $TBB_INCLUDE"
98
99STDLIBS="-L $GCC_BASE/lib64"
100
101CLANG_BIN="$CLANG_BASE/bin"
102CLANG_LIB="$CLANG_BASE/lib"
103CLANG_SRC="$CLANG_BASE/../../src"
104
105CLANG_ANALYZER="$CLANG_BIN/clang++"
106CLANG_SCAN_BUILD="$CLANG_SRC/llvm/tools/clang/tools/scan-build/bin/scan-build"
107
108if [ -z "$USE_CLANG" ]; then
109  # gcc
110  CC="$GCC_BASE/bin/gcc"
111  CXX="$GCC_BASE/bin/g++"
112  AR="$GCC_BASE/bin/gcc-ar"
113
114  CFLAGS+=" -B$BINUTILS/gold"
115  CFLAGS+=" -isystem $GLIBC_INCLUDE"
116  CFLAGS+=" -isystem $LIBGCC_INCLUDE"
117  JEMALLOC=1
118else
119  # clang
120  CLANG_INCLUDE="$CLANG_LIB/clang/stable/include"
121  CC="$CLANG_BIN/clang"
122  CXX="$CLANG_BIN/clang++"
123  AR="$CLANG_BIN/llvm-ar"
124
125  KERNEL_HEADERS_INCLUDE="$KERNEL_HEADERS_BASE/include"
126
127  CFLAGS+=" -B$BINUTILS/gold -nostdinc -nostdlib"
128  CFLAGS+=" -isystem $LIBGCC_BASE/include/c++/5.x "
129  CFLAGS+=" -isystem $LIBGCC_BASE/include/c++/5.x/x86_64-facebook-linux "
130  CFLAGS+=" -isystem $GLIBC_INCLUDE"
131  CFLAGS+=" -isystem $LIBGCC_INCLUDE"
132  CFLAGS+=" -isystem $CLANG_INCLUDE"
133  CFLAGS+=" -isystem $KERNEL_HEADERS_INCLUDE/linux "
134  CFLAGS+=" -isystem $KERNEL_HEADERS_INCLUDE "
135  CFLAGS+=" -Wno-expansion-to-defined "
136  CXXFLAGS="-nostdinc++"
137fi
138
139CFLAGS+=" $DEPS_INCLUDE"
140CFLAGS+=" -DROCKSDB_PLATFORM_POSIX -DROCKSDB_LIB_IO_POSIX -DROCKSDB_FALLOCATE_PRESENT -DROCKSDB_MALLOC_USABLE_SIZE -DROCKSDB_RANGESYNC_PRESENT -DROCKSDB_SCHED_GETCPU_PRESENT -DROCKSDB_SUPPORT_THREAD_LOCAL -DHAVE_SSE42"
141CXXFLAGS+=" $CFLAGS"
142
143EXEC_LDFLAGS=" $SNAPPY_LIBS $ZLIB_LIBS $BZIP_LIBS $LZ4_LIBS $ZSTD_LIBS $GFLAGS_LIBS $NUMA_LIB $TBB_LIBS"
144EXEC_LDFLAGS+=" -B$BINUTILS/gold"
145EXEC_LDFLAGS+=" -Wl,--dynamic-linker,/usr/local/fbcode/gcc-5-glibc-2.23/lib/ld.so"
146EXEC_LDFLAGS+=" $LIBUNWIND"
147EXEC_LDFLAGS+=" -Wl,-rpath=/usr/local/fbcode/gcc-5-glibc-2.23/lib"
148# required by libtbb
149EXEC_LDFLAGS+=" -ldl"
150
151PLATFORM_LDFLAGS="$LIBGCC_LIBS $GLIBC_LIBS $STDLIBS -lgcc -lstdc++"
152
153EXEC_LDFLAGS_SHARED="$SNAPPY_LIBS $ZLIB_LIBS $BZIP_LIBS $LZ4_LIBS $ZSTD_LIBS $GFLAGS_LIBS $TBB_LIBS"
154
155VALGRIND_VER="$VALGRIND_BASE/bin/"
156
157LUA_PATH="$LUA_BASE"
158
159if test -z $PIC_BUILD; then
160  LUA_LIB=" $LUA_PATH/lib/liblua.a"
161else
162  LUA_LIB=" $LUA_PATH/lib/liblua_pic.a"
163fi
164
165USE_FOLLY_DISTRIBUTED_MUTEX=1
166
167export CC CXX AR CFLAGS CXXFLAGS EXEC_LDFLAGS EXEC_LDFLAGS_SHARED VALGRIND_VER JEMALLOC_LIB JEMALLOC_INCLUDE CLANG_ANALYZER CLANG_SCAN_BUILD LUA_PATH LUA_LIB
168