Usermode/IRC - Added basic PTY support
[tpg/acess2.git] / Usermode / Applications / irc_src / main.c
1 /*\r
2  * Acess2 IRC Client\r
3  */\r
4 #include <acess/sys.h>\r
5 #include <stdlib.h>\r
6 #include <stdio.h>\r
7 #include <string.h>\r
8 #include <net.h>\r
9 #include <readline.h>\r
10 #include <acess/devices/pty.h>\r
11 \r
12 // === TYPES ===\r
13 typedef struct sServer {\r
14         struct sServer  *Next;\r
15          int    FD;\r
16         char    InBuf[BUFSIZ+1];\r
17          int    ReadPos;\r
18         char    Name[];\r
19 } tServer;\r
20 \r
21 typedef struct sMessage\r
22 {\r
23         struct sMessage *Next;\r
24         time_t  Timestamp;\r
25         tServer *Server;\r
26          int    Type;\r
27         char    *Source;        // Pointer into `Data`\r
28         char    Data[];\r
29 }       tMessage;\r
30 \r
31 typedef struct sWindow\r
32 {\r
33         struct sWindow  *Next;\r
34         tMessage        *Messages;\r
35         tServer *Server;        //!< Canoical server (can be NULL)\r
36          int    ActivityLevel;\r
37         char    Name[]; // Channel name / remote user\r
38 }       tWindow;\r
39 \r
40 enum eMessageTypes\r
41 {\r
42         MSG_TYPE_NULL,\r
43         MSG_TYPE_SERVER,        // Server message\r
44         \r
45         MSG_TYPE_NOTICE,        // NOTICE command\r
46         MSG_TYPE_JOIN,  // JOIN command\r
47         MSG_TYPE_PART,  // PART command\r
48         MSG_TYPE_QUIT,  // QUIT command\r
49         \r
50         MSG_TYPE_STANDARD,      // Standard line\r
51         MSG_TYPE_ACTION,        // /me\r
52         \r
53         MSG_TYPE_UNK\r
54 };\r
55 \r
56 // === PROTOTYPES ===\r
57  int    ParseArguments(int argc, const char *argv[]);\r
58  int    ParseUserCommand(char *String);\r
59 // --- \r
60 tServer *Server_Connect(const char *Name, const char *AddressString, short PortNumber);\r
61 tMessage        *Message_AppendF(tServer *Server, int Type, const char *Source, const char *Dest, const char *Message, ...);\r
62 tMessage        *Message_Append(tServer *Server, int Type, const char *Source, const char *Dest, const char *Message);\r
63 tWindow *Window_Create(tServer *Server, const char *Name);\r
64 void    Redraw_Screen(void);\r
65 \r
66  int    ProcessIncoming(tServer *Server);\r
67 // --- Helpers\r
68  int    SetCursorPos(int Row, int Col);\r
69  int    writef(int FD, const char *Format, ...);\r
70  int    OpenTCP(const char *AddressString, short PortNumber);\r
71 char    *GetValue(char *Str, int *Ofs);\r
72 static inline int       isdigit(int ch);\r
73 \r
74 // === GLOBALS ===\r
75 char    *gsUsername = "user";\r
76 char    *gsHostname = "acess";\r
77 char    *gsRealName = "Acess2 IRC Client";\r
78 char    *gsNickname = "acess";\r
79 tServer *gpServers;\r
80 tWindow gWindow_Status = {\r
81         NULL, NULL, NULL,       // No next, empty list, no server\r
82         0, {""} // No activity, empty name (rendered as status)\r
83 };\r
84 tWindow *gpWindows = &gWindow_Status;\r
85 tWindow *gpCurrentWindow = &gWindow_Status;\r
86  int    giTerminal_Width = 80;\r
87  int    giTerminal_Height = 25;\r
88 \r
89 // ==== CODE ====\r
90 void ExitHandler(void)\r
91 {\r
92         printf("\x1B[?1047l");\r
93         printf("Quit\n");\r
94 }\r
95 \r
96 int main(int argc, const char *argv[], const char *envp[])\r
97 {\r
98          int    tmp;\r
99         tReadline       *readline_info;\r
100         \r
101         // Parse Command line\r
102         if( (tmp = ParseArguments(argc, argv)) )        return tmp;\r
103         \r
104         atexit(ExitHandler);\r
105         \r
106         if( _SysIOCtl(1, DRV_IOCTL_TYPE, NULL) != DRV_TYPE_TERMINAL ) {\r
107                 printf(stderr, "note: assuming 80x25, can't get terminal dimensions\n");\r
108                 giTerminal_Width = 80;\r
109                 giTerminal_Height = 25;\r
110         }\r
111         else {\r
112                 struct ptydims  dims;\r
113                 _SysIOCtl(1, PTY_IOCTL_GETDIMS, &dims);\r
114                 giTerminal_Width = dims.W;\r
115                 giTerminal_Height = dims.H;\r
116         }\r
117         \r
118         printf("\x1B[?1047h");\r
119         printf("\x1B[%i;%ir", 0, giTerminal_Height-1);\r
120         \r
121         SetCursorPos(giTerminal_Height-1, 0);\r
122         printf("[(status)] ");\r
123         \r
124         // HACK: Static server entry\r
125         // UCC (University [of Western Australia] Computer Club) IRC Server\r
126         gWindow_Status.Server = Server_Connect( "UCC", "130.95.13.18", 6667 );\r
127         // Freenode (#osdev)\r
128 //      gWindow_Status.Server = Server_Connect( "Freenode", "89.16.176.16", 6667 );\r
129 //      gWindow_Status.Server = Server_Connect( "Host", "10.0.2.2", 6667 );\r
130         // Local server\r
131 //      gWindow_Status.Server = Server_Connect( "BitlBee", "192.168.1.39", 6667 );\r
132         \r
133         if( !gWindow_Status.Server )\r
134                 return -1;\r
135         \r
136         SetCursorPos(giTerminal_Height-1, 0);\r
137         printf("[(status)] ");\r
138         readline_info = Readline_Init(1);\r
139         \r
140         for( ;; )\r
141         {\r
142                 fd_set  readfds, errorfds;\r
143                  int    rv, maxFD = 0;\r
144                 tServer *srv;\r
145                 \r
146                 FD_ZERO(&readfds);\r
147                 FD_ZERO(&errorfds);\r
148                 FD_SET(0, &readfds);    // stdin\r
149                 \r
150                 // Fill server FDs in fd_set\r
151                 for( srv = gpServers; srv; srv = srv->Next )\r
152                 {\r
153                         FD_SET(srv->FD, &readfds);\r
154                         FD_SET(srv->FD, &errorfds);\r
155                         if( srv->FD > maxFD )\r
156                                 maxFD = srv->FD;\r
157                 }\r
158                 \r
159                 rv = select(maxFD+1, &readfds, 0, &errorfds, NULL);\r
160                 if( rv == -1 )  break;\r
161                 \r
162                 if(FD_ISSET(0, &readfds))\r
163                 {\r
164                         // User input\r
165                         char    *cmd = Readline_NonBlock(readline_info);\r
166                         if( cmd )\r
167                         {\r
168                                 if( cmd[0] )\r
169                                 {\r
170                                         ParseUserCommand(cmd);\r
171                                 }\r
172                                 free(cmd);\r
173                                 // Prompt\r
174                                 SetCursorPos(giTerminal_Height-1, 0);\r
175                                 printf("\x1B[K");       // Clear line\r
176                                 if( gpCurrentWindow->Name[0] )\r
177                                         printf("[%s:%s] ", gpCurrentWindow->Server->Name, gpCurrentWindow->Name);\r
178                                 else\r
179                                         printf("[(status)] ");\r
180                         }\r
181                 }\r
182                 \r
183                 // Server response\r
184                 for( srv = gpServers; srv; srv = srv->Next )\r
185                 {\r
186                         if(FD_ISSET(srv->FD, &readfds))\r
187                         {\r
188                                 if( ProcessIncoming(srv) != 0 ) {\r
189                                         // Oops, error\r
190                                         break;\r
191                                 }\r
192                         }\r
193                         \r
194                         if(FD_ISSET(srv->FD, &errorfds))\r
195                         {\r
196                                 break;\r
197                         }\r
198                 }\r
199                 \r
200                 // Oops, an error\r
201                 if( srv )       break;\r
202         }\r
203         \r
204         {\r
205                 tServer *srv;\r
206                 for( srv = gpServers; srv; srv = srv->Next )\r
207                         _SysClose(srv->FD);\r
208         }\r
209         return 0;\r
210 }\r
211 \r
212 /**\r
213  * \todo Actually implement correctly :)\r
214  */\r
215 int ParseArguments(int argc, const char *argv[])\r
216 {\r
217         return 0;\r
218 }\r
219 \r
220 \r
221 void Cmd_join(char *ArgString)\r
222 {\r
223          int    pos=0;\r
224         char    *channel_name = GetValue(ArgString, &pos);\r
225         \r
226         if( gpCurrentWindow->Server )\r
227         {\r
228                 writef(gpCurrentWindow->Server->FD, "JOIN :%s\n", channel_name);\r
229         }\r
230 }\r
231 \r
232 void Cmd_quit(char *ArgString)\r
233 {\r
234         const char *quit_message = ArgString;\r
235         if( quit_message == NULL || quit_message[0] == '\0' )\r
236                 quit_message = "/quit - Acess2 IRC Client";\r
237         \r
238         for( tServer *srv = gpServers; srv; srv = srv->Next )\r
239         {\r
240                 writef(srv->FD, "QUIT :%s\n", quit_message);\r
241         }\r
242         \r
243         exit(0);\r
244 }\r
245 \r
246 void Cmd_window(char *ArgString)\r
247 {\r
248          int    pos = 0;\r
249         char    *window_id = GetValue(ArgString, &pos);\r
250          int    window_num = atoi(window_id);\r
251         \r
252         if( window_num > 0 )\r
253         {\r
254                 tWindow *win;\r
255                 window_num --;  // Move to base 0\r
256                 // Get `window_num`th window\r
257                 for( win = gpWindows; win && window_num--; win = win->Next );\r
258                 if( win ) {\r
259                         gpCurrentWindow = win;\r
260                         Redraw_Screen();\r
261                 }\r
262                 // Otherwise, silently ignore\r
263         }\r
264         else\r
265         {\r
266                 window_num = 1;\r
267                 for( tWindow *win = gpWindows; win; win = win->Next, window_num ++ )\r
268                 {\r
269                         if( win->Name[0] ) {\r
270                                 Message_AppendF(NULL, MSG_TYPE_SERVER, "client", "",\r
271                                         "%i: %s/%s", window_num, win->Server->Name, win->Name);\r
272                         }\r
273                         else {\r
274                                 Message_AppendF(NULL, MSG_TYPE_SERVER, "client", "",\r
275                                         "%i: (status)", window_num);\r
276                         }\r
277                 }\r
278         }\r
279 }\r
280 \r
281 const struct {\r
282         const char *Name;\r
283         void    (*Fcn)(char *ArgString);\r
284 } caCommands[] = {\r
285         {"join", Cmd_join},\r
286         {"quit", Cmd_quit},\r
287         {"window", Cmd_window},\r
288         {"win",    Cmd_window},\r
289         {"w",      Cmd_window},\r
290 };\r
291 const int ciNumCommands = sizeof(caCommands)/sizeof(caCommands[0]);\r
292 \r
293 /**\r
294  * \brief Handle a line from the prompt\r
295  */\r
296 int ParseUserCommand(char *String)\r
297 {\r
298         if( String[0] == '/' )\r
299         {\r
300                 char    *command;\r
301                  int    pos = 0;\r
302                 \r
303                 command = GetValue(String, &pos)+1;\r
304 \r
305                 // TODO: Prefix matches\r
306                  int    cmdIdx = -1;\r
307                 for( int i = 0; i < ciNumCommands; i ++ )\r
308                 {\r
309                         if( strcmp(command, caCommands[i].Name) == 0 ) {\r
310                                 cmdIdx = i;\r
311                                 break;\r
312                         }\r
313                 }\r
314                 if( cmdIdx != -1 ) {\r
315                         caCommands[cmdIdx].Fcn(String+pos);\r
316                 }\r
317                 else\r
318                 {\r
319                         Message_AppendF(NULL, MSG_TYPE_SERVER, "client", "", "Unknown command %s", command);\r
320                 }\r
321         }\r
322         else\r
323         {\r
324                 // Message\r
325                 // - Only send if server is valid and window name is non-empty\r
326                 if( gpCurrentWindow->Server && gpCurrentWindow->Name[0] )\r
327                 {\r
328                         Message_Append(gpCurrentWindow->Server, MSG_TYPE_STANDARD,\r
329                                 gsNickname, gpCurrentWindow->Name, String);\r
330                         writef(gpCurrentWindow->Server->FD,\r
331                                 "PRIVMSG %s :%s\n", gpCurrentWindow->Name,\r
332                                 String\r
333                                 );\r
334                 }\r
335         }\r
336         \r
337         return 0;\r
338 }\r
339 \r
340 /**\r
341  * \brief Connect to a server\r
342  */\r
343 tServer *Server_Connect(const char *Name, const char *AddressString, short PortNumber)\r
344 {\r
345         tServer *ret;\r
346         \r
347         ret = calloc(1, sizeof(tServer) + strlen(Name) + 1);\r
348         \r
349         strcpy(ret->Name, Name);\r
350         \r
351         // Connect to the remove server\r
352         ret->FD = OpenTCP( AddressString, PortNumber );\r
353         if( ret->FD == -1 ) {\r
354                 fprintf(stderr, "%s: Unable to create socket\n", Name);\r
355                 return NULL;\r
356         }\r
357         \r
358         // Append to open list\r
359         ret->Next = gpServers;\r
360         gpServers = ret;\r
361         \r
362         // Read some initial data\r
363         Message_Append(NULL, MSG_TYPE_SERVER, Name, "", "Connection opened");\r
364         ProcessIncoming(ret);\r
365         \r
366         // Identify\r
367         writef(ret->FD, "USER %s %s %s : %s\n", gsUsername, gsHostname, AddressString, gsRealName);\r
368         writef(ret->FD, "NICK %s\n", gsNickname);\r
369         Message_Append(NULL, MSG_TYPE_SERVER, Name, "", "Identified");\r
370         //printf("%s: Identified\n", Name);\r
371         \r
372         return ret;\r
373 }\r
374 \r
375 tMessage *Message_AppendF(tServer *Server, int Type, const char *Source, const char *Dest, const char *Message, ...)\r
376 {\r
377         va_list args;\r
378          int    len;\r
379         va_start(args, Message);\r
380         len = vsnprintf(NULL, 0, Message, args);\r
381         {\r
382                 char    buf[len+1];\r
383                 vsnprintf(buf, len+1, Message, args);\r
384                 return Message_Append(Server, Type, Source, Dest, buf);\r
385         }\r
386 }\r
387 \r
388 tMessage *Message_Append(tServer *Server, int Type, const char *Source, const char *Dest, const char *Message)\r
389 {\r
390         tMessage        *ret;\r
391         tWindow *win = NULL;\r
392          int    msgLen = strlen(Message);\r
393         \r
394         // NULL servers are internal messages\r
395         if( Server == NULL || Source[0] == '\0' )\r
396         {\r
397                 win = &gWindow_Status;\r
398         }\r
399         // Determine if it's a channel or PM message\r
400         else if( Dest[0] == '#' || Dest[0] == '&' )     // TODO: Better determining here\r
401         {\r
402                 for(win = gpWindows; win; win = win->Next)\r
403                 {\r
404                         if( win->Server == Server && strcmp(win->Name, Dest) == 0 )\r
405                         {\r
406                                 break;\r
407                         }\r
408                 }\r
409                 if( !win ) {\r
410                         win = Window_Create(Server, Dest);\r
411                 }\r
412         }\r
413         #if 0\r
414         else if( strcmp(Dest, Server->Nick) != 0 )\r
415         {\r
416                 // Umm... message for someone who isn't us?\r
417                 win = &gWindow_Status;  // Stick it in the status window, just in case\r
418         }\r
419         #endif\r
420         // Server message?\r
421         else if( strchr(Source, '.') )  // TODO: And again, less hack please\r
422         {\r
423                 #if 1\r
424                 for(win = gpWindows; win; win = win->Next)\r
425                 {\r
426                         if( win->Server == Server && strcmp(win->Name, Source) == 0 )\r
427                         {\r
428                                 break;\r
429                         }\r
430                 }\r
431                 #endif\r
432                 if( !win ) {\r
433                         win = &gWindow_Status;\r
434                 }\r
435                 \r
436         }\r
437         // Private message\r
438         else\r
439         {\r
440                 for(win = gpWindows; win; win = win->Next)\r
441                 {\r
442                         if( win->Server == Server && strcmp(win->Name, Source) == 0 )\r
443                         {\r
444                                 break;\r
445                         }\r
446                 }\r
447                 if( !win ) {\r
448                         win = Window_Create(Server, Dest);\r
449                 }\r
450         }\r
451         \r
452         ret = malloc( sizeof(tMessage) + msgLen + 1 + strlen(Source) + 1 );\r
453         ret->Source = ret->Data + msgLen + 1;\r
454         strcpy(ret->Source, Source);\r
455         strcpy(ret->Data, Message);\r
456         ret->Type = Type;\r
457         ret->Server = Server;\r
458         \r
459         // TODO: Append to window message list\r
460         ret->Next = win->Messages;\r
461         win->Messages = ret;\r
462         \r
463         //TODO: Set location\r
464         \r
465         {\r
466                 int pos = SetCursorPos(giTerminal_Height-2, 0);\r
467                 if( win == gpCurrentWindow ) {\r
468                          int    prefixlen = strlen(Source) + 3;\r
469                          int    avail = giTerminal_Width - prefixlen;\r
470                          int    msglen = strlen(Message);\r
471                         printf("\x1B[T");       // Scroll down 1 (free space below)\r
472                         printf("[%s] %.*s\n", Source, avail, Message);\r
473                         while( msglen > avail ) {\r
474                                 msglen -= avail;\r
475                                 Message += avail;\r
476                                 printf("\x1B[T");\r
477                                 SetCursorPos(giTerminal_Height-2, prefixlen);\r
478                                 printf("%.*s\n", avail, Message);\r
479                         }\r
480                 }\r
481                 SetCursorPos(-1, pos);\r
482         }\r
483         \r
484         return ret;\r
485 }\r
486 \r
487 tWindow *Window_Create(tServer *Server, const char *Name)\r
488 {\r
489         tWindow *ret, *prev = NULL;\r
490          int    num = 1;\r
491         \r
492         // Get the end of the list\r
493         // TODO: Cache this instead\r
494         for( ret = gpCurrentWindow; ret; prev = ret, ret = ret->Next )\r
495                 num ++;\r
496         \r
497         ret = malloc(sizeof(tWindow) + strlen(Name) + 1);\r
498         ret->Messages = NULL;\r
499         ret->Server = Server;\r
500         ret->ActivityLevel = 1;\r
501         strcpy(ret->Name, Name);\r
502         \r
503         if( prev ) {\r
504                 ret->Next = prev->Next;\r
505                 prev->Next = ret;\r
506         }\r
507         else {  // Shouldn't happen really\r
508                 ret->Next = gpWindows;\r
509                 gpWindows = ret;\r
510         }\r
511         \r
512 //      printf("Win %i %s:%s created\n", num, Server->Name, Name);\r
513         \r
514         return ret;\r
515 }\r
516 \r
517 void Redraw_Screen(void)\r
518 {\r
519          int    y = 0;\r
520         tMessage        *msg;\r
521 \r
522         printf("\x1B[2J");      // Clear screen\r
523         printf("\x1B[0;0H");    // Reset cursor\r
524 \r
525         msg = gpCurrentWindow->Messages;\r
526         \r
527         // TODO: Title bar?\r
528 \r
529         // Note: This renders from the bottom up\r
530         for( y = giTerminal_Height - 1; y -- && msg; msg = msg->Next)\r
531         {\r
532                  int    msglen = strlen(msg->Data);\r
533                  int    prefix_len = 3 + strlen(msg->Source);\r
534                  int    line_avail = giTerminal_Width - prefix_len;\r
535                  int    i = 0, done = 0;\r
536                 \r
537                 y -= msglen / line_avail;       // Extra lines (y-- above handles the 1 line case)\r
538                 SetCursorPos(y, 0);\r
539                 printf("[%s] ", msg->Source);\r
540                 \r
541                 while(done < msglen) {\r
542                         done += printf("%.*s", line_avail, msg->Data+done);\r
543                         i ++;\r
544                         SetCursorPos(y+i, prefix_len);\r
545                 }\r
546         }\r
547 \r
548         // Bottom line is rendered by the prompt\r
549 }\r
550 \r
551 void Cmd_PRIVMSG(tServer *Server, const char *Dest, const char *Src, const char *Message)\r
552 {\r
553         Message_Append(Server, MSG_TYPE_STANDARD, Dest, Src, Message);\r
554         //printf("<%s:%s:%s> %s\n", Server->Name, Dest, Src, Message);\r
555 }\r
556 \r
557 /**\r
558  */\r
559 void ParseServerLine(tServer *Server, char *Line)\r
560 {\r
561          int    pos = 0;\r
562         char    *ident, *cmd;\r
563 \r
564         _SysDebug("[%s] %s", Server->Name, Line);       \r
565         \r
566         // Message?\r
567         if( *Line == ':' )\r
568         {\r
569                 pos ++;\r
570                 ident = GetValue(Line, &pos);   // Ident (user or server)\r
571                 cmd = GetValue(Line, &pos);\r
572                 \r
573                 // Numeric command\r
574                 if( isdigit(cmd[0]) && isdigit(cmd[1]) && isdigit(cmd[2]) )\r
575                 {\r
576                         char    *user, *message;\r
577                          int    num;\r
578                         num  = (cmd[0] - '0') * 100;\r
579                         num += (cmd[1] - '0') * 10;\r
580                         num += (cmd[2] - '0') * 1;\r
581                         \r
582                         user = GetValue(Line, &pos);\r
583                         \r
584                         if( Line[pos] == ':' ) {\r
585                                 message = Line + pos + 1;\r
586                         }\r
587                         else {\r
588                                 message = GetValue(Line, &pos);\r
589                         }\r
590                         \r
591                         switch(num)\r
592                         {\r
593                         case 332:       // Topic\r
594                                 user = message; // Channel\r
595                                 message = Line + pos + 1;       // Topic\r
596                                 Message_AppendF(Server, MSG_TYPE_SERVER, user, user, "Topic: %s", message);\r
597                                 break;\r
598                         case 333:       // Topic set by\r
599                                 user = message; // Channel\r
600                                 message = GetValue(Line, &pos); // User\r
601                                 GetValue(Line, &pos);   // Timestamp\r
602                                 Message_AppendF(Server, MSG_TYPE_SERVER, user, user, "Set by %s", message);\r
603                                 break;\r
604                         case 353:       // /NAMES list\r
605                                 // <user> = <channel> :list\r
606                                 // '=' was eaten in and set to message\r
607                                 user = GetValue(Line, &pos);    // Actually channel\r
608                                 message = Line + pos + 1;       // List\r
609                                 Message_AppendF(Server, MSG_TYPE_SERVER, user, user, "Names: %s", message);\r
610                                 break;\r
611                         case 366:       // end of /NAMES list\r
612                                 // <user> <channel> :msg\r
613                                 user = message;\r
614                                 message = Line + pos + 1;\r
615                                 Message_Append(Server, MSG_TYPE_SERVER, user, user, message);\r
616                                 break;\r
617                         case 372:       // MOTD Data\r
618                         case 375:       // MOTD Start\r
619                         case 376:       // MOTD End\r
620                                 \r
621                         default:\r
622                                 //printf("[%s] %i %s\n", Server->Name, num, message);\r
623                                 Message_Append(Server, MSG_TYPE_SERVER, ident, user, message);\r
624                                 break;\r
625                         }\r
626                 }\r
627                 else if( strcmp(cmd, "NOTICE") == 0 )\r
628                 {\r
629                         char    *class, *message;\r
630                         \r
631                         class = GetValue(Line, &pos);\r
632                         _SysDebug("NOTICE class='%s'", class);\r
633                         \r
634                         if( Line[pos] == ':' ) {\r
635                                 message = Line + pos + 1;\r
636                         }\r
637                         else {\r
638                                 message = GetValue(Line, &pos);\r
639                         }\r
640                         \r
641                         //printf("[%s] NOTICE %s: %s\n", Server->Name, ident, message);\r
642                         char *ident_bang = strchr(ident, '!');\r
643                         if( ident_bang ) {\r
644                                 *ident_bang = '\0';\r
645                         }\r
646                         Message_Append(Server, MSG_TYPE_NOTICE, ident, "", message);\r
647                 }\r
648                 else if( strcmp(cmd, "PRIVMSG") == 0 )\r
649                 {\r
650                         char    *dest, *message;\r
651                         dest = GetValue(Line, &pos);\r
652                         \r
653                         if( Line[pos] == ':' ) {\r
654                                 message = Line + pos + 1;\r
655                         }\r
656                         else {\r
657                                 message = GetValue(Line, &pos);\r
658                         }\r
659 \r
660                         // TODO: Catch when the privmsg is addressed to the user\r
661 \r
662 //                      Cmd_PRIVMSG(Server, dest, ident, message);\r
663                         char *ident_bang = strchr(ident, '!');\r
664                         if( ident_bang ) {\r
665                                 *ident_bang = '\0';\r
666                         }\r
667                         Message_Append(Server, MSG_TYPE_STANDARD, ident, dest, message);\r
668                 }\r
669                 else if( strcmp(cmd, "JOIN" ) == 0 )\r
670                 {\r
671                         char    *channel;\r
672                         channel = Line + pos + 1;\r
673                         Window_Create(Server, channel);\r
674                 }\r
675                 else\r
676                 {\r
677                         Message_AppendF(Server, MSG_TYPE_SERVER, "", "", "Unknown message %s (%s)\n", cmd, Line+pos);\r
678                 }\r
679         }\r
680         else {          \r
681                 cmd = GetValue(Line, &pos);\r
682                 \r
683                 if( strcmp(cmd, "PING") == 0 ) {\r
684                         writef(Server->FD, "PONG %s\n", gsHostname);\r
685                         \r
686                 }\r
687                 else {\r
688                         // Command to client\r
689                         Message_AppendF(NULL, MSG_TYPE_UNK, "", "", "Client Command: %s", Line);\r
690                 }\r
691         }\r
692 }\r
693 \r
694 /**\r
695  * \brief Process incoming lines from the server\r
696  */\r
697 int ProcessIncoming(tServer *Server)\r
698 {       \r
699         char    *ptr, *newline;\r
700          int    len;\r
701         \r
702         // While there is data in the buffer, read it into user memory and \r
703         // process it line by line\r
704         // ioctl#8 on a TCP client gets the number of bytes in the recieve buffer\r
705         // - Used to avoid blocking\r
706         #if NON_BLOCK_READ\r
707         while( (len = _SysIOCtl(Server->FD, 8, NULL)) > 0 )\r
708         {\r
709         #endif\r
710                 // Read data\r
711                 len = _SysRead(Server->FD, &Server->InBuf[Server->ReadPos], BUFSIZ - Server->ReadPos);\r
712                 if( len == -1 ) {\r
713                         return -1;\r
714                 }\r
715                 Server->InBuf[Server->ReadPos + len] = '\0';\r
716                 \r
717                 // Break into lines\r
718                 ptr = Server->InBuf;\r
719                 while( (newline = strchr(ptr, '\n')) )\r
720                 {\r
721                         *newline = '\0';\r
722                         if( newline[-1] == '\r' )       newline[-1] = '\0';\r
723                         ParseServerLine(Server, ptr);\r
724                         ptr = newline + 1;\r
725                 }\r
726                 \r
727                 // Handle incomplete lines\r
728                 if( ptr - Server->InBuf < len + Server->ReadPos ) {\r
729                         // Update the read position\r
730                         // InBuf ReadPos    ptr          ReadPos+len\r
731                         // | old | new used | new unused |\r
732                         Server->ReadPos = len + Server->ReadPos - (ptr - Server->InBuf);\r
733                         // Copy stuff back (moving "new unused" to the start of the buffer)\r
734                         memcpy(Server->InBuf, ptr, Server->ReadPos);\r
735                 }\r
736                 else {\r
737                         Server->ReadPos = 0;\r
738                 }\r
739         #if NON_BLOCK_READ\r
740         }\r
741         #endif\r
742         \r
743         return 0;\r
744 }\r
745 \r
746 /**\r
747  * \brief Write a formatted string to a file descriptor\r
748  * \r
749  */\r
750 int writef(int FD, const char *Format, ...)\r
751 {\r
752         va_list args;\r
753          int    len;\r
754         \r
755         va_start(args, Format);\r
756         len = vsnprintf(NULL, 0, Format, args);\r
757         va_end(args);\r
758         \r
759         {\r
760                 char    buf[len+1];\r
761                 va_start(args, Format);\r
762                 vsnprintf(buf, len+1, Format, args);\r
763                 va_end(args);\r
764                 \r
765                 return _SysWrite(FD, buf, len);\r
766         }\r
767 }\r
768 \r
769 /**\r
770  * \brief Initialise a TCP connection to \a AddressString on port \a PortNumber\r
771  */\r
772 int OpenTCP(const char *AddressString, short PortNumber)\r
773 {\r
774          int    fd, addrType;\r
775         char    addrBuffer[8];\r
776         \r
777         // Parse IP Address\r
778         addrType = Net_ParseAddress(AddressString, addrBuffer);\r
779         if( addrType == 0 ) {\r
780                 fprintf(stderr, "Unable to parse '%s' as an IP address\n", AddressString);\r
781                 _SysDebug("Unable to parse '%s' as an IP address\n", AddressString);\r
782                 return -1;\r
783         }\r
784         \r
785         // Finds the interface for the destination address\r
786         fd = Net_OpenSocket(addrType, addrBuffer, "tcpc");\r
787         if( fd == -1 ) {\r
788                 fprintf(stderr, "Unable to open TCP Client to '%s'\n", AddressString);\r
789                 _SysDebug("Unable to open TCP client to '%s'\n", AddressString);\r
790                 return -1;\r
791         }\r
792         \r
793         // Set remote port and address\r
794 //      printf("Setting port and remote address\n");\r
795         _SysIOCtl(fd, 5, &PortNumber);\r
796         _SysIOCtl(fd, 6, addrBuffer);\r
797         \r
798         // Connect\r
799 //      printf("Initiating connection\n");\r
800         if( _SysIOCtl(fd, 7, NULL) == 0 ) {\r
801                 // Shouldn't happen :(\r
802                 fprintf(stderr, "Unable to start connection\n");\r
803                 return -1;\r
804         }\r
805         \r
806         // Return descriptor\r
807         return fd;\r
808 }\r
809 \r
810 /**\r
811  * \brief Read a space-separated value from a string\r
812  */\r
813 char *GetValue(char *Src, int *Ofs)\r
814 {\r
815          int    pos = *Ofs;\r
816         char    *ret = Src + pos;\r
817         char    *end;\r
818         \r
819         if( !Src )      return NULL;\r
820         \r
821         while( *ret == ' ' )    ret ++;\r
822         \r
823         end = strchr(ret, ' ');\r
824         if( end ) {\r
825                 *end = '\0';\r
826         }\r
827         else {\r
828                 end = ret + strlen(ret) - 1;\r
829         }\r
830         \r
831         end ++ ;\r
832         while( *ret == ' ' )    end ++;\r
833         *Ofs = end - Src;\r
834         \r
835         return ret;\r
836 }\r
837 \r
838 int SetCursorPos(int Row, int Col)\r
839 {\r
840          int    rv;\r
841         if( Row == -1 ) {\r
842                 Row = Col / giTerminal_Width;\r
843                 Col = Col % giTerminal_Width;\r
844         }\r
845         rv = _SysIOCtl(1, 9, NULL);     // Ugh, constants\r
846         printf("\x1B[%i;%iH", Col, Row);\r
847         return rv;\r
848 }\r
849 \r
850 static inline int isdigit(int ch)\r
851 {\r
852         return '0' <= ch && ch < '9';\r
853 }\r

UCC git Repository :: git.ucc.asn.au