Fixing the bug of CGPT when primary entry table is invalid.

cgpt/cmd_boot.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 "cgpt.h"
 
 #include <errno.h>
 #include <fcntl.h>
 #include <getopt.h>
 #include <stdio.h>
@@ skipping 114 matching lines @@
             gpt_retval, GptError(gpt_retval));
       goto done;
     }
 
     if (partition > GetNumberOfEntries(&drive.gpt)) {
       Error("invalid partition number: %d\n", partition);
       goto done;
     }
 
     uint32_t index = partition - 1;
-    GptEntry *entry = GetEntry(&drive.gpt, PRIMARY, index);
+    GptEntry *entry = GetEntry(&drive.gpt, ANY_VALID, index);
     memcpy(&drive.pmbr.boot_guid, &entry->unique, sizeof(Guid));
   }
 
   if (bootfile) {
     int fd = open(bootfile, O_RDONLY);
     if (fd < 0) {
       Error("Can't read %s: %s\n", bootfile, strerror(errno));
       goto done;
     }
 
@@ skipping 12 matching lines @@
   printf("%s\n", buf);
 
   // Write it all out
   if (CGPT_OK == WritePMBR(&drive))
     retval = 0;
 
 done:
   (void) DriveClose(&drive, 1);
   return retval;
 }