#include <stdio.h>\r
#include <string.h>\r
#include <net.h>\r
+#include <readline.h>\r
\r
#define BUFSIZ 1023\r
\r
// === TYPES ===\r
typedef struct sServer {\r
+ struct sServer *Next;\r
int FD;\r
char InBuf[BUFSIZ+1];\r
int ReadPos;\r
+ char Name[];\r
} tServer;\r
\r
+typedef struct sMessage\r
+{\r
+ struct sMessage *Next;\r
+ time_t Timestamp;\r
+ tServer *Server;\r
+ int Type;\r
+ char *Source; // Pointer into `Data`\r
+ char Data[];\r
+} tMessage;\r
+\r
+typedef struct sWindow\r
+{\r
+ struct sWindow *Next;\r
+ tMessage *Messages;\r
+ tServer *Server; //!< Canoical server (can be NULL)\r
+ int ActivityLevel;\r
+ char Name[]; // Channel name / remote user\r
+} tWindow;\r
+\r
+enum eMessageTypes\r
+{\r
+ MSG_TYPE_NULL,\r
+ MSG_TYPE_SERVER, // Server message\r
+ \r
+ MSG_TYPE_NOTICE, // NOTICE command\r
+ MSG_TYPE_JOIN, // JOIN command\r
+ MSG_TYPE_PART, // PART command\r
+ MSG_TYPE_QUIT, // QUIT command\r
+ \r
+ MSG_TYPE_STANDARD, // Standard line\r
+ MSG_TYPE_ACTION, // /me\r
+ \r
+ MSG_TYPE_UNK\r
+};\r
+\r
// === PROTOTYPES ===\r
int ParseArguments(int argc, const char *argv[]);\r
-void ProcessIncoming(tServer *Server);\r
+ int ParseUserCommand(char *String);\r
+// --- \r
+tServer *Server_Connect(const char *Name, const char *AddressString, short PortNumber);\r
+tMessage *Message_Append(tServer *Server, int Type, const char *Source, const char *Dest, const char *Message);\r
+tWindow *Window_Create(tServer *Server, const char *Name);\r
+\r
+ int ProcessIncoming(tServer *Server);\r
+// --- Helpers\r
+ int SetCursorPos(int Row, int Col);\r
int writef(int FD, const char *Format, ...);\r
int OpenTCP(const char *AddressString, short PortNumber);\r
+char *GetValue(char *Str, int *Ofs);\r
+static inline int isdigit(int ch);\r
\r
// === GLOBALS ===\r
-char *gsUsername = "root";\r
+char *gsUsername = "user";\r
char *gsHostname = "acess";\r
-char *gsRemoteAddress = NULL;\r
char *gsRealName = "Acess2 IRC Client";\r
char *gsNickname = "acess";\r
-short giRemotePort = 6667;\r
+tServer *gpServers;\r
+tWindow gWindow_Status = {\r
+ NULL, NULL, NULL, // No next, empty list, no server\r
+ 0, "" // No activity, empty name (rendered as status)\r
+};\r
+tWindow *gpWindows = &gWindow_Status;\r
+tWindow *gpCurrentWindow = &gWindow_Status;\r
+ int giTerminal_Width = 80;\r
+ int giTerminal_Height = 25;\r
\r
// ==== CODE ====\r
+void ExitHandler(void)\r
+{\r
+ printf("\x1B[?1047l");\r
+}\r
+\r
int main(int argc, const char *argv[], const char *envp[])\r
{\r
int tmp;\r
- tServer srv;\r
- \r
- memset(&srv, 0, sizeof(srv));\r
+ tReadline *readline_info;\r
\r
// Parse Command line\r
- // - Sets the server configuration globals\r
- if( (tmp = ParseArguments(argc, argv)) ) {\r
- return tmp;\r
- }\r
+ if( (tmp = ParseArguments(argc, argv)) ) return tmp;\r
\r
- // Connect to the remove server\r
- srv.FD = OpenTCP( gsRemoteAddress, giRemotePort );\r
- if( srv.FD == -1 ) {\r
- fprintf(stderr, "Unable to create socket\n");\r
+ atexit(ExitHandler);\r
+ \r
+ giTerminal_Width = ioctl(1, 5, NULL); // getset_width\r
+ giTerminal_Height = ioctl(1, 6, NULL); // getset_height\r
+ \r
+ printf("\x1B[?1047h");\r
+ printf("\x1B[%i;%ir", 0, giTerminal_Height-1);\r
+ \r
+ // HACK: Static server entry\r
+ // UCC (University [of Western Australia] Computer Club) IRC Server\r
+ gWindow_Status.Server = Server_Connect( "UCC", "130.95.13.18", 6667 );\r
+ \r
+ if( !gWindow_Status.Server )\r
return -1;\r
- }\r
\r
- printf("Connection opened\n");\r
+ readline_info = Readline_Init(1);\r
\r
- writef(srv.FD, "USER %s %s %s : %s\n", gsUsername, gsHostname, gsRemoteAddress, gsRealName);\r
- writef(srv.FD, "NICK %s", gsNickname);\r
+ SetCursorPos(giTerminal_Height-1, 0);\r
+ printf("[(status)] ");\r
\r
- ProcessIncoming(&srv);\r
+ for( ;; )\r
+ {\r
+ fd_set readfds, errorfds;\r
+ int rv, maxFD = 0;\r
+ tServer *srv;\r
+ \r
+ FD_ZERO(&readfds);\r
+ FD_ZERO(&errorfds);\r
+ FD_SET(0, &readfds); // stdin\r
+ \r
+ // Fill server FDs in fd_set\r
+ for( srv = gpServers; srv; srv = srv->Next )\r
+ {\r
+ FD_SET(srv->FD, &readfds);\r
+ FD_SET(srv->FD, &errorfds);\r
+ if( srv->FD > maxFD )\r
+ maxFD = srv->FD;\r
+ }\r
+ \r
+ rv = select(maxFD+1, &readfds, 0, &errorfds, NULL);\r
+ if( rv == -1 ) break;\r
+ \r
+ if(FD_ISSET(0, &readfds))\r
+ {\r
+ // User input\r
+ char *cmd = Readline_NonBlock(readline_info);\r
+ if( cmd )\r
+ {\r
+ if( cmd[0] )\r
+ {\r
+ ParseUserCommand(cmd);\r
+ }\r
+ free(cmd);\r
+ // Prompt\r
+ SetCursorPos(giTerminal_Height-1, 0);\r
+ if( gpCurrentWindow->Name[0] )\r
+ printf("[%s:%s] ", gpCurrentWindow->Server->Name, gpCurrentWindow->Name);\r
+ else\r
+ printf("[(status)] ");\r
+ }\r
+ }\r
+ \r
+ // Server response\r
+ for( srv = gpServers; srv; srv = srv->Next )\r
+ {\r
+ if(FD_ISSET(srv->FD, &readfds))\r
+ {\r
+ if( ProcessIncoming(srv) != 0 ) {\r
+ // Oops, error\r
+ break;\r
+ }\r
+ }\r
+ \r
+ if(FD_ISSET(srv->FD, &errorfds))\r
+ {\r
+ break;\r
+ }\r
+ }\r
+ \r
+ // Oops, an error\r
+ if( srv ) break;\r
+ }\r
\r
- close(srv.FD);\r
+ {\r
+ tServer *srv;\r
+ for( srv = gpServers; srv; srv = srv->Next )\r
+ close(srv->FD);\r
+ }\r
return 0;\r
}\r
\r
*/\r
int ParseArguments(int argc, const char *argv[])\r
{\r
- gsRemoteAddress = "130.95.13.18"; // irc.ucc.asn.au\r
- \r
return 0;\r
}\r
\r
-void Cmd_PRIVMSG(tServer *Server, const char *Dest, const char *Src, const char *Message)\r
+int ParseUserCommand(char *String)\r
{\r
- printf("%p: %s => %s\t%s\n", Server, Src, Dest, Message);\r
+ if( String[0] == '/' )\r
+ {\r
+ char *command;\r
+ int pos = 0;\r
+ \r
+ command = GetValue(String, &pos);\r
+ \r
+ if( strcmp(command, "/join") == 0 )\r
+ {\r
+ char *channel_name = GetValue(String, &pos);\r
+ \r
+ if( gpCurrentWindow->Server )\r
+ {\r
+ writef(gpCurrentWindow->Server->FD, "JOIN :%s\n", channel_name);\r
+ }\r
+ }\r
+ else if( strcmp(command, "/quit") == 0 )\r
+ {\r
+ char *quit_message = GetValue(String, &pos);\r
+ tServer *srv;\r
+ \r
+ if( quit_message == NULL || quit_message[0] == '\0' )\r
+ quit_message = "/quit - Acess2 IRC Client";\r
+ \r
+ for( srv = gpServers; srv; srv = srv->Next )\r
+ {\r
+ writef(srv->FD, "QUIT :%s\n", quit_message);\r
+ }\r
+ \r
+ exit(0);\r
+ }\r
+ else if( strcmp(command, "/window") == 0 || strcmp(command, "/win") == 0 || strcmp(command, "/w") == 0 )\r
+ {\r
+ char *window_id = GetValue(String, &pos);\r
+ int window_num = atoi(window_id);\r
+ \r
+ if( window_num > 0 )\r
+ {\r
+ tWindow *win;\r
+ window_num --; // Move to base 0\r
+ // Get `window_num`th window\r
+ for( win = gpWindows; win && window_num--; win = win->Next );\r
+ if( win ) {\r
+ gpCurrentWindow = win;\r
+ }\r
+ // Otherwise, silently ignore\r
+ }\r
+ else\r
+ {\r
+ tWindow *win;\r
+ window_num = 1;\r
+ for( win = gpWindows; win; win = win->Next, window_num ++ )\r
+ {\r
+ char tmp[snprintf(NULL, 1000, "%i: %s/%s", window_num, win->Server->Name, win->Name)+1];\r
+ snprintf(tmp, 1000, "%i: %s/%s", window_num, win->Server->Name, win->Name);\r
+ Message_Append(NULL, MSG_TYPE_SERVER, "client", "", tmp);\r
+ }\r
+ }\r
+ }\r
+ else\r
+ {\r
+ int len = snprintf(NULL, 0, "Unknown command %s", command);\r
+ char buf[len+1];\r
+ snprintf(buf, len+1, "Unknown command %s", command);\r
+ Message_Append(NULL, MSG_TYPE_SERVER, "client", "", buf);\r
+ }\r
+ }\r
+ else\r
+ {\r
+ // Message\r
+ // - Only send if server is valid and window name is non-empty\r
+ if( gpCurrentWindow->Server && gpCurrentWindow->Name[0] )\r
+ {\r
+ writef(gpCurrentWindow->Server->FD,\r
+ "PRIVMSG %s :%s\n", gpCurrentWindow->Name,\r
+ String\r
+ );\r
+ }\r
+ }\r
+ \r
+ return 0;\r
}\r
\r
/**\r
- * \brief Read a space-separated value from a string\r
+ * \brief Connect to a server\r
*/\r
-char *GetValue(char *Src, int *Ofs)\r
+tServer *Server_Connect(const char *Name, const char *AddressString, short PortNumber)\r
{\r
- int pos = *Ofs;\r
- char *ret = Src + pos;\r
- char *end;\r
+ tServer *ret;\r
\r
- while( *ret == ' ' ) ret ++;\r
+ ret = calloc(1, sizeof(tServer) + strlen(Name) + 1);\r
\r
- end = strchr(ret, ' ');\r
- if( end ) {\r
- *end = '\0';\r
+ strcpy(ret->Name, Name);\r
+ \r
+ // Connect to the remove server\r
+ ret->FD = OpenTCP( AddressString, PortNumber );\r
+ if( ret->FD == -1 ) {\r
+ fprintf(stderr, "%s: Unable to create socket\n", Name);\r
+ return NULL;\r
}\r
- else {\r
- end = ret + strlen(ret) - 1;\r
+ \r
+ // Append to open list\r
+ ret->Next = gpServers;\r
+ gpServers = ret;\r
+ \r
+ // Read some initial data\r
+ printf("%s: Connection opened\n", Name);\r
+ ProcessIncoming(ret);\r
+ \r
+ // Identify\r
+ writef(ret->FD, "USER %s %s %s : %s\n", gsUsername, gsHostname, AddressString, gsRealName);\r
+ writef(ret->FD, "NICK %s\n", gsNickname);\r
+ printf("%s: Identified\n", Name);\r
+ \r
+ return ret;\r
+}\r
+\r
+tMessage *Message_Append(tServer *Server, int Type, const char *Source, const char *Dest, const char *Message)\r
+{\r
+ tMessage *ret;\r
+ tWindow *win = NULL;\r
+ int msgLen = strlen(Message);\r
+ \r
+ // NULL servers are internal messages\r
+ if( Server == NULL )\r
+ {\r
+ win = &gWindow_Status;\r
+ }\r
+ // Determine if it's a channel or PM message\r
+ else if( Dest[0] == '#' || Dest[0] == '&' ) // TODO: Better determining here\r
+ {\r
+ tWindow *prev = NULL;\r
+ for(win = gpWindows; win; prev = win, win = win->Next)\r
+ {\r
+ if( win->Server == Server && strcmp(win->Name, Dest) == 0 )\r
+ {\r
+ break;\r
+ }\r
+ }\r
+ if( !win ) {\r
+ win = Window_Create(Server, Dest);\r
+ }\r
+ }\r
+ #if 0\r
+ else if( strcmp(Dest, Server->Nick) != 0 )\r
+ {\r
+ // Umm... message for someone who isn't us?\r
+ win = &gWindow_Status; // Stick it in the status window, just in case\r
+ }\r
+ #endif\r
+ // Server message?\r
+ else if( strchr(Source, '.') ) // TODO: And again, less hack please\r
+ {\r
+ #if 1\r
+ for(win = gpWindows; win; win = win->Next)\r
+ {\r
+ if( win->Server == Server && strcmp(win->Name, Source) == 0 )\r
+ {\r
+ break;\r
+ }\r
+ }\r
+ #endif\r
+ if( !win ) {\r
+ win = &gWindow_Status;\r
+ }\r
+ \r
+ }\r
+ // Private message\r
+ else\r
+ {\r
+ for(win = gpWindows; win; win = win->Next)\r
+ {\r
+ if( win->Server == Server && strcmp(win->Name, Source) == 0 )\r
+ {\r
+ break;\r
+ }\r
+ }\r
+ if( !win ) {\r
+ win = Window_Create(Server, Dest);\r
+ }\r
+ }\r
+ \r
+ ret = malloc( sizeof(tMessage) + msgLen + 1 + strlen(Source) + 1 );\r
+ ret->Source = ret->Data + msgLen + 1;\r
+ strcpy(ret->Source, Source);\r
+ strcpy(ret->Data, Message);\r
+ ret->Type = Type;\r
+ ret->Server = Server;\r
+ \r
+ // TODO: Append to window message list\r
+ ret->Next = win->Messages;\r
+ win->Messages = ret;\r
+ \r
+ //TODO: Set location\r
+ \r
+ {\r
+ int pos = SetCursorPos(giTerminal_Height-2, 0);\r
+ printf("\x1B[T"); // Scroll down 1 (free space below)\r
+ #if 1\r
+ if( win == gpCurrentWindow ) {\r
+ printf("[%s] %s\n", Source, Message);\r
+ }\r
+ #else\r
+ if(win->Name[0])\r
+ printf("%s/%s [%s] %s\n", win->Server->Name, win->Name, Source, Message);\r
+ else\r
+ printf("(status) [%s] %s\n", Source, Message);\r
+ #endif\r
+ SetCursorPos(-1, pos);\r
+ }\r
+ \r
+ return ret;\r
+}\r
+\r
+tWindow *Window_Create(tServer *Server, const char *Name)\r
+{\r
+ tWindow *ret, *prev = NULL;\r
+ int num = 1;\r
+ \r
+ // Get the end of the list\r
+ // TODO: Cache this instead\r
+ for( ret = gpCurrentWindow; ret; prev = ret, ret = ret->Next )\r
+ num ++;\r
+ \r
+ ret = malloc(sizeof(tWindow) + strlen(Name) + 1);\r
+ ret->Messages = NULL;\r
+ ret->Server = Server;\r
+ ret->ActivityLevel = 1;\r
+ strcpy(ret->Name, Name);\r
+ \r
+ if( prev ) {\r
+ ret->Next = prev->Next;\r
+ prev->Next = ret;\r
+ }\r
+ else { // Shouldn't happen really\r
+ ret->Next = gpWindows;\r
+ gpWindows = ret;\r
}\r
- *Ofs = end - Src + 1;\r
+ \r
+// printf("Win %i %s:%s created\n", num, Server->Name, Name);\r
\r
return ret;\r
}\r
\r
+void Cmd_PRIVMSG(tServer *Server, const char *Dest, const char *Src, const char *Message)\r
+{\r
+ Message_Append(Server, MSG_TYPE_STANDARD, Dest, Src, Message);\r
+ //printf("<%s:%s:%s> %s\n", Server->Name, Dest, Src, Message);\r
+}\r
+\r
/**\r
*/\r
void ParseServerLine(tServer *Server, char *Line)\r
{\r
- int pos;\r
+ int pos = 0;\r
char *ident, *cmd;\r
- if( *Line == ':' ) {\r
- // Message\r
- ident = GetValue(Line, &pos);\r
- pos ++; // Space\r
+ \r
+ // Message?\r
+ if( *Line == ':' )\r
+ {\r
+ ident = GetValue(Line, &pos); // Ident (user or server)\r
cmd = GetValue(Line, &pos);\r
\r
- if( strcmp(cmd, "PRIVMSG") == 0 ) {\r
+ // Numeric command\r
+ if( isdigit(cmd[0]) && isdigit(cmd[1]) && isdigit(cmd[2]) )\r
+ {\r
+ char *user, *message;\r
+ int num;\r
+ num = (cmd[0] - '0') * 100;\r
+ num += (cmd[1] - '0') * 10;\r
+ num += (cmd[2] - '0') * 1;\r
+ \r
+ user = GetValue(Line, &pos);\r
+ \r
+ if( Line[pos] == ':' ) {\r
+ message = Line + pos + 1;\r
+ }\r
+ else {\r
+ message = GetValue(Line, &pos);\r
+ }\r
+ \r
+ switch(num)\r
+ {\r
+ default:\r
+ //printf("[%s] %i %s\n", Server->Name, num, message);\r
+ Message_Append(Server, MSG_TYPE_SERVER, ident, user, message);\r
+ break;\r
+ }\r
+ }\r
+ else if( strcmp(cmd, "NOTICE") == 0 )\r
+ {\r
+ char *class, *message;\r
+ \r
+ class = GetValue(Line, &pos);\r
+ \r
+ if( Line[pos] == ':' ) {\r
+ message = Line + pos + 1;\r
+ }\r
+ else {\r
+ message = GetValue(Line, &pos);\r
+ }\r
+ \r
+ //printf("[%s] NOTICE %s: %s\n", Server->Name, ident, message);\r
+ Message_Append(Server, MSG_TYPE_NOTICE, ident, "", message);\r
+ }\r
+ else if( strcmp(cmd, "PRIVMSG") == 0 )\r
+ {\r
char *dest, *message;\r
- pos ++;\r
dest = GetValue(Line, &pos);\r
- pos ++;\r
+ \r
if( Line[pos] == ':' ) {\r
message = Line + pos + 1;\r
}\r
message = GetValue(Line, &pos);\r
}\r
Cmd_PRIVMSG(Server, dest, ident, message);\r
+ Message_Append(Server, MSG_TYPE_STANDARD, ident, dest, message);\r
+ }\r
+ else if( strcmp(cmd, "JOIN" ) == 0 )\r
+ {\r
+ char *channel;\r
+ channel = GetValue(Line, &pos) + 1;\r
+ Window_Create(Server, channel);\r
+ }\r
+ else\r
+ {\r
+ printf("Unknown message %s (%s)\n", cmd, Line+pos);\r
}\r
}\r
else {\r
+ \r
// Command to client\r
+ printf("Client Command: %s", Line);\r
}\r
}\r
\r
/**\r
* \brief Process incoming lines from the server\r
*/\r
-void ProcessIncoming(tServer *Server)\r
+int ProcessIncoming(tServer *Server)\r
{ \r
char *ptr, *newline;\r
int len;\r
// process it line by line\r
// ioctl#8 on a TCP client gets the number of bytes in the recieve buffer\r
// - Used to avoid blocking\r
- while( ioctl(Server->FD, 8, NULL) )\r
+ #if NON_BLOCK_READ\r
+ while( (len = ioctl(Server->FD, 8, NULL)) > 0 )\r
{\r
+ #endif\r
// Read data\r
- len = read(Server->FD, BUFSIZ - Server->ReadPos, Server->InBuf + Server->ReadPos);\r
+ len = read(Server->FD, BUFSIZ - Server->ReadPos, &Server->InBuf[Server->ReadPos]);\r
+ if( len == -1 ) {\r
+ return -1;\r
+ }\r
Server->InBuf[Server->ReadPos + len] = '\0';\r
\r
// Break into lines\r
while( (newline = strchr(ptr, '\n')) )\r
{\r
*newline = '\0';\r
- printf("%s\n", ptr);\r
+ if( newline[-1] == '\r' ) newline[-1] = '\0';\r
ParseServerLine(Server, ptr);\r
ptr = newline + 1;\r
}\r
\r
// Handle incomplete lines\r
if( ptr - Server->InBuf < len + Server->ReadPos ) {\r
- Server->ReadPos = ptr - Server->InBuf;\r
+ // Update the read position\r
+ // InBuf ReadPos ptr ReadPos+len\r
+ // | old | new used | new unused |\r
+ Server->ReadPos = len + Server->ReadPos - (ptr - Server->InBuf);\r
+ // Copy stuff back (moving "new unused" to the start of the buffer)\r
memcpy(Server->InBuf, ptr, Server->ReadPos);\r
}\r
else {\r
Server->ReadPos = 0;\r
}\r
+ #if NON_BLOCK_READ\r
}\r
+ #endif\r
+ \r
+ return 0;\r
}\r
\r
/**\r
// Return descriptor\r
return fd;\r
}\r
+\r
+/**\r
+ * \brief Read a space-separated value from a string\r
+ */\r
+char *GetValue(char *Src, int *Ofs)\r
+{\r
+ int pos = *Ofs;\r
+ char *ret = Src + pos;\r
+ char *end;\r
+ \r
+ if( !Src ) return NULL;\r
+ \r
+ while( *ret == ' ' ) ret ++;\r
+ \r
+ end = strchr(ret, ' ');\r
+ if( end ) {\r
+ *end = '\0';\r
+ }\r
+ else {\r
+ end = ret + strlen(ret) - 1;\r
+ }\r
+ \r
+ end ++ ;\r
+ while( *ret == ' ' ) end ++;\r
+ *Ofs = end - Src;\r
+ \r
+ return ret;\r
+}\r
+\r
+int SetCursorPos(int Row, int Col)\r
+{\r
+ if( Row == -1 ) {\r
+ Row = Col / giTerminal_Width;\r
+ Col = Col % giTerminal_Width;\r
+ }\r
+ printf("\x1B[%i;%iH", Col, Row);\r
+ return ioctl(1, 9, NULL); // Ugh, constants\r
+}\r
+\r
+static inline int isdigit(int ch)\r
+{\r
+ return '0' <= ch && ch < '9';\r
+}\r