Implementation of the NetworkSocketAPI for LWIP

Dependencies:   lwip-eth lwip-sys lwip

Dependents:   HelloLWIPInterface HelloLWIPInterfaceNonBlocking LWIPInterfaceTests SimpleHTTPExample ... more

LWIPInterface.cpp

Committer:
geky
Date:
2016-04-05
Revision:
12:899403b675fe
Parent:
11:82796df87b0a
Child:
13:57d9e1721826

File content as of revision 12:899403b675fe:

/* LWIP implementation of NetworkInterfaceAPI
 * Copyright (c) 2015 ARM Limited
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

#include "LWIPInterface.h"

#include "mbed.h"
#include "lwip/inet.h"
#include "lwip/netif.h"
#include "lwip/dhcp.h"
#include "lwip/tcpip.h"
#include "lwip/sockets.h"
#include "lwip/netdb.h"
#include "netif/etharp.h"
#include "eth_arch.h"

#if 0
/* TCP/IP and Network Interface Initialisation */
static struct netif netif;

static char ip_addr[NS_IP_SIZE] = "\0";
static char mac_addr[NS_MAC_SIZE] = "\0";

static Semaphore tcpip_inited(0);
static Semaphore netif_linked(0);
static Semaphore netif_up(0);

static void tcpip_init_done(void *)
{
    tcpip_inited.release();
}

static void netif_link_callback(struct netif *netif)
{
    if (netif_is_link_up(netif)) {
        netif_linked.release();
    }
}

static void netif_status_callback(struct netif *netif)
{
    if (netif_is_up(netif)) {
        strcpy(ip_addr, inet_ntoa(netif->ip_addr));
        netif_up.release();
    }
}

static void init_netif(ip_addr_t *ipaddr, ip_addr_t *netmask, ip_addr_t *gw)
{
    tcpip_init(tcpip_init_done, NULL);
    tcpip_inited.wait();

    memset((void*) &netif, 0, sizeof(netif));
    netif_add(&netif, ipaddr, netmask, gw, NULL, eth_arch_enetif_init, tcpip_input);
    netif_set_default(&netif);

    netif_set_link_callback  (&netif, netif_link_callback);
    netif_set_status_callback(&netif, netif_status_callback);
}

static void set_mac_address(void)
{
#if (MBED_MAC_ADDRESS_SUM != MBED_MAC_ADDR_INTERFACE)
    snprintf(mac_addr, 19, "%02x:%02x:%02x:%02x:%02x:%02x", MBED_MAC_ADDR_0, MBED_MAC_ADDR_1, MBED_MAC_ADDR_2,
             MBED_MAC_ADDR_3, MBED_MAC_ADDR_4, MBED_MAC_ADDR_5);
#else
    char mac[6];
    mbed_mac_address(mac);
    snprintf(mac_addr, 19, "%02x:%02x:%02x:%02x:%02x:%02x", mac[0], mac[1], mac[2], mac[3], mac[4], mac[5]);
#endif
}

#if 0
// LWIPInterface implementation
int32_t LWIPInterface::connect()
{
    // Set up network
    set_mac_address();
    init_netif(0, 0, 0);

    // Connect to network
    eth_arch_enable_interrupts();

    dhcp_start(&netif);

    // Wait for an IP Address
    // -1: error, 0: timeout
    if (netif_up.wait(LWIP_TIMEOUT) < 0) {
        return NS_ERROR_TIMEOUT;
    }

    return 0;
}

int32_t LWIPInterface::disconnect()
{
    dhcp_release(&netif);
    dhcp_stop(&netif);

    eth_arch_disable_interrupts();

    return 0;
}

const char *LWIPInterface::getIPAddress()
{
    return ip_addr;
}

const char *LWIPInterface::getMACAddress()
{
    return mac_addr;
}

SocketInterface *LWIPInterface::createSocket(ns_protocol_t proto)
{
    int type = (proto == NS_UDP) ? SOCK_DGRAM : SOCK_STREAM;
    int fd = lwip_socket(AF_INET, type, 0);
    if (fd < 0) {
        return 0;
    }

    return new LWIPSocket(fd);
}

void LWIPInterface::destroySocket(SocketInterface *siface)
{
    LWIPSocket *socket = (LWIPSocket *)siface;
    lwip_close(socket->fd);

    delete socket;
}


// TCP SocketInterface implementation
int32_t LWIPInterface::LWIPSocket::open(const char *ip, uint16_t port)
{
    struct sockaddr_in host;
    memset(&host, 0, sizeof host);
    inet_aton(ip, &host.sin_addr);
    host.sin_family = AF_INET;
    host.sin_port = htons(port);

    if (lwip_connect(fd, (const struct sockaddr *)&host, sizeof host) < 0) {
        return NS_ERROR_NO_CONNECTION;
    }

    return 0;
}

int32_t LWIPInterface::LWIPSocket::close()
{
    return 0;
}

int32_t LWIPInterface::LWIPSocket::send(const void *voiddata, uint32_t size)
{
    uint8_t *data = (uint8_t *)voiddata;
    uint32_t writtenLen = 0;

    while (writtenLen < size) {
        int ret = lwip_send(fd, data + writtenLen, size - writtenLen, 0);

        if (ret > 0) {
            writtenLen += ret;
        } else if (ret == 0) {
            return NS_ERROR_NO_CONNECTION;
        } else {
            return NS_ERROR_DEVICE_ERROR;
        }
    }

    return writtenLen;
}

int32_t LWIPInterface::LWIPSocket::recv(void *data, uint32_t size)
{
    int ret = lwip_recv(fd, data, size, MSG_DONTWAIT);

    if (ret > 0) {
        return ret;
    } else if (ret == 0) {
        return NS_ERROR_NO_CONNECTION;
    } else if (ret == -1) {
        return NS_ERROR_WOULD_BLOCK;
    } else {
        return NS_ERROR_DEVICE_ERROR;
    }
}
#endif

    /** Start the interface
     *  @return     0 on success, negative on failure
     */
    virtual int connect() = 0;

    /** Stop the interface
     *  @return     0 on success, negative on failure
     */
    virtual int disconnect() = 0;

class NetworkInterface
{
public:
    virtual ~NetworkInterface() {};

    /** Get the internally stored IP address
    /return     IP address of the interface or null if not yet connected
    */
    virtual const char *get_ip_address() = 0;

    /** Get the internally stored MAC address
    /return     MAC address of the interface
    */
    virtual const char *get_mac_address() = 0;

    /** Get the current status of the interface
    /return     true if connected
    */
    virtual bool is_connected() {
        return get_ip_address() != NULL;
    }

    /** Looks up the specified host's IP address
    /param name Hostname to lookup
    /param dest Destination for IP address, must have space for SocketAddress::IP_SIZE
    /return     0 on success, negative on failure
    */
    virtual int gethostbyname(const char *name, char *dest);

protected:
    friend class Socket;
    friend class UDPSocket;
    friend class TCPSocket;
    friend class TCPServer;

    /** Enum of socket protocols
    /enum protocol_t
    */
    enum protocol_t {
        TCP, /*!< Socket is of TCP type */
        UDP, /*!< Socket is of UDP type */
    };

    /** Create a socket
    /param proto    The type of socket to open, TCP or UDP
    /return         The alocated socket or null on failure
    */
    virtual void *socket_create(protocol_t proto) = 0;

    /** Destroy a socket
    /param socket   Previously allocated socket
    */
    virtual void socket_destroy(void *handle) = 0;

    /** Set socket options
    \param handle   Socket handle
    \param optname  Option ID
    \param optval   Option value
    \param optlen   Length of the option value
    \return         0 on success, negative on failure
    */    
    virtual int socket_set_option(void *handle, int optname, const void *optval, unsigned int optlen) = 0;

    /** Get socket options
    \param handle   Socket handle
    \param optname  Option ID
    \param optval   Buffer pointer where to write the option value
    \param optlen   Length of the option value
    \return         0 on success, negative on failure
    */
    virtual int socket_get_option(void *handle, int optname, void *optval, unsigned int *optlen) = 0;

    /** Bind a server socket to a specific port
    \param handle   Socket handle
    \param port     The port to listen for incoming connections on
    \return         0 on success, negative on failure.
    */
    virtual int socket_bind(void *handle, int port) = 0;

    /** Start listening for incoming connections
    \param handle   Socket handle
    \param backlog  Number of pending connections that can be queued up at any
                    one time [Default: 1]
    \return         0 on success, negative on failure
    */
    virtual int socket_listen(void *handle, int backlog) = 0;

    /** Connects this TCP socket to the server
    \param handle   Socket handle
    \param address  SocketAddress to connect to
    \return         0 on success, negative on failure
    */
    virtual int socket_connect(void *handle, const SocketAddress &address) = 0;
    
    /** Check if the socket is connected
    \param handle   Socket handle
    \return         true if connected, false otherwise
    */
    virtual bool socket_is_connected(void *handle) = 0;

    /** Accept a new connection.
    \param handle   Socket handle
    \param socket   A TCPSocket instance that will handle the incoming connection.
    \return         0 on success, negative on failure.
    \note This call is not-blocking, if this call would block, must
          immediately return NSAPI_ERROR_WOULD_WAIT
    */
    virtual int socket_accept(void *handle, void **connection) = 0;

    /** Send data to the remote host
    \param handle   Socket handle
    \param data     The buffer to send to the host
    \param size     The length of the buffer to send
    \return         Number of written bytes on success, negative on failure
    \note This call is not-blocking, if this call would block, must
          immediately return NSAPI_ERROR_WOULD_WAIT
    */
    virtual int socket_send(void *handle, const void *data, unsigned size) = 0;

    /** Receive data from the remote host
    \param handle   Socket handle
    \param data     The buffer in which to store the data received from the host
    \param size     The maximum length of the buffer
    \return         Number of received bytes on success, negative on failure
    \note This call is not-blocking, if this call would block, must
          immediately return NSAPI_ERROR_WOULD_WAIT
    */
    virtual int socket_recv(void *handle, void *data, unsigned size) = 0;

    /** Send a packet to a remote endpoint
    \param handle   Socket handle
    \param address  The remote SocketAddress
    \param data     The packet to be sent
    \param size     The length of the packet to be sent
    \return the number of written bytes on success, negative on failure
    \note This call is not-blocking, if this call would block, must
          immediately return NSAPI_ERROR_WOULD_WAIT
    */
    virtual int socket_sendto(void *handle, const SocketAddress &address, const void *data, unsigned size) = 0;

    /** Receive a packet from a remote endpoint
    \param handle   Socket handle
    \param address  Destination for the remote SocketAddress or null
    \param buffer   The buffer for storing the incoming packet data
                    If a packet is too long to fit in the supplied buffer,
                    excess bytes are discarded
    \param size     The length of the buffer
    \return the number of received bytes on success, negative on failure
    \note This call is not-blocking, if this call would block, must
          immediately return NSAPI_ERROR_WOULD_WAIT
    */
    virtual int socket_recvfrom(void *handle, SocketAddress *address, void *buffer, unsigned size) = 0;

    /** Close the socket
    \param handle   Socket handle
    \param shutdown  free the left-over data in message queues
    */
    virtual int socket_close(void *handle, bool shutdown) = 0;

    /** Register a callback on when a new connection is ready
    \param handle   Socket handle
    \param callback Function to call when accept will succeed, may be called in
                    interrupt context.
    \param id       Argument to pass to callback
    */
    virtual void socket_attach_accept(void *handle, void (*callback)(void *), void *id) = 0;

    /** Register a callback on when send is ready
    \param handle   Socket handle
    \param callback Function to call when accept will succeed, may be called in
                    interrupt context.
    \param id       Argument to pass to callback
    */
    virtual void socket_attach_send(void *handle, void (*callback)(void *), void *id) = 0;

    /** Register a callback on when recv is ready
    \param handle   Socket handle
    \param callback Function to call when accept will succeed, may be called in
                    interrupt context.
    \param id       Argument to pass to callback
    */
    virtual void socket_attach_recv(void *handle, void (*callback)(void *), void *id) = 0;
}; 
#endif