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

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