wolfssl-w32/mqx/wolfssl_client/Sources/main.c

294 lines
8.2 KiB
C

/*
* main.c
*/
#include "main.h"
#include "util.h"
#if !BSPCFG_ENABLE_IO_SUBSYSTEM
#error This application requires BSPCFG_ENABLE_IO_SUBSYSTEM defined \
non-zero in user_config.h. Please recompile BSP with this option.
#endif
#ifndef BSP_DEFAULT_IO_CHANNEL_DEFINED
#error This application requires BSP_DEFAULT_IO_CHANNEL to be not NULL. \
Please set corresponding BSPCFG_ENABLE_TTYx to non-zero in \
user_config.h and recompile BSP with this option.
#endif
#if defined BSP_SDCARD_ESDHC_CHANNEL
#if ! BSPCFG_ENABLE_ESDHC
#error This application requires BSPCFG_ENABLE_ESDHC defined non-zero in \
user_config.h. Please recompile libraries with this option.
#endif
#elif defined BSP_SDCARD_SDHC_CHANNEL
#if ! BSPCFG_ENABLE_SDHC
#error This application requires BSPCFG_ENABLE_SDHC defined non-zero in \
user_config.h. Please recompile libraries with this option.
#endif
#endif
#if defined (BSP_SDCARD_SPI_CHANNEL)
#define SDCARD_COM_CHANNEL BSP_SDCARD_SPI_CHANNEL
#elif defined (BSP_SDCARD_ESDHC_CHANNEL)
#define SDCARD_COM_CHANNEL BSP_SDCARD_ESDHC_CHANNEL
#elif defined (BSP_SDCARD_SDHC_CHANNEL)
#define SDCARD_COM_CHANNEL BSP_SDCARD_SDHC_CHANNEL
#else
#error "SDCARD low level communication device not defined!"
#endif
TASK_TEMPLATE_STRUCT MQX_template_list[] =
{
/* Task number, Entry point, Stack, Pri, String, Auto? */
{MAIN_TASK, Main_task, 20000, 9, "main", MQX_AUTO_START_TASK},
{0, 0, 0, 0, 0, 0, }
};
/*TASK*-----------------------------------------------------
*
* Task Name : Main_task
* Comments :
* This task sets up the SD card and Ethernet devices,
* then starts the example wolfSSL client. The example
* wolfSSL client connects to a server over TLS and sends
* a simple HTTP GET message, then prints out the reply
* from the server.
*
* To change the IP address and port of the server,
* change the wolfsslIP and wolfsslPort variables in
* client_test(). Note that wolfsslIP needs to be given
* in hexadecimal.
*
*END*-----------------------------------------------------*/
void Main_task(uint32_t initial_data)
{
int ret = 0;
char filesystem_name[] = "a:";
char partman_name[] = "pm:";
MQX_FILE_PTR com_handle, sdcard_handle, filesystem_handle, partman_handle;
printf("Starting client example... \n");
ret = sdcard_open(&com_handle, &sdcard_handle, &partman_handle,
&filesystem_handle, partman_name, filesystem_name);
if (ret != 0) {
printf("error: sdcard_open(), ret = %d\n", ret);
_mqx_exit(1);
}
printf("SD card installed to %s\n", filesystem_name);
setup_ethernet();
setup_clock();
client_test();
ret = sdcard_close(&sdcard_handle, &partman_handle, &filesystem_handle,
partman_name, filesystem_name);
if (ret != 0) {
printf("error: sdcard_close(), ret = %d\n", ret);
_mqx_exit(1);
}
printf("SD card uninstalled.\n");
_mqx_exit(0);
}
void setup_ethernet(void)
{
int error;
_enet_handle ehandle; /* for Ethernet driver */
_rtcs_if_handle ihandle;
_enet_address address;
error = RTCS_create();
if (error) {
err_sys("failed to create RTCS");
}
ENET_get_mac_address(BSP_DEFAULT_ENET_DEVICE, ENET_IPADDR, address);
/* Set up the Ethernet driver */
error = ENET_initialize(BSP_DEFAULT_ENET_DEVICE, address, 0, &ehandle);
if (error)
err_sys("failed to initialize Ethernet driver");
error = RTCS_if_add(ehandle, RTCS_IF_ENET, &ihandle);
if (error)
err_sys("failed to add interface for Ethernet");
error = RTCS_if_bind(ihandle, ENET_IPADDR, ENET_IPMASK);
if (error)
err_sys("failed to bind interface for Ethernet");
#ifdef GATE_IPADDR
RTCS_gate_add(GATE_IPADDR, INADDR_ANY, INADDR_ANY);
#endif
printf("Ethernet device %d bound to %X\n", BSP_DEFAULT_ENET_DEVICE,
ENET_IPADDR);
}
void setup_clock(void)
{
uint32_t ret = 0, i = 0;
uint32_t sntp_connected = 0;
uint32_t sntp_max_tries = 3;
TIME_STRUCT time_s;
DATE_STRUCT date_s;
/* NTP server: nist1-lnk.binary.net */
_ip_address ipaddr = IPADDR(216,229,0,179);
for (i = 0; i < sntp_max_tries; i++) {
printf("Getting time from NTP server [ attempt %u of %u ]...\n",
i+1, sntp_max_tries);
/* update time from NTP server */
ret = SNTP_oneshot(ipaddr, 5000);
if (ret == RTCS_OK) {
sntp_connected = 1;
printf("SNTP successfully updated device time\n");
break;
} else if (ret == RTCSERR_TIMEOUT) {
printf("SNTP attempt timed out.\n");
}
_time_delay(1000);
}
if (sntp_connected == 0) {
err_sys("SNTP failed to update device time");
}
/* print device time, for debug purposes */
_time_get(&time_s);
_time_to_date(&time_s, &date_s);
printf("Current time: %02d/%02d/%02d %02d:%02d:%02d\n",
date_s.YEAR, date_s.MONTH, date_s.DAY, date_s.HOUR, date_s.MINUTE,
date_s.SECOND);
return;
}
int myVerify(int preverify, WOLFSSL_X509_STORE_CTX* store)
{
(void)preverify;
char buffer[80];
printf("In verification callback, error = %d, %s\n",
store->error, wolfSSL_ERR_error_string(store->error, buffer));
return 0;
}
void client_test(void)
{
char msg[64];
char reply[1024];
int sockfd, input;
int ret = 0, msgSz = 0;
struct sockaddr_in servaddr;
WOLFSSL_CTX* ctx;
WOLFSSL* ssl;
long wolfsslIP = IPADDR(192,168,1,125);
long wolfsslPort = 11111;
/* for debug, compile wolfSSL with DEBUG_WOLFSSL defined */
wolfSSL_Debugging_ON();
wolfSSL_Init();
ctx = wolfSSL_CTX_new(wolfTLSv1_2_client_method());
if (ctx == 0)
err_sys("setting up ctx");
wolfSSL_CTX_set_verify(ctx, SSL_VERIFY_PEER, myVerify);
ret = wolfSSL_CTX_use_certificate_file(ctx, clientCert, SSL_FILETYPE_PEM);
if (ret != SSL_SUCCESS)
err_sys("can't load client cert file, check file");
ret = wolfSSL_CTX_use_PrivateKey_file(ctx, clientKey, SSL_FILETYPE_PEM);
if (ret != SSL_SUCCESS)
err_sys("can't load client key file, check file");
ret = wolfSSL_CTX_load_verify_locations(ctx, caCert, 0);
if (ret != SSL_SUCCESS)
err_sys("can't load CA cert file, check file");
/* create socket descriptor */
sockfd = socket(AF_INET, SOCK_STREAM, 0);
if (sockfd == RTCS_SOCKET_ERROR) {
err_sys("socket creation failed");
} else {
printf("socket created successfully\n");
}
/*
* Unlike most TCP/IP stacks, RTCS requires that sin_port and
* sin_addr needs to be in Host Byte Order, not Network Byte Order.
* This means we shouldn't use htons() when setting these values.
*/
memset((char*)&servaddr, 0, sizeof(servaddr));
servaddr.sin_family = AF_INET;
servaddr.sin_port = wolfsslPort;
servaddr.sin_addr.s_addr = wolfsslIP;
ret = connect(sockfd, &servaddr, sizeof(servaddr));
if (ret != RTCS_OK) {
err_sys("connect() failed");
} else {
printf("Connected to %lx, port %d.\n", servaddr.sin_addr.s_addr,
servaddr.sin_port);
}
if ( (ssl = wolfSSL_new(ctx)) == NULL)
err_sys("wolfSSL_new failed");
ret = wolfSSL_set_fd(ssl, sockfd);
if (ret != SSL_SUCCESS)
err_sys("wolfSSL_set_fd failed");
ret = wolfSSL_connect(ssl);
if (ret != SSL_SUCCESS)
err_sys("wolfSSL_connect failed");
printf("wolfSSL_connect() ok, sending GET...\n");
msgSz = 28;
strncpy(msg, "GET /index.html HTTP/1.0\r\n\r\n", msgSz);
if (wolfSSL_write(ssl, msg, msgSz) != msgSz)
err_sys("wolfSSL_write() failed");
input = wolfSSL_read(ssl, reply, sizeof(reply)-1);
if (input > 0) {
reply[input] = 0;
printf("Server response: %s\n", reply);
while (1) {
input = wolfSSL_read(ssl, reply, sizeof(reply)-1);
if (input > 0) {
reply[input] = 0;
printf("%s\n", reply);
} else {
break;
}
}
}
wolfSSL_shutdown(ssl);
wolfSSL_free(ssl);
wolfSSL_CTX_free(ctx);
wolfSSL_Cleanup();
}
/* EOF */