2 * Copyright (C) 2013-2020 Red Hat Inc.
4 * Redistribution and use in source and binary forms, with or without
5 * modification, are permitted provided that the following conditions are
8 * * Redistributions of source code must retain the above copyright
9 * notice, this list of conditions and the following disclaimer.
11 * * Redistributions in binary form must reproduce the above copyright
12 * notice, this list of conditions and the following disclaimer in the
13 * documentation and/or other materials provided with the distribution.
15 * * Neither the name of Red Hat nor the names of its contributors may be
16 * used to endorse or promote products derived from this software without
17 * specific prior written permission.
19 * THIS SOFTWARE IS PROVIDED BY RED HAT AND CONTRIBUTORS ''AS IS'' AND
20 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO,
21 * THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A
22 * PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL RED HAT OR
23 * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
24 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
25 * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF
26 * USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
27 * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
28 * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT
29 * OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
46 #include "web-server.h"
53 "echo iteration=$iteration"
56 check_request (const char *request
)
60 /* Check the Cookie header. */
61 snprintf (expected
, sizeof expected
,
62 "\r\nCookie: iteration=%u\r\n", iteration
);
63 if (strcasestr (request
, expected
) == NULL
) {
64 fprintf (stderr
, "%s: no/incorrect iteration cookie in request\n",
73 main (int argc
, char *argv
[])
76 struct nbd_handle
*nbd
;
77 CLEANUP_FREE
char *usp_param
= NULL
;
79 #ifndef HAVE_CURLOPT_UNIX_SOCKET_PATH
80 fprintf (stderr
, "%s: curl does not support CURLOPT_UNIX_SOCKET_PATH\n",
85 sockpath
= web_server ("disk", check_request
);
86 if (sockpath
== NULL
) {
87 fprintf (stderr
, "%s: could not start web server thread\n", program_name
);
93 fprintf (stderr
, "%s\n", nbd_get_error ());
97 /* We expect that connecting will cause a HEAD request (to find the
98 * size). $iteration will be 0.
103 if (asprintf (&usp_param
, "unix-socket-path=%s", sockpath
) == -1) {
107 if (nbd_connect_command (nbd
,
109 "nbdkit", "-s", "--exit-with-parent", "-v",
111 "-D", "curl.verbose=1",
112 "http://localhost/disk",
113 "cookie-script=" SCRIPT
,
114 "cookie-script-renew=1",
115 usp_param
, /* unix-socket-path=... */
117 fprintf (stderr
, "%s\n", nbd_get_error ());
121 /* Sleep the script will be called again. $iteration will be 1. */
125 /* Make a request. */
126 if (nbd_pread (nbd
, buf
, sizeof buf
, 0, 0) == -1) {
127 fprintf (stderr
, "%s\n", nbd_get_error ());
131 /* Sleep again and make another request. $iteration will be 2. */
135 if (nbd_pread (nbd
, buf
, sizeof buf
, 0, 0) == -1) {
136 fprintf (stderr
, "%s\n", nbd_get_error ());