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_select.c
23 * @brief Testcase for libmicrohttpd HTTPS GET operations
24 * @author Sagie Amir
25 */
26
27 #include "platform.h"
28 #include "microhttpd.h"
29 #include <limits.h>
30 #include <sys/stat.h>
31 #include <curl/curl.h>
32 #include <gcrypt.h>
33 #include "tls_test_common.h"
34
35 extern const char srv_key_pem[];
36 extern const char srv_self_signed_cert_pem[];
37 extern const char srv_signed_cert_pem[];
38 extern const char srv_signed_key_pem[];
39
40 static int oneone;
41
42 static int
ahc_echo(void * cls,struct MHD_Connection * connection,const char * url,const char * method,const char * version,const char * upload_data,size_t * upload_data_size,void ** unused)43 ahc_echo (void *cls,
44 struct MHD_Connection *connection,
45 const char *url,
46 const char *method,
47 const char *version,
48 const char *upload_data, size_t *upload_data_size,
49 void **unused)
50 {
51 static int ptr;
52 const char *me = cls;
53 struct MHD_Response *response;
54 int ret;
55
56 if (0 != strcmp (me, method))
57 return MHD_NO; /* unexpected method */
58 if (&ptr != *unused)
59 {
60 *unused = &ptr;
61 return MHD_YES;
62 }
63 *unused = NULL;
64 response = MHD_create_response_from_buffer (strlen (url),
65 (void *) url,
66 MHD_RESPMEM_MUST_COPY);
67 ret = MHD_queue_response (connection, MHD_HTTP_OK, response);
68 MHD_destroy_response (response);
69 if (ret == MHD_NO)
70 abort ();
71 return ret;
72 }
73
74
75 static int
testExternalGet(int flags)76 testExternalGet (int flags)
77 {
78 struct MHD_Daemon *d;
79 CURL *c;
80 char buf[2048];
81 struct CBC cbc;
82 CURLM *multi;
83 CURLMcode mret;
84 fd_set rs;
85 fd_set ws;
86 fd_set es;
87 MHD_socket max;
88 int running;
89 struct CURLMsg *msg;
90 time_t start;
91 struct timeval tv;
92 const char *aes256_sha = "AES256-SHA";
93
94 multi = NULL;
95 cbc.buf = buf;
96 cbc.size = 2048;
97 cbc.pos = 0;
98 d = MHD_start_daemon (MHD_USE_DEBUG | MHD_USE_SSL | flags,
99 1082, NULL, NULL, &ahc_echo, "GET",
100 MHD_OPTION_HTTPS_MEM_KEY, srv_key_pem,
101 MHD_OPTION_HTTPS_MEM_CERT, srv_self_signed_cert_pem,
102 MHD_OPTION_END);
103 if (d == NULL)
104 return 256;
105
106 if (curl_uses_nss_ssl() == 0)
107 aes256_sha = "rsa_aes_256_sha";
108
109 c = curl_easy_init ();
110 curl_easy_setopt (c, CURLOPT_URL, "https://127.0.0.1:1082/hello_world");
111 curl_easy_setopt (c, CURLOPT_WRITEFUNCTION, ©Buffer);
112 curl_easy_setopt (c, CURLOPT_WRITEDATA, &cbc);
113 /* TLS options */
114 curl_easy_setopt (c, CURLOPT_SSLVERSION, CURL_SSLVERSION_TLSv1);
115 curl_easy_setopt (c, CURLOPT_SSL_CIPHER_LIST, aes256_sha);
116 curl_easy_setopt (c, CURLOPT_SSL_VERIFYPEER, 0);
117 curl_easy_setopt (c, CURLOPT_SSL_VERIFYHOST, 0);
118 curl_easy_setopt (c, CURLOPT_FAILONERROR, 1);
119 if (oneone)
120 curl_easy_setopt (c, CURLOPT_HTTP_VERSION, CURL_HTTP_VERSION_1_1);
121 else
122 curl_easy_setopt (c, CURLOPT_HTTP_VERSION, CURL_HTTP_VERSION_1_0);
123 curl_easy_setopt (c, CURLOPT_TIMEOUT, 150L);
124 curl_easy_setopt (c, CURLOPT_CONNECTTIMEOUT, 150L);
125 /* NOTE: use of CONNECTTIMEOUT without also
126 setting NOSIGNAL results in really weird
127 crashes on my system! */
128 curl_easy_setopt (c, CURLOPT_NOSIGNAL, 1);
129
130
131 multi = curl_multi_init ();
132 if (multi == NULL)
133 {
134 curl_easy_cleanup (c);
135 MHD_stop_daemon (d);
136 return 512;
137 }
138 mret = curl_multi_add_handle (multi, c);
139 if (mret != CURLM_OK)
140 {
141 curl_multi_cleanup (multi);
142 curl_easy_cleanup (c);
143 MHD_stop_daemon (d);
144 return 1024;
145 }
146 start = time (NULL);
147 while ((time (NULL) - start < 5) && (multi != NULL))
148 {
149 max = 0;
150 FD_ZERO (&rs);
151 FD_ZERO (&ws);
152 FD_ZERO (&es);
153 mret = curl_multi_fdset (multi, &rs, &ws, &es, &max);
154 if (mret != CURLM_OK)
155 {
156 curl_multi_remove_handle (multi, c);
157 curl_multi_cleanup (multi);
158 curl_easy_cleanup (c);
159 MHD_stop_daemon (d);
160 return 2048;
161 }
162 if (MHD_YES != MHD_get_fdset (d, &rs, &ws, &es, &max))
163 {
164 curl_multi_remove_handle (multi, c);
165 curl_multi_cleanup (multi);
166 curl_easy_cleanup (c);
167 MHD_stop_daemon (d);
168 return 4096;
169 }
170 tv.tv_sec = 0;
171 tv.tv_usec = 1000;
172 select (max + 1, &rs, &ws, &es, &tv);
173 curl_multi_perform (multi, &running);
174 if (running == 0)
175 {
176 msg = curl_multi_info_read (multi, &running);
177 if (msg == NULL)
178 break;
179 if (msg->msg == CURLMSG_DONE)
180 {
181 if (msg->data.result != CURLE_OK)
182 printf ("%s failed at %s:%d: `%s'\n",
183 "curl_multi_perform",
184 __FILE__,
185 __LINE__, curl_easy_strerror (msg->data.result));
186 curl_multi_remove_handle (multi, c);
187 curl_multi_cleanup (multi);
188 curl_easy_cleanup (c);
189 c = NULL;
190 multi = NULL;
191 }
192 }
193 MHD_run (d);
194 }
195 if (multi != NULL)
196 {
197 curl_multi_remove_handle (multi, c);
198 curl_easy_cleanup (c);
199 curl_multi_cleanup (multi);
200 }
201 MHD_stop_daemon (d);
202 if (cbc.pos != strlen ("/hello_world"))
203 return 8192;
204 if (0 != strncmp ("/hello_world", cbc.buf, strlen ("/hello_world")))
205 return 16384;
206 return 0;
207 }
208
209
210 int
main(int argc,char * const * argv)211 main (int argc, char *const *argv)
212 {
213 unsigned int errorCount = 0;
214
215 if (0 != curl_global_init (CURL_GLOBAL_ALL))
216 {
217 fprintf (stderr, "Error: %s\n", strerror (errno));
218 return -1;
219 }
220 #if EPOLL_SUPPORT
221 if (0 != (errorCount = testExternalGet (MHD_USE_EPOLL_LINUX_ONLY)))
222 fprintf (stderr, "Fail: %d\n", errorCount);
223 #endif
224 if (0 != (errorCount = testExternalGet (0)))
225 fprintf (stderr, "Fail: %d\n", errorCount);
226 curl_global_cleanup ();
227 return errorCount != 0;
228 }
229