o
    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g diZg d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d dd!d"Zeeeeef f ed#< d$e d%dfd&d'Z!d(d) Z"G d*d+ d+e#Z$d,d- Z%G d.d/ d/e&Z'G d0d1 d1Z(G d2d3 d3e'd4Z)d5d6 Z*dCd8d9Z+d:d; Z,d<d= Z-	>	dDd?e d@e.d%e(fdAdBZ/dS )E    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-arp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-timepriorityzforward-delayz
hello-timezmax-agez	path-costzport-prioritystp)bridge_ageingbridge_bridgeprio	bridge_fdbridge_gcintbridge_hellobridge_maxagebridge_maxwaitbridge_pathcostbridge_portprio
bridge_stpbridge_waitport)bondbridgeNET_CONFIG_TO_V2diktreturnc                 C   s(   d| v sd| v rt jdddd dS 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)
deprecateddeprecated_versionextra_messageN)r   	deprecate)r:    r@   =/usr/lib/python3/dist-packages/cloudinit/net/network_state.pywarn_deprecated_all_devicesV   s   
rB   c                 C   s    t | }|D ]}|| q|S N)setdiscard)expectedactualmissingkeyr@   r@   rA   	diff_keysa   s   rJ   c                   @   s   e Zd ZdS )InvalidCommandN)__name__
__module____qualname__r@   r@   r@   rA   rK   h   s    rK   c                    s    fdd}|S )Nc                    s   t   fdd}|S )Nc                    s:   rt |}|rtd|f  | |g|R i |S )Nz&Command missing %s of required keys %s)rJ   rK   )selfcommandargskwargsmissing_keys)funcrequired_keysr@   rA   	decoratorn   s   
z7ensure_command_keys.<locals>.wrapper.<locals>.decorator)	functoolswraps)rT   rV   rU   )rT   rA   wrapperm   s   
z$ensure_command_keys.<locals>.wrapperr@   )rU   rZ   r@   rY   rA   ensure_command_keysl   s   r[   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 ]\}}t|r!|dr!|tdd  }|r!|||< q||d< tt| | |||S )Nhandle_command_handlers)itemscallable
startswithlensuperr\   __new__)clsnameparentsdctr^   	attr_nameattrhandles_what	__class__r@   rA   rd      s   zCommandHandlerMeta.__new__)rL   rM   rN   __doc__rd   __classcell__r@   r@   rl   rA   r\   ~   s    r\   c                   @   s   e Zd ZefdedefddZedef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dedd fddZdS )NetworkStater   r   c                 C   s*   t || _|| _|dd| _d | _d S )Nuse_ipv6F)copydeepcopy_network_state_versiongetrq   _has_default_route)rO   r   r   r@   r@   rA   __init__   s   
zNetworkState.__init__r;   c                 C   s
   | j d S )Nr   )rt   rO   r@   r@   rA   r         
zNetworkState.configc                 C   s   | j S rC   )ru   ry   r@   r@   rA   r      s   zNetworkState.versionc                 C   (   z| j d d W S  ty   g  Y S w )Ndnsr"   rt   KeyErrorry   r@   r@   rA   dns_nameservers   
   zNetworkState.dns_nameserversc                 C   r{   )Nr|   searchr}   ry   r@   r@   rA   dns_searchdomains   r   zNetworkState.dns_searchdomainsc                 C   s   | j d u r
|  | _ | j S rC   )rw   _maybe_has_default_routery   r@   r@   rA   has_default_route   s   

zNetworkState.has_default_routeNc                 c   s@    | j di }| D ]}|d u r|V  q||r|V  qd S )Nr   )rt   rv   values)rO   filter_funcifacesifacer@   r@   rA   iter_interfaces   s   zNetworkState.iter_interfacesc                 c   s8    | j dg D ]}|d ur||r|V  q|V  qd S Nroutesrt   rv   )rO   r   router@   r@   rA   iter_routes   s   zNetworkState.iter_routesc                 C   sh   |   D ]
}| |r dS q|  D ]}|dg D ]}|dg D ]}| |r/   dS q#qqdS )NTsubnetsr   F)r   _is_default_router   rv   )rO   r   r   subnetr@   r@   rA   r      s   


z%NetworkState._maybe_has_default_routec                 C   s    d}| ddko| d|v S )N)z::z0.0.0.0prefixr   network)rv   )rO   r   default_netsr@   r@   rA   r      s   zNetworkState._is_default_routec                 C   s,   i }d|v r|d |d< | d|ifi |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@   )re   r   rR   r@   r@   rA   to_passthrough   s   	zNetworkState.to_passthroughrC   )rL   rM   rN   NETWORK_STATE_VERSIONdictintrx   propertyr   r   r   r   r   r   r   r   r   classmethodr   r@   r@   r@   rA   rp      s.    






	rp   c                   @   s  e Zd Zi g g g ddddZeddf	dHddZed	efd
dZedd Z	e	j
dd Z	dd Zdd Zdd Zdd ZdIddZdIddZdIddZedgdd  Zedgd!d" Zeg d#d$d% Zeg d&d'd( Zedd)gd*d+ Zedgd,d- Zd.d/ Zed0gd1d2 Zed0gd3d4 Zed5gd6d7 Zd8d9 Zd:d; Zd<d= Zd>d? Z d@dA Z!dJdBdCZ"dKdDdEZ#dFdG Z$dS )LNetworkStateInterpreterr"   r   FN)r   r   r|   rq   r   r#   Optional[Renderer]c                 C   s:   || _ || _t| j| _|| jd< d| _i | _|| _d S )Nr   F)	ru   _configrr   rs   initial_network_statert   _parsed_interface_dns_map	_renderer)rO   r   r   r#   r@   r@   rA   rx      s   

z NetworkStateInterpreter.__init__r;   c                 C   sH   ddl m} | jdkrt| j|rtd t| j	S t| j
| jdS )Nr   r      zPassthrough netplan v2 configr   )cloudinit.net.netplanr   ru   
isinstancer   LOGdebugrp   r   r   rt   )rO   NetplanRendererr@   r@   rA   r     s
   
z%NetworkStateInterpreter.network_statec                 C   s   | j dS Nrq   r   ry   r@   r@   rA   rq   
  s   z NetworkStateInterpreter.use_ipv6c                 C   s   | j d|i d S r   )rt   update)rO   valr@   r@   rA   rq     s   c                 C   s   | j | j| jd}t|S )Nr   )ru   r   rt   r   dumps)rO   stater@   r@   rA   dump  s
   
zNetworkStateInterpreter.dumpc                 C   sv   d|vrt d tdt|d  }t||}|r'd| }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 vr|qS )r   r@   ).0kr@   r@   rA   
<listcomp>'  s    z0NetworkStateInterpreter.load.<locals>.<listcomp>)r   error
ValueErrorNETWORK_STATE_REQUIRED_KEYSrJ   setattr)rO   r   rU   rS   msgrI   r@   r@   rA   load  s   


zNetworkStateInterpreter.loadc                 C   s   t | jS rC   )r   r   rt   ry   r@   r@   rA   dump_network_state*  s   z*NetworkStateInterpreter.dump_network_statec                 C   s   | j | jdS )N)r   r   )ru   r   ry   r@   r@   rA   as_dict-  s   zNetworkStateInterpreter.as_dictTc                 C   sD   | j dkr| j|d d| _d S | j dkr | j|d d| _d S d S )Nr   skip_brokenTr   )ru   parse_config_v1r   parse_config_v2)rO   r   r@   r@   rA   parse_config0  s   



z$NetworkStateInterpreter.parse_configc                 C   s   | j D ]C}|d }z| j| }W n ty# } ztd| |d }~ww z|| | W q tyF   |s5 tjd|dd t|   Y qw | j	
 D ]1\}}d }z	| jd | }W n tyo } ztd||d }~ww |r}|\}	}
|	|
d|d	< qLd 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   r   r|   )r   r^   r~   RuntimeErrorrK   r   warningr   r   r   r_   rt   r   format)rO   r   rP   command_typehandlere	interfacer|   r   r"   r   r@   r@   rA   r   8  sT   

z'NetworkStateInterpreter.parse_config_v1c                 C   s   ddl m} t| j|rd S | j D ]K\}}|dv rqz| j| }W n ty6 } ztd| |d }~ww z|| | | 	| W q t
y^   |sM tjd|dd t|   Y qw d S )Nr   r   )r   r#   z!No handler found for command '%s'r   Tr   )r   r   r   r   r   r_   r^   r~   r   
_v2_commonrK   r   r   r   r   )rO   r   r   r   rP   r   r   r@   r@   rA   r   [  s8   
z'NetworkStateInterpreter.parse_config_v2rf   c                 C   s
   |  |S rC   handle_physicalrO   rP   r@   r@   rA   handle_loopbackw  rz   z'NetworkStateInterpreter.handle_loopbackc           
      C   s(  | j di }||d i }|di  D ]\}}|||i qt|d}| jsF|D ]}|dds@t|drEd| _ nq/|d	d
}|d
urUt	|}|dd
}	|	d
urdt	|	}	||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   rf   paramsr   r   6addressTr&   Nr%   mac_addressinetmanualr!   )rf   r   r   r   r'   r!   r   gatewayr   r&   r%   )rt   rv   r_   r   _normalize_subnetsrq   endswithr   r   is_truer   )
rO   rP   r   r   paramr   r   r   	accept_rar%   r@   r@   rA   r   {  sF   

z'NetworkStateInterpreter.handle_physical)rf   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   rf   r   zvlan-raw-devicer   N)rt   rv   r   r   )rO   rP   r   r   r@   r@   rA   handle_vlan  s   

z#NetworkStateInterpreter.handle_vlan)rf   bond_interfacesr   c           	      C   s  |  | | jd}||di }|d D ]\}}|||i q|ddi | jd |d |i |dD ]@}||vrO|dd}|  | | jdi }||}|d|d	< |d D ]\}}|||i qi| 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   rf   r   zbond-slavesnoner   r7   )rf   r   zbond-masterN)r   rt   rv   r_   r   )	rO   rP   r   r   r   r   ifnamecmdbond_ifr@   r@   rA   handle_bond  s(   


z#NetworkStateInterpreter.handle_bondbridge_interfacesc           	      C   s  | j di }|dD ]}||v rqd|i}| | q| j di }| | ||di }|d |d< |di  D ]\}}|||i q@|d}|durwt|tsw|dv rad	}n|d
v rhd}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   rf   bridge_portsr   r5   N)on1r   T)off0r   Fz2Cannot convert bridge_stp value ({stp}) to boolean)r+   )	rt   rv   r   r_   r   r   boolr   r   )	rO   rP   r   r   r   r   r   r   r5   r@   r@   rA   handle_bridge  s4   

z%NetworkStateInterpreter.handle_bridgec                 C   s   |  | d S rC   r   r   r@   r@   rA   handle_infiniband9  s   z)NetworkStateInterpreter.handle_infinibandc                 C   sx   g }g }d|v r|d }t |ts|g}|D ]}|| qd|v r8|d }t |ts.|g}|D ]}|| q0||fS )Nr   r   )r   listappend)rO   rP   r"   r   addrsaddrpathspathr@   r@   rA   
_parse_dns=  s   

z"NetworkStateInterpreter._parse_dnsr   c                 C   sX   | j d}| |\}}d|v r||f| j|d < d S |d | |d | d S )Nr|   r   r"   r   )rt   rv   r   r   extend)rO   rP   r|   r"   r   r@   r@   rA   handle_nameserverN  s   z)NetworkStateInterpreter.handle_nameserverc                 C   s0   | j d}| |\}}||d|| d< d S )Nr   r   r|   )rt   rv   r   )rO   rP   r   _ifacer"   r   r@   r@   rA   _handle_individual_nameserver[  s   z5NetworkStateInterpreter._handle_individual_nameserverdestinationc                 C   s   | j d t| d S r   )rt   r   _normalize_router   r@   r@   rA   handle_routea  s   z$NetworkStateInterpreter.handle_routec                 C      | 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',
            }
        }

        r7   cmd_typeN_handle_bond_bridger   r@   r@   rA   handle_bondsf  s   z$NetworkStateInterpreter.handle_bondsc                 C   r   )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
            }
        }

        r8   r   Nr   r   r@   r@   rA   handle_bridges  s   z&NetworkStateInterpreter.handle_bridgesc                 C   s  t  }| D ]\}}ddi}|di }|dd}|s&td|t| ||d< |}|d}	|	r6|	}n|rF|rF| }
t|
}|rF|}||d	< |d
d}|rXd
|i|d< dD ]}||v rf|| ||< qZt| | 	|}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   physicalr    
macaddressNzHNetworkState Version2: missing "macaddress" info in config entry: %s: %sr   r$   rf   driverr   )r!   r    r%   r&   r   r   z!v2(ethernets) -> v1(physical):
%s)r	   r_   rv   r   r   strlowerr   rB   _v2_to_v1_ipcfgrb   r   r   )rO   rP   ifaces_by_macethcfgphy_cmdr    r   rf   set_namelcase_mac_addressmacr  rI   r   r@   r@   rA   handle_ethernets  sJ   -

z(NetworkStateInterpreter.handle_ethernetsc                 C   s   |  D ]<\}}d||d|dd}d|v r|d |d< t| | |}t|dkr5|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   rf   r   r   r!   r   r   zv2(vlans) -> v1(vlan):
%sN)	r_   rv   rB   r  rb   r   r   r   r   )rO   rP   r  r  vlan_cmdr   r@   r@   rA   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@   rA   handle_wifis  s   z$NetworkStateInterpreter.handle_wifisc           
      C   s   t d| | D ]c\}}d|v r|d}|r|}d|v rm|ddg }|ddg }ddi}t|dkrB|d|i t|dkrO|d	|i | | |d
i d}|rgt|}	|	rg|	}| || q
d S )Nzv2_common: handling config:
%sr$   r"   r   r   r   
nameserverr   r   r    r  )	r   r   r_   rv   rb   r   r   r   r   )
rO   r  r   dev_cfgset_name_ifacer   r|   name_cmdr   real_if_namer@   r@   rA   r   $  s2   

z"NetworkStateInterpreter._v2_commonc           
         s&  t dd t| D  | D ]~\}}t dd | D }|di }|dd}|r3||d< d|d	||d
 |ddt  fdd| D i}d|v rW|d |d< t| | |}	t|	dkrm|d|	i t	
d||| |dkr| | q|dkr| | qtdj|ddS )z(Common handler for bond and bridge typesc                 s   s    | ]	\}}||fV  qd S rC   r@   r   r   vr@   r@   rA   	<genexpr>C  s    
z>NetworkStateInterpreter._handle_bond_bridge.<locals>.<genexpr>c                 s   s$    | ]\}}|t vr||fV  qd S rC   )NETWORK_V2_KEY_FILTER)r   rI   valuer@   r@   rA   r  H  s    
parameterszgratuitious-arpNr(   r   rf   _interfacesr   r   c                 3   s     | ]\}} | |fV  qd S rC   r@   r  v2key_to_v1r@   rA   r  X      r!   r   r   zv2(%s) -> v1(%s):
%sr8   r7   z Unknown command type: {cmd_type}r   )r   r9   rv   r_   poprB   r  rb   r   r   r   r   r   r   r   )
rO   rP   r   	item_nameitem_cfgitem_paramsr   
grat_valuev1_cmdr   r@   r#  rA   r   ?  sB   
z+NetworkStateInterpreter._handle_bond_bridgec              	   C   s  dd }g }| drddi}|| di | || | dr7ddi}d| _|| di | || d	}d	}i }| d
g D ]`}d|d}d|v rcd|v rb|d	u rb| d}|d|i nd|v rw|d	u rw| d}|d|i d|v r|s| d d
}	|	r|	|d< | d d}
|
r|
|d< || || qCg }| dg D ]}|t| d| d| dd qt|rt|r||d d< |S )z7Common ipconfig extraction from v2 to v1 subnets array.c                 S   s   d| v r| d |d< d S d S )Nzroute-metricmetricr@   )	overridesr   r@   r@   rA   _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"   r   r   
dns_searchr   toviar,  )r   r   r,  r   )rv   r   rq   r   r   rb   )rO   r  r.  r   r   r   r   r"   r   r   r   r   r   r@   r@   rA   r  n  s`   






z'NetworkStateInterpreter._v2_to_v1_ipcfg)r#   r   )T)r;   NrC   )%rL   rM   rN   r   r   rx   r   rp   r   rq   setterr   r   r   r   r   r   r   r[   r   r   r   r   r   r   r   r   r   r   r  r  r  r  r  r   r   r  r@   r@   r@   rA   r      sn    





#


6



4
@



^"

/r   )	metaclassc                 C   sz   t | } tdd |  D }| ddv r |t|dd dd | d	g D |d	< d
d }dD ]}||| q3|S )Nc                 s   s     | ]\}}|r||fV  qd S rC   r@   r  r@   r@   rA   r    r%  z$_normalize_subnet.<locals>.<genexpr>r   )r/  static6)r   
ip_addressaddress_keysc                 S      g | ]}t |qS r@   )r   )r   rr@   r@   rA   r     s    z%_normalize_subnet.<locals>.<listcomp>r   c                 S   s2   || v rt | | ts| |  | |< d S d S d S rC   )r   r   split)snetrf   r@   r@   rA   listify  s   z"_normalize_subnet.<locals>.listify)r1  r   )rr   rs   r   r_   rv   r   _normalize_net_keys)r   normal_subnetr>  r   r@   r@   rA   _normalize_subnet  s    
	

rA  r@   c                 C   s  dd |   D }d}|D ]}||r|} nq|s-dd|| f }t| t|t||}t|sFtd| td| dt|}t	|}|d	}	d
|v r|
d
\}
}}|
||< |rjt|}n8|rqt|}n1td| td| dd|v rt|d }n|	r|rt|	}n|	r|rt|	}n|rdnd}d|v rt|d t|krtd|| ||d< |rd	|v r|d	= |S |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@   rA   
<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/r   @      z;Overwriting existing 'prefix' with '%s' in network info: %s)r_   rv   joinr   r   r   r  r   r   r   	partitionr   r
   r   r   r   )r   r9  netaddr_keyrI   messager   ipv6ipv4rD  	addr_part_maybe_prefixr   r@   r@   rA   r?    sf   






r?  c              
   C   s   t dd |  D }d|v r|d |d< |d= |t|dd |d}|rFz	t||d< W |S  tyE } ztd||d	}~ww |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 vr||fV  qdS )) NNr@   r  r@   r@   rA   r  $  s    z#_normalize_route.<locals>.<genexpr>r   r   )r   r   r8  r,  z(Route config metric {} is not an integerN)	r   r_   r   r?  rv   r   r   	TypeErrorr   )r   normal_router,  r   r@   r@   rA   r     s2   

r   c                 C   s   | sg } dd | D S )Nc                 S   r:  r@   )rA  )r   sr@   r@   rA   r   ?  s    z&_normalize_subnets.<locals>.<listcomp>r@   )r   r@   r@   rA   r   <  s   r   T
net_configr   c                 C   sf   d}|  d}|  d}|dkr| }|r(|dur(t|||d}|j|d |j}|s1t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:
)rv   r   r   r   r   )rV  r   r#   r   r   r   nsir@   r@   rA   parse_net_config_dataB  s$   	

rX  )r@   )TN)0rr   rW   loggingtypingr   r   r   r   	cloudinitr   r   cloudinit.netr   r	   r
   r   r   r   r   r   r   cloudinit.net.rendererr   	getLoggerrL   r   r   r   r  r9   r  __annotations__r   rB   rJ   	ExceptionrK   r[   r   r\   rp   r   rA  r?  r   r   r   rX  r@   r@   r@   rA   <module>   s   
,
"V     O
L"