techninja1008 · March 30, 2018 18:32 · Mar 30, 2018 · Mar 30, 2018 · Mar 30, 2018 · Mar 30, 2018
diff --git a/HttpProxy.go b/HttpProxy.go
@@ -68,7 +68,7 @@ func (p *proxy) ServeHTTP(wr http.ResponseWriter, req *http.Request) {
 	//http://golang.org/src/pkg/net/http/client.go
 	req.RequestURI = ""
 
-	u, err := url.Parse("http://127.0.0.1:8001/api/v1/namespaces/" + p.ns + "/services/" + p.svc + "/proxy" + req.URL)
+	u, err := url.Parse("http://127.0.0.1:8001/api/v1/namespaces/" + p.ns + "/services/" + p.svc + "/proxy" + req.URL.RequestURI())
 	if err != nil {
 		http.Error(wr, "Server Error", http.StatusInternalServerError)
 		log.Fatal("ServeHTTP:", err)

diff --git a/HttpProxy.go b/HttpProxy.go
@@ -68,7 +68,7 @@ func (p *proxy) ServeHTTP(wr http.ResponseWriter, req *http.Request) {
 	//http://golang.org/src/pkg/net/http/client.go
 	req.RequestURI = ""
 
-	u, err := url.Parse("http://127.0.0.1:8001/api/v1/namespaces/" + p.ns + "/services/" + p.svc + "/proxy" + req.URL.Path)
+	u, err := url.Parse("http://127.0.0.1:8001/api/v1/namespaces/" + p.ns + "/services/" + p.svc + "/proxy" + req.URL)
 	if err != nil {
 		http.Error(wr, "Server Error", http.StatusInternalServerError)
 		log.Fatal("ServeHTTP:", err)

diff --git a/HttpProxy.go b/HttpProxy.go
@@ -99,7 +99,7 @@ func (p *proxy) ServeHTTP(wr http.ResponseWriter, req *http.Request) {
 }
 
 func main() {
-	var addr = flag.String("addr", "127.0.0.1:8001", "The addr of the application.")
+	var addr = flag.String("addr", "127.0.0.1:8002", "The addr of the application.")
 	var namespace = flag.String("ns", "longhorn-system", "Kube namespace")
 	var service = flag.String("svc", "longhorn-frontend:longhorn-ui", "Kube service")
 	flag.Parse()

diff --git a/HttpProxy.go b/HttpProxy.go
@@ -53,14 +53,14 @@ type proxy struct {
 }
 
 func (p *proxy) ServeHTTP(wr http.ResponseWriter, req *http.Request) {
-	log.Println(req.URL.Scheme, " ", req.RemoteAddr, " ", req.Method, " ", req.URL)
+	log.Println(req.RemoteAddr, " ", req.Method, " ", req.URL)
 
-	if req.URL.Scheme != "http" && req.URL.Scheme != "https" {
+	/*if req.URL.Scheme != "http" && req.URL.Scheme != "https" {
 	  	msg := "unsupported protocal scheme "+req.URL.Scheme
 		http.Error(wr, msg, http.StatusBadRequest)
 		log.Println(msg)
 		return
-	}
+	}*/
 
 	client := &http.Client{}
 

diff --git a/HttpProxy.go b/HttpProxy.go
@@ -68,8 +68,14 @@ func (p *proxy) ServeHTTP(wr http.ResponseWriter, req *http.Request) {
 	//http://golang.org/src/pkg/net/http/client.go
 	req.RequestURI = ""
 
-	req.URL = url.Parse("http://127.0.0.1:8001/api/v1/namespaces/" + p.ns + "/services/" + p.svc + "/proxy" + req.URL.Path)
-
+	u, err := url.Parse("http://127.0.0.1:8001/api/v1/namespaces/" + p.ns + "/services/" + p.svc + "/proxy" + req.URL.Path)
+	if err != nil {
+		http.Error(wr, "Server Error", http.StatusInternalServerError)
+		log.Fatal("ServeHTTP:", err)
+	}
+
+	req.URL = u
+
 	delHopHeaders(req.Header)
 
 	if clientIP, _, err := net.SplitHostPort(req.RemoteAddr); err == nil {

diff --git a/HttpProxy.go b/HttpProxy.go
@@ -7,6 +7,7 @@ import (
 	"net"
 	"net/http"
 	"strings"
+	"net/url"
 )
 
 // Hop-by-hop headers. These are removed when sent to the backend.
@@ -52,7 +53,7 @@ type proxy struct {
 }
 
 func (p *proxy) ServeHTTP(wr http.ResponseWriter, req *http.Request) {
-	log.Println(req.RemoteAddr, " ", req.Method, " ", req.URL)
+	log.Println(req.URL.Scheme, " ", req.RemoteAddr, " ", req.Method, " ", req.URL)
 
 	if req.URL.Scheme != "http" && req.URL.Scheme != "https" {
 	  	msg := "unsupported protocal scheme "+req.URL.Scheme
@@ -67,7 +68,7 @@ func (p *proxy) ServeHTTP(wr http.ResponseWriter, req *http.Request) {
 	//http://golang.org/src/pkg/net/http/client.go
 	req.RequestURI = ""
 
-	req.URL.Path = "/api/v1/namespaces/" + p.ns + "/services/" + p.svc + "/proxy" + req.URL.Path
+	req.URL = url.Parse("http://127.0.0.1:8001/api/v1/namespaces/" + p.ns + "/services/" + p.svc + "/proxy" + req.URL.Path)
 
 	delHopHeaders(req.Header)
 

diff --git a/HttpProxy.go b/HttpProxy.go
@@ -47,6 +47,8 @@ func appendHostToXForwardHeader(header http.Header, host string) {
 }
 
 type proxy struct {
+	ns string
+	svc string
 }
 
 func (p *proxy) ServeHTTP(wr http.ResponseWriter, req *http.Request) {
@@ -64,6 +66,8 @@ func (p *proxy) ServeHTTP(wr http.ResponseWriter, req *http.Request) {
 	//http: Request.RequestURI can't be set in client requests.
 	//http://golang.org/src/pkg/net/http/client.go
 	req.RequestURI = ""
+
+	req.URL.Path = "/api/v1/namespaces/" + p.ns + "/services/" + p.svc + "/proxy" + req.URL.Path
 
 	delHopHeaders(req.Header)
 
@@ -88,10 +92,12 @@ func (p *proxy) ServeHTTP(wr http.ResponseWriter, req *http.Request) {
 }
 
 func main() {
-	var addr = flag.String("addr", "127.0.0.1:8080", "The addr of the application.")
+	var addr = flag.String("addr", "127.0.0.1:8001", "The addr of the application.")
+	var namespace = flag.String("ns", "longhorn-system", "Kube namespace")
+	var service = flag.String("svc", "longhorn-frontend:longhorn-ui", "Kube service")
 	flag.Parse()
 
-	handler := &proxy{}
+	handler := &proxy{ns: *namespace, svc: *service}
 
 	log.Println("Starting proxy server on", *addr)
 	if err := http.ListenAndServe(*addr, handler); err != nil {

diff --git a/HttpProxy.go b/HttpProxy.go
@@ -53,7 +53,7 @@ func (p *proxy) ServeHTTP(wr http.ResponseWriter, req *http.Request) {
 	log.Println(req.RemoteAddr, " ", req.Method, " ", req.URL)
 
 	if req.URL.Scheme != "http" && req.URL.Scheme != "https" {
-	  msg := "unsupported protocal scheme "+req.URL.Scheme
+	  	msg := "unsupported protocal scheme "+req.URL.Scheme
 		http.Error(wr, msg, http.StatusBadRequest)
 		log.Println(msg)
 		return

diff --git a/HttpProxy.go b/HttpProxy.go
@@ -88,7 +88,7 @@ func (p *proxy) ServeHTTP(wr http.ResponseWriter, req *http.Request) {
 }
 
 func main() {
-	var addr = flag.String("addr", ":8080", "The addr of the application.")
+	var addr = flag.String("addr", "127.0.0.1:8080", "The addr of the application.")
 	flag.Parse()
 
 	handler := &proxy{}

diff --git a/HttpProxy.go b/HttpProxy.go
@@ -6,7 +6,6 @@ import (
 	"log"
 	"net"
 	"net/http"
-	"net/url"
 	"strings"
 )
 

diff --git a/HttpProxy.go b/HttpProxy.go
@@ -0,0 +1,101 @@
+package main
+
+import (
+	"flag"
+	"io"
+	"log"
+	"net"
+	"net/http"
+	"net/url"
+	"strings"
+)
+
+// Hop-by-hop headers. These are removed when sent to the backend.
+// http://www.w3.org/Protocols/rfc2616/rfc2616-sec13.html
+var hopHeaders = []string{
+	"Connection",
+	"Keep-Alive",
+	"Proxy-Authenticate",
+	"Proxy-Authorization",
+	"Te", // canonicalized version of "TE"
+	"Trailers",
+	"Transfer-Encoding",
+	"Upgrade",
+}
+
+func copyHeader(dst, src http.Header) {
+	for k, vv := range src {
+		for _, v := range vv {
+			dst.Add(k, v)
+		}
+	}
+}
+
+func delHopHeaders(header http.Header) {
+	for _, h := range hopHeaders {
+		header.Del(h)
+	}
+}
+
+func appendHostToXForwardHeader(header http.Header, host string) {
+	// If we aren't the first proxy retain prior
+	// X-Forwarded-For information as a comma+space
+	// separated list and fold multiple headers into one.
+	if prior, ok := header["X-Forwarded-For"]; ok {
+		host = strings.Join(prior, ", ") + ", " + host
+	}
+	header.Set("X-Forwarded-For", host)
+}
+
+type proxy struct {
+}
+
+func (p *proxy) ServeHTTP(wr http.ResponseWriter, req *http.Request) {
+	log.Println(req.RemoteAddr, " ", req.Method, " ", req.URL)
+
+	if req.URL.Scheme != "http" && req.URL.Scheme != "https" {
+	  msg := "unsupported protocal scheme "+req.URL.Scheme
+		http.Error(wr, msg, http.StatusBadRequest)
+		log.Println(msg)
+		return
+	}
+
+	client := &http.Client{}
+
+	//http: Request.RequestURI can't be set in client requests.
+	//http://golang.org/src/pkg/net/http/client.go
+	req.RequestURI = ""
+
+	delHopHeaders(req.Header)
+
+	if clientIP, _, err := net.SplitHostPort(req.RemoteAddr); err == nil {
+		appendHostToXForwardHeader(req.Header, clientIP)
+	}
+
+	resp, err := client.Do(req)
+	if err != nil {
+		http.Error(wr, "Server Error", http.StatusInternalServerError)
+		log.Fatal("ServeHTTP:", err)
+	}
+	defer resp.Body.Close()
+
+	log.Println(req.RemoteAddr, " ", resp.Status)
+
+	delHopHeaders(resp.Header)
+
+	copyHeader(wr.Header(), resp.Header)
+	wr.WriteHeader(resp.StatusCode)
+	io.Copy(wr, resp.Body)
+}
+
+func main() {
+	var addr = flag.String("addr", ":8080", "The addr of the application.")
+	flag.Parse()
+
+	handler := &proxy{}
+
+	log.Println("Starting proxy server on", *addr)
+	if err := http.ListenAndServe(*addr, handler); err != nil {
+		log.Fatal("ListenAndServe:", err)
+	}
+}
No results found