apps/irc/proto.c

   1 /*
   2  *  ZeX/OS
   3  *  Copyright (C) 2008  Tomas 'ZeXx86' Jedrzejek (zexx86@zexos.org)
   4  *
   5  *  This program is free software: you can redistribute it and/or modify
   6  *  it under the terms of the GNU General Public License as published by
   7  *  the Free Software Foundation, either version 3 of the License, or
   8  *  (at your option) any later version.
   9  *
  10  *  This program is distributed in the hope that it will be useful,
  11  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
  12  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  13  *  GNU General Public License for more details.
  14  *
  15  *  You should have received a copy of the GNU General Public License
  16  *  along with this program.  If not, see <http://www.gnu.org/licenses/>.
  17  */
  18
  19
  20 #include <time.h>
  21 #include <stdio.h>
  22 #include <string.h>
  23 #include <stdlib.h>
  24 #include "commands.h"
  25 #include "config.h"
  26 #include "proto.h"
  27 #include "net.h"
  28
  29 char *parser;
  30 char *parser_tmp;
  31 char *pong_msg;
  32 unsigned parser_len;
  33
  34 #ifdef LINUX
  35 int setcolor (unsigned char fg, unsigned char bg)
  36 {
  37         return 1;
  38 }
  39 #endif
  40
  41 int proto_command (char *buffer, unsigned len)
  42 {
  43         buffer[len] = '\0';
  44
  45         if (buffer[0] == ':') {
  46                 /* server message */
  47
  48                 if (!strncmp (config.server, buffer+1, config.server_len)) {
  49                         char *str = strchr (buffer+2+config.server_len, ':');
  50
  51                         /* servername 353 - user list */
  52                         setcolor (2, 0);
  53                         printf ("SERVER -> %s\n", str ? str+1 : "!MESSAGE PARSE ERROR!\n");
  54                         setcolor (15, 0);
  55                         return 1;
  56                 }
  57
  58                 char *str = strchr (buffer+2, '!');
  59
  60                 /* cut of nick */
  61                 if (str)
  62                         str[0] = '\0';
  63
  64                 char *str2 = strchr (str+1, ' ');
  65
  66                 if (!str2) {
  67                         printf ("ERROR -> unknown command from server\n");
  68                         return 1;
  69                 }
  70
  71                 /* chat message */
  72                 if (!cstrcmp ("PRIVMSG", str2+1)) {
  73                         char *msg = strchr (str2+10, ':');
  74
  75                         if (!msg)
  76                                 return 1;
  77
  78                         msg --;
  79                         /* cut of channel name */
  80                         msg[0] = '\0';
  81
  82                         msg += 2;
  83
  84                         //if (!strncmp (str2+9, config.channel, config.channel_len)) {
  85                                 if (strncmp (msg+1, "ACTION", 6)) {
  86                                         setcolor (14, 0);
  87                                         printf ("> %s: %s\n", buffer+1, msg);
  88                                 } else {
  89                                         setcolor (10, 0);
  90                                         printf ("> ***%s %s\n", buffer+1, msg+8);
  91                                 }
  92                         /*} else {
  93                                 if (strncmp (msg+1, "ACTION", 6)) {
  94                                         setcolor (11, 0);
  95                                         printf ("> (%s) %s: %s\n", str2+9, buffer+1, msg);
  96                                 } else {
  97                                         setcolor (3, 0);
  98                                         printf ("> (%s) ***%s %s\n", str2+9, buffer+1, msg+8);
  99                                 }
 100                         }*/
 101
 102                         setcolor (15, 0);
 103                         return 1;
 104                 } else if (!cstrcmp ("QUIT", str2+1)) {
 105                         setcolor (4, 0);
 106                         printf ("User %s was disconnected\n", buffer+1);
 107                         setcolor (15, 0);
 108                         return 1;
 109                 }
 110
 111                 return 1;
 112
 113         }
 114
 115         if (!strncmp (buffer, "PING", 4))
 116                 return net_send (pong_msg, 10+(3*config.server_len));
 117
 118         setcolor (7, 0);
 119         printf (">> %s\n", buffer);
 120         setcolor (15, 0);
 121
 122         return 1;
 123 }
 124
 125 int proto_parser ()
 126 {
 127         unsigned i = 0;
 128
 129         while (i < parser_len) {
 130                 if (parser[i] == '\n') {
 131                         proto_command (parser, i);
 132
 133                         i ++;
 134
 135                         //memcpy (parser+parser_len-len+i, buffer, len);
 136                         if (i == parser_len) {
 137                                 parser_len = 0;
 138                                 parser = parser_tmp;
 139                                 //printf ("parser_len == i - NULL\n");
 140                                 //memset (parser, 0, PROTO_BUFSIZE_PARSER);
 141                                 break;
 142                         } else {
 143                                 //printf ("parser_len: %d / %d\n", i, parser_len);
 144
 145                                 parser += i;
 146                                 parser_len -= i;
 147
 148                                 proto_parser ();
 149                                 break;
 150                         }
 151                 }
 152
 153                 i ++;
 154         }
 155
 156         return 1;
 157 }
 158
 159 int proto_handler (char *buffer, unsigned len)
 160 {
 161         if (parser_len > (PROTO_BUFSIZE_PARSER-len)) {
 162                 printf ("ERROR -> receive buffer is full, this cause loss of last packets\n");
 163
 164                 return 0;
 165         }
 166
 167         memcpy (parser+parser_len, buffer, len);
 168         parser_len += len;
 169
 170         //printf ("proto_handler () - %d\n", len);
 171
 172         return 1;
 173 }
 174
 175 int init_proto ()
 176 {
 177         parser = (char *) malloc (sizeof (char) * PROTO_BUFSIZE_PARSER);
 178
 179         if (!parser)
 180                 return 0;
 181
 182         parser_tmp = parser;
 183
 184         parser_len = 0;
 185
 186         char str[64];
 187         sprintf (str, "USER %s blah blah :zexos irc client\nNICK %s\n", config.nick, config.nick);
 188
 189         net_send (str, strlen (str));
 190
 191         pong_msg = (char *) malloc (sizeof (char) * (config.server_len*3 + 10));
 192
 193         if (!pong_msg)
 194                 return 0;
 195
 196         pong_msg[0] = ':';
 197         memcpy (pong_msg+1, config.server, config.server_len);
 198         memcpy (pong_msg+1+config.server_len, " PONG ", 6);
 199         memcpy (pong_msg+7+config.server_len, config.server, config.server_len);
 200         memcpy (pong_msg+7+(2*config.server_len), " :", 2);
 201         memcpy (pong_msg+9+(2*config.server_len), config.server, config.server_len);
 202         pong_msg[9+(3*config.server_len)] = '\n';
 203
 204         return 1;
 205 }