mirror of
https://github.com/firestar5683/StarPilot.git
synced 2026-11-08 05:23:41 +08:00
firestar5683
39 lines
1.2 KiB
Python
39 lines
1.2 KiB
Python
"""Authentication for the existing statistics service."""
|
|
import os
|
|
from pathlib import Path
|
|
import stat
|
|
|
|
|
|
CLIENT_TOKEN = 'StarPilot'
|
|
TOKEN_FILE = Path('/data/starpilot/analytics/token')
|
|
|
|
|
|
def _token(value):
|
|
if not isinstance(value, str):
|
|
return None
|
|
value = value.strip()
|
|
return value if 16 <= len(value) <= 4096 and all(33 <= ord(c) <= 126 for c in value) else None
|
|
|
|
|
|
def read_token():
|
|
value = os.environ.get('STARPILOT_STATS_TOKEN')
|
|
if value is not None:
|
|
return _token(value)
|
|
explicit_path = os.environ.get('STARPILOT_STATS_TOKEN_FILE')
|
|
path = Path(explicit_path) if explicit_path is not None else TOKEN_FILE
|
|
descriptor = None
|
|
try:
|
|
descriptor = os.open(path, os.O_RDONLY | os.O_NOFOLLOW | os.O_NONBLOCK)
|
|
info = os.fstat(descriptor)
|
|
if (not stat.S_ISREG(info.st_mode) or info.st_mode & 0o077 or
|
|
info.st_uid not in (0, os.getuid()) or not 16 <= info.st_size <= 4097):
|
|
return None
|
|
return _token(os.read(descriptor, 4097).decode('ascii'))
|
|
except FileNotFoundError:
|
|
return CLIENT_TOKEN if explicit_path is None else None
|
|
except (OSError, UnicodeError):
|
|
return None
|
|
finally:
|
|
if descriptor is not None:
|
|
os.close(descriptor)
|