зеркало из https://github.com/mozilla/MozDef.git
58 строки
1.8 KiB
Python
58 строки
1.8 KiB
Python
|
#!/usr/bin/env python
|
||
|
|
|
||
|
|
# 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/.
|
||
|
|
# Copyright (c) 2017 Mozilla Corporation
|
||
|
|
#
|
||
|
|
# Contributors:
|
||
|
|
# Brandon Myers bmyers@mozilla.com
|
||
|
|
|
||
|
|
from git import Repo, cmd
|
||
|
|
|
||
|
|
import sys
|
||
|
|
import os
|
||
|
|
from configlib import getConfig, OptionParser
|
||
|
|
|
||
|
|
sys.path.append(os.path.join(os.path.dirname(os.path.abspath(__file__)), '../lib'))
|
||
|
|
from utilities.logger import logger, initLogger
|
||
|
|
|
||
|
|
|
||
|
|
def download_generic_alerts(repo_url, save_location):
|
||
|
|
if not os.path.isdir(save_location):
|
||
|
|
logger.debug("Cloning " + str(repo_url) + " into " + str(save_location))
|
||
|
|
Repo.clone_from(repo_url, save_location)
|
||
|
|
else:
|
||
|
|
logger.debug("Updating " + str(save_location))
|
||
|
|
git_repo = cmd.Git(save_location)
|
||
|
|
git_repo.pull()
|
||
|
|
|
||
|
|
|
||
|
|
def main():
|
||
|
|
logger.debug('Starting')
|
||
|
|
logger.debug(options)
|
||
|
|
download_generic_alerts(options.alert_repo_url, options.alert_data_location)
|
||
|
|
|
||
|
|
|
||
|
|
def initConfig():
|
||
|
|
# output our log to stdout or syslog
|
||
|
|
options.output = getConfig('output', 'stdout', options.configfile)
|
||
|
|
options.sysloghostname = getConfig('sysloghostname', 'localhost', options.configfile)
|
||
|
|
options.syslogport = getConfig('syslogport', 514, options.configfile)
|
||
|
|
|
||
|
|
options.alert_repo_url = getConfig('alert_repo_url', '', options.configfile)
|
||
|
|
options.alert_data_location = getConfig('alert_data_location', '', options.configfile)
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
parser = OptionParser()
|
||
|
|
parser.add_option(
|
||
|
|
"-c",
|
||
|
|
dest='configfile',
|
||
|
|
default=sys.argv[0].replace('.py', '.conf'),
|
||
|
|
help="configuration file to use")
|
||
|
|
(options, args) = parser.parse_args()
|
||
|
|
initConfig()
|
||
|
|
initLogger(options)
|
||
|
|
main()
|