1 // Copyright (c) 2012 The Chromium OS Authors. All rights reserved.
2 // Use of this source code is governed by a BSD-style license that can be
3 // found in the LICENSE file.
4
5 #include <getopt.h>
6 #include <stdio.h>
7 #include <stdlib.h>
8 #include <string.h>
9 #include <uuid/uuid.h>
10
11 #include "cgpt.h"
12 #include "vboot_host.h"
13
14 extern const char* progname;
15
Usage(void)16 static void Usage(void)
17 {
18 printf("\nUsage: %s prioritize [OPTIONS] DRIVE\n\n"
19 "Reorder the priority of all active ChromeOS Kernel partitions.\n\n"
20 "Options:\n"
21 " -D NUM Size (in bytes) of the disk where partitions reside\n"
22 " default 0, meaning partitions and GPT structs are\n"
23 " both on DRIVE\n"
24 " -P NUM Highest priority to use in the new ordering. The\n"
25 " other partitions will be ranked in decreasing\n"
26 " priority while preserving their original order.\n"
27 " If necessary the lowest ranks will be coalesced.\n"
28 " No active kernels will be lowered to priority 0.\n"
29 " -i NUM Specify the partition to make the highest in the new\n"
30 " order.\n"
31 " -f Friends of the given partition (those with the same\n"
32 " starting priority) are also updated to the new\n"
33 " highest priority.\n"
34 "\n"
35 "With no options this will set the lowest active kernel to\n"
36 "priority 1 while maintaining the original order.\n"
37 "\n", progname);
38 }
39
cmd_prioritize(int argc,char * argv[])40 int cmd_prioritize(int argc, char *argv[]) {
41 CgptPrioritizeParams params;
42 memset(¶ms, 0, sizeof(params));
43
44 int c;
45 int errorcnt = 0;
46 char *e = 0;
47
48 opterr = 0; // quiet, you
49 while ((c=getopt(argc, argv, ":hi:fP:D:")) != -1)
50 {
51 switch (c)
52 {
53 case 'D':
54 params.drive_size = strtoull(optarg, &e, 0);
55 if (!*optarg || (e && *e))
56 {
57 Error("invalid argument to -%c: \"%s\"\n", c, optarg);
58 errorcnt++;
59 }
60 break;
61 case 'i':
62 params.set_partition = (uint32_t)strtoul(optarg, &e, 0);
63 if (!*optarg || (e && *e))
64 {
65 Error("invalid argument to -%c: \"%s\"\n", c, optarg);
66 errorcnt++;
67 }
68 break;
69 case 'f':
70 params.set_friends = 1;
71 break;
72 case 'P':
73 params.max_priority = (int)strtol(optarg, &e, 0);
74 if (!*optarg || (e && *e))
75 {
76 Error("invalid argument to -%c: \"%s\"\n", c, optarg);
77 errorcnt++;
78 }
79 if (params.max_priority < 1 || params.max_priority > 15) {
80 Error("value for -%c must be between 1 and 15\n", c);
81 errorcnt++;
82 }
83 break;
84
85 case 'h':
86 Usage();
87 return CGPT_OK;
88 case '?':
89 Error("unrecognized option: -%c\n", optopt);
90 errorcnt++;
91 break;
92 case ':':
93 Error("missing argument to -%c\n", optopt);
94 errorcnt++;
95 break;
96 default:
97 errorcnt++;
98 break;
99 }
100 }
101 if (errorcnt)
102 {
103 Usage();
104 return CGPT_FAILED;
105 }
106
107 if (params.set_friends && !params.set_partition) {
108 Error("the -f option is only useful with the -i option\n");
109 Usage();
110 return CGPT_FAILED;
111 }
112
113 if (optind >= argc) {
114 Error("missing drive argument\n");
115 return CGPT_FAILED;
116 }
117
118 params.drive_name = argv[optind];
119
120 return CgptPrioritize(¶ms);
121 }
122