def _get_bro_version(self):
from BroControl import execute
bro = self.config["bro"]
if not os.path.lexists(bro):
raise ConfigurationError("cannot find Bro binary: %s" % bro)
version = ""
success, output = execute.run_localcmd("%s -v" % bro)
if success and output:
version = output.splitlines()[-1]
else:
msg = " with no output"
if output:
msg = " with output:\n%s" % output
raise RuntimeEnvironmentError('running "bro -v" failed%s' % msg)
match = re.search(".* version ([^ ]*).*$", version)
if not match:
raise RuntimeEnvironmentError(
'cannot determine Bro version ("bro -v" output: %s)' %
version.strip())
version = match.group(1)
# If bro is built with the "--enable-debug" configure option, then it
# appends "-debug" to the version string.
if version.endswith("-debug"):
version = version[:-6]
return version
def __init__(self, path):
self.path = path
try:
self.db = sqlite3.connect(self.path)
except sqlite3.Error as err:
raise RuntimeEnvironmentError(
"%s: %s\nCheck if the user running BroControl has both write and search permission to\nthe directory containing the database file and has both read and write\npermission to the database file itself."
% (err, path))
self.c = self.db.cursor()
try:
self.setup()
except sqlite3.Error as err:
raise RuntimeEnvironmentError(
"%s: %s\nCheck if the user running BroControl has write access to the database file.\nOtherwise, the database file is possibly corrupt."
% (err, path))
def set(self, key, value):
value = json.dumps(value)
try:
self.c.execute("REPLACE INTO state (key, value) VALUES (?,?)",
[key, value])
except sqlite3.Error as err:
raise RuntimeEnvironmentError(
"%s: %s\nCheck if the user running BroControl has write access to the database file."
% (err, self.path))
self.db.commit()
def _initialize_options(self):
from BroControl import execute
# Set defaults for options we get passed in.
self.init_option("brobase", self.basedir)
self.init_option("broscriptdir", self.broscriptdir)
self.init_option("version", VERSION)
# Initialize options that are not already set.
for opt in options.options:
if not opt.dontinit:
self.init_option(opt.name, opt.default)
# Set defaults for options we derive dynamically.
self.init_option("mailto", "%s" % os.getenv("USER"))
self.init_option("mailfrom",
"Big Brother <bro@%s>" % socket.gethostname())
self.init_option("mailalarmsto", self.config["mailto"])
# Determine operating system.
success, output = execute.run_localcmd("uname")
if not success or not output:
raise RuntimeEnvironmentError("failed to run uname: %s" % output)
self.init_option("os", output.strip())
# Determine the CPU pinning command.
pin_cmd = ""
if self.config["os"] == "Linux":
pin_cmd = "taskset -c"
elif self.config["os"] == "FreeBSD":
pin_cmd = "cpuset -l"
self.init_option("pin_command", pin_cmd)
# Find the time command (should be a GNU time for best results).
time_cmd = ""
success, output = execute.run_localcmd("which time")
if success and output:
# On redhat-based systems, path to cmd is prefixed with '\t' on 2nd
# line when alias is defined.
time_cmd = output.splitlines()[-1].strip()
self.init_option("time", time_cmd)
# Calculate the log expire interval (in minutes).
minutes = self._get_interval_minutes("logexpireinterval")
self.init_option("logexpireminutes", minutes)