Issues (69)

scripts/gen-random-targets.gmp.py (2 issues)

1
# -*- coding: utf-8 -*-
2
# Copyright (C) 2017-2021 Greenbone Networks GmbH
3
#
4
# SPDX-License-Identifier: GPL-3.0-or-later
5
#
6
# This program is free software: you can redistribute it and/or modify
7
# it under the terms of the GNU General Public License as published by
8
# the Free Software Foundation, either version 3 of the License, or
9
# (at your option) any later version.
10
#
11
# This program is distributed in the hope that it will be useful,
12
# but WITHOUT ANY WARRANTY; without even the implied warranty of
13
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
14
# GNU General Public License for more details.
15
#
16
# You should have received a copy of the GNU General Public License
17
# along with this program.  If not, see <http://www.gnu.org/licenses/>.
18
19
import sys
20
from random import choice, gauss
21
22
from gvmtools.helper import generate_random_ips
23
24
25
def check_args(args):
26
    len_args = len(args.script) - 1
27
    if len_args < 2:
28
        message = """
29
        This script generates random task data and feeds it to\
30
    a desired GSM
31
        It needs two parameters after the script name.
32
33
        1. <host_number> -- number of dummy hosts to select from
34
        2. <number>      -- number of targets to be generated
35
36
        In addition, if you would like for the number of targets generated
37
    to be randomized on a Gaussian distribution, add 'with-gauss'
38
39
        Example:
40
            $ gvm-script --gmp-username name --gmp-password pass \
41
    ssh --hostname <gsm> scripts/gen-random-targets.gmp.py 3 40 with-gauss
42
        """
43
        print(message)
44
        sys.exit()
45
46
47
def generate(gmp, args, n_targets, n_ips):
48
    ips = generate_random_ips(n_ips)
49
50
    if 'with-gauss' in args.script:
51
        n_targets = int(gauss(n_targets, 2))
52
53
    for i in range(n_targets):
54
        host_ip = choice(ips)
55
        index = '{{0:0>{}}}'.format(len(str(n_targets)))
56
        name = 'Target_{}'.format(index.format(i + 1))
57
58
        gmp.create_target(name=name, make_unique=True, hosts=[host_ip])
59
60
61
def main(gmp, args):
62
    # pylint: disable=undefined-variable
63
64
    check_args(args)
65
66
    host_number = int(args.script[1])
67
    number_targets = int(args.script[2])
68
69
    print('Generating random targets...')
70
71
    generate(gmp, args, number_targets, host_number)
72
73
74
if __name__ == '__gmp__':
75
    main(gmp, args)
0 ignored issues
show
Comprehensibility Best Practice introduced by
The variable gmp does not seem to be defined.
Loading history...
Comprehensibility Best Practice introduced by
The variable args does not seem to be defined.
Loading history...
76