1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
|
/*
This file is part of libmicrohttpd
Copyright (C) 2007-2018 Daniel Pittman and Christian Grothoff
This library is free software; you can redistribute it and/or
modify it under the terms of the GNU Lesser General Public
License as published by the Free Software Foundation; either
version 2.1 of the License, or (at your option) any later version.
This library is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
Lesser General Public License for more details.
You should have received a copy of the GNU Lesser General Public
License along with this library; if not, write to the Free Software
Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
*/
/**
* @file lib/daemon_quiesce.c
* @brief main functions to quiesce a daemon
* @author Christian Grothoff
*/
#include "internal.h"
/**
* Stop accepting connections from the listening socket. Allows
* clients to continue processing, but stops accepting new
* connections. Note that the caller is responsible for closing the
* returned socket; however, if MHD is run using threads (anything but
* external select mode), it must not be closed until AFTER
* #MHD_stop_daemon has been called (as it is theoretically possible
* that an existing thread is still using it).
*
* Note that some thread modes require the caller to have passed
* #MHD_USE_ITC when using this API. If this daemon is
* in one of those modes and this option was not given to
* #MHD_start_daemon, this function will return #MHD_INVALID_SOCKET.
*
* @param daemon daemon to stop accepting new connections for
* @return old listen socket on success, #MHD_INVALID_SOCKET if
* the daemon was already not listening anymore, or
* was never started
* @ingroup specialized
*/
MHD_socket
MHD_daemon_quiesce (struct MHD_Daemon *daemon)
{
MHD_socket listen_socket;
if (MHD_INVALID_SOCKET == (listen_socket = daemon->listen_socket))
return MHD_INVALID_SOCKET;
if ( (daemon->disable_itc) &&
(MHD_TM_EXTERNAL_EVENT_LOOP != daemon->threading_mode) )
{
#ifdef HAVE_MESSAGES
MHD_DLOG (daemon,
MHD_SC_SYSCALL_QUIESCE_REQUIRES_ITC,
"Using MHD_quiesce_daemon in this mode requires ITC\n");
#endif
return MHD_INVALID_SOCKET;
}
if (NULL != daemon->worker_pool)
{
unsigned int i;
for (i = 0; i < daemon->worker_pool_size; i++)
{
struct MHD_Daemon *worker = &daemon->worker_pool[i];
worker->was_quiesced = true;
#ifdef EPOLL_SUPPORT
if ( (MHD_ELS_EPOLL == daemon->event_loop_syscall) &&
(-1 != worker->epoll_fd) &&
(worker->listen_socket_in_epoll) )
{
if (0 != epoll_ctl (worker->epoll_fd,
EPOLL_CTL_DEL,
listen_socket,
NULL))
MHD_PANIC (_ ("Failed to remove listen FD from epoll set\n"));
worker->listen_socket_in_epoll = false;
}
else
#endif
if (MHD_ITC_IS_VALID_ (worker->itc))
{
if (! MHD_itc_activate_ (worker->itc,
"q"))
MHD_PANIC (_ (
"Failed to signal quiesce via inter-thread communication channel"));
}
}
daemon->was_quiesced = true;
#ifdef EPOLL_SUPPORT
if ( (MHD_ELS_EPOLL == daemon->event_loop_syscall) &&
(-1 != daemon->epoll_fd) &&
(daemon->listen_socket_in_epoll) )
{
if (0 != epoll_ctl (daemon->epoll_fd,
EPOLL_CTL_DEL,
listen_socket,
NULL))
MHD_PANIC ("Failed to remove listen FD from epoll set\n");
daemon->listen_socket_in_epoll = false;
}
#endif
}
if ( (MHD_ITC_IS_VALID_ (daemon->itc)) &&
(! MHD_itc_activate_ (daemon->itc,
"q")) )
MHD_PANIC (_ (
"Failed to signal quiesce via inter-thread communication channel"));
/* FIXME: we might want some bi-directional communication here
(in both the thread-pool and single-thread case!)
to be sure that the threads have stopped using the listen
socket, otherwise there is still the possibility of a race
between a thread accept()ing and the caller closing and
re-binding the socket. */return listen_socket;
}
/* end of daemon_quiesce.c */
|