Emulation/maryo
1
0
Fork 0
mirror of https://github.com/PretendoNetwork/maryo.git synced 2024-06-11 17:17:33 -04:00
Code Issues Packages Projects Releases Wiki Activity

updated a few things to avoid errors

Browse source
This commit is contained in:
superwhiskers 2018-03-21 00:12:37 -05:00
parent 87ee36060e
commit 2d380996ac
1 changed files with 37 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

42
proxy.go
View file

@ -89,7 +89,24 @@ func startProxy(configName string, logging bool) {
writeFile("maryo/proxy.log", fmt.Sprintf("-> got request to %s\n", r.URL.Host))
// get prettified request
reqData, err := httputil.DumpRequest(r, true)
// define these variables so no errors
var reqData []byte
var err error
// no errors for post requests
if r.Method == "POST" {
// if it is, then tell the dumper it is
reqData, err = httputil.DumpRequest(r, true)
} else {
// otherwise, don't
reqData, err = httputil.DumpRequest(r, false)
}
if err != nil {
// output error
@ -103,14 +120,14 @@ func startProxy(configName string, logging bool) {
// log the request data, then
fmt.Printf("\n-- request data\n")
fmt.Printf("%s", string(reqData))
fmt.Printf("%s", string(reqData[:]))
fmt.Printf("\n\n")
}
// always log to file
writeFile("maryo/proxy.log", fmt.Sprintf("-> request data to %s\n", r.URL.Host))
writeFile("maryo/proxy.log", fmt.Sprintf("%s", string(reqData)))
writeFile("maryo/proxy.log", fmt.Sprintf("%s", string(reqData[:])))
writeFile("maryo/proxy.log", fmt.Sprintf("\n\n"))
// attempt to proxy it to the servers listed in config
@ -165,7 +182,22 @@ func startProxy(configName string, logging bool) {
}
// dump response
fmtResp, err := httputil.DumpResponse(resp, true)
// make these variables earlier on so no errors
var fmtResp []byte
// check if the request is a post request
if r.Method == "POST" {
// if so, tell the dumper that it is one
fmtResp, err = httputil.DumpResponse(resp, true)
} else {
// otherwise, don't
fmtResp, err = httputil.DumpResponse(resp, false)
}
// error handling
if err != nil {
@ -181,7 +213,7 @@ func startProxy(configName string, logging bool) {
// log it if they do
fmt.Printf("\n-- response data\n")
fmt.Printf("%s\n", string(fmtResp))
fmt.Printf("%s\n", string(fmtResp[:]))
fmt.Printf("\n\n")
}