configure.py

import os
import json

'''
Supported scanners:
    - w3af
'''


def main():
    with open("/root/config.json", "r") as f_conf:
        config = json.load(f_conf)

    # configure cscan target
    target = config["target"]
    if "url" in target:
        f_t = open("/root/cscan/websites.txt", "w")
        f_t.write(target["url"])
        f_t.write(os.linesep)
        f_t.close()
    if "ip" in target:
        f_t = open("/root/cscan/ips.txt", "w")
        f_t.write(target["ip"])
        f_t.write(os.linesep)
        f_t.close()

    # configure scanners
    cscan_config = {}
    for scanner in config["config"]:
        profile = config["config"][scanner]["profile"]
        if scanner == "w3af":
            cscan_config["CS_W3AF"] = "/root/w3af/w3af_api"
            if profile == "fast_scan":
                cscan_config["CS_W3AF_PROFILE"] = "/root/w3af/profiles/fast_scan.pw3af"
            else:
                raise UnsupportedProfileException()
            # params = config["config"][scanner]["parameters"]
        elif scanner == "zap":
            cscan_config["CS_ZAP"] = "/root/ZAP_2.7.0/zap.sh"
            if profile != "basic":
                raise UnsupportedProfileException()
        else:
            raise UnsupportedScannerException()
    
    with open("/root/cscan/config.py", "w") as f_csconf:
        f_csconf.write("config = %s\n" % cscan_config)

class UnsupportedProfileException(Exception):
    pass

class UnsupportedScannerException(Exception):
    pass

if __name__ == "__main__":
    main()