apps/irssi/src/core/servers-reconnect.c

   1 /*
   2  servers-reconnect.c : irssi
   3
   4     Copyright (C) 1999-2000 Timo Sirainen
   5
   6     This program is free software; you can redistribute it and/or modify
   7     it under the terms of the GNU General Public License as published by
   8     the Free Software Foundation; either version 2 of the License, or
   9     (at your option) any later version.
  10
  11     This program is distributed in the hope that it will be useful,
  12     but WITHOUT ANY WARRANTY; without even the implied warranty of
  13     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14     GNU General Public License for more details.
  15
  16     You should have received a copy of the GNU General Public License
  17     along with this program; if not, write to the Free Software
  18     Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  19 */
  20
  21 #include "module.h"
  22 #include "commands.h"
  23 #include "network.h"
  24 #include "signals.h"
  25
  26 #include "chat-protocols.h"
  27 #include "servers.h"
  28 #include "servers-setup.h"
  29 #include "servers-reconnect.h"
  30
  31 #include "settings.h"
  32
  33 GSList *reconnects;
  34 static int last_reconnect_tag;
  35 static int reconnect_timeout_tag;
  36 static int reconnect_time;
  37
  38 void reconnect_save_status(SERVER_CONNECT_REC *conn, SERVER_REC *server)
  39 {
  40         g_free_not_null(conn->tag);
  41         conn->tag = g_strdup(server->tag);
  42
  43         g_free_not_null(conn->away_reason);
  44         conn->away_reason = !server->usermode_away ? NULL :
  45                 g_strdup(server->away_reason);
  46
  47         signal_emit("server reconnect save status", 2, conn, server);
  48 }
  49
  50 static void server_reconnect_add(SERVER_CONNECT_REC *conn,
  51                                  time_t next_connect)
  52 {
  53         RECONNECT_REC *rec;
  54
  55         g_return_if_fail(IS_SERVER_CONNECT(conn));
  56
  57         rec = g_new(RECONNECT_REC, 1);
  58         rec->tag = ++last_reconnect_tag;
  59         rec->next_connect = next_connect;
  60
  61         rec->conn = conn;
  62         server_connect_ref(conn);
  63
  64         reconnects = g_slist_append(reconnects, rec);
  65 }
  66
  67 void server_reconnect_destroy(RECONNECT_REC *rec)
  68 {
  69         g_return_if_fail(rec != NULL);
  70
  71         reconnects = g_slist_remove(reconnects, rec);
  72
  73         signal_emit("server reconnect remove", 1, rec);
  74         server_connect_unref(rec->conn);
  75         g_free(rec);
  76
  77         if (reconnects == NULL)
  78             last_reconnect_tag = 0;
  79 }
  80
  81 static int server_reconnect_timeout(void)
  82 {
  83         SERVER_CONNECT_REC *conn;
  84         GSList *list, *tmp;
  85         time_t now;
  86
  87         /* If server_connect() removes the next reconnection in queue,
  88            we're screwed. I don't think this should happen anymore, but just
  89            to be sure we don't crash, do this safely. */
  90         list = g_slist_copy(reconnects);
  91         now = time(NULL);
  92         for (tmp = list; tmp != NULL; tmp = tmp->next) {
  93                 RECONNECT_REC *rec = tmp->data;
  94
  95                 if (g_slist_find(reconnects, rec) == NULL)
  96                         continue;
  97
  98                 if (rec->next_connect <= now) {
  99                         conn = rec->conn;
 100                         server_connect_ref(conn);
 101                         server_reconnect_destroy(rec);
 102                         CHAT_PROTOCOL(conn)->server_connect(conn);
 103                         server_connect_unref(conn);
 104                 }
 105         }
 106
 107         g_slist_free(list);
 108         return 1;
 109 }
 110
 111 static void sserver_connect(SERVER_SETUP_REC *rec, SERVER_CONNECT_REC *conn)
 112 {
 113         conn->family = rec->family;
 114         conn->address = g_strdup(rec->address);
 115         if (conn->port == 0) conn->port = rec->port;
 116
 117         server_setup_fill_reconn(conn, rec);
 118         server_reconnect_add(conn, rec->last_connect+reconnect_time);
 119         server_connect_unref(conn);
 120 }
 121
 122 static SERVER_CONNECT_REC *
 123 server_connect_copy_skeleton(SERVER_CONNECT_REC *src, int connect_info)
 124 {
 125         SERVER_CONNECT_REC *dest;
 126
 127         dest = NULL;
 128         signal_emit("server connect copy", 2, &dest, src);
 129         g_return_val_if_fail(dest != NULL, NULL);
 130
 131         server_connect_ref(dest);
 132         dest->type = module_get_uniq_id("SERVER CONNECT", 0);
 133         dest->reconnection = src->reconnection;
 134         dest->proxy = g_strdup(src->proxy);
 135         dest->proxy_port = src->proxy_port;
 136         dest->proxy_string = g_strdup(src->proxy_string);
 137         dest->proxy_string_after = g_strdup(src->proxy_string_after);
 138         dest->proxy_password = g_strdup(src->proxy_password);
 139
 140         dest->tag = g_strdup(src->tag);
 141
 142         if (connect_info) {
 143                 dest->family = src->family;
 144                 dest->address = g_strdup(src->address);
 145                 dest->port = src->port;
 146                 dest->password = g_strdup(src->password);
 147         }
 148
 149         dest->chatnet = g_strdup(src->chatnet);
 150         dest->nick = g_strdup(src->nick);
 151         dest->username = g_strdup(src->username);
 152         dest->realname = g_strdup(src->realname);
 153
 154         if (src->own_ip4 != NULL) {
 155                 dest->own_ip4 = g_new(IPADDR, 1);
 156                 memcpy(dest->own_ip4, src->own_ip4, sizeof(IPADDR));
 157         }
 158         if (src->own_ip6 != NULL) {
 159                 dest->own_ip6 = g_new(IPADDR, 1);
 160                 memcpy(dest->own_ip6, src->own_ip6, sizeof(IPADDR));
 161         }
 162
 163         dest->channels = g_strdup(src->channels);
 164         dest->away_reason = g_strdup(src->away_reason);
 165         dest->no_autojoin_channels = src->no_autojoin_channels;
 166
 167         return dest;
 168 }
 169
 170 #define server_should_reconnect(server) \
 171         ((server)->connection_lost && !(server)->no_reconnect && \
 172         ((server)->connrec->chatnet != NULL || \
 173                 (!(server)->banned && !(server)->dns_error)))
 174
 175 #define sserver_connect_ok(rec, net) \
 176         (!(rec)->banned && !(rec)->dns_error && (rec)->chatnet != NULL && \
 177         g_strcasecmp((rec)->chatnet, (net)) == 0)
 178
 179 static void sig_reconnect(SERVER_REC *server)
 180 {
 181         SERVER_CONNECT_REC *conn;
 182         SERVER_SETUP_REC *sserver;
 183         GSList *tmp;
 184         int use_next, through;
 185         time_t now;
 186
 187         g_return_if_fail(IS_SERVER(server));
 188
 189         if (reconnect_time == -1 || !server_should_reconnect(server))
 190                 return;
 191
 192         conn = server_connect_copy_skeleton(server->connrec, FALSE);
 193         g_return_if_fail(conn != NULL);
 194
 195         /* save the server status */
 196         if (server->connected) {
 197                 conn->reconnection = TRUE;
 198
 199                 reconnect_save_status(conn, server);
 200         }
 201
 202         sserver = server_setup_find(server->connrec->address,
 203                                     server->connrec->port,
 204                                     server->connrec->chatnet);
 205
 206         if (sserver != NULL) {
 207                 /* save the last connection time/status */
 208                 sserver->last_connect = server->connect_time == 0 ?
 209                         time(NULL) : server->connect_time;
 210                 sserver->last_failed = !server->connected;
 211                 if (server->banned) sserver->banned = TRUE;
 212                 if (server->dns_error) sserver->dns_error = TRUE;
 213         }
 214
 215         if (sserver == NULL || conn->chatnet == NULL) {
 216                 /* not in any chatnet, just reconnect back to same server */
 217                 conn->family = server->connrec->family;
 218                 conn->address = g_strdup(server->connrec->address);
 219                 conn->port = server->connrec->port;
 220                 conn->password = g_strdup(server->connrec->password);
 221
 222                 server_reconnect_add(conn, (server->connect_time == 0 ? time(NULL) :
 223                                             server->connect_time) + reconnect_time);
 224                 server_connect_unref(conn);
 225                 return;
 226         }
 227
 228         /* always try to first connect to the first on the list where we
 229            haven't got unsuccessful connection attempts for the past half
 230            an hour. */
 231
 232         now = time(NULL);
 233         for (tmp = setupservers; tmp != NULL; tmp = tmp->next) {
 234                 SERVER_SETUP_REC *rec = tmp->data;
 235
 236                 if (sserver_connect_ok(rec, conn->chatnet) &&
 237                     (!rec->last_connect || !rec->last_failed ||
 238                      rec->last_connect < now-FAILED_RECONNECT_WAIT)) {
 239                         if (rec == sserver)
 240                                 conn->port = server->connrec->port;
 241                         sserver_connect(rec, conn);
 242                         return;
 243                 }
 244         }
 245
 246         /* just try the next server in list */
 247         use_next = through = FALSE;
 248         for (tmp = setupservers; tmp != NULL; ) {
 249                 SERVER_SETUP_REC *rec = tmp->data;
 250
 251                 if (!use_next && server->connrec->port == rec->port &&
 252                     g_strcasecmp(rec->address, server->connrec->address) == 0)
 253                         use_next = TRUE;
 254                 else if (use_next && sserver_connect_ok(rec, conn->chatnet)) {
 255                         if (rec == sserver)
 256                                 conn->port = server->connrec->port;
 257                         sserver_connect(rec, conn);
 258                         break;
 259                 }
 260
 261                 if (tmp->next != NULL) {
 262                         tmp = tmp->next;
 263                         continue;
 264                 }
 265
 266                 if (through) {
 267                         /* shouldn't happen unless there's no servers in
 268                            this chatnet in setup.. */
 269                         server_connect_unref(conn);
 270                         break;
 271                 }
 272
 273                 tmp = setupservers;
 274                 use_next = through = TRUE;
 275         }
 276 }
 277
 278 static void sig_connected(SERVER_REC *server)
 279 {
 280         g_return_if_fail(IS_SERVER(server));
 281         if (!server->connrec->reconnection)
 282                 return;
 283
 284         if (server->connrec->channels != NULL)
 285                 server->channels_join(server, server->connrec->channels, TRUE);
 286 }
 287
 288 /* Remove all servers from reconnect list */
 289 /* SYNTAX: RMRECONNS */
 290 static void cmd_rmreconns(void)
 291 {
 292         while (reconnects != NULL)
 293                 server_reconnect_destroy(reconnects->data);
 294 }
 295
 296 static RECONNECT_REC *reconnect_find_tag(int tag)
 297 {
 298         GSList *tmp;
 299
 300         for (tmp = reconnects; tmp != NULL; tmp = tmp->next) {
 301                 RECONNECT_REC *rec = tmp->data;
 302
 303                 if (rec->tag == tag)
 304                         return rec;
 305         }
 306
 307         return NULL;
 308 }
 309
 310 static void reconnect_all(void)
 311 {
 312         GSList *list;
 313         SERVER_CONNECT_REC *conn;
 314         RECONNECT_REC *rec;
 315
 316         /* first move reconnects to another list so if server_connect()
 317            fails and goes to reconnection list again, we won't get stuck
 318            here forever */
 319         list = NULL;
 320         while (reconnects != NULL) {
 321                 rec = reconnects->data;
 322
 323                 list = g_slist_append(list, rec->conn);
 324                 server_connect_ref(rec->conn);
 325                 server_reconnect_destroy(rec);
 326         }
 327
 328
 329         while (list != NULL) {
 330                 conn = list->data;
 331
 332                 CHAT_PROTOCOL(conn)->server_connect(conn);
 333                 server_connect_unref(conn);
 334                 list = g_slist_remove(list, conn);
 335         }
 336 }
 337
 338 /* SYNTAX: RECONNECT <tag> */
 339 static void cmd_reconnect(const char *data, SERVER_REC *server)
 340 {
 341         SERVER_CONNECT_REC *conn;
 342         RECONNECT_REC *rec;
 343         int tag;
 344
 345         if (*data == '\0' && server != NULL) {
 346                 /* reconnect back to same server */
 347                 conn = server_connect_copy_skeleton(server->connrec, TRUE);
 348
 349                 if (server->connected)
 350                         reconnect_save_status(conn, server);
 351                 signal_emit("command disconnect", 2, "* Reconnecting", server);
 352
 353                 conn->reconnection = TRUE;
 354                 CHAT_PROTOCOL(conn)->server_connect(conn);
 355                 server_connect_unref(conn);
 356                 return;
 357         }
 358
 359         if (g_strcasecmp(data, "all") == 0) {
 360                 /* reconnect all servers in reconnect queue */
 361                 reconnect_all();
 362                 return;
 363         }
 364
 365         if (*data == '\0') {
 366                 /* reconnect to first server in reconnection list */
 367                 if (reconnects == NULL)
 368                         cmd_return_error(CMDERR_NOT_CONNECTED);
 369                 rec = reconnects->data;
 370         } else {
 371                 if (g_strncasecmp(data, "RECON-", 6) == 0)
 372                         data += 6;
 373
 374                 tag = atoi(data);
 375                 rec = tag <= 0 ? NULL : reconnect_find_tag(tag);
 376
 377                 if (rec == NULL) {
 378                         signal_emit("server reconnect not found", 1, data);
 379                         return;
 380                 }
 381         }
 382
 383         conn = rec->conn;
 384         server_connect_ref(conn);
 385         server_reconnect_destroy(rec);
 386         CHAT_PROTOCOL(conn)->server_connect(conn);
 387         server_connect_unref(conn);
 388 }
 389
 390 static void cmd_disconnect(const char *data, SERVER_REC *server)
 391 {
 392         RECONNECT_REC *rec;
 393
 394         if (g_strncasecmp(data, "RECON-", 6) != 0)
 395                 return; /* handle only reconnection removing */
 396
 397         rec = reconnect_find_tag(atoi(data+6));
 398
 399         if (rec == NULL)
 400                 signal_emit("server reconnect not found", 1, data);
 401         else
 402                 server_reconnect_destroy(rec);
 403         signal_stop();
 404 }
 405
 406 static void sig_chat_protocol_deinit(CHAT_PROTOCOL_REC *proto)
 407 {
 408         GSList *tmp, *next;
 409
 410         for (tmp = reconnects; tmp != NULL; tmp = next) {
 411                 RECONNECT_REC *rec = tmp->data;
 412
 413                 next = tmp->next;
 414                 if (rec->conn->chat_type == proto->id)
 415                         server_reconnect_destroy(rec);
 416         }
 417 }
 418
 419 static void read_settings(void)
 420 {
 421         reconnect_time = settings_get_int("server_reconnect_time");
 422 }
 423
 424 void servers_reconnect_init(void)
 425 {
 426         settings_add_int("server", "server_reconnect_time", 300);
 427
 428         reconnects = NULL;
 429         last_reconnect_tag = 0;
 430
 431         reconnect_timeout_tag = g_timeout_add(1000, (GSourceFunc) server_reconnect_timeout, NULL);
 432         read_settings();
 433
 434         signal_add("server connect failed", (SIGNAL_FUNC) sig_reconnect);
 435         signal_add("server disconnected", (SIGNAL_FUNC) sig_reconnect);
 436         signal_add("event connected", (SIGNAL_FUNC) sig_connected);
 437         signal_add("chat protocol deinit", (SIGNAL_FUNC) sig_chat_protocol_deinit);
 438         signal_add("setup changed", (SIGNAL_FUNC) read_settings);
 439
 440         command_bind("rmreconns", NULL, (SIGNAL_FUNC) cmd_rmreconns);
 441         command_bind("reconnect", NULL, (SIGNAL_FUNC) cmd_reconnect);
 442         command_bind_first("disconnect", NULL, (SIGNAL_FUNC) cmd_disconnect);
 443 }
 444
 445 void servers_reconnect_deinit(void)
 446 {
 447         g_source_remove(reconnect_timeout_tag);
 448
 449         signal_remove("server connect failed", (SIGNAL_FUNC) sig_reconnect);
 450         signal_remove("server disconnected", (SIGNAL_FUNC) sig_reconnect);
 451         signal_remove("event connected", (SIGNAL_FUNC) sig_connected);
 452         signal_remove("chat protocol deinit", (SIGNAL_FUNC) sig_chat_protocol_deinit);
 453         signal_remove("setup changed", (SIGNAL_FUNC) read_settings);
 454
 455         command_unbind("rmreconns", (SIGNAL_FUNC) cmd_rmreconns);
 456         command_unbind("reconnect", (SIGNAL_FUNC) cmd_reconnect);
 457         command_unbind("disconnect", (SIGNAL_FUNC) cmd_disconnect);
 458 }