72 lines
1.4 KiB
Go
72 lines
1.4 KiB
Go
package srnd
|
|
|
|
import (
|
|
"bufio"
|
|
"errors"
|
|
"fmt"
|
|
"io"
|
|
"log"
|
|
"net"
|
|
"os/user"
|
|
"strings"
|
|
)
|
|
|
|
var ErrSpamFilterNotEnabled = errors.New("spam filter access attempted when disabled")
|
|
var ErrSpamFilterFailed = errors.New("spam filter failed")
|
|
var ErrMessageIsSpam = errors.New("message is spam")
|
|
|
|
type SpamFilter struct {
|
|
addr string
|
|
enabled bool
|
|
}
|
|
|
|
func (sp *SpamFilter) Configure(c SpamConfig) {
|
|
sp.enabled = c.enabled
|
|
sp.addr = c.addr
|
|
}
|
|
|
|
func (sp *SpamFilter) Enabled(newsgroup string) bool {
|
|
return sp.enabled && newsgroup != "ctl"
|
|
}
|
|
|
|
func (sp *SpamFilter) Rewrite(msg io.Reader, out io.WriteCloser, group string) error {
|
|
var buff [65636]byte
|
|
if !sp.Enabled(group) {
|
|
return ErrSpamFilterNotEnabled
|
|
}
|
|
addr, err := net.ResolveTCPAddr("tcp", sp.addr)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
c, err := net.DialTCP("tcp", nil, addr)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
u, err := user.Current()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
fmt.Fprintf(c, "PROCESS SPAMC/1.5\r\nUser: %s\r\n\r\n", u.Username)
|
|
io.CopyBuffer(c, msg, buff[:])
|
|
c.CloseWrite()
|
|
r := bufio.NewReader(c)
|
|
for {
|
|
l, err := r.ReadString(10)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
l = strings.TrimSpace(l)
|
|
if strings.HasPrefix(l, "Spam: True ") {
|
|
return ErrMessageIsSpam
|
|
}
|
|
log.Println("SpamFilter:", l)
|
|
if l == "" {
|
|
_, err = io.CopyBuffer(out, r, buff[:])
|
|
c.Close()
|
|
out.Close()
|
|
return err
|
|
}
|
|
}
|
|
return ErrSpamFilterFailed
|
|
}
|