mbed OS5

Fork of UIPEthernet by Zoltan Hudak

UIPEthernet.cpp

Committer:
hudakz
Date:
2015-03-08
Revision:
4:d774541a34da
Parent:
2:049ce85163c5
Child:
8:4acb22344932

File content as of revision 4:d774541a34da:

/*
 UIPEthernet.cpp - Arduino implementation of a uIP wrapper class.
 Copyright (c) 2013 Norbert Truchsess <norbert.truchsess@t-online.de>
 All rights reserved.

 Modified (ported to mbed) by Zoltan Hudak <hudakz@inbox.com>

 This program is free software: you can redistribute it and/or modify
 it under the terms of the GNU General Public License as published by
 the Free Software Foundation, either version 3 of the License, or
 (at your option) any later version.

 This program 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 General Public License for more details.

 You should have received a copy of the GNU General Public License
 along with this program.  If not, see <http://www.gnu.org/licenses/>.
  */
#include <mbed.h>
#include "UIPEthernet.h"
#include "utility/Enc28J60Network.h"

#if (defined UIPETHERNET_DEBUG || defined UIPETHERNET_DEBUG_CHKSUM)
    #include "HardwareSerial.h"
#endif
#include "UIPUdp.h"

extern "C"
{
#include "utility/uip-conf.h"
#include "utility/uip.h"
#include "utility/uip_arp.h"
#include "utility/uip_timer.h"
#include "utility/millis.h"
}
#define ETH_HDR ((struct uip_eth_hdr*) &uip_buf[0])

memhandle UIPEthernetClass::        in_packet(NOBLOCK);
memhandle UIPEthernetClass::        uip_packet(NOBLOCK);
uint8_t UIPEthernetClass::          uip_hdrlen(0);
uint8_t UIPEthernetClass::          packetstate(0);

IPAddress UIPEthernetClass::        _dnsServerAddress;
DhcpClass* UIPEthernetClass::       _dhcp(NULL);

unsigned long UIPEthernetClass::    periodic_timer;

/**
 * @brief
 * @note
 * @param
 * @retval
 */
void enc28J60_mempool_block_move_callback(memaddress dest, memaddress src, memaddress len) {

    //void

    //Enc28J60Network::memblock_mv_cb(uint16_t dest, uint16_t src, uint16_t len)
    //{
    //as ENC28J60 DMA is unable to copy single bytes:
    if(len == 1) {
        UIPEthernet.network.writeByte(dest, UIPEthernet.network.readByte(src));
    }
    else {

        // calculate address of last byte
        len += src - 1;

        /*  1. Appropriately program the EDMAST, EDMAND
       and EDMADST register pairs. The EDMAST
       registers should point to the first byte to copy
       from, the EDMAND registers should point to the
       last byte to copy and the EDMADST registers
       should point to the first byte in the destination
       range. The destination range will always be
       linear, never wrapping at any values except from
       8191 to 0 (the 8-Kbyte memory boundary).
       Extreme care should be taken when
       programming the start and end pointers to
       prevent a never ending DMA operation which
       would overwrite the entire 8-Kbyte buffer.
       */
        UIPEthernet.network.writeRegPair(EDMASTL, src);
        UIPEthernet.network.writeRegPair(EDMADSTL, dest);

        if((src <= RXSTOP_INIT) && (len > RXSTOP_INIT))
            len -= (RXSTOP_INIT - RXSTART_INIT);
        UIPEthernet.network.writeRegPair(EDMANDL, len);

        /*
       2. If an interrupt at the end of the copy process is
       desired, set EIE.DMAIE and EIE.INTIE and
       clear EIR.DMAIF.

       3. Verify that ECON1.CSUMEN is clear. */
        UIPEthernet.network.writeOp(ENC28J60_BIT_FIELD_CLR, ECON1, ECON1_CSUMEN);

        /* 4. Start the DMA copy by setting ECON1.DMAST. */
        UIPEthernet.network.writeOp(ENC28J60_BIT_FIELD_SET, ECON1, ECON1_DMAST);

        // wait until runnig DMA is completed
        while(UIPEthernet.network.readOp(ENC28J60_READ_CTRL_REG, ECON1) & ECON1_DMAST);
    }
}

// Because uIP isn't encapsulated within a class we have to use global

// variables, so we can only have one TCP/IP stack per program.
UIPEthernetClass::UIPEthernetClass(PinName mosi, PinName miso, PinName sck, PinName cs) :
    network(mosi, miso, sck, cs) {
    millis_start();
}

#if UIP_UDP

/**
 * @brief
 * @note
 * @param
 * @retval
 */
int UIPEthernetClass::begin(const uint8_t* mac) {
    static DhcpClass    s_dhcp;
    _dhcp = &s_dhcp;

    // Initialise the basic info
    init(mac);

    // Now try to get our config info from a DHCP server
    int ret = _dhcp->beginWithDHCP((uint8_t*)mac);
    if(ret == 1) {

        // We've successfully found a DHCP server and got our configuration info, so set things
        // accordingly
        configure(_dhcp->getLocalIp(), _dhcp->getDnsServerIp(), _dhcp->getGatewayIp(), _dhcp->getSubnetMask());
    }

    return ret;
}
#endif

/**
 * @brief
 * @note
 * @param
 * @retval
 */
void UIPEthernetClass::begin(const uint8_t* mac, IPAddress ip) {
    IPAddress   dns = ip;
    dns[3] = 1;
    begin(mac, ip, dns);
}

/**
 * @brief
 * @note
 * @param
 * @retval
 */
void UIPEthernetClass::begin(const uint8_t* mac, IPAddress ip, IPAddress dns) {
    IPAddress   gateway = ip;
    gateway[3] = 1;
    begin(mac, ip, dns, gateway);
}

/**
 * @brief
 * @note
 * @param
 * @retval
 */
void UIPEthernetClass::begin(const uint8_t* mac, IPAddress ip, IPAddress dns, IPAddress gateway) {
    IPAddress   subnet(255, 255, 255, 0);
    begin(mac, ip, dns, gateway, subnet);
}

/**
 * @brief
 * @note
 * @param
 * @retval
 */
void UIPEthernetClass::begin(const uint8_t* mac, IPAddress ip, IPAddress dns, IPAddress gateway, IPAddress subnet) {
    init(mac);
    configure(ip, dns, gateway, subnet);
}

/**
 * @brief
 * @note
 * @param
 * @retval
 */
int UIPEthernetClass::maintain(void) {
    tick();

    int rc = DHCP_CHECK_NONE;
#if UIP_UDP
    if(_dhcp != NULL) {

        //we have a pointer to dhcp, use it
        rc = _dhcp->checkLease();
        switch(rc) {
        case DHCP_CHECK_NONE:
            //nothing done
            break;

        case DHCP_CHECK_RENEW_OK:
        case DHCP_CHECK_REBIND_OK:
            //we might have got a new IP.
            configure(_dhcp->getLocalIp(), _dhcp->getDnsServerIp(), _dhcp->getGatewayIp(), _dhcp->getSubnetMask());
            break;

        default:
            //this is actually a error, it will retry though
            break;
        }
    }

    return rc;
#endif
}

/**
 * @brief
 * @note
 * @param
 * @retval
 */
IPAddress UIPEthernetClass::localIP(void) {
    IPAddress       ret;
    uip_ipaddr_t    a;
    uip_gethostaddr(a);
    return ip_addr_uip(a);
}

/**
 * @brief
 * @note
 * @param
 * @retval
 */
IPAddress UIPEthernetClass::subnetMask(void) {
    IPAddress       ret;
    uip_ipaddr_t    a;
    uip_getnetmask(a);
    return ip_addr_uip(a);
}

/**
 * @brief
 * @note
 * @param
 * @retval
 */
IPAddress UIPEthernetClass::gatewayIP(void) {
    IPAddress       ret;
    uip_ipaddr_t    a;
    uip_getdraddr(a);
    return ip_addr_uip(a);
}

/**
 * @brief
 * @note
 * @param
 * @retval
 */
IPAddress UIPEthernetClass::dnsServerIP(void) {
    return _dnsServerAddress;
}

/**
 * @brief
 * @note
 * @param
 * @retval
 */
void UIPEthernetClass::tick(void)
{
    if(in_packet == NOBLOCK) {
        in_packet = network.receivePacket();
#ifdef UIPETHERNET_DEBUG
        if(in_packet != NOBLOCK) {
            Serial.print(F("--------------\nreceivePacket: "));
            Serial.println(in_packet);
        }
#endif
    }
    if(in_packet != NOBLOCK) {
        packetstate = UIPETHERNET_FREEPACKET;
        uip_len = network.blockSize(in_packet);
        if(uip_len > 0) {
            network.readPacket(in_packet, 0, (uint8_t*)uip_buf, UIP_BUFSIZE);
            if(ETH_HDR->type == HTONS(UIP_ETHTYPE_IP)) {
                uip_packet = in_packet; //required for upper_layer_checksum of in_packet!
#ifdef UIPETHERNET_DEBUG
                Serial.print(F("readPacket type IP, uip_len: "));
                Serial.println(uip_len);
#endif
                uip_arp_ipin();
                uip_input();
                if(uip_len > 0) {
                    uip_arp_out();
                    network_send();
                }
            }
            else
            if(ETH_HDR->type == HTONS(UIP_ETHTYPE_ARP))
            {
#ifdef UIPETHERNET_DEBUG
                Serial.print(F("readPacket type ARP, uip_len: "));
                Serial.println(uip_len);
#endif
                uip_arp_arpin();
                if(uip_len > 0) {
                    network_send();
                }
            }
        }
        if(in_packet != NOBLOCK && (packetstate & UIPETHERNET_FREEPACKET))
        {
#ifdef UIPETHERNET_DEBUG
            Serial.print(F("freeing packet: "));
            Serial.println(in_packet);
#endif
            network.freePacket();
            in_packet = NOBLOCK;
        }
    }
    unsigned long   now = millis();

#if UIP_CLIENT_TIMER >= 0
    bool            periodic = (long)(now - periodic_timer) >= 0;
    for(int i = 0; i < UIP_CONNS; i++)
    {
#else
        if((long)(now - periodic_timer) >= 0) {
            periodic_timer = now + UIP_PERIODIC_TIMER;

            for(int i = 0; i < UIP_CONNS; i++)
            {
#endif
                uip_conn = &uip_conns[i];
#if UIP_CLIENT_TIMER >= 0
                if(periodic)
                {
#endif
                    uip_process(UIP_TIMER);
#if UIP_CLIENT_TIMER >= 0
                }
                else {
                    if((long)(now - ((uip_userdata_t*)uip_conn->appstate)->timer) >= 0)
                        uip_process(UIP_POLL_REQUEST);
                    else
                        continue;
                }
#endif
                // If the above function invocation resulted in data that

                // should be sent out on the Enc28J60Network, the global variable
                // uip_len is set to a value > 0.
                if(uip_len > 0) {
                    uip_arp_out();
                    network_send();
                }
            }
#if UIP_CLIENT_TIMER >= 0
            if(periodic) {
                periodic_timer = now + UIP_PERIODIC_TIMER;
#endif
#if UIP_UDP
                for(int i = 0; i < UIP_UDP_CONNS; i++) {
                    uip_udp_periodic(i);

                    // If the above function invocation resulted in data that
                    // should be sent out on the Enc28J60Network, the global variable
                    // uip_len is set to a value > 0. */
                    if(uip_len > 0) {
                        UIPUDP::_send((uip_udp_userdata_t *) (uip_udp_conns[i].appstate));
                    }
                }
#endif /* UIP_UDP */
            }
        }

        /**
 * @brief
 * @note
 * @param
 * @retval
 */
        bool UIPEthernetClass::network_send(void) {
            if(packetstate & UIPETHERNET_SENDPACKET)
            {
#ifdef UIPETHERNET_DEBUG
                Serial.print(F("Enc28J60Network_send uip_packet: "));
                Serial.print(uip_packet);
                Serial.print(F(", hdrlen: "));
                Serial.println(uip_hdrlen);
#endif
                UIPEthernet.network.writePacket(uip_packet, 0, uip_buf, uip_hdrlen);
                packetstate &= ~UIPETHERNET_SENDPACKET;
                goto sendandfree;
            }

            uip_packet = Enc28J60Network::allocBlock(uip_len);
            if(uip_packet != NOBLOCK)
            {
#ifdef UIPETHERNET_DEBUG
                Serial.print(F("Enc28J60Network_send uip_buf (uip_len): "));
                Serial.print(uip_len);
                Serial.print(F(", packet: "));
                Serial.println(uip_packet);
#endif
                UIPEthernet.network.writePacket(uip_packet, 0, uip_buf, uip_len);
                goto sendandfree;
            }

            return false;
sendandfree:
            network.sendPacket(uip_packet);
            Enc28J60Network::freeBlock(uip_packet);
            uip_packet = NOBLOCK;
            return true;
        }

        /**
 * @brief
 * @note
 * @param
 * @retval
 */
        void UIPEthernetClass::init(const uint8_t* mac) {
            periodic_timer = millis() + UIP_PERIODIC_TIMER;

            network.init((uint8_t*)mac);
            uip_seteth_addr(mac);

            uip_init();
            uip_arp_init();
        }

        /**
 * @brief
 * @note
 * @param
 * @retval
 */
        void UIPEthernetClass::configure(IPAddress ip, IPAddress dns, IPAddress gateway, IPAddress subnet) {
            uip_ipaddr_t    ipaddr;

            uip_ip_addr(ipaddr, ip);
            uip_sethostaddr(ipaddr);

            uip_ip_addr(ipaddr, gateway);
            uip_setdraddr(ipaddr);

            uip_ip_addr(ipaddr, subnet);
            uip_setnetmask(ipaddr);

            _dnsServerAddress = dns;
        }

        //UIPEthernetClass UIPEthernet;

        /*---------------------------------------------------------------------------*/
        uint16_t UIPEthernetClass::chksum(uint16_t sum, const uint8_t* data, uint16_t len) {
            uint16_t        t;
            const uint8_t*  dataptr;
            const uint8_t*  last_byte;

            dataptr = data;
            last_byte = data + len - 1;

            while(dataptr < last_byte) {

                /* At least two more bytes */
                t = (dataptr[0] << 8) + dataptr[1];
                sum += t;
                if(sum < t) {
                    sum++;  /* carry */
                }

                dataptr += 2;
            }

            if(dataptr == last_byte) {
                t = (dataptr[0] << 8) + 0;
                sum += t;
                if(sum < t) {
                    sum++;  /* carry */
                }
            }

            /* Return sum in host byte order. */
            return sum;
        }

        /*---------------------------------------------------------------------------*/
        uint16_t UIPEthernetClass::ipchksum(void) {
            uint16_t    sum;

            sum = chksum(0, &uip_buf[UIP_LLH_LEN], UIP_IPH_LEN);
            return(sum == 0) ? 0xffff : htons(sum);
        }

        /*---------------------------------------------------------------------------*/
        uint16_t
#if UIP_UDP
        UIPEthernetClass::upper_layer_chksum(uint8_t proto)
#else
        uip_tcpchksum (void)
#endif
        {
            uint16_t    upper_layer_len;
            uint16_t    sum;

#if UIP_CONF_IPV6
            upper_layer_len = (((u16_t) (BUF->len[0]) << 8) + BUF->len[1]);
#else /* UIP_CONF_IPV6 */
            upper_layer_len = (((u16_t) (BUF->len[0]) << 8) + BUF->len[1]) - UIP_IPH_LEN;
#endif /* UIP_CONF_IPV6 */

            /* First sum pseudoheader. */

            /* IP protocol and length fields. This addition cannot carry. */
#if UIP_UDP
            sum = upper_layer_len + proto;
#else
            sum = upper_layer_len + UIP_PROTO_TCP;
#endif
            /* Sum IP source and destination addresses. */

            sum = UIPEthernetClass::chksum(sum, (u8_t*) &BUF->srcipaddr[0], 2 * sizeof(uip_ipaddr_t));

            uint8_t upper_layer_memlen;
#if UIP_UDP
            switch(proto) {
            //    case UIP_PROTO_ICMP:
            //    case UIP_PROTO_ICMP6:
            //      upper_layer_memlen = upper_layer_len;
            //      break;
            case UIP_PROTO_UDP:
                upper_layer_memlen = UIP_UDPH_LEN;
                break;

            default:
                //  case UIP_PROTO_TCP:
#endif
                upper_layer_memlen = (BUF->tcpoffset >> 4) << 2;
#if UIP_UDP
                break;
            }
#endif
            sum = UIPEthernetClass::chksum(sum, &uip_buf[UIP_IPH_LEN + UIP_LLH_LEN], upper_layer_memlen);
#ifdef UIPETHERNET_DEBUG_CHKSUM
            Serial.print(F("chksum uip_buf["));
            Serial.print(UIP_IPH_LEN + UIP_LLH_LEN);
            Serial.print(F("-"));
            Serial.print(UIP_IPH_LEN + UIP_LLH_LEN + upper_layer_memlen);
            Serial.print(F("]: "));
            Serial.println(htons(sum), HEX);
#endif
            if(upper_layer_memlen < upper_layer_len) {
                sum = network.chksum
                    (
                        sum, UIPEthernetClass::uip_packet, UIP_IPH_LEN +
                        UIP_LLH_LEN +
                        upper_layer_memlen, upper_layer_len -
                        upper_layer_memlen
                    );
#ifdef UIPETHERNET_DEBUG_CHKSUM
                Serial.print(F("chksum uip_packet("));
                Serial.print(uip_packet);
                Serial.print(F(")["));
                Serial.print(UIP_IPH_LEN + UIP_LLH_LEN + upper_layer_memlen);
                Serial.print(F("-"));
                Serial.print(UIP_IPH_LEN + UIP_LLH_LEN + upper_layer_len);
                Serial.print(F("]: "));
                Serial.println(htons(sum), HEX);
#endif
            }
            return(sum == 0) ? 0xffff : htons(sum);
        }

        /**
 * @brief
 * @note
 * @param
 * @retval
 */
        uint16_t uip_ipchksum(void) {
            return UIPEthernet.ipchksum();
        }

#if UIP_UDP
        /**
 * @brief
 * @note
 * @param
 * @retval
 */
        uint16_t uip_tcpchksum(void) {
            uint16_t    sum = UIPEthernet.upper_layer_chksum(UIP_PROTO_TCP);
            return sum;
        }

        /**
 * @brief
 * @note
 * @param
 * @retval
 */
        uint16_t uip_udpchksum(void) {
            uint16_t    sum = UIPEthernet.upper_layer_chksum(UIP_PROTO_UDP);
            return sum;
        }
#endif