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

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