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

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