o
    ÷¢iŒ
  ã                   @   s¦   d dl Z d dlZd dlZd dlZd dlmZ d dlmZ e e	¡Z
dedefdd„Zdedefdd	„Zdefd
d„Zdefdd„Zdefdd„ZG dd„ de jƒZdS )é    N)ÚOptional)ÚdmiÚsystem_uuidÚreturnc                 C   s>   zt  | ¡}W n ty   t d| ¡ ‚ w tt j|jdƒS )aà  Byte 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.
    zFailed to parse system uuid: %r)Úbytes)ÚuuidÚUUIDÚ
ValueErrorÚLOGÚerrorÚstrÚbytes_le)r   Úoriginal_uuid© r   úB/usr/lib/python3/dist-packages/cloudinit/sources/azure/identity.pyÚbyte_swap_system_uuid   s   þr   c                 C   s   t ƒ rt| ƒS | S )z!Determine VM ID from system uuid.)Ú
is_vm_gen1r   ©r   r   r   r   Úconvert_system_uuid_to_vm_id'   s   r   c                   C   s$   t j d¡rdS t j d¡rdS dS )zZDetermine if VM is gen1 or gen2.

    Gen2 guests use UEFI while gen1 is legacy BIOS.
    z/sys/firmware/efiFz/dev/efiT)ÚosÚpathÚexistsr   r   r   r   r   /   s
   r   c                  C   s2   t  d¡} | du rtdƒ‚|  ¡ } t d| ¡ | S )z Query system uuid in lower-case.zsystem-uuidNzfailed to read system-uuidzRead product uuid: %s)r   Úread_dmi_dataÚRuntimeErrorÚlowerr
   Údebugr   r   r   r   Úquery_system_uuid?   s   
r   c                  C   s   t ƒ } t| ƒS )zQuery VM ID from system.)r   r   r   r   r   r   Úquery_vm_idK   s   r   c                   @   s&   e Zd ZdZeded  fdd„ƒZdS )ÚChassisAssetTagz 7783-7084-3265-9085-8269-3286-77r   c                 C   sL   t  d¡}z| |ƒ}W n ty   t d|¡ Y dS w t d||j¡ |S )zŽCheck platform environment to report if this datasource may run.

        :returns: ChassisAssetTag if matching tag found, else None.
        zchassis-asset-tagzNon-Azure chassis asset tag: %rNz Azure chassis asset tag: %r (%s))r   r   r	   r
   r   Úname)ÚclsÚ	asset_tagÚtagr   r   r   Úquery_systemT   s   
þzChassisAssetTag.query_systemN)Ú__name__Ú
__module__Ú__qualname__ÚAZURE_CLOUDÚclassmethodr   r#   r   r   r   r   r   Q   s    r   )ÚenumÚloggingr   r   Útypingr   Ú	cloudinitr   Ú	getLoggerr$   r
   r   r   r   Úboolr   r   r   ÚEnumr   r   r   r   r   Ú<module>   s   
