24024
af7c11d7f9
- 新增图像生成接口，支持试用、积分和自定义API Key模式 - 实现生成图片结果异步上传至MinIO存储，带重试机制 - 优化积分预扣除和异常退还逻辑，保障用户积分准确 - 添加获取生成历史记录接口，支持时间范围和分页 - 提供本地字典配置接口，支持模型、比例、提示模板和尺寸 - 实现图片批量上传接口，支持S3兼容对象存储 feat(admin): 增加管理员角色管理与权限分配接口 - 实现角色列表查询、角色创建、更新及删除功能 - 增加权限列表查询接口 - 实现用户角色分配接口，便于统一管理用户权限 - 增加系统字典增删查改接口，支持分类过滤和排序 - 权限控制全面覆盖管理接口，保证安全访问 feat(auth): 完善用户登录注册及权限相关接口与页面 - 实现手机号验证码发送及校验功能，保障注册安全 - 支持手机号注册、登录及退出接口，集成日志记录 - 增加修改密码功能，验证原密码后更新 - 提供动态导航菜单接口，基于权限展示不同菜单 - 实现管理界面路由及日志、角色、字典管理页面访问权限控制 - 添加系统日志查询接口，支持关键词和等级筛选 feat(app): 初始化Flask应用并配置蓝图与数据库 - 创建应用程序工厂，加载配置，初始化数据库和Redis客户端 - 注册认证、API及管理员蓝图，整合路由 - 根路由渲染主页模板 - 应用上下文中自动创建数据库表，保证运行环境准备完毕 feat(database): 提供数据库创建与迁移支持脚本 - 新增数据库创建脚本，支持自动检测是否已存在 - 添加数据库表初始化脚本，支持创建和删除所有表 - 实现RBAC权限初始化，包含基础权限和角色创建 - 新增字段手动修复脚本，添加用户API Key和积分字段 - 强制迁移脚本支持清理连接和修复表结构，初始化默认数据及角色分配 feat(config): 新增系统配置参数 - 配置数据库、Redis、Session和MinIO相关参数 - 添加AI接口地址及试用Key配置 - 集成阿里云短信服务配置及开发模式相关参数 feat(extensions): 初始化数据库、Redis和MinIO客户端 - 创建全局SQLAlchemy数据库实例和Redis客户端 - 配置基于boto3的MinIO兼容S3客户端 chore(logs): 添加示例系统日志文件 - 记录用户请求、验证码发送成功与失败的日志信息
575 lines
20 KiB
Python
575 lines
20 KiB
Python
from __future__ import annotations
|
|
|
|
import getpass
|
|
import hashlib
|
|
import json
|
|
import os
|
|
import pkgutil
|
|
import re
|
|
import sys
|
|
import time
|
|
import typing as t
|
|
import uuid
|
|
from contextlib import ExitStack
|
|
from io import BytesIO
|
|
from itertools import chain
|
|
from multiprocessing import Value
|
|
from os.path import basename
|
|
from os.path import join
|
|
from zlib import adler32
|
|
|
|
from .._internal import _log
|
|
from ..exceptions import NotFound
|
|
from ..exceptions import SecurityError
|
|
from ..http import parse_cookie
|
|
from ..sansio.utils import host_is_trusted
|
|
from ..security import gen_salt
|
|
from ..utils import send_file
|
|
from ..wrappers.request import Request
|
|
from ..wrappers.response import Response
|
|
from .console import Console
|
|
from .tbtools import DebugFrameSummary
|
|
from .tbtools import DebugTraceback
|
|
from .tbtools import render_console_html
|
|
|
|
if t.TYPE_CHECKING:
|
|
from _typeshed.wsgi import StartResponse
|
|
from _typeshed.wsgi import WSGIApplication
|
|
from _typeshed.wsgi import WSGIEnvironment
|
|
|
|
# A week
|
|
PIN_TIME = 60 * 60 * 24 * 7
|
|
|
|
|
|
def hash_pin(pin: str) -> str:
|
|
return hashlib.sha1(f"{pin} added salt".encode("utf-8", "replace")).hexdigest()[:12]
|
|
|
|
|
|
_machine_id: str | bytes | None = None
|
|
|
|
|
|
def get_machine_id() -> str | bytes | None:
|
|
global _machine_id
|
|
|
|
if _machine_id is not None:
|
|
return _machine_id
|
|
|
|
def _generate() -> str | bytes | None:
|
|
linux = b""
|
|
|
|
# machine-id is stable across boots, boot_id is not.
|
|
for filename in "/etc/machine-id", "/proc/sys/kernel/random/boot_id":
|
|
try:
|
|
with open(filename, "rb") as f:
|
|
value = f.readline().strip()
|
|
except OSError:
|
|
continue
|
|
|
|
if value:
|
|
linux += value
|
|
break
|
|
|
|
# Containers share the same machine id, add some cgroup
|
|
# information. This is used outside containers too but should be
|
|
# relatively stable across boots.
|
|
try:
|
|
with open("/proc/self/cgroup", "rb") as f:
|
|
linux += f.readline().strip().rpartition(b"/")[2]
|
|
except OSError:
|
|
pass
|
|
|
|
if linux:
|
|
return linux
|
|
|
|
# On OS X, use ioreg to get the computer's serial number.
|
|
try:
|
|
# subprocess may not be available, e.g. Google App Engine
|
|
# https://github.com/pallets/werkzeug/issues/925
|
|
from subprocess import PIPE
|
|
from subprocess import Popen
|
|
|
|
dump = Popen(
|
|
["ioreg", "-c", "IOPlatformExpertDevice", "-d", "2"], stdout=PIPE
|
|
).communicate()[0]
|
|
match = re.search(b'"serial-number" = <([^>]+)', dump)
|
|
|
|
if match is not None:
|
|
return match.group(1)
|
|
except (OSError, ImportError):
|
|
pass
|
|
|
|
# On Windows, use winreg to get the machine guid.
|
|
if sys.platform == "win32":
|
|
import winreg
|
|
|
|
try:
|
|
with winreg.OpenKey(
|
|
winreg.HKEY_LOCAL_MACHINE,
|
|
"SOFTWARE\\Microsoft\\Cryptography",
|
|
0,
|
|
winreg.KEY_READ | winreg.KEY_WOW64_64KEY,
|
|
) as rk:
|
|
guid: str | bytes
|
|
guid_type: int
|
|
guid, guid_type = winreg.QueryValueEx(rk, "MachineGuid")
|
|
|
|
if guid_type == winreg.REG_SZ:
|
|
return guid.encode()
|
|
|
|
return guid
|
|
except OSError:
|
|
pass
|
|
|
|
return None
|
|
|
|
_machine_id = _generate()
|
|
return _machine_id
|
|
|
|
|
|
class _ConsoleFrame:
|
|
"""Helper class so that we can reuse the frame console code for the
|
|
standalone console.
|
|
"""
|
|
|
|
def __init__(self, namespace: dict[str, t.Any]):
|
|
self.console = Console(namespace)
|
|
self.id = 0
|
|
|
|
def eval(self, code: str) -> t.Any:
|
|
return self.console.eval(code)
|
|
|
|
|
|
def get_pin_and_cookie_name(
|
|
app: WSGIApplication,
|
|
) -> tuple[str, str] | tuple[None, None]:
|
|
"""Given an application object this returns a semi-stable 9 digit pin
|
|
code and a random key. The hope is that this is stable between
|
|
restarts to not make debugging particularly frustrating. If the pin
|
|
was forcefully disabled this returns `None`.
|
|
|
|
Second item in the resulting tuple is the cookie name for remembering.
|
|
"""
|
|
pin = os.environ.get("WERKZEUG_DEBUG_PIN")
|
|
rv = None
|
|
num = None
|
|
|
|
# Pin was explicitly disabled
|
|
if pin == "off":
|
|
return None, None
|
|
|
|
# Pin was provided explicitly
|
|
if pin is not None and pin.replace("-", "").isdecimal():
|
|
# If there are separators in the pin, return it directly
|
|
if "-" in pin:
|
|
rv = pin
|
|
else:
|
|
num = pin
|
|
|
|
modname = getattr(app, "__module__", t.cast(object, app).__class__.__module__)
|
|
username: str | None
|
|
|
|
try:
|
|
# getuser imports the pwd module, which does not exist in Google
|
|
# App Engine. It may also raise a KeyError if the UID does not
|
|
# have a username, such as in Docker.
|
|
username = getpass.getuser()
|
|
# Python >= 3.13 only raises OSError
|
|
except (ImportError, KeyError, OSError):
|
|
username = None
|
|
|
|
mod = sys.modules.get(modname)
|
|
|
|
# This information only exists to make the cookie unique on the
|
|
# computer, not as a security feature.
|
|
probably_public_bits = [
|
|
username,
|
|
modname,
|
|
getattr(app, "__name__", type(app).__name__),
|
|
getattr(mod, "__file__", None),
|
|
]
|
|
|
|
# This information is here to make it harder for an attacker to
|
|
# guess the cookie name. They are unlikely to be contained anywhere
|
|
# within the unauthenticated debug page.
|
|
private_bits = [str(uuid.getnode()), get_machine_id()]
|
|
|
|
h = hashlib.sha1()
|
|
for bit in chain(probably_public_bits, private_bits):
|
|
if not bit:
|
|
continue
|
|
if isinstance(bit, str):
|
|
bit = bit.encode()
|
|
h.update(bit)
|
|
h.update(b"cookiesalt")
|
|
|
|
cookie_name = f"__wzd{h.hexdigest()[:20]}"
|
|
|
|
# If we need to generate a pin we salt it a bit more so that we don't
|
|
# end up with the same value and generate out 9 digits
|
|
if num is None:
|
|
h.update(b"pinsalt")
|
|
num = f"{int(h.hexdigest(), 16):09d}"[:9]
|
|
|
|
# Format the pincode in groups of digits for easier remembering if
|
|
# we don't have a result yet.
|
|
if rv is None:
|
|
for group_size in 5, 4, 3:
|
|
if len(num) % group_size == 0:
|
|
rv = "-".join(
|
|
num[x : x + group_size].rjust(group_size, "0")
|
|
for x in range(0, len(num), group_size)
|
|
)
|
|
break
|
|
else:
|
|
rv = num
|
|
|
|
return rv, cookie_name
|
|
|
|
|
|
class DebuggedApplication:
|
|
"""Enables debugging support for a given application::
|
|
|
|
from werkzeug.debug import DebuggedApplication
|
|
from myapp import app
|
|
app = DebuggedApplication(app, evalex=True)
|
|
|
|
The ``evalex`` argument allows evaluating expressions in any frame
|
|
of a traceback. This works by preserving each frame with its local
|
|
state. Some state, such as context globals, cannot be restored with
|
|
the frame by default. When ``evalex`` is enabled,
|
|
``environ["werkzeug.debug.preserve_context"]`` will be a callable
|
|
that takes a context manager, and can be called multiple times.
|
|
Each context manager will be entered before evaluating code in the
|
|
frame, then exited again, so they can perform setup and cleanup for
|
|
each call.
|
|
|
|
:param app: the WSGI application to run debugged.
|
|
:param evalex: enable exception evaluation feature (interactive
|
|
debugging). This requires a non-forking server.
|
|
:param request_key: The key that points to the request object in this
|
|
environment. This parameter is ignored in current
|
|
versions.
|
|
:param console_path: the URL for a general purpose console.
|
|
:param console_init_func: the function that is executed before starting
|
|
the general purpose console. The return value
|
|
is used as initial namespace.
|
|
:param show_hidden_frames: by default hidden traceback frames are skipped.
|
|
You can show them by setting this parameter
|
|
to `True`.
|
|
:param pin_security: can be used to disable the pin based security system.
|
|
:param pin_logging: enables the logging of the pin system.
|
|
|
|
.. versionchanged:: 2.2
|
|
Added the ``werkzeug.debug.preserve_context`` environ key.
|
|
"""
|
|
|
|
_pin: str | None
|
|
_pin_cookie: str
|
|
|
|
def __init__(
|
|
self,
|
|
app: WSGIApplication,
|
|
evalex: bool = False,
|
|
request_key: str = "werkzeug.request",
|
|
console_path: str = "/console",
|
|
console_init_func: t.Callable[[], dict[str, t.Any]] | None = None,
|
|
show_hidden_frames: bool = False,
|
|
pin_security: bool = True,
|
|
pin_logging: bool = True,
|
|
) -> None:
|
|
if not console_init_func:
|
|
console_init_func = None
|
|
self.app = app
|
|
self.evalex = evalex
|
|
self.frames: dict[int, DebugFrameSummary | _ConsoleFrame] = {}
|
|
self.frame_contexts: dict[int, list[t.ContextManager[None]]] = {}
|
|
self.request_key = request_key
|
|
self.console_path = console_path
|
|
self.console_init_func = console_init_func
|
|
self.show_hidden_frames = show_hidden_frames
|
|
self.secret = gen_salt(20)
|
|
self._failed_pin_auth = Value("B")
|
|
|
|
self.pin_logging = pin_logging
|
|
if pin_security:
|
|
# Print out the pin for the debugger on standard out.
|
|
if os.environ.get("WERKZEUG_RUN_MAIN") == "true" and pin_logging:
|
|
_log("warning", " * Debugger is active!")
|
|
if self.pin is None:
|
|
_log("warning", " * Debugger PIN disabled. DEBUGGER UNSECURED!")
|
|
else:
|
|
_log("info", " * Debugger PIN: %s", self.pin)
|
|
else:
|
|
self.pin = None
|
|
|
|
self.trusted_hosts: list[str] = [".localhost", "127.0.0.1"]
|
|
"""List of domains to allow requests to the debugger from. A leading dot
|
|
allows all subdomains. This only allows ``".localhost"`` domains by
|
|
default.
|
|
|
|
.. versionadded:: 3.0.3
|
|
"""
|
|
|
|
@property
|
|
def pin(self) -> str | None:
|
|
if not hasattr(self, "_pin"):
|
|
pin_cookie = get_pin_and_cookie_name(self.app)
|
|
self._pin, self._pin_cookie = pin_cookie # type: ignore
|
|
return self._pin
|
|
|
|
@pin.setter
|
|
def pin(self, value: str | None) -> None:
|
|
if value is None:
|
|
# Set _pin to None explicitly to prevent regeneration by the getter
|
|
self._pin = None
|
|
else:
|
|
self._pin = value
|
|
|
|
@property
|
|
def pin_cookie_name(self) -> str:
|
|
"""The name of the pin cookie."""
|
|
if not hasattr(self, "_pin_cookie"):
|
|
pin_cookie = get_pin_and_cookie_name(self.app)
|
|
self._pin, self._pin_cookie = pin_cookie # type: ignore
|
|
return self._pin_cookie
|
|
|
|
def debug_application(
|
|
self, environ: WSGIEnvironment, start_response: StartResponse
|
|
) -> t.Iterator[bytes]:
|
|
"""Run the application and conserve the traceback frames."""
|
|
contexts: list[t.ContextManager[t.Any]] = []
|
|
|
|
if self.evalex:
|
|
environ["werkzeug.debug.preserve_context"] = contexts.append
|
|
|
|
app_iter = None
|
|
try:
|
|
app_iter = self.app(environ, start_response)
|
|
yield from app_iter
|
|
if hasattr(app_iter, "close"):
|
|
app_iter.close()
|
|
except Exception as e:
|
|
if hasattr(app_iter, "close"):
|
|
app_iter.close() # type: ignore
|
|
|
|
tb = DebugTraceback(e, skip=1, hide=not self.show_hidden_frames)
|
|
|
|
for frame in tb.all_frames:
|
|
self.frames[id(frame)] = frame
|
|
self.frame_contexts[id(frame)] = contexts
|
|
|
|
is_trusted = bool(self.check_pin_trust(environ))
|
|
html = tb.render_debugger_html(
|
|
evalex=self.evalex and self.check_host_trust(environ),
|
|
secret=self.secret,
|
|
evalex_trusted=is_trusted,
|
|
)
|
|
response = Response(html, status=500, mimetype="text/html")
|
|
|
|
try:
|
|
yield from response(environ, start_response)
|
|
except Exception:
|
|
# if we end up here there has been output but an error
|
|
# occurred. in that situation we can do nothing fancy any
|
|
# more, better log something into the error log and fall
|
|
# back gracefully.
|
|
environ["wsgi.errors"].write(
|
|
"Debugging middleware caught exception in streamed "
|
|
"response at a point where response headers were already "
|
|
"sent.\n"
|
|
)
|
|
|
|
environ["wsgi.errors"].write("".join(tb.render_traceback_text()))
|
|
|
|
def execute_command(
|
|
self,
|
|
request: Request,
|
|
command: str,
|
|
frame: DebugFrameSummary | _ConsoleFrame,
|
|
) -> Response:
|
|
"""Execute a command in a console."""
|
|
if not self.check_host_trust(request.environ):
|
|
return SecurityError() # type: ignore[return-value]
|
|
|
|
contexts = self.frame_contexts.get(id(frame), [])
|
|
|
|
with ExitStack() as exit_stack:
|
|
for cm in contexts:
|
|
exit_stack.enter_context(cm)
|
|
|
|
return Response(frame.eval(command), mimetype="text/html")
|
|
|
|
def display_console(self, request: Request) -> Response:
|
|
"""Display a standalone shell."""
|
|
if not self.check_host_trust(request.environ):
|
|
return SecurityError() # type: ignore[return-value]
|
|
|
|
if 0 not in self.frames:
|
|
if self.console_init_func is None:
|
|
ns = {}
|
|
else:
|
|
ns = dict(self.console_init_func())
|
|
ns.setdefault("app", self.app)
|
|
self.frames[0] = _ConsoleFrame(ns)
|
|
is_trusted = bool(self.check_pin_trust(request.environ))
|
|
return Response(
|
|
render_console_html(secret=self.secret, evalex_trusted=is_trusted),
|
|
mimetype="text/html",
|
|
)
|
|
|
|
def get_resource(self, request: Request, filename: str) -> Response:
|
|
"""Return a static resource from the shared folder."""
|
|
path = join("shared", basename(filename))
|
|
|
|
try:
|
|
data = pkgutil.get_data(__package__, path)
|
|
except OSError:
|
|
return NotFound() # type: ignore[return-value]
|
|
else:
|
|
if data is None:
|
|
return NotFound() # type: ignore[return-value]
|
|
|
|
etag = str(adler32(data) & 0xFFFFFFFF)
|
|
return send_file(
|
|
BytesIO(data), request.environ, download_name=filename, etag=etag
|
|
)
|
|
|
|
def check_pin_trust(self, environ: WSGIEnvironment) -> bool | None:
|
|
"""Checks if the request passed the pin test. This returns `True` if the
|
|
request is trusted on a pin/cookie basis and returns `False` if not.
|
|
Additionally if the cookie's stored pin hash is wrong it will return
|
|
`None` so that appropriate action can be taken.
|
|
"""
|
|
if self.pin is None:
|
|
return True
|
|
|
|
# If we failed too many times, then we're locked out.
|
|
if self._failed_pin_auth.value >= 10:
|
|
return False
|
|
|
|
val = parse_cookie(environ).get(self.pin_cookie_name)
|
|
if not val or "|" not in val:
|
|
return False
|
|
ts_str, pin_hash = val.split("|", 1)
|
|
|
|
try:
|
|
ts = int(ts_str)
|
|
except ValueError:
|
|
return False
|
|
|
|
if pin_hash != hash_pin(self.pin):
|
|
return None
|
|
return (time.time() - PIN_TIME) < ts
|
|
|
|
def check_host_trust(self, environ: WSGIEnvironment) -> bool:
|
|
return host_is_trusted(environ.get("HTTP_HOST"), self.trusted_hosts)
|
|
|
|
def _fail_pin_auth(self) -> None:
|
|
with self._failed_pin_auth.get_lock():
|
|
count = self._failed_pin_auth.value
|
|
self._failed_pin_auth.value = count + 1
|
|
|
|
time.sleep(5.0 if count > 5 else 0.5)
|
|
|
|
def pin_auth(self, request: Request) -> Response:
|
|
"""Authenticates with the pin."""
|
|
if not self.check_host_trust(request.environ):
|
|
return SecurityError() # type: ignore[return-value]
|
|
|
|
exhausted = False
|
|
auth = False
|
|
trust = self.check_pin_trust(request.environ)
|
|
pin = t.cast(str, self.pin)
|
|
|
|
# If the trust return value is `None` it means that the cookie is
|
|
# set but the stored pin hash value is bad. This means that the
|
|
# pin was changed. In this case we count a bad auth and unset the
|
|
# cookie. This way it becomes harder to guess the cookie name
|
|
# instead of the pin as we still count up failures.
|
|
bad_cookie = False
|
|
if trust is None:
|
|
self._fail_pin_auth()
|
|
bad_cookie = True
|
|
|
|
# If we're trusted, we're authenticated.
|
|
elif trust:
|
|
auth = True
|
|
|
|
# If we failed too many times, then we're locked out.
|
|
elif self._failed_pin_auth.value >= 10:
|
|
exhausted = True
|
|
|
|
# Otherwise go through pin based authentication
|
|
else:
|
|
entered_pin = request.args["pin"]
|
|
|
|
if entered_pin.strip().replace("-", "") == pin.replace("-", ""):
|
|
self._failed_pin_auth.value = 0
|
|
auth = True
|
|
else:
|
|
self._fail_pin_auth()
|
|
|
|
rv = Response(
|
|
json.dumps({"auth": auth, "exhausted": exhausted}),
|
|
mimetype="application/json",
|
|
)
|
|
if auth:
|
|
rv.set_cookie(
|
|
self.pin_cookie_name,
|
|
f"{int(time.time())}|{hash_pin(pin)}",
|
|
httponly=True,
|
|
samesite="Strict",
|
|
secure=request.is_secure,
|
|
)
|
|
elif bad_cookie:
|
|
rv.delete_cookie(self.pin_cookie_name)
|
|
return rv
|
|
|
|
def log_pin_request(self, request: Request) -> Response:
|
|
"""Log the pin if needed."""
|
|
if not self.check_host_trust(request.environ):
|
|
return SecurityError() # type: ignore[return-value]
|
|
|
|
if self.pin_logging and self.pin is not None:
|
|
_log(
|
|
"info", " * To enable the debugger you need to enter the security pin:"
|
|
)
|
|
_log("info", " * Debugger pin code: %s", self.pin)
|
|
return Response("")
|
|
|
|
def __call__(
|
|
self, environ: WSGIEnvironment, start_response: StartResponse
|
|
) -> t.Iterable[bytes]:
|
|
"""Dispatch the requests."""
|
|
# important: don't ever access a function here that reads the incoming
|
|
# form data! Otherwise the application won't have access to that data
|
|
# any more!
|
|
request = Request(environ)
|
|
response = self.debug_application
|
|
if request.args.get("__debugger__") == "yes":
|
|
cmd = request.args.get("cmd")
|
|
arg = request.args.get("f")
|
|
secret = request.args.get("s")
|
|
frame = self.frames.get(request.args.get("frm", type=int)) # type: ignore
|
|
if cmd == "resource" and arg:
|
|
response = self.get_resource(request, arg) # type: ignore
|
|
elif cmd == "pinauth" and secret == self.secret:
|
|
response = self.pin_auth(request) # type: ignore
|
|
elif cmd == "printpin" and secret == self.secret:
|
|
response = self.log_pin_request(request) # type: ignore
|
|
elif (
|
|
self.evalex
|
|
and cmd is not None
|
|
and frame is not None
|
|
and self.secret == secret
|
|
and self.check_pin_trust(environ)
|
|
):
|
|
response = self.execute_command(request, cmd, frame) # type: ignore
|
|
elif (
|
|
self.evalex
|
|
and self.console_path is not None
|
|
and request.path == self.console_path
|
|
):
|
|
response = self.display_console(request) # type: ignore
|
|
return response(environ, start_response)
|