Go Hijack黑科技

最新推荐文章于 2023-05-25 09:49:27 发布

Maratrix

最新推荐文章于 2023-05-25 09:49:27 发布

阅读量3k

点赞数 5

分类专栏： Go 文章标签： go

本文链接：https://blog.csdn.net/ya_feng/article/details/104354690

版权

本文介绍了Go语言中的Hijack接口，该接口允许接管HTTP连接的TCP层，接管后的连接不再受HTTP库管理。文章讨论了Hijack与普通HTTP请求的区别，并举例说明了其在RPC和WebSocket场景中的应用，强调了使用Hijack需要注意手动关闭连接以及不能继续写入HTTP数据等要点。

摘要由CSDN通过智能技术生成

最近在看Go标准库里面的rpc源码，发现了下面一段代码：

// ServeHTTP implements an http.Handler that answers RPC requests.
func (server *Server) ServeHTTP(w http.ResponseWriter, req *http.Request) {
	if req.Method != "CONNECT" {
		w.Header().Set("Content-Type", "text/plain; charset=utf-8")
		w.WriteHeader(http.StatusMethodNotAllowed)
		io.WriteString(w, "405 must CONNECT\n")
		return
	}
	conn, _, err := w.(http.Hijacker).Hijack()	//注意看这里
	if err != nil {
		log.Print("rpc hijacking ", req.RemoteAddr, ": ", err.Error())
		return
	}
	io.WriteString(conn, "HTTP/1.0 "+connected+"\n\n")
	server.ServeConn(conn)
}

这是一段接管 HTTP 连接的代码，所谓的接管 HTTP 连接是指这里接管了 HTTP 的 TCP 连接，也就是说 Golang 的内置 HTTP 库和 HTTPServer 库将不会管理这个 TCP 连接的生命周期，这个生命周期已经划给 Hijacker 了。

`Hijack` 介绍

首先看下官方的http.Hijacker接口介绍：

type Hijacker interface {
	// Hijack lets the caller take over the connection.
	// After a call to Hijack the HTTP server library
	// will not do anything else with the connection.
	//
	// It becomes the caller's responsibility to manage
	// and close the connection.
	//
	// The returned net.Conn may have read or write deadlines
	// already