src/irc.c

   1 /*
   2  *      irc.c - this file is part of vomak - a very simple IRC bot
   3  *
   4  *      Copyright 2008 Enrico Tröger <enrico(dot)troeger(at)uvena(dot)de>
   5  *      Copyright 2008 Dominic Hopf     <dh(at)dmaphy(dot)de>
   6  *
   7  *      This program is free software; you can redistribute it and/or modify
   8  *      it under the terms of the GNU General Public License as published by
   9  *      the Free Software Foundation; version 2 of the License.
  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., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
  19  */
  20
  21 #define _GNU_SOURCE
  22
  23 #include <stdio.h>
  24 #include <stdlib.h>
  25 #include <errno.h>
  26 #include <netdb.h>
  27 #include <sys/types.h>
  28 #include <sys/stat.h>
  29 #include <netinet/in.h>
  30 #include <sys/socket.h>
  31 #include <unistd.h>
  32 #include <string.h>
  33 #include <strings.h>
  34 #include <signal.h>
  35 #include <time.h>
  36 #ifndef DEBUG
  37 # include <syslog.h>
  38 #endif
  39
  40 #include <glib.h>
  41 #include <glib/gstdio.h>
  42
  43 #include "vomak.h"
  44 #include "socket.h"
  45 #include "irc.h"
  46
  47
  48 config_t *config;
  49
  50
  51 static void irc_kick(irc_conn_t *irc_conn, const gchar *nickname);
  52 static gboolean irc_is_user_op(irc_conn_t *irc_conn, const gchar *nickname);
  53
  54 enum
  55 {
  56         GOODIES_COFFEE,
  57         GOODIES_COKE,
  58         GOODIES_BEER,
  59         GOODIES_TEA,
  60         GOODIES_PIZZA,
  61 };
  62 typedef struct
  63 {
  64         const gchar *command;
  65         const gchar *message;
  66 } goodies_t;
  67
  68 const goodies_t goodies[] = {
  69         { "!coffee", "A nice sexy waitress brings %s a big cup of coffee!" },
  70         { "!coke", "A nice sexy waitress brings %s a cool bottle of coke!" },
  71         { "!beer", "A nice sexy waitress brings %s a nice bottle of beer!" },
  72         { "!tea", "A nice sexy waitress brings %s a cup of hot tea!" },
  73         { "!pizza", "Someone calls Mario, and he brings %s a tasty hawaiian pizza!" },
  74         { NULL, NULL }
  75 };
  76
  77 gboolean irc_query_names(gpointer data)
  78 {
  79         irc_conn_t *irc = data;
  80         static gchar msg[1024];
  81         guint msg_len;
  82
  83         TRACE
  84
  85         msg_len = g_snprintf(msg, sizeof msg, "NAMES #%s\r\n", config->channel);
  86         // send the message directly to avoid logging (prevent log file spamming)
  87         send(irc->socket, msg, msg_len, MSG_NOSIGNAL);
  88
  89         return TRUE;
  90 }
  91
  92
  93 /* 'line' should be terminated by "\r\n" (CRLF) */
  94 static void irc_log(irc_conn_t *irc_conn, const gchar *line, gboolean send)
  95 {
  96         time_t t;
  97         struct tm *tp;
  98         static gchar str[256];
  99         GString *log_line;
 100
 101         if (! irc_conn->log_fd)
 102                 return;
 103
 104         t = time(NULL);
 105         tp = localtime(&t);
 106         strftime(str, sizeof str, "%F %T %z ", tp);
 107
 108         log_line = g_string_new(str);
 109         if (send) // if we are sending a message, add our ident string
 110         {
 111                 g_string_append_printf(log_line, ":%s!n=%s@%s ",
 112                         config->nickname, config->username, config->servername);
 113         }
 114         g_string_append(log_line, line);
 115         /* add \r\n if it is missing */
 116         if (strncmp(log_line->str + log_line->len - 2, "\r\n", 2) != 0)
 117         {
 118                 g_string_append(log_line, "\r\n");
 119         }
 120         fwrite(log_line->str, log_line->len, 1, irc_conn->log_fd);
 121         fflush(irc_conn->log_fd);
 122         g_string_free(log_line, TRUE);
 123 }
 124
 125
 126 static void irc_send_private_message(irc_conn_t *irc_conn, const gchar *target, const gchar *format, ...)
 127 {
 128         static gchar tmp_msg[1024];
 129         static gchar msg[1024];
 130         guint msg_len;
 131         va_list ap;
 132
 133         if (target == NULL)
 134                 return;
 135
 136         va_start(ap, format);
 137         g_vsnprintf(tmp_msg, sizeof tmp_msg, format, ap);
 138         va_end(ap);
 139
 140         msg_len = g_snprintf(msg, sizeof msg, "PRIVMSG %s :%s, %s\r\n", target, target, tmp_msg);
 141         irc_send(irc_conn, msg, msg_len);
 142 }
 143
 144
 145 static gchar *get_nickname(const gchar *line, guint len)
 146 {
 147         static gchar result[20];
 148         guint i, j = 0;
 149
 150         // :eht16!n=enrico@uvena.de PRIVMSG #eht16 :df
 151         for (i = 0; i < len; i++)
 152         {
 153                 if (line[i] == '!' || line[i] == '=' || j >= 19)
 154                         break;
 155
 156                 if (line[i] == ':')
 157                         continue;
 158
 159                 result[j++] = line[i];
 160         }
 161         result[j] = '\0';
 162
 163         return result;
 164 }
 165
 166
 167 static gchar *get_command_sender(const gchar *line, guint len, const gchar *command)
 168 {
 169         guint i;
 170         gsize cmd_len = strlen(command);
 171
 172         // :eht16!n=enrico@uvena.de PRIVMSG GeanyTestBot :hi
 173         for (i = 0; i < len; i++)
 174         {
 175                 if (line[i] != ' ')
 176                         continue;
 177
 178                 if (strncmp(command, line + i + 1, cmd_len) == 0)
 179                 {
 180                         static gchar name[20];
 181                         g_snprintf(name, sizeof(name), "%s :", config->nickname);
 182                         // if the receiver of the message is me, then it's a private message and
 183                         // we return the sender's nickname, otherwise NULL
 184                         if (strncmp(name, line + i + cmd_len + 2, strlen(config->nickname) + 2) == 0 ||
 185                                 strncmp("KICK", command, 4) == 0)
 186                         {
 187                                 return get_nickname(line, len);
 188                         }
 189                         else
 190                                 return NULL;
 191                 }
 192         }
 193
 194         return NULL;
 195 }
 196
 197
 198 static gint get_response(const gchar *line, guint len)
 199 {
 200         static gchar result[4];
 201         guint i, j = 0;
 202         gboolean in_response = FALSE;
 203
 204         // :kornbluth.freenode.net 353 GeanyTestBot @ #eht16 :GeanyTestBot eht16
 205         // :kornbluth.freenode.net 366 GeanyTestBot #eht16 :End of /NAMES list.
 206         for (i = 0; i < len; i++)
 207         {
 208                 // before a response code
 209                 if (line[i] != ' ' && ! in_response)
 210                         continue;
 211
 212                 // after a response code
 213                 if (line[i] == ' ' && in_response)
 214                 {
 215                         in_response = FALSE;
 216                         break;
 217                 }
 218
 219                 if (line[i] == ' ' )
 220                         i++; // skip the space
 221
 222                 result[j++] = line[i];
 223                 in_response = TRUE;
 224         }
 225         result[j] = '\0';
 226
 227         return atoi(result);
 228 }
 229
 230
 231 const gchar *irc_get_message(const gchar *line, guint len)
 232 {
 233         static gchar result[1024] = { 0 };
 234         guint i, j = 0;
 235         gint state = 0; // represents the current part of the whole line, separated by spaces
 236
 237         // :eht16!n=enrico@uvena.de PRIVMSG #eht16 :df foo: var
 238         // -> df foo: var
 239         for (i = 0; i < len; i++)
 240         {
 241                 if (state < 3)
 242                 {
 243                         if (line[i] == ' ')
 244                         {
 245                                 state++;
 246                                 i++; // skip the ':'
 247                                 continue;
 248                         }
 249                 }
 250                 else if (line[i] != '\r' && line[i] != '\n')
 251                 {
 252                         result[j++] = line[i];
 253                 }
 254         }
 255         result[j] = '\0';
 256
 257         return result;
 258 }
 259
 260
 261 static const gchar *irc_get_message_with_name(const gchar *line, guint len, const gchar *name)
 262 {
 263         const gchar *tmp;
 264         gsize name_len;
 265
 266         tmp = irc_get_message(line, len);
 267         name_len = strlen(name);
 268
 269         if (strncmp(tmp, name, name_len) == 0)
 270                 tmp += name_len;
 271
 272         return tmp;
 273 }
 274
 275
 276 // returns a nickname argument given to cmd, it may also be "cmd for nickname", then the "for" is
 277 // skipped
 278 static const gchar *get_argument_target(const gchar *line, guint len, const gchar *cmd)
 279 {
 280         static gchar result[20];
 281         const gchar *tmp;
 282         gchar **parts;
 283
 284         // :eht16!n=enrico@uvena.de PRIVMSG #eht16 :!beer for me
 285         tmp = irc_get_message(line, len);
 286
 287         // -> !beer for me
 288         parts = g_strsplit(tmp, " ", -1);
 289         if (parts == NULL || parts[0] == NULL)
 290         {
 291                 g_strfreev(parts);
 292                 return NULL;
 293         }
 294
 295         // if cmd doesn't match, skip it
 296         if (parts[0] != NULL && strcmp(parts[0], cmd) != 0)
 297         {
 298                 g_strfreev(parts);
 299                 return NULL;
 300         }
 301
 302         if (parts[1] == NULL)
 303         {
 304                 g_strfreev(parts);
 305                 return NULL;
 306         }
 307
 308         if (strcmp("for", parts[1]) == 0)
 309         {
 310                 if (parts[2] != NULL)
 311                 {
 312                         if (strcmp(parts[2], "me") == 0)
 313                         {
 314                                 g_strfreev(parts);
 315                                 return NULL; // if we return NULL, the nickname of the caller is used, aka "me"
 316                         }
 317                         else
 318                                 g_strlcpy(result, parts[2], sizeof(result));
 319                 }
 320                 else
 321                         g_strlcpy(result, parts[1], sizeof(result));
 322         }
 323         else if (strcmp(parts[1], "me") == 0)
 324         {
 325                 g_strfreev(parts);
 326                 return NULL; // if we return NULL, the nickname of the caller is used, aka "me"
 327         }
 328         else
 329         {
 330                 g_strlcpy(result, parts[1], sizeof(result));
 331         }
 332
 333         g_strfreev(parts);
 334         return result;
 335 }
 336
 337
 338 // Parses the line and puts the first argument in arg1, all further arguments are concatenated
 339 // in arg2. arg1 and arg2 should be freed when no longer needed.
 340 // If arg1 and arg2 were set successfully, TRUE is returned, if any error occurs, FALSE is returned
 341 // and arg1 and arg2 are set to NULL.
 342 static gboolean get_argument_two(const gchar *line, guint len, const gchar *cmd,
 343                                                                  gchar **arg1, gchar **arg2)
 344 {
 345         const gchar *tmp;
 346         gchar **parts;
 347
 348         *arg1 = NULL;
 349         *arg2 = NULL;
 350
 351         // :eht16!n=enrico@uvena.de PRIVMSG #eht16 :!learn keyword text to be added
 352         tmp = irc_get_message(line, len);
 353
 354         // -> !learn keyword text to be added
 355         parts = g_strsplit(tmp, " ", 3);
 356         if (parts == NULL || parts[0] == NULL)
 357         {
 358                 g_strfreev(parts);
 359                 return FALSE;
 360         }
 361
 362         // if cmd doesn't match, skip it
 363         if (parts[0] != NULL && strcmp(parts[0], cmd) != 0)
 364         {
 365                 g_strfreev(parts);
 366                 return FALSE;
 367         }
 368
 369         if (parts[1] == NULL || parts[2] == NULL)
 370         {
 371                 g_strfreev(parts);
 372                 return FALSE;
 373         }
 374
 375         *arg1 = g_strdup(parts[1]);
 376         *arg2 = g_strdup(parts[2]);
 377
 378         g_strfreev(parts);
 379
 380         return TRUE;
 381 }
 382
 383
 384 static void command_fortune(irc_conn_t *irc_conn)
 385 {
 386         GError *error = NULL;
 387         gchar *out = NULL;
 388         gchar *err = NULL;
 389
 390         if (g_spawn_command_line_sync(config->fortune_cmd, &out, &err, NULL, &error))
 391         {
 392                 if (NZV(out))
 393                 {
 394                         gchar **lines = g_strsplit(out, "\n", -1);
 395                         gsize i, len;
 396
 397                         len = g_strv_length(lines);
 398                         for (i = 0; i < len; i++)
 399                         {
 400                                 if (strlen(g_strchomp(lines[i])) > 0)
 401                                         irc_send_message(irc_conn, NULL, lines[i]);
 402                         }
 403                         g_strfreev(lines);
 404                 }
 405                 else
 406                 {
 407                         vomak_warning("Executing fortune failed (%s)", err);
 408                 }
 409                 g_free(out);
 410                 g_free(err);
 411         }
 412         else
 413         {
 414                 vomak_warning("Executing fortune failed (%s)", error->message);
 415                 g_error_free(error);
 416         }
 417 }
 418
 419
 420 static void command_moo(irc_conn_t *irc_conn)
 421 {
 422         gint32 rand = g_random_int();
 423
 424         if (rand % 2 == 0)
 425         {
 426                 gsize i;
 427                 const gchar *moo_str[] = {
 428                                 "         ^__^\r\n",
 429                                 "         (oo)\r\n",
 430                                 "   /-----(__)\r\n",
 431                                 "  / |    ||\r\n",
 432                                 " *  /\\---/\\\r\n",
 433                                 "    ~~   ~~\n\r\n",
 434                                 "..\"Have you mooed today?\"..\r\n",
 435                                 NULL
 436                 };
 437
 438                 for (i = 0; moo_str[i] != NULL; i++)
 439                 {
 440                         irc_send_message(irc_conn, NULL, moo_str[i]);
 441                 }
 442         }
 443         else
 444         {
 445                 irc_send_message(irc_conn, NULL, "I have Super Cow Powers. Have you mooed today?");
 446         }
 447 }
 448
 449
 450 static void command_learn(irc_conn_t *irc_conn, const gchar *line, guint len)
 451 {
 452         gchar *arg1, *arg2;
 453
 454         if (get_argument_two(line, len, "!learn", &arg1, &arg2))
 455         {
 456                 gint result = help_system_learn(arg1, arg2);
 457                 gchar *text;
 458
 459                 switch (result)
 460                 {
 461                         case 0:
 462                         {
 463                                 text = g_strdup_printf("new keyword \"%s\" was added.", arg1);
 464                                 break;
 465                         }
 466                         case 1:
 467                         {
 468                                 text = g_strdup_printf("existing keyword \"%s\" was updated.", arg1);
 469                                 break;
 470                         }
 471                         default:
 472                         {
 473                                 text = g_strdup("an error occurred. Database not updated.");
 474                                 break;
 475                         }
 476                 }
 477                 irc_send_message(irc_conn, get_nickname(line, len), "%s", text);
 478
 479                 g_free(text);
 480                 g_free(arg1);
 481                 g_free(arg2);
 482         }
 483         else
 484                 irc_send_message(irc_conn, get_nickname(line, len),
 485                         "wrong usage of !learn. Use \"?? learn\" for usage information.");
 486 }
 487
 488
 489 static void command_alias(irc_conn_t *irc_conn, const gchar *line, guint len)
 490 {
 491         gchar *arg1, *arg2;
 492
 493         if (get_argument_two(line, len, "!alias", &arg1, &arg2))
 494         {
 495                 // detect if arg2 has more than one word by scanning for spaces in
 496                 // the string
 497                 if (strchr(arg2, ' '))
 498                 {
 499                         irc_send_message(irc_conn, get_nickname(line, len),
 500                                 "You gave me more than two arguments for !alias. I can not handle this.");
 501                 }
 502                 // check if the target actually exist and refuse if it doesn't exist
 503                 else if (g_hash_table_lookup(config->data, arg2) == NULL)
 504                 {
 505                         irc_send_message(irc_conn, get_nickname(line, len),
 506                                 "The given target for the alias does not exist. I will refuse your request.");
 507                 }
 508                 else
 509                 {
 510                         gint result;
 511                         gchar *text;
 512                         gchar *alias = g_strconcat("@", arg2, NULL);
 513
 514                         result = help_system_learn(arg1, alias);
 515
 516                         switch (result)
 517                         {
 518                                 case 0:
 519                                 {
 520                                         text = g_strdup_printf("new alias \"%s\" was added.", arg1);
 521                                         break;
 522                                 }
 523                                 case 1:
 524                                 {
 525                                         text = g_strdup_printf("existing alias \"%s\" was updated.", arg1);
 526                                         break;
 527                                 }
 528                                 default:
 529                                 {
 530                                         text = g_strdup("An error occurred. Database not updated.");
 531                                         break;
 532                                 }
 533                         }
 534
 535                         irc_send_message(irc_conn, get_nickname(line, len), "%s", text);
 536
 537                         g_free(alias);
 538                         g_free(text);
 539                         g_free(arg1);
 540                         g_free(arg2);
 541                 }
 542         }
 543         else
 544                 irc_send_message(irc_conn, get_nickname(line, len),
 545                         "wrong usage of !alias. Use \"?? alias\" for usage information.");
 546 }
 547
 548
 549 static void command_goodies(irc_conn_t *irc_conn, const gchar *line, guint len, gint goodie)
 550 {
 551         const goodies_t *g = &goodies[goodie];
 552         const gchar *arg = get_argument_target(line, len, g->command);
 553
 554         if (arg == NULL)
 555                 arg = get_nickname(line, len);
 556
 557         irc_send_message(irc_conn, NULL,
 558                 g->message, arg);
 559 }
 560
 561
 562 static void process_command(irc_conn_t *irc_conn, const gchar *line, guint len, const gchar *content)
 563 {
 564         // !test
 565         if (strncmp(content, "!test", 5) == 0)
 566         {
 567                 irc_send_message(irc_conn, get_nickname(line, len), "I don't like tests!");
 568         }
 569         // !moo
 570         else if (strncmp(content, "!moo", 4) == 0)
 571         {
 572                 command_moo(irc_conn);
 573         }
 574         // !fortune
 575         else if (config->fortune_cmd != NULL && strncmp(content, "!fortune", 8) == 0)
 576         {
 577                 command_fortune(irc_conn);
 578         }
 579         // !coffee
 580         else if (strncmp(content, "!coffee", 7) == 0)
 581         {
 582                 command_goodies(irc_conn, line, len, GOODIES_COFFEE);
 583         }
 584         // !coke
 585         else if (strncmp(content, "!coke", 5) == 0)
 586         {
 587                 command_goodies(irc_conn, line, len, GOODIES_COKE);
 588         }
 589         // !beer
 590         else if (strncmp(content, "!beer", 5) == 0)
 591         {
 592                 command_goodies(irc_conn, line, len, GOODIES_BEER);
 593         }
 594         // !pizza
 595         else if (strncmp(content, "!pizza", 5) == 0)
 596         {
 597                 command_goodies(irc_conn, line, len, GOODIES_PIZZA);
 598         }
 599         // !tea
 600         else if (strncmp(content, "!tea", 5) == 0)
 601         {
 602                 command_goodies(irc_conn, line, len, GOODIES_TEA);
 603         }
 604         // !help
 605         else if (strncmp(content, "!help", 5) == 0)
 606         {
 607                 help_system_query("?? help");
 608         }
 609         /*
 610          * Fun with !roulette
 611          * You have to register your bot with nickserv and add it to the access-list
 612          * of your channel to make the !roulette-command work.
 613          * This is just tested on FreeNode. Please feel free to write patches, that
 614          * will make this work on other Networks.
 615          */
 616         else if (strncmp(content, "!roulette", 9) == 0)
 617         {
 618                 static gint32 rand = -1;
 619                 static gint bullets_fired = 0;
 620
 621                 if (rand == -1)
 622                         rand = g_random_int() % 6;
 623                 if (rand == bullets_fired)
 624                 {
 625                         irc_send_message(irc_conn, NULL, "*bang*");
 626                         irc_kick(irc_conn, get_nickname(line, len));
 627                         rand = g_random_int() % 6;
 628                         bullets_fired = 0;
 629                 }
 630
 631                 else
 632                 {
 633                         irc_send_message(irc_conn, NULL, "*click*");
 634                         bullets_fired++;
 635                 }
 636         }
 637         // !learn
 638         /// TODO require op privileges for !learn
 639         else if (strncmp(content, "!learn", 6) == 0)
 640         {
 641                 command_learn(irc_conn, line, len);
 642         }
 643         // !alias
 644         else if (strncmp(content, "!alias", 6) == 0)
 645         {
 646                 command_alias(irc_conn, line, len);
 647         }
 648 }
 649
 650
 651 static gboolean process_line(irc_conn_t *irc_conn, const gchar *line, guint len)
 652 {
 653         static gchar msg[1024];
 654         guint msg_len;
 655         gint response = get_response(line, len);
 656         static gchar tmp_userlist[1024];
 657         gchar *priv_sender;
 658         const gchar *content;
 659         static gboolean connected = FALSE;
 660
 661         // don't log the NAMES command's output (prevent log file spam)
 662         if (response != 353 && response != 366)
 663                 irc_log(irc_conn, line, FALSE);
 664
 665         // An error occurred, try to quit cleanly and print the error
 666         if ((response > 400 && response < 503))
 667         {
 668                 // ignore Freenode's info messages sent with error code 477
 669                 // (see http://freenode.net/faq.shtml#freenode-info)
 670                 if (response != 477 || strstr(line, "[freenode-info]") == NULL)
 671                 {
 672 #ifndef DEBUG
 673                         syslog(LOG_WARNING, "received error: %d (%s)", response, g_strstrip((gchar*) line));
 674 #else
 675                         g_print("Error: %s", line);
 676 #endif
 677                         g_free(irc_conn->quit_msg);
 678                         irc_conn->quit_msg = g_strdup("I got an error and better leave in advance. Bye.");
 679                         irc_goodbye(irc_conn);
 680                         main_quit();
 681                         return FALSE;
 682                 }
 683         }
 684
 685         if (! connected)
 686         {
 687                 if (response == 376)
 688                         connected = TRUE;
 689                 else
 690                         return TRUE;
 691         }
 692
 693         content = irc_get_message(line, len);
 694
 695         // retrieve user name list
 696         if (response == 353)
 697         {
 698                 if (tmp_userlist[0] == '\0')
 699                         g_strlcpy(tmp_userlist, strchr(content, ':') + 1, sizeof(tmp_userlist));
 700                 else
 701                         g_strlcat(tmp_userlist, strchr(content, ':') + 1, sizeof(tmp_userlist));
 702         }
 703         // retrieve end of user name list
 704         else if (response == 366)
 705         {
 706                 if (tmp_userlist[0] != '\0')
 707                 {
 708                         set_user_list(tmp_userlist);
 709                         tmp_userlist[0] = '\0';
 710                 }
 711         }
 712         else if (! connected)
 713         {
 714                 // don't do anything else until we got finished connecting (to skip MOTD messages)
 715         }
 716         // PING-PONG
 717         else if (strncmp("PING :", line, 6) == 0)
 718         {
 719                 msg_len = g_snprintf(msg, sizeof msg, "PONG %s\r\n", line + 6); // 7 = "PING :"
 720                 debug("PONG: -%s-\n", msg);
 721                 irc_send(irc_conn, msg, msg_len);
 722         }
 723         // handle private message
 724         else if ((priv_sender = get_command_sender(line, len, "PRIVMSG")) != NULL)
 725         {
 726                 // to be able to send private messages to users, you need to register your bot's
 727                 // nickname with Nickserv (at least on Freenode)
 728                 irc_send_private_message(irc_conn, priv_sender, "I don't like private messages!");
 729         }
 730         // handle kicks (we were kicked, bastards)
 731         else if (get_command_sender(line, len, "KICK") != NULL)
 732         {
 733                 if (config->auto_rejoin)
 734                 {
 735                         g_usleep(5000000); // just wait a little bit
 736                         msg_len = g_snprintf(msg, sizeof msg, "JOIN #%s\r\n", config->channel);
 737                         irc_send(irc_conn, msg, msg_len);
 738                 }
 739                 else
 740                 {
 741                         main_quit();
 742                 }
 743         }
 744         // Hi /me, acts on "hello $nickname" and "hi $nickname", hi and hello are case-insensitive
 745         // Thanks /me
 746         else if (strstr(content, config->nickname) != NULL)
 747         {
 748                 const gchar *tmp_msg = irc_get_message_with_name(line, len, config->nickname);
 749
 750                 if (strncasecmp("hi", content, 2) == 0 || strncasecmp("hello", content, 5) == 0 || strncasecmp("hey", content, 3) == 0 ||
 751                         strcasecmp(", hi", tmp_msg) == 0 || strcasecmp(", hello", tmp_msg) == 0 || strcasecmp(", hey", tmp_msg) == 0 ||
 752                         strcasecmp(": hi", tmp_msg) == 0 || strcasecmp(": hello", tmp_msg) == 0 || strcasecmp(": hey", tmp_msg) == 0)
 753                 {
 754                         irc_send_message(irc_conn, NULL,
 755         "Hi %s. My name is %s and I'm here to offer additional services to you! "
 756         "Try \"?? help\" for general information and \"?? vomak\" for information about me.",
 757                                 get_nickname(line, len), config->nickname);
 758                 }
 759                 else if (strncasecmp("thanks", content, 6) == 0 || strncasecmp("thx", content, 3) == 0 ||
 760                                  strcasecmp(", thanks", tmp_msg) == 0 || strcasecmp(", thx", tmp_msg) == 0 ||
 761                                  strcasecmp(": thanks", tmp_msg) == 0 || strcasecmp(": thx", tmp_msg) == 0)
 762                 {
 763                         irc_send_message(irc_conn, get_nickname(line, len),
 764                                 "no problem. It was a pleasure to serve you.");
 765                 }
 766         }
 767         // ?? ...
 768         else if (strncmp(content, "?? ", 3) == 0)
 769         {
 770                 help_system_query(content);
 771         }
 772         // pass to process_command() to process other commands (!beer, !test, !learn, ...)
 773         else if (*content == '!')
 774         {
 775                 process_command(irc_conn, line, len, content);
 776         }
 777
 778         return TRUE;
 779 }
 780
 781
 782 /*
 783  * Please note that this will not work on Networks without ChanServ, e.g. on
 784  * Quakenet or IRCnet. Your Bot has to be registered with NickServ and to be
 785  * added to the channel access list for this to work.
 786  */
 787 static gboolean irc_toggle_op(irc_conn_t *irc_conn, gboolean request_op)
 788 {
 789         const gchar *cmd;
 790         static gchar msg[1024];
 791         guint msg_len;
 792
 793         if (irc_is_user_op(irc_conn, "ChanServ"))
 794         {
 795                 cmd = (request_op) ? "op" : "deop";
 796                 msg_len = g_snprintf(msg, sizeof msg, "PRIVMSG ChanServ :%s #%s\r\n", cmd, config->channel);
 797                 irc_send(irc_conn, msg, msg_len);
 798
 799                 return TRUE;
 800         }
 801         return FALSE; /* it seems we don't have a ChanServ bot */
 802 }
 803
 804
 805 static gboolean irc_is_user_op(irc_conn_t *irc_conn, const gchar *nickname)
 806 {
 807         const gchar *pos;
 808         const gchar *userlist;
 809
 810         if (nickname == NULL)
 811                 return FALSE;
 812
 813         userlist = get_user_list();
 814
 815         if ( (pos = strstr(userlist, nickname)) )
 816         {
 817                 if ( (pos - 1) >= userlist && (*(pos - 1) == '@') )
 818                 {
 819                         return TRUE;
 820                 }
 821         }
 822         else
 823         {
 824 #ifdef DEBUG
 825                 irc_send_message(irc_conn, NULL,
 826                                 "Hey. There are crazy things going on here. (O.o)");
 827 #else
 828                 syslog(LOG_WARNING, "user %s not found in names list of #%s", nickname, config->channel);
 829 #endif
 830         }
 831
 832         return FALSE;
 833 }
 834
 835
 836 static void irc_kick(irc_conn_t *irc_conn, const gchar *nickname)
 837 {
 838         static gchar msg[1024];
 839         gboolean need_deop = FALSE;
 840         guint msg_len;
 841
 842         TRACE
 843
 844         if (! irc_is_user_op(irc_conn, config->nickname))
 845         {
 846                 // if irc_toggle_op fails, most probably we don't have a ChanServ and at this point we
 847                 // know we are not an op, so fail silently and don't try to kick
 848                 if (! irc_toggle_op(irc_conn, TRUE)) /// TODO: prüfen, ob das auch erfolreich war
 849                         return;
 850                 need_deop = TRUE;
 851         }
 852
 853         // give the server a chance to set the op status for us before we make us of it,
 854         // and let the victim read his *bang* message
 855         g_usleep(2000000);
 856
 857         msg_len = g_snprintf(msg, sizeof msg, "KICK #%s %s\r\n", config->channel, nickname);
 858         irc_send(irc_conn, msg, msg_len);
 859         if (need_deop)
 860                 irc_toggle_op(irc_conn, FALSE);
 861 }
 862
 863
 864 static gboolean input_cb(GIOChannel *source, GIOCondition cond, gpointer data)
 865 {
 866 #if 1
 867         gchar buf[1024];
 868         guint buf_len;
 869         irc_conn_t *irc = data;
 870         gboolean ret = TRUE;
 871
 872         if (cond & (G_IO_ERR | G_IO_HUP))
 873                 return FALSE;
 874
 875         if ((buf_len = socket_fd_gets(irc->socket, buf, sizeof(buf))) != -1)
 876         {
 877                 ret = process_line(irc, buf, buf_len);
 878         }
 879 #else
 880         gsize buf_len;
 881         irc_conn_t *irc = data;
 882
 883         if (cond & (G_IO_IN | G_IO_PRI))
 884         {
 885                 gchar *buf = NULL;
 886                 GIOStatus rv;
 887                 GError *err = NULL;
 888
 889                 do
 890                 {
 891                         rv = g_io_channel_read_line(source, &buf, &buf_len, NULL, &err);
 892                         if (buf != NULL)
 893                         {
 894                                 buf_len -= 2;
 895                                 buf[buf_len] = '\0'; // skip trailing \r\n
 896
 897                                 process_line(irc, buf, buf_len);
 898                                 g_free(buf);
 899                         }
 900                         if (err != NULL)
 901                         {
 902                                 debug("%s: error: %s", __func__, err->message);
 903                                 g_error_free(err);
 904                                 err = NULL;
 905                         }
 906                 }
 907                 while (rv == G_IO_STATUS_NORMAL || rv == G_IO_STATUS_AGAIN);
 908                 debug("%s: status %d\n", __func__, rv);
 909         }
 910 #endif
 911         return ret;
 912 }
 913
 914
 915 void irc_send_message(irc_conn_t *irc_conn, const gchar *target, const gchar *format, ...)
 916 {
 917         static gchar tmp_msg[1024];
 918         static gchar msg[1024];
 919         guint msg_len;
 920         va_list ap;
 921
 922         va_start(ap, format);
 923         g_vsnprintf(tmp_msg, sizeof tmp_msg, format, ap);
 924         va_end(ap);
 925
 926         if (target)
 927                 msg_len = g_snprintf(msg, sizeof msg, "PRIVMSG #%s :%s, %s\r\n", config->channel, target, tmp_msg);
 928         else
 929                 msg_len = g_snprintf(msg, sizeof msg, "PRIVMSG #%s :%s\r\n", config->channel, tmp_msg);
 930
 931         irc_send(irc_conn, msg, msg_len);
 932 }
 933
 934
 935 /*
 936  * target should not be NULL!
 937  */
 938 void irc_send_notice(irc_conn_t *irc_conn, const gchar *target, const gchar *format, ...)
 939 {
 940         static gchar tmp_msg[1024];
 941         static gchar msg[1024];
 942         guint msg_len;
 943         va_list ap;
 944
 945         va_start(ap, format);
 946         g_vsnprintf(tmp_msg, sizeof tmp_msg, format, ap);
 947         va_end(ap);
 948
 949         msg_len = g_snprintf(msg, sizeof msg, "NOTICE #%s :%s, %s\r\n", config->channel, target, tmp_msg);
 950
 951         irc_send(irc_conn, msg, msg_len);
 952 }
 953
 954
 955 // simple wrapper for send() to enable logging for sent commands
 956 gint irc_send(irc_conn_t *irc_conn, const gchar *msg, guint msg_len)
 957 {
 958         irc_log(irc_conn, msg, TRUE);
 959         return send(irc_conn->socket, msg, msg_len, MSG_NOSIGNAL);
 960 }
 961
 962
 963 void irc_goodbye(irc_conn_t *irc)
 964 {
 965         guint len;
 966         gchar msg[256];
 967
 968         if (NZV(irc->quit_msg))
 969                 len = g_snprintf(msg, sizeof msg, "QUIT :%s\r\n", irc->quit_msg);
 970         else
 971                 len = g_strlcpy(msg, "QUIT :Good bye. It was a pleasure to serve you\r\n", sizeof msg);
 972         irc_send(irc, msg, len);
 973 }
 974
 975
 976 void irc_logfile_reopen(irc_conn_t *irc_conn)
 977 {
 978         TRACE
 979
 980         if (irc_conn->log_fd != NULL)
 981                 fclose(irc_conn->log_fd);
 982
 983         if (NZV(config->logfile))
 984         {
 985                 irc_conn->log_fd = g_fopen(config->logfile, "a");
 986                 if (! irc_conn->log_fd)
 987                         vomak_warning("Logfile could not be opened.");
 988         }
 989 }
 990
 991
 992 gint irc_finalize(irc_conn_t *irc_conn)
 993 {
 994         if (irc_conn->socket < 0)
 995                 return -1;
 996
 997         if (irc_conn->lock_tag > 0)
 998                 g_source_remove(irc_conn->lock_tag);
 999
1000         if (irc_conn->log_fd != NULL)
1001         {
1002                 irc_log(irc_conn, "Stop logging\r\n", FALSE);
1003                 fclose(irc_conn->log_fd);
1004         }
1005
1006         if (irc_conn->read_ioc)
1007         {
1008                 g_io_channel_shutdown(irc_conn->read_ioc, TRUE, NULL);
1009                 g_io_channel_unref(irc_conn->read_ioc);
1010                 irc_conn->read_ioc = NULL;
1011         }
1012         socket_fd_close(irc_conn->socket);
1013         irc_conn->socket = -1;
1014
1015         g_free(irc_conn->quit_msg);
1016
1017         return 0;
1018 }
1019
1020
1021 void irc_connect(irc_conn_t *irc_conn)
1022 {
1023         struct hostent *he;
1024         struct sockaddr_in their_addr;
1025         gchar msg[256];
1026         guint msg_len;
1027
1028         TRACE
1029
1030         // Connect the socket to the server
1031         if ((he = gethostbyname(config->server)) == NULL)
1032         {
1033                 perror("gethostbyname");
1034                 exit(1);
1035         }
1036
1037         if ((irc_conn->socket = socket(PF_INET, SOCK_STREAM, 0)) == -1)
1038         {
1039                 perror("socket");
1040                 exit(1);
1041         }
1042
1043         their_addr.sin_family = PF_INET;
1044         their_addr.sin_port = htons(6667);
1045         their_addr.sin_addr = *((struct in_addr *)he->h_addr_list[0]);
1046         memset(&(their_addr.sin_zero), '\0', 8);
1047
1048         if (connect(irc_conn->socket, (struct sockaddr *)&their_addr, sizeof(struct sockaddr)) == -1)
1049         {
1050                 perror("connect");
1051                 exit(1);
1052         }
1053         // Logging
1054         if (NZV(config->logfile))
1055         {
1056                 irc_conn->log_fd = g_fopen(config->logfile, "a");
1057                 if (irc_conn->log_fd)
1058                         irc_log(irc_conn, "Start logging\r\n", FALSE);
1059                 else
1060                         vomak_warning("Logfile could not be opened.");
1061         }
1062         // say who we are
1063         msg_len = g_snprintf(msg, sizeof(msg), "USER %s %s %s :%s\r\n",
1064                         config->username, config->servername, config->servername, config->realname);
1065         if (irc_send(irc_conn, msg, msg_len) == -1)
1066         {
1067                 perror("send USER");
1068         }
1069         // and how we are called
1070         msg_len = g_snprintf(msg, sizeof(msg), "NICK %s\r\n", config->nickname);
1071         if (irc_send(irc_conn, msg, msg_len) == -1)
1072         {
1073                 perror("send NICK");
1074         }
1075         // identify our nick
1076         if (NZV(config->nickserv_password))
1077         {
1078                 msg_len = g_snprintf(msg, sizeof(msg), "PRIVMSG nickserv :identify %s\r\n", config->nickserv_password);
1079                 // don't use irc_send() here, no need to log our password
1080                 if (send(irc_conn->socket, msg, msg_len, MSG_NOSIGNAL) == -1)
1081                 {
1082                         perror("send NICKSERV");
1083                 }
1084         }
1085         // join the channel
1086         msg_len = g_snprintf(msg, sizeof msg, "JOIN #%s\r\n", config->channel);
1087         if (irc_send(irc_conn, msg, msg_len) == -1)
1088         {
1089                 perror("send");
1090         }
1091
1092         // input callback, attached to the main loop
1093         irc_conn->read_ioc = g_io_channel_unix_new(irc_conn->socket);
1094         //~ g_io_channel_set_flags(irc_conn.read_ioc, G_IO_FLAG_NONBLOCK, NULL);
1095         g_io_channel_set_encoding(irc_conn->read_ioc, NULL, NULL);
1096         irc_conn->lock_tag = g_io_add_watch(irc_conn->read_ioc, G_IO_IN|G_IO_PRI|G_IO_ERR, input_cb, irc_conn);
1097 }