infosecn1nja
/
Empire
mirror of https://github.com/infosecn1nja/Empire.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Fix for path errors in some cases for ./setup/setup_database.py

1.6
Harmj0y 2016-04-01 15:51:58 -04:00
parent e43fb94634
commit 8e44188350
1 changed files with 15 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
setup/setup_database.py
View File

@ -56,17 +56,27 @@ DEFAULT_CERT_PATH = ''
DEFAULT_PORT = 8080
# the installation path for Empire, defaults to auto-calculating it
INSTALL_PATH = "/".join(os.getcwd().split("/")[0:-1])+"/"
# set manually if issues arise
currentPath = os.path.dirname(os.path.realpath(__file__))
empireIndex = currentPath.rfind("Empire")
if empireIndex < 0:
empireIndex = currentPath.rfind("empire")
if empireIndex < 0:
INSTALL_PATH = "/".join(os.getcwd().split("/")[0:-1])+"/"
else:
endIndex = currentPath.find("/", empireIndex)
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
# 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
# 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
@ -86,8 +96,8 @@ API_PERMANENT_TOKEN = ''.join(random.choice(string.ascii_lowercase + string.digi
#
###################################################
conn = sqlite3.connect('../data/empire.db')
dbPath = INSTALL_PATH + "data/empire.db"
conn = sqlite3.connect(dbPath)
c = conn.cursor()