o fFE @sVdZddlZddlZddlZddlZddlZddlmZmZddl m Z ddl m Z m Z mZmZmZmZmZddlZddlmZddlmZddlmZdd lmZmZmZmZmZdd l m!Z!e"e#Z$d Z%d Z&d Z'dddddddZ(de)fddZ* d;deee)de e)e ffddZ+GdddeZ,GdddeZ-GdddeZ.de)de/fdd Z0Gd!d"d"ej1Z2de3fd#d$Z4 %dd5d6Z?e#d7kr)ddl@Z@d8ZAe@jBeAd9ZCeCDeEeFertrrrT) rlog_timer)r? read_metadatarsrogetmetadataupdate userdata_rawrrvendordata_raw)rKr> user_metadatar2r2r3 _get_datas.        zDataSourceLXD._get_datacCsdjttdS)z.Return subplatform details for this datasourcez"LXD socket API v. {ver} ({socket}))verrO)rkLXD_SOCKET_API_VERSIONr[rSr2r2r3_get_subplatformszDataSourceLXD._get_subplatformcCsBttjd}|di}t|tst|}|d|jdkS)z%Return True if instance_id unchanged. metadata_keysr~z instance-id) r MetaDataKeys META_DATArrfrgrrhr)rKsys_cfgresponsemdr2r2r3check_instance_ids    zDataSourceLXD.check_instance_idcCs|jtjkr?|jtjkr|t|jtr?|jdr(t d|jd|_n|jdr?dd|jd D}t ||_|jtjkrNt dt |_t t|jS)zNetwork config read from LXD socket config/user.network-config. If none is present, then we generate fallback configuration. rz,LXD datasource using provided network configdevicescSs g|] \}}|ddkr|qS)r9nicr2).0kvr2r2r3 s  z0DataSourceLXD.network_config..z8LXD datasource generating network config using fallback.) rrrUNSETrsrrfrgrr)r?itemsrAr )rKrr2r2r3network_configs(          zDataSourceLXD.network_config)rTrUrVdsnamerrrrr rstr__annotations__rsr DataSourcerur rxintrw staticmethodboolrzrrrpropertyrgrrWr2r2rLr3rps"  rpcCs"tjtrtttjSdS)z=Return True when this platform appears to have an LXD socket.F)ospathexistsr[statS_ISSOCKlstatst_moder2r2r2r3rys ryTsessionr\do_raisec Csrt|||}|jstd||j|jdiSz|WSty8}zt dj ||jdd|d}~ww)NSkipping %s on [HTTP:%d]:%sutf-8zFUnable to process LXD config at {url}. Expected JSON but found: {resp})r\resp) _do_requestokr)r? status_codecontentdecodejsonrrrjrk)rr\r url_responsernr2r2r3_get_json_responses,    rcCstdddD]}||}d|jkr!tdtd|j||qtd|j||r?|js?t dj |j||j d d |S) Nrig?z,[GET] [HTTP:%d] %s, retrying %d more time(s)z[GET] [HTTP:%d] %sz3Invalid HTTP response [{code}] from {route}: {resp}r)coderouter)rangerrtimesleepr)r*r?rrrjrkrr)rr\rretriesrr2r2r3r1s*     rc@s0eZdZeZeZeZeZeeBeBZdS)rN) rTrUrVrNONECONFIGDEVICESrALLr2r2r2r3rNs rc@sDeZdZefdefddZdejdefddZ de defd d Z d S) _MetaDataReader api_versioncCs||_tt|j|_dSrN)rr combine_urlLXD_URL _version_url)rKrr2r2r3rGWsz_MetaDataReader.__init__rrc Csdii}t|jd}t||}t|D]L}tt|}t||dd}|jd}|j s6t d||j |q| dd} ||d| <| tvr`t| |vrT||t| <qt d| | d d d q|S) aIterate on LXD API config items. Promoting CONFIG_KEY_ALIASES Any CONFIG_KEY_ALIASES which affect cloud-init behavior are promoted as top-level configuration keys: user-data, network-data, vendor-data. LXD's cloud-init.* config keys override any user.* config keys. Log debug messages if any user.* keys are overridden by the related cloud-init.* key. r>Frrr/rz,Ignoring LXD config %s in favor of %s value.userz cloud-initr6)rrrrsortedrrrrrr)r?r rpartitionCONFIG_KEY_ALIASESr*replace) rKrr> config_url config_routes config_routeconfig_route_urlconfig_route_response response_textcfg_keyr2r2r3_process_config[s:        z_MetaDataReader._process_configrcCstT}||jtd|ji}tj|vr)t |jd}t ||j d|d<tj |vr6|||tj|vrOt |jd}t||dd}|rO||d<|WdS1s[wYdS)N_metadata_api_versionr~rrFr)requestsSessionmountrrZrrrrrrrrrrrrr)rKrrrmd_router\rr2r2r3__call__s*     $z_MetaDataReader.__call__N) rTrUrVrrrGrrrgrrrr2r2r2r3rVs6rrrcCst|d|dS)a8Fetch metadata from the /dev/lxd/socket routes. Perform a number of HTTP GETs on known routes on the devlxd socket API. Minimally all containers must respond to /meta-data when the LXD configuration setting `security.devlxd` is true. When `security.devlxd` is false, no /dev/lxd/socket file exists. This datasource will return False from `is_platform_viable` in that case. Perform a GET of /config` and walk all `user.*` configuration keys, storing all keys and values under a dict key LXD_SOCKET_API_VERSION: config {...}. In the presence of the following optional user config keys, create top level aliases: - user.user-data -> user-data - user.vendor-data -> vendor-data - user.network-config -> network-config :param api_version: LXD API version to operated with. :param metadata_keys: Instance of `MetaDataKeys` indicating what keys to fetch. :return: A dict with the following optional keys: meta-data, user-data, vendor-data, network-config, network_mode, devices. Below is a dict representation of all raw configuration keys and values provided to the container surfaced by the socket under the /1.0/config/ route. )rr)r)rrr2r2r3rs#rcCs t|tSrN)rlist_from_depends datasources)dependsr2r2r3get_datasource_lists r__main__z*Query LXD metadata and emit a JSON object.) descriptionrrN)T)G__doc__loggingrrOrrenumrr json.decoderrtypingrrrrr r r rrequests.adaptersr urllib3.connectionr urllib3.connectionpoolr cloudinitrrrrr cloudinit.netr getLoggerrTr)r[rrrrr4rArCrXrZrgrorrprryrrResponserrrrrDEP_FILESYSTEMrrargparserArgumentParserparser parse_argsprint json_dumpsr2r2r2r3s  $         &   g  R *