OPENCORES.ORG

From: cvs at opencores.org<cvs@o...>
Date: Thu Apr 26 03:31:22 CEST 2007
Subject: [cvs-checkins] MODIFIED: mlite ...

Date: 00/07/04 26:03:31

Added: mlite/kernel netutil.c Log: FTP server; FTP client; TFTP server; TFTP client; Telnet server; console Revision Changes Path 1.1 mlite/kernel/netutil.c http://www.opencores.org/cvsweb.shtml/mlite/kernel/netutil.c?rev=1.1&content-type=text/x-cvsweb-markup Index: netutil.c =================================================================== /*-------------------------------------------------------------------- * TITLE: Plasma TCP/IP Network Utilities * AUTHOR: Steve Rhoads (rhoadss@y...) * DATE CREATED: 4/20/07 * FILENAME: netutil.c * PROJECT: Plasma CPU core * COPYRIGHT: Software placed into the public domain by the author. * Software 'as is' without warranty. Author liable for nothing. * DESCRIPTION: * Plasma FTP server and FTP client and TFTP server and client * and Telnet server. *--------------------------------------------------------------------*/ #ifdef WIN32 #include <stdio.h> #include <stdlib.h> #include <string.h> #include <ctype.h> #define _LIBC #endif #include "rtos.h" #include "tcpip.h" #ifdef WIN32 #define UartPrintf printf #define OS_MQueueCreate(A,B,C) 0 #define OS_MQueueGet(A,B,C) 0 #define OS_ThreadCreate(A,B,C,D,E) 0 #endif typedef struct { IPSocket *socket; int ip, port, bytes, done, canReceive; FILE *file; } FtpdInfo; static OS_MQueue_t *FtpMQueue; static void FtpdSender(IPSocket *socket) { unsigned char buf[600]; int i, bytes, bytes2; FtpdInfo *info = (FtpdInfo*)socket->userPtr; if(info == NULL || info->done) return; fseek(info->file, info->bytes, 0); for(i = 0; i < 5; ++i) { bytes = fread(buf, 1, 512, info->file); bytes2 = IPWrite(socket, buf, bytes); info->bytes += bytes2; if(bytes != bytes2) return; if(bytes < 512) { fclose(info->file); IPClose(socket); info->done = 1; IPPrintf(info->socket, "226 Done\r\n"); return; } } } static void FtpdReceiver(IPSocket *socket) { unsigned char buf[600]; int bytes; FtpdInfo *info = (FtpdInfo*)socket->userPtr; if(info == NULL || info->done) return; if(socket->state > IP_TCP) { fclose(info->file); IPClose(socket); info->done = 1; IPPrintf(info->socket, "226 Done\r\n"); return; } bytes = IPRead(socket, buf, sizeof(buf)); fwrite(buf, 1, bytes, info->file); } static void FtpdServerAction(IPSocket *socket) { uint8 buf[600]; int bytes; int ip0, ip1, ip2, ip3, port0, port1; IPSocket *socketOut; FtpdInfo *info = (FtpdInfo*)socket->userPtr; if(socket == NULL) return; bytes = IPRead(socket, buf, sizeof(buf)-1); buf[bytes] = 0; //printf("(%s)\n", buf); if(socket->userPtr == NULL) { info = (FtpdInfo*)malloc(sizeof(FtpdInfo)); if(info == NULL) return; memset(info, 0, sizeof(FtpdInfo)); socket->userPtr = info; info->socket = socket; socket->timeout = 0; IPPrintf(socket, "220 Connected to Plasma\r\n"); } else if(strstr((char*)buf, "USER")) { if(strstr((char*)buf, "PlasmaSend")) info->canReceive = 1; socket->timeout = 0; IPPrintf(socket, "331 Password?\r\n"); } else if(strstr((char*)buf, "PASS")) { IPPrintf(socket, "230 Logged in\r\n"); } else if(strstr((char*)buf, "PORT")) { if(info == NULL) return; socket->timeout = 0; sscanf((char*)buf + 5, "%d,%d,%d,%d,%d,%d", &ip0, &ip1, &ip2, &ip3, &port0, &port1); info->ip = (ip0 << 24) | (ip1 << 16) | (ip2 << 8) | ip3; info->port = (port0 << 8) | port1; //printf("ip=0x%x port=%d\n", info->ip, info->port); IPPrintf(socket, "200 OK\r\n"); } else if(strstr((char*)buf, "RETR") || strstr((char*)buf, "STOR")) { char *ptr = strstr((char*)buf, "\r"); if(ptr) *ptr = 0; if(info == NULL) return; info->file = NULL; info->bytes = 0; info->done = 0; if(strstr((char*)buf, "RETR")) info->file = fopen((char*)buf + 5, "rb"); else if(info->canReceive) info->file = fopen((char*)buf + 5, "wb"); if(info->file) { IPPrintf(socket, "150 File ready\r\n"); if(strstr((char*)buf, "RETR")) socketOut = IPOpen(IP_MODE_TCP, info->ip, info->port, FtpdSender); else socketOut = IPOpen(IP_MODE_TCP, info->ip, info->port, FtpdReceiver); if(socketOut) socketOut->userPtr = info; } else { IPPrintf(socket, "500 Error\r\n"); } } else if(strstr((char*)buf, "QUIT")) { if(socket->userPtr) free(socket->userPtr); IPPrintf(socket, "221 Bye\r\n"); IPClose(socket); } else if(bytes) { IPPrintf(socket, "500 Error\r\n"); } } #ifndef WIN32 static void FtpdThread(void *Arg) { IPSocket *socket=NULL; (void)Arg; for(;;) { OS_MQueueGet(HttpMQueue, &socket, OS_WAIT_FOREVER); FtpdServerAction(socket); } } #endif static void FtpdServer(IPSocket *socket) { #ifdef WIN32 FtpdServerAction(socket); #else OS_MQueueSend(FtpMQueue, &socket); #endif } void FtpdInit(int UseFiles) { (void)UseFiles; FtpMQueue = OS_MQueueCreate("ftp", FRAME_COUNT, 4); OS_ThreadCreate("ftp", HttpThread, NULL, 50, 0); IPOpen(IP_MODE_TCP, 0, 21, FtpdServer); } //******************* FTP Client ************************ typedef struct { uint32 ip, port; char user[80], passwd[80], filename[80]; uint8 *buf; int size, bytes, send, state; } FtpInfo; static void FtpCallbackTransfer(IPSocket *socket) { int bytes; FtpInfo *info = (FtpInfo*)socket->userPtr; //printf("FtpCallbackTransfer\n"); if(info == NULL) return; bytes = info->size - info->bytes; if(info->send == 0) bytes = IPRead(socket, info->buf + info->bytes, bytes); else bytes = IPWrite(socket, info->buf + info->bytes, bytes); info->bytes += bytes; if(info->bytes == info->size || socket->state > IP_TCP) { socket->userFunc(info->buf, info->bytes); free(info); socket->userPtr = NULL; IPClose(socket); } } static void FtpCallback(IPSocket *socket) { char buf[600]; FtpInfo *info = (FtpInfo*)socket->userPtr; int bytes, value; bytes = IPRead(socket, (uint8*)buf, sizeof(buf)-1); if(bytes == 0) return; buf[bytes] = 0; sscanf(buf, "%d", &value); if(bytes > 2) buf[bytes-2] = 0; //printf("FtpCallback(%d:%s)\n", socket->userData, buf); if(value / 100 != 2 && value / 100 != 3) return; buf[0] = 0; switch(socket->userData) { case 0: sprintf(buf, "USER %s\r\n", info->user); socket->userData = 1; break; case 1: sprintf(buf, "PASS %s\r\n", info->passwd); socket->userData = 2; if(value == 331) break; //possible fall-through case 2: sprintf(buf, "PORT %d,%d,%d,%d,%d,%d\r\n", info->ip >> 24, (uint8)(info->ip >> 16), (uint8)(info->ip >> 8), (uint8)info->ip, (uint8)(info->port >> 8), (uint8)info->port); socket->userData = 3; break; case 3: if(info->send == 0) sprintf(buf, "RETR %s\r\n", info->filename); else sprintf(buf, "STOR %s\r\n", info->filename); socket->userData = 4; break; case 4: sprintf(buf, "QUIT\r\n"); socket->userData = 9; break; } IPWrite(socket, (uint8*)buf, strlen(buf)); IPWriteFlush(socket); if(socket->userData == 9) IPClose(socket); } IPSocket *FtpTransfer(uint32 ip, char *user, char *passwd, char *filename, uint8 *buf, int size, int send, void (*callback)(uint8 *data, int size)) { IPSocket *socket, *socketTransfer; FtpInfo *info; uint8 *ptr; info = (FtpInfo*)malloc(sizeof(FtpInfo)); if(info == NULL) return NULL; strncpy(info->user, user, 80); strncpy(info->passwd, passwd, 80); strncpy(info->filename, filename, 80); info->buf = buf; info->size = size; info->send = send; info->bytes = 0; info->state = 0; info->port = 2000; socketTransfer = IPOpen(IP_MODE_TCP, 0, info->port, FtpCallbackTransfer); socketTransfer->userPtr = info; socketTransfer->userFunc = callback; socket = IPOpen(IP_MODE_TCP, ip, 21, FtpCallback); socket->userPtr = info; socket->userFunc = callback; ptr = socket->headerSend; info->ip = IPAddressSelf(); return socket; } //******************* TFTP Server ************************ static void TftpdCallback(IPSocket *socket) { unsigned char buf[512+4]; int bytes, blockNum; FILE *file = (FILE*)socket->userPtr; bytes = IPRead(socket, buf, sizeof(buf)); //printf("TfptdCallback bytes=%d\n", bytes); if(bytes < 4 || buf[0]) return; if(buf[1] == 1) //RRQ = Read Request { if(file) fclose(file); file = fopen((char*)buf+2, "rb"); socket->userPtr = file; if(file == NULL) { buf[0] = 0; buf[1] = 5; //ERROR buf[2] = 0; buf[3] = 0; buf[4] = 'X'; //Error string buf[5] = 0; IPWrite(socket, buf, 6); return; } } if(buf[1] == 1 || buf[1] == 4) //ACK { if(file == NULL) return; if(buf[1] == 1) blockNum = 0; else blockNum = (buf[2] << 8) | buf[3]; ++blockNum; buf[0] = 0; buf[1] = 3; //DATA buf[2] = (uint8)(blockNum >> 8); buf[3] = (uint8)blockNum; fseek(file, (blockNum-1)*512, 0); bytes = fread(buf+4, 1, 512, file); IPWrite(socket, buf, bytes+4); } } void TftpdInit(void) { IPSocket *socket; socket = IPOpen(IP_MODE_UDP, 0, 69, TftpdCallback); } //******************* TFTP Client ************************ static void TftpCallback(IPSocket *socket) { unsigned char buf[512+4]; int bytes, blockNum, length; bytes = IPRead(socket, buf, sizeof(buf)); if(bytes < 4 || buf[0]) return; blockNum = (buf[2] << 8) | buf[3]; length = blockNum * 512 - 512 + bytes - 4; //printf("TftpCallback(%d,%d)\n", buf[1], blockNum); if(length > (int)socket->userData) { bytes -= length - (int)socket->userData; length = (int)socket->userData; } if(buf[1] == 3) //DATA { memcpy((uint8*)socket->userPtr + blockNum * 512 - 512, buf+4, bytes-4); buf[1] = 4; //ACK IPWrite(socket, buf, 4); if(bytes-4 < 512) { printf("Done %d bytes\n", length); socket->userFunc(socket->userPtr, length); IPClose(socket); } } } IPSocket *TftpTransfer(uint32 ip, char *filename, uint8 *buffer, int size, void (*callback)(uint8 *data, int bytes)) { IPSocket *socket; uint8 buf[512+4]; int bytes; socket = IPOpen(IP_MODE_UDP, ip, 69, TftpCallback); socket->userPtr = buffer; socket->userData = size; socket->userFunc = callback; buf[0] = 0; buf[1] = 1; //read strcpy((char*)buf+2, filename); bytes = strlen(filename); strcpy((char*)buf+bytes+3, "octet"); IPWrite(socket, buf, bytes+9); return socket; } //******************* Telnet Server ************************ static TelnetFunc_t *TelnetFuncList; static void TelnetServer(IPSocket *socket) { uint8 buf[512+4]; int bytes, i; char *ptr; static char command[80]; char command2[80]; bytes = IPRead(socket, buf, sizeof(buf)-1); if(bytes == 0) { if(socket->userData) return; socket->userData = 1; socket->timeout = 0; socket->headerSend[47] |= 0x08; //PSH buf[0] = 255; //IAC buf[1] = 251; //WILL buf[2] = 3; //suppress go ahead buf[3] = 255; //IAC buf[4] = 251; //WILL buf[5] = 1; //echo strcpy((char*)buf+6, "Welcome to Plasma.\r\n-> "); IPWrite(socket, buf, 6+23); IPWriteFlush(socket); command[0] = 0; return; } buf[bytes] = 0; if(buf[0] == 255 || buf[0] == 27) return; if(buf[0] != 8) { IPWrite(socket, buf, bytes); IPWriteFlush(socket); strncat(command, (char*)buf, sizeof(command)); } else { //backspace bytes = strlen(command); if(bytes) { command[bytes-1] = 0; buf[0] = 8; buf[1] = ' '; buf[2] = 8; IPWrite(socket, buf, 3); IPWriteFlush(socket); } } ptr = strstr(command, "\r"); if(ptr) { ptr[0] = 0; buf[0] = 0; if(strncmp(command, "help", 4) == 0) { sprintf((char*)buf, "Commands: help, exit"); for(i = 0; TelnetFuncList[i].name; ++i) { strcat((char*)buf, ", "); strcat((char*)buf, TelnetFuncList[i].name); } strcat((char*)buf, "\r\n"); } else if(strncmp(command, "exit", 4) == 0) IPClose(socket); else if(command[0]) { strcpy(command2, command); ptr = strstr(command2, " "); if(ptr) ptr[0] = 0; for(i = 0; TelnetFuncList[i].name; ++i) { if(strcmp(command2, TelnetFuncList[i].name) == 0) { TelnetFuncList[i].func(socket, command); break; } } if(TelnetFuncList[i].name == NULL) sprintf((char*)buf, "Unknown command (%s)\r\n", command); } strcat((char*)buf, "-> "); IPPrintf(socket, (char*)buf); command[0] = 0; } } void TelnetInit(TelnetFunc_t *funcList) { IPSocket *socket; TelnetFuncList = funcList; socket = IPOpen(IP_MODE_TCP, 0, 23, TelnetServer); } //******************* Console ************************ static uint8 myBuffer[1024*3]; IPSocket *socketTelnet; void TransferDone(uint8 *data, int bytes) { printf("TransferDone(0x%x, %d)\n", data, bytes); data[bytes] = 0; if(bytes > 500) data[500] = 0; printf("%s\n", data); } static void TelnetInfo(IPSocket *socket, char *command) { (void)command; IPPrintf(socket, "Steve was here!\r\n"); } static void TelnetMath(IPSocket *socket, char *command) { int v1, v2; char buf[20], buf2[20]; (void)socket; if(strlen(command) < 6) { IPPrintf(socket, "Usage: math <number> <operator> <value>\r\n"); return; } sscanf(command, "%s%d%s%d", buf, &v1, buf2, &v2); if(buf2[0] == '+') v1 += v2; else if(buf2[0] == '-') v1 -= v2; else if(buf2[0] == '*') v1 *= v2; else if(buf2[0] == '/') { if(v2 != 0) v1 /= v2; } sprintf((char*)myBuffer, "%d\r\n", v1); IPPrintf(socket, myBuffer); } void PingCallback(IPSocket *socket) { IPSocket *socket2 = socket->userPtr; //printf("Ping Reply\n"); IPClose(socket); if(socket2) IPPrintf(socket2, "Ping Reply\r\n"); } static void DnsResultCallback(IPSocket *socket, uint32 ip, void *arg) { char buf[80]; IPSocket *socketTelnet = arg; IPSocket *socketPing; sprintf(buf, "ip=%d.%d.%d.%d\r\n", (uint8)(ip >> 24), (uint8)(ip >> 16), (uint8)(ip >> 8), (uint8)ip); IPPrintf(socketTelnet, buf); socketPing = IPOpen(IP_MODE_PING, ip, 0, PingCallback); socketPing->userPtr = socketTelnet; myBuffer[0] = 'A'; IPWrite(socketPing, myBuffer, 1); } static void TelnetPing(IPSocket *socket, char *command) { int ip0, ip1, ip2, ip3; char buf[20]; IPSocket *socketPing; if('0' <= command[5] && command[5] <= '9') { sscanf(command, "%s %d.%d.%d.%d", buf, &ip0, &ip1, &ip2, &ip3); ip0 = (ip0 << 24) | (ip1 << 16) | (ip2 << 8) | ip3; socketPing = IPOpen(IP_MODE_PING, ip0, 0, PingCallback); socketPing->userPtr = socket; myBuffer[0] = 'A'; IPWrite(socketPing, myBuffer, 1); IPPrintf(socket, "Sent ping\r\n"); } else { IPResolve(command+5, DnsResultCallback, socket); IPPrintf(socket, "Sent DNS request\r\n"); } } void TelnetTransferDone(uint8 *data, int length) { data[length] = 0; IPPrintf(socketTelnet, "Transfer Done\r\n"); } static void TelnetFtp(IPSocket *socket, char *command) { char buf[40], user[40], passwd[40], name[80]; int ip0, ip1, ip2, ip3; if(strlen(command) < 10) { IPPrintf(socket, "ftp #.#.#.# User Password File\r\n"); return; } sscanf(command, "%s %d.%d.%d.%d %s %s %s", buf, &ip0, &ip1, &ip2, &ip3, user, passwd, name); ip0 = (ip0 << 24) | (ip1 << 16) | (ip2 << 8) | ip3; socketTelnet = socket; FtpTransfer(ip0, user, passwd, name, myBuffer, sizeof(myBuffer)-1, 0, TelnetTransferDone); } static void TelnetTftp(IPSocket *socket, char *command) { char buf[80], name[80]; int ip0, ip1, ip2, ip3; if(strlen(command) < 10) { IPPrintf(socket, "tftp #.#.#.# File\r\n"); return; } sscanf(command, "%s %d.%d.%d.%d %s %s %s", buf, &ip0, &ip1, &ip2, &ip3, name); ip0 = (ip0 << 24) | (ip1 << 16) | (ip2 << 8) | ip3; socketTelnet = socket; TftpTransfer(ip0, name, myBuffer, sizeof(myBuffer)-1, TelnetTransferDone); } static void TelnetShow(IPSocket *socket, char *command) { (char)command; IPPrintf(socket, myBuffer); IPPrintf(socket, "\r\n"); } static void TelnetClear(IPSocket *socket, char *command) { memset(myBuffer, 0, sizeof(myBuffer)); } static TelnetFunc_t MyFuncs[] = { "info", 0, TelnetInfo, "math", 0, TelnetMath, "ping", 0, TelnetPing, "ftp", 0, TelnetFtp, "tftp", 0, TelnetTftp, "show", 0, TelnetShow, "clear", 0, TelnetClear, NULL, 0, NULL }; void ConsoleInit(void) { FtpdInit(1); TftpdInit(); TelnetInit(MyFuncs); }