mirror of
https://github.com/yunginnanet/HellPot
synced 2024-06-30 17:50:53 +00:00
75 lines
1.6 KiB
Go
75 lines
1.6 KiB
Go
/*
|
|
Package heffalump attempts to encapsulate the original work by carlmjohnson on heffalump
|
|
https://github.com/carlmjohnson/heffalump
|
|
*/
|
|
package heffalump
|
|
|
|
import (
|
|
"bufio"
|
|
"io"
|
|
"sync"
|
|
|
|
"github.com/yunginnanet/HellPot/config"
|
|
)
|
|
|
|
var log = config.GetLogger()
|
|
|
|
// DefaultHeffalump represents a Heffalump type
|
|
var DefaultHeffalump = NewHeffalump(DefaultMarkovMap, 100*1<<10)
|
|
|
|
// Heffalump represents our buffer pool and markov map from Heffalump
|
|
// https://github.com/carlmjohnson/heffalump
|
|
type Heffalump struct {
|
|
pool sync.Pool
|
|
buffsize int
|
|
mm MarkovMap
|
|
}
|
|
|
|
// NewHeffalump instantiates a new Heffalump for markov generation and buffer/io operations
|
|
// https://github.com/carlmjohnson/heffalump
|
|
func NewHeffalump(mm MarkovMap, buffsize int) *Heffalump {
|
|
return &Heffalump{
|
|
pool: sync.Pool{},
|
|
buffsize: buffsize,
|
|
mm: mm,
|
|
}
|
|
}
|
|
|
|
func (h *Heffalump) getBuffer() []byte {
|
|
x := h.pool.Get()
|
|
if buf, ok := x.([]byte); ok {
|
|
return buf
|
|
}
|
|
return make([]byte, h.buffsize)
|
|
}
|
|
|
|
func (h *Heffalump) putBuffer(buf []byte) {
|
|
h.pool.Put(buf)
|
|
}
|
|
|
|
// WriteHell writes markov chain heffalump hell to the provided io.Writer
|
|
// https://github.com/carlmjohnson/heffalump
|
|
func (h *Heffalump) WriteHell(bw *bufio.Writer) (int64, error) {
|
|
var n int64
|
|
var err error
|
|
|
|
defer func() {
|
|
if r := recover(); r != nil {
|
|
log.Error().Interface("caller", r).Msg("panic recovered!")
|
|
}
|
|
}()
|
|
|
|
buf := h.getBuffer()
|
|
defer h.putBuffer(buf)
|
|
|
|
if _, err = io.WriteString(bw, "<HTML>\n<BODY>\n"); err != nil {
|
|
return n, err
|
|
}
|
|
|
|
if n, err = io.CopyBuffer(bw, h.mm, buf); err != nil {
|
|
return n, nil
|
|
}
|
|
|
|
return n, nil
|
|
}
|