63 lines
1.2 KiB
Go
63 lines
1.2 KiB
Go
|
package http
|
||
|
|
|
||
|
|
import (
|
||
|
|
"time"
|
||
|
|
|
||
|
|
"github.com/fasthttp/router"
|
||
|
|
"github.com/valyala/fasthttp"
|
||
|
|
|
||
|
|
"github.com/yunginnanet/HellPot/config"
|
||
|
|
"github.com/yunginnanet/HellPot/heffalump"
|
||
|
|
)
|
||
|
|
|
||
|
|
func getRealRemote(ctx *fasthttp.RequestCtx) string {
|
||
|
|
xrealip := string(ctx.Request.Header.Peek("X-Real-IP"))
|
||
|
|
if len(xrealip) > 0 {
|
||
|
|
return xrealip
|
||
|
|
}
|
||
|
|
return ctx.RemoteIP().String()
|
||
|
|
}
|
||
|
|
|
||
|
|
func scopeCheck(ctx *fasthttp.RequestCtx) {
|
||
|
|
var inscope = false
|
||
|
|
reqpath := ctx.QueryArgs().Peek("path")
|
||
|
|
|
||
|
|
remoteAddr := getRealRemote(ctx)
|
||
|
|
|
||
|
|
slog := log.With().
|
||
|
|
Str("UserAgent", string(ctx.UserAgent())).
|
||
|
|
Str("REMOTE_ADDR", remoteAddr).
|
||
|
|
Interface("URL", string(ctx.RequestURI())).Logger()
|
||
|
|
|
||
|
|
for _, p := range config.Paths {
|
||
|
|
if string(reqpath) == p {
|
||
|
|
inscope = true
|
||
|
|
break
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
if !inscope {
|
||
|
|
slog.Warn().Msg("Request outside of configured scope!")
|
||
|
|
ctx.Error("", 404)
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
s := time.Now()
|
||
|
|
|
||
|
|
n := heffalump.DefaultHeffalump.WriteHell(ctx.Response.BodyWriter())
|
||
|
|
|
||
|
|
slog.Info().
|
||
|
|
Int64("BYTES", n).
|
||
|
|
Dur("DURATION", time.Since(s)).
|
||
|
|
Msg("FINISH")
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
// Serve starts our HTTP server and request router
|
||
|
|
func Serve() error {
|
||
|
|
r := router.New()
|
||
|
|
r.GET("/robots.txt", robotsTXT)
|
||
|
|
r.GET("/{path}", scopeCheck)
|
||
|
|
return fasthttp.ListenAndServe(":8080", r.Handler)
|
||
|
|
}
|