File: //proc/self/root/usr/local/CyberCP/containerization/__pycache__/containerManager.cpython-310.pyc
o
��hW � @ sj d dl mZ d dlZd dlmZ d dlmZ d dl m
Z
d dl Z d dl Z d dl
mZ G dd� dej�ZdS )� )�ProcessUtilitiesN)�
ACLManager)�ServerStatusUtil)�httpProcc @ sb e Zd ZdZddd�Zdd� Zedd� �Zed d
� �Zd d � Z d
d� Z
dd� Z dd� Z dd� Z
dS )�ContainerManagera� group {groupName}{
cpu {
cpu.cfs_quota_us = {cfs_quota_us};
cpu.cfs_period_us = {cfs_period_us};
}
memory {
memory.limit_in_bytes = {memory}m;
}
blkio {
blkio.throttle.read_bps_device = "{major}:{minor} {io}";
blkio.throttle.write_bps_device = "{major}:{minor} {io}";
blkio.throttle.read_iops_device = "{major}:{minor} {iops}";
blkio.throttle.write_iops_device = "{major}:{minor} {iops}";
}
net_cls
{
net_cls.classid = 0x10{net_cls};
}
}Nc C s( t j�| � || _|| _|| _|| _d S )N)�multi�Thread�__init__�request� templateName�function�data)�selfr
r r r
� r �7/usr/local/CyberCP/containerization/containerManager.pyr s
zContainerManager.__init__c
C s� z'| j dkr
| �� W d S | j dkr| �� W d S | j dkr%| �� W d S W d S tyD } ztj�t|�d � W Y d }~d S d }~ww )N�submitContainerInstall�addTrafficController� removeLimitsz [ContainerManager.run]) r r r r �
BaseException�logging�CyberCPLogFileWriter� writeToFile�str)r �msgr r r �run% s
�"��zContainerManager.runc C sZ t | d�}t|�dkrd| S t|�dkrd| S t|�dkr#d| S t|�dkr+|S d S )N�02x� �000� �00� �0)�format�len)�count�hexValuer r r �
fetchHexValue0 s
�z ContainerManager.fetchHexValuec
C s� z[t �d�tj }tt �|��}td�} tt|�d d �}
tj�d| �} | �d|�} | �d|�} | �d|�} | �d|�} | �d | �} | �d
|
�} | �d t|��} | �d t|��} | W S t yv } zt
j � t| �� W Y d } ~ dS d } ~ ww )
N�/r i z {groupName}z{cfs_quota_us}z{cfs_period_us}z{memory}z{major}z{minor}z{io}z{iops}z {net_cls})
�os�stat�ST_DEVr �major�intr � defaultConf�replacer r r r )
� groupName� cfs_quota_us�
cfs_period_us�memory�io�iops�net_cls�devr+ �minor�finalIO�ioConfr r r r �prepConf= s( ��zContainerManager.prepConfc C s� i }d|d< d|d< t �� t jkr%d|d< d|d< t| jd|d�}|�� S t �� s=d|d< d|d< t| jd|d�}|�� S | jd krEi | _d| jd< d| jd< t| j| j|d�}|�� S )Nr �OLS� notInstalledr z"containerization/notAvailable.html�admin) r � decideServerr; r r
�render�containerCheckr
r )r r
�procr r r �renderCT s&
zContainerManager.renderCc
C s� z,| j jd }t�|�}|d dkrn tj�tjdd� W dS d}|d }t �
|� W d S t yL } ztj�tjt |�d d� W Y d }~d S d }~ww ) N�userIDr= r z4Not authorized to install container packages. [404].r zN/usr/local/CyberCP/bin/python /usr/local/CyberCP/containerization/container.pyz" --function submitContainerInstallz [404].)
r
�sessionr � loadedACLr r � statusWriterr �lswsInstallStatusPathr �outputExecutionerr r )r rC �
currentACL�execPathr r r r r l s"
�(��z'ContainerManager.submitContainerInstallc C s d}t �|� d}t �|� d S )Nz sudo systemctl restart cgconfigz sudo systemctl restart cgred)r � executioner�r �commandr r r �restartServices� s
z ContainerManager.restartServicesc C s� d}t �|� zdt| jd � }t �|� W n Y d}t �|� dt| jd � d t| jd � }t �|� d}t �|� | �� d S ) Nz;sudo tc qdisc add dev eth0 root handle 10: htb default 1000�&sudo tc class del dev eth0 classid 10:�classIDzFsudo tc class add dev eth0 parent 10: classid 10:1000 htb rate 100mbitz1sudo tc class add dev eth0 parent 10: classid 10:z
htb rate � rateLimitzKsudo tc filter add dev eth0 parent 10: protocol ip prio 10 handle 1: cgroup)r rK r r
rN rL r r r r � s
$
z%ContainerManager.addTrafficControllerc C s( dt | jd � }t�|� | �� d S )NrO rP )r r
r rK rN rL r r r r � s
z ContainerManager.removeLimits)NNNN)�__name__�
__module__� __qualname__r- r r � staticmethodr& r: rB r rN r r r r r r r s
r )�plogical.processUtilitiesr � threadingr � plogical.aclr � plogical.CyberCPLogFileWriterr r � serverStatus.serverStatusUtilr r( r) �plogical.httpProcr r r r r r r �<module> s