一致性检查、文档同步

Author: ShiJiashuai

README.md

@@ -79,6 +79,10 @@ go run ./cmd/server
 
 ## 配置
 
+可以通过环境变量调整服务监听地址：
+
+- `ADDR`：HTTP 服务监听地址（默认 `:8080`）。
+
 默认情况下，信令服务只允许本地开发来源（`localhost` / `127.0.0.1`）。  
 如需在其他域名下访问，可以设置环境变量 `WS_ALLOWED_ORIGINS`，例如：
 
@@ -87,6 +91,13 @@ $env:WS_ALLOWED_ORIGINS="https://example.com,https://foo.bar"
 go run ./cmd/server
 ```
 
+如需临时允许所有来源（不推荐用于生产环境），可设置：
+
+```powershell
+$env:WS_ALLOWED_ORIGINS="*"
+go run ./cmd/server
+```
+
 在生产环境中请根据实际域名进行配置。
 
 ## 文档

ROADMAP.md

@@ -44,15 +44,15 @@
 
 ### 3.1 后端（Go 信令服务）
 
-- [ ] **增加基础日志输出**  
+- [x] **增加基础日志输出**  
   在 `internal/signal/hub.go` 中：
-  - [ ] WebSocket 升级失败时，打印请求路径和错误信息。
-  - [ ] 读取 JSON 消息失败时，打印错误并包含房间/客户端信息（若有）。
-  - [ ] 写入 JSON 失败时，同样记录日志，便于调试。
+  - [x] WebSocket 升级失败时，打印请求路径和错误信息。
+  - [x] 读取 JSON 消息失败时，打印错误并包含房间/客户端信息（若有）。
+  - [x] 写入 JSON 失败时，同样记录日志，便于调试。
 
-- [ ] **更稳健的连接清理**  
-  - [ ] 在 `HandleWS` 的循环中，只要出现读/写错误，就调用 `removeClient`，确保房间中不会残留“僵尸连接”。
-  - [ ] 在 `removeClient` 中确认：当房间成员为空时删除该房间。
+- [x] **更稳健的连接清理**  
+  - [x] 在 `HandleWS` 的循环中，只要出现读/写错误，就调用 `removeClient`，确保房间中不会残留“僵尸连接”。
+  - [x] 在 `removeClient` 中确认：当房间成员为空时删除该房间。
 
 - [ ] **（可选）简单心跳机制**  
   - 前端每隔 N 秒发送一个 `type: "ping"` 的消息。

changelog/2025-12-18.md

@@ -0,0 +1,19 @@
+# 2025-12-18
+
+- 初始化：新增 changelog 目录，用于记录后续变更。
+
+- 前端：补齐 DataChannel 聊天发送（Send 按钮 + Enter），未建立连接时给出提示。
+- 前端：成员列表渲染优化（胶囊按钮、禁用自己、点击自动填入 Remote）。
+- 前端：状态机与按钮状态统一（Join/Leave 文案切换、输入框启用禁用逻辑）。
+- 前端：信令与通话流程修复（offer 到达后自动进入 calling，区分手动关闭与异常断开）。
+- 前端：资源清理与可重复使用（Hangup 不再停止本地轨道；新增 Leave 清理 PC/WS/媒体；屏幕共享与再次呼叫兼容）。
+
+- 后端：信令 Hub 支持通过 `Options` 注入 Origin 校验配置（`AllowedOrigins`/`AllowAllOrigins`），避免包内隐式全局 env 状态。
+- 后端：入口统一解析 `WS_ALLOWED_ORIGINS` 并注入 Hub；服务启动改用 `ServeMux` + `http.Server`，避免使用默认全局路由。
+- 后端：`writePump` 写失败时主动关闭连接，触发读循环退出并完成资源清理。
+- 后端：升级失败日志补充 `path`；连接结束时关闭 `send` 通道以退出写协程；`removeClient` 调整为不关闭通道，避免 `leave` 后同连接再次 `join` 引发 panic。
+
+- 文档：同步更新 `docs/signaling.md` 中的 Hub 结构与 `writePump` 示例代码。
+- 文档：更新 `README.md` 配置章节（补充 `ADDR`、`WS_ALLOWED_ORIGINS="*"` 用法），并同步更新 `ROADMAP.md` 阶段 1 后端任务勾选状态。
+
+- 文档：同步补充 `docs/guide.md` 中 Hub 的 Origin 校验配置说明（`allowedOrigins`/`allowAllOrigins`），强调由入口解析 `WS_ALLOWED_ORIGINS` 并通过 `signal.Options` 注入。

cmd/server/main.go

@@ -5,22 +5,47 @@ import (
 	"net/http"
 	"os"
 	"path/filepath"
+	"strings"
 
 	"lessup/webrtc/internal/signal"
 )
 
 func main() {
-	hub := signal.NewHub()
-	http.HandleFunc("/ws", hub.HandleWS)
-
-	webDir := filepath.Join("web")
-	fs := http.FileServer(http.Dir(webDir))
-	http.Handle("/", fs)
-
 	addr := ":8080"
 	if v := os.Getenv("ADDR"); v != "" {
 		addr = v
 	}
-	log.Println("listening", addr)
-	log.Fatal(http.ListenAndServe(addr, nil))
+
+	var wsAllowAll bool
+	var wsAllowed []string
+	if v := os.Getenv("WS_ALLOWED_ORIGINS"); v != "" {
+		if v == "*" {
+			wsAllowAll = true
+		} else {
+			parts := strings.Split(v, ",")
+			wsAllowed = make([]string, 0, len(parts))
+			for _, p := range parts {
+				p = strings.TrimSpace(p)
+				if p != "" {
+					wsAllowed = append(wsAllowed, p)
+				}
+			}
+		}
+	}
+
+	hub := signal.NewHubWithOptions(signal.Options{
+		AllowedOrigins:  wsAllowed,
+		AllowAllOrigins: wsAllowAll,
+	})
+
+	mux := http.NewServeMux()
+	mux.HandleFunc("/ws", hub.HandleWS)
+
+	webDir := filepath.Join("web")
+	fs := http.FileServer(http.Dir(webDir))
+	mux.Handle("/", fs)
+
+	srv := &http.Server{Addr: addr, Handler: mux}
+	log.Println("server: listening", addr)
+	log.Fatal(srv.ListenAndServe())
 }

docs/guide.md

@@ -104,6 +104,7 @@ type Message struct {
 - `Hub`：
   - `rooms map[string]map[string]*Client`：`房间名 -> (客户端 ID -> Client)`。
   - `upg websocket.Upgrader`：升级 HTTP 为 WebSocket。
+  - `allowedOrigins` / `allowAllOrigins`：WebSocket `Origin` 校验配置（由服务入口解析 `WS_ALLOWED_ORIGINS` 并通过 `signal.Options` 注入）。
 
 - `Client`：
   - `id`：客户端 ID（由前端生成，例如随机字符串）。

docs/signaling.md

@@ -84,6 +84,9 @@ type Hub struct {
     mu    sync.RWMutex
     rooms map[string]map[string]*Client
     upg   websocket.Upgrader
+
+	allowedOrigins  []string
+	allowAllOrigins bool
 }
 
 type Client struct {
@@ -125,6 +128,7 @@ type Client struct {
   func (h *Hub) writePump(c *Client) {
       for msg := range c.send {
           if err := c.conn.WriteJSON(msg); err != nil {
+              c.conn.Close()
               break
           }
       }
@@ -147,14 +151,15 @@ type Client struct {
 func (h *Hub) HandleWS(w http.ResponseWriter, r *http.Request) {
     c, err := h.upg.Upgrade(w, r, nil)
     if err != nil {
-        log.Printf("signal: ws upgrade failed from %s: %v", r.RemoteAddr, err)
+        log.Printf("signal: ws upgrade failed from %s path=%s: %v", r.RemoteAddr, r.URL.Path, err)
         return
     }
     log.Printf("signal: ws connected from %s", r.RemoteAddr)
     client := &Client{conn: c, send: make(chan Message, 32)}
     go h.writePump(client)
     defer func() {
         h.removeClient(client)
+        close(client.send)
         c.Close()
     }()
     for {
@@ -188,6 +193,7 @@ func (h *Hub) HandleWS(w http.ResponseWriter, r *http.Request) {
 3. 启动 `writePump` 协程，负责异步写消息。
 4. 使用 `defer` 确保函数结束时：
    - 调用 `removeClient` 移除客户端；
+   - 关闭 `send` 通道，结束 `writePump` 写协程；
    - 关闭连接 `c.Close()`。
 
 ### 4.2 消息读取与分发

internal/signal/hub.go

@@ -4,7 +4,6 @@ import (
 	"log"
 	"net/http"
 	"net/url"
-	"os"
 	"strings"
 	"sync"
 
@@ -15,6 +14,9 @@ type Hub struct {
 	mu    sync.RWMutex
 	rooms map[string]map[string]*Client
 	upg   websocket.Upgrader
+
+	allowedOrigins  []string
+	allowAllOrigins bool
 }
 
 type Client struct {
@@ -24,31 +26,31 @@ type Client struct {
 	send chan Message
 }
 
-var (
-	allowedOrigins  []string
-	allowAllOrigins bool
-)
+type Options struct {
+	AllowedOrigins  []string
+	AllowAllOrigins bool
+}
 
-func init() {
-	v := os.Getenv("WS_ALLOWED_ORIGINS")
-	if v == "" {
-		return
-	}
-	if v == "*" {
-		allowAllOrigins = true
-		return
-	}
-	parts := strings.Split(v, ",")
-	for _, p := range parts {
-		p = strings.TrimSpace(p)
-		if p != "" {
-			allowedOrigins = append(allowedOrigins, p)
-		}
+func NewHub() *Hub {
+	return NewHubWithOptions(Options{})
+}
+
+func NewHubWithOptions(opts Options) *Hub {
+	h := &Hub{
+		rooms:           make(map[string]map[string]*Client),
+		allowedOrigins:  append([]string(nil), opts.AllowedOrigins...),
+		allowAllOrigins: opts.AllowAllOrigins,
+	}
+	h.upg = websocket.Upgrader{
+		CheckOrigin: func(r *http.Request) bool {
+			return h.isOriginAllowed(r)
+		},
 	}
+	return h
 }
 
-func isOriginAllowed(r *http.Request) bool {
-	if allowAllOrigins {
+func (h *Hub) isOriginAllowed(r *http.Request) bool {
+	if h.allowAllOrigins {
 		return true
 	}
 	origin := r.Header.Get("Origin")
@@ -59,7 +61,7 @@ func isOriginAllowed(r *http.Request) bool {
 		}
 		return false
 	}
-	if len(allowedOrigins) == 0 {
+	if len(h.allowedOrigins) == 0 {
 		u, err := url.Parse(origin)
 		if err != nil {
 			return false
@@ -70,36 +72,26 @@ func isOriginAllowed(r *http.Request) bool {
 		}
 		return false
 	}
-	for _, o := range allowedOrigins {
+	for _, o := range h.allowedOrigins {
 		if o == origin {
 			return true
 		}
 	}
 	return false
 }
 
-func NewHub() *Hub {
-	return &Hub{
-		rooms: make(map[string]map[string]*Client),
-		upg: websocket.Upgrader{
-			CheckOrigin: func(r *http.Request) bool {
-				return isOriginAllowed(r)
-			},
-		},
-	}
-}
-
 func (h *Hub) HandleWS(w http.ResponseWriter, r *http.Request) {
 	c, err := h.upg.Upgrade(w, r, nil)
 	if err != nil {
-		log.Printf("signal: ws upgrade failed from %s: %v", r.RemoteAddr, err)
+		log.Printf("signal: ws upgrade failed from %s path=%s: %v", r.RemoteAddr, r.URL.Path, err)
 		return
 	}
 	log.Printf("signal: ws connected from %s", r.RemoteAddr)
 	client := &Client{conn: c, send: make(chan Message, 32)}
 	go h.writePump(client)
 	defer func() {
 		h.removeClient(client)
+		close(client.send)
 		c.Close()
 	}()
 	for {
@@ -162,30 +154,34 @@ func (h *Hub) removeClient(c *Client) {
 	if c.room == "" || c.id == "" {
 		return
 	}
-	if m, ok := h.rooms[c.room]; ok {
-		if existing, ok2 := m[c.id]; ok2 {
-			delete(m, c.id)
-			close(existing.send)
+	room := c.room
+	if m, ok := h.rooms[room]; ok {
+		if _, ok2 := m[c.id]; !ok2 {
+			c.room = ""
+			return
 		}
+		delete(m, c.id)
+		c.room = ""
 		if len(m) == 0 {
-			delete(h.rooms, c.room)
-			log.Printf("signal: room %s closed", c.room)
-		} else {
-			members := make([]string, 0, len(m))
-			for id := range m {
-				members = append(members, id)
-			}
-			msg := Message{
-				Type:    "room_members",
-				Room:    c.room,
-				Members: members,
-			}
-			for _, cli := range m {
-				if cli != nil && cli.conn != nil {
-					select {
-					case cli.send <- msg:
-					default:
-					}
+			delete(h.rooms, room)
+			log.Printf("signal: room %s closed", room)
+			return
+		}
+
+		members := make([]string, 0, len(m))
+		for id := range m {
+			members = append(members, id)
+		}
+		msg := Message{
+			Type:    "room_members",
+			Room:    room,
+			Members: members,
+		}
+		for _, cli := range m {
+			if cli != nil && cli.conn != nil {
+				select {
+				case cli.send <- msg:
+				default:
 				}
 			}
 		}
@@ -210,6 +206,7 @@ func (h *Hub) writePump(c *Client) {
 	for msg := range c.send {
 		if err := c.conn.WriteJSON(msg); err != nil {
 			log.Printf("signal: write message error room=%s id=%s: %v", c.room, c.id, err)
+			c.conn.Close()
 			break
 		}
 	}