update_conf.py

import os
import sys
import ConfigParser


#Setting auxiliary.conf
virtualbox_cfg = ConfigParser.ConfigParser()
virtualbox_cfg.read("/cuckoo/conf/virtualbox_websrv.conf")
with open("/cuckoo/conf/virtualbox_websrv.conf", 'w') as cfile:
    if os.environ.get('VBOX_URL'):
        virtualbox_cfg.set('virtualbox_websrv', 'url', os.environ['VBOX_URL'])
    if os.environ.get('VBOX_USER'):
        virtualbox_cfg.set('virtualbox_websrv', 'user', os.environ['VBOX_USER'])
    if os.environ.get('VBOX_PASSWORD'):
        virtualbox_cfg.set('virtualbox_websrv', 'password', os.environ['VBOX_PASSWORD'])
    virtualbox_cfg.write(cfile)


#Setting reporting.conf
reporting_cfg = ConfigParser.ConfigParser()
reporting_cfg.read("/cuckoo/conf/reporting.conf")
with open("/cuckoo/conf/reporting.conf", 'w') as cfile:
    # if os.environ.get('ES_HOST'):
    #     reporting_cfg.set('elasticsearch', 'enabled', "yes")
    #     if os.environ.get('ES_PORT'):
    #         reporting_cfg.set('elasticsearch', 'hosts', "%s:%s" % (os.environ['ES_HOST'],os.environ['ES_PORT']))
    #     else:
    #         reporting_cfg.set('elasticsearch', 'hosts', os.environ['ES_HOST'])

    if os.environ.get('MONGO_HOST'):
        reporting_cfg.set('mongodb', 'enabled', "yes")
        reporting_cfg.set('mongodb', 'host', os.environ['MONGO_HOST'])
    if os.environ.get('MONGO_TCP_PORT'):
        reporting_cfg.set('mongodb', 'port', os.environ['MONGO_TCP_PORT'])

    reporting_cfg.write(cfile)

#Setting cuckoo.conf
cuckoo_cfg = ConfigParser.ConfigParser()
cuckoo_cfg.read("/cuckoo/conf/cuckoo.conf")
with open("/cuckoo/conf/cuckoo.conf", 'w') as cfile:
    if os.environ.get('RESULTSERVER_HOST'):
        cuckoo_cfg.set('resultserver', 'ip', os.environ['RESULTSERVER_HOST'])
    if os.environ.get('RESULTSERVER_PORT'):
        cuckoo_cfg.set('resultserver', 'port', os.environ['RESULTSERVER_PORT'])

    if os.environ.get('MACHINERY'):
        cuckoo_cfg.set('cuckoo', 'machinery', os.environ['MACHINERY'])

    if os.environ.get('DATABASE_CONNECTION'):
        cuckoo_cfg.set('database', 'connection', os.environ['DATABASE_CONNECTION'])

    cuckoo_cfg.write(cfile)


#Setting processing.conf
processing_cfg = ConfigParser.ConfigParser()
processing_cfg.read("/cuckoo/conf/processing.conf")
with open("/cuckoo/conf/processing.conf", 'w') as cfile:
    if os.environ.get('ALLOWED_DNS'):
        processing_cfg.set('network', 'whitelist_dns', 'yes')
        processing_cfg.set('network','allowed_dns',os.environ['ALLOWED_DNS'])

    # if os.environ.get('EXTRACT_DLL'):
    #     processing_cfg.set('procmemory', 'extract_dll', os.environ['EXTRACT_DLL'])

    if os.environ.get('ENABLE_STRINGS'):
        processing_cfg.set('strings', 'enabled', os.environ['ENABLE_STRINGS'])

    # if os.environ.get('ENABLE_SURICATA'):
    #     processing_cfg.set('suricata', 'enabled', os.environ['ENABLE_SURICATA'])

    processing_cfg.write(cfile)


#Setting auxiliary.conf
# auxiliary_cfg = ConfigParser.ConfigParser()
# auxiliary_cfg.read("/cuckoo/conf/auxiliary.conf")
# with open("/cuckoo/conf/auxiliary.conf", 'w') as cfile:
#     if os.environ.get('ENABLE_MITM'):
#         auxiliary_cfg.set('mitm', 'enabled', os.environ['ENABLE_MITM'])

#     auxiliary_cfg.write(cfile)


sys.exit()