o f @sddlZddlZddlZddlZddlmZddlmZddlm Z e e Z de de fddZde de fd d Zdefd d Zde fd dZde fddZGdddejZdS)N)Optional)dmi)report_diagnostic_event system_uuidreturncCsJzt|}Wntyd|}t|tjdwttj|jdS)aByte swap system uuid. Azure always uses little-endian for the first three fields in the uuid. This behavior was made strict in SMBIOS 2.6+, but Linux and dmidecode follow RFC 4122 and assume big-endian for earlier SMBIOS versions. Azure's gen1 VMs use SMBIOS 2.3 which requires byte swapping to match compute.vmId presented by IMDS. Azure's gen2 VMs use SMBIOS 3.1 which does not require byte swapping. :raises ValueError: if UUID is invalid. zFailed to parse system uuid:  logger_func)bytes)uuidUUID ValueErrorrLOGerrorstrbytes_le)r original_uuidmsgrB/usr/lib/python3/dist-packages/cloudinit/sources/azure/identity.pybyte_swap_system_uuids  rcCstrt|S|S)z!Determine VM ID from system uuid.) is_vm_gen1rrrrrconvert_system_uuid_to_vm_id)srcCs$tjdrdStjdrdSdS)zZDetermine if VM is gen1 or gen2. Gen2 guests use UEFI while gen1 is legacy BIOS. z/sys/firmware/efiFz/dev/efiT)ospathexistsrrrrr1s  rcCs2td}|dur td|}td||S)z Query system uuid in lower-case.z system-uuidNzfailed to read system-uuidzRead product uuid: %s)r read_dmi_data RuntimeErrorlowerr debugrrrrquery_system_uuidAs  r cCst}t|S)zQuery VM ID from system.)r rrrrr query_vm_idMsr!c@s&eZdZdZededfddZdS)ChassisAssetTagz 7783-7084-3265-9085-8269-3286-77rcCsZtd}z||}Wntytd|tjdYdSwtd||jftjd|S)zCheck platform environment to report if this datasource may run. :returns: ChassisAssetTag if matching tag found, else None. zchassis-asset-tagzNon-Azure chassis asset tag: %rrNz Azure chassis asset tag: %r (%s))rrr rr rname)cls asset_tagtagrrr query_systemVs    zChassisAssetTag.query_systemN)__name__ __module__ __qualname__ AZURE_CLOUD classmethodrr'rrrrr"Ssr")enumloggingrr typingr cloudinitrcloudinit.sources.helpers.azurer getLoggerr(r rrrboolrr r!Enumr"rrrrs