ó ½-'Nc@s9ddlZddlZddlZddlmZddlmZddlmZddl m Z m Z ddl m Z ddlmZddlmZd Zd efZd efd „ƒYZd „Zd„Zdefd„ƒYZdefd„ƒYZdefd„ƒYZd„Zd„Ze d„ZdS(iÿÿÿÿN(tDeferred(tloads(t log_failure(tfetcht FetchError(tis_valid_tag_list(tget_fqdn(t get_vm_infos169.254.169.254shttp://%s/latesttInvalidCredentialsErrorcBseZdZRS(s} Raised when an invalid account title and/or registration password is used with L{RegistrationManager.register}. (t__name__t __module__t__doc__(((sA/usr/lib/python2.7/dist-packages/landscape/broker/registration.pyRscs+‡fd†}‡fd†}t||ƒS(Ncs|jjˆƒS(N(t_persisttget(tself(tname(sA/usr/lib/python2.7/dist-packages/landscape/broker/registration.pyR scs|jjˆ|ƒdS(N(R tset(Rtvalue(R(sA/usr/lib/python2.7/dist-packages/landscape/broker/registration.pyRs(tproperty(RR R((RsA/usr/lib/python2.7/dist-packages/landscape/broker/registration.pytpersist_propertyscs‡fd†}t|ƒS(Ncst|jˆƒS(N(tgetattrt_config(R(R(sA/usr/lib/python2.7/dist-packages/landscape/broker/registration.pyR 's(R(RR ((RsA/usr/lib/python2.7/dist-packages/landscape/broker/registration.pytconfig_property%stIdentitycBs_eZdZedƒZedƒZedƒZedƒZedƒZ edƒZ d„Z RS(s½Maintains details about the identity of this Landscape client. @ivar secure_id: A server-provided ID for secure message exchange. @ivar insecure_id: Non-secure server-provided ID, mainly used with the ping server. @ivar computer_title: See L{BrokerConfiguration}. @ivar account_name: See L{BrokerConfiguration}. @ivar registration_password: See L{BrokerConfiguration}. @ivar tags: See L{BrokerConfiguration} s secure-ids insecure-idtcomputer_titlet account_nametregistration_passwordttagscCs||_|jdƒ|_dS(s½ @param config: A L{BrokerConfiguration} object, used to set the C{computer_title}, C{account_name} and C{registration_password} instance variables. t registrationN(Rtroot_atR (Rtconfigtpersist((sA/usr/lib/python2.7/dist-packages/landscape/broker/registration.pyt__init__@s ( R R R Rt secure_idt insecure_idRRRRRR (((sA/usr/lib/python2.7/dist-packages/landscape/broker/registration.pyR-s       tRegistrationHandlercBskeZdZd d„Zd„Zd„Zd„Zd„Zd„Z d„Z d„Z d „Z d „Z RS( sÉ An object from which registration can be requested of the server, and which will handle forced ID changes from the server. L{register} should be used to perform initial registration. cCsâ||_||_||_||_||_||_|jjd|jƒ|jjd|jƒ|jjd|j ƒ|jj d|j ƒ|jj d|j ƒ|jj d|j ƒd|_||_d|_d|_dS(Ntruns pre-exchanges exchange-donesset-ids unknown-idR(Rt _identityt_reactort _exchanget_pingert_message_storetcall_ont_fetch_ec2_datat_handle_pre_exchanget_handle_exchange_donetregister_messaget_handle_set_idt_handle_unknown_idt_handle_registrationtNonet_should_registert _fetch_asynct_otpt _ec2_data(RRtidentitytreactortexchangetpingert message_storet fetch_async((sA/usr/lib/python2.7/dist-packages/landscape/broker/registration.pyR Rs"           cCsg|j}|jjr5t|j o1|jjdƒƒSt|j oc|joc|joc|jjdƒƒS(Nsregister-cloud-vmtregister( R%RtcloudtboolR!R)tacceptsRR(Rtid((sA/usr/lib/python2.7/dist-packages/landscape/broker/registration.pytshould_registerfs    cCs;d|j_d|j_t|jƒj}|jjƒ|S(sô Attempt to register with the Landscape server. @return: A L{Deferred} which will either be fired with None if registration was successful or will fail with an L{InvalidCredentialsError} if not. N( R2R%R!R"tRegistrationResponseR&tdeferredR'R9(Rtresult((sA/usr/lib/python2.7/dist-packages/landscape/broker/registration.pyR=os    cCs|jt|ƒj|jƒS(st Get data at C{path} on the EC2 API endpoint, and add the result to the C{accumulate} list. (R4tEC2_APIt addCallbacktappend(Rtpatht accumulate((sA/usr/lib/python2.7/dist-packages/landscape/broker/registration.pyt _get_data}scs÷ˆj}ˆjjró|j róg‰ˆjtdƒjtƒjˆj ƒ}dddddddg}x*|D]"}|j|‡‡fd †ƒqoW|j‡fd †ƒ|jˆj ƒ‡‡fd †}d „}|j|ƒ|j|ƒnd S(sARetrieve available EC2 information, if in a EC2 compatible cloud.s /user-datas/meta-data/instance-ids/meta-data/reservation-ids/meta-data/local-hostnames/meta-data/public-hostnames/meta-data/ami-launch-indexs/meta-data/kernel-ids/meta-data/ami-idcsˆj|ˆƒS(N(RK(tignoreRI(Rtec2_data(sA/usr/lib/python2.7/dist-packages/landscape/broker/registration.pyt›scsˆjtdƒjtƒS(Ns/meta-data/ramdisk-id(R4RFt addErrbackR(RL(R(sA/usr/lib/python2.7/dist-packages/landscape/broker/registration.pyRNžsc sЈ\ }}}}}}}}} i|d6|d6|d6|d6|d6|d6| d6|d6ˆ_xQˆjjƒD]@\} } | d krœ| dkrœqrn| jd ƒˆj| R!R4RFRORRGRH(RRARDtpathsRIRmRp((RRMsA/usr/lib/python2.7/dist-packages/landscape/broker/registration.pyR+„s,   +  cCs*|jƒr&|j r&|jjƒndS(sMRegistered handler for the C{"exchange-done"} event. If we are not registered yet, schedule another message exchange. The first exchange made us accept the message type "register", so the next "pre-exchange" event will make L{_handle_pre_exchange} queue a registration message for delivery. N(RBR3R'R9(R((sA/usr/lib/python2.7/dist-packages/landscape/broker/registration.pyR-Ôs cCsl|jƒ|_|jrh|j}|jjƒ|j}t|ƒsYd}tj dƒn|j j r£|j dk r£|j rítjdƒidd6|j d6tƒd6dd6dd6|d 6tƒd 6}|j|j ƒ|jj|ƒqe|jrd d |gt|ƒ}tjd |j|fƒidd6dd6tƒd6|jd6|jd6|d 6tƒd 6}|j|j ƒ|jj|ƒqe|jjdƒqh|jrUddgt|jƒ}d d |gt|ƒ}tjd|j||fƒidd6|jd6|jd6|jd6tƒd6|d 6tƒd 6}|jj|ƒqh|jjdƒndS(sÈ An exchange is about to happen. If we don't have a secure id already set, and we have the needed information available, queue a registration message with the server. s-Invalid tags provided for cloud registration.s%Queueing message to register with OTPsregister-cloud-vmttypeRXthostnameRRRsvm-infotu and tags %s uBQueueing message to register with account %r %sas an EC2 instance.sregistration-failedtwithouttwithu=Queueing message to register with account %r %s%s a password.R=RN(RBR3R%R)tdelete_all_messagesRRR2tloggingRoRR>R6R5tinfoRRtupdateR'tsendRR?RR&tfireR(RRARtmessaget with_tagst with_word((sA/usr/lib/python2.7/dist-packages/landscape/broker/registration.pyR,àsd                            cCs|j}|jdƒ|_|jdƒ|_tjd|jd|jƒtjd|jƒ|jj dƒ|jj dƒdS( s×Registered handler for the C{"set-id"} event. Record and start using the secure and insecure IDs from the given message. Fire C{"registration-done"} and C{"resynchronize-clients"}. RAs insecure-ids2Using new secure-id ending with %s for account %s.iöÿÿÿsUsing new secure-id: %ssregistration-donesresynchronize-clientsN( R%R R!R"RxRyRtdebugR&R|(RR}RA((sA/usr/lib/python2.7/dist-packages/landscape/broker/registration.pyR/#s  cCs'|ddkr#|jjdƒndS(NRysunknown-accountsregistration-failed(R&R|(RR}((sA/usr/lib/python2.7/dist-packages/landscape/broker/registration.pyR14scCs3|j}tjd|jƒd|_d|_dS(Ns,Client has unknown secure-id for account %s.(R%RxRyRR2R!R"(RR}RA((sA/usr/lib/python2.7/dist-packages/landscape/broker/registration.pyR08s     N(R R R R2R RBR=RKR+R-R,R/R1R0(((sA/usr/lib/python2.7/dist-packages/landscape/broker/registration.pyR#Js    P C  RCcBs2eZdZd„Zd„Zd„Zd„ZRS(sµA helper for dealing with the response of a single registration request. @ivar deferred: The L{Deferred} that will be fired as per L{RegistrationHandler.register}. cCsI||_|jd|jƒ|_|jd|jƒ|_tƒ|_dS(Nsregistration-donesregistration-failed(R&R*t_donet_done_idt_failedt _failed_idRRD(RR8((sA/usr/lib/python2.7/dist-packages/landscape/broker/registration.pyR Gs cCs*|jj|jƒ|jj|jƒdS(N(R&t cancel_callR‚R„(R((sA/usr/lib/python2.7/dist-packages/landscape/broker/registration.pyt _cancel_callsMscCs|jjdƒ|jƒdS(N(RDtcallbackR2R†(R((sA/usr/lib/python2.7/dist-packages/landscape/broker/registration.pyRQscCs!|jjtƒƒ|jƒdS(N(RDterrbackRR†(R((sA/usr/lib/python2.7/dist-packages/landscape/broker/registration.pyRƒUs(R R R R R†RRƒ(((sA/usr/lib/python2.7/dist-packages/landscape/broker/registration.pyRC@s    cCsyt|ƒ}Wn&tk r8tjd|fƒdSXt|tƒs`tjd|fƒdSx5d D]-}||krgtjd||fƒdSqgWt|dƒ|krÉtjd||fƒdSi|d|d 6|dd6|dd6}d |kr|d |d s(     ö "