blob: 1daa68363b1f996cf29317aa49504b2de1ff4244 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
|
import random
import socket
import struct
# Generator handler class
# Manage generators' activity through this class.
class Generator:
# Init
# Args : s=settings object
def __init__(self, s):
self.g = []
self.g.append(IPv4Generator())
# Get actives
# Return a list of all active generators.
# Args : None
def get_actives(self):
r = []
for g in self.g:
if g.active is True:
r.append(g)
return r
# Activate
# Manually activate a generator.
# Args : v=generator class name
def activate(self, v):
for g in self.g:
if g.__class__.__name__ == v:
g.active = True
return True
return False
# Generate
# Generate an internet address with a random generator.
# Args : None
def generate(self):
return random.choice(self.get_actives()).generate()
# To string
# Return list of generators loaded as a string.
# Args : None
def tostring(self):
r = ""
for g in self.get_actives():
r += g.__class__.__name__ + " "
return r
# Base class for generators
# Specific generators are inherited from that class
class aGenerator:
# Init
# Args : None
def __init__(self):
self.active = False
self.max = 0
# IP address generator class
# Generates IP addresses.
class IPv4Generator(aGenerator):
# Init
# Args : None
def __init__(self):
aGenerator.__init__(self)
self.max = (255**4)
# Generate
# Generate a random IP address.
# Args : None
def generate(self):
return socket.inet_ntoa(struct.pack('>I', random.randint(1, 0xffffffff)))
|