tls.c

   1 /*
   2  * uhttpd - Tiny single-threaded httpd - Main header
   3  *
   4  *   Copyright (C) 2010-2012 Jo-Philipp Wich <xm@subsignal.org>
   5  *   Copyright (C) 2012 Felix Fietkau <nbd@openwrt.org>
   6  *
   7  *  Licensed under the Apache License, Version 2.0 (the "License");
   8  *  you may not use this file except in compliance with the License.
   9  *  You may obtain a copy of the License at
  10  *
  11  *      http://www.apache.org/licenses/LICENSE-2.0
  12  *
  13  *  Unless required by applicable law or agreed to in writing, software
  14  *  distributed under the License is distributed on an "AS IS" BASIS,
  15  *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  16  *  See the License for the specific language governing permissions and
  17  *  limitations under the License.
  18  */
  19
  20 #include <dlfcn.h>
  21 #include "uhttpd.h"
  22 #include "tls.h"
  23
  24 #ifdef __APPLE__
  25 #define LIB_EXT "dylib"
  26 #else
  27 #define LIB_EXT "so"
  28 #endif
  29
  30 static struct ustream_ssl_ops *ops;
  31 static void *dlh;
  32 static void *ctx;
  33
  34 int uh_tls_init(const char *key, const char *crt)
  35 {
  36         static bool _init = false;
  37
  38         if (_init)
  39                 return 0;
  40
  41         _init = true;
  42         dlh = dlopen("libustream-ssl." LIB_EXT, RTLD_LAZY | RTLD_LOCAL);
  43         if (!dlh) {
  44                 fprintf(stderr, "Failed to load ustream-ssl library: %s\n", dlerror());
  45                 return -ENOENT;
  46         }
  47
  48         ops = dlsym(dlh, "ustream_ssl_ops");
  49         if (!ops) {
  50                 fprintf(stderr, "Could not find required symbol 'ustream_ssl_ops' in ustream-ssl library\n");
  51                 return -ENOENT;
  52         }
  53
  54         ctx = ops->context_new(true);
  55         if (!ctx) {
  56                 fprintf(stderr, "Failed to initialize ustream-ssl\n");
  57                 return -EINVAL;
  58         }
  59
  60         if (ops->context_set_crt_file(ctx, crt) ||
  61             ops->context_set_key_file(ctx, key)) {
  62                 fprintf(stderr, "Failed to load certificate/key files\n");
  63                 return -EINVAL;
  64         }
  65
  66         return 0;
  67 }
  68
  69 static void tls_ustream_read_cb(struct ustream *s, int bytes)
  70 {
  71         struct client *cl = container_of(s, struct client, ssl);
  72
  73         uh_client_read_cb(cl);
  74 }
  75
  76 static void tls_ustream_write_cb(struct ustream *s, int bytes)
  77 {
  78         struct client *cl = container_of(s, struct client, ssl);
  79
  80         if (cl->dispatch.write_cb)
  81                 cl->dispatch.write_cb(cl);
  82 }
  83
  84 static void tls_notify_state(struct ustream *s)
  85 {
  86         struct client *cl = container_of(s, struct client, ssl);
  87
  88         uh_client_notify_state(cl);
  89 }
  90
  91 void uh_tls_client_attach(struct client *cl)
  92 {
  93         cl->us = &cl->ssl.stream;
  94         ops->init(&cl->ssl, &cl->sfd.stream, ctx, true);
  95         cl->us->notify_read = tls_ustream_read_cb;
  96         cl->us->notify_write = tls_ustream_write_cb;
  97         cl->us->notify_state = tls_notify_state;
  98 }
  99
 100 void uh_tls_client_detach(struct client *cl)
 101 {
 102         ustream_free(&cl->ssl.stream);
 103 }