#!/usr/bin/python

import sqlite3, os, string, hashlib
from Crypto.Random import random


###################################################
#
# Default values for the config
#
###################################################

# Staging Key is set up via environmental variable
# or via command line. By setting RANDOM a randomly
# selected password will automatically be selected
# or it can be set to any bash acceptable character
# set for a password.

STAGING_KEY = os.getenv('STAGING_KEY', "BLANK")
punctuation = '!#$%&()*+,-./:;<=>?@[\]^_`{|}~'

# otherwise prompt the user for a set value to hash for the negotiation password
if STAGING_KEY == "BLANK":
    choice = raw_input("\n [>] Enter server negotiation password, enter for random generation: ")
    if choice == "":
        # if no password is entered, generation something random
        STAGING_KEY = ''.join(random.sample(string.ascii_letters + string.digits + punctuation, 32))
    else:
        STAGING_KEY = hashlib.md5(choice).hexdigest()
elif STAGING_KEY == "RANDOM":
    STAGING_KEY = ''.join(random.sample(string.ascii_letters + string.digits + punctuation, 32))

# the resource requested by the initial launcher
STAGE0_URI = "index.asp"

# the resource used by the DH key post
STAGE1_URI = "index.jsp"

# the resource used by the sysinfo checkin that returns the agent.ps1
STAGE2_URI = "index.php"

# the default delay (in seconds) for agent callback
DEFAULT_DELAY = 5

# the default jitter (0.0-1.0) to apply to the callback delay
DEFAULT_JITTER = 0.0

# the default traffic profile to use for agent communications
#   format -> requestUris|user_agent|additionalHeaders
DEFAULT_PROFILE = "/admin/get.php,/news.asp,/login/process.jsp|Mozilla/5.0 (Macintosh; Intel Mac OS X 10.11; rv:45.0) Gecko/20100101 Firefox/45.0"

# default https cert to use
DEFAULT_CERT_PATH = ''

# the default port for listeners
DEFAULT_PORT = 8080

# the installation path for EmPyre, defaults to auto-calculating it
# NOTE: set manually if issues arise
currentPath = os.path.dirname(os.path.realpath(__file__))
empyreIndex = currentPath.rfind("EmPyre")
if empyreIndex < 0:
    empyreIndex = currentPath.rfind("empyre")
if empyreIndex < 0:
    INSTALL_PATH = "/".join(os.getcwd().split("/")[0:-1])+"/"
else:
    endIndex = currentPath.find("/", empyreIndex)
    endIndex = None if endIndex < 0 else endIndex
    INSTALL_PATH = currentPath[0:endIndex] + "/"

# the version version to appear as
SERVER_VERSION = "Microsoft-IIS/7.5"

# an IP white list to ONLY accept clients from
#   format is "192.168.1.1,192.168.1.10-192.168.1.100,10.0.0.0/8"
IP_WHITELIST = ""

# an IP black list to reject accept clients from
#   format is "192.168.1.1,192.168.1.10-192.168.1.100,10.0.0.0/8"
IP_BLACKLIST = ""

# number of times an agent will call back without an answer prior to exiting
DEFAULT_LOST_LIMIT = 60 

# default credentials used to log into the RESTful API
API_USERNAME = "empyreadmin"
API_PASSWORD = ''.join(random.sample(string.ascii_letters + string.digits + punctuation, 32))

# the 'permanent' API token (doesn't change)
API_PERMANENT_TOKEN = ''.join(random.choice(string.ascii_lowercase + string.digits) for x in range(40))


###################################################
#
# Database setup.
#
###################################################


conn = sqlite3.connect('../data/empyre.db')

c = conn.cursor()

# try to prevent some of the weird sqlite I/O errors
c.execute('PRAGMA journal_mode = OFF')

c.execute('DROP TABLE IF EXISTS config')
c.execute('''CREATE TABLE config (
    "staging_key" text,
    "stage0_uri" text,
    "stage1_uri" text,
    "stage2_uri" text,
    "default_delay" integer,
    "default_jitter" real,
    "default_profile" text,
    "default_cert_path" text,
    "default_port" text,
    "install_path" text,
    "server_version" text,
    "ip_whitelist" text,
    "ip_blacklist" text,
    "default_lost_limit" integer,
    "autorun_command" text,
    "autorun_data" text,
    "rootuser" boolean,
    "api_username" text,
    "api_password" text,
    "api_current_token" text,
    "api_permanent_token" text
    )''')

# kick off the config component of the database
c.execute("INSERT INTO config VALUES (?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?)", (STAGING_KEY,STAGE0_URI,STAGE1_URI,STAGE2_URI,DEFAULT_DELAY,DEFAULT_JITTER,DEFAULT_PROFILE,DEFAULT_CERT_PATH,DEFAULT_PORT,INSTALL_PATH,SERVER_VERSION,IP_WHITELIST,IP_BLACKLIST, DEFAULT_LOST_LIMIT, "", "", False, API_USERNAME, API_PASSWORD, "", API_PERMANENT_TOKEN))

c.execute('DROP TABLE IF EXISTS "agents"')
c.execute('''CREATE TABLE "agents" (
    "id" integer PRIMARY KEY,
    "session_id" text,
    "listener" text,
    "name" text,
    "delay" integer,
    "jitter" real,
    "external_ip" text,
    "internal_ip" text,
    "username" text,
    "high_integrity" integer,
    "process_id" text,
    "hostname" text,
    "os_details" text,
    "session_key" text,
    "nonce" text,
    "checkin_time" text,
    "lastseen_time" text,
    "servers" text,
    "uris" text,
    "old_uris" text,
    "user_agent" text,
    "headers" text,
    "kill_date" text,
    "working_hours" text,
    "py_version" text,
    "lost_limit" integer,
    "taskings" text,
    "results" text
    )''')

c.execute('DROP TABLE IF EXISTS "listeners"')
c.execute('''CREATE TABLE "listeners" (
    "id" integer PRIMARY KEY,
    "name" text,
    "host" text,
    "port" integer,
    "cert_path" text,
    "staging_key" text,
    "default_delay" integer,
    "default_jitter" real,
    "default_profile" text,
    "kill_date" text,
    "working_hours" text,
    "listener_type" text,
    "redirect_target" text,
    "default_lost_limit" integer
    )''')


# type = hash, plaintext, token
#   for tokens, the data is base64'ed and stored in pass
c.execute('DROP TABLE IF EXISTS "credentials"')
c.execute('''CREATE TABLE "credentials" (
    "id" integer PRIMARY KEY,
    "credtype" text,
    "domain" text,
    "username" text,
    "password" text,
    "host" text,
    "sid" text,
    "notes" text
    )''')


# event_types -> checkin, task, result, rename
c.execute('DROP TABLE IF EXISTS "reporting"')
c.execute('''CREATE TABLE "reporting" (
    "id" integer PRIMARY KEY,
    "name" text,
    "event_type" text,
    "message" text,
    "time_stamp" text
    )''')


# commit the changes and close everything off
conn.commit()
conn.close()

print "\n [*] Database setup completed!\n"