"T`'8@shddlZddlZddlZddlmZddlmZddlmZddlmZej e Z dZ d$d%d&d'd(gZ GdddejZddZddZddZdde ddddZddZGdddeZGdddeZeejejffgZdd Ze d!krdd"d#ZedS))N)log)sources) url_helper)utilz 2012-03-01meta-data/instance-idFmeta-data/local-hostnamemeta-data/public-keysTmeta-data/vendor-data vendor-data user-datac@seZdZdZdZdZdZddZeddZ dd Z d d Z d d Z ddZ ddZddZdS)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 MAASNcCsStjj||||d|_tjj|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.pyr.s zDataSourceMAAS.__init__cCs"|jst|j|_|jS)N) _oauth_helperget_oauth_helperr)rrrr oauth_helper4s zDataSourceMAAS.oauth_helpercCs#tjj|}d||jfS)Nz%s [%s])rr__str__r)rrootrrrr :szDataSourceMAAS.__str__cCs|j}y$|j|jt|jdSWnOtk rAYn>tk r~}ztjd|j|WYdd}~XnX|jdd}|sdSyL|j |sdS|j|t |d|j j d|j dddSWn)tk rtjtd |dSYnXdS) NTz%s was malformed: %s metadata_urlFread_file_or_urlrretriesz$Failed fetching metadata from url %s)r _set_datarread_maas_seed_dirMAASSeedDirNoneMAASSeedDirMalformedLOGwarninggetwait_for_metadata_serviceread_maas_seed_urlrreadurlr Exceptionrlogexc)rmcfgexcurlrrr _get_data>s.   zDataSourceMAAS._get_datacCs||_|\}}}||_||_||_|rytj||_Wn>tk r}ztj d|d|_WYdd}~XnXdS)Nz"Invalid content in vendor-data: %s) rZ userdata_rawmetadataZvendordata_purerZconvert_vendordataZvendordata_raw ValueErrorr*r+)rr4dataZudmdvderrrr&`s    zDataSourceMAAS._set_datacCs d|jS)z/Return the subplatform metadata source details.z seed-dir (%s))r)rrrr_get_subplatformnszDataSourceMAAS._get_subplatformc Csg|j}d}yt|jd|}Wn%tk rRtjtd|YnX|dkrcdSd}y(||krt|jd|}Wn"tk rtjd|YnXtj}|j d r|dd}d |t f}|g}|j j d |d|d|\}}|r:tj d |n#tjd|ttj|t|S)Nxmax_waitz Failed to get max wait. using %srF2timeoutzFailed to get timeout, using %s/r%z%s/%s/meta-data/instance-idurlszUsing metadata source: '%s'z(Giving up on md from %s after %i seconds)rintr,r0rr1r*r+timeendswith MD_VERSIONrZ wait_for_urldebugZcriticalbool) rr4r2r>r@Z starttimeZ check_urlrBZ _responserrrr-rs4         z(DataSourceMAAS.wait_for_metadata_servicecCsD|jdkrdStj|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_idsz DataSourceMAAS.check_instance_id)__name__ __module__ __qualname____doc__rKrrrpropertyrr r5r&r<r-rLrrrrr s    "   "r cs2d}tfdd|D}tj|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_secretcs%g|]}|j|fqSr)r,).0r)cfgrr s z$get_oauth_helper..)rRrSrTrU)dictrZOauthUrlHelper)rXkeyskwargsr)rXrrsrcsHd }djfdd|D}dtj|jdjS) z;Given a config, generate a unique identifier for this node.rRrTrUcs"g|]}j|dqS))r,)rVf)rrrrYs z&get_id_from_ds_cfg..zv1:zutf-8)z consumer_keyz token_keyz token_secret)rhashlibZsha256encodeZ hexdigest)rZfieldsZidstrr)rrrs"rcCsm|jdr|dd}tjj| sMttj|dkrYtdtd|ddS)Nzfile://rz%s: not a directoryz file://%sversion) startswithrrisdirlenlistdirr(r.)Zseed_drrrr's . r'cCs|dkrtj}|jdr4|dd }i}x@tD]8\}}} } |dkrrd||f} nd|||f} ytj|} || d|d|d| } | jr| r| j||//meta-data/instance-id * //meta-data/local-hostname * //user-data If version is None, then / will not be used. NrAr%z%s/%sz%s/%s/%sr$r@ ssl_detailsz4Fetching from %s resulted in an invalid http code %sizMissing required %s: %srC)rr#rF DS_FIELDSrZfetch_ssl_detailsokcontentsZ decode_binaryr*r+codeZUrlErrorr)check_seed_contents)Zseed_urlr#r@rcrr$r9rZ _dictnamebinaryoptionalr4rhZrespr;rrrr.s6       r.c Cs_i}g}xtD]\}}}}||krG|s|j|qd|kr|jd\}} } ||kri||<||||| | jD]0} | | kr=| | dkr=| | | | |d?d@\}}}t!dAt!|dkrdBn |j"t!dC|j|t!dD|j|dkrdBn|n]|jdkr7|jn>|jdkru|jj#d<shdE|j|_|jdS)Kz Call with single argument of directory or http or https url. If url is given additional arguments are allowed, which will be interpreted as consumer_key, token_key, token_secret, consumer_secret rNZ descriptionzInteract with MAAS DSz--configmetavarfilehelpzspecify DS config filerpz--ckeykeyzthe consumer key to auth withz--tkeyzthe token key to auth withz--csecZsecretzthe consumer secret (likely '')r^z--tseczthe token secret to auth withz--apiverrcz the apiver to use ( can be used)titleZ subcommandsdestsubcmdcrawlcrawl the datasourcer,do a single GET of provided url check-seedread and verify seed at urlr4zthe datasource urlnargs?rRrTrUrS91_kernel_cmdline_url 90_dpkg_maasz/etc/cloud/cloud.cfg.d/z.cfgzUsed config in %s. rJr r"z)Must provide a url or a config with url. r%csj|ddjS)Nr$r%)r/rk)r4)rrrgeturldszmain..geturlcs$td||jfdS)Nz == %s == %s )printdecode)r4)rrrprinturlhszmain..printurlcs|jdrx|jjD]c}|jdrTd||fq(|dkrwd|dfq(d||fq(Wn |dS)NrAz%s%sz meta-dataz meta-data/)rFr splitlines)r4line)rrrrrrks zmain..crawlzChecking seed at %s rAzfile://r#r$z=== user-data ===zN/Az=== meta-data ===z=== vendor-data ===z%s/zcrawlrzgetrrr)rrr)rr)$argparsepprintsysArgumentParser add_argumentrGZadd_subparsersZ add_parser parse_argsZckeyZtkeyZtsecZcsecconfigrrexistsaccessR_OKstderrwriterZ read_confr[r4exitrrr/rdr'r.ZapiverrrrF)rrrparserZsubcmdsnamer}rvargsZcredsfnameZfpathrXr~r/Zuserdatar6r:r)rrrrrmain's           '        % "   "r)rrFF)rrFF)rrFT)r r TT)r r TT)r`rrEZ cloudinitrZloggingrrrZ getLoggerrMr*rGrirr rrr'r.rmr0r(r)ZDEP_FILESYSTEMZ DEP_NETWORKrxryrrrrr s6     / &   i