1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
|
/* 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.
*
* Update GPT attribute bits.
*/
#include <getopt.h>
#include <stdio.h>
#include <stdlib.h>
#include "cgpt.h"
#include "cgptlib_internal.h"
#include "utility.h"
static struct number_range
range_1_0 = {1, 0},
range_15_0 = {15, 0};
/* Integers to store parsed argument. */
static int help, partition, bad, successful, tries, priority;
/* The structure for getopt_long(). When you add/delete any line, please refine
* attribute_comments[] and third parameter of getopt_long() too. */
static struct option attribute_options[] = {
{.name = "help", .has_arg = no_argument, .flag = 0, .val = 'h'},
{.name = "partition", .has_arg = required_argument, .flag = 0, .val = 'i'},
{.name = "bad", .has_arg = required_argument, .flag = 0, .val = 'b'},
{.name = "successful", .has_arg = required_argument, .flag = 0, .val = 's'},
{.name = "tries", .has_arg = required_argument, .flag = 0, .val = 't'},
{.name = "priority", .has_arg = required_argument, .flag = 0, .val = 'p'},
};
/* Extra information than struct option, please update this structure if you
* add/remove any line in attribute_options[]. */
static struct option_details attribute_options_details[] = {
/* help */
{ .comment = "print this help",
.validator = AssignTrue,
.valid_range = 0,
.parsed = &help},
/* partition */
{ .comment = "partition number "
"(defualt: first ChromeOS kernel)",
.validator = InNumberRange,
.valid_range = &range_15_0,
.parsed = &partition},
/* bad */
{ .comment = "mark partition bad",
.validator = InNumberRange,
.valid_range = &range_1_0,
.parsed = &bad},
/* successful */
{ .comment = "mark partition successful",
.validator = InNumberRange,
.valid_range = &range_1_0,
.parsed = &successful},
/* tries */
{ .comment = "tries",
.validator = InNumberRange,
.valid_range = &range_15_0,
.parsed = &tries},
/* priority */
{ .comment = "priority to boot",
.validator = InNumberRange,
.valid_range = &range_15_0,
.parsed = &priority},
};
void AttributeHelp() {
printf("\nUsage: %s attribute [OPTIONS] device_name\n\n", progname);
ShowOptions(attribute_options, attribute_options_details,
ARRAY_COUNT(attribute_options));
printf("\n* Attribute command only applies on ChromeOS kernel entry.\n\n");
}
/* Parses all options (and validates them), then opens the drive and sets
* corresponding bits in GPT entry. */
int CgptAttribute(int argc, char *argv[]) {
struct drive drive;
/* I know this is NOT the perfect place to put code to make options[] and
* details[] are synced. But this is the best place we have right now since C
* preprocessor doesn't know sizeof() for #if directive. */
assert(ARRAY_COUNT(attribute_options) ==
ARRAY_COUNT(attribute_options_details));
help = partition = bad = successful = tries = priority = NOT_INITED;
if (CGPT_OK != HandleOptions(argc, argv,
"hi:b:s:t:p:",
ARRAY_COUNT(attribute_options),
attribute_options,
attribute_options_details))
return CGPT_FAILED;
if (help != NOT_INITED) {
AttributeHelp();
return CGPT_FAILED;
}
if (CGPT_OK != OpenDriveInLastArgument(argc, argv, &drive))
return CGPT_FAILED;
debug("[OPTION] i:%d b:%d s:%d t:%d p:%d\n", partition, bad, successful, tries, priority); /* FIXME */
/* partition is not specified, search for the first Chromeos kernel. */
if (partition == NOT_INITED) {
int i;
for (i = 0; i < GetNumberOfEntries(&drive.gpt); ++i) {
static Guid chromeos_kernel = GPT_ENT_TYPE_CHROMEOS_KERNEL;
GptEntry *entry;
entry = GetEntry(&drive.gpt, PRIMARY, i);
if (!Memcmp(&chromeos_kernel, &entry->type, sizeof(Guid))) {
partition = i;
break;
}
}
if (partition == NOT_INITED) {
printf("[ERROR] No ChromeOS kernel is found. "
"Please use --partition to specify.\n");
return CGPT_FAILED;
} else {
debug("No --partition is specified. "
"Found the first ChromeOS kernel at index [%d].\n",
partition);
}
}
if (bad != NOT_INITED)
SetBad(&drive.gpt, PRIMARY, partition, bad);
if (successful != NOT_INITED)
SetSuccessful(&drive.gpt, PRIMARY, partition, successful);
if (tries != NOT_INITED)
SetTries(&drive.gpt, PRIMARY, partition, tries);
if (priority != NOT_INITED)
SetPriority(&drive.gpt, PRIMARY, partition, priority);
/* Claims primary is good, then secondary will be overwritten. */
RepairEntries(&drive.gpt, MASK_PRIMARY);
RepairHeader(&drive.gpt, MASK_PRIMARY);
/* Forces headers and entries are modified so that CRC32 will be re-calculated
* and headers and entries will be updated to drive. */
drive.gpt.modified |= (GPT_MODIFIED_HEADER1 | GPT_MODIFIED_ENTRIES1 |
GPT_MODIFIED_HEADER2 | GPT_MODIFIED_ENTRIES2);
UpdateCrc(&drive.gpt);
DriveClose(&drive);
return CGPT_OK;
}
|
