File: //usr/lib64/python3.9/site-packages/rhsmlib/dbus/__pycache__/server.cpython-39.opt-1.pyc
a
X�Zh> � @ s2 d dl Z d dlmZmZmZmZmZmZ d dlZ d dl
Z d dl Z d dl Z d dl
mZ d dlmZ d dlmZ d dlmZ d dlmZ d dlmZ d d lmZ d d
lm Z m Z d d lm Z m Z m Z m!Z!m"Z" d d l#m$Z% d d
l&m'Z' er�d dl(Z d dl)m*Z* e �+e,�Z-e� Z.e�/e.�Z0G dd� d�Z1G dd� d�Z2dS )� N)�Any�Dict�List�Optional�
TYPE_CHECKING�Type)� constants)�
connection)�GLib)�partial)�config)�
pid_of_sender)�get_config_parser)�create_filesystem_watcher�DirectoryWatch)�CONSUMER_WATCHER�ENTITLEMENT_WATCHER�CONFIG_WATCHER�PRODUCT_WATCHER�SYSPURPOSE_WATCHER)� injection)� init_logger)�
BaseObjectc @ s� e Zd ZdZdZdd� Zded ee ee d ee
d�dd �Z dd
d �Z e
d d
� �Zdd� Zededd�dd��Zededd�dd��ZdS )�Serverz9
Class used for rhsm.service providing D-Bus API
Nc O s t �| �| _| jS )zK
Function called, when new instance of Server is requested
)�object�__new__�INSTANCE)�cls�args�kwargs� r �9/usr/lib64/python3.9/site-packages/rhsmlib/dbus/server.pyr = s zServer.__new__zdbus._dbus.Busr )� bus_class�bus_name�object_classes�
bus_kwargsc C s� dt _tt� tjj�� |p"tj | _
|p.tj }|p6i }|p>g }g | _ i | _
z|f i |��| _W n" tjjy� t�d� � Y n0 t�tj�| _d}td d }td d }d}tj�| j
| j�| _t� � | _|D ]l} z| d | d }
} W n t �y
| }
i } Y n0 |
f |
j | jd
�| ��} | j � | � | | j
t |
j!�< q�g }
g }g }g }g }d | j
v �r�|� | j
d j"� |
� | j
d j"� |� | j
d j"� |� | j
d j"� |� | j
d j#� d | j
v �r�|
� | j
d j$� d
| j
v �r |� | j
d
j%j"� |� | j
d
j&� d| j
v �r*|� | j
d j'� d| j
v �rH|� | j
d j(� t)| jj*|
�}t)||�}t)||�}t)||�}t)||�}t+t,|t-|t.|t/|t0|i�| _1t2j3| j1j4dd�| _5| j5�6� dS )a�
Create a connection to a bus defined by bus_class and bus_kwargs; instantiate objects in
object_classes; expose them under bus_name and enter a GLib mainloop. bus_kwargs are generally
only necessary if you're using dbus.bus.BusConnection
The object_classes argument is a list. The list can contain either a class or a tuple consisting
of a class and a dictionary of arguments to send that class's constructor.
FzCould not create bus classz/etc/rhsm/rhsm.conf�rhsmZproductCertDirZentitlementCertDirz$/etc/rhsm/syspurpose/syspurpose.jsonr � )Z object_pathr# ZEntitlementDBusObjectZConsumerDBusObjectZConfigDBusObjectZProductsDBusObjectZSyspurposeDBusObjectzThread-FileSystemWatcher)�target�nameN)7r ZREUSE_CONNECTIONr �parser�dbus�mainloopZglibZ threads_initr ZBUS_NAMEr# � SystemBus�objectsZ
object_map�bus�
exceptionsZ
DBusException�log� exception�injZrequireZIDENTITY�identity�confZserviceZBusNameZconnection_namer
ZMainLoop� TypeErrorZdefault_dbus_path�append�str�__name__�reloadZEntitlementChangedZConsumerChanged�implZ
ConfigChangedZInstalledProductsChangedZSyspurposeChangedr Z
cert_dir_pathr r r r r r �filesystem_watcher� threading�ThreadZloop�_thread�start)�selfr" r# r$ r% Zconfig_cert_dir_pathZproducts_cert_dir_pathZentitlement_cert_dir_pathZsyspurpose_cert_dir_path�item�clazzr Zclazz_instanceZconsumer_dir_listZentitlement_dir_listZconfig_dir_listZproducts_dir_listZsyspurpose_dir_listZconsumer_dir_watchZentitlement_dir_watchZconfig_dir_watchZproducts_dir_watchZsyspurpose_dir_watchr r r! �__init__D s�
��
�� �zServer.__init__c
C s� z�z t �| j|� | j�� W n� tyL } zt�|� W Y d}~nbd}~0 tyx } zt�|� W Y d}~n6d}~0 t y� } zt�|� W Y d}~n
d}~0 0 W | j
� � |r�|� � n| j
� � |r�|� � 0 dS )z�
The two arguments, started_event and stopped_event, should be instances of threading.
Event that will be set when the mainloop has finished starting and stopping.
N)
r
Zidle_add�notify_startedr, �run�KeyboardInterruptr1 r2 �
SystemExit� Exceptionr< �stop�set)rA �
started_eventZ
stopped_event�er r r! rF � s "
�
z
Server.runc C s | r | � � dS )z�
This callback will be run once the mainloop is up and running. It's only purpose is to alert
other blocked threads that the mainloop is ready.
F)rK )rL r r r! rE � s zServer.notify_startedc C sX | j �� | j�� | j�d� | jD ] }|�� q&t� d| j
� �� | j � | j
� dS )z�
This method is primarily intended for uses of Server in a thread such as during testing since
in a single-threaded program, the execution would be blocked on the mainloop and therefore
preclude even calling this method.
� zReleasing Bus name: N)
r, �quitr< rJ r? �joinr. �remove_from_connectionr1 �debugr# r/ Z release_name)rA �objr r r! �shutdown� s
zServer.shutdown)� watcher_set�returnc C sV | j }|du rdS |jj�� D ]2\}}|dur8||vr8q t�d|� �� |�� q dS )z�
This function temporary disables file system directory watchers
:param watcher_set: Set of watchers. If the watcher is None, then all watchers are disabled
:return: None
Nz Disabling directory watcher: )r r< � dir_watches�itemsr1 rR Ztemporary_disable�r rU Zserver_instanceZdir_watcher_idZ dir_watcherr r r! � temporary_disable_dir_watchers� s z%Server.temporary_disable_dir_watchersc C sV | j }|du rdS |jj�� D ]2\}}|dur8||vr8q t�d|� �� |�� q dS )z�
This function enables file system directory watchers
:param watcher_set: Set of watcher. If the watcher is None, then all watchers are enabled
:return: None
Nz Enabling directory watcher: )r r< rW rX r1 rR �enablerY r r r! �enable_dir_watchers� s zServer.enable_dir_watchers)NNNN)NN)N)N)r9 �
__module__� __qualname__�__doc__r r r r8 r r �dictrD rF � staticmethodrE rT � classmethodrK rZ r\ r r r r! r 6 s* �
� i
r c @ s� e Zd ZU dZdZeed< dZeed< edd� �Z e
dd � �Z e
d
d � �Z ed d
� �Z
d dd�Zedd�dd�Zeed�dd�Zed�dd�Zdd� Zdd � ZdS ) �DomainSocketServera� This class sets up a DBus server on a domain socket. That server can then be used to perform
registration. The issue is that we can't send registration credentials over the regular system or
session bus since those aren't really locked down. The work-around is the client asks our service
to open another server on a domain socket, gets socket information back, and then connects and sends
the register command (with the credentials) to the server on the domain socket.z unix:tmpdir=�_server_socket_ifacez/run�_server_socket_pathc C s | j | j S �N)rd re �rA r r r! �_server_socket s z!DomainSocketServer._server_socketc C sr ||| j | jd�}t�d||f � |�|� | j� | jd7 _W d � n1 sX0 Y t�d|� d S )N)�conn�sender�cmd_linezInstance: %s of %s createdr' zNew connection: %s)rj rk r1 rR r7 �lock�connection_count)�domain_socket_serverZ
service_classZ object_listri rS r r r! �connection_added s �
,z#DomainSocketServer.connection_addedc C s� t �d|� | j�Z | jd8 _| jdkr8t �d� n&| jdkrNt �d� nt �d| j � W d � n1 sr0 Y d S )NzClosed connection: %sr' r zNo connections remainz Server still has one connectionz Server still has %d connections)r1 rR rl rm )rn ri r r r! �connection_removed( s
z%DomainSocketServer.connection_removedc C s | j r| j jS d S d S rf )�_server�addressrg r r r! rr 5 s zDomainSocketServer.addressNc C s� |pg | _ g | _d| _|| _t� | _t�d|� d�� | j�|� || _ t
� � | _ | j � d| _
W d � n1 st0 Y dS )a� Create a connection to a bus defined by bus_class and bus_kwargs; instantiate objects in
object_classes; expose them under bus_name and enter a GLib mainloop. bus_kwargs are generally
only necessary if you're using dbus.bus.BusConnection
The object_classes argument is a list. The list can contain either a class or a tuple consisting
of a class and a dictionary of arguments to send that class's constructor.
NzAdding sender z to the set of sendersr )r$ r. rq rj rK �_sendersr1 rR �addrk r= �Lockrl rm )rA r$ rj rk r r r! rD < s
zDomainSocketServer.__init__)rj rV c C s | j �|� dS )z3
Add sender to the list of senders
N)rs rt �rA rj r r r! �
add_senderQ s z DomainSocketServer.add_senderc C sR z| j �|� W n& ty6 t�d|� d�� Y dS 0 t�d|� d�� dS dS )z=
Try to remove sender from the set of sender
zSender z? wasn't removed from the set of senders (not member of the set)Fz removed from the set of sendersTN)rs �remove�KeyErrorr1 rR rv r r r! �
remove_senderW s z DomainSocketServer.remove_sender)rV c C sR d}t � }t�� }| jD ]&}t||�}|du r:|�|� qd}q| j�|�| _|S )z�
Check if other users are still running. It sender in the set is not
running, then remove sender from the set, because sender could
crash, or it was terminated since it called Start() method.
FNT)rK r+ r- rs r
rt �
difference)rA Zis_one_sender_runningZ not_runningr/ rj �pidr r r! � are_other_senders_still_runningd s
z2DomainSocketServer.are_other_senders_still_runningc C s: | j D ] }|�� q| j�� d | _ d | _d | _d | _d S rf )r. rQ rq Z
disconnectrj rk )rA �or r r! rT v s
zDomainSocketServer.shutdownc C sT t j�| j�| _| jD ] }| jj�tt j
| || j �� q| jj �tt j
| �� | jS rf )r+ Zserverr rh rq r$ Zon_connection_addedr7 r rc ro r. Zon_connection_removedrp rr )rA rC r r r! rF � s
�zDomainSocketServer.run)NNN)r9 r] r^ r_ rd r8 �__annotations__re �propertyrh ra ro rp rr rD rw �boolrz r} rT rF r r r r! rc s"
rc )3Zlogging�typingr r r r r r Z dbus.servicer+ Z dbus.serverZdbus.mainloop.glibr= Z rhsmlib.dbusr r& r Z
gi.repositoryr
� functoolsr Zrhsmlib.servicesr Zrhsmlib.dbus.dbus_utilsr
Z rhsm.configr Zrhsmlib.file_monitorr r r r r r r Zsubscription_managerr r3 Z rhsm.logutilr Z
dbus._dbusZrhsmlib.dbus.base_objectr Z getLoggerr9 r1 r* ZConfigr5 r rc r r r r! �<module> s2
T