Laszlo Ersek
2023-May-30 14:06 UTC
[Libguestfs] [libnbd PATCH v3 04/22] states: Prepare to send 64-bit requests
On 5/25/23 15:00, Eric Blake wrote:> Support sending 64-bit requests if extended headers were negotiated. > This includes setting NBD_CMD_FLAG_PAYLOAD_LEN any time we send an > extended NBD_CMD_WRITE; this is such a fundamental part of the > protocol that for now it is easier to silently ignore whatever value > the user passes in for that bit in the flags parameter of nbd_pwrite > regardless of the current settings in set_strict_mode, rather than > trying to force the user to pass in the correct value to match whether > extended mode is negotiated. However, when we later add APIs to give > the user more control for interoperability testing, it may be worth > adding a new set_strict_mode control knob to explicitly enable the > client to intentionally violate the protocol (the testsuite added in > this patch would then be updated to match). > > At this point, h->extended_headers is permanently false (we can't > enable it until all other aspects of the protocol have likewise been > converted). > > Support for using FLAG_PAYLOAD_LEN with NBD_CMD_BLOCK_STATUS is less > fundamental, and deserves to be in its own patch. > > Signed-off-by: Eric Blake <eblake at redhat.com> > --- > lib/internal.h | 10 ++- > generator/API.ml | 20 +++-- > generator/states-issue-command.c | 29 ++++--- > generator/states-reply-structured.c | 2 +- > lib/rw.c | 17 +++-- > tests/Makefile.am | 4 + > tests/pwrite-extended.c | 112 ++++++++++++++++++++++++++++ > .gitignore | 1 + > 8 files changed, 169 insertions(+), 26 deletions(-) > create mode 100644 tests/pwrite-extended.c > > diff --git a/lib/internal.h b/lib/internal.h > index c71980ef..8a5f93d4 100644 > --- a/lib/internal.h > +++ b/lib/internal.h > @@ -110,6 +110,9 @@ struct nbd_handle { > char *tls_username; /* Username, NULL = use current username */ > char *tls_psk_file; /* PSK filename, NULL = no PSK */ > > + /* Extended headers. */ > + bool extended_headers; /* If we negotiated NBD_OPT_EXTENDED_HEADERS */ > + > /* Desired metadata contexts. */ > bool request_sr; > bool request_meta; > @@ -263,7 +266,10 @@ struct nbd_handle { > /* Issuing a command must use a buffer separate from sbuf, for the > * case when we interrupt a request to service a reply. > */ > - struct nbd_request request; > + union { > + struct nbd_request compact; > + struct nbd_request_ext extended; > + } req; > bool in_write_payload; > bool in_write_shutdown; > > @@ -364,7 +370,7 @@ struct command { > uint16_t type; > uint64_t cookie; > uint64_t offset; > - uint32_t count; > + uint64_t count; > void *data; /* Buffer for read/write */ > struct command_cb cb; > bool initialized; /* For read, true if getting a hole may skip memset */(1) Are there places in the code where we currently assign this "count" field back to a uint32_t object, and assume truncation impossible?> diff --git a/generator/API.ml b/generator/API.ml > index 5fcb0e1f..02c1260d 100644 > --- a/generator/API.ml > +++ b/generator/API.ml > @@ -198,11 +198,12 @@ let cmd_flags > flag_prefix = "CMD_FLAG"; > guard = Some "((h->strict & LIBNBD_STRICT_FLAGS) || flags > UINT16_MAX)"; > flags = [ > - "FUA", 1 lsl 0; > - "NO_HOLE", 1 lsl 1; > - "DF", 1 lsl 2; > - "REQ_ONE", 1 lsl 3; > - "FAST_ZERO", 1 lsl 4; > + "FUA", 1 lsl 0; > + "NO_HOLE", 1 lsl 1; > + "DF", 1 lsl 2; > + "REQ_ONE", 1 lsl 3; > + "FAST_ZERO", 1 lsl 4; > + "PAYLOAD_LEN", 1 lsl 5; > ] > } > let handshake_flags = { > @@ -2507,7 +2508,7 @@ "pread_structured", { > "pwrite", { > default_call with > args = [ BytesIn ("buf", "count"); UInt64 "offset" ]; > - optargs = [ OFlags ("flags", cmd_flags, Some ["FUA"]) ]; > + optargs = [ OFlags ("flags", cmd_flags, Some ["FUA"; "PAYLOAD_LEN"]) ]; > ret = RErr; > permitted_states = [ Connected ]; > shortdesc = "write to the NBD server"; > @@ -2530,7 +2531,10 @@ "pwrite", { > C<LIBNBD_CMD_FLAG_FUA> meaning that the server should not > return until the data has been committed to permanent storage > (if that is supported - some servers cannot do this, see > -L<nbd_can_fua(3)>)." > +L<nbd_can_fua(3)>). For convenience, libnbd ignores the presence > +or absence of the flag C<LIBNBD_CMD_FLAG_PAYLOAD_LEN> in C<flags>, > +while correctly using the flag over the wire according to whether > +extended headers were negotiated." > ^ strict_call_description; > see_also = [Link "can_fua"; Link "is_read_only"; > Link "aio_pwrite"; Link "get_block_size"; > @@ -3220,7 +3224,7 @@ "aio_pwrite", { > default_call with > args = [ BytesPersistIn ("buf", "count"); UInt64 "offset" ]; > optargs = [ OClosure completion_closure; > - OFlags ("flags", cmd_flags, Some ["FUA"]) ]; > + OFlags ("flags", cmd_flags, Some ["FUA"; "PAYLOAD_LEN"]) ]; > ret = RCookie; > permitted_states = [ Connected ]; > shortdesc = "write to the NBD server"; > diff --git a/generator/states-issue-command.c b/generator/states-issue-command.c > index 111e131c..79136b61 100644 > --- a/generator/states-issue-command.c > +++ b/generator/states-issue-command.c > @@ -41,15 +41,24 @@ ISSUE_COMMAND.START: > return 0; > } > > - h->request.magic = htobe32 (NBD_REQUEST_MAGIC); > - h->request.flags = htobe16 (cmd->flags); > - h->request.type = htobe16 (cmd->type); > - h->request.handle = htobe64 (cmd->cookie); > - h->request.offset = htobe64 (cmd->offset); > - h->request.count = htobe32 (cmd->count); > + /* These fields are coincident between req.compact and req.extended */ > + h->req.compact.flags = htobe16 (cmd->flags); > + h->req.compact.type = htobe16 (cmd->type); > + h->req.compact.handle = htobe64 (cmd->cookie); > + h->req.compact.offset = htobe64 (cmd->offset);What's more, this is a "by the book" common initial sequence! :)> + if (h->extended_headers) { > + h->req.extended.magic = htobe32 (NBD_EXTENDED_REQUEST_MAGIC); > + h->req.extended.count = htobe64 (cmd->count); > + h->wlen = sizeof (h->req.extended); > + } > + else { > + assert (cmd->count <= UINT32_MAX); > + h->req.compact.magic = htobe32 (NBD_REQUEST_MAGIC); > + h->req.compact.count = htobe32 (cmd->count); > + h->wlen = sizeof (h->req.compact); > + } > h->chunks_sent++; > - h->wbuf = &h->request; > - h->wlen = sizeof (h->request); > + h->wbuf = &h->req; > if (cmd->type == NBD_CMD_WRITE || cmd->next) > h->wflags = MSG_MORE; > SET_NEXT_STATE (%SEND_REQUEST); > @@ -74,7 +83,7 @@ ISSUE_COMMAND.PREPARE_WRITE_PAYLOAD: > > assert (h->cmds_to_issue != NULL); > cmd = h->cmds_to_issue; > - assert (cmd->cookie == be64toh (h->request.handle)); > + assert (cmd->cookie == be64toh (h->req.compact.handle)); > if (cmd->type == NBD_CMD_WRITE) { > h->wbuf = cmd->data; > h->wlen = cmd->count; > @@ -120,7 +129,7 @@ ISSUE_COMMAND.FINISH: > assert (!h->wlen); > assert (h->cmds_to_issue != NULL); > cmd = h->cmds_to_issue; > - assert (cmd->cookie == be64toh (h->request.handle)); > + assert (cmd->cookie == be64toh (h->req.compact.handle)); > h->cmds_to_issue = cmd->next; > if (h->cmds_to_issue_tail == cmd) > h->cmds_to_issue_tail = NULL; > diff --git a/generator/states-reply-structured.c b/generator/states-reply-structured.c > index 6f96945a..df509216 100644 > --- a/generator/states-reply-structured.c > +++ b/generator/states-reply-structured.c > @@ -34,7 +34,7 @@ structured_reply_in_bounds (uint64_t offset, uint32_t length, > offset + length > cmd->offset + cmd->count) { > set_error (0, "range of structured reply is out of bounds, " > "offset=%" PRIu64 ", cmd->offset=%" PRIu64 ", " > - "length=%" PRIu32 ", cmd->count=%" PRIu32 ": " > + "length=%" PRIu32 ", cmd->count=%" PRIu64 ": " > "this is likely to be a bug in the NBD server", > offset, cmd->offset, length, cmd->count); > return false; > diff --git a/lib/rw.c b/lib/rw.c > index 3dc3499e..8b2bd4cc 100644 > --- a/lib/rw.c > +++ b/lib/rw.c > @@ -223,13 +223,11 @@ nbd_internal_command_common (struct nbd_handle *h, > } > break; > > - /* Other commands are currently limited by the 32 bit field in the > - * command structure on the wire, but in future we hope to support > - * 64 bit values here with a change to the NBD protocol which is > - * being discussed upstream. > + /* Other commands are limited by the 32 bit field in the command > + * structure on the wire, unless extended headers were negotiated. > */ > default: > - if (count > UINT32_MAX) { > + if (!h->extended_headers && count > UINT32_MAX) { > set_error (ERANGE, "request too large: maximum request size is %" PRIu32, > UINT32_MAX); > goto err; > @@ -358,6 +356,15 @@ nbd_unlocked_aio_pwrite (struct nbd_handle *h, const void *buf, > return -1; > } > } > + /* It is more convenient to manage PAYLOAD_LEN by what was negotiated > + * than to require the user to have to set it correctly. > + * TODO: Add new h->strict bit to allow intentional protocol violation > + * for interoperability testing. > + */ > + if (h->extended_headers) > + flags |= LIBNBD_CMD_FLAG_PAYLOAD_LEN; > + else > + flags &= ~LIBNBD_CMD_FLAG_PAYLOAD_LEN;Nice -- I wanted to ask for: flags &= ~(uint32_t)LIBNBD_CMD_FLAG_PAYLOAD_LEN; due to LIBNBD_CMD_FLAG_PAYLOAD_LEN having type "int". However: in patch#3, what has type "int" is: +#define NBD_CMD_FLAG_PAYLOAD_LEN (1<<5) and here we have LIBNBD_CMD_FLAG_PAYLOAD_LEN instead -- and the latter has type unsigned int already, from your recent commit 69eecae2c03a ("api: Generate flag values as unsigned", 2022-11-11). And I think we're fine assuming that uint32_t is unsigned int.> > SET_CALLBACK_TO_NULL (*completion); > return nbd_internal_command_common (h, flags, NBD_CMD_WRITE, offset, count, > diff --git a/tests/Makefile.am b/tests/Makefile.am > index 3a93251e..8b839bf5 100644 > --- a/tests/Makefile.am > +++ b/tests/Makefile.am > @@ -232,6 +232,7 @@ check_PROGRAMS += \ > closure-lifetimes \ > pread-initialize \ > socket-activation-name \ > + pwrite-extended \ > $(NULL) > > TESTS += \(2) Incorrect indentation: two spaces rather than one tab.> @@ -650,6 +651,9 @@ socket_activation_name_SOURCES = \ > requires.h > socket_activation_name_LDADD = $(top_builddir)/lib/libnbd.la > > +pwrite_extended_SOURCES = pwrite-extended.c > +pwrite_extended_LDADD = $(top_builddir)/lib/libnbd.la > + > #---------------------------------------------------------------------- > # Testing TLS support. > > diff --git a/tests/pwrite-extended.c b/tests/pwrite-extended.c > new file mode 100644 > index 00000000..f0b5a3f3 > --- /dev/null > +++ b/tests/pwrite-extended.c > @@ -0,0 +1,112 @@ > +/* NBD client library in userspace > + * Copyright Red Hat > + * > + * 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 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 > + */ > + > +/* Check behavior of pwrite with PAYLOAD_LEN flag for extended headers. */ > + > +#include <config.h> > + > +#include <stdio.h> > +#include <stdlib.h> > +#include <string.h> > +#include <errno.h> > +#include <unistd.h> > +#include <sys/stat.h> > + > +#include <libnbd.h> > + > +static char *progname; > +static char buf[512]; > + > +static void > +check (int experr, const char *prefix) > +{ > + const char *msg = nbd_get_error (); > + int errnum = nbd_get_errno (); > + > + fprintf (stderr, "error: \"%s\"\n", msg); > + fprintf (stderr, "errno: %d (%s)\n", errnum, strerror (errnum)); > + if (strncmp (msg, prefix, strlen (prefix)) != 0) { > + fprintf (stderr, "%s: test failed: missing context prefix: %s\n", > + progname, msg); > + exit (EXIT_FAILURE); > + } > + if (errnum != experr) { > + fprintf (stderr, "%s: test failed: " > + "expected errno = %d (%s), but got %d\n", > + progname, experr, strerror (experr), errnum); > + exit (EXIT_FAILURE); > + } > +} > + > +int > +main (int argc, char *argv[]) > +{ > + struct nbd_handle *nbd; > + const char *cmd[] = { > + "nbdkit", "-s", "-v", "--exit-with-parent", "memory", "1048576", NULL > + }; > + uint32_t strict; > + > + progname = argv[0]; > + > + nbd = nbd_create (); > + if (nbd == NULL) { > + fprintf (stderr, "%s\n", nbd_get_error ());(3) Minor inconsistency with check(): we're not printing "progname" here.> + exit (EXIT_FAILURE); > + } > + > + /* Connect to the server. */ > + if (nbd_connect_command (nbd, (char **)cmd) == -1) { > + fprintf (stderr, "%s: %s\n", argv[0], nbd_get_error ()); > + exit (EXIT_FAILURE); > + }(4) Another kind of inconsistency: we could use "progname" here, in place of argv[0]. (This applies to all other fprintf()s below.)> + > + /* FIXME: future API addition to test if server negotiated extended mode. > + * Until then, strict flags must ignore the PAYLOAD_LEN flag for pwrite, > + * even though it is rejected for other commands. > + */ > + strict = nbd_get_strict_mode (nbd); > + if (!(strict & LIBNBD_STRICT_FLAGS)) { > + fprintf (stderr, "%s: test failed: " > + "nbd_get_strict_mode did not have expected flag set\n", > + argv[0]); > + exit (EXIT_FAILURE); > + }Not sure if I understand this check. Per <https://libguestfs.org/nbd_set_strict_mode.3.html>, I take it that LIBNBD_STRICT_FLAGS should be "on" by default. Are you enforcing that? And if so: is it your intent that, *even with* LIBNBD_STRICT_FLAGS, an invalid PAYLOAD_LEN is not rejected (as seen by the libnbd client app), but fixed up silently?> + if (nbd_aio_pread (nbd, buf, 512, 0, NBD_NULL_COMPLETION, > + LIBNBD_CMD_FLAG_PAYLOAD_LEN) != -1) { > + fprintf (stderr, "%s: test failed: " > + "nbd_aio_pread did not fail with unexpected flag\n", > + argv[0]); > + exit (EXIT_FAILURE); > + } > + check (EINVAL, "nbd_aio_pread: ");Ah, got it now. We do want APIs other than pwrite to fail.> + > + if (nbd_aio_pwrite (nbd, buf, 512, 0, NBD_NULL_COMPLETION, > + LIBNBD_CMD_FLAG_PAYLOAD_LEN) == -1) { > + fprintf (stderr, "%s: %s\n", argv[0], nbd_get_error ()); > + exit (EXIT_FAILURE); > + } > + > + if (nbd_aio_pwrite (nbd, buf, 512, 0, NBD_NULL_COMPLETION, 0) == -1) { > + fprintf (stderr, "%s: %s\n", argv[0], nbd_get_error ()); > + exit (EXIT_FAILURE); > + }You could contract these into: if (nbd_aio_pwrite (nbd, buf, 512, 0, NBD_NULL_COMPLETION, LIBNBD_CMD_FLAG_PAYLOAD_LEN) == -1 || nbd_aio_pwrite (nbd, buf, 512, 0, NBD_NULL_COMPLETION, 0) == -1) { fprintf (stderr, "%s: %s\n", argv[0], nbd_get_error ()); exit (EXIT_FAILURE); }> + > + nbd_close (nbd); > + exit (EXIT_SUCCESS); > +}In general, I think it's good practice to reach nbd_close() whenever nbd_create() succeeds (that is, on error paths as well, after nbd_create() succeeds). For example, if we connected to the server with systemd socket activation in this test, and (say) one of the pwrites failed, then we'd leak the unix domain socket in the filesystem (from the bind() in "generator/states-connect-socket-activation.c"). "sact_sockpath" is unlinked in nbd_close(). (As written, this test should not be affected, because, according to unix(7), unix domain sockets created with socketpair(2) are unnamed.)> diff --git a/.gitignore b/.gitignore > index fe7feffa..bc7c2c37 100644 > --- a/.gitignore > +++ b/.gitignore > @@ -250,6 +250,7 @@ Makefile.in > /tests/pki/ > /tests/pread-initialize > /tests/private-data > +/tests/pwrite-extended > /tests/read-only-flag > /tests/read-write-flag > /tests/server-deathThanks Laszlo
Eric Blake
2023-May-30 18:18 UTC
[Libguestfs] [libnbd PATCH v3 04/22] states: Prepare to send 64-bit requests
On Tue, May 30, 2023 at 04:06:32PM +0200, Laszlo Ersek wrote:> On 5/25/23 15:00, Eric Blake wrote: > > Support sending 64-bit requests if extended headers were negotiated. > > This includes setting NBD_CMD_FLAG_PAYLOAD_LEN any time we send an > > extended NBD_CMD_WRITE; this is such a fundamental part of the > > protocol that for now it is easier to silently ignore whatever value > > the user passes in for that bit in the flags parameter of nbd_pwrite > > regardless of the current settings in set_strict_mode, rather than > > trying to force the user to pass in the correct value to match whether > > extended mode is negotiated. However, when we later add APIs to give > > the user more control for interoperability testing, it may be worth > > adding a new set_strict_mode control knob to explicitly enable the > > client to intentionally violate the protocol (the testsuite added in > > this patch would then be updated to match). > > > > At this point, h->extended_headers is permanently false (we can't > > enable it until all other aspects of the protocol have likewise been > > converted). > > > > Support for using FLAG_PAYLOAD_LEN with NBD_CMD_BLOCK_STATUS is less > > fundamental, and deserves to be in its own patch. > > > > Signed-off-by: Eric Blake <eblake at redhat.com> > > ---> > @@ -364,7 +370,7 @@ struct command { > > uint16_t type; > > uint64_t cookie; > > uint64_t offset; > > - uint32_t count; > > + uint64_t count; > > void *data; /* Buffer for read/write */ > > struct command_cb cb; > > bool initialized; /* For read, true if getting a hole may skip memset */ > > (1) Are there places in the code where we currently assign this "count" > field back to a uint32_t object, and assume truncation impossible?Grepping for '->count' in lib/ and generator/ shows we need to check at least: generator/states-reply-simple.c: h->rlen = cmd->count; generator/states-reply-simple.c: cmd->data_seen += cmd->count; which are adjustments to size_t and uint32_t variables respectively, in response to a server's reply to an NBD_CMD_READ command. But since we never send a server a read request larger than 64M, truncation and overflow are not possible in those lines of code (at most one simple reply is possible, and code in states-reply-structured.c ensures that cmd->data_seen is a saturating variable that never exceeds 2*MAX_REQUEST_SIZE). There is also pre-series: generator/states-issue-command.c: h->request.count = htobe32 (cmd->count); which is specifically updated in this patch to cover extended headers.> > +++ b/generator/states-issue-command.c > > @@ -41,15 +41,24 @@ ISSUE_COMMAND.START: > > return 0; > > } > > > > - h->request.magic = htobe32 (NBD_REQUEST_MAGIC); > > - h->request.flags = htobe16 (cmd->flags); > > - h->request.type = htobe16 (cmd->type); > > - h->request.handle = htobe64 (cmd->cookie); > > - h->request.offset = htobe64 (cmd->offset); > > - h->request.count = htobe32 (cmd->count); > > + /* These fields are coincident between req.compact and req.extended */ > > + h->req.compact.flags = htobe16 (cmd->flags); > > + h->req.compact.type = htobe16 (cmd->type); > > + h->req.compact.handle = htobe64 (cmd->cookie); > > + h->req.compact.offset = htobe64 (cmd->offset); > > What's more, this is a "by the book" common initial sequence! :)> > > + if (h->extended_headers) { > > + h->req.extended.magic = htobe32 (NBD_EXTENDED_REQUEST_MAGIC); > > + h->req.extended.count = htobe64 (cmd->count); > > + h->wlen = sizeof (h->req.extended); > > + } > > + else { > > + assert (cmd->count <= UINT32_MAX); > > + h->req.compact.magic = htobe32 (NBD_REQUEST_MAGIC); > > + h->req.compact.count = htobe32 (cmd->count); > > + h->wlen = sizeof (h->req.compact); > > + }Indeed, and shows why my efforts to get a sane layout early in the series matter, even if it will cause me a bit more rebase churn here based on my response to your comments earlier in the series.> > @@ -358,6 +356,15 @@ nbd_unlocked_aio_pwrite (struct nbd_handle *h, const void *buf, > > return -1; > > } > > } > > + /* It is more convenient to manage PAYLOAD_LEN by what was negotiated > > + * than to require the user to have to set it correctly. > > + * TODO: Add new h->strict bit to allow intentional protocol violation > > + * for interoperability testing. > > + */ > > + if (h->extended_headers) > > + flags |= LIBNBD_CMD_FLAG_PAYLOAD_LEN; > > + else > > + flags &= ~LIBNBD_CMD_FLAG_PAYLOAD_LEN; > > Nice -- I wanted to ask for: > > flags &= ~(uint32_t)LIBNBD_CMD_FLAG_PAYLOAD_LEN; > > due to LIBNBD_CMD_FLAG_PAYLOAD_LEN having type "int". > > However: in patch#3, what has type "int" is: > > +#define NBD_CMD_FLAG_PAYLOAD_LEN (1<<5) > > and here we have LIBNBD_CMD_FLAG_PAYLOAD_LEN instead -- and the latter > has type unsigned int already, from your recent commit 69eecae2c03a > ("api: Generate flag values as unsigned", 2022-11-11).Still, worth a (separate) cleanup patch to nbd-protocol.h to prefer unsigned constants for the flag values where they are not generated.> > And I think we're fine assuming that uint32_t is unsigned int.Not true of all generic C platforms, but certainly true for the POSIX-like platforms we target (anyone that defines uint32_t as 'unsigned long' on a platform with 32-bit longs is unusual, but even then we should still be okay).> > > > > SET_CALLBACK_TO_NULL (*completion); > > return nbd_internal_command_common (h, flags, NBD_CMD_WRITE, offset, count, > > diff --git a/tests/Makefile.am b/tests/Makefile.am > > index 3a93251e..8b839bf5 100644 > > --- a/tests/Makefile.am > > +++ b/tests/Makefile.am > > @@ -232,6 +232,7 @@ check_PROGRAMS += \ > > closure-lifetimes \ > > pread-initialize \ > > socket-activation-name \ > > + pwrite-extended \ > > $(NULL) > > > > TESTS += \ > > (2) Incorrect indentation: two spaces rather than one tab.Arrgh. ./.editorconfig is supposed to do this correctly, but obviously its interaction with emacs is a bit botched when it comes to Makefile syntax. Will clean up.> > +++ b/tests/pwrite-extended.c> > +static void > > +check (int experr, const char *prefix) > > +{ > > + const char *msg = nbd_get_error (); > > + int errnum = nbd_get_errno (); > > + > > + fprintf (stderr, "error: \"%s\"\n", msg); > > + fprintf (stderr, "errno: %d (%s)\n", errnum, strerror (errnum)); > > + if (strncmp (msg, prefix, strlen (prefix)) != 0) { > > + fprintf (stderr, "%s: test failed: missing context prefix: %s\n", > > + progname, msg); > > + exit (EXIT_FAILURE); > > + } > > + if (errnum != experr) { > > + fprintf (stderr, "%s: test failed: " > > + "expected errno = %d (%s), but got %d\n", > > + progname, experr, strerror (experr), errnum); > > + exit (EXIT_FAILURE); > > + } > > +} > > + > > +int > > +main (int argc, char *argv[]) > > +{ > > + struct nbd_handle *nbd; > > + const char *cmd[] = { > > + "nbdkit", "-s", "-v", "--exit-with-parent", "memory", "1048576", NULL > > + }; > > + uint32_t strict; > > + > > + progname = argv[0]; > > + > > + nbd = nbd_create (); > > + if (nbd == NULL) { > > + fprintf (stderr, "%s\n", nbd_get_error ()); > > (3) Minor inconsistency with check(): we're not printing "progname" here. > > > + exit (EXIT_FAILURE); > > + } > > + > > + /* Connect to the server. */ > > + if (nbd_connect_command (nbd, (char **)cmd) == -1) { > > + fprintf (stderr, "%s: %s\n", argv[0], nbd_get_error ()); > > + exit (EXIT_FAILURE); > > + } > > (4) Another kind of inconsistency: we could use "progname" here, in > place of argv[0]. > > (This applies to all other fprintf()s below.)Probably copy-and-paste from other similar tests, but I don't mind cleaning those up.> > > + > > + /* FIXME: future API addition to test if server negotiated extended mode. > > + * Until then, strict flags must ignore the PAYLOAD_LEN flag for pwrite, > > + * even though it is rejected for other commands. > > + */ > > + strict = nbd_get_strict_mode (nbd); > > + if (!(strict & LIBNBD_STRICT_FLAGS)) { > > + fprintf (stderr, "%s: test failed: " > > + "nbd_get_strict_mode did not have expected flag set\n", > > + argv[0]); > > + exit (EXIT_FAILURE); > > + } > > Not sure if I understand this check. Per > <https://libguestfs.org/nbd_set_strict_mode.3.html>, I take it that > LIBNBD_STRICT_FLAGS should be "on" by default. Are you enforcing that? > And if so: is it your intent that, *even with* LIBNBD_STRICT_FLAGS, an > invalid PAYLOAD_LEN is not rejected (as seen by the libnbd client app), > but fixed up silently?Rather, until we can tell if the server negotiated extended mode, we are ASSUMING that the server did NOT negotiate it, and therefore we are in violation of the spec if we send the flag over the wire anyways. We can flag all other API where it is inappropriate to ever use...> > > + if (nbd_aio_pread (nbd, buf, 512, 0, NBD_NULL_COMPLETION, > > + LIBNBD_CMD_FLAG_PAYLOAD_LEN) != -1) { > > + fprintf (stderr, "%s: test failed: " > > + "nbd_aio_pread did not fail with unexpected flag\n", > > + argv[0]); > > + exit (EXIT_FAILURE); > > + } > > + check (EINVAL, "nbd_aio_pread: "); > > Ah, got it now. We do want APIs other than pwrite to fail....but because we don't want to require clients to correctly decide when to pass or omit the flag to their API calls (by us masking out the user's choice and then hardcoding our actual wire behavior based on negotiated mode), passing the flag to pread works even when it would be technically wrong over the wire. The FIXME does get modified again later in the series, when I do add in support for detecting when the server supports extended headers.> > > + > > + if (nbd_aio_pwrite (nbd, buf, 512, 0, NBD_NULL_COMPLETION, > > + LIBNBD_CMD_FLAG_PAYLOAD_LEN) == -1) { > > + fprintf (stderr, "%s: %s\n", argv[0], nbd_get_error ()); > > + exit (EXIT_FAILURE); > > + } > > + > > + if (nbd_aio_pwrite (nbd, buf, 512, 0, NBD_NULL_COMPLETION, 0) == -1) { > > + fprintf (stderr, "%s: %s\n", argv[0], nbd_get_error ()); > > + exit (EXIT_FAILURE); > > + } > > You could contract these into: > > if (nbd_aio_pwrite (nbd, buf, 512, 0, NBD_NULL_COMPLETION, > LIBNBD_CMD_FLAG_PAYLOAD_LEN) == -1 || > nbd_aio_pwrite (nbd, buf, 512, 0, NBD_NULL_COMPLETION, 0) == -1) { > fprintf (stderr, "%s: %s\n", argv[0], nbd_get_error ()); > exit (EXIT_FAILURE); > }Sure.> > > + > > + nbd_close (nbd); > > + exit (EXIT_SUCCESS); > > +} > > In general, I think it's good practice to reach nbd_close() whenever > nbd_create() succeeds (that is, on error paths as well, after > nbd_create() succeeds). For example, if we connected to the server with > systemd socket activation in this test, and (say) one of the pwrites > failed, then we'd leak the unix domain socket in the filesystem (from > the bind() in "generator/states-connect-socket-activation.c"). > "sact_sockpath" is unlinked in nbd_close(). > > (As written, this test should not be affected, because, according to > unix(7), unix domain sockets created with socketpair(2) are unnamed.)Pre-existing in other tests, but a good observation for a followup patch. -- Eric Blake, Principal Software Engineer Red Hat, Inc. +1-919-301-3266 Virtualization: qemu.org | libvirt.org
Maybe Matching Threads
- [libnbd PATCH] tests: Enhance errors test
- Re: [PATCH libnbd 1/4] api: Combine callback and user_data into a single struct.
- [libnbd PATCH 0/2] Fix docs and testing of completion callback
- Re: [PATCH libnbd 1/4] api: Combine callback and user_data into a single struct.
- [libnbd PATCH v3 0/7] Avoid deadlock with in-flight commands