g7ddlZddlZddlZddlZddlmZgdZejdejZ ejdejZ ejdZ ejjdZejjdZGdd ejj"ZGd d eZGd d eZGddeZGddeZGddeZGddeZddZy)N) GNTPRegister GNTPNotice GNTPSubscribeGNTPOK GNTPError parse_gntpzGNTP/(?P\d+\.\d+) (?PREGISTER|NOTIFY|SUBSCRIBE|\-OK|\-ERROR) (?P[A-Z0-9]+(:(?P[A-F0-9]+))?) ?((?P[A-Z0-9]+):(?P[A-F0-9]+).(?P[A-F0-9]+))? zRGNTP/(?P\d+\.\d+) (?PREGISTER|NOTIFY|SUBSCRIBE|\-OK|\-ERROR)z ([\w-]+):(.+) z: ceZdZdZddZdZy) _GNTPBufferzGNTP Buffer classNc|r.|jtjj||jtyN)writegntpshimbGNTP_EOL)selfvalues /opt/Tautulli/lib/gntp/core.pywritelnz_GNTPBuffer.writeln)s* ::diikk% !**XcFt|ts t|}|jtjj ||jt |jtjj ||jtyr ) isinstancestrrrrrGNTP_SEPrrkeyrs r writeheaderz_GNTPBuffer.writeheader.s_ E3  u:5**TYY[[ **X**TYY[[  **Xrr )__name__ __module__ __qualname____doc__rrrrr r 's rr ceZdZdZddZej jrdZndZdZ ddZ dZ d Z d Z d Zd Zd ZdZdZddZdZy) _GNTPBasezBase initilization :param string messagetype: GNTP Message type :param string version: GNTP Protocol version :param string encription: Encryption protocol Nc|||d|_tjtjtjtj d|_i|_i|_y)N)version messagetypeencryptionAlgorithmID)MD5SHA1SHA256SHA512) infohashlibmd5sha1sha256sha512 hash_algoheaders resources)rr(r' encryptions r__init__z_GNTPBase.__init__>sK&$) ++ << ^^ ^^ $. $,$.rc"|jSr )encoders r__str__z_GNTPBase.__str__Qs ++-rc\tjj|jSr )rrur:r;s rr<z_GNTPBase.__str__Ts ))++dkkm $$rctj|}|stjd|j }|ddk(rd|d<|S)zParse the first line of a GNTP message to get security and other info values :param string data: GNTP Message :return dict: Parsed GNTP Info line ERROR_PARSING_INFO_LINEr)NONEN)GNTP_INFO_LINEmatcherrors ParseError groupdict)rdatarCr.s r _parse_infoz_GNTPBase._parse_infoWsS   t $%    4 55  $ !"f,#'4 +rc>|sd|jd<d|jd<ytjj||_|j |_|j |jvr"tjd|j z|jj|j }|jd}tjjd}||j}||j}||z}||j}||j} |j |jd<| j |jd<|j |jd<y) zSet a password for a GNTP Message :param string password: Null to clear password :param string encryptAlgo: Supports MD5, SHA1, SHA256, SHA512 Nr)keyHashAlgorithmzINVALID HASH "%s"utf8keyHashAlgorithmIDkeyHashsalt)r.rrrpasswordupper encryptAlgor4rDUnsupportedErrorgetr:timectime hexdigestdigest) rrOrQ hashfunctionseedrNsaltHashkeyBasisrrMs r set_passwordz_GNTPBase.set_passwordisM (,499 $%#'499   ))++h'$- &&($  T^^ +  !4t7G7G!G HH##D$4$45, __V $(   V $$ d  % % '$ $  & & ((  (X%%'#   ' ' )'$($4$4$)) ! $))Ijjl$))Frcd}tdt|dD]"}t|||dzd}|t|z }$|S)zHelper function to decode hex string to `proper` hex string :param string value: Human readable hex string :return string: Hex string r)rangelenintchr)rrresultitmps r _decode_hexz_GNTPBase._decode_hexsP & CJ "a U1QU^R 3 SX6 -rc|dz }t|d}|jj|t|z}||z}|j||}t||k(s$t j d|dt||S)N LengthzINVALID_DATA_LENGTH Expected: z Recieved )rcrawfindrbrDrE)r rawIdentifier identifier dataLength pointerStart pointerEndrGs r_decode_binaryz_GNTPBase._decode_binarys:-:h'(*}-M0BB,j(* ,z *$ Tj   ZY\]aYbc dd +rc||_|tjd|jj dd}| |jy|tjd|jtjd|jj dd}|jj d}|j |jd }||z}|j||j|_ |j||jj}|j|jdjk(stjd y) z-Validate GNTP Message against stored passwordNzMissing passwordrMTzInvalid keyHashrLr*rKrNz Invalid Hash) rOrD AuthErrorr.rSr:rhr4rWrrVrP)rrOrMrLrZr[s r_validate_passwordz_GNTPBase._validate_passwords>$-    , -- IIMM)T *' _.  _   + ,, ]]   , --yy}}%9%@ ]] ! !& )(   dii/ 0(  ( /T^^. / 9 @ @ B$( .DNN- .txx 8 B B D' DIIi0668 8   . )) rc|jD]6}|jj|dr tjd|zy)zVerify required headersFMissing Notification Header: N)_requiredHeadersr5rSrDrE)rheaders rvalidatez_GNTPBase.validatesD%%Ff ,,  65 )   ;fD EEFrc "d|jjdd|jjd}|jjddr>|d|jjdd|jjdz }n|d z }|jjd drY|d|jjd d|jjd d |jjd z }|S)z9Generate info line for GNTP Message :return string: zGNTP/r' r(r)N:ivValuez NONErLrM.rN)r.rS)rr.s r _format_infoz_GNTPBase._format_infos 99==99== $ YY]]*D1IIMM)*IIMM) 4 7?4 YY]]'.IIMM&'IIMM)IIMM& 4 +rci}|jdD]]}tj|}|s|jdj }|jdj }|||<_|S)zHelper function to parse blocks of GNTP headers into a dictionary :param string data: :return dict: Dictionary of parsed GNTP Headers r r_)split GNTP_HEADERrCgroupstrip)rrGdlinerCrvals r _parse_dictz_GNTPBase._parse_dictss  !jj d   T "5  Q   3 Q   3 1S6 (rc"||j|<yr )r5rs r add_headerz_GNTPBase.add_headers$,,srctjj|}tj|j }||j |<d|zS)z9Add binary resource :param string data: Binary Data zx-growl-resource://%s)rrrr/r0rVr6)rrGros r add_resourcez_GNTPBase.add_resourcesF T ${{4 **,*#$.. : --rc||_tjj||_|jj d}|j |j|_|j|d|_ y)z-Decode GNTP Message :param string data: rjrN) rOrrr>rlrrHr.rr5)rrGrOpartss rdecodez_GNTPBase.decodes] $- YY[[ $( ((.. $%txx($)!!%(+$,rct}|j|j|jj D]\}}|j |||j|j j D]s\}}|j d||j dt||j|j||j|ju|jS)zlEncode a generic GNTP Message :return string: GNTP Message ready to be sent. Returned as a byte string Identifierrk) r rrr5itemsrr6rbrgetvalue)rbuffkvresourcerGs rr:z_GNTPBase.encodes $,,t  "#ll  "daAq,,.,,.nhL(+Hc$i(<<>::d<<><<>  r)Nz1.0N)r*r )rr r!r"r8rrPY2r<rHr\rhrsrvr{rrrrrr:r#rrr%r%7s` $ IIMM%$#> 4F 4 ". ,rr%c@eZdZdZddgZdgZd dZdZdZd d Z d Z y) rzRepresents a GNTP Registration Command :param string data: (Optional) See decode() :param string password: (Optional) Password to use while encoding/decoding messages Application-NameNotifications-CountNotification-NameNctj|dg|_|r|j||y|j ||j dd|j ddy)NREGISTERrpygntprr)r%r8 notificationsrr\rrrGrOs rr8zGNTPRegister.__init__(sW T:&$ ;;tXX??%x0??(!,rc&|jD]6}|jj|dr tjd|z|j D]=}|j D],}|j|drtjd|z?y)z;Validate required headers and validate notification headersFzMissing Registration Header: rxN)ryr5rSrDrEr_requiredNotificationHeaders)rrznotices rr{zGNTPRegister.validate3s%%Ff ,,  65 )   ;fD EEF""Gf22Gv ::fe $   rlrrHr.rvrr5 enumeraterrSrappendrsr6rrGrOrrfpartrs rrzGNTPRegister.decode=s YY[[ $( ((.. $%txx($)(#!!%(+$,5! 6ga1f  jjlb   T "6 jj$e,f% <'((v6F6N/5DNN6::l+, 6rci}||d<||d<|jj||jdt|jy)zAdd new Notification to Registration message :param string name: Notification Name :param boolean enabled: Enable this notification by default rzNotification-EnabledrN)rrrrb)rnameenabledrs radd_notificationzGNTPRegister.add_notificationUsM & $& #*&  F#//'T-?-?)@Arct}|j|j|jj D]\}}|j |||jt |jdkDrK|jD]<}|j D]\}}|j |||j>|jj D]s\}}|j d||j dt ||j|j||j|ju|jS)zuEncode a GNTP Registration Message :return string: Encoded GNTP Registration message. Returned as a byte string rrrk) r rrr5rrrbrr6rr)rrrrrrrGs rr:zGNTPRegister.encodebs2 $,,t  "#ll  "daAq,,.   q ##v 1 aLLN ,,.nhL(+Hc$i(<<>::d<<><<>  rNN)T) rr r!r"ryrr8r{rrr:r#rrrrs= "5 5 -G60 Brrc&eZdZdZgdZddZdZy)raMRepresents a GNTP Notification Command :param string data: (Optional) See decode() :param string app: (Optional) Set Application-Name :param string name: (Optional) Set Notification-Name :param string title: (Optional) Set Notification Title :param string password: (Optional) Password to use while encoding/decoding messages )rrNotification-TitleNctj|d|r|j||y|j||r|j d||r|j d||r|j d|yy)NNOTIFYrrr)r%r8rr\r)rrGapprtitlerOs rr8zGNTPNotice.__init__sl T8$ ;;tXX OO&, OO'. OO(%0 rctjj||_|jj d}|j |j|_|j||j|d|_ t|D]v\}}|dk(r |jdk(r |j|}|jddsD|j|||d<||j|jd<xy)zVDecode existing GNTP Notification message :param string data: Message to decode. rjrr^rFrN)rrr>rlrrHr.rvrr5rrrSrsr6rs rrzGNTPNotice.decodes YY[[ $( ((.. $%txx($)(#!!%(+$,5! 6ga1f  jjlb   T "6 jju%((v6F6N/5DNN6::l+, 6r)NNNNN)rr r!r"ryr8rr#rrrrs 16rrc eZdZdZddgZddZy)rzRepresents a GNTP Subscribe Command :param string data: (Optional) See decode() :param string password: (Optional) Password to use while encoding/decoding messages z Subscriber-IDzSubscriber-NameNc|tj|d|r|j||y|j|y)N SUBSCRIBE)r%r8rr\rs rr8zGNTPSubscribe.__init__s1 T;' ;;tXXrrrr r!r"ryr8r#rrrrs  rrceZdZdZdgZddZy)rzRepresents a GNTP OK Response :param string data: (Optional) See _GNTPResponse.decode() :param string action: (Optional) Set type of action the OK Response is for Response-ActionNctj|d|r|j||r|jd|yy)N-OKrr%r8rr)rrGactions rr8zGNTPOK.__init__s8 T5! ;;t ??$f- rrrr#rrrrs ''.rrc&eZdZdZddgZddZdZy)rzRepresents a GNTP Error response :param string data: (Optional) See _GNTPResponse.decode() :param string errorcode: (Optional) Error code :param string errordesc: (Optional) Error Description Error-CodeError-DescriptionNctj|d|r|j||r%|jd||jd|yy)N-ERRORrrr)rrG errorcode errordescs rr8zGNTPError.__init__sF T8$ ;;t??<+??& 2rcr|jjdd|jjddfS)Nrr)r5rSr;s rerrorzGNTPError.errors5 ,,  < .<<'. 00r)NNN)rr r!r"ryr8rr#rrrrs "#6730rrctjj|}tj |}|st j d|j}|ddk(r t||S|ddk(r t||S|ddk(r t||S|ddk(r t|S|ddk(r t|St j d ) zAttempt to parse a message as a GNTP message :param string data: Message to be parsed :param string password: Optional password to be used to verify the message INVALID_GNTP_INFOr(r)rOrrrrINVALID_GNTP_MESSAGE) rrr>GNTP_INFO_LINE_SHORTrCrDrErFrrrrr)rGrOrCr.s rrrs D  # #D ) -.. :% dX .. =X% D8 ,, =[( th // =U"  =X% 4 /00rr )r/rerT gntp.shimr gntp.errorsrD__all__compile IGNORECASErBrrrrrrStringIOr objectr%rrrrrrr#rrrs  W "rzzU bjj) 99;;v  99;;t  $))$$  bbJe9eP1616hI& .Y . 0 0,1r