2015-10-11 17:43:33 -06:00
|
|
|
/*
|
|
|
|
* This file is part of the Micro Python project, http://micropython.org/
|
|
|
|
*
|
|
|
|
* The MIT License (MIT)
|
|
|
|
*
|
|
|
|
* Copyright (c) 2013, 2014 Damien P. George
|
|
|
|
* Copyright (c) 2015 Galen Hazelwood
|
|
|
|
*
|
|
|
|
* Permission is hereby granted, free of charge, to any person obtaining a copy
|
|
|
|
* of this software and associated documentation files (the "Software"), to deal
|
|
|
|
* in the Software without restriction, including without limitation the rights
|
|
|
|
* to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
|
|
|
* copies of the Software, and to permit persons to whom the Software is
|
|
|
|
* furnished to do so, subject to the following conditions:
|
|
|
|
*
|
|
|
|
* The above copyright notice and this permission notice shall be included in
|
|
|
|
* all copies or substantial portions of the Software.
|
|
|
|
*
|
|
|
|
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
|
|
|
* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
|
|
|
* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
|
|
|
* AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
|
|
|
* LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
|
|
|
* OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
|
|
|
|
* THE SOFTWARE.
|
|
|
|
*/
|
|
|
|
|
|
|
|
#include <string.h>
|
|
|
|
#include <errno.h>
|
2015-12-31 13:54:28 -07:00
|
|
|
#include <stdio.h>
|
2015-10-11 17:43:33 -06:00
|
|
|
|
|
|
|
#include "py/nlr.h"
|
|
|
|
#include "py/objlist.h"
|
|
|
|
#include "py/runtime.h"
|
2015-10-26 15:04:33 -06:00
|
|
|
#include "py/stream.h"
|
2015-11-01 23:46:58 -07:00
|
|
|
#include "py/mphal.h"
|
2015-10-11 17:43:33 -06:00
|
|
|
|
|
|
|
#include "netutils.h"
|
|
|
|
|
|
|
|
#include "lwip/init.h"
|
|
|
|
#include "lwip/timers.h"
|
|
|
|
#include "lwip/tcp.h"
|
|
|
|
#include "lwip/udp.h"
|
|
|
|
//#include "lwip/raw.h"
|
|
|
|
#include "lwip/dns.h"
|
|
|
|
|
2016-04-13 16:15:52 -06:00
|
|
|
#if 0 // print debugging info
|
|
|
|
#define DEBUG_printf DEBUG_printf
|
|
|
|
#else // don't print debugging info
|
|
|
|
#define DEBUG_printf(...) (void)0
|
|
|
|
#endif
|
|
|
|
|
2016-03-25 09:44:24 -06:00
|
|
|
// For compatibilily with older lwIP versions.
|
|
|
|
#ifndef ip_set_option
|
|
|
|
#define ip_set_option(pcb, opt) ((pcb)->so_options |= (opt))
|
|
|
|
#endif
|
2016-03-25 12:53:52 -06:00
|
|
|
#ifndef ip_reset_option
|
|
|
|
#define ip_reset_option(pcb, opt) ((pcb)->so_options &= ~(opt))
|
|
|
|
#endif
|
2016-03-25 09:44:24 -06:00
|
|
|
|
2015-10-11 17:43:33 -06:00
|
|
|
#ifdef MICROPY_PY_LWIP_SLIP
|
|
|
|
#include "netif/slipif.h"
|
2015-10-26 15:04:33 -06:00
|
|
|
#include "lwip/sio.h"
|
2015-10-11 17:43:33 -06:00
|
|
|
#endif
|
|
|
|
|
|
|
|
#ifdef MICROPY_PY_LWIP_SLIP
|
|
|
|
/******************************************************************************/
|
|
|
|
// Slip object for modlwip. Requires a serial driver for the port that supports
|
|
|
|
// the lwip serial callback functions.
|
|
|
|
|
|
|
|
typedef struct _lwip_slip_obj_t {
|
|
|
|
mp_obj_base_t base;
|
|
|
|
struct netif lwip_netif;
|
|
|
|
} lwip_slip_obj_t;
|
|
|
|
|
|
|
|
// Slip object is unique for now. Possibly can fix this later. FIXME
|
|
|
|
STATIC lwip_slip_obj_t lwip_slip_obj;
|
|
|
|
|
|
|
|
// Declare these early.
|
2015-10-26 16:23:07 -06:00
|
|
|
void mod_lwip_register_poll(void (*poll)(void *arg), void *poll_arg);
|
|
|
|
void mod_lwip_deregister_poll(void (*poll)(void *arg), void *poll_arg);
|
2015-10-11 17:43:33 -06:00
|
|
|
|
|
|
|
STATIC void slip_lwip_poll(void *netif) {
|
2015-10-26 16:23:07 -06:00
|
|
|
slipif_poll((struct netif*)netif);
|
2015-10-11 17:43:33 -06:00
|
|
|
}
|
|
|
|
|
|
|
|
STATIC const mp_obj_type_t lwip_slip_type;
|
|
|
|
|
2015-10-26 15:04:33 -06:00
|
|
|
// lwIP SLIP callback functions
|
|
|
|
sio_fd_t sio_open(u8_t dvnum) {
|
|
|
|
// We support singleton SLIP interface, so just return any truish value.
|
|
|
|
return (sio_fd_t)1;
|
|
|
|
}
|
|
|
|
|
|
|
|
void sio_send(u8_t c, sio_fd_t fd) {
|
|
|
|
mp_obj_type_t *type = mp_obj_get_type(MP_STATE_VM(lwip_slip_stream));
|
|
|
|
int error;
|
|
|
|
type->stream_p->write(MP_STATE_VM(lwip_slip_stream), &c, 1, &error);
|
|
|
|
}
|
|
|
|
|
|
|
|
u32_t sio_tryread(sio_fd_t fd, u8_t *data, u32_t len) {
|
|
|
|
mp_obj_type_t *type = mp_obj_get_type(MP_STATE_VM(lwip_slip_stream));
|
|
|
|
int error;
|
|
|
|
mp_uint_t out_sz = type->stream_p->read(MP_STATE_VM(lwip_slip_stream), data, len, &error);
|
|
|
|
if (out_sz == MP_STREAM_ERROR) {
|
|
|
|
if (mp_is_nonblocking_error(error)) {
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
// Can't do much else, can we?
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
return out_sz;
|
|
|
|
}
|
|
|
|
|
2015-10-11 17:43:33 -06:00
|
|
|
// constructor lwip.slip(device=integer, iplocal=string, ipremote=string)
|
|
|
|
STATIC mp_obj_t lwip_slip_make_new(mp_obj_t type_in, mp_uint_t n_args, mp_uint_t n_kw, const mp_obj_t *args) {
|
|
|
|
mp_arg_check_num(n_args, n_kw, 3, 3, false);
|
|
|
|
|
|
|
|
lwip_slip_obj.base.type = &lwip_slip_type;
|
|
|
|
|
2015-10-26 15:04:33 -06:00
|
|
|
MP_STATE_VM(lwip_slip_stream) = args[0];
|
2015-10-11 17:43:33 -06:00
|
|
|
|
|
|
|
ip_addr_t iplocal, ipremote;
|
|
|
|
if (!ipaddr_aton(mp_obj_str_get_str(args[1]), &iplocal)) {
|
|
|
|
nlr_raise(mp_obj_new_exception_msg(&mp_type_ValueError, "not a valid local IP"));
|
|
|
|
}
|
|
|
|
if (!ipaddr_aton(mp_obj_str_get_str(args[2]), &ipremote)) {
|
|
|
|
nlr_raise(mp_obj_new_exception_msg(&mp_type_ValueError, "not a valid remote IP"));
|
|
|
|
}
|
|
|
|
|
2015-10-26 16:37:35 -06:00
|
|
|
struct netif *n = &lwip_slip_obj.lwip_netif;
|
2015-10-26 15:04:33 -06:00
|
|
|
if (netif_add(n, &iplocal, IP_ADDR_BROADCAST, &ipremote, NULL, slipif_init, ip_input) == NULL) {
|
2015-10-11 17:43:33 -06:00
|
|
|
nlr_raise(mp_obj_new_exception_msg(&mp_type_ValueError, "out of memory"));
|
|
|
|
}
|
|
|
|
netif_set_up(n);
|
|
|
|
netif_set_default(n);
|
|
|
|
mod_lwip_register_poll(slip_lwip_poll, n);
|
|
|
|
|
|
|
|
return (mp_obj_t)&lwip_slip_obj;
|
|
|
|
}
|
|
|
|
|
|
|
|
STATIC mp_obj_t lwip_slip_status(mp_obj_t self_in) {
|
|
|
|
// Null function for now.
|
|
|
|
return mp_const_none;
|
|
|
|
}
|
|
|
|
|
|
|
|
STATIC MP_DEFINE_CONST_FUN_OBJ_1(lwip_slip_status_obj, lwip_slip_status);
|
|
|
|
|
|
|
|
STATIC const mp_map_elem_t lwip_slip_locals_dict_table[] = {
|
|
|
|
{ MP_OBJ_NEW_QSTR(MP_QSTR_status), (mp_obj_t)&lwip_slip_status_obj },
|
|
|
|
};
|
|
|
|
|
|
|
|
STATIC MP_DEFINE_CONST_DICT(lwip_slip_locals_dict, lwip_slip_locals_dict_table);
|
|
|
|
|
|
|
|
STATIC const mp_obj_type_t lwip_slip_type = {
|
|
|
|
{ &mp_type_type },
|
|
|
|
.name = MP_QSTR_slip,
|
|
|
|
.make_new = lwip_slip_make_new,
|
|
|
|
.locals_dict = (mp_obj_t)&lwip_slip_locals_dict,
|
|
|
|
};
|
|
|
|
|
|
|
|
#endif // MICROPY_PY_LWIP_SLIP
|
|
|
|
|
|
|
|
/******************************************************************************/
|
|
|
|
// Table to convert lwIP err_t codes to socket errno codes, from the lwIP
|
|
|
|
// socket API.
|
|
|
|
|
2015-12-30 07:43:35 -07:00
|
|
|
// Extension to lwIP error codes
|
|
|
|
#define _ERR_BADF -16
|
2016-03-25 12:32:01 -06:00
|
|
|
// TODO: We just know that change happened somewhere between 1.4.0 and 1.4.1,
|
|
|
|
// investigate in more detail.
|
|
|
|
#if LWIP_VERSION < 0x01040100
|
2015-10-11 17:43:33 -06:00
|
|
|
static const int error_lookup_table[] = {
|
|
|
|
0, /* ERR_OK 0 No error, everything OK. */
|
|
|
|
ENOMEM, /* ERR_MEM -1 Out of memory error. */
|
|
|
|
ENOBUFS, /* ERR_BUF -2 Buffer error. */
|
|
|
|
EWOULDBLOCK, /* ERR_TIMEOUT -3 Timeout */
|
|
|
|
EHOSTUNREACH, /* ERR_RTE -4 Routing problem. */
|
|
|
|
EINPROGRESS, /* ERR_INPROGRESS -5 Operation in progress */
|
|
|
|
EINVAL, /* ERR_VAL -6 Illegal value. */
|
|
|
|
EWOULDBLOCK, /* ERR_WOULDBLOCK -7 Operation would block. */
|
2016-03-25 12:32:01 -06:00
|
|
|
|
|
|
|
ECONNABORTED, /* ERR_ABRT -8 Connection aborted. */
|
|
|
|
ECONNRESET, /* ERR_RST -9 Connection reset. */
|
|
|
|
ENOTCONN, /* ERR_CLSD -10 Connection closed. */
|
|
|
|
ENOTCONN, /* ERR_CONN -11 Not connected. */
|
|
|
|
EIO, /* ERR_ARG -12 Illegal argument. */
|
|
|
|
EADDRINUSE, /* ERR_USE -13 Address in use. */
|
|
|
|
-1, /* ERR_IF -14 Low-level netif error */
|
|
|
|
EALREADY, /* ERR_ISCONN -15 Already connected. */
|
|
|
|
EBADF, /* _ERR_BADF -16 Closed socket (null pcb) */
|
|
|
|
};
|
|
|
|
#else
|
|
|
|
static const int error_lookup_table[] = {
|
|
|
|
0, /* ERR_OK 0 No error, everything OK. */
|
|
|
|
ENOMEM, /* ERR_MEM -1 Out of memory error. */
|
|
|
|
ENOBUFS, /* ERR_BUF -2 Buffer error. */
|
|
|
|
EWOULDBLOCK, /* ERR_TIMEOUT -3 Timeout */
|
|
|
|
EHOSTUNREACH, /* ERR_RTE -4 Routing problem. */
|
|
|
|
EINPROGRESS, /* ERR_INPROGRESS -5 Operation in progress */
|
|
|
|
EINVAL, /* ERR_VAL -6 Illegal value. */
|
|
|
|
EWOULDBLOCK, /* ERR_WOULDBLOCK -7 Operation would block. */
|
|
|
|
|
2015-10-11 17:43:33 -06:00
|
|
|
EADDRINUSE, /* ERR_USE -8 Address in use. */
|
|
|
|
EALREADY, /* ERR_ISCONN -9 Already connected. */
|
|
|
|
ECONNABORTED, /* ERR_ABRT -10 Connection aborted. */
|
|
|
|
ECONNRESET, /* ERR_RST -11 Connection reset. */
|
|
|
|
ENOTCONN, /* ERR_CLSD -12 Connection closed. */
|
|
|
|
ENOTCONN, /* ERR_CONN -13 Not connected. */
|
|
|
|
EIO, /* ERR_ARG -14 Illegal argument. */
|
|
|
|
-1, /* ERR_IF -15 Low-level netif error */
|
2015-12-30 07:43:35 -07:00
|
|
|
EBADF, /* _ERR_BADF -16 Closed socket (null pcb) */
|
2015-10-11 17:43:33 -06:00
|
|
|
};
|
2016-03-25 12:32:01 -06:00
|
|
|
#endif
|
2015-10-11 17:43:33 -06:00
|
|
|
|
|
|
|
/*******************************************************************************/
|
|
|
|
// The socket object provided by lwip.socket.
|
|
|
|
|
|
|
|
#define MOD_NETWORK_AF_INET (2)
|
|
|
|
#define MOD_NETWORK_AF_INET6 (10)
|
|
|
|
|
|
|
|
#define MOD_NETWORK_SOCK_STREAM (1)
|
|
|
|
#define MOD_NETWORK_SOCK_DGRAM (2)
|
|
|
|
#define MOD_NETWORK_SOCK_RAW (3)
|
|
|
|
|
|
|
|
typedef struct _lwip_socket_obj_t {
|
|
|
|
mp_obj_base_t base;
|
|
|
|
|
2015-12-29 12:00:52 -07:00
|
|
|
volatile union {
|
2015-11-01 23:46:58 -07:00
|
|
|
struct tcp_pcb *tcp;
|
|
|
|
struct udp_pcb *udp;
|
|
|
|
} pcb;
|
2015-12-29 12:00:52 -07:00
|
|
|
volatile union {
|
2015-11-01 23:46:58 -07:00
|
|
|
struct pbuf *pbuf;
|
|
|
|
struct tcp_pcb *connection;
|
|
|
|
} incoming;
|
2016-04-17 09:06:45 -06:00
|
|
|
mp_obj_t callback;
|
2015-10-11 17:43:33 -06:00
|
|
|
byte peer[4];
|
|
|
|
mp_uint_t peer_port;
|
|
|
|
mp_uint_t timeout;
|
|
|
|
uint16_t leftover_count;
|
|
|
|
|
|
|
|
uint8_t domain;
|
|
|
|
uint8_t type;
|
|
|
|
|
2015-12-30 07:26:23 -07:00
|
|
|
#define STATE_NEW 0
|
|
|
|
#define STATE_CONNECTING 1
|
|
|
|
#define STATE_CONNECTED 2
|
|
|
|
#define STATE_PEER_CLOSED 3
|
|
|
|
// Negative value is lwIP error
|
|
|
|
int8_t state;
|
2015-10-11 17:43:33 -06:00
|
|
|
} lwip_socket_obj_t;
|
|
|
|
|
2015-12-29 11:35:41 -07:00
|
|
|
static inline void poll_sockets(void) {
|
2015-12-28 09:58:14 -07:00
|
|
|
#ifdef MICROPY_EVENT_POLL_HOOK
|
|
|
|
MICROPY_EVENT_POLL_HOOK;
|
|
|
|
#else
|
2015-12-29 11:35:41 -07:00
|
|
|
mp_hal_delay_ms(1);
|
2015-12-28 09:58:14 -07:00
|
|
|
#endif
|
2015-12-29 11:35:41 -07:00
|
|
|
}
|
|
|
|
|
2015-10-11 17:43:33 -06:00
|
|
|
/*******************************************************************************/
|
|
|
|
// Callback functions for the lwIP raw API.
|
|
|
|
|
2016-04-17 09:06:45 -06:00
|
|
|
static inline void exec_user_callback(lwip_socket_obj_t *socket) {
|
|
|
|
if (socket->callback != MP_OBJ_NULL) {
|
|
|
|
mp_call_function_1(socket->callback, socket);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2015-10-11 17:43:33 -06:00
|
|
|
// Callback for incoming UDP packets. We simply stash the packet and the source address,
|
|
|
|
// in case we need it for recvfrom.
|
|
|
|
STATIC void _lwip_udp_incoming(void *arg, struct udp_pcb *upcb, struct pbuf *p, ip_addr_t *addr, u16_t port) {
|
2015-10-26 16:23:07 -06:00
|
|
|
lwip_socket_obj_t *socket = (lwip_socket_obj_t*)arg;
|
2015-10-11 17:43:33 -06:00
|
|
|
|
2015-11-01 23:46:58 -07:00
|
|
|
if (socket->incoming.pbuf != NULL) {
|
2015-10-11 17:43:33 -06:00
|
|
|
// That's why they call it "unreliable". No room in the inn, drop the packet.
|
|
|
|
pbuf_free(p);
|
|
|
|
} else {
|
2015-11-01 23:46:58 -07:00
|
|
|
socket->incoming.pbuf = p;
|
2015-10-11 17:43:33 -06:00
|
|
|
socket->peer_port = (mp_uint_t)port;
|
2015-12-30 07:40:23 -07:00
|
|
|
memcpy(&socket->peer, addr, sizeof(socket->peer));
|
2015-10-11 17:43:33 -06:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
// Callback for general tcp errors.
|
|
|
|
STATIC void _lwip_tcp_error(void *arg, err_t err) {
|
2015-10-26 16:23:07 -06:00
|
|
|
lwip_socket_obj_t *socket = (lwip_socket_obj_t*)arg;
|
2015-10-11 17:43:33 -06:00
|
|
|
|
|
|
|
// Pass the error code back via the connection variable.
|
2015-12-30 07:26:23 -07:00
|
|
|
socket->state = err;
|
2015-10-11 17:43:33 -06:00
|
|
|
// If we got here, the lwIP stack either has deallocated or will deallocate the pcb.
|
2015-11-01 23:46:58 -07:00
|
|
|
socket->pcb.tcp = NULL;
|
2015-10-11 17:43:33 -06:00
|
|
|
}
|
|
|
|
|
|
|
|
// Callback for tcp connection requests. Error code err is unused. (See tcp.h)
|
|
|
|
STATIC err_t _lwip_tcp_connected(void *arg, struct tcp_pcb *tpcb, err_t err) {
|
2015-10-26 16:23:07 -06:00
|
|
|
lwip_socket_obj_t *socket = (lwip_socket_obj_t*)arg;
|
2015-10-11 17:43:33 -06:00
|
|
|
|
2015-12-30 07:26:23 -07:00
|
|
|
socket->state = STATE_CONNECTED;
|
2015-10-11 17:43:33 -06:00
|
|
|
return ERR_OK;
|
|
|
|
}
|
|
|
|
|
2016-04-10 16:21:34 -06:00
|
|
|
// By default, a child socket of listen socket is created with recv
|
|
|
|
// handler which discards incoming pbuf's. We don't want to do that,
|
|
|
|
// so set this handler which requests lwIP to keep pbuf's and deliver
|
|
|
|
// them later. We cannot cache pbufs in child socket on Python side,
|
|
|
|
// until it is created in accept().
|
|
|
|
STATIC err_t _lwip_tcp_recv_unaccepted(void *arg, struct tcp_pcb *pcb, struct pbuf *p, err_t err) {
|
|
|
|
return ERR_BUF;
|
|
|
|
}
|
|
|
|
|
2015-10-11 17:43:33 -06:00
|
|
|
// Callback for incoming tcp connections.
|
|
|
|
STATIC err_t _lwip_tcp_accept(void *arg, struct tcp_pcb *newpcb, err_t err) {
|
2015-10-26 16:23:07 -06:00
|
|
|
lwip_socket_obj_t *socket = (lwip_socket_obj_t*)arg;
|
2016-04-10 16:21:34 -06:00
|
|
|
tcp_recv(newpcb, _lwip_tcp_recv_unaccepted);
|
2015-10-11 17:43:33 -06:00
|
|
|
|
2015-11-01 23:46:58 -07:00
|
|
|
if (socket->incoming.connection != NULL) {
|
2016-04-14 15:35:47 -06:00
|
|
|
DEBUG_printf("_lwip_tcp_accept: Tried to queue >1 pcb waiting for accept\n");
|
2015-10-11 17:43:33 -06:00
|
|
|
// We need to handle this better. This single-level structure makes the
|
|
|
|
// backlog setting kind of pointless. FIXME
|
|
|
|
return ERR_BUF;
|
|
|
|
} else {
|
2015-11-01 23:46:58 -07:00
|
|
|
socket->incoming.connection = newpcb;
|
2016-04-17 09:06:45 -06:00
|
|
|
exec_user_callback(socket);
|
2015-10-11 17:43:33 -06:00
|
|
|
return ERR_OK;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
// Callback for inbound tcp packets.
|
|
|
|
STATIC err_t _lwip_tcp_recv(void *arg, struct tcp_pcb *tcpb, struct pbuf *p, err_t err) {
|
2015-10-26 16:23:07 -06:00
|
|
|
lwip_socket_obj_t *socket = (lwip_socket_obj_t*)arg;
|
2015-10-11 17:43:33 -06:00
|
|
|
|
|
|
|
if (p == NULL) {
|
|
|
|
// Other side has closed connection.
|
2016-04-13 16:15:52 -06:00
|
|
|
DEBUG_printf("_lwip_tcp_recv[%p]: other side closed connection\n", socket);
|
2015-12-30 07:26:23 -07:00
|
|
|
socket->state = STATE_PEER_CLOSED;
|
2016-04-17 09:06:45 -06:00
|
|
|
exec_user_callback(socket);
|
2015-10-11 17:43:33 -06:00
|
|
|
return ERR_OK;
|
2015-11-01 23:46:58 -07:00
|
|
|
} else if (socket->incoming.pbuf != NULL) {
|
2015-10-11 17:43:33 -06:00
|
|
|
// No room in the inn, let LWIP know it's still responsible for delivery later
|
|
|
|
return ERR_BUF;
|
|
|
|
}
|
2015-11-01 23:46:58 -07:00
|
|
|
socket->incoming.pbuf = p;
|
2016-04-17 09:06:45 -06:00
|
|
|
|
|
|
|
exec_user_callback(socket);
|
|
|
|
|
2015-10-11 17:43:33 -06:00
|
|
|
return ERR_OK;
|
|
|
|
}
|
|
|
|
|
|
|
|
/*******************************************************************************/
|
|
|
|
// Functions for socket send/recieve operations. Socket send/recv and friends call
|
|
|
|
// these to do the work.
|
|
|
|
|
|
|
|
// Helper function for send/sendto to handle UDP packets.
|
|
|
|
STATIC mp_uint_t lwip_udp_send(lwip_socket_obj_t *socket, const byte *buf, mp_uint_t len, byte *ip, mp_uint_t port, int *_errno) {
|
|
|
|
if (len > 0xffff) {
|
|
|
|
// Any packet that big is probably going to fail the pbuf_alloc anyway, but may as well try
|
|
|
|
len = 0xffff;
|
|
|
|
}
|
|
|
|
|
|
|
|
// FIXME: maybe PBUF_ROM?
|
|
|
|
struct pbuf *p = pbuf_alloc(PBUF_TRANSPORT, len, PBUF_RAM);
|
|
|
|
if (p == NULL) {
|
|
|
|
*_errno = ENOMEM;
|
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
|
|
|
|
memcpy(p->payload, buf, len);
|
|
|
|
|
|
|
|
err_t err;
|
|
|
|
if (ip == NULL) {
|
2015-11-01 23:46:58 -07:00
|
|
|
err = udp_send(socket->pcb.udp, p);
|
2015-10-11 17:43:33 -06:00
|
|
|
} else {
|
|
|
|
ip_addr_t dest;
|
|
|
|
IP4_ADDR(&dest, ip[0], ip[1], ip[2], ip[3]);
|
2015-11-01 23:46:58 -07:00
|
|
|
err = udp_sendto(socket->pcb.udp, p, &dest, port);
|
2015-10-11 17:43:33 -06:00
|
|
|
}
|
|
|
|
|
|
|
|
pbuf_free(p);
|
|
|
|
|
|
|
|
if (err != ERR_OK) {
|
|
|
|
*_errno = error_lookup_table[-err];
|
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
|
|
|
|
return len;
|
|
|
|
}
|
|
|
|
|
|
|
|
// Helper function for recv/recvfrom to handle UDP packets
|
|
|
|
STATIC mp_uint_t lwip_udp_receive(lwip_socket_obj_t *socket, byte *buf, mp_uint_t len, byte *ip, mp_uint_t *port, int *_errno) {
|
|
|
|
|
2015-11-01 23:46:58 -07:00
|
|
|
if (socket->incoming.pbuf == NULL) {
|
2015-10-11 17:43:33 -06:00
|
|
|
if (socket->timeout != -1) {
|
|
|
|
for (mp_uint_t retries = socket->timeout / 100; retries--;) {
|
2015-10-27 14:31:42 -06:00
|
|
|
mp_hal_delay_ms(100);
|
2015-11-01 23:46:58 -07:00
|
|
|
if (socket->incoming.pbuf != NULL) break;
|
2015-10-11 17:43:33 -06:00
|
|
|
}
|
2015-11-01 23:46:58 -07:00
|
|
|
if (socket->incoming.pbuf == NULL) {
|
2015-10-11 17:43:33 -06:00
|
|
|
*_errno = ETIMEDOUT;
|
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
} else {
|
2015-11-01 23:46:58 -07:00
|
|
|
while (socket->incoming.pbuf == NULL) {
|
2015-12-28 09:58:14 -07:00
|
|
|
poll_sockets();
|
2015-10-11 17:43:33 -06:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
if (ip != NULL) {
|
2015-12-30 07:40:23 -07:00
|
|
|
memcpy(ip, &socket->peer, sizeof(socket->peer));
|
2015-10-11 17:43:33 -06:00
|
|
|
*port = socket->peer_port;
|
|
|
|
}
|
|
|
|
|
2015-11-01 23:46:58 -07:00
|
|
|
struct pbuf *p = socket->incoming.pbuf;
|
2015-10-11 17:43:33 -06:00
|
|
|
|
|
|
|
u16_t result = pbuf_copy_partial(p, buf, ((p->tot_len > len) ? len : p->tot_len), 0);
|
|
|
|
pbuf_free(p);
|
2015-11-01 23:46:58 -07:00
|
|
|
socket->incoming.pbuf = NULL;
|
2015-10-11 17:43:33 -06:00
|
|
|
|
|
|
|
return (mp_uint_t) result;
|
|
|
|
}
|
|
|
|
|
2016-04-16 17:20:05 -06:00
|
|
|
// For use in stream virtual methods
|
|
|
|
#define STREAM_ERROR_CHECK(socket) \
|
|
|
|
if (socket->state < 0) { \
|
|
|
|
*_errno = error_lookup_table[-socket->state]; \
|
|
|
|
return MP_STREAM_ERROR; \
|
|
|
|
} \
|
|
|
|
assert(socket->pcb.tcp);
|
|
|
|
|
|
|
|
|
2015-10-11 17:43:33 -06:00
|
|
|
// Helper function for send/sendto to handle TCP packets
|
2015-10-26 16:26:05 -06:00
|
|
|
STATIC mp_uint_t lwip_tcp_send(lwip_socket_obj_t *socket, const byte *buf, mp_uint_t len, int *_errno) {
|
2016-04-16 17:20:05 -06:00
|
|
|
// Check for any pending errors
|
|
|
|
STREAM_ERROR_CHECK(socket);
|
|
|
|
|
2015-11-01 23:46:58 -07:00
|
|
|
u16_t available = tcp_sndbuf(socket->pcb.tcp);
|
2016-03-25 10:38:13 -06:00
|
|
|
|
|
|
|
if (available == 0) {
|
|
|
|
// Non-blocking socket
|
|
|
|
if (socket->timeout == 0) {
|
|
|
|
*_errno = EAGAIN;
|
2016-04-16 17:20:05 -06:00
|
|
|
return MP_STREAM_ERROR;
|
2016-03-25 10:38:13 -06:00
|
|
|
}
|
|
|
|
|
|
|
|
mp_uint_t start = mp_hal_ticks_ms();
|
|
|
|
// Assume that STATE_PEER_CLOSED may mean half-closed connection, where peer closed it
|
|
|
|
// sending direction, but not receiving. Consequently, check for both STATE_CONNECTED
|
|
|
|
// and STATE_PEER_CLOSED as normal conditions and still waiting for buffers to be sent.
|
|
|
|
// If peer fully closed socket, we would have socket->state set to ERR_RST (connection
|
|
|
|
// reset) by error callback.
|
|
|
|
// Avoid sending too small packets, so wait until at least 16 bytes available
|
|
|
|
while (socket->state >= STATE_CONNECTED && (available = tcp_sndbuf(socket->pcb.tcp)) < 16) {
|
|
|
|
if (socket->timeout != -1 && mp_hal_ticks_ms() - start > socket->timeout) {
|
|
|
|
*_errno = ETIMEDOUT;
|
2016-04-16 17:20:05 -06:00
|
|
|
return MP_STREAM_ERROR;
|
2016-03-25 10:38:13 -06:00
|
|
|
}
|
|
|
|
poll_sockets();
|
|
|
|
}
|
|
|
|
|
2016-04-16 17:20:05 -06:00
|
|
|
// While we waited, something could happen
|
|
|
|
STREAM_ERROR_CHECK(socket);
|
2016-03-25 10:38:13 -06:00
|
|
|
}
|
|
|
|
|
2015-10-26 16:26:05 -06:00
|
|
|
u16_t write_len = MIN(available, len);
|
2015-10-11 17:43:33 -06:00
|
|
|
|
2015-11-01 23:46:58 -07:00
|
|
|
err_t err = tcp_write(socket->pcb.tcp, buf, write_len, TCP_WRITE_FLAG_COPY);
|
2015-10-11 17:43:33 -06:00
|
|
|
|
|
|
|
if (err != ERR_OK) {
|
|
|
|
*_errno = error_lookup_table[-err];
|
2016-04-16 17:20:05 -06:00
|
|
|
return MP_STREAM_ERROR;
|
2015-10-11 17:43:33 -06:00
|
|
|
}
|
|
|
|
|
2015-10-26 16:26:05 -06:00
|
|
|
return write_len;
|
2015-10-11 17:43:33 -06:00
|
|
|
}
|
|
|
|
|
|
|
|
// Helper function for recv/recvfrom to handle TCP packets
|
|
|
|
STATIC mp_uint_t lwip_tcp_receive(lwip_socket_obj_t *socket, byte *buf, mp_uint_t len, int *_errno) {
|
2016-04-16 17:22:26 -06:00
|
|
|
// Check for any pending errors
|
|
|
|
STREAM_ERROR_CHECK(socket);
|
|
|
|
|
2015-11-01 23:46:58 -07:00
|
|
|
if (socket->incoming.pbuf == NULL) {
|
2016-01-02 14:56:27 -07:00
|
|
|
|
|
|
|
// Non-blocking socket
|
|
|
|
if (socket->timeout == 0) {
|
2016-04-13 15:58:51 -06:00
|
|
|
if (socket->state == STATE_PEER_CLOSED) {
|
|
|
|
return 0;
|
|
|
|
}
|
2016-01-02 14:56:27 -07:00
|
|
|
*_errno = EAGAIN;
|
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
|
2015-12-29 11:35:41 -07:00
|
|
|
mp_uint_t start = mp_hal_ticks_ms();
|
2015-12-30 12:03:58 -07:00
|
|
|
while (socket->state == STATE_CONNECTED && socket->incoming.pbuf == NULL) {
|
2015-12-29 11:35:41 -07:00
|
|
|
if (socket->timeout != -1 && mp_hal_ticks_ms() - start > socket->timeout) {
|
2015-10-11 17:43:33 -06:00
|
|
|
*_errno = ETIMEDOUT;
|
|
|
|
return -1;
|
|
|
|
}
|
2015-12-29 11:35:41 -07:00
|
|
|
poll_sockets();
|
2015-10-11 17:43:33 -06:00
|
|
|
}
|
2016-04-16 17:22:26 -06:00
|
|
|
|
2015-12-30 12:03:58 -07:00
|
|
|
if (socket->state == STATE_PEER_CLOSED) {
|
2015-12-30 15:49:50 -07:00
|
|
|
if (socket->incoming.pbuf == NULL) {
|
|
|
|
// socket closed and no data left in buffer
|
|
|
|
return 0;
|
|
|
|
}
|
2015-12-30 12:03:58 -07:00
|
|
|
} else if (socket->state != STATE_CONNECTED) {
|
2016-02-27 17:14:49 -07:00
|
|
|
assert(socket->state < 0);
|
|
|
|
*_errno = error_lookup_table[-socket->state];
|
2015-12-30 12:03:58 -07:00
|
|
|
return -1;
|
|
|
|
}
|
2015-10-11 17:43:33 -06:00
|
|
|
}
|
|
|
|
|
2016-04-16 17:22:26 -06:00
|
|
|
assert(socket->pcb.tcp != NULL);
|
|
|
|
|
2015-11-01 23:46:58 -07:00
|
|
|
struct pbuf *p = socket->incoming.pbuf;
|
2015-10-11 17:43:33 -06:00
|
|
|
|
|
|
|
if (socket->leftover_count == 0) {
|
|
|
|
socket->leftover_count = p->tot_len;
|
|
|
|
}
|
|
|
|
|
|
|
|
u16_t result = pbuf_copy_partial(p, buf, ((socket->leftover_count >= len) ? len : socket->leftover_count), (p->tot_len - socket->leftover_count));
|
|
|
|
if (socket->leftover_count > len) {
|
|
|
|
// More left over...
|
|
|
|
socket->leftover_count -= len;
|
|
|
|
} else {
|
|
|
|
pbuf_free(p);
|
2015-11-01 23:46:58 -07:00
|
|
|
socket->incoming.pbuf = NULL;
|
2015-10-11 17:43:33 -06:00
|
|
|
socket->leftover_count = 0;
|
|
|
|
}
|
|
|
|
|
2015-11-01 23:46:58 -07:00
|
|
|
tcp_recved(socket->pcb.tcp, result);
|
2015-10-11 17:43:33 -06:00
|
|
|
return (mp_uint_t) result;
|
|
|
|
}
|
|
|
|
|
|
|
|
/*******************************************************************************/
|
|
|
|
// The socket functions provided by lwip.socket.
|
|
|
|
|
|
|
|
STATIC const mp_obj_type_t lwip_socket_type;
|
|
|
|
|
2015-12-28 16:08:30 -07:00
|
|
|
STATIC void lwip_socket_print(const mp_print_t *print, mp_obj_t self_in, mp_print_kind_t kind) {
|
|
|
|
lwip_socket_obj_t *self = self_in;
|
|
|
|
mp_printf(print, "<socket state=%d timeout=%d incoming=%p remaining=%d>", self->state, self->timeout,
|
|
|
|
self->incoming.pbuf, self->leftover_count);
|
|
|
|
}
|
|
|
|
|
2015-10-11 17:43:33 -06:00
|
|
|
// FIXME: Only supports two arguments at present
|
2016-01-24 11:59:45 -07:00
|
|
|
STATIC mp_obj_t lwip_socket_make_new(const mp_obj_type_t *type, mp_uint_t n_args,
|
2015-10-11 17:43:33 -06:00
|
|
|
mp_uint_t n_kw, const mp_obj_t *args) {
|
|
|
|
mp_arg_check_num(n_args, n_kw, 0, 4, false);
|
|
|
|
|
|
|
|
lwip_socket_obj_t *socket = m_new_obj_with_finaliser(lwip_socket_obj_t);
|
|
|
|
socket->base.type = (mp_obj_t)&lwip_socket_type;
|
|
|
|
socket->domain = MOD_NETWORK_AF_INET;
|
|
|
|
socket->type = MOD_NETWORK_SOCK_STREAM;
|
2016-04-17 09:06:45 -06:00
|
|
|
socket->callback = MP_OBJ_NULL;
|
2015-10-11 17:43:33 -06:00
|
|
|
if (n_args >= 1) {
|
|
|
|
socket->domain = mp_obj_get_int(args[0]);
|
|
|
|
if (n_args >= 2) {
|
|
|
|
socket->type = mp_obj_get_int(args[1]);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
switch (socket->type) {
|
2015-11-01 23:46:58 -07:00
|
|
|
case MOD_NETWORK_SOCK_STREAM: socket->pcb.tcp = tcp_new(); break;
|
|
|
|
case MOD_NETWORK_SOCK_DGRAM: socket->pcb.udp = udp_new(); break;
|
|
|
|
//case MOD_NETWORK_SOCK_RAW: socket->pcb.raw = raw_new(); break;
|
2015-10-11 17:43:33 -06:00
|
|
|
default: nlr_raise(mp_obj_new_exception_arg1(&mp_type_OSError, MP_OBJ_NEW_SMALL_INT(EINVAL)));
|
|
|
|
}
|
|
|
|
|
2015-11-01 23:46:58 -07:00
|
|
|
if (socket->pcb.tcp == NULL) {
|
2015-10-11 17:43:33 -06:00
|
|
|
nlr_raise(mp_obj_new_exception_arg1(&mp_type_OSError, MP_OBJ_NEW_SMALL_INT(ENOMEM)));
|
|
|
|
}
|
|
|
|
|
|
|
|
switch (socket->type) {
|
|
|
|
case MOD_NETWORK_SOCK_STREAM: {
|
|
|
|
// Register the socket object as our callback argument.
|
2015-11-01 23:46:58 -07:00
|
|
|
tcp_arg(socket->pcb.tcp, (void*)socket);
|
2015-10-11 17:43:33 -06:00
|
|
|
// Register our error callback.
|
2015-11-01 23:46:58 -07:00
|
|
|
tcp_err(socket->pcb.tcp, _lwip_tcp_error);
|
2015-10-11 17:43:33 -06:00
|
|
|
break;
|
|
|
|
}
|
|
|
|
case MOD_NETWORK_SOCK_DGRAM: {
|
|
|
|
// Register our receive callback now. Since UDP sockets don't require binding or connection
|
|
|
|
// before use, there's no other good time to do it.
|
2015-11-01 23:46:58 -07:00
|
|
|
udp_recv(socket->pcb.udp, _lwip_udp_incoming, (void*)socket);
|
2015-10-11 17:43:33 -06:00
|
|
|
break;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2015-11-01 23:46:58 -07:00
|
|
|
socket->incoming.pbuf = NULL;
|
2015-10-11 17:43:33 -06:00
|
|
|
socket->timeout = -1;
|
2015-12-30 07:26:23 -07:00
|
|
|
socket->state = STATE_NEW;
|
2015-10-11 17:43:33 -06:00
|
|
|
socket->leftover_count = 0;
|
|
|
|
return socket;
|
|
|
|
}
|
|
|
|
|
|
|
|
STATIC mp_obj_t lwip_socket_close(mp_obj_t self_in) {
|
|
|
|
lwip_socket_obj_t *socket = self_in;
|
|
|
|
bool socket_is_listener = false;
|
|
|
|
|
2015-11-01 23:46:58 -07:00
|
|
|
if (socket->pcb.tcp == NULL) {
|
2015-10-11 17:43:33 -06:00
|
|
|
return mp_const_none;
|
|
|
|
}
|
|
|
|
switch (socket->type) {
|
|
|
|
case MOD_NETWORK_SOCK_STREAM: {
|
2015-11-01 23:46:58 -07:00
|
|
|
if (socket->pcb.tcp->state == LISTEN) {
|
2015-10-11 17:43:33 -06:00
|
|
|
socket_is_listener = true;
|
|
|
|
}
|
2015-11-01 23:46:58 -07:00
|
|
|
if (tcp_close(socket->pcb.tcp) != ERR_OK) {
|
2016-04-14 15:35:47 -06:00
|
|
|
DEBUG_printf("lwip_close: had to call tcp_abort()\n");
|
2015-11-01 23:46:58 -07:00
|
|
|
tcp_abort(socket->pcb.tcp);
|
2015-10-11 17:43:33 -06:00
|
|
|
}
|
|
|
|
break;
|
|
|
|
}
|
2015-11-01 23:46:58 -07:00
|
|
|
case MOD_NETWORK_SOCK_DGRAM: udp_remove(socket->pcb.udp); break;
|
|
|
|
//case MOD_NETWORK_SOCK_RAW: raw_remove(socket->pcb.raw); break;
|
2015-10-11 17:43:33 -06:00
|
|
|
}
|
2015-11-01 23:46:58 -07:00
|
|
|
socket->pcb.tcp = NULL;
|
2015-12-30 07:43:35 -07:00
|
|
|
socket->state = _ERR_BADF;
|
2015-11-01 23:46:58 -07:00
|
|
|
if (socket->incoming.pbuf != NULL) {
|
2015-10-11 17:43:33 -06:00
|
|
|
if (!socket_is_listener) {
|
2015-11-01 23:46:58 -07:00
|
|
|
pbuf_free(socket->incoming.pbuf);
|
2015-10-11 17:43:33 -06:00
|
|
|
} else {
|
2015-11-01 23:46:58 -07:00
|
|
|
tcp_abort(socket->incoming.connection);
|
2015-10-11 17:43:33 -06:00
|
|
|
}
|
2015-11-01 23:46:58 -07:00
|
|
|
socket->incoming.pbuf = NULL;
|
2015-10-11 17:43:33 -06:00
|
|
|
}
|
|
|
|
|
|
|
|
return mp_const_none;
|
|
|
|
}
|
|
|
|
STATIC MP_DEFINE_CONST_FUN_OBJ_1(lwip_socket_close_obj, lwip_socket_close);
|
|
|
|
|
|
|
|
STATIC mp_obj_t lwip_socket_bind(mp_obj_t self_in, mp_obj_t addr_in) {
|
|
|
|
lwip_socket_obj_t *socket = self_in;
|
|
|
|
|
|
|
|
uint8_t ip[NETUTILS_IPV4ADDR_BUFSIZE];
|
|
|
|
mp_uint_t port = netutils_parse_inet_addr(addr_in, ip, NETUTILS_BIG);
|
|
|
|
|
|
|
|
ip_addr_t bind_addr;
|
|
|
|
IP4_ADDR(&bind_addr, ip[0], ip[1], ip[2], ip[3]);
|
|
|
|
|
|
|
|
err_t err = ERR_ARG;
|
|
|
|
switch (socket->type) {
|
|
|
|
case MOD_NETWORK_SOCK_STREAM: {
|
2015-11-01 23:46:58 -07:00
|
|
|
err = tcp_bind(socket->pcb.tcp, &bind_addr, port);
|
2015-10-11 17:43:33 -06:00
|
|
|
break;
|
|
|
|
}
|
|
|
|
case MOD_NETWORK_SOCK_DGRAM: {
|
2015-11-01 23:46:58 -07:00
|
|
|
err = udp_bind(socket->pcb.udp, &bind_addr, port);
|
2015-10-11 17:43:33 -06:00
|
|
|
break;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
if (err != ERR_OK) {
|
|
|
|
nlr_raise(mp_obj_new_exception_arg1(&mp_type_OSError, MP_OBJ_NEW_SMALL_INT(error_lookup_table[-err])));
|
|
|
|
}
|
|
|
|
|
|
|
|
return mp_const_none;
|
|
|
|
}
|
|
|
|
STATIC MP_DEFINE_CONST_FUN_OBJ_2(lwip_socket_bind_obj, lwip_socket_bind);
|
|
|
|
|
|
|
|
STATIC mp_obj_t lwip_socket_listen(mp_obj_t self_in, mp_obj_t backlog_in) {
|
|
|
|
lwip_socket_obj_t *socket = self_in;
|
|
|
|
mp_int_t backlog = mp_obj_get_int(backlog_in);
|
|
|
|
|
2015-11-01 23:46:58 -07:00
|
|
|
if (socket->pcb.tcp == NULL) {
|
2015-10-11 17:43:33 -06:00
|
|
|
nlr_raise(mp_obj_new_exception_arg1(&mp_type_OSError, MP_OBJ_NEW_SMALL_INT(EBADF)));
|
|
|
|
}
|
|
|
|
if (socket->type != MOD_NETWORK_SOCK_STREAM) {
|
|
|
|
nlr_raise(mp_obj_new_exception_arg1(&mp_type_OSError, MP_OBJ_NEW_SMALL_INT(EOPNOTSUPP)));
|
|
|
|
}
|
|
|
|
|
2015-11-01 23:46:58 -07:00
|
|
|
struct tcp_pcb *new_pcb = tcp_listen_with_backlog(socket->pcb.tcp, (u8_t)backlog);
|
2015-10-11 17:43:33 -06:00
|
|
|
if (new_pcb == NULL) {
|
|
|
|
nlr_raise(mp_obj_new_exception_arg1(&mp_type_OSError, MP_OBJ_NEW_SMALL_INT(ENOMEM)));
|
|
|
|
}
|
2015-11-01 23:46:58 -07:00
|
|
|
socket->pcb.tcp = new_pcb;
|
2015-10-11 17:43:33 -06:00
|
|
|
tcp_accept(new_pcb, _lwip_tcp_accept);
|
|
|
|
|
|
|
|
return mp_const_none;
|
|
|
|
}
|
|
|
|
STATIC MP_DEFINE_CONST_FUN_OBJ_2(lwip_socket_listen_obj, lwip_socket_listen);
|
|
|
|
|
|
|
|
STATIC mp_obj_t lwip_socket_accept(mp_obj_t self_in) {
|
|
|
|
lwip_socket_obj_t *socket = self_in;
|
|
|
|
|
2015-11-01 23:46:58 -07:00
|
|
|
if (socket->pcb.tcp == NULL) {
|
2015-10-11 17:43:33 -06:00
|
|
|
nlr_raise(mp_obj_new_exception_arg1(&mp_type_OSError, MP_OBJ_NEW_SMALL_INT(EBADF)));
|
|
|
|
}
|
|
|
|
if (socket->type != MOD_NETWORK_SOCK_STREAM) {
|
|
|
|
nlr_raise(mp_obj_new_exception_arg1(&mp_type_OSError, MP_OBJ_NEW_SMALL_INT(EOPNOTSUPP)));
|
|
|
|
}
|
|
|
|
// I need to do this because "tcp_accepted", later, is a macro.
|
2015-11-01 23:46:58 -07:00
|
|
|
struct tcp_pcb *listener = socket->pcb.tcp;
|
2015-10-11 17:43:33 -06:00
|
|
|
if (listener->state != LISTEN) {
|
|
|
|
nlr_raise(mp_obj_new_exception_arg1(&mp_type_OSError, MP_OBJ_NEW_SMALL_INT(EINVAL)));
|
|
|
|
}
|
|
|
|
|
|
|
|
// accept incoming connection
|
2015-11-01 23:46:58 -07:00
|
|
|
if (socket->incoming.connection == NULL) {
|
2015-10-11 17:43:33 -06:00
|
|
|
if (socket->timeout != -1) {
|
|
|
|
for (mp_uint_t retries = socket->timeout / 100; retries--;) {
|
2015-10-27 14:31:42 -06:00
|
|
|
mp_hal_delay_ms(100);
|
2015-11-01 23:46:58 -07:00
|
|
|
if (socket->incoming.connection != NULL) break;
|
2015-10-11 17:43:33 -06:00
|
|
|
}
|
2015-11-01 23:46:58 -07:00
|
|
|
if (socket->incoming.connection == NULL) {
|
2015-10-11 17:43:33 -06:00
|
|
|
nlr_raise(mp_obj_new_exception_arg1(&mp_type_OSError, MP_OBJ_NEW_SMALL_INT(ETIMEDOUT)));
|
|
|
|
}
|
|
|
|
} else {
|
2015-11-01 23:46:58 -07:00
|
|
|
while (socket->incoming.connection == NULL) {
|
2015-12-28 09:58:14 -07:00
|
|
|
poll_sockets();
|
2015-10-11 17:43:33 -06:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
// create new socket object
|
|
|
|
lwip_socket_obj_t *socket2 = m_new_obj_with_finaliser(lwip_socket_obj_t);
|
|
|
|
socket2->base.type = (mp_obj_t)&lwip_socket_type;
|
|
|
|
|
|
|
|
// We get a new pcb handle...
|
2015-11-01 23:46:58 -07:00
|
|
|
socket2->pcb.tcp = socket->incoming.connection;
|
|
|
|
socket->incoming.connection = NULL;
|
2015-10-11 17:43:33 -06:00
|
|
|
|
|
|
|
// ...and set up the new socket for it.
|
|
|
|
socket2->domain = MOD_NETWORK_AF_INET;
|
|
|
|
socket2->type = MOD_NETWORK_SOCK_STREAM;
|
2015-11-01 23:46:58 -07:00
|
|
|
socket2->incoming.pbuf = NULL;
|
2015-10-11 17:43:33 -06:00
|
|
|
socket2->timeout = socket->timeout;
|
2015-12-30 07:26:23 -07:00
|
|
|
socket2->state = STATE_CONNECTED;
|
2015-10-11 17:43:33 -06:00
|
|
|
socket2->leftover_count = 0;
|
2016-04-17 09:06:45 -06:00
|
|
|
socket2->callback = MP_OBJ_NULL;
|
2015-11-01 23:46:58 -07:00
|
|
|
tcp_arg(socket2->pcb.tcp, (void*)socket2);
|
|
|
|
tcp_err(socket2->pcb.tcp, _lwip_tcp_error);
|
|
|
|
tcp_recv(socket2->pcb.tcp, _lwip_tcp_recv);
|
2015-10-11 17:43:33 -06:00
|
|
|
|
|
|
|
tcp_accepted(listener);
|
|
|
|
|
|
|
|
// make the return value
|
|
|
|
uint8_t ip[NETUTILS_IPV4ADDR_BUFSIZE];
|
2015-12-30 07:40:23 -07:00
|
|
|
memcpy(ip, &(socket2->pcb.tcp->remote_ip), sizeof(ip));
|
2015-11-01 23:46:58 -07:00
|
|
|
mp_uint_t port = (mp_uint_t)socket2->pcb.tcp->remote_port;
|
2015-10-11 17:43:33 -06:00
|
|
|
mp_obj_tuple_t *client = mp_obj_new_tuple(2, NULL);
|
|
|
|
client->items[0] = socket2;
|
|
|
|
client->items[1] = netutils_format_inet_addr(ip, port, NETUTILS_BIG);
|
|
|
|
|
|
|
|
return client;
|
|
|
|
}
|
|
|
|
STATIC MP_DEFINE_CONST_FUN_OBJ_1(lwip_socket_accept_obj, lwip_socket_accept);
|
|
|
|
|
|
|
|
STATIC mp_obj_t lwip_socket_connect(mp_obj_t self_in, mp_obj_t addr_in) {
|
|
|
|
lwip_socket_obj_t *socket = self_in;
|
|
|
|
|
2015-11-01 23:46:58 -07:00
|
|
|
if (socket->pcb.tcp == NULL) {
|
2015-10-11 17:43:33 -06:00
|
|
|
nlr_raise(mp_obj_new_exception_arg1(&mp_type_OSError, MP_OBJ_NEW_SMALL_INT(EBADF)));
|
|
|
|
}
|
|
|
|
|
|
|
|
// get address
|
|
|
|
uint8_t ip[NETUTILS_IPV4ADDR_BUFSIZE];
|
|
|
|
mp_uint_t port = netutils_parse_inet_addr(addr_in, ip, NETUTILS_BIG);
|
|
|
|
|
|
|
|
ip_addr_t dest;
|
|
|
|
IP4_ADDR(&dest, ip[0], ip[1], ip[2], ip[3]);
|
|
|
|
|
|
|
|
err_t err = ERR_ARG;
|
|
|
|
switch (socket->type) {
|
|
|
|
case MOD_NETWORK_SOCK_STREAM: {
|
2015-12-30 07:26:23 -07:00
|
|
|
if (socket->state != STATE_NEW) {
|
|
|
|
if (socket->state == STATE_CONNECTED) {
|
2015-10-11 17:43:33 -06:00
|
|
|
nlr_raise(mp_obj_new_exception_arg1(&mp_type_OSError, MP_OBJ_NEW_SMALL_INT(EALREADY)));
|
|
|
|
} else {
|
|
|
|
nlr_raise(mp_obj_new_exception_arg1(&mp_type_OSError, MP_OBJ_NEW_SMALL_INT(EINPROGRESS)));
|
|
|
|
}
|
|
|
|
}
|
|
|
|
// Register our recieve callback.
|
2015-11-01 23:46:58 -07:00
|
|
|
tcp_recv(socket->pcb.tcp, _lwip_tcp_recv);
|
2015-12-30 07:26:23 -07:00
|
|
|
socket->state = STATE_CONNECTING;
|
2015-11-01 23:46:58 -07:00
|
|
|
err = tcp_connect(socket->pcb.tcp, &dest, port, _lwip_tcp_connected);
|
2015-10-11 17:43:33 -06:00
|
|
|
if (err != ERR_OK) {
|
2015-12-30 07:26:23 -07:00
|
|
|
socket->state = STATE_NEW;
|
2015-10-11 17:43:33 -06:00
|
|
|
nlr_raise(mp_obj_new_exception_arg1(&mp_type_OSError, MP_OBJ_NEW_SMALL_INT(error_lookup_table[-err])));
|
|
|
|
}
|
|
|
|
socket->peer_port = (mp_uint_t)port;
|
2015-12-30 07:40:23 -07:00
|
|
|
memcpy(socket->peer, &dest, sizeof(socket->peer));
|
2015-10-11 17:43:33 -06:00
|
|
|
// And now we wait...
|
|
|
|
if (socket->timeout != -1) {
|
|
|
|
for (mp_uint_t retries = socket->timeout / 100; retries--;) {
|
2015-10-27 14:31:42 -06:00
|
|
|
mp_hal_delay_ms(100);
|
2015-12-30 07:26:23 -07:00
|
|
|
if (socket->state != STATE_CONNECTING) break;
|
2015-10-11 17:43:33 -06:00
|
|
|
}
|
2015-12-30 07:26:23 -07:00
|
|
|
if (socket->state == STATE_CONNECTING) {
|
2015-10-11 17:43:33 -06:00
|
|
|
nlr_raise(mp_obj_new_exception_arg1(&mp_type_OSError, MP_OBJ_NEW_SMALL_INT(ETIMEDOUT)));
|
|
|
|
}
|
|
|
|
} else {
|
2015-12-30 07:26:23 -07:00
|
|
|
while (socket->state == STATE_CONNECTING) {
|
2015-12-28 09:58:14 -07:00
|
|
|
poll_sockets();
|
2015-10-11 17:43:33 -06:00
|
|
|
}
|
|
|
|
}
|
2015-12-30 07:26:23 -07:00
|
|
|
if (socket->state == STATE_CONNECTED) {
|
2015-10-11 17:43:33 -06:00
|
|
|
err = ERR_OK;
|
|
|
|
} else {
|
2015-12-30 07:26:23 -07:00
|
|
|
err = socket->state;
|
2015-10-11 17:43:33 -06:00
|
|
|
}
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
case MOD_NETWORK_SOCK_DGRAM: {
|
2015-11-01 23:46:58 -07:00
|
|
|
err = udp_connect(socket->pcb.udp, &dest, port);
|
2015-10-11 17:43:33 -06:00
|
|
|
break;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
if (err != ERR_OK) {
|
|
|
|
nlr_raise(mp_obj_new_exception_arg1(&mp_type_OSError, MP_OBJ_NEW_SMALL_INT(error_lookup_table[-err])));
|
|
|
|
}
|
|
|
|
|
|
|
|
return mp_const_none;
|
|
|
|
}
|
|
|
|
STATIC MP_DEFINE_CONST_FUN_OBJ_2(lwip_socket_connect_obj, lwip_socket_connect);
|
|
|
|
|
2016-02-27 16:13:14 -07:00
|
|
|
STATIC void lwip_socket_check_connected(lwip_socket_obj_t *socket) {
|
2015-11-01 23:46:58 -07:00
|
|
|
if (socket->pcb.tcp == NULL) {
|
2015-10-11 17:43:33 -06:00
|
|
|
// not connected
|
2016-02-27 16:13:14 -07:00
|
|
|
int _errno = error_lookup_table[-socket->state];
|
2015-12-30 07:43:35 -07:00
|
|
|
socket->state = _ERR_BADF;
|
2015-10-11 17:43:33 -06:00
|
|
|
nlr_raise(mp_obj_new_exception_arg1(&mp_type_OSError, MP_OBJ_NEW_SMALL_INT(_errno)));
|
|
|
|
}
|
2016-02-27 16:13:14 -07:00
|
|
|
}
|
|
|
|
|
|
|
|
STATIC mp_obj_t lwip_socket_send(mp_obj_t self_in, mp_obj_t buf_in) {
|
|
|
|
lwip_socket_obj_t *socket = self_in;
|
|
|
|
int _errno;
|
|
|
|
|
|
|
|
lwip_socket_check_connected(socket);
|
2015-10-11 17:43:33 -06:00
|
|
|
|
|
|
|
mp_buffer_info_t bufinfo;
|
|
|
|
mp_get_buffer_raise(buf_in, &bufinfo, MP_BUFFER_READ);
|
|
|
|
|
|
|
|
mp_uint_t ret = 0;
|
|
|
|
switch (socket->type) {
|
|
|
|
case MOD_NETWORK_SOCK_STREAM: {
|
|
|
|
ret = lwip_tcp_send(socket, bufinfo.buf, bufinfo.len, &_errno);
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
case MOD_NETWORK_SOCK_DGRAM: {
|
|
|
|
ret = lwip_udp_send(socket, bufinfo.buf, bufinfo.len, NULL, 0, &_errno);
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
if (ret == -1) {
|
|
|
|
nlr_raise(mp_obj_new_exception_arg1(&mp_type_OSError, MP_OBJ_NEW_SMALL_INT(_errno)));
|
|
|
|
}
|
|
|
|
|
|
|
|
return mp_obj_new_int_from_uint(ret);
|
|
|
|
}
|
|
|
|
STATIC MP_DEFINE_CONST_FUN_OBJ_2(lwip_socket_send_obj, lwip_socket_send);
|
|
|
|
|
|
|
|
STATIC mp_obj_t lwip_socket_recv(mp_obj_t self_in, mp_obj_t len_in) {
|
|
|
|
lwip_socket_obj_t *socket = self_in;
|
|
|
|
int _errno;
|
|
|
|
|
2016-02-27 16:13:14 -07:00
|
|
|
lwip_socket_check_connected(socket);
|
2015-10-11 17:43:33 -06:00
|
|
|
|
|
|
|
mp_int_t len = mp_obj_get_int(len_in);
|
|
|
|
vstr_t vstr;
|
|
|
|
vstr_init_len(&vstr, len);
|
|
|
|
|
|
|
|
mp_uint_t ret = 0;
|
|
|
|
switch (socket->type) {
|
|
|
|
case MOD_NETWORK_SOCK_STREAM: {
|
|
|
|
ret = lwip_tcp_receive(socket, (byte*)vstr.buf, len, &_errno);
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
case MOD_NETWORK_SOCK_DGRAM: {
|
|
|
|
ret = lwip_udp_receive(socket, (byte*)vstr.buf, len, NULL, NULL, &_errno);
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
if (ret == -1) {
|
|
|
|
nlr_raise(mp_obj_new_exception_arg1(&mp_type_OSError, MP_OBJ_NEW_SMALL_INT(_errno)));
|
|
|
|
}
|
|
|
|
|
|
|
|
if (ret == 0) {
|
|
|
|
return mp_const_empty_bytes;
|
|
|
|
}
|
|
|
|
vstr.len = ret;
|
|
|
|
return mp_obj_new_str_from_vstr(&mp_type_bytes, &vstr);
|
|
|
|
}
|
|
|
|
STATIC MP_DEFINE_CONST_FUN_OBJ_2(lwip_socket_recv_obj, lwip_socket_recv);
|
|
|
|
|
|
|
|
STATIC mp_obj_t lwip_socket_sendto(mp_obj_t self_in, mp_obj_t data_in, mp_obj_t addr_in) {
|
|
|
|
lwip_socket_obj_t *socket = self_in;
|
|
|
|
int _errno;
|
|
|
|
|
2016-02-27 16:13:14 -07:00
|
|
|
lwip_socket_check_connected(socket);
|
2015-10-11 17:43:33 -06:00
|
|
|
|
|
|
|
mp_buffer_info_t bufinfo;
|
|
|
|
mp_get_buffer_raise(data_in, &bufinfo, MP_BUFFER_READ);
|
|
|
|
|
|
|
|
uint8_t ip[NETUTILS_IPV4ADDR_BUFSIZE];
|
|
|
|
mp_uint_t port = netutils_parse_inet_addr(addr_in, ip, NETUTILS_BIG);
|
|
|
|
|
|
|
|
mp_uint_t ret = 0;
|
|
|
|
switch (socket->type) {
|
|
|
|
case MOD_NETWORK_SOCK_STREAM: {
|
|
|
|
ret = lwip_tcp_send(socket, bufinfo.buf, bufinfo.len, &_errno);
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
case MOD_NETWORK_SOCK_DGRAM: {
|
|
|
|
ret = lwip_udp_send(socket, bufinfo.buf, bufinfo.len, ip, port, &_errno);
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
if (ret == -1) {
|
|
|
|
nlr_raise(mp_obj_new_exception_arg1(&mp_type_OSError, MP_OBJ_NEW_SMALL_INT(_errno)));
|
|
|
|
}
|
|
|
|
|
|
|
|
return mp_obj_new_int_from_uint(ret);
|
|
|
|
}
|
|
|
|
STATIC MP_DEFINE_CONST_FUN_OBJ_3(lwip_socket_sendto_obj, lwip_socket_sendto);
|
|
|
|
|
|
|
|
STATIC mp_obj_t lwip_socket_recvfrom(mp_obj_t self_in, mp_obj_t len_in) {
|
|
|
|
lwip_socket_obj_t *socket = self_in;
|
|
|
|
int _errno;
|
|
|
|
|
2016-02-27 16:13:14 -07:00
|
|
|
lwip_socket_check_connected(socket);
|
2015-10-11 17:43:33 -06:00
|
|
|
|
|
|
|
mp_int_t len = mp_obj_get_int(len_in);
|
|
|
|
vstr_t vstr;
|
|
|
|
vstr_init_len(&vstr, len);
|
|
|
|
byte ip[4];
|
|
|
|
mp_uint_t port;
|
|
|
|
|
|
|
|
mp_uint_t ret = 0;
|
|
|
|
switch (socket->type) {
|
|
|
|
case MOD_NETWORK_SOCK_STREAM: {
|
2015-12-30 07:40:23 -07:00
|
|
|
memcpy(ip, &socket->peer, sizeof(socket->peer));
|
2015-10-11 17:43:33 -06:00
|
|
|
port = (mp_uint_t) socket->peer_port;
|
|
|
|
ret = lwip_tcp_receive(socket, (byte*)vstr.buf, len, &_errno);
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
case MOD_NETWORK_SOCK_DGRAM: {
|
|
|
|
ret = lwip_udp_receive(socket, (byte*)vstr.buf, len, ip, &port, &_errno);
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
if (ret == -1) {
|
|
|
|
nlr_raise(mp_obj_new_exception_arg1(&mp_type_OSError, MP_OBJ_NEW_SMALL_INT(_errno)));
|
|
|
|
}
|
|
|
|
|
|
|
|
mp_obj_t tuple[2];
|
|
|
|
if (ret == 0) {
|
|
|
|
tuple[0] = mp_const_empty_bytes;
|
|
|
|
} else {
|
|
|
|
vstr.len = ret;
|
|
|
|
tuple[0] = mp_obj_new_str_from_vstr(&mp_type_bytes, &vstr);
|
|
|
|
}
|
|
|
|
tuple[1] = netutils_format_inet_addr(ip, port, NETUTILS_BIG);
|
|
|
|
return mp_obj_new_tuple(2, tuple);
|
|
|
|
}
|
|
|
|
STATIC MP_DEFINE_CONST_FUN_OBJ_2(lwip_socket_recvfrom_obj, lwip_socket_recvfrom);
|
|
|
|
|
|
|
|
STATIC mp_obj_t lwip_socket_settimeout(mp_obj_t self_in, mp_obj_t timeout_in) {
|
|
|
|
lwip_socket_obj_t *socket = self_in;
|
|
|
|
mp_uint_t timeout;
|
|
|
|
if (timeout_in == mp_const_none) {
|
|
|
|
timeout = -1;
|
|
|
|
} else {
|
|
|
|
#if MICROPY_PY_BUILTIN_FLOAT
|
|
|
|
timeout = 1000 * mp_obj_get_float(timeout_in);
|
|
|
|
#else
|
|
|
|
timeout = 1000 * mp_obj_get_int(timeout_in);
|
|
|
|
#endif
|
|
|
|
}
|
|
|
|
socket->timeout = timeout;
|
|
|
|
return mp_const_none;
|
|
|
|
}
|
|
|
|
STATIC MP_DEFINE_CONST_FUN_OBJ_2(lwip_socket_settimeout_obj, lwip_socket_settimeout);
|
|
|
|
|
2016-02-26 17:04:36 -07:00
|
|
|
STATIC mp_obj_t lwip_socket_setblocking(mp_obj_t self_in, mp_obj_t flag_in) {
|
|
|
|
lwip_socket_obj_t *socket = self_in;
|
|
|
|
bool val = mp_obj_is_true(flag_in);
|
|
|
|
if (val) {
|
|
|
|
socket->timeout = -1;
|
|
|
|
} else {
|
|
|
|
socket->timeout = 0;
|
|
|
|
}
|
|
|
|
return mp_const_none;
|
|
|
|
}
|
|
|
|
STATIC MP_DEFINE_CONST_FUN_OBJ_2(lwip_socket_setblocking_obj, lwip_socket_setblocking);
|
|
|
|
|
2015-12-31 13:54:28 -07:00
|
|
|
STATIC mp_obj_t lwip_socket_setsockopt(mp_uint_t n_args, const mp_obj_t *args) {
|
|
|
|
(void)n_args; // always 4
|
2016-03-25 09:44:24 -06:00
|
|
|
lwip_socket_obj_t *socket = args[0];
|
2016-04-17 09:06:45 -06:00
|
|
|
|
|
|
|
int opt = mp_obj_get_int(args[2]);
|
|
|
|
if (opt == 20) {
|
|
|
|
if (args[3] == mp_const_none) {
|
|
|
|
socket->callback = MP_OBJ_NULL;
|
|
|
|
} else {
|
|
|
|
socket->callback = args[3];
|
|
|
|
}
|
|
|
|
return mp_const_none;
|
|
|
|
}
|
|
|
|
|
|
|
|
// Integer options
|
2016-03-25 09:44:24 -06:00
|
|
|
mp_int_t val = mp_obj_get_int(args[3]);
|
2016-04-17 09:06:45 -06:00
|
|
|
switch (opt) {
|
2016-03-25 09:44:24 -06:00
|
|
|
case SOF_REUSEADDR:
|
|
|
|
// Options are common for UDP and TCP pcb's.
|
2016-03-25 12:53:52 -06:00
|
|
|
if (val) {
|
|
|
|
ip_set_option(socket->pcb.tcp, SOF_REUSEADDR);
|
|
|
|
} else {
|
|
|
|
ip_reset_option(socket->pcb.tcp, SOF_REUSEADDR);
|
|
|
|
}
|
2016-03-25 09:44:24 -06:00
|
|
|
break;
|
|
|
|
default:
|
|
|
|
printf("Warning: lwip.setsockopt() not implemented\n");
|
|
|
|
}
|
2015-12-31 13:54:28 -07:00
|
|
|
return mp_const_none;
|
|
|
|
}
|
|
|
|
STATIC MP_DEFINE_CONST_FUN_OBJ_VAR_BETWEEN(lwip_socket_setsockopt_obj, 4, 4, lwip_socket_setsockopt);
|
|
|
|
|
2015-12-31 22:31:16 -07:00
|
|
|
STATIC mp_obj_t lwip_socket_makefile(mp_uint_t n_args, const mp_obj_t *args) {
|
|
|
|
(void)n_args;
|
|
|
|
return args[0];
|
|
|
|
}
|
|
|
|
STATIC MP_DEFINE_CONST_FUN_OBJ_VAR_BETWEEN(lwip_socket_makefile_obj, 1, 3, lwip_socket_makefile);
|
|
|
|
|
2015-12-31 17:36:30 -07:00
|
|
|
STATIC mp_uint_t lwip_socket_read(mp_obj_t self_in, void *buf, mp_uint_t size, int *errcode) {
|
|
|
|
lwip_socket_obj_t *socket = self_in;
|
|
|
|
|
|
|
|
switch (socket->type) {
|
|
|
|
case MOD_NETWORK_SOCK_STREAM:
|
|
|
|
return lwip_tcp_receive(socket, buf, size, errcode);
|
|
|
|
case MOD_NETWORK_SOCK_DGRAM:
|
|
|
|
return lwip_udp_receive(socket, buf, size, NULL, NULL, errcode);
|
|
|
|
}
|
|
|
|
// Unreachable
|
|
|
|
return MP_STREAM_ERROR;
|
|
|
|
}
|
|
|
|
|
2016-01-02 14:22:37 -07:00
|
|
|
STATIC mp_uint_t lwip_socket_write(mp_obj_t self_in, const void *buf, mp_uint_t size, int *errcode) {
|
|
|
|
lwip_socket_obj_t *socket = self_in;
|
|
|
|
|
|
|
|
switch (socket->type) {
|
|
|
|
case MOD_NETWORK_SOCK_STREAM:
|
|
|
|
return lwip_tcp_send(socket, buf, size, errcode);
|
|
|
|
case MOD_NETWORK_SOCK_DGRAM:
|
|
|
|
return lwip_udp_send(socket, buf, size, NULL, 0, errcode);
|
|
|
|
}
|
|
|
|
// Unreachable
|
|
|
|
return MP_STREAM_ERROR;
|
|
|
|
}
|
|
|
|
|
2015-10-11 17:43:33 -06:00
|
|
|
STATIC const mp_map_elem_t lwip_socket_locals_dict_table[] = {
|
|
|
|
{ MP_OBJ_NEW_QSTR(MP_QSTR___del__), (mp_obj_t)&lwip_socket_close_obj },
|
|
|
|
{ MP_OBJ_NEW_QSTR(MP_QSTR_close), (mp_obj_t)&lwip_socket_close_obj },
|
|
|
|
{ MP_OBJ_NEW_QSTR(MP_QSTR_bind), (mp_obj_t)&lwip_socket_bind_obj },
|
|
|
|
{ MP_OBJ_NEW_QSTR(MP_QSTR_listen), (mp_obj_t)&lwip_socket_listen_obj },
|
|
|
|
{ MP_OBJ_NEW_QSTR(MP_QSTR_accept), (mp_obj_t)&lwip_socket_accept_obj },
|
|
|
|
{ MP_OBJ_NEW_QSTR(MP_QSTR_connect), (mp_obj_t)&lwip_socket_connect_obj },
|
|
|
|
{ MP_OBJ_NEW_QSTR(MP_QSTR_send), (mp_obj_t)&lwip_socket_send_obj },
|
|
|
|
{ MP_OBJ_NEW_QSTR(MP_QSTR_recv), (mp_obj_t)&lwip_socket_recv_obj },
|
|
|
|
{ MP_OBJ_NEW_QSTR(MP_QSTR_sendto), (mp_obj_t)&lwip_socket_sendto_obj },
|
|
|
|
{ MP_OBJ_NEW_QSTR(MP_QSTR_recvfrom), (mp_obj_t)&lwip_socket_recvfrom_obj },
|
|
|
|
{ MP_OBJ_NEW_QSTR(MP_QSTR_settimeout), (mp_obj_t)&lwip_socket_settimeout_obj },
|
2016-02-26 17:04:36 -07:00
|
|
|
{ MP_OBJ_NEW_QSTR(MP_QSTR_setblocking), (mp_obj_t)&lwip_socket_setblocking_obj },
|
2015-12-31 13:54:28 -07:00
|
|
|
{ MP_OBJ_NEW_QSTR(MP_QSTR_setsockopt), (mp_obj_t)&lwip_socket_setsockopt_obj },
|
2015-12-31 22:31:16 -07:00
|
|
|
{ MP_OBJ_NEW_QSTR(MP_QSTR_makefile), (mp_obj_t)&lwip_socket_makefile_obj },
|
2016-01-01 12:38:26 -07:00
|
|
|
|
|
|
|
{ MP_OBJ_NEW_QSTR(MP_QSTR_read), (mp_obj_t)&mp_stream_read_obj },
|
|
|
|
{ MP_OBJ_NEW_QSTR(MP_QSTR_readline), (mp_obj_t)&mp_stream_unbuffered_readline_obj},
|
2016-01-02 14:22:37 -07:00
|
|
|
{ MP_OBJ_NEW_QSTR(MP_QSTR_write), (mp_obj_t)&mp_stream_write_obj },
|
2015-10-11 17:43:33 -06:00
|
|
|
};
|
|
|
|
STATIC MP_DEFINE_CONST_DICT(lwip_socket_locals_dict, lwip_socket_locals_dict_table);
|
|
|
|
|
2015-12-31 17:36:30 -07:00
|
|
|
STATIC const mp_stream_p_t lwip_socket_stream_p = {
|
|
|
|
.read = lwip_socket_read,
|
2016-01-02 14:22:37 -07:00
|
|
|
.write = lwip_socket_write,
|
2015-12-31 17:36:30 -07:00
|
|
|
};
|
|
|
|
|
2015-10-11 17:43:33 -06:00
|
|
|
STATIC const mp_obj_type_t lwip_socket_type = {
|
|
|
|
{ &mp_type_type },
|
|
|
|
.name = MP_QSTR_socket,
|
2015-12-28 16:08:30 -07:00
|
|
|
.print = lwip_socket_print,
|
2015-10-11 17:43:33 -06:00
|
|
|
.make_new = lwip_socket_make_new,
|
2015-12-31 17:36:30 -07:00
|
|
|
.stream_p = &lwip_socket_stream_p,
|
2015-10-11 17:43:33 -06:00
|
|
|
.locals_dict = (mp_obj_t)&lwip_socket_locals_dict,
|
|
|
|
};
|
|
|
|
|
|
|
|
/******************************************************************************/
|
|
|
|
// Support functions for memory protection. lwIP has its own memory management
|
|
|
|
// routines for its internal structures, and since they might be called in
|
|
|
|
// interrupt handlers, they need some protection.
|
|
|
|
sys_prot_t sys_arch_protect() {
|
|
|
|
return (sys_prot_t)MICROPY_BEGIN_ATOMIC_SECTION();
|
|
|
|
}
|
|
|
|
|
|
|
|
void sys_arch_unprotect(sys_prot_t state) {
|
|
|
|
MICROPY_END_ATOMIC_SECTION((mp_uint_t)state);
|
|
|
|
}
|
|
|
|
|
|
|
|
/******************************************************************************/
|
|
|
|
// Polling callbacks for the interfaces connected to lwIP. Right now it calls
|
|
|
|
// itself a "list" but isn't; we only support a single interface.
|
|
|
|
|
|
|
|
typedef struct nic_poll {
|
|
|
|
void (* poll)(void *arg);
|
|
|
|
void *poll_arg;
|
|
|
|
} nic_poll_t;
|
|
|
|
|
|
|
|
STATIC nic_poll_t lwip_poll_list;
|
|
|
|
|
|
|
|
void mod_lwip_register_poll(void (* poll)(void *arg), void *poll_arg) {
|
|
|
|
lwip_poll_list.poll = poll;
|
|
|
|
lwip_poll_list.poll_arg = poll_arg;
|
|
|
|
}
|
|
|
|
|
|
|
|
void mod_lwip_deregister_poll(void (* poll)(void *arg), void *poll_arg) {
|
|
|
|
lwip_poll_list.poll = NULL;
|
|
|
|
}
|
|
|
|
|
|
|
|
/******************************************************************************/
|
|
|
|
// The lwip global functions.
|
|
|
|
|
|
|
|
STATIC mp_obj_t mod_lwip_reset() {
|
|
|
|
lwip_init();
|
|
|
|
lwip_poll_list.poll = NULL;
|
|
|
|
return mp_const_none;
|
|
|
|
}
|
|
|
|
MP_DEFINE_CONST_FUN_OBJ_0(mod_lwip_reset_obj, mod_lwip_reset);
|
|
|
|
|
|
|
|
STATIC mp_obj_t mod_lwip_callback() {
|
|
|
|
if (lwip_poll_list.poll != NULL) {
|
|
|
|
lwip_poll_list.poll(lwip_poll_list.poll_arg);
|
|
|
|
}
|
|
|
|
sys_check_timeouts();
|
|
|
|
return mp_const_none;
|
|
|
|
}
|
|
|
|
MP_DEFINE_CONST_FUN_OBJ_0(mod_lwip_callback_obj, mod_lwip_callback);
|
|
|
|
|
2016-02-18 05:50:29 -07:00
|
|
|
typedef struct _getaddrinfo_state_t {
|
|
|
|
volatile int status;
|
|
|
|
volatile ip_addr_t ipaddr;
|
|
|
|
} getaddrinfo_state_t;
|
|
|
|
|
|
|
|
// Callback for incoming DNS requests.
|
|
|
|
STATIC void lwip_getaddrinfo_cb(const char *name, ip_addr_t *ipaddr, void *arg) {
|
|
|
|
getaddrinfo_state_t *state = arg;
|
|
|
|
if (ipaddr != NULL) {
|
|
|
|
state->status = 1;
|
|
|
|
state->ipaddr = *ipaddr;
|
|
|
|
} else {
|
|
|
|
// error
|
|
|
|
state->status = -2;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2015-10-11 17:43:33 -06:00
|
|
|
// lwip.getaddrinfo
|
|
|
|
STATIC mp_obj_t lwip_getaddrinfo(mp_obj_t host_in, mp_obj_t port_in) {
|
|
|
|
mp_uint_t hlen;
|
|
|
|
const char *host = mp_obj_str_get_data(host_in, &hlen);
|
|
|
|
mp_int_t port = mp_obj_get_int(port_in);
|
|
|
|
|
2016-02-18 05:50:29 -07:00
|
|
|
getaddrinfo_state_t state;
|
|
|
|
state.status = 0;
|
2015-10-11 17:43:33 -06:00
|
|
|
|
2016-02-18 05:50:29 -07:00
|
|
|
err_t ret = dns_gethostbyname(host, (ip_addr_t*)&state.ipaddr, lwip_getaddrinfo_cb, &state);
|
|
|
|
switch (ret) {
|
|
|
|
case ERR_OK:
|
|
|
|
// cached
|
|
|
|
state.status = 1;
|
|
|
|
break;
|
|
|
|
case ERR_INPROGRESS:
|
|
|
|
while (state.status == 0) {
|
2015-12-28 09:58:14 -07:00
|
|
|
poll_sockets();
|
2015-10-11 17:43:33 -06:00
|
|
|
}
|
|
|
|
break;
|
2016-02-18 05:50:29 -07:00
|
|
|
default:
|
|
|
|
state.status = ret;
|
|
|
|
}
|
|
|
|
|
|
|
|
if (state.status < 0) {
|
|
|
|
// TODO: CPython raises gaierror, we raise with native lwIP negative error
|
|
|
|
// values, to differentiate from normal errno's at least in such way.
|
|
|
|
nlr_raise(mp_obj_new_exception_arg1(&mp_type_OSError, MP_OBJ_NEW_SMALL_INT(state.status)));
|
2015-10-11 17:43:33 -06:00
|
|
|
}
|
|
|
|
|
|
|
|
mp_obj_tuple_t *tuple = mp_obj_new_tuple(5, NULL);
|
|
|
|
tuple->items[0] = MP_OBJ_NEW_SMALL_INT(MOD_NETWORK_AF_INET);
|
|
|
|
tuple->items[1] = MP_OBJ_NEW_SMALL_INT(MOD_NETWORK_SOCK_STREAM);
|
|
|
|
tuple->items[2] = MP_OBJ_NEW_SMALL_INT(0);
|
|
|
|
tuple->items[3] = MP_OBJ_NEW_QSTR(MP_QSTR_);
|
2016-02-18 05:50:29 -07:00
|
|
|
tuple->items[4] = netutils_format_inet_addr((uint8_t*)&state.ipaddr, port, NETUTILS_BIG);
|
2015-10-11 17:43:33 -06:00
|
|
|
return mp_obj_new_list(1, (mp_obj_t*)&tuple);
|
|
|
|
}
|
|
|
|
STATIC MP_DEFINE_CONST_FUN_OBJ_2(lwip_getaddrinfo_obj, lwip_getaddrinfo);
|
|
|
|
|
|
|
|
#ifdef MICROPY_PY_LWIP
|
|
|
|
|
|
|
|
STATIC const mp_map_elem_t mp_module_lwip_globals_table[] = {
|
|
|
|
{ MP_OBJ_NEW_QSTR(MP_QSTR___name__), MP_OBJ_NEW_QSTR(MP_QSTR_lwip) },
|
|
|
|
{ MP_OBJ_NEW_QSTR(MP_QSTR_reset), (mp_obj_t)&mod_lwip_reset_obj },
|
|
|
|
{ MP_OBJ_NEW_QSTR(MP_QSTR_callback), (mp_obj_t)&mod_lwip_callback_obj },
|
|
|
|
{ MP_OBJ_NEW_QSTR(MP_QSTR_getaddrinfo), (mp_obj_t)&lwip_getaddrinfo_obj },
|
|
|
|
// objects
|
|
|
|
{ MP_OBJ_NEW_QSTR(MP_QSTR_socket), (mp_obj_t)&lwip_socket_type },
|
|
|
|
#ifdef MICROPY_PY_LWIP_SLIP
|
|
|
|
{ MP_OBJ_NEW_QSTR(MP_QSTR_slip), (mp_obj_t)&lwip_slip_type },
|
|
|
|
#endif
|
|
|
|
// class constants
|
|
|
|
{ MP_OBJ_NEW_QSTR(MP_QSTR_AF_INET), MP_OBJ_NEW_SMALL_INT(MOD_NETWORK_AF_INET) },
|
|
|
|
{ MP_OBJ_NEW_QSTR(MP_QSTR_AF_INET6), MP_OBJ_NEW_SMALL_INT(MOD_NETWORK_AF_INET6) },
|
|
|
|
|
|
|
|
{ MP_OBJ_NEW_QSTR(MP_QSTR_SOCK_STREAM), MP_OBJ_NEW_SMALL_INT(MOD_NETWORK_SOCK_STREAM) },
|
|
|
|
{ MP_OBJ_NEW_QSTR(MP_QSTR_SOCK_DGRAM), MP_OBJ_NEW_SMALL_INT(MOD_NETWORK_SOCK_DGRAM) },
|
|
|
|
{ MP_OBJ_NEW_QSTR(MP_QSTR_SOCK_RAW), MP_OBJ_NEW_SMALL_INT(MOD_NETWORK_SOCK_RAW) },
|
2016-03-25 07:06:12 -06:00
|
|
|
|
|
|
|
{ MP_OBJ_NEW_QSTR(MP_QSTR_SOL_SOCKET), MP_OBJ_NEW_SMALL_INT(1) },
|
|
|
|
{ MP_OBJ_NEW_QSTR(MP_QSTR_SO_REUSEADDR), MP_OBJ_NEW_SMALL_INT(SOF_REUSEADDR) },
|
2015-10-11 17:43:33 -06:00
|
|
|
};
|
|
|
|
|
|
|
|
STATIC MP_DEFINE_CONST_DICT(mp_module_lwip_globals, mp_module_lwip_globals_table);
|
|
|
|
|
|
|
|
const mp_obj_module_t mp_module_lwip = {
|
|
|
|
.base = { &mp_type_module },
|
|
|
|
.name = MP_QSTR_lwip,
|
|
|
|
.globals = (mp_obj_dict_t*)&mp_module_lwip_globals,
|
|
|
|
};
|
|
|
|
|
|
|
|
#endif // MICROPY_PY_LWIP
|