Mini Shell
3
�Mf�X � @ sP d dl mZ d dl mZ d dl mZ d dlmZ d dlmZ d dl mZ ddlZddl Z ddl
Z
ddl Z ddl Z dd l
mZ ddlZejd
d � dd lmZ ddlZejd
�ZdadZdddgZdZdZdZdZ e d Z ed Z dd� Z edd d g�Z G d d � d e!�Z"G d d!� d!e!�Z#d"d#� Z$G d$d%� d%ej%�Z&e&� Z'G d&d'� d'e!�Z(e(� Z)dS )(� )�errors)�udev)�util)�flags)�_)� safe_dbus� N)�
namedtuple�GLibz2.0)r
Zblivet� z /etc/iscsi/initiatorname.iscsiZcxgb3iZbnx2iZbe2iscsizorg.freedesktop.UDisks2z/org/freedesktop/UDisks2z /org/freedesktop/UDisks2/Managerz org.freedesktop.UDisks2.Managerz.ISCSI.Initiatorz.ISCSI.Sessionc C s< t jdt j�sdS ts8tjd�} | s(dS | atjdt� dS )Nz/sys/module/iscsi_tcpF�iscsidz ISCSID is %sT)�os�access�X_OK�ISCSID�shutil�which�log�info)�location� r �/usr/lib/python3.6/iscsi.py� has_iscsi8 s
r �
TargetInfo�ipaddr�portc @ s$ e Zd ZdZdd� Zedd� �ZdS )�NodeInfoz*Simple representation of node information.c C s: || _ || _|| _|| _|| _d | _d | _d | _d | _d S )N) �name�tpgt�addressr �iface�username�password�
r_username�
r_password)�selfr r r r r r r r �__init__M s zNodeInfo.__init__c C s | j | j| j| j| jfS )zrThe 5-tuple of connection info (no auth info). This form
is useful for interacting with storaged.
)r r r r r )r% r r r � conn_infoY s zNodeInfo.conn_infoN)�__name__�
__module__� __qualname__�__doc__r& �propertyr' r r r r r K s r c @ s e Zd Zdd� ZdS )� LoginInfoc C s || _ || _d S )N)�node� logged_in)r% r. r/ r r r r& b s zLoginInfo.__init__N)r( r) r* r& r r r r r- a s r- c C s dd� | D �S )zETransforms an 'a(sisis)' GLib.Variant into a list of NodeInfo objectsc S s g | ] }t |� �qS r )r )�.0r r r r �
<listcomp>i s z"_to_node_infos.<locals>.<listcomp>r )Zvariantr r r �_to_node_infosg s r2 c @ s e Zd ZdZdd� ZdS )�iSCSIDependencyGuardz*storaged iSCSI functionality not availablec C sV y0t jttt�sdS t jtttdtjdd�� W n t jk
rF dS X t jttt �S )NFZ
EnableModulesz(b)T)T)
r Zcheck_object_available�STORAGED_SERVICE�STORAGED_MANAGER_PATH�
MANAGER_IFACE� call_syncr
�Variant�
DBusCallError�INITIATOR_IFACE)r% r r r � _check_availo s
z!iSCSIDependencyGuard._check_availN)r( r) r* Z error_msgr; r r r r r3 l s r3 c @ sb e Zd ZdZdd� Zdd� Zdd� Zeede j
j d �d
d � ��Z ed d
� �Z
ede j
j d �d8dd��Zedd� �Zeede j
j d �dd� ��Zejede j
j d �dd� ��Zd9dd�Zedd� �Zd:dd �Zd;d d �Zede j
j d �d d � �Zede j
j d �d!d"� �Zd#d$� Zd%d&� Zd'd(� Zd)d*� Zd<d,d-�Z d=d.d/�Z d>d0d1�Z d?d2d3�Z d4d5� Z d6d7� Z!dS )@�iSCSIa� iSCSI utility class.
This class will automatically discover and login to iBFT (or
other firmware) configured iscsi devices when the startup() method
gets called. It can also be used to manually configure iscsi devices
through the add_target() method.
As this class needs to make sure certain things like starting iscsid
and logging in to firmware discovered disks only happens once
and as it keeps a global list of all iSCSI devices it is implemented as
a Singleton.
.. warning::
Since this is a singleton class, calling deepcopy() on the instance
just returns ``self`` with no copy being created.
c C s� i | _ g | _d| _d| _i | _d | _tjr�y| jd�d }|| _W n2 t k
rt } zt
j dt |�� W Y d d }~X n,X t
j d| j
� tjd|d f�}| jd|� d S ) Nr FZGetFirmwareInitiatorNamer z4failed to get initiator name from iscsi firmware: %sz+Setting up firmware iSCSI initiator name %sz(sa{sv})�SetInitiatorName)�discovered_targets�
ibft_nodes�
_initiator�started�ifaces�_iSCSI__connectionr �ibft�_call_initiator_method� Exceptionr r �str� initiatorr
r8 )r% Z
initiatorname�e�argsr r r r& � s
"ziSCSI.__init__c C s | S )Nr )r% r r r �__call__� s ziSCSI.__call__c C s | S )Nr )r% Z memo_dictr r r � __deepcopy__� s ziSCSI.__deepcopy__F)ZcriticalZ eval_modec C s dS )NTr )r% r r r � available� s ziSCSI.availablec C s | j stj� | _ | j S )N)rC r Zget_new_system_connection)r% r r r � _connection� s
ziSCSI._connectionTNc C s t jttt||| jd�S )z�Class a method of the ISCSI.Initiator DBus object
:param str method: name of the method to call
:param params: arguments to pass to the method
:type params: GLib.Variant
)Z
connection)r r7 r4 r5 r: rN )r% �methodrJ r r r rE � s z iSCSI._call_initiator_methodc C s
| j dkS )z&True if initiator is set at our level.r )r@ )r% r r r �
initiator_set� s ziSCSI.initiator_setc C s8 | j dkr| j S t| jd�d d d� �}|jddd�S ) Nr ZGetInitiatorNameRawr r zutf-8�replace)r ���)r@ �bytesrE �decode)r% Z
raw_initiatorr r r rH � s
ziSCSI.initiatorc C s� t |�dkrttd���| j� }|r2tjtd���tjd| j� t j
d|d f�}| j d|� | j r�|| j
kr�tjd� tjdd d
g�}|dkr�tjtd ���|| _
d S ) Nr z$Must provide an iSCSI initiator namez3Cannot change initiator name with an active sessionz"Setting up iSCSI initiator name %sz(sa{sv})r= z-Restarting iscsid after initiator name change� systemctlZrestartr z4Failed to restart iscsid after initiator name change)�len�
ValueErrorr �_get_active_sessionsr �
ISCSIErrorr r rH r
r8 rE rP r@ r � run_program)r% �val�activerJ �rcr r r rH � s
c C sD |r dd� | j j|g �D �S dd� tjt| j j� �� D �| j S dS )zNodes logged in toc S s g | ]}|j r|j�qS r )r/ r. )r0 r r r r r1 � s z&iSCSI.active_nodes.<locals>.<listcomp>c S s g | ]}|j r|j�qS r )r/ r. )r0 r r r r r1 � s N)r> �get� itertools�chain�list�valuesr? )r% �targetr r r � active_nodes� s ziSCSI.active_nodesc C s | j � s dS | jrdS dS d S )NZnoneZbind�default)rd rB )r% r r r �mode� s
z
iSCSI.modec C s: x4| j j� D ]&}x |D ]}|j|kr||_dS qW q W dS )zMMark node as one logged in to
Returns False if not found
TF)r> rb r. r/ )r% r. r\ Z login_infosr r r r �_mark_node_active s
ziSCSI._mark_node_activec C sB |dkrt � }tjdd�|d< tjd|j|f �}| jd|� dS )z�Try to login to the iSCSI node
:type node_info: :class:`NodeInfo`
:param dict extra: extra configuration for the node (e.g. authentication info)
:raises :class:`~.safe_dbus.DBusCallError`: if login fails
N�sZ automaticz node.startupz (sisisa{sv})ZLogin)�dictr
r8 r' rE )r% Z node_infoZextrarJ r r r �_login
s
z iSCSI._loginc C s� yt jttddd �d }W n2 t jk
rL } ztjdt|�� g S d }~X nX dd� |j� D �}g }x>|D ]6}|| t }|j
t |d |d |d |d
d �� qjW |S ) Nz"org.freedesktop.DBus.ObjectManagerZGetManagedObjectsr z(iscsi: Failed to get active sessions: %sc s s | ]}t jd |�r|V qdS )z.*/iscsi/session[0-9]+$N)�re�match)r0 �objr r r � <genexpr>) s z-iSCSI._get_active_sessions.<locals>.<genexpr>Z target_namer Zpersistent_addressZpersistent_port) r r7 r4 �
STORAGED_PATHr9 r r rG �keys�
SESSION_IFACE�appendr )r% ZobjectsrI Zsessionsr\ ZsessionZ
propertiesr r r rX s&
ziSCSI._get_active_sessionsc s6 t js
d S tjdg f�}y| jd|�\}}W n2 tjk
r^ } ztjdt |�� d S d }~X nX t
|�}| j � }x�|D ]�� t � fdd�|D ��r�tjd� j
� j� j� j� | jj� � y4| j� � tjd� j
� j� j� j� | jj� � W qv tjk
�r$ } ztjd� j
t |�� W Y d d }~X qvX qvW | j� d S ) Nz(a{sv})ZDiscoverFirmwarez iscsi: No IBFT info found: %sc 3 s> | ]6}� j |j ko4� j|jko4� j|jko4� j|jkV qd S )N)r r r r )r0 �a)r. r r rn F s z$iSCSI._start_ibft.<locals>.<genexpr>z9iscsi IBFT: already logged in node %s at %s:%s through %sz.iscsi IBFT: logged into %s at %s:%s through %sz+Could not log into ibft iscsi target %s: %s)r rD r
r8 rE r r9 r r rG r2 rX �anyr r r r r? rr rj �error� stabilize)r% rJ � found_nodes�_n_nodesrI rd r )r. r � _start_ibft6 s2
$ziSCSI._start_ibftc C s t jd� tj� d S )N� )�time�sleepr Zsettle)r% r r r rv V s
ziSCSI.stabilizec C sl xf|D ]^}dt | j� }tjdddd|dg� tjdddd|ddd d
|g
� || j|< tjd ||� qW d S ) Nziface%d�iscsiadmz-mr z-Iz--op=newz --op=updatez-nziface.net_ifacenamez-vzcreated_interface %s:%s)rV rB r rZ r �debug)r% rB r �iscsi_iface_namer r r �create_interfaces^ s
ziSCSI.create_interfacesc C s: | j s
d S x$| j D ]}tjdddd|dg� qW i | _ d S )Nr} z-mr z-Iz --op=delete)rB r rZ )r% r r r r �delete_interfacesl s
ziSCSI.delete_interfacesc C s� | j r
d S t� sd S tjdddg� | jdkr<tjd� d S x.dd� dD �D ] }tjj |�sLtj
|d� qLW tjd� tjddgt � t j
d�}|r�tjd|� tj|g� n
tjd� tjtg� tjd� | j� d| _ d S )NrU �startziscsi-init.servicer zno initiator setc s s | ]}t jjd |�V qdS )z/var/lib/iscsiN)r
�path�join)r0 �dr r r rn � s z iSCSI.startup.<locals>.<genexpr>rB �isns�nodes� send_targets�slp�statici� z
iSCSI startupZmodprobez-a�iscsiuioziscsi: iscsiuio is at %sziscsi: iscsiuio not found.r T)rB r� r� r� r� r� )rA r r rZ r@ r r r
r� �isdir�makedirs�
ISCSI_MODULESr r r~ r r{ r| ry )r% Zfulldirr� r r r �startupu s.
z
iSCSI.startup�3260c C s6 t � stjtd���| jdkr*ttd���| jt||��rJtj d||� n�| j
� t � }|rlt j
d|�|d< |r�t j
d|�|d< |r�t j
d|�|d< |r�t j
d|�|d < t j
d
|t|�|f�}| jd |�\} }
t| �} t||�} g | j| < x0| D ](}
| j| jt|
d �� tj d
|
j� q�W dd� | jt||� D �S )a�
Discover iSCSI nodes on the target available for login.
If we are logged in a node discovered for specified target
do not do the discovery again as it can corrupt credentials
stored for the node (set_auth and get_auth are using database
in /var/lib/iscsi/nodes which is filled by discovery). Just
return nodes obtained and stored in the first discovery
instead.
Returns list of nodes user can log in.
ziSCSI not availabler zNo initiator name setz6iSCSI: skipping discovery of %s:%s due to active nodesrh r! r" zreverse-usernamezreverse-passwordz (sqa{sv})ZDiscoverSendTargetsFzdiscovered iSCSI node: %sc S s g | ]}|j s|j�qS r )r/ r. )r0 r r r r r1 � s z"iSCSI.discover.<locals>.<listcomp>)r r rY r r@ rW rd r r r~ r� ri r
r8 �intrE r2 r> rr r- r )r% r r r! r" r# r$ � auth_inforJ r� rx rw Zt_infor. r r r �discover� s4
ziSCSI.discoverc
C s d}d}t � }|r"tjd|�|d< |r6tjd|�|d< |rJtjd|�|d< |r^tjd|�|d< yj| j||� d}tjd |j|j|j|j � | j
|�s�tj d
� |r�||_ |r�||_
|r�||_|r�||_W n> tjk
�r } z t| �}tjd |j|� W Y d d } ~ X nX ||fS )
aQ
:param node: node to log into
:type node: :class:`NodeInfo`
:param str username: username to use when logging in
:param str password: password to use when logging in
:param str r_username: r_username to use when logging in
:param str r_password: r_password to use when logging in
Fr rh r! r" zreverse-usernamezreverse-passwordTz)iSCSI: logged into %s at %s:%s through %sz&iSCSI: node not found among discoveredz iSCSI: could not log into %s: %sN)ri r
r8 rj r r r r r r rg ru r! r" r# r$ r r9 rG Zwarning)
r% r. r! r" r# r$ r] �msgr� rI r r r �
log_into_node� s<
"ziSCSI.log_into_nodec
C s� d}
d}| j ||| |
| | �}|dkr2tjtd���x�|D ]|}|rZ||jkrZtjd|j� q8|r�| jj|j |j �}||kr�tjd|j|� q8|
d }
| j
|||||�\}}|r8|d }q8W |
dkr�tjtd���|dkr�tjtd���| j � dS ) a�
Connect to iSCSI server specified by IP address and port
and add all targets found on the server and authenticate if necessary.
If the target parameter is set, connect only to this target.
NOTE: the iSCSI target can have two sets of different authentication
credentials - one for discovery and one for logging into nodes
:param str ipaddr: target IP address
:param str port: target port
:param user: CHAP username for node login
:type user: str or NoneType
:param pw: CHAP password for node login
:type pw: str or NoneType
:param user_in: reverse CHAP username for node login
:type user: str or NoneType
:param pw_in: reverse CHAP password for node login
:type pw_in: str or NoneType
:param target: only add this target (if present)
:type target: str or NoneType
:param iface: interface to use
:type iface: str or NoneType
:param discover_user: CHAP username for discovery
:type discover_user: str or NoneType
:param discover_pw: CHAP password for discovery
:type discover_pw: str or NoneType
:param discover_user_in: reverse CHAP username for discovery
:type discover_user: str or NoneType
:param discover_pw_in: reverse CHAP password for discovery
:type discover_pw_in: str or NoneType
r NzNo iSCSI nodes discoveredz*iscsi: skipping logging to iscsi node '%s'z1iscsi: skipping logging to iscsi node '%s' via %sr z No new iSCSI nodes discoveredz/Could not log in to any of the discovered nodes) r� r rY r r r r~ rB r^ r r� rv )r% r r �userZpwZuser_inZpw_inrc r Z
discover_userZ discover_pwZdiscover_user_inZdiscover_pw_in�foundr/ rw r. Znode_net_ifacer] Z_msgr r r �
add_target� s2 $
ziSCSI.add_targetc C s| | j s
d S tjj|d �r(tj|d � tjjd�rHtjd|d dd� tjj|d �shtj|d d� tjt |t � d S )Nz/var/lib/iscsiT)Zsymlinksz
/etc/iscsii� )
rP r
r� r� r ZrmtreeZcopytreer� Zcopyfile�INITIATOR_FILE)r% �root�storager r r �write= s z iSCSI.writec C sF x@| j � D ]4}|j|kr
|j|kr
|jt|�kr
|j|kr
|S q
W d S )N)rd r r r r� r )r% r r r r r. r r r �get_nodeM s
ziSCSI.get_nodec C s: g }dd� |j D �}x |D ]}|j|kr|j|� qW |S )Nc s s | ]}|j d kr|V qdS )�iscsiN)�type)r0 r� r r r rn W s z'iSCSI.get_node_disks.<locals>.<genexpr>)Zdevicesr. rr )r% r. r� Z
node_disksZ iscsi_disksZdiskr r r �get_node_disksU s
ziSCSI.get_node_disks)N)N)T)N)r� NNNN)NNNN) r� NNNNNNNNNN)N)"r( r) r* r+ r& rK rL r, �storaged_iscsi_requiredr ZEvalModeZonetimerM rN rE rP rH �setterrd rf rg rj rX ry rv r� r� r� r� r� r� r� r� r� r r r r r< sF
%
0
,
B
r< )*r r r r r Zi18nr r r
rk r r{ r_ � collectionsr ZgiZrequire_versionZ
gi.repositoryr
ZloggingZ getLoggerr r r� r� r4 ro r5 r6 r: rq r r �objectr r- r2 ZDependencyGuardr3 r� r< r� r r r r �<module> sL
d