gvm-tools 23.10.0__tar.gz → 23.11.0__tar.gz

{gvm_tools-23.10.0 → gvm_tools-23.11.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: gvm-tools
-Version: 23.10.0
+Version: 23.11.0
 Summary: Tools to control a GSM/GVM over GMP or OSP
 Home-page: https://github.com/greenbone/gvm-tools/
 License: GPL-3.0-or-later
@@ -18,6 +18,7 @@ Classifier: Programming Language :: Python :: 3
 Classifier: Programming Language :: Python :: 3.9
 Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
 Classifier: Topic :: Software Development :: Libraries :: Python Modules
 Requires-Dist: python-gvm (>=23.4.2)
 Project-URL: Documentation, https://greenbone.github.io/gvm-tools/

{gvm_tools-23.10.0 → gvm_tools-23.11.0}/gvmtools/__version__.py

@@ -2,4 +2,4 @@
 
 # THIS IS AN AUTOGENERATED FILE. DO NOT TOUCH!
 
-__version__ = "23.10.0"
+__version__ = "23.11.0"

{gvm_tools-23.10.0 → gvm_tools-23.11.0}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "poetry.core.masonry.api"
 
 [tool.poetry]
 name = "gvm-tools"
-version = "23.10.0"
+version = "23.11.0"
 authors = ["Greenbone AG <info@greenbone.net>"]
 description = "Tools to control a GSM/GVM over GMP or OSP"
 license = "GPL-3.0-or-later"

{gvm_tools-23.10.0 → gvm_tools-23.11.0}/scripts/generate-random-reports.gmp.py

@@ -1,5 +1,5 @@
 # -*- coding: utf-8 -*-
-# Copyright (C) 2017-2021 Greenbone AG
+# Copyright (C) 2017-2023 Greenbone AG
 #
 # SPDX-License-Identifier: GPL-3.0-or-later
 #
@@ -20,20 +20,20 @@
 
 import json
 import textwrap
-import time
 from argparse import ArgumentParser, Namespace, RawTextHelpFormatter
 from pathlib import Path
 from random import choice, gauss, randrange, seed
+from datetime import datetime, timedelta
 
 from gvm.protocols.gmp import Gmp
 from lxml import etree as e
 
 from gvmtools.helper import generate_id, generate_random_ips, generate_uuid
 
-__version__ = "0.2.0"
+__version__ = "0.3.1"
 
 HELP_TEXT = f"""
-    Random Report Generation Script {__version__} (C) 2017-2021 Greenbone AG
+    Random Report Generation Script {__version__} (C) 2017-2023 Greenbone AG
 
     This program is free software: you can redistribute it and/or modify
     it under the terms of the GNU General Public License as published by
@@ -59,6 +59,17 @@ voluptate velit esse cillum dolore eu fugiat nulla pariatur.
 Excepteur sint occaecat cupidatat non proident, sunt in culpa qui officia
 deserunt mollit anim id est laborum."""
 
+local_date_time = datetime.now()
+
+
+def calculate_timestamp(delta: int = 0) -> str:
+    return (
+        (local_date_time + timedelta(seconds=delta))
+        .astimezone()
+        .replace(microsecond=0)
+        .isoformat()
+    )
+
 
 def generate_ports(n_ports):
     protocol = ["/tcp", "/udp"]
@@ -164,17 +175,9 @@ def generate_result_elem(
     e.SubElement(own, "name").text = generate_id()
 
     elem = e.Element("modification_time")
-    e.SubElement(result_elem, "modification_time").text = (
-        time.strftime("%Y-%m-%dT%H:%M:%S%z", time.localtime(time.time()))[:-2]
-        + ":00"
-    )  # Hell of a Timeformat :D
+    e.SubElement(result_elem, "modification_time").text = calculate_timestamp(0)
     e.SubElement(result_elem, "comment").text = ""
-    e.SubElement(result_elem, "creation_time").text = (
-        time.strftime("%Y-%m-%dT%H:%M:%S%z", time.localtime(time.time() - 20))[
-            :-2
-        ]
-        + ":00"
-    )
+    e.SubElement(result_elem, "creation_time").text = calculate_timestamp(-20)
 
     host_elem = e.Element("host")
     host_elem.text = host_ip
@@ -260,18 +263,8 @@ def generate_host_elem(
     e.SubElement(host_elem, "ip").text = host_ip
     e.SubElement(host_elem, "asset", {"asset_id": host_asset}).text = ""
 
-    e.SubElement(host_elem, "start").text = (
-        time.strftime(
-            "%Y-%m-%dT%H:%M:%S%z", time.localtime(time.time() - 1000)
-        )[:-2]
-        + ":00"
-    )
-    e.SubElement(host_elem, "end").text = (
-        time.strftime("%Y-%m-%dT%H:%M:%S%z", time.localtime(time.time() - 30))[
-            :-2
-        ]
-        + ":00"
-    )
+    e.SubElement(host_elem, "start").text = calculate_timestamp(-1000)
+    e.SubElement(host_elem, "end").text = calculate_timestamp(-30)
 
     app = choice(list(data["apps"]))
     os = choice(list(data["oss"]))
@@ -304,7 +297,7 @@ def generate_host_elem(
     host_elem.append(
         generate_host_detail_elem(
             "best_os_txt",
-            list(os)[0],
+            os,
             source_name=oid,
             source_description="Host Details",
             source_type="nvt",
@@ -359,22 +352,58 @@ def generate_reports(task, n_reports, n_results, with_gauss, **kwargs):
     return reports
 
 
-def generate_data(gmp, n_tasks, **kwargs):
+def generate_target():
+    target_name = f"Random_Report_Generation_Target_{calculate_timestamp(0)}"
+    target_comment = (
+        f"Generated by Random Report Generation Script Version {__version__}."
+    )
+    target_hosts = ["198.18.0.0/24"]
+    target_port_list_id = "c7e03b6c-3bbe-11e1-a057-406186ea4fc5"
+
+    gmp.create_target(
+        name=target_name,
+        comment=target_comment,
+        hosts=target_hosts,
+        port_list_id=target_port_list_id,
+    )
+
+
+def generate_data(gmp, n_tasks, task_type, **kwargs):
+    if task_type == "scan":
+        scanner_id = "08b69003-5fc2-4037-a479-93b440211c73"
+        config_id = "085569ce-73ed-11df-83c3-002264764cea"
+        generate_target()
+        target_id = gmp.get_targets(
+            filter_string=f'"Random_Report_Generation_Target_{calculate_timestamp(0)}"'
+        ).xpath("//@id")[0]
+
     for i in range(n_tasks):
         index = f"{{0:0>{len(str(n_tasks))}}}"
-        task_name = f"Task_for_GenReport:_{index.format(i + 1)}"
+        task_name = f"Random_Report_Generation_{task_type.capitalize()}_Task_{calculate_timestamp(0)}_{index.format(i + 1)}"
+        task_comment = f"Generated by Random Report Generation Script Version {__version__}."
 
-        gmp.create_container_task(task_name)
+        gmp.create_container_task(name=task_name, comment=task_comment)
 
         task_id = gmp.get_tasks(filter_string=f"name={task_name}").xpath(
             "//@id"
         )[0]
 
+        if task_type == "scan":
+            gmp.modify_task(task_id=task_id, alterable=1)
+
         reports = generate_reports(task=(task_id, task_name), **kwargs)
 
         for report in reports[0:]:
             gmp.import_report(report, task_id=task_id, in_assets=True)
 
+        if task_type == "scan":
+            gmp.modify_task(
+                task_id=task_id,
+                target_id=target_id,
+                scanner_id=scanner_id,
+                config_id=config_id,
+            )
+
 
 def main(gmp: Gmp, args: Namespace) -> None:
     # pylint: disable=undefined-variable, line-too-long
@@ -474,6 +503,16 @@ def main(gmp: Gmp, args: Namespace) -> None:
         "--seed", help="RNG Seed, in case the same data should be generated."
     )
 
+    parser.add_argument(
+        "--task-type",
+        dest="task_type",
+        type=str,
+        choices=["container", "scan"],
+        default="container",
+        help="Type of Task(s) to store the generated Reports. Can either "
+        "be 'container' or 'scan', default: 'container'.",
+    )
+
     script_args = parser.parse_args(args.script_args)
 
     if not script_args.seed:
@@ -484,7 +523,7 @@ def main(gmp: Gmp, args: Namespace) -> None:
     with open(str(script_args.datafile), encoding="utf-8") as file:
         data = json.load(file)
 
-    print("\n  Generating randomized data(s)...\n")
+    print("\n  Generating randomized data...\n")
 
     generate_data(
         gmp,
@@ -497,9 +536,10 @@ def main(gmp: Gmp, args: Namespace) -> None:
         data=data,
         with_gauss=script_args.with_gauss,
         with_descriptions=script_args.with_descriptions,
+        task_type=script_args.task_type,
     )
 
-    print("\n  Generation done.\n")
+    print("\n  Generation completed.\n")
 
 
 if __name__ == "__gmp__":