mirror of
https://github.com/yunginnanet/HellPot
synced 2024-06-30 17:50:53 +00:00
39 lines
833 B
Go
39 lines
833 B
Go
package heff
|
|
|
|
import (
|
|
"io"
|
|
"log"
|
|
"net/http"
|
|
"sync"
|
|
)
|
|
|
|
// DefaultHoneypot is an http.HandlerFunc that serves random HTML from the
|
|
// DefaultMarkovMap, 100KB at a time.
|
|
var DefaultHoneypot = NewHoneypot(DefaultMarkovMap, 100*1<<10)
|
|
|
|
// NewHoneypot creates an http.HandlerFunc from a MarkovMap
|
|
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
|
|
}
|
|
return make([]byte, buffsize)
|
|
}
|
|
|
|
putBuffer := func(buf []byte) {
|
|
pool.Put(buf)
|
|
}
|
|
|
|
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, mm, buf)
|
|
log.Printf("Wrote: %d (%v)", n, err)
|
|
}
|
|
}
|