ruạṛ
a ���fG � @ sj d dl Z d dlZd dlZd dlZd dlmZ e� �d�durJddlm Z nd dl Z da dZ dZ e j �e d �ZdaG d d � d �Zd d � Zdd� Zdd� Zdd� Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d!� Z d"d#� Z d$d%� Z d&d'� Z d(d)� Z d*d+� Z!eeed,�Z"eed-�Z#eeed.�Z$eed-�Z%e&d/k�rfe � Z'd0a e"e'j(d e'j)ee'j*��\Z+Z,e-e,� dS )1� N)�ArgumentParser�__spec__� )� im_commonFz(/var/lib/oracle-cloud-agent/plugins/osmsz .osms_readyz.osmsv2_pid_filec @ s& e Zd ZdZdZdZdZddd�ZdS )� OsmsxParamsN� c C s || _ || _|| _|| _d S �N)�osms_pid� oca_version�instance_metadata�logger)�selfr r r r � r �4/sparta/input/plugins/osms/agent/shared/osmsx_ctl.py�__init__ s zOsmsxParams.__init__)r NN)�__name__� __module__� __qualname__r r r r r r r r r r s r c C s� z�g }| dkr�d}t d� tD ]~}z*t| |�\}}|dkrJd}|�|� W q ty� } z6t dt|� � t |� d}|�t|�� W Y d}~q d}~0 0 q d�|�|fW S t| |�W S W n: t� y� } z t |� t|�dfW Y d}~S d}~0 0 dS )z� Enable given plugin :param plugin: target plugin name, can be all, alx, osmsv2, or im :param osmsx_params: instance of OsmsxParams class :return: Tuple[str, int] error string and int 1 if fails, 0 if success �allr z$enabling all plugins managed by OSMS� zFailed to enable N�. )� print_in_cli�enable_plugins_mapping�append� Exception�str�join��plugin� osmsx_params� error_msgs�exit_val� single_plugin�single_error_message�single_exit_code�er r r �enable' s* &r& c C s. d}t � s d}t|� |dfS | jdkr<d}t|� |dfS td� tj�t��rtj�tt�}zZt t� |tj tj B d�d ��*}|� d� |� t| j�� W d � n1 s�0 Y W nJ t�y } z0t|� td t|� � d }|dfW Y d }~S d }~0 0 td � |dfS d}t|� |dfS d S )z� Enable alx by creating the flag file :param osmsx_params: instance of OsmsxParams class :return: Tuple[str, int] error string and int 1 if fails, 0 if success r �(alx can only operate on autonomous linuxr ����Bcannot find OSMS_PID, please provide correct OSMS_PID with -p flagr zenabling alx...� �wNz(Get IOError when creating flag file at: z alx failed to create flag filezalx is enabledz%alx flag file directory does not exit)�is_alxr r �os�path�isdir� osms_lib_dirr �osms_ready_flag_file�open�O_WRONLY�O_CREAT�truncate�writer �IOError)r � error_msg�file�f�errorr r r � enable_alxF s4 2 r< c C sh t � sd}t|� |dfS td� � z�tj�� }d|v rF|d |d< n |�dd� | j�d|� tj dd gtj tj |d �}z>t t d �� }|� t|j�� W d � n1 s�0 Y W n` t�y } zFz |�� W n ty� Y n0 d | }t|� |d fW Y d}~W S d}~0 0 W n> t�yZ } z$d| }t|� |d fW Y d}~S d}~0 0 td� dS )z� Enable osmsv2 :param osmsx_params: instance of OsmsxParams class :return: Tuple[str, int] error string and int 1 if fails, 0 if success �2osmsv2 can only operate on Oracle Linux 8 or abover zenabling osmsv2...�LD_LIBRARY_PATH_ORIG�LD_LIBRARY_PATHNzosmsv2 env: %sz5/usr/libexec/oracle-cloud-agent/plugins/osmsv2/osmsv2z -plugin=false)�stdout�stderr�envr+ z#failed to write osmsv2 pid file: %sr zfailed to enable osmsv2: %szosmsv2 is enabled�r r )�is_ol8_or_abover r- �environ�copy�popr �debug� subprocess�Popen�DEVNULLr2 �osmsv2_pid_filer6 r �pidr �kill)r r8 rB �procr: r; r r r � enable_osmsv2k sD � 2 $ rP c C s� z�g }| dkr�d}t D ]~}z*t | |�\}}|dkrBd}|�|� W q ty� } z6tdt|� � t|� d}|�t|�� W Y d}~qd}~0 0 qd�|�|fW S t | |�W S W n8 ty� } z t|� t|�dfW Y d}~S d}~0 0 dS )a Check given plugin, called after every success OSMS check in :param plugin: target plugin name, can be all, alx, osmsv2, or im :param osmsx_params: instance of OsmsxParams class :return: Tuple[str, int] error string and int 1 if fails, 0 if success r r r zFailed to check Nr )�check_plugins_mappingr r r r r r r r r �check� s( &rR c C sd d}t � s d}t|� |dfS | jdkr<d}t|� |dfS td� tj�tt�}tj�|��rXz�t t� |tj tj B d�d ��x}|� � � � }|t| j�kr�|�d� |�d� |�t| j�� td |� d | j� �� ntd � |dfW d � W S 1 � s�0 Y W nJ t�yT } z0t|� tdt|� � d}|dfW Y d }~S d }~0 0 nt| �S d S )z� Check alx flag file exists and contains correct pid. If not, create the file/change the pid :param osmsx_params: instance of OsmsxParams class :return: Tuple[str, int] error string and int 1 if fails, 0 if success r r' r r( r) r zchecking alx flag file...r* zw+z Updating flag file PPID from z to z"no change needed for alx flag fileNz,Get IOError when checking alx flag file at: z alx failed to check flag file)r, r r r- r. r r0 r1 �existsr2 �O_RDWRr4 �readline�stripr �seekr5 r6 r7 r< )r r8 r9 r: �old_ppidr; r r r � check_alx� s8 . rY c C s* t � sd}t|� |dfS td� tj�t�s6t| �S z�td�\}}}|dkrXt| �W S t|� � �}t t�� }t|� � � � �}W d � n1 s�0 Y ||kr�t td�� }|� t |�� W d � n1 s�0 Y W n> t�y } z$d| }t|� |dfW Y d}~S d}~0 0 td � d S ) z� Check osmsv2 process and start it if needed. :param osmsx_params: instance of OsmsxParams class :return: Tuple[str, int] error string and int 1 if fails, 0 if success r= r zchecking osmsv2...�?pgrep -nf /usr/libexec/oracle-cloud-agent/plugins/osmsv2/osmsv2Nr+ zfailed to check osmsv2: %sr zcheck osmsv2 finishedrC )rD r r- r. rS rL rP �run_cmd�intrV r2 �readr6 r r ) r r8 �out�err� return_code� current_pidr: � saved_pidr; r r r � check_osmsv2� s. . 0 rc c C s� da tj�d�s$d}t|� |dfS tdu r@t�| j| j | j �at� � r\d}t|� |dfS t� � sxd}t|� |dfS t� � }t|� t�� s�dnd }t�|�\}}t|� ||fS ) a; Check if IM plugin is already deployed, if not deploy it First we call activate endpoint that will return 404, or 202 and fill workRequestId If activate call succeeded (202 and workRequestId returned), the next call to check_im will be checking the status of the workRequest until it's completed. If work request fails, new activate call be we done at next backoff interval followed by workRequest check. Once workrequest completed successfully, osms_im_ready_flag_file gets created and check_im becomes no operation Exponential backoff details - after 10 backed off calls (now, 2/4/8/16/32/64/128/256/512/1024mins later), we try once 24hrs (every 1440mins) :param osmsx_params: instance of OsmsxParams class :return: Tuple[str, int] error string and int 1 if fails, 0 if success T�linuxz im can only operate on linuxr Nzim already activatedz)im skipping request due to required delay�activate� workrequest)�cli_mode�sys�platform� startswithr �osms_imr �OsmsImr r r � has_flag_file�is_ready_to_run�*reset_work_request_if_max_attempts_reached�get_work_request_id� call_action)r �cli_msg�action�statusr r r �check_im s* ru c C s� z�g }| dkr�d}t d� tD ]~}z*t| |�\}}|dkrJd}|�|� W q ty� } z6t dt|� � t |� d}|�t|�� W Y d}~q d}~0 0 q d�|�|fW S t| |�W S W n: t� y� } z t |� t|�dfW Y d}~S d}~0 0 dS )z� Disable the target plugin :param plugin: target plugin name, can be all, alx, osmsv2, or im :param osmsx_params: instance of OsmsxParams class :return: Tuple[str, int] error string and int 1 if fails, 0 if success r r z%disabling all plugins managed by OSMSr zFailed to disable Nr )r �disable_plugins_mappingr r r r r r r r �disable9 s* &rw c C s� d}t � s d}t|� |dfS td� tj�tt�}tj�|�r�z t�|� td� |dfW S t y� } z,t|� d| }t|� |dfW Y d}~S d}~0 0 nd }t|� |dfS dS ) z� Disable alx by removing flag file :param osmsx_params: instance of OsmsxParams class :return: Tuple[str, int] error string and int 1 if fails, 0 if success r r' r zdisabling alx...zalx is disabledz failed to remove alx flag filer Nz alx flag file does not exist) r, r r- r. r r0 r1 rS �remove�OSError)r r8 r9 r% r r r � disable_alxZ s( rz c C s t � sd}t|� |dfS td� tj�t�sBd}t|� |dfS z�tt�� }t|�� � � �}W d � n1 sr0 Y t� |t j � t� t� td�\}}}|dkr�t|� � �}t� |t j � W n> t�y } z$d| }t|� |dfW Y d}~S d}~0 0 td � d S ) z� Disable osmsv2 :param osmsx_params: instance of OsmsxParams class :return: Tuple[str, int] error string and int 1 if fails, 0 if success r= r zdisabling osmsv2...z osmsv2 pid file does not existNrZ z failed to disable osmsv2: %sr zosmsv2 is disabledrC )rD r r- r. rS rL r2 r\ r] rV rN �signal�SIGKILL�unlinkr[ r )r r8 r: rM r^ r_ r` r; r r r �disable_osmsv2y s0 . r~ c C sZ t dd�} | jdddg d�dd� | jdd dg d �d d d � | jddtt� dd� | �� S )z; Parse CLI arguments :return: parsed arguments zAScript called by OSMS agent to control plugins depending on OSMS.)� descriptionrs r �r& rR rw zenable/check/disable)�nargs�metavar�choices�helpr �?)�alx�osmsv2�imr z<target plugin name, can be alx/osmsv2/im/all, default to allr )r� r� r� r� �defaultz-p�OSMS_PIDz$OSMS_PID, should be OSMS Agent's PID)r� �typer� r� )r � add_argumentr\ � get_osms_pid� parse_args)�parserr r r � init_argparse� s ���r� c C sT z:t j�d�r6d} t| �\}}}|dkr6t|�� �W S W dS tyN Y dS 0 dS )zX pgrep -nf /usr/libexec/oracle-cloud-agent/plugins/osms/osms-agent :return: rd zApgrep -nf /usr/libexec/oracle-cloud-agent/plugins/osms/osms-agentr r( N)rh ri rj r[ r\ rV r )�commandr^ r_ r` r r r r� � s r� c C sX t j�� }d|v r |d |d< n |�dd � tj| tjd|d�}|�� \}}|||jfS )Nr> r? T)r@ �shellrB ) r- rE rF rG rI rJ �PIPE� communicate� returncode)r� rB rO r^ r_ r r r r[ � s � r[ c C sH z.t j�d�sW dS d} t| �\}}}|dkW S tyB Y dS 0 dS )zC Check whether the system is autonomous linux :return: rd Fzrpm -q alx-utilsr N)rh ri rj r[ r )�cmdr^ r_ r` r r r r, � s r, c C sF z,t d�\} }}|dkr W dS | �d�dkW S ty@ Y dS 0 d S )Nz,rpm -q --qf '%{version}' oraclelinux-releaser Fzutf-8�8)r[ �decoder )r^ r_ r` r r r rD � s rD c C s t r| rt| � d S r )rg �print)�stringr r r r � s r r� )r� r� )r� r� r� �__main__T).r- r{ rI rh �argparser �globals�get�sharedr rg r0 r1 r. r rL rk r r& r< rP rR rY rc ru rw rz r~ r� r� r[ r, rD r �action_choicesr rQ rv r �argsrs r �p�msg� exit_code�exitr r r r �<module> s` %, - 6! ����
cải xoăn