On Fri, 9 Dec 2022 14:48:08 +0100
Nico Boehr <nrb@xxxxxxxxxxxxx> wrote:
> migrate_once() can simplify the control flow in migration-skey and
> migration-cmm.
>
> Signed-off-by: Nico Boehr <nrb@xxxxxxxxxxxxx>
looks very good, just some small nits
> ---
> s390x/Makefile | 1 +
> s390x/migration-cmm.c | 25 ++++++++-----------------
> s390x/migration-sck.c | 4 ++--
> s390x/migration-skey.c | 15 +++++----------
> s390x/migration.c | 7 ++-----
> 5 files changed, 18 insertions(+), 34 deletions(-)
>
> diff --git a/s390x/Makefile b/s390x/Makefile
> index bf1504f9d58c..52a9d821974e 100644
> --- a/s390x/Makefile
> +++ b/s390x/Makefile
> @@ -85,6 +85,7 @@ cflatobjs += lib/alloc_page.o
> cflatobjs += lib/vmalloc.o
> cflatobjs += lib/alloc_phys.o
> cflatobjs += lib/getchar.o
> +cflatobjs += lib/migrate.o
> cflatobjs += lib/s390x/io.o
> cflatobjs += lib/s390x/stack.o
> cflatobjs += lib/s390x/sclp.o
> diff --git a/s390x/migration-cmm.c b/s390x/migration-cmm.c
> index aa7910ca76bf..eef45f5f8fb7 100644
> --- a/s390x/migration-cmm.c
> +++ b/s390x/migration-cmm.c
> @@ -9,6 +9,7 @@
> */
>
> #include <libcflat.h>
> +#include <migrate.h>
> #include <asm/interrupt.h>
> #include <asm/page.h>
> #include <asm/cmm.h>
> @@ -39,8 +40,7 @@ static void test_migration(void)
> essa(ESSA_SET_POT_VOLATILE, (unsigned long)pagebuf[i + 3]);
> }
>
> - puts("Please migrate me, then press return\n");
> - (void)getchar();
> + migrate_once();
>
> for (i = 0; i < NUM_PAGES; i++) {
> actual_state = essa(ESSA_GET_STATE, (unsigned long)pagebuf[i]);
> @@ -53,25 +53,16 @@ static void test_migration(void)
>
> int main(void)
> {
> - bool has_essa = check_essa_available();
> -
> report_prefix_push("migration-cmm");
> - if (!has_essa) {
> - report_skip("ESSA is not available");
> -
> - /*
> - * If we just exit and don't ask migrate_cmd to migrate us, it
> - * will just hang forever. Hence, also ask for migration when we
> - * skip this test alltogether.
> - */
> - puts("Please migrate me, then press return\n");
> - (void)getchar();
>
> - goto done;
> + if (!check_essa_available()) {
> + report_skip("ESSA is not available");
> + } else {
> + test_migration();
> }
I think you don't need the {} any longer here?
>
> - test_migration();
> -done:
> + migrate_once();
> +
> report_prefix_pop();
> return report_summary();
> }
> diff --git a/s390x/migration-sck.c b/s390x/migration-sck.c
> index 2d9a195ab4c4..2a9c87071643 100644
> --- a/s390x/migration-sck.c
> +++ b/s390x/migration-sck.c
> @@ -9,6 +9,7 @@
> */
>
> #include <libcflat.h>
> +#include <migrate.h>
> #include <asm/time.h>
>
> static void test_sck_migration(void)
> @@ -30,8 +31,7 @@ static void test_sck_migration(void)
> report(!cc, "clock running after set");
> report(now_after_set >= time_to_set, "TOD clock value is larger than what has been set");
>
> - puts("Please migrate me, then press return\n");
> - (void)getchar();
> + migrate_once();
>
> cc = stckf(&now_after_migration);
> report(!cc, "clock still set");
> diff --git a/s390x/migration-skey.c b/s390x/migration-skey.c
> index b7bd82581abe..438a4be95702 100644
> --- a/s390x/migration-skey.c
> +++ b/s390x/migration-skey.c
> @@ -9,6 +9,7 @@
> */
>
> #include <libcflat.h>
> +#include <migrate.h>
> #include <asm/facility.h>
> #include <asm/page.h>
> #include <asm/mem.h>
> @@ -35,8 +36,7 @@ static void test_migration(void)
> set_storage_key(pagebuf[i], key_to_set, 1);
> }
>
> - puts("Please migrate me, then press return\n");
> - (void)getchar();
> + migrate_once();
>
> for (i = 0; i < NUM_PAGES; i++) {
> actual_key.val = get_storage_key(pagebuf[i]);
> @@ -64,20 +64,15 @@ static void test_migration(void)
> int main(void)
> {
> report_prefix_push("migration-skey");
> +
> if (test_facility(169)) {
> report_skip("storage key removal facility is active");
> -
> - /*
> - * If we just exit and don't ask migrate_cmd to migrate us, it
> - * will just hang forever. Hence, also ask for migration when we
> - * skip this test altogether.
> - */
> - puts("Please migrate me, then press return\n");
> - (void)getchar();
> } else {
> test_migration();
> }
same here {} ^
>
> + migrate_once();
> +
> report_prefix_pop();
> return report_summary();
> }
> diff --git a/s390x/migration.c b/s390x/migration.c
> index a45296374cd8..fe6ea8369edb 100644
> --- a/s390x/migration.c
> +++ b/s390x/migration.c
> @@ -8,6 +8,7 @@
> * Nico Boehr <nrb@xxxxxxxxxxxxx>
> */
> #include <libcflat.h>
> +#include <migrate.h>
> #include <asm/arch_def.h>
> #include <asm/vector.h>
> #include <asm/barrier.h>
> @@ -178,11 +179,7 @@ int main(void)
> mb();
> flag_thread_complete = 0;
>
> - /* ask migrate_cmd to migrate (it listens for 'migrate') */
> - puts("Please migrate me, then press return\n");
> -
> - /* wait for migration to finish, we will read a newline */
> - (void)getchar();
> + migrate_once();
>
> flag_migration_complete = 1;
>
