crabapples
/
websockify
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Use logging instead of directly printing messages to sys.stderr

This commit is contained in:
Javier Cacheiro 2022-04-13 16:12:31 +02:00
parent e4cff3746d
commit 62ac4aeb03
1 changed files with 27 additions and 23 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

40
websockify/token_plugins.py
View File

@ -1,8 +1,12 @@
import logging
import os import os
import sys import sys
import time import time
import re import re
logger = logging.getLogger(__name__)
class BasePlugin(): class BasePlugin():
def __init__(self, src): def __init__(self, src):
self.source = src self.source = src
@ -35,7 +39,7 @@ class ReadOnlyTokenFile(BasePlugin):
tok, target = re.split(':\s', line) tok, target = re.split(':\s', line)
self._targets[tok] = target.strip().rsplit(':', 1) self._targets[tok] = target.strip().rsplit(':', 1)
except ValueError: except ValueError:
print("Syntax error in %s on line %d" % (self.source, index), file=sys.stderr) logger.error("Syntax error in %s on line %d" % (self.source, index))
index += 1 index += 1
def lookup(self, token): def lookup(self, token):
@ -108,7 +112,7 @@ class JWTTokenApi(BasePlugin):
with open(self.source, 'rb') as key_file: with open(self.source, 'rb') as key_file:
key_data = key_file.read() key_data = key_file.read()
except Exception as e: except Exception as e:
print("Error loading key file: %s" % str(e), file=sys.stderr) logger.error("Error loading key file: %s" % str(e))
return None return None
try: try:
@ -117,7 +121,7 @@ class JWTTokenApi(BasePlugin):
try: try:
key.import_key(k=key_data.decode('utf-8'),kty='oct') key.import_key(k=key_data.decode('utf-8'),kty='oct')
except: except:
print('Failed to correctly parse key data!', file=sys.stderr) logger.error('Failed to correctly parse key data!')
return None return None
try: try:
@ -133,21 +137,21 @@ class JWTTokenApi(BasePlugin):
if 'nbf' in parsed: if 'nbf' in parsed:
# Not Before is present, so we need to check it # Not Before is present, so we need to check it
if time.time() < parsed['nbf']: if time.time() < parsed['nbf']:
print('Token can not be used yet!', file=sys.stderr) logger.warning('Token can not be used yet!')
return None return None
if 'exp' in parsed: if 'exp' in parsed:
# Expiration time is present, so we need to check it # Expiration time is present, so we need to check it
if time.time() > parsed['exp']: if time.time() > parsed['exp']:
print('Token has expired!', file=sys.stderr) logger.warning('Token has expired!')
return None return None
return (parsed['host'], parsed['port']) return (parsed['host'], parsed['port'])
except Exception as e: except Exception as e:
print("Failed to parse token: %s" % str(e), file=sys.stderr) logger.error("Failed to parse token: %s" % str(e))
return None return None
except ImportError as e: except ImportError:
print("package jwcrypto not found, are you sure you've installed it correctly?", file=sys.stderr) logger.error("package jwcrypto not found, are you sure you've installed it correctly?")
return None return None
class TokenRedis(): class TokenRedis():
@ -172,14 +176,14 @@ class TokenRedis():
import redis import redis
import simplejson import simplejson
self._server, self._port = src.split(":") self._server, self._port = src.split(":")
print("TokenRedis backend initilized (%s:%s)" % logger.info("TokenRedis backend initilized (%s:%s)" %
(self._server, self._port), file=sys.stderr) (self._server, self._port))
except ValueError: except ValueError:
print("The provided --token-source='%s' is not in an expected format <host>:<port>" % logger.error("The provided --token-source='%s' is not in an expected format <host>:<port>" %
src, file=sys.stderr) src)
sys.exit() sys.exit()
except ImportError: except ImportError:
print("package redis or simplejson not found, are you sure you've installed them correctly?", file=sys.stderr) logger.error("package redis or simplejson not found, are you sure you've installed them correctly?")
sys.exit() sys.exit()
def lookup(self, token): def lookup(self, token):
@ -187,20 +191,20 @@ class TokenRedis():
import redis import redis
import simplejson import simplejson
except ImportError: except ImportError:
print("package redis or simplejson not found, are you sure you've installed them correctly?", file=sys.stderr) logger.error("package redis or simplejson not found, are you sure you've installed them correctly?")
sys.exit() sys.exit()
print("resolving token '%s'" % token, file=sys.stderr) logger.info("resolving token '%s'" % token)
client = redis.Redis(host=self._server, port=self._port) client = redis.Redis(host=self._server, port=self._port)
stuff = client.get(token) stuff = client.get(token)
if stuff is None: if stuff is None:
return None return None
else: else:
responseStr = stuff.decode("utf-8") responseStr = stuff.decode("utf-8")
print("response from redis : %s" % responseStr, file=sys.stderr) logger.debug("response from redis : %s" % responseStr)
combo = simplejson.loads(responseStr) combo = simplejson.loads(responseStr)
(host, port) = combo["host"].split(':') (host, port) = combo["host"].split(':')
print("host: %s, port: %s" % (host,port), file=sys.stderr) logger.debug("host: %s, port: %s" % (host,port))
return [host, port] return [host, port]
@ -228,5 +232,5 @@ class UnixDomainSocketDirectory(BasePlugin):
return [ 'unix_socket', uds_path ] return [ 'unix_socket', uds_path ]
except Exception as e: except Exception as e:
print("Error finding unix domain socket: %s" % str(e), file=sys.stderr) logger.error("Error finding unix domain socket: %s" % str(e))
return None return None