157 lines
4 KiB
Arduino
157 lines
4 KiB
Arduino
|
/* wolfssl_client.ino
|
||
|
*
|
||
|
* Copyright (C) 2006-2020 wolfSSL Inc.
|
||
|
*
|
||
|
* This file is part of wolfSSL.
|
||
|
*
|
||
|
* wolfSSL 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.
|
||
|
*
|
||
|
* wolfSSL 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-1335, USA
|
||
|
*/
|
||
|
|
||
|
|
||
|
#include <wolfssl.h>
|
||
|
#include <wolfssl/ssl.h>
|
||
|
#include <Ethernet.h>
|
||
|
|
||
|
const char host[] = "192.168.1.148"; // server to connect to
|
||
|
const int port = 11111; // port on server to connect to
|
||
|
|
||
|
int EthernetSend(WOLFSSL* ssl, char* msg, int sz, void* ctx);
|
||
|
int EthernetReceive(WOLFSSL* ssl, char* reply, int sz, void* ctx);
|
||
|
int reconnect = 10;
|
||
|
|
||
|
EthernetClient client;
|
||
|
|
||
|
WOLFSSL_CTX* ctx = NULL;
|
||
|
WOLFSSL* ssl = NULL;
|
||
|
|
||
|
void setup() {
|
||
|
WOLFSSL_METHOD* method;
|
||
|
|
||
|
Serial.begin(9600);
|
||
|
|
||
|
method = wolfTLSv1_2_client_method();
|
||
|
if (method == NULL) {
|
||
|
Serial.println("unable to get method");
|
||
|
return;
|
||
|
}
|
||
|
ctx = wolfSSL_CTX_new(method);
|
||
|
if (ctx == NULL) {
|
||
|
Serial.println("unable to get ctx");
|
||
|
return;
|
||
|
}
|
||
|
// initialize wolfSSL using callback functions
|
||
|
wolfSSL_CTX_set_verify(ctx, SSL_VERIFY_NONE, 0);
|
||
|
wolfSSL_SetIOSend(ctx, EthernetSend);
|
||
|
wolfSSL_SetIORecv(ctx, EthernetReceive);
|
||
|
|
||
|
return;
|
||
|
}
|
||
|
|
||
|
int EthernetSend(WOLFSSL* ssl, char* msg, int sz, void* ctx) {
|
||
|
int sent = 0;
|
||
|
|
||
|
sent = client.write((byte*)msg, sz);
|
||
|
|
||
|
return sent;
|
||
|
}
|
||
|
|
||
|
int EthernetReceive(WOLFSSL* ssl, char* reply, int sz, void* ctx) {
|
||
|
int ret = 0;
|
||
|
|
||
|
while (client.available() > 0 && ret < sz) {
|
||
|
reply[ret++] = client.read();
|
||
|
}
|
||
|
|
||
|
return ret;
|
||
|
}
|
||
|
|
||
|
void loop() {
|
||
|
int err = 0;
|
||
|
int input = 0;
|
||
|
int total_input = 0;
|
||
|
char msg[32] = "hello wolfssl!";
|
||
|
int msgSz = (int)strlen(msg);
|
||
|
char errBuf[80];
|
||
|
char reply[80];
|
||
|
const char* cipherName;
|
||
|
|
||
|
if (reconnect) {
|
||
|
reconnect--;
|
||
|
|
||
|
if (client.connect(host, port)) {
|
||
|
|
||
|
Serial.print("Connected to ");
|
||
|
Serial.println(host);
|
||
|
|
||
|
ssl = wolfSSL_new(ctx);
|
||
|
if (ssl == NULL) {
|
||
|
Serial.println("Unable to allocate SSL object");
|
||
|
return;
|
||
|
}
|
||
|
|
||
|
err = wolfSSL_connect(ssl);
|
||
|
if (err != WOLFSSL_SUCCESS) {
|
||
|
err = wolfSSL_get_error(ssl, 0);
|
||
|
wolfSSL_ERR_error_string(err, errBuf);
|
||
|
Serial.print("TLS Connect Error: ");
|
||
|
Serial.println(errBuf);
|
||
|
}
|
||
|
|
||
|
Serial.print("SSL version is ");
|
||
|
Serial.println(wolfSSL_get_version(ssl));
|
||
|
|
||
|
cipherName = wolfSSL_get_cipher(ssl);
|
||
|
Serial.print("SSL cipher suite is ");
|
||
|
Serial.println(cipherName);
|
||
|
|
||
|
if ((wolfSSL_write(ssl, msg, msgSz)) == msgSz) {
|
||
|
|
||
|
Serial.print("Server response: ");
|
||
|
while (client.available() || wolfSSL_pending(ssl)) {
|
||
|
input = wolfSSL_read(ssl, reply, sizeof(reply) - 1);
|
||
|
total_input += input;
|
||
|
if (input < 0) {
|
||
|
err = wolfSSL_get_error(ssl, 0);
|
||
|
wolfSSL_ERR_error_string(err, errBuf);
|
||
|
Serial.print("TLS Read Error: ");
|
||
|
Serial.println(errBuf);
|
||
|
break;
|
||
|
} else if (input > 0) {
|
||
|
reply[input] = '\0';
|
||
|
Serial.print(reply);
|
||
|
} else {
|
||
|
Serial.println();
|
||
|
}
|
||
|
}
|
||
|
} else {
|
||
|
err = wolfSSL_get_error(ssl, 0);
|
||
|
wolfSSL_ERR_error_string(err, errBuf);
|
||
|
Serial.print("TLS Write Error: ");
|
||
|
Serial.println(errBuf);
|
||
|
}
|
||
|
|
||
|
wolfSSL_shutdown(ssl);
|
||
|
wolfSSL_free(ssl);
|
||
|
|
||
|
client.stop();
|
||
|
Serial.println("Connection complete.");
|
||
|
reconnect = 0;
|
||
|
} else {
|
||
|
Serial.println("Trying to reconnect...");
|
||
|
}
|
||
|
}
|
||
|
delay(1000);
|
||
|
}
|