crypto/async/arch/async_win.c

   1 /*
   2  * Written by Matt Caswell (matt@openssl.org) for the OpenSSL project.
   3  */
   4 /* ====================================================================
   5  * Copyright (c) 2015 The OpenSSL Project.  All rights reserved.
   6  *
   7  * Redistribution and use in source and binary forms, with or without
   8  * modification, are permitted provided that the following conditions
   9  * are met:
  10  *
  11  * 1. Redistributions of source code must retain the above copyright
  12  *    notice, this list of conditions and the following disclaimer.
  13  *
  14  * 2. Redistributions in binary form must reproduce the above copyright
  15  *    notice, this list of conditions and the following disclaimer in
  16  *    the documentation and/or other materials provided with the
  17  *    distribution.
  18  *
  19  * 3. All advertising materials mentioning features or use of this
  20  *    software must display the following acknowledgment:
  21  *    "This product includes software developed by the OpenSSL Project
  22  *    for use in the OpenSSL Toolkit. (http://www.OpenSSL.org/)"
  23  *
  24  * 4. The names "OpenSSL Toolkit" and "OpenSSL Project" must not be used to
  25  *    endorse or promote products derived from this software without
  26  *    prior written permission. For written permission, please contact
  27  *    licensing@OpenSSL.org.
  28  *
  29  * 5. Products derived from this software may not be called "OpenSSL"
  30  *    nor may "OpenSSL" appear in their names without prior written
  31  *    permission of the OpenSSL Project.
  32  *
  33  * 6. Redistributions of any form whatsoever must retain the following
  34  *    acknowledgment:
  35  *    "This product includes software developed by the OpenSSL Project
  36  *    for use in the OpenSSL Toolkit (http://www.OpenSSL.org/)"
  37  *
  38  * THIS SOFTWARE IS PROVIDED BY THE OpenSSL PROJECT ``AS IS'' AND ANY
  39  * EXPRESSED OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  40  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
  41  * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE OpenSSL PROJECT OR
  42  * ITS CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
  43  * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
  44  * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
  45  * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  46  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT,
  47  * STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
  48  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED
  49  * OF THE POSSIBILITY OF SUCH DAMAGE.
  50  * ====================================================================
  51  */
  52
  53 /* This must be the first #include file */
  54 #include "../async_locl.h"
  55
  56 #ifdef ASYNC_WIN
  57
  58 # include <windows.h>
  59 # include "internal/cryptlib.h"
  60
  61 struct winpool {
  62     STACK_OF(ASYNC_JOB) *pool;
  63     size_t curr_size;
  64     size_t max_size;
  65 };
  66
  67 static DWORD asyncwinpool = 0;
  68 static DWORD asyncwinctx = 0;
  69 static DWORD asyncwindispatch = 0;
  70
  71
  72 void async_start_func(void);
  73
  74 int async_global_init(void)
  75 {
  76     asyncwinpool = TlsAlloc();
  77     asyncwinctx = TlsAlloc();
  78     asyncwindispatch = TlsAlloc();
  79     if (asyncwinpool == TLS_OUT_OF_INDEXES || asyncwinctx == TLS_OUT_OF_INDEXES
  80             || asyncwindispatch == TLS_OUT_OF_INDEXES) {
  81         if (asyncwinpool != TLS_OUT_OF_INDEXES) {
  82             TlsFree(asyncwinpool);
  83         }
  84         if (asyncwinctx != TLS_OUT_OF_INDEXES) {
  85             TlsFree(asyncwinctx);
  86         }
  87         if (asyncwindispatch != TLS_OUT_OF_INDEXES) {
  88             TlsFree(asyncwindispatch);
  89         }
  90         return 0;
  91     }
  92     return 1;
  93 }
  94
  95 int async_local_init(void)
  96 {
  97     return (TlsSetValue(asyncwinpool, NULL) != 0)
  98         && (TlsSetValue(asyncwinctx, NULL) != 0)
  99         && (TlsSetValue(asyncwindispatch, NULL) != 0);
 100 }
 101
 102 void async_local_cleanup(void)
 103 {
 104     async_ctx *ctx = async_get_ctx();
 105     if (ctx != NULL) {
 106         async_fibre *fibre = &ctx->dispatcher;
 107         if(fibre != NULL && fibre->fibre != NULL && fibre->converted) {
 108             ConvertFiberToThread();
 109             fibre->fibre = NULL;
 110         }
 111     }
 112 }
 113
 114 void async_global_cleanup(void)
 115 {
 116     TlsFree(asyncwinpool);
 117     TlsFree(asyncwinctx);
 118     TlsFree(asyncwindispatch);
 119     asyncwinpool = 0;
 120     asyncwinctx = 0;
 121     asyncwindispatch = 0;
 122 }
 123
 124 int async_fibre_init_dispatcher(async_fibre *fibre)
 125 {
 126     LPVOID dispatcher;
 127
 128     dispatcher = (LPVOID)TlsGetValue(asyncwindispatch);
 129     if (dispatcher == NULL) {
 130         fibre->fibre = ConvertThreadToFiber(NULL);
 131         if (fibre->fibre == NULL) {
 132             fibre->converted = 0;
 133             fibre->fibre = GetCurrentFiber();
 134             if (fibre->fibre == NULL)
 135                 return 0;
 136         } else {
 137             fibre->converted = 1;
 138         }
 139         if (TlsSetValue(asyncwindispatch, (LPVOID)fibre->fibre) == 0)
 140             return 0;
 141     } else {
 142         fibre->fibre = dispatcher;
 143     }
 144     return 1;
 145 }
 146
 147 VOID CALLBACK async_start_func_win(PVOID unused)
 148 {
 149     async_start_func();
 150 }
 151
 152 int async_pipe(OSSL_ASYNC_FD *pipefds)
 153 {
 154     if (CreatePipe(&pipefds[0], &pipefds[1], NULL, 256) == 0)
 155         return 0;
 156
 157     return 1;
 158 }
 159
 160 int async_close_fd(OSSL_ASYNC_FD fd)
 161 {
 162     if (CloseHandle(fd) == 0)
 163         return 0;
 164
 165     return 1;
 166 }
 167
 168 int async_write1(OSSL_ASYNC_FD fd, const void *buf)
 169 {
 170     DWORD numwritten = 0;
 171
 172     if (WriteFile(fd, buf, 1, &numwritten, NULL) && numwritten == 1)
 173         return 1;
 174
 175     return 0;
 176 }
 177
 178 int async_read1(OSSL_ASYNC_FD fd, void *buf)
 179 {
 180     DWORD numread = 0;
 181
 182     if (ReadFile(fd, buf, 1, &numread, NULL) && numread == 1)
 183         return 1;
 184
 185     return 0;
 186 }
 187
 188 async_pool *async_get_pool(void)
 189 {
 190     return (async_pool *)TlsGetValue(asyncwinpool);
 191 }
 192
 193
 194 int async_set_pool(async_pool *pool)
 195 {
 196     return TlsSetValue(asyncwinpool, (LPVOID)pool) != 0;
 197 }
 198
 199 async_ctx *async_get_ctx(void)
 200 {
 201     return (async_ctx *)TlsGetValue(asyncwinctx);
 202 }
 203
 204 int async_set_ctx(async_ctx *ctx)
 205 {
 206     return TlsSetValue(asyncwinctx, (LPVOID)ctx) != 0;
 207 }
 208
 209 #endif