ustream-io-openssl.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 static int
  27 s_ustream_new(BIO *b)
  28 {
  29         b->init = 1;
  30         b->num = 0;
  31         b->ptr = NULL;
  32         b->flags = 0;
  33         return 1;
  34 }
  35
  36 static int
  37 s_ustream_free(BIO *b)
  38 {
  39         if (!b)
  40                 return 0;
  41
  42         b->ptr = NULL;
  43         b->init = 0;
  44         b->flags = 0;
  45         return 1;
  46 }
  47
  48 static int
  49 s_ustream_read(BIO *b, char *buf, int len)
  50 {
  51         struct ustream *s;
  52         char *sbuf;
  53         int slen;
  54
  55         if (!buf || len <= 0)
  56                 return 0;
  57
  58         s = (struct ustream *)b->ptr;
  59         if (!s)
  60                 return 0;
  61
  62         sbuf = ustream_get_read_buf(s, &slen);
  63
  64         BIO_clear_retry_flags(b);
  65         if (!slen) {
  66                 BIO_set_retry_read(b);
  67                 return -1;
  68         }
  69
  70         if (slen > len)
  71                 slen = len;
  72
  73         memcpy(buf, sbuf, slen);
  74         ustream_consume(s, slen);
  75
  76         return slen;
  77 }
  78
  79 static int
  80 s_ustream_write(BIO *b, const char *buf, int len)
  81 {
  82         struct ustream *s;
  83
  84         if (!buf || len <= 0)
  85                 return 0;
  86
  87         s = (struct ustream *)b->ptr;
  88         if (!s)
  89                 return 0;
  90
  91         return ustream_write(s, buf, len, false);
  92 }
  93
  94 static int
  95 s_ustream_gets(BIO *b, char *buf, int len)
  96 {
  97         return -1;
  98 }
  99
 100 static int
 101 s_ustream_puts(BIO *b, const char *str)
 102 {
 103         return s_ustream_write(b, str, strlen(str));
 104 }
 105
 106 static long s_ustream_ctrl(BIO *b, int cmd, long num, void *ptr)
 107 {
 108         switch (cmd) {
 109         case BIO_CTRL_FLUSH:
 110                 return 1;
 111         default:
 112                 return 0;
 113         };
 114 }
 115
 116 static BIO_METHOD methods_ustream = {
 117         100 | BIO_TYPE_SOURCE_SINK,
 118         "ustream",
 119         s_ustream_write,
 120         s_ustream_read,
 121         s_ustream_puts,
 122         s_ustream_gets,
 123         s_ustream_ctrl,
 124         s_ustream_new,
 125         s_ustream_free,
 126         NULL,
 127 };
 128
 129 static BIO *ustream_bio_new(struct ustream *s)
 130 {
 131         BIO *bio;
 132
 133         bio = BIO_new(&methods_ustream);
 134         bio->ptr = s;
 135         return bio;
 136 }
 137
 138 __hidden void ustream_set_io(void *ctx, void *ssl, struct ustream *conn)
 139 {
 140         BIO *bio = ustream_bio_new(conn);
 141         SSL_set_bio(ssl, bio, bio);
 142 }