�
d�bc@s
dd
lZdd
l
Z
dd
lZdd
lZdd
ljjZddlmZ
ddl	m
Z

ddlmZ
ddl
mZmZmZmZmZmZmZ
dZd�Zdef
d	��YZd
�Zd�Zd�Zd
�Zd�Zd�Zd�Zd
S(i����N(
t	DhcpError(
t
get_osutil(
tKNOWN_WIRESERVER_IP(thex_dumpt	hex_dump2t	hex_dump3t
compare_bytest
str_to_ordtunpack_big_endiantint_to_ip4_addrt10813FA8c
Cst�S(
N(
tDhcpHandler(((s?/usr/lib/python2.7/site-packages/azurelinuxagent/common/dhcp.pytget_dhcp_handler$s
RcBsbeZ
dZd
�Zd�Zd�Zed��
Zed��
Zd�Z	d�Z
d�ZRS(	s?
    Azure use DHCP option 245 to pass endpoint ip to VMs.
    c

Cs=t�|_
d|_d|_d|_t|_t|_dS(
N(	R
tosutiltNonetendpointtgatewaytroutestFalset_request_broadcastt
skip_cache(
tself((s?/usr/lib/python2.7/site-packages/azurelinuxagent/common/dhcp.pyt__init__-s

	
	
	
	
c

Cs.|js|j
rd
S|j�
|j�
d
S(s{
        Send dhcp request
        Configure default gateway and routes
        Save wire server endpoint if found
        N(twireserver_route_existstdhcp_cache_existst
send_dhcp_reqtconf_routes(
R((s?/usr/lib/python2.7/site-packages/azurelinuxagent/common/dhcp.pytrun5s


c
Csu|jj
�}
x_|
d
ks*|
dkrptjd�

tjd�

tjd�

|jj�
|jj
�}
qWdS(s;
        Wait for network stack to be initialized.
        ts0.0.0.0sWaiting for network.i
sTry to start network interface.N(R
tget_ip4_addrtloggertinfottimetsleept
start_network(Rtipv4((s?/usr/lib/python2.7/site-packages/azurelinuxagent/common/dhcp.pytwait_for_networkAs









c
Cs�t}
t
jd
jt�
�

yy|jj�}td�|D�
�
r~t|_d|_
d|_t}
t
jdjt�
�

nt
j
djt�
�

Wn,tk
r�
}
t
jdjt|��

n
X|
S(s�
        Determine whether a route to the known wireserver
        ip already exists, and if so use that as the endpoint.
        This is true when running in a virtual network.
        :return: True if a route to KNOWN_WIRESERVER_IP exists.
        sTest for route to {0}c
ss|]}
t|
kV
qdS(
N(
tKNOWN_WIRESERVER_IP_ENTRY(t.0troute((s?/usr/lib/python2.7/site-packages/azurelinuxagent/common/dhcp.pys	<genexpr>YssRoute to {0} existssNo route exists to {0}s4Could not determine whether route exists to {0}: {1}N(RRRtformatRR
tread_route_tabletanyRRRRtTruetwarnt	Exceptionterror(Rtroute_existstroute_tablet
e((s?/usr/lib/python2.7/site-packages/azurelinuxagent/common/dhcp.pyRMs 



	
	
	





c
Csg|jr
t
St
}
tjd
�

|jj�}|dk	rM||_t}
ntjdj	|
�
�

|
S(s�
        Check whether the dhcp options cache exists and contains the
        wireserver endpoint, unless skip_cache is True.
        :return: True if the cached endpoint was found in the dhcp lease
        sChecking for dhcp lease cachesCache exists [{0}]N(
RRRRR
tget_dhcp_lease_endpointRRR+R((Rtexiststcached_endpoint((s?/usr/lib/python2.7/site-packages/azurelinuxagent/common/dhcp.pyRjs	




	
	

c
Cs�tj
d
�

tj
d|j�
tj
d|j�
|jdk	rm|jj�rm|jjdd|j�
n|jdk	r�x6|jD](}
|jj|
d|
d|
d�
q�WndS(NsConfigure routessGateway:{0}s
Routes:{0}ii
i(RRRRRR
tis_missing_default_routet	route_add(RR'((s?/usr/lib/python2.7/site-packages/azurelinuxagent/common/dhcp.pyR~s







cCs�d
ddddg}xo|D]g}y.|jj
�
t|
�
}t|
|�
|SWn#tk
ru
}
tjd|�
n
Xtj|�

qWdS(Nii
ii<s Failed to send DHCP request: {0}(
R
tallow_dhcp_broadcasttsocket_sendtvalidate_dhcp_respRRR,R R!R(Rtrequestt__waiting_duration__tdurationtresponseR1((s?/usr/lib/python2.7/site-packages/azurelinuxagent/common/dhcp.pyt_send_dhcp_req�s













c
CsL
|jj
�}
|
s/tjd
�

t|_dStjd�

|jj�}t||j�}|jsrt	|_n|jj
�}|jj�}|r�|jj|�

n|jj
�r�|jj�
n|j|�
}|jj
�r�|jj�
n|r
|jj|�

n|dkr*
td�
�
nt|�
\|_|_|_dS(s,
        Check if DHCP is available
        s!send_dhcp_req: DHCP not availableNsSend dhcp requests Failed to receive dhcp response.(R
tis_dhcp_availableRRRRtget_mac_addrtbuild_dhcp_requestRR+R5tget_if_nametset_route_for_dhcp_broadcasttis_dhcp_enabledtstop_dhcp_serviceR>tstart_dhcp_servicetremove_route_for_dhcp_broadcastRRtparse_dhcp_respRR(Rtdhcp_availabletmac_addrtreqtmissing_default_routetifnametresp((s?/usr/lib/python2.7/site-packages/azurelinuxagent/common/dhcp.pyR�s0



	




	








(t__name__t
__module__t__doc__RRR$tpropertyRRRR>R(((s?/usr/lib/python2.7/site-packages/azurelinuxagent/common/dhcp.pyR(s					
cCsL
t|
�
}|d
kr,t
jd|�
tSt
jdt|�
�
t
jdt|
|��
t||
dd�s�t
jdt|dd�t|
dd��
t	d�
�
nt||
dd�s�t
jd	t|dd�t|
dd��
t	d
�
�
nt||
dd�sH
t
jd
t|dd�t|
dd��
t	d�
�
ndS(Ni�s.HandleDhcpResponse: Too few bytes received:{0}sBytesReceived:{0}sDHCP response:{0}i�is'Cookie not match:
send={0},
receive={1}s1Cookie in dhcp respones doesn't match the requests.TransactionID not match:
send={0},
receive={1}s8TransactionID in dhcp respones doesn't match the requestiis,Mac Address not match:
send={0},
receive={1}s3Mac Addr in dhcp respones doesn't match the request(
tlenRR.RtverbosethexRRRR(R:R=t
bytes_recv((s?/usr/lib/python2.7/site-packages/azurelinuxagent/common/dhcp.pyR9�s,


	



	



	



	


cCs9
tj
d
t|�
t|�
�
g}|dkrDtjd|
�
n|d}x�|||dkr
t||�
}|dd
@d?}ddd|>@}	|d	7}t|||�}
|
d|d
K}
|
|	M}
||7}t||d�}|d7}|j|
|	|f�

qQW|||dkr5
tjd�

n|S(Ns%Routes at offset: {0} with length:{1}isData too small for option:{0}iiiI����i i
iisUnable to parse routesi����(RRTRUR.RRtappend(R=toptiont
itlengthRVRt
jt
mask_len_bitstmask_len_bytestmasktnetR((s?/usr/lib/python2.7/site-packages/azurelinuxagent/common/dhcp.pytparse_route�s*

























cCsg|d
|krS|dkr-tj
d�

dSt||dd�}t|�
}|Stj
d|
�
dS(Niis'Endpoint or Default Gateway not 4 bytesisData too small for option:{0}(RR.RRR	(R=RXRYRZRVtaddrtip_addr((s?/usr/lib/python2.7/site-packages/azurelinuxagent/common/dhcp.pyt
parse_ip_addr
s








c
Cs�
tj
d
�

t|�
}
d}d}d}d}xx
||
kr�
t||�
}d}|d|
kr}t||d�
}ntj
dt|�
t|�
t|�
�
|dkr�tj
dt|�
�
Pn�|dkr�t|||||
�}n�|d	kr2
t|||||
�}tj
d
|t|�
�
nh|dkrr
t|||||
�}tj
d|t|�
�
n(tj
d
t|�
t|�
t|�
�
||d7}q4W|||fS(sL
    Parse DHCP response:
    Returns endpoint server or None on error.
    sparse Dhcp Responsei�ii
s-DHCP option {0} at offset:{1} with length:{2}i�sDHCP packet ended at offset:{0}i�isDefault gateway:{0}, at {1}i�s(Azure wire protocol endpoint:{0}, at {1}s/Skipping DHCP option:{0} at {1} with length {2}iN(RRTRSRRRUR`Rc(R=RVRRRRYRXRZ((s?/usr/lib/python2.7/site-packages/azurelinuxagent/common/dhcp.pyRH
s<










	











	

	


c
Cs�d}
z�y�t
j
t
jt
jt
j�}
|
jt
jt
jd
�
|
jt
jt
jd
�
|
j	d
�

|
j
|d�
|
jd�

tj
d�

|
jd�
}|SWn(tk
r�
}
td	j|�
�
�
n
XWd|
dk	r�|
j�
nXdS(Ni
s0.0.0.0iDs<broadcast>iCi
s;Send DHCP request: Setting socket.timeout=10, entering recvis{0}(s0.0.0.0iD(s<broadcast>iC(RtsockettAF_INETt
SOCK_DGRAMtIPPROTO_UDPt
setsockoptt
SOL_SOCKETtSO_BROADCASTtSO_REUSEADDRtbindtsendtot
settimeoutRRTtrecvtIOErrorRR(tclose(R:tsockR=R1((s?/usr/lib/python2.7/site-packages/azurelinuxagent/common/dhcp.pyR8<
s"
















c	Cs;
d
g
d}t�}x.t
d
d�D]}dddg|||<q&Wx/t
d
d�D]}t||�
|d|<qWWtjdt|�
t|dd�f�

|
r�d|d	<nx/t
d
d�D]}t||�
|d
|<q�WxAt
d
d�D]0}dd
ddddddg||d|<q�Wtjd|�S(s$
    Build DHCP request string.
    ii�ii
iis'BuildDhcpRequest: transactionId:%s,%04Xi�i
iiici�iSi5i�i�t
B(tgen_trans_idtrangeRRRTRRtarray(RJtrequest_broadcastR:ttrans_idt
a((s?/usr/lib/python2.7/site-packages/azurelinuxagent/common/dhcp.pyRAQ
s !
	

	
	



.
cCs
tj
d
�
S(Ni(tosturandom(((s?/usr/lib/python2.7/site-packages/azurelinuxagent/common/dhcp.pyRt�
s
( RvRzRdR tazurelinuxagent.common.loggertcommonRt azurelinuxagent.common.exceptionRtazurelinuxagent.common.osutilR
t%azurelinuxagent.common.utils.restutilRt%azurelinuxagent.common.utils.textutilRRRRRRR	R%RtobjectRR9R`RcRHR8RARt(((s?/usr/lib/python2.7/site-packages/azurelinuxagent/common/dhcp.pyt<module>s$






4	�	"		
	,		G