diff mbox series

[v2,02/10] KVM: selftests: simplify setup_demand_paging error handling

Message ID 20210519200339.829146-3-axelrasmussen@google.com
State Accepted
Commit 25408e5a0246048e3e36d2cd513565ebcc481f51
Headers show
Series KVM: selftests: exercise userfaultfd minor faults | expand

Commit Message

Axel Rasmussen May 19, 2021, 8:03 p.m. UTC
A small cleanup. Our caller writes:

  r = setup_demand_paging(...);
  if (r < 0) exit(-r);

Since we're just going to exit anyway, instead of returning an error we
can just re-use TEST_ASSERT. This makes the caller simpler, as well as
the function itself - no need to write our branches, etc.

Signed-off-by: Axel Rasmussen <axelrasmussen@google.com>
---
 .../selftests/kvm/demand_paging_test.c        | 51 +++++++------------
 1 file changed, 19 insertions(+), 32 deletions(-)

Comments

Axel Rasmussen May 19, 2021, 10:14 p.m. UTC | #1
On Wed, May 19, 2021 at 2:45 PM Ben Gardon <bgardon@google.com> wrote:
>
> On Wed, May 19, 2021 at 1:03 PM Axel Rasmussen <axelrasmussen@google.com> wrote:
> >
> > A small cleanup. Our caller writes:
> >
> >   r = setup_demand_paging(...);
> >   if (r < 0) exit(-r);
> >
> > Since we're just going to exit anyway, instead of returning an error we
> > can just re-use TEST_ASSERT. This makes the caller simpler, as well as
> > the function itself - no need to write our branches, etc.
> >
> > Signed-off-by: Axel Rasmussen <axelrasmussen@google.com>
> > ---
> >  .../selftests/kvm/demand_paging_test.c        | 51 +++++++------------
> >  1 file changed, 19 insertions(+), 32 deletions(-)
> >
> > diff --git a/tools/testing/selftests/kvm/demand_paging_test.c b/tools/testing/selftests/kvm/demand_paging_test.c
> > index 9398ba6ef023..601a1df24dd2 100644
> > --- a/tools/testing/selftests/kvm/demand_paging_test.c
> > +++ b/tools/testing/selftests/kvm/demand_paging_test.c
> > @@ -9,6 +9,8 @@
> >
> >  #define _GNU_SOURCE /* for pipe2 */
> >
> > +#include <inttypes.h>
> > +#include <stdint.h>
>
> Why do the includes need to change in this commit? Is it for the PRIu64 below?

Right, I didn't actually try compiling without these, but inttypes.h
defines PRIu64 and stdint.h defines uint64_t. In general I tend to
prefer including things like this because we're using their
definitions directly, even if we might be picking them up transiently
some other way.

>
> >  #include <stdio.h>
> >  #include <stdlib.h>
> >  #include <time.h>
> > @@ -198,42 +200,32 @@ static void *uffd_handler_thread_fn(void *arg)
> >         return NULL;
> >  }
> >
> > -static int setup_demand_paging(struct kvm_vm *vm,
> > -                              pthread_t *uffd_handler_thread, int pipefd,
> > -                              useconds_t uffd_delay,
> > -                              struct uffd_handler_args *uffd_args,
> > -                              void *hva, uint64_t len)
> > +static void setup_demand_paging(struct kvm_vm *vm,
> > +                               pthread_t *uffd_handler_thread, int pipefd,
> > +                               useconds_t uffd_delay,
> > +                               struct uffd_handler_args *uffd_args,
> > +                               void *hva, uint64_t len)
> >  {
> >         int uffd;
> >         struct uffdio_api uffdio_api;
> >         struct uffdio_register uffdio_register;
> >
> >         uffd = syscall(__NR_userfaultfd, O_CLOEXEC | O_NONBLOCK);
> > -       if (uffd == -1) {
> > -               pr_info("uffd creation failed\n");
> > -               return -1;
> > -       }
> > +       TEST_ASSERT(uffd >= 0, "uffd creation failed, errno: %d", errno);
> >
> >         uffdio_api.api = UFFD_API;
> >         uffdio_api.features = 0;
> > -       if (ioctl(uffd, UFFDIO_API, &uffdio_api) == -1) {
> > -               pr_info("ioctl uffdio_api failed\n");
> > -               return -1;
> > -       }
> > +       TEST_ASSERT(ioctl(uffd, UFFDIO_API, &uffdio_api) != -1,
> > +                   "ioctl UFFDIO_API failed: %" PRIu64,
> > +                   (uint64_t)uffdio_api.api);
> >
> >         uffdio_register.range.start = (uint64_t)hva;
> >         uffdio_register.range.len = len;
> >         uffdio_register.mode = UFFDIO_REGISTER_MODE_MISSING;
> > -       if (ioctl(uffd, UFFDIO_REGISTER, &uffdio_register) == -1) {
> > -               pr_info("ioctl uffdio_register failed\n");
> > -               return -1;
> > -       }
> > -
> > -       if ((uffdio_register.ioctls & UFFD_API_RANGE_IOCTLS) !=
> > -                       UFFD_API_RANGE_IOCTLS) {
> > -               pr_info("unexpected userfaultfd ioctl set\n");
> > -               return -1;
> > -       }
> > +       TEST_ASSERT(ioctl(uffd, UFFDIO_REGISTER, &uffdio_register) != -1,
> > +                   "ioctl UFFDIO_REGISTER failed");
> > +       TEST_ASSERT((uffdio_register.ioctls & UFFD_API_RANGE_IOCTLS) ==
> > +                   UFFD_API_RANGE_IOCTLS, "unexpected userfaultfd ioctl set");
> >
> >         uffd_args->uffd = uffd;
> >         uffd_args->pipefd = pipefd;
> > @@ -243,8 +235,6 @@ static int setup_demand_paging(struct kvm_vm *vm,
> >
> >         PER_VCPU_DEBUG("Created uffd thread for HVA range [%p, %p)\n",
> >                        hva, hva + len);
> > -
> > -       return 0;
> >  }
> >
> >  struct test_params {
> > @@ -321,13 +311,10 @@ static void run_test(enum vm_guest_mode mode, void *arg)
> >                                   O_CLOEXEC | O_NONBLOCK);
> >                         TEST_ASSERT(!r, "Failed to set up pipefd");
> >
> > -                       r = setup_demand_paging(vm,
> > -                                               &uffd_handler_threads[vcpu_id],
> > -                                               pipefds[vcpu_id * 2],
> > -                                               p->uffd_delay, &uffd_args[vcpu_id],
> > -                                               vcpu_hva, vcpu_mem_size);
> > -                       if (r < 0)
> > -                               exit(-r);
> > +                       setup_demand_paging(vm, &uffd_handler_threads[vcpu_id],
> > +                                           pipefds[vcpu_id * 2], p->uffd_delay,
> > +                                           &uffd_args[vcpu_id], vcpu_hva,
> > +                                           vcpu_mem_size);
> >                 }
> >         }
> >
> > --
> > 2.31.1.751.gd2f1c929bd-goog
> >
Ben Gardon May 19, 2021, 10:23 p.m. UTC | #2
On Wed, May 19, 2021 at 3:14 PM Axel Rasmussen <axelrasmussen@google.com> wrote:
>
> On Wed, May 19, 2021 at 2:45 PM Ben Gardon <bgardon@google.com> wrote:
> >
> > On Wed, May 19, 2021 at 1:03 PM Axel Rasmussen <axelrasmussen@google.com> wrote:
> > >
> > > A small cleanup. Our caller writes:
> > >
> > >   r = setup_demand_paging(...);
> > >   if (r < 0) exit(-r);
> > >
> > > Since we're just going to exit anyway, instead of returning an error we
> > > can just re-use TEST_ASSERT. This makes the caller simpler, as well as
> > > the function itself - no need to write our branches, etc.
> > >
> > > Signed-off-by: Axel Rasmussen <axelrasmussen@google.com>

Reviewed-by: Ben Gardon <bgardon@google.com>

> > > ---
> > >  .../selftests/kvm/demand_paging_test.c        | 51 +++++++------------
> > >  1 file changed, 19 insertions(+), 32 deletions(-)
> > >
> > > diff --git a/tools/testing/selftests/kvm/demand_paging_test.c b/tools/testing/selftests/kvm/demand_paging_test.c
> > > index 9398ba6ef023..601a1df24dd2 100644
> > > --- a/tools/testing/selftests/kvm/demand_paging_test.c
> > > +++ b/tools/testing/selftests/kvm/demand_paging_test.c
> > > @@ -9,6 +9,8 @@
> > >
> > >  #define _GNU_SOURCE /* for pipe2 */
> > >
> > > +#include <inttypes.h>
> > > +#include <stdint.h>
> >
> > Why do the includes need to change in this commit? Is it for the PRIu64 below?
>
> Right, I didn't actually try compiling without these, but inttypes.h
> defines PRIu64 and stdint.h defines uint64_t. In general I tend to
> prefer including things like this because we're using their
> definitions directly, even if we might be picking them up transiently
> some other way.

Makes sense to me.

>
> >
> > >  #include <stdio.h>
> > >  #include <stdlib.h>
> > >  #include <time.h>
> > > @@ -198,42 +200,32 @@ static void *uffd_handler_thread_fn(void *arg)
> > >         return NULL;
> > >  }
> > >
> > > -static int setup_demand_paging(struct kvm_vm *vm,
> > > -                              pthread_t *uffd_handler_thread, int pipefd,
> > > -                              useconds_t uffd_delay,
> > > -                              struct uffd_handler_args *uffd_args,
> > > -                              void *hva, uint64_t len)
> > > +static void setup_demand_paging(struct kvm_vm *vm,
> > > +                               pthread_t *uffd_handler_thread, int pipefd,
> > > +                               useconds_t uffd_delay,
> > > +                               struct uffd_handler_args *uffd_args,
> > > +                               void *hva, uint64_t len)
> > >  {
> > >         int uffd;
> > >         struct uffdio_api uffdio_api;
> > >         struct uffdio_register uffdio_register;
> > >
> > >         uffd = syscall(__NR_userfaultfd, O_CLOEXEC | O_NONBLOCK);
> > > -       if (uffd == -1) {
> > > -               pr_info("uffd creation failed\n");
> > > -               return -1;
> > > -       }
> > > +       TEST_ASSERT(uffd >= 0, "uffd creation failed, errno: %d", errno);
> > >
> > >         uffdio_api.api = UFFD_API;
> > >         uffdio_api.features = 0;
> > > -       if (ioctl(uffd, UFFDIO_API, &uffdio_api) == -1) {
> > > -               pr_info("ioctl uffdio_api failed\n");
> > > -               return -1;
> > > -       }
> > > +       TEST_ASSERT(ioctl(uffd, UFFDIO_API, &uffdio_api) != -1,
> > > +                   "ioctl UFFDIO_API failed: %" PRIu64,
> > > +                   (uint64_t)uffdio_api.api);
> > >
> > >         uffdio_register.range.start = (uint64_t)hva;
> > >         uffdio_register.range.len = len;
> > >         uffdio_register.mode = UFFDIO_REGISTER_MODE_MISSING;
> > > -       if (ioctl(uffd, UFFDIO_REGISTER, &uffdio_register) == -1) {
> > > -               pr_info("ioctl uffdio_register failed\n");
> > > -               return -1;
> > > -       }
> > > -
> > > -       if ((uffdio_register.ioctls & UFFD_API_RANGE_IOCTLS) !=
> > > -                       UFFD_API_RANGE_IOCTLS) {
> > > -               pr_info("unexpected userfaultfd ioctl set\n");
> > > -               return -1;
> > > -       }
> > > +       TEST_ASSERT(ioctl(uffd, UFFDIO_REGISTER, &uffdio_register) != -1,
> > > +                   "ioctl UFFDIO_REGISTER failed");
> > > +       TEST_ASSERT((uffdio_register.ioctls & UFFD_API_RANGE_IOCTLS) ==
> > > +                   UFFD_API_RANGE_IOCTLS, "unexpected userfaultfd ioctl set");
> > >
> > >         uffd_args->uffd = uffd;
> > >         uffd_args->pipefd = pipefd;
> > > @@ -243,8 +235,6 @@ static int setup_demand_paging(struct kvm_vm *vm,
> > >
> > >         PER_VCPU_DEBUG("Created uffd thread for HVA range [%p, %p)\n",
> > >                        hva, hva + len);
> > > -
> > > -       return 0;
> > >  }
> > >
> > >  struct test_params {
> > > @@ -321,13 +311,10 @@ static void run_test(enum vm_guest_mode mode, void *arg)
> > >                                   O_CLOEXEC | O_NONBLOCK);
> > >                         TEST_ASSERT(!r, "Failed to set up pipefd");
> > >
> > > -                       r = setup_demand_paging(vm,
> > > -                                               &uffd_handler_threads[vcpu_id],
> > > -                                               pipefds[vcpu_id * 2],
> > > -                                               p->uffd_delay, &uffd_args[vcpu_id],
> > > -                                               vcpu_hva, vcpu_mem_size);
> > > -                       if (r < 0)
> > > -                               exit(-r);
> > > +                       setup_demand_paging(vm, &uffd_handler_threads[vcpu_id],
> > > +                                           pipefds[vcpu_id * 2], p->uffd_delay,
> > > +                                           &uffd_args[vcpu_id], vcpu_hva,
> > > +                                           vcpu_mem_size);
> > >                 }
> > >         }
> > >
> > > --
> > > 2.31.1.751.gd2f1c929bd-goog
> > >
Paolo Bonzini May 24, 2021, 1:25 p.m. UTC | #3
On 20/05/21 00:14, Axel Rasmussen wrote:
> On Wed, May 19, 2021 at 2:45 PM Ben Gardon <bgardon@google.com> wrote:

>>

>> On Wed, May 19, 2021 at 1:03 PM Axel Rasmussen <axelrasmussen@google.com> wrote:

>>>

>>> A small cleanup. Our caller writes:

>>>

>>>    r = setup_demand_paging(...);

>>>    if (r < 0) exit(-r);

>>>

>>> Since we're just going to exit anyway, instead of returning an error we

>>> can just re-use TEST_ASSERT. This makes the caller simpler, as well as

>>> the function itself - no need to write our branches, etc.

>>>

>>> Signed-off-by: Axel Rasmussen <axelrasmussen@google.com>

>>> ---

>>>   .../selftests/kvm/demand_paging_test.c        | 51 +++++++------------

>>>   1 file changed, 19 insertions(+), 32 deletions(-)

>>>

>>> diff --git a/tools/testing/selftests/kvm/demand_paging_test.c b/tools/testing/selftests/kvm/demand_paging_test.c

>>> index 9398ba6ef023..601a1df24dd2 100644

>>> --- a/tools/testing/selftests/kvm/demand_paging_test.c

>>> +++ b/tools/testing/selftests/kvm/demand_paging_test.c

>>> @@ -9,6 +9,8 @@

>>>

>>>   #define _GNU_SOURCE /* for pipe2 */

>>>

>>> +#include <inttypes.h>

>>> +#include <stdint.h>

>>

>> Why do the includes need to change in this commit? Is it for the PRIu64 below?

> 

> Right, I didn't actually try compiling without these, but inttypes.h

> defines PRIu64 and stdint.h defines uint64_t. In general I tend to

> prefer including things like this because we're using their

> definitions directly, even if we might be picking them up transiently

> some other way.


inttypes.h is defined to include stdint.h (stdint.h is mostly useful in 
freestanding environments and is usually provided by the C compiler, 
while inttypes.h is provided by libc).

Paolo
diff mbox series

Patch

diff --git a/tools/testing/selftests/kvm/demand_paging_test.c b/tools/testing/selftests/kvm/demand_paging_test.c
index 9398ba6ef023..601a1df24dd2 100644
--- a/tools/testing/selftests/kvm/demand_paging_test.c
+++ b/tools/testing/selftests/kvm/demand_paging_test.c
@@ -9,6 +9,8 @@ 
 
 #define _GNU_SOURCE /* for pipe2 */
 
+#include <inttypes.h>
+#include <stdint.h>
 #include <stdio.h>
 #include <stdlib.h>
 #include <time.h>
@@ -198,42 +200,32 @@  static void *uffd_handler_thread_fn(void *arg)
 	return NULL;
 }
 
-static int setup_demand_paging(struct kvm_vm *vm,
-			       pthread_t *uffd_handler_thread, int pipefd,
-			       useconds_t uffd_delay,
-			       struct uffd_handler_args *uffd_args,
-			       void *hva, uint64_t len)
+static void setup_demand_paging(struct kvm_vm *vm,
+				pthread_t *uffd_handler_thread, int pipefd,
+				useconds_t uffd_delay,
+				struct uffd_handler_args *uffd_args,
+				void *hva, uint64_t len)
 {
 	int uffd;
 	struct uffdio_api uffdio_api;
 	struct uffdio_register uffdio_register;
 
 	uffd = syscall(__NR_userfaultfd, O_CLOEXEC | O_NONBLOCK);
-	if (uffd == -1) {
-		pr_info("uffd creation failed\n");
-		return -1;
-	}
+	TEST_ASSERT(uffd >= 0, "uffd creation failed, errno: %d", errno);
 
 	uffdio_api.api = UFFD_API;
 	uffdio_api.features = 0;
-	if (ioctl(uffd, UFFDIO_API, &uffdio_api) == -1) {
-		pr_info("ioctl uffdio_api failed\n");
-		return -1;
-	}
+	TEST_ASSERT(ioctl(uffd, UFFDIO_API, &uffdio_api) != -1,
+		    "ioctl UFFDIO_API failed: %" PRIu64,
+		    (uint64_t)uffdio_api.api);
 
 	uffdio_register.range.start = (uint64_t)hva;
 	uffdio_register.range.len = len;
 	uffdio_register.mode = UFFDIO_REGISTER_MODE_MISSING;
-	if (ioctl(uffd, UFFDIO_REGISTER, &uffdio_register) == -1) {
-		pr_info("ioctl uffdio_register failed\n");
-		return -1;
-	}
-
-	if ((uffdio_register.ioctls & UFFD_API_RANGE_IOCTLS) !=
-			UFFD_API_RANGE_IOCTLS) {
-		pr_info("unexpected userfaultfd ioctl set\n");
-		return -1;
-	}
+	TEST_ASSERT(ioctl(uffd, UFFDIO_REGISTER, &uffdio_register) != -1,
+		    "ioctl UFFDIO_REGISTER failed");
+	TEST_ASSERT((uffdio_register.ioctls & UFFD_API_RANGE_IOCTLS) ==
+		    UFFD_API_RANGE_IOCTLS, "unexpected userfaultfd ioctl set");
 
 	uffd_args->uffd = uffd;
 	uffd_args->pipefd = pipefd;
@@ -243,8 +235,6 @@  static int setup_demand_paging(struct kvm_vm *vm,
 
 	PER_VCPU_DEBUG("Created uffd thread for HVA range [%p, %p)\n",
 		       hva, hva + len);
-
-	return 0;
 }
 
 struct test_params {
@@ -321,13 +311,10 @@  static void run_test(enum vm_guest_mode mode, void *arg)
 				  O_CLOEXEC | O_NONBLOCK);
 			TEST_ASSERT(!r, "Failed to set up pipefd");
 
-			r = setup_demand_paging(vm,
-						&uffd_handler_threads[vcpu_id],
-						pipefds[vcpu_id * 2],
-						p->uffd_delay, &uffd_args[vcpu_id],
-						vcpu_hva, vcpu_mem_size);
-			if (r < 0)
-				exit(-r);
+			setup_demand_paging(vm, &uffd_handler_threads[vcpu_id],
+					    pipefds[vcpu_id * 2], p->uffd_delay,
+					    &uffd_args[vcpu_id], vcpu_hva,
+					    vcpu_mem_size);
 		}
 	}