The right implementation of CGPT label conversion between UTF8 and UTF16.

cgpt/cmd_find.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 <getopt.h>
 #include <stdio.h>
 #include <stdlib.h>
 #include <string.h>
@@ skipping 163 matching lines @@
     entry = GetEntry(&drive.gpt, ANY_VALID, i);
 
     if (IsZero(&entry->type))
       continue;
 
     int found = 0;
     if ((set_unique && !memcmp(&unique_guid, &entry->unique, sizeof(Guid))) ||
         (set_type && !memcmp(&type_guid, &entry->type, sizeof(Guid)))) {
       found = 1;
     } else if (set_label) {
-      UTF16ToUTF8(entry->name, sizeof(entry->name) / sizeof(entry->name[0]),
-                  (uint8_t *)partlabel, sizeof(partlabel));
+      if (CGPT_OK != UTF16ToUTF8(entry->name,
+                                 sizeof(entry->name) / sizeof(entry->name[0]),
+                                 (uint8_t *)partlabel, sizeof(partlabel))) {
+        Error("The label cannot be converted to UTF16, so abort.\n");
+        return 0;
+      }
       if (!strncmp(label, partlabel, sizeof(partlabel))) {
         found = 1;
       }
     }
     if (found && match_content(&drive, entry)) {
       hits++;
       retval++;
       showmatch(filename, i+1, entry);
       if (!match_partnum) {
         match_partnum = i+1;
@@ skipping 181 matching lines @@
   if (oneonly && hits != 1) {
     return CGPT_FAILED;
   }
 
   if (match_partnum) {
     return CGPT_OK;
   }
 
   return CGPT_FAILED;
 }