aboutsummaryrefslogtreecommitdiff
path: root/firmwares/wifishield/wifiHD/src/ard_tcp.c
diff options
context:
space:
mode:
authorDavid Madison <dmadison@users.noreply.github.com>2019-02-15 12:08:19 -0500
committerDavid Madison <dmadison@users.noreply.github.com>2019-02-15 12:08:31 -0500
commit4a7ef06b516dba4cdda939a16e5458c59592f6b1 (patch)
treeb00e15e665a2f456c1fb207aeb9ac12283181a00 /firmwares/wifishield/wifiHD/src/ard_tcp.c
parent98f00eb533e0ed571b90f8e014d1eb59052a0597 (diff)
Delete 'firmwares' folder
Diffstat (limited to 'firmwares/wifishield/wifiHD/src/ard_tcp.c')
-rw-r--r--firmwares/wifishield/wifiHD/src/ard_tcp.c987
1 files changed, 0 insertions, 987 deletions
diff --git a/firmwares/wifishield/wifiHD/src/ard_tcp.c b/firmwares/wifishield/wifiHD/src/ard_tcp.c
deleted file mode 100644
index 0a73b20..0000000
--- a/firmwares/wifishield/wifiHD/src/ard_tcp.c
+++ /dev/null
@@ -1,987 +0,0 @@
-/*
- * ard_tcp.c
- *
- * Created on: May 27, 2010
- * Author: mlf by Metodo2 srl
- */
-
-//#define _APP_DEBUG_
-#include "lwip/opt.h"
-
-#include "lwip/mem.h"
-#include "lwip/raw.h"
-#include "lwip/icmp.h"
-#include "lwip/netif.h"
-#include "lwip/sys.h"
-#include "lwip/sockets.h"
-#include "lwip/inet.h"
-#include "lwip/inet_chksum.h"
-#include "lwip/tcp.h"
-#include "lwip/udp.h"
-
-#include "ard_tcp.h"
-#include "ard_spi.h"
-#include "timer.h"
-#include "util.h"
-
-#include "getopt.h"
-#include "ard_utils.h"
-#include "debug.h"
-#include "trace.h"
-
-unsigned int startTime = 0;
-extern bool ifStatus;
-
-static err_t tcp_data_sent(void *arg, struct tcp_pcb *pcb, u16_t len);
-
-static void atcp_init_pend_flags(struct ttcp* _ttcp)
-{
- int i = 0;
- for (; i<MAX_CLIENT_ACCEPTED; ++i)
- {
- if (_ttcp) _ttcp->pending_close[i] = false;
- }
-}
-
-/**
- * Clean up and free the ttcp structure
- */
-static void ard_tcp_destroy(struct ttcp* ttcp) {
- err_t err = ERR_OK;
- DUMP_TCP_STATE(ttcp);
-
- uint8_t sock = getSock(ttcp);
- if (sock == -1)
- WARN("ttcp already deallocated!\n");
-
- freeAllTcpData(sock);
- int i = 0;
- for (; i<MAX_CLIENT_ACCEPTED; ++i)
- {
- if (ttcp->tpcb[i]) {
- tcp_arg(ttcp->tpcb[i], NULL);
- tcp_sent(ttcp->tpcb[i], NULL);
- tcp_recv(ttcp->tpcb[i], NULL);
- tcp_err(ttcp->tpcb[i], NULL);
- //TEMPORAQARY
- //err = tcp_close(ttcp->tpcb);
- INFO_TCP("Closing tpcb: state:0x%x err:%d\n", ttcp->tpcb[i]->state, err);
- }
- }
-
- if (ttcp->lpcb) {
- tcp_arg(ttcp->lpcb, NULL);
- tcp_accept(ttcp->lpcb, NULL);
- err = tcp_close(ttcp->lpcb);
- INFO_TCP("Closing lpcb: state:0x%x err:%d\n", ttcp->lpcb->state, err);
- }
-
- if (ttcp->upcb) {
- udp_disconnect(ttcp->upcb);
- udp_remove(ttcp->upcb);
- }
-
- FREE_PAYLOAD(ttcp);
- free(ttcp);
-}
-
-/**
- * Invoked when transfer is done or aborted (non-zero result).
- */
-static void ard_tcp_done(struct ttcp* ttcp, int result) {
-// if (result == 0)
-// ard_tcp_print_stats(ttcp);
-
- if (ttcp->done_cb)
- ttcp->done_cb(ttcp->opaque, result);
-
- ard_tcp_destroy(ttcp);
- clearMapSockTcp(getSock(ttcp), GET_TCP_MODE(ttcp));
-}
-
-/**
- * Only used in TCP mode.
- * Will transmit a maximum of pbuf->tot_len bytes.
- * Called upon connect and when there's space available in the TCP send window
- *
- */
-static err_t tcp_send_data_pcb(struct ttcp *ttcp, struct tcp_pcb *pcb) {
- err_t err = ERR_OK;
- uint32_t len;
-
- GET_CLIENT_ID(ttcp, pcb);
-
- len = ttcp->left[id];
- ttcp->buff_sent[id] = 0;
-
- if (len == 0) return ERR_MEM;
-
- INFO_TCP_VER("left=%d len:%d\n", ttcp->left[id], len);
-
- /* don't send more than we have in the payload */
- if (len > ttcp->buflen)
- len = ttcp->buflen;
-
- /* We cannot send more data than space available in the send
- buffer. */
- if (len > tcp_sndbuf(pcb))
- len = tcp_sndbuf(pcb);
-
- IF_TCP(startTime = timer_get_ms());
- err = tcp_write(pcb, ttcp->payload[id], len, TCP_WRITE_FLAG_COPY);
- if (err != ERR_OK)
- {
- INFO_TCP("tcp_write failed %p state:%d len:%d err:%d\n",
- pcb, pcb->state, len, err);
- ttcp->buff_sent[id] = 0;
- }else{
- ttcp->buff_sent[id] = 1;
- ttcp->left[id] -= len;
- }
-
- return err;
-}
-
-
-/**
- * Only used in TCP mode.
- */
-static err_t tcp_connect_cb(void *arg, struct tcp_pcb *tpcb, err_t err) {
- struct ttcp* _ttcp = arg;
-
- if (_ttcp == NULL) return ERR_ARG;
-
- GET_CLIENT_ID(_ttcp, tpcb);
- INFO_TCP("TTCP [%p-%p]: connect %d %d\n", _ttcp, tpcb, err, tpcb->state);
-
- _connected = ( tpcb->state == ESTABLISHED) ? 1 : 0;
- _ttcp->tcp_poll_retries[id] = 0;
-
- _ttcp->start_time = timer_get_ms();
-
- return ERR_OK;
-}
-
-static void cleanSockState_cb(void *ctx) {
- struct ttcp* _ttcp = ctx;
-
- if (_ttcp == NULL) return;
-
- int sock = getSock(_ttcp);
- if (sock != -1)
- clearMapSockTcp(sock, GET_TCP_MODE(_ttcp));
- INFO_TCP("TTCP [%p]: cleanSockState_cb %d\n", _ttcp, sock);
- _connected = false;
-}
-
-/**
- * Only used in TCP mode.
- */
-
-static err_t close_conn_pcb(struct tcp_pcb* tpcb) {
-
- err_t err = tcp_close(tpcb);
- if (err== ERR_OK)
- {
- tcp_arg(tpcb, NULL);
- tcp_sent(tpcb, NULL);
- tcp_recv(tpcb, NULL);
- }
-
- INFO_TCP("Closing tpcb[%p]: state:0x%x err:%d\n", tpcb, tpcb->state, err);
- return err;
-}
-
-static void atcp_conn_err_cb(void *arg, err_t err) {
- struct ttcp* _ttcp = arg;
-
- WARN("TTCP [%p]: connection error: %d currId:%d\n",
- _ttcp, err, getCurrClientConnId());
-
- if (ifStatus == false)
- printk("Abort connection\n");
-
- if (err == ERR_ABRT)
- {
- removeNewClientConn(_ttcp, GET_CURR_PCB(_ttcp));
- FREE_PAYLOAD_ID(_ttcp, getCurrClientConnId());
- }
-}
-
-static void atcp_conn_cli_err_cb(void *arg, err_t err) {
- struct ttcp* _ttcp = arg;
-
- if (_ttcp == NULL) return;
-
- WARN("TTCP [%p]: connection error: %d arg:%p\n",
- _ttcp, err, arg);
-
- if (ifStatus == false)
- printk("Abort connection\n");
-
- if ((_ttcp)&&(err == ERR_ABRT))
- {
- WARN("TTCP [%p]: free memory\n", _ttcp);
- cleanSockState_cb(_ttcp);
- // TODO
- FREE_PAYLOAD(_ttcp);
- }
-
- //atcp_init_pend_flags(_ttcp);
-}
-
-static err_t close_conn(struct ttcp *_ttcp, struct tcp_pcb* tpcb) {
-
- if (_ttcp == NULL) return ERR_MEM;
-
- GET_CLIENT_ID(_ttcp, tpcb);
-
- err_t err = close_conn_pcb(_ttcp->tpcb[id]);
-
- if (err == ERR_MEM)
- {
- WARN("Cannot close id:%d-%p put pending\n", id, _ttcp->tpcb[id]);
- _ttcp->pending_close[id] = true;
- }
- else{
- _ttcp->pending_close[id] = false;
- removeNewClientConn(_ttcp, _ttcp->tpcb[id]);
- FREE_PAYLOAD_ID(_ttcp, id);
- INFO_TCP("----------------------\n");
- }
- return err;
-}
-
-void closeConnections()
-{
- int ii=0;
- for (; ii<MAX_MODE_NUM; ii++)
- {
- int i = 0;
- for (; i<MAX_SOCK_NUM; i++)
- {
- void* p = getTTCP(i, ii);
- if (p)
- {
- ttcp_t* _ttcp = (ttcp_t* )p;
- if (_ttcp->udp == TCP_MODE)
- {
- ard_tcp_destroy(_ttcp);
- clearMapSockTcp(getSock(_ttcp), GET_TCP_MODE(_ttcp));
- }
- }
- }
- }
-}
-
-/**
- * Only used in TCP mode.
- */
-static err_t atcp_recv_cb(void *arg, struct tcp_pcb *pcb, struct pbuf *p,
- err_t err) {
- struct ttcp* ttcp = arg;
-
- if (err == ERR_OK && p != NULL) {
- DATA_LED_ON();
- /* for print_stats() */
- ttcp->recved += p->tot_len;
-
- if ((ttcp->verbose)||(verboseDebug & INFO_TCP_FLAG)) {
- INFO_TCP("len:%d\n",p->tot_len);
- DUMP_TCP(p->payload, p->tot_len);
- ttcp->print_cnt++;
- }
-
- uint8_t* pBufferStore = insert_pBuf(p, ttcp->sock, (void*) pcb);
- INFO_TCP("sock:%d pcb:%p pbuf:%p err:%d bufStore:%p len:%d\n",
- ttcp->sock, pcb, p, err, pBufferStore, p->tot_len);
- pbuf_free(p);
- DATA_LED_OFF();
- }
-
- /* p will be NULL when remote end is done */
- if (err == ERR_OK && p == NULL) {
- INFO_TCP("atcp_recv_cb p=NULL on sock:%d pcb:%p\n", ttcp->sock, pcb);
- close_conn(ttcp, pcb);
- }
-
- if (err!=ERR_OK)
- WARN("err=%d p=%p\n", err, p);
- return ERR_OK;
-}
-
-void ack_recved(void* pcb, int len) {
- // Comment the call because it is activated on atcp_recv_cb
- INFO_TCP("Received %p len:%d\n", pcb, len);
- tcp_recved(pcb, len);
-}
-
-static err_t atcp_poll(void *arg, struct tcp_pcb *pcb) {
- struct ttcp* _ttcp = arg;
-
- if (_ttcp == NULL) return ERR_ARG;
-
- GET_CLIENT_ID(_ttcp, pcb);
-
- if (_ttcp->left[id]>0)
- ++_ttcp->tcp_poll_retries[id];
-
- if (_ttcp->tcp_poll_retries[id] > 4) {
- WARN("ARD TCP [%p] arg=%p retries=%d abort\n",
- pcb, arg, _ttcp->tcp_poll_retries[id]);
- _ttcp->tcp_poll_retries[id] = 0;
- tcp_abort(pcb);
- _ttcp->pending_close[id] = false;
- return ERR_ABRT;
- }
-
- if (pcb)
- INFO_TCP_POLL("keepAliveCnt:%d keep_idle:%d persist_cnt:%d\n",
- pcb->keep_cnt_sent, pcb->keep_idle, pcb->persist_cnt);
-
- if (_ttcp->left[id] > 0)
- INFO_TCP("ARD TCP [%p-%p] arg=%p retries=%d pend.close:%d len:%d\n",
- (_ttcp)?GET_FIRST_CLIENT_TCP(_ttcp):0, pcb, arg,
- _ttcp->tcp_poll_retries[id], _ttcp->pending_close[id], (_ttcp)?_ttcp->left[id]:0);
- tcp_send_data_pcb(_ttcp, pcb);
-
- if (_ttcp->pending_close[id])
- {
- err_t err = ERR_OK;
- if (id >=0){
- err = tcp_close(pcb);
- if (err == ERR_MEM)
- {
- _ttcp->pending_close[id] = true;
- }
- else
- {
- _ttcp->pending_close[id] = false;
- removeNewClientConn(_ttcp, _ttcp->tpcb[id]);
- FREE_PAYLOAD_ID(_ttcp, id);
- INFO_TCP("----------------------\n");
- }
- }
- INFO_TCP("ARD TCP [%p-%p] try to close pending:%d err:%d id:%d\n", pcb,
- (_ttcp)?GET_FIRST_CLIENT_TCP(_ttcp):0, _ttcp->pending_close[id], err, id);
- }
- return ERR_OK;
-}
-
-static err_t atcp_poll_conn(void *arg, struct tcp_pcb *pcb) {
- struct ttcp* _ttcp = arg;
-
- if (_ttcp == NULL) return ERR_ARG;
-
- GET_CLIENT_ID(_ttcp, pcb)
-
- INFO_TCP_POLL("ARD TCP [%p-%p] arg=%p retries=%d pend.close:%d conn:%d\n",
- (_ttcp)?GET_FIRST_CLIENT_TCP(_ttcp):0, pcb, arg,
- _ttcp->tcp_poll_retries[id], _ttcp->pending_close[id], _connected);
-
- if (id != NO_VALID_ID)
- {
- if (_ttcp->pending_close[id])
- ++(_ttcp->tcp_poll_retries[id]);
- }
-
- if (_ttcp->tcp_poll_retries[id] > 8) {
- WARN("ARD TCP [%p-%p] arg=%p retries=%d\n",
- pcb, GET_FIRST_CLIENT_TCP(_ttcp), arg, _ttcp->tcp_poll_retries[id]);
- _ttcp->tcp_poll_retries[id] = 0;
- tcp_abort(pcb);
- return ERR_ABRT;
- }
-
- if ((_ttcp)&&(_connected)) tcp_send_data_pcb(_ttcp, pcb);
-
- if ((id != NO_VALID_ID) && (_ttcp->pending_close[id]))
- {
- err_t err = tcp_close(pcb);
- if (err == ERR_MEM)
- {
- _ttcp->pending_close[id] = true;
- }
- else
- {
- cleanSockState_cb(_ttcp);
- FREE_PAYLOAD_ID(_ttcp, id);
- _ttcp->pending_close[id] = false;
- }
-
- INFO_TCP("ARD TCP [%p-%p] try to close pending:%d\n", pcb, (_ttcp)?GET_FIRST_CLIENT_TCP(_ttcp):0, _ttcp->pending_close[id]);
- }
- return ERR_OK;
-}
-
-int8_t currConnId = 0;
-
-int8_t getCurrClientConnId() { return currConnId;}
-
-int8_t getNewClientConnId(struct ttcp* _ttcp, struct tcp_pcb *newpcb)
-{
- if (_ttcp != NULL){
- int i = 0;
- for (; i<MAX_CLIENT_ACCEPTED; ++i)
- {
- int idx = GET_IDX_CONN(i);
-
- if (_ttcp->tpcb[idx] == newpcb)
- {
- INFO_TCP_VER("ttcp:%p id=%d, tpcb=%p\n", _ttcp, idx, newpcb);
- return idx;
- }
- }
- }
- WARN("No Valid Id for ttcp:%p pcb:%p\n", _ttcp, newpcb);
- return NO_VALID_ID;
-}
-
-struct tcp_pcb * getFirstClient(struct ttcp* _ttcp, bool verbose)
-{
- if (_ttcp != NULL){
- int i = 0;
- for (; i<MAX_CLIENT_ACCEPTED; ++i)
- {
- int idx = GET_IDX_CONN(i);
- if (_ttcp->tpcb[idx] != NULL)
- {
- if (verbose) INFO_TCP("ttcp:%p id=%d, tpcb=%p\n", _ttcp, idx, _ttcp->tpcb[idx]);
- currConnId = idx;
- return _ttcp->tpcb[idx];
- }
- }
- }
- if (verbose) WARN("No Valid client for ttcp:%p\n", _ttcp);
- return NULL;
-}
-
-
-int8_t setNewClientConn(struct ttcp* _ttcp, struct tcp_pcb *newpcb, uint8_t id)
-{
- if ((_ttcp != NULL)&&(id>=0)&&(id<MAX_CLIENT_ACCEPTED)){
- INFO_TCP("ttcp:%p id=%d, tpcb=%p\n", _ttcp, id, newpcb);
- _ttcp->tpcb[id] = newpcb;
- return id;
- }
- return NO_VALID_ID;
-}
-
-int8_t insertNewClientConn(struct ttcp* _ttcp, struct tcp_pcb *newpcb)
-{
- if (_ttcp != NULL){
- int i = 0;
- for (; i<MAX_CLIENT_ACCEPTED; ++i)
- {
- int idx = GET_IDX_CONN(i);
- if ((_ttcp->tpcb[idx] == NULL)||(_ttcp->tpcb[idx] == newpcb))
- {
- INFO_TCP("ttcp:%p id=%d, tpcb=%p\n", _ttcp, idx, newpcb);
- _ttcp->tpcb[idx] = newpcb;
- return idx;
- }
- }
- }
- return NO_VALID_ID;
-}
-
-int8_t removeNewClientConn(struct ttcp* _ttcp, struct tcp_pcb *newpcb)
-{
- if (_ttcp != NULL){
- int i = 0;
- for (; i<MAX_CLIENT_ACCEPTED; ++i)
- {
- int idx = GET_IDX_CONN(i);
- if (_ttcp->tpcb[idx] == newpcb)
- {
- INFO_TCP("ttcp:%p id=%d, tpcb=%p\n", _ttcp, idx, newpcb);
- _ttcp->tpcb[idx] = NULL;
- return idx;
- }
- }
- }
- return NO_VALID_ID;
-}
-
-bool cleanNewClientConn(struct ttcp* _ttcp)
-{
- if (_ttcp != NULL){
- int i = 0;
- for (; i<MAX_CLIENT_ACCEPTED; ++i)
- _ttcp->tpcb[i] = NULL;
- return true;
- }
- return false;
-}
-
-
-/**
- * Only used in TCP mode.
- */
-static err_t atcp_accept_cb(void *arg, struct tcp_pcb *newpcb, err_t err) {
- struct ttcp* _ttcp = arg;
-
- if (_ttcp == NULL) return ERR_ARG;
-
- INFO_TCP("ARD TCP [%p]: accept new [%p]\n", _ttcp, newpcb);
- INFO_TCP("local:%d remote:%d state:%d\n", newpcb->local_port, newpcb->remote_port, newpcb->state);
-
- int8_t id = insertNewClientConn(_ttcp, newpcb);
-
- ASSERT((_ttcp->payload[id]==NULL), "payload not freed!");
- _ttcp->payload[id] = malloc(_ttcp->buflen);
- INFO_TCP("Alloc payload %d-%p\n", id, _ttcp->payload[id]);
- if (_ttcp->payload[id] == NULL) {
- WARN("TTCP [%p]: could not allocate payload\n", _ttcp);
- return -1;
- }
- tcp_arg(_ttcp->tpcb[id], _ttcp);
- tcp_recv(_ttcp->tpcb[id], atcp_recv_cb);
- tcp_err(_ttcp->tpcb[id], atcp_conn_err_cb);
- tcp_poll(_ttcp->tpcb[id], atcp_poll, 4);
- // Copy the pointer to ttcp also to TRANSMIT mode for the clients connected to the server
- int _sock = getSock(_ttcp);
- if ((_sock != -1)&&(IS_VALID_SOCK(_sock)))
- setMapSockMode(_sock, _ttcp, TTCP_MODE_TRANSMIT);
- _ttcp->start_time = timer_get_ms();
- return ERR_OK;
-}
-
-/**
- * Start TCP transfer.
- */
-static int atcp_start(struct ttcp* ttcp) {
- err_t err = ERR_OK;
-
- struct tcp_pcb * p = tcp_new();
-
- if (p == NULL) {
- WARN("TTCP [%p]: could not allocate pcb\n", ttcp);
- return -1;
- }
-
- currConnId = 0;
- tcp_arg(p, ttcp);
- atcp_init_pend_flags(ttcp);
-
- if (ttcp->mode == TTCP_MODE_TRANSMIT) {
- int8_t id = insertNewClientConn(ttcp, p);
- ttcp->payload[id] = malloc(ttcp->buflen);
- INFO_TCP("Alloc payload %d-%p\n", id, ttcp->payload[id]);
- if (ttcp->payload[id] == NULL) {
- WARN("TTCP [%p]: could not allocate payload\n", ttcp);
- return -1;
- }
-
- struct tcp_pcb * pcb = p;
- tcp_err(pcb, atcp_conn_cli_err_cb);
- tcp_recv(pcb, atcp_recv_cb);
- tcp_sent(pcb, tcp_data_sent);
- tcp_poll(pcb, atcp_poll_conn, 4);
- _connected = false;
- INFO_TCP("[tpcb]-%p payload:%p\n", pcb, ttcp->payload[id]);
- DUMP_TCP_STATE(ttcp);
- if (tcp_connect(pcb, &ttcp->addr, ttcp->port, tcp_connect_cb)
- != ERR_OK) {
- WARN("TTCP [%p]: tcp connect failed\n", ttcp);
- return -1;
- }
-
- } else {
- INFO_TCP("BEFORE BIND ttcp:%p lpcb:%p pcb:%p\n", ttcp, ttcp->lpcb, GET_FIRST_CLIENT_TCP(ttcp));
-
- err = tcp_bind(p, IP_ADDR_ANY, ttcp->port);
- if (err != ERR_OK){
- WARN("TTCP [%p]: bind failed err=%d Port already used\n", ttcp, err);
- return -1;
- }
-
- ttcp->lpcb = tcp_listen(p);
- if (ttcp->lpcb == NULL) {
- WARN("TTCP [%p]: listen failed\n", ttcp);
- return -1;
- }
-
- DUMP_TCP_STATE(ttcp);
- tcp_accept(ttcp->lpcb, atcp_accept_cb);
- }
-
- return 0;
-}
-
-/**
- * Only used in UDP mode. Will finalize the ttcp process when an end marker
- * is seen.
- */
-static void audp_recv_cb(void *arg, struct udp_pcb *upcb, struct pbuf *p,
- struct ip_addr *addr, u16_t port) {
- struct ttcp* ttcp = arg;
-
- /* for print_stats() */
- ttcp->recved += p->tot_len;
- DUMP(p->payload,p->tot_len);
- if (ttcp->verbose) {
- printk(".");
- if (ttcp->print_cnt % 80 == 0)
- printk("\n");
- ttcp->print_cnt++;
- }
- INFO_TCP("UDP Insert %p sock:%d addr:%s port:%d\n", p, ttcp->sock,
- ip2str(*addr), port);
- insert_pBuf(p, ttcp->sock, (void*) upcb);
- setRemoteClient(ttcp->sock, addr->addr, port);
-
- pbuf_free(p);
-}
-
-/**
- * Start UDP transfer.
- */
-static int udp_start(struct ttcp* ttcp) {
- err_t err = ERR_OK;
- ttcp->udp_end_marker_left = 5;
- ttcp->upcb = udp_new();
- if (ttcp->upcb == NULL) {
- WARN("TTCP [%p]: could not allocate pcb\n", ttcp);
- return -1;
- }
-
- INFO_TCP("%s, upcb:%p %s:%d\n", __FUNCTION__, ttcp->upcb, ip2str(ttcp->addr), ttcp->port);
- if (ttcp->mode == TTCP_MODE_TRANSMIT) {
- if (udp_connect(ttcp->upcb, &(ttcp->addr), ttcp->port) != ERR_OK) {
- WARN("TTCP [%p]: udp connect failed\n", ttcp);
- return -1;
- }
- udp_recv(ttcp->upcb, audp_recv_cb, ttcp);
- } else {
- /* bind to any IP address on port specified */
- err = udp_bind(ttcp->upcb, IP_ADDR_ANY, ttcp->port);
- if (err!= ERR_OK) {
- WARN("TTCP [%p]: bind failed err=%d Port already used\n", ttcp, err);
- return -1;
- }
- // clear remote client data
- setRemoteClient(ttcp->sock, 0, 0);
- udp_recv(ttcp->upcb, audp_recv_cb, ttcp);
- }
- INFO_TCP("%s, loc:0x%x-%d rem:0x%x-%d\n", __FUNCTION__,
- ttcp->upcb->local_ip.addr, ttcp->upcb->local_port,
- ttcp->upcb->remote_ip.addr, ttcp->upcb->remote_port);
- return 0;
-}
-
-/**
- * Start a new ttcp transfer. It should be possible to call this function
- * multiple times in order to get multiple ttcp streams. done_cb() will be
- * invoked upon completion.
- *
- */
-int ard_tcp_start(struct ip_addr addr, uint16_t port, void *opaque,
- ard_tcp_done_cb_t *done_cb, int mode, uint16_t nbuf, uint16_t buflen,
- int udp, int verbose, uint8_t sock, void** _ttcp) {
- struct ttcp* ttcp;
- int status;
-
- if (mode != TTCP_MODE_TRANSMIT && mode != TTCP_MODE_RECEIVE) {
- WARN("TTCP [-]: invalid mode\n");
- return -1;
- }
-
- if (nbuf == 0) {
- WARN("TTCP [-]: invalid nbuf\n");
- return -1;
- }
-
- if (buflen == 0) {
- WARN("TTCP [-]: invalid buflen\n");
- return -1;
- }
-
- ttcp = calloc(1, sizeof(struct ttcp));
- if (ttcp == NULL) {
- WARN("TTCP [-]: could not allocate memory for ttcp\n");
- return -1;
- }
-
- ttcp->addr = addr;
- ttcp->port = port;
- ttcp->nbuf = nbuf;
- ttcp->mode = mode;
- ttcp->done_cb = done_cb;
- ttcp->opaque = opaque;
- ttcp->udp = udp;
- ttcp->verbose = verbose;
- ttcp->buflen = buflen;
- cleanNewClientConn(ttcp);
-
- if (ttcp->udp)
- status = udp_start(ttcp);
- else
- status = atcp_start(ttcp);
-
- if (status) {
- WARN("Start server FAILED!\n");
- goto fail;
- }
- INFO_TCP("TTCP [%p-%p]: nbuf=%d, buflen=%d, port=%d (%s/%s)\n", ttcp,
- ((ttcp->udp==1)?(void*)ttcp->upcb:GET_FIRST_CLIENT_TCP(ttcp)), ttcp->nbuf, ttcp->buflen,
- ttcp->port, ProtMode2Str(ttcp->udp), Mode2Str(ttcp->mode));
-
- *_ttcp = (void*) ttcp;
- ttcp->sock = sock;
-
- return 0;
-
- fail: ard_tcp_destroy(ttcp);
- return -1;
-}
-
-void ard_tcp_stop(void* ttcp) {
- struct ttcp* _ttcp = (struct ttcp*) ttcp;
- if (_ttcp == NULL)
- {
- WARN("ttcp = NULL!\n");
- return;
- }
- if (_ttcp->mode == TTCP_MODE_TRANSMIT) {
- int i = getCurrClientConnId();
- ard_tcp_destroy(_ttcp);
- clearMapSockTcp(getSock(_ttcp), GET_TCP_MODE(_ttcp));
- _ttcp->tcp_poll_retries[i] = 0;
- }else{
- DUMP_TCP_STATE(_ttcp);
-
- int i = getCurrClientConnId();
- if ((_ttcp)&&(_ttcp->tpcb[i])&&(_ttcp->tpcb[i]->state!=LAST_ACK)&&(_ttcp->tpcb[i]->state!=CLOSED))
- {
- // Flush all the data
- err_t err=tcp_output(_ttcp->tpcb[i]);
- INFO_TCP("flush data: tpcb:%p err:%d\n", _ttcp->tpcb[i], err);
- // if any socket cannot be close stop the close connection
- close_conn(_ttcp, _ttcp->tpcb[i]);
- }
- }
-}
-
-uint8_t getStateTcp(void* p, bool client) {
- struct ttcp* _ttcp = (struct ttcp*) p;
-
- if (ifStatus == false)
- return CLOSED;
- struct tcp_pcb * pcb = GET_FIRST_CLIENT_TCP_NV(_ttcp);
- if ((_ttcp != NULL) && ((pcb != NULL) || (client==0))) {
- IF_SPI_POLL(DUMP_TCP_STATE(_ttcp));
- if (client)
- {
- if ((pcb->state != ESTABLISHED)&&(pcb->state != CLOSED))
- DUMP_TCP_STATE(_ttcp);
- return pcb->state;
- }
- else
- {
- return _ttcp->lpcb->state;
- }
- } else {
- WARN_POLL("TCP not initialized ttcp:%p tpcb:%p lpcb:%p\n",
- _ttcp, ((_ttcp)?pcb:0), ((_ttcp)?_ttcp->lpcb:0));
- }
- return CLOSED;
-}
-
-uint8_t getModeTcp(void* p) {
- struct ttcp* _ttcp = (struct ttcp*) p;
-
- if (_ttcp != NULL)
- return _ttcp->mode;
- return 0;
-}
-
-uint8_t isDataSent(void* p) {
- struct ttcp *_ttcp = (struct ttcp *)p;
-
- int8_t id = getCurrClientConnId();
- if ((_ttcp)&&(!_ttcp->buff_sent[id]))
- {
- return 0;
- }
-
- return 1;
-}
-
-static err_t tcp_data_sent(void *arg, struct tcp_pcb *pcb, u16_t len) {
- struct ttcp *_ttcp;
-
- LWIP_UNUSED_ARG(len);
-
- _ttcp = arg;
-
- if (_ttcp == NULL) return ERR_ARG;
-
- GET_CLIENT_ID(_ttcp, pcb);
- _ttcp->tcp_poll_retries[id] = 0;
- _ttcp->buff_sent[id] = 1;
-
- INFO_TCP("Packet sent pcb:%p len:%d dur:%d left:%d\n", pcb, len, timer_get_ms() - startTime,
- (_ttcp)?(_ttcp->left[id]):0);
-
- if ((_ttcp)&&(_ttcp->left[id] > 0)) {
- tcp_send_data_pcb(_ttcp, pcb);
- }
-
- return ERR_OK;
-}
-
-int sendTcpData(void* p, uint8_t* buf, uint16_t len)
-{
- struct ttcp* _ttcp = (struct ttcp*) p;
-
- if (_ttcp==NULL)
- {
- WARN("ttcp == NULL!\n");
- return WL_FAILURE;
- }
-
- struct tcp_pcb * pcb = GET_FIRST_CLIENT_TCP_NV(_ttcp);
- GET_CLIENT_ID(_ttcp, pcb);
-
- INFO_TCP_VER("ttcp:%p pcb:%p buf:%p len:%d\n", _ttcp, pcb, buf, len);
- DUMP_TCP(buf,len);
- IF_TCP_VER(DUMP_TCP_STATE(_ttcp));
-
- if ((_ttcp != NULL) && (pcb != NULL) &&
- (buf != NULL) && (len != 0) && (_ttcp->payload[id] != NULL)) {
- if (pcb->state == ESTABLISHED || pcb->state == CLOSE_WAIT ||
- pcb->state == SYN_SENT || pcb->state == SYN_RCVD) {
-
- memcpy(_ttcp->payload[id], buf, len);
- _ttcp->payload[id][len]='\0';
- INFO_TCP_VER("'%s'\n", _ttcp->payload[id]);
- _ttcp->left[id] = len;
- tcp_sent(pcb, tcp_data_sent);
- tcp_send_data_pcb(_ttcp, pcb);
-
- return WL_SUCCESS;
- }
- }
- //printk("Write failure _ttcp=%p _ttcp->tpcb=%p buf=%p len=%d\n", _ttcp, _ttcp->tpcb, buf, len);
- return WL_FAILURE;
-}
-
-int sendUdpData(void* ttcp, uint8_t* buf, uint16_t len) {
- struct ttcp* _ttcp = (struct ttcp*) ttcp;
- if ((_ttcp != NULL) && (buf != NULL) && (len != 0))
- {
- INFO_TCP("buf:%p len:%d\n", buf, len);
- DUMP_TCP(buf,len);
- }else{
- return WL_FAILURE;
- }
-
- struct pbuf* p = pbuf_alloc(PBUF_TRANSPORT, len, PBUF_RAM);
- if (p == NULL) {
- WARN("TTCP [%p]: could not allocate pbuf\n", ttcp);
- return WL_FAILURE;
- }
- memcpy(p->payload, buf, len);
- if (udp_send(_ttcp->upcb, p) != ERR_OK) {
- WARN("TTCP [%p]: udp_send() failed\n", _ttcp);
- pbuf_free(p);
- return WL_FAILURE;
- }
-
- pbuf_free(p);
- return WL_SUCCESS;
-}
-
-
-
-char
- usage[] =
- "Usage: ttcp -t/-r [-options] host\n\
- -l length of bufs written to network (default 1024)\n\
- -n number of bufs written to network (default 1024)\n\
- -p port number to send to (default 2000)\n\
- -u udp\n\
- -v verbose\n";
-
-/**
- *
- */
-cmd_state_t cmd_ttcp(int argc, char* argv[], void* ctx) {
-
- int c;
- int mode = TTCP_MODE_TRANSMIT;
- int verbose = 0;
- uint16_t buflen = 1024;
- uint16_t nbuf = 1024;
- uint16_t port = 2000;
- int udp = 0;
- struct ip_addr addr = { 0 };
-
- optind = 1;
- while ((c = getopt(argc, argv, "utrl:n:p:v")) != -1) {
- switch (c) {
- case 't':
- mode = TTCP_MODE_TRANSMIT;
- break;
- case 'r':
- mode = TTCP_MODE_RECEIVE;
- break;
- case 'l':
- buflen = atoi(optarg);
- break;
- case 'v':
- verbose = 1;
- break;
- case 'n':
- nbuf = atoi(optarg);
- break;
- case 'u':
- udp = 1;
- break;
- case 'p':
- port = atoi(optarg);
- break;
- }
- }
-
- if (mode == TTCP_MODE_TRANSMIT) {
- if (optind >= argc) {
- printk("%s", usage);
- return CMD_DONE;
- }
-
- addr = str2ip(argv[optind]);
- if (!addr.addr) {
- printk("%s", usage);
- return CMD_DONE;
- }
- }
- void* _ttcp = NULL;
- if (ard_tcp_start(addr, port, NULL, NULL, mode, nbuf, buflen, udp, verbose,
- 0, &_ttcp))
- return CMD_DONE;
-
- return CMD_DONE;
-}
-
-
-#if 0
-#include "lwip/sockets.h"
-
-void testlwip()
-{
- int Sock;
- fd_set fdsetR;
- FD_ZERO(&fdsetR);
- FD_SET(Sock, &fdsetR);
- fd_set fdsetE = fdsetR;
-
- int rc;
- const int cMillies = 10000;
- struct timeval timeout;
- timeout.tv_sec = cMillies / 1000;
- timeout.tv_usec = (cMillies % 1000) * 1000;
- //rc = lwip_select(Sock + 1, &fdsetR, NULL, &fdsetE, &timeout);
-}
-#endif