CasperSecurity
o
`�wiM- � @ s� d dl Z d dlZd dlZddlmZ ddlmZ dZG dd� de�ZG dd � d e�Z G d
d � d e�Z
G d d
� d
e�Z G dd� de �Z
G dd� dej�ZG dd� de
�Zejdkr�G dd� dej�ZG dd� dej�ZG dd� dej�ZG dd � d e
�ZG d d � d e
�ZG d d � d e
�Ze� e� e� d!�Zejd"kr�eed# �Zn#eed$ �Zn G d%d� dej�ZG d&d � d e
�Zd#e� iZeed# �Zd'd(� Ze�� Z d)d*� Z d+d,� Z d-d.� Z dS )/� N� )�process)� reduction� c @ � e Zd ZdS )� ProcessErrorN��__name__�
__module__� __qualname__r r r �./usr/lib/python3.10/multiprocessing/context.pyr � r c @ r )�BufferTooShortNr r r r r r r
r c @ r )� TimeoutErrorNr r r r r r r
r c @ r )�AuthenticationErrorNr r r r r r r
r c @ s\ e Zd ZeZeZeZeZeej �Z eej
�Z
eej �Z dd� Z dd� Z
dCdd�Zdd � Zd
d � ZdDd
d�ZdEdd�ZdEdd�Zdd� ZdFdd�ZdGdd�ZdGdd �Zd d � Z dHd d!�Zd"d#� Zd$d%� Zdd&�d'd(�Z dd&�d)d*�Z d+d,� Z d-d.� Z dDd/d0�Z d1d2� Z!d3d4� Z"d5d6� Z#dDd7d8�Z$dId:d;�Z%dId<d=�Z&e'd>d?� �Z(e(j)d@d?� �Z(dAdB� Z*d S )J� BaseContextc C s t �� }|du r td��|S )z(Returns the number of CPUs in the systemNz cannot determine number of cpus)�os� cpu_count�NotImplementedError)�self�numr r r r ) s zBaseContext.cpu_countc C s& ddl m} || �� d�}|�� |S )z�Returns a manager associated with a running server process
The managers methods such as `Lock()`, `Condition()` and `Queue()`
can be used to create shared objects.
r )� SyncManager��ctx)�managersr � get_context�start)r r �mr r r �Manager1 s zBaseContext.ManagerTc C � ddl m} ||�S )z1Returns two connection object connected by a piper )�Pipe)�
connectionr )r �duplexr r r r r < � zBaseContext.Pipec C � ddl m} || �� d�S )z#Returns a non-recursive lock objectr )�Lockr )� synchronizer% r )r r% r r r r% A � zBaseContext.Lockc C r$ )z Returns a recursive lock objectr )�RLockr )r& r( r )r r( r r r r( F r' zBaseContext.RLockNc C � ddl m} ||| �� d�S )zReturns a condition objectr )� Conditionr )r& r* r )r �lockr* r r r r* K � zBaseContext.Conditionr c C r) )zReturns a semaphore objectr )� Semaphorer )r& r- r )r �valuer- r r r r- P r, zBaseContext.Semaphorec C r) )z"Returns a bounded semaphore objectr )�BoundedSemaphorer )r& r/ r )r r. r/ r r r r/ U r, z BaseContext.BoundedSemaphorec C r$ )zReturns an event objectr )�Eventr )r& r0 r )r r0 r r r r0 Z r' zBaseContext.Eventc C � ddl m} ||||| �� d�S )zReturns a barrier objectr )�Barrierr )r& r2 r )r �parties�action�timeoutr2 r r r r2 _ s zBaseContext.Barrierr c C r) )�Returns a queue objectr )�Queuer )�queuesr7 r )r �maxsizer7 r r r r7 d r, zBaseContext.Queuec C r) )r6 r )�
JoinableQueuer )r8 r: r )r r9 r: r r r r: i r, zBaseContext.JoinableQueuec C r$ )r6 r )� SimpleQueuer )r8 r; r )r r; r r r r; n r' zBaseContext.SimpleQueuer c C s" ddl m} |||||| �� d�S )z Returns a process pool objectr )�Pool)�context)�poolr< r )r � processes� initializer�initargs�maxtasksperchildr< r r r r< s s
�zBaseContext.Poolc G s ddl m} ||g|�R � S )zReturns a shared objectr )�RawValue)� sharedctypesrC )r �typecode_or_type�argsrC r r r rC z r, zBaseContext.RawValuec C s ddl m} |||�S )zReturns a shared arrayr )�RawArray)rD rG )r rE �size_or_initializerrG r r r rG s
zBaseContext.RawArray)r+ c G s( ddl m} ||g|�R || �� d��S )z$Returns a synchronized shared objectr )�Value�r+ r )rD rI r )r rE r+ rF rI r r r rI � s �zBaseContext.Valuec C r1 )z#Returns a synchronized shared arrayr )�ArrayrJ )rD rK r )r rE rH r+ rK r r r rK � s �zBaseContext.Arrayc C s4 t jdkrtt dd�rddlm} |� dS dS dS )z�Check whether this is a fake forked process in a frozen executable.
If so then run code specified by commandline and exit.
�win32�frozenFr )�freeze_supportN)�sys�platform�getattr�spawnrN )r rN r r r rN � s
�zBaseContext.freeze_supportc C s ddl m} |� S )zZReturn package logger -- if it does not already exist then
it is created.
r )�
get_logger)�utilrS )r rS r r r rS � s zBaseContext.get_loggerc C r )z8Turn on logging and add a handler which prints to stderrr )�
log_to_stderr)rT rU )r �levelrU r r r rU � r# zBaseContext.log_to_stderrc C s ddl m} dS )zVInstall support for sending connections and sockets
between processes
r )r! N)� r! )r r! r r r �allow_connection_pickling� s z%BaseContext.allow_connection_picklingc C � ddl m} ||� dS )z�Sets the path to a python.exe or pythonw.exe binary used to run
child processes instead of sys.executable when using the 'spawn'
start method. Useful for people embedding Python.
r )�set_executableN)rR rZ )r �
executablerZ r r r rZ � s zBaseContext.set_executablec C rY )zkSet list of module names to try to load in forkserver process.
This is really just a hint.
r )�set_forkserver_preloadN)�
forkserverr\ )r � module_namesr\ r r r r\ � s z"BaseContext.set_forkserver_preloadc C sB |d u r| S zt | }W n ty td| �d �w |�� |S )Nzcannot find context for %r)�_concrete_contexts�KeyError�
ValueError�_check_available)r �methodr r r r r � s �zBaseContext.get_contextFc C s | j S �N)�_name�r �
allow_noner r r �get_start_method� s z BaseContext.get_start_methodc C s t d��)Nz+cannot set start method of concrete context)ra �r rc �forcer r r �set_start_method� s z BaseContext.set_start_methodc C s t � �d�S )z_Controls how objects will be reduced to a form that can be
shared with other processes.r )�globals�get�r r r r �reducer� s zBaseContext.reducerc C s |t � d< d S )Nr )rl )r r r r r ro � � c C � d S rd r rn r r r rb � s z BaseContext._check_available)Trd )r )NN)r )NNr N�F)+r r
r r r r r � staticmethodr �current_process�parent_process�active_childrenr r r r% r( r* r- r/ r0 r2 r7 r: r; r< rC rG rI rK rN rS rU rX rZ r\ r rh rk �propertyro �setterrb r r r r r sT
�
r c @ s( e Zd ZdZedd� �Zedd� �ZdS )�ProcessNc C s t �� j�| �S rd )�_default_contextr ry �_Popen)� process_objr r r r{ � s zProcess._Popenc C s t �� j�� S rd )rz r ry � _after_forkr r r r r} � rp zProcess._after_fork�r r
r �
_start_methodrs r{ r} r r r r ry � s
ry c sF e Zd ZeZdd� Zd
� fdd� Zddd�Zdd d
�Zd d � Z� Z S )�DefaultContextc C s || _ d | _d S rd )rz �_actual_context)r r= r r r �__init__� s
zDefaultContext.__init__Nc s, |d u r| j d u r
| j| _ | j S t� �|�S rd )r� rz �superr )r rc �� __class__r r r � s
zDefaultContext.get_contextFc C s< | j d ur |s td��|d u r|rd | _ d S | �|�| _ d S )Nz context has already been set)r� � RuntimeErrorr ri r r r rk � s z DefaultContext.set_start_methodc C s" | j d u r
|r d S | j| _ | j jS rd )r� rz re rf r r r rh � s
z DefaultContext.get_start_methodc C s> t jdkrdgS t jdkrddgnddg}tjr |�d� |S )NrL rR �darwin�forkr] )rO rP r �HAVE_SEND_HANDLE�append)r �methodsr r r �get_all_start_methods s
z$DefaultContext.get_all_start_methodsrd rr )
r r
r ry r� r rk rh r� �
__classcell__r r r� r r� � s
r� rL c @ � e Zd ZdZedd� �ZdS )� ForkProcessr� c C � ddl m} || �S �Nr )�Popen)�
popen_forkr� �r| r� r r r r{ r# zForkProcess._PopenN�r r
r r rs r{ r r r r r� � r� c @ �( e Zd ZdZedd� �Zedd� �ZdS )� SpawnProcessrR c C r� r� )�popen_spawn_posixr� r� r r r r{ r# �SpawnProcess._Popenc C rq rd r r r r r r} " � �SpawnProcess._after_forkNr~ r r r r r� �
r� c @ r� )�ForkServerProcessr] c C r� r� )�popen_forkserverr� r� r r r r{ ) r# zForkServerProcess._PopenNr� r r r r r� ' r� r� c @ � e Zd ZdZeZdS )� ForkContextr� N)r r
r re r� ry r r r r r� . � r� c @ r� �� SpawnContextrR N�r r
r re r� ry r r r r r� 2 r� r� c @ s e Zd ZdZeZdd� ZdS )�ForkServerContextr] c C s t jstd��d S )Nz%forkserver start method not available)r r� ra rn r r r rb 9 s �z"ForkServerContext._check_availableN)r r
r re r� ry rb r r r r r� 6 s r� )r� rR r] r� rR r� c @ r� )r� rR c C r� r� )�popen_spawn_win32r� r� r r r r{ M r# r� c C rq rd r r r r r r} R r� r� Nr~ r r r r r� K r� c @ r� r� r� r r r r r� W r� c C s t | t_d S rd )r_ rz r� )rc r r r �_force_start_methodd s r� c C s t tdd �S )N�spawning_popen)rQ �_tlsr r r r �get_spawning_popenm s r� c C s
| t _d S rd )r� r� )�popenr r r �set_spawning_popenp s
r� c C s t � d u rtdt| �j ��d S )NzF%s objects should only be shared between processes through inheritance)r� r� �typer )�objr r r �assert_spawnings s
���r� ) r rO � threadingrW r r �__all__� Exceptionr r r r �objectr � BaseProcessry r� rP r� r� r� r� r� r� r_ rz r� �localr� r� r� r� r r r r �<module> sL ?
, �
�