Handler: Make handler configurable

heff/http.go

@@ -7,22 +7,25 @@ import (
 	"sync"
 )
 
+var DefaultHoneypot = NewHoneypot(DefaultMarkovMap, 100*1<<10)
+
+func NewHoneypot(mm MarkovMap, buffsize int) http.HandlerFunc {
 	var pool sync.Pool
 
-func getBuffer() []byte {
+	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)
@@ -30,3 +33,4 @@ func Honeypot(w http.ResponseWriter, r *http.Request) {
 		n, err := io.CopyBuffer(w, DefaultMarkovMap, buf)
 		log.Printf("Wrote: %d (%v)", n, err)
 	}
+}

heffalump.go

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