U
    w7e                     @   s  U d dl Z d dlZd dlZd dlmZmZmZmZ d dlm	Z	m
Z
 d dlmZmZmZmZmZmZmZmZmZ er~d dlmZ eeZdZdddd	giZd
ddddddddddddddgZdddddddd d!d"d#d$d%d&d'd(d)d*d+dd,d-dd.d/d0dd1d2Zeeeeef f ed3< e dd4d5d6Z!d7d8 Z"G d9d: d:e#Z$d;d< Z%G d=d> d>e&Z'G d?d@ d@Z(G dAdB dBe'dCZ)dDdE Z*dQdGdHZ+dIdJ Z,dKdL Z-dRe e.e(dNdOdPZ/dS )S    N)TYPE_CHECKINGAnyDictOptional)safeyamlutil)	find_interface_name_from_macget_interfaces_by_macipv4_mask_to_net_prefixipv6_mask_to_net_prefixis_ip_networkis_ipv4_networkis_ipv6_addressis_ipv6_networknet_prefix_to_ipv4_maskRenderer   versionconfignetwork_state	addressesdhcp4dhcp4-overridesdhcp6dhcp6-overridesgateway4gateway6
interfacesmatchmtunameserversrendererset-name	wakeonlan	accept-raz	ad-selectzarp-intervalzarp-ip-targetzarp-validatez
down-delayzfail-over-mac-policyz	lacp-ratezmii-monitor-intervalz	min-linksmodegratuitous-arpZprimaryzprimary-reselect-policyzup-delayztransmit-hash-policy)zbond-ad-selectzbond-arp-intervalzbond-arp-ip-targetzbond-arp-validatezbond-downdelayzbond-fail-over-maczbond-lacp-ratezbond-miimonzbond-min-linksz	bond-modezbond-num-grat-arpzbond-primaryzbond-primary-reselectzbond-updelayzbond-xmit-hash-policyzageing-timeZpriorityzforward-delayz
hello-timezmax-agez	path-costzport-prioritystp)Zbridge_ageingZbridge_bridgeprioZ	bridge_fdZbridge_gcintZbridge_helloZbridge_maxageZbridge_maxwaitZbridge_pathcostZbridge_portprio
bridge_stpZbridge_waitport)bondbridgeNET_CONFIG_TO_V2)diktreturnc                 C   s$   d| ksd| kr t jdddd dS )z8Warn about deprecations of v2 properties for all devicesr   r   z$The use of `gateway4` and `gateway6`z22.4zhFor more info check out: https://cloudinit.readthedocs.io/en/latest/topics/network-config-format-v2.html)Z
deprecatedZdeprecated_versionZextra_messageN)r   Z	deprecate)r-    r/   =/usr/lib/python3/dist-packages/cloudinit/net/network_state.pywarn_deprecated_all_devicesV   s    r1   c                 C   s    t | }|D ]}|| q|S N)setdiscard)ZexpectedZactualZmissingkeyr/   r/   r0   	diff_keysa   s    r6   c                   @   s   e Zd ZdS )InvalidCommandN)__name__
__module____qualname__r/   r/   r/   r0   r7   h   s   r7   c                    s    fdd}|S )Nc                    s   t   fdd}|S )Nc                    s4   r"t |}|r"td|f  | |f||S )Nz&Command missing %s of required keys %s)r6   r7   )selfcommandargskwargsmissing_keys)funcrequired_keysr/   r0   	decoratorn   s    
z7ensure_command_keys.<locals>.wrapper.<locals>.decorator)	functoolswraps)r@   rB   rA   )r@   r0   wrapperm   s    
z$ensure_command_keys.<locals>.wrapperr/   )rA   rF   r/   rE   r0   ensure_command_keysl   s    rG   c                       s    e Zd ZdZ fddZ  ZS )CommandHandlerMetaa  Metaclass that dynamically creates a 'command_handlers' attribute.

    This will scan the to-be-created class for methods that start with
    'handle_' and on finding those will populate a class attribute mapping
    so that those methods can be quickly located and called.
    c                    sb   i }|  D ]6\}}t|r|dr|tdd  }|r|||< q||d< tt| | |||S )NZhandle_command_handlers)itemscallable
startswithlensuperrH   __new__)clsnameparentsZdctrI   Z	attr_nameattrZhandles_what	__class__r/   r0   rO      s    
zCommandHandlerMeta.__new__)r8   r9   r:   __doc__rO   __classcell__r/   r/   rT   r0   rH   ~   s   rH   c                   @   s   e Zd ZefeedddZeedddZedd Z	ed	d
 Z
edd Zedd ZdddZdddZdd Zdd Zeed dddZdS )NetworkState)r   r   c                 C   s*   t || _|| _|dd| _d | _d S )Nuse_ipv6F)copydeepcopy_network_state_versiongetrY   _has_default_route)r;   r   r   r/   r/   r0   __init__   s    zNetworkState.__init__r.   c                 C   s
   | j d S )Nr   )r\   r;   r/   r/   r0   r      s    zNetworkState.configc                 C   s   | j S r2   )r]   rb   r/   r/   r0   r      s    zNetworkState.versionc                 C   s.   z| j d d W S  tk
r(   g  Y S X d S )Ndnsr!   r\   KeyErrorrb   r/   r/   r0   dns_nameservers   s    zNetworkState.dns_nameserversc                 C   s.   z| j d d W S  tk
r(   g  Y S X d S )Nrc   searchrd   rb   r/   r/   r0   dns_searchdomains   s    zNetworkState.dns_searchdomainsc                 C   s   | j d kr|  | _ | j S r2   )r_   _maybe_has_default_routerb   r/   r/   r0   has_default_route   s    

zNetworkState.has_default_routeNc                 c   s>   | j di }| D ]"}|d kr*|V  q||r|V  qd S )Nr   )r\   r^   values)r;   filter_funcZifacesifacer/   r/   r0   iter_interfaces   s    zNetworkState.iter_interfacesc                 c   s6   | j dg D ]"}|d k	r*||r0|V  q|V  qd S Nroutesr\   r^   )r;   rl   router/   r/   r0   iter_routes   s
    zNetworkState.iter_routesc                 C   sh   |   D ]}| |r dS q|  D ]<}|dg D ]*}|dg D ]}| |rF   dS qFq6q&dS )NTsubnetsrp   F)rs   _is_default_routern   r^   )r;   rr   rm   subnetr/   r/   r0   ri      s    

z%NetworkState._maybe_has_default_routec                 C   s    d}| ddko| d|kS )N)z::z0.0.0.0prefixr   network)r^   )r;   rr   Zdefault_netsr/   r/   r0   ru      s    zNetworkState._is_default_route)r   r.   c                 C   s(   i }d|kr|d |d< | d|if|S )zInstantiates a `NetworkState` without interpreting its data.

        That means only `config` and `version` are copied.

        :param network_state: Network state data.
        :return: Instance of `NetworkState`.
        r   r   r/   )rP   r   r>   r/   r/   r0   to_passthrough   s    	zNetworkState.to_passthrough)N)N)r8   r9   r:   NETWORK_STATE_VERSIONdictintr`   propertyr   r   rf   rh   rj   rn   rs   ri   ru   classmethodry   r/   r/   r/   r0   rX      s*    




	
rX   c                   @   s  e Zd Zi g g g ddddZeddfddddZeed	d
dZedd Z	e	j
dd Z	dd Zdd Zdd Zdd ZdJddZdKddZdLddZedgdd  Zedgd!d" Zedd#d$gd%d& Zedd'd(gd)d* Zedd+gd,d- Zedgd.d/ Zd0d1 Zed2gd3d4 Zed2gd5d6 Zed7gd8d9 Zd:d; Zd<d= Zd>d? Zd@dA Z dBdC Z!dd	dDdEZ"dMdFdGZ#dHdI Z$dS )NNetworkStateInterpreterr!   rg   FN)r   rp   rc   rY   r   zOptional[Renderer])r"   c                 C   s:   || _ || _t| j| _|| jd< d| _i | _|| _d S )Nr   F)	r]   _configrZ   r[   initial_network_stater\   _parsed_interface_dns_map	_renderer)r;   r   r   r"   r/   r/   r0   r`      s    
z NetworkStateInterpreter.__init__ra   c                 C   sH   ddl m} | jdkr8t| j|r8td t| j	S t| j
| jdS )Nr   r      zPassthrough netplan v2 configr   )cloudinit.net.netplanr   r]   
isinstancer   LOGdebugrX   ry   r   r\   )r;   NetplanRendererr/   r/   r0   r     s
    
z%NetworkStateInterpreter.network_statec                 C   s   | j dS NrY   rq   rb   r/   r/   r0   rY   
  s    z NetworkStateInterpreter.use_ipv6c                 C   s   | j d|i d S r   )r\   update)r;   valr/   r/   r0   rY     s    c                 C   s   | j | j| jd}t|S )N)r   r   r   )r]   r   r\   r   dumps)r;   stater/   r/   r0   dump  s
    zNetworkStateInterpreter.dumpc                 C   sv   d|krt d tdt|d  }t||}|rNd| }t | t|dd |D D ]}t| |||  q\d S )Nr   z$Invalid state, missing version fieldzInvalid state, missing keys: %sc                 S   s   g | ]}|d kr|qS )r   r/   ).0kr/   r/   r0   
<listcomp>'  s      z0NetworkStateInterpreter.load.<locals>.<listcomp>)r   error
ValueErrorNETWORK_STATE_REQUIRED_KEYSr6   setattr)r;   r   rA   r?   msgr5   r/   r/   r0   load  s    


zNetworkStateInterpreter.loadc                 C   s   t | jS r2   )r   r   r\   rb   r/   r/   r0   dump_network_state*  s    z*NetworkStateInterpreter.dump_network_statec                 C   s   | j | jdS )N)r   r   )r]   r   rb   r/   r/   r0   as_dict-  s    zNetworkStateInterpreter.as_dictTc                 C   s>   | j dkr| j|d d| _n| j dkr:| j|d d| _d S )Nr   skip_brokenTr   )r]   parse_config_v1r   parse_config_v2)r;   r   r/   r/   r0   parse_config0  s    

z$NetworkStateInterpreter.parse_configc                 C   s  | j D ]}|d }z| j| }W n0 tk
rP } ztd| |W 5 d }~X Y nX z|| | W q tk
r   |sx ntjd|dd t|   Y qX q| j	
 D ]l\}}d }z| jd | }W n2 tk
r } ztd||W 5 d }~X Y nX |r|\}	}
|	|
d|d	< qd S )
Ntypez"No handler found for  command '%s'Skipping invalid command: %sTexc_infor   zINameserver specified for interface {0}, but interface {0} does not exist!)r   rg   rc   )r   rI   re   RuntimeErrorr7   r   warningr   r   r   rJ   r\   r   format)r;   r   r<   command_typehandlere	interfacerc   rm   r!   rg   r/   r/   r0   r   8  sL    
  z'NetworkStateInterpreter.parse_config_v1c                 C   s   ddl m} t| j|rd S | j D ]\}}|dkr8q&z| j| }W n0 tk
rv } ztd| |W 5 d }~X Y nX z|| | | 	| W q& t
k
r   |s ntjd|dd t|   Y q&X q&d S )Nr   r   )r   r"   z!No handler found for command '%s'r   Tr   )r   r   r   r   r   rJ   rI   re   r   
_v2_commonr7   r   r   r   r   )r;   r   r   r   r<   r   r   r/   r/   r0   r   [  s4    
  z'NetworkStateInterpreter.parse_config_v2rQ   c                 C   s
   |  |S r2   handle_physicalr;   r<   r/   r/   r0   handle_loopbackw  s    z'NetworkStateInterpreter.handle_loopbackc           
      C   s(  | j di }||d i }|di  D ]\}}|||i q.t|d}| js|D ],}|ddst|dr^d| _ qq^|d	d
}|d
k	rt	|}|dd
}	|	d
k	rt	|	}	||d|d|ddd|dd
d
|||	d | j d |d|i | 
  d
S )z
        command = {
            'type': 'physical',
            'mac_address': 'c0:d6:9f:2c:e8:80',
            'name': 'eth0',
            'subnets': [
                {'type': 'dhcp4'}
             ],
            'accept-ra': 'true'
        }
        r   rQ   paramsrt   r   6addressTr%   Nr$   mac_addressinetZmanualr    )rQ   r   r   r   r&   r    r   gatewayrt   r%   r$   )r\   r^   rJ   r   _normalize_subnetsrY   endswithr   r   Zis_truer   )
r;   r<   r   rm   paramr   rt   rv   Z	accept_rar$   r/   r/   r0   r   {  sD    

z'NetworkStateInterpreter.handle_physicalvlan_id	vlan_linkc                 C   s\   | j di }| | ||di }|d|d< |d|d< ||d |i dS )z
        auto eth0.222
        iface eth0.222 inet static
                address 10.10.10.1
                netmask 255.255.255.0
                hwaddress ether BC:76:4E:06:96:B3
                vlan-raw-device eth0
        r   rQ   r   zvlan-raw-devicer   N)r\   r^   r   r   )r;   r<   r   rm   r/   r/   r0   handle_vlan  s    

z#NetworkStateInterpreter.handle_vlanbond_interfacesr   c           	      C   s  |  | | jd}||di }|d D ]\}}|||i q6|ddi | jd |d |i |dD ]}||kr|dd}|  | | jdi }||}|d|d	< |d D ]\}}|||i q| jd ||i q~d
S )aU  
        #/etc/network/interfaces
        auto eth0
        iface eth0 inet manual
            bond-master bond0
            bond-mode 802.3ad

        auto eth1
        iface eth1 inet manual
            bond-master bond0
            bond-mode 802.3ad

        auto bond0
        iface bond0 inet static
             address 192.168.0.10
             gateway 192.168.0.1
             netmask 255.255.255.0
             bond-slaves none
             bond-mode 802.3ad
             bond-miimon 100
             bond-downdelay 200
             bond-updelay 200
             bond-lacp-rate 4
        r   rQ   r   zbond-slavesZnoner   r*   )rQ   r   zbond-masterN)r   r\   r^   rJ   r   )	r;   r<   r   rm   r   r   ifnamecmdZbond_ifr/   r/   r0   handle_bond  s&    


z#NetworkStateInterpreter.handle_bondbridge_interfacesc           	      C   s  | j di }|dD ] }||kr&qd|i}| | q| j di }| | ||di }|d |d< |di  D ]\}}|||i q|d}|dk	rt|ts|dkrd	}n|d
krd}ntdj|d|d|i ||d |i dS )a  
            auto br0
            iface br0 inet static
                    address 10.10.10.1
                    netmask 255.255.255.0
                    bridge_ports eth0 eth1
                    bridge_stp off
                    bridge_fd 0
                    bridge_maxwait 0

        bridge_params = [
            "bridge_ports",
            "bridge_ageing",
            "bridge_bridgeprio",
            "bridge_fd",
            "bridge_gcint",
            "bridge_hello",
            "bridge_hw",
            "bridge_maxage",
            "bridge_maxwait",
            "bridge_pathcost",
            "bridge_portprio",
            "bridge_stp",
            "bridge_waitport",
        ]
        r   r   rQ   Zbridge_portsr   r)   N)Zon1r   T)Zoff0r   Fz2Cannot convert bridge_stp value ({stp}) to boolean)r(   )	r\   r^   r   rJ   r   r   boolr   r   )	r;   r<   r   r   r   rm   r   r   r)   r/   r/   r0   handle_bridge  s6     

z%NetworkStateInterpreter.handle_bridgec                 C   s   |  | d S r2   r   r   r/   r/   r0   handle_infiniband9  s    z)NetworkStateInterpreter.handle_infinibandc                 C   sx   g }g }d|kr<|d }t |ts(|g}|D ]}|| q,d|krp|d }t |ts\|g}|D ]}|| q`||fS )Nr   rg   )r   listappend)r;   r<   r!   rg   Zaddrsaddrpathspathr/   r/   r0   
_parse_dns=  s    

z"NetworkStateInterpreter._parse_dnsr   c                 C   sV   | j d}| |\}}d|kr6||f| j|d < n|d | |d | d S )Nrc   r   r!   rg   )r\   r^   r   r   extend)r;   r<   rc   r!   rg   r/   r/   r0   handle_nameserverN  s    z)NetworkStateInterpreter.handle_nameserverc                 C   s0   | j d}| |\}}||d|| d< d S )Nr   r   rc   )r\   r^   r   )r;   r<   rm   Z_ifacer!   rg   r/   r/   r0   _handle_individual_nameserver[  s    z5NetworkStateInterpreter._handle_individual_nameserverdestinationc                 C   s   | j d t| d S ro   )r\   r   _normalize_router   r/   r/   r0   handle_routea  s    z$NetworkStateInterpreter.handle_routec                 C   s   | j |dd dS )a  
        v2_command = {
          bond0: {
            'interfaces': ['interface0', 'interface1'],
            'parameters': {
               'mii-monitor-interval': 100,
               'mode': '802.3ad',
               'xmit_hash_policy': 'layer3+4'}},
          bond1: {
            'bond-slaves': ['interface2', 'interface7'],
            'parameters': {
                'mode': 1,
            }
          }
        }

        v1_command = {
            'type': 'bond'
            'name': 'bond0',
            'bond_interfaces': [interface0, interface1],
            'params': {
                'bond-mode': '802.3ad',
                'bond_miimon: 100,
                'bond_xmit_hash_policy': 'layer3+4',
            }
        }

        r*   cmd_typeN_handle_bond_bridger   r/   r/   r0   handle_bondsf  s    z$NetworkStateInterpreter.handle_bondsc                 C   s   | j |dd dS )a  
        v2_command = {
          br0: {
            'interfaces': ['interface0', 'interface1'],
            'forward-delay': 0,
            'stp': False,
            'maxwait': 0,
          }
        }

        v1_command = {
            'type': 'bridge'
            'name': 'br0',
            'bridge_interfaces': [interface0, interface1],
            'params': {
                'bridge_stp': 'off',
                'bridge_fd: 0,
                'bridge_maxwait': 0
            }
        }

        r+   r   Nr   r   r/   r/   r0   handle_bridges  s    z&NetworkStateInterpreter.handle_bridgesc                 C   s  t  }| D ]\}}ddi}|di }|dd}|sNtd|t| ||d< |}|d}	|	rn|	}n |r|r| }
t|
}|r|}||d	< |d
d}|rd
|i|d< dD ]}||kr|| ||< qt| | 	|}t
|dkr|d|i td| | | qdS )a  
        ethernets:
          eno1:
            match:
              macaddress: 00:11:22:33:44:55
              driver: hv_netvsc
            wakeonlan: true
            dhcp4: true
            dhcp6: false
            addresses:
              - 192.168.14.2/24
              - 2001:1::1/64
            gateway4: 192.168.14.1
            gateway6: 2001:1::2
            nameservers:
              search: [foo.local, bar.local]
              addresses: [8.8.8.8, 8.8.4.4]
          lom:
            match:
              driver: ixgbe
            set-name: lom1
            dhcp6: true
            accept-ra: true
          switchports:
            match:
              name: enp2*
            mtu: 1280

        command = {
            'type': 'physical',
            'mac_address': 'c0:d6:9f:2c:e8:80',
            'name': 'eth0',
            'subnets': [
                {'type': 'dhcp4'}
             ]
        }
        r   Zphysicalr   
macaddressNzHNetworkState Version2: missing "macaddress" info in config entry: %s: %sr   r#   rQ   driverr   )r    r   r$   r%   r   rt   z!v2(ethernets) -> v1(physical):
%s)r	   rJ   r^   r   r   strlowerr   r1   _v2_to_v1_ipcfgrM   r   r   )r;   r<   Zifaces_by_macZethcfgZphy_cmdr   r   rQ   Zset_nameZlcase_mac_addressZmacr   r5   rt   r/   r/   r0   handle_ethernets  sH    - 

z(NetworkStateInterpreter.handle_ethernetsc                 C   s   |  D ]x\}}d||d|dd}d|kr>|d |d< t| | |}t|dkrj|d|i td| | | qd	S )
aq  
        v2_vlans = {
            'eth0.123': {
                'id': 123,
                'link': 'eth0',
                'dhcp4': True,
            }
        }

        v1_command = {
            'type': 'vlan',
            'name': 'eth0.123',
            'vlan_link': 'eth0',
            'vlan_id': 123,
            'subnets': [{'type': 'dhcp4'}],
        }
        vlanidlink)r   rQ   r   r   r    r   rt   zv2(vlans) -> v1(vlan):
%sN)	rJ   r^   r1   r   rM   r   r   r   r   )r;   r<   r   r   Zvlan_cmdrt   r/   r/   r0   handle_vlans  s    
z$NetworkStateInterpreter.handle_vlansc                 C   s   t d d S )NzOWifi configuration is only available to distros with netplan rendering support.)r   r   r   r/   r/   r0   handle_wifis  s    z$NetworkStateInterpreter.handle_wifisc           
      C   s   t d| | D ]\}}d|kr6|d}|r6|}d|kr|ddg }|ddg }ddi}t|dkr|d|i t|dkr|d	|i | | |d
i d}|rt|}	|	r|	}| || qd S )Nzv2_common: handling config:
%sr#   r!   rg   r   r   Z
nameserverr   r   r   r   )	r   r   rJ   r^   rM   r   r   r   r   )
r;   r   rm   Zdev_cfgZset_name_ifacerg   rc   Zname_cmdr   Zreal_if_namer/   r/   r0   r   $  s.    

z"NetworkStateInterpreter._v2_commonc           
         s.  t dd t| D  | D ]\}}t dd | D }|di }|dd}|rh||d< d|d	||d
 |ddt  fdd| D i}d|kr|d |d< t| | |}	t|	dkr|d|	i t	
d||| |dkr| | q$|dkr| | q$tdj|dq$dS )z(Common handler for bond and bridge typesc                 s   s   | ]\}}||fV  qd S r2   r/   r   r   vr/   r/   r0   	<genexpr>C  s    z>NetworkStateInterpreter._handle_bond_bridge.<locals>.<genexpr>c                 s   s"   | ]\}}|t kr||fV  qd S r2   )NETWORK_V2_KEY_FILTER)r   r5   valuer/   r/   r0   r   H  s   Z
parameterszgratuitious-arpNr'   r   rQ   Z_interfacesr   r   c                 3   s   | ]\}} | |fV  qd S r2   r/   r   Zv2key_to_v1r/   r0   r   X  s     r    r   rt   zv2(%s) -> v1(%s):
%sr+   r*   z Unknown command type: {cmd_type}r   )r{   r,   r^   rJ   popr1   r   rM   r   r   r   r   r   r   r   )
r;   r<   r   Z	item_nameZitem_cfgZitem_paramsr   Z
grat_valueZv1_cmdrt   r/   r   r0   r   ?  sH        


z+NetworkStateInterpreter._handle_bond_bridgec              	   C   s  dd }g }| dr:ddi}|| di | || | drnddi}d| _|| di | || d	}d	}i }| d
g D ]}d|d}d|krd|kr|d	kr| d}|d|i n(d|kr|d	kr| d}|d|i d|krD|sD| d d
}	|	r|	|d< | d d}
|
r:|
|d< || || qg }| dg D ].}|t| d| d| dd q`t|rt|r||d d< |S )z7Common ipconfig extraction from v2 to v1 subnets array.c                 S   s   d| kr| d |d< d S )Nzroute-metricmetricr/   )Z	overridesrv   r/   r/   r0   _add_dhcp_overridesq  s    zDNetworkStateInterpreter._v2_to_v1_ipcfg.<locals>._add_dhcp_overridesr   r   r   r   Tr   Nr   static)r   r   :r   r   r   r!   rf   rg   
dns_searchrp   toZviar   )r   r   r   r   )r^   r   rY   r   r   rM   )r;   r   r   rt   rv   r   r   r!   r   r   rg   rp   rr   r/   r/   r0   r   n  s^    






z'NetworkStateInterpreter._v2_to_v1_ipcfg)T)T)T)N)%r8   r9   r:   r   rz   r`   r}   rX   r   rY   setterr   r   r   r   r   r   r   rG   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r/   r/   r/   r0   r      sl   



#


6

4

@



^"
/r   )	metaclassc                 C   sz   t | } tdd |  D }| ddkr@|t|dd dd | d	g D |d	< d
d }dD ]}||| qf|S )Nc                 s   s   | ]\}}|r||fV  qd S r2   r/   r   r/   r/   r0   r     s      z$_normalize_subnet.<locals>.<genexpr>r   )r   Zstatic6)r   Z
ip_addressaddress_keysc                 S   s   g | ]}t |qS r/   )r   )r   rr/   r/   r0   r     s    z%_normalize_subnet.<locals>.<listcomp>rp   c                 S   s*   || kr&t | | ts&| |  | |< d S r2   )r   r   split)ZsnetrQ   r/   r/   r0   listify  s    z"_normalize_subnet.<locals>.listify)r   rf   )rZ   r[   r{   rJ   r^   r   _normalize_net_keys)rv   Znormal_subnetr   r   r/   r/   r0   _normalize_subnet  s     
	

r   r/   c                 C   s  dd |   D }d}|D ]}||r|} q2q|sZdd|| f }t| t|t||}t|std| td| dt|}t	|}|d	}	d
|kr|
d
\}
}}|
||< |rt|}n*|rt|}ntd| td| dnRd|krt|d }n:|	r0|r0t|	}n$|	rF|rFt|	}n|rPdnd}d|krt|d t|krtd|| ||d< |rd	|kr|d	= n|rt|d |d	< |S )a  Normalize dictionary network keys returning prefix and address keys.

    @param network: A dict of network-related definition containing prefix,
        netmask and address_keys.
    @param address_keys: A tuple of keys to search for representing the address
        or cidr. The first address_key discovered will be used for
        normalization.

    @returns: A dict containing normalized prefix and matching addr_key.
    c                 S   s"   i | ]\}}|s|d kr||qS )r   r/   r   r/   r/   r0   
<dictcomp>  s
        z'_normalize_net_keys.<locals>.<dictcomp>Nz/No config network address keys [%s] found in %s,z$Address %s is not a valid ip networkzAddress z is not a valid ip addressnetmask/rw   @      z;Overwriting existing 'prefix' with '%s' in network info: %s)rJ   r^   joinr   r   r   r   r   r   r   	partitionr   r
   r|   r   r   )rx   r   ZnetZaddr_keyr5   messager   Zipv6Zipv4r   Z	addr_part_Zmaybe_prefixrw   r/   r/   r0   r     sb    








 
r   c              
   C   s   t dd |  D }d|kr0|d |d< |d= |t|dd |d}|rzt||d< W n2 tk
r } ztd||W 5 d	}~X Y nX |S )
a  normalize a route.
    return a dictionary with only:
       'type': 'route' (only present if it was present in input)
       'network': the network portion of the route as a string.
       'prefix': the network prefix for address as an integer.
       'metric': integer metric (only if present in input).
       'netmask': netmask (string) equivalent to prefix iff network is ipv4.
    c                 s   s"   | ]\}}|d kr||fV  qdS )) NNr/   r   r/   r/   r0   r   $  s     z#_normalize_route.<locals>.<genexpr>r   rx   )rx   r   r   r   z(Route config metric {} is not an integerN)	r{   rJ   r   r   r^   r|   r   	TypeErrorr   )rr   Znormal_router   r   r/   r/   r0   r     s.    
 
r   c                 C   s   | sg } dd | D S )Nc                 S   s   g | ]}t |qS r/   )r   )r   sr/   r/   r0   r   ?  s     z&_normalize_subnets.<locals>.<listcomp>r/   )rt   r/   r/   r0   r   <  s    r   T)
net_configr   r.   c                 C   sf   d}|  d}|  d}|dkr$| }|rP|dk	rPt|||d}|j|d |j}|sbtd|  |S )zfParses the config, returns NetworkState object

    :param net_config: curtin network config dict
    Nr   r   r   )r   r   r"   r   zpNo valid network_state object created from network config. Did you specify the correct version? Network config:
)r^   r   r   r   r   )r
  r   r"   r   r   r   Znsir/   r/   r0   parse_net_config_dataB  s$    	

  r  )r/   )TN)0rZ   rC   Zloggingtypingr   r   r   r   Z	cloudinitr   r   Zcloudinit.netr   r	   r
   r   r   r   r   r   r   Zcloudinit.net.rendererr   Z	getLoggerr8   r   rz   r   r   r,   r   __annotations__r{   r1   r6   	Exceptionr7   rG   r   rH   rX   r   r   r   r   r   r   r  r/   r/   r/   r0   <module>   s    ,
 "V     O
L"  