Add AVR32 port and demo files.

Demo/lwIP_AVR32_UC3/NETWORK/BasicTFTP/BasicTFTP.c

@@ -0,0 +1,473 @@
+/* This source file is part of the ATMEL FREERTOS-0.9.0 Release */
+
+/*This file has been prepared for Doxygen automatic documentation generation.*/
+/*! \file *********************************************************************
+ *
+ * \brief Basic TFTP Server for AVR32 UC3.
+ *
+ * - Compiler:           GNU GCC for AVR32
+ * - Supported devices:  All AVR32 devices can be used.
+ * - AppNote:
+ *
+ * \author               Atmel Corporation: http://www.atmel.com \n
+ *                       Support email: avr32@atmel.com
+ *
+ *****************************************************************************/
+
+/* Copyright (c) 2007, Atmel Corporation All rights reserved.
+ *
+ * Redistribution and use in source and binary forms, with or without
+ * modification, are permitted provided that the following conditions are met:
+ *
+ * 1. Redistributions of source code must retain the above copyright notice,
+ * this list of conditions and the following disclaimer.
+ *
+ * 2. Redistributions in binary form must reproduce the above copyright notice,
+ * this list of conditions and the following disclaimer in the documentation
+ * and/or other materials provided with the distribution.
+ *
+ * 3. The name of ATMEL may not be used to endorse or promote products derived
+ * from this software without specific prior written permission.
+ *
+ * THIS SOFTWARE IS PROVIDED BY ATMEL ``AS IS'' AND ANY EXPRESS OR IMPLIED
+ * WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF
+ * MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE EXPRESSLY AND
+ * SPECIFICALLY DISCLAIMED. IN NO EVENT SHALL ATMEL BE LIABLE FOR ANY DIRECT,
+ * INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
+ * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
+ * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
+ * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
+ * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
+ * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
+ */
+
+
+/*
+  Implements a simplistic TFTP server.
+  
+  In order to put data on the TFTP server (not over 2048 bytes)
+  tftp 192.168.0.2 PUT <src_filename>
+  this will copy file from your hard drive to the RAM buffer of the application
+
+  tftp 192.168.0.2 GET <dst_filename>
+  this will copy file from the RAM buffer of the application to your hard drive
+  You can then check that src_filename and dst_filename are identical    
+*/
+
+#if (TFTP_USED == 1)
+
+#include <string.h>
+
+
+/* Scheduler includes. */
+#include "FreeRTOS.h"
+#include "task.h"
+#include "partest.h"
+#include "BasicTFTP.h"
+
+
+/* Demo includes. */
+#include "AVR32_EMAC.h"
+#include "portmacro.h"
+
+/* lwIP includes. */
+#include "lwip/api.h"
+#include "lwip/tcpip.h"
+#include "lwip/memp.h"
+#include "lwip/stats.h"
+#include "lwip/opt.h"
+#include "lwip/api.h"
+#include "lwip/arch.h"
+#include "lwip/sys.h"
+#include "netif/loopif.h"
+#include "lwip/sockets.h"
+
+#define O_WRONLY 1
+#define O_RDONLY 2
+
+
+/* The port on which we listen. */
+#define TFTP_PORT    ( 69 )
+
+/* Delay on close error. */
+#define TFTP_DELAY         ( 10 )
+
+/* Delay on bind error. */
+#define TFTP_ERROR_DELAY    ( 40 )
+
+#define TFTP_LED     ( 4 )
+
+char data_out[SEGSIZE+sizeof(struct tftphdr)];
+char data_in[SEGSIZE+sizeof(struct tftphdr)];
+
+//struct tftp_server *server;
+
+/*------------------------------------------------------------*/
+static char * errmsg[] = {
+  "Undefined error code",               // 0 nothing defined
+  "File not found",                     // 1 TFTP_ENOTFOUND 
+  "Access violation",                   // 2 TFTP_EACCESS   
+  "Disk full or allocation exceeded",   // 3 TFTP_ENOSPACE  
+  "Illegal TFTP operation",             // 4 TFTP_EBADOP    
+  "Unknown transfer ID",                // 5 TFTP_EBADID    
+  "File already exists",                // 6 TFTP_EEXISTS   
+  "No such user",                       // 7 TFTP_ENOUSER   
+};
+
+
+/* Send an error packet to the client */
+static void 
+tftpd_send_error(int s, struct tftphdr * reply, int err,
+     struct sockaddr_in *from_addr, int from_len)
+{
+    if ( ( 0 <= err ) && ( sizeof(errmsg)/sizeof(errmsg[0]) > err) ) {
+    reply->th_opcode = htons(ERROR);
+    reply->th_code = htons(err);
+    if ( (0 > err) || (sizeof(errmsg)/sizeof(errmsg[0]) <= err) )
+        err = 0; // Do not copy a random string from hyperspace
+    strcpy(reply->th_msg, errmsg[err]);
+    sendto(s, reply, 4+strlen(reply->th_msg)+1, 0, 
+     (struct sockaddr *)from_addr, from_len);
+    }
+}
+
+portCHAR cRamBuffer[2048];
+int lCurrentBlock = 0;
+int lTotalLength = 0;
+
+
+int tftpd_close_data_file(int fd)
+{
+  lCurrentBlock = 0;
+  return (5);
+}
+
+int tftpd_open_data_file(int fd, int mode)
+{
+  lCurrentBlock = 0; 
+  return (5);
+}
+
+int tftpd_read_data_file(int fd, portCHAR * buffer, int length)
+{
+int lReturnValue;
+
+  if ((lTotalLength -= length) >= 0) {
+    lReturnValue = length;
+  }
+  else
+  {
+    lReturnValue = lTotalLength + length;
+    lTotalLength = 0;
+  }
+  memcpy(buffer, &cRamBuffer[lCurrentBlock * SEGSIZE], lReturnValue);
+  lCurrentBlock++;
+  return (lReturnValue);
+}
+
+//
+// callback to store data to the RAM buffer
+//
+int tftpd_write_data_file(int fd, portCHAR * buffer, int length)
+{
+  lTotalLength += length;
+  memcpy(&cRamBuffer[lCurrentBlock * SEGSIZE], buffer, length);
+  lCurrentBlock++;
+  return (length);
+}
+
+//
+// Receive a file from the client
+//
+static void
+tftpd_write_file(struct tftphdr *hdr,
+                 struct sockaddr_in *from_addr, int from_len)
+{
+
+    struct tftphdr *reply = (struct tftphdr *)data_out;
+    struct tftphdr *response = (struct tftphdr *)data_in;
+    int fd, len, ok, tries, closed, data_len, s;
+    unsigned short block;
+    struct timeval timeout;
+    fd_set fds;
+    int total_timeouts = 0;
+    struct sockaddr_in client_addr, local_addr;
+    int client_len;
+
+
+    s = socket(AF_INET, SOCK_DGRAM, 0);
+    if (s < 0) {
+        return;
+    }
+
+    memset((char *)&local_addr, 0, sizeof(local_addr));
+    local_addr.sin_family = AF_INET;
+    local_addr.sin_len = sizeof(local_addr);
+    local_addr.sin_addr.s_addr = htonl(INADDR_ANY);
+    local_addr.sin_port = htons(INADDR_ANY);
+
+    if (bind(s, (struct sockaddr *)&local_addr, sizeof(local_addr)) < 0) {
+        // Problem setting up my end
+        close(s);
+        return;
+    }
+
+    if ((fd = tftpd_open_data_file((int)hdr->th_stuff, O_WRONLY)) < 0) {
+        tftpd_send_error(s,reply,TFTP_ENOTFOUND,from_addr, from_len);
+        close(s);
+        return;
+    }
+
+    ok = pdTRUE;
+    closed = pdFALSE;
+    block = 0;
+    while (ok) {
+        // Send ACK telling client he can send data
+        reply->th_opcode = htons(ACK);
+        reply->th_block = htons(block++); // postincrement
+        for (tries = 0;  tries < TFTP_RETRIES_MAX;  tries++) {
+            sendto(s, reply, 4, 0, (struct sockaddr *)from_addr, from_len);
+        repeat_select:
+            timeout.tv_sec = TFTP_TIMEOUT_PERIOD;
+            timeout.tv_usec = 0;
+            FD_ZERO(&fds);
+            FD_SET(s, &fds);
+           vParTestToggleLED( TFTP_LED );
+           if (lwip_select(s+1, &fds, 0, 0, &timeout) <= 0) {
+                if (++total_timeouts > TFTP_TIMEOUT_MAX) {
+                    tftpd_send_error(s,reply,TFTP_EBADOP,from_addr, from_len);
+                    ok = pdFALSE;
+                    break;
+                }
+                continue; // retry the send, using up one retry.
+            }
+            vParTestToggleLED( TFTP_LED );
+            // Some data has arrived
+            data_len = sizeof(data_in);
+            client_len = sizeof(client_addr);
+            if ((data_len = recvfrom(s, data_in, data_len, 0, 
+                      (struct sockaddr *)&client_addr, &client_len)) < 0) {
+                // What happened?  No data here!
+                continue; // retry the send, using up one retry.
+            }
+            if (ntohs(response->th_opcode) == DATA &&
+                ntohs(response->th_block) < block) {
+                // Then it is repeat DATA with an old block; listen again,
+                // but do not repeat sending the current ack, and do not
+                // use up a retry count.  (we do re-send the ack if
+                // subsequently we time out)
+                goto repeat_select;
+            }
+            if (ntohs(response->th_opcode) == DATA &&
+                ntohs(response->th_block) == block) {
+                // Good data - write to file
+                len = tftpd_write_data_file(fd, response->th_data, data_len-4);
+                if (len < (data_len-4)) {
+                    // File is "full"
+                    tftpd_send_error(s,reply,TFTP_ENOSPACE,
+                                     from_addr, from_len);     
+                    ok = pdFALSE;  // Give up
+                    break; // out of the retries loop
+                }
+                if (data_len < (SEGSIZE+4)) {
+                    // End of file
+                    closed = pdTRUE;
+                    ok = pdFALSE;
+                    vParTestSetLED( 0 , 0 );
+
+                    if (tftpd_close_data_file(fd) == -1) {
+                        tftpd_send_error(s,reply,TFTP_EACCESS,
+                                         from_addr, from_len);
+
+                       break;  // out of the retries loop
+                    }
+                    // Exception to the loop structure: we must ACK the last
+                    // packet, the one that implied EOF:
+                    reply->th_opcode = htons(ACK);
+                    reply->th_block = htons(block++); // postincrement
+                    sendto(s, reply, 4, 0, (struct sockaddr *)from_addr, from_len);
+                    break; // out of the retries loop
+                }
+                // Happy!  Break out of the retries loop.
+                break;
+            }
+        } // End of the retries loop.
+        if (TFTP_RETRIES_MAX <= tries) {
+            tftpd_send_error(s,reply,TFTP_EBADOP,from_addr, from_len);
+            ok = pdFALSE;
+        }
+    }
+    close(s);
+    if (!closed) {
+      tftpd_close_data_file(fd);
+    }
+}
+
+
+//
+// Send a file to the client
+//
+static void
+tftpd_read_file(struct tftphdr *hdr,
+                struct sockaddr_in *from_addr, int from_len)
+{
+    struct tftphdr *reply = (struct tftphdr *)data_out;
+    struct tftphdr *response = (struct tftphdr *)data_in;
+    int fd, len, tries, ok, data_len, s;
+    unsigned short block;
+    struct timeval timeout;
+    fd_set fds;
+    int total_timeouts = 0;
+    struct sockaddr_in client_addr, local_addr;
+    int client_len;
+
+    s = socket(AF_INET, SOCK_DGRAM, 0);
+    if (s < 0) {
+        return;
+    }
+    memset((char *)&local_addr, 0, sizeof(local_addr));
+    local_addr.sin_family = AF_INET;
+    local_addr.sin_len = sizeof(local_addr);
+    local_addr.sin_addr.s_addr = htonl(INADDR_ANY);
+    local_addr.sin_port = htons(INADDR_ANY);
+    if (bind(s, (struct sockaddr *)&local_addr, sizeof(local_addr)) < 0) {
+        // Problem setting up my end
+        close(s);
+        return;
+    }
+    if ((fd = tftpd_open_data_file((int)hdr->th_stuff, O_RDONLY)) < 0) {
+        tftpd_send_error(s,reply,TFTP_ENOTFOUND,from_addr, from_len);
+        close(s);
+        return;
+    }
+    block = 0;
+    ok = pdTRUE;
+    while (ok) {
+        // Read next chunk of file
+        len = tftpd_read_data_file(fd, reply->th_data, SEGSIZE);
+        reply->th_block = htons(++block); // preincrement
+        reply->th_opcode = htons(DATA);
+        for (tries = 0;  tries < TFTP_RETRIES_MAX;  tries++) {
+            if (sendto(s, reply, 4+len, 0,
+                       (struct sockaddr *)from_addr, from_len) < 0) {
+                // Something went wrong with the network!
+                ok = pdFALSE;
+                break;
+            }
+        repeat_select:
+            timeout.tv_sec = TFTP_TIMEOUT_PERIOD;
+            timeout.tv_usec = 0;
+            FD_ZERO(&fds);
+            FD_SET(s, &fds);
+            vParTestToggleLED( TFTP_LED );
+            if (select(s+1, &fds, 0, 0, &timeout) <= 0) {
+                if (++total_timeouts > TFTP_TIMEOUT_MAX) {
+                    tftpd_send_error(s,reply,TFTP_EBADOP,from_addr, from_len);
+                    ok = pdFALSE;
+                    break;
+                }
+                continue; // retry the send, using up one retry.
+            }
+            vParTestToggleLED( TFTP_LED );
+            data_len = sizeof(data_in);
+            client_len = sizeof(client_addr);
+            if ((data_len = recvfrom(s, data_in, data_len, 0, 
+                                     (struct sockaddr *)&client_addr,
+                                     &client_len)) < 0) {
+                // What happened?  Maybe someone lied to us...
+                continue; // retry the send, using up one retry.
+            }
+            if ((ntohs(response->th_opcode) == ACK) &&
+                (ntohs(response->th_block) < block)) {
+                // Then it is a repeat ACK for an old block; listen again,
+                // but do not repeat sending the current block, and do not
+                // use up a retry count.  (we do re-send the data if
+                // subsequently we time out)
+                goto repeat_select;
+            }
+            if ((ntohs(response->th_opcode) == ACK) &&
+                (ntohs(response->th_block) == block)) {
+                // Happy!  Break out of the retries loop.
+                break;
+            }
+        } // End of the retries loop.
+        if (TFTP_RETRIES_MAX <= tries) {
+            tftpd_send_error(s,reply,TFTP_EBADOP,from_addr, from_len);
+            ok = pdFALSE;
+        }
+        if (len < SEGSIZE) {
+            break; // That's end of file then.
+        }
+    }
+    close(s);
+    tftpd_close_data_file(fd);
+}
+
+
+
+portTASK_FUNCTION( vBasicTFTPServer, pvParameters )
+{
+    int lSocket;
+    int lDataLen, lRecvLen, lFromLen;
+    struct sockaddr_in sLocalAddr, sFromAddr;
+    portCHAR cData[SEGSIZE+sizeof(struct tftphdr)];
+    struct tftphdr *sHdr = (struct tftphdr *)cData;
+
+    // Set up port
+    // Network order in info; host order in server:
+
+    for (;;) {
+        // Create socket
+        lSocket = socket(AF_INET, SOCK_DGRAM, 0);
+        if (lSocket < 0) {
+            return;
+        }
+        memset((char *)&sLocalAddr, 0, sizeof(sLocalAddr));
+        sLocalAddr.sin_family = AF_INET;
+        sLocalAddr.sin_len = sizeof(sLocalAddr);
+        sLocalAddr.sin_addr.s_addr = htonl(INADDR_ANY);
+        sLocalAddr.sin_port = TFTP_PORT;
+
+        if (bind(lSocket, (struct sockaddr *)&sLocalAddr, sizeof(sLocalAddr)) < 0) {
+            // Problem setting up my end
+            close(lSocket);
+            return;
+        }
+
+
+        lRecvLen = sizeof(cData);
+        lFromLen = sizeof(sFromAddr);
+        lDataLen = recvfrom(lSocket, sHdr, lRecvLen, 0,
+                            (struct sockaddr *)&sFromAddr, &lFromLen);
+        vParTestSetLED( TFTP_LED , pdTRUE );
+        close(lSocket); // so that other servers can bind to the TFTP socket
+
+        if ( lDataLen < 0) {
+
+        } else {
+            switch (ntohs(sHdr->th_opcode)) {
+            case WRQ:
+                tftpd_write_file(sHdr, &sFromAddr, lFromLen);
+                vParTestSetLED( TFTP_LED , pdFALSE );
+                break;
+            case RRQ:
+                tftpd_read_file(sHdr, &sFromAddr, lFromLen);
+                vParTestSetLED( TFTP_LED , pdFALSE );
+                break;
+            case ACK:
+            case DATA:
+            case ERROR:
+                vParTestSetLED( TFTP_LED , pdFALSE );
+                // Ignore
+                break;
+            default:
+                for(;;)
+                {
+                  vParTestToggleLED( TFTP_LED );
+                  vTaskDelay(200);                    
+                }
+             }
+        }
+    }
+}
+#endif