magic-agent 2.4.3 → 2.6.0

package/internal/client/ab_test.go

@@ -0,0 +1,287 @@
+package client
+
+// ab_test.go - agent-browser 封装与 shot-remote 端点的测试。
+//
+// 不连真 daemon（那要真 Chromium）：用 go build 现场编一个**假 agent-browser exe**
+// 验证探测链、--json 解析与端点行为。go 编译器不在 PATH 时相关用例跳过
+// （本机开发环境一定有；CI 沙箱也总有）。
+
+import (
+	"context"
+	"encoding/json"
+	"io"
+	"log"
+	"net/http"
+	"net/http/httptest"
+	"net/url"
+	"os"
+	"os/exec"
+	"path/filepath"
+	"runtime"
+	"strings"
+	"testing"
+)
+
+// abNativeExeName 当前平台 agent-browser 原生 exe 的文件名（与 ab.go 同规则）。
+func abNativeExeName() string {
+	name := "agent-browser-" + runtime.GOOS + "-" + runtime.GOARCH
+	if runtime.GOOS == "windows" {
+		name += ".exe"
+	}
+	return name
+}
+
+// fakeABBin 现场编一个假 agent-browser exe，返回其路径。
+//
+// 行为由环境变量 AB_FAKE_MODE 控制（abRun 会把调用方环境透传给子进程）：
+//
+//	ok      → 按子命令吐 {"success":true,"data":…}；screenshot 会真写出文件
+//	err     → 吐 {"success":false,"error":"fake boom"}
+//	garbage → 吐非 JSON 乱码
+//	missing → 无输出直接退出 1
+func fakeABBin(t *testing.T) string {
+	t.Helper()
+	src := `package main
+
+import (
+	"fmt"
+	"os"
+	"strings"
+)
+
+func main() {
+	mode := os.Getenv("AB_FAKE_MODE")
+	switch mode {
+	case "err":
+		fmt.Println(` + "`" + `{"success":false,"error":"fake boom"}` + "`" + `)
+		os.Exit(1)
+	case "garbage":
+		fmt.Println("garbage not json at all")
+		os.Exit(0)
+	case "missing":
+		os.Exit(1)
+	}
+	// 按子命令给一点可断言的 data（跳过全局 flag）。
+	sub := ""
+	for i, a := range os.Args[1:] {
+		if strings.HasPrefix(a, "-") || a == "magic-client" {
+			continue
+		}
+		switch a {
+		case "get", "open", "eval", "pdf", "snapshot":
+			continue
+		case "screenshot":
+			// 真的把文件写出来：screenshot 的**最后一个非 flag 参数**是路径
+			// （与真 agent-browser 一致 —— flag 必须放在路径前面）。
+			out := ""
+			for _, b := range os.Args[i+1:] {
+				if !strings.HasPrefix(b, "-") {
+					out = b
+				}
+			}
+			if out != "" {
+				_ = os.WriteFile(out, []byte("png"), 0o600)
+			}
+			fmt.Println(` + "`" + `{"success":true,"data":{"path":"x"}}` + "`" + `)
+			return
+		case "url":
+			fmt.Println(` + "`" + `{"success":true,"data":"https://example.com/page"}` + "`" + `)
+			return
+		case "title":
+			fmt.Println(` + "`" + `{"success":true,"data":"Example"}` + "`" + `)
+			return
+		}
+		if sub == "" {
+			sub = a
+		}
+	}
+	fmt.Println(` + "`" + `{"success":true,"data":{"ok":true}}` + "`" + `)
+}
+`
+	dir := t.TempDir()
+	srcFile := filepath.Join(dir, "main.go")
+	if err := os.WriteFile(srcFile, []byte(src), 0o600); err != nil {
+		t.Fatal(err)
+	}
+	exe := filepath.Join(dir, abNativeExeName())
+	cmd := exec.Command("go", "build", "-o", exe, srcFile)
+	if out, err := cmd.CombinedOutput(); err != nil {
+		t.Skipf("go build 假 exe 失败（跳过，不影响其它用例）: %v: %s", err, out)
+	}
+	return exe
+}
+
+// TestABBin探测链 环境变量 > npm 前缀 > 都没有时空。
+func TestABBin探测链(t *testing.T) {
+	exe := fakeABBin(t)
+	// 1) 显式环境变量最优先。
+	t.Setenv(EnvAgentBrowserBin, exe)
+	if got := abBin(); got != exe {
+		t.Fatalf("环境变量应最优先，实际 %q", got)
+	}
+	// 2) npm 全局前缀（<prefix>/node_modules/agent-browser/bin/<平台名>）。
+	t.Setenv(EnvAgentBrowserBin, "")
+	prefix := t.TempDir()
+	binDir := filepath.Join(prefix, "node_modules", "agent-browser", "bin")
+	if err := os.MkdirAll(binDir, 0o755); err != nil {
+		t.Fatal(err)
+	}
+	b, err := os.ReadFile(exe)
+	if err != nil {
+		t.Fatal(err)
+	}
+	if err := os.WriteFile(filepath.Join(binDir, abNativeExeName()), b, 0o755); err != nil {
+		t.Fatal(err)
+	}
+	t.Setenv("NPM_CONFIG_PREFIX", prefix)
+	got := abBin()
+	if !strings.Contains(filepath.ToSlash(got), "node_modules/agent-browser/bin/"+abNativeExeName()) {
+		t.Fatalf("应从 npm 前缀探测到，实际 %q", got)
+	}
+}
+
+// TestABRunJSON 解析：--json 成功 / 失败 / 乱码 / 无输出四种形状。
+func TestABRunJSON(t *testing.T) {
+	exe := fakeABBin(t)
+	t.Setenv(EnvAgentBrowserBin, exe)
+
+	t.Setenv("AB_FAKE_MODE", "ok")
+	data, err := abRun(context.Background(), "get", "url")
+	if err != nil {
+		t.Fatalf("ok 模式失败: %v", err)
+	}
+	if string(data) != `"https://example.com/page"` {
+		t.Fatalf("data 不对: %s", data)
+	}
+
+	t.Setenv("AB_FAKE_MODE", "err")
+	if _, err := abRun(context.Background(), "get", "url"); err == nil || !strings.Contains(err.Error(), "fake boom") {
+		t.Fatalf("err 模式应报 fake boom: %v", err)
+	}
+
+	t.Setenv("AB_FAKE_MODE", "garbage")
+	if _, err := abRun(context.Background(), "get", "url"); err == nil {
+		t.Fatal("乱码输出应报错")
+	}
+
+	t.Setenv("AB_FAKE_MODE", "missing")
+	if _, err := abRun(context.Background(), "get", "url"); err == nil {
+		t.Fatal("无输出退出应报错")
+	}
+}
+
+// TestABShot 落盘：假 exe 真写出文件，abShot 返回 nil 且文件在。
+func TestABShot(t *testing.T) {
+	exe := fakeABBin(t)
+	t.Setenv(EnvAgentBrowserBin, exe)
+	t.Setenv("AB_FAKE_MODE", "ok")
+	out := filepath.Join(t.TempDir(), "s.png")
+	if err := abShot(context.Background(), out, true); err != nil {
+		t.Fatalf("截图失败: %v", err)
+	}
+	if b, err := os.ReadFile(out); err != nil || string(b) != "png" {
+		t.Fatalf("截图文件没落盘: %v", err)
+	}
+}
+
+// abHideAll 把 abBin 的全部探测路径指到空目录（测「没装 agent-browser」的分支）。
+// t.Setenv 自动在测试结束时还原。PATH 上可能有真安装（npm 全局 bin），一并清掉 ——
+// 清 PATH 后 go build 假 exe 的用例会受影响，但 hide 只在 501 分支用，不冲突。
+func abHideAll(t *testing.T) {
+	t.Helper()
+	t.Setenv(EnvAgentBrowserBin, "")
+	t.Setenv("NPM_CONFIG_PREFIX", "")
+	t.Setenv("npm_config_prefix", "")
+	t.Setenv("APPDATA", t.TempDir()) // Windows 默认前缀指空
+	t.Setenv("PATH", t.TempDir())    // LookPath 兜底也指空
+}
+
+// TestShotRemoteEndpoint shot-remote 端点：能力位 gate / 参数校验 / 正常路径。
+func TestShotRemoteEndpoint(t *testing.T) {
+	exe := fakeABBin(t)
+	s := &Service{log: log.New(io.Discard, "", 0), contract: Contract{Features: []string{FeatureRequirementShots}}}
+	mux := http.NewServeMux()
+	mux.HandleFunc("/desk/requirements/shot-remote", s.handleRequirementShotRemote)
+	srv := httptest.NewServer(mux)
+	defer srv.Close()
+
+	// 没配 agent-browser → 501 + 安装提示。
+	// ⚠️ 本机可能真装了 agent-browser（npm 默认前缀 / PATH 都能命中）：
+	// 「没装」分支要把**所有**探测路径都指到空目录，否则测的是真安装。
+	abHideAll(t)
+	resp, err := http.Post(srv.URL+"/desk/requirements/shot-remote", "application/json",
+		strings.NewReader(`{"url":"https://example.com"}`))
+	if err != nil {
+		t.Fatal(err)
+	}
+	body := readAllBody(resp)
+	if resp.StatusCode != http.StatusNotImplemented || !strings.Contains(body, "agent-browser") {
+		t.Fatalf("没装 agent-browser 应 501 + 提示，实际 %d %s", resp.StatusCode, body)
+	}
+
+	// 配上假 exe → 正常路径回 data（截图落 uploads 目录并真实存在）。
+	t.Setenv(EnvAgentBrowserBin, exe)
+	t.Setenv("AB_FAKE_MODE", "ok")
+	t.Setenv("TMP", t.TempDir()) // uploads 目录走 os.TempDir()
+	t.Setenv("TEMP", os.Getenv("TMP"))
+	resp2, err := http.Post(srv.URL+"/desk/requirements/shot-remote", "application/json",
+		strings.NewReader(`{"url":"https://example.com"}`))
+	if err != nil {
+		t.Fatal(err)
+	}
+	body2 := readAllBody(resp2)
+	if resp2.StatusCode != http.StatusOK {
+		t.Fatalf("正常路径应 200，实际 %d %s", resp2.StatusCode, body2)
+	}
+	var out struct {
+		OK   bool   `json:"ok"`
+		Data string `json:"data"`
+	}
+	if err := json.Unmarshal([]byte(body2), &out); err != nil || !out.OK || !strings.HasPrefix(out.Data, "/desk/uploads/") {
+		t.Fatalf("响应形状不对: %s", body2)
+	}
+	// data 指向的文件真实存在。
+	name, err := url.PathUnescape(strings.TrimPrefix(out.Data, "/desk/uploads/"))
+	if err != nil {
+		t.Fatal(err)
+	}
+	if _, err := os.Stat(filepath.Join(os.TempDir(), "magic-client-uploads", name)); err != nil {
+		t.Fatalf("截图文件不在 uploads: %v", err)
+	}
+
+	// 非法 url → 400。
+	resp3, err := http.Post(srv.URL+"/desk/requirements/shot-remote", "application/json",
+		strings.NewReader(`{"url":"javascript:alert(1)"}`))
+	if err != nil {
+		t.Fatal(err)
+	}
+	resp3.Body.Close()
+	if resp3.StatusCode != http.StatusBadRequest {
+		t.Fatalf("非法 url 应 400，实际 %d", resp3.StatusCode)
+	}
+}
+
+// TestShotRemote能力位 缺 feature.requirement.shots 必须 501（与 /shots 端点同一杆秤）。
+func TestShotRemote能力位(t *testing.T) {
+	s := &Service{log: log.New(io.Discard, "", 0), contract: Contract{Features: []string{}}}
+	mux := http.NewServeMux()
+	mux.HandleFunc("/desk/requirements/shot-remote", s.handleRequirementShotRemote)
+	srv := httptest.NewServer(mux)
+	defer srv.Close()
+	resp, err := http.Post(srv.URL+"/desk/requirements/shot-remote", "application/json",
+		strings.NewReader(`{"url":"https://example.com"}`))
+	if err != nil {
+		t.Fatal(err)
+	}
+	defer resp.Body.Close()
+	if resp.StatusCode != http.StatusNotImplemented {
+		t.Fatalf("缺能力位应 501，实际 %d", resp.StatusCode)
+	}
+}
+
+// readAllBody 把响应体读完转成字符串（测试用，错误一律 fatal）。
+func readAllBody(resp *http.Response) string {
+	b, _ := io.ReadAll(resp.Body)
+	resp.Body.Close()
+	return string(b)
+}

package/internal/client/browse.go

@@ -26,8 +26,18 @@ import (
  *              → 回给手机
  *
  * 用**路径前缀**承载目标地址（而不是 ?url= 全量编码）：浏览器解析相对地址时会自动保留
- * 前缀，配上注入的 <base> 就能让同站链接继续走中转。令牌只放在**首帧**的查询串里，
- * 首帧响应下发 cookie，后续子资源请求靠 cookie 过鉴权（否则每个 URL 都得带令牌）。
+ * 前缀，配上注入的 <base> 就能让同站链接继续走中转。
+ *
+ * ── 鉴权：令牌进每一条改写出的 URL ──
+ * 早先令牌只放在**首帧**的查询串里，首帧响应下发 cookie，后续子资源请求靠 cookie 过鉴权。
+ * 那个方案在 APK（Capacitor 壳）里必然失败：壳的首屏 origin 是它自己的本地服务器
+ * （localhost），与被代理 iframe 所在的桌面端 LAN origin **跨站**，而 cookie 是
+ * SameSite=Lax —— 跨站子资源请求压根不带它，于是改写过的图片/脚本/样式全部 401
+ * （首帧有 ?magicToken= 所以页面能出来，百度移动页 CSS 又全内联，看起来就是「图片全裂」）。
+ * SameSite=None 要求 Secure，而客户端在局域网是明文 HTTP，此路不通。
+ * 现在令牌**直接拼进每条改写出的 URL**（?magicToken=…），上游转发前由 stripMagicToken
+ * 剥掉，任何嵌入上下文（同源 / 跨站 / 第三方 cookie 被墙）都不受影响。
+ * cookie 仍然照发（同源场景下聊胜于无），但不再是唯一凭据。
  *
  * ── 刻意不做的事 ──
  *   - 不转发上游的 Set-Cookie：上游会话存桌面端的 cookie jar，别去污染手机对本服务的 cookie。
@@ -40,8 +50,11 @@ import (
 const (
 	// browsePrefix 中转路径前缀。
 	browsePrefix = "/b/"
-	// browseCookie 首帧下发、供子资源请求复用的鉴权 cookie。
+	// browseCookie 首帧下发、供子资源请求复用的鉴权 cookie（辅助凭据，见文件头注释）。
 	browseCookie = "magic_browse"
+	// browseTokenParam 令牌查询参数名：改写出的每条 URL 都带它（stripMagicToken 会在
+	// 转发上游前剥掉）。不用 cookie 独扛鉴权，就是为了跨站 iframe（APK 壳）也能过。
+	browseTokenParam = "magicToken"
 	// browseMaxBytes HTML 改写单页size上限；超了直接报错（不静默截断成坏页面）。
 	browseMaxBytes = 12 << 20
 	// browseUA 伪装成 Android Chrome —— 直接拿 Go 默认 UA 会被不少移动站点降级甚至拒绝。
@@ -111,6 +124,16 @@ func (s *Service) browseUserAgent() string {
 
 // handleBrowse 中转入口。
 func (s *Service) handleBrowse(w http.ResponseWriter, r *http.Request) {
+	// relay 脚本本体：/b/relay.js。特判放在最前面 —— 它是**我们自己的**资源，
+	// 不能让 browseTarget 把 "relay.js" 当主机名去抓。
+	if strings.TrimPrefix(r.URL.Path, browsePrefix) == "relay.js" {
+		w.Header().Set("Content-Type", "text/javascript; charset=utf-8")
+		w.Header().Set("Cache-Control", "no-store")
+		w.WriteHeader(http.StatusOK)
+		_, _ = w.Write([]byte(browseRelayJS))
+		return
+	}
+	// （/b/shotlib.js 已随页面内截图退役：截图改走服务端 agent-browser，见 ab.go。）
 	target, ok := browseTarget(r)
 	if !ok {
 		writeErr(w, http.StatusBadRequest, "用法：/b/<http|https>/<host>/<path>")
@@ -118,9 +141,11 @@ func (s *Service) handleBrowse(w http.ResponseWriter, r *http.Request) {
 	}
 	target = remapBrowseHost(target)
 
-	// 鉴权：查询令牌（首帧）或 cookie（子资源）。
-	if want := strings.TrimSpace(s.opts.Token); want != "" {
-		got := strings.TrimSpace(r.URL.Query().Get("magicToken"))
+	// 鉴权：查询令牌（每条改写出的 URL 都带）或 cookie（同源场景下的辅助凭据）。
+	// want 同时是改写 URL 时要拼进去的令牌 —— 空表示没开鉴权，URL 保持干净。
+	want := strings.TrimSpace(s.opts.Token)
+	if want != "" {
+		got := strings.TrimSpace(r.URL.Query().Get(browseTokenParam))
 		if got == "" {
 			if c, err := r.Cookie(browseCookie); err == nil {
 				got = c.Value
@@ -130,7 +155,7 @@ func (s *Service) handleBrowse(w http.ResponseWriter, r *http.Request) {
 			writeErr(w, http.StatusUnauthorized, "缺少或错误的访问令牌（?magicToken= 或 cookie）")
 			return
 		}
-		if r.URL.Query().Get("magicToken") != "" {
+		if r.URL.Query().Get(browseTokenParam) != "" {
 			http.SetCookie(w, &http.Cookie{
 				Name: browseCookie, Value: want, Path: "/",
 				HttpOnly: true, SameSite: http.SameSiteLaxMode,
@@ -192,7 +217,7 @@ func (s *Service) handleBrowse(w http.ResponseWriter, r *http.Request) {
 	// 重定向：把 Location 也改成中转地址，否则浏览器一跳出前缀就离开中转了。
 	if loc := resp.Header.Get("Location"); loc != "" && resp.StatusCode >= 300 && resp.StatusCode < 400 {
 		if abs, perr := target.Parse(loc); perr == nil {
-			w.Header().Set("Location", browseURL(abs, proxyBase))
+			w.Header().Set("Location", browseURL(abs, proxyBase, want))
 		} else {
 			w.Header().Set("Location", loc)
 		}
@@ -210,7 +235,7 @@ func (s *Service) handleBrowse(w http.ResponseWriter, r *http.Request) {
 		w.Header().Set("Content-Type", "text/html; charset=utf-8")
 		w.Header().Set("Cache-Control", "no-store")
 		w.WriteHeader(resp.StatusCode)
-		_, _ = w.Write(rewriteBrowseHTML(body, target, proxyBase))
+		_, _ = w.Write(rewriteBrowseHTML(body, target, proxyBase, want))
 		return
 	}
 	// CSS 也要改写：样式表里 background-image / @import 引用的图不经过 HTML，
@@ -223,7 +248,7 @@ func (s *Service) handleBrowse(w http.ResponseWriter, r *http.Request) {
 		}
 		w.Header().Set("Content-Type", "text/css; charset=utf-8")
 		w.WriteHeader(resp.StatusCode)
-		_, _ = w.Write([]byte(rewriteCSS(string(body), target, proxyBase)))
+		_, _ = w.Write([]byte(rewriteCSS(string(body), target, proxyBase, want)))
 		return
 	}
 
@@ -314,11 +339,14 @@ func stripMagicToken(q string) string {
 // 原因：根相对地址按「所在文档 / 样式表的源」解析，而页面里常有**直连加载**的外站样式表
 // 或脚本（由 JS 动态插入，我们改写不到）。它们消费到 url(/b/...) 时会解析回自己的源
 // （实测：被解析成 ms.bdstatic.com/b/...，图片全裂）。绝对地址在任何上下文都指向中转。
-func browseURL(abs *url.URL, proxyBase string) string {
-	if abs == nil || (abs.Scheme != "http" && abs.Scheme != "https") {
-		if abs == nil {
-			return ""
-		}
+//
+// token 非空时拼上 ?magicToken=…：子资源鉴权不能只靠 cookie —— 跨站 iframe（APK 壳）
+// 下 SameSite=Lax 的 cookie 根本不随子资源请求发送，详见文件头注释。
+func browseURL(abs *url.URL, proxyBase, token string) string {
+	if abs == nil {
+		return ""
+	}
+	if abs.Scheme != "http" && abs.Scheme != "https" {
 		return abs.String()
 	}
 	p := abs.EscapedPath()
@@ -326,15 +354,22 @@ func browseURL(abs *url.URL, proxyBase string) string {
 		p = "/"
 	}
 	out := proxyBase + browsePrefix + abs.Scheme + "/" + abs.Host + p
-	if abs.RawQuery != "" {
-		out += "?" + abs.RawQuery
+	q := abs.RawQuery
+	if token != "" {
+		if q != "" {
+			q += "&"
+		}
+		q += browseTokenParam + "=" + url.QueryEscape(token)
+	}
+	if q != "" {
+		out += "?" + q
 	}
 	return out
 }
 
 // rewriteRef 把一个页面里的引用地址解析成绝对地址后再映射到中转。
 // data:/javascript:/mailto: 等非网络协议与纯锚点一律不动，避免把页面改坏。
-func rewriteRef(raw string, page *url.URL, proxyBase string) string {
+func rewriteRef(raw string, page *url.URL, proxyBase, token string) string {
 	t := strings.TrimSpace(raw)
 	if t == "" {
 		return raw
@@ -353,13 +388,16 @@ func rewriteRef(raw string, page *url.URL, proxyBase string) string {
 	if abs.Scheme != "http" && abs.Scheme != "https" {
 		return raw
 	}
-	return browseURL(abs, proxyBase)
+	return browseURL(abs, proxyBase, token)
 }
 
 // rewriteBrowseHTML 就地改写 HTML：剥掉会妨碍内嵌的头、把所有 URL 指回中转、注入 <base>。
 //
-// 分片处理：<script> 段原样保留（在 JS 里替换 URL 会砸脚本），其余段做属性 / srcset / CSS url() 改写。
-func rewriteBrowseHTML(body []byte, page *url.URL, proxyBase string) []byte {
+// 分片处理：<script> 元素拆成「开标签 + 内容」—— 开标签的属性（src 等）要改写
+// （否则 <script src="//cdn.example.com/a.js"> 这类协议相对地址会绕过中转，直连手机网络，
+// 桌面端才能访问的内网站点照样裂），内容原样保留（在 JS 里替换 URL 会砸脚本）。
+// 其余段做属性 / srcset / CSS url() 改写。
+func rewriteBrowseHTML(body []byte, page *url.URL, proxyBase, token string) []byte {
 	html := string(body)
 	html = browseBaseRe.ReplaceAllString(html, "")
 	html = browseCSPRe.ReplaceAllString(html, "")
@@ -367,27 +405,37 @@ func rewriteBrowseHTML(body []byte, page *url.URL, proxyBase string) []byte {
 	var b strings.Builder
 	last := 0
 	for _, loc := range browseScriptRe.FindAllStringIndex(html, -1) {
-		b.WriteString(rewriteChunk(html[last:loc[0]], page, proxyBase))
-		b.WriteString(html[loc[0]:loc[1]])
+		b.WriteString(rewriteChunk(html[last:loc[0]], page, proxyBase, token))
+		elem := html[loc[0]:loc[1]]
+		if gt := strings.IndexByte(elem, '>'); gt >= 0 {
+			b.WriteString(rewriteChunk(elem[:gt+1], page, proxyBase, token))
+			b.WriteString(elem[gt+1:])
+		} else {
+			b.WriteString(elem)
+		}
 		last = loc[1]
 	}
-	b.WriteString(rewriteChunk(html[last:], page, proxyBase))
+	b.WriteString(rewriteChunk(html[last:], page, proxyBase, token))
 	out := b.String()
 
 	// 注入 <base>：给**没被改写**的相对地址一个基准，指向本中转的页面地址。
 	// 改写过的地址现在是绝对的、不依赖 base（早先用根相对地址 + 指向原站的 base，
 	// 会被解析回原站导致图片全裂，已修）。
-	base := `<base href="` + htmlEscapeAttr(browseURL(page, proxyBase)) + `">`
+	// 同时注入 relay 脚本（/b/relay.js，见 browseinput.go）：连回 /browse-input
+	// 收指令，在被代理页面里派发合成鼠标/键盘事件 —— 为浏览器 MCP 驱动鼠标铺路。
+	// 脚本地址是绝对的（带 proxyBase）：页面 URL 可能是任何源，相对地址会解析回原站。
+	inject := `<base href="` + htmlEscapeAttr(browseURL(page, proxyBase, "")) + `">` +
+		`<script src="` + htmlEscapeAttr(proxyBase+"/b/relay.js") + `" defer></script>`
 	if i := strings.Index(strings.ToLower(out), "<head"); i >= 0 {
 		if j := strings.IndexByte(out[i:], '>'); j >= 0 {
 			k := i + j + 1
-			return []byte(out[:k] + base + out[k:])
+			return []byte(out[:k] + inject + out[k:])
 		}
 	}
-	return []byte(base + out)
+	return []byte(inject + out)
 }
 
-func rewriteChunk(chunk string, page *url.URL, proxyBase string) string {
+func rewriteChunk(chunk string, page *url.URL, proxyBase, token string) string {
 	chunk = browseAttrRe.ReplaceAllStringFunc(chunk, func(m string) string {
 		sub := browseAttrRe.FindStringSubmatch(m)
 		name := sub[1]
@@ -401,7 +449,7 @@ func rewriteChunk(chunk string, page *url.URL, proxyBase string) string {
 			q = '\''
 			val = sub[4]
 		}
-		return name + "=" + string(q) + rewriteRef(val, page, proxyBase) + string(q)
+		return name + "=" + string(q) + rewriteRef(val, page, proxyBase, token) + string(q)
 	})
 	chunk = browseSrcsetRe.ReplaceAllStringFunc(chunk, func(m string) string {
 		sub := browseSrcsetRe.FindStringSubmatch(m)
@@ -411,17 +459,17 @@ func rewriteChunk(chunk string, page *url.URL, proxyBase string) string {
 			val = sub[3]
 			q = `'`
 		}
-		return "srcset=" + q + rewriteSrcset(val, page, proxyBase) + q
+		return "srcset=" + q + rewriteSrcset(val, page, proxyBase, token) + q
 	})
-	return rewriteCSS(chunk, page, proxyBase)
+	return rewriteCSS(chunk, page, proxyBase, token)
 }
 
 // rewriteCSS 改写样式里的资源引用：url(...) 与 @import "..."。
-func rewriteCSS(css string, page *url.URL, proxyBase string) string {
+func rewriteCSS(css string, page *url.URL, proxyBase, token string) string {
 	css = browseCSSURLRe.ReplaceAllStringFunc(css, func(m string) string {
 		sub := browseCSSURLRe.FindStringSubmatch(m)
 		// 统一重建成不带引号的形式：Go 的 RE2 不支持反向引用，没法配对首尾引号。
-		return "url(" + rewriteRef(sub[1], page, proxyBase) + ")"
+		return "url(" + rewriteRef(sub[1], page, proxyBase, token) + ")"
 	})
 	css = browseImportRe.ReplaceAllStringFunc(css, func(m string) string {
 		sub := browseImportRe.FindStringSubmatch(m)
@@ -430,20 +478,20 @@ func rewriteCSS(css string, page *url.URL, proxyBase string) string {
 		if q == "'" {
 			val = sub[3]
 		}
-		return "@import " + q + rewriteRef(val, page, proxyBase) + q
+		return "@import " + q + rewriteRef(val, page, proxyBase, token) + q
 	})
 	return css
 }
 
 // rewriteSrcset 处理 srcset：逗号分隔，每段「URL [descriptor]」，只改 URL 那一段。
-func rewriteSrcset(val string, page *url.URL, proxyBase string) string {
+func rewriteSrcset(val string, page *url.URL, proxyBase, token string) string {
 	items := strings.Split(val, ",")
 	for i, it := range items {
 		fields := strings.Fields(strings.TrimSpace(it))
 		if len(fields) == 0 {
 			continue
 		}
-		fields[0] = rewriteRef(fields[0], page, proxyBase)
+		fields[0] = rewriteRef(fields[0], page, proxyBase, token)
 		items[i] = strings.Join(fields, " ")
 	}
 	return strings.Join(items, ", ")