File: //usr/share/rhn/osad/jabber_lib.pyc
�
D�)Xc @ s� d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l m Z m
Z
d d l Z d d l m
Z
y* d d l m Z d d l m Z m Z Wn7 e k
r� d d l m Z d d l m Z m Z n Xd d l m Z d d l m Z d d l Z z# e j d d e �d d l Z Wd e j � Xd
Z d e Z d e Z! d
e Z" d e Z# d e Z$ d e Z% d e Z& d e Z' d Z( d Z) d f d � � YZ* d e
j
j+ f d � � YZ, d � Z- d � Z. d e j/ f d � � YZ0 d e j1 e0 f d � � YZ2 d e j3 e0 f d � � YZ4 d! e j5 e0 f d" � � YZ6 d# f d$ � � YZ7 d% � Z8 d& e9 f d' � � YZ: d( e9 f d) � � YZ; d* e9 f d+ � � YZ< d, e; f d- � � YZ= d. e9 f d/ � � YZ> d0 e> f d1 � � YZ? d2 e> f d3 � � YZ@ d4 e9 f d5 � � YZA d6 e; f d7 � � YZB d8 e jC eD f d9 � � YZE d: e
jF f d; � � YZF d< e
j
j+ f d= � � YZG d> d? � ZH d@ � ZI dA f dB � � YZJ dC � ZK dD � ZL dE f dF � � YZM d S(G i����N( t OptionParsert Option( t SSL( t StringIO( t log_debugt log_error( t
raise_with_tb( t bstrt ignoret categorys# http://jabber.rhn.redhat.com/jabbers %s/signeds %s/presence/subscribes %s/presence/subscribeds %s/presence/unsubscribes %s/message/request/checkins %s/message/response/checkins %s/message/request/pings %s/message/response/pings urn:ietf:params:xml:ns:xmpp-tlss# urn:ietf:params:xml:ns:xmpp-stanzast Runnerc B s� e Z e Z e Z d Z d Z d Z d � Z
d � Z d � Z d � Z
d � Z d � Z d � Z d � Z d
� Z d � Z d � Z d
� Z d � Z d d � Z d � Z d � Z RS( i< iZ c C s� | j d d d d d d �| j d d d d d d
�| j d d d d d
�| j d d d d d �g | _ d | _ d | _ g | _ d | _ d | _ d | _ d | _ d | _
d | _ d S( Ns -vs --verboset actiont countt helps Increase verbositys -Ns
--nodetacht
store_trues4 Suppress backgrounding and detachment of the processs
--pid-filet stores Write to this PID files --logfiles" Write log information to this filei ( t optiont
options_tablet Nonet ssl_certt debug_levelt _jabber_serverst _connected_jabber_servert _usernamet _passwordt _resourcet _in_backgroundt
_use_proxy( t self( ( s! /usr/share/rhn/osad/jabber_lib.pyt __init__F s" c C s7 | j d | j � | _ | j j � \ | _ | _ d S( s Process command line optionst option_listN( t
option_parserR t _parsert
parse_argst optionst args( R ( ( s! /usr/share/rhn/osad/jabber_lib.pyt process_cli_options\ s c C s� | j � d } d } x�| d k r. d } n d } yu | j � } | ro t d d � | j | d � d } n
| j | � | j d | � } | j | � | j | � Wq t k
r� y | j � Wn n Xt
j d � q t k
r� � q t
k
r=t
j � d } t d | � | j � s | j � q q q t k
rst
j � d } t d d � d } q q t k
r�t j | j | j � } t d d | � | j � s�| j � n y t j | � Wq�t k
r�t
j d � q�Xq t k
r$t
j � d } t d
� � q t d � t t � � t j | j | j � } t d d | d
� | j � s�| j � n y t j | � Wq�t k
r�t
j d � q�Xq Xq d S( s� Method that starts up everything
- processes command line options
- big loop to reconnect if necessary
- read config
- setup config
- setup jabber connection
- process requests
i i i s ###Forcing setupt no_forks Restart requestedi s Need Restarts8 Unable to connect to jabber servers, sleeping %s secondss Invalid Cert Error:s
Error caught:t Sleepingt secondsN( R$ R t read_configR t setup_configt setup_connectiont fix_connectiont process_forevert KeyboardInterruptt
disconnectt syst exitt
SystemExitt RestartRequestedt exc_infoR t is_in_backgroundt push_to_backgroundt NeedRestartt JabberConnectionErrort randomt randintt
_min_sleept
_max_sleept timet sleept InvalidCertErrort extract_traceback( R t force_setupR% t configt ct et
time_to_sleep( ( s! /usr/share/rhn/osad/jabber_lib.pyt maina sz
c C s | S( s> After setting up the connection, do whatever else is necessary( ( R t client( ( s! /usr/share/rhn/osad/jabber_lib.pyR+ � s c C s | S( N( ( R RF ( ( s! /usr/share/rhn/osad/jabber_lib.pyt preprocess_once� s c C s� t d � | j | � xg y0 | j | � t j d d � } t j | � Wq t k
rt | j � t j
d � q � q Xq d S( s% Big loop to process requests
i i i
i N( R RG t process_onceR8 R9 R< R= R- R. R/ R0 ( R RF RD ( ( s! /usr/share/rhn/osad/jabber_lib.pyR, � s
c C s
t � d S( s" To be overridden in a client classN( t NotImplementedError( R RF ( ( s! /usr/share/rhn/osad/jabber_lib.pyRH � s c C s d S( N( ( R RA ( ( s! /usr/share/rhn/osad/jabber_lib.pyR) � s c C s i S( N( ( R ( ( s! /usr/share/rhn/osad/jabber_lib.pyR( � s c C s | j S( N( R ( R ( ( s! /usr/share/rhn/osad/jabber_lib.pyR4 � s c C s | j j s t � d | _ n | j j } | ry t j | � Wn6 t k
rz t j � d } | j
d k r{ � q{ n Xyv t j | t j t j
Bt j Bt d d � � } t j | t d t j � � � t j | t d � � t j | � Wqt k
rqXn d S( Ni i t 0644i s %ds
( R" t nodetachR5 R t pid_filet ost unlinkt OSErrorR/ R3 t errnot opent O_WRONLYt O_APPENDt O_CREATt intt writeR t getpidt close( R RL RC t fd( ( s! /usr/share/rhn/osad/jabber_lib.pyR5 � s$
/
c C s
t | � S( N( t
check_cert( R t cert( ( s! /usr/share/rhn/osad/jabber_lib.pyRZ � s c C s! t d | � t d d | � d S( Ni s" Could not connect to jabber server( R ( R t jsRC ( ( s! /usr/share/rhn/osad/jabber_lib.pyt
print_message� s
i c C s� xX| j D]'} t d d | � y- | j | � } t d d | � | | _ PWq
t k
rm t j d � q
t j k
r� t j � d } | j
| d � t t � � q
q
t
k
r� t j � d } | j
| d � t t � � q
q
t k
r0t j � d } | j
| d � t t � � q
q
Xq
Wt d d � | sU| j � n t � | sn| j � n | j d | j d
| j d | j � | j j | j d � | j j | j d
� | j j | j d � | S( s
- initializes a Jabber connection (by instantiating a Jabber client)
- if necessary, pushes to background
- authentication and resource binding (by calling start())
Possible causes for this function to return None:
- jabber server is not started
- jabber server is started but did not initialize SSL just yet
This function will kill the process with exit code 1 if the SSL
handshake failed (an indication of a mismatching CA cert). We do this
so starting the program as a daemon to fail if this happens. Of
course, if the server is down and the CA cert is bad, then the daemon
will start but will silently fail afterwards; the error log should
have a traceback though.
i s
Connecting toi s Connected to jabber servers socket errort JabberErrort SSLErrors& Could not connect to any jabber servert usernamet passwordt resourcet presencet messaget error( R R t _get_jabber_clientR t SSLHandshakeErrorR/ R0 t socketRe R3 R] R R? R^ R_ R5 R7 t startR R R t custom_handlert register_callbackt _presence_callbackt _message_callbackt _error_callback( R R% R\ RB RC ( ( s! /usr/share/rhn/osad/jabber_lib.pyR* � sH
c C sT t d � | j d d � } | d } | j � } d } | j r� d | k r� i | d j d � d d 6t | d j d � d � d 6} | d r� | d | d
<| d | d <q� n t | � d
k rt | d � } t d
d | | � | j | | d | �} n% t d
d | � | j | d | �} | j | _ | j | j
� | j � | S( s] Returns a connected Jabber client, or raises an exception if it was
unable to connecti t :i i t proxy_urlt hostt portt enable_proxy_autht
proxy_usert usert proxy_passwordRa i s
Connecting tot proxyN( R t splitR( R R RU t lent client_factoryR t add_trusted_certR t connect( R t
jabber_servert arrt cft jabberpy_proxy_dictt jabber_portRB ( ( s! /usr/share/rhn/osad/jabber_lib.pyRf 8 s*
!
c C sJ t d | � x6 | j D]+ } | j � d k r t d � t � q q Wd S( s2 Logs error stanza messages for diagnostic purposess Received an error stanza: t conflicts6 Received an conflict. Restarting with new credentials.N( R t kidst getNameR6 ( R RF t stanzat kid( ( s! /usr/share/rhn/osad/jabber_lib.pyRn U s
N( t __name__t
__module__R R R R R Rz R: R; R R$ RE R+ RG R, RH R) R( R4 R5 RZ R] R* Rf Rn ( ( ( s! /usr/share/rhn/osad/jabber_lib.pyR
; s* N D R> c B s e Z d � Z e Z RS( c C s d j | j � S( Nt ( t joinR# ( R ( ( s! /usr/share/rhn/osad/jabber_lib.pyt __str__^ s ( R� R� R� t __repr__( ( ( s! /usr/share/rhn/osad/jabber_lib.pyR> ] s c C s� | d k r t d � � n y t | � j � } Wn1 t k
rd t t d | � t j � d � n Xy t j j
t j j | � } Wn7 t j j k
r� t t d | � t j � d � n Xt
d d | j � � | j � r� t d | � � n d S( Ns& Cannot pass None as a certificate paths Unable to read filei s Unable to open certificatei s Loading certs Expired certificate( R R> RQ t readt IOErrorR R/ R3 R t cryptot load_certificatet FILETYPE_PEMt ErrorR t get_subjectt has_expired( t cert_pathR[ t x509( ( s! /usr/share/rhn/osad/jabber_lib.pyRZ b s
$ $ c G s[ t j d t | � � j � } x6 | D]. } t j d t d | | f � � j � } q% W| S( Nt sha1s %s%s( t hashlibt newR t hexdigest( t
secret_keyt valuest ht v( ( s! /usr/share/rhn/osad/jabber_lib.pyt signq s
,t JabberProtocolNodec B s e Z d � Z RS( c C s | j j | d � S( N( t attrst getR ( R t key( ( s! /usr/share/rhn/osad/jabber_lib.pyt getAttry s ( R� R� R� ( ( ( s! /usr/share/rhn/osad/jabber_lib.pyR� x s t JabberIqNodec B s e Z e j Z RS( ( R� R� R� R� ( ( ( s! /usr/share/rhn/osad/jabber_lib.pyR� | s t JabberMessageNodec B s e Z e j Z RS( ( R� R� R� R� ( ( ( s! /usr/share/rhn/osad/jabber_lib.pyR� s t JabberPresenceNodec B s e Z e j Z RS( ( R� R� R� R� ( ( ( s! /usr/share/rhn/osad/jabber_lib.pyR� � s t Handlersc B se e Z d � Z d � Z d � Z d � Z d d d d d � Z d � Z d � Z d � Z
d � Z RS(
c C s t d � i | _ d S( Ni ( R t _handlers( R ( ( s! /usr/share/rhn/osad/jabber_lib.pyR � s
c C su t d | � | j � | j | � } | s@ t d d | � d Sx. | D]& } t d d | | � | | | � qG Wd S( Ni i s Unhandled stanzai s Calling callback( R t cleanup_expired_callbackst _get_callbacks( R RF R� t callbackst callback( ( s! /usr/share/rhn/osad/jabber_lib.pyt dispatch� s
c C s t d | � | j � } | | j k r, g S| j � } | j � } i } | j | \ } } } } | d k r� | r� | j | | f g � }
| j |
| � n | d k r� | j | g � }
| j |
| � n | r� | j | g � }
| j |
| � n | j | | � t | j � � S( Ni (
R R� R� t getIDt getNamespaceR R� t _get_callbacks_from_listt listt keys( R R� t stanza_namet stanza_idt stanza_nst resultt h_idnst h_idt h_nst l_deft cbs( ( s! /usr/share/rhn/osad/jabber_lib.pyR� � s&
c C s� x� | D]� } | d \ } } } | d k s8 | d k rW d | | <| d k rW q qW n | d } | d k r� | j | � q n | d | d <q Wd S( Ni i i i ( R t remove( R t lt result_hasht entR� t expiryt usage_count( ( s! /usr/share/rhn/osad/jabber_lib.pyR� � s
c C s* t d | | | | | | � | r5 t j � | } n d } | | | g } | j | j | d i i i g f �\ } }
} } | d k r� | r� | j | | | f g � }
|
j | � d S| d k r� | j |
| g � }
|
j | � d S| r| j | | g � }
|
j | � d S| j | � d S( Ni t
default_value( R R< R t _get_from_hashR� t append( R R� R� R� R� t timeoutR� R� t callback_entryR� R� R� R� R� ( ( s! /usr/share/rhn/osad/jabber_lib.pyRk � s* '
c C s+ | | k r | | } n | } | | <| S( N( ( R R� R� R� t val( ( s! /usr/share/rhn/osad/jabber_lib.pyR� � s
c C s� t d � t j � } xi | j j � D]X \ } } | \ } } } } x' | | | f D] } | j | | � qT W| j | | � q& Wd S( Ni ( R R< R� t itemst _expire_callbacks_hasht _expire_callbacks_list( R t nowR� t valsR� R� R� R� R� ( ( s! /usr/share/rhn/osad/jabber_lib.pyR� � s
c C s> t d | � x* | j � D] \ } } | j | | � q Wd S( Ni ( R R� R� ( R R� R� R� R� ( ( s! /usr/share/rhn/osad/jabber_lib.pyR� � s
c C s_ t d | | � xH | D]@ } | \ } } } | s8 q n | | k rJ q n | j | � q Wd S( Ni ( R R� ( R R� R� R� R� R� R� ( ( s! /usr/share/rhn/osad/jabber_lib.pyR� � s
N( R� R� R R� R� R� R Rk R� R� R� R� ( ( ( s! /usr/share/rhn/osad/jabber_lib.pyR� � s
c G s d | f GHd S( Ns
Debugging:( ( R# ( ( s! /usr/share/rhn/osad/jabber_lib.pyt my_debug� s R2 c B s e Z RS( ( R� R� ( ( ( s! /usr/share/rhn/osad/jabber_lib.pyR2 s R^ c B s e Z RS( ( R� R� ( ( ( s! /usr/share/rhn/osad/jabber_lib.pyR^ s R6 c B s e Z RS( ( R� R� ( ( ( s! /usr/share/rhn/osad/jabber_lib.pyR6 s t TimeoutErrorc B s e Z RS( ( R� R� ( ( ( s! /usr/share/rhn/osad/jabber_lib.pyR� s R_ c B s e Z d Z RS( s- Raised when a lower-level SSL error is caught( R� R� t __doc__( ( ( s! /usr/share/rhn/osad/jabber_lib.pyR_ s Rg c B s e Z d Z RS( s$ Raised when the SSL handshake failed( R� R� R� ( ( ( s! /usr/share/rhn/osad/jabber_lib.pyRg s t SSLDisabledErrorc B s e Z d Z RS( s) Raised if the server does not support SSL( R� R� R� ( ( ( s! /usr/share/rhn/osad/jabber_lib.pyR� s R7 c B s e Z d Z RS( s6 Raised when we were unable to make a jabber connection( R� R� R� ( ( ( s! /usr/share/rhn/osad/jabber_lib.pyR7 s t JabberQualifiedErrorc B s e Z d � Z d � Z e Z RS( c G s& | | _ | | _ t j | | � d S( N( t errcodet errR^ R ( R R� R� R# ( ( s! /usr/share/rhn/osad/jabber_lib.pyR s c C s& d | j j t | � | j | j f S( Ns' <%s instance at %s; errcode=%s; err=%s>( t __class__R� t idR� R� ( R ( ( s! /usr/share/rhn/osad/jabber_lib.pyR� $ s ( R� R� R R� R� ( ( ( s! /usr/share/rhn/osad/jabber_lib.pyR� s t JabberClientc B s� e Z d Z e j j Z d � Z d � Z d � Z d � Z d � Z
d � Z d � Z d � Z
d � Z d
� Z d* d � Z d � Z d
� Z d � Z d d � Z d � Z d � Z d* d* d* d � Z d � Z d � Z d � Z d* d � Z d � Z e j d � Z e j d � Z d � Z! d � Z" d � Z# d � Z$ d � Z% d � Z& d* d! � Z' d" � Z( d# � Z) d$ � Z* d% � Z+ d& � Z, d' � Z- d* d( � Z. d) � Z/ RS(+ i c O sU t d � t j j | | | � d | _ g | _ d | _ g | _ | j d t
� | j d t � | j d t � | j d t
� | j d t
� | j d | j d t �| j d | j d t j d t �t � } | | _ | j d | j � | j d | j � | j d | j � | j d | j � d | _ t � | _ t d
� | _ d S( Ni i t unknownt iqRd Rc Re t systemR� i ( R t jabbert ClientR R t jidt _incoming_node_queueR t
trusted_certst registerProtocolR� R� R� R� t registerHandlert _expectedIqHandlert Truet _IqRegisterResultt NS_REGISTERR� Rj R� t
_non_ssl_sockt Rostert _rostert generate_random_stringt _uniq_client_string( R R# t kwargsR� ( ( s! /usr/share/rhn/osad/jabber_lib.pyR . s.
c C s t | � | j j | � d S( N( RZ R� R� ( R t trusted_cert( ( s! /usr/share/rhn/osad/jabber_lib.pyR{ N s
c C s t d � | j s" t d � � n | j | _ t d d � x� t d d � D]� } y t j j | � WnA t j
k
r� t j � d } t
d | � t j
| � � n Xt d d � | j � } t d d
| � | j � d k r t d d � | j � t j d � qK PqK Wt
d
� t � | j d � } t d d | � | d k rht
d � | j � t � n | j d t � | j � } t d d | � | j � d k r�t
d � | j � t � n t d d � t | j d | j �} t j j | _ | j | _ | j � y | j � | j | � WnE t j j! k
rpt
d � t
t" � � t# t$ t j � d � n Xt j% j& j | � | j | _' | j( | _ | j) � | j* | j+ � � | j, � | j � } | j � d k r�| j � t- d � � n | j. | _ t d d � d S( Ni s No trusted certs addedi s Attempting to connecti i i sl Error connecting to jabber server: %s. See https://access.redhat.com/solutions/327903 for more information. t Connecteds Expecting features stanza, got:t featuress9 Server did not return a <features /> stanza, reconnectings^ Not able to reconnect - See https://access.redhat.com/solutions/45332 for possible solutions.
t starttlss
starttls nodesE Server does not support TLS - <starttls /> not in <features /> stanzas <starttls xmlns='%s' />s Expecting proceed stanza, got:t proceeds3 Server broke TLS negotiation - <proceed /> not senti s Preparing for TLS handshakeR� s Traceback caught:s! Server did not pass any features?s connect returning(/ R R� t SSLVerifyErrort _auth_dispatchR� t rangeR� R� R| Rh Re R/ R3 R t get_one_stanzaR� R. R< R= R� t getTagR RV t NS_STARTTLSt SSLSockett _sockR t TLSv1_METHODt _ssl_methodt ssl_verify_callbackt init_sslt do_handshaket verify_peerR� R? R Rg t xmlstreamt StreamR� t _connectiont _setupCommst sendt _header_stringt processt Exceptiont _orig_dispatch( R t retryRC R� t
starttls_nodet ssl( ( s! /usr/share/rhn/osad/jabber_lib.pyR| R sx
c C s2 y t j j | � Wn t j j k
r- n Xd S( N( R� R� R. R R� ( R ( ( s! /usr/share/rhn/osad/jabber_lib.pyR. � s c C sX | j j | _ t | j d � r3 | j j | _ n t | j � j | _ | j | _ d S( Nt sendall( R� t recvt _readt hasattrR t _writet Sendallt _reader( R ( ( s! /usr/share/rhn/osad/jabber_lib.pyR � s
c C sR t d d | | | � | sN t d � | j d � | j � | j j � | S| S( Ni t Calleds# SSL certificate verification faileds </stream:stream>( R R RV RX R� ( R t connR[ t errnumt deptht ok( ( s! /usr/share/rhn/osad/jabber_lib.pyR� � s
c C s� | j � } | d k r' t d � � n | j � } t | d � sQ t d � � n | j } | d d k rw | d } n | j } | d d k r� | d } n | | k r� t j | | � r� t d | j | f � � n d S( Ns Unable to retrieve peer certt CNs Certificate has no Common Namei����t .s( Mismatch: peer name: %s; common name: %s( t get_peer_certificateR R� R� R R t _hostt fnmatcht fnmatchcase( R R
R[ t subjectt common_namet hdot( ( s! /usr/share/rhn/osad/jabber_lib.pyR� � s
c C s | j j | j d � d | j � } t d d � } | j t j � | j | � | j | � d } x. | j � } | j d � } | | k rh Pqh qh xh | j
j � j � D]Q \ } } d | k r� | d d k r� | j | d d �q� | j | d d �q� Wd S(
s} Request the roster. Will register the roster callback,
but the call will wait for the roster to be properly populatedR� s
iq-request-%st typeR� R� t askt subscribet
subscribedN( Rj Rk t _roster_callbackt
get_unique_idR� t setQueryR� t NS_ROSTERt setIDR R R� R� R� t get_subscribed_fromR� t
send_presence( R t
iq_node_idt iq_nodeR� t node_idt kR� ( ( s! /usr/share/rhn/osad/jabber_lib.pyt retrieve_roster� s
" c C s� t d d | � | j d � } | d k s@ | j � t j k rQ t d d � d S| j d � } | d k r} t d d � d Sx' | j d
� D] } | j j | � q� Wd S( Ni s Updating the rostert queryi s Query node not found, skippingR R� t sets Not a result or a set, skippingt item( s results set(
R R� R R� R� R' R� t getTagsR� t add_item( R RF R� t qnodet node_typet node( ( s! /usr/share/rhn/osad/jabber_lib.pyR$ s !
c C s� | s
d St d � } | j t j � xA | D]9 } i | d 6d d 6} t d d | �} | j | � q- Wt d d � } d | j � } | j | � | j | � | j | � d S(
NR0 R� R� t subscriptionR2 R� R R1 s remove-%s( R� t setNamespaceR� R' t
insertNodeR� R% R( R ( R t jidsR5 R� R� t inodeR7 t remove_iq_id( ( s! /usr/share/rhn/osad/jabber_lib.pyt cancel_subscription s
c C s� | r t j � } n xp | j s� | rn t j � } | | | k r] t d d | | | � d S| | | } n d } | j d | � q W| j d } | j d =| S( s. Returns one stanza (or None if timeout is set)i s timed outR� i N( R< R� R R R ( R R� Ri R� t tmR7 ( ( s! /usr/share/rhn/osad/jabber_lib.pyR� 1 s
c C sB | j � } | | j k r$ d } n | j | t d | � } | S( sn Builds one stanza according to the handlers we have registered via
registerHandler or registerProtocolR� R7 ( R� t handlersR ( R R� t name( ( s! /usr/share/rhn/osad/jabber_lib.pyt
_build_stanzaG s
c C s� t d | � | j d k rU t j j d � t j j t | � � t j j d � n | j | � } t j j | | � | j
j | � d S( Ni i s <-- s
( R R R/ t stderrRV t strRB R� R� R� R� R� ( R R� ( ( s! /usr/share/rhn/osad/jabber_lib.pyR Q s
c C sx t d | � | j d k rU t j j d � t j j t | � � t j j d � n | j | � } | j j | � d S( Ni i s <-- s
( R R R/ RC RV RD RB R� R� ( R R� ( ( s! /usr/share/rhn/osad/jabber_lib.pyR� _ s
i c C sG t d | | | | � d | j � } t j d d � } | j | � | j t j � } | j d � j | � | j | � t d d | � y | j
| d d �} Wnu t k
rt j
� d
} | s� � n | j d k rt d d � | j | | � | j | | | d
d �S� n Xt d d | � | j d � }
d | j � } t j d d � } | j | � | j t j � } | j d � j | � | j d � j | � |
j d � r\|
j d � j � }
|
j d � j � } t j d t j d | � j � |
� j � } x2 t t | � � D] } t j d | � j � } q!W| j d � j | � nf |
j d � r�| j d � } | j t j d t | j � | � � j � � n | j d � j | � t d d | � y | j | � WnP t k
r5t j
� d
} | j d k r/t d d � | j | | � S� n Xt d d � t S( s� Try to authenticate the username with the specified password
If the authentication fails, try to register the user.
If that fails as well, then JabberQualifiedError is raised
i s auth-get-%sR R� R` i s Sending auth requestR� i< i t 401s Need to registert registeri s
Auth responseR0 s auth-set-%sR1 Rb t tokent sequenceR� t hasht digestRa s Sending auth infot
Authenticated( R R% R� t IqR( R& t NS_AUTHt insertTagt
insertDataR t waitForResponseR� R/ R3 R� RF t authR� t getDataR� R� R� R� RU R t
getIncomingIDt SendAndWaitForResponseR� ( R R` Ra Rb RF t
auth_iq_idt auth_get_iqt qt
auth_responseRC t auth_ret_queryt auth_set_idt auth_set_iqRG t seqR� t iRJ ( ( s! /usr/share/rhn/osad/jabber_lib.pyRQ j sd
. #
c C s[ | j d k rH t j j d � t j j t | � � t j j d � n t j j | | � S( Ni s --> s
( R R/ RC RV RD R� R� R ( R R� ( ( s! /usr/share/rhn/osad/jabber_lib.pyR � s
c C sb | j j � } t d d | � | j j � } t d d | � | j j � } t d d | � | j j � } t d d | � x�| D]�} | j | � } t | � } | | k r� t d d | � q� n | | k r� t d d � q� n | | k r-| | } d | k r-| d d k r-t d d � q� q-n | | k ru| | } d | k ru| d d k rut d d
� q� qun | | j j k r�d } | j j }
n d } | j j }
i | d
6| d 6d d 6|
| <t d | � | j | � } t
d | d d � } | j d | j � � | j
| t � }
|
r=| j |
� n t d d | � | j | � q� Wd S( s� Subscribe to these nodes' presence
The subscription in jabber works like this:
Contact 1 State Contact 1 State Contact 2 Contact 2
----------+-------------------+------------------+----------
subscribe -> [ none + ask ]
[ from ] <- subscribed
[ to ]
[ from + ask ] <- subscribe
subscribed -> [ both ]
[ both ]
----------+-------------------+------------------+----------
Enclosed in square brackets is the state when the communication took
place.
i s
Subscribed tos Subscribed boths Subscribed nones Subscribed froms* Already subscribed to the presence of nodeR! R" s Subscribed none + ask=subscribes Subscribed from + ask=subscribet fromt noneR� R8 t toR s presence-%si s% Sending presence subscription requestN( R� t get_subscribed_toR t get_subscribed_botht get_subscribed_noneR) t _strip_resourceRD t _subscribed_fromt _subscribed_noneR� R( R% t _create_signaturet NS_RHN_PRESENCE_SUBSCRIBER: R ( R R; t
subscribed_tot subscribed_botht subscribed_nonet subscribed_fromt full_jidR� R� R8 t hashdt stripped_jidt
presence_nodet sig( ( s! /usr/share/rhn/osad/jabber_lib.pyt subscribe_to_presence� sZ
c C st t d | | � | d k r( t � } n t d | � } | rM | j | � n | rc | j | � n | j | � d S( Ni R` ( R R R� t setTypeR( R ( R R� R t xidR7 ( ( s! /usr/share/rhn/osad/jabber_lib.pyR* s c C s
| j j � S( N( R t fileno( R ( ( s! /usr/share/rhn/osad/jabber_lib.pyRu s c C s� d } x] t j | j � g g g d � \ } } } | s= Pn | j | j � } | sY Pn | | } q | s| | j | � n | S( Nt i ( t selectRu R
t
BLOCK_SIZEt disconnected( R t receivedt rfdst wfdst exfdst buff( ( s! /usr/share/rhn/osad/jabber_lib.pyR� s *
c C s d S( N( ( R ( ( s! /usr/share/rhn/osad/jabber_lib.pyt process_loop_hook( s c C s@ t d | � g | _ | j � } t j � } x t j � } | rh | | | k rW d S| | | } n d } | j � t d d | � t j | g g g | � \ } } } t d d � | s� d St | j d � r�t d d | j � y | j
| j � } Wng t j j k
rht
j � d }
t d d � | j j � t t d
t |
� � t
j � d � n Xt d d t | � � | s�t d
� � n | j j | � | j j � } | r| j
| � } | j j | � qn7 | j
| j � } | st d
� � n | j j | � | j s/| r+q1 n d St | j � Sd S( Ni i i s before select(); timeouts select() returnedt pendings Reading %s bytes from ssl socketi s Closing sockets OpenSSL error; will retryi s
Read %s bytess
Premature EOF( R R� Ru R< R R Rw R R� Rx R
R t SysCallErrorR/ R3 R� RX R R_ RD Ry R^ R t ParseR� ( R R� Ru Ri R� R? R{ R| R} t dataRC R� ( ( s! /usr/share/rhn/osad/jabber_lib.pyR + sX
$
*
c C s� t d | | � | j � | j � } d | k rE | j d | � n d | k rd | j d | � n y | j � WnW t k
r� t j � d } | j d k r� t d � | j
� t j d � n � n Xt S( Ni R` Ra i t 409s Invalid passwordi (
R t requestRegInfot
getRegInfot
setRegInfot sendRegInfoR� R/ R3 R� R R. R0 R� ( R R` Ra t dRC ( ( s! /usr/share/rhn/osad/jabber_lib.pyRF n s"
c C sd t d | | � t j | � } d | _ d | _ | d k rm t j � | } | j d | | f t j � n | j d | t j � x� | d k r� d } n | t j � } | d k r� Pn t d d � | j | � } t d d � | s| d k s� t
� Pn | j � } | r/| j � | _ | | _ d St j | j
� � } | | k r� | Sq� d | _ d S( Ni Rv i s waiting with timeout:%s for %ss waiting for %ss before get_one_stanzas after get_one_stanzat Timeout( R R� t ustrt lastErrt lastErrCodeR R< t DEBUGt DBG_NODE_IQR� t AssertionErrort getErrorCodet getErrorR� ( R t IDR� t
abort_timet tmoutR� t
error_codet tid( ( s! /usr/share/rhn/osad/jabber_lib.pyt _waitForResponse� s>
c C sw | j | d | �} | d k r% | S| j d k r@ t � � n | j ra t | j | j � � n t d | j � � d S( NR� R� s
Unknown error( R� R R� R� R� R� R^ ( R R� R� R� ( ( s! /usr/share/rhn/osad/jabber_lib.pyRP � s c C s' | j } | d t _ d | j | f S( Ni s %s-%s( t _seqR� R� ( R R\ ( ( s! /usr/share/rhn/osad/jabber_lib.pyR% � s
c C s d S( N( ( R R ( ( s! /usr/share/rhn/osad/jabber_lib.pyt disconnectHandler� s c C s� | j d t j � d } i | j d 6| j d 6d d 6d d 6} | j rZ | j | d <n g } x1 | j � D]# \ } } | j d
| | f � qm Wd j | � } | | S( Ns6 jabber_lib.JabberClient.header: sending initial headers8 <?xml version='1.0' encoding='UTF-8'?><stream:stream %s>R` t xmlnss http://etherx.jabber.org/streamss xmlns:streams 1.0t versionR� s %s='%s'R� ( R� R� t DBG_INITR t
_namespacet _outgoingIDR� R� R� ( R t templR� t additionR. R� ( ( s! /usr/share/rhn/osad/jabber_lib.pyR � s
c C s- | j � } | j | � | j t j � d S( N( R R R R� R� ( R t header( ( s! /usr/share/rhn/osad/jabber_lib.pyR� � s
c C s | S( N( ( R R� ( ( s! /usr/share/rhn/osad/jabber_lib.pyt _fix_jid� s c C sP | j � } | j � } t d | j | | � | j � } | j � d k sR t � | d k sj | d k r� t d d | | � | j | � | j | g � d S| d k r� t d d | | � | j
| � d S| d
k rt d d | � | j | d d d
| �| j | g � d S| d k rLt d d | � | j t
d | � � n d S( s�
If the roster is enabled, presence stanzas with type="subscribed"
should never be received - the server will initiate a roster push
instead
i Rc R# i s Node is availableNt unsubscribedt unavailables Node is unavailableR" s Subscription request approvedR Rt t probes Presence probeR` ( R� R� ( t getFromt getTypeR R� R� R� R� R t set_jid_availableRr t set_jid_unavailableR* R R� ( R RF R� R� t
presence_typeR� ( ( s! /usr/share/rhn/osad/jabber_lib.pyRl � s,
c C s d S( Ni ( ( R R� t actions( ( s! /usr/share/rhn/osad/jabber_lib.pyt _check_signature s c C s
t | � S( N( t strip_resource( R R� ( ( s! /usr/share/rhn/osad/jabber_lib.pyRd s c C s d S( N( R ( R R� R ( ( s! /usr/share/rhn/osad/jabber_lib.pyRg s c C sN t d | d d � } | j | | � } | r= | j | � n | j | � d S( NR` R t normal( R� Rg R: R ( R R� R R7 Rq ( ( s! /usr/share/rhn/osad/jabber_lib.pyt send_message s
c C s | j j | � S( N( R� t
jid_available( R R� ( ( s! /usr/share/rhn/osad/jabber_lib.pyR� % s c C s | j j | � S( N( R� t
set_available( R R� ( ( s! /usr/share/rhn/osad/jabber_lib.pyR� ( s c C s | j j | � S( N( R� t set_unavailable( R R� ( ( s! /usr/share/rhn/osad/jabber_lib.pyR� + s c C sR | j | � } | s g S| d k r) | Sg | D] } | j � | k r0 | ^ q0 S( s[ Get the matching (child) tags of this stanza, possibly with the
specified namespaceN( R3 R R� ( R R� t tag_namet namespacet tagst x( ( s! /usr/share/rhn/osad/jabber_lib.pyt match_stanza_tags. s c C s t d | � | j � d k s% t � | j � } | j � } | d k rd t d d | | f � d S| d k r� t d d | � d S| j | d d t j �} | r� t d d
� d S| j | d | �} | s| j
d k r� t d � � n t d d
� d S| S( Ni Rd Re i s Received error from %s: %sR� s# Unsupported message type %s ignoredR� R� s Ignoring delayed stanzaR� i s Mismatching signatures( R R� R� R� R� R R� R� t NS_DELAYR� R R ( R R� R� t message_fromt message_typet x_delayed_nodesRq ( ( s! /usr/share/rhn/osad/jabber_lib.pyt _check_signature_from_message9 s,
N(0 R� R� R� R� R� Rx R R{ R| R. R R� R� R/ R$ R> R R� RB R R� RQ R Rr R* Ru R� R R RF R� R� RP R% R� R R� R� Rl R� Rd Rg R� R� R� R� R� R� ( ( ( s! /usr/share/rhn/osad/jabber_lib.pyR� * sT g
C Q C 3
/ R� c B s e Z RS( ( R� R� ( ( ( s! /usr/share/rhn/osad/jabber_lib.pyR� W s R� c B s e Z RS( ( R� R� ( ( ( s! /usr/share/rhn/osad/jabber_lib.pyR� Z s i c C s� | s
d Sd } t | � } t j d � } | j t d t j � � � | j t d t j � � � t d d � } g } d } xR | j | j | � � | j
� } | j | � | t | � } | | k r� Pq� q� | j
� d j | � | j � S( NRv i R� s %.8fs %ss /dev/urandomt rbi ( RU R� R� t updateR R< RM RW RQ R� R� R� Ry RX R� t lower( t lengtht random_bytest st devrandomR� t
cur_lengtht buf( ( s! /usr/share/rhn/osad/jabber_lib.pyR� ] s&
c C s t d d � t j � } | d k r5 t j d � n t j � t j d � t j d � x$ t j t j f D] } | j
� ql Wt d d � } t d d � } t d d d � } t j | j
� t j j
� � t j | j
� t j j
� � t j | j
� t j j
� � d S( Ni s Pushing process into backgroundi t /s /dev/nullt rs ab+( R RM t forkt _exitt setpgrpt chdirt umaskR/ t stdoutRC t flushRQ t dup2Ru t stdin( t pidt ft sit sot se( ( s! /usr/share/rhn/osad/jabber_lib.pyR5 u s
R� c B s� e Z d � Z d � Z d � Z d � Z d � Z d � Z d � Z d � Z d � Z
d � Z d
� Z d � Z
d � Z d
� Z RS( c C s1 i | _ i | _ i | _ i | _ i | _ d S( N( t _subscribed_toRe t _subscribed_bothRf t _available_nodes( R ( ( s! /usr/share/rhn/osad/jabber_lib.pyR � s
c C s� | j d � } | j d � } t | � } t | � } i | d 6| d 6} | j d � } | rl | | d <n d d d d g } | | k r� xS | D]H } t | d | � } | | k r� | | | <q� | | k r� | | =q� q� Wn d S( NR8 R� R! R` R^ t bothR_ t _subscribed_( R� R� RD t getattr( R R2 t subscrR� t entryR! R� t aR� ( ( s! /usr/share/rhn/osad/jabber_lib.pyR4 � s$
c C s
| j j � S( N( Re t copy( R ( ( s! /usr/share/rhn/osad/jabber_lib.pyR) � s c C s
| j j � S( N( R� R� ( R ( ( s! /usr/share/rhn/osad/jabber_lib.pyRa � s c C s
| j j � S( N( R� R� ( R ( ( s! /usr/share/rhn/osad/jabber_lib.pyRb � s c C s
| j j � S( N( Rf R� ( R ( ( s! /usr/share/rhn/osad/jabber_lib.pyRc � s c C s# | j j � } | j | j � | S( N( R� R� R� R� ( R t ret( ( s! /usr/share/rhn/osad/jabber_lib.pyt get_subscribed_to_jids� s c C s# | j j � } | j | j � | S( N( Re R� R� R� ( R R� ( ( s! /usr/share/rhn/osad/jabber_lib.pyt get_subscribed_from_jids� s c C s
| j j � S( N( R� R� ( R ( ( s! /usr/share/rhn/osad/jabber_lib.pyt get_available_nodes� s c C s t | � } d | j | <d S( Ni ( RD R� ( R R� ( ( s! /usr/share/rhn/osad/jabber_lib.pyR� � s c C s, t | � } | | j k r( | j | =n d S( N( RD R� ( R R� ( ( s! /usr/share/rhn/osad/jabber_lib.pyR� � s c C s
| | j k S( N( R� ( R R� ( ( s! /usr/share/rhn/osad/jabber_lib.pyR� � s c C s8 | j j � | j j � | j j � | j j � d S( N( R� t clearRe R� Rf ( R ( ( s! /usr/share/rhn/osad/jabber_lib.pyR� � s
c C s8 d | j j � | j j � | j j � | j j � f S( Ns- Roster:
to: %s
from: %s
both: %s
none: %s( R� R� Re R� Rf ( R ( ( s! /usr/share/rhn/osad/jabber_lib.pyR� � s
( R� R� R R4 R) Ra Rb Rc R� R� R� R� R� R� R� R� ( ( ( s! /usr/share/rhn/osad/jabber_lib.pyR� � s c C s. t | t j � s$ t j | � } n | j � S( N( t
isinstanceR� t JIDt getStripped( R� ( ( s! /usr/share/rhn/osad/jabber_lib.pyR� � s c C s# t � } t j d | � | j � S( N( R t tracebackt print_excR t getvalue( t sio( ( s! /usr/share/rhn/osad/jabber_lib.pyR? � s R c B s# e Z d Z d � Z d d � Z RS( s_ This class exists here because python 1.5.2 does not support a
sendall() method for socketsc C s
| | _ d S( N( t sock( R R� ( ( s! /usr/share/rhn/osad/jabber_lib.pyR s i c C s] t | � } | s d Sd } x: | j j | | | � } | | | k rL Pn | | } q | S( Ni ( Ry R� R ( R R� t flagst to_sendt
bytes_sentR� ( ( s! /usr/share/rhn/osad/jabber_lib.pyR s
( R� R� R� R R ( ( ( s! /usr/share/rhn/osad/jabber_lib.pyR � s (N RM R� R/ R< Rw Rh R8 R t optparseR R R� t rhnR t ioR t osad.rhn_logR R t ImportErrort cStringIOt rhn_logt spacewalk.common.usixR t rhn.i18nR t warningst filterwarningst DeprecationWarningR� t
resetwarningst NS_RHNt
NS_RHN_SIGNEDRh t NS_RHN_PRESENCE_SUBSCRIBEDt NS_RHN_PRESENCE_UNSUBSCRIBEt NS_RHN_MESSAGE_REQUEST_CHECKINt NS_RHN_MESSAGE_RESPONSE_CHECKINt NS_RHN_MESSAGE_REQUEST_PINGt NS_RHN_MESSAGE_RESPONSE_PINGR� t
NS_STANZASR
R� R> RZ R� t ProtocolR� RL R� t MessageR� t PresenceR� R� R� R R2 R^ R6 R� R_ Rg R� R7 R� R� t objectR� R� R� R� R5 R� R� R? R ( ( ( s! /usr/share/rhn/osad/jabber_lib.pyt <module> s�
� # z � � � 0 *T