ViPNet_IPCalc/main.py at master · MaxSH97/ViPNet_IPCalc · GitHub

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
#!/usr/bin/env python3.7
# -*- coding: utf-8 -*-

import ipaddress


class RealIPNotInRange(Exception):
    def __init__(self, text):
        self.text = text


def get_virt_ip(real_ip, r_real_ip_s, r_real_ip_e, r_virt_ip_s):
    real_ip = int(ipaddress.IPv4Address(real_ip))
    r_real_ip_s = int(ipaddress.IPv4Address(r_real_ip_s))
    r_real_ip_e = int(ipaddress.IPv4Address(r_real_ip_e))
    r_virt_ip_s = int(ipaddress.IPv4Address(r_virt_ip_s))

    d_result = r_real_ip_e - real_ip
    r_result = r_real_ip_e - r_real_ip_s

    if real_ip < r_real_ip_s or real_ip > r_real_ip_e:
        raise RealIPNotInRange('Real IP address not in range of Real IP addresses')

    r_virt_ip_e = ipaddress.IPv4Address(r_virt_ip_s + r_result)

    virt_ip = ipaddress.IPv4Address(int(r_virt_ip_e) - d_result)
    return str(virt_ip), str(r_virt_ip_e)


real_ip = input('Real IP address: ')
real_ip = int(ipaddress.IPv4Address(real_ip))

r_real_ip_s = input('Start range of Real IP addresses: ')
r_real_ip_s = int(ipaddress.IPv4Address(r_real_ip_s))

r_real_ip_e = input('End range of Real IP addresses: ')
r_real_ip_e = int(ipaddress.IPv4Address(r_real_ip_e))

d_result = r_real_ip_e - real_ip
r_result = r_real_ip_e - r_real_ip_s

r_virt_ip_s = input('Start range of Virtual IP addresses: ')
r_virt_ip_s = int(ipaddress.IPv4Address(r_virt_ip_s))

r_virt_ip_e = int(ipaddress.IPv4Address(r_virt_ip_s + r_result))

virt_ip = ipaddress.IPv4Address(r_virt_ip_e - d_result)

result = '\nReal IP address: ' + str(ipaddress.IPv4Address(real_ip)) + '\nRange of Real IP addresses: ' \
         + str(ipaddress.IPv4Address(r_real_ip_s)) + ' - ' + str(ipaddress.IPv4Address(r_real_ip_e)) \
         + '\nRange of Virtual IP addresses: ' + str(ipaddress.IPv4Address(r_virt_ip_s)) + ' - ' \
         + str(ipaddress.IPv4Address(r_virt_ip_e)) + '\nVirtual IP address: ' + str(virt_ip)
print(result)