add proxy
This commit is contained in:
80
proxy/Proxy.go
Normal file
80
proxy/Proxy.go
Normal file
@@ -0,0 +1,80 @@
|
||||
package main
|
||||
|
||||
// code from : https://gist.github.com/fabrizioc1/4327250
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"io"
|
||||
"log"
|
||||
"net/http"
|
||||
)
|
||||
|
||||
type HttpConnection struct {
|
||||
Request *http.Request
|
||||
Response *http.Response
|
||||
}
|
||||
|
||||
type HttpConnectionChannel chan *HttpConnection
|
||||
|
||||
var connChannel = make(HttpConnectionChannel)
|
||||
|
||||
func PrintHTTP(conn *HttpConnection) {
|
||||
fmt.Println("[INFO] HTTP Request-------------------------------------------------------")
|
||||
fmt.Printf("%v %v\n", conn.Request.Method, conn.Request.RequestURI)
|
||||
for k, v := range conn.Request.Header {
|
||||
fmt.Println(k, ":", v)
|
||||
}
|
||||
fmt.Println("[INFO] HTTP Response------------------------------------------------------")
|
||||
fmt.Printf("HTTP/1.1 %v\n", conn.Response.Status)
|
||||
for k, v := range conn.Response.Header {
|
||||
fmt.Println(k, ":", v)
|
||||
}
|
||||
fmt.Println(conn.Response.Body)
|
||||
fmt.Println("[INFO] HTTP Ended---------------------------------------------------------")
|
||||
}
|
||||
|
||||
type Proxy struct {
|
||||
}
|
||||
|
||||
func NewProxy() *Proxy { return &Proxy{} }
|
||||
|
||||
func (p *Proxy) ServeHTTP(wr http.ResponseWriter, r *http.Request) {
|
||||
var resp *http.Response
|
||||
var err error
|
||||
var req *http.Request
|
||||
client := &http.Client{}
|
||||
|
||||
//log.Printf("%v %v", r.Method, r.RequestURI)
|
||||
req, err = http.NewRequest(r.Method, r.RequestURI, r.Body)
|
||||
for name, value := range r.Header {
|
||||
req.Header.Set(name, value[0])
|
||||
}
|
||||
resp, err = client.Do(req)
|
||||
r.Body.Close()
|
||||
|
||||
// combined for GET/POST
|
||||
if err != nil {
|
||||
http.Error(wr, err.Error(), http.StatusInternalServerError)
|
||||
return
|
||||
}
|
||||
|
||||
conn := &HttpConnection{r, resp}
|
||||
|
||||
for k, v := range resp.Header {
|
||||
wr.Header().Set(k, v[0])
|
||||
}
|
||||
wr.WriteHeader(resp.StatusCode)
|
||||
io.Copy(wr, resp.Body)
|
||||
resp.Body.Close()
|
||||
|
||||
PrintHTTP(conn)
|
||||
}
|
||||
|
||||
func main() {
|
||||
proxy := NewProxy()
|
||||
fmt.Println("==============================")
|
||||
err := http.ListenAndServe(":12345", proxy)
|
||||
if err != nil {
|
||||
log.Fatal("ListenAndServe: ", err.Error())
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user