From 3b6dd3fdef2c27b3cc4d81bca1692e5237c462e4 Mon Sep 17 00:00:00 2001 From: eleksir Date: Fri, 19 Apr 2024 21:29:20 +0300 Subject: [PATCH] Do not write data to dead socket --- event_parser.go | 27 ++++++++++-------- main.go | 68 ++++++++++++++++++++++----------------------- read_config.go | 10 +++---- settings-db-util.go | 4 +-- 4 files changed, 56 insertions(+), 53 deletions(-) diff --git a/event_parser.go b/event_parser.go index 919c75b..b70a7c8 100644 --- a/event_parser.go +++ b/event_parser.go @@ -54,7 +54,7 @@ func parseEvent(e interface{}) { //nolint:maintidx,gocognit,gocyclo lastMucActivity.Set(muc, lastServerActivity) } - // Приватный чятик + // Приватный чятик. case "chat": // Здесь у нас может быть 2 вида From: // slackware-current@conference.jabber.ru/eleksir - это если сообщение из публичного чятика @@ -70,7 +70,7 @@ func parseEvent(e interface{}) { //nolint:maintidx,gocognit,gocyclo lastActivity = lastServerActivity // Внезапно, ошибки. По идее они должны ассоциироваться с отправляемыми сообщениями, но по ходу это - // не реализовано, поэтому мы получаем ошибки в форме отдельного чятика + // не реализовано, поэтому мы получаем ошибки в форме отдельного чятика. case "error": // Это ошибка со стороны сервера, видимо, в ответ на наши действия и поскольку такие вещи происходят // асинхронно, связать их с конкретными нашими действиями довольно сложно. @@ -80,7 +80,7 @@ func parseEvent(e interface{}) { //nolint:maintidx,gocognit,gocyclo // IQ-сообщение - пинг, понг, что-то ещё... case xmpp.IQ: - // По правилам IQ обязательно должна содержать ID, причём не пустой + // По правилам IQ обязательно должна содержать ID, причём не пустой. if v.ID == "" { log.Info("Got an IQ stanza with empty id, discarding") @@ -363,9 +363,9 @@ func parseEvent(e interface{}) { //nolint:maintidx,gocognit,gocyclo } } + // Поскольку никакой логики у нас на этот счёт не предусмотрено, то просто пропускаем ответ. if mucNickMatch { log.Debugf("Got 2-nd stage MUC pong answer (xep-0410) from %s to %s", v.From, v.To) - // Поскольку никакой логики у нас на этот счёт не предусмотрено, то просто пропускаем ответ } else { log.Infof("Got unknown pong from %s to %s", v.From, v.To) log.Debug(spew.Sdump(e)) @@ -376,7 +376,7 @@ func parseEvent(e interface{}) { //nolint:maintidx,gocognit,gocyclo log.Debug(spew.Sdump(e)) } - // Этот бот не управляется со стороны сервера, поэтому все попытки порулить игнорируем + // Этот бот не управляется со стороны сервера, поэтому все попытки порулить игнорируем. case xmpp.IQTypeSet: if muc, _ := strings.CutSuffix(v.To, "/"); muc != "" { if slices.Contains(roomsConnected, muc) { @@ -404,7 +404,7 @@ func parseEvent(e interface{}) { //nolint:maintidx,gocognit,gocyclo log.Debug(spew.Sdump(e)) - // Нам прилетело сообщение об ошибке + // Нам прилетело сообщение об ошибке. case xmpp.IQTypeError: // Если сервер не хочет пинговаться и отвечает ошибкой на пинг, то наверно он не умеет в пинги, // хотя если мы его пингуем, значит он анонсировал такой capability. Вот, засранец! @@ -432,7 +432,7 @@ func parseEvent(e interface{}) { //nolint:maintidx,gocognit,gocyclo parseError = false } - // Это у нас пинг xep-0410 и мы не в комнате, предполагается, что надо бы заджойниться + // Это у нас пинг xep-0410 и мы не в комнате, предполагается, что надо бы заджойниться. if parseError { var iqErrorCancelNotAcceptable jabberIqErrorCancelNotAcceptable @@ -484,7 +484,7 @@ func parseEvent(e interface{}) { //nolint:maintidx,gocognit,gocyclo log.Error("Unhandled IQ Error message") log.Error(spew.Sdump(e)) } - // Нам прилетело что-то неизвестное из семейства IQ stanza + // Нам прилетело что-то неизвестное из семейства IQ stanza. default: log.Info("Got an unknown IQ request. Dunno how deal with it, discarding") log.Info(spew.Sdump(e)) @@ -505,7 +505,7 @@ func parseEvent(e interface{}) { //nolint:maintidx,gocognit,gocyclo // клиента забанят из-за спама "вошёл-вышел". // Однако, это не значит, что такую ситуацию мы не должны корректным образом обрабатывать. if v.Type == "unavailable" { - // Считаем, что мы больше не в комнате, поэтому не знаем, кто там есть + // Считаем, что мы больше не в комнате, поэтому не знаем, кто там есть. roomPresences.Delete(v.From) log.Error("Presence notification - looks like another instance of client leaves room") @@ -548,6 +548,7 @@ func parseEvent(e interface{}) { //nolint:maintidx,gocognit,gocyclo case "none": if presenceJSONInterface, present := roomPresences.Get(room); present { presenceJSONStrings := interfaceToStringSlice(presenceJSONInterface) + var newPresenceJSONStrings []string for _, presenceJSONstring := range presenceJSONStrings { @@ -565,9 +566,11 @@ func parseEvent(e interface{}) { //nolint:maintidx,gocognit,gocyclo } // Участник пришёл default: - var presenceJSONStrings []string - var newPresenceJSONStrings []string - var presenceJSONBytes []byte + var ( + presenceJSONStrings []string + newPresenceJSONStrings []string + presenceJSONBytes []byte + ) if presenceJSONInterface, present := roomPresences.Get(room); present { presenceJSONStrings = interfaceToStringSlice(presenceJSONInterface) diff --git a/main.go b/main.go index a097e63..045f97b 100644 --- a/main.go +++ b/main.go @@ -26,7 +26,7 @@ func init() { TimestampFormat: "2006-01-02 15:04:05", }) - // Добавим динамически формируемый путь до конфига в список возможных положений конфига в системе + // Добавим динамически формируемый путь до конфига в список возможных положений конфига в системе. executablePath, err := os.Executable() if err != nil { @@ -84,14 +84,14 @@ func main() { verboseClient = true } - // github.com/mattn/go-xmpp пишет в stdio, нам этого не надо, ловим выхлоп его в logrus с уровнем trace + // github.com/mattn/go-xmpp пишет в stdio, нам этого не надо, ловим выхлоп его в logrus с уровнем trace. xmpp.DebugWriter = log.WithFields(log.Fields{"logger": "stdlib"}).WriterLevel(log.TraceLevel) - // Хэндлер сигналов не надо трогать, он нужен для завершения программы целиком + // Хэндлер сигналов не надо трогать, он нужен для завершения программы целиком. go sigHandler() signal.Notify(sigChan, os.Interrupt) - // Иницализируем redis-клиента + // Иницализируем redis-клиента. redisClient = redis.NewClient(&redis.Options{ //nolint:exhaustruct Addr: fmt.Sprintf("%s:%d", config.Redis.Server, config.Redis.Port), }) @@ -121,10 +121,10 @@ func main() { DialTimeout: time.Duration(config.Jabber.ConnectionTimeout) * time.Second, } - // Через tomb попробуем сделать выход горутинок управляемым + // Через tomb попробуем сделать выход горутинок управляемым. gTomb = tomb.Tomb{} - // Устанавливаем соединение и гребём события, посылаемые сервером + // Устанавливаем соединение и гребём события, посылаемые сервером. myLoop() log.Error(gTomb.Wait()) @@ -132,6 +132,7 @@ func main() { // Если у нас wire error, то вызов .Close() повлечёт за собой ошибку, но мы вынуждены звать .Close(), чтоб // закрыть tls контекст и почистить всё что связанно с прерванным соединением. log.Infoln("Closing connection to jabber server") + _ = talk.Close() time.Sleep(time.Duration(config.Jabber.ReconnectDelay) * time.Second) @@ -143,10 +144,11 @@ func myLoop() { select { case <-gTomb.Dying(): gTomb.Done() + return default: - // Зададим начальное значение глобальным переменным + // Зададим начальное значение глобальным переменным. serverPingTimestampRx = 0 serverPingTimestampTx = 0 roomsConnected = make([]string, 1) @@ -160,65 +162,63 @@ func myLoop() { serverPingTimestampRx = 0 roomPresences = NewCollection() - // Установим коннект + // Установим коннект. establishConnection() - serverPingTimestampRx = time.Now().Unix() // Считаем, что если коннект запустился, то первый пинг успешен + serverPingTimestampRx = time.Now().Unix() // Считаем, что если коннект запустился, то первый пинг успешен. - // Тыкаем сервер палочкой, проверяем, что коннект жив и переустанавливаем его, если он не жив + // Тыкаем сервер палочкой, проверяем, что коннект жив и переустанавливаем его, если он не жив. go probeServerLiveness() - // Тыкаем muc-и палочкой, проверяем, что они живы и пере-заходим в них, если пинги пропали + // Тыкаем muc-и палочкой, проверяем, что они живы и пере-заходим в них, если пинги пропали. go probeMUCLiveness() // Гребём сообщения... for { - // Стриггерилось завершение работы приложения, или соединение не установлено (порвалось, например) - // грести не надо + // Стриггерилось завершение работы приложения, или соединение не установлено (порвалось, например). + // грести не надо. if shutdown { break } if !isConnected { + // Tight loop - это наверно не очень хорошо, думаю, ничего страшного не будет, если мы поспим 100мс. + time.Sleep(100 * time.Millisecond) + continue } + // Вынимаем ивент из "провода". chat, err := talk.Recv() if err != nil { log.Errorf("Unable to get events from server: %s", err) + // Попробуем скрасить печальные будни generic-ошибок... switch { - // Стрим не читается, он закрылся с той стороны во время чтения + // Стрим не читается, он закрылся с той стороны во время чтения. case errors.Is(err, io.EOF): - err := fmt.Errorf("tcp stream closed while reading, err=%w", err) - gTomb.Kill(err) + err = fmt.Errorf("tcp stream closed while reading, err=%w", err) - continue - - // Пытаемся читать закрытый сокет + // Пытаемся читать закрытый сокет. case errors.Is(err, net.ErrClosed): - err := fmt.Errorf("unable to read closed socket, err=%w", err) - gTomb.Kill(err) + err = fmt.Errorf("unable to read closed socket, err=%w", err) - continue - - // Не смогли записать в сокет + // Не смогли записать в сокет. case errors.Is(err, net.ErrWriteToConnected): - err := fmt.Errorf("unable to write to socket, err=%w", err) - gTomb.Kill(err) + err = fmt.Errorf("unable to write to socket, err=%w", err) - continue - - // Не сетевая проблема + // Не сетевая проблема. default: // Это уже что-то странное. - // Вероятно, ошибка парсинга xml. Собственно, баг сервера, тут мы ничего поделать не можем - err := fmt.Errorf("error during parsing received message, err=%w", err) - gTomb.Kill(err) - - continue + // Вероятно, ошибка парсинга xml. Собственно, баг сервера, тут мы ничего поделать не можем. + err = fmt.Errorf("error during parsing received message, err=%w", err) } + + gTomb.Kill(err) + + // Не забываем выходить на ошибке :( + return } parseEvent(chat) diff --git a/read_config.go b/read_config.go index 4133770..9fdb6e8 100644 --- a/read_config.go +++ b/read_config.go @@ -189,15 +189,15 @@ func readConfig() error { //nolint:gocognit,gocyclo for n, channel := range sampleConfig.Jabber.Channels { if channel.Name == "" { - return errors.New("no \"name\" entry in jabber channel config") + return fmt.Errorf("no \"name\" entry in jabber channel config") } if channel.Nick == "" { sampleConfig.Jabber.Channels[n].Nick = sampleConfig.Jabber.Nick } - // channel.Password может быть пустым, тогда пароля нет - } + // channel.Password может быть пустым, тогда пароля нет. + } //nolint:wsl // Если список фраз с которыми стартует бот пустой, вносим в него 1 запись с пустой строкой if len(sampleConfig.Jabber.StartupStatus) == 0 { @@ -235,13 +235,13 @@ func readConfig() error { //nolint:gocognit,gocyclo config = sampleConfig configLoaded = true - log.Infof("Using %s as config file", location) //nolint:wsl + log.Infof("Using %s as config file", location) break } if !configLoaded { - return errors.New("config was not loaded!") //nolint:goerr113,revive,stylecheck + return fmt.Errorf("config was not loaded!") //nolint:goerr113,revive,stylecheck } return err //nolint:wrapcheck diff --git a/settings-db-util.go b/settings-db-util.go index b43b6e8..0652ee6 100644 --- a/settings-db-util.go +++ b/settings-db-util.go @@ -14,7 +14,7 @@ import ( // StoreKV сохраняет в указанной бд ключ и значение. func StoreKV(db *pebble.DB, key string, value string) error { var ( - kArray = []byte(key) //nolint:revive + kArray = []byte(key) vArray = []byte(value) ) @@ -26,7 +26,7 @@ func StoreKV(db *pebble.DB, key string, value string) error { // FetchV достаёт значение по ключу. func FetchV(db *pebble.DB, key string) (string, error) { var ( - kArray = []byte(key) //nolint:revive + kArray = []byte(key) vArray []byte valueString string )