-
Notifications
You must be signed in to change notification settings - Fork 79
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Merge pull request #15 from chkp-alexeybu/ab_default_configs
Added flexible build configurations to enable running without command line. Fixed issues #12 and #14.
- Loading branch information
Showing
10 changed files
with
218 additions
and
62 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file was deleted.
Oops, something went wrong.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -1,97 +1,126 @@ | ||
| __author__ = 'arl' | ||
|
|
||
|
|
||
| import os | ||
| from json import load | ||
| from sys import argv | ||
| from os import path | ||
|
|
||
|
|
||
| REPORT = "report.html" | ||
| BOOTSTRAP = "bootstrap.css" | ||
| CONFIG_PATH = "../config/" | ||
|
|
||
| PATH_CONF_CUCKOO = 'cuckoo' | ||
| PATH_REPORT = 'report' | ||
| PATH_BOOTSTRAP = 'bootstrap' | ||
| OUT_DEFAULT_CONFIG = "default_configs.h" | ||
| OUT_DATA_REPORT = "data_report_html.h" | ||
| OUT_DATA_BOOTSTRAP = "data_bootstrap_css.h" | ||
|
|
||
|
|
||
| def make_hex_array(data): | ||
| return ','.join(['0x%02x' % ord(x) + ('\r\n' if (i + 1) % 16 == 0 else '') for i, x in enumerate(data)]) | ||
| return ",".join( | ||
| [ | ||
| "0x%02x" % ord(x) + ("\n" if (i + 1) % 16 == 0 else "") | ||
| for i, x in enumerate(data) | ||
| ] | ||
| ) | ||
|
|
||
|
|
||
| def escape_file_data(data): | ||
| data = data.replace("\\", "\\\\") | ||
| data = data.replace("\"", "\\\"") | ||
| dv = data.split('\n') | ||
| data = '' | ||
| for ds in dv: | ||
| data += ds.rstrip() + '\\\r\n' | ||
|
|
||
| return data | ||
| data = data.replace('"', '\\"') | ||
| return "\\\n".join(map(str.rstrip, data.split("\n"))) | ||
|
|
||
|
|
||
| def read_data(fn): | ||
| try: | ||
| return open(fn, 'rb').read() | ||
| with open(fn, "r") as f: | ||
| return f.read() | ||
| except Exception as e: | ||
| print 'File Read Exception: %s' % e | ||
| print("File Read Exception: %s" % e) | ||
| return | ||
|
|
||
|
|
||
| def write_data(fn, data): | ||
| try: | ||
| open(fn, 'wb').write(data) | ||
| with open(fn, "w") as f: | ||
| f.write(data) | ||
| except Exception as e: | ||
| print 'File Write Exception: %s' % e | ||
| print("File Write Exception: %s" % e) | ||
| return False | ||
|
|
||
| return True | ||
|
|
||
|
|
||
| def get_file_data(conf, option): | ||
| fn = conf.get(option, '') | ||
| fn = conf.get(option, "") | ||
| if not fn: | ||
| print 'Configuration parameter: `%s\' is absent' % option | ||
| print("Configuration parameter: `%s' is absent" % option) | ||
| return False | ||
|
|
||
| return read_data(fn) | ||
|
|
||
|
|
||
| def create_includes(conf): | ||
| cuckoo_d = get_file_data(conf, PATH_CONF_CUCKOO) | ||
| report_d = get_file_data(conf, PATH_REPORT) | ||
| bootstrap_d = get_file_data(conf, PATH_BOOTSTRAP) | ||
| def create_includes(sandboxes): | ||
| report_d = read_data(REPORT) | ||
| bootstrap_d = read_data(BOOTSTRAP) | ||
|
|
||
| # escape characters | ||
| cuckoo_file = "static const char *cuckoo_conf = \"%s\";" % escape_file_data(cuckoo_d) | ||
| report_file = "static const char *report_data = \"%s\";" % escape_file_data(report_d) | ||
| bootstrap_file = "static const char bootstrap_data[] = {\r\n%s\r\n};" % make_hex_array(bootstrap_d) | ||
| sandbox_file = "std::list<std::pair<std::string, const char *>> default_configs;\n" | ||
| for sandbox in sandboxes: | ||
| sandbox_d = read_data(path.join(CONFIG_PATH, sandbox + ".conf")) | ||
| sandbox_file += f"""static const char {sandbox}_conf[] = %s; | ||
| \ndefault_configs.push_back(std::pair<std::string, const char *>(std::string("{sandbox}"), {sandbox}_conf)); | ||
| """ % ( | ||
| "{" + make_hex_array(sandbox_d) + "}" | ||
| ) | ||
|
|
||
| if not write_data("code_cuckoo.conf", cuckoo_file): | ||
| report_file = 'static const char *report_data = "%s";' % escape_file_data(report_d) | ||
| bootstrap_file = "static const char bootstrap_data[] = {\n%s\n};" % make_hex_array( | ||
| bootstrap_d | ||
| ) | ||
|
|
||
| if not write_data(OUT_DEFAULT_CONFIG, sandbox_file): | ||
| return False | ||
| if not write_data("data_report.html", report_file): | ||
| if not write_data(OUT_DATA_REPORT, report_file): | ||
| return False | ||
| if not write_data("data_bootstrap.css", bootstrap_file): | ||
| if not write_data(OUT_DATA_BOOTSTRAP, bootstrap_file): | ||
| return False | ||
|
|
||
| return True | ||
|
|
||
|
|
||
| def read_conf(fn): | ||
| try: | ||
| return load(open(fn, 'rb')) | ||
| with open(fn, "r") as f: | ||
| return load(f) | ||
| except Exception as e: | ||
| print 'Configuration Read Exception: %s' % e | ||
| print("Configuration Read Exception: %s" % e) | ||
| return | ||
|
|
||
|
|
||
| def list_configs(): | ||
| return [ | ||
| x.split(".")[0] | ||
| for x in filter(lambda x: x.endswith(".conf"), os.listdir(CONFIG_PATH)) | ||
| ] | ||
|
|
||
|
|
||
| def main(): | ||
| configs = list_configs() | ||
| if len(argv) < 2: | ||
| print 'GenDefaultData usage: %s [config]' % argv[0] | ||
| print( | ||
| "GenDefaultData usage: %s [configuration_name1] [configuration_name2] [...]" | ||
| % argv[0] | ||
| ) | ||
| print("List of detection sets:\n" + "\n".join(configs)) | ||
| exit(1) | ||
|
|
||
| conf = read_conf(argv[1]) | ||
| if not conf: | ||
| exit(2) | ||
| for ds in argv[1:]: | ||
| if ds not in configs: | ||
| print(f"Invalid configuration name: {ds}") | ||
| exit(2) | ||
|
|
||
| if not create_includes(conf): | ||
| if not create_includes(argv[1:]): | ||
| exit(3) | ||
|
|
||
|
|
||
| if __name__ == '__main__': | ||
| if __name__ == "__main__": | ||
| main() |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.