11 KiB
TLSSocket - Design document
Overview and Background
This document describes design of TLSSocket class that provides simple interface for Mbed OS user to create TLS connections over TCP socket.
This class greatly simplifies the usage of TLS but limits itself to only one use case. This design limitation is accepted as other users can continue using Mbed TLS API directly. TLSSocket also exposes internal Mbed TLS structures allowing use of Mbed TLS API to configure the underlying library.
High level goal is to demonstrate Mbed OS users that secure connections are not hard to do.
Design Limitations
Following design limitations exist in the current design of TLSSocket
TLSSocket::connect()
is always blocking- Can only use server and client certificates through
set_root_ca_cert()
andset_client_cert_key()
methods. For other use cases, internal Mbed TLS structures are exposed. - No PSK mode
System Architecture and High-Level Design
Secure socket consist of two classes:
TLSocketWrapper
Which handles initialization of TLS library and does TLS handsake. Takes any Socket as a trasport.TLSSocket
Which inherits TLSocketWrapper and has TCP socket as a transport. Addsconnect(char *hostname)
for initiating the TCP and TLS handshakes at one call.
,--------------.
|Socket |
|--------------|
|--------------|
|int connect();|
|int recv(); |
|int send(); |
`--------------'
|
|
,----------------------------.
|TLSocketWrapper |
|----------------------------|
|Socket *transport; |
|----------------------------|
|TLSocketWrapper(*transport);|
|int set_root_ca_cert(...); |
|void set_hostname(...); |
|int do_handshake(); |
`----------------------------'
|
,----------------------------.
|TLSSocket |
|----------------------------|
|TCPSocket transport |
|----------------------------|
|int connect(char *hostname);|
`----------------------------'
High Level Design Goal 1: Abstract socket API
Secure socket both uses Socket
interface as its transport layer and implements it. This makes it transport independent and there is no direct dependency to IP stack.
When TLSSocket implements Socket API it is able to be used instead of TCP connection in any Mbed OS library. For example MQTT library is made secure without any code changes: https://github.com/coisme/Mbed-to-Azure-IoT-Hub/tree/new-TLSSocket
High Level Design Goal 2: Only certificate based authentication
Aim is to first support only certificate based authentication, so we implement only set_root_ca_cert()
and set_client_cert_key()
functions. For later on, different types of authentication methods can be added.
High Level Design Goal 3: Support both blocking and non-blocking operations
As the Mbed TLS is already work with both socket types, we are able to TLSocketWrapper
that can handle both types as well.
Functions set_blocking()
and set_timeout()
just pass the information for underlying
transport socket. Extra handling on the TLS state machine is not required.
High Level Design Goal 4: Expose Mbed TLS structures
Exposing Mbed TLS configuration structures allows user to configure the underlying TLS instance using Mbed TLS API. This allows greater usability as TLSSocket is not limited to only one use case.
Also configuration structures can be shared between sockets which leads to RAM saving then two or more TLSSockets are used with same parameters.
System Architecture and Component Interaction
Description and diagrams showing overall architecture of how the above-mentioned components/resources interface with each other.
Detailed Design
This section provides detailed design on implementation of components/modules mentioned in High Level Design section.
Detailed Design for Abstract socket API
Mbed OS Socket interface is a abstract C++ inteface that follows POSIX socket API.
Receiving and sending data from Mbed TLS
TLSSocketWrapper
contains static wrappers ssl_recv()
and ssl_send()
functions which are
registered to Mbed TLS library in mbedtls_ssl_set_bio()
call.
There functions then call the transport socket's Socket::send()
and Socket::recv()
calls
respectively. Error coded are passed through, except NSAPI_ERROR_WOULD_BLOCK
which is translated to MBEDTLS_ERR_SSL_WANT_WRITE
or MBEDTLS_ERR_SSL_WANT_READ
.
Providing Socket API
virtual nsapi_error_t close();
Destroys the memory allocated by TLS library.
Alternatively also closes the transport socket, unless TLSSocketWrapper::keep_transport_open()
has been called earlier.
virtual nsapi_error_t connect(const SocketAddress &address);
Initiates the TCP connection and continues to TLS hanshake. This is currently forced to blocking mode.
virtual nsapi_size_or_error_t send(const void *data, nsapi_size_t size);
virtual nsapi_size_or_error_t recv(void *data, nsapi_size_t size);
virtual nsapi_size_or_error_t sendto(const SocketAddress &address, const void *data, nsapi_size_t size);
virtual nsapi_size_or_error_t recvfrom(SocketAddress *address, void *data, nsapi_size_t size);
These work as expected, but SocketAddress
parameters are ignored. TLS connection cannot
change the peer. Also recvfrom()
call does not set the peer address.
Mbed TLS error codes MBEDTLS_ERR_SSL_WANT_READ
and MBEDTLS_ERR_SSL_WANT_WRITE
are
translated to NSAPI_ERROR_WOULD_BLOCK
before passing to user.
MBEDTLS_ERR_SSL_PEER_CLOSE_NOTIFY
is ignored and zero is returned to user (connection closed). Other error codes are passed through.
virtual nsapi_error_t bind(const SocketAddress &address);
virtual void set_blocking(bool blocking);
virtual void set_timeout(int timeout);
virtual void sigio(mbed::Callback<void()> func);
virtual nsapi_error_t setsockopt(int level, int optname, const void *optval, unsigned optlen);
virtual nsapi_error_t getsockopt(int level, int optname, void *optval, unsigned *optlen);
These are passed through to transport socket.
virtual Socket *accept(nsapi_error_t *error = NULL);
virtual nsapi_error_t listen(int backlog = 1);
These are returning NSAPI_ERROR_UNSUPPORTED
as TLS socket cannot be set to listening mode.
Detailed Design for certificate based authentication
TLSSocketWrapper
provides following API to set server certificate. You can use either BASE64 formatted PEM certificate, or binary DER certificates. Later form just assumes root_ca_pem
or client_cert_pem
to be standard C string and counts its lenght and passes to method which takes just void*
and len
.
/** Sets the certification of Root CA.
*
* @param root_ca Root CA Certificate in any mbed-TLS supported format.
* @param len Length of certificate (including terminating 0 for PEM).
*/
nsapi_error_t TLSSocketWrapper::set_root_ca_cert(const void *root_ca, size_t len);
/** Sets the certification of Root CA.
*
* @param root_ca_pem Root CA Certificate in PEM format
*/
nsapi_error_t TLSSocketWrapper::set_root_ca_cert(const char *root_ca_pem);
If client authentication is required, following API allows you to set the client certificate and private key:
/** Sets client certificate, and client private key.
*
* @param client_cert Client certification in any mbed-TLS supported format.
* @param client_private_key Client private key in PEM format.
*/
nsapi_error_t TLSSocketWrapper::set_client_cert_key(const void *client_cert_pem, size_t client_cert_len,
const void *client_private_key_pem, size_t client_private_key_len);
/** Sets client certificate, and client private key.
*
* @param client_cert_pem Client certification in PEM format.
* @param client_private_key Client private key in PEM format.
*/
nsapi_error_t TLSSocketWrapper::set_client_cert_key(const char *client_cert_pem, const char *client_private_key_pem);
Certificate is then passed unmodified to mbedtls_x509_crt_parse()
function.
Detailed Design for Support both blocking and non-blocking operations
send()
and receive()
methods do not need to know whether underlying socket is in
blocking mode as Mbed OS return values are enough to tell that.
Data path when application send data to TLS socket:
- Application call
TLSSocketWrapper::send()
function. - Which calls
mbedtls_ssl_write()
function - Which calls
TLSSocketWrapper::ssl_send()
- Which calls transport socket's
Socket::send()
which either blocks or returnsNSAPI_ERROR_WOULD_BLOCK
When this NSAPI_ERROR_WOULD_BLOCK
is returned:
TLSSocketWrapper::ssl_send()
translates error toMBEDTLS_ERR_SSL_WANT_WRITE
- returned to
mbedtls_ssl_write()
which return the same error TLSSocketWrapper::send()
gets theMBEDTLS_ERR_SSL_WANT_WRITE
and translates that toNSAPI_ERROR_WOULD_BLOCK
- Application gets
NSAPI_ERROR_WOULD_BLOCK
and back off.
When transport socket is in blocking mode, it never returns NSAPI_ERROR_WOULD_BLOCK
and therefore mbedtls_ssl_write()
never gets MBEDTLS_ERR_SSL_WANT_WRITE
, so any translation does not happen, but code path stays unchanged.
Detailed Design for exposing Mbed TLS structures
TLSSocket exposes following API to provide access to internal Mbed TLS data structures:
mbedtls_x509_crt *get_own_cert();
int set_own_cert(mbedtls_x509_crt *);
mbedtls_x509_crt *get_ca_chain();
void set_ca_chain(mbedtls_x509_crt *);
mbedtls_ssl_config *get_ssl_config();
void set_ssl_config(mbedtls_ssl_config *);
This allows sockets to share same configuration and allow user to fine tune TLS configuration, for example:
TLSSocket a;
TLSSocket b;
a.set_root_ca_cert(<cert>);
b.set_ssl_config(a.get_ssl_config());
Usage Scenarios and Examples
Scenario 1: Connecting to secure server:
NetworkInterface *net = NetworkInterface::get_default_instance();
TLSSocket sock(net);
sock.add_root_CA(...);
sock.connect("my_server.mbed.com", 8080); // Connect & handle TLS handshake
// Now Socket is connected and can be used as any socket
sock.send("hello", 5);
Example user: https://github.com/coisme/Mbed-to-Azure-IoT-Hub/blob/new-TLSSocket/MQTTNetwork.h
Tools and Configuration Changes
No tool changes required
Other Information
Reusability
Parts of the state machine are probably relevant when implementing DTLS socket.
TLSSocketWrapper is entirely reusable when doing TLS handshake using any socket type. It does not have tight bindings to TCP.
Assumptions
We are assuming that server certificate is given from the application to TLSSocket::set_root_ca_cert()
interface in a format that is understood by Mbed TLS.
Deprecations
No deprecations