o zqf )ã@sÌddlZddlZddlZddlZddlZddlZe e¡ZGdd„dƒZ Gdd„de ƒZ Gdd„de ƒZ Gdd „d e ƒZ Gd d „d e ƒZ Gd d „d e ƒZGdd„de ƒZGdd„de ƒZGdd„de ƒZdS)éNc@óeZdZdd„Zdd„ZdS)Ú BasePlugincCs ||_dS©N)Úsource©ÚselfÚsrc©r úW/root/parts/websockify/install/lib/python3.10/site-packages/websockify/token_plugins.pyÚ__init__ s zBasePlugin.__init__cCsdSrr ©rÚtokenr r r ÚlookupszBasePlugin.lookupN)Ú__name__Ú __module__Ú __qualname__r rr r r r r s rcs,eZdZ‡fdd„Zdd„Zdd„Z‡ZS)ÚReadOnlyTokenFilecstƒj|i|¤Žd|_dSr)Úsuperr Ú_targets©rÚargsÚkwargs©Ú __class__r r r s zReadOnlyTokenFile.__init__c sÎtj ˆj¡r‡fdd„t ˆj¡Dƒ}nˆjg}iˆ_d}|D]D}dd„t|ƒ ¡DƒD]6}|r_| d¡s_zt   d|¡\}}|  ¡  dd¡ˆj|<Wnt y^t dˆj|f¡Ynw|d7}q-q dS) Ncsg|] }tj ˆj|¡‘qSr )ÚosÚpathÚjoinr)Ú.0Úf©rr r Ú sÿz3ReadOnlyTokenFile._load_targets..écSsg|]}| ¡‘qSr )Ústrip)rÚlr r r r %sú#z:\sú:zSyntax error in %s on line %d)rrÚisdirrÚlistdirrÚopenÚ readlinesÚ startswithÚreÚsplitr"ÚrsplitÚ ValueErrorÚloggerÚerror)rÚ cfg_filesÚindexrÚlineÚtokÚtargetr rr Ú _load_targetss&  ÿ ÿ ùÿzReadOnlyTokenFile._load_targetscCs*|jdur | ¡||jvr|j|SdSr)rr6r r r r r.s   zReadOnlyTokenFile.lookup)rrrr r6rÚ __classcell__r r rr rs rcseZdZ‡fdd„Z‡ZS)Ú TokenFilecs| ¡tƒ |¡Sr)r6rrr rr r r?s zTokenFile.lookup)rrrrr7r r rr r8;sr8có$eZdZ‡fdd„Zdd„Z‡ZS)Ú TokenFileNamecs$tƒ |¡tj |¡stdƒ‚dS)Nz)TokenFileName plugin requires a directory)rr rrr&Ú Exceptionrrr r r Hs  ÿzTokenFileName.__init__cCsBtj |¡}tj |j|¡}tj |¡rt|ƒ ¡ ¡  d¡SdS)Nr%) rrÚbasenamerrÚexistsr(Úreadr"r,)rr rr r r rMs  zTokenFileName.lookup©rrrr rr7r r rr r:Ds r:c@r)Ú BaseTokenAPIcCs$|j d¡\}}| dd¡}||gS)Nr%ÚasciiÚignore)Útextr,Úencode)rÚrespÚhostÚportr r r Úprocess_result]s zBaseTokenAPI.process_resultcCs,ddl}| |j|¡}|jr| |¡SdS)Nr)ÚrequestsÚgetrÚokrH)rr rIrEr r r rbs  zBaseTokenAPI.lookupN)rrrrHrr r r r r@Vs r@c@óeZdZdd„ZdS)Ú JSONTokenApicCs| ¡}|d|dfS)NrFrG)Újson)rrEÚ resp_jsonr r r rHqszJSONTokenApi.process_resultN)rrrrHr r r r rMmó rMc@rL)Ú JWTTokenApic Csâzáddlm}m}ddl}| ¡}zt|jdƒ }| ¡}Wdƒn1s'wYWntyJ}zt   dt |ƒ¡WYd}~WdSd}~wwz|  |¡Wnz |j | d¡ddWn t   d¡YYWdSYzS|j||d }| |j¡} d | vr‹|j||jd }| |j¡} d | vr¦t ¡| d kr¦t  d ¡WWdSd | vr»t ¡| d kr»t  d¡WWdS| d| dfWWStyá}zt   dt |ƒ¡WYd}~WdSd}~wwtyðt   d¡YdSw)Nr)ÚjwtÚjwkÚrbzError loading key file: %súutf-8Úoct)ÚkÚktyz#Failed to correctly parse key data!)ÚkeyrRÚencÚnbfzToken can not be used yet!ÚexpzToken has expired!rFrGzFailed to parse token: %szGpackage jwcrypto not found, are you sure you've installed it correctly?)ÚjwcryptorRrSrNÚJWKr(rr>r;r/r0ÚstrÚimport_from_pemÚ import_keyÚdecodeÚJWTÚloadsÚheaderÚclaimsÚtimeÚwarningÚ ImportError) rr rRrSrNrYÚkey_fileÚkey_dataÚeÚ parsed_headerÚparsedr r r rzs^ ÿ€€þ  ý    €þ  þzJWTTokenApi.lookupN)rrrrr r r r rQvrPrQc@s eZdZdZdd„Zdd„ZdS)Ú TokenRedisažToken plugin based on the Redis in-memory data store. The token source is in the format: host[:port[:db[:password]]] where port, db and password are optional. If port or db are left empty they will take its default value, ie. 6379 and 0 respectively. If your redis server is using the default port (6379) then you can use: my-redis-host In case you need to authenticate with the redis server and you are using the default database and port you can use: my-redis-host:::verysecretpass In the more general case you will use: my-redis-host:6380:1:verysecretpass The TokenRedis plugin expects the format of the target in one of these two formats: - JSON {"host": "target-host:target-port"} - Plain text target-host:target-port Prepare data with: redis-cli set my-token '{"host": "127.0.0.1:5000"}' Verify with: redis-cli --raw get my-token Spawn a test "server" using netcat nc -l 5000 -v Note: This Token Plugin depends on the 'redis' module, so you have to install it before using this plugin: pip install redis cCslzddl}Wntyt d¡t ¡Ynwd|_d|_d|_z~|  d¡}t |ƒdkr4|d|_ nSt |ƒdkrG|\|_ |_|jsFd|_n@t |ƒdkrb|\|_ |_|_|js[d|_|jsad|_n%t |ƒdkr…|\|_ |_|_|_|jsxd|_|js~d|_|js„d|_nt ‚t |jƒ|_t |jƒ|_t d |j |jf¡WdSt yµt d |¡t ¡YdSw) NrzUnable to load redis moduleiër%r!éééz%TokenRedis backend initilized (%s:%s)zbThe provided --token-source='%s' is not in the expected format [:[:[:]]])Úredisrir/r0ÚsysÚexitÚ_portÚ_dbÚ _passwordr,ÚlenÚ_serverr.ÚintÚinfo)rrrsÚfieldsr r r r ãs\    þ     € € €    ÿ þüzTokenRedis.__init__c CsBzddl}Wntyt d¡t ¡Ynwt d|¡|j|j|j |j |j d}|  |¡}|dur8dS|  d¡ ¡}t d|¡| d¡r}zt |¡}|d d ¡\}}Wn8tylt d |¡YdSty|t d |¡YdSwt d |¡r‹| d ¡\}}n t d |¡dSt d||f¡||gS)NrzFpackage redis not found, are you sure you've installed them correctly?zresolving token '%s')rFrGÚdbÚpasswordrUzresponse from redis : %sÚ{rFr%zUnable to decode JSON token: %sz+Unable to find 'host' key in JSON token: %sz\S+:\S+zUnable to parse token: %szhost: %s, port: %s)rsrir/r0rtrur|ÚRedisrzrvrwrxrJrbr"Údebugr*rNrdr,r.ÚKeyErrorr+Úmatch) rr rsÚclientÚstuffÚ responseStrÚcomborFrGr r r rsH    þ ÿ    ÿ ÿý zTokenRedis.lookupN)rrrÚ__doc__r rr r r r ro°s2 ,rocr9)ÚUnixDomainSocketDirectorycs&tƒj|i|¤Žtj |j¡|_dSr)rr rrÚabspathrÚ _dir_pathrrr r r 5sz"UnixDomainSocketDirectory.__init__c Cs´z=ddl}tj |j¡sWdStj tj |j|¡¡}| |j¡s$WdStj |¡s-WdS|  t |¡j ¡s9WdSd|gWSt yY}zt   dt|ƒ¡WYd}~dSd}~ww)NrÚ unix_socketz$Error finding unix domain socket: %s)Ústatrrr&rŒr‹rr*r=ÚS_ISSOCKÚst_moder;r/r0r_)rr rŽÚuds_pathrlr r r r9s"   €þz UnixDomainSocketDirectory.lookupr?r r rr rŠ4s rŠ)Úloggingrrtrgr+rNÚ getLoggerrr/rrr8r:r@rMrQrorŠr r r r Ús" (  :