Spdk/test/unit/lib/bdev/gpt/gpt.c/gpt_ut.c
paul luse a6dbe3721e update Intel copyright notices
per Intel policy to include file commit date using git cmd
below.  The policy does not apply to non-Intel (C) notices.

git log --follow -C90% --format=%ad --date default <file> | tail -1

and then pull just the 4 digit year from the result.

Intel copyrights were not added to files where Intel either had
no contribution ot the contribution lacked substance (ie license
header updates, formatting changes, etc).  Contribution date used
"--follow -C95%" to get the most accurate date.

Note that several files in this patch didn't end the license/(c)
block with a blank comment line so these were added as the vast
majority of files do have this last blank line.  Simply there for
consistency.

Signed-off-by: paul luse <paul.e.luse@intel.com>
Change-Id: Id5b7ce4f658fe87132f14139ead58d6e285c04d4
Reviewed-on: https://review.spdk.io/gerrit/c/spdk/spdk/+/15192
Tested-by: SPDK CI Jenkins <sys_sgci@intel.com>
Reviewed-by: Jim Harris <james.r.harris@intel.com>
Reviewed-by: Ben Walker <benjamin.walker@intel.com>
Community-CI: Mellanox Build Bot
2022-11-10 08:28:53 +00:00

336 lines
9.3 KiB
C

/* SPDX-License-Identifier: BSD-3-Clause
* Copyright (C) 2017 Intel Corporation.
* All rights reserved.
*/
#include "spdk_cunit.h"
#include "common/lib/test_env.c"
#include "bdev/gpt/gpt.c"
static void
test_check_mbr(void)
{
struct spdk_gpt *gpt;
struct spdk_mbr *mbr;
unsigned char a[SPDK_GPT_BUFFER_SIZE];
int re;
/* Set gpt is NULL */
re = gpt_parse_mbr(NULL);
CU_ASSERT(re == -1);
/* Set gpt->buf is NULL */
gpt = calloc(1, sizeof(*gpt));
SPDK_CU_ASSERT_FATAL(gpt != NULL);
re = gpt_parse_mbr(gpt);
CU_ASSERT(re == -1);
/* Set *gpt is "aaa...", all are mismatch include mbr_signature */
memset(a, 'a', sizeof(a));
gpt->buf = &a[0];
re = gpt_check_mbr(gpt);
CU_ASSERT(re == -1);
/* Set mbr->mbr_signature matched, start lba mismatch */
mbr = (struct spdk_mbr *)gpt->buf;
mbr->mbr_signature = 0xAA55;
re = gpt_check_mbr(gpt);
CU_ASSERT(re == -1);
/* Set mbr->partitions[0].start lba matched, os_type mismatch */
mbr->partitions[0].start_lba = 1;
re = gpt_check_mbr(gpt);
CU_ASSERT(re == -1);
/* Set mbr->partitions[0].os_type matched, size_lba mismatch */
mbr->partitions[0].os_type = 0xEE;
re = gpt_check_mbr(gpt);
CU_ASSERT(re == -1);
/* Set mbr->partitions[0].size_lba matched, passing case */
mbr->partitions[0].size_lba = 0xFFFFFFFF;
re = gpt_check_mbr(gpt);
CU_ASSERT(re == 0);
free(gpt);
}
static void
test_read_header(void)
{
struct spdk_gpt *gpt;
struct spdk_gpt_header *head;
unsigned char a[SPDK_GPT_BUFFER_SIZE];
int re;
/* gpt_read_header(NULL) does not exist, NULL is filtered out in gpt_parse_mbr() */
gpt = calloc(1, sizeof(*gpt));
SPDK_CU_ASSERT_FATAL(gpt != NULL);
gpt->parse_phase = SPDK_GPT_PARSE_PHASE_PRIMARY;
gpt->sector_size = 512;
/* Set *gpt is "aaa..." */
memset(a, 'a', sizeof(a));
gpt->buf = &a[0];
gpt->buf_size = sizeof(a);
/* Set header_size mismatch */
gpt->sector_size = 512;
head = (struct spdk_gpt_header *)(gpt->buf + GPT_PRIMARY_PARTITION_TABLE_LBA * gpt->sector_size);
to_le32(&head->header_size, 0x258);
re = gpt_read_header(gpt);
CU_ASSERT(re == -1);
/* Set head->header_size matched, header_crc32 mismatch */
head->header_size = sizeof(*head);
to_le32(&head->header_crc32, 0x22D18C80);
re = gpt_read_header(gpt);
CU_ASSERT(re == -1);
/* Set head->header_crc32 matched, gpt_signature mismatch */
to_le32(&head->header_crc32, 0xC5B2117E);
re = gpt_read_header(gpt);
CU_ASSERT(re == -1);
/* Set head->gpt_signature matched, head->my_lba mismatch */
to_le32(&head->header_crc32, 0xD637335A);
head->gpt_signature[0] = 'E';
head->gpt_signature[1] = 'F';
head->gpt_signature[2] = 'I';
head->gpt_signature[3] = ' ';
head->gpt_signature[4] = 'P';
head->gpt_signature[5] = 'A';
head->gpt_signature[6] = 'R';
head->gpt_signature[7] = 'T';
re = gpt_read_header(gpt);
CU_ASSERT(re == -1);
/* Set head->my_lba matched, lba_end usable_lba mismatch */
to_le32(&head->header_crc32, 0xB3CDB2D2);
to_le64(&head->my_lba, 0x1);
re = gpt_read_header(gpt);
CU_ASSERT(re == -1);
/* Set gpt->lba_end usable_lba matched, passing case */
to_le32(&head->header_crc32, 0x5531F2F0);
to_le64(&gpt->lba_start, 0x0);
to_le64(&gpt->lba_end, 0x2E935FFE);
to_le64(&head->first_usable_lba, 0xA);
to_le64(&head->last_usable_lba, 0xF4240);
re = gpt_read_header(gpt);
CU_ASSERT(re == 0);
free(gpt);
}
static void
test_read_partitions(void)
{
struct spdk_gpt *gpt;
struct spdk_gpt_header *head;
unsigned char a[SPDK_GPT_BUFFER_SIZE];
int re;
/* gpt_read_partitions(NULL) does not exist, NULL is filtered out in gpt_parse_mbr() */
gpt = calloc(1, sizeof(*gpt));
SPDK_CU_ASSERT_FATAL(gpt != NULL);
gpt->parse_phase = SPDK_GPT_PARSE_PHASE_PRIMARY;
gpt->sector_size = 512;
/* Set *gpt is "aaa..." */
memset(a, 'a', sizeof(a));
gpt->buf = &a[0];
gpt->buf_size = sizeof(a);
/* Set num_partition_entries exceeds Max value of entries GPT supported */
gpt->sector_size = 512;
head = (struct spdk_gpt_header *)(gpt->buf + GPT_PRIMARY_PARTITION_TABLE_LBA * gpt->sector_size);
gpt->header = head;
to_le32(&head->num_partition_entries, 0x100);
re = gpt_read_partitions(gpt);
CU_ASSERT(re == -1);
/* Set num_partition_entries within Max value, size_of_partition_entry mismatch */
to_le32(&head->header_crc32, 0x573857BE);
to_le32(&head->num_partition_entries, 0x40);
to_le32(&head->size_of_partition_entry, 0x0);
re = gpt_read_partitions(gpt);
CU_ASSERT(re == -1);
/* Set size_of_partition_entry matched, partition_entry_lba mismatch */
to_le32(&head->header_crc32, 0x5279B712);
to_le32(&head->size_of_partition_entry, 0x80);
to_le64(&head->partition_entry_lba, 0x64);
re = gpt_read_partitions(gpt);
CU_ASSERT(re == -1);
/* Set partition_entry_lba matched, partition_entry_array_crc32 mismatch */
to_le32(&head->header_crc32, 0xEC093B43);
to_le64(&head->partition_entry_lba, 0x20);
to_le32(&head->partition_entry_array_crc32, 0x0);
re = gpt_read_partitions(gpt);
CU_ASSERT(re == -1);
/* Set partition_entry_array_crc32 matched, passing case */
to_le32(&head->header_crc32, 0xE1A08822);
to_le32(&head->partition_entry_array_crc32, 0xEBEE44FB);
to_le32(&head->num_partition_entries, 0x80);
re = gpt_read_partitions(gpt);
CU_ASSERT(re == 0);
free(gpt);
}
static void
test_parse_mbr_and_primary(void)
{
struct spdk_gpt *gpt;
struct spdk_mbr *mbr;
struct spdk_gpt_header *head;
unsigned char a[SPDK_GPT_BUFFER_SIZE];
int re;
/* Set gpt is NULL */
re = gpt_parse_mbr(NULL);
CU_ASSERT(re == -1);
/* Set gpt->buf is NULL */
gpt = calloc(1, sizeof(*gpt));
SPDK_CU_ASSERT_FATAL(gpt != NULL);
gpt->parse_phase = SPDK_GPT_PARSE_PHASE_PRIMARY;
gpt->sector_size = 512;
re = gpt_parse_mbr(gpt);
CU_ASSERT(re == -1);
/* Set *gpt is "aaa...", check_mbr failed */
memset(a, 'a', sizeof(a));
gpt->buf = &a[0];
gpt->buf_size = sizeof(a);
re = gpt_parse_mbr(gpt);
CU_ASSERT(re == -1);
/* Set check_mbr passed */
mbr = (struct spdk_mbr *)gpt->buf;
mbr->mbr_signature = 0xAA55;
mbr->partitions[0].start_lba = 1;
mbr->partitions[0].os_type = 0xEE;
mbr->partitions[0].size_lba = 0xFFFFFFFF;
re = gpt_parse_mbr(gpt);
CU_ASSERT(re == 0);
/* Expect read_header failed */
re = gpt_parse_partition_table(gpt);
CU_ASSERT(re == -1);
/* Set read_header passed, read_partitions failed */
head = (struct spdk_gpt_header *)(gpt->buf + GPT_PRIMARY_PARTITION_TABLE_LBA * gpt->sector_size);
head->header_size = sizeof(*head);
head->gpt_signature[0] = 'E';
head->gpt_signature[1] = 'F';
head->gpt_signature[2] = 'I';
head->gpt_signature[3] = ' ';
head->gpt_signature[4] = 'P';
head->gpt_signature[5] = 'A';
head->gpt_signature[6] = 'R';
head->gpt_signature[7] = 'T';
to_le32(&head->header_crc32, 0x5531F2F0);
to_le64(&head->my_lba, 0x1);
to_le64(&gpt->lba_start, 0x0);
to_le64(&gpt->lba_end, 0x2E935FFE);
to_le64(&head->first_usable_lba, 0xA);
to_le64(&head->last_usable_lba, 0xF4240);
re = gpt_parse_partition_table(gpt);
CU_ASSERT(re == -1);
/* Set read_partitions passed, all passed */
to_le32(&head->size_of_partition_entry, 0x80);
to_le64(&head->partition_entry_lba, 0x20);
to_le32(&head->header_crc32, 0x845A09AA);
to_le32(&head->partition_entry_array_crc32, 0xEBEE44FB);
to_le32(&head->num_partition_entries, 0x80);
re = gpt_parse_partition_table(gpt);
CU_ASSERT(re == 0);
free(gpt);
}
static void
test_parse_secondary(void)
{
struct spdk_gpt *gpt;
struct spdk_gpt_header *head;
unsigned char a[SPDK_GPT_BUFFER_SIZE];
int re;
/* gpt_parse_partition_table(NULL) does not exist, NULL is filtered out in gpt_parse_mbr() */
gpt = calloc(1, sizeof(*gpt));
SPDK_CU_ASSERT_FATAL(gpt != NULL);
gpt->parse_phase = SPDK_GPT_PARSE_PHASE_SECONDARY;
gpt->sector_size = 512;
/* Set *gpt is "aaa...", read_header failed */
memset(a, 'a', sizeof(a));
gpt->buf = &a[0];
gpt->buf_size = sizeof(a);
re = gpt_parse_partition_table(gpt);
CU_ASSERT(re == -1);
/* Set read_header passed, read_partitions failed */
head = (struct spdk_gpt_header *)(gpt->buf + gpt->buf_size - gpt->sector_size);
head->header_size = sizeof(*head);
head->gpt_signature[0] = 'E';
head->gpt_signature[1] = 'F';
head->gpt_signature[2] = 'I';
head->gpt_signature[3] = ' ';
head->gpt_signature[4] = 'P';
head->gpt_signature[5] = 'A';
head->gpt_signature[6] = 'R';
head->gpt_signature[7] = 'T';
to_le32(&head->header_crc32, 0xAA68A167);
to_le64(&head->my_lba, 0x63FFFFF);
to_le64(&gpt->lba_start, 0x0);
to_le64(&gpt->lba_end, 0x63FFFFF);
to_le64(&gpt->total_sectors, 0x6400000);
to_le64(&head->first_usable_lba, 0xA);
to_le64(&head->last_usable_lba, 0x63FFFDE);
re = gpt_parse_partition_table(gpt);
CU_ASSERT(re == -1);
/* Set read_partitions passed, all passed */
to_le32(&head->size_of_partition_entry, 0x80);
to_le64(&head->partition_entry_lba, 0x63FFFDF);
to_le32(&head->header_crc32, 0x204129E8);
to_le32(&head->partition_entry_array_crc32, 0xEBEE44FB);
to_le32(&head->num_partition_entries, 0x80);
re = gpt_parse_partition_table(gpt);
CU_ASSERT(re == 0);
free(gpt);
}
int
main(int argc, char **argv)
{
CU_pSuite suite = NULL;
unsigned int num_failures;
CU_set_error_action(CUEA_ABORT);
CU_initialize_registry();
suite = CU_add_suite("gpt_parse", NULL, NULL);
CU_ADD_TEST(suite, test_parse_mbr_and_primary);
CU_ADD_TEST(suite, test_parse_secondary);
CU_ADD_TEST(suite, test_check_mbr);
CU_ADD_TEST(suite, test_read_header);
CU_ADD_TEST(suite, test_read_partitions);
CU_basic_set_mode(CU_BRM_VERBOSE);
CU_basic_run_tests();
num_failures = CU_get_number_of_failures();
CU_cleanup_registry();
return num_failures;
}