forked from jpillora/cloud-torrent
-
Notifications
You must be signed in to change notification settings - Fork 0
/
reverse_proxy.go
79 lines (74 loc) · 1.54 KB
/
reverse_proxy.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
package httptoo
import (
"encoding/gob"
"io"
"net/http"
"net/url"
"sync"
)
func deepCopy(dst, src interface{}) error {
r, w := io.Pipe()
e := gob.NewEncoder(w)
d := gob.NewDecoder(r)
var decErr, encErr error
var wg sync.WaitGroup
wg.Add(1)
go func() {
defer wg.Done()
decErr = d.Decode(dst)
r.Close()
}()
encErr = e.Encode(src)
// Always returns nil.
w.CloseWithError(encErr)
wg.Wait()
if encErr != nil {
return encErr
}
return decErr
}
// Takes a request, and alters its destination fields, for proxying.
func RedirectedRequest(r *http.Request, newUrl string) (ret *http.Request, err error) {
u, err := url.Parse(newUrl)
if err != nil {
return
}
ret = new(http.Request)
*ret = *r
ret.Header = nil
err = deepCopy(&ret.Header, r.Header)
if err != nil {
return
}
ret.URL = u
ret.RequestURI = ""
return
}
func ForwardResponse(w http.ResponseWriter, r *http.Response) {
for h, vs := range r.Header {
for _, v := range vs {
w.Header().Add(h, v)
}
}
w.WriteHeader(r.StatusCode)
// Errors frequently occur writing the body when the client hangs up.
io.Copy(w, r.Body)
r.Body.Close()
}
func ReverseProxy(w http.ResponseWriter, r *http.Request, originUrl string, client *http.Client) (err error) {
if client == nil {
client = http.DefaultClient
}
originRequest, err := RedirectedRequest(r, originUrl)
if err != nil {
return
}
// b, _ := httputil.DumpRequest(originRequest, false)
// os.Stderr.Write(b)
originResp, err := client.Do(originRequest)
if err != nil {
return
}
ForwardResponse(w, originResp)
return
}