adbfuzzconfig.py

#!/usr/bin/env python
# ***** BEGIN LICENSE BLOCK *****
# Version: MPL 2.0
#
# This Source Code Form is subject to the terms of the Mozilla Public
# License, v. 2.0. If a copy of the MPL was not distributed with this
# file, You can obtain one at http://mozilla.org/MPL/2.0/.
#
# The Original Code is ADBFuzz.
#
# The Initial Developer of the Original Code is Christian Holler (decoder).
#
# Contributors:
#  Christian Holler <decoder@mozilla.com> (Original Developer)
#
# ***** END LICENSE BLOCK *****

from ConfigParser import SafeConfigParser

class ADBFuzzConfig:

  def __init__(self, cfgFile):
    cfgDefaults = {}
    cfgDefaults['id'] = None
    cfgDefaults['remoteHost'] = None
    cfgDefaults['localPort'] = '8088'
    cfgDefaults['useWebSockets'] = False
    cfgDefaults['localWebSocketPort'] = '8089'
    cfgDefaults['localListenPort'] = '8090'
    cfgDefaults['localPortOffset'] = '0'
    cfgDefaults['libDir'] = None
    cfgDefaults['knownPath'] = None
    cfgDefaults['debug'] = str(False)
    cfgDefaults['useMail'] = str(False)
    cfgDefaults['runTimeout'] = str(5)
    cfgDefaults['maxLogSize'] = str(1024*1024) # Default to 1 MB maximum log
    cfgDefaults['addon'] = None

    self.cfg = SafeConfigParser(cfgDefaults)
    if (len(self.cfg.read(cfgFile)) == 0):
      raise Exception("Unable to read configuration file: " + cfgFile)

    self.id = self.cfg.get('main', 'id')
    self.fuzzerFile = self.cfg.get('main', 'fuzzer')
    self.runTimeout = self.cfg.getint('main', 'runTimeout')
    self.maxLogSize = self.cfg.getint('main', 'maxLogSize')
    self.remoteAddr = self.cfg.get('main', 'remoteHost')
    self.localAddr = self.cfg.get('main', 'localHost')
    self.localPort = self.cfg.get('main', 'localPort')
    self.debug = self.cfg.getboolean('main', 'debug')
    self.knownPath = self.cfg.get('main', 'knownPath')
    self.addon = self.cfg.get('main', 'addon')

    self.useWebSockets = self.cfg.getboolean('main', 'useWebSockets')
    self.localWebSocketPort = self.cfg.get('main', 'localWebSocketPort')
    self.localListenPort = self.cfg.get('main', 'localListenPort')
    self.localPortOffset = self.cfg.get('main', 'localPortOffset')
    self.libDir = self.cfg.get('main', 'libDir')

    # Mail configuration
    self.useMail = self.cfg.getboolean('main', 'useMail')
    if self.useMail:
      self.mailFrom = self.cfg.get('main', 'mailFrom')
      self.mailTo = self.cfg.get('main', 'mailTo')
      self.SMTPHost = self.cfg.get('main', 'SMTPHost')
      
    # Add our port offset to all local ports
    portOffset = int(self.localPortOffset)
    self.localPort = str(int(self.localPort) + portOffset)
    self.localWebSocketPort = str(int(self.localWebSocketPort) + portOffset)
    self.localListenPort = str(int(self.localListenPort) + portOffset)

if __name__ == "__main__":
  raise Exception("This module cannot run standalone, but is used by ADBFuzz")