1
4
mirror of https://github.com/yunginnanet/HellPot synced 2024-06-25 15:28:01 +00:00

Add documentation

This commit is contained in:
Carl Johnson 2016-12-11 19:15:14 -05:00
parent 77796bd6b3
commit cfeee439d2
3 changed files with 12 additions and 2 deletions

@ -7,8 +7,11 @@ import (
"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
@ -16,9 +19,8 @@ func NewHoneypot(mm MarkovMap, buffsize int) http.HandlerFunc {
x := pool.Get()
if buf, ok := x.([]byte); ok {
return buf
} else {
return make([]byte, buffsize)
}
return make([]byte, buffsize)
}
putBuffer := func(buf []byte) {

@ -53,16 +53,20 @@ func ScanHTML(data []byte, atEOF bool) (advance int, token []byte, err error) {
type tokenPair [2]string
// DefaultMarkovMap is a Markov chain based on Src.
var DefaultMarkovMap = MakeMarkovMap(strings.NewReader(Src))
// MarkovMap is a map that acts as a Markov chain generator.
type MarkovMap map[tokenPair][]string
// MakeMarkovMap makes an empty MakeMarkov and fills it with r.
func MakeMarkovMap(r io.Reader) MarkovMap {
m := MarkovMap{}
m.Fill(r)
return m
}
// Fill adds all the tokens in r to a MarkovMap
func (mm MarkovMap) Fill(r io.Reader) {
var w1, w2, w3 string
@ -77,11 +81,13 @@ func (mm MarkovMap) Fill(r io.Reader) {
mm.Add(w1, w2, w3)
}
// Add adds a three token sequence to the map.
func (mm MarkovMap) Add(w1, w2, w3 string) {
p := tokenPair{w1, w2}
mm[p] = append(mm[p], w3)
}
// Get psuedo-randomly chooses a possible suffix to w1 and w2.
func (mm MarkovMap) Get(w1, w2 string) string {
p := tokenPair{w1, w2}
suffix, ok := mm[p]
@ -93,6 +99,7 @@ func (mm MarkovMap) Get(w1, w2 string) string {
return suffix[r]
}
// Read fills p with data from calling Get on the MarkovMap.
func (mm MarkovMap) Read(p []byte) (n int, err error) {
var w1, w2, w3 string

@ -1,5 +1,6 @@
package heff
// Src is A. A. Milne's "Once on a Time" as transcribed by Project Gutenberg.
const Src = `<H1 ALIGN="center">
ONCE ON A TIME
</H1>