OwlCyberSecurity - MANAGER
Edit File: amppush.pyc
� �t�ac @ s� d d l m Z d d l m Z d d l m Z d d l m Z m Z d d l m Z d d l m Z m Z d d l m Z d d l Z d d l Z e � Z d Z d e j f d � � YZ d e j f d � � YZ d e j f d � � YZ d e f d � � YZ d e f d � � YZ d e j f d � � YZ d e f d � � YZ d e f d � � YZ d e j f d � � YZ d e f d � � YZ d e j f d � � YZ! d e f d! � � YZ" e d d" � � Z$ d S(# i����( t PushPriority( t PushScheduler( t Logger( t inlineCallbackst returnValue( t TCP4ClientEndpoint( t Factoryt ServerFactory( t ampNt pusht SubscribeToIDc B sA e Z d e j � f d e j � f g Z d e j � f g Z RS( t tokent idt status( t __name__t __module__R t Stringt argumentst response( ( ( s. /opt/cpanel-ccs/calendarserver/push/amppush.pyR % s $t UnsubscribeFromIDc B s2 e Z d e j � f g Z d e j � f g Z RS( R R ( R R R R R R ( ( ( s. /opt/cpanel-ccs/calendarserver/push/amppush.pyR * s t NotificationForIDc B s\ e Z d e j � f d e j d e � f d e j d e � f g Z d e j � f g Z RS( R t dataChangedTimestampt optionalt priorityR ( R R R R t Integert TrueR R ( ( ( s. /opt/cpanel-ccs/calendarserver/push/amppush.pyR 1 s t AMPPushForwardingFactoryc B s# e Z e � Z d � Z d � Z RS( c C s | | _ d S( N( t forwarder( t selfR ( ( s. /opt/cpanel-ccs/calendarserver/push/amppush.pyt __init__= s c C s# t j � } | j j j | � | S( N( R t AMPR t protocolst append( R t addrt protocol( ( s. /opt/cpanel-ccs/calendarserver/push/amppush.pyt buildProtocol@ s ( R R R t logR R# ( ( ( s. /opt/cpanel-ccs/calendarserver/push/amppush.pyR : s t AMPPushForwarderc B s8 e Z d Z e � Z d � Z e d e j d � � Z RS( sJ Runs in the slaves, forwards notifications to the master via AMP c C s# g | _ | j t t | � � d S( N( R t addFactoryt PUSH_ROUTER ( R t controlSocket( ( s. /opt/cpanel-ccs/calendarserver/push/amppush.pyR L s c c s\ | d k r! t t j � � } n x4 | j D]) } | j t d | d | d | j �Vq+ Wd S( NR R R ( t Nonet intt timeR t callRemoteR t value( R t transactionR R R R" ( ( s. /opt/cpanel-ccs/calendarserver/push/amppush.pyt enqueueP s N( R R t __doc__R R$ R R R) R t highR/ ( ( ( s. /opt/cpanel-ccs/calendarserver/push/amppush.pyR% F s t AMPPushMasterListeningProtocolc B s> e Z d Z e � Z d � Z e j d e j j d � � Z RS( sF Listens for notifications coming in over AMP from the slaves c C s t t | � j � | | _ d S( N( t superR2 R t master( R R4 ( ( s. /opt/cpanel-ccs/calendarserver/push/amppush.pyR d s c C sT | d k r! t t j � � } n | j j d | d | d t j | � �i d d 6S( NR R t OKR ( R) R* R+ R4 R/ R t lookupByValue( R R R R ( ( s. /opt/cpanel-ccs/calendarserver/push/amppush.pyt enqueueFromWorkerh s N( R R R0 R R$ R R t responderR) R R1 R- R7 ( ( ( s. /opt/cpanel-ccs/calendarserver/push/amppush.pyR2 ^ s t AMPPushMasterListenerFactoryc B s# e Z e � Z d � Z d � Z RS( c C s | | _ d S( N( R4 ( R R4 ( ( s. /opt/cpanel-ccs/calendarserver/push/amppush.pyR x s c C s t | j � } | S( N( R2 R4 ( R R! R" ( ( s. /opt/cpanel-ccs/calendarserver/push/amppush.pyR# { s ( R R R R$ R R# ( ( ( s. /opt/cpanel-ccs/calendarserver/push/amppush.pyR9 u s t AMPPushMasterc B se e Z d Z e � Z d d � Z d � Z d � Z d e j d � Z e d � � Z e d � � Z RS( sr AMPPushNotifierService allows clients to use AMP to subscribe to, and receive, change notifications. c C s� | d k r d d l m } n d d l m } | r` | t | � t | � d | �j | � n | d k r� | j t t | � � n g | _ | r� t | | j d | �| _ n d | _ d S( Ni����( t reactor( t serviceR; t staggerSeconds( R) t twisted.internetR; t twisted.application.strportsR< t strt AMPPushNotifierFactoryt setServiceParentR& R' R9 t subscribersR t sendNotificationt scheduler( R R( t parentServicet portt enableStaggeringR= R; t strPortsService( ( s. /opt/cpanel-ccs/calendarserver/push/amppush.pyR � s c C s$ | j j d � | j j | � d S( Ns Added subscriber( R$ t debugRC R ( R t p( ( s. /opt/cpanel-ccs/calendarserver/push/amppush.pyt addSubscriber� s c C s$ | j j d � | j j | � d S( Ns Removed subscriber( R$ RJ RC t remove( R RK ( ( s. /opt/cpanel-ccs/calendarserver/push/amppush.pyt removeSubscriber� s c C s� | d k r! t t j � � } n g } x< | j D]1 } | j | � } | d k r1 | j | � q1 q1 W| r� | j | | | | � Sd S( s� Sends an AMP push notification to any clients subscribing to this pushKey. @param pushKey: The identifier of the resource that was updated, including a prefix indicating whether this is CalDAV or CardDAV related. "/CalDAV/abc/def/" @type pushKey: C{str} @param dataChangedTimestamp: Timestamp (epoch seconds) for the data change which triggered this notification (Only used for unit tests) @type key: C{int} N( R) R* R+ RC t subscribedToIDR t scheduleNotifications( R R. t pushKeyR R t tokenst subscriberR ( ( s. /opt/cpanel-ccs/calendarserver/push/amppush.pyR/ � s c c sA x: | j D]/ } | j | � r | j | | | | � Vq q Wd S( N( RC RO t notify( R R R R R RS ( ( s. /opt/cpanel-ccs/calendarserver/push/amppush.pyRD � s c c sW | j d k r+ | j j | | | | � n( x% | D] } | j | | | | � Vq2 Wd S( N( RE R) t scheduleRD ( R RR R R R R ( ( s. /opt/cpanel-ccs/calendarserver/push/amppush.pyRP � s N( R R R0 R R$ R) R RL RN R R1 R/ R RD RP ( ( ( s. /opt/cpanel-ccs/calendarserver/push/amppush.pyR: � s t AMPPushNotifierProtocolc B sd e Z e � Z d � Z d � Z e j e � d � Z e j e � d � Z d � Z d d � Z RS( c C s2 t t | � j � | | _ i | _ d | _ d S( N( R3 RV R R< t subscriptionsR) t any( R R< ( ( s. /opt/cpanel-ccs/calendarserver/push/amppush.pyR � s c C s0 | d k r | | _ n | | j | <i d d 6S( NRX R5 R ( RX RW ( R R R ( ( s. /opt/cpanel-ccs/calendarserver/push/amppush.pyt subscribe� s c C s- y | j | =Wn t k r! n Xi d d 6S( NR5 R ( RW t KeyError( R R ( ( s. /opt/cpanel-ccs/calendarserver/push/amppush.pyt unsubscribe� s c C sW | j | � | k rS | j j d d | d | �| j t d | d | d | j �Sd S( Ns( Sending notification for {id} to {token}R R R R ( RO R$ RJ R, R R- ( R R R R R ( ( s. /opt/cpanel-ccs/calendarserver/push/amppush.pyRT � s c C s) | j d k r | j S| j j | d � S( N( RX R) RW t get( R R ( ( s. /opt/cpanel-ccs/calendarserver/push/amppush.pyRO s c C s | j j | � d S( N( R< RN ( R t reason( ( s. /opt/cpanel-ccs/calendarserver/push/amppush.pyt connectionLost s N( R R R R$ R RY R R8 R[ R RT RO R) R^ ( ( ( s. /opt/cpanel-ccs/calendarserver/push/amppush.pyRV � s RA c B s) e Z e � Z e Z d � Z d � Z RS( c C s | | _ d S( N( R< ( R R< ( ( s. /opt/cpanel-ccs/calendarserver/push/amppush.pyR s c C s2 | j | j � } | j j | � | j | _ | S( N( R" R< RL ( R R! RK ( ( s. /opt/cpanel-ccs/calendarserver/push/amppush.pyR# s ( R R R R$ RV R" R R# ( ( ( s. /opt/cpanel-ccs/calendarserver/push/amppush.pyRA s t AMPPushClientProtocolc B s3 e Z d Z d � Z e d � � Z e j e � RS( s� Implements the client side of the AMP push protocol. Whenever the NotificationForID Command arrives, the registered callback will be called with the id. c C s t t | � j � | | _ d S( N( R3 R_ R t callback( R R` ( ( s. /opt/cpanel-ccs/calendarserver/push/amppush.pyR " s c c s2 | j | | t j | � � Vt i d d 6� d S( NR5 R ( R` R R6 R ( R R R R ( ( s. /opt/cpanel-ccs/calendarserver/push/amppush.pyt notificationForID& s ( R R R0 R R Ra R R8 ( ( ( s. /opt/cpanel-ccs/calendarserver/push/amppush.pyR_ s t AMPPushClientFactoryc B s) e Z e � Z e Z d � Z d � Z RS( c C s | | _ d S( N( R` ( R R` ( ( s. /opt/cpanel-ccs/calendarserver/push/amppush.pyR 3 s c C s | j | j � } | S( N( R" R` ( R R! RK ( ( s. /opt/cpanel-ccs/calendarserver/push/amppush.pyR# 6 s ( R R R R$ R_ R" R R# ( ( ( s. /opt/cpanel-ccs/calendarserver/push/amppush.pyRb . s c c s� | d k r d d l m } n t t j � � } t | | | � } t | � } | j | � V} x( | D] } | j t d | d | �Vqf Wt | � d S( s Clients can call this helper method to register a callback which will get called whenever a push notification is fired for any id in the ids list. @param host: AMP host name to connect to @type host: string @param port: AMP port to connect to @type port: integer @param ids: The push IDs to subscribe to @type ids: list of strings @param callback: The method to call whenever a notification is received. @type callback: callable which is passed an id (string) i����( R; R R N( R) R> R; R@ t uuidt uuid4R Rb t connectR, R R ( t hostRG t idsR` R; R t endpointt factoryR" R ( ( s. /opt/cpanel-ccs/calendarserver/push/amppush.pyt subscribeToIDs= s (% t calendarserver.push.ipushR t calendarserver.push.utilR t twext.python.logR t twisted.internet.deferR R t twisted.internet.endpointsR t twisted.internet.protocolR R t twisted.protocolsR R+ Rc R$ R' t CommandR R R R t objectR% R R2 R9 R: RV RA R_ Rb R) Rj ( ( ( s. /opt/cpanel-ccs/calendarserver/push/amppush.pyt <module> s0 `*