Home | History | Annotate | Download | only in https
      1 /*
      2   This file is part of libmicrohttpd
      3   Copyright (C) 2007 Christian Grothoff
      4 
      5   libmicrohttpd is free software; you can redistribute it and/or modify
      6   it under the terms of the GNU General Public License as published
      7   by the Free Software Foundation; either version 2, or (at your
      8   option) any later version.
      9 
     10   libmicrohttpd is distributed in the hope that it will be useful, but
     11   WITHOUT ANY WARRANTY; without even the implied warranty of
     12   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
     13   General Public License for more details.
     14 
     15   You should have received a copy of the GNU General Public License
     16   along with libmicrohttpd; see the file COPYING.  If not, write to the
     17   Free Software Foundation, Inc., 59 Temple Place - Suite 330,
     18   Boston, MA 02111-1307, USA.
     19 */
     20 
     21 /**
     22  * @file test_https_get_parallel.c
     23  * @brief  Testcase for libmicrohttpd HTTPS GET operations
     24  * @author Sagie Amir
     25  * @author Christian Grothoff
     26  */
     27 
     28 #include "platform.h"
     29 #include "microhttpd.h"
     30 #include <sys/stat.h>
     31 #include <limits.h>
     32 #include <curl/curl.h>
     33 #include <pthread.h>
     34 #include <gcrypt.h>
     35 #include "tls_test_common.h"
     36 
     37 #if defined(CPU_COUNT) && (CPU_COUNT+0) < 4
     38 #undef CPU_COUNT
     39 #endif
     40 #if !defined(CPU_COUNT)
     41 #define CPU_COUNT 4
     42 #endif
     43 
     44 extern const char srv_key_pem[];
     45 extern const char srv_self_signed_cert_pem[];
     46 
     47 int curl_check_version (const char *req_version, ...);
     48 
     49 
     50 /**
     51  * used when spawning multiple threads executing curl server requests
     52  *
     53  */
     54 static void *
     55 https_transfer_thread_adapter (void *args)
     56 {
     57   static int nonnull;
     58   struct https_test_data *cargs = args;
     59   int ret;
     60 
     61   /* time spread incomming requests */
     62   usleep ((useconds_t) 10.0 * ((double) rand ()) / ((double) RAND_MAX));
     63   ret = test_https_transfer (NULL,
     64                              cargs->cipher_suite, cargs->proto_version);
     65   if (ret == 0)
     66     return NULL;
     67   return &nonnull;
     68 }
     69 
     70 
     71 /**
     72  * Test non-parallel requests.
     73  *
     74  * @return: 0 upon all client requests returning '0', -1 otherwise.
     75  *
     76  * TODO : make client_count a parameter - number of curl client threads to spawn
     77  */
     78 static int
     79 test_single_client (void *cls, const char *cipher_suite,
     80                     int curl_proto_version)
     81 {
     82   void *client_thread_ret;
     83   struct https_test_data client_args =
     84     { NULL, cipher_suite, curl_proto_version };
     85 
     86   client_thread_ret = https_transfer_thread_adapter (&client_args);
     87   if (client_thread_ret != NULL)
     88     return -1;
     89   return 0;
     90 }
     91 
     92 
     93 /**
     94  * Test parallel request handling.
     95  *
     96  * @return: 0 upon all client requests returning '0', -1 otherwise.
     97  *
     98  * TODO : make client_count a parameter - numver of curl client threads to spawn
     99  */
    100 static int
    101 test_parallel_clients (void * cls, const char *cipher_suite,
    102                        int curl_proto_version)
    103 {
    104   int i;
    105   int client_count = (CPU_COUNT - 1);
    106   void *client_thread_ret;
    107   pthread_t client_arr[client_count];
    108   struct https_test_data client_args =
    109     { NULL, cipher_suite, curl_proto_version };
    110 
    111   for (i = 0; i < client_count; ++i)
    112     {
    113       if (pthread_create (&client_arr[i], NULL,
    114                           &https_transfer_thread_adapter, &client_args) != 0)
    115         {
    116           fprintf (stderr, "Error: failed to spawn test client threads.\n");
    117           return -1;
    118         }
    119     }
    120 
    121   /* check all client requests fulfilled correctly */
    122   for (i = 0; i < client_count; ++i)
    123     {
    124       if ((pthread_join (client_arr[i], &client_thread_ret) != 0) ||
    125           (client_thread_ret != NULL))
    126         return -1;
    127     }
    128 
    129   return 0;
    130 }
    131 
    132 
    133 int
    134 main (int argc, char *const *argv)
    135 {
    136   unsigned int errorCount = 0;
    137   const char *aes256_sha = "AES256-SHA";
    138 
    139   /* initialize random seed used by curl clients */
    140   unsigned int iseed = (unsigned int) time (NULL);
    141   srand (iseed);
    142   if (0 != curl_global_init (CURL_GLOBAL_ALL))
    143     {
    144       fprintf (stderr, "Error: %s\n", strerror (errno));
    145       return -1;
    146     }
    147 
    148   if (curl_uses_nss_ssl() == 0)
    149     aes256_sha = "rsa_aes_256_sha";
    150 #if EPOLL_SUPPORT
    151   errorCount +=
    152     test_wrap ("single threaded daemon, single client, epoll", &test_single_client,
    153                NULL,
    154                MHD_USE_SELECT_INTERNALLY | MHD_USE_SSL | MHD_USE_DEBUG | MHD_USE_EPOLL_LINUX_ONLY,
    155                aes256_sha, CURL_SSLVERSION_TLSv1, MHD_OPTION_HTTPS_MEM_KEY,
    156                srv_key_pem, MHD_OPTION_HTTPS_MEM_CERT,
    157                srv_self_signed_cert_pem, MHD_OPTION_END);
    158 #endif
    159   errorCount +=
    160     test_wrap ("single threaded daemon, single client", &test_single_client,
    161                NULL,
    162                MHD_USE_SELECT_INTERNALLY | MHD_USE_SSL | MHD_USE_DEBUG,
    163                aes256_sha, CURL_SSLVERSION_TLSv1, MHD_OPTION_HTTPS_MEM_KEY,
    164                srv_key_pem, MHD_OPTION_HTTPS_MEM_CERT,
    165                srv_self_signed_cert_pem, MHD_OPTION_END);
    166 #if EPOLL_SUPPORT
    167   errorCount +=
    168     test_wrap ("single threaded daemon, parallel clients, epoll",
    169                &test_parallel_clients, NULL,
    170                MHD_USE_SELECT_INTERNALLY | MHD_USE_SSL | MHD_USE_DEBUG | MHD_USE_EPOLL_LINUX_ONLY,
    171                aes256_sha, CURL_SSLVERSION_TLSv1, MHD_OPTION_HTTPS_MEM_KEY,
    172                srv_key_pem, MHD_OPTION_HTTPS_MEM_CERT,
    173                srv_self_signed_cert_pem, MHD_OPTION_END);
    174 #endif
    175   errorCount +=
    176     test_wrap ("single threaded daemon, parallel clients",
    177                &test_parallel_clients, NULL,
    178                MHD_USE_SELECT_INTERNALLY | MHD_USE_SSL | MHD_USE_DEBUG,
    179                aes256_sha, CURL_SSLVERSION_TLSv1, MHD_OPTION_HTTPS_MEM_KEY,
    180                srv_key_pem, MHD_OPTION_HTTPS_MEM_CERT,
    181                srv_self_signed_cert_pem, MHD_OPTION_END);
    182 
    183   curl_global_cleanup ();
    184   return errorCount != 0;
    185 }
    186