EVOLUTION-MANAGER
Edit File: ip_map.cpython-36.pyc
3 �f� � @ s0 d dl Z d dlZd dlmZ G dd� de�ZdS )� N)�SoSMapc @ s| e Zd ZdZdddddddd gZi Zd Zd d d dgZdZdd� Z dd� Z dd� Z dd� Z dd� Z dd� Zd d � Zd d � Zd S )!�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 Z127Z169Z172Z192Fc C s< x6| j j� D ](}t|�jd�d |jd�d kr dS q W dS )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 TF)�dataset�values�str�split)�self�ipaddr�_ip� r � /usr/lib/python3.6/ip_map.py� ip_in_dataset1 s zSoSIPMap.ip_in_datasetc C s� d}|j |�r |jdj|��}|| jj� kr6| j| S | j|�sJ| j|�rN|S d|kr�x0| jj� D ]"}|j |�rb| j| jd�d S qbW | 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 ) � startswith�lstrip�joinr �keysZ ignore_itemr r �add)r r Z filt_start�keyr r r �get= 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 �<lambda>h s z;SoSIPMap.set_ip_cidr_from_existing_subnet.<locals>.<lambda>T)r �reverser )� _networks�ip�broadcast_address�network�append�sort)r �addrZnetsZnetr r r � set_ip_cidr_from_existing_subnetV s z)SoSIPMap.set_ip_cidr_from_existing_subnetc C sf yt j|�}W n" tk r0 | jj|� � Y nX |j}t|j�dkrR| 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� ValueError�ignore_matchesr# r" r Znetmaskr& �sanitize_network�sanitize_ipaddr)r �itemr% r" r r r � sanitize_itemk 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 r* � s zSoSIPMap.sanitize_networkc C sf |j | jkr^| j|j }|j|j jkr0t|j�S x,|j� D ] }| j|�s:dt|�|jf S q:W | j� S )zEObfuscate the IP address within the known obfuscated network z%s/%s) r" r r r! r Zhostsr r � _new_obfuscated_single_address)r r% � _obf_networkr r r r r+ � s zSoSIPMap.sanitize_ipaddrc C s( dd� }|� }|| j j� kr$| j� S |S )Nc S s6 g } x$t dd�D ]}| jtjdd�� qW dt| � S )Nr � � �c z %s.%s.%s.%s)�ranger# �randomZrandint�tuple)Z_octets�ir r r � _gen_address� s z=SoSIPMap._new_obfuscated_single_address.<locals>._gen_address)r r r/ )r r8 Z_addrr r r r/ � s z'SoSIPMap._new_obfuscated_single_addressc C s� d}t |tj�rd| j| jkr*| jd7 _d| j }|jjd�d }tjd||f �}| jd7 _t |tj�rp|r�|| j|< t |�| j t |�< 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 N� z%s.0.0.0r z%s/%s) � isinstancer' Z IPv4Network�network_first_octet�skip_network_octetsZ with_netmaskr Z IPv6Networkr r r )r r" r0 Z _obf_addressZ _obf_maskr r r r. � s z SoSIPMap._new_obfuscated_networkN)�__name__� __module__� __qualname__�__doc__r) r r; r< Zcompile_regexesr r r& r- r* r+ r/ r. r r r r r s* r )r' r5 Zsos.cleaner.mappingsr r r r r r �<module> s