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 mhds_get_test.c
23 * @brief: daemon TLS alert response test-case
24 *
25 * @author Sagie Amir
26 */
27
28 #include "platform.h"
29 #include "microhttpd.h"
30 #include "internal.h"
31 #include "tls_test_common.h"
32 #include <gcrypt.h>
33
34 #ifdef _WIN32
35 #ifndef WIN32_LEAN_AND_MEAN
36 #define WIN32_LEAN_AND_MEAN 1
37 #endif /* !WIN32_LEAN_AND_MEAN */
38 #include <windows.h>
39 #endif
40
41 extern const char srv_key_pem[];
42 extern const char srv_self_signed_cert_pem[];
43
44 static const int TIME_OUT = 3;
45
46 static int
test_tls_session_time_out(gnutls_session_t session)47 test_tls_session_time_out (gnutls_session_t session)
48 {
49 int ret;
50 MHD_socket sd;
51 struct sockaddr_in sa;
52
53 sd = socket (AF_INET, SOCK_STREAM, 0);
54 if (sd == -1)
55 {
56 fprintf (stderr, "Failed to create socket: %s\n", strerror (errno));
57 return -1;
58 }
59
60 memset (&sa, '\0', sizeof (struct sockaddr_in));
61 sa.sin_family = AF_INET;
62 sa.sin_port = htons (DEAMON_TEST_PORT);
63 sa.sin_addr.s_addr = htonl (INADDR_LOOPBACK);
64
65 gnutls_transport_set_ptr (session, (gnutls_transport_ptr_t) (intptr_t) sd);
66
67 ret = connect (sd, &sa, sizeof (struct sockaddr_in));
68
69 if (ret < 0)
70 {
71 fprintf (stderr, "Error: %s\n", MHD_E_FAILED_TO_CONNECT);
72 close (sd);
73 return -1;
74 }
75
76 ret = gnutls_handshake (session);
77 if (ret < 0)
78 {
79 fprintf (stderr, "Handshake failed\n");
80 close (sd);
81 return -1;
82 }
83
84 sleep (TIME_OUT + 1);
85
86 /* check that server has closed the connection */
87 /* TODO better RST trigger */
88 if (send (sd, "", 1, 0) == 0)
89 {
90 fprintf (stderr, "Connection failed to time-out\n");
91 close (sd);
92 return -1;
93 }
94
95 close (sd);
96 return 0;
97 }
98
99
100 int
main(int argc,char * const * argv)101 main (int argc, char *const *argv)
102 {
103 int errorCount = 0;;
104 struct MHD_Daemon *d;
105 gnutls_session_t session;
106 gnutls_datum_t key;
107 gnutls_datum_t cert;
108 gnutls_certificate_credentials_t xcred;
109
110
111 gcry_control (GCRYCTL_ENABLE_QUICK_RANDOM, 0);
112 #ifdef GCRYCTL_INITIALIZATION_FINISHED
113 gcry_control (GCRYCTL_INITIALIZATION_FINISHED, 0);
114 #endif
115 gnutls_global_init ();
116 gnutls_global_set_log_level (11);
117
118 d = MHD_start_daemon (MHD_USE_THREAD_PER_CONNECTION | MHD_USE_SSL |
119 MHD_USE_DEBUG, DEAMON_TEST_PORT,
120 NULL, NULL, &http_dummy_ahc, NULL,
121 MHD_OPTION_CONNECTION_TIMEOUT, TIME_OUT,
122 MHD_OPTION_HTTPS_MEM_KEY, srv_key_pem,
123 MHD_OPTION_HTTPS_MEM_CERT, srv_self_signed_cert_pem,
124 MHD_OPTION_END);
125
126 if (d == NULL)
127 {
128 fprintf (stderr, MHD_E_SERVER_INIT);
129 return -1;
130 }
131
132 if (0 != setup_session (&session, &key, &cert, &xcred))
133 {
134 fprintf (stderr, "failed to setup session\n");
135 return 1;
136 }
137 errorCount += test_tls_session_time_out (session);
138 teardown_session (session, &key, &cert, xcred);
139
140 print_test_result (errorCount, argv[0]);
141
142 MHD_stop_daemon (d);
143 gnutls_global_deinit ();
144
145 return errorCount != 0;
146 }
147