add mbedtls variant
[project/ustream-ssl.git] / ustream-io-cyassl.c
1 /*
2  * ustream-ssl - library for SSL over ustream
3  *
4  * Copyright (C) 2012 Felix Fietkau <nbd@openwrt.org>
5  *
6  * Permission to use, copy, modify, and/or distribute this software for any
7  * purpose with or without fee is hereby granted, provided that the above
8  * copyright notice and this permission notice appear in all copies.
9  *
10  * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
11  * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
12  * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
13  * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
14  * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
15  * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
16  * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
17  */
18
19 #include <string.h>
20
21 #include <libubox/ustream.h>
22
23 #include "ustream-ssl.h"
24 #include "ustream-internal.h"
25
26 #ifdef HAVE_CYASSL_VERSION_H
27 #include <cyassl/version.h>
28 #else
29 #define LIBCYASSL_VERSION_HEX 0
30 #endif
31
32 static int s_ustream_read(char *buf, int len, void *ctx)
33 {
34         struct ustream *s = ctx;
35         char *sbuf;
36         int slen;
37
38         if (s->eof)
39                 return -3;
40
41         sbuf = ustream_get_read_buf(s, &slen);
42         if (slen > len)
43                 slen = len;
44
45         if (!slen)
46                 return -2;
47
48         memcpy(buf, sbuf, slen);
49         ustream_consume(s, slen);
50
51         return slen;
52 }
53
54 static int s_ustream_write(char *buf, int len, void *ctx)
55 {
56         struct ustream *s = ctx;
57
58         if (s->write_error)
59                 return len;
60
61         return ustream_write(s, buf, len, false);
62 }
63
64 #if (LIBCYASSL_VERSION_HEX > 0)
65 static int io_recv_cb(SSL* ssl, char *buf, int sz, void *ctx)
66 {
67         return s_ustream_read(buf, sz, ctx);
68 }
69
70 static int io_send_cb(SSL* ssl, char *buf, int sz, void *ctx)
71 {
72         return s_ustream_write(buf, sz, ctx);
73 }
74 #else
75 /* not defined in the header file */
76 typedef int (*CallbackIORecv)(char *buf, int sz, void *ctx);
77 typedef int (*CallbackIOSend)(char *buf, int sz, void *ctx);
78
79 void SetCallbackIORecv_Ctx(SSL_CTX*, CallbackIORecv);
80 void SetCallbackIOSend_Ctx(SSL_CTX*, CallbackIOSend);
81 void SetCallbackIO_ReadCtx(SSL* ssl, void *rctx);
82 void SetCallbackIO_WriteCtx(SSL* ssl, void *wctx);
83
84 #define CyaSSL_SetIOReadCtx SetCallbackIO_ReadCtx
85 #define CyaSSL_SetIOWriteCtx SetCallbackIO_WriteCtx
86 #define CyaSSL_SetIORecv SetCallbackIORecv_Ctx
87 #define CyaSSL_SetIOSend SetCallbackIOSend_Ctx
88
89 static int io_recv_cb(char *buf, int sz, void *ctx)
90 {
91         return s_ustream_read(buf, sz, ctx);
92 }
93
94 static int io_send_cb(char *buf, int sz, void *ctx)
95 {
96         return s_ustream_write(buf, sz, ctx);
97 }
98 #endif
99
100 __hidden void ustream_set_io(struct ustream_ssl_ctx *ctx, void *ssl, struct ustream *conn)
101 {
102         CyaSSL_SetIOReadCtx(ssl, conn);
103         CyaSSL_SetIOWriteCtx(ssl, conn);
104         CyaSSL_SetIORecv((void *) ctx, io_recv_cb);
105         CyaSSL_SetIOSend((void *) ctx, io_send_cb);
106 }