dx:ddlZddlZddlZddlZddlmcmZddlmZddl m Z ddl m Z ddl mZmZmZmZmZmZmZdZdZGdd eZd Zd Zd Zd ZdZdZdZy)N) DhcpError) get_osutil)KNOWN_WIRESERVER_IP)hex_dump hex_dump2 hex_dump3 compare_bytes str_to_ordunpack_big_endianint_to_ip4_addr10813FA8ctSN) DhcpHandler=/usr/lib/python3/dist-packages/azurelinuxagent/common/dhcp.pyget_dhcp_handlerr$s =rcTeZdZdZdZdZdZedZedZ dZ dZ d Z y ) rz? Azure use DHCP option 245 to pass endpoint ip to VMs. cht|_d|_d|_d|_d|_d|_y)NF)rosutilendpointgatewayroutes_request_broadcast skip_cacheselfs r__init__zDhcpHandler.__init__-s/ l    "'rcv|js |jry|j|jy)z{ Send dhcp request Configure default gateway and routes Save wire server endpoint if found N)wireserver_route_existsdhcp_cache_exists send_dhcp_req conf_routesrs rrunzDhcpHandler.run5s0  ' '4+A+A   rcL|jj}|dk(s|dk(rtjdt j dtjd|jj |jj}|dk(ry|dk(ryy)z; Wait for network stack to be initialized. 0.0.0.0zWaiting for network. zTry to start network interface.N)r get_ip4_addrloggerinfotimesleep start_network)ripv4s rwait_for_networkzDhcpHandler.wait_for_networkAs{{{'')bjDI- KK. / JJrN KK9 : KK % % ';;++-D bjDI-rcd}tjdjt |jj }t d|DrEt|_d|_d|_ d}tjdjt|Stjdjt |S#t$r4}tjdjt|Yd}~|Sd}~wwxYw) z 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. FzTest for route to {0}c3,K|] }t|vywr)KNOWN_WIRESERVER_IP_ENTRY).0routes r z6DhcpHandler.wireserver_route_exists..YsQE-6QsNTzRoute to {0} existszNo route exists to {0}z4Could not determine whether route exists to {0}: {1}) r+r,formatrrread_route_tableanyrrrwarn Exceptionerror)r route_exists route_tablees rr!z#DhcpHandler.wireserver_route_existsMs  +223FGH -++668KQ[QQ!4 # " #  1889LMN  4;;BB C JJx s2A A+ A&&A+cT|jj}|s!tjdt|_y tjd|jj }t||j}|jsd|_|jj}|jj}|r|jj||jjr|jj|j|}|jjr|jj|r|jj!|| t#dt%|\|_|_|_y)z, Check if DHCP is available z!send_dhcp_req: DHCP not availableNzSend dhcp requestTz Failed to receive dhcp response.)ris_dhcp_availabler+r,rr get_mac_addrbuild_dhcp_requestrrH get_if_nameset_route_for_dhcp_broadcastis_dhcp_enabledstop_dhcp_servicerTstart_dhcp_serviceremove_route_for_dhcp_broadcastrparse_dhcp_resprr)rdhcp_availablemac_addrreqmissing_default_routeifnameresps rr#zDhcpHandler.send_dhcp_reqsF++779 KK; </DM    '(;;++-!4+B+BC&&&*D #!% D D F((* KK 4 4V < ;; & & ( KK ) ) +""3' ;; & & ( KK * * , KK 7 7 ? <>? ?3B43H0 t|T[rN) __name__ __module__ __qualname____doc__rr%r1propertyr!r"r$rTr#rrrrr(sO   .8& D /Irrc tt|}|dkrtjd|ytjdt |tjdt ||t ||dds8tjdt|ddt|ddtd t ||dds8tjd t|ddt|ddtd t ||d d s8tjdt|d d t|d d tdy)Nz.HandleDhcpResponse: Too few bytes received:{0}FzBytesReceived:{0}zDHCP response:{0}z'Cookie not match: send={0}, receive={1}z1Cookie in dhcp respones doesn't match the requestz.TransactionID not match: send={0}, receive={1}z8TransactionID in dhcp respones doesn't match the requestz,Mac Address not match: send={0}, receive={1}z3Mac Addr in dhcp respones doesn't match the request) lenr+r=verbosehexrr rr)rPrS bytes_recvs rrOrOs/XJD E ! NN&J8 NN&:(FG (D! 4B $2 43 5KLL (Aq 1I !Q/ 1a0 245 5 (D! 4G $2 43 545 5 5rctjdt|t|g}|dkrtjd||dz}|||zdzkr}t ||}|dzdzdz }ddd |z zz} |d z }t |||} | d |d zz z} | | z} ||z }t ||d } |d z }|j | | | f|||zdzkr}|||zdzk7rtjd |S)Nz%Routes at offset: {0} with length:{1}Data too small for option:{0}rGil rFrnzUnable to parse routes)r+rrrsr=r r append) rSoptionilengthrtrj mask_len_bitsmask_len_bytesmasknetrs r parse_routers& NN:CFv; F z 4f= AA q6zA~ "8A;/ )A-39ZB,>?@ Q!^< nq(() t  ^#Ha3 Q sD'*+ q6zA~  QZ!^ -. Mrc|dz|kr8|dk7rtjdyt||dzd}t|}|Stjd|y)Nrvrnz'Endpoint or Default Gateway not 4 bytesrGrw)r+r=r r )rSr}r~rrtaddrip_addrs r parse_ip_addrrsX1uz Q; LLB C 1q5!4!$' 4f= rctjdt|}d}d}d}d}||krBt||}d}|dz|krt||dz}tjdt |t |t ||dk(r tjdt |n|d k(rt |||||}n|d k(r0t |||||}tjd |t |nh|d k(r0t |||||}tjd |t |n3tjdt |t |t |||dzz }||krB|||fS)zL Parse DHCP response: Returns endpoint server or None on error. zparse Dhcp ResponseNrrFz-DHCP option {0} at offset:{1} with length:{2}zDHCP packet ended at offset:{0}ryzDefault gateway:{0}, at {1}z(Azure wire protocol endpoint:{0}, at {1}z/Skipping DHCP option:{0} at {1} with length {2}rG)r+rrrqr rsrr)rSrtrrrr~r}rs rr_r_si  NN()XJHG F A j.HQK( EZ Q0FF6{CFCK 9 S= NNCr)z;Send DHCP request: Setting socket.timeout=10, entering recviz{0})socketAF_INET SOCK_DGRAM IPPROTO_UDP setsockopt SOL_SOCKET SO_BROADCAST SO_REUSEADDRbindsendto settimeoutr+rrrecvcloseIOErrorrr8)rPsockrSr@s rrNrN<s D}}V^^V->->#//1 ))6+>+>B ))6+>+>B /" G01 ' (99T?   JJL  ) Q(()   JJL s$C:D D:D55D::D==Ec dgdz}t}tddD] }gd|||<tddD]}t|||d|z<tjdt |t |ddfz|rd|d<tdd D]}t|||d |z<tdd D]}gd ||d |z<tjd|S)z$ Build DHCP request string. rry)rFrFrprnz'BuildDhcpRequest: transactionId:%s,%04Xr)rpror{)cSr5rFrFrrmB) gen_trans_idranger r+rrrr array)rarequest_broadcastrPtrans_idas rrXrXQsBcCiG~H 1a["q\ "1a[1#HQK0A1 NN<('1a(@**+ 1a[4&x{3q41a[@rsv" ..64E']I&]I@5D2 )%X*D%Nr