X-Git-Url: https://git.ucc.asn.au/?a=blobdiff_plain;f=Usermode%2FApplications%2Ftelnetd_src%2Fmain.c;h=c7f1978c6c9f67af547abc862664d39567273731;hb=f829eb1731bb6e0b9f29d285603731cb36edaeb3;hp=a62d0dd3958b246b9a514fa24a4257664314403d;hpb=507f19941f45bca7c1de783a394f56b15f10be4d;p=tpg%2Facess2.git diff --git a/Usermode/Applications/telnetd_src/main.c b/Usermode/Applications/telnetd_src/main.c index a62d0dd3..c7f1978c 100644 --- a/Usermode/Applications/telnetd_src/main.c +++ b/Usermode/Applications/telnetd_src/main.c @@ -1,16 +1,35 @@ /* + * Acess2 Telnet Server (TCP server test case) + * - By John Hodge (thePowersGang) + * + * main.c + * - All */ #include #include -#include #include +#include +#include +#include +#include +#include // === TYPES === +enum eTelnetMode +{ + MODE_DATA, + MODE_IAC, + MODE_WILL, + MODE_WONT, + MODE_DO, + MODE_DONT +}; + typedef struct sClient { + enum eTelnetMode Mode; int Socket; - int stdout; - int stdin; + int pty; } tClient; // === PROTOTYPES === @@ -41,7 +60,7 @@ int main(int argc, char *argv[]) int addrtype = Net_ParseAddress("10.0.2.10", data); int port = 23; giServerFD = Net_OpenSocket(addrtype, data, "tcps"); - ioctl(giServerFD, 4, &port); // Set port and start listening + _SysIOCtl(giServerFD, 4, &port); // Set port and start listening } // Event loop @@ -50,17 +69,17 @@ int main(int argc, char *argv[]) return 0; } +static void FD_SET_MAX(fd_set *set, int fd, int *maxfd) +{ + FD_SET(fd, set); + if(*maxfd < fd) *maxfd = fd; +} + void EventLoop(void) { fd_set fds; int maxfd; - void FD_SET_MAX(fd_set *set, int fd, int *maxfd) - { - FD_SET(fd, set); - if(*maxfd < fd) *maxfd = fd; - } - for( ;; ) { FD_ZERO(&fds); @@ -70,14 +89,12 @@ void EventLoop(void) for( int i = 0; i < giConfig_MaxClients; i ++ ) { if( gaClients[i].Socket == 0 ) continue ; - _SysDebug("Socket = %i, stdout = %i", - gaClients[i].Socket, gaClients[i].stdout); FD_SET_MAX(&fds, gaClients[i].Socket, &maxfd); - FD_SET_MAX(&fds, gaClients[i].stdout, &maxfd); + FD_SET_MAX(&fds, gaClients[i].pty, &maxfd); } // Select! - select( maxfd+1, &fds, NULL, NULL, NULL ); + _SysSelect( maxfd+1, &fds, NULL, NULL, NULL, 0 ); // Check events if( FD_ISSET(giServerFD, &fds) ) @@ -91,7 +108,7 @@ void EventLoop(void) // Handle client data HandleServerBoundData(&gaClients[i]); } - if( FD_ISSET(gaClients[i].stdout, &fds) ) + if( FD_ISSET(gaClients[i].pty, &fds) ) { // Handle output from terminal HandleClientBoundData(&gaClients[i]); @@ -102,13 +119,13 @@ void EventLoop(void) void Server_NewClient(int FD) { - tClient *clt; + tClient *clt = NULL; // TODO: Is this done in the IPStack? if( giNumClients == giConfig_MaxClients ) { // Open, reject - close( _SysOpenChild(FD, "", O_RDWR) ); + _SysClose( _SysOpenChild(FD, "", OPENFLAG_READ) ); return ; } @@ -120,30 +137,127 @@ void Server_NewClient(int FD) break; } } + assert(clt); // Accept the connection - clt->Socket = _SysOpenChild(FD, "", O_RDWR); + clt->Socket = _SysOpenChild(FD, "", OPENFLAG_READ|OPENFLAG_WRITE); giNumClients ++; - // Create stdin/stdout - clt->stdin = open("/Devices/fifo/anon", O_RDWR); - clt->stdout = open("/Devices/fifo/anon", O_RDWR); + // Create PTY + // TODO: Use PTYs + clt->pty = _SysOpen("/Devices/pts/ptmx", OPENFLAG_READ|OPENFLAG_WRITE); + if( clt->pty < 0 ) { + perror("Unable to create/open PTY"); + _SysDebug("Unable to create/open PTY: %s", strerror(errno)); + _SysClose(clt->Socket); + clt->Socket = 0; + return ; + } + // - Initialise + { + _SysIOCtl(clt->pty, PTY_IOCTL_SETID, "telnetd0"); + struct ptymode mode = {.InputMode = 0, .OutputMode=0}; + struct ptydims dims = {.W = 80, .H = 25}; + _SysIOCtl(clt->pty, PTY_IOCTL_SETMODE, &mode); + _SysIOCtl(clt->pty, PTY_IOCTL_SETDIMS, &dims); + } // TODO: Arguments and envp { - int fds[3] = {clt->stdin, clt->stdout, clt->stdout}; - const char *argv[] = {NULL}; + int clientfd = _SysOpen("/Devices/pts/telnetd0", OPENFLAG_READ|OPENFLAG_WRITE); + if(clientfd < 0) { + perror("Unable to open login PTY"); + _SysClose(clt->Socket); + _SysClose(clt->pty); + clt->Socket = 0; + return ; + } + int fds[3] = {clientfd, clientfd, clientfd}; + const char *argv[] = {"login", NULL}; _SysSpawn("/Acess/SBin/login", argv, argv, 3, fds, NULL); } } void HandleServerBoundData(tClient *Client) { - char buf[BUFSIZ]; - int len; + uint8_t buf[BUFSIZ]; + size_t len; - len = read(Client->Socket, buf, BUFSIZ); - if( len <= 0 ) return ; - write(Client->stdin, buf, len); + len = _SysRead(Client->Socket, buf, BUFSIZ); + if( len == 0 ) return ; + if( len == -1 ) { + return ; + } + // handle options + int last_flush = 0; + for( int i = 0; i < len; i ++ ) + { + switch(Client->Mode) + { + case MODE_IAC: + Client->Mode = MODE_DATA; + switch(buf[i]) + { + case 240: // End of subnegotiation parameters + _SysDebug("End Subnegotiation"); + break; + case 241: // Nop + break; + case 242: // SYNCH + case 243: // NVT Break + case 244: // Function 'IP' (Ctrl-C) + + break; + case 245: // Function 'AO' + case 246: // Function 'AYT' + case 247: // Function 'EC' + case 248: // Function 'EL' + case 249: // GA Signal + break; + case 250: // Subnegotation + _SysDebug("Subnegotiation"); + break; + case 251: // WILL + Client->Mode = MODE_WILL; + break; + case 252: // WONT + Client->Mode = MODE_WONT; + break; + case 253: // DO + Client->Mode = MODE_DO; + break; + case 254: // DONT + Client->Mode = MODE_DONT; + break; + case 255: // Literal 255 + Client->Mode = MODE_DATA; + i --; // hacky! + break; + } + break; + case MODE_WILL: + _SysDebug("Option %i WILL", buf[i]); + Client->Mode = MODE_DATA; + break; + case MODE_WONT: + _SysDebug("Option %i WONT", buf[i]); + Client->Mode = MODE_DATA; + break; + case MODE_DO: + _SysDebug("Option %i DO", buf[i]); + Client->Mode = MODE_DATA; + break; + case MODE_DONT: + _SysDebug("Option %i DONT", buf[i]); + Client->Mode = MODE_DATA; + break; + case MODE_DATA: + if( buf[i] == 255 ) + Client->Mode = MODE_IAC; + else + _SysWrite(Client->pty, buf+i, 1); + break; + } + } } void HandleClientBoundData(tClient *Client) @@ -151,8 +265,8 @@ void HandleClientBoundData(tClient *Client) char buf[BUFSIZ]; int len; - len = read(Client->stdout, buf, BUFSIZ); + len = _SysRead(Client->pty, buf, BUFSIZ); if( len <= 0 ) return ; - write(Client->Socket, buf, len); + _SysWrite(Client->Socket, buf, len); }