aboutsummaryrefslogtreecommitdiff
path: root/firmwares/wifishield/wifiHD/src/ard_tcp.h
blob: 7d0bf7bd44ab3783c7555676cdda57ce7fa418a7 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
/*
 * ard_tcp.h
 *
 *  Created on: May 27, 2010
 *      Author: mlf by Metodo2 srl
 */

#ifndef ARD_TCP_H
#define ARD_TCP_H

#include "console.h"
#include "lwip/tcp.h"

typedef void (ard_tcp_done_cb_t)(void *opaque, int result);

#define TTCP_MODE_TRANSMIT          0
#define TTCP_MODE_RECEIVE           1
#define MAX_MODE_NUM 				2

#define GET_TCP_MODE(X)	((X!=NULL)?((struct ttcp*)(X))->mode:0)
#define IS_VALID_SOCK(SOCK) ((SOCK>=0)&&(SOCK<MAX_SOCK_NUM))
#define IS_UDP_SOCK(SOCK)	((getTTCP(SOCK, TTCP_MODE_RECEIVE)!=NULL)?((struct ttcp*)(getTTCP(SOCK, TTCP_MODE_RECEIVE)))->udp:0)

// Maximum number of client connection accepted by server
#define MAX_CLIENT_ACCEPTED			4
#define NO_VALID_ID					-1
#define GET_FIRST_CLIENT_TCP(TTCP)	((TTCP!=NULL)?TTCP->tpcb[0] : NULL)
#define GET_CLIENT_TCP(TTCP,ID)	(((TTCP!=NULL)&&(ID>=0)&&(ID<MAX_CLIENT_ACCEPTED))?TTCP->tpcb[ID] : NULL)


typedef struct ttcp {

	/* options */
	struct ip_addr addr; /* host */
	uint16_t port; /* -p */
	uint16_t nbuf; /* -n */
	int mode; /* -t */
	int verbose; /* -v */
	int udp; /* -u */
	uint8_t sock;
	uint8_t buff_sent;

	/* common */
	uint16_t print_cnt;
	uint32_t start_time;
	uint32_t left;
	uint32_t recved;
	ard_tcp_done_cb_t* done_cb;
	void* opaque;
	uint32_t buflen; /* -l */
	uint32_t tid;

	/* TCP specific */
	struct tcp_pcb* tpcb[MAX_CLIENT_ACCEPTED];
	struct tcp_pcb* lpcb;
	char* payload;
	uint8_t tcp_poll_retries;
	bool pending_close[MAX_CLIENT_ACCEPTED];

	/* UDP specific */
	int udp_started;
	uint16_t udp_end_marker_left;
	struct udp_pcb* upcb;
}ttcp_t;

bool _connected;

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);

void ard_tcp_stop(void* ttcp);

uint8_t getStateTcp(void* p, bool client );

uint8_t getModeTcp(void* p);

int sendTcpData(void* p, uint8_t* buf, uint16_t len);

int sendUdpData(void* p, uint8_t* buf, uint16_t len);

uint8_t isDataSent(void* p );

cmd_state_t cmd_ttcp(int argc, char* argv[], void* ctx);

int8_t setNewClientConn(struct ttcp* _ttcp, struct tcp_pcb *newpcb, uint8_t id);

int8_t insertNewClientConn(struct ttcp* _ttcp, struct tcp_pcb *newpcb);

int8_t removeNewClientConn(struct ttcp* _ttcp, struct tcp_pcb *newpcb);

bool cleanNewClientConn(struct ttcp* _ttcp);

int8_t getNewClientConnId(struct ttcp* _ttcp, struct tcp_pcb *newpcb);

void closeConnections();

#endif