0x1949 Team - FAZEMRX - MANAGER
Edit File: cfreactor.cpython-310.pyc
o �b[D � @ s� d Z ddgZddlZddlmZ ddlmZmZmZm Z m Z m Z m Z m Z mZmZmZ ddlmZmZmZmZmZmZmZmZmZmZm Z ddl m Z dd l m Z m!Z!m"Z" dd l#m$Z$ dZ%d Z&d Z'G d d� de"�Z(ee �G dd� de!��Z)ddd�Z*dS )z� A reactor for integrating with U{CFRunLoop<http://bit.ly/cfrunloop>}, the CoreFoundation main loop used by macOS. This is useful for integrating Twisted with U{PyObjC<http://pyobjc.sf.net/>} applications. �install� CFReactor� N)� implementer) �CFSocketCreateRunLoopSource�CFSocketCreateWithNative�CFSocketDisableCallBacks�CFSocketEnableCallBacks�CFSocketInvalidate�CFSocketSetSocketFlags�*kCFSocketAutomaticallyReenableReadCallBack�+kCFSocketAutomaticallyReenableWriteCallBack�kCFSocketConnectCallBack�kCFSocketReadCallBack�kCFSocketWriteCallBack) �CFAbsoluteTimeGetCurrent�CFRunLoopAddSource�CFRunLoopAddTimer�CFRunLoopGetMain�CFRunLoopRemoveSource� CFRunLoopRun� CFRunLoopStop�CFRunLoopTimerCreate�CFRunLoopTimerInvalidate�kCFAllocatorDefault�kCFRunLoopCommonModes)� IReactorFDSet)� _NO_FILEDESC�PosixReactorBase�_Waker)�log� �@ c @ s e Zd ZdZdd� ZdS )� _WakerPlusa� The normal Twisted waker will simply wake up the main loop, which causes an iteration to run, which in turn causes L{ReactorBase.runUntilCurrent} to get invoked. L{CFReactor} has a slightly different model of iteration, though: rather than have each iteration process the thread queue, then timed calls, then file descriptors, each callback is run as it is dispatched by the CFRunLoop observer which triggered it. So this waker needs to not only unblock the loop, but also make sure the work gets done; so, it reschedules the invocation of C{runUntilCurrent} to be immediate (0 seconds from now) even if there is no timed call work to do. c C s t �| �}| j�d� |S )zw Wake up the loop and force C{runUntilCurrent} to run immediately in the next timed iteration. T)r �doRead�reactor�_scheduleSimulate)�self�result� r( �</usr/lib/python3/dist-packages/twisted/internet/cfreactor.pyr# H s z_WakerPlus.doReadN)�__name__� __module__� __qualname__�__doc__r# r( r( r( r) r" 7 s r" c @ s� e Zd ZdZd/dd�Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� Z dd� Z dd� Z dd� Z dd� Zdd� Zdd � Zd d � Zd Zd d!� ZdZd0d"d#�Zd$d%� Zd&d'� Zd(d)� Zd*d+� Zd1d-d.�ZdS )2r a& The CoreFoundation reactor. You probably want to use this via the L{install} API. @ivar _fdmap: a dictionary, mapping an integer (a file descriptor) to a 4-tuple of: - source: a C{CFRunLoopSource}; the source associated with this socket. - socket: a C{CFSocket} wrapping the file descriptor. - descriptor: an L{IReadDescriptor} and/or L{IWriteDescriptor} provider. - read-write: a 2-C{list} of booleans: respectively, whether this descriptor is currently registered for reading or registered for writing. @ivar _idmap: a dictionary, mapping the id() of an L{IReadDescriptor} or L{IWriteDescriptor} to a C{fd} in L{_fdmap}. Implemented in this manner so that we don't have to rely (even more) on the hashability of L{IReadDescriptor} providers, and we know that they won't be collected since these are kept in sync with C{_fdmap}. Necessary because the .fileno() of a file descriptor may change at will, so we need to be able to look up what its file descriptor I{used} to be, so that we can look it up in C{_fdmap} @ivar _cfrunloop: the C{CFRunLoop} pyobjc object wrapped by this reactor. @ivar _inCFLoop: Is C{CFRunLoopRun} currently running? @type _inCFLoop: L{bool} @ivar _currentSimulator: if a CFTimer is currently scheduled with the CF run loop to run Twisted callLater calls, this is a reference to it. Otherwise, it is L{None} Nc C s@ i | _ i | _|d u r t}|| _|d u rt� }|| _t�| � d S �N)�_fdmap�_idmapr �_runnerr � _cfrunloopr �__init__)r& �runLoop�runnerr( r( r) r3 z s zCFReactor.__init__c C s2 | j st| �| _ | j�| j � | �| j � dS dS )z� Override C{installWaker} in order to use L{_WakerPlus}; otherwise this should be exactly the same as the parent implementation. N)�wakerr" �_internalReaders�add� addReader�r& r( r( r) � installWaker� s �zCFReactor.installWakerc sX |\}}|�j vrt�j|t� dS �j | \}} ��� ���fdd�} t��| � dS )a� The socket callback issued by CFRunLoop. This will issue C{doRead} or C{doWrite} calls to the L{IReadDescriptor} and L{IWriteDescriptor} registered with the file descriptor that we are being notified of. @param cfSocket: The C{CFSocket} which has got some activity. @param callbackType: The type of activity that we are being notified of. Either C{kCFSocketReadCallBack} or C{kCFSocketWriteCallBack}. @param ignoredAddress: Unused, because this is not used for either of the callback types we register for. @param ignoredData: Unused, because this is not used for either of the callback types we register for. @param context: The data associated with this callback by C{CFSocketCreateWithNative} (in C{CFReactor._watchFD}). A 2-tuple of C{(int, CFRunLoopSource)}. Nc s� d } d}z"�� � dkrt} n� tk}|r �t r ��� } n�t r%��� } W n ty9 t� � d } t � � Y nw | rE�� �| |� d S d S )NF���r ) �filenor r �_READr# �_WRITE�doWrite� BaseException�sys�exc_infor �err�_disconnectSelectable)�why�isRead�� callbackType�readWriteDescriptor�rwr& r( r) �_drdw� s( �� ��z(CFReactor._socketCallback.<locals>._drdw)r/ r r2 r r �callWithLogger) r& �cfSocketrI �ignoredAddress� ignoredData�context�fd� smugglesrc�src�sktrL r( rH r) �_socketCallback� s zCFReactor._socketCallbackc C s� |dkrt d��|| jv r| j| \}}}}nLg }|�|� tt|ttB tB | j|�}t |t t B t B � t t|d�}|�|� t| j|t� t|ttB tB � ddg}|| jt|�< ||||f| j|< d|| �|�< t||� dS )a� Register a file descriptor with the C{CFRunLoop}, or modify its state so that it's listening for both notifications (read and write) rather than just one; used to implement C{addReader} and C{addWriter}. @param fd: The file descriptor. @type fd: L{int} @param descr: the L{IReadDescriptor} or L{IWriteDescriptor} @param flag: the flag to register for callbacks on, either C{kCFSocketReadCallBack} or C{kCFSocketWriteCallBack} r< zInvalid file descriptor.r FTN)� RuntimeErrorr/ �appendr r r r r rV r r r �_preserveSOErrorr r r2 r r r0 �id� _flag2idxr ) r& rR �descr�flagrT �cfs�gotdescrrK �ctxr( r( r) �_watchFD� sR ��� ��� ���zCFReactor._watchFDc C s t ttti| S )a- Convert a C{kCFSocket...} constant to an index into the read/write state list (C{_READ} or C{_WRITE}) (the 4th element of the value of C{self._fdmap}). @param flag: C{kCFSocketReadCallBack} or C{kCFSocketWriteCallBack} @return: C{_READ} or C{_WRITE} )r r> r r? )r& r] r( r( r) r[ s zCFReactor._flag2idxc C s� t |�| jvr dS |dkr| jt |� }n|}| j| \}}}}t||� d|| �|�< |t sK|t sM| jt |�= | j|= t| j|t � t |� dS dS dS )a� Unregister a file descriptor with the C{CFRunLoop}, or modify its state so that it's listening for only one notification (read or write) as opposed to both; used to implement C{removeReader} and C{removeWriter}. @param fd: a file descriptor @type fd: C{int} @param descr: an L{IReadDescriptor} or L{IWriteDescriptor} @param flag: C{kCFSocketWriteCallBack} C{kCFSocketReadCallBack} Nr< F) rZ r0 r/ r r[ r> r? r r2 r r )r& rR r\ r] �realfdrT r^ rK r( r( r) � _unwatchFD s �zCFReactor._unwatchFDc C � | � |�� |t� dS )z7 Implement L{IReactorFDSet.addReader}. N)ra r= r �r& �readerr( r( r) r9 7 � zCFReactor.addReaderc C rd )z7 Implement L{IReactorFDSet.addWriter}. N)ra r= r �r& �writerr( r( r) � addWriter= rg zCFReactor.addWriterc C rd )z: Implement L{IReactorFDSet.removeReader}. N)rc r= r re r( r( r) � removeReaderC rg zCFReactor.removeReaderc C rd )z: Implement L{IReactorFDSet.removeWriter}. N)rc r= r rh r( r( r) � removeWriterI rg zCFReactor.removeWriterc C sH dd� | j �� D �}|t| j�8 }|D ] }| �|� | �|� qt|�S )z7 Implement L{IReactorFDSet.removeAll}. c S s h | ]\}}}}|�qS r( r( ��.0rT r^ r\ rK r( r( r) � <setcomp>S s z&CFReactor.removeAll.<locals>.<setcomp>)r/ �values�setr7 rk rl �list)r& �allDesc�descr( r( r) � removeAllO s zCFReactor.removeAllc C � dd� | j �� D �S )z8 Implement L{IReactorFDSet.getReaders}. c S � g | ] \}}}}|t r|�qS r( )r> rm r( r( r) � <listcomp>^ � z(CFReactor.getReaders.<locals>.<listcomp>�r/ rp r: r( r( r) � getReadersZ � zCFReactor.getReadersc C rv )z8 Implement L{IReactorFDSet.getWriters}. c S rw r( )r? rm r( r( r) rx d ry z(CFReactor.getWriters.<locals>.<listcomp>rz r: r( r( r) � getWriters` r| zCFReactor.getWritersc C s t �| |�}| �� |S )aw Override L{PosixReactorBase}'s implementation of L{IDelayedCall.reset} so that it will immediately reschedule. Normally C{_moveCallLaterSooner} depends on the fact that C{runUntilCurrent} is always run before the mainloop goes back to sleep, so this forces it to immediately recompute how long the loop needs to stay asleep. )r �_moveCallLaterSoonerr% )r& �tpler' r( r( r) r~ f s z CFReactor._moveCallLaterSoonerFc C s$ d| _ z | �� W d| _ dS d| _ w )z� Run the runner (C{CFRunLoopRun} or something that calls it), which runs the run loop until C{crash()} is called. TFN)� _inCFLoopr1 r: r( r( r) �mainLoopt s zCFReactor.mainLoopc sx � j dur t� j � d� _ � �� }|rd}|dur:t� | }� fdd�}tt|ddd|d� }� _ t� j|t� dS dS )aV Schedule a call to C{self.runUntilCurrent}. This will cancel the currently scheduled call if it is already scheduled. @param force: Even if there are no timed calls, make sure that C{runUntilCurrent} runs immediately (in a 0-seconds-from-now C{CFRunLoopTimer}). This is necessary for calls which need to trigger behavior of C{runUntilCurrent} other than running timed calls, such as draining the thread call queue or calling C{crash()} when the appropriate flags are set. @type force: C{bool} Ng c s d � _ � �� � �� d S r. )�_currentSimulator�runUntilCurrentr% )�cftimer�extrar: r( r) �simulate� s z-CFReactor._scheduleSimulate.<locals>.simulater ) r� r �timeoutr r r r r2 r )r& �forcer� �fireDater� �cr( r: r) r% � s ��zCFReactor._scheduleSimulatec O s( t j| ||g|�R i |��}| �� |S )z6 Implement L{IReactorTime.callLater}. )r � callLaterr% )r& �_seconds�_f�args�kw� delayedCallr( r( r) r� � s zCFReactor.callLaterc C s t �| � | �d� dS )z1 Implement L{IReactorCore.stop}. TN)r �stopr% r: r( r( r) r� � s zCFReactor.stopc C s< | j }t�| � | jr| �� dS |r | �d| j� dS dS )z1 Implement L{IReactorCore.crash} r N)�_startedr �crashr� �_stopNowr� )r& � wasStartedr( r( r) r� � s �zCFReactor.crashc C s t | j� dS )zJ Immediately stop the CFRunLoop (which must be running!). N)r r2 r: r( r( r) r� � s zCFReactor._stopNowr c C s | � || j� | �� dS )z� Emulate the behavior of C{iterate()} for things that want to call it, by letting the loop run for a little while and then scheduling a timed call to exit it. N)r� r� r� )r& �delayr( r( r) �iterate� s zCFReactor.iterate�NN)F)r )r* r+ r, r- r3 r; rV ra r[ rc r9 rj rk rl ru r{ r} r~ r� r� r� r% r� r� r� r� r� r( r( r( r) r R s2 & B; ! c C s$ t | |d�}ddlm} ||� |S )a6 Configure the twisted mainloop to be run inside CFRunLoop. @param runLoop: the run loop to use. @param runner: the function to call in order to actually invoke the main loop. This will default to C{CFRunLoopRun} if not specified. However, this is not an appropriate choice for GUI applications, as you need to run NSApplicationMain (or something like it). For example, to run the Twisted mainloop in a PyObjC application, your C{main.py} should look something like this:: from PyObjCTools import AppHelper from twisted.internet.cfreactor import install install(runner=AppHelper.runEventLoop) # initialize your application reactor.run() @return: The installed reactor. @rtype: C{CFReactor} )r4 r5 r )�installReactor)r �twisted.internet.mainr� )r4 r5 r$ r� r( r( r) r � s r� )+r- �__all__rB �zope.interfacer � CFNetworkr r r r r r r r r r r �CoreFoundationr r r r r r r r r r r �twisted.internet.interfacesr �twisted.internet.posixbaser r r �twisted.pythonr r> r? rY r" r r r( r( r( r) �<module> s$ 44 |