455 lines
12 KiB
Go
455 lines
12 KiB
Go
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"regexp"
|
|
"strings"
|
|
"time"
|
|
|
|
"github.com/davecgh/go-spew/spew"
|
|
"github.com/eleksir/go-xmpp"
|
|
log "github.com/sirupsen/logrus"
|
|
)
|
|
|
|
// bunyPresense производит проверку по бело-чёрным спискам. Если presence пришёл от злодея (из чёрного списка), то
|
|
// отправляет его в бан.
|
|
func bunyPresense(v xmpp.Presence) error { //nolint:gocognit,gocyclo
|
|
var err error
|
|
|
|
// Если у presence-а есть JID и presence из одной из комнат, в которой мы есть и если его домен в чёрном
|
|
// списке, заносим пидора в список outcast-ов
|
|
if v.JID != "" {
|
|
// На всякий случай: себя никогда не баним, явным образом
|
|
if v.JID == talk.JID() {
|
|
return err
|
|
}
|
|
|
|
room := strings.SplitN(v.From, "/", 2)[0]
|
|
|
|
if room == "" {
|
|
log.Infof("We got empty room field in presence event, which kinda strange: %s", spew.Sdump(v))
|
|
|
|
return err
|
|
}
|
|
|
|
goodJid := strings.SplitN(v.JID, "/", 2)[0]
|
|
|
|
// Обрабатываем правила белого списка
|
|
for _, good := range whiteList.Whitelist {
|
|
// Глобальный белый список
|
|
if good.RoomName == "" {
|
|
for _, person := range good.Jid {
|
|
if person == goodJid {
|
|
return err
|
|
}
|
|
}
|
|
}
|
|
|
|
// Белый список для конкретных каналов
|
|
if good.RoomName == room {
|
|
for _, person := range good.Jid {
|
|
if person == goodJid {
|
|
return err
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
evilJid := strings.SplitN(v.JID, "/", 2)[0]
|
|
evilNick := ""
|
|
evilNicks := strings.SplitN(v.From, "/", 2)
|
|
|
|
// Давайте-ка похэндлим нонсенс - когда у нас в строке нету разделителя
|
|
if len(evilNicks) > 1 {
|
|
evilNick = evilNicks[1]
|
|
}
|
|
|
|
// Обрабатываем правила чёрного списка
|
|
for _, cRoom := range roomsConnected {
|
|
if cRoom == room {
|
|
for _, bEntry := range blackList.Blacklist {
|
|
// Обработаем правила глобального чёрного списка
|
|
if bEntry.RoomName == "" {
|
|
for _, jidRegexp := range bEntry.JidRe {
|
|
if jidRegexp == "" {
|
|
continue
|
|
}
|
|
|
|
re, err := regexp.Compile(jidRegexp)
|
|
|
|
if err != nil {
|
|
log.Errorf("Incorrect regexp in global blacklist: %s, skipping", jidRegexp)
|
|
|
|
continue
|
|
}
|
|
|
|
log.Debugf("Checking jid %s vs global blacklist regex %s", evilJid, jidRegexp)
|
|
|
|
if re.MatchString(evilJid) {
|
|
log.Warnf(
|
|
"Hammer falls on %s (%s): jid matches with global blacklist entry: %s",
|
|
v.From,
|
|
evilJid,
|
|
jidRegexp,
|
|
)
|
|
|
|
id, err := squash(room, evilJid, bEntry.ReasonEnable, v.Type)
|
|
|
|
if err != nil {
|
|
err := fmt.Errorf(
|
|
"unable to ban user: id=%s, err=%w",
|
|
id,
|
|
err,
|
|
)
|
|
|
|
gTomb.Kill(err)
|
|
}
|
|
|
|
return err
|
|
}
|
|
}
|
|
|
|
for _, nickRegexp := range bEntry.NickRe {
|
|
if nickRegexp == "" {
|
|
continue
|
|
}
|
|
|
|
re, err := regexp.Compile(nickRegexp)
|
|
|
|
if err != nil {
|
|
log.Errorf("Incorrect regexp in global blacklist: %s, skipping", nickRegexp)
|
|
|
|
continue
|
|
}
|
|
|
|
log.Debugf("Checking nick %s vs global blacklist regex %s", evilNick, nickRegexp)
|
|
|
|
if re.MatchString(evilNick) {
|
|
log.Warnf(
|
|
"Hammer falls on %s (%s): nick matches with global blacklist entry: %s",
|
|
v.From,
|
|
evilJid,
|
|
nickRegexp,
|
|
)
|
|
|
|
// Баним именно jid
|
|
id, err := squash(room, evilJid, bEntry.ReasonEnable, v.Type)
|
|
|
|
if err != nil {
|
|
err := fmt.Errorf(
|
|
"unable to ban user: id=%s, err=%w",
|
|
id,
|
|
err,
|
|
)
|
|
|
|
gTomb.Kill(err)
|
|
}
|
|
|
|
return err
|
|
}
|
|
}
|
|
|
|
continue
|
|
}
|
|
|
|
// Обработаем правила конкретного канала, комнаты конференции
|
|
if bEntry.RoomName == room {
|
|
for _, jidRegexp := range bEntry.JidRe {
|
|
if jidRegexp == "" {
|
|
continue
|
|
}
|
|
|
|
re, err := regexp.Compile(jidRegexp)
|
|
|
|
if err != nil {
|
|
log.Errorf("Incorrect regexp in room %s blacklist: %s, skipping", room, jidRegexp)
|
|
|
|
continue
|
|
}
|
|
|
|
log.Debugf("Checking jid %s vs room %s blacklist regex %s", evilJid, room, jidRegexp)
|
|
|
|
if re.MatchString(evilJid) {
|
|
log.Warnf(
|
|
"Hammer falls on %s (%s): jid matches with room blacklist entry: %s",
|
|
v.From,
|
|
evilJid,
|
|
jidRegexp,
|
|
)
|
|
|
|
id, err := squash(room, evilJid, bEntry.ReasonEnable, v.Type)
|
|
|
|
if err != nil {
|
|
err := fmt.Errorf(
|
|
"unable to ban user: id=%s, err=%w",
|
|
id,
|
|
err,
|
|
)
|
|
|
|
gTomb.Kill(err)
|
|
}
|
|
|
|
return err
|
|
}
|
|
}
|
|
|
|
if bEntry.RoomName == room {
|
|
for _, nickRegexp := range bEntry.NickRe {
|
|
if nickRegexp == "" {
|
|
continue
|
|
}
|
|
|
|
re, err := regexp.Compile(nickRegexp)
|
|
|
|
if err != nil {
|
|
log.Errorf("Incorrect regexp in room %s blacklist: %s, skipping", room, nickRegexp)
|
|
|
|
continue
|
|
}
|
|
|
|
log.Debugf("Checking nick %s vs room %s blacklist regex %s", evilJid, room, nickRegexp)
|
|
|
|
if re.MatchString(evilNick) {
|
|
log.Warnf(
|
|
"Hammer falls on %s (%s): nick matches with room blacklist entry: %s",
|
|
v.From,
|
|
evilJid,
|
|
nickRegexp,
|
|
)
|
|
|
|
// Баним именно jid
|
|
id, err := squash(room, evilJid, bEntry.ReasonEnable, v.Type)
|
|
|
|
if err != nil {
|
|
err := fmt.Errorf(
|
|
"unable to ban user: id=%s, err=%w",
|
|
id,
|
|
err,
|
|
)
|
|
|
|
gTomb.Kill(err)
|
|
}
|
|
|
|
return err
|
|
}
|
|
}
|
|
}
|
|
|
|
continue
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
return err
|
|
}
|
|
|
|
// bunyChat производит проверку сообщений участников чата по списку забаненных фраз и в случае нахождения запрещённого
|
|
// шаблона банит участника чата.
|
|
func bunyChat(v xmpp.Chat) error {
|
|
var (
|
|
room = (strings.SplitN(v.Remote, "/", 2))[0]
|
|
// nick = (strings.SplitN(v.Remote, "/", 2))[1]
|
|
err error
|
|
)
|
|
|
|
// Действовать мы можем только в рамках тех комнат, где явно присуствуем.
|
|
for _, cRoom := range roomsConnected {
|
|
if cRoom == room {
|
|
// Перебирём правила чёрных списков.
|
|
for _, bEntry := range blackList.Blacklist {
|
|
// Обработаем правила глобального чёрного списка
|
|
if bEntry.RoomName == "" {
|
|
for _, phraseRegexp := range bEntry.PhraseRe {
|
|
if phraseRegexp == "" {
|
|
continue
|
|
}
|
|
|
|
re, err := regexp.Compile(phraseRegexp)
|
|
|
|
if err != nil {
|
|
log.Errorf("Incorrect regexp in room %s blacklist: %s, skipping", room, phraseRegexp)
|
|
|
|
continue
|
|
}
|
|
|
|
log.Debugf("Checking phrase %s vs room %s blacklist regex %s", v.Text, room, phraseRegexp)
|
|
|
|
if re.MatchString(v.Text) {
|
|
realJID := getRealJIDfromNick(v.Remote)
|
|
|
|
log.Warnf(
|
|
"Hammer falls on %s (%s): phrase matches with global blacklist entry: %s vs %s",
|
|
v.Remote,
|
|
realJID,
|
|
v.Text,
|
|
phraseRegexp,
|
|
)
|
|
|
|
if id, err := squash(room, realJID, bEntry.ReasonEnable, v.Type); err != nil {
|
|
err := fmt.Errorf(
|
|
"unable to ban user: id=%s, err=%w",
|
|
id,
|
|
err,
|
|
)
|
|
|
|
gTomb.Kill(err)
|
|
}
|
|
|
|
return err
|
|
}
|
|
}
|
|
}
|
|
|
|
// Обработаем правила чёрного списка для конкретной комнаты
|
|
if bEntry.RoomName == room {
|
|
for _, phraseRegexp := range bEntry.PhraseRe {
|
|
if phraseRegexp == "" {
|
|
continue
|
|
}
|
|
|
|
re, err := regexp.Compile(phraseRegexp)
|
|
|
|
if err != nil {
|
|
log.Errorf("Incorrect regexp in room %s blacklist: %s, skipping", room, phraseRegexp)
|
|
|
|
continue
|
|
}
|
|
|
|
log.Debugf("Checking phrase %s vs room %s blacklist regex %s", v.Text, room, phraseRegexp)
|
|
|
|
if re.MatchString(v.Text) {
|
|
realJID := getRealJIDfromNick(v.Remote)
|
|
|
|
log.Warnf(
|
|
"Hammer falls on %s (%s): phrase matches with room blacklist entry: %s vs %s",
|
|
v.Remote,
|
|
realJID,
|
|
v.Text,
|
|
phraseRegexp,
|
|
)
|
|
|
|
if id, err := squash(room, realJID, bEntry.ReasonEnable, v.Type); err != nil {
|
|
err := fmt.Errorf(
|
|
"unable to ban user: id=%s, err=%w",
|
|
id,
|
|
err,
|
|
)
|
|
|
|
gTomb.Kill(err)
|
|
}
|
|
|
|
return err
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
// Если включено, проверяем фразу на КАПС.
|
|
for _, channel := range config.Jabber.Channels {
|
|
if channel.Name == room && channel.AllCaps.Enabled {
|
|
// Нормализуем строку и вырежем из неё пробелы
|
|
normPhrase := strings.ReplaceAll(nString(v.Text), " ", "")
|
|
|
|
// Проверяем согласно тому, что длина фразы более чем сколько-то символов
|
|
if len(normPhrase) >= channel.AllCaps.MinLength {
|
|
normPhraseUpper := strings.ReplaceAll(nStringUpper(v.Text), " ", "")
|
|
|
|
if normPhrase == normPhraseUpper {
|
|
realJID := getRealJIDfromNick(v.Remote)
|
|
id, err := squash(room, realJID, false, v.Type)
|
|
|
|
if err != nil {
|
|
err := fmt.Errorf(
|
|
"unable to ban user: id=%s, err=%w",
|
|
id,
|
|
err,
|
|
)
|
|
|
|
gTomb.Kill(err)
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
break
|
|
}
|
|
}
|
|
|
|
return err
|
|
}
|
|
|
|
// squash банит указанный jid в указанной комнате.
|
|
// reasonEnable указывает, надо ли писать дату автобана в банлисте в поле reason (это единственная причина, в которую
|
|
// умеет бот).
|
|
func squash(room, jid string, reasonEnable bool, vType string) (string, error) {
|
|
var (
|
|
id string
|
|
err error
|
|
)
|
|
|
|
if config.Jabber.BanPhrasesEnable {
|
|
phrase := randomPhrase(config.Jabber.BanPhrases)
|
|
|
|
if _, err = talk.Send(
|
|
xmpp.Chat{ //nolint:exhaustruct
|
|
Remote: room,
|
|
Text: phrase,
|
|
Type: vType,
|
|
},
|
|
); err != nil {
|
|
err = fmt.Errorf("unable to send phrase to room %s: %w", room, err)
|
|
|
|
// Здесь возвращаем nil, т.к. за нас ошибку залоггирует код выше
|
|
return id, err
|
|
}
|
|
}
|
|
|
|
// https://xmpp.org/extensions/xep-0045.html#ban баним вот таким сообщением
|
|
ban := "<item affiliation='outcast' jid='" + jid + "'>"
|
|
|
|
if reasonEnable {
|
|
var t = time.Now()
|
|
ban += fmt.Sprintf(
|
|
"<reason>autoban at %04d.%02d.%02d %02d:%02d:%02d</reason>",
|
|
t.Year(),
|
|
t.Month(),
|
|
t.Day(),
|
|
t.Hour(),
|
|
t.Minute(),
|
|
t.Second(),
|
|
)
|
|
} else {
|
|
ban += "<reason />"
|
|
}
|
|
|
|
ban += "</item>"
|
|
|
|
// Выжидаем некоторое время перед баном. А то можно настолько рано забанить, что сервер не внесёт злодея в банлист
|
|
// комнаты и пришлёт affiliation: none вместо affiliation: outcast.
|
|
if config.Jabber.BanDelay > 0 {
|
|
time.Sleep(time.Duration(config.Jabber.BanDelay) * time.Millisecond)
|
|
}
|
|
|
|
if id, err = talk.RawInformationQuery(
|
|
talk.JID(),
|
|
room,
|
|
"ban1",
|
|
xmpp.IQTypeSet,
|
|
"http://jabber.org/protocol/muc#admin",
|
|
ban,
|
|
); err != nil {
|
|
err = fmt.Errorf(
|
|
"unable to ban user: id=%s, err=%w",
|
|
id,
|
|
err,
|
|
)
|
|
}
|
|
|
|
return id, err
|
|
}
|
|
|
|
/* vim: set ft=go noet ai ts=4 sw=4 sts=4: */
|