osquery-1/tools/provision.sh

#!/usr/bin/env bash

set -e

SCRIPT_DIR="$( cd "$( dirname "${BASH_SOURCE[0]}" )" && pwd )"
WORKING_DIR="$SCRIPT_DIR/../.sources"
export PATH="$PATH:/usr/local/bin"

. $SCRIPT_DIR/lib.sh


# cmake
# downloads: http://www.cmake.org/download/

function install_cmake() {
  if [ $OS = "centos" ] || [ $OS = "ubuntu" ] || [ $OS = "darwin" ]; then
    if [[ ! -f cmake-2.8.12.2.tar.gz ]]; then
      log "downloading the cmake source"
      wget http://www.cmake.org/files/v2.8/cmake-2.8.12.2.tar.gz
    fi
    if [[ ! -d cmake-2.8.12.2 ]]; then
      log "unpacking the cmake source"
      tar -xf cmake-2.8.12.2.tar.gz
    fi
    if [[ -f /usr/local/bin/cmake ]]; then
      log "cmake is already installed. skipping."
    else
      log "building cmake"
      pushd cmake-2.8.12.2 > /dev/null
      CC=clang CXX=clang++ ./configure
      make
      make install
      popd
    fi
  fi
}

function install_thrift() {
  if [[ -f /usr/local/lib/libthrift.a ]]; then
    log "thrift is installed. skipping."
    return
  fi
  if [[ ! -f 0.9.1.tar.gz ]]; then
    wget https://github.com/apache/thrift/archive/0.9.1.tar.gz
  fi
  if [[ ! -d thrift-0.9.1 ]]; then
    tar -xf 0.9.1.tar.gz
  fi
  if [[ ! -f /usr/local/lib/libthrift.a ]]; then
    pushd thrift-0.9.1
    ./bootstrap.sh
    ./configure
    make
    sudo make install
    popd
  else
    log "thrift is installed. skipping."
  fi
}

function install_rocksdb() {
  if [[ -f /usr/local/lib/librocksdb.a ]]; then
    log "rocksdb is installed. skipping."
    return
  fi
  if [[ ! -f rocksdb-3.5.tar.gz ]]; then
    wget https://github.com/facebook/rocksdb/archive/rocksdb-3.5.tar.gz
  fi
  if [[ ! -d rocksdb-rocksdb-3.5 ]]; then
    tar -xf rocksdb-3.5.tar.gz
  fi
  if [ $OS = "ubuntu" ] || [ $OS = "centos" ]; then
    if [[ ! -f rocksdb-rocksdb-3.5/librocksdb.a ]]; then
      pushd rocksdb-rocksdb-3.5
      make static_lib
      popd
    fi
    if [[ ! -f /usr/local/lib/librocksdb.a ]]; then
      sudo cp rocksdb-rocksdb-3.5/librocksdb.a /usr/local/lib
    else
      log "librocksdb already installed. skipping."
    fi
    if [[ ! -d /usr/local/include/rocksdb ]]; then
      mkdir -p /usr/local/include
      sudo cp -R rocksdb-rocksdb-3.5/include/rocksdb /usr/local/include
    else
      log "rocksdb header already installed. skipping."
    fi
  elif [[ $OS = "darwin" ]]; then
    if [[ ! -f rocksdb-rocksdb-3.5/librocksdb.a ]]; then
      pushd rocksdb-rocksdb-3.5
      make static_lib
      popd
    fi
    if [[ ! -f /usr/local/lib/librocksdb.a ]]; then
      sudo cp rocksdb-rocksdb-3.5/librocksdb.a /usr/local/lib
      sudo cp -R rocksdb-rocksdb-3.5/include/rocksdb /usr/local/include
    else
      log "rocksdb already installed. skipping."
    fi
  fi
}

function package() {
  if [[ $OS = "ubuntu" ]]; then
    if dpkg --get-selections | grep --quiet $1; then
      log "$1 is already installed. skipping."
    else
      sudo apt-get install $@ -y
    fi
  elif [[ $OS = "centos" ]]; then
    if rpm -qa | grep --quiet $1; then
      log "$1 is already installed. skipping."
    else
      sudo yum install $@ -y
    fi
  elif [[ $OS = "darwin" ]]; then
    if brew list | grep --quiet $1; then
      log "$1 is already installed. skipping."
    else
      brew install $@ || brew upgrade $@
    fi
  fi
}

function main() {
  platform OS

  mkdir -p $WORKING_DIR
  cd $WORKING_DIR

  if [ $OS = "ubuntu" ] || [ $OS = "centos" ]; then
    if [[ $EUID -ne 0 ]]; then
      fatal "this script must be run as root. exiting."
    fi
  fi

  if [[ $OS = "centos" ]]; then
    log "detected centos"
  elif [[ $OS = "ubuntu" ]]; then
    log "detected ubuntu"
    DISTRO=`cat /etc/*-release | grep DISTRIB_CODENAME | awk '{split($0,bits,"="); print bits[2]}'`
  elif [[ $OS = "darwin" ]]; then
    log "detected mac os x"
  else
    fatal "could not detect the current operating system. exiting."
  fi

  threads THREADS

  if [[ $OS = "ubuntu" ]]; then

    if [[ $DISTRO = "precise" ]]; then
      add-apt-repository http://ppa.launchpad.net/boost-latest/ppa/ubuntu
    fi
    apt-get update

    package git
    package unzip
    package build-essential
    package cmake
    package devscripts
    package debhelper
    if [[ $DISTRO = "precise" ]]; then
      package libunwind7-dev
    fi
    if [[ $DISTRO = "trusty" ]]; then
      package libunwind8-dev
    fi

    package python-pip
    package python-dev

    package clang-3.4
    package clang-format-3.4

    set_cc clang
    set_cxx clang++

    package libboost1.55-all-dev

    if [[ $DISTRO = "precise" ]]; then
      if [[ ! -f libgflags-dev_2.1.0-1_amd64.deb ]]; then
        wget https://github.com/schuhschuh/gflags/releases/download/v2.1.0/libgflags-dev_2.1.0-1_amd64.deb
      else
        log "gflags deb is already downloaded. skipping."
      fi
      if [[ ! -f /usr/lib/libgflags.a ]]; then
        dpkg -i libgflags-dev_2.1.0-1_amd64.deb
      else
        log "gflags is already installed. skipping."
      fi
    else
      package libgoogle-glog-dev
    fi

    if [[ $DISTRO = "precise" ]]; then
      if [[ ! -f glog-0.3.3.tar.gz ]]; then
        wget https://google-glog.googlecode.com/files/glog-0.3.3.tar.gz
      fi
      if [[ ! -d glog-0.3.3 ]]; then
        tar -xf glog-0.3.3.tar.gz
      fi
      if [[ ! -f "glog-0.3.3-gflags-namespace.patch" ]]; then
        wget https://gist.githubusercontent.com/marpaia/02312b7bd25502f5319a/raw/7d3e50c5079a085fe08822fd6952d5fb19c2fe1e/glog-0.3.3-gflags-namespace.patch
        pushd glog-0.3.3
        patch -p1 < ../glog-0.3.3-gflags-namespace.patch
        popd
      fi
      pushd glog-0.3.3
      ./configure
      make
      sudo make install
      popd
    else
      package libgoogle-glog-dev
    fi
    package libsnappy-dev
    package libbz2-dev
    package libreadline-dev
    package libprocps3-dev

    package libtool
    package autoconf
    package automake
    package pkg-config
    package libssl-dev
    package bison
    package flex
    install_thrift

    install_rocksdb

    package liblzma-dev
    package libprocps3-dev
  elif [[ $OS = "centos" ]]; then
    sudo yum update -y

    package git-all
    package unzip
    package xz

  elif [[ $OS = "darwin" ]]; then
    if [[ ! -f "/usr/local/bin/brew" ]]; then
      fatal "could not find homebrew. please install it from http://brew.sh/"
    fi

    if brew list | grep --quiet wget; then
      log "wget is already installed. skipping"
    else
      brew install wget
    fi

    package cmake
    package boost --c++11
    package gflags
    package glog
    package snappy
    package readline
    package thrift
    package rocksdb
    package lz4
  fi

  cd $SCRIPT_DIR/../
  sudo pip install -r requirements.txt
  git submodule init
  git submodule update
}

main