4 * This file is part of BeRTOS.
6 * Bertos is free software; you can redistribute it and/or modify
7 * it under the terms of the GNU General Public License as published by
8 * the Free Software Foundation; either version 2 of the License, or
9 * (at your option) any later version.
11 * This program is distributed in the hope that it will be useful,
12 * but WITHOUT ANY WARRANTY; without even the implied warranty of
13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
14 * GNU General Public License for more details.
16 * You should have received a copy of the GNU General Public License
17 * along with this program; if not, write to the Free Software
18 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA
20 * As a special exception, you may use this file as part of a free software
21 * library without restriction. Specifically, if other files instantiate
22 * templates or use macros or inline functions from this file, or you compile
23 * this file and link it with other files to produce an executable, this
24 * file does not by itself cause the resulting executable to be covered by
25 * the GNU General Public License. This exception does not however
26 * invalidate any other reasons why the executable file might be covered by
27 * the GNU General Public License.
29 * Copyright 2012 Develer S.r.l. (http://www.develer.com/)
33 * \brief TCP sockect with kfile interface.
35 * \author Luca Ottaviano <lottaviano@develer.com>
36 * \author Daniele Basile <asterix@develer.com>
41 #include "tcp_socket.h"
43 #define LOG_LEVEL LOG_LVL_INFO
44 #define LOG_FORMAT LOG_FMT_TERSE
46 #include <cpu/byteorder.h>
48 #include <lwip/ip_addr.h>
50 #include <lwip/netif.h>
51 #include <lwip/netbuf.h>
52 #include <lwip/tcpip.h>
55 INLINE int close_socket(TcpSocket *socket)
57 /* Clean all previuos states */
58 netbuf_delete(socket->rx_buf_conn);
59 socket->rx_buf_conn = NULL;
60 socket->remaning_data_len = 0;
66 /* Close socket if was opened */
67 socket->error = netconn_delete(socket->sock);
70 if (socket->error != ERR_OK)
72 LOG_ERR("Closing socket\n");
79 static bool tcpsocket_reconnect(TcpSocket *socket)
81 LOG_INFO("Reconnecting...\n");
83 /* Close socket if was opened */
86 /* Start with new connection */
87 socket->sock = netconn_new(NETCONN_TCP);
90 LOG_ERR("Unabe to alloc new connection\n");
95 socket->error = netconn_bind(socket->sock, socket->local_addr, socket->port);
96 if(socket->error != ERR_OK)
98 LOG_ERR("Connection error\n");
102 socket->error = netconn_connect(socket->sock, socket->remote_addr, socket->port);
103 if(socket->error != ERR_OK)
105 LOG_ERR("Cannot create socket\n");
109 LOG_INFO("connected ip=%d.%d.%d.%d\n", IP_ADDR_TO_INT_TUPLE(socket->remote_addr->addr));
113 netconn_delete(socket->sock);
118 static int tcpsocket_close(KFile *fd)
120 TcpSocket *socket = TCPSOCKET_CAST(fd);
121 return close_socket(socket);
124 static KFile *tcpsocket_reopen(KFile *fd)
126 TcpSocket *socket = TCPSOCKET_CAST(fd);
127 if (tcpsocket_reconnect(socket))
134 * Read data from socket.
136 * The read return the bytes that had been received if they are less than we request too.
137 * Otherwise if the byte that we want read are less that the received bytes, we return only
138 * the requested bytes. To get the remaning bytes we need to make an others read, until the
140 * When there are not any more bytes, a new read takes data from remote socket.
142 static size_t tcpsocket_read(KFile *fd, void *buf, size_t len)
144 TcpSocket *socket = TCPSOCKET_CAST(fd);
147 uint16_t read_len = 0;
149 if (socket->remaning_data_len <= 0)
151 LOG_INFO("No byte left.\n");
152 netbuf_delete(socket->rx_buf_conn);
154 else /* We had byte into buffer use that */
156 LOG_INFO("Read stored bytes.\n");
157 if (!socket->rx_buf_conn)
159 LOG_ERR("Byte stored are corrupted!\n");
160 socket->remaning_data_len = 0;
163 uint16_t tot_data_len = 0;
164 netbuf_data(socket->rx_buf_conn, (void **)&data, &tot_data_len);
168 ASSERT(((int)tot_data_len - (int)socket->remaning_data_len) >= 0);
169 size_t chunk_len = MIN((size_t)(socket->remaning_data_len), len);
170 memcpy((char *)buf, &data[tot_data_len - socket->remaning_data_len], chunk_len);
172 socket->remaning_data_len -= chunk_len;
177 LOG_ERR("No valid data to read\n");
178 socket->remaning_data_len = 0;
179 netbuf_delete(socket->rx_buf_conn);
184 /* Try reconnecting if our socket isn't valid */
185 LOG_INFO("sock[%s]\n", socket->sock ? "valido":"nullo");
186 if (!socket->sock && !tcpsocket_reconnect(socket))
188 LOG_INFO("sock1[%s]\n", socket->sock ? "valido":"nullo");
192 LOG_INFO("Get bytes from socket.\n");
193 socket->rx_buf_conn = netconn_recv(socket->sock);
195 socket->error = netconn_err(socket->sock);
196 if (socket->error != ERR_OK)
198 LOG_ERR("While recv %d\n", socket->error);
199 close_socket(socket);
203 size_t chunk_len = 0;
204 uint16_t data_len = 0;
205 if (socket->rx_buf_conn)
207 netbuf_data(socket->rx_buf_conn, (void **)&data, &data_len);
211 chunk_len = MIN((size_t)data_len, len);
212 memcpy(buf, data, chunk_len);
214 socket->remaning_data_len = data_len - chunk_len;
217 if (socket->remaning_data_len <= 0)
219 netbuf_delete(socket->rx_buf_conn);
220 socket->rx_buf_conn = NULL;
221 socket->remaning_data_len = 0;
227 read_len += chunk_len;
233 static size_t tcpsocket_write(KFile *fd, const void *buf, size_t len)
235 TcpSocket *socket = TCPSOCKET_CAST(fd);
237 /* Try reconnecting if our socket isn't valid */
238 if (!socket->sock && !tcpsocket_reconnect(socket))
241 socket->error = netconn_write(socket->sock, buf, len, NETCONN_COPY);
242 if (socket->error != ERR_OK)
244 LOG_ERR("While writing %d\n", socket->error);
245 close_socket(socket);
252 static int tcpsocket_error(KFile *fd)
254 TcpSocket *socket = TCPSOCKET_CAST(fd);
255 return socket->error;
258 static void tcpsocket_clearerr(KFile *fd)
260 TcpSocket *socket = TCPSOCKET_CAST(fd);
264 void tcpsocket_init(TcpSocket *socket, struct ip_addr *local_addr, struct ip_addr *remote_addr, uint16_t port)
267 socket->local_addr = local_addr;
268 socket->remote_addr = remote_addr;
271 socket->fd._type = KFT_TCPSOCKET;
272 socket->fd.read = tcpsocket_read;
273 socket->fd.error = tcpsocket_error;
274 socket->fd.close = tcpsocket_close;
275 socket->fd.write = tcpsocket_write;
276 socket->fd.clearerr = tcpsocket_clearerr;
277 socket->fd.reopen = tcpsocket_reopen;