at_device/class/bc28/at_socket_bc28.c

774 lines
23 KiB
C

/*
* File : at_socket_bc28.c
* This file is part of RT-Thread RTOS
* Copyright (c) 2020, RudyLo <luhuadong@163.com>
*
* This program is free software; you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation; either version 2 of the License, or
* (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License along
* with this program; if not, write to the Free Software Foundation, Inc.,
* 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
*
* Change Logs:
* Date Author Notes
* 2020-02-13 luhuadong first version
*/
#include <stdio.h>
#include <string.h>
#include <at_device_bc28.h>
#define LOG_TAG "at.skt.bc28"
#include <at_log.h>
#if defined(AT_DEVICE_USING_BC28) && defined(AT_USING_SOCKET)
#define BC28_MODULE_SEND_MAX_SIZE 1024
/* set real event by current socket and current state */
#define SET_EVENT(socket, event) (((socket + 1) << 16) | (event))
/* AT socket event type */
#define BC28_EVENT_CONN_OK (1L << 0)
#define BC28_EVENT_SEND_OK (1L << 1)
#define BC28_EVENT_RECV_OK (1L << 2)
#define BC28_EVNET_CLOSE_OK (1L << 3)
#define BC28_EVENT_CONN_FAIL (1L << 4)
#define BC28_EVENT_SEND_FAIL (1L << 5)
#define BC28_EVENT_DOMAIN_OK (1L << 6)
static at_evt_cb_t at_evt_cb_set[] = {
[AT_SOCKET_EVT_RECV] = NULL,
[AT_SOCKET_EVT_CLOSED] = NULL,
};
static void at_tcp_ip_errcode_parse(int result)//TCP/IP_QIGETERROR
{
switch(result)
{
case 0 : LOG_D("%d : Operation successful", result); break;
case 550 : LOG_E("%d : Unknown error", result); break;
case 551 : LOG_E("%d : Operation blocked", result); break;
case 552 : LOG_E("%d : Invalid parameters", result); break;
case 553 : LOG_E("%d : Memory not enough", result); break;
case 554 : LOG_E("%d : Create socket failed", result); break;
case 555 : LOG_E("%d : Operation not supported", result); break;
case 556 : LOG_E("%d : Socket bind failed", result); break;
case 557 : LOG_E("%d : Socket listen failed", result); break;
case 558 : LOG_E("%d : Socket write failed", result); break;
case 559 : LOG_E("%d : Socket read failed", result); break;
case 560 : LOG_E("%d : Socket accept failed", result); break;
case 561 : LOG_E("%d : Open PDP context failed", result); break;
case 562 : LOG_E("%d : Close PDP context failed", result); break;
case 563 : LOG_W("%d : Socket identity has been used", result); break;
case 564 : LOG_E("%d : DNS busy", result); break;
case 565 : LOG_E("%d : DNS parse failed", result); break;
case 566 : LOG_E("%d : Socket connect failed", result); break;
// case 567 : LOG_W("%d : Socket has been closed", result); break;
case 567 : break;
case 568 : LOG_E("%d : Operation busy", result); break;
case 569 : LOG_E("%d : Operation timeout", result); break;
case 570 : LOG_E("%d : PDP context broken down", result); break;
case 571 : LOG_E("%d : Cancel send", result); break;
case 572 : LOG_E("%d : Operation not allowed", result); break;
case 573 : LOG_E("%d : APN not configured", result); break;
case 574 : LOG_E("%d : Port busy", result); break;
default : LOG_E("%d : Unknown err code", result); break;
}
}
static int bc28_socket_event_send(struct at_device *device, uint32_t event)
{
return (int) rt_event_send(device->socket_event, event);
}
static int bc28_socket_event_recv(struct at_device *device, uint32_t event, uint32_t timeout, rt_uint8_t option)
{
int result = RT_EOK;
rt_uint32_t recved;
result = rt_event_recv(device->socket_event, event, option | RT_EVENT_FLAG_CLEAR, timeout, &recved);
if (result != RT_EOK)
{
return -RT_ETIMEOUT;
}
return recved;
}
/**
* close socket by AT commands.
*
* @param current socket
*
* @return 0: close socket success
* -1: send AT commands error
* -2: wait socket event timeout
* -5: no memory
*/
static int bc28_socket_close(struct at_socket *socket)
{
int result = RT_EOK;
at_response_t resp = RT_NULL;
int device_socket = (int) socket->user_data;
struct at_device *device = (struct at_device *) socket->device;
resp = at_create_resp(64, 0, rt_tick_from_millisecond(300));
if (resp == RT_NULL)
{
LOG_E("no memory for resp create.");
return -RT_ENOMEM;
}
result = at_obj_exec_cmd(device->client, resp, "AT+NSOCL=%d", device_socket);
at_delete_resp(resp);
return result;
}
/**
* create TCP/UDP client or server connect by AT commands.
*
* @param socket current socket
* @param ip server or client IP address
* @param port server or client port
* @param type connect socket type(tcp, udp)
* @param is_client connection is client
*
* @return 0: connect success
* -1: connect failed, send commands error or type error
* -2: wait socket event timeout
* -5: no memory
*/
static int bc28_socket_connect(struct at_socket *socket, char *ip, int32_t port,
enum at_socket_type type, rt_bool_t is_client)
{
#define CONN_RETRY 2
int i = 0;
const char *type_str = RT_NULL;
uint32_t event = 0;
at_response_t resp = RT_NULL;
int result = 0, event_result = 0;
int device_socket = (int) socket->user_data;
struct at_device *device = (struct at_device *) socket->device;
RT_ASSERT(ip);
RT_ASSERT(port >= 0);
if ( ! is_client)
{
return -RT_ERROR;
}
if (type != AT_SOCKET_TCP)
{
return -RT_ERROR;
}
resp = at_create_resp(128, 0, rt_tick_from_millisecond(300));
if (resp == RT_NULL)
{
LOG_E("no memory for resp create.");
return -RT_ENOMEM;
}
for(i=0; i<CONN_RETRY; i++)
{
/* clear socket connect event */
event = SET_EVENT(device_socket, BC28_EVENT_CONN_OK | BC28_EVENT_CONN_FAIL);
bc28_socket_event_recv(device, event, 0, RT_EVENT_FLAG_OR);
if (at_obj_exec_cmd(device->client, resp, "AT+NSOCO=%d,%s,%d",
device_socket, ip, port) < 0)
{
result = -RT_ERROR;
break;
}
/* waiting result event from AT URC, the device default connection timeout is 60 seconds*/
if (bc28_socket_event_recv(device, SET_EVENT(device_socket, 0),
60 * RT_TICK_PER_SECOND, RT_EVENT_FLAG_OR) < 0)
{
LOG_E("%s device socket(%d) wait connect result timeout.", device->name, device_socket);
result = -RT_ETIMEOUT;
break;
}
/* waiting OK or failed result */
event_result = bc28_socket_event_recv(device, BC28_EVENT_CONN_OK | BC28_EVENT_CONN_FAIL,
1 * RT_TICK_PER_SECOND, RT_EVENT_FLAG_OR);
if (event_result < 0)
{
LOG_E("%s device socket(%d) wait connect OK|FAIL timeout.", device->name, device_socket);
result = -RT_ETIMEOUT;
break;
}
/* check result */
if (event_result & BC28_EVENT_CONN_OK)
{
result = RT_EOK;
break;
}
LOG_D("%s device socket(%d) connect failed, the socket was not be closed and now will connect retry.",
device->name, device_socket);
if (bc28_socket_close(socket) < 0)
{
result = -RT_ERROR;
break;
}
}
if (i == CONN_RETRY)
{
LOG_E("%s device socket(%d) connect failed.", device->name, device_socket);
result = -RT_ERROR;
}
if (resp)
{
at_delete_resp(resp);
}
return result;
}
static int at_get_send_size(struct at_socket *socket, size_t *size, size_t *acked, size_t *nacked)
{
int result = 0;
at_response_t resp = RT_NULL;
int device_socket = (int) socket->user_data;
struct at_device *device = (struct at_device *) socket->device;
resp = at_create_resp(128, 0, rt_tick_from_millisecond(300));
if (resp == RT_NULL)
{
LOG_E("no memory for resp create.", device->name);
return -RT_ENOMEM;
}
if (at_obj_exec_cmd(device->client, resp, "AT+QISEND=%d,0", device_socket) < 0)
{
result = -RT_ERROR;
goto __exit;
}
if (at_resp_parse_line_args_by_kw(resp, "+QISEND:", "+QISEND: %d,%d,%d", size, acked, nacked) <= 0)
{
result = -RT_ERROR;
goto __exit;
}
__exit:
if (resp)
{
at_delete_resp(resp);
}
return result;
}
static int at_wait_send_finish(struct at_socket *socket, size_t settings_size)
{
/* get the timeout by the input data size */
rt_tick_t timeout = rt_tick_from_millisecond(settings_size);
rt_tick_t last_time = rt_tick_get();
size_t size = 0, acked = 0, nacked = 0xFFFF;
while (rt_tick_get() - last_time <= timeout)
{
at_get_send_size(socket, &size, &acked, &nacked);
if (nacked == 0)
{
return RT_EOK;
}
rt_thread_mdelay(100);
}
return -RT_ETIMEOUT;
}
/**
* send data to server or client by AT commands.
*
* @param socket current socket
* @param buff send buffer
* @param bfsz send buffer size
* @param type connect socket type(tcp, udp)
*
* @return >=0: the size of send success
* -1: send AT commands error or send data error
* -2: waited socket event timeout
* -5: no memory
*/
static int bc28_socket_send(struct at_socket *socket, const char *buff, size_t bfsz, enum at_socket_type type)
{
uint32_t event = 0;
int result = 0, event_result = 0;
size_t cur_pkt_size = 0, sent_size = 0;
at_response_t resp = RT_NULL;
int device_socket = (int) socket->user_data;
struct at_device *device = (struct at_device *) socket->device;
struct at_device_bc28 *bc28 = (struct at_device_bc28 *) device->user_data;
rt_mutex_t lock = device->client->lock;
RT_ASSERT(buff);
resp = at_create_resp(128, 2, rt_tick_from_millisecond(300));
if (resp == RT_NULL)
{
LOG_E("no memory for resp create.");
return -RT_ENOMEM;
}
rt_mutex_take(lock, RT_WAITING_FOREVER);
/* set current socket for send URC event */
bc28->user_data = (void *) device_socket;
/* clear socket send event */
event = SET_EVENT(device_socket, BC28_EVENT_SEND_OK | BC28_EVENT_SEND_FAIL);
bc28_socket_event_recv(device, event, 0, RT_EVENT_FLAG_OR);
/* set AT client end sign to deal with '>' sign.*/
at_obj_set_end_sign(device->client, '>');
while (sent_size < bfsz)
{
if (bfsz - sent_size < BC28_MODULE_SEND_MAX_SIZE)
{
cur_pkt_size = bfsz - sent_size;
}
else
{
cur_pkt_size = BC28_MODULE_SEND_MAX_SIZE;
}
/* send the "AT+QISEND" commands to AT server than receive the '>' response on the first line. */
if (at_obj_exec_cmd(device->client, resp, "AT+QISEND=%d,%d", device_socket, (int)cur_pkt_size) < 0)
{
result = -RT_ERROR;
goto __exit;
}
rt_thread_mdelay(5);//delay at least 4ms
/* send the real data to server or client */
result = (int) at_client_obj_send(device->client, buff + sent_size, cur_pkt_size);
if (result == 0)
{
result = -RT_ERROR;
goto __exit;
}
/* waiting result event from AT URC */
event = SET_EVENT(device_socket, 0);
event_result = bc28_socket_event_recv(device, event, 10 * RT_TICK_PER_SECOND, RT_EVENT_FLAG_OR);
if (event_result < 0)
{
LOG_E("%s device socket(%d) wait event timeout.", device->name, device_socket);
result = -RT_ETIMEOUT;
goto __exit;
}
/* waiting OK or failed result */
event = BC28_EVENT_SEND_OK | BC28_EVENT_SEND_FAIL;
event_result = bc28_socket_event_recv(device, event, 1 * RT_TICK_PER_SECOND, RT_EVENT_FLAG_OR);
if (event_result < 0)
{
LOG_E("%s device socket(%d) wait sned OK|FAIL timeout.", device->name, device_socket);
result = -RT_ETIMEOUT;
goto __exit;
}
/* check result */
if (event_result & BC28_EVENT_SEND_FAIL)
{
LOG_E("%s device socket(%d) send failed.", device->name, device_socket);
result = -RT_ERROR;
goto __exit;
}
if (type == AT_SOCKET_TCP)
{
at_wait_send_finish(socket, 2*cur_pkt_size);
}
sent_size += cur_pkt_size;
}
__exit:
/* reset the end sign for data conflict */
at_obj_set_end_sign(device->client, 0);
rt_mutex_release(lock);
if (resp)
{
at_delete_resp(resp);
}
return result > 0 ? sent_size : result;
}
/**
* domain resolve by AT commands.
*
* @param name domain name
* @param ip parsed IP address, it's length must be 16
*
* @return 0: domain resolve success
* -1: send AT commands error or response error
* -2: wait socket event timeout
* -5: no memory
*/
int bc28_domain_resolve(const char *name, char ip[16])
{
#define RESOLVE_RETRY 3
int i, result;
at_response_t resp = RT_NULL;
struct at_device *device = RT_NULL;
struct at_device_bc28 *bc28 = RT_NULL;
RT_ASSERT(name);
RT_ASSERT(ip);
device = at_device_get_first_initialized();
if (device == RT_NULL)
{
LOG_E("get first init device failed.");
return -RT_ERROR;
}
/* the maximum response time is 60 seconds, but it set to 10 seconds is convenient to use. */
resp = at_create_resp(128, 0, rt_tick_from_millisecond(300));
if (!resp)
{
LOG_E("no memory for resp create.");
return -RT_ENOMEM;
}
/* clear BC28_EVENT_DOMAIN_OK */
bc28_socket_event_recv(device, BC28_EVENT_DOMAIN_OK, 0, RT_EVENT_FLAG_OR);
bc28 = (struct at_device_bc28 *) device->user_data;
bc28->socket_data = ip;
if (at_obj_exec_cmd(device->client, resp, "AT+QDNS=0,%s", name) != RT_EOK)
{
result = -RT_ERROR;
goto __exit;
}
for(i = 0; i < RESOLVE_RETRY; i++)
{
/* waiting result event from AT URC, the device default connection timeout is 30 seconds.*/
if (bc28_socket_event_recv(device, BC28_EVENT_DOMAIN_OK, 10 * RT_TICK_PER_SECOND, RT_EVENT_FLAG_OR) < 0)
{
result = -RT_ETIMEOUT;
continue;
}
else
{
if (rt_strlen(ip) < 8)
{
rt_thread_mdelay(100);
/* resolve failed, maybe receive an URC CRLF */
result = -RT_ERROR;
continue;
}
else
{
result = RT_EOK;
break;
}
}
}
__exit:
bc28->socket_data = RT_NULL;
if (resp)
{
at_delete_resp(resp);
}
return result;
}
/**
* set AT socket event notice callback
*
* @param event notice event
* @param cb notice callback
*/
static void bc28_socket_set_event_cb(at_socket_evt_t event, at_evt_cb_t cb)
{
if (event < sizeof(at_evt_cb_set) / sizeof(at_evt_cb_set[1]))
{
at_evt_cb_set[event] = cb;
}
}
static void urc_connect_func(struct at_client *client, const char *data, rt_size_t size)
{
int device_socket = 0, result = 0;
struct at_device *device = RT_NULL;
char *client_name = client->device->parent.name;
RT_ASSERT(data && size);
device = at_device_get_by_name(AT_DEVICE_NAMETYPE_CLIENT, client_name);
if (device == RT_NULL)
{
LOG_E("get device(%s) failed.", client_name);
return;
}
sscanf(data, "+QIOPEN: %d,%d", &device_socket , &result);
if (result == 0)
{
bc28_socket_event_send(device, SET_EVENT(device_socket, BC28_EVENT_CONN_OK));
}
else
{
at_tcp_ip_errcode_parse(result);
bc28_socket_event_send(device, SET_EVENT(device_socket, BC28_EVENT_CONN_FAIL));
}
}
static void urc_send_func(struct at_client *client, const char *data, rt_size_t size)
{
int device_socket = 0;
struct at_device *device = RT_NULL;
struct at_device_bc28 *bc28 = RT_NULL;
char *client_name = client->device->parent.name;
RT_ASSERT(data && size);
device = at_device_get_by_name(AT_DEVICE_NAMETYPE_CLIENT, client_name);
if (device == RT_NULL)
{
LOG_E("get device(%s) failed.", client_name);
return;
}
bc28 = (struct at_device_bc28 *) device->user_data;
device_socket = (int) bc28->user_data;
if (rt_strstr(data, "SEND OK"))
{
bc28_socket_event_send(device, SET_EVENT(device_socket, BC28_EVENT_SEND_OK));
}
else if (rt_strstr(data, "SEND FAIL"))
{
bc28_socket_event_send(device, SET_EVENT(device_socket, BC28_EVENT_SEND_FAIL));
}
}
static void urc_close_func(struct at_client *client, const char *data, rt_size_t size)
{
int device_socket = 0;
struct at_socket *socket = RT_NULL;
struct at_device *device = RT_NULL;
char *client_name = client->device->parent.name;
RT_ASSERT(data && size);
device = at_device_get_by_name(AT_DEVICE_NAMETYPE_CLIENT, client_name);
if (device == RT_NULL)
{
LOG_E("get device(%s) failed.", client_name);
return;
}
sscanf(data, "+QIURC: \"closed\",%d", &device_socket);
/* get at socket object by device socket descriptor */
socket = &(device->sockets[device_socket]);
/* notice the socket is disconnect by remote */
if (at_evt_cb_set[AT_SOCKET_EVT_CLOSED])
{
at_evt_cb_set[AT_SOCKET_EVT_CLOSED](socket, AT_SOCKET_EVT_CLOSED, NULL, 0);
}
}
static void urc_recv_func(struct at_client *client, const char *data, rt_size_t size)
{
int device_socket = 0;
rt_int32_t timeout;
rt_size_t bfsz = 0, temp_size = 0;
char *recv_buf = RT_NULL, temp[8] = {0};
struct at_socket *socket = RT_NULL;
struct at_device *device = RT_NULL;
char *client_name = client->device->parent.name;
RT_ASSERT(data && size);
device = at_device_get_by_name(AT_DEVICE_NAMETYPE_CLIENT, client_name);
if (device == RT_NULL)
{
LOG_E("get device(%s) failed.", client_name);
return;
}
/* get the current socket and receive buffer size by receive data */
sscanf(data, "+QIURC: \"recv\",%d,%d", &device_socket, (int *) &bfsz);
/* set receive timeout by receive buffer length, not less than 10 ms */
timeout = bfsz > 10 ? bfsz : 10;
if (device_socket < 0 || bfsz == 0)
{
return;
}
recv_buf = (char *) rt_calloc(1, bfsz);
if (recv_buf == RT_NULL)
{
LOG_E("no memory for URC receive buffer(%d).", bfsz);
/* read and clean the coming data */
while (temp_size < bfsz)
{
if (bfsz - temp_size > sizeof(temp))
{
at_client_obj_recv(client, temp, sizeof(temp), timeout);
}
else
{
at_client_obj_recv(client, temp, bfsz - temp_size, timeout);
}
temp_size += sizeof(temp);
}
return;
}
/* sync receive data */
if (at_client_obj_recv(client, recv_buf, bfsz, timeout) != bfsz)
{
LOG_E("%s device receive size(%d) data failed.", device->name, bfsz);
rt_free(recv_buf);
return;
}
/* get at socket object by device socket descriptor */
socket = &(device->sockets[device_socket]);
/* notice the receive buffer and buffer size */
if (at_evt_cb_set[AT_SOCKET_EVT_RECV])
{
at_evt_cb_set[AT_SOCKET_EVT_RECV](socket, AT_SOCKET_EVT_RECV, recv_buf, bfsz);
}
}
static void urc_dnsqip_func(struct at_client *client, const char *data, rt_size_t size)
{
int i = 0, j = 0;
char recv_ip[16] = {0};
int result, ip_count, dns_ttl;
struct at_device *device = RT_NULL;
struct at_device_bc28 *bc28 = RT_NULL;
char *client_name = client->device->parent.name;
RT_ASSERT(data && size);
device = at_device_get_by_name(AT_DEVICE_NAMETYPE_CLIENT, client_name);
if (device == RT_NULL)
{
LOG_E("get device(%s) failed.", client_name);
return;
}
bc28 = (struct at_device_bc28 *) device->user_data;
if (bc28->socket_data == RT_NULL)
{
LOG_D("%s device socket_data no config.", bc28->device_name);
return;
}
sscanf(data, "+QDNS:%s", recv_ip);
recv_ip[15] = '\0';
rt_memcpy(bc28->socket_data, recv_ip, sizeof(recv_ip));
bc28_socket_event_send(device, BC28_EVENT_DOMAIN_OK);
}
static void urc_func(struct at_client *client, const char *data, rt_size_t size)
{
RT_ASSERT(data);
LOG_I("URC data : %.*s", size, data);
}
static void urc_dns_func(struct at_client *client, const char *data, rt_size_t size)
{
RT_ASSERT(data && size);
urc_dnsqip_func(client, data, size);
}
static void urc_qiurc_func(struct at_client *client, const char *data, rt_size_t size)
{
RT_ASSERT(data && size);
switch(*(data + 9))
{
case 'c' : urc_close_func(client, data, size); break;//+QIURC: "closed"
case 'r' : urc_recv_func(client, data, size); break;//+QIURC: "recv"
case 'd' : urc_dnsqip_func(client, data, size); break;//+QIURC: "dnsgip"
default : urc_func(client, data, size); break;
}
}
static const struct at_urc urc_table[] =
{
{"SEND OK", "\r\n", urc_send_func},
{"SEND FAIL", "\r\n", urc_send_func},
{"+QIOPEN:", "\r\n", urc_connect_func},
{"+QIURC:", "\r\n", urc_qiurc_func},
{"+QDNS:", "\r\n", urc_dns_func},
};
static const struct at_socket_ops bc28_socket_ops =
{
bc28_socket_connect,
bc28_socket_close,
bc28_socket_send,
bc28_domain_resolve,
bc28_socket_set_event_cb,
};
int bc28_socket_init(struct at_device *device)
{
RT_ASSERT(device);
/* register URC data execution function */
at_obj_set_urc_table(device->client, urc_table, sizeof(urc_table) / sizeof(urc_table[0]));
return RT_EOK;
}
int bc28_socket_class_register(struct at_device_class *class)
{
RT_ASSERT(class);
class->socket_num = AT_DEVICE_BC28_SOCKETS_NUM;
class->socket_ops = &bc28_socket_ops;
return RT_EOK;
}
#endif /* AT_DEVICE_USING_BC28 && AT_USING_SOCKET */