3
ܓiA" @ s( d dl Z d dlmZ G dd deZdS ) N)SoSMapc s e Zd ZdZdddddddd gZi Ze Zd
Zd d d
dgZ dZ
dZ fddZ dd Z
dd Zdd Zdd Zdd Zd d Zd d Zd!d" Zd#d$ Z ZS )%SoSIPMapa A mapping store for IP addresses
Each IP address added to this map is chcked for subnet membership. If that
subnet already exists in the map, then IP addresses are deterministically
generated sequentially within that subnet. For example, if a given IP is
matched to subnet 192.168.1.0/24 then 192.168.1 may be obfuscated to
100.11.12.0/24. Each IP address in the original 192.168.1.0/24 subnet
will then be assigned an address in 100.11.12.0/24 sequentially, such as
100.11.12.1, 100.11.12.2, etc...
Internally, the ipaddress library is used to manipulate the address objects
however, when retrieved by SoSCleaner any values will be strings.
z127.*z::1z0\.(.*)?z1\.(.*)?z8.8.8.8z8.8.4.4z 169.254.*z255.*d Z127Z169Z172Z192Fl "X c s< x*|j D ] }| jj|jdddd q
W t j| dS )z^Override the base conf_update() so that we can add items into
obfuscated_ips.
/ )maxsplitr N)valuesobfuscated_ipsaddsplitsuper conf_update)selfconfigvalue) __class__ /usr/lib/python3.6/ip_map.pyr
6 s zSoSIPMap.conf_updatec C s( | j j|jdddd || j|< d S )Nr r )r r )r r
r dataset)r itemr r r r insert_to_dataset> s zSoSIPMap.insert_to_datasetc C s t |jdddd }|| jkS )a There are multiple ways in which an ip address could be handed to us
in a way where we're matching against a previously obfuscated address.
Here, match the ip address to any of the obfuscated addresses we've
already created
r r )r r )strr r )r ZipaddrZaddr_strr r r
ip_in_datasetB s zSoSIPMap.ip_in_datasetc C s d}|j |r |jdj|}|| jkr2| j| S | j|sF| j|rJ|S d|krx.| jj D ] \}}|j |r^|jdd S q^W | j|S )zEnsure that when requesting an obfuscated address, we return a str
object instead of an IPv(4|6)Address object
r =]) r )r r r r )
startswithlstripjoinr Z ignore_itemr itemsr r
)r r Z
filt_startkeyr r r r getL s
z SoSIPMap.getc C sb g }x8| j D ].}|j|jkr&||_dS |j|kr |j| q W |r^|jdd dd |d |_dS )a/ Determine if a given address is in a subnet of an already obfuscated
network and if it is, then set the address' network to the network
object we're tracking. This allows us to match ip addresses with or
without a CIDR notation and maintain proper network relationships.
Nc S s | j S )N) prefixlen)nr r r w s z;SoSIPMap.set_ip_cidr_from_existing_subnet..T)r! reverser ) _networksipbroadcast_addressnetworkappendsort)r addrZnetsZnetr r r set_ip_cidr_from_existing_subnete s
z)SoSIPMap.set_ip_cidr_from_existing_subnetc C sf yt j|}W n" tk
r0 | jj| Y nX |j}t|jdkrR| j| n
| j | | j
|S )ziGiven an IP address, sanitize it to an obfuscated network or host
address as appropriate
z255.255.255.255) ipaddressZ ip_interface
ValueErrorignore_matchesr+ r* r Znetmaskr. sanitize_networksanitize_ipaddr)r r r- r* r r r
sanitize_itemz s
zSoSIPMap.sanitize_itemc C s || j kr| j| dS )zObfuscate the network address provided, and if there are host bits
in the address then obfuscate those as well
N)r' _new_obfuscated_network)r r* r r r r2 s
zSoSIPMap.sanitize_networkc C sh |j | jkr`| j|j }|j|j jkr0t|jS x.|j D ]"}| j|s:t| d|j S q:W | j S )zEObfuscate the IP address within the known obfuscated network
r ) r* r' r( r) r Zhostsr r# _new_obfuscated_single_address)r r- _obf_networkZ_ipr r r r3 s
zSoSIPMap.sanitize_ipaddrc C s | j d7 _ x | j d d kr.| j d7 _ qW | j d? d| j d? d d| j d? d d| j d }|| jj kr| j S |S )
Nr r . )r r9 )
_saddr_cntr r r6 )r Z_addrr r r r6 s :z'SoSIPMap._new_obfuscated_single_addressc C s d}t |tjrh| j| jkr*| jd7 _| j d}|jjdd }tj| d| }| jd7 _t |tjrt|rt|}|| j |< || j
t|< | j j |jdddd dS )a Generate an obfuscated network address for the network address given
which will allow us to maintain network relationships without divulging
actual network details
Positional arguments:
:param network: An ipaddress.IPv{4|6)Network object
Nr z.0.0.0r )r r )
isinstancer/ Z IPv4Networknetwork_first_octetskip_network_octetsZ with_netmaskr Z IPv6Networkr r' r r r
)r r* r7 Z _obf_addressZ _obf_maskZ_obf_network_sr r r r5 s
z SoSIPMap._new_obfuscated_network)__name__
__module__ __qualname____doc__r1 r' setr r@ rA Zcompile_regexesr> r
r r r" r. r4 r2 r3 r6 r5
__classcell__r r )r r r s2
r )r/ Zsos.cleaner.mappingsr r r r r r s