On Mon, 12 Dec 2022 12:17:30 +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>
Reviewed-by: Claudio Imbrenda <imbrenda@xxxxxxxxxxxxx>
> ---
> s390x/Makefile | 1 +
> s390x/migration-cmm.c | 24 +++++++-----------------
> s390x/migration-sck.c | 4 ++--
> s390x/migration-skey.c | 20 +++++++-------------
> s390x/migration.c | 7 ++-----
> 5 files changed, 19 insertions(+), 37 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..43673f18e45a 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,15 @@ 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();
> + if (!check_essa_available())
> + report_skip("ESSA is not available");
> + else
> + test_migration();
>
> - goto done;
> - }
> + migrate_once();
>
> - test_migration();
> -done:
> 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..a91eb6b5a63e 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,19 +64,13 @@ 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 {
> + if (test_facility(169))
> + report_skip("storage key removal facility is active");
> + else
> test_migration();
> - }
> +
> + 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;
>
