On Thu, 1 Dec 2022 09:46:40 +0100 Nico Boehr <nrb@xxxxxxxxxxxxx> wrote: > Upcoming changes will add a test which is very similar to the existing > skey migration test. To reduce code duplication, move the common > functions to a library which can be re-used by both tests. > > Signed-off-by: Nico Boehr <nrb@xxxxxxxxxxxxx> a few nits, otherwise looks pretty straightforward > --- > lib/s390x/skey.c | 92 ++++++++++++++++++++++++++++++++++++++++++ > lib/s390x/skey.h | 32 +++++++++++++++ > s390x/Makefile | 1 + > s390x/migration-skey.c | 44 +++----------------- > 4 files changed, 131 insertions(+), 38 deletions(-) > create mode 100644 lib/s390x/skey.c > create mode 100644 lib/s390x/skey.h > > diff --git a/lib/s390x/skey.c b/lib/s390x/skey.c > new file mode 100644 > index 000000000000..100f0949a244 > --- /dev/null > +++ b/lib/s390x/skey.c > @@ -0,0 +1,92 @@ > +/* SPDX-License-Identifier: GPL-2.0-only */ > +/* > + * Storage key migration test library > + * > + * Copyright IBM Corp. 2022 > + * > + * Authors: > + * Nico Boehr <nrb@xxxxxxxxxxxxx> > + */ > + > +#include <libcflat.h> > +#include <asm/facility.h> > +#include <asm/mem.h> > +#include <skey.h> > + > +/* > + * Set storage keys on pagebuf. surely you should explain better what the function does (e.g. how are you setting the keys and why) > + * pagebuf must point to page_count consecutive pages. > + */ > +void skey_set_keys(uint8_t *pagebuf, unsigned long page_count) this name does not make clear what the function is doing. at first one would think that it sets the same key for all pages. maybe something like set_storage_keys_test_pattern or skey_set_test_pattern or something like that > +{ > + unsigned char key_to_set; > + unsigned long i; > + > + for (i = 0; i < page_count; i++) { > + /* > + * Storage keys are 7 bit, lowest bit is always returned as zero > + * by iske. > + * This loop will set all 7 bits which means we set fetch > + * protection as well as reference and change indication for > + * some keys. > + */ > + key_to_set = i * 2; > + set_storage_key(pagebuf + i * PAGE_SIZE, key_to_set, 1); why not just i * 2 instead of using key_to_set ? > + } > +} > + > +/* > + * Verify storage keys on pagebuf. > + * Storage keys must have been set by skey_set_keys on pagebuf before. > + * > + * If storage keys match the expected result, will return a skey_verify_result > + * with verify_failed false. All other fields are then invalid. > + * If there is a mismatch, returned struct will have verify_failed true and will > + * be filled with the details on the first mismatch encountered. > + */ > +struct skey_verify_result skey_verify_keys(uint8_t *pagebuf, unsigned long page_count) and here then adjust the function name accordingly > +{ > + union skey expected_key, actual_key; > + struct skey_verify_result result = { > + .verify_failed = true > + }; > + uint8_t *cur_page; > + unsigned long i; > + > + for (i = 0; i < page_count; i++) { > + cur_page = pagebuf + i * PAGE_SIZE; > + actual_key.val = get_storage_key(cur_page); > + expected_key.val = i * 2; > + > + /* > + * The PoP neither gives a guarantee that the reference bit is > + * accurate nor that it won't be cleared by hardware. Hence we > + * don't rely on it and just clear the bits to avoid compare > + * errors. > + */ > + actual_key.str.rf = 0; > + expected_key.str.rf = 0; > + > + if (actual_key.val != expected_key.val) { > + result.expected_key.val = expected_key.val; > + result.actual_key.val = actual_key.val; > + result.page_mismatch_idx = i; > + result.page_mismatch_addr = (unsigned long)cur_page; > + return result; > + } > + } > + > + result.verify_failed = false; > + return result; > +} > + > +void skey_report_verify(struct skey_verify_result * const result) > +{ > + if (result->verify_failed) > + report_fail("page skey mismatch: first page idx = %lu, addr = 0x%lx, " > + "expected_key = 0x%x, actual_key = 0x%x", > + result->page_mismatch_idx, result->page_mismatch_addr, > + result->expected_key.val, result->actual_key.val); > + else > + report_pass("skeys match"); > +} > diff --git a/lib/s390x/skey.h b/lib/s390x/skey.h > new file mode 100644 > index 000000000000..a0f8caa1270b > --- /dev/null > +++ b/lib/s390x/skey.h > @@ -0,0 +1,32 @@ > +/* SPDX-License-Identifier: GPL-2.0-only */ > +/* > + * Storage key migration test library > + * > + * Copyright IBM Corp. 2022 > + * > + * Authors: > + * Nico Boehr <nrb@xxxxxxxxxxxxx> > + */ > +#ifndef S390X_SKEY_H > +#define S390X_SKEY_H > + > +#include <libcflat.h> > +#include <asm/facility.h> > +#include <asm/page.h> > +#include <asm/mem.h> > + > +struct skey_verify_result { > + bool verify_failed; > + union skey expected_key; > + union skey actual_key; > + unsigned long page_mismatch_idx; > + unsigned long page_mismatch_addr; > +}; > + > +void skey_set_keys(uint8_t *pagebuf, unsigned long page_count); > + > +struct skey_verify_result skey_verify_keys(uint8_t *pagebuf, unsigned long page_count); > + > +void skey_report_verify(struct skey_verify_result * const result); > + > +#endif /* S390X_SKEY_H */ > diff --git a/s390x/Makefile b/s390x/Makefile > index bf1504f9d58c..d097b7071dfb 100644 > --- a/s390x/Makefile > +++ b/s390x/Makefile > @@ -99,6 +99,7 @@ cflatobjs += lib/s390x/malloc_io.o > cflatobjs += lib/s390x/uv.o > cflatobjs += lib/s390x/sie.o > cflatobjs += lib/s390x/fault.o > +cflatobjs += lib/s390x/skey.o > > OBJDIRS += lib/s390x > > diff --git a/s390x/migration-skey.c b/s390x/migration-skey.c > index b7bd82581abe..fed6fc1ed0f8 100644 > --- a/s390x/migration-skey.c > +++ b/s390x/migration-skey.c > @@ -10,55 +10,23 @@ > > #include <libcflat.h> > #include <asm/facility.h> > -#include <asm/page.h> > -#include <asm/mem.h> > -#include <asm/interrupt.h> > #include <hardware.h> > +#include <skey.h> > > #define NUM_PAGES 128 > -static uint8_t pagebuf[NUM_PAGES][PAGE_SIZE] __attribute__((aligned(PAGE_SIZE))); > +static uint8_t pagebuf[NUM_PAGES * PAGE_SIZE] __attribute__((aligned(PAGE_SIZE))); > > static void test_migration(void) > { > - union skey expected_key, actual_key; > - int i, key_to_set, key_mismatches = 0; > + struct skey_verify_result result; > > - for (i = 0; i < NUM_PAGES; i++) { > - /* > - * Storage keys are 7 bit, lowest bit is always returned as zero > - * by iske. > - * This loop will set all 7 bits which means we set fetch > - * protection as well as reference and change indication for > - * some keys. > - */ > - key_to_set = i * 2; ah I see, you have simply moved this code :) > - set_storage_key(pagebuf[i], key_to_set, 1); > - } > + skey_set_keys(pagebuf, NUM_PAGES); > > puts("Please migrate me, then press return\n"); > (void)getchar(); > > - for (i = 0; i < NUM_PAGES; i++) { > - actual_key.val = get_storage_key(pagebuf[i]); > - expected_key.val = i * 2; > - > - /* > - * The PoP neither gives a guarantee that the reference bit is > - * accurate nor that it won't be cleared by hardware. Hence we > - * don't rely on it and just clear the bits to avoid compare > - * errors. > - */ > - actual_key.str.rf = 0; > - expected_key.str.rf = 0; > - > - /* don't log anything when key matches to avoid spamming the log */ > - if (actual_key.val != expected_key.val) { > - key_mismatches++; > - report_fail("page %d expected_key=0x%x actual_key=0x%x", i, expected_key.val, actual_key.val); > - } > - } > - > - report(!key_mismatches, "skeys after migration match"); > + result = skey_verify_keys(pagebuf, NUM_PAGES); > + skey_report_verify(&result); > } > > int main(void)