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
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
|
/*
* gptsync/gptsync.h
* Common header for gptsync and showpart
*
* Copyright (c) 2006 Christoph Pfisterer
* All rights reserved.
*
* Redistribution and use in source and binary forms, with or without
* modification, are permitted provided that the following conditions are
* met:
*
* * Redistributions of source code must retain the above copyright
* notice, this list of conditions and the following disclaimer.
*
* * Redistributions in binary form must reproduce the above copyright
* notice, this list of conditions and the following disclaimer in the
* documentation and/or other materials provided with the
* distribution.
*
* * Neither the name of Christoph Pfisterer nor the names of the
* contributors may be used to endorse or promote products derived
* from this software without specific prior written permission.
*
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
* "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
* LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
* A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
* OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
* SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
* LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
* DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
* THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
* (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
* OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
*/
//
// config
//
#if defined(EFI32) || defined(EFIX64)
#define CONFIG_EFI
#endif
//
// platform-dependent types
//
#ifdef CONFIG_EFI
#include <efi.h>
#include <efilib.h>
#define copy_guid(destguid, srcguid) (CopyMem(destguid, srcguid, 16))
#define guids_are_equal(guid1, guid2) (CompareMem(guid1, guid2, 16) == 0)
typedef CHAR16 CHARN;
#define STR(x) L##x
#endif
#ifndef CONFIG_EFI
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <stdint.h>
#include <unistd.h>
#include <errno.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <sys/time.h>
#include <fcntl.h>
typedef int INTN;
typedef unsigned int UINTN;
typedef uint8_t UINT8;
typedef uint16_t UINT16;
typedef uint32_t UINT32;
typedef uint64_t UINT64;
typedef void VOID;
typedef int BOOLEAN;
#ifndef FALSE
#define FALSE (0)
#endif
#ifndef TRUE
#define TRUE (1)
#endif
typedef unsigned short CHAR16;
typedef char CHARN;
#define STR(x) x
void Print(wchar_t *format, ...);
// FUTURE: use STR(), #define Print printf
#define CopyMem memcpy
#define SetMem memset
#define CompareMem memcmp
#define copy_guid(destguid, srcguid) (memcpy(destguid, srcguid, 16))
#define guids_are_equal(guid1, guid2) (memcmp(guid1, guid2, 16) == 0)
#endif
//
// platform-independent types
//
typedef struct {
UINT8 flags;
UINT8 start_chs[3];
UINT8 type;
UINT8 end_chs[3];
UINT32 start_lba;
UINT32 size;
} MBR_PARTITION_INFO;
typedef struct {
UINT8 type;
CHARN *name;
} MBR_PARTTYPE;
typedef struct {
UINT64 signature;
UINT32 spec_revision;
UINT32 header_size;
UINT32 header_crc32;
UINT32 reserved;
UINT64 header_lba;
UINT64 alternate_header_lba;
UINT64 first_usable_lba;
UINT64 last_usable_lba;
UINT8 disk_guid[16];
UINT64 entry_lba;
UINT32 entry_count;
UINT32 entry_size;
UINT32 entry_crc32;
} GPT_HEADER;
typedef struct {
UINT8 type_guid[16];
UINT8 partition_guid[16];
UINT64 start_lba;
UINT64 end_lba;
UINT64 attributes;
CHAR16 name[36];
} GPT_ENTRY;
#define GPT_KIND_SYSTEM (0)
#define GPT_KIND_DATA (1)
#define GPT_KIND_BASIC_DATA (2)
#define GPT_KIND_FATAL (3)
typedef struct {
UINT8 guid[16];
UINT8 mbr_type;
CHARN *name;
UINTN kind;
} GPT_PARTTYPE;
typedef struct {
UINTN index;
UINT64 start_lba;
UINT64 end_lba;
UINTN mbr_type;
UINT8 gpt_type[16];
GPT_PARTTYPE *gpt_parttype;
BOOLEAN active;
} PARTITION_INFO;
//
// functions provided by the OS-specific module
//
UINTN read_sector(UINT64 lba, UINT8 *buffer);
UINTN write_sector(UINT64 lba, UINT8 *buffer);
UINTN input_boolean(CHARN *prompt, BOOLEAN *bool_out);
//
// vars and functions provided by the common lib module
//
extern UINT8 empty_guid[16];
extern PARTITION_INFO mbr_parts[4];
extern UINTN mbr_part_count;
extern PARTITION_INFO gpt_parts[128];
extern UINTN gpt_part_count;
extern PARTITION_INFO new_mbr_parts[4];
extern UINTN new_mbr_part_count;
extern UINT8 sector[512];
extern MBR_PARTTYPE mbr_types[];
extern GPT_PARTTYPE gpt_types[];
extern GPT_PARTTYPE gpt_dummy_type;
CHARN * mbr_parttype_name(UINT8 type);
UINTN read_mbr(VOID);
GPT_PARTTYPE * gpt_parttype(UINT8 *type_guid);
UINTN read_gpt(VOID);
UINTN detect_mbrtype_fs(UINT64 partlba, UINTN *parttype, CHARN **fsname);
//
// actual platform-independent programs
//
UINTN gptsync(VOID);
UINTN showpart(VOID);
/* EOF */
|
