375 lines
8.6 KiB
C
375 lines
8.6 KiB
C
/*
|
|
* Copyright (C) 2015-2018 Alibaba Group Holding Limited
|
|
*/
|
|
|
|
#include "infra_config.h"
|
|
|
|
#ifdef INFRA_NET
|
|
#include <stdio.h>
|
|
#include <string.h>
|
|
#include <stdlib.h>
|
|
#include "infra_state.h"
|
|
#include "infra_defs.h"
|
|
#include "infra_net.h"
|
|
#include "wrappers_defs.h"
|
|
#include "wrappers.h"
|
|
|
|
#ifdef INFRA_LOG
|
|
#include "infra_log.h"
|
|
#define net_err(...) log_err("infra_net", __VA_ARGS__)
|
|
#else
|
|
#define net_err(...)
|
|
#endif
|
|
|
|
/*** SSL connection ***/
|
|
#if defined(SUPPORT_TLS)
|
|
#define CERT_HOST_ADDRESS "124.126.19.5"
|
|
|
|
static int is_cert_host_addr = 0;
|
|
#if defined(AT_SSL_ENABLED)
|
|
#include "at_ssl.h"
|
|
|
|
static int read_ssl(utils_network_pt pNetwork, char *buffer, uint32_t len, uint32_t timeout_ms)
|
|
{
|
|
if (NULL == pNetwork)
|
|
{
|
|
return STATE_SYS_DEPEND_NWK_INVALID_HANDLE;
|
|
}
|
|
|
|
return AT_SSL_Read((uintptr_t)pNetwork->handle, buffer, len, timeout_ms);
|
|
}
|
|
|
|
static int write_ssl(utils_network_pt pNetwork, const char *buffer, uint32_t len, uint32_t timeout_ms)
|
|
{
|
|
if (NULL == pNetwork)
|
|
{
|
|
return STATE_SYS_DEPEND_NWK_INVALID_HANDLE;
|
|
}
|
|
|
|
return AT_SSL_Write((uintptr_t)pNetwork->handle, buffer, len, timeout_ms);
|
|
}
|
|
|
|
static int disconnect_ssl(utils_network_pt pNetwork)
|
|
{
|
|
if (NULL == pNetwork)
|
|
{
|
|
return STATE_SYS_DEPEND_NWK_INVALID_HANDLE;
|
|
}
|
|
|
|
AT_SSL_Destroy((uintptr_t)pNetwork->handle);
|
|
pNetwork->handle = 0;
|
|
|
|
return STATE_SUCCESS;
|
|
}
|
|
|
|
static int connect_ssl(utils_network_pt pNetwork)
|
|
{
|
|
if (NULL == pNetwork)
|
|
{
|
|
return STATE_SYS_DEPEND_NWK_INVALID_HANDLE;
|
|
}
|
|
|
|
pNetwork->handle = AT_SSL_Establish(
|
|
pNetwork->pHostAddress,
|
|
pNetwork->port,
|
|
pNetwork->ca_crt,
|
|
pNetwork->ca_crt_len + 1);
|
|
if (pNetwork->handle == (uintptr_t)(-1))
|
|
{
|
|
return STATE_SYS_DEPEND_NWK_INVALID_HANDLE;
|
|
}
|
|
|
|
return STATE_SUCCESS;
|
|
}
|
|
#else /* AT_SSL_ENABLED */
|
|
#ifdef INFRA_MEM_STATS
|
|
#include "infra_mem_stats.h"
|
|
#define NET_MALLOC(size) LITE_malloc(size, MEM_MAGIC, "infra_net")
|
|
#define NET_FREE(ptr) LITE_free(ptr)
|
|
extern int ssl_hooks_set(ssl_hooks_t *hooks);
|
|
#else
|
|
#define NET_MALLOC(size) HAL_Malloc(size)
|
|
#define NET_FREE(ptr) HAL_Free(ptr)
|
|
#endif
|
|
|
|
static void *ssl_malloc(uint32_t size)
|
|
{
|
|
return NET_MALLOC(size);
|
|
}
|
|
static void ssl_free(void *ptr)
|
|
{
|
|
NET_FREE(ptr);
|
|
}
|
|
|
|
static int read_ssl(utils_network_pt pNetwork, char *buffer, uint32_t len, uint32_t timeout_ms)
|
|
{
|
|
if (NULL == pNetwork)
|
|
{
|
|
return STATE_SYS_DEPEND_NWK_INVALID_HANDLE;
|
|
}
|
|
|
|
return HAL_SSL_Read((uintptr_t)pNetwork->handle, buffer, len, timeout_ms);
|
|
}
|
|
|
|
static int write_ssl(utils_network_pt pNetwork, const char *buffer, uint32_t len, uint32_t timeout_ms)
|
|
{
|
|
if (NULL == pNetwork)
|
|
{
|
|
return STATE_SYS_DEPEND_NWK_INVALID_HANDLE;
|
|
}
|
|
|
|
return HAL_SSL_Write((uintptr_t)pNetwork->handle, buffer, len, timeout_ms);
|
|
}
|
|
|
|
static int disconnect_ssl(utils_network_pt pNetwork)
|
|
{
|
|
if (NULL == pNetwork)
|
|
{
|
|
return STATE_SYS_DEPEND_NWK_INVALID_HANDLE;
|
|
}
|
|
|
|
HAL_SSL_Destroy((uintptr_t)pNetwork->handle);
|
|
pNetwork->handle = 0;
|
|
|
|
return STATE_SUCCESS;
|
|
}
|
|
|
|
static int connect_ssl(utils_network_pt pNetwork)
|
|
{
|
|
ssl_hooks_t ssl_hooks;
|
|
|
|
if (NULL == pNetwork)
|
|
{
|
|
return STATE_SYS_DEPEND_NWK_INVALID_HANDLE;
|
|
}
|
|
|
|
#ifdef INFRA_MEM_STATS
|
|
memset(&ssl_hooks, 0, sizeof(ssl_hooks_t));
|
|
ssl_hooks.malloc = ssl_malloc;
|
|
ssl_hooks.free = ssl_free;
|
|
|
|
ssl_hooks_set(&ssl_hooks);
|
|
#else
|
|
(void)ssl_hooks;
|
|
#endif
|
|
|
|
if (0 != (pNetwork->handle = (intptr_t)HAL_SSL_Establish(
|
|
pNetwork->pHostAddress,
|
|
pNetwork->port,
|
|
pNetwork->ca_crt,
|
|
pNetwork->ca_crt_len + 1)))
|
|
{
|
|
return STATE_SUCCESS;
|
|
}
|
|
else
|
|
{
|
|
/* TODO SHOLUD not remove this handle space */
|
|
/* The space will be freed by calling disconnect_ssl() */
|
|
/* utils_memory_free((void *)pNetwork->handle); */
|
|
return STATE_SYS_DEPEND_NWK_INVALID_HANDLE;
|
|
}
|
|
}
|
|
#endif /* AT_SSL_ENABLED */
|
|
#elif defined(AT_TCP_ENABLED)
|
|
#include "at_tcp.h"
|
|
/*** TCP connection ***/
|
|
static int read_tcp(utils_network_pt pNetwork, char *buffer, uint32_t len, uint32_t timeout_ms)
|
|
{
|
|
return AT_TCP_Read(pNetwork->handle, buffer, len, timeout_ms);
|
|
}
|
|
|
|
static int write_tcp(utils_network_pt pNetwork, const char *buffer, uint32_t len, uint32_t timeout_ms)
|
|
{
|
|
return AT_TCP_Write(pNetwork->handle, buffer, len, timeout_ms);
|
|
}
|
|
|
|
static int disconnect_tcp(utils_network_pt pNetwork)
|
|
{
|
|
if (pNetwork->handle == (uintptr_t)(-1))
|
|
{
|
|
return STATE_SYS_DEPEND_NWK_INVALID_HANDLE;
|
|
}
|
|
|
|
AT_TCP_Destroy(pNetwork->handle);
|
|
pNetwork->handle = (uintptr_t)(-1);
|
|
return STATE_SUCCESS;
|
|
}
|
|
|
|
static int connect_tcp(utils_network_pt pNetwork)
|
|
{
|
|
if (NULL == pNetwork)
|
|
{
|
|
return STATE_SYS_DEPEND_NWK_INVALID_HANDLE;
|
|
}
|
|
|
|
pNetwork->handle = AT_TCP_Establish(pNetwork->pHostAddress, pNetwork->port);
|
|
if (pNetwork->handle == (uintptr_t)(-1))
|
|
{
|
|
return STATE_SYS_DEPEND_NWK_INVALID_HANDLE;
|
|
}
|
|
|
|
return STATE_SUCCESS;
|
|
}
|
|
|
|
#endif /* #ifdef SUPPORT_TLS */
|
|
/*** TCP connection ***/
|
|
static int read_tcp(utils_network_pt pNetwork, char *buffer, uint32_t len, uint32_t timeout_ms)
|
|
{
|
|
return HAL_TCP_Read(pNetwork->handle, buffer, len, timeout_ms);
|
|
}
|
|
|
|
static int write_tcp(utils_network_pt pNetwork, const char *buffer, uint32_t len, uint32_t timeout_ms)
|
|
{
|
|
return HAL_TCP_Write(pNetwork->handle, buffer, len, timeout_ms);
|
|
}
|
|
|
|
static int disconnect_tcp(utils_network_pt pNetwork)
|
|
{
|
|
if (pNetwork->handle == (uintptr_t)(-1))
|
|
{
|
|
return STATE_SYS_DEPEND_NWK_INVALID_HANDLE;
|
|
}
|
|
|
|
HAL_TCP_Destroy(pNetwork->handle);
|
|
pNetwork->handle = (uintptr_t)(-1);
|
|
return STATE_SUCCESS;
|
|
}
|
|
|
|
static int connect_tcp(utils_network_pt pNetwork)
|
|
{
|
|
if (NULL == pNetwork)
|
|
{
|
|
return STATE_SYS_DEPEND_NWK_INVALID_HANDLE;
|
|
}
|
|
|
|
pNetwork->handle = HAL_TCP_Establish(pNetwork->pHostAddress, pNetwork->port);
|
|
if (pNetwork->handle == (uintptr_t)(-1))
|
|
{
|
|
return STATE_SYS_DEPEND_NWK_INVALID_HANDLE;
|
|
}
|
|
|
|
return STATE_SUCCESS;
|
|
}
|
|
|
|
/****** network interface ******/
|
|
int utils_net_read(utils_network_pt pNetwork, char *buffer, uint32_t len, uint32_t timeout_ms)
|
|
{
|
|
int ret = 0;
|
|
#if defined(SUPPORT_TLS)
|
|
if ((is_cert_host_addr == 0)&&(NULL != pNetwork->ca_crt))
|
|
{
|
|
ret = read_ssl(pNetwork, buffer, len, timeout_ms);
|
|
}
|
|
else
|
|
{
|
|
ret = read_tcp(pNetwork, buffer, len, timeout_ms);
|
|
}
|
|
#else
|
|
ret = read_tcp(pNetwork, buffer, len, timeout_ms);
|
|
#endif
|
|
|
|
return ret;
|
|
}
|
|
|
|
int utils_net_write(utils_network_pt pNetwork, const char *buffer, uint32_t len, uint32_t timeout_ms)
|
|
{
|
|
int ret = 0;
|
|
#if defined(SUPPORT_TLS)
|
|
if ((is_cert_host_addr == 0)&&(NULL != pNetwork->ca_crt))
|
|
{
|
|
ret = write_ssl(pNetwork, buffer, len, timeout_ms);
|
|
}
|
|
else
|
|
{
|
|
ret = write_tcp(pNetwork, buffer, len, timeout_ms);
|
|
}
|
|
#else
|
|
ret = write_tcp(pNetwork, buffer, len, timeout_ms);
|
|
#endif
|
|
|
|
return ret;
|
|
}
|
|
|
|
int iotx_net_disconnect(utils_network_pt pNetwork)
|
|
{
|
|
int ret = 0;
|
|
#if defined(SUPPORT_TLS)
|
|
if ((is_cert_host_addr == 0)&&(NULL != pNetwork->ca_crt))
|
|
{
|
|
ret = disconnect_ssl(pNetwork);
|
|
}
|
|
else
|
|
{
|
|
ret = disconnect_tcp(pNetwork);
|
|
}
|
|
#else
|
|
ret = disconnect_tcp(pNetwork);
|
|
#endif
|
|
|
|
return ret;
|
|
}
|
|
|
|
int iotx_net_connect(utils_network_pt pNetwork)
|
|
{
|
|
int ret = 0;
|
|
#if defined(SUPPORT_TLS)
|
|
if ((is_cert_host_addr == 0)&&(NULL != pNetwork->ca_crt))
|
|
{
|
|
ret = connect_ssl(pNetwork);
|
|
}
|
|
else
|
|
{
|
|
ret = connect_tcp(pNetwork);
|
|
}
|
|
|
|
#else
|
|
ret = connect_tcp(pNetwork);
|
|
#endif
|
|
|
|
return ret;
|
|
}
|
|
|
|
int iotx_net_init(utils_network_pt pNetwork, const char *host, uint16_t port, const char *ca_crt)
|
|
{
|
|
if (!pNetwork || !host)
|
|
{
|
|
return STATE_USER_INPUT_INVALID;
|
|
}
|
|
pNetwork->pHostAddress = host;
|
|
pNetwork->port = port;
|
|
|
|
#if defined(SUPPORT_TLS)
|
|
if (memcmp(pNetwork->pHostAddress, CERT_HOST_ADDRESS, strlen(pNetwork->pHostAddress)))
|
|
{
|
|
pNetwork->ca_crt = ca_crt;
|
|
printf("\n\n\nTLS==================================\n\n\n\n");
|
|
is_cert_host_addr = 0;
|
|
}
|
|
else
|
|
{
|
|
pNetwork->ca_crt = NULL;
|
|
printf("\n\n\n========TCP==================================\n\n\n\n");
|
|
is_cert_host_addr = 1;
|
|
}
|
|
#else
|
|
pNetwork->ca_crt = ca_crt;
|
|
#endif
|
|
if (NULL == ca_crt)
|
|
{
|
|
pNetwork->ca_crt_len = 0;
|
|
}
|
|
else
|
|
{
|
|
pNetwork->ca_crt_len = strlen(ca_crt);
|
|
}
|
|
|
|
pNetwork->handle = 0;
|
|
pNetwork->read = utils_net_read;
|
|
pNetwork->write = utils_net_write;
|
|
pNetwork->disconnect = iotx_net_disconnect;
|
|
pNetwork->connect = iotx_net_connect;
|
|
|
|
return STATE_SUCCESS;
|
|
}
|
|
#endif
|