112 lines
2.9 KiB
Python
Executable file
112 lines
2.9 KiB
Python
Executable file
#! /usr/bin/env python
|
|
# -*- coding: iso-8859-15 -*-
|
|
"""
|
|
Manipulation d'IPv4
|
|
|
|
Copyright (C) Frédéric Pauget
|
|
Licence : GPLv2
|
|
"""
|
|
|
|
|
|
# Pour accélérer QuadToDec...
|
|
__QuadToDecDone = {}
|
|
|
|
def QuadToDec(ip) :
|
|
"""
|
|
Retourne la représentation décimale d'une ip
|
|
ip est de la forme xxx.xxx.xxx.xxx
|
|
"""
|
|
if ip in __QuadToDecDone:
|
|
return __QuadToDecDone[ip]
|
|
test = ip.split('.')
|
|
if len(test)!=4 : raise ValueError('IP Invalide')
|
|
ip_dec = 0
|
|
for z in range(0,4) :
|
|
n = int(test[z])
|
|
if n<0 or n>255 : raise ValueError('IP Invalide')
|
|
ip_dec += n * ( 256**(3-z) )
|
|
|
|
__QuadToDecDone[ip] = ip_dec
|
|
return ip_dec
|
|
|
|
def DecToQuad(ip_dec) :
|
|
"""
|
|
Retourne la représentation habituelle d'une ip (xxx.xxx.xxx.xxx)
|
|
ip_dec est l'IP en base 10
|
|
"""
|
|
try :
|
|
return "%d.%d.%d.%d" % ( \
|
|
ip_dec/(256**3) ,
|
|
(ip_dec%(256**3)) / (256**2) ,
|
|
( (ip_dec%(256**3)) % (256**2) ) / 256 ,
|
|
( (ip_dec%(256**3)) % (256**2) ) % 256 )
|
|
except :
|
|
raise ValueError('IP Invalide')
|
|
|
|
def param(net) :
|
|
"""
|
|
net est un résau fourni sous la forme xxx.xxx.xxx.xxx/yy
|
|
si donnée valide retourne un dictionnaire :
|
|
{ 'network' : xxx.xxx.xxx.xxx ,
|
|
'netmask' : yyy.yyy.yyy.yyy ,
|
|
'broadcast' : zzz.zzz.zzz.zzz }
|
|
sinon retourne {}
|
|
"""
|
|
reseau = {}
|
|
ip, mask = net.split('/')
|
|
|
|
try :
|
|
mask = int(mask)
|
|
dec_ip = QuadToDec(ip)
|
|
if dec_ip == -1 : raise
|
|
except :
|
|
return {}
|
|
|
|
# Calcul du netmask
|
|
dec_netmask=0
|
|
non_dec_netmask=0 # On calcule aussi le complémentaire
|
|
for i in range(0,32) :
|
|
if i < mask :
|
|
dec_netmask += 2**(31-i)
|
|
else :
|
|
non_dec_netmask += 2**(31-i)
|
|
|
|
reseau['netmask'] = DecToQuad(dec_netmask)
|
|
|
|
# Calcul du network
|
|
reseau['network'] = DecToQuad( dec_ip & dec_netmask )
|
|
|
|
# Calcul du broadcast
|
|
reseau['broadcast'] = DecToQuad( dec_ip | non_dec_netmask )
|
|
return reseau
|
|
|
|
def AddrInNet(ip,net) :
|
|
"""
|
|
ip est de la forme xxx.xxx.xxx.xxx
|
|
net est de la forme xxx.xxx.xxx.xxx/yy
|
|
net peut être une liste de chaînes ci-dessus
|
|
Retourne True si l'ip est dans un des réseaux.
|
|
Note : retourne False si l'IP est une adresse de réseau ou broadcast
|
|
"""
|
|
if type(net)==str : net = [ net ]
|
|
|
|
r = False
|
|
for ne in net :
|
|
n = param(ne)
|
|
if ip == n['broadcast'] or ip ==n['network'] :
|
|
return False
|
|
r = r or QuadToDec(n['netmask']) & QuadToDec(ip) == QuadToDec(n['network'])
|
|
|
|
return r
|
|
|
|
def AddrInNets(ip,nets) :
|
|
""" Vérifie si l'ip est au moins dans un des réseaux
|
|
de la liste nets (voir AddrInNet) """
|
|
for net in nets :
|
|
if AddrInNet(ip,net) :
|
|
return True
|
|
return False
|
|
|
|
if __name__ == '__main__' :
|
|
import sys
|
|
print param(sys.argv[1])
|