create-filters-from-csv.gmp.py

# SPDX-FileCopyrightText: 2024 Martin Boller
#
# SPDX-License-Identifier: GPL-3.0-or-later
#

# Loosely based on the create-filter-from-host-list
#
# Run with gvm-script --gmp-username admin-user --gmp-password password socket create-filters-from-csv.gmp.py hostname-server filters.csv

import csv
import sys
import time
from argparse import ArgumentParser, Namespace, RawTextHelpFormatter
from pathlib import Path

from gvm.errors import GvmResponseError
from gvm.protocols.gmp import Gmp
from gvmtools.helper import error_and_exit

HELP_TEXT = (
    "This script pulls filtername, hostnames/IP addresses, and credentials "
    "from a csv file and creates a filter for each row. \n\n"
    "csv file to contain name of filter, ips, and up to 4 credentials previously created \n"
    "name,ip-addresses,credential1,credential2,credential3,credential4"
)


def check_args(args):
    len_args = len(args.script) - 1
    if len_args != 1:
        message = """
        This script pulls filter names from a csv file and creates a filter \
for each row.
        One parameter after the script name is required.

        1. <filters_csvfile>  -- text file containing filtername and hostnames or IP-addresses

        Example:
            $ gvm-script --gmp-username name --gmp-password pass \
ssh --hostname <gsm> scripts/create_filters_from_csv.gmp <filters_csvfile>
        """
        print(message)
        sys.exit()


def parse_args(args: Namespace) -> Namespace:  # pylint: disable=unused-argument
    """Parsing args ..."""

    parser = ArgumentParser(
        prefix_chars="+",
        add_help=False,
        formatter_class=RawTextHelpFormatter,
        description=HELP_TEXT,
    )

    parser.add_argument(
        "+h",
        "++help",
        action="help",
        help="Show this help message and exit.",
    )

    parser.add_argument(
        "filters_csv_file",
        type=str,
        help=("File containing host names / IPs"),
    )

    script_args, _ = parser.parse_known_args(args)
    return script_args


def filter_id(
    gmp: Gmp,
    filter_name: str,
):
    response_xml = gmp.get_filters(filter_string="rows=-1, name=" + filter_name)
    filters_xml = response_xml.xpath("filter")
    filter_id = ""

    for filter in filters_xml:
        filter_id = filter.get("id")
    return filter_id


def create_filters(
    gmp: Gmp,
    filter_csv_file: Path,
):
    try:
        numberfilters = 0
        with open(filter_csv_file, encoding="utf-8") as csvFile:
            content = csv.reader(csvFile, delimiter=",")  # read the data
            for row in content:  # loop through each row
                if len(row) == 0:
                    continue
                filterType = row[0]
                filterName = row[1]
                filterDescription = row[2]
                filterTerm = row[3]
                filterNameFull = (
                    filterName + ":" + filterDescription + ":" + filterType
                )
                comment = f"Created: {time.strftime('%Y/%m/%d-%H:%M:%S')}"
                if filterType == "FAIL!":
                    print(filterType.upper())
                elif filterType.upper() == "ALERT":
                    resource_type = gmp.types.FilterType.ALERT
                elif filterType.upper() == "ASSET":
                    resource_type = gmp.types.FilterType.ASSET
                elif filterType.upper() == "CONFIG":
                    resource_type = gmp.types.FilterType.SCAN_CONFIG
                elif filterType.upper() == "CREDENTIAL":
                    resource_type = gmp.types.FilterType.CREDENTIAL
                elif filterType.upper() == "HOST":
                    resource_type = gmp.types.FilterType.HOST
                elif filterType.upper() == "SECINFO":
                    resource_type = gmp.types.FilterType.ALL_SECINFO
                elif filterType.upper() == "NOTE":
                    resource_type = gmp.types.FilterType.NOTE
                elif filterType.upper() == "OS":
                    resource_type = gmp.types.FilterType.OPERATING_SYSTEM
                elif filterType.upper() == "OVERRIDE":
                    resource_type = gmp.types.FilterType.OVERRIDE
                elif filterType.upper() == "PERMISSION":
                    resource_type = gmp.types.FilterType.PERMISSION
                elif filterType.upper() == "PORT_LIST":
                    resource_type = gmp.types.FilterType.PORT_LIST
                elif filterType.upper() == "REPORT":
                    resource_type = gmp.types.FilterType.REPORT
                elif filterType.upper() == "REPORT_FORMAT":
                    resource_type = gmp.types.FilterType.REPORT_FORMAT
                elif filterType.upper() == "RESULT":
                    resource_type = gmp.types.FilterType.RESULT
                elif filterType.upper() == "ROLE":
                    resource_type = gmp.types.FilterType.ROLE
                elif filterType.upper() == "SCHEDULE":
                    resource_type = gmp.types.FilterType.SCHEDULE
                elif filterType.upper() == "TAG":
                    resource_type = gmp.types.FilterType.TAG
                elif filterType.upper() == "TARGET":
                    resource_type = gmp.types.FilterType.TARGET
                elif filterType.upper() == "TASK":
                    resource_type = gmp.types.FilterType.TASK
                elif filterType.upper() == "TICKET":
                    resource_type = gmp.types.FilterType.TICKET
                elif filterType.upper() == "TLS_CERTIFICATE":
                    resource_type = gmp.types.FilterType.TLS_CERTIFICATE
                elif filterType.upper() == "USER":
                    resource_type = gmp.types.FilterType.USER
                elif filterType.upper() == "VULNERABILITY":
                    resource_type = gmp.types.FilterType.VULNERABILITY
                else:
                    print(
                        "FilterType: " + filterType.upper() + " Not supported"
                    )
                try:
                    if filter_id(gmp, filterNameFull):
                        print(
                            f"Filter: {filterNameFull} exist, not creating..."
                        )
                        continue

                    print("Creating filter: " + filterNameFull)
                    gmp.create_filter(
                        name=filterNameFull,
                        comment=comment,
                        filter_type=resource_type,
                        term=filterTerm,
                    )
                    numberfilters = numberfilters + 1
                except GvmResponseError as gvmerr:
                    print(f"{gvmerr=}, name: {filterNameFull}")
                    pass
        csvFile.close()  # close the csv file
    except IOError as e:
        error_and_exit(f"Failed to read filter_csv_file: {str(e)} (exit)")

    if len(row) == 0:
        error_and_exit("filter file is empty (exit)")

    return numberfilters


def main(gmp: Gmp, args: Namespace) -> None:
    # pylint: disable=undefined-variable
    if args.script:
        args = args.script[1:]

    parsed_args = parse_args(args=args)

    print("Creating filters.\n")

    numberfilters = create_filters(
        gmp,
        parsed_args.filters_csv_file,
    )

    numberfilters = str(numberfilters)
    print("    [" + numberfilters + "] filter(s) created!\n")


if __name__ == "__gmp__":
    main(gmp, args)