complete 'cgpt show' and refactor for incoming commands.

src/platform/vboot_reference/cgptlib/tests/cgptlib_test.c

 /* Copyright (c) 2010 The Chromium OS Authors. All rights reserved.
  * Use of this source code is governed by a BSD-style license that can be
  * found in the LICENSE file.
  */
 
 #include "cgptlib_test.h"
 #include <string.h>
 #include "cgptlib.h"
 #include "cgptlib_internal.h"
 #include "crc32.h"
@@ skipping 96 matching lines @@
  * this file. */
 void BuildTestGptData(GptData *gpt) {
   GptHeader *header, *header2;
   GptEntry *entries, *entries2;
   Guid chromeos_kernel = GPT_ENT_TYPE_CHROMEOS_KERNEL;
   Guid chromeos_rootfs = GPT_ENT_TYPE_CHROMEOS_ROOTFS;
 
   gpt->sector_bytes = DEFAULT_SECTOR_SIZE;
   gpt->drive_sectors = DEFAULT_DRIVE_SECTORS;
   gpt->current_kernel = CGPT_KERNEL_ENTRY_NOT_FOUND;
+  gpt->valid_headers = MASK_BOTH;
+  gpt->valid_entries = MASK_BOTH;
 
   /* build primary */
   header = (GptHeader*)gpt->primary_header;
   entries = (GptEntry*)gpt->primary_entries;
   Memcpy(header->signature, GPT_HEADER_SIGNATURE, sizeof(GPT_HEADER_SIGNATURE));
   header->revision = GPT_HEADER_REVISION;
   header->size = sizeof(GptHeader) - sizeof(header->padding);
   header->reserved = 0;
   header->my_lba = 1;
   header->first_usable_lba = 34;
@@ skipping 466 matching lines @@
   /* Modify the last byte of secondary header, and expect the CRC is wrong. */
   BuildTestGptData(gpt);
   gpt->secondary_header[secondary_header->size-1] ^= 0x5a;
   EXPECT(MASK_PRIMARY == CheckHeaderCrc(gpt));
 
   /* Modify out of CRC range, expect CRC is still right. */
   BuildTestGptData(gpt);
   gpt->primary_header[primary_header->size] ^= 0x87;
   EXPECT(MASK_BOTH == CheckHeaderCrc(gpt));
 
+  /* Very long header (actually invalid header). Expect will be ignored. */
+  primary_header->size = 0x12345678;
+  secondary_header->size = 0x87654321;
+  gpt->valid_headers = MASK_NONE;
+  EXPECT(MASK_NONE == CheckHeaderCrc(gpt));
+
   return TEST_OK;
 }
 
 /* Tests if PartitionEntryArrayCRC32 is checked.
  * PartitionEntryArrayCRC32 must be calculated over SizeOfPartitionEntry *
  * NumberOfPartitionEntries bytes.
  */
 int EntriesCrcTest() {
   GptData *gpt;
 
@@ skipping 99 matching lines @@
   secondary_header->disk_uuid.u.raw[0] ^= 0x56;
   EXPECT(GPT_MODIFIED_HEADER2 == RepairHeader(gpt, MASK_BOTH));
   EXPECT(0 == Memcmp(&primary_header->disk_uuid,
                      &secondary_header->disk_uuid, sizeof(Guid)));
 
   /* Consider header repairing in GptInit(). */
   BuildTestGptData(gpt);
   ++secondary_header->first_usable_lba;
   RefreshCrc32(gpt);
   EXPECT(GPT_SUCCESS == GptInit(gpt));
-  EXPECT(GPT_MODIFIED_HEADER2 == gpt->modified);
+  EXPECT((gpt->modified & (GPT_MODIFIED_HEADER1 | GPT_MODIFIED_HEADER2)) ==
+         GPT_MODIFIED_HEADER2);
   EXPECT(primary_header->first_usable_lba ==
          secondary_header->first_usable_lba);
 
   return TEST_OK;
 }
 
 /* Tests if partition geometry is checked.
  * All active (non-zero PartitionTypeGUID) partition entries should have:
  *   entry.StartingLBA >= header.FirstUsableLBA
  *   entry.EndingLBA <= header.LastUsableLBA
@@ skipping 152 matching lines @@
   EXPECT(0 == Memcmp(primary_entries, secondary_entries, TOTAL_ENTRIES_SIZE));
   /* We expect the modified header/entries can pass GptInit(). */
   EXPECT(GPT_SUCCESS == GptInit(gpt));
   EXPECT(0 == gpt->modified);
 
   /* Make primary header invalid (the entries is not damaged actually). */
   BuildTestGptData(gpt);
   primary_header->entries_crc32 ^= 0x73;
   EXPECT(GPT_SUCCESS == GptInit(gpt));
   /* After header is repaired, the entries are valid actually. */
-  EXPECT((GPT_MODIFIED_HEADER1) == gpt->modified);
+  EXPECT((gpt->modified & (GPT_MODIFIED_HEADER1 | GPT_MODIFIED_HEADER2)) ==
+         GPT_MODIFIED_HEADER1);
   /* We expect the modified header/entries can pass GptInit(). */
   EXPECT(GPT_SUCCESS == GptInit(gpt));
   EXPECT(0 == gpt->modified);
 
   return TEST_OK;
 }
 
 /* Invalidate all kernel entries and expect GptNextKernelEntry() cannot find
  * any usable kernel entry.
  */
@@ skipping 265 matching lines @@
     printf(COL_RED "The following %d test cases are failed:\n" COL_STOP,
            error_count);
     for (i = 0; i < sizeof(test_cases)/sizeof(test_cases[0]); ++i) {
       if (test_cases[i].retval)
         printf("  %s()\n", test_cases[i].name);
     }
   }
 
   return (error_count) ? 1 : 0;
 }