_lsy-key-scan.py 2.7 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374
  1. # -*- coding: utf-8 -*-
  2. """提交前体检:logs/ 下的产物里有没有混进机构密钥。
  3. 只打计数与文件名(文件名不是秘密),绝不打印密钥本身或长度以外的信息。"""
  4. import io, os, re, sys
  5. YML = "java/src/main/resources/application.yml"
  6. VAR = re.compile(r"^\$\{([A-Za-z0-9_]+):(.*)\}$")
  7. def read_block(path, top):
  8. lines = io.open(path, encoding="utf-8", errors="replace").read().split("\n")
  9. start = None
  10. for i, ln in enumerate(lines):
  11. if re.match(r"^%s:\s*$" % re.escape(top), ln):
  12. start = i; break
  13. if start is None:
  14. raise SystemExit("找不到顶层段 %s:" % top)
  15. out = {}
  16. for j in range(start + 1, len(lines)):
  17. ln = lines[j]
  18. if ln.strip() == "" or ln.lstrip().startswith("#"):
  19. continue
  20. if not ln.startswith((" ", "\t")):
  21. break
  22. m = re.match(r"^\s+([A-Za-z0-9_.\-]+):\s*(.*)$", ln)
  23. if m:
  24. out[m.group(1)] = re.split(r"\s+#", m.group(2), maxsplit=1)[0].strip().strip("'\"")
  25. return out
  26. cfg = read_block(YML, "leshouying")
  27. raw = cfg.get("key", "")
  28. m = VAR.match(raw)
  29. KEY = (os.environ.get(m.group(1)) or m.group(2)) if m else raw
  30. print("密钥长度 = %d(只打长度)" % len(KEY))
  31. if len(KEY) < 8:
  32. raise SystemExit("密钥太短,无法作为搜索针,体检作废")
  33. needle = KEY.encode("utf-8")
  34. # 正对照:密钥必然在 application.yml 里;扫不到说明扫描器/针是坏的
  35. def hits(path):
  36. try:
  37. return io.open(path, "rb").read().count(needle)
  38. except Exception:
  39. return 0
  40. pos = hits(YML)
  41. print("正对照 application.yml 命中 = %d %s" % (pos, "OK" if pos >= 1 else "!! 扫描器失效,体检作废"))
  42. if pos < 1:
  43. raise SystemExit(1)
  44. # ★ 2026-09-28 改成**递归扫全部 logs/_lsy\***,不再用硬编码文件表。
  45. # 原表漏了后来新增的 logs/_lsy-lc4/ 与几个 query 探针 —— 于是「体检通过」只覆盖了表里那些,
  46. # 而表和新增产物**各改各的**(同一类失真:量具自己的覆盖面没人量)。
  47. # 实测过它漏的后果:ed310fc 之后新增的 11 个产物不在表里,体检说「干净」而它们根本没被读过。
  48. # 递归之后新增产物自动在射程内,这个失效模式整类消失。
  49. files = []
  50. for root, _dirs, names in os.walk("logs"):
  51. for n in sorted(names):
  52. if n.startswith("_lsy"):
  53. files.append(os.path.join(root, n))
  54. files.sort()
  55. bad, n = [], 0
  56. for f in files:
  57. n += 1
  58. c = hits(f)
  59. if c:
  60. bad.append((f, c))
  61. print("扫描文件数 = %d" % n)
  62. print("含密钥的文件数 = %d" % len(bad))
  63. for f, c in bad:
  64. print(" !! %s 命中 %d 次" % (f, c))
  65. print("结论: %s" % ("干净,可提交" if not bad else "**有泄漏,不得提交**"))