Subversion Repositories amber

/*----------------------------------------------------------------
//                                                              //
//  boot-loader-ethmac.c                                        //
//                                                              //
//  This file is part of the Amber project                      //
//  http://www.opencores.org/project,amber                      //
//                                                              //
//  Description                                                 //
//  The main functions for the boot loader application. This    //
//  application is embedded in the FPGA's SRAM and is used      //
//  to load larger applications into the DDR3 memory on         //
//  the development board.                                      //
//                                                              //
//  Author(s):                                                  //
//      - Conor Santifort, csantifort.amber@gmail.com           //
//                                                              //
//////////////////////////////////////////////////////////////////
//                                                              //
// Copyright (C) 2011 Authors and OPENCORES.ORG                 //
//                                                              //
// This source file may be used and distributed without         //
// restriction provided that this copyright statement is not    //
// removed from the file and that any derivative work contains  //
// the original copyright notice and the associated disclaimer. //
//                                                              //
// This source file is free software; you can redistribute it   //
// and/or modify it under the terms of the GNU Lesser General   //
// Public License as published by the Free Software Foundation; //
// either version 2.1 of the License, or (at your option) any   //
// later version.                                               //
//                                                              //
// This source is distributed in the hope that it will be       //
// useful, but WITHOUT ANY WARRANTY; without even the implied   //
// warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR      //
// PURPOSE.  See the GNU Lesser General Public License for more //
// details.                                                     //
//                                                              //
// You should have received a copy of the GNU Lesser General    //
// Public License along with this source; if not, download it   //
// from http://www.opencores.org/lgpl.shtml                     //
//                                                              //
----------------------------------------------------------------*/
 
 
#include "address_map.h"
#include "line-buffer.h"
#include "timer.h"
#include "utilities.h"
#include "packet.h"
#include "udp.h"
 
 
int         udp_checksum_errors_g = 0;
block_t*    udp_file_g = NULL;
block_t*    udp_current_block_g = NULL;
 
 
void parse_udp_packet(char * buf, packet_t* rx_packet)
{
    unsigned int udp_src_port    = buf[0]<<8|buf[1];
    unsigned int udp_dst_port    = buf[2]<<8|buf[3];
    unsigned int udp_len         = buf[4]<<8|buf[5];
 
    unsigned short prot_udp=17;
    unsigned short word16;
    unsigned long  sum = 0;
    unsigned int   checksum;
    int mode_offset;
    int binary_mode;
    int i;
 
 
    for (i=0;i<4;i=i+2){
            word16 =((rx_packet->src_ip[i]<<8)&0xFF00)+(rx_packet->src_ip[i+1]&0xFF);
            sum=sum+word16;
    }
    for (i=0;i<4;i=i+2){
            word16 =((rx_packet->dst_ip[i]<<8)&0xFF00)+(rx_packet->dst_ip[i+1]&0xFF);
            sum=sum+word16;
    }
    // the protocol number and the length of the TCP packet
    sum = sum + prot_udp + udp_len;
 
    checksum = header_checksum16(buf, udp_len, sum);
 
    if (checksum)
        udp_checksum_errors_g++;
 
 
     /* TFTP */
    if (udp_dst_port == 69 && !checksum) {
        unsigned int opcode = buf[8]<<8|buf[9];
        unsigned int block  = buf[10]<<8|buf[11];
        int tftp_len = udp_len - 12;
 
        mode_offset = next_string(&buf[10]);
        binary_mode = strcmp("octet", &buf[10+mode_offset]);
 
        switch (opcode) {
 
            case  UDP_READ:
                udp_reply(rx_packet, udp_dst_port, udp_src_port, 0, UDP_ERROR);
                break;
 
            case  UDP_WRITE:
                udp_file_g = init_buffer_512();
                udp_file_g->filename = malloc(256);
                strcpy(udp_file_g->filename, &buf[10]);
 
                if (strncmp(&buf[10], "vmlinux", 7) == 0)
                    udp_file_g->linux_boot = 1;
 
                udp_current_block_g = udp_file_g;
 
                if (binary_mode)
                    udp_reply(rx_packet, udp_dst_port, udp_src_port, 0, UDP_ACK);
                else
                    udp_reply(rx_packet, udp_dst_port, udp_src_port, 0, UDP_ERROR);
                break;
 
 
            case  UDP_DATA:
                udp_reply(rx_packet, udp_dst_port, udp_src_port, block, UDP_ACK);
 
                if (block > udp_file_g->last_block) {
                    // Have not already received this block
                    udp_file_g->last_block = block;
 
                    /* receive and save a block */
                    udp_current_block_g->bytes = tftp_len;
                    udp_file_g->total_bytes += tftp_len;
                    udp_file_g->total_blocks++;
 
                    memcpy(udp_current_block_g->buf512, &buf[12], tftp_len);
 
                    /* Prepare the next block */
                    if (tftp_len == 512) {
                        udp_current_block_g->next = init_buffer_512();
                        udp_current_block_g = udp_current_block_g->next;
                        }
                    else { /* Last block */
                        udp_file_g->ready = 1;
                        }
                    }
                break;
 
 
            default: break;
            }
        }
}
 
 
 
void udp_reply(packet_t* rx_packet, int udp_src_port, int udp_dst_port, int block, int reply_type)
{
    char* buf = (char*)ETHMAC_TX_BUFFER;
    unsigned short checksum;
    unsigned short prot_udp=17;
    unsigned short udp_len;
    unsigned short word16;
    unsigned long  sum = 0;
    int i;
    mac_ip_t target;
 
    target.mac[0] = rx_packet->src_mac[0];
    target.mac[1] = rx_packet->src_mac[1];
    target.mac[2] = rx_packet->src_mac[2];
    target.mac[3] = rx_packet->src_mac[3];
    target.mac[4] = rx_packet->src_mac[4];
    target.mac[5] = rx_packet->src_mac[5];
    target.ip[0]  = rx_packet->src_ip[0];
    target.ip[1]  = rx_packet->src_ip[1];
    target.ip[2]  = rx_packet->src_ip[2];
    target.ip[3]  = rx_packet->src_ip[3];
 
    /* udp header */
    buf[34] = (udp_src_port & 0xff00)>>8;
    buf[35] =  udp_src_port & 0xff;
    buf[36] = (udp_dst_port & 0xff00)>>8;
    buf[37] =  udp_dst_port & 0xff;
 
    if (reply_type == UDP_ACK)
        udp_len = 8+4;
    else /* error */
        udp_len = 8+2+2+14;
 
 
    buf[38] = (udp_len      & 0xff00)>>8;
    buf[39] =  udp_len      & 0xff;
 
    buf[40] = 0;  // checksum
    buf[41] = 0;  // checksum
 
    // -------------------------------------
    // tftf payload
    // -------------------------------------
    // Opcode
    buf[42] = 0;
    buf[43] = reply_type & 0xff;  // Acknowledgment
 
    if (reply_type == UDP_ACK) {
        // block
        buf[44] = (block & 0xff00)>>8;
        buf[45] =  block & 0xff;
        }
    else {/* error */
        // Error Code
        buf[44] = 0;
        buf[45] = 0;
        // 46 to 59
        strncpy(&buf[46], "Not supported", 14);
        }
 
 
    // -------------------------------------
    // UDP Checksum calculation
    // -------------------------------------
    for (i=0;i<4;i=i+2){
            word16 =((rx_packet->src_ip[i]<<8)&0xFF00)+(rx_packet->src_ip[i+1]&0xFF);
            sum=sum+word16;
    }
    for (i=0;i<4;i=i+2){
            word16 =((rx_packet->dst_ip[i]<<8)&0xFF00)+(rx_packet->dst_ip[i+1]&0xFF);
            sum=sum+word16;
    }
    // the protocol number and the length of the TCP packet
    sum = sum + prot_udp + udp_len;
 
    checksum = header_checksum16(&buf[34], udp_len, sum);
    buf[40] = (checksum & 0xff00)>>8;  // checksum
    buf[41] =  checksum & 0xff;        // checksum
 
    ip_header(&buf[14], &target, 20+udp_len, 17); /* 20 byes of tcp  options, bytes 14 to 33, ip_proto = 17, UDP */
    ethernet_header(buf, &target, 0x0800);  /*bytes 0 to 13*/
    tx_packet(34+udp_len);  // packet length in bytes
}
 
 
block_t* init_buffer_512()
{
    block_t* block = malloc(sizeof(block_t));
    block->buf512 = malloc(512);
    block->next   = NULL;
    block->bytes  = 0;
    block->last_block  = 0;
    block->total_bytes  = 0;
    block->total_blocks  = 0;
    block->ready  = 0;
    block->filename  = NULL;
    block->linux_boot = 0;
    return block;
}