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
41extern const char srv_key_pem[];
42extern const char srv_self_signed_cert_pem[];
43
44static const int TIME_OUT = 3;
45
46static int
47test_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
100int
101main (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