Function process_thread

python3_cron_scripts/zgrab_port_ip.py:672–728

Runs zgrab and stores the result if necessary

(
    logger, ips, port, run_command, zones_struct, zgrab_collection, tnum
)

def process_thread(
    logger, ips, port, run_command, zones_struct, zgrab_collection, tnum
):
    """
    Runs zgrab and stores the result if necessary
    """
    json_output = run_command(ips, tnum)

    if ("success_count" in json_output and json_output["success_count"] > 0) or (
        "statuses" in json_output
        and json_output["statuses"][global_port_names[port]]["successes"] > 0
    ):
        result_file = open(
            "./json_p" + port + "/p" + port + "-" + str(tnum) + ".json", "r"
        )
        results = []
        for result in result_file:
            results.append(json.loads(result))
        result_file.close()
        for result in results:
            if (
                "zgrab2" in global_zgrab_path
                and "error" in result["data"][global_port_names[port]]
            ) or "error" in result:
                if port == "25":
                    if (
                        (
                            "zgrab2" in global_zgrab_path
                            and "result" in result["data"]["smtp"]
                            and "starttls" in result["data"]["smtp"]["result"]
                        )
                        or "error_component" in result
                        and result["error_component"] == "starttls"
                    ):
                        logger.debug("Adding " + str(result["ip"]) + " to retest list")
                        global_retest_list.append(result["ip"])
                    else:
                        logger.warning("Failed " + port + ": " + str(result["ip"]))
                else:
                    logger.warning("Failed " + port + ": " + str(result["ip"]))
            else:
                result["aws"] = zones_struct["ip_manager"].is_aws_ip(result["ip"])
                result["azure"] = zones_struct["ip_manager"].is_azure_ip(result["ip"])
                result["gcp"] = zones_struct["ip_manager"].is_gcp_ip(result["ip"])
                result["tracked"] = zones_struct["ip_manager"].is_tracked_ip(
                    result["ip"]
                )
                insert_result(
                    result,
                    port,
                    zones_struct["ip_context"],
                    zones_struct["zones"],
                    zgrab_collection,
                )
                logger.debug("Inserted " + port + ": " + result["ip"])
    else:
        logger.warning("Failed " + port + ": " + str(ips))