U Ӈgi;@sddlZddlZddlZddlZddlmZmZmZddlm Z ddl m Z e e ZdZdddd d gZGd d d ejZGd ddeZddZddZddZddeddfddZddZGdddeZGdddeZeejffeejejffgZddZe dkrd d!Z e dS)"N)sources url_helperutil)KlibcNetworkConfigSource)NetworkConfigSourcez 2012-03-01)meta-data/instance-idrFF)meta-data/local-hostnamerFF)meta-data/public-keysr FT)zmeta-data/vendor-data vendor-dataTT) user-datar TTc@sheZdZdZdZdZdZdZdZddZ e dd Z d d Z d d Z ddZddZddZddZdS)DataSourceMAASz DataSourceMAAS reads instance information from MAAS. Given a config metadata_url, and oauth tokens, it expects to find files under the root named: instance-id user-data hostname vendor-data MAASNx2cCs:tj||||d|_tj|jd|_t|j |_ dS)NZmaas) r DataSource__init__base_urlospathjoinseed_dirget_id_from_ds_cfgds_cfgid_hash)selfsys_cfgZdistropathsrB/usr/lib/python3/dist-packages/cloudinit/sources/DataSourceMAAS.pyr2szDataSourceMAAS.__init__cCs|jst|j|_|jSN) _oauth_helperget_oauth_helperrrrrr oauth_helper8s zDataSourceMAAS.oauth_helpercCstj|}d||jfS)Nz%s [%s])rr__str__r)rrootrrrr$>s zDataSourceMAAS.__str__c Cs|j}z||jt|jWdStk r4Yn4tk rf}ztd|j|W5d}~XYnX|dd}|s|dSz4| |sWdS||t ||j j |j ddWdStk rttd|YdSXdS)NTz%s was malformed: %s metadata_urlF)read_file_or_urlrretriesz$Failed fetching metadata from url %s)r _set_datarread_maas_seed_dirMAASSeedDirNoneMAASSeedDirMalformedLOGwarninggetwait_for_metadata_serviceread_maas_seed_urlr#readurlr ExceptionrZlogexc)rZmcfgexcurlrrr _get_dataBs8   zDataSourceMAAS._get_datac Csj||_|\}}}||_||_|rfzt||_Wn4tk rd}ztd|d|_W5d}~XYnXdS)Nz"Invalid content in vendor-data: %s) rZ userdata_rawmetadatarZconvert_vendordataZvendordata_raw ValueErrorr.r/)rr6dataZudmdvderrrr*is  zDataSourceMAAS._set_datacCs d|jS)z/Return the subplatform metadata source details.z seed-dir (%s))rr"rrr_get_subplatformvszDataSourceMAAS._get_subplatformcCs|}|jdkrdSt}|d}d|tf}|g}|jj||j|jd\}}|rft d|nt d|t t|t |S)NrF/z%s/%s/meta-data/instance-id)urlsZmax_waittimeoutzUsing metadata source: '%s'z(Giving up on md from %s after %i seconds)Zget_url_paramsZmax_wait_secondstime monotonicrstrip MD_VERSIONr#Z wait_for_urlZtimeout_secondsr.debugZcriticalintbool)rr6Z url_paramsZ starttimeZ check_urlr@Z _responserrrr1zs(    z(DataSourceMAAS.wait_for_metadata_servicecCs0|jdkrdSt|d|jfi}|jt|kS)alocally check if the current system is the same instance. MAAS doesn't provide a real instance-id, and if it did, it is still only available over the network. We need to check based only on local resources. So compute a hash based on Oauth tokens.NF datasource)rrZget_cfg_by_pathdsnamer)rrZncfgrrrcheck_instance_ids z DataSourceMAAS.check_instance_id)__name__ __module__ __qualname____doc__rJrr Z url_max_waitZ url_timeoutrpropertyr#r$r7r*r>r1rKrrrrr s  ' r cs0eZdZejejejejfZfddZ Z S)DataSourceMAASLocalcs,tstddStdtS)NzNo initramfs applicable configFz!Found initramfs applicable config)rZ is_applicabler.rFsuperr7r" __class__rrr7s    zDataSourceMAASLocal._get_data) rLrMrNrZCMD_LINEZ SYSTEM_CFGZDSZ INITRAMFSZnetwork_config_sourcesr7 __classcell__rrrSrrQs rQcs&d}tfdd|D}tjf|S)zReturn an oauth helper instance for values in cfg. @raises ValueError from OauthUrlHelper if some required fields have true-ish values but others do not.) consumer_keyconsumer_secret token_key token_secretcsg|]}||fqSrr0).0rcfgrr sz$get_oauth_helper..)dictrZOauthUrlHelper)r^keyskwargsrr]rr!sr!cs4d}dfdd|D}dt|dS)z;Given a config, generate a unique identifier for this node.)rVrXrYcsg|]}|dqS)rZ)r[frrrr_sz&get_id_from_ds_cfg..zv1:zutf-8)rhashlibZsha256encodeZ hexdigest)rZfieldsZidstrrrfrrsrcCsL|dr|dd}tj|r4tt|dkr//meta-data/instance-id * //meta-data/local-hostname * //user-data If version is None, then / will not be used. Nr?z%s/%sz%s/%s/%s)r)rA ssl_detailsz4Fetching from %s resulted in an invalid http code %sizMissing required %s: %s)rr(endswith DS_FIELDSrZfetch_ssl_detailsokcontentsZ decode_binaryr.r/codeZUrlErrorr-check_seed_contents)Zseed_urlr(rArkrr)r;rZ _dictnamebinaryoptionalr6rqZrespr=rrrr2sH      r2c Csi}g}tD]j\}}}}||kr0|s ||q d|krj|d\}} } ||krXi||<||||| <q ||||<q t|dkrtd||rtd||fd} |drt} tj |d| td} | | krtd|d |d | fS) zValidate if dictionary content valid as a return for a datasource. Either return a (userdata, metadata, vendordata) tuple or Raise MAASSeedDirMalformed or MAASSeedDirNone r?rz%s: no data files foundz%s: missing files %sNr )defaultZallowedz%vendor-data was not loadable as yaml.r meta-data) rsappend partitionrnr,r-r0objectrZ load_yaml) ZcontentZseedZretZmissingZspathZdpathZ_binaryrytop_pZvd_dataerrrrrrw s8    rwc@s eZdZdS)r,NrLrMrNrrrrr,0sr,c@s eZdZdS)r-Nrrrrrr-4sr-cCs t|tSr)rZlist_from_depends datasources)Zdependsrrrget_datasource_list@sr__main__csddl}ddl}ddl}|jdd}|jddddd|jd d d dd|jd d d dd|jddddd|jddddd|jdddtd|jddd}dD](\}}|j||d}|jdddddq|}|j |j |j |j d } |j dkr.geturlcstd||fdS)Nz == %s == %s )printdecoder)rrrprinturlszmain..printurlcst|drh|D]J}|dr:d||fq|dkrTd|dfqd||fqn|dS)Nr?z%s%sr{z meta-data/)rrr splitlines)r6line)rrrrrrs  zmain..crawlrzChecking seed at %s r?ri)rkr(r)z=== user-data ===zN/Az=== meta-data ===z=== vendor-data ===r0rz%s/)$argparsepprintsysArgumentParser add_argumentrEZadd_subparsersZ add_parser parse_argsZckeyZtkeyZtsecZcsecZconfigrrexistsaccessR_OKstderrwriterZ read_confrar6exitr!rr3rlr+r2Zapiverrrrr)rrrparserZsubcmdsnamerrargsZcredsfnameZfpathr^rr3Zuserdatar8r<r)rrr#rrmainFs                     r)!rgZloggingrrBZ cloudinitrrrZcloudinit.net.cmdlinerZcloudinit.sourcesrZ getLoggerrLr.rErsrr rQr!rr+r2rwr4r,r-ZDEP_FILESYSTEMZ DEP_NETWORKrrrrrrr sJ        :'