yunginnanet
/
HellPot
mirror of https://github.com/yunginnanet/HellPot.git
1
1
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Handler: Make handler configurable

This commit is contained in:
Carl Johnson 2016-12-11 18:18:38 -05:00
parent 6a94b58990
commit b5b634b579
2 changed files with 25 additions and 21 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
heff/http.go
View File

@ -7,26 +7,30 @@ import (
"sync"
)
var pool sync.Pool
var DefaultHoneypot = NewHoneypot(DefaultMarkovMap, 100*1<<10)
func getBuffer() []byte {
func NewHoneypot(mm MarkovMap, buffsize int) http.HandlerFunc {
var pool sync.Pool
getBuffer := func() []byte {
x := pool.Get()
if buf, ok := x.([]byte); ok {
return buf
} else {
return make([]byte, 100*1<<10)
return make([]byte, buffsize)
}
}
}
func putBuffer(buf []byte) {
putBuffer := func(buf []byte) {
pool.Put(buf)
}
}
func Honeypot(w http.ResponseWriter, r *http.Request) {
return func(w http.ResponseWriter, r *http.Request) {
log.Printf("Started writing: %v", r.URL)
buf := getBuffer()
defer putBuffer(buf)
io.WriteString(w, "<HTML>\n<BODY>\n")
n, err := io.CopyBuffer(w, DefaultMarkovMap, buf)
log.Printf("Wrote: %d (%v)", n, err)
}
}

2
heffalump.go
View File

@ -37,7 +37,7 @@ func main() {
path = "/"
}
http.HandleFunc(path, heff.Honeypot)
http.HandleFunc(path, heff.DefaultHoneypot)
log.Fatal(http.ListenAndServe(addr, nil))