github-starred/go-http-tunnel
2
0
Fork 0
mirror of https://github.com/mmatczuk/go-http-tunnel.git synced 2026-05-15 14:16:17 -06:00
Code Issues 61 Projects Packages Wiki Activity Actions

proto: ControlMessage improvements

Browse source
This commit is contained in:
mmatczuk 2017-01-23 14:20:10 +01:00
parent 9a7c36ff5a
commit 2857955688
9 changed files with 67 additions and 95 deletions
Download patch file Download diff file Expand all files Collapse all files

16
client.go
View file

@ -131,19 +131,27 @@ func (c *Client) dial(network, addr string, config *tls.Config) (net.Conn, error
func (c *Client) serveHTTP(w http.ResponseWriter, r *http.Request) {
if r.Method == http.MethodConnect {
c.log.Info("Handshake: hello from server")
c.log.Info("Connected to server: %s", r.RemoteAddr)
http.Error(w, "Nice to see you", http.StatusOK)
return
}
msg, err := proto.ParseControlMessage(r.Header)
if err != nil {
c.log.Warning("Parsing control message failed: %s", err)
http.Error(w, err.Error(), http.StatusBadRequest)
return
}
c.log.Debug("Start proxying %v", msg)
c.config.Proxy(w, r.Body, msg)
c.log.Debug("Done proxying %v", msg)
c.log.Debug("Start %s", msg)
switch msg.Action {
case proto.Proxy:
c.config.Proxy(w, r.Body, msg)
default:
c.log.Warning("Unknown action: %s", msg)
http.Error(w, err.Error(), http.StatusBadRequest)
}
c.log.Debug("Done %s", msg)
}
// Stop closes the connection between client and server. After stopping client