| /** |
| * @file |
| * Sequential API Internal module |
| * |
| */ |
| |
| /* |
| * Copyright (c) 2001-2004 Swedish Institute of Computer Science. |
| * 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 the author may not be used to endorse or promote products |
| * derived from this software without specific prior written permission. |
| * |
| * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``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 DISCLAIMED. IN NO EVENT |
| * SHALL THE AUTHOR 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. |
| * |
| * This file is part of the lwIP TCP/IP stack. |
| * |
| * Author: Adam Dunkels <adam@sics.se> |
| * |
| */ |
| |
| #include "lwip/opt.h" |
| |
| #if LWIP_NETCONN /* don't build if not configured for use in lwipopts.h */ |
| |
| #include "lwip/priv/api_msg.h" |
| |
| #include "lwip/ip.h" |
| #include "lwip/ip_addr.h" |
| #include "lwip/udp.h" |
| #include "lwip/tcp.h" |
| #include "lwip/raw.h" |
| |
| #include "lwip/memp.h" |
| #include "lwip/igmp.h" |
| #include "lwip/dns.h" |
| #include "lwip/mld6.h" |
| #include "lwip/priv/tcpip_priv.h" |
| |
| #include <string.h> |
| |
| /* netconns are polled once per second (e.g. continue write on memory error) */ |
| #define NETCONN_TCP_POLL_INTERVAL 2 |
| |
| #define SET_NONBLOCKING_CONNECT(conn, val) do { if (val) { \ |
| netconn_set_flags(conn, NETCONN_FLAG_IN_NONBLOCKING_CONNECT); \ |
| } else { \ |
| netconn_clear_flags(conn, NETCONN_FLAG_IN_NONBLOCKING_CONNECT); }} while(0) |
| #define IN_NONBLOCKING_CONNECT(conn) netconn_is_flag_set(conn, NETCONN_FLAG_IN_NONBLOCKING_CONNECT) |
| |
| #if LWIP_NETCONN_FULLDUPLEX |
| #define NETCONN_MBOX_VALID(conn, mbox) (sys_mbox_valid(mbox) && ((conn->flags & NETCONN_FLAG_MBOXINVALID) == 0)) |
| #else |
| #define NETCONN_MBOX_VALID(conn, mbox) sys_mbox_valid(mbox) |
| #endif |
| |
| /* forward declarations */ |
| #if LWIP_TCP |
| #if LWIP_TCPIP_CORE_LOCKING |
| #define WRITE_DELAYED , 1 |
| #define WRITE_DELAYED_PARAM , u8_t delayed |
| #else /* LWIP_TCPIP_CORE_LOCKING */ |
| #define WRITE_DELAYED |
| #define WRITE_DELAYED_PARAM |
| #endif /* LWIP_TCPIP_CORE_LOCKING */ |
| static err_t lwip_netconn_do_writemore(struct netconn *conn WRITE_DELAYED_PARAM); |
| static err_t lwip_netconn_do_close_internal(struct netconn *conn WRITE_DELAYED_PARAM); |
| #endif |
| |
| static void netconn_drain(struct netconn *conn); |
| |
| #if LWIP_TCPIP_CORE_LOCKING |
| #define TCPIP_APIMSG_ACK(m) |
| #else /* LWIP_TCPIP_CORE_LOCKING */ |
| #define TCPIP_APIMSG_ACK(m) do { sys_sem_signal(LWIP_API_MSG_SEM(m)); } while(0) |
| #endif /* LWIP_TCPIP_CORE_LOCKING */ |
| |
| #if LWIP_NETCONN_FULLDUPLEX |
| const u8_t netconn_deleted = 0; |
| |
| int |
| lwip_netconn_is_deallocated_msg(void *msg) |
| { |
| if (msg == &netconn_deleted) { |
| return 1; |
| } |
| return 0; |
| } |
| #endif /* LWIP_NETCONN_FULLDUPLEX */ |
| |
| #if LWIP_TCP |
| const u8_t netconn_aborted = 0; |
| const u8_t netconn_reset = 0; |
| const u8_t netconn_closed = 0; |
| |
| /** Translate an error to a unique void* passed via an mbox */ |
| static void * |
| lwip_netconn_err_to_msg(err_t err) |
| { |
| switch (err) { |
| case ERR_ABRT: |
| return LWIP_CONST_CAST(void *, &netconn_aborted); |
| case ERR_RST: |
| return LWIP_CONST_CAST(void *, &netconn_reset); |
| case ERR_CLSD: |
| return LWIP_CONST_CAST(void *, &netconn_closed); |
| default: |
| LWIP_ASSERT("unhandled error", err == ERR_OK); |
| return NULL; |
| } |
| } |
| |
| int |
| lwip_netconn_is_err_msg(void *msg, err_t *err) |
| { |
| LWIP_ASSERT("err != NULL", err != NULL); |
| |
| if (msg == &netconn_aborted) { |
| *err = ERR_ABRT; |
| return 1; |
| } else if (msg == &netconn_reset) { |
| *err = ERR_RST; |
| return 1; |
| } else if (msg == &netconn_closed) { |
| *err = ERR_CLSD; |
| return 1; |
| } |
| return 0; |
| } |
| #endif /* LWIP_TCP */ |
| |
| |
| #if LWIP_RAW |
| /** |
| * Receive callback function for RAW netconns. |
| * Doesn't 'eat' the packet, only copies it and sends it to |
| * conn->recvmbox |
| * |
| * @see raw.h (struct raw_pcb.recv) for parameters and return value |
| */ |
| static u8_t |
| recv_raw(void *arg, struct raw_pcb *pcb, struct pbuf *p, |
| const ip_addr_t *addr) |
| { |
| struct pbuf *q; |
| struct netbuf *buf; |
| struct netconn *conn; |
| |
| LWIP_UNUSED_ARG(addr); |
| conn = (struct netconn *)arg; |
| |
| if ((conn != NULL) && NETCONN_MBOX_VALID(conn, &conn->recvmbox)) { |
| #if LWIP_SO_RCVBUF |
| int recv_avail; |
| SYS_ARCH_GET(conn->recv_avail, recv_avail); |
| if ((recv_avail + (int)(p->tot_len)) > conn->recv_bufsize) { |
| return 0; |
| } |
| #endif /* LWIP_SO_RCVBUF */ |
| /* copy the whole packet into new pbufs */ |
| q = pbuf_clone(PBUF_RAW, PBUF_RAM, p); |
| if (q != NULL) { |
| u16_t len; |
| buf = (struct netbuf *)memp_malloc(MEMP_NETBUF); |
| if (buf == NULL) { |
| pbuf_free(q); |
| return 0; |
| } |
| |
| buf->p = q; |
| buf->ptr = q; |
| ip_addr_copy(buf->addr, *ip_current_src_addr()); |
| buf->port = pcb->protocol; |
| |
| len = q->tot_len; |
| if (sys_mbox_trypost(&conn->recvmbox, buf) != ERR_OK) { |
| netbuf_delete(buf); |
| return 0; |
| } else { |
| #if LWIP_SO_RCVBUF |
| SYS_ARCH_INC(conn->recv_avail, len); |
| #endif /* LWIP_SO_RCVBUF */ |
| /* Register event with callback */ |
| API_EVENT(conn, NETCONN_EVT_RCVPLUS, len); |
| } |
| } |
| } |
| |
| return 0; /* do not eat the packet */ |
| } |
| #endif /* LWIP_RAW*/ |
| |
| #if LWIP_UDP |
| /** |
| * Receive callback function for UDP netconns. |
| * Posts the packet to conn->recvmbox or deletes it on memory error. |
| * |
| * @see udp.h (struct udp_pcb.recv) for parameters |
| */ |
| static void |
| recv_udp(void *arg, struct udp_pcb *pcb, struct pbuf *p, |
| const ip_addr_t *addr, u16_t port) |
| { |
| struct netbuf *buf; |
| struct netconn *conn; |
| u16_t len; |
| #if LWIP_SO_RCVBUF |
| int recv_avail; |
| #endif /* LWIP_SO_RCVBUF */ |
| |
| LWIP_UNUSED_ARG(pcb); /* only used for asserts... */ |
| LWIP_ASSERT("recv_udp must have a pcb argument", pcb != NULL); |
| LWIP_ASSERT("recv_udp must have an argument", arg != NULL); |
| conn = (struct netconn *)arg; |
| |
| if (conn == NULL) { |
| pbuf_free(p); |
| return; |
| } |
| |
| LWIP_ASSERT("recv_udp: recv for wrong pcb!", conn->pcb.udp == pcb); |
| |
| #if LWIP_SO_RCVBUF |
| SYS_ARCH_GET(conn->recv_avail, recv_avail); |
| if (!NETCONN_MBOX_VALID(conn, &conn->recvmbox) || |
| ((recv_avail + (int)(p->tot_len)) > conn->recv_bufsize)) { |
| #else /* LWIP_SO_RCVBUF */ |
| if (!NETCONN_MBOX_VALID(conn, &conn->recvmbox)) { |
| #endif /* LWIP_SO_RCVBUF */ |
| pbuf_free(p); |
| return; |
| } |
| |
| buf = (struct netbuf *)memp_malloc(MEMP_NETBUF); |
| if (buf == NULL) { |
| pbuf_free(p); |
| return; |
| } else { |
| buf->p = p; |
| buf->ptr = p; |
| ip_addr_set(&buf->addr, addr); |
| buf->port = port; |
| #if LWIP_NETBUF_RECVINFO |
| if (conn->flags & NETCONN_FLAG_PKTINFO) { |
| /* get the UDP header - always in the first pbuf, ensured by udp_input */ |
| const struct udp_hdr *udphdr = (const struct udp_hdr *)ip_next_header_ptr(); |
| buf->flags = NETBUF_FLAG_DESTADDR; |
| ip_addr_set(&buf->toaddr, ip_current_dest_addr()); |
| buf->toport_chksum = udphdr->dest; |
| } |
| #endif /* LWIP_NETBUF_RECVINFO */ |
| } |
| |
| len = p->tot_len; |
| if (sys_mbox_trypost(&conn->recvmbox, buf) != ERR_OK) { |
| netbuf_delete(buf); |
| return; |
| } else { |
| #if LWIP_SO_RCVBUF |
| SYS_ARCH_INC(conn->recv_avail, len); |
| #endif /* LWIP_SO_RCVBUF */ |
| /* Register event with callback */ |
| API_EVENT(conn, NETCONN_EVT_RCVPLUS, len); |
| } |
| } |
| #endif /* LWIP_UDP */ |
| |
| #if LWIP_TCP |
| /** |
| * Receive callback function for TCP netconns. |
| * Posts the packet to conn->recvmbox, but doesn't delete it on errors. |
| * |
| * @see tcp.h (struct tcp_pcb.recv) for parameters and return value |
| */ |
| static err_t |
| recv_tcp(void *arg, struct tcp_pcb *pcb, struct pbuf *p, err_t err) |
| { |
| struct netconn *conn; |
| u16_t len; |
| void *msg; |
| |
| LWIP_UNUSED_ARG(pcb); |
| LWIP_ASSERT("recv_tcp must have a pcb argument", pcb != NULL); |
| LWIP_ASSERT("recv_tcp must have an argument", arg != NULL); |
| LWIP_ASSERT("err != ERR_OK unhandled", err == ERR_OK); |
| LWIP_UNUSED_ARG(err); /* for LWIP_NOASSERT */ |
| conn = (struct netconn *)arg; |
| |
| if (conn == NULL) { |
| return ERR_VAL; |
| } |
| LWIP_ASSERT("recv_tcp: recv for wrong pcb!", conn->pcb.tcp == pcb); |
| |
| if (!NETCONN_MBOX_VALID(conn, &conn->recvmbox)) { |
| /* recvmbox already deleted */ |
| if (p != NULL) { |
| tcp_recved(pcb, p->tot_len); |
| pbuf_free(p); |
| } |
| return ERR_OK; |
| } |
| /* Unlike for UDP or RAW pcbs, don't check for available space |
| using recv_avail since that could break the connection |
| (data is already ACKed) */ |
| |
| if (p != NULL) { |
| msg = p; |
| len = p->tot_len; |
| } else { |
| msg = LWIP_CONST_CAST(void *, &netconn_closed); |
| len = 0; |
| } |
| |
| if (sys_mbox_trypost(&conn->recvmbox, msg) != ERR_OK) { |
| /* don't deallocate p: it is presented to us later again from tcp_fasttmr! */ |
| return ERR_MEM; |
| } else { |
| #if LWIP_SO_RCVBUF |
| SYS_ARCH_INC(conn->recv_avail, len); |
| #endif /* LWIP_SO_RCVBUF */ |
| /* Register event with callback */ |
| API_EVENT(conn, NETCONN_EVT_RCVPLUS, len); |
| } |
| |
| return ERR_OK; |
| } |
| |
| /** |
| * Poll callback function for TCP netconns. |
| * Wakes up an application thread that waits for a connection to close |
| * or data to be sent. The application thread then takes the |
| * appropriate action to go on. |
| * |
| * Signals the conn->sem. |
| * netconn_close waits for conn->sem if closing failed. |
| * |
| * @see tcp.h (struct tcp_pcb.poll) for parameters and return value |
| */ |
| static err_t |
| poll_tcp(void *arg, struct tcp_pcb *pcb) |
| { |
| struct netconn *conn = (struct netconn *)arg; |
| |
| LWIP_UNUSED_ARG(pcb); |
| LWIP_ASSERT("conn != NULL", (conn != NULL)); |
| |
| if (conn->state == NETCONN_WRITE) { |
| lwip_netconn_do_writemore(conn WRITE_DELAYED); |
| } else if (conn->state == NETCONN_CLOSE) { |
| #if !LWIP_SO_SNDTIMEO && !LWIP_SO_LINGER |
| if (conn->current_msg && conn->current_msg->msg.sd.polls_left) { |
| conn->current_msg->msg.sd.polls_left--; |
| } |
| #endif /* !LWIP_SO_SNDTIMEO && !LWIP_SO_LINGER */ |
| lwip_netconn_do_close_internal(conn WRITE_DELAYED); |
| } |
| /* @todo: implement connect timeout here? */ |
| |
| /* Did a nonblocking write fail before? Then check available write-space. */ |
| if (conn->flags & NETCONN_FLAG_CHECK_WRITESPACE) { |
| /* If the queued byte- or pbuf-count drops below the configured low-water limit, |
| let select mark this pcb as writable again. */ |
| if ((conn->pcb.tcp != NULL) && (tcp_sndbuf(conn->pcb.tcp) > TCP_SNDLOWAT) && |
| (tcp_sndqueuelen(conn->pcb.tcp) < TCP_SNDQUEUELOWAT)) { |
| netconn_clear_flags(conn, NETCONN_FLAG_CHECK_WRITESPACE); |
| API_EVENT(conn, NETCONN_EVT_SENDPLUS, 0); |
| } |
| } |
| |
| return ERR_OK; |
| } |
| |
| /** |
| * Sent callback function for TCP netconns. |
| * Signals the conn->sem and calls API_EVENT. |
| * netconn_write waits for conn->sem if send buffer is low. |
| * |
| * @see tcp.h (struct tcp_pcb.sent) for parameters and return value |
| */ |
| static err_t |
| sent_tcp(void *arg, struct tcp_pcb *pcb, u16_t len) |
| { |
| struct netconn *conn = (struct netconn *)arg; |
| |
| LWIP_UNUSED_ARG(pcb); |
| LWIP_ASSERT("conn != NULL", (conn != NULL)); |
| |
| if (conn) { |
| if (conn->state == NETCONN_WRITE) { |
| lwip_netconn_do_writemore(conn WRITE_DELAYED); |
| } else if (conn->state == NETCONN_CLOSE) { |
| lwip_netconn_do_close_internal(conn WRITE_DELAYED); |
| } |
| |
| /* If the queued byte- or pbuf-count drops below the configured low-water limit, |
| let select mark this pcb as writable again. */ |
| if ((conn->pcb.tcp != NULL) && (tcp_sndbuf(conn->pcb.tcp) > TCP_SNDLOWAT) && |
| (tcp_sndqueuelen(conn->pcb.tcp) < TCP_SNDQUEUELOWAT)) { |
| netconn_clear_flags(conn, NETCONN_FLAG_CHECK_WRITESPACE); |
| API_EVENT(conn, NETCONN_EVT_SENDPLUS, len); |
| } |
| } |
| |
| return ERR_OK; |
| } |
| |
| /** |
| * Error callback function for TCP netconns. |
| * Signals conn->sem, posts to all conn mboxes and calls API_EVENT. |
| * The application thread has then to decide what to do. |
| * |
| * @see tcp.h (struct tcp_pcb.err) for parameters |
| */ |
| static void |
| err_tcp(void *arg, err_t err) |
| { |
| struct netconn *conn; |
| enum netconn_state old_state; |
| void *mbox_msg; |
| SYS_ARCH_DECL_PROTECT(lev); |
| |
| conn = (struct netconn *)arg; |
| LWIP_ASSERT("conn != NULL", (conn != NULL)); |
| |
| SYS_ARCH_PROTECT(lev); |
| |
| /* when err is called, the pcb is deallocated, so delete the reference */ |
| conn->pcb.tcp = NULL; |
| /* store pending error */ |
| conn->pending_err = err; |
| /* prevent application threads from blocking on 'recvmbox'/'acceptmbox' */ |
| conn->flags |= NETCONN_FLAG_MBOXCLOSED; |
| |
| /* reset conn->state now before waking up other threads */ |
| old_state = conn->state; |
| conn->state = NETCONN_NONE; |
| |
| SYS_ARCH_UNPROTECT(lev); |
| |
| /* Notify the user layer about a connection error. Used to signal select. */ |
| API_EVENT(conn, NETCONN_EVT_ERROR, 0); |
| /* Try to release selects pending on 'read' or 'write', too. |
| They will get an error if they actually try to read or write. */ |
| API_EVENT(conn, NETCONN_EVT_RCVPLUS, 0); |
| API_EVENT(conn, NETCONN_EVT_SENDPLUS, 0); |
| |
| mbox_msg = lwip_netconn_err_to_msg(err); |
| /* pass error message to recvmbox to wake up pending recv */ |
| if (NETCONN_MBOX_VALID(conn, &conn->recvmbox)) { |
| /* use trypost to prevent deadlock */ |
| sys_mbox_trypost(&conn->recvmbox, mbox_msg); |
| } |
| /* pass error message to acceptmbox to wake up pending accept */ |
| if (NETCONN_MBOX_VALID(conn, &conn->acceptmbox)) { |
| /* use trypost to preven deadlock */ |
| sys_mbox_trypost(&conn->acceptmbox, mbox_msg); |
| } |
| |
| if ((old_state == NETCONN_WRITE) || (old_state == NETCONN_CLOSE) || |
| (old_state == NETCONN_CONNECT)) { |
| /* calling lwip_netconn_do_writemore/lwip_netconn_do_close_internal is not necessary |
| since the pcb has already been deleted! */ |
| int was_nonblocking_connect = IN_NONBLOCKING_CONNECT(conn); |
| SET_NONBLOCKING_CONNECT(conn, 0); |
| |
| if (!was_nonblocking_connect) { |
| sys_sem_t *op_completed_sem; |
| /* set error return code */ |
| LWIP_ASSERT("conn->current_msg != NULL", conn->current_msg != NULL); |
| if (old_state == NETCONN_CLOSE) { |
| /* let close succeed: the connection is closed after all... */ |
| conn->current_msg->err = ERR_OK; |
| } else { |
| /* Write and connect fail */ |
| conn->current_msg->err = err; |
| } |
| op_completed_sem = LWIP_API_MSG_SEM(conn->current_msg); |
| LWIP_ASSERT("inavlid op_completed_sem", sys_sem_valid(op_completed_sem)); |
| conn->current_msg = NULL; |
| /* wake up the waiting task */ |
| sys_sem_signal(op_completed_sem); |
| } else { |
| /* @todo: test what happens for error on nonblocking connect */ |
| } |
| } else { |
| LWIP_ASSERT("conn->current_msg == NULL", conn->current_msg == NULL); |
| } |
| } |
| |
| /** |
| * Setup a tcp_pcb with the correct callback function pointers |
| * and their arguments. |
| * |
| * @param conn the TCP netconn to setup |
| */ |
| static void |
| setup_tcp(struct netconn *conn) |
| { |
| struct tcp_pcb *pcb; |
| |
| pcb = conn->pcb.tcp; |
| tcp_arg(pcb, conn); |
| tcp_recv(pcb, recv_tcp); |
| tcp_sent(pcb, sent_tcp); |
| tcp_poll(pcb, poll_tcp, NETCONN_TCP_POLL_INTERVAL); |
| tcp_err(pcb, err_tcp); |
| } |
| |
| /** |
| * Accept callback function for TCP netconns. |
| * Allocates a new netconn and posts that to conn->acceptmbox. |
| * |
| * @see tcp.h (struct tcp_pcb_listen.accept) for parameters and return value |
| */ |
| static err_t |
| accept_function(void *arg, struct tcp_pcb *newpcb, err_t err) |
| { |
| struct netconn *newconn; |
| struct netconn *conn = (struct netconn *)arg; |
| |
| if (conn == NULL) { |
| return ERR_VAL; |
| } |
| if (!NETCONN_MBOX_VALID(conn, &conn->acceptmbox)) { |
| LWIP_DEBUGF(API_MSG_DEBUG, ("accept_function: acceptmbox already deleted\n")); |
| return ERR_VAL; |
| } |
| |
| if (newpcb == NULL) { |
| /* out-of-pcbs during connect: pass on this error to the application */ |
| if (sys_mbox_trypost(&conn->acceptmbox, lwip_netconn_err_to_msg(ERR_ABRT)) == ERR_OK) { |
| /* Register event with callback */ |
| API_EVENT(conn, NETCONN_EVT_RCVPLUS, 0); |
| } |
| return ERR_VAL; |
| } |
| LWIP_ASSERT("expect newpcb == NULL or err == ERR_OK", err == ERR_OK); |
| LWIP_UNUSED_ARG(err); /* for LWIP_NOASSERT */ |
| |
| LWIP_DEBUGF(API_MSG_DEBUG, ("accept_function: newpcb->state: %s\n", tcp_debug_state_str(newpcb->state))); |
| |
| /* We have to set the callback here even though |
| * the new socket is unknown. newconn->socket is marked as -1. */ |
| newconn = netconn_alloc(conn->type, conn->callback); |
| if (newconn == NULL) { |
| /* outof netconns: pass on this error to the application */ |
| if (sys_mbox_trypost(&conn->acceptmbox, lwip_netconn_err_to_msg(ERR_ABRT)) == ERR_OK) { |
| /* Register event with callback */ |
| API_EVENT(conn, NETCONN_EVT_RCVPLUS, 0); |
| } |
| return ERR_MEM; |
| } |
| newconn->pcb.tcp = newpcb; |
| setup_tcp(newconn); |
| |
| /* handle backlog counter */ |
| tcp_backlog_delayed(newpcb); |
| |
| if (sys_mbox_trypost(&conn->acceptmbox, newconn) != ERR_OK) { |
| /* When returning != ERR_OK, the pcb is aborted in tcp_process(), |
| so do nothing here! */ |
| /* remove all references to this netconn from the pcb */ |
| struct tcp_pcb *pcb = newconn->pcb.tcp; |
| tcp_arg(pcb, NULL); |
| tcp_recv(pcb, NULL); |
| tcp_sent(pcb, NULL); |
| tcp_poll(pcb, NULL, 0); |
| tcp_err(pcb, NULL); |
| /* remove reference from to the pcb from this netconn */ |
| newconn->pcb.tcp = NULL; |
| /* no need to drain since we know the recvmbox is empty. */ |
| sys_mbox_free(&newconn->recvmbox); |
| sys_mbox_set_invalid(&newconn->recvmbox); |
| netconn_free(newconn); |
| return ERR_MEM; |
| } else { |
| /* Register event with callback */ |
| API_EVENT(conn, NETCONN_EVT_RCVPLUS, 0); |
| } |
| |
| return ERR_OK; |
| } |
| #endif /* LWIP_TCP */ |
| |
| /** |
| * Create a new pcb of a specific type. |
| * Called from lwip_netconn_do_newconn(). |
| * |
| * @param msg the api_msg describing the connection type |
| */ |
| static void |
| pcb_new(struct api_msg *msg) |
| { |
| enum lwip_ip_addr_type iptype = IPADDR_TYPE_V4; |
| |
| LWIP_ASSERT("pcb_new: pcb already allocated", msg->conn->pcb.tcp == NULL); |
| |
| #if LWIP_IPV6 && LWIP_IPV4 |
| /* IPv6: Dual-stack by default, unless netconn_set_ipv6only() is called */ |
| if (NETCONNTYPE_ISIPV6(netconn_type(msg->conn))) { |
| iptype = IPADDR_TYPE_ANY; |
| } |
| #endif |
| |
| /* Allocate a PCB for this connection */ |
| switch (NETCONNTYPE_GROUP(msg->conn->type)) { |
| #if LWIP_RAW |
| case NETCONN_RAW: |
| msg->conn->pcb.raw = raw_new_ip_type(iptype, msg->msg.n.proto); |
| if (msg->conn->pcb.raw != NULL) { |
| #if LWIP_IPV6 |
| /* ICMPv6 packets should always have checksum calculated by the stack as per RFC 3542 chapter 3.1 */ |
| if (NETCONNTYPE_ISIPV6(msg->conn->type) && msg->conn->pcb.raw->protocol == IP6_NEXTH_ICMP6) { |
| msg->conn->pcb.raw->chksum_reqd = 1; |
| msg->conn->pcb.raw->chksum_offset = 2; |
| } |
| #endif /* LWIP_IPV6 */ |
| raw_recv(msg->conn->pcb.raw, recv_raw, msg->conn); |
| } |
| break; |
| #endif /* LWIP_RAW */ |
| #if LWIP_UDP |
| case NETCONN_UDP: |
| msg->conn->pcb.udp = udp_new_ip_type(iptype); |
| if (msg->conn->pcb.udp != NULL) { |
| #if LWIP_UDPLITE |
| if (NETCONNTYPE_ISUDPLITE(msg->conn->type)) { |
| udp_setflags(msg->conn->pcb.udp, UDP_FLAGS_UDPLITE); |
| } |
| #endif /* LWIP_UDPLITE */ |
| if (NETCONNTYPE_ISUDPNOCHKSUM(msg->conn->type)) { |
| udp_setflags(msg->conn->pcb.udp, UDP_FLAGS_NOCHKSUM); |
| } |
| udp_recv(msg->conn->pcb.udp, recv_udp, msg->conn); |
| } |
| break; |
| #endif /* LWIP_UDP */ |
| #if LWIP_TCP |
| case NETCONN_TCP: |
| msg->conn->pcb.tcp = tcp_new_ip_type(iptype); |
| if (msg->conn->pcb.tcp != NULL) { |
| setup_tcp(msg->conn); |
| } |
| break; |
| #endif /* LWIP_TCP */ |
| default: |
| /* Unsupported netconn type, e.g. protocol disabled */ |
| msg->err = ERR_VAL; |
| return; |
| } |
| if (msg->conn->pcb.ip == NULL) { |
| msg->err = ERR_MEM; |
| } |
| } |
| |
| /** |
| * Create a new pcb of a specific type inside a netconn. |
| * Called from netconn_new_with_proto_and_callback. |
| * |
| * @param m the api_msg describing the connection type |
| */ |
| void |
| lwip_netconn_do_newconn(void *m) |
| { |
| struct api_msg *msg = (struct api_msg *)m; |
| |
| msg->err = ERR_OK; |
| if (msg->conn->pcb.tcp == NULL) { |
| pcb_new(msg); |
| } |
| /* Else? This "new" connection already has a PCB allocated. */ |
| /* Is this an error condition? Should it be deleted? */ |
| /* We currently just are happy and return. */ |
| |
| TCPIP_APIMSG_ACK(msg); |
| } |
| |
| /** |
| * Create a new netconn (of a specific type) that has a callback function. |
| * The corresponding pcb is NOT created! |
| * |
| * @param t the type of 'connection' to create (@see enum netconn_type) |
| * @param callback a function to call on status changes (RX available, TX'ed) |
| * @return a newly allocated struct netconn or |
| * NULL on memory error |
| */ |
| struct netconn * |
| netconn_alloc(enum netconn_type t, netconn_callback callback) |
| { |
| struct netconn *conn; |
| int size; |
| u8_t init_flags = 0; |
| |
| conn = (struct netconn *)memp_malloc(MEMP_NETCONN); |
| if (conn == NULL) { |
| return NULL; |
| } |
| |
| conn->pending_err = ERR_OK; |
| conn->type = t; |
| conn->pcb.tcp = NULL; |
| #if LWIP_NETCONN_FULLDUPLEX |
| conn->mbox_threads_waiting = 0; |
| #endif |
| |
| /* If all sizes are the same, every compiler should optimize this switch to nothing */ |
| switch (NETCONNTYPE_GROUP(t)) { |
| #if LWIP_RAW |
| case NETCONN_RAW: |
| size = DEFAULT_RAW_RECVMBOX_SIZE; |
| break; |
| #endif /* LWIP_RAW */ |
| #if LWIP_UDP |
| case NETCONN_UDP: |
| size = DEFAULT_UDP_RECVMBOX_SIZE; |
| #if LWIP_NETBUF_RECVINFO |
| init_flags |= NETCONN_FLAG_PKTINFO; |
| #endif /* LWIP_NETBUF_RECVINFO */ |
| break; |
| #endif /* LWIP_UDP */ |
| #if LWIP_TCP |
| case NETCONN_TCP: |
| size = DEFAULT_TCP_RECVMBOX_SIZE; |
| break; |
| #endif /* LWIP_TCP */ |
| default: |
| LWIP_ASSERT("netconn_alloc: undefined netconn_type", 0); |
| goto free_and_return; |
| } |
| |
| if (sys_mbox_new(&conn->recvmbox, size) != ERR_OK) { |
| goto free_and_return; |
| } |
| #if !LWIP_NETCONN_SEM_PER_THREAD |
| if (sys_sem_new(&conn->op_completed, 0) != ERR_OK) { |
| sys_mbox_free(&conn->recvmbox); |
| goto free_and_return; |
| } |
| #endif |
| |
| #if LWIP_TCP |
| sys_mbox_set_invalid(&conn->acceptmbox); |
| #endif |
| conn->state = NETCONN_NONE; |
| #if LWIP_SOCKET |
| /* initialize socket to -1 since 0 is a valid socket */ |
| conn->socket = -1; |
| #endif /* LWIP_SOCKET */ |
| conn->callback = callback; |
| #if LWIP_TCP |
| conn->current_msg = NULL; |
| #endif /* LWIP_TCP */ |
| #if LWIP_SO_SNDTIMEO |
| conn->send_timeout = 0; |
| #endif /* LWIP_SO_SNDTIMEO */ |
| #if LWIP_SO_RCVTIMEO |
| conn->recv_timeout = 0; |
| #endif /* LWIP_SO_RCVTIMEO */ |
| #if LWIP_SO_RCVBUF |
| conn->recv_bufsize = RECV_BUFSIZE_DEFAULT; |
| conn->recv_avail = 0; |
| #endif /* LWIP_SO_RCVBUF */ |
| #if LWIP_SO_LINGER |
| conn->linger = -1; |
| #endif /* LWIP_SO_LINGER */ |
| conn->flags = init_flags; |
| return conn; |
| free_and_return: |
| memp_free(MEMP_NETCONN, conn); |
| return NULL; |
| } |
| |
| /** |
| * Delete a netconn and all its resources. |
| * The pcb is NOT freed (since we might not be in the right thread context do this). |
| * |
| * @param conn the netconn to free |
| */ |
| void |
| netconn_free(struct netconn *conn) |
| { |
| LWIP_ASSERT("PCB must be deallocated outside this function", conn->pcb.tcp == NULL); |
| |
| #if LWIP_NETCONN_FULLDUPLEX |
| /* in fullduplex, netconn is drained here */ |
| netconn_drain(conn); |
| #endif /* LWIP_NETCONN_FULLDUPLEX */ |
| |
| LWIP_ASSERT("recvmbox must be deallocated before calling this function", |
| !sys_mbox_valid(&conn->recvmbox)); |
| #if LWIP_TCP |
| LWIP_ASSERT("acceptmbox must be deallocated before calling this function", |
| !sys_mbox_valid(&conn->acceptmbox)); |
| #endif /* LWIP_TCP */ |
| |
| #if !LWIP_NETCONN_SEM_PER_THREAD |
| sys_sem_free(&conn->op_completed); |
| sys_sem_set_invalid(&conn->op_completed); |
| #endif |
| |
| memp_free(MEMP_NETCONN, conn); |
| } |
| |
| /** |
| * Delete rcvmbox and acceptmbox of a netconn and free the left-over data in |
| * these mboxes |
| * |
| * @param conn the netconn to free |
| * @bytes_drained bytes drained from recvmbox |
| * @accepts_drained pending connections drained from acceptmbox |
| */ |
| static void |
| netconn_drain(struct netconn *conn) |
| { |
| void *mem; |
| |
| /* This runs when mbox and netconn are marked as closed, |
| so we don't need to lock against rx packets */ |
| #if LWIP_NETCONN_FULLDUPLEX |
| LWIP_ASSERT("netconn marked closed", conn->flags & NETCONN_FLAG_MBOXINVALID); |
| #endif /* LWIP_NETCONN_FULLDUPLEX */ |
| |
| /* Delete and drain the recvmbox. */ |
| if (sys_mbox_valid(&conn->recvmbox)) { |
| while (sys_mbox_tryfetch(&conn->recvmbox, &mem) != SYS_MBOX_EMPTY) { |
| #if LWIP_NETCONN_FULLDUPLEX |
| if (!lwip_netconn_is_deallocated_msg(mem)) |
| #endif /* LWIP_NETCONN_FULLDUPLEX */ |
| { |
| #if LWIP_TCP |
| if (NETCONNTYPE_GROUP(conn->type) == NETCONN_TCP) { |
| err_t err; |
| if (!lwip_netconn_is_err_msg(mem, &err)) { |
| pbuf_free((struct pbuf *)mem); |
| } |
| } else |
| #endif /* LWIP_TCP */ |
| { |
| netbuf_delete((struct netbuf *)mem); |
| } |
| } |
| } |
| sys_mbox_free(&conn->recvmbox); |
| sys_mbox_set_invalid(&conn->recvmbox); |
| } |
| |
| /* Delete and drain the acceptmbox. */ |
| #if LWIP_TCP |
| if (sys_mbox_valid(&conn->acceptmbox)) { |
| while (sys_mbox_tryfetch(&conn->acceptmbox, &mem) != SYS_MBOX_EMPTY) { |
| #if LWIP_NETCONN_FULLDUPLEX |
| if (!lwip_netconn_is_deallocated_msg(mem)) |
| #endif /* LWIP_NETCONN_FULLDUPLEX */ |
| { |
| err_t err; |
| if (!lwip_netconn_is_err_msg(mem, &err)) { |
| struct netconn *newconn = (struct netconn *)mem; |
| /* Only tcp pcbs have an acceptmbox, so no need to check conn->type */ |
| /* pcb might be set to NULL already by err_tcp() */ |
| /* drain recvmbox */ |
| netconn_drain(newconn); |
| if (newconn->pcb.tcp != NULL) { |
| tcp_abort(newconn->pcb.tcp); |
| newconn->pcb.tcp = NULL; |
| } |
| netconn_free(newconn); |
| } |
| } |
| } |
| sys_mbox_free(&conn->acceptmbox); |
| sys_mbox_set_invalid(&conn->acceptmbox); |
| } |
| #endif /* LWIP_TCP */ |
| } |
| |
| #if LWIP_NETCONN_FULLDUPLEX |
| static void |
| netconn_mark_mbox_invalid(struct netconn *conn) |
| { |
| int i, num_waiting; |
| void *msg = LWIP_CONST_CAST(void *, &netconn_deleted); |
| |
| /* Prevent new calls/threads from reading from the mbox */ |
| conn->flags |= NETCONN_FLAG_MBOXINVALID; |
| |
| SYS_ARCH_LOCKED(num_waiting = conn->mbox_threads_waiting); |
| for (i = 0; i < num_waiting; i++) { |
| if (sys_mbox_valid_val(conn->recvmbox)) { |
| sys_mbox_trypost(&conn->recvmbox, msg); |
| } else { |
| sys_mbox_trypost(&conn->acceptmbox, msg); |
| } |
| } |
| } |
| #endif /* LWIP_NETCONN_FULLDUPLEX */ |
| |
| #if LWIP_TCP |
| /** |
| * Internal helper function to close a TCP netconn: since this sometimes |
| * doesn't work at the first attempt, this function is called from multiple |
| * places. |
| * |
| * @param conn the TCP netconn to close |
| */ |
| static err_t |
| lwip_netconn_do_close_internal(struct netconn *conn WRITE_DELAYED_PARAM) |
| { |
| err_t err; |
| u8_t shut, shut_rx, shut_tx, shut_close; |
| u8_t close_finished = 0; |
| struct tcp_pcb *tpcb; |
| #if LWIP_SO_LINGER |
| u8_t linger_wait_required = 0; |
| #endif /* LWIP_SO_LINGER */ |
| |
| LWIP_ASSERT("invalid conn", (conn != NULL)); |
| LWIP_ASSERT("this is for tcp netconns only", (NETCONNTYPE_GROUP(conn->type) == NETCONN_TCP)); |
| LWIP_ASSERT("conn must be in state NETCONN_CLOSE", (conn->state == NETCONN_CLOSE)); |
| LWIP_ASSERT("pcb already closed", (conn->pcb.tcp != NULL)); |
| LWIP_ASSERT("conn->current_msg != NULL", conn->current_msg != NULL); |
| |
| tpcb = conn->pcb.tcp; |
| shut = conn->current_msg->msg.sd.shut; |
| shut_rx = shut & NETCONN_SHUT_RD; |
| shut_tx = shut & NETCONN_SHUT_WR; |
| /* shutting down both ends is the same as closing |
| (also if RD or WR side was shut down before already) */ |
| if (shut == NETCONN_SHUT_RDWR) { |
| shut_close = 1; |
| } else if (shut_rx && |
| ((tpcb->state == FIN_WAIT_1) || |
| (tpcb->state == FIN_WAIT_2) || |
| (tpcb->state == CLOSING))) { |
| shut_close = 1; |
| } else if (shut_tx && ((tpcb->flags & TF_RXCLOSED) != 0)) { |
| shut_close = 1; |
| } else { |
| shut_close = 0; |
| } |
| |
| /* Set back some callback pointers */ |
| if (shut_close) { |
| tcp_arg(tpcb, NULL); |
| } |
| if (tpcb->state == LISTEN) { |
| tcp_accept(tpcb, NULL); |
| } else { |
| /* some callbacks have to be reset if tcp_close is not successful */ |
| if (shut_rx) { |
| tcp_recv(tpcb, NULL); |
| tcp_accept(tpcb, NULL); |
| } |
| if (shut_tx) { |
| tcp_sent(tpcb, NULL); |
| } |
| if (shut_close) { |
| tcp_poll(tpcb, NULL, 0); |
| tcp_err(tpcb, NULL); |
| } |
| } |
| /* Try to close the connection */ |
| if (shut_close) { |
| #if LWIP_SO_LINGER |
| /* check linger possibilites before calling tcp_close */ |
| err = ERR_OK; |
| /* linger enabled/required at all? (i.e. is there untransmitted data left?) */ |
| if ((conn->linger >= 0) && (conn->pcb.tcp->unsent || conn->pcb.tcp->unacked)) { |
| if ((conn->linger == 0)) { |
| /* data left but linger prevents waiting */ |
| tcp_abort(tpcb); |
| tpcb = NULL; |
| } else if (conn->linger > 0) { |
| /* data left and linger says we should wait */ |
| if (netconn_is_nonblocking(conn)) { |
| /* data left on a nonblocking netconn -> cannot linger */ |
| err = ERR_WOULDBLOCK; |
| } else if ((s32_t)(sys_now() - conn->current_msg->msg.sd.time_started) >= |
| (conn->linger * 1000)) { |
| /* data left but linger timeout has expired (this happens on further |
| calls to this function through poll_tcp */ |
| tcp_abort(tpcb); |
| tpcb = NULL; |
| } else { |
| /* data left -> need to wait for ACK after successful close */ |
| linger_wait_required = 1; |
| } |
| } |
| } |
| if ((err == ERR_OK) && (tpcb != NULL)) |
| #endif /* LWIP_SO_LINGER */ |
| { |
| err = tcp_close(tpcb); |
| } |
| } else { |
| err = tcp_shutdown(tpcb, shut_rx, shut_tx); |
| } |
| if (err == ERR_OK) { |
| close_finished = 1; |
| #if LWIP_SO_LINGER |
| if (linger_wait_required) { |
| /* wait for ACK of all unsent/unacked data by just getting called again */ |
| close_finished = 0; |
| err = ERR_INPROGRESS; |
| } |
| #endif /* LWIP_SO_LINGER */ |
| } else { |
| if (err == ERR_MEM) { |
| /* Closing failed because of memory shortage, try again later. Even for |
| nonblocking netconns, we have to wait since no standard socket application |
| is prepared for close failing because of resource shortage. |
| Check the timeout: this is kind of an lwip addition to the standard sockets: |
| we wait for some time when failing to allocate a segment for the FIN */ |
| #if LWIP_SO_SNDTIMEO || LWIP_SO_LINGER |
| s32_t close_timeout = LWIP_TCP_CLOSE_TIMEOUT_MS_DEFAULT; |
| #if LWIP_SO_SNDTIMEO |
| if (conn->send_timeout > 0) { |
| close_timeout = conn->send_timeout; |
| } |
| #endif /* LWIP_SO_SNDTIMEO */ |
| #if LWIP_SO_LINGER |
| if (conn->linger >= 0) { |
| /* use linger timeout (seconds) */ |
| close_timeout = conn->linger * 1000U; |
| } |
| #endif |
| if ((s32_t)(sys_now() - conn->current_msg->msg.sd.time_started) >= close_timeout) { |
| #else /* LWIP_SO_SNDTIMEO || LWIP_SO_LINGER */ |
| if (conn->current_msg->msg.sd.polls_left == 0) { |
| #endif /* LWIP_SO_SNDTIMEO || LWIP_SO_LINGER */ |
| close_finished = 1; |
| if (shut_close) { |
| /* in this case, we want to RST the connection */ |
| tcp_abort(tpcb); |
| err = ERR_OK; |
| } |
| } |
| } else { |
| /* Closing failed for a non-memory error: give up */ |
| close_finished = 1; |
| } |
| } |
| if (close_finished) { |
| /* Closing done (succeeded, non-memory error, nonblocking error or timeout) */ |
| sys_sem_t *op_completed_sem = LWIP_API_MSG_SEM(conn->current_msg); |
| conn->current_msg->err = err; |
| conn->current_msg = NULL; |
| conn->state = NETCONN_NONE; |
| if (err == ERR_OK) { |
| if (shut_close) { |
| /* Set back some callback pointers as conn is going away */ |
| conn->pcb.tcp = NULL; |
| /* Trigger select() in socket layer. Make sure everybody notices activity |
| on the connection, error first! */ |
| API_EVENT(conn, NETCONN_EVT_ERROR, 0); |
| } |
| if (shut_rx) { |
| API_EVENT(conn, NETCONN_EVT_RCVPLUS, 0); |
| } |
| if (shut_tx) { |
| API_EVENT(conn, NETCONN_EVT_SENDPLUS, 0); |
| } |
| } |
| #if LWIP_TCPIP_CORE_LOCKING |
| if (delayed) |
| #endif |
| { |
| /* wake up the application task */ |
| sys_sem_signal(op_completed_sem); |
| } |
| return ERR_OK; |
| } |
| if (!close_finished) { |
| /* Closing failed and we want to wait: restore some of the callbacks */ |
| /* Closing of listen pcb will never fail! */ |
| LWIP_ASSERT("Closing a listen pcb may not fail!", (tpcb->state != LISTEN)); |
| if (shut_tx) { |
| tcp_sent(tpcb, sent_tcp); |
| } |
| /* when waiting for close, set up poll interval to 500ms */ |
| tcp_poll(tpcb, poll_tcp, 1); |
| tcp_err(tpcb, err_tcp); |
| tcp_arg(tpcb, conn); |
| /* don't restore recv callback: we don't want to receive any more data */ |
| } |
| /* If closing didn't succeed, we get called again either |
| from poll_tcp or from sent_tcp */ |
| LWIP_ASSERT("err != ERR_OK", err != ERR_OK); |
| return err; |
| } |
| #endif /* LWIP_TCP */ |
| |
| /** |
| * Delete the pcb inside a netconn. |
| * Called from netconn_delete. |
| * |
| * @param m the api_msg pointing to the connection |
| */ |
| void |
| lwip_netconn_do_delconn(void *m) |
| { |
| struct api_msg *msg = (struct api_msg *)m; |
| |
| enum netconn_state state = msg->conn->state; |
| LWIP_ASSERT("netconn state error", /* this only happens for TCP netconns */ |
| (state == NETCONN_NONE) || (NETCONNTYPE_GROUP(msg->conn->type) == NETCONN_TCP)); |
| #if LWIP_NETCONN_FULLDUPLEX |
| /* In full duplex mode, blocking write/connect is aborted with ERR_CLSD */ |
| if (state != NETCONN_NONE) { |
| if ((state == NETCONN_WRITE) || |
| ((state == NETCONN_CONNECT) && !IN_NONBLOCKING_CONNECT(msg->conn))) { |
| /* close requested, abort running write/connect */ |
| sys_sem_t *op_completed_sem; |
| LWIP_ASSERT("msg->conn->current_msg != NULL", msg->conn->current_msg != NULL); |
| op_completed_sem = LWIP_API_MSG_SEM(msg->conn->current_msg); |
| msg->conn->current_msg->err = ERR_CLSD; |
| msg->conn->current_msg = NULL; |
| msg->conn->state = NETCONN_NONE; |
| sys_sem_signal(op_completed_sem); |
| } |
| } |
| #else /* LWIP_NETCONN_FULLDUPLEX */ |
| if (((state != NETCONN_NONE) && |
| (state != NETCONN_LISTEN) && |
| (state != NETCONN_CONNECT)) || |
| ((state == NETCONN_CONNECT) && !IN_NONBLOCKING_CONNECT(msg->conn))) { |
| /* This means either a blocking write or blocking connect is running |
| (nonblocking write returns and sets state to NONE) */ |
| msg->err = ERR_INPROGRESS; |
| } else |
| #endif /* LWIP_NETCONN_FULLDUPLEX */ |
| { |
| LWIP_ASSERT("blocking connect in progress", |
| (state != NETCONN_CONNECT) || IN_NONBLOCKING_CONNECT(msg->conn)); |
| msg->err = ERR_OK; |
| #if LWIP_NETCONN_FULLDUPLEX |
| /* Mark mboxes invalid */ |
| netconn_mark_mbox_invalid(msg->conn); |
| #else /* LWIP_NETCONN_FULLDUPLEX */ |
| netconn_drain(msg->conn); |
| #endif /* LWIP_NETCONN_FULLDUPLEX */ |
| |
| if (msg->conn->pcb.tcp != NULL) { |
| |
| switch (NETCONNTYPE_GROUP(msg->conn->type)) { |
| #if LWIP_RAW |
| case NETCONN_RAW: |
| raw_remove(msg->conn->pcb.raw); |
| break; |
| #endif /* LWIP_RAW */ |
| #if LWIP_UDP |
| case NETCONN_UDP: |
| msg->conn->pcb.udp->recv_arg = NULL; |
| udp_remove(msg->conn->pcb.udp); |
| break; |
| #endif /* LWIP_UDP */ |
| #if LWIP_TCP |
| case NETCONN_TCP: |
| LWIP_ASSERT("already writing or closing", msg->conn->current_msg == NULL); |
| msg->conn->state = NETCONN_CLOSE; |
| msg->msg.sd.shut = NETCONN_SHUT_RDWR; |
| msg->conn->current_msg = msg; |
| #if LWIP_TCPIP_CORE_LOCKING |
| if (lwip_netconn_do_close_internal(msg->conn, 0) != ERR_OK) { |
| LWIP_ASSERT("state!", msg->conn->state == NETCONN_CLOSE); |
| UNLOCK_TCPIP_CORE(); |
| sys_arch_sem_wait(LWIP_API_MSG_SEM(msg), 0); |
| LOCK_TCPIP_CORE(); |
| LWIP_ASSERT("state!", msg->conn->state == NETCONN_NONE); |
| } |
| #else /* LWIP_TCPIP_CORE_LOCKING */ |
| lwip_netconn_do_close_internal(msg->conn); |
| #endif /* LWIP_TCPIP_CORE_LOCKING */ |
| /* API_EVENT is called inside lwip_netconn_do_close_internal, before releasing |
| the application thread, so we can return at this point! */ |
| return; |
| #endif /* LWIP_TCP */ |
| default: |
| break; |
| } |
| msg->conn->pcb.tcp = NULL; |
| } |
| /* tcp netconns don't come here! */ |
| |
| /* @todo: this lets select make the socket readable and writable, |
| which is wrong! errfd instead? */ |
| API_EVENT(msg->conn, NETCONN_EVT_RCVPLUS, 0); |
| API_EVENT(msg->conn, NETCONN_EVT_SENDPLUS, 0); |
| } |
| if (sys_sem_valid(LWIP_API_MSG_SEM(msg))) { |
| TCPIP_APIMSG_ACK(msg); |
| } |
| } |
| |
| /** |
| * Bind a pcb contained in a netconn |
| * Called from netconn_bind. |
| * |
| * @param m the api_msg pointing to the connection and containing |
| * the IP address and port to bind to |
| */ |
| void |
| lwip_netconn_do_bind(void *m) |
| { |
| struct api_msg *msg = (struct api_msg *)m; |
| err_t err; |
| |
| if (msg->conn->pcb.tcp != NULL) { |
| switch (NETCONNTYPE_GROUP(msg->conn->type)) { |
| #if LWIP_RAW |
| case NETCONN_RAW: |
| err = raw_bind(msg->conn->pcb.raw, API_EXPR_REF(msg->msg.bc.ipaddr)); |
| break; |
| #endif /* LWIP_RAW */ |
| #if LWIP_UDP |
| case NETCONN_UDP: |
| err = udp_bind(msg->conn->pcb.udp, API_EXPR_REF(msg->msg.bc.ipaddr), msg->msg.bc.port); |
| break; |
| #endif /* LWIP_UDP */ |
| #if LWIP_TCP |
| case NETCONN_TCP: |
| err = tcp_bind(msg->conn->pcb.tcp, API_EXPR_REF(msg->msg.bc.ipaddr), msg->msg.bc.port); |
| break; |
| #endif /* LWIP_TCP */ |
| default: |
| err = ERR_VAL; |
| break; |
| } |
| } else { |
| err = ERR_VAL; |
| } |
| msg->err = err; |
| TCPIP_APIMSG_ACK(msg); |
| } |
| /** |
| * Bind a pcb contained in a netconn to an interface |
| * Called from netconn_bind_if. |
| * |
| * @param m the api_msg pointing to the connection and containing |
| * the IP address and port to bind to |
| */ |
| void |
| lwip_netconn_do_bind_if(void *m) |
| { |
| struct netif *netif; |
| struct api_msg *msg = (struct api_msg *)m; |
| err_t err; |
| |
| netif = netif_get_by_index(msg->msg.bc.if_idx); |
| |
| if ((netif != NULL) && (msg->conn->pcb.tcp != NULL)) { |
| err = ERR_OK; |
| switch (NETCONNTYPE_GROUP(msg->conn->type)) { |
| #if LWIP_RAW |
| case NETCONN_RAW: |
| raw_bind_netif(msg->conn->pcb.raw, netif); |
| break; |
| #endif /* LWIP_RAW */ |
| #if LWIP_UDP |
| case NETCONN_UDP: |
| udp_bind_netif(msg->conn->pcb.udp, netif); |
| break; |
| #endif /* LWIP_UDP */ |
| #if LWIP_TCP |
| case NETCONN_TCP: |
| tcp_bind_netif(msg->conn->pcb.tcp, netif); |
| break; |
| #endif /* LWIP_TCP */ |
| default: |
| err = ERR_VAL; |
| break; |
| } |
| } else { |
| err = ERR_VAL; |
| } |
| msg->err = err; |
| TCPIP_APIMSG_ACK(msg); |
| } |
| |
| #if LWIP_TCP |
| /** |
| * TCP callback function if a connection (opened by tcp_connect/lwip_netconn_do_connect) has |
| * been established (or reset by the remote host). |
| * |
| * @see tcp.h (struct tcp_pcb.connected) for parameters and return values |
| */ |
| static err_t |
| lwip_netconn_do_connected(void *arg, struct tcp_pcb *pcb, err_t err) |
| { |
| struct netconn *conn; |
| int was_blocking; |
| sys_sem_t *op_completed_sem = NULL; |
| |
| LWIP_UNUSED_ARG(pcb); |
| |
| conn = (struct netconn *)arg; |
| |
| if (conn == NULL) { |
| return ERR_VAL; |
| } |
| |
| LWIP_ASSERT("conn->state == NETCONN_CONNECT", conn->state == NETCONN_CONNECT); |
| LWIP_ASSERT("(conn->current_msg != NULL) || conn->in_non_blocking_connect", |
| (conn->current_msg != NULL) || IN_NONBLOCKING_CONNECT(conn)); |
| |
| if (conn->current_msg != NULL) { |
| conn->current_msg->err = err; |
| op_completed_sem = LWIP_API_MSG_SEM(conn->current_msg); |
| } |
| if ((NETCONNTYPE_GROUP(conn->type) == NETCONN_TCP) && (err == ERR_OK)) { |
| setup_tcp(conn); |
| } |
| was_blocking = !IN_NONBLOCKING_CONNECT(conn); |
| SET_NONBLOCKING_CONNECT(conn, 0); |
| LWIP_ASSERT("blocking connect state error", |
| (was_blocking && op_completed_sem != NULL) || |
| (!was_blocking && op_completed_sem == NULL)); |
| conn->current_msg = NULL; |
| conn->state = NETCONN_NONE; |
| API_EVENT(conn, NETCONN_EVT_SENDPLUS, 0); |
| |
| if (was_blocking) { |
| sys_sem_signal(op_completed_sem); |
| } |
| return ERR_OK; |
| } |
| #endif /* LWIP_TCP */ |
| |
| /** |
| * Connect a pcb contained inside a netconn |
| * Called from netconn_connect. |
| * |
| * @param m the api_msg pointing to the connection and containing |
| * the IP address and port to connect to |
| */ |
| void |
| lwip_netconn_do_connect(void *m) |
| { |
| struct api_msg *msg = (struct api_msg *)m; |
| err_t err; |
| |
| if (msg->conn->pcb.tcp == NULL) { |
| /* This may happen when calling netconn_connect() a second time */ |
| err = ERR_CLSD; |
| } else { |
| switch (NETCONNTYPE_GROUP(msg->conn->type)) { |
| #if LWIP_RAW |
| case NETCONN_RAW: |
| err = raw_connect(msg->conn->pcb.raw, API_EXPR_REF(msg->msg.bc.ipaddr)); |
| break; |
| #endif /* LWIP_RAW */ |
| #if LWIP_UDP |
| case NETCONN_UDP: |
| err = udp_connect(msg->conn->pcb.udp, API_EXPR_REF(msg->msg.bc.ipaddr), msg->msg.bc.port); |
| break; |
| #endif /* LWIP_UDP */ |
| #if LWIP_TCP |
| case NETCONN_TCP: |
| /* Prevent connect while doing any other action. */ |
| if (msg->conn->state == NETCONN_CONNECT) { |
| err = ERR_ALREADY; |
| } else if (msg->conn->state != NETCONN_NONE) { |
| err = ERR_ISCONN; |
| } else { |
| setup_tcp(msg->conn); |
| err = tcp_connect(msg->conn->pcb.tcp, API_EXPR_REF(msg->msg.bc.ipaddr), |
| msg->msg.bc.port, lwip_netconn_do_connected); |
| if (err == ERR_OK) { |
| u8_t non_blocking = netconn_is_nonblocking(msg->conn); |
| msg->conn->state = NETCONN_CONNECT; |
| SET_NONBLOCKING_CONNECT(msg->conn, non_blocking); |
| if (non_blocking) { |
| err = ERR_INPROGRESS; |
| } else { |
| msg->conn->current_msg = msg; |
| /* sys_sem_signal() is called from lwip_netconn_do_connected (or err_tcp()), |
| when the connection is established! */ |
| #if LWIP_TCPIP_CORE_LOCKING |
| LWIP_ASSERT("state!", msg->conn->state == NETCONN_CONNECT); |
| UNLOCK_TCPIP_CORE(); |
| sys_arch_sem_wait(LWIP_API_MSG_SEM(msg), 0); |
| LOCK_TCPIP_CORE(); |
| LWIP_ASSERT("state!", msg->conn->state != NETCONN_CONNECT); |
| #endif /* LWIP_TCPIP_CORE_LOCKING */ |
| return; |
| } |
| } |
| } |
| break; |
| #endif /* LWIP_TCP */ |
| default: |
| LWIP_ERROR("Invalid netconn type", 0, do { |
| err = ERR_VAL; |
| } while (0)); |
| break; |
| } |
| } |
| msg->err = err; |
| /* For all other protocols, netconn_connect() calls netconn_apimsg(), |
| so use TCPIP_APIMSG_ACK() here. */ |
| TCPIP_APIMSG_ACK(msg); |
| } |
| |
| /** |
| * Disconnect a pcb contained inside a netconn |
| * Only used for UDP netconns. |
| * Called from netconn_disconnect. |
| * |
| * @param m the api_msg pointing to the connection to disconnect |
| */ |
| void |
| lwip_netconn_do_disconnect(void *m) |
| { |
| struct api_msg *msg = (struct api_msg *)m; |
| |
| #if LWIP_UDP |
| if (NETCONNTYPE_GROUP(msg->conn->type) == NETCONN_UDP) { |
| udp_disconnect(msg->conn->pcb.udp); |
| msg->err = ERR_OK; |
| } else |
| #endif /* LWIP_UDP */ |
| { |
| msg->err = ERR_VAL; |
| } |
| TCPIP_APIMSG_ACK(msg); |
| } |
| |
| #if LWIP_TCP |
| /** |
| * Set a TCP pcb contained in a netconn into listen mode |
| * Called from netconn_listen. |
| * |
| * @param m the api_msg pointing to the connection |
| */ |
| void |
| lwip_netconn_do_listen(void *m) |
| { |
| struct api_msg *msg = (struct api_msg *)m; |
| err_t err; |
| |
| if (msg->conn->pcb.tcp != NULL) { |
| if (NETCONNTYPE_GROUP(msg->conn->type) == NETCONN_TCP) { |
| if (msg->conn->state == NETCONN_NONE) { |
| struct tcp_pcb *lpcb; |
| if (msg->conn->pcb.tcp->state != CLOSED) { |
| /* connection is not closed, cannot listen */ |
| err = ERR_VAL; |
| } else { |
| u8_t backlog; |
| #if TCP_LISTEN_BACKLOG |
| backlog = msg->msg.lb.backlog; |
| #else /* TCP_LISTEN_BACKLOG */ |
| backlog = TCP_DEFAULT_LISTEN_BACKLOG; |
| #endif /* TCP_LISTEN_BACKLOG */ |
| #if LWIP_IPV4 && LWIP_IPV6 |
| /* "Socket API like" dual-stack support: If IP to listen to is IP6_ADDR_ANY, |
| * and NETCONN_FLAG_IPV6_V6ONLY is NOT set, use IP_ANY_TYPE to listen |
| */ |
| if (ip_addr_cmp(&msg->conn->pcb.ip->local_ip, IP6_ADDR_ANY) && |
| (netconn_get_ipv6only(msg->conn) == 0)) { |
| /* change PCB type to IPADDR_TYPE_ANY */ |
| IP_SET_TYPE_VAL(msg->conn->pcb.tcp->local_ip, IPADDR_TYPE_ANY); |
| IP_SET_TYPE_VAL(msg->conn->pcb.tcp->remote_ip, IPADDR_TYPE_ANY); |
| } |
| #endif /* LWIP_IPV4 && LWIP_IPV6 */ |
| |
| lpcb = tcp_listen_with_backlog_and_err(msg->conn->pcb.tcp, backlog, &err); |
| |
| if (lpcb == NULL) { |
| /* in this case, the old pcb is still allocated */ |
| } else { |
| /* delete the recvmbox and allocate the acceptmbox */ |
| if (sys_mbox_valid(&msg->conn->recvmbox)) { |
| /** @todo: should we drain the recvmbox here? */ |
| sys_mbox_free(&msg->conn->recvmbox); |
| sys_mbox_set_invalid(&msg->conn->recvmbox); |
| } |
| err = ERR_OK; |
| if (!sys_mbox_valid(&msg->conn->acceptmbox)) { |
| err = sys_mbox_new(&msg->conn->acceptmbox, DEFAULT_ACCEPTMBOX_SIZE); |
| } |
| if (err == ERR_OK) { |
| msg->conn->state = NETCONN_LISTEN; |
| msg->conn->pcb.tcp = lpcb; |
| tcp_arg(msg->conn->pcb.tcp, msg->conn); |
| tcp_accept(msg->conn->pcb.tcp, accept_function); |
| } else { |
| /* since the old pcb is already deallocated, free lpcb now */ |
| tcp_close(lpcb); |
| msg->conn->pcb.tcp = NULL; |
| } |
| } |
| } |
| } else if (msg->conn->state == NETCONN_LISTEN) { |
| /* already listening, allow updating of the backlog */ |
| err = ERR_OK; |
| tcp_backlog_set(msg->conn->pcb.tcp, msg->msg.lb.backlog); |
| } else { |
| err = ERR_CONN; |
| } |
| } else { |
| err = ERR_ARG; |
| } |
| } else { |
| err = ERR_CONN; |
| } |
| msg->err = err; |
| TCPIP_APIMSG_ACK(msg); |
| } |
| #endif /* LWIP_TCP */ |
| |
| /** |
| * Send some data on a RAW or UDP pcb contained in a netconn |
| * Called from netconn_send |
| * |
| * @param m the api_msg pointing to the connection |
| */ |
| void |
| lwip_netconn_do_send(void *m) |
| { |
| struct api_msg *msg = (struct api_msg *)m; |
| |
| err_t err = netconn_err(msg->conn); |
| if (err == ERR_OK) { |
| if (msg->conn->pcb.tcp != NULL) { |
| switch (NETCONNTYPE_GROUP(msg->conn->type)) { |
| #if LWIP_RAW |
| case NETCONN_RAW: |
| if (ip_addr_isany(&msg->msg.b->addr) || IP_IS_ANY_TYPE_VAL(msg->msg.b->addr)) { |
| err = raw_send(msg->conn->pcb.raw, msg->msg.b->p); |
| } else { |
| err = raw_sendto(msg->conn->pcb.raw, msg->msg.b->p, &msg->msg.b->addr); |
| } |
| break; |
| #endif |
| #if LWIP_UDP |
| case NETCONN_UDP: |
| #if LWIP_CHECKSUM_ON_COPY |
| if (ip_addr_isany(&msg->msg.b->addr) || IP_IS_ANY_TYPE_VAL(msg->msg.b->addr)) { |
| err = udp_send_chksum(msg->conn->pcb.udp, msg->msg.b->p, |
| msg->msg.b->flags & NETBUF_FLAG_CHKSUM, msg->msg.b->toport_chksum); |
| } else { |
| err = udp_sendto_chksum(msg->conn->pcb.udp, msg->msg.b->p, |
| &msg->msg.b->addr, msg->msg.b->port, |
| msg->msg.b->flags & NETBUF_FLAG_CHKSUM, msg->msg.b->toport_chksum); |
| } |
| #else /* LWIP_CHECKSUM_ON_COPY */ |
| if (ip_addr_isany_val(msg->msg.b->addr) || IP_IS_ANY_TYPE_VAL(msg->msg.b->addr)) { |
| err = udp_send(msg->conn->pcb.udp, msg->msg.b->p); |
| } else { |
| err = udp_sendto(msg->conn->pcb.udp, msg->msg.b->p, &msg->msg.b->addr, msg->msg.b->port); |
| } |
| #endif /* LWIP_CHECKSUM_ON_COPY */ |
| break; |
| #endif /* LWIP_UDP */ |
| default: |
| err = ERR_CONN; |
| break; |
| } |
| } else { |
| err = ERR_CONN; |
| } |
| } |
| msg->err = err; |
| TCPIP_APIMSG_ACK(msg); |
| } |
| |
| #if LWIP_TCP |
| /** |
| * Indicate data has been received from a TCP pcb contained in a netconn |
| * Called from netconn_recv |
| * |
| * @param m the api_msg pointing to the connection |
| */ |
| void |
| lwip_netconn_do_recv(void *m) |
| { |
| struct api_msg *msg = (struct api_msg *)m; |
| |
| msg->err = ERR_OK; |
| if (msg->conn->pcb.tcp != NULL) { |
| if (NETCONNTYPE_GROUP(msg->conn->type) == NETCONN_TCP) { |
| size_t remaining = msg->msg.r.len; |
| do { |
| u16_t recved = (u16_t)((remaining > 0xffff) ? 0xffff : remaining); |
| tcp_recved(msg->conn->pcb.tcp, recved); |
| remaining -= recved; |
| } while (remaining != 0); |
| } |
| } |
| TCPIP_APIMSG_ACK(msg); |
| } |
| |
| #if TCP_LISTEN_BACKLOG |
| /** Indicate that a TCP pcb has been accepted |
| * Called from netconn_accept |
| * |
| * @param m the api_msg pointing to the connection |
| */ |
| void |
| lwip_netconn_do_accepted(void *m) |
| { |
| struct api_msg *msg = (struct api_msg *)m; |
| |
| msg->err = ERR_OK; |
| if (msg->conn->pcb.tcp != NULL) { |
| if (NETCONNTYPE_GROUP(msg->conn->type) == NETCONN_TCP) { |
| tcp_backlog_accepted(msg->conn->pcb.tcp); |
| } |
| } |
| TCPIP_APIMSG_ACK(msg); |
| } |
| #endif /* TCP_LISTEN_BACKLOG */ |
| |
| /** |
| * See if more data needs to be written from a previous call to netconn_write. |
| * Called initially from lwip_netconn_do_write. If the first call can't send all data |
| * (because of low memory or empty send-buffer), this function is called again |
| * from sent_tcp() or poll_tcp() to send more data. If all data is sent, the |
| * blocking application thread (waiting in netconn_write) is released. |
| * |
| * @param conn netconn (that is currently in state NETCONN_WRITE) to process |
| * @return ERR_OK |
| * ERR_MEM if LWIP_TCPIP_CORE_LOCKING=1 and sending hasn't yet finished |
| */ |
| static err_t |
| lwip_netconn_do_writemore(struct netconn *conn WRITE_DELAYED_PARAM) |
| { |
| err_t err; |
| const void *dataptr; |
| u16_t len, available; |
| u8_t write_finished = 0; |
| size_t diff; |
| u8_t dontblock; |
| u8_t apiflags; |
| u8_t write_more; |
| |
| LWIP_ASSERT("conn != NULL", conn != NULL); |
| LWIP_ASSERT("conn->state == NETCONN_WRITE", (conn->state == NETCONN_WRITE)); |
| LWIP_ASSERT("conn->current_msg != NULL", conn->current_msg != NULL); |
| LWIP_ASSERT("conn->pcb.tcp != NULL", conn->pcb.tcp != NULL); |
| LWIP_ASSERT("conn->current_msg->msg.w.offset < conn->current_msg->msg.w.len", |
| conn->current_msg->msg.w.offset < conn->current_msg->msg.w.len); |
| LWIP_ASSERT("conn->current_msg->msg.w.vector_cnt > 0", conn->current_msg->msg.w.vector_cnt > 0); |
| |
| apiflags = conn->current_msg->msg.w.apiflags; |
| dontblock = netconn_is_nonblocking(conn) || (apiflags & NETCONN_DONTBLOCK); |
| |
| #if LWIP_SO_SNDTIMEO |
| if ((conn->send_timeout != 0) && |
| ((s32_t)(sys_now() - conn->current_msg->msg.w.time_started) >= conn->send_timeout)) { |
| write_finished = 1; |
| if (conn->current_msg->msg.w.offset == 0) { |
| /* nothing has been written */ |
| err = ERR_WOULDBLOCK; |
| } else { |
| /* partial write */ |
| err = ERR_OK; |
| } |
| } else |
| #endif /* LWIP_SO_SNDTIMEO */ |
| { |
| do { |
| dataptr = (const u8_t *)conn->current_msg->msg.w.vector->ptr + conn->current_msg->msg.w.vector_off; |
| diff = conn->current_msg->msg.w.vector->len - conn->current_msg->msg.w.vector_off; |
| if (diff > 0xffffUL) { /* max_u16_t */ |
| len = 0xffff; |
| apiflags |= TCP_WRITE_FLAG_MORE; |
| } else { |
| len = (u16_t)diff; |
| } |
| available = tcp_sndbuf(conn->pcb.tcp); |
| if (available < len) { |
| /* don't try to write more than sendbuf */ |
| len = available; |
| if (dontblock) { |
| if (!len) { |
| /* set error according to partial write or not */ |
| err = (conn->current_msg->msg.w.offset == 0) ? ERR_WOULDBLOCK : ERR_OK; |
| goto err_mem; |
| } |
| } else { |
| apiflags |= TCP_WRITE_FLAG_MORE; |
| } |
| } |
| LWIP_ASSERT("lwip_netconn_do_writemore: invalid length!", |
| ((conn->current_msg->msg.w.vector_off + len) <= conn->current_msg->msg.w.vector->len)); |
| /* we should loop around for more sending in the following cases: |
| 1) We couldn't finish the current vector because of 16-bit size limitations. |
| tcp_write() and tcp_sndbuf() both are limited to 16-bit sizes |
| 2) We are sending the remainder of the current vector and have more */ |
| if ((len == 0xffff && diff > 0xffffUL) || |
| (len == (u16_t)diff && conn->current_msg->msg.w.vector_cnt > 1)) { |
| write_more = 1; |
| apiflags |= TCP_WRITE_FLAG_MORE; |
| } else { |
| write_more = 0; |
| } |
| err = tcp_write(conn->pcb.tcp, dataptr, len, apiflags); |
| if (err == ERR_OK) { |
| conn->current_msg->msg.w.offset += len; |
| conn->current_msg->msg.w.vector_off += len; |
| /* check if current vector is finished */ |
| if (conn->current_msg->msg.w.vector_off == conn->current_msg->msg.w.vector->len) { |
| conn->current_msg->msg.w.vector_cnt--; |
| /* if we have additional vectors, move on to them */ |
| if (conn->current_msg->msg.w.vector_cnt > 0) { |
| conn->current_msg->msg.w.vector++; |
| conn->current_msg->msg.w.vector_off = 0; |
| } |
| } |
| } |
| } while (write_more && err == ERR_OK); |
| /* if OK or memory error, check available space */ |
| if ((err == ERR_OK) || (err == ERR_MEM)) { |
| err_mem: |
| if (dontblock && (conn->current_msg->msg.w.offset < conn->current_msg->msg.w.len)) { |
| /* non-blocking write did not write everything: mark the pcb non-writable |
| and let poll_tcp check writable space to mark the pcb writable again */ |
| API_EVENT(conn, NETCONN_EVT_SENDMINUS, 0); |
| conn->flags |= NETCONN_FLAG_CHECK_WRITESPACE; |
| } else if ((tcp_sndbuf(conn->pcb.tcp) <= TCP_SNDLOWAT) || |
| (tcp_sndqueuelen(conn->pcb.tcp) >= TCP_SNDQUEUELOWAT)) { |
| /* The queued byte- or pbuf-count exceeds the configured low-water limit, |
| let select mark this pcb as non-writable. */ |
| API_EVENT(conn, NETCONN_EVT_SENDMINUS, 0); |
| } |
| } |
| |
| if (err == ERR_OK) { |
| err_t out_err; |
| if ((conn->current_msg->msg.w.offset == conn->current_msg->msg.w.len) || dontblock) { |
| /* return sent length (caller reads length from msg.w.offset) */ |
| write_finished = 1; |
| } |
| out_err = tcp_output(conn->pcb.tcp); |
| if (out_err == ERR_RTE) { |
| /* If tcp_output fails because no route is found, |
| don't try writing any more but return the error |
| to the application thread. */ |
| err = out_err; |
| write_finished = 1; |
| } |
| } else if (err == ERR_MEM) { |
| /* If ERR_MEM, we wait for sent_tcp or poll_tcp to be called. |
| For blocking sockets, we do NOT return to the application |
| thread, since ERR_MEM is only a temporary error! Non-blocking |
| will remain non-writable until sent_tcp/poll_tcp is called */ |
| |
| /* tcp_write returned ERR_MEM, try tcp_output anyway */ |
| err_t out_err = tcp_output(conn->pcb.tcp); |
| if (out_err == ERR_RTE) { |
| /* If tcp_output fails because no route is found, |
| don't try writing any more but return the error |
| to the application thread. */ |
| err = out_err; |
| write_finished = 1; |
| } else if (dontblock) { |
| /* non-blocking write is done on ERR_MEM, set error according |
| to partial write or not */ |
| err = (conn->current_msg->msg.w.offset == 0) ? ERR_WOULDBLOCK : ERR_OK; |
| write_finished = 1; |
| } |
| } else { |
| /* On errors != ERR_MEM, we don't try writing any more but return |
| the error to the application thread. */ |
| write_finished = 1; |
| } |
| } |
| if (write_finished) { |
| /* everything was written: set back connection state |
| and back to application task */ |
| sys_sem_t *op_completed_sem = LWIP_API_MSG_SEM(conn->current_msg); |
| conn->current_msg->err = err; |
| conn->current_msg = NULL; |
| conn->state = NETCONN_NONE; |
| #if LWIP_TCPIP_CORE_LOCKING |
| if (delayed) |
| #endif |
| { |
| sys_sem_signal(op_completed_sem); |
| } |
| } |
| #if LWIP_TCPIP_CORE_LOCKING |
| else { |
| return ERR_MEM; |
| } |
| #endif |
| return ERR_OK; |
| } |
| #endif /* LWIP_TCP */ |
| |
| /** |
| * Send some data on a TCP pcb contained in a netconn |
| * Called from netconn_write |
| * |
| * @param m the api_msg pointing to the connection |
| */ |
| void |
| lwip_netconn_do_write(void *m) |
| { |
| struct api_msg *msg = (struct api_msg *)m; |
| |
| err_t err = netconn_err(msg->conn); |
| if (err == ERR_OK) { |
| if (NETCONNTYPE_GROUP(msg->conn->type) == NETCONN_TCP) { |
| #if LWIP_TCP |
| if (msg->conn->state != NETCONN_NONE) { |
| /* netconn is connecting, closing or in blocking write */ |
| err = ERR_INPROGRESS; |
| } else if (msg->conn->pcb.tcp != NULL) { |
| msg->conn->state = NETCONN_WRITE; |
| /* set all the variables used by lwip_netconn_do_writemore */ |
| LWIP_ASSERT("already writing or closing", msg->conn->current_msg == NULL); |
| LWIP_ASSERT("msg->msg.w.len != 0", msg->msg.w.len != 0); |
| msg->conn->current_msg = msg; |
| #if LWIP_TCPIP_CORE_LOCKING |
| if (lwip_netconn_do_writemore(msg->conn, 0) != ERR_OK) { |
| LWIP_ASSERT("state!", msg->conn->state == NETCONN_WRITE); |
| UNLOCK_TCPIP_CORE(); |
| sys_arch_sem_wait(LWIP_API_MSG_SEM(msg), 0); |
| LOCK_TCPIP_CORE(); |
| LWIP_ASSERT("state!", msg->conn->state != NETCONN_WRITE); |
| } |
| #else /* LWIP_TCPIP_CORE_LOCKING */ |
| lwip_netconn_do_writemore(msg->conn); |
| #endif /* LWIP_TCPIP_CORE_LOCKING */ |
| /* for both cases: if lwip_netconn_do_writemore was called, don't ACK the APIMSG |
| since lwip_netconn_do_writemore ACKs it! */ |
| return; |
| } else { |
| err = ERR_CONN; |
| } |
| #else /* LWIP_TCP */ |
| err = ERR_VAL; |
| #endif /* LWIP_TCP */ |
| #if (LWIP_UDP || LWIP_RAW) |
| } else { |
| err = ERR_VAL; |
| #endif /* (LWIP_UDP || LWIP_RAW) */ |
| } |
| } |
| msg->err = err; |
| TCPIP_APIMSG_ACK(msg); |
| } |
| |
| /** |
| * Return a connection's local or remote address |
| * Called from netconn_getaddr |
| * |
| * @param m the api_msg pointing to the connection |
| */ |
| void |
| lwip_netconn_do_getaddr(void *m) |
| { |
| struct api_msg *msg = (struct api_msg *)m; |
| |
| if (msg->conn->pcb.ip != NULL) { |
| if (msg->msg.ad.local) { |
| ip_addr_copy(API_EXPR_DEREF(msg->msg.ad.ipaddr), |
| msg->conn->pcb.ip->local_ip); |
| } else { |
| ip_addr_copy(API_EXPR_DEREF(msg->msg.ad.ipaddr), |
| msg->conn->pcb.ip->remote_ip); |
| } |
| |
| msg->err = ERR_OK; |
| switch (NETCONNTYPE_GROUP(msg->conn->type)) { |
| #if LWIP_RAW |
| case NETCONN_RAW: |
| if (msg->msg.ad.local) { |
| API_EXPR_DEREF(msg->msg.ad.port) = msg->conn->pcb.raw->protocol; |
| } else { |
| /* return an error as connecting is only a helper for upper layers */ |
| msg->err = ERR_CONN; |
| } |
| break; |
| #endif /* LWIP_RAW */ |
| #if LWIP_UDP |
| case NETCONN_UDP: |
| if (msg->msg.ad.local) { |
| API_EXPR_DEREF(msg->msg.ad.port) = msg->conn->pcb.udp->local_port; |
| } else { |
| if ((msg->conn->pcb.udp->flags & UDP_FLAGS_CONNECTED) == 0) { |
| msg->err = ERR_CONN; |
| } else { |
| API_EXPR_DEREF(msg->msg.ad.port) = msg->conn->pcb.udp->remote_port; |
| } |
| } |
| break; |
| #endif /* LWIP_UDP */ |
| #if LWIP_TCP |
| case NETCONN_TCP: |
| if ((msg->msg.ad.local == 0) && |
| ((msg->conn->pcb.tcp->state == CLOSED) || (msg->conn->pcb.tcp->state == LISTEN))) { |
| /* pcb is not connected and remote name is requested */ |
| msg->err = ERR_CONN; |
| } else { |
| API_EXPR_DEREF(msg->msg.ad.port) = (msg->msg.ad.local ? msg->conn->pcb.tcp->local_port : msg->conn->pcb.tcp->remote_port); |
| } |
| break; |
| #endif /* LWIP_TCP */ |
| default: |
| LWIP_ASSERT("invalid netconn_type", 0); |
| break; |
| } |
| } else { |
| msg->err = ERR_CONN; |
| } |
| TCPIP_APIMSG_ACK(msg); |
| } |
| |
| /** |
| * Close or half-shutdown a TCP pcb contained in a netconn |
| * Called from netconn_close |
| * In contrast to closing sockets, the netconn is not deallocated. |
| * |
| * @param m the api_msg pointing to the connection |
| */ |
| void |
| lwip_netconn_do_close(void *m) |
| { |
| struct api_msg *msg = (struct api_msg *)m; |
| |
| #if LWIP_TCP |
| enum netconn_state state = msg->conn->state; |
| /* First check if this is a TCP netconn and if it is in a correct state |
| (LISTEN doesn't support half shutdown) */ |
| if ((msg->conn->pcb.tcp != NULL) && |
| (NETCONNTYPE_GROUP(msg->conn->type) == NETCONN_TCP) && |
| ((msg->msg.sd.shut == NETCONN_SHUT_RDWR) || (state != NETCONN_LISTEN))) { |
| /* Check if we are in a connected state */ |
| if (state == NETCONN_CONNECT) { |
| /* TCP connect in progress: cannot shutdown */ |
| msg->err = ERR_CONN; |
| } else if (state == NETCONN_WRITE) { |
| #if LWIP_NETCONN_FULLDUPLEX |
| if (msg->msg.sd.shut & NETCONN_SHUT_WR) { |
| /* close requested, abort running write */ |
| sys_sem_t *write_completed_sem; |
| LWIP_ASSERT("msg->conn->current_msg != NULL", msg->conn->current_msg != NULL); |
| write_completed_sem = LWIP_API_MSG_SEM(msg->conn->current_msg); |
| msg->conn->current_msg->err = ERR_CLSD; |
| msg->conn->current_msg = NULL; |
| msg->conn->state = NETCONN_NONE; |
| state = NETCONN_NONE; |
| sys_sem_signal(write_completed_sem); |
| } else { |
| LWIP_ASSERT("msg->msg.sd.shut == NETCONN_SHUT_RD", msg->msg.sd.shut == NETCONN_SHUT_RD); |
| /* In this case, let the write continue and do not interfere with |
| conn->current_msg or conn->state! */ |
| msg->err = tcp_shutdown(msg->conn->pcb.tcp, 1, 0); |
| } |
| } |
| if (state == NETCONN_NONE) { |
| #else /* LWIP_NETCONN_FULLDUPLEX */ |
| msg->err = ERR_INPROGRESS; |
| } else { |
| #endif /* LWIP_NETCONN_FULLDUPLEX */ |
| if (msg->msg.sd.shut & NETCONN_SHUT_RD) { |
| #if LWIP_NETCONN_FULLDUPLEX |
| /* Mark mboxes invalid */ |
| netconn_mark_mbox_invalid(msg->conn); |
| #else /* LWIP_NETCONN_FULLDUPLEX */ |
| netconn_drain(msg->conn); |
| #endif /* LWIP_NETCONN_FULLDUPLEX */ |
| } |
| LWIP_ASSERT("already writing or closing", msg->conn->current_msg == NULL); |
| msg->conn->state = NETCONN_CLOSE; |
| msg->conn->current_msg = msg; |
| #if LWIP_TCPIP_CORE_LOCKING |
| if (lwip_netconn_do_close_internal(msg->conn, 0) != ERR_OK) { |
| LWIP_ASSERT("state!", msg->conn->state == NETCONN_CLOSE); |
| UNLOCK_TCPIP_CORE(); |
| sys_arch_sem_wait(LWIP_API_MSG_SEM(msg), 0); |
| LOCK_TCPIP_CORE(); |
| LWIP_ASSERT("state!", msg->conn->state == NETCONN_NONE); |
| } |
| #else /* LWIP_TCPIP_CORE_LOCKING */ |
| lwip_netconn_do_close_internal(msg->conn); |
| #endif /* LWIP_TCPIP_CORE_LOCKING */ |
| /* for tcp netconns, lwip_netconn_do_close_internal ACKs the message */ |
| return; |
| } |
| } else |
| #endif /* LWIP_TCP */ |
| { |
| msg->err = ERR_CONN; |
| } |
| TCPIP_APIMSG_ACK(msg); |
| } |
| |
| #if LWIP_IGMP || (LWIP_IPV6 && LWIP_IPV6_MLD) |
| /** |
| * Join multicast groups for UDP netconns. |
| * Called from netconn_join_leave_group |
| * |
| * @param m the api_msg pointing to the connection |
| */ |
| void |
| lwip_netconn_do_join_leave_group(void *m) |
| { |
| struct api_msg *msg = (struct api_msg *)m; |
| |
| msg->err = ERR_CONN; |
| if (msg->conn->pcb.tcp != NULL) { |
| if (NETCONNTYPE_GROUP(msg->conn->type) == NETCONN_UDP) { |
| #if LWIP_UDP |
| #if LWIP_IPV6 && LWIP_IPV6_MLD |
| if (NETCONNTYPE_ISIPV6(msg->conn->type)) { |
| if (msg->msg.jl.join_or_leave == NETCONN_JOIN) { |
| msg->err = mld6_joingroup(ip_2_ip6(API_EXPR_REF(msg->msg.jl.netif_addr)), |
| ip_2_ip6(API_EXPR_REF(msg->msg.jl.multiaddr))); |
| } else { |
| msg->err = mld6_leavegroup(ip_2_ip6(API_EXPR_REF(msg->msg.jl.netif_addr)), |
| ip_2_ip6(API_EXPR_REF(msg->msg.jl.multiaddr))); |
| } |
| } else |
| #endif /* LWIP_IPV6 && LWIP_IPV6_MLD */ |
| { |
| #if LWIP_IGMP |
| if (msg->msg.jl.join_or_leave == NETCONN_JOIN) { |
| msg->err = igmp_joingroup(ip_2_ip4(API_EXPR_REF(msg->msg.jl.netif_addr)), |
| ip_2_ip4(API_EXPR_REF(msg->msg.jl.multiaddr))); |
| } else { |
| msg->err = igmp_leavegroup(ip_2_ip4(API_EXPR_REF(msg->msg.jl.netif_addr)), |
| ip_2_ip4(API_EXPR_REF(msg->msg.jl.multiaddr))); |
| } |
| #endif /* LWIP_IGMP */ |
| } |
| #endif /* LWIP_UDP */ |
| #if (LWIP_TCP || LWIP_RAW) |
| } else { |
| msg->err = ERR_VAL; |
| #endif /* (LWIP_TCP || LWIP_RAW) */ |
| } |
| } |
| TCPIP_APIMSG_ACK(msg); |
| } |
| /** |
| * Join multicast groups for UDP netconns. |
| * Called from netconn_join_leave_group_netif |
| * |
| * @param m the api_msg pointing to the connection |
| */ |
| void |
| lwip_netconn_do_join_leave_group_netif(void *m) |
| { |
| struct api_msg *msg = (struct api_msg *)m; |
| struct netif *netif; |
| |
| netif = netif_get_by_index(msg->msg.jl.if_idx); |
| if (netif == NULL) { |
| msg->err = ERR_IF; |
| goto done; |
| } |
| |
| msg->err = ERR_CONN; |
| if (msg->conn->pcb.tcp != NULL) { |
| if (NETCONNTYPE_GROUP(msg->conn->type) == NETCONN_UDP) { |
| #if LWIP_UDP |
| #if LWIP_IPV6 && LWIP_IPV6_MLD |
| if (NETCONNTYPE_ISIPV6(msg->conn->type)) { |
| if (msg->msg.jl.join_or_leave == NETCONN_JOIN) { |
| msg->err = mld6_joingroup_netif(netif, |
| ip_2_ip6(API_EXPR_REF(msg->msg.jl.multiaddr))); |
| } else { |
| msg->err = mld6_leavegroup_netif(netif, |
| ip_2_ip6(API_EXPR_REF(msg->msg.jl.multiaddr))); |
| } |
| } else |
| #endif /* LWIP_IPV6 && LWIP_IPV6_MLD */ |
| { |
| #if LWIP_IGMP |
| if (msg->msg.jl.join_or_leave == NETCONN_JOIN) { |
| msg->err = igmp_joingroup_netif(netif, |
| ip_2_ip4(API_EXPR_REF(msg->msg.jl.multiaddr))); |
| } else { |
| msg->err = igmp_leavegroup_netif(netif, |
| ip_2_ip4(API_EXPR_REF(msg->msg.jl.multiaddr))); |
| } |
| #endif /* LWIP_IGMP */ |
| } |
| #endif /* LWIP_UDP */ |
| #if (LWIP_TCP || LWIP_RAW) |
| } else { |
| msg->err = ERR_VAL; |
| #endif /* (LWIP_TCP || LWIP_RAW) */ |
| } |
| } |
| |
| done: |
| TCPIP_APIMSG_ACK(msg); |
| } |
| #endif /* LWIP_IGMP || (LWIP_IPV6 && LWIP_IPV6_MLD) */ |
| |
| #if LWIP_DNS |
| /** |
| * Callback function that is called when DNS name is resolved |
| * (or on timeout). A waiting application thread is waked up by |
| * signaling the semaphore. |
| */ |
| static void |
| lwip_netconn_do_dns_found(const char *name, const ip_addr_t *ipaddr, void *arg) |
| { |
| struct dns_api_msg *msg = (struct dns_api_msg *)arg; |
| |
| /* we trust the internal implementation to be correct :-) */ |
| LWIP_UNUSED_ARG(name); |
| |
| if (ipaddr == NULL) { |
| /* timeout or memory error */ |
| API_EXPR_DEREF(msg->err) = ERR_VAL; |
| } else { |
| /* address was resolved */ |
| API_EXPR_DEREF(msg->err) = ERR_OK; |
| API_EXPR_DEREF(msg->addr) = *ipaddr; |
| } |
| /* wake up the application task waiting in netconn_gethostbyname */ |
| sys_sem_signal(API_EXPR_REF_SEM(msg->sem)); |
| } |
| |
| /** |
| * Execute a DNS query |
| * Called from netconn_gethostbyname |
| * |
| * @param arg the dns_api_msg pointing to the query |
| */ |
| void |
| lwip_netconn_do_gethostbyname(void *arg) |
| { |
| struct dns_api_msg *msg = (struct dns_api_msg *)arg; |
| u8_t addrtype = |
| #if LWIP_IPV4 && LWIP_IPV6 |
| msg->dns_addrtype; |
| #else |
| LWIP_DNS_ADDRTYPE_DEFAULT; |
| #endif |
| |
| API_EXPR_DEREF(msg->err) = dns_gethostbyname_addrtype(msg->name, |
| API_EXPR_REF(msg->addr), lwip_netconn_do_dns_found, msg, addrtype); |
| #if LWIP_TCPIP_CORE_LOCKING |
| /* For core locking, only block if we need to wait for answer/timeout */ |
| if (API_EXPR_DEREF(msg->err) == ERR_INPROGRESS) { |
| UNLOCK_TCPIP_CORE(); |
| sys_sem_wait(API_EXPR_REF_SEM(msg->sem)); |
| LOCK_TCPIP_CORE(); |
| LWIP_ASSERT("do_gethostbyname still in progress!!", API_EXPR_DEREF(msg->err) != ERR_INPROGRESS); |
| } |
| #else /* LWIP_TCPIP_CORE_LOCKING */ |
| if (API_EXPR_DEREF(msg->err) != ERR_INPROGRESS) { |
| /* on error or immediate success, wake up the application |
| * task waiting in netconn_gethostbyname */ |
| sys_sem_signal(API_EXPR_REF_SEM(msg->sem)); |
| } |
| #endif /* LWIP_TCPIP_CORE_LOCKING */ |
| } |
| #endif /* LWIP_DNS */ |
| |
| #endif /* LWIP_NETCONN */ |