mirror of
https://github.com/Security-Onion-Solutions/securityonion.git
synced 2025-12-06 09:12:45 +01:00
80 lines
2.2 KiB
Python
Executable File
80 lines
2.2 KiB
Python
Executable File
import json
|
|
import helpers
|
|
import os
|
|
import sys
|
|
import argparse
|
|
import csv
|
|
|
|
|
|
def checkConfigRequirements(conf):
|
|
if "file_path" not in conf or len(conf['file_path']) == 0:
|
|
sys.exit(126)
|
|
else:
|
|
return True
|
|
|
|
|
|
def searchFile(artifact, csvfiles):
|
|
dir = os.path.dirname(os.path.realpath(__file__))
|
|
found = []
|
|
for f in csvfiles:
|
|
filename = dir + "/" + f
|
|
with open(filename, "r") as csvfile:
|
|
csvdata = csv.DictReader(csvfile)
|
|
for row in csvdata:
|
|
first_key = list(row.keys())[0]
|
|
if artifact in row[first_key]:
|
|
row.update({"filename": filename})
|
|
found.append(row)
|
|
if len(found) != 0:
|
|
if len(found) == 1:
|
|
results = found[0]
|
|
else:
|
|
results = found
|
|
else:
|
|
results = "No results"
|
|
|
|
return results
|
|
|
|
|
|
def prepareResults(raw):
|
|
if len(raw) > 0:
|
|
if "No results" in raw:
|
|
status = "ok"
|
|
summary = "no_results"
|
|
else:
|
|
status = "info"
|
|
summary = "suspicious"
|
|
else:
|
|
raw = {}
|
|
status = "caution"
|
|
summary = "internal_failure"
|
|
response = raw
|
|
results = {'response': response, 'status': status, 'summary': summary}
|
|
return results
|
|
|
|
|
|
def analyze(conf, input):
|
|
checkConfigRequirements(conf)
|
|
meta = helpers.loadMetadata(__file__)
|
|
data = helpers.parseArtifact(input)
|
|
helpers.checkSupportedType(meta, data["artifactType"])
|
|
search = searchFile(data["value"], conf['file_path'])
|
|
results = prepareResults(search)
|
|
return results
|
|
|
|
|
|
def main():
|
|
dir = os.path.dirname(os.path.realpath(__file__))
|
|
parser = argparse.ArgumentParser(description='Search CSV file for a given artifact')
|
|
parser.add_argument('artifact', help='the artifact represented in JSON format')
|
|
parser.add_argument('-c', '--config', metavar="CONFIG_FILE", default=dir + "/localfile.yaml", help='optional config file to use instead of the default config file')
|
|
|
|
args = parser.parse_args()
|
|
if args.artifact:
|
|
results = analyze(helpers.loadConfig(args.config), args.artifact)
|
|
print(json.dumps(results))
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|