You cannot select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

76 lines
1.5 KiB
Go

10 years ago
package main
import (
10 years ago
"flag"
10 years ago
"fmt"
10 years ago
"github.com/jordanorelli/moon/lib"
10 years ago
"io"
"net/http"
"net/http/httputil"
"os"
"strings"
"unicode"
)
var client = new(http.Client)
func httpHandler(w http.ResponseWriter, r *http.Request) {
10 years ago
fmt.Println("from:", r.RemoteAddr)
10 years ago
b, err := httputil.DumpRequest(r, true)
if err != nil {
fmt.Printf("error dumping request: %s\n", err)
return
}
os.Stdout.Write(b)
r.RequestURI = ""
r.URL.Scheme = strings.Map(unicode.ToLower, r.URL.Scheme)
res, err := client.Do(r)
if err != nil {
fmt.Printf("error forwarding request: %s\n", err)
return
}
defer res.Body.Close()
for k, v := range res.Header {
w.Header()[k] = v
}
10 years ago
if _, ok := w.Header()["Proxy-Connection"]; ok {
delete(w.Header(), "Proxy-Connection")
}
10 years ago
w.WriteHeader(res.StatusCode)
if _, err := io.Copy(w, res.Body); err != nil {
fmt.Printf("error copying body: %s\n", err)
}
}
10 years ago
func bail(status int, t string, args ...interface{}) {
if status == 0 {
fmt.Fprintf(os.Stdout, t+"\n", args...)
} else {
fmt.Fprintf(os.Stderr, t+"\n", args...)
}
os.Exit(status)
}
10 years ago
func main() {
10 years ago
var configPath string
flag.StringVar(&configPath, "config", "./prox_config.moon", "path to configuration file")
flag.Parse()
conf, err := moon.ReadFile(configPath)
if err != nil {
bail(1, "unable to read config: %s", err)
}
var addr string
if err := conf.Get("proxy_addr", &addr); err != nil {
bail(1, "error reading proxy_addr from config: %s", err)
}
10 years ago
http.HandleFunc("/", httpHandler)
10 years ago
http.ListenAndServe(addr, nil)
10 years ago
}