mirror of
https://github.com/Sean-Der/fail2rest.git
synced 2024-12-22 13:42:17 +01:00
39 lines
694 B
Go
39 lines
694 B
Go
package main
|
|
|
|
import (
|
|
"bytes"
|
|
"errors"
|
|
"github.com/kisielk/og-rek"
|
|
"net"
|
|
)
|
|
|
|
func fail2banRequest(input []string) (interface{}, error) {
|
|
c, err := net.Dial("unix", "/var/run/fail2ban/fail2ban.sock")
|
|
|
|
if err != nil {
|
|
return nil, errors.New("Failed to contact fail2ban socket")
|
|
}
|
|
|
|
p := &bytes.Buffer{}
|
|
ogórek.NewEncoder(p).Encode(input)
|
|
c.Write(p.Bytes())
|
|
c.Write([]byte("<F2B_END_COMMAND>"))
|
|
|
|
buf := make([]byte, 0)
|
|
tmpBuf := make([]byte, 1)
|
|
for {
|
|
bufRead, _ := c.Read(tmpBuf)
|
|
|
|
if bufRead != 0 {
|
|
buf = append(buf, tmpBuf...)
|
|
} else {
|
|
buf = buf[:len(buf)-17]
|
|
break
|
|
}
|
|
|
|
}
|
|
|
|
dec := ogórek.NewDecoder(bytes.NewBuffer(buf))
|
|
v, err := dec.Decode()
|
|
return v, err
|
|
}
|