commit
23b253ccf5
@ -31,6 +31,15 @@ if GetDepend(['AT_DEVICE_USING_ESP8266']):
|
||||
if GetDepend(['AT_DEVICE_ESP8266_SAMPLE']):
|
||||
src += Glob('samples/at_sample_esp8266.c')
|
||||
|
||||
# ESP32
|
||||
if GetDepend(['AT_DEVICE_USING_ESP32']):
|
||||
path += [cwd + '/class/esp32']
|
||||
src += Glob('class/esp32/at_device_esp32.c')
|
||||
if GetDepend(['AT_USING_SOCKET']):
|
||||
src += Glob('class/esp32/at_socket_esp32.c')
|
||||
if GetDepend(['AT_DEVICE_ESP32_SAMPLE']):
|
||||
src += Glob('samples/at_sample_esp32.c')
|
||||
|
||||
# MW31
|
||||
if GetDepend(['AT_DEVICE_USING_MW31']):
|
||||
path += [cwd + '/class/mw31']
|
||||
|
942
class/esp32/at_device_esp32.c
Normal file
942
class/esp32/at_device_esp32.c
Normal file
@ -0,0 +1,942 @@
|
||||
/*
|
||||
* File : at_socket_esp32.c
|
||||
* This file is part of RT-Thread RTOS
|
||||
* COPYRIGHT (C) 2006 - 2018, RT-Thread Development Team
|
||||
*
|
||||
* 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
|
||||
* 2019-10-26 zylx first version
|
||||
*/
|
||||
|
||||
#include <stdio.h>
|
||||
#include <string.h>
|
||||
|
||||
#include <at_device_esp32.h>
|
||||
|
||||
#define LOG_TAG "at.dev.esp"
|
||||
|
||||
#include <at_log.h>
|
||||
|
||||
#ifdef AT_DEVICE_USING_ESP32
|
||||
|
||||
#define ESP32_WAIT_CONNECT_TIME 5000
|
||||
#define ESP32_THREAD_STACK_SIZE 2048
|
||||
#define ESP32_THREAD_PRIORITY (RT_THREAD_PRIORITY_MAX / 2)
|
||||
|
||||
/* ============================= esp32 network interface operations ============================= */
|
||||
|
||||
static int esp32_netdev_set_dns_server(struct netdev *netdev, uint8_t dns_num, ip_addr_t *dns_server);
|
||||
|
||||
static void esp32_get_netdev_info(struct rt_work *work, void *work_data)
|
||||
{
|
||||
#define AT_ADDR_LEN 32
|
||||
#define AT_ERR_DNS_SERVER "255.255.255.255"
|
||||
#define AT_DEF_DNS_SERVER "114.114.114.114"
|
||||
|
||||
at_response_t resp = RT_NULL;
|
||||
char ip[AT_ADDR_LEN] = {0}, mac[AT_ADDR_LEN] = {0};
|
||||
char gateway[AT_ADDR_LEN] = {0}, netmask[AT_ADDR_LEN] = {0};
|
||||
char dns_server1[AT_ADDR_LEN] = {0}, dns_server2[AT_ADDR_LEN] = {0};
|
||||
const char *resp_expr = "%*[^\"]\"%[^\"]\"";
|
||||
const char *resp_dns = "+CIPDNS:%s";
|
||||
ip_addr_t ip_addr;
|
||||
rt_uint32_t mac_addr[6] = {0};
|
||||
rt_uint32_t num = 0;
|
||||
rt_uint8_t dhcp_stat = 0;
|
||||
struct rt_delayed_work *delay_work = (struct rt_delayed_work *)work;
|
||||
struct at_device *device = (struct at_device *)work_data;
|
||||
struct netdev *netdev = device->netdev;
|
||||
struct at_client *client = device->client;
|
||||
|
||||
if (delay_work)
|
||||
{
|
||||
rt_free(delay_work);
|
||||
}
|
||||
|
||||
resp = at_create_resp(512, 0, rt_tick_from_millisecond(300));
|
||||
if (resp == RT_NULL)
|
||||
{
|
||||
LOG_E("no memory for resp create.");
|
||||
return;
|
||||
}
|
||||
|
||||
/* send mac addr query commond "AT+CIFSR" and wait response */
|
||||
if (at_obj_exec_cmd(client, resp, "AT+CIFSR") < 0)
|
||||
{
|
||||
goto __exit;
|
||||
}
|
||||
|
||||
if (at_resp_parse_line_args_by_kw(resp, "STAMAC", resp_expr, mac) <= 0)
|
||||
{
|
||||
LOG_E("%s device parse \"AT+CIFSR\" cmd error.", device->name);
|
||||
goto __exit;
|
||||
}
|
||||
|
||||
/* send addr info query commond "AT+CIPSTA?" and wait response */
|
||||
if (at_obj_exec_cmd(client, resp, "AT+CIPSTA?") < 0)
|
||||
{
|
||||
LOG_E("%s device send \"AT+CIPSTA?\" cmd error.", device->name);
|
||||
goto __exit;
|
||||
}
|
||||
|
||||
if (at_resp_parse_line_args_by_kw(resp, "ip", resp_expr, ip) <= 0 ||
|
||||
at_resp_parse_line_args_by_kw(resp, "gateway", resp_expr, gateway) <= 0 ||
|
||||
at_resp_parse_line_args_by_kw(resp, "netmask", resp_expr, netmask) <= 0)
|
||||
{
|
||||
LOG_E("%s device prase \"AT+CIPSTA?\" cmd error.", device->name);
|
||||
goto __exit;
|
||||
}
|
||||
|
||||
/* set netdev info */
|
||||
inet_aton(gateway, &ip_addr);
|
||||
netdev_low_level_set_gw(netdev, &ip_addr);
|
||||
inet_aton(netmask, &ip_addr);
|
||||
netdev_low_level_set_netmask(netdev, &ip_addr);
|
||||
inet_aton(ip, &ip_addr);
|
||||
netdev_low_level_set_ipaddr(netdev, &ip_addr);
|
||||
sscanf(mac, "%x:%x:%x:%x:%x:%x",
|
||||
&mac_addr[0], &mac_addr[1], &mac_addr[2], &mac_addr[3], &mac_addr[4], &mac_addr[5]);
|
||||
for (num = 0; num < netdev->hwaddr_len; num++)
|
||||
{
|
||||
netdev->hwaddr[num] = mac_addr[num];
|
||||
}
|
||||
|
||||
/* send dns server query commond "AT+CIPDNS?" and wait response */
|
||||
if (at_obj_exec_cmd(device->client, resp, "AT+CIPDNS?") < 0)
|
||||
{
|
||||
LOG_W("please check and update %s device firmware to support the \"AT+CIPDNS?\" cmd.", device->name);
|
||||
goto __exit;
|
||||
}
|
||||
|
||||
if (at_resp_parse_line_args(resp, 1, resp_dns, dns_server1) <= 0 &&
|
||||
at_resp_parse_line_args(resp, 2, resp_dns, dns_server2) <= 0)
|
||||
{
|
||||
LOG_E("%s device prase \"AT+CIPDNS?\" cmd error.", device->name);
|
||||
goto __exit;
|
||||
}
|
||||
|
||||
/* set primary DNS server address */
|
||||
if (rt_strlen(dns_server1) > 0 &&
|
||||
rt_strncmp(dns_server1, AT_ERR_DNS_SERVER, rt_strlen(AT_ERR_DNS_SERVER)) != 0)
|
||||
{
|
||||
inet_aton(dns_server1, &ip_addr);
|
||||
netdev_low_level_set_dns_server(netdev, 0, &ip_addr);
|
||||
}
|
||||
else
|
||||
{
|
||||
inet_aton(AT_DEF_DNS_SERVER, &ip_addr);
|
||||
esp32_netdev_set_dns_server(netdev, 0, &ip_addr);
|
||||
}
|
||||
|
||||
/* set secondary DNS server address */
|
||||
if (rt_strlen(dns_server2) > 0 )
|
||||
{
|
||||
inet_aton(dns_server2, &ip_addr);
|
||||
netdev_low_level_set_dns_server(netdev, 1, &ip_addr);
|
||||
}
|
||||
|
||||
/* send DHCP query commond " AT+CWDHCP_CUR?" and wait response */
|
||||
if (at_obj_exec_cmd(client, resp, "AT+CWDHCP?") < 0)
|
||||
{
|
||||
goto __exit;
|
||||
}
|
||||
|
||||
/* parse response data, get the DHCP status */
|
||||
if (at_resp_parse_line_args_by_kw(resp, "+CWDHCP:", "+CWDHCP:%d", &dhcp_stat) < 0)
|
||||
{
|
||||
LOG_E("%s device prase DHCP status error.", device->name);
|
||||
goto __exit;
|
||||
}
|
||||
|
||||
/* Bit0 - SoftAP DHCP status, Bit1 - Station DHCP status */
|
||||
netdev_low_level_set_dhcp_status(netdev, dhcp_stat & 0x01 ? RT_TRUE : RT_FALSE);
|
||||
|
||||
__exit:
|
||||
if (resp)
|
||||
{
|
||||
at_delete_resp(resp);
|
||||
}
|
||||
}
|
||||
|
||||
static int esp32_net_init(struct at_device *device);
|
||||
|
||||
static int esp32_netdev_set_up(struct netdev *netdev)
|
||||
{
|
||||
struct at_device *device = RT_NULL;
|
||||
|
||||
device = at_device_get_by_name(AT_DEVICE_NAMETYPE_NETDEV, netdev->name);
|
||||
if (device == RT_NULL)
|
||||
{
|
||||
LOG_E("get device(%s) failed.", netdev->name);
|
||||
return -RT_ERROR;
|
||||
}
|
||||
|
||||
if (device->is_init == RT_FALSE)
|
||||
{
|
||||
esp32_net_init(device);
|
||||
netdev_low_level_set_status(netdev, RT_TRUE);
|
||||
LOG_D("network interface device(%s) set up status", netdev->name);
|
||||
}
|
||||
|
||||
return RT_EOK;
|
||||
}
|
||||
|
||||
static int esp32_netdev_set_down(struct netdev *netdev)
|
||||
{
|
||||
struct at_device *device = RT_NULL;
|
||||
|
||||
device = at_device_get_by_name(AT_DEVICE_NAMETYPE_NETDEV, netdev->name);
|
||||
if (device == RT_NULL)
|
||||
{
|
||||
LOG_E("get device by netdev(%s) failed.", netdev->name);
|
||||
return -RT_ERROR;
|
||||
}
|
||||
|
||||
if (device->is_init == RT_TRUE)
|
||||
{
|
||||
device->is_init = RT_FALSE;
|
||||
netdev_low_level_set_status(netdev, RT_FALSE);
|
||||
LOG_D("network interface device(%s) set down status", netdev->name);
|
||||
}
|
||||
|
||||
return RT_EOK;
|
||||
}
|
||||
|
||||
static int esp32_netdev_set_addr_info(struct netdev *netdev, ip_addr_t *ip_addr, ip_addr_t *netmask, ip_addr_t *gw)
|
||||
{
|
||||
#define IPADDR_RESP_SIZE 128
|
||||
#define IPADDR_SIZE 16
|
||||
|
||||
int result = RT_EOK;
|
||||
at_response_t resp = RT_NULL;
|
||||
struct at_device *device = RT_NULL;
|
||||
char ip_str[IPADDR_SIZE] = {0};
|
||||
char gw_str[IPADDR_SIZE] = {0};
|
||||
char netmask_str[IPADDR_SIZE] = {0};
|
||||
|
||||
RT_ASSERT(netdev);
|
||||
RT_ASSERT(ip_addr || netmask || gw);
|
||||
|
||||
device = at_device_get_by_name(AT_DEVICE_NAMETYPE_NETDEV, netdev->name);
|
||||
if (device == RT_NULL)
|
||||
{
|
||||
LOG_E("get device(%s) failed.", netdev->name);
|
||||
return -RT_ERROR;
|
||||
}
|
||||
|
||||
resp = at_create_resp(IPADDR_RESP_SIZE, 0, rt_tick_from_millisecond(300));
|
||||
if (resp == RT_NULL)
|
||||
{
|
||||
LOG_E("no memory for resp create.");
|
||||
result = -RT_ENOMEM;
|
||||
goto __exit;
|
||||
}
|
||||
|
||||
/* Convert numeric IP address into decimal dotted ASCII representation. */
|
||||
if (ip_addr)
|
||||
rt_memcpy(ip_str, inet_ntoa(*ip_addr), IPADDR_SIZE);
|
||||
else
|
||||
rt_memcpy(ip_str, inet_ntoa(netdev->ip_addr), IPADDR_SIZE);
|
||||
|
||||
if (gw)
|
||||
rt_memcpy(gw_str, inet_ntoa(*gw), IPADDR_SIZE);
|
||||
else
|
||||
rt_memcpy(gw_str, inet_ntoa(netdev->gw), IPADDR_SIZE);
|
||||
|
||||
if (netmask)
|
||||
rt_memcpy(netmask_str, inet_ntoa(*netmask), IPADDR_SIZE);
|
||||
else
|
||||
rt_memcpy(netmask_str, inet_ntoa(netdev->netmask), IPADDR_SIZE);
|
||||
|
||||
/* send addr info set commond "AT+CIPSTA_CUR=<ip>[,<gateway>,<netmask>]" and wait response */
|
||||
if (at_obj_exec_cmd(device->client, resp, "AT+CIPSTA_CUR=\"%s\",\"%s\",\"%s\"",
|
||||
ip_str, gw_str, netmask_str) < 0)
|
||||
{
|
||||
LOG_E("%s device set address failed.", device->name);
|
||||
result = -RT_ERROR;
|
||||
}
|
||||
else
|
||||
{
|
||||
/* Update netdev information */
|
||||
if (ip_addr)
|
||||
netdev_low_level_set_ipaddr(netdev, ip_addr);
|
||||
|
||||
if (gw)
|
||||
netdev_low_level_set_gw(netdev, gw);
|
||||
|
||||
if (netmask)
|
||||
netdev_low_level_set_netmask(netdev, netmask);
|
||||
|
||||
LOG_D("%s device set address success.", device->name);
|
||||
}
|
||||
|
||||
__exit:
|
||||
if (resp)
|
||||
{
|
||||
at_delete_resp(resp);
|
||||
}
|
||||
|
||||
return result;
|
||||
}
|
||||
|
||||
static int esp32_netdev_set_dns_server(struct netdev *netdev, uint8_t dns_num, ip_addr_t *dns_server)
|
||||
{
|
||||
#define DNS_RESP_SIZE 128
|
||||
|
||||
int result = RT_EOK;
|
||||
at_response_t resp = RT_NULL;
|
||||
struct at_device *device = RT_NULL;
|
||||
|
||||
RT_ASSERT(netdev);
|
||||
RT_ASSERT(dns_server);
|
||||
|
||||
device = at_device_get_by_name(AT_DEVICE_NAMETYPE_NETDEV, netdev->name);
|
||||
if (device == RT_NULL)
|
||||
{
|
||||
LOG_E("get device by netdev(%s) failed.", netdev->name);
|
||||
return -RT_ERROR;
|
||||
}
|
||||
|
||||
resp = at_create_resp(DNS_RESP_SIZE, 0, rt_tick_from_millisecond(300));
|
||||
if (resp == RT_NULL)
|
||||
{
|
||||
LOG_E("no memory for resp create.");
|
||||
return -RT_ENOMEM;
|
||||
}
|
||||
|
||||
/* send dns server set commond "AT+CIPDNS_CUR=<enable>[,<DNS server0>,<DNS server1>]" and wait response */
|
||||
if (at_obj_exec_cmd(device->client, resp, "AT+CIPDNS_CUR=1,\"%s\"", inet_ntoa(*dns_server)) < 0)
|
||||
{
|
||||
LOG_E("%s device set DNS failed.", device->name);
|
||||
result = -RT_ERROR;
|
||||
}
|
||||
else
|
||||
{
|
||||
netdev_low_level_set_dns_server(netdev, dns_num, dns_server);
|
||||
LOG_D("%s device set DNS(%s) success.", device->name, inet_ntoa(*dns_server));
|
||||
}
|
||||
|
||||
if (resp)
|
||||
{
|
||||
at_delete_resp(resp);
|
||||
}
|
||||
|
||||
return result;
|
||||
}
|
||||
|
||||
static int esp32_netdev_set_dhcp(struct netdev *netdev, rt_bool_t is_enabled)
|
||||
{
|
||||
#define ESP32_STATION 1
|
||||
#define RESP_SIZE 128
|
||||
|
||||
int result = RT_EOK;
|
||||
at_response_t resp = RT_NULL;
|
||||
struct at_device *device = RT_NULL;
|
||||
|
||||
RT_ASSERT(netdev);
|
||||
|
||||
device = at_device_get_by_name(AT_DEVICE_NAMETYPE_NETDEV, netdev->name);
|
||||
if (device == RT_NULL)
|
||||
{
|
||||
LOG_E("get device by netdev(%s) failed.", netdev->name);
|
||||
return -RT_ERROR;
|
||||
}
|
||||
|
||||
resp = at_create_resp(RESP_SIZE, 0, rt_tick_from_millisecond(300));
|
||||
if (resp == RT_NULL)
|
||||
{
|
||||
LOG_E("no memory for resp struct.");
|
||||
return -RT_ENOMEM;
|
||||
}
|
||||
|
||||
/* send dhcp set commond "AT+CWDHCP_CUR=<mode>,<en>" and wait response */
|
||||
if (at_obj_exec_cmd(device->client, resp, "AT+CWDHCP_CUR=%d,%d", ESP32_STATION, is_enabled) < 0)
|
||||
{
|
||||
LOG_E("%s device set DHCP status(%d) failed.", device->name, is_enabled);
|
||||
result = -RT_ERROR;
|
||||
goto __exit;
|
||||
}
|
||||
else
|
||||
{
|
||||
netdev_low_level_set_dhcp_status(netdev, is_enabled);
|
||||
LOG_D("%s device set DHCP status(%d) ok.", device->name, is_enabled);
|
||||
}
|
||||
|
||||
__exit:
|
||||
if (resp)
|
||||
{
|
||||
at_delete_resp(resp);
|
||||
}
|
||||
|
||||
return result;
|
||||
}
|
||||
|
||||
#ifdef NETDEV_USING_PING
|
||||
static int esp32_netdev_ping(struct netdev *netdev, const char *host,
|
||||
size_t data_len, uint32_t timeout, struct netdev_ping_resp *ping_resp)
|
||||
{
|
||||
#define ESP32_PING_IP_SIZE 16
|
||||
|
||||
rt_err_t result = RT_EOK;
|
||||
at_response_t resp = RT_NULL;
|
||||
struct at_device *device = RT_NULL;
|
||||
char ip_addr[ESP32_PING_IP_SIZE] = {0};
|
||||
int req_time;
|
||||
|
||||
RT_ASSERT(netdev);
|
||||
RT_ASSERT(host);
|
||||
RT_ASSERT(ping_resp);
|
||||
|
||||
device = at_device_get_by_name(AT_DEVICE_NAMETYPE_NETDEV, netdev->name);
|
||||
if (device == RT_NULL)
|
||||
{
|
||||
LOG_E("get device(%s) failed.", netdev->name);
|
||||
return -RT_ERROR;
|
||||
}
|
||||
|
||||
resp = at_create_resp(64, 0, timeout);
|
||||
if (resp == RT_NULL)
|
||||
{
|
||||
LOG_E("no memory for resp create.");
|
||||
return -RT_ENOMEM;
|
||||
}
|
||||
|
||||
/* send domain commond "AT+CIPDOMAIN=<domain name>" and wait response */
|
||||
if (at_obj_exec_cmd(device->client, resp, "AT+CIPDOMAIN=\"%s\"", host) < 0)
|
||||
{
|
||||
result = -RT_ERROR;
|
||||
goto __exit;
|
||||
}
|
||||
|
||||
/* parse the third line of response data, get the IP address */
|
||||
if (at_resp_parse_line_args_by_kw(resp, "+CIPDOMAIN:", "+CIPDOMAIN:%s", ip_addr) < 0)
|
||||
{
|
||||
result = -RT_ERROR;
|
||||
goto __exit;
|
||||
}
|
||||
|
||||
/* send ping commond "AT+PING=<IP>" and wait response */
|
||||
if (at_obj_exec_cmd(device->client, resp, "AT+PING=\"%s\"", host) < 0)
|
||||
{
|
||||
result = -RT_ERROR;
|
||||
goto __exit;
|
||||
}
|
||||
|
||||
if (at_resp_parse_line_args_by_kw(resp, "+", "+PING:%d", &req_time) < 0)
|
||||
{
|
||||
result = -RT_ERROR;
|
||||
goto __exit;
|
||||
}
|
||||
|
||||
if (req_time)
|
||||
{
|
||||
inet_aton(ip_addr, &(ping_resp->ip_addr));
|
||||
ping_resp->data_len = data_len;
|
||||
ping_resp->ttl = 0;
|
||||
ping_resp->ticks = req_time;
|
||||
}
|
||||
|
||||
__exit:
|
||||
if (resp)
|
||||
{
|
||||
at_delete_resp(resp);
|
||||
}
|
||||
|
||||
return result;
|
||||
}
|
||||
#endif /* NETDEV_USING_PING */
|
||||
|
||||
#ifdef NETDEV_USING_NETSTAT
|
||||
void esp32_netdev_netstat(struct netdev *netdev)
|
||||
{
|
||||
#define ESP32_NETSTAT_RESP_SIZE 320
|
||||
#define ESP32_NETSTAT_TYPE_SIZE 4
|
||||
#define ESP32_NETSTAT_IPADDR_SIZE 17
|
||||
#define ESP32_NETSTAT_EXPRESSION "+CIPSTATUS:%*d,\"%[^\"]\",\"%[^\"]\",%d,%d,%*d"
|
||||
|
||||
at_response_t resp = RT_NULL;
|
||||
struct at_device *device = RT_NULL;
|
||||
int remote_port, local_port, i;
|
||||
char *type = RT_NULL;
|
||||
char *ipaddr = RT_NULL;
|
||||
|
||||
device = at_device_get_by_name(AT_DEVICE_NAMETYPE_NETDEV, netdev->name);
|
||||
if (device == RT_NULL)
|
||||
{
|
||||
LOG_E("get device(%s) failed.", netdev->name);
|
||||
return;
|
||||
}
|
||||
|
||||
type = (char *) rt_calloc(1, ESP32_NETSTAT_TYPE_SIZE);
|
||||
ipaddr = (char *) rt_calloc(1, ESP32_NETSTAT_IPADDR_SIZE);
|
||||
if ((type && ipaddr) == RT_NULL)
|
||||
{
|
||||
LOG_E("no memory for ipaddr create.");
|
||||
goto __exit;
|
||||
}
|
||||
|
||||
resp = at_create_resp(ESP32_NETSTAT_RESP_SIZE, 0, 5 * RT_TICK_PER_SECOND);
|
||||
if (resp == RT_NULL)
|
||||
{
|
||||
LOG_E("no memory for resp create.", device->name);
|
||||
goto __exit;
|
||||
}
|
||||
|
||||
/* send network connection information commond "AT+CIPSTATUS" and wait response */
|
||||
if (at_obj_exec_cmd(device->client, resp, "AT+CIPSTATUS") < 0)
|
||||
{
|
||||
goto __exit;
|
||||
}
|
||||
|
||||
for (i = 1; i <= resp->line_counts; i++)
|
||||
{
|
||||
if (strstr(at_resp_get_line(resp, i), "+CIPSTATUS"))
|
||||
{
|
||||
/* parse the third line of response data, get the network connection information */
|
||||
if (at_resp_parse_line_args(resp, i, ESP32_NETSTAT_EXPRESSION, type, ipaddr, &remote_port, &local_port) < 0)
|
||||
{
|
||||
goto __exit;
|
||||
}
|
||||
else
|
||||
{
|
||||
LOG_RAW("%s: %s:%d ==> %s:%d\n", type, inet_ntoa(netdev->ip_addr), local_port, ipaddr, remote_port);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
__exit:
|
||||
if (resp)
|
||||
{
|
||||
at_delete_resp(resp);
|
||||
}
|
||||
|
||||
if (type)
|
||||
{
|
||||
rt_free(type);
|
||||
}
|
||||
|
||||
if (ipaddr)
|
||||
{
|
||||
rt_free(ipaddr);
|
||||
}
|
||||
}
|
||||
#endif /* NETDEV_USING_NETSTAT */
|
||||
|
||||
static const struct netdev_ops esp32_netdev_ops =
|
||||
{
|
||||
esp32_netdev_set_up,
|
||||
esp32_netdev_set_down,
|
||||
|
||||
esp32_netdev_set_addr_info,
|
||||
esp32_netdev_set_dns_server,
|
||||
esp32_netdev_set_dhcp,
|
||||
|
||||
#ifdef NETDEV_USING_PING
|
||||
esp32_netdev_ping,
|
||||
#endif
|
||||
#ifdef NETDEV_USING_NETSTAT
|
||||
esp32_netdev_netstat,
|
||||
#endif
|
||||
};
|
||||
|
||||
static struct netdev *esp32_netdev_add(const char *netdev_name)
|
||||
{
|
||||
#define ETHERNET_MTU 1500
|
||||
#define HWADDR_LEN 6
|
||||
struct netdev *netdev = RT_NULL;
|
||||
|
||||
RT_ASSERT(netdev_name);
|
||||
|
||||
netdev = (struct netdev *) rt_calloc(1, sizeof(struct netdev));
|
||||
if (netdev == RT_NULL)
|
||||
{
|
||||
LOG_E("no memory for netdev create.");
|
||||
return RT_NULL;
|
||||
}
|
||||
|
||||
netdev->mtu = ETHERNET_MTU;
|
||||
netdev->ops = &esp32_netdev_ops;
|
||||
netdev->hwaddr_len = HWADDR_LEN;
|
||||
|
||||
#ifdef SAL_USING_AT
|
||||
extern int sal_at_netdev_set_pf_info(struct netdev *netdev);
|
||||
/* set the network interface socket/netdb operations */
|
||||
sal_at_netdev_set_pf_info(netdev);
|
||||
#endif
|
||||
|
||||
netdev_register(netdev, netdev_name, RT_NULL);
|
||||
|
||||
return netdev;
|
||||
}
|
||||
|
||||
/* ============================= esp32 device operations ============================= */
|
||||
|
||||
#define AT_SEND_CMD(client, resp, cmd) \
|
||||
do { \
|
||||
(resp) = at_resp_set_info((resp), 256, 0, 5 * RT_TICK_PER_SECOND); \
|
||||
if (at_obj_exec_cmd((client), (resp), (cmd)) < 0) \
|
||||
{ \
|
||||
result = -RT_ERROR; \
|
||||
goto __exit; \
|
||||
} \
|
||||
} while(0) \
|
||||
|
||||
static void esp32_netdev_start_delay_work(struct at_device *device)
|
||||
{
|
||||
struct rt_delayed_work *net_work = RT_NULL;
|
||||
net_work = (struct rt_delayed_work *)rt_calloc(1, sizeof(struct rt_delayed_work));
|
||||
if (net_work == RT_NULL)
|
||||
{
|
||||
return;
|
||||
}
|
||||
|
||||
rt_delayed_work_init(net_work, esp32_get_netdev_info, (void *)device);
|
||||
rt_work_submit(&(net_work->work), RT_TICK_PER_SECOND);
|
||||
}
|
||||
|
||||
static void esp32_init_thread_entry(void *parameter)
|
||||
{
|
||||
#define INIT_RETRY 5
|
||||
|
||||
struct at_device *device = (struct at_device *) parameter;
|
||||
struct at_device_esp32 *esp32 = (struct at_device_esp32 *) device->user_data;
|
||||
struct at_client *client = device->client;
|
||||
at_response_t resp = RT_NULL;
|
||||
rt_err_t result = RT_EOK;
|
||||
rt_size_t i = 0, retry_num = INIT_RETRY;
|
||||
rt_bool_t wifi_is_conn = RT_FALSE;
|
||||
|
||||
LOG_D("%s device initialize start.", device->name);
|
||||
|
||||
/* wait esp32 device startup finish */
|
||||
if (at_client_obj_wait_connect(client, ESP32_WAIT_CONNECT_TIME))
|
||||
{
|
||||
return;
|
||||
}
|
||||
|
||||
resp = at_create_resp(128, 0, 5 * RT_TICK_PER_SECOND);
|
||||
if (resp == RT_NULL)
|
||||
{
|
||||
LOG_E("no memory for resp create.");
|
||||
return;
|
||||
}
|
||||
|
||||
while (retry_num--)
|
||||
{
|
||||
/* reset module */
|
||||
AT_SEND_CMD(client, resp, "AT+RST");
|
||||
/* reset waiting delay */
|
||||
rt_thread_mdelay(1000);
|
||||
/* disable echo */
|
||||
AT_SEND_CMD(client, resp, "ATE0");
|
||||
/* set current mode to Wi-Fi station */
|
||||
AT_SEND_CMD(client, resp, "AT+CWMODE=1");
|
||||
/* get module version */
|
||||
AT_SEND_CMD(client, resp, "AT+GMR");
|
||||
/* show module version */
|
||||
for (i = 0; i < resp->line_counts - 1; i++)
|
||||
{
|
||||
LOG_D("%s", at_resp_get_line(resp, i + 1));
|
||||
}
|
||||
|
||||
AT_SEND_CMD(client, resp, "AT+CIPMUX=1");
|
||||
|
||||
/* initialize successfully */
|
||||
result = RT_EOK;
|
||||
break;
|
||||
|
||||
__exit:
|
||||
if (result != RT_EOK)
|
||||
{
|
||||
rt_thread_mdelay(1000);
|
||||
LOG_I("%s device initialize retry...", device->name);
|
||||
}
|
||||
}
|
||||
|
||||
/* connect to WiFi AP */
|
||||
if (at_obj_exec_cmd(client, at_resp_set_info(resp, 512, 0, 20 * RT_TICK_PER_SECOND),
|
||||
"AT+CWJAP=\"%s\",\"%s\"", esp32->wifi_ssid, esp32->wifi_password) != RT_EOK)
|
||||
{
|
||||
LOG_W("%s device wifi connect failed, check ssid(%s) and password(%s).",
|
||||
device->name, esp32->wifi_ssid, esp32->wifi_password);
|
||||
}
|
||||
else
|
||||
{
|
||||
wifi_is_conn = RT_TRUE;
|
||||
}
|
||||
|
||||
if (resp)
|
||||
{
|
||||
at_delete_resp(resp);
|
||||
}
|
||||
|
||||
if (result != RT_EOK)
|
||||
{
|
||||
netdev_low_level_set_status(device->netdev, RT_FALSE);
|
||||
LOG_E("%s device network initialize failed(%d).", device->name, result);
|
||||
}
|
||||
else
|
||||
{
|
||||
device->is_init = RT_TRUE;
|
||||
netdev_low_level_set_status(device->netdev, RT_TRUE);
|
||||
if (wifi_is_conn)
|
||||
{
|
||||
netdev_low_level_set_link_status(device->netdev, RT_TRUE);
|
||||
}
|
||||
esp32_netdev_start_delay_work(device);
|
||||
LOG_I("%s device network initialize successfully.", device->name);
|
||||
}
|
||||
}
|
||||
|
||||
static int esp32_net_init(struct at_device *device)
|
||||
{
|
||||
#ifdef AT_DEVICE_ESP32_INIT_ASYN
|
||||
rt_thread_t tid;
|
||||
|
||||
tid = rt_thread_create("esp_net", esp32_init_thread_entry, (void *) device,
|
||||
ESP32_THREAD_STACK_SIZE, ESP32_THREAD_PRIORITY, 20);
|
||||
if (tid)
|
||||
{
|
||||
rt_thread_startup(tid);
|
||||
}
|
||||
else
|
||||
{
|
||||
LOG_E("create %s device init thread failed.", device->name);
|
||||
return -RT_ERROR;
|
||||
}
|
||||
#else
|
||||
esp32_init_thread_entry(device);
|
||||
#endif /* AT_DEVICE_ESP32_INIT_ASYN */
|
||||
|
||||
return RT_EOK;
|
||||
}
|
||||
|
||||
static void urc_busy_p_func(struct at_client *client, const char *data, rt_size_t size)
|
||||
{
|
||||
LOG_D("system is processing a commands...");
|
||||
}
|
||||
|
||||
static void urc_busy_s_func(struct at_client *client, const char *data, rt_size_t size)
|
||||
{
|
||||
LOG_D("system is sending data...");
|
||||
}
|
||||
|
||||
static void urc_func(struct at_client *client, const char *data, rt_size_t size)
|
||||
{
|
||||
struct at_device *device = RT_NULL;
|
||||
char *client_name = client->device->parent.name;
|
||||
|
||||
RT_ASSERT(client && 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;
|
||||
}
|
||||
|
||||
if (rt_strstr(data, "WIFI CONNECTED"))
|
||||
{
|
||||
LOG_I("%s device wifi is connected.", device->name);
|
||||
|
||||
if (device->is_init)
|
||||
{
|
||||
netdev_low_level_set_link_status(device->netdev, RT_TRUE);
|
||||
|
||||
esp32_netdev_start_delay_work(device);
|
||||
}
|
||||
}
|
||||
else if (rt_strstr(data, "WIFI DISCONNECT"))
|
||||
{
|
||||
LOG_I("%s device wifi is disconnect.", device->name);
|
||||
|
||||
if (device->is_init)
|
||||
{
|
||||
netdev_low_level_set_link_status(device->netdev, RT_FALSE);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
static const struct at_urc urc_table[] =
|
||||
{
|
||||
{"busy p", "\r\n", urc_busy_p_func},
|
||||
{"busy s", "\r\n", urc_busy_s_func},
|
||||
{"WIFI CONNECTED", "\r\n", urc_func},
|
||||
{"WIFI DISCONNECT", "\r\n", urc_func},
|
||||
};
|
||||
|
||||
static int esp32_init(struct at_device *device)
|
||||
{
|
||||
struct at_device_esp32 *esp32 = (struct at_device_esp32 *) device->user_data;
|
||||
|
||||
/* initialize AT client */
|
||||
at_client_init(esp32->client_name, esp32->recv_line_num);
|
||||
|
||||
device->client = at_client_get(esp32->client_name);
|
||||
if (device->client == RT_NULL)
|
||||
{
|
||||
LOG_E("get AT client(%s) failed.", esp32->client_name);
|
||||
return -RT_ERROR;
|
||||
}
|
||||
|
||||
/* register URC data execution function */
|
||||
at_obj_set_urc_table(device->client, urc_table, sizeof(urc_table) / sizeof(urc_table[0]));
|
||||
|
||||
#ifdef AT_USING_SOCKET
|
||||
esp32_socket_init(device);
|
||||
#endif
|
||||
|
||||
/* add esp32 device to the netdev list */
|
||||
device->netdev = esp32_netdev_add(esp32->device_name);
|
||||
if (device->netdev == RT_NULL)
|
||||
{
|
||||
LOG_E("add netdev(%s) failed.", esp32->device_name);
|
||||
return -RT_ERROR;
|
||||
}
|
||||
|
||||
/* initialize esp32 device network */
|
||||
return esp32_netdev_set_up(device->netdev);
|
||||
}
|
||||
|
||||
static int esp32_deinit(struct at_device *device)
|
||||
{
|
||||
return esp32_netdev_set_down(device->netdev);
|
||||
}
|
||||
|
||||
/* reset eap8266 device and initialize device network again */
|
||||
static int esp32_reset(struct at_device *device)
|
||||
{
|
||||
int result = RT_EOK;
|
||||
struct at_client *client = device->client;
|
||||
|
||||
/* send "AT+RST" commonds to esp32 device */
|
||||
result = at_obj_exec_cmd(client, RT_NULL, "AT+RST");
|
||||
rt_thread_mdelay(1000);
|
||||
|
||||
/* waiting 10 seconds for esp32 device reset */
|
||||
device->is_init = RT_FALSE;
|
||||
if (at_client_obj_wait_connect(client, ESP32_WAIT_CONNECT_TIME))
|
||||
{
|
||||
return -RT_ETIMEOUT;
|
||||
}
|
||||
|
||||
/* initialize esp32 device network */
|
||||
esp32_net_init(device);
|
||||
|
||||
device->is_init = RT_TRUE;
|
||||
|
||||
return result;
|
||||
}
|
||||
|
||||
/* change eap8266 wifi ssid and password information */
|
||||
static int esp32_wifi_info_set(struct at_device *device, struct at_device_ssid_pwd *info)
|
||||
{
|
||||
int result = RT_EOK;
|
||||
struct at_response *resp = RT_NULL;
|
||||
|
||||
if (info->ssid == RT_NULL || info->password == RT_NULL)
|
||||
{
|
||||
LOG_E("input wifi ssid(%s) and password(%s) error.", info->ssid, info->password);
|
||||
return -RT_ERROR;
|
||||
}
|
||||
|
||||
resp = at_create_resp(128, 0, 20 * RT_TICK_PER_SECOND);
|
||||
if (resp == RT_NULL)
|
||||
{
|
||||
LOG_E("no memory for resp create.");
|
||||
return -RT_ENOMEM;
|
||||
}
|
||||
|
||||
/* connect to input wifi ap */
|
||||
if (at_obj_exec_cmd(device->client, resp, "AT+CWJAP=\"%s\",\"%s\"", info->ssid, info->password) != RT_EOK)
|
||||
{
|
||||
LOG_E("%s device wifi connect failed, check ssid(%s) and password(%s).",
|
||||
device->name, info->ssid, info->password);
|
||||
result = -RT_ERROR;
|
||||
}
|
||||
|
||||
if (resp)
|
||||
{
|
||||
at_delete_resp(resp);
|
||||
}
|
||||
|
||||
return result;
|
||||
}
|
||||
|
||||
static int esp32_control(struct at_device *device, int cmd, void *arg)
|
||||
{
|
||||
int result = -RT_ERROR;
|
||||
|
||||
RT_ASSERT(device);
|
||||
|
||||
switch (cmd)
|
||||
{
|
||||
case AT_DEVICE_CTRL_POWER_ON:
|
||||
case AT_DEVICE_CTRL_POWER_OFF:
|
||||
case AT_DEVICE_CTRL_LOW_POWER:
|
||||
case AT_DEVICE_CTRL_SLEEP:
|
||||
case AT_DEVICE_CTRL_WAKEUP:
|
||||
case AT_DEVICE_CTRL_NET_CONN:
|
||||
case AT_DEVICE_CTRL_NET_DISCONN:
|
||||
case AT_DEVICE_CTRL_GET_SIGNAL:
|
||||
case AT_DEVICE_CTRL_GET_GPS:
|
||||
case AT_DEVICE_CTRL_GET_VER:
|
||||
LOG_W("not support the control cmd(%d).", cmd);
|
||||
break;
|
||||
case AT_DEVICE_CTRL_RESET:
|
||||
result = esp32_reset(device);
|
||||
break;
|
||||
case AT_DEVICE_CTRL_SET_WIFI_INFO:
|
||||
result = esp32_wifi_info_set(device, (struct at_device_ssid_pwd *) arg);
|
||||
break;
|
||||
default:
|
||||
LOG_E("input error control cmd(%d).", cmd);
|
||||
break;
|
||||
}
|
||||
|
||||
return result;
|
||||
}
|
||||
|
||||
static const struct at_device_ops esp32_device_ops =
|
||||
{
|
||||
esp32_init,
|
||||
esp32_deinit,
|
||||
esp32_control,
|
||||
};
|
||||
|
||||
static int esp32_device_class_register(void)
|
||||
{
|
||||
struct at_device_class *class = RT_NULL;
|
||||
|
||||
class = (struct at_device_class *) rt_calloc(1, sizeof(struct at_device_class));
|
||||
if (class == RT_NULL)
|
||||
{
|
||||
LOG_E("no memory for class create.");
|
||||
return -RT_ENOMEM;
|
||||
}
|
||||
|
||||
/* fill ESP32 device class object */
|
||||
#ifdef AT_USING_SOCKET
|
||||
esp32_socket_class_register(class);
|
||||
#endif
|
||||
class->device_ops = &esp32_device_ops;
|
||||
|
||||
return at_device_class_register(class, AT_DEVICE_CLASS_ESP32);
|
||||
}
|
||||
INIT_DEVICE_EXPORT(esp32_device_class_register);
|
||||
|
||||
#endif /* AT_DEVICE_USING_ESP32 */
|
66
class/esp32/at_device_esp32.h
Normal file
66
class/esp32/at_device_esp32.h
Normal file
@ -0,0 +1,66 @@
|
||||
/*
|
||||
* File : at_device_esp32.h
|
||||
* This file is part of RT-Thread RTOS
|
||||
* COPYRIGHT (C) 2006 - 2018, RT-Thread Development Team
|
||||
*
|
||||
* 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
|
||||
* 2019-05-16 chenyong first version
|
||||
*/
|
||||
|
||||
#ifndef __AT_DEVICE_ESP32_H__
|
||||
#define __AT_DEVICE_ESP32_H__
|
||||
|
||||
#ifdef __cplusplus
|
||||
extern "C" {
|
||||
#endif
|
||||
|
||||
#include <stdlib.h>
|
||||
|
||||
#include <at_device.h>
|
||||
|
||||
/* The maximum number of sockets supported by the esp32 device */
|
||||
#define AT_DEVICE_ESP32_SOCKETS_NUM 5
|
||||
|
||||
struct at_device_esp32
|
||||
{
|
||||
char *device_name;
|
||||
char *client_name;
|
||||
|
||||
char *wifi_ssid;
|
||||
char *wifi_password;
|
||||
size_t recv_line_num;
|
||||
struct at_device device;
|
||||
|
||||
void *user_data;
|
||||
};
|
||||
|
||||
#ifdef AT_USING_SOCKET
|
||||
|
||||
/* esp32 device socket initialize */
|
||||
int esp32_socket_init(struct at_device *device);
|
||||
|
||||
/* esp32 device class socket register */
|
||||
int esp32_socket_class_register(struct at_device_class *class);
|
||||
|
||||
#endif /* AT_USING_SOCKET */
|
||||
|
||||
#ifdef __cplusplus
|
||||
}
|
||||
#endif
|
||||
|
||||
#endif /* __AT_DEVICE_ESP32_H__ */
|
556
class/esp32/at_socket_esp32.c
Normal file
556
class/esp32/at_socket_esp32.c
Normal file
@ -0,0 +1,556 @@
|
||||
/*
|
||||
* File : at_socket_esp32.c
|
||||
* This file is part of RT-Thread RTOS
|
||||
* COPYRIGHT (C) 2006 - 2018, RT-Thread Development Team
|
||||
*
|
||||
* 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
|
||||
* 2018-06-20 chenyong first version
|
||||
* 2019-05-09 chenyong multi AT socket client support
|
||||
*/
|
||||
|
||||
#include <stdio.h>
|
||||
#include <string.h>
|
||||
|
||||
#include <at_device_esp32.h>
|
||||
|
||||
#define LOG_TAG "at.skt.esp"
|
||||
#include <at_log.h>
|
||||
|
||||
#if defined(AT_DEVICE_USING_ESP32) && defined(AT_USING_SOCKET)
|
||||
|
||||
#define ESP32_MODULE_SEND_MAX_SIZE 2048
|
||||
/* set real event by current socket and current state */
|
||||
#define SET_EVENT(socket, event) (((socket + 1) << 16) | (event))
|
||||
|
||||
/* AT socket event type */
|
||||
#define ESP32_EVENT_CONN_OK (1L << 0)
|
||||
#define ESP32_EVENT_SEND_OK (1L << 1)
|
||||
#define ESP32_EVENT_RECV_OK (1L << 2)
|
||||
#define ESP32_EVNET_CLOSE_OK (1L << 3)
|
||||
#define ESP32_EVENT_CONN_FAIL (1L << 4)
|
||||
#define ESP32_EVENT_SEND_FAIL (1L << 5)
|
||||
|
||||
static at_evt_cb_t at_evt_cb_set[] = {
|
||||
[AT_SOCKET_EVT_RECV] = NULL,
|
||||
[AT_SOCKET_EVT_CLOSED] = NULL,
|
||||
};
|
||||
|
||||
static int esp32_socket_event_send(struct at_device *device, uint32_t event)
|
||||
{
|
||||
return (int) rt_event_send(device->socket_event, event);
|
||||
}
|
||||
|
||||
static int esp32_socket_event_recv(struct at_device *device, uint32_t event, uint32_t timeout, rt_uint8_t option)
|
||||
{
|
||||
int result = 0;
|
||||
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 esp32_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+CIPCLOSE=%d", device_socket);
|
||||
|
||||
if (resp)
|
||||
{
|
||||
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 esp32_socket_connect(struct at_socket *socket, char *ip, int32_t port, enum at_socket_type type, rt_bool_t is_client)
|
||||
{
|
||||
int result = RT_EOK;
|
||||
rt_bool_t retryed = RT_FALSE;
|
||||
at_response_t resp = RT_NULL;
|
||||
int device_socket = (int) socket->user_data;
|
||||
struct at_device *device = (struct at_device *) socket->device;
|
||||
|
||||
RT_ASSERT(ip);
|
||||
RT_ASSERT(port >= 0);
|
||||
|
||||
resp = at_create_resp(128, 0, 5 * RT_TICK_PER_SECOND);
|
||||
if (resp == RT_NULL)
|
||||
{
|
||||
LOG_E("no memory for resp create.");
|
||||
return -RT_ENOMEM;
|
||||
}
|
||||
|
||||
__retry:
|
||||
if (is_client)
|
||||
{
|
||||
switch (type)
|
||||
{
|
||||
case AT_SOCKET_TCP:
|
||||
/* send AT commands to connect TCP server */
|
||||
if (at_obj_exec_cmd(device->client, resp,
|
||||
"AT+CIPSTART=%d,\"TCP\",\"%s\",%d,60", device_socket, ip, port) < 0)
|
||||
{
|
||||
result = -RT_ERROR;
|
||||
}
|
||||
break;
|
||||
|
||||
case AT_SOCKET_UDP:
|
||||
if (at_obj_exec_cmd(device->client, resp,
|
||||
"AT+CIPSTART=%d,\"UDP\",\"%s\",%d", device_socket, ip, port) < 0)
|
||||
{
|
||||
result = -RT_ERROR;
|
||||
}
|
||||
break;
|
||||
|
||||
default:
|
||||
LOG_E("not supported connect type %d.", type);
|
||||
result = -RT_ERROR;
|
||||
goto __exit;
|
||||
}
|
||||
}
|
||||
|
||||
if (result != RT_EOK && retryed == RT_FALSE)
|
||||
{
|
||||
LOG_D("%s device socket (%d) connect failed, the socket was not be closed and now will connect retry.",
|
||||
device->name, device_socket);
|
||||
if (esp32_socket_close(socket) < 0)
|
||||
{
|
||||
goto __exit;
|
||||
}
|
||||
retryed = RT_TRUE;
|
||||
result = RT_EOK;
|
||||
goto __retry;
|
||||
}
|
||||
|
||||
__exit:
|
||||
if (resp)
|
||||
{
|
||||
at_delete_resp(resp);
|
||||
}
|
||||
|
||||
return result;
|
||||
}
|
||||
|
||||
/**
|
||||
* 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 esp32_socket_send(struct at_socket *socket, const char *buff, size_t bfsz, enum at_socket_type type)
|
||||
{
|
||||
int result = RT_EOK;
|
||||
int 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_esp32 *esp32 = (struct at_device_esp32 *) device->user_data;
|
||||
rt_mutex_t lock = device->client->lock;
|
||||
|
||||
RT_ASSERT(buff);
|
||||
RT_ASSERT(bfsz > 0);
|
||||
|
||||
resp = at_create_resp(128, 2, 5 * RT_TICK_PER_SECOND);
|
||||
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 */
|
||||
esp32->user_data = (void *) device_socket;
|
||||
|
||||
/* set AT client end sign to deal with '>' sign */
|
||||
at_obj_set_end_sign(device->client, '>');
|
||||
|
||||
while (sent_size < bfsz)
|
||||
{
|
||||
if (bfsz - sent_size < ESP32_MODULE_SEND_MAX_SIZE)
|
||||
{
|
||||
cur_pkt_size = bfsz - sent_size;
|
||||
}
|
||||
else
|
||||
{
|
||||
cur_pkt_size = ESP32_MODULE_SEND_MAX_SIZE;
|
||||
}
|
||||
|
||||
/* send the "AT+CIPSEND" commands to AT server than receive the '>' response on the first line */
|
||||
if (at_obj_exec_cmd(device->client, resp, "AT+CIPSEND=%d,%d", device_socket, cur_pkt_size) < 0)
|
||||
{
|
||||
result = -RT_ERROR;
|
||||
goto __exit;
|
||||
}
|
||||
|
||||
/* 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 */
|
||||
if (esp32_socket_event_recv(device, SET_EVENT(device_socket, 0),
|
||||
10 * 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;
|
||||
goto __exit;
|
||||
}
|
||||
/* waiting OK or failed result */
|
||||
event_result = esp32_socket_event_recv(device, ESP32_EVENT_SEND_OK | ESP32_EVENT_SEND_FAIL,
|
||||
5 * 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;
|
||||
goto __exit;
|
||||
}
|
||||
/* check result */
|
||||
if (event_result & ESP32_EVENT_SEND_FAIL)
|
||||
{
|
||||
LOG_E("%s device socket(%d) send failed.", device->name, device_socket);
|
||||
result = -RT_ERROR;
|
||||
goto __exit;
|
||||
}
|
||||
|
||||
sent_size += cur_pkt_size;
|
||||
}
|
||||
|
||||
__exit:
|
||||
/* reset the end sign for data */
|
||||
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
|
||||
* -2: wait socket event timeout
|
||||
* -5: no memory
|
||||
*/
|
||||
static int esp32_domain_resolve(const char *name, char ip[16])
|
||||
{
|
||||
#define RESOLVE_RETRY 5
|
||||
|
||||
int i, result = RT_EOK;
|
||||
char recv_ip[16] = { 0 };
|
||||
at_response_t resp = RT_NULL;
|
||||
struct at_device *device = 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;
|
||||
}
|
||||
|
||||
resp = at_create_resp(128, 0, 20 * RT_TICK_PER_SECOND);
|
||||
if (resp == RT_NULL)
|
||||
{
|
||||
LOG_E("no memory for resp create.");
|
||||
return -RT_ENOMEM;
|
||||
}
|
||||
|
||||
for (i = 0; i < RESOLVE_RETRY; i++)
|
||||
{
|
||||
if (at_obj_exec_cmd(device->client, resp, "AT+CIPDOMAIN=\"%s\"", name) < 0)
|
||||
{
|
||||
result = -RT_ERROR;
|
||||
goto __exit;
|
||||
}
|
||||
|
||||
/* parse the third line of response data, get the IP address */
|
||||
if (at_resp_parse_line_args_by_kw(resp, "+CIPDOMAIN:", "+CIPDOMAIN:%s", recv_ip) < 0)
|
||||
{
|
||||
rt_thread_mdelay(100);
|
||||
/* resolve failed, maybe receive an URC CRLF */
|
||||
continue;
|
||||
}
|
||||
|
||||
if (rt_strlen(recv_ip) < 8)
|
||||
{
|
||||
rt_thread_mdelay(100);
|
||||
/* resolve failed, maybe receive an URC CRLF */
|
||||
continue;
|
||||
}
|
||||
else
|
||||
{
|
||||
rt_strncpy(ip, recv_ip, 15);
|
||||
ip[15] = '\0';
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
__exit:
|
||||
if (resp)
|
||||
{
|
||||
at_delete_resp(resp);
|
||||
}
|
||||
|
||||
return result;
|
||||
|
||||
}
|
||||
|
||||
/**
|
||||
* set AT socket event notice callback
|
||||
*
|
||||
* @param event notice event
|
||||
* @param cb notice callback
|
||||
*/
|
||||
static void esp32_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 const struct at_socket_ops esp32_socket_ops =
|
||||
{
|
||||
esp32_socket_connect,
|
||||
esp32_socket_close,
|
||||
esp32_socket_send,
|
||||
esp32_domain_resolve,
|
||||
esp32_socket_set_event_cb,
|
||||
};
|
||||
|
||||
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_esp32 *esp32 = 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;
|
||||
}
|
||||
esp32 = (struct at_device_esp32 *) device->user_data;
|
||||
device_socket = (int) esp32->user_data;
|
||||
|
||||
if (rt_strstr(data, "SEND OK"))
|
||||
{
|
||||
esp32_socket_event_send(device, SET_EVENT(device_socket, ESP32_EVENT_SEND_OK));
|
||||
}
|
||||
else if (rt_strstr(data, "SEND FAIL"))
|
||||
{
|
||||
esp32_socket_event_send(device, SET_EVENT(device_socket, ESP32_EVENT_SEND_FAIL));
|
||||
}
|
||||
}
|
||||
|
||||
static void urc_send_bfsz_func(struct at_client *client, const char *data, rt_size_t size)
|
||||
{
|
||||
static int cur_send_bfsz = 0;
|
||||
|
||||
RT_ASSERT(data && size);
|
||||
|
||||
sscanf(data, "Recv %d bytes", &cur_send_bfsz);
|
||||
}
|
||||
|
||||
static void urc_close_func(struct at_client *client, const char *data, rt_size_t size)
|
||||
{
|
||||
int index = 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, "%d,CLOSED", &index);
|
||||
socket = &(device->sockets[index]);
|
||||
|
||||
/* 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, RT_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 = 0;
|
||||
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 at deveice socket and receive buffer size by receive data */
|
||||
sscanf(data, "+IPD,%d,%d:", &device_socket, (int *) &bfsz);
|
||||
|
||||
/* set receive timeout by receive buffer length, not less than 10ms */
|
||||
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 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 const struct at_urc urc_table[] =
|
||||
{
|
||||
{"SEND OK", "\r\n", urc_send_func},
|
||||
{"SEND FAIL", "\r\n", urc_send_func},
|
||||
{"Recv", "bytes\r\n", urc_send_bfsz_func},
|
||||
{"", ",CLOSED\r\n", urc_close_func},
|
||||
{"+IPD", ":", urc_recv_func},
|
||||
};
|
||||
|
||||
int esp32_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 esp32_socket_class_register(struct at_device_class *class)
|
||||
{
|
||||
RT_ASSERT(class);
|
||||
|
||||
class->socket_num = AT_DEVICE_ESP32_SOCKETS_NUM;
|
||||
class->socket_ops = &esp32_socket_ops;
|
||||
|
||||
return RT_EOK;
|
||||
}
|
||||
|
||||
#endif /* AT_DEVICE_USING_ESP32 && AT_USING_SOCKET */
|
@ -51,6 +51,7 @@ extern "C" {
|
||||
#define AT_DEVICE_CLASS_SIM76XX 0x05U
|
||||
#define AT_DEVICE_CLASS_RW007 0x06U
|
||||
#define AT_DEVICE_CLASS_MW31 0x07U
|
||||
#define AT_DEVICE_CLASS_ESP32 0x08U
|
||||
|
||||
/* Options and Commands for AT device control opreations */
|
||||
#define AT_DEVICE_CTRL_POWER_ON 0x01L
|
||||
|
52
samples/at_sample_esp32.c
Normal file
52
samples/at_sample_esp32.c
Normal file
@ -0,0 +1,52 @@
|
||||
/*
|
||||
* File : at_sample_esp32.c
|
||||
* This file is part of RT-Thread RTOS
|
||||
* COPYRIGHT (C) 2006 - 2018, RT-Thread Development Team
|
||||
*
|
||||
* 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
|
||||
* 2019-05-10 chenyong first version
|
||||
*/
|
||||
|
||||
#include <at_device_esp32.h>
|
||||
|
||||
#define LOG_TAG "at.sample.esp"
|
||||
#include <at_log.h>
|
||||
|
||||
#define ESP32_SAMPLE_DEIVCE_NAME "esp0"
|
||||
|
||||
static struct at_device_esp32 esp0 =
|
||||
{
|
||||
ESP32_SAMPLE_DEIVCE_NAME,
|
||||
ESP32_SAMPLE_CLIENT_NAME,
|
||||
|
||||
ESP32_SAMPLE_WIFI_SSID,
|
||||
ESP32_SAMPLE_WIFI_PASSWORD,
|
||||
ESP32_SAMPLE_RECV_BUFF_LEN,
|
||||
};
|
||||
|
||||
static int esp32_device_register(void)
|
||||
{
|
||||
struct at_device_esp32 *esp32 = &esp0;
|
||||
|
||||
return at_device_register(&(esp32->device),
|
||||
esp32->device_name,
|
||||
esp32->client_name,
|
||||
AT_DEVICE_CLASS_ESP32,
|
||||
(void *) esp32);
|
||||
}
|
||||
INIT_APP_EXPORT(esp32_device_register);
|
Loading…
x
Reference in New Issue
Block a user