Golang中WebSocket和WSS的支持_golang wss，费时6个月成功入职阿里

机器学习之心AI
51 阅读5分钟

func main() { http.HandleFunc("/websocket", WebSocketHandler) log.Println("WebSocket server is running on :8080") http.ListenAndServe(":8080", nil) }


在上面的代码中，我们首先导入`github.com/gorilla/websocket`包，并定义了一个`upgrader`对象。然后，我们实现了一个`WebSocketHandler`函数，用于处理WebSocket请求。在该函数中，我们使用`upgrader.Upgrade`方法将HTTP连接升级为WebSocket连接，并通过`conn.ReadMessage`方法读取客户端发送的消息，并使用`conn.WriteMessage`方法向客户端发送消息。最后，我们使用`http.HandleFunc`函数将`WebSocketHandler`函数注册为处理WebSocket请求的处理程序，并使用`http.ListenAndServe`函数启动WebSocket服务器。


#### 构建WebSocket客户端


要构建一个WebSocket客户端，我们可以使用`github.com/gorilla/websocket`包提供的`Dial`函数来建立与WebSocket服务器的连接。下面是一个简单的示例，演示了如何使用Golang构建一个WebSocket客户端：

package main

import ( "fmt" "log" "net/url" "github.com/gorilla/websocket" )

func main() { u := url.URL{Scheme: "ws", Host: "localhost:8080", Path: "/websocket"} conn, _, err := websocket.DefaultDialer.Dial(u.String(), nil) if err != nil { log.Println("Failed to connect to WebSocket server:", err) return }

defer conn.Close()

// 向服务器发送消息
err = conn.WriteMessage(websocket.TextMessage, []byte("Hello, server!"))
if err != nil {
	log.Println("Failed to write message:", err)
	return
}

// 读取服务器发送的消息
\_, message, err := conn.ReadMessage()
if err != nil {
	log.Println("Failed to read message:", err)
	return
}

fmt.Println("Received message:", string(message))

}


在上面的代码中，我们首先导入`github.com/gorilla/websocket`包，并使用`websocket.DefaultDialer.Dial`方法建立与WebSocket服务器的连接。然后，我们使用`conn.WriteMessage`方法向服务器发送消息，并使用`conn.ReadMessage`方法读取服务器发送的消息。最后，我们将接收到的消息打印出来。


### Golang中的WSS


#### 生成自签名证书


要在Golang中使用WSS，首先需要生成一个自签名证书。可以使用OpenSSL工具来生成自签名证书。下面是一个简单的示例，演示了如何生成自签名证书：

$ openssl req -x509 -nodes -days 365 -newkey rsa:2048 -keyout key.pem -out cert.pem


在上面的命令中，我们使用`openssl req`命令生成一个自签名证书，并将私钥保存到`key.pem`文件中，将公钥保存到`cert.pem`文件中。在实际使用中，你可能需要根据自己的需求来生成证书。


#### 使用WSS的WebSocket服务器和客户端


要在Golang中使用WSS，我们只需要在HTTP服务器和客户端中使用TLS配置即可。下面是一个简单的示例，演示了如何在Golang中构建一个使用WSS的WebSocket服务器和客户端：

package main

import ( "fmt" "log" "net/http" "github.com/gorilla/websocket" )

var upgrader = websocket.Upgrader{ ReadBufferSize: 1024, WriteBufferSize: 1024, }

func WebSocketHandler(w http.ResponseWriter, r *http.Request) { conn, err := upgrader.Upgrade(w, r, nil) if err != nil { log.Println("Failed to upgrade to WebSocket:", err) return }

defer conn.Close()

for {
	\_, message, err := conn.ReadMessage()
	if err != nil {
		log.Println("Failed to read message:", err)
		break
	}

	fmt.Println("Received message:", string(message))

	err = conn.WriteMessage(websocket.TextMessage, []byte("Hello, client!"))
	if err != nil {
		log.Println("Failed to write message:", err)
		break
	}
}

}

func main() { http.HandleFunc("/websocket", WebSocketHandler) log.Println("WebSocket server is running on :8080") err := http.ListenAndServeTLS(":8080", "cert.pem", "key.pem", nil) if err != nil { log.Fatal("Failed to start WebSocket server:", err) } }


在上面的代码中，我们首先导入`github.com/gorilla/websocket`包，并定义了一个`upgrader`对象。然后，我们实现了一个`WebSocketHandler`函数，用于处理WebSocket请求。最后，我们使用`http.ListenAndServeTLS`函数启动一个使用WSS的WebSocket服务器，并提供证书文件`cert.pem`和私钥文件`key.pem`。


对于WebSocket客户端，只需要在建立连接时使用`DialTLS`方法，并提供证书的URL即可。下面是一个简单的示例：

package main

import ( "fmt" "log" "net/url" "github.com/gorilla/websocket" )

func main() { u := url.URL{Scheme: "wss", Host: "localhost:8080", Path: "/websocket"} conn, _, err := websocket.DefaultDialer.Dial(u.String(), nil) if err != nil { log.Println("Failed to connect to WebSocket server:", err) return }

defer conn.Close()

err = conn.WriteMessage(websocket.TextMessage, []byte("Hello, server!"))
if err != nil {
	log.Println("Failed to write message:", err)
	return
}

\_, message, err := conn.ReadMessage()
if err != nil {
	log.Println("Failed to read message:", err)
	return
}

fmt.Println("Received message:", string(message))

}


在上面的代码中，我们首先导入`github.com/gorilla/websocket`包，并使用`websocket.DefaultDialer.Dial`方法建立一个使用WSS的WebSocket连接。


### 安全性的考虑


在使用Golang构建WebSocket和WSS的过程中，保证数据的安全性是至关重要的。以下是几个在实现WebSocket和WSS时应考虑的安全性问题。


#### 1. SSL/TLS加密


WSS协议通过在HTTP协议基础上添加SSL/TLS加密层来确保数据的安全传输。在Golang中，可以使用`http.ListenAndServeTLS`函数来启动一个使用WSS的WebSocket服务器，通过提供证书和私钥来实现加密传输。


#### 2. 跨站脚本攻击（XSS）防护


跨站脚本攻击（XSS）是一种常见的网络攻击方式，攻击者通过在网页中插入恶意脚本来窃取用户的敏感信息或进行其他恶意行为。为了保护WebSocket应用程序免受XSS攻击，可以采取以下措施：


* 输入验证和过滤：确保所有用户输入的数据都进行验证和过滤，防止恶意脚本的注入。
* 安全的内容解析：使用安全的方式解析和渲染用户提供的内容，如使用HTML转义来防止注入攻击。


#### 3. 跨站请求伪造（CSRF）防护



![img](https://p3-xtjj-sign.byteimg.com/tos-cn-i-73owjymdk6/e6850d90d31e4236af53574756a28679~tplv-73owjymdk6-jj-mark-v1:0:0:0:0:5o6Y6YeR5oqA5pyv56S-5Yy6IEAg5py65Zmo5a2m5Lmg5LmL5b-DQUk=:q75.awebp?rk3s=f64ab15b&x-expires=1771857280&x-signature=RX57ABrR7r8HQ2DpQogdz%2Fqpgmg%3D)
![img](https://p3-xtjj-sign.byteimg.com/tos-cn-i-73owjymdk6/e413dddcc520473caf173f95e0a9d973~tplv-73owjymdk6-jj-mark-v1:0:0:0:0:5o6Y6YeR5oqA5pyv56S-5Yy6IEAg5py65Zmo5a2m5Lmg5LmL5b-DQUk=:q75.awebp?rk3s=f64ab15b&x-expires=1771857280&x-signature=4697Ue3iA1X4POyMY6U6sgC9CkE%3D)

**网上学习资料一大堆，但如果学到的知识不成体系，遇到问题时只是浅尝辄止，不再深入研究，那么很难做到真正的技术提升。**

**[需要这份系统化的资料的朋友，可以添加戳这里获取](https://gitee.com/vip204888)**


**一个人可以走的很快，但一群人才能走的更远！不论你是正从事IT行业的老鸟或是对IT行业感兴趣的新人，都欢迎加入我们的的圈子（技术交流、学习资源、职场吐槽、大厂内推、面试辅导），让我们一起学习成长！**
avatar
文章
阅读
粉丝