Go语言WebSocket跨域通信中的Origin头部处理教程

package main

import (
    "log"
    "net/http"
    "strings" // 用于字符串操作，如检查子串或前缀

    "github.com/gorilla/websocket"
)

// 定义一个WebSocket升级器，允许所有来源进行测试
// 在生产环境中，应限制CheckOrigin函数以增强安全性
var upgrader = websocket.Upgrader{
    ReadBufferSize:  1024,
    WriteBufferSize: 1024,
    CheckOrigin: func(r *http.Request) bool {
        // 生产环境中，这里应该进行严格的源验证
        // 例如：
        // allowedOrigins := []string{"http://localhost:8080", "https://yourdomain.com"}
        // origin := r.Header.Get("Origin")
        // for _, ao := range allowedOrigins {
        //  if origin == ao {
        //      return true
        //  }
        // }
        // return false

        // 示例：允许来自特定域名的请求
        origin := r.Header.Get("Origin")
        log.Printf("Received WebSocket connection request from Origin: %s", origin)

        // 假设我们只允许来自 localhost 或某个特定 IP 的请求
        // 对于 PhoneGap 应用，如果 Origin 是 file:// 或者为空，可能需要特殊处理
        if origin == "http://localhost:8080" ||
           strings.HasPrefix(origin, "http://192.168.") || // 允许局域网内的特定IP
           origin == "" { // 如果某些客户端不发送 Origin 头部，可以考虑允许（不推荐生产环境）
            return true
        }

        // 如果是 PhoneGap 应用，其 Origin 可能是 file://
        // 或者在某些 WebView 环境下可能根本不发送 Origin 头部
        // 此时，需要根据实际情况决定是否允许
        if strings.HasPrefix(origin, "file://") {
            log.Printf("Allowing file:// origin for PhoneGap/local development: %s", origin)
            return true
        }

        log.Printf("Rejected WebSocket connection from Origin: %s", origin)
        return false // 默认拒绝未知来源
    },
}

// WebSocket处理函数
func handleConnections(w http.ResponseWriter, r *http.Request) {
    // 尝试将HTTP连接升级为WebSocket连接
    conn, err := upgrader.Upgrade(w, r, nil)
    if err != nil {
        log.Printf("Failed to upgrade to WebSocket: %v", err)
        return
    }
    defer conn.Close() // 确保连接在函数结束时关闭

    log.Println("Client Connected")

    // 循环读取客户端消息
    for {
        messageType, message, err := conn.ReadMessage()
        if err != nil {
            log.Printf("Read error: %v", err)
            break
        }
        log.Printf("Received: %s", message)

        // 将收到的消息原样发回客户端
        err = conn.WriteMessage(messageType, message)
        if err != nil {
            log.Printf("Write error: %v", err)
            break
        }
    }
    log.Println("Client Disconnected")
}

func main() {
    http.HandleFunc("/ws", handleConnections) // 注册WebSocket路由

    log.Println("WebSocket server starting on :8080")
    err := http.ListenAndServe(":8080", nil)
    if err != nil {
        log.Fatalf("Server failed to start: %v", err)
    }
}