17c478bd9Sstevel@tonic-gate /*
27c478bd9Sstevel@tonic-gate * CDDL HEADER START
37c478bd9Sstevel@tonic-gate *
47c478bd9Sstevel@tonic-gate * The contents of this file are subject to the terms of the
5d33344bbSsy25831 * Common Development and Distribution License (the "License").
6d33344bbSsy25831 * You may not use this file except in compliance with the License.
77c478bd9Sstevel@tonic-gate *
87c478bd9Sstevel@tonic-gate * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
97c478bd9Sstevel@tonic-gate * or http://www.opensolaris.org/os/licensing.
107c478bd9Sstevel@tonic-gate * See the License for the specific language governing permissions
117c478bd9Sstevel@tonic-gate * and limitations under the License.
127c478bd9Sstevel@tonic-gate *
137c478bd9Sstevel@tonic-gate * When distributing Covered Code, include this CDDL HEADER in each
147c478bd9Sstevel@tonic-gate * file and include the License file at usr/src/OPENSOLARIS.LICENSE.
157c478bd9Sstevel@tonic-gate * If applicable, add the following below this CDDL HEADER, with the
167c478bd9Sstevel@tonic-gate * fields enclosed by brackets "[]" replaced with your own identifying
177c478bd9Sstevel@tonic-gate * information: Portions Copyright [yyyy] [name of copyright owner]
187c478bd9Sstevel@tonic-gate *
197c478bd9Sstevel@tonic-gate * CDDL HEADER END
207c478bd9Sstevel@tonic-gate */
217c478bd9Sstevel@tonic-gate /*
22af28f636SEnrico Perla - Sun Microsystems * Copyright (c) 2005, 2010, Oracle and/or its affiliates. All rights reserved.
2333f5ff17SMilan Jurik * Copyright 2012 Milan Jurik. All rights reserved.
241a902ef8SHans Rosenfeld * Copyright 2015 Nexenta Systems, Inc. All rights reserved.
257c478bd9Sstevel@tonic-gate */
267c478bd9Sstevel@tonic-gate
277c478bd9Sstevel@tonic-gate #include <stdio.h>
287c478bd9Sstevel@tonic-gate #include <stdlib.h>
297c478bd9Sstevel@tonic-gate #include <libgen.h>
307c478bd9Sstevel@tonic-gate #include <malloc.h>
317c478bd9Sstevel@tonic-gate #include <string.h>
327c478bd9Sstevel@tonic-gate #include <fcntl.h>
337c478bd9Sstevel@tonic-gate #include <unistd.h>
347c478bd9Sstevel@tonic-gate #include <strings.h>
35af28f636SEnrico Perla - Sun Microsystems #include <libintl.h>
36af28f636SEnrico Perla - Sun Microsystems #include <locale.h>
37af28f636SEnrico Perla - Sun Microsystems #include <errno.h>
38af28f636SEnrico Perla - Sun Microsystems #include <libfdisk.h>
39af28f636SEnrico Perla - Sun Microsystems #include <stdarg.h>
40af28f636SEnrico Perla - Sun Microsystems #include <assert.h>
41af28f636SEnrico Perla - Sun Microsystems
427c478bd9Sstevel@tonic-gate #include <sys/mount.h>
437c478bd9Sstevel@tonic-gate #include <sys/mnttab.h>
447c478bd9Sstevel@tonic-gate #include <sys/dktp/fdisk.h>
45d33344bbSsy25831 #include <sys/dkio.h>
46d33344bbSsy25831 #include <sys/vtoc.h>
47af28f636SEnrico Perla - Sun Microsystems #include <sys/types.h>
48af28f636SEnrico Perla - Sun Microsystems #include <sys/stat.h>
49af28f636SEnrico Perla - Sun Microsystems #include <sys/multiboot.h>
50af28f636SEnrico Perla - Sun Microsystems #include <sys/sysmacros.h>
511a902ef8SHans Rosenfeld #include <sys/efi_partition.h>
521a902ef8SHans Rosenfeld
531a902ef8SHans Rosenfeld #include <libnvpair.h>
541a902ef8SHans Rosenfeld #include <libfstyp.h>
557c478bd9Sstevel@tonic-gate
567c478bd9Sstevel@tonic-gate #include "message.h"
57af28f636SEnrico Perla - Sun Microsystems #include "installgrub.h"
58af28f636SEnrico Perla - Sun Microsystems #include "./../common/bblk_einfo.h"
59af28f636SEnrico Perla - Sun Microsystems #include "./../common/boot_utils.h"
60af28f636SEnrico Perla - Sun Microsystems #include "./../common/mboot_extra.h"
617c478bd9Sstevel@tonic-gate
627c478bd9Sstevel@tonic-gate #ifndef TEXT_DOMAIN
637c478bd9Sstevel@tonic-gate #define TEXT_DOMAIN "SUNW_OST_OSCMD"
647c478bd9Sstevel@tonic-gate #endif
657c478bd9Sstevel@tonic-gate
66af28f636SEnrico Perla - Sun Microsystems /*
67af28f636SEnrico Perla - Sun Microsystems * Variables to track installgrub desired mode of operation.
68af28f636SEnrico Perla - Sun Microsystems * 'nowrite' and 'boot_debug' come from boot_common.h.
69af28f636SEnrico Perla - Sun Microsystems */
70af28f636SEnrico Perla - Sun Microsystems static boolean_t write_mbr = B_FALSE;
71af28f636SEnrico Perla - Sun Microsystems static boolean_t force_mbr = B_FALSE;
72af28f636SEnrico Perla - Sun Microsystems static boolean_t force_update = B_FALSE;
73af28f636SEnrico Perla - Sun Microsystems static boolean_t do_getinfo = B_FALSE;
74af28f636SEnrico Perla - Sun Microsystems static boolean_t do_version = B_FALSE;
75af28f636SEnrico Perla - Sun Microsystems static boolean_t do_mirror_bblk = B_FALSE;
76af28f636SEnrico Perla - Sun Microsystems static boolean_t strip = B_FALSE;
77af28f636SEnrico Perla - Sun Microsystems static boolean_t verbose_dump = B_FALSE;
787c478bd9Sstevel@tonic-gate
79af28f636SEnrico Perla - Sun Microsystems /* Installing the bootblock is the default operation. */
80af28f636SEnrico Perla - Sun Microsystems static boolean_t do_install = B_TRUE;
817c478bd9Sstevel@tonic-gate
82af28f636SEnrico Perla - Sun Microsystems /* Versioning string, if present. */
83af28f636SEnrico Perla - Sun Microsystems static char *update_str;
847c478bd9Sstevel@tonic-gate
85af28f636SEnrico Perla - Sun Microsystems /*
86af28f636SEnrico Perla - Sun Microsystems * Temporary buffer to store the first 32K of data looking for a multiboot
87af28f636SEnrico Perla - Sun Microsystems * signature.
88af28f636SEnrico Perla - Sun Microsystems */
89af28f636SEnrico Perla - Sun Microsystems char mboot_scan[MBOOT_SCAN_SIZE];
907ce76caaSEnrico Perla - Sun Microsystems
91af28f636SEnrico Perla - Sun Microsystems /* Function prototypes. */
92af28f636SEnrico Perla - Sun Microsystems static void check_options(char *);
93af28f636SEnrico Perla - Sun Microsystems static int handle_install(char *, char **);
94af28f636SEnrico Perla - Sun Microsystems static int handle_mirror(char *, char **);
95af28f636SEnrico Perla - Sun Microsystems static int handle_getinfo(char *, char **);
96af28f636SEnrico Perla - Sun Microsystems static int commit_to_disk(ig_data_t *, char *);
97af28f636SEnrico Perla - Sun Microsystems static int init_device(ig_device_t *, char *path);
98af28f636SEnrico Perla - Sun Microsystems static void cleanup_device(ig_device_t *);
99af28f636SEnrico Perla - Sun Microsystems static void cleanup_stage2(ig_stage2_t *);
100af28f636SEnrico Perla - Sun Microsystems static int get_start_sector(ig_device_t *);
101af28f636SEnrico Perla - Sun Microsystems static int get_disk_fd(ig_device_t *device);
102af28f636SEnrico Perla - Sun Microsystems static int get_raw_partition_fd(ig_device_t *);
103af28f636SEnrico Perla - Sun Microsystems static char *get_raw_partition_path(ig_device_t *);
104af28f636SEnrico Perla - Sun Microsystems static boolean_t gather_stage2_from_dev(ig_data_t *);
105af28f636SEnrico Perla - Sun Microsystems static int propagate_bootblock(ig_data_t *, ig_data_t *, char *);
106af28f636SEnrico Perla - Sun Microsystems static int find_x86_bootpar(struct mboot *, int *, uint32_t *);
107af28f636SEnrico Perla - Sun Microsystems static int copy_stage2_to_pcfs(ig_data_t *);
108af28f636SEnrico Perla - Sun Microsystems static int write_stage2(ig_data_t *);
109af28f636SEnrico Perla - Sun Microsystems static int write_stage1(ig_data_t *);
1107c478bd9Sstevel@tonic-gate static void usage(char *);
111af28f636SEnrico Perla - Sun Microsystems static int read_stage1_from_file(char *, ig_data_t *);
112af28f636SEnrico Perla - Sun Microsystems static int read_stage2_from_file(char *, ig_data_t *);
113af28f636SEnrico Perla - Sun Microsystems static int read_stage1_from_disk(int, char *);
1141a902ef8SHans Rosenfeld static int read_stage2_from_disk(int, ig_stage2_t *, int);
115af28f636SEnrico Perla - Sun Microsystems static int prepare_stage1(ig_data_t *);
116af28f636SEnrico Perla - Sun Microsystems static int prepare_stage2(ig_data_t *, char *);
117af28f636SEnrico Perla - Sun Microsystems static void prepare_fake_multiboot(ig_stage2_t *);
118af28f636SEnrico Perla - Sun Microsystems static void add_stage2_einfo(ig_stage2_t *, char *updt_str);
119af28f636SEnrico Perla - Sun Microsystems static boolean_t is_update_necessary(ig_data_t *, char *);
1207c478bd9Sstevel@tonic-gate
121342440ecSPrasad Singamsetty extern int read_stage2_blocklist(int, unsigned int *);
1227c478bd9Sstevel@tonic-gate
1237c478bd9Sstevel@tonic-gate int
main(int argc,char * argv[])1247c478bd9Sstevel@tonic-gate main(int argc, char *argv[])
1257c478bd9Sstevel@tonic-gate {
126af28f636SEnrico Perla - Sun Microsystems int opt;
127af28f636SEnrico Perla - Sun Microsystems int params = 3;
128af28f636SEnrico Perla - Sun Microsystems int ret;
129af28f636SEnrico Perla - Sun Microsystems char **handle_args;
130af28f636SEnrico Perla - Sun Microsystems char *progname;
1317c478bd9Sstevel@tonic-gate
1327c478bd9Sstevel@tonic-gate (void) setlocale(LC_ALL, "");
1337c478bd9Sstevel@tonic-gate (void) textdomain(TEXT_DOMAIN);
1347c478bd9Sstevel@tonic-gate
135af28f636SEnrico Perla - Sun Microsystems /*
136af28f636SEnrico Perla - Sun Microsystems * retro-compatibility: installing the bootblock is the default
137af28f636SEnrico Perla - Sun Microsystems * and there is no switch for it.
138af28f636SEnrico Perla - Sun Microsystems */
139af28f636SEnrico Perla - Sun Microsystems do_install = B_TRUE;
140af28f636SEnrico Perla - Sun Microsystems
141af28f636SEnrico Perla - Sun Microsystems while ((opt = getopt(argc, argv, "dVMFfmneiu:")) != EOF) {
1427c478bd9Sstevel@tonic-gate switch (opt) {
1437c478bd9Sstevel@tonic-gate case 'm':
144af28f636SEnrico Perla - Sun Microsystems write_mbr = B_TRUE;
1457c478bd9Sstevel@tonic-gate break;
1467c478bd9Sstevel@tonic-gate case 'n':
147af28f636SEnrico Perla - Sun Microsystems nowrite = B_TRUE;
1487c478bd9Sstevel@tonic-gate break;
1497c478bd9Sstevel@tonic-gate case 'f':
150af28f636SEnrico Perla - Sun Microsystems force_mbr = B_TRUE;
1517c478bd9Sstevel@tonic-gate break;
1527ce76caaSEnrico Perla - Sun Microsystems case 'i':
153af28f636SEnrico Perla - Sun Microsystems do_getinfo = B_TRUE;
154af28f636SEnrico Perla - Sun Microsystems do_install = B_FALSE;
1557ce76caaSEnrico Perla - Sun Microsystems params = 1;
1567ce76caaSEnrico Perla - Sun Microsystems break;
157af28f636SEnrico Perla - Sun Microsystems case 'V':
158af28f636SEnrico Perla - Sun Microsystems verbose_dump = B_TRUE;
1597ce76caaSEnrico Perla - Sun Microsystems break;
160af28f636SEnrico Perla - Sun Microsystems case 'd':
161af28f636SEnrico Perla - Sun Microsystems boot_debug = B_TRUE;
162af28f636SEnrico Perla - Sun Microsystems break;
163af28f636SEnrico Perla - Sun Microsystems case 'F':
164af28f636SEnrico Perla - Sun Microsystems force_update = B_TRUE;
165af28f636SEnrico Perla - Sun Microsystems break;
166af28f636SEnrico Perla - Sun Microsystems case 'e':
167af28f636SEnrico Perla - Sun Microsystems strip = B_TRUE;
168af28f636SEnrico Perla - Sun Microsystems break;
169af28f636SEnrico Perla - Sun Microsystems case 'M':
170af28f636SEnrico Perla - Sun Microsystems do_mirror_bblk = B_TRUE;
171af28f636SEnrico Perla - Sun Microsystems do_install = B_FALSE;
172af28f636SEnrico Perla - Sun Microsystems params = 2;
173af28f636SEnrico Perla - Sun Microsystems break;
174af28f636SEnrico Perla - Sun Microsystems case 'u':
175af28f636SEnrico Perla - Sun Microsystems do_version = B_TRUE;
176af28f636SEnrico Perla - Sun Microsystems
177af28f636SEnrico Perla - Sun Microsystems update_str = malloc(strlen(optarg) + 1);
178af28f636SEnrico Perla - Sun Microsystems if (update_str == NULL) {
179af28f636SEnrico Perla - Sun Microsystems (void) fprintf(stderr, gettext("Unable to "
180af28f636SEnrico Perla - Sun Microsystems "allocate memory\n"));
181af28f636SEnrico Perla - Sun Microsystems exit(BC_ERROR);
182af28f636SEnrico Perla - Sun Microsystems }
183af28f636SEnrico Perla - Sun Microsystems (void) strlcpy(update_str, optarg, strlen(optarg) + 1);
1847ce76caaSEnrico Perla - Sun Microsystems break;
1857c478bd9Sstevel@tonic-gate default:
1867c478bd9Sstevel@tonic-gate /* fall through to process non-optional args */
1877c478bd9Sstevel@tonic-gate break;
1887c478bd9Sstevel@tonic-gate }
1897c478bd9Sstevel@tonic-gate }
1907c478bd9Sstevel@tonic-gate
1917c478bd9Sstevel@tonic-gate /* check arguments */
1927ce76caaSEnrico Perla - Sun Microsystems if (argc != optind + params) {
1937c478bd9Sstevel@tonic-gate usage(argv[0]);
194af28f636SEnrico Perla - Sun Microsystems exit(BC_ERROR);
1957c478bd9Sstevel@tonic-gate }
1967c478bd9Sstevel@tonic-gate
197af28f636SEnrico Perla - Sun Microsystems /*
198af28f636SEnrico Perla - Sun Microsystems * clean up options (and bail out if an unrecoverable combination is
199af28f636SEnrico Perla - Sun Microsystems * requested.
200af28f636SEnrico Perla - Sun Microsystems */
201af28f636SEnrico Perla - Sun Microsystems progname = argv[0];
202af28f636SEnrico Perla - Sun Microsystems check_options(progname);
203af28f636SEnrico Perla - Sun Microsystems handle_args = argv + optind;
204af28f636SEnrico Perla - Sun Microsystems
205af28f636SEnrico Perla - Sun Microsystems if (nowrite)
2067c478bd9Sstevel@tonic-gate (void) fprintf(stdout, DRY_RUN);
207af28f636SEnrico Perla - Sun Microsystems
208af28f636SEnrico Perla - Sun Microsystems if (do_getinfo) {
209af28f636SEnrico Perla - Sun Microsystems ret = handle_getinfo(progname, handle_args);
210af28f636SEnrico Perla - Sun Microsystems } else if (do_mirror_bblk) {
211af28f636SEnrico Perla - Sun Microsystems ret = handle_mirror(progname, handle_args);
212af28f636SEnrico Perla - Sun Microsystems } else {
213af28f636SEnrico Perla - Sun Microsystems ret = handle_install(progname, handle_args);
214af28f636SEnrico Perla - Sun Microsystems }
215af28f636SEnrico Perla - Sun Microsystems return (ret);
2167c478bd9Sstevel@tonic-gate }
2177c478bd9Sstevel@tonic-gate
218af28f636SEnrico Perla - Sun Microsystems #define MEANINGLESS_OPT gettext("%s specified but meaningless, ignoring\n")
219af28f636SEnrico Perla - Sun Microsystems static void
check_options(char * progname)220af28f636SEnrico Perla - Sun Microsystems check_options(char *progname)
2217c478bd9Sstevel@tonic-gate {
222af28f636SEnrico Perla - Sun Microsystems if (do_getinfo && do_mirror_bblk) {
223af28f636SEnrico Perla - Sun Microsystems (void) fprintf(stderr, gettext("Only one of -M and -i can be "
224af28f636SEnrico Perla - Sun Microsystems "specified at the same time\n"));
225af28f636SEnrico Perla - Sun Microsystems usage(progname);
226af28f636SEnrico Perla - Sun Microsystems exit(BC_ERROR);
227af28f636SEnrico Perla - Sun Microsystems }
228af28f636SEnrico Perla - Sun Microsystems
229af28f636SEnrico Perla - Sun Microsystems if (do_mirror_bblk) {
230af28f636SEnrico Perla - Sun Microsystems /*
231af28f636SEnrico Perla - Sun Microsystems * -u and -F may actually reflect a user intent that is not
232af28f636SEnrico Perla - Sun Microsystems * correct with this command (mirror can be interpreted
233af28f636SEnrico Perla - Sun Microsystems * "similar" to install. Emit a message and continue.
234af28f636SEnrico Perla - Sun Microsystems * -e and -V have no meaning, be quiet here and only report the
235af28f636SEnrico Perla - Sun Microsystems * incongruence if a debug output is requested.
236af28f636SEnrico Perla - Sun Microsystems */
237af28f636SEnrico Perla - Sun Microsystems if (do_version) {
238af28f636SEnrico Perla - Sun Microsystems (void) fprintf(stderr, MEANINGLESS_OPT, "-u");
239af28f636SEnrico Perla - Sun Microsystems do_version = B_FALSE;
240af28f636SEnrico Perla - Sun Microsystems }
241af28f636SEnrico Perla - Sun Microsystems if (force_update) {
242af28f636SEnrico Perla - Sun Microsystems (void) fprintf(stderr, MEANINGLESS_OPT, "-F");
243af28f636SEnrico Perla - Sun Microsystems force_update = B_FALSE;
244af28f636SEnrico Perla - Sun Microsystems }
245af28f636SEnrico Perla - Sun Microsystems if (strip || verbose_dump) {
246af28f636SEnrico Perla - Sun Microsystems BOOT_DEBUG(MEANINGLESS_OPT, "-e|-V");
247af28f636SEnrico Perla - Sun Microsystems strip = B_FALSE;
248af28f636SEnrico Perla - Sun Microsystems verbose_dump = B_FALSE;
249af28f636SEnrico Perla - Sun Microsystems }
250af28f636SEnrico Perla - Sun Microsystems }
251af28f636SEnrico Perla - Sun Microsystems
252af28f636SEnrico Perla - Sun Microsystems if (do_getinfo) {
253af28f636SEnrico Perla - Sun Microsystems if (write_mbr || force_mbr || do_version || force_update) {
254af28f636SEnrico Perla - Sun Microsystems BOOT_DEBUG(MEANINGLESS_OPT, "-m|-f|-u|-F");
255af28f636SEnrico Perla - Sun Microsystems write_mbr = force_mbr = do_version = B_FALSE;
256af28f636SEnrico Perla - Sun Microsystems force_update = B_FALSE;
257af28f636SEnrico Perla - Sun Microsystems }
258af28f636SEnrico Perla - Sun Microsystems }
259af28f636SEnrico Perla - Sun Microsystems }
260af28f636SEnrico Perla - Sun Microsystems
261af28f636SEnrico Perla - Sun Microsystems /*
262af28f636SEnrico Perla - Sun Microsystems * Install a new stage1/stage2 pair on the specified device. handle_install()
263af28f636SEnrico Perla - Sun Microsystems * expects argv to contain 3 parameters (the path to stage1, the path to stage2,
264af28f636SEnrico Perla - Sun Microsystems * the target device).
265af28f636SEnrico Perla - Sun Microsystems *
266af28f636SEnrico Perla - Sun Microsystems * Returns: BC_SUCCESS - if the installation is successful
267af28f636SEnrico Perla - Sun Microsystems * BC_ERROR - if the installation failed
268af28f636SEnrico Perla - Sun Microsystems * BC_NOUPDT - if no installation was performed because the GRUB
269af28f636SEnrico Perla - Sun Microsystems * version currently installed is more recent than the
270af28f636SEnrico Perla - Sun Microsystems * supplied one.
271af28f636SEnrico Perla - Sun Microsystems *
272af28f636SEnrico Perla - Sun Microsystems */
273af28f636SEnrico Perla - Sun Microsystems static int
handle_install(char * progname,char ** argv)274af28f636SEnrico Perla - Sun Microsystems handle_install(char *progname, char **argv)
275af28f636SEnrico Perla - Sun Microsystems {
276af28f636SEnrico Perla - Sun Microsystems ig_data_t install_data;
277af28f636SEnrico Perla - Sun Microsystems char *stage1_path = NULL;
278af28f636SEnrico Perla - Sun Microsystems char *stage2_path = NULL;
279af28f636SEnrico Perla - Sun Microsystems char *device_path = NULL;
280af28f636SEnrico Perla - Sun Microsystems int ret = BC_ERROR;
281af28f636SEnrico Perla - Sun Microsystems
282af28f636SEnrico Perla - Sun Microsystems stage1_path = strdup(argv[0]);
283af28f636SEnrico Perla - Sun Microsystems stage2_path = strdup(argv[1]);
284af28f636SEnrico Perla - Sun Microsystems device_path = strdup(argv[2]);
285af28f636SEnrico Perla - Sun Microsystems
286af28f636SEnrico Perla - Sun Microsystems bzero(&install_data, sizeof (ig_data_t));
287af28f636SEnrico Perla - Sun Microsystems
288af28f636SEnrico Perla - Sun Microsystems if (!stage1_path || !stage2_path || !device_path) {
289af28f636SEnrico Perla - Sun Microsystems (void) fprintf(stderr, gettext("Missing parameter"));
290af28f636SEnrico Perla - Sun Microsystems usage(progname);
291af28f636SEnrico Perla - Sun Microsystems goto out;
292af28f636SEnrico Perla - Sun Microsystems }
293af28f636SEnrico Perla - Sun Microsystems
294af28f636SEnrico Perla - Sun Microsystems BOOT_DEBUG("stage1 path: %s, stage2 path: %s, device: %s\n",
295af28f636SEnrico Perla - Sun Microsystems stage1_path, stage2_path, device_path);
296af28f636SEnrico Perla - Sun Microsystems
297af28f636SEnrico Perla - Sun Microsystems if (init_device(&install_data.device, device_path) != BC_SUCCESS) {
298af28f636SEnrico Perla - Sun Microsystems (void) fprintf(stderr, gettext("Unable to gather device "
299af28f636SEnrico Perla - Sun Microsystems "information for %s\n"), device_path);
300af28f636SEnrico Perla - Sun Microsystems goto out;
301af28f636SEnrico Perla - Sun Microsystems }
302af28f636SEnrico Perla - Sun Microsystems
303af28f636SEnrico Perla - Sun Microsystems /* read in stage1 and stage2. */
304af28f636SEnrico Perla - Sun Microsystems if (read_stage1_from_file(stage1_path, &install_data) != BC_SUCCESS) {
305af28f636SEnrico Perla - Sun Microsystems (void) fprintf(stderr, gettext("Error opening %s\n"),
306af28f636SEnrico Perla - Sun Microsystems stage1_path);
307af28f636SEnrico Perla - Sun Microsystems goto out_dev;
308af28f636SEnrico Perla - Sun Microsystems }
309af28f636SEnrico Perla - Sun Microsystems
310af28f636SEnrico Perla - Sun Microsystems if (read_stage2_from_file(stage2_path, &install_data) != BC_SUCCESS) {
311af28f636SEnrico Perla - Sun Microsystems (void) fprintf(stderr, gettext("Error opening %s\n"),
312af28f636SEnrico Perla - Sun Microsystems stage2_path);
313af28f636SEnrico Perla - Sun Microsystems goto out_dev;
314af28f636SEnrico Perla - Sun Microsystems }
315af28f636SEnrico Perla - Sun Microsystems
316af28f636SEnrico Perla - Sun Microsystems /* We do not support versioning on PCFS. */
317af28f636SEnrico Perla - Sun Microsystems if (is_bootpar(install_data.device.type) && do_version)
318af28f636SEnrico Perla - Sun Microsystems do_version = B_FALSE;
319af28f636SEnrico Perla - Sun Microsystems
320af28f636SEnrico Perla - Sun Microsystems /*
321af28f636SEnrico Perla - Sun Microsystems * is_update_necessary() will take care of checking if versioning and/or
322af28f636SEnrico Perla - Sun Microsystems * forcing the update have been specified. It will also emit a warning
323af28f636SEnrico Perla - Sun Microsystems * if a non-versioned update is attempted over a versioned bootblock.
324af28f636SEnrico Perla - Sun Microsystems */
325af28f636SEnrico Perla - Sun Microsystems if (!is_update_necessary(&install_data, update_str)) {
326af28f636SEnrico Perla - Sun Microsystems (void) fprintf(stderr, gettext("GRUB version installed "
327af28f636SEnrico Perla - Sun Microsystems "on %s is more recent or identical\n"
328af28f636SEnrico Perla - Sun Microsystems "Use -F to override or install without the -u option\n"),
329af28f636SEnrico Perla - Sun Microsystems device_path);
330af28f636SEnrico Perla - Sun Microsystems ret = BC_NOUPDT;
331af28f636SEnrico Perla - Sun Microsystems goto out_dev;
332af28f636SEnrico Perla - Sun Microsystems }
333af28f636SEnrico Perla - Sun Microsystems /*
334af28f636SEnrico Perla - Sun Microsystems * We get here if:
335af28f636SEnrico Perla - Sun Microsystems * - the installed GRUB version is older than the one about to be
336af28f636SEnrico Perla - Sun Microsystems * installed.
337af28f636SEnrico Perla - Sun Microsystems * - no versioning string has been passed through the command line.
338af28f636SEnrico Perla - Sun Microsystems * - a forced update is requested (-F).
339af28f636SEnrico Perla - Sun Microsystems */
340af28f636SEnrico Perla - Sun Microsystems BOOT_DEBUG("Ready to commit to disk\n");
341af28f636SEnrico Perla - Sun Microsystems ret = commit_to_disk(&install_data, update_str);
342af28f636SEnrico Perla - Sun Microsystems
343af28f636SEnrico Perla - Sun Microsystems out_dev:
344af28f636SEnrico Perla - Sun Microsystems cleanup_device(&install_data.device);
345af28f636SEnrico Perla - Sun Microsystems out:
346af28f636SEnrico Perla - Sun Microsystems free(stage1_path);
347af28f636SEnrico Perla - Sun Microsystems free(stage2_path);
348af28f636SEnrico Perla - Sun Microsystems free(device_path);
349af28f636SEnrico Perla - Sun Microsystems return (ret);
350af28f636SEnrico Perla - Sun Microsystems }
351af28f636SEnrico Perla - Sun Microsystems
352af28f636SEnrico Perla - Sun Microsystems /*
353af28f636SEnrico Perla - Sun Microsystems * Retrieves from a device the extended information (einfo) associated to the
354af28f636SEnrico Perla - Sun Microsystems * installed stage2.
355af28f636SEnrico Perla - Sun Microsystems * Expects one parameter, the device path, in the form: /dev/rdsk/c?[t?]d?s0.
356af28f636SEnrico Perla - Sun Microsystems * Returns:
357af28f636SEnrico Perla - Sun Microsystems * - BC_SUCCESS (and prints out einfo contents depending on 'flags')
358af28f636SEnrico Perla - Sun Microsystems * - BC_ERROR (on error)
359af28f636SEnrico Perla - Sun Microsystems * - BC_NOEINFO (no extended information available)
360af28f636SEnrico Perla - Sun Microsystems */
361af28f636SEnrico Perla - Sun Microsystems static int
handle_getinfo(char * progname,char ** argv)362af28f636SEnrico Perla - Sun Microsystems handle_getinfo(char *progname, char **argv)
363af28f636SEnrico Perla - Sun Microsystems {
364af28f636SEnrico Perla - Sun Microsystems ig_data_t data;
365af28f636SEnrico Perla - Sun Microsystems ig_stage2_t *stage2 = &data.stage2;
366af28f636SEnrico Perla - Sun Microsystems ig_device_t *device = &data.device;
367af28f636SEnrico Perla - Sun Microsystems bblk_einfo_t *einfo;
368af28f636SEnrico Perla - Sun Microsystems uint8_t flags = 0;
369af28f636SEnrico Perla - Sun Microsystems uint32_t size;
370af28f636SEnrico Perla - Sun Microsystems char *device_path;
371af28f636SEnrico Perla - Sun Microsystems int retval = BC_ERROR;
372af28f636SEnrico Perla - Sun Microsystems int ret;
373af28f636SEnrico Perla - Sun Microsystems
374af28f636SEnrico Perla - Sun Microsystems device_path = strdup(argv[0]);
375af28f636SEnrico Perla - Sun Microsystems if (!device_path) {
376af28f636SEnrico Perla - Sun Microsystems (void) fprintf(stderr, gettext("Missing parameter"));
377af28f636SEnrico Perla - Sun Microsystems usage(progname);
378af28f636SEnrico Perla - Sun Microsystems goto out;
379af28f636SEnrico Perla - Sun Microsystems }
380af28f636SEnrico Perla - Sun Microsystems
381af28f636SEnrico Perla - Sun Microsystems bzero(&data, sizeof (ig_data_t));
382af28f636SEnrico Perla - Sun Microsystems BOOT_DEBUG("device path: %s\n", device_path);
383af28f636SEnrico Perla - Sun Microsystems
384af28f636SEnrico Perla - Sun Microsystems if (init_device(device, device_path) != BC_SUCCESS) {
385af28f636SEnrico Perla - Sun Microsystems (void) fprintf(stderr, gettext("Unable to gather device "
386af28f636SEnrico Perla - Sun Microsystems "information for %s\n"), device_path);
387af28f636SEnrico Perla - Sun Microsystems goto out_dev;
388af28f636SEnrico Perla - Sun Microsystems }
389af28f636SEnrico Perla - Sun Microsystems
390af28f636SEnrico Perla - Sun Microsystems if (is_bootpar(device->type)) {
391af28f636SEnrico Perla - Sun Microsystems (void) fprintf(stderr, gettext("Versioning not supported on "
392af28f636SEnrico Perla - Sun Microsystems "PCFS\n"));
393af28f636SEnrico Perla - Sun Microsystems goto out_dev;
394af28f636SEnrico Perla - Sun Microsystems }
395af28f636SEnrico Perla - Sun Microsystems
3961a902ef8SHans Rosenfeld ret = read_stage2_from_disk(device->part_fd, stage2, device->type);
397af28f636SEnrico Perla - Sun Microsystems if (ret == BC_ERROR) {
398af28f636SEnrico Perla - Sun Microsystems (void) fprintf(stderr, gettext("Error reading stage2 from "
399af28f636SEnrico Perla - Sun Microsystems "%s\n"), device_path);
400af28f636SEnrico Perla - Sun Microsystems goto out_dev;
401af28f636SEnrico Perla - Sun Microsystems }
402af28f636SEnrico Perla - Sun Microsystems
403af28f636SEnrico Perla - Sun Microsystems if (ret == BC_NOEXTRA) {
404af28f636SEnrico Perla - Sun Microsystems (void) fprintf(stdout, gettext("No multiboot header found on "
405af28f636SEnrico Perla - Sun Microsystems "%s, unable to locate extra information area\n"),
406af28f636SEnrico Perla - Sun Microsystems device_path);
407af28f636SEnrico Perla - Sun Microsystems retval = BC_NOEINFO;
408af28f636SEnrico Perla - Sun Microsystems goto out_dev;
409af28f636SEnrico Perla - Sun Microsystems }
410af28f636SEnrico Perla - Sun Microsystems
41114d44f22SHans Rosenfeld einfo = find_einfo(stage2->extra, stage2->extra_size);
412af28f636SEnrico Perla - Sun Microsystems if (einfo == NULL) {
413af28f636SEnrico Perla - Sun Microsystems retval = BC_NOEINFO;
414af28f636SEnrico Perla - Sun Microsystems (void) fprintf(stderr, gettext("No extended information "
415af28f636SEnrico Perla - Sun Microsystems "found\n"));
416af28f636SEnrico Perla - Sun Microsystems goto out_dev;
417af28f636SEnrico Perla - Sun Microsystems }
418af28f636SEnrico Perla - Sun Microsystems
419af28f636SEnrico Perla - Sun Microsystems /* Print the extended information. */
420af28f636SEnrico Perla - Sun Microsystems if (strip)
421af28f636SEnrico Perla - Sun Microsystems flags |= EINFO_EASY_PARSE;
422af28f636SEnrico Perla - Sun Microsystems if (verbose_dump)
423af28f636SEnrico Perla - Sun Microsystems flags |= EINFO_PRINT_HEADER;
424af28f636SEnrico Perla - Sun Microsystems
425af28f636SEnrico Perla - Sun Microsystems size = stage2->buf_size - P2ROUNDUP(stage2->file_size, 8);
426af28f636SEnrico Perla - Sun Microsystems print_einfo(flags, einfo, size);
427af28f636SEnrico Perla - Sun Microsystems retval = BC_SUCCESS;
428af28f636SEnrico Perla - Sun Microsystems
429af28f636SEnrico Perla - Sun Microsystems out_dev:
430af28f636SEnrico Perla - Sun Microsystems cleanup_device(&data.device);
431af28f636SEnrico Perla - Sun Microsystems out:
432af28f636SEnrico Perla - Sun Microsystems free(device_path);
433af28f636SEnrico Perla - Sun Microsystems return (retval);
434af28f636SEnrico Perla - Sun Microsystems }
435af28f636SEnrico Perla - Sun Microsystems
436af28f636SEnrico Perla - Sun Microsystems /*
437af28f636SEnrico Perla - Sun Microsystems * Attempt to mirror (propagate) the current stage2 over the attaching disk.
438af28f636SEnrico Perla - Sun Microsystems *
439af28f636SEnrico Perla - Sun Microsystems * Returns:
440af28f636SEnrico Perla - Sun Microsystems * - BC_SUCCESS (a successful propagation happened)
441af28f636SEnrico Perla - Sun Microsystems * - BC_ERROR (an error occurred)
442af28f636SEnrico Perla - Sun Microsystems * - BC_NOEXTRA (it is not possible to dump the current bootblock since
443af28f636SEnrico Perla - Sun Microsystems * there is no multiboot information)
444af28f636SEnrico Perla - Sun Microsystems */
445af28f636SEnrico Perla - Sun Microsystems static int
handle_mirror(char * progname,char ** argv)446af28f636SEnrico Perla - Sun Microsystems handle_mirror(char *progname, char **argv)
447af28f636SEnrico Perla - Sun Microsystems {
448af28f636SEnrico Perla - Sun Microsystems ig_data_t curr_data;
449af28f636SEnrico Perla - Sun Microsystems ig_data_t attach_data;
450af28f636SEnrico Perla - Sun Microsystems ig_device_t *curr_device = &curr_data.device;
451af28f636SEnrico Perla - Sun Microsystems ig_device_t *attach_device = &attach_data.device;
452af28f636SEnrico Perla - Sun Microsystems ig_stage2_t *stage2_curr = &curr_data.stage2;
453af28f636SEnrico Perla - Sun Microsystems ig_stage2_t *stage2_attach = &attach_data.stage2;
454af28f636SEnrico Perla - Sun Microsystems bblk_einfo_t *einfo_curr = NULL;
455af28f636SEnrico Perla - Sun Microsystems char *curr_device_path;
456af28f636SEnrico Perla - Sun Microsystems char *attach_device_path;
457af28f636SEnrico Perla - Sun Microsystems char *updt_str = NULL;
458af28f636SEnrico Perla - Sun Microsystems int retval = BC_ERROR;
459af28f636SEnrico Perla - Sun Microsystems int ret;
460af28f636SEnrico Perla - Sun Microsystems
461af28f636SEnrico Perla - Sun Microsystems curr_device_path = strdup(argv[0]);
462af28f636SEnrico Perla - Sun Microsystems attach_device_path = strdup(argv[1]);
463af28f636SEnrico Perla - Sun Microsystems
464af28f636SEnrico Perla - Sun Microsystems if (!curr_device_path || !attach_device_path) {
465af28f636SEnrico Perla - Sun Microsystems (void) fprintf(stderr, gettext("Missing parameter"));
466af28f636SEnrico Perla - Sun Microsystems usage(progname);
467af28f636SEnrico Perla - Sun Microsystems goto out;
468af28f636SEnrico Perla - Sun Microsystems }
469af28f636SEnrico Perla - Sun Microsystems BOOT_DEBUG("Current device path is: %s, attaching device path is: "
470af28f636SEnrico Perla - Sun Microsystems " %s\n", curr_device_path, attach_device_path);
471af28f636SEnrico Perla - Sun Microsystems
472af28f636SEnrico Perla - Sun Microsystems bzero(&curr_data, sizeof (ig_data_t));
473af28f636SEnrico Perla - Sun Microsystems bzero(&attach_data, sizeof (ig_data_t));
474af28f636SEnrico Perla - Sun Microsystems
475af28f636SEnrico Perla - Sun Microsystems if (init_device(curr_device, curr_device_path) != BC_SUCCESS) {
476af28f636SEnrico Perla - Sun Microsystems (void) fprintf(stderr, gettext("Unable to gather device "
477af28f636SEnrico Perla - Sun Microsystems "information for %s (current device)\n"), curr_device_path);
478af28f636SEnrico Perla - Sun Microsystems goto out_currdev;
479af28f636SEnrico Perla - Sun Microsystems }
480af28f636SEnrico Perla - Sun Microsystems
481af28f636SEnrico Perla - Sun Microsystems if (init_device(attach_device, attach_device_path) != BC_SUCCESS) {
482af28f636SEnrico Perla - Sun Microsystems (void) fprintf(stderr, gettext("Unable to gather device "
483af28f636SEnrico Perla - Sun Microsystems "information for %s (attaching device)\n"),
484af28f636SEnrico Perla - Sun Microsystems attach_device_path);
485af28f636SEnrico Perla - Sun Microsystems goto out_devs;
486af28f636SEnrico Perla - Sun Microsystems }
487af28f636SEnrico Perla - Sun Microsystems
488af28f636SEnrico Perla - Sun Microsystems if (is_bootpar(curr_device->type) || is_bootpar(attach_device->type)) {
489af28f636SEnrico Perla - Sun Microsystems (void) fprintf(stderr, gettext("boot block mirroring is not "
490af28f636SEnrico Perla - Sun Microsystems "supported on PCFS\n"));
491af28f636SEnrico Perla - Sun Microsystems goto out_devs;
492af28f636SEnrico Perla - Sun Microsystems }
493af28f636SEnrico Perla - Sun Microsystems
4941a902ef8SHans Rosenfeld ret = read_stage2_from_disk(curr_device->part_fd, stage2_curr,
4951a902ef8SHans Rosenfeld curr_device->type);
496af28f636SEnrico Perla - Sun Microsystems if (ret == BC_ERROR) {
497af28f636SEnrico Perla - Sun Microsystems BOOT_DEBUG("Error reading first stage2 blocks from %s\n",
498af28f636SEnrico Perla - Sun Microsystems curr_device->path);
499af28f636SEnrico Perla - Sun Microsystems retval = BC_ERROR;
500af28f636SEnrico Perla - Sun Microsystems goto out_devs;
501af28f636SEnrico Perla - Sun Microsystems }
502af28f636SEnrico Perla - Sun Microsystems
503af28f636SEnrico Perla - Sun Microsystems if (ret == BC_NOEXTRA) {
504af28f636SEnrico Perla - Sun Microsystems BOOT_DEBUG("No multiboot header found on %s, unable to grab "
505af28f636SEnrico Perla - Sun Microsystems "stage2\n", curr_device->path);
506af28f636SEnrico Perla - Sun Microsystems retval = BC_NOEXTRA;
507af28f636SEnrico Perla - Sun Microsystems goto out_devs;
508af28f636SEnrico Perla - Sun Microsystems }
509af28f636SEnrico Perla - Sun Microsystems
51014d44f22SHans Rosenfeld einfo_curr = find_einfo(stage2_curr->extra, stage2_curr->extra_size);
511af28f636SEnrico Perla - Sun Microsystems if (einfo_curr != NULL)
512af28f636SEnrico Perla - Sun Microsystems updt_str = einfo_get_string(einfo_curr);
513af28f636SEnrico Perla - Sun Microsystems
514af28f636SEnrico Perla - Sun Microsystems write_mbr = B_TRUE;
515af28f636SEnrico Perla - Sun Microsystems force_mbr = B_TRUE;
516af28f636SEnrico Perla - Sun Microsystems retval = propagate_bootblock(&curr_data, &attach_data, updt_str);
517af28f636SEnrico Perla - Sun Microsystems cleanup_stage2(stage2_curr);
518af28f636SEnrico Perla - Sun Microsystems cleanup_stage2(stage2_attach);
519af28f636SEnrico Perla - Sun Microsystems
520af28f636SEnrico Perla - Sun Microsystems out_devs:
521af28f636SEnrico Perla - Sun Microsystems cleanup_device(attach_device);
522af28f636SEnrico Perla - Sun Microsystems out_currdev:
523af28f636SEnrico Perla - Sun Microsystems cleanup_device(curr_device);
524af28f636SEnrico Perla - Sun Microsystems out:
525af28f636SEnrico Perla - Sun Microsystems free(curr_device_path);
526af28f636SEnrico Perla - Sun Microsystems free(attach_device_path);
527af28f636SEnrico Perla - Sun Microsystems return (retval);
528af28f636SEnrico Perla - Sun Microsystems }
529af28f636SEnrico Perla - Sun Microsystems
530af28f636SEnrico Perla - Sun Microsystems static int
commit_to_disk(ig_data_t * install,char * updt_str)531af28f636SEnrico Perla - Sun Microsystems commit_to_disk(ig_data_t *install, char *updt_str)
532af28f636SEnrico Perla - Sun Microsystems {
533af28f636SEnrico Perla - Sun Microsystems assert(install != NULL);
534af28f636SEnrico Perla - Sun Microsystems /*
535af28f636SEnrico Perla - Sun Microsystems * vanilla stage1 and stage2 need to be updated at runtime.
536af28f636SEnrico Perla - Sun Microsystems * Update stage2 before stage1 because stage1 needs to know the first
537af28f636SEnrico Perla - Sun Microsystems * sector stage2 will be written to.
538af28f636SEnrico Perla - Sun Microsystems */
539af28f636SEnrico Perla - Sun Microsystems if (prepare_stage2(install, updt_str) != BC_SUCCESS) {
540af28f636SEnrico Perla - Sun Microsystems (void) fprintf(stderr, gettext("Error building stage2\n"));
541af28f636SEnrico Perla - Sun Microsystems return (BC_ERROR);
542af28f636SEnrico Perla - Sun Microsystems }
543af28f636SEnrico Perla - Sun Microsystems if (prepare_stage1(install) != BC_SUCCESS) {
544af28f636SEnrico Perla - Sun Microsystems (void) fprintf(stderr, gettext("Error building stage1\n"));
545af28f636SEnrico Perla - Sun Microsystems return (BC_ERROR);
546af28f636SEnrico Perla - Sun Microsystems }
547af28f636SEnrico Perla - Sun Microsystems
548af28f636SEnrico Perla - Sun Microsystems /* Write stage2 out to disk. */
549af28f636SEnrico Perla - Sun Microsystems if (write_stage2(install) != BC_SUCCESS) {
550af28f636SEnrico Perla - Sun Microsystems (void) fprintf(stderr, gettext("Error writing stage2 to "
551af28f636SEnrico Perla - Sun Microsystems "disk\n"));
552af28f636SEnrico Perla - Sun Microsystems return (BC_ERROR);
553af28f636SEnrico Perla - Sun Microsystems }
554af28f636SEnrico Perla - Sun Microsystems
555af28f636SEnrico Perla - Sun Microsystems /* Write stage1 to disk and, if requested, to the MBR. */
556af28f636SEnrico Perla - Sun Microsystems if (write_stage1(install) != BC_SUCCESS) {
557af28f636SEnrico Perla - Sun Microsystems (void) fprintf(stderr, gettext("Error writing stage1 to "
558af28f636SEnrico Perla - Sun Microsystems "disk\n"));
559af28f636SEnrico Perla - Sun Microsystems return (BC_ERROR);
560af28f636SEnrico Perla - Sun Microsystems }
561af28f636SEnrico Perla - Sun Microsystems
562af28f636SEnrico Perla - Sun Microsystems return (BC_SUCCESS);
563af28f636SEnrico Perla - Sun Microsystems }
564af28f636SEnrico Perla - Sun Microsystems
565af28f636SEnrico Perla - Sun Microsystems /*
566af28f636SEnrico Perla - Sun Microsystems * Propagate the bootblock on the source disk to the destination disk and
567af28f636SEnrico Perla - Sun Microsystems * version it with 'updt_str' in the process. Since we cannot trust any data
568af28f636SEnrico Perla - Sun Microsystems * on the attaching disk, we do not perform any specific check on a potential
569af28f636SEnrico Perla - Sun Microsystems * target extended information structure and we just blindly update.
570af28f636SEnrico Perla - Sun Microsystems */
571af28f636SEnrico Perla - Sun Microsystems static int
propagate_bootblock(ig_data_t * source,ig_data_t * target,char * updt_str)572af28f636SEnrico Perla - Sun Microsystems propagate_bootblock(ig_data_t *source, ig_data_t *target, char *updt_str)
573af28f636SEnrico Perla - Sun Microsystems {
574af28f636SEnrico Perla - Sun Microsystems ig_device_t *src_device = &source->device;
575af28f636SEnrico Perla - Sun Microsystems ig_device_t *dest_device = &target->device;
576af28f636SEnrico Perla - Sun Microsystems ig_stage2_t *src_stage2 = &source->stage2;
577af28f636SEnrico Perla - Sun Microsystems ig_stage2_t *dest_stage2 = &target->stage2;
578af28f636SEnrico Perla - Sun Microsystems uint32_t buf_size;
579af28f636SEnrico Perla - Sun Microsystems int retval;
580af28f636SEnrico Perla - Sun Microsystems
581af28f636SEnrico Perla - Sun Microsystems assert(source != NULL);
582af28f636SEnrico Perla - Sun Microsystems assert(target != NULL);
583af28f636SEnrico Perla - Sun Microsystems
584af28f636SEnrico Perla - Sun Microsystems /* read in stage1 from the source disk. */
585af28f636SEnrico Perla - Sun Microsystems if (read_stage1_from_disk(src_device->part_fd, target->stage1_buf)
586af28f636SEnrico Perla - Sun Microsystems != BC_SUCCESS)
587af28f636SEnrico Perla - Sun Microsystems return (BC_ERROR);
588af28f636SEnrico Perla - Sun Microsystems
589af28f636SEnrico Perla - Sun Microsystems /* Prepare target stage2 for commit_to_disk. */
590af28f636SEnrico Perla - Sun Microsystems cleanup_stage2(dest_stage2);
591af28f636SEnrico Perla - Sun Microsystems
592af28f636SEnrico Perla - Sun Microsystems if (updt_str != NULL)
593af28f636SEnrico Perla - Sun Microsystems do_version = B_TRUE;
594af28f636SEnrico Perla - Sun Microsystems else
595af28f636SEnrico Perla - Sun Microsystems do_version = B_FALSE;
596af28f636SEnrico Perla - Sun Microsystems
597af28f636SEnrico Perla - Sun Microsystems buf_size = src_stage2->file_size + SECTOR_SIZE;
598af28f636SEnrico Perla - Sun Microsystems
599af28f636SEnrico Perla - Sun Microsystems dest_stage2->buf_size = P2ROUNDUP(buf_size, SECTOR_SIZE);
600af28f636SEnrico Perla - Sun Microsystems dest_stage2->buf = malloc(dest_stage2->buf_size);
601af28f636SEnrico Perla - Sun Microsystems if (dest_stage2->buf == NULL) {
602af28f636SEnrico Perla - Sun Microsystems perror(gettext("Memory allocation failed"));
603af28f636SEnrico Perla - Sun Microsystems return (BC_ERROR);
604af28f636SEnrico Perla - Sun Microsystems }
605af28f636SEnrico Perla - Sun Microsystems dest_stage2->file = dest_stage2->buf;
606af28f636SEnrico Perla - Sun Microsystems dest_stage2->file_size = src_stage2->file_size;
607af28f636SEnrico Perla - Sun Microsystems memcpy(dest_stage2->file, src_stage2->file, dest_stage2->file_size);
608af28f636SEnrico Perla - Sun Microsystems dest_stage2->extra = dest_stage2->buf +
609af28f636SEnrico Perla - Sun Microsystems P2ROUNDUP(dest_stage2->file_size, 8);
610af28f636SEnrico Perla - Sun Microsystems
611af28f636SEnrico Perla - Sun Microsystems /* If we get down here we do have a mboot structure. */
612af28f636SEnrico Perla - Sun Microsystems assert(src_stage2->mboot);
613af28f636SEnrico Perla - Sun Microsystems
614af28f636SEnrico Perla - Sun Microsystems dest_stage2->mboot_off = src_stage2->mboot_off;
615af28f636SEnrico Perla - Sun Microsystems dest_stage2->mboot = (multiboot_header_t *)(dest_stage2->buf +
616af28f636SEnrico Perla - Sun Microsystems dest_stage2->mboot_off);
617af28f636SEnrico Perla - Sun Microsystems
618af28f636SEnrico Perla - Sun Microsystems (void) fprintf(stdout, gettext("Propagating %s stage1/stage2 to %s\n"),
619af28f636SEnrico Perla - Sun Microsystems src_device->path, dest_device->path);
620af28f636SEnrico Perla - Sun Microsystems retval = commit_to_disk(target, updt_str);
621af28f636SEnrico Perla - Sun Microsystems
622af28f636SEnrico Perla - Sun Microsystems return (retval);
623af28f636SEnrico Perla - Sun Microsystems }
624af28f636SEnrico Perla - Sun Microsystems
625af28f636SEnrico Perla - Sun Microsystems /*
626af28f636SEnrico Perla - Sun Microsystems * open the device and fill the various members of ig_device_t.
627af28f636SEnrico Perla - Sun Microsystems */
628af28f636SEnrico Perla - Sun Microsystems static int
init_device(ig_device_t * device,char * path)629af28f636SEnrico Perla - Sun Microsystems init_device(ig_device_t *device, char *path)
630af28f636SEnrico Perla - Sun Microsystems {
6311a902ef8SHans Rosenfeld struct dk_gpt *vtoc;
6321a902ef8SHans Rosenfeld fstyp_handle_t fhdl;
6331a902ef8SHans Rosenfeld const char *fident;
6341a902ef8SHans Rosenfeld
635af28f636SEnrico Perla - Sun Microsystems bzero(device, sizeof (*device));
636af28f636SEnrico Perla - Sun Microsystems device->part_fd = -1;
637af28f636SEnrico Perla - Sun Microsystems device->disk_fd = -1;
638af28f636SEnrico Perla - Sun Microsystems device->path_p0 = NULL;
639af28f636SEnrico Perla - Sun Microsystems
640af28f636SEnrico Perla - Sun Microsystems device->path = strdup(path);
641af28f636SEnrico Perla - Sun Microsystems if (device->path == NULL) {
642af28f636SEnrico Perla - Sun Microsystems perror(gettext("Memory allocation failed"));
643af28f636SEnrico Perla - Sun Microsystems return (BC_ERROR);
644af28f636SEnrico Perla - Sun Microsystems }
645af28f636SEnrico Perla - Sun Microsystems
646af28f636SEnrico Perla - Sun Microsystems if (strstr(device->path, "diskette")) {
647af28f636SEnrico Perla - Sun Microsystems (void) fprintf(stderr, gettext("installing GRUB to a floppy "
648af28f636SEnrico Perla - Sun Microsystems "disk is no longer supported\n"));
649af28f636SEnrico Perla - Sun Microsystems return (BC_ERROR);
650af28f636SEnrico Perla - Sun Microsystems }
651af28f636SEnrico Perla - Sun Microsystems
652af28f636SEnrico Perla - Sun Microsystems /* Detect if the target device is a pcfs partition. */
653af28f636SEnrico Perla - Sun Microsystems if (strstr(device->path, "p0:boot"))
654af28f636SEnrico Perla - Sun Microsystems device->type = IG_DEV_X86BOOTPAR;
655af28f636SEnrico Perla - Sun Microsystems
656af28f636SEnrico Perla - Sun Microsystems if (get_disk_fd(device) != BC_SUCCESS)
657af28f636SEnrico Perla - Sun Microsystems return (BC_ERROR);
658af28f636SEnrico Perla - Sun Microsystems
659af28f636SEnrico Perla - Sun Microsystems /* read in the device boot sector. */
660af28f636SEnrico Perla - Sun Microsystems if (read(device->disk_fd, device->boot_sector, SECTOR_SIZE)
661af28f636SEnrico Perla - Sun Microsystems != SECTOR_SIZE) {
662af28f636SEnrico Perla - Sun Microsystems (void) fprintf(stderr, gettext("Error reading boot sector\n"));
663af28f636SEnrico Perla - Sun Microsystems perror("read");
664af28f636SEnrico Perla - Sun Microsystems return (BC_ERROR);
665af28f636SEnrico Perla - Sun Microsystems }
666af28f636SEnrico Perla - Sun Microsystems
6671a902ef8SHans Rosenfeld if (efi_alloc_and_read(device->disk_fd, &vtoc) > 0) {
6681a902ef8SHans Rosenfeld device->type = IG_DEV_EFI;
6691a902ef8SHans Rosenfeld efi_free(vtoc);
6701a902ef8SHans Rosenfeld }
6711a902ef8SHans Rosenfeld
672af28f636SEnrico Perla - Sun Microsystems if (get_raw_partition_fd(device) != BC_SUCCESS)
673af28f636SEnrico Perla - Sun Microsystems return (BC_ERROR);
674af28f636SEnrico Perla - Sun Microsystems
675*9348d232SHans Rosenfeld if (is_efi(device->type)) {
6761a902ef8SHans Rosenfeld if (fstyp_init(device->part_fd, 0, NULL, &fhdl) != 0)
6771a902ef8SHans Rosenfeld return (BC_ERROR);
6781a902ef8SHans Rosenfeld
6791a902ef8SHans Rosenfeld if (fstyp_ident(fhdl, "zfs", &fident) != 0) {
6801a902ef8SHans Rosenfeld fstyp_fini(fhdl);
681*9348d232SHans Rosenfeld (void) fprintf(stderr, gettext("Booting of EFI labeled "
682*9348d232SHans Rosenfeld "disks is only supported with ZFS\n"));
6831a902ef8SHans Rosenfeld return (BC_ERROR);
6841a902ef8SHans Rosenfeld }
6851a902ef8SHans Rosenfeld fstyp_fini(fhdl);
686*9348d232SHans Rosenfeld }
6871a902ef8SHans Rosenfeld
688af28f636SEnrico Perla - Sun Microsystems if (get_start_sector(device) != BC_SUCCESS)
689af28f636SEnrico Perla - Sun Microsystems return (BC_ERROR);
690af28f636SEnrico Perla - Sun Microsystems
691af28f636SEnrico Perla - Sun Microsystems return (BC_SUCCESS);
692af28f636SEnrico Perla - Sun Microsystems }
693af28f636SEnrico Perla - Sun Microsystems
694af28f636SEnrico Perla - Sun Microsystems static void
cleanup_device(ig_device_t * device)695af28f636SEnrico Perla - Sun Microsystems cleanup_device(ig_device_t *device)
696af28f636SEnrico Perla - Sun Microsystems {
697af28f636SEnrico Perla - Sun Microsystems if (device->path)
698af28f636SEnrico Perla - Sun Microsystems free(device->path);
699af28f636SEnrico Perla - Sun Microsystems if (device->path_p0)
700af28f636SEnrico Perla - Sun Microsystems free(device->path_p0);
701af28f636SEnrico Perla - Sun Microsystems
702af28f636SEnrico Perla - Sun Microsystems if (device->part_fd != -1)
703af28f636SEnrico Perla - Sun Microsystems (void) close(device->part_fd);
704af28f636SEnrico Perla - Sun Microsystems if (device->disk_fd != -1)
705af28f636SEnrico Perla - Sun Microsystems (void) close(device->disk_fd);
706af28f636SEnrico Perla - Sun Microsystems
707af28f636SEnrico Perla - Sun Microsystems bzero(device, sizeof (ig_device_t));
708af28f636SEnrico Perla - Sun Microsystems device->part_fd = -1;
709af28f636SEnrico Perla - Sun Microsystems device->disk_fd = -1;
710af28f636SEnrico Perla - Sun Microsystems }
711af28f636SEnrico Perla - Sun Microsystems
712af28f636SEnrico Perla - Sun Microsystems static void
cleanup_stage2(ig_stage2_t * stage2)713af28f636SEnrico Perla - Sun Microsystems cleanup_stage2(ig_stage2_t *stage2)
714af28f636SEnrico Perla - Sun Microsystems {
715af28f636SEnrico Perla - Sun Microsystems if (stage2->buf)
716af28f636SEnrico Perla - Sun Microsystems free(stage2->buf);
717af28f636SEnrico Perla - Sun Microsystems bzero(stage2, sizeof (ig_stage2_t));
718af28f636SEnrico Perla - Sun Microsystems }
719af28f636SEnrico Perla - Sun Microsystems
720af28f636SEnrico Perla - Sun Microsystems static int
get_start_sector(ig_device_t * device)721af28f636SEnrico Perla - Sun Microsystems get_start_sector(ig_device_t *device)
722af28f636SEnrico Perla - Sun Microsystems {
7237fc5d2a4SVikram Hegde uint32_t secnum = 0, numsec = 0;
7247fc5d2a4SVikram Hegde int i, pno, rval, log_part = 0;
7257c478bd9Sstevel@tonic-gate struct mboot *mboot;
7267c478bd9Sstevel@tonic-gate struct ipart *part;
727aa1b14e7SSheshadri Vasudevan ext_part_t *epp;
7287fc5d2a4SVikram Hegde struct part_info dkpi;
7297fc5d2a4SVikram Hegde struct extpart_info edkpi;
7307c478bd9Sstevel@tonic-gate
7311a902ef8SHans Rosenfeld if (is_efi(device->type)) {
7321a902ef8SHans Rosenfeld struct dk_gpt *vtoc;
7331a902ef8SHans Rosenfeld
7341a902ef8SHans Rosenfeld if (efi_alloc_and_read(device->disk_fd, &vtoc) <= 0)
7351a902ef8SHans Rosenfeld return (BC_ERROR);
7361a902ef8SHans Rosenfeld
7371a902ef8SHans Rosenfeld device->start_sector = vtoc->efi_parts[device->slice].p_start;
7381a902ef8SHans Rosenfeld /* GPT doesn't use traditional slice letters */
7391a902ef8SHans Rosenfeld device->slice = 0xff;
7401a902ef8SHans Rosenfeld device->partition = 0;
7411a902ef8SHans Rosenfeld
7421a902ef8SHans Rosenfeld efi_free(vtoc);
7431a902ef8SHans Rosenfeld goto found_part;
7441a902ef8SHans Rosenfeld }
7451a902ef8SHans Rosenfeld
746af28f636SEnrico Perla - Sun Microsystems mboot = (struct mboot *)device->boot_sector;
7477c478bd9Sstevel@tonic-gate
748af28f636SEnrico Perla - Sun Microsystems if (is_bootpar(device->type)) {
749af28f636SEnrico Perla - Sun Microsystems if (find_x86_bootpar(mboot, &pno, &secnum) != BC_SUCCESS) {
750af28f636SEnrico Perla - Sun Microsystems (void) fprintf(stderr, NOBOOTPAR);
751af28f636SEnrico Perla - Sun Microsystems return (BC_ERROR);
752af28f636SEnrico Perla - Sun Microsystems } else {
753af28f636SEnrico Perla - Sun Microsystems device->start_sector = secnum;
754af28f636SEnrico Perla - Sun Microsystems device->partition = pno;
7557fc5d2a4SVikram Hegde goto found_part;
7567fc5d2a4SVikram Hegde }
757d33344bbSsy25831 }
758d33344bbSsy25831
759aa1b14e7SSheshadri Vasudevan /*
760af28f636SEnrico Perla - Sun Microsystems * Search for Solaris fdisk partition
761d33344bbSsy25831 * Get the solaris partition information from the device
762d33344bbSsy25831 * and compare the offset of S2 with offset of solaris partition
763d33344bbSsy25831 * from fdisk partition table.
764d33344bbSsy25831 */
765af28f636SEnrico Perla - Sun Microsystems if (ioctl(device->part_fd, DKIOCEXTPARTINFO, &edkpi) < 0) {
766af28f636SEnrico Perla - Sun Microsystems if (ioctl(device->part_fd, DKIOCPARTINFO, &dkpi) < 0) {
767d33344bbSsy25831 (void) fprintf(stderr, PART_FAIL);
768af28f636SEnrico Perla - Sun Microsystems return (BC_ERROR);
769342440ecSPrasad Singamsetty } else {
770342440ecSPrasad Singamsetty edkpi.p_start = dkpi.p_start;
771342440ecSPrasad Singamsetty }
772d33344bbSsy25831 }
773d33344bbSsy25831
774d33344bbSsy25831 for (i = 0; i < FD_NUMPART; i++) {
775d33344bbSsy25831 part = (struct ipart *)mboot->parts + i;
776d33344bbSsy25831
777d33344bbSsy25831 if (part->relsect == 0) {
778d33344bbSsy25831 (void) fprintf(stderr, BAD_PART, i);
779af28f636SEnrico Perla - Sun Microsystems return (BC_ERROR);
780d33344bbSsy25831 }
781aa1b14e7SSheshadri Vasudevan
782342440ecSPrasad Singamsetty if (edkpi.p_start >= part->relsect &&
783342440ecSPrasad Singamsetty edkpi.p_start < (part->relsect + part->numsect)) {
784d33344bbSsy25831 /* Found the partition */
7857c478bd9Sstevel@tonic-gate break;
786d33344bbSsy25831 }
7877c478bd9Sstevel@tonic-gate }
7887c478bd9Sstevel@tonic-gate
7897fc5d2a4SVikram Hegde if (i == FD_NUMPART) {
7907fc5d2a4SVikram Hegde /* No solaris fdisk partitions (primary or logical) */
7917fc5d2a4SVikram Hegde (void) fprintf(stderr, NOSOLPAR);
792af28f636SEnrico Perla - Sun Microsystems return (BC_ERROR);
7937c478bd9Sstevel@tonic-gate }
7947c478bd9Sstevel@tonic-gate
7957fc5d2a4SVikram Hegde /*
7967fc5d2a4SVikram Hegde * We have found a Solaris fdisk partition (primary or extended)
7977fc5d2a4SVikram Hegde * Handle the simple case first: Solaris in a primary partition
7987fc5d2a4SVikram Hegde */
7997fc5d2a4SVikram Hegde if (!fdisk_is_dos_extended(part->systid)) {
800af28f636SEnrico Perla - Sun Microsystems device->start_sector = part->relsect;
801af28f636SEnrico Perla - Sun Microsystems device->partition = i;
8027fc5d2a4SVikram Hegde goto found_part;
8037fc5d2a4SVikram Hegde }
8047fc5d2a4SVikram Hegde
8057fc5d2a4SVikram Hegde /*
8067fc5d2a4SVikram Hegde * Solaris in a logical partition. Find that partition in the
8077fc5d2a4SVikram Hegde * extended part.
8087fc5d2a4SVikram Hegde */
809af28f636SEnrico Perla - Sun Microsystems if ((rval = libfdisk_init(&epp, device->path_p0, NULL, FDISK_READ_DISK))
8107fc5d2a4SVikram Hegde != FDISK_SUCCESS) {
8117fc5d2a4SVikram Hegde switch (rval) {
8127fc5d2a4SVikram Hegde /*
8136cb5747bSSharath M Srinivasan * The first 3 cases are not an error per-se, just that
8147fc5d2a4SVikram Hegde * there is no Solaris logical partition
8157fc5d2a4SVikram Hegde */
8167fc5d2a4SVikram Hegde case FDISK_EBADLOGDRIVE:
8177fc5d2a4SVikram Hegde case FDISK_ENOLOGDRIVE:
8186cb5747bSSharath M Srinivasan case FDISK_EBADMAGIC:
8197fc5d2a4SVikram Hegde (void) fprintf(stderr, NOSOLPAR);
820af28f636SEnrico Perla - Sun Microsystems return (BC_ERROR);
8217fc5d2a4SVikram Hegde case FDISK_ENOVGEOM:
8227fc5d2a4SVikram Hegde (void) fprintf(stderr, NO_VIRT_GEOM);
823af28f636SEnrico Perla - Sun Microsystems return (BC_ERROR);
8247fc5d2a4SVikram Hegde case FDISK_ENOPGEOM:
8257fc5d2a4SVikram Hegde (void) fprintf(stderr, NO_PHYS_GEOM);
826af28f636SEnrico Perla - Sun Microsystems return (BC_ERROR);
8277fc5d2a4SVikram Hegde case FDISK_ENOLGEOM:
8287fc5d2a4SVikram Hegde (void) fprintf(stderr, NO_LABEL_GEOM);
829af28f636SEnrico Perla - Sun Microsystems return (BC_ERROR);
8307fc5d2a4SVikram Hegde default:
8317fc5d2a4SVikram Hegde (void) fprintf(stderr, LIBFDISK_INIT_FAIL);
832af28f636SEnrico Perla - Sun Microsystems return (BC_ERROR);
8337fc5d2a4SVikram Hegde }
8347fc5d2a4SVikram Hegde }
8357fc5d2a4SVikram Hegde
8367fc5d2a4SVikram Hegde rval = fdisk_get_solaris_part(epp, &pno, &secnum, &numsec);
8376cb5747bSSharath M Srinivasan libfdisk_fini(&epp);
8387fc5d2a4SVikram Hegde if (rval != FDISK_SUCCESS) {
8397fc5d2a4SVikram Hegde /* No solaris logical partition */
8407fc5d2a4SVikram Hegde (void) fprintf(stderr, NOSOLPAR);
841af28f636SEnrico Perla - Sun Microsystems return (BC_ERROR);
8427fc5d2a4SVikram Hegde }
8437fc5d2a4SVikram Hegde
844af28f636SEnrico Perla - Sun Microsystems device->start_sector = secnum;
845af28f636SEnrico Perla - Sun Microsystems device->partition = pno - 1;
8467fc5d2a4SVikram Hegde log_part = 1;
8477fc5d2a4SVikram Hegde
8487fc5d2a4SVikram Hegde found_part:
8497c478bd9Sstevel@tonic-gate /* get confirmation for -m */
850af28f636SEnrico Perla - Sun Microsystems if (write_mbr && !force_mbr) {
8517c478bd9Sstevel@tonic-gate (void) fprintf(stdout, MBOOT_PROMPT);
8527c478bd9Sstevel@tonic-gate if (getchar() != 'y') {
853af28f636SEnrico Perla - Sun Microsystems write_mbr = 0;
8547c478bd9Sstevel@tonic-gate (void) fprintf(stdout, MBOOT_NOT_UPDATED);
855af28f636SEnrico Perla - Sun Microsystems return (BC_ERROR);
8567c478bd9Sstevel@tonic-gate }
8577c478bd9Sstevel@tonic-gate }
8587c478bd9Sstevel@tonic-gate
8597fc5d2a4SVikram Hegde /*
8607fc5d2a4SVikram Hegde * Currently if Solaris is in an extended partition we need to
8617fc5d2a4SVikram Hegde * write GRUB to the MBR. Check for this.
8627fc5d2a4SVikram Hegde */
863af28f636SEnrico Perla - Sun Microsystems if (log_part && !write_mbr) {
864af28f636SEnrico Perla - Sun Microsystems (void) fprintf(stdout, gettext("Installing Solaris on an "
865af28f636SEnrico Perla - Sun Microsystems "extended partition... forcing MBR update\n"));
866af28f636SEnrico Perla - Sun Microsystems write_mbr = 1;
867aa1b14e7SSheshadri Vasudevan }
868aa1b14e7SSheshadri Vasudevan
8697fc5d2a4SVikram Hegde /*
8707fc5d2a4SVikram Hegde * warn, if Solaris in primary partition and GRUB not in MBR and
8717fc5d2a4SVikram Hegde * partition is not active
8727fc5d2a4SVikram Hegde */
873af28f636SEnrico Perla - Sun Microsystems if (!log_part && part->bootid != 128 && !write_mbr) {
874af28f636SEnrico Perla - Sun Microsystems (void) fprintf(stdout, SOLPAR_INACTIVE, device->partition + 1);
8757c478bd9Sstevel@tonic-gate }
8767c478bd9Sstevel@tonic-gate
877af28f636SEnrico Perla - Sun Microsystems return (BC_SUCCESS);
8787c478bd9Sstevel@tonic-gate }
8797c478bd9Sstevel@tonic-gate
880af28f636SEnrico Perla - Sun Microsystems static int
get_disk_fd(ig_device_t * device)881af28f636SEnrico Perla - Sun Microsystems get_disk_fd(ig_device_t *device)
882af28f636SEnrico Perla - Sun Microsystems {
883af28f636SEnrico Perla - Sun Microsystems int i;
884af28f636SEnrico Perla - Sun Microsystems char save[2];
885af28f636SEnrico Perla - Sun Microsystems char *end = NULL;
886af28f636SEnrico Perla - Sun Microsystems
887af28f636SEnrico Perla - Sun Microsystems assert(device != NULL);
888af28f636SEnrico Perla - Sun Microsystems assert(device->path != NULL);
889af28f636SEnrico Perla - Sun Microsystems
890af28f636SEnrico Perla - Sun Microsystems if (is_bootpar(device->type)) {
891af28f636SEnrico Perla - Sun Microsystems end = strstr(device->path, "p0:boot");
892af28f636SEnrico Perla - Sun Microsystems /* tested at the start of init_device() */
893af28f636SEnrico Perla - Sun Microsystems assert(end != NULL);
894af28f636SEnrico Perla - Sun Microsystems /* chop off :boot */
895af28f636SEnrico Perla - Sun Microsystems save[0] = end[2];
896af28f636SEnrico Perla - Sun Microsystems end[2] = '\0';
897af28f636SEnrico Perla - Sun Microsystems } else {
898af28f636SEnrico Perla - Sun Microsystems i = strlen(device->path);
899af28f636SEnrico Perla - Sun Microsystems save[0] = device->path[i - 2];
900af28f636SEnrico Perla - Sun Microsystems save[1] = device->path[i - 1];
901af28f636SEnrico Perla - Sun Microsystems device->path[i - 2] = 'p';
902af28f636SEnrico Perla - Sun Microsystems device->path[i - 1] = '0';
903af28f636SEnrico Perla - Sun Microsystems }
904af28f636SEnrico Perla - Sun Microsystems
905af28f636SEnrico Perla - Sun Microsystems if (nowrite)
906af28f636SEnrico Perla - Sun Microsystems device->disk_fd = open(device->path, O_RDONLY);
907af28f636SEnrico Perla - Sun Microsystems else
908af28f636SEnrico Perla - Sun Microsystems device->disk_fd = open(device->path, O_RDWR);
909af28f636SEnrico Perla - Sun Microsystems
910af28f636SEnrico Perla - Sun Microsystems device->path_p0 = strdup(device->path);
911af28f636SEnrico Perla - Sun Microsystems if (device->path_p0 == NULL) {
912af28f636SEnrico Perla - Sun Microsystems perror("strdup");
913af28f636SEnrico Perla - Sun Microsystems return (BC_ERROR);
914af28f636SEnrico Perla - Sun Microsystems }
915af28f636SEnrico Perla - Sun Microsystems
916af28f636SEnrico Perla - Sun Microsystems if (is_bootpar(device->type)) {
917af28f636SEnrico Perla - Sun Microsystems end[2] = save[0];
918af28f636SEnrico Perla - Sun Microsystems } else {
919af28f636SEnrico Perla - Sun Microsystems device->path[i - 2] = save[0];
920af28f636SEnrico Perla - Sun Microsystems device->path[i - 1] = save[1];
921af28f636SEnrico Perla - Sun Microsystems }
922af28f636SEnrico Perla - Sun Microsystems
923af28f636SEnrico Perla - Sun Microsystems if (device->disk_fd == -1) {
924af28f636SEnrico Perla - Sun Microsystems perror("open");
925af28f636SEnrico Perla - Sun Microsystems return (BC_ERROR);
926af28f636SEnrico Perla - Sun Microsystems }
927af28f636SEnrico Perla - Sun Microsystems
928af28f636SEnrico Perla - Sun Microsystems return (BC_SUCCESS);
929af28f636SEnrico Perla - Sun Microsystems }
930af28f636SEnrico Perla - Sun Microsystems
931af28f636SEnrico Perla - Sun Microsystems static void
prepare_fake_multiboot(ig_stage2_t * stage2)932af28f636SEnrico Perla - Sun Microsystems prepare_fake_multiboot(ig_stage2_t *stage2)
933af28f636SEnrico Perla - Sun Microsystems {
934af28f636SEnrico Perla - Sun Microsystems multiboot_header_t *mboot;
935af28f636SEnrico Perla - Sun Microsystems
936af28f636SEnrico Perla - Sun Microsystems assert(stage2 != NULL);
937af28f636SEnrico Perla - Sun Microsystems assert(stage2->mboot != NULL);
938af28f636SEnrico Perla - Sun Microsystems assert(stage2->buf != NULL);
939af28f636SEnrico Perla - Sun Microsystems
940af28f636SEnrico Perla - Sun Microsystems mboot = stage2->mboot;
941af28f636SEnrico Perla - Sun Microsystems
942af28f636SEnrico Perla - Sun Microsystems /*
943af28f636SEnrico Perla - Sun Microsystems * Currently we expect find_multiboot() to have located a multiboot
944af28f636SEnrico Perla - Sun Microsystems * header with the AOUT kludge flag set.
945af28f636SEnrico Perla - Sun Microsystems */
946af28f636SEnrico Perla - Sun Microsystems assert(mboot->flags & BB_MBOOT_AOUT_FLAG);
947af28f636SEnrico Perla - Sun Microsystems
948af28f636SEnrico Perla - Sun Microsystems /* Insert the information necessary to locate stage2. */
949af28f636SEnrico Perla - Sun Microsystems mboot->header_addr = stage2->mboot_off;
950af28f636SEnrico Perla - Sun Microsystems mboot->load_addr = 0;
951af28f636SEnrico Perla - Sun Microsystems mboot->load_end_addr = stage2->file_size;
952af28f636SEnrico Perla - Sun Microsystems }
953af28f636SEnrico Perla - Sun Microsystems
954af28f636SEnrico Perla - Sun Microsystems static void
add_stage2_einfo(ig_stage2_t * stage2,char * updt_str)955af28f636SEnrico Perla - Sun Microsystems add_stage2_einfo(ig_stage2_t *stage2, char *updt_str)
956af28f636SEnrico Perla - Sun Microsystems {
957af28f636SEnrico Perla - Sun Microsystems bblk_hs_t hs;
958af28f636SEnrico Perla - Sun Microsystems uint32_t avail_space;
959af28f636SEnrico Perla - Sun Microsystems
960af28f636SEnrico Perla - Sun Microsystems assert(stage2 != NULL);
961af28f636SEnrico Perla - Sun Microsystems
962af28f636SEnrico Perla - Sun Microsystems /* Fill bootblock hashing source information. */
963af28f636SEnrico Perla - Sun Microsystems hs.src_buf = (unsigned char *)stage2->file;
964af28f636SEnrico Perla - Sun Microsystems hs.src_size = stage2->file_size;
965af28f636SEnrico Perla - Sun Microsystems /* How much space for the extended information structure? */
966af28f636SEnrico Perla - Sun Microsystems avail_space = stage2->buf_size - P2ROUNDUP(stage2->file_size, 8);
967af28f636SEnrico Perla - Sun Microsystems add_einfo(stage2->extra, updt_str, &hs, avail_space);
968af28f636SEnrico Perla - Sun Microsystems }
969af28f636SEnrico Perla - Sun Microsystems
970af28f636SEnrico Perla - Sun Microsystems
971af28f636SEnrico Perla - Sun Microsystems static int
write_stage2(ig_data_t * install)972af28f636SEnrico Perla - Sun Microsystems write_stage2(ig_data_t *install)
973af28f636SEnrico Perla - Sun Microsystems {
974af28f636SEnrico Perla - Sun Microsystems ig_device_t *device = &install->device;
975af28f636SEnrico Perla - Sun Microsystems ig_stage2_t *stage2 = &install->stage2;
976af28f636SEnrico Perla - Sun Microsystems off_t offset;
977af28f636SEnrico Perla - Sun Microsystems
978af28f636SEnrico Perla - Sun Microsystems assert(install != NULL);
979af28f636SEnrico Perla - Sun Microsystems
980af28f636SEnrico Perla - Sun Microsystems if (is_bootpar(device->type)) {
981af28f636SEnrico Perla - Sun Microsystems /*
982af28f636SEnrico Perla - Sun Microsystems * stage2 is already on the filesystem, we only need to update
983af28f636SEnrico Perla - Sun Microsystems * the first two blocks (that we have modified during
984af28f636SEnrico Perla - Sun Microsystems * prepare_stage2())
985af28f636SEnrico Perla - Sun Microsystems */
986af28f636SEnrico Perla - Sun Microsystems if (write_out(device->part_fd, stage2->file, SECTOR_SIZE,
987af28f636SEnrico Perla - Sun Microsystems stage2->pcfs_first_sectors[0] * SECTOR_SIZE)
988af28f636SEnrico Perla - Sun Microsystems != BC_SUCCESS ||
989af28f636SEnrico Perla - Sun Microsystems write_out(device->part_fd, stage2->file + SECTOR_SIZE,
990af28f636SEnrico Perla - Sun Microsystems SECTOR_SIZE, stage2->pcfs_first_sectors[1] * SECTOR_SIZE)
991af28f636SEnrico Perla - Sun Microsystems != BC_SUCCESS) {
992af28f636SEnrico Perla - Sun Microsystems (void) fprintf(stderr, WRITE_FAIL_STAGE2);
993af28f636SEnrico Perla - Sun Microsystems return (BC_ERROR);
994af28f636SEnrico Perla - Sun Microsystems }
995af28f636SEnrico Perla - Sun Microsystems (void) fprintf(stdout, WRITE_STAGE2_PCFS);
996af28f636SEnrico Perla - Sun Microsystems return (BC_SUCCESS);
997af28f636SEnrico Perla - Sun Microsystems }
998af28f636SEnrico Perla - Sun Microsystems
999af28f636SEnrico Perla - Sun Microsystems /*
1000af28f636SEnrico Perla - Sun Microsystems * For disk, write stage2 starting at STAGE2_BLKOFF sector.
1001af28f636SEnrico Perla - Sun Microsystems * Note that we use stage2->buf rather than stage2->file, because we
1002af28f636SEnrico Perla - Sun Microsystems * may have extended information after the latter.
1003*9348d232SHans Rosenfeld *
1004*9348d232SHans Rosenfeld * If we're writing to an EFI-labeled disk where stage2 lives in the
1005*9348d232SHans Rosenfeld * 3.5MB boot loader gap following the ZFS vdev labels, make sure the
1006*9348d232SHans Rosenfeld * size of the buffer doesn't exceed the size of the gap.
1007af28f636SEnrico Perla - Sun Microsystems */
1008*9348d232SHans Rosenfeld if (is_efi(device->type) && stage2->buf_size > STAGE2_MAXSIZE) {
1009*9348d232SHans Rosenfeld (void) fprintf(stderr, WRITE_FAIL_STAGE2);
1010*9348d232SHans Rosenfeld return (BC_ERROR);
1011*9348d232SHans Rosenfeld }
1012*9348d232SHans Rosenfeld
10131a902ef8SHans Rosenfeld offset = STAGE2_BLKOFF(device->type) * SECTOR_SIZE;
10141a902ef8SHans Rosenfeld
1015af28f636SEnrico Perla - Sun Microsystems if (write_out(device->part_fd, stage2->buf, stage2->buf_size,
1016af28f636SEnrico Perla - Sun Microsystems offset) != BC_SUCCESS) {
1017af28f636SEnrico Perla - Sun Microsystems perror("write");
1018af28f636SEnrico Perla - Sun Microsystems return (BC_ERROR);
1019af28f636SEnrico Perla - Sun Microsystems }
1020af28f636SEnrico Perla - Sun Microsystems
1021af28f636SEnrico Perla - Sun Microsystems /* Simulate the "old" installgrub output. */
1022af28f636SEnrico Perla - Sun Microsystems (void) fprintf(stdout, WRITE_STAGE2_DISK, device->partition,
10231a902ef8SHans Rosenfeld (stage2->buf_size / SECTOR_SIZE) + 1, STAGE2_BLKOFF(device->type),
1024af28f636SEnrico Perla - Sun Microsystems stage2->first_sector);
1025af28f636SEnrico Perla - Sun Microsystems
1026af28f636SEnrico Perla - Sun Microsystems return (BC_SUCCESS);
1027af28f636SEnrico Perla - Sun Microsystems }
1028af28f636SEnrico Perla - Sun Microsystems
1029af28f636SEnrico Perla - Sun Microsystems static int
write_stage1(ig_data_t * install)1030af28f636SEnrico Perla - Sun Microsystems write_stage1(ig_data_t *install)
1031af28f636SEnrico Perla - Sun Microsystems {
1032af28f636SEnrico Perla - Sun Microsystems ig_device_t *device = &install->device;
1033af28f636SEnrico Perla - Sun Microsystems
1034af28f636SEnrico Perla - Sun Microsystems assert(install != NULL);
1035af28f636SEnrico Perla - Sun Microsystems
1036af28f636SEnrico Perla - Sun Microsystems if (write_out(device->part_fd, install->stage1_buf,
1037af28f636SEnrico Perla - Sun Microsystems sizeof (install->stage1_buf), 0) != BC_SUCCESS) {
1038af28f636SEnrico Perla - Sun Microsystems (void) fprintf(stdout, WRITE_FAIL_PBOOT);
1039af28f636SEnrico Perla - Sun Microsystems perror("write");
1040af28f636SEnrico Perla - Sun Microsystems return (BC_ERROR);
1041af28f636SEnrico Perla - Sun Microsystems }
1042af28f636SEnrico Perla - Sun Microsystems
1043af28f636SEnrico Perla - Sun Microsystems /* Simulate "old" installgrub output. */
1044af28f636SEnrico Perla - Sun Microsystems (void) fprintf(stdout, WRITE_PBOOT, device->partition,
1045af28f636SEnrico Perla - Sun Microsystems device->start_sector);
1046af28f636SEnrico Perla - Sun Microsystems
1047af28f636SEnrico Perla - Sun Microsystems if (write_mbr) {
1048af28f636SEnrico Perla - Sun Microsystems if (write_out(device->disk_fd, install->stage1_buf,
1049af28f636SEnrico Perla - Sun Microsystems sizeof (install->stage1_buf), 0) != BC_SUCCESS) {
1050af28f636SEnrico Perla - Sun Microsystems (void) fprintf(stdout, WRITE_FAIL_BOOTSEC);
1051af28f636SEnrico Perla - Sun Microsystems perror("write");
1052af28f636SEnrico Perla - Sun Microsystems return (BC_ERROR);
1053af28f636SEnrico Perla - Sun Microsystems }
1054af28f636SEnrico Perla - Sun Microsystems /* Simulate "old" installgrub output. */
1055af28f636SEnrico Perla - Sun Microsystems (void) fprintf(stdout, WRITE_MBOOT);
1056af28f636SEnrico Perla - Sun Microsystems }
1057af28f636SEnrico Perla - Sun Microsystems
1058af28f636SEnrico Perla - Sun Microsystems return (BC_SUCCESS);
1059af28f636SEnrico Perla - Sun Microsystems }
1060af28f636SEnrico Perla - Sun Microsystems
1061af28f636SEnrico Perla - Sun Microsystems #define USAGE_STRING "%s [-m|-f|-n|-F|-u verstr] stage1 stage2 device\n" \
1062af28f636SEnrico Perla - Sun Microsystems "%s -M [-n] device1 device2\n" \
1063af28f636SEnrico Perla - Sun Microsystems "%s [-V|-e] -i device\n" \
1064af28f636SEnrico Perla - Sun Microsystems
1065af28f636SEnrico Perla - Sun Microsystems #define CANON_USAGE_STR gettext(USAGE_STRING)
1066af28f636SEnrico Perla - Sun Microsystems
10677c478bd9Sstevel@tonic-gate static void
usage(char * progname)10687c478bd9Sstevel@tonic-gate usage(char *progname)
10697c478bd9Sstevel@tonic-gate {
1070af28f636SEnrico Perla - Sun Microsystems (void) fprintf(stdout, CANON_USAGE_STR, progname, progname, progname);
10717c478bd9Sstevel@tonic-gate }
10727c478bd9Sstevel@tonic-gate
1073af28f636SEnrico Perla - Sun Microsystems
10747c478bd9Sstevel@tonic-gate static int
read_stage1_from_file(char * path,ig_data_t * dest)1075af28f636SEnrico Perla - Sun Microsystems read_stage1_from_file(char *path, ig_data_t *dest)
10767c478bd9Sstevel@tonic-gate {
10777c478bd9Sstevel@tonic-gate int fd;
10787c478bd9Sstevel@tonic-gate
1079af28f636SEnrico Perla - Sun Microsystems assert(dest);
1080af28f636SEnrico Perla - Sun Microsystems
10817c478bd9Sstevel@tonic-gate /* read the stage1 file from filesystem */
1082af28f636SEnrico Perla - Sun Microsystems fd = open(path, O_RDONLY);
1083af28f636SEnrico Perla - Sun Microsystems if (fd == -1 ||
1084af28f636SEnrico Perla - Sun Microsystems read(fd, dest->stage1_buf, SECTOR_SIZE) != SECTOR_SIZE) {
1085af28f636SEnrico Perla - Sun Microsystems (void) fprintf(stderr, READ_FAIL_STAGE1, path);
1086af28f636SEnrico Perla - Sun Microsystems return (BC_ERROR);
10877c478bd9Sstevel@tonic-gate }
10887c478bd9Sstevel@tonic-gate (void) close(fd);
1089af28f636SEnrico Perla - Sun Microsystems return (BC_SUCCESS);
10907c478bd9Sstevel@tonic-gate }
10917c478bd9Sstevel@tonic-gate
1092af28f636SEnrico Perla - Sun Microsystems static int
read_stage2_from_file(char * path,ig_data_t * dest)1093af28f636SEnrico Perla - Sun Microsystems read_stage2_from_file(char *path, ig_data_t *dest)
10947c478bd9Sstevel@tonic-gate {
1095af28f636SEnrico Perla - Sun Microsystems int fd;
1096af28f636SEnrico Perla - Sun Microsystems struct stat sb;
1097af28f636SEnrico Perla - Sun Microsystems ig_stage2_t *stage2 = &dest->stage2;
1098af28f636SEnrico Perla - Sun Microsystems ig_device_t *device = &dest->device;
1099af28f636SEnrico Perla - Sun Microsystems uint32_t buf_size;
1100af28f636SEnrico Perla - Sun Microsystems
1101af28f636SEnrico Perla - Sun Microsystems assert(dest);
1102af28f636SEnrico Perla - Sun Microsystems assert(stage2->buf == NULL);
1103af28f636SEnrico Perla - Sun Microsystems
1104af28f636SEnrico Perla - Sun Microsystems fd = open(path, O_RDONLY);
1105af28f636SEnrico Perla - Sun Microsystems if (fstat(fd, &sb) == -1) {
1106af28f636SEnrico Perla - Sun Microsystems perror("fstat");
1107af28f636SEnrico Perla - Sun Microsystems goto out;
11087c478bd9Sstevel@tonic-gate }
11097c478bd9Sstevel@tonic-gate
1110af28f636SEnrico Perla - Sun Microsystems stage2->file_size = sb.st_size;
11117c478bd9Sstevel@tonic-gate
1112af28f636SEnrico Perla - Sun Microsystems if (!is_bootpar(device->type)) {
1113af28f636SEnrico Perla - Sun Microsystems /*
1114af28f636SEnrico Perla - Sun Microsystems * buffer size needs to account for stage2 plus the extra
1115af28f636SEnrico Perla - Sun Microsystems * versioning information at the end of it. We reserve one
1116af28f636SEnrico Perla - Sun Microsystems * extra sector (plus we round up to the next sector boundary).
1117af28f636SEnrico Perla - Sun Microsystems */
1118af28f636SEnrico Perla - Sun Microsystems buf_size = stage2->file_size + SECTOR_SIZE;
11197c478bd9Sstevel@tonic-gate } else {
1120af28f636SEnrico Perla - Sun Microsystems /* In the PCFS case we only need to read in stage2. */
1121af28f636SEnrico Perla - Sun Microsystems buf_size = stage2->file_size;
11227c478bd9Sstevel@tonic-gate }
11237c478bd9Sstevel@tonic-gate
1124af28f636SEnrico Perla - Sun Microsystems stage2->buf_size = P2ROUNDUP(buf_size, SECTOR_SIZE);
1125af28f636SEnrico Perla - Sun Microsystems
1126af28f636SEnrico Perla - Sun Microsystems BOOT_DEBUG("stage2 buffer size = %d (%d sectors)\n", stage2->buf_size,
1127af28f636SEnrico Perla - Sun Microsystems stage2->buf_size / SECTOR_SIZE);
1128af28f636SEnrico Perla - Sun Microsystems
1129af28f636SEnrico Perla - Sun Microsystems stage2->buf = malloc(stage2->buf_size);
1130af28f636SEnrico Perla - Sun Microsystems if (stage2->buf == NULL) {
1131af28f636SEnrico Perla - Sun Microsystems perror(gettext("Memory allocation failed"));
1132af28f636SEnrico Perla - Sun Microsystems goto out_fd;
1133af28f636SEnrico Perla - Sun Microsystems }
1134af28f636SEnrico Perla - Sun Microsystems
1135af28f636SEnrico Perla - Sun Microsystems stage2->file = stage2->buf;
11367c478bd9Sstevel@tonic-gate
11377c478bd9Sstevel@tonic-gate /*
1138af28f636SEnrico Perla - Sun Microsystems * Extra information (e.g. the versioning structure) is placed at the
1139af28f636SEnrico Perla - Sun Microsystems * end of stage2, aligned on a 8-byte boundary.
11407c478bd9Sstevel@tonic-gate */
1141af28f636SEnrico Perla - Sun Microsystems if (!(is_bootpar(device->type)))
1142af28f636SEnrico Perla - Sun Microsystems stage2->extra = stage2->file + P2ROUNDUP(stage2->file_size, 8);
11437c478bd9Sstevel@tonic-gate
1144af28f636SEnrico Perla - Sun Microsystems if (lseek(fd, 0, SEEK_SET) == -1) {
1145af28f636SEnrico Perla - Sun Microsystems perror("lseek");
1146af28f636SEnrico Perla - Sun Microsystems goto out_alloc;
11477c478bd9Sstevel@tonic-gate }
11487c478bd9Sstevel@tonic-gate
1149af28f636SEnrico Perla - Sun Microsystems if (read(fd, stage2->file, stage2->file_size) < 0) {
1150af28f636SEnrico Perla - Sun Microsystems perror(gettext("unable to read stage2"));
1151af28f636SEnrico Perla - Sun Microsystems goto out_alloc;
11527c478bd9Sstevel@tonic-gate }
11537c478bd9Sstevel@tonic-gate
1154af28f636SEnrico Perla - Sun Microsystems (void) close(fd);
1155af28f636SEnrico Perla - Sun Microsystems return (BC_SUCCESS);
11567ce76caaSEnrico Perla - Sun Microsystems
1157af28f636SEnrico Perla - Sun Microsystems out_alloc:
1158af28f636SEnrico Perla - Sun Microsystems free(stage2->buf);
1159af28f636SEnrico Perla - Sun Microsystems stage2->buf = NULL;
1160af28f636SEnrico Perla - Sun Microsystems out_fd:
1161af28f636SEnrico Perla - Sun Microsystems (void) close(fd);
1162af28f636SEnrico Perla - Sun Microsystems out:
1163af28f636SEnrico Perla - Sun Microsystems return (BC_ERROR);
11647ce76caaSEnrico Perla - Sun Microsystems }
11657ce76caaSEnrico Perla - Sun Microsystems
11667ce76caaSEnrico Perla - Sun Microsystems static int
prepare_stage1(ig_data_t * install)1167af28f636SEnrico Perla - Sun Microsystems prepare_stage1(ig_data_t *install)
11687ce76caaSEnrico Perla - Sun Microsystems {
1169af28f636SEnrico Perla - Sun Microsystems ig_device_t *device = &install->device;
11707ce76caaSEnrico Perla - Sun Microsystems
1171af28f636SEnrico Perla - Sun Microsystems assert(install != NULL);
11727ce76caaSEnrico Perla - Sun Microsystems
1173af28f636SEnrico Perla - Sun Microsystems /* If PCFS add the BIOS Parameter Block. */
1174af28f636SEnrico Perla - Sun Microsystems if (is_bootpar(device->type)) {
1175af28f636SEnrico Perla - Sun Microsystems char bpb_sect[SECTOR_SIZE];
1176af28f636SEnrico Perla - Sun Microsystems
1177af28f636SEnrico Perla - Sun Microsystems if (pread(device->part_fd, bpb_sect, SECTOR_SIZE, 0)
1178af28f636SEnrico Perla - Sun Microsystems != SECTOR_SIZE) {
1179af28f636SEnrico Perla - Sun Microsystems (void) fprintf(stderr, READ_FAIL_BPB);
1180af28f636SEnrico Perla - Sun Microsystems return (BC_ERROR);
1181af28f636SEnrico Perla - Sun Microsystems }
1182af28f636SEnrico Perla - Sun Microsystems bcopy(bpb_sect + STAGE1_BPB_OFFSET,
1183af28f636SEnrico Perla - Sun Microsystems install->stage1_buf + STAGE1_BPB_OFFSET, STAGE1_BPB_SIZE);
11847ce76caaSEnrico Perla - Sun Microsystems }
11857ce76caaSEnrico Perla - Sun Microsystems
1186af28f636SEnrico Perla - Sun Microsystems /* copy MBR to stage1 in case of overwriting MBR sector. */
1187af28f636SEnrico Perla - Sun Microsystems bcopy(device->boot_sector + BOOTSZ, install->stage1_buf + BOOTSZ,
1188af28f636SEnrico Perla - Sun Microsystems SECTOR_SIZE - BOOTSZ);
1189af28f636SEnrico Perla - Sun Microsystems /* modify default stage1 file generated by GRUB. */
1190af28f636SEnrico Perla - Sun Microsystems *((unsigned char *)(install->stage1_buf + STAGE1_FORCE_LBA)) = 1;
1191af28f636SEnrico Perla - Sun Microsystems *((ulong_t *)(install->stage1_buf + STAGE1_STAGE2_SECTOR))
1192af28f636SEnrico Perla - Sun Microsystems = install->stage2.first_sector;
1193af28f636SEnrico Perla - Sun Microsystems *((ushort_t *)(install->stage1_buf + STAGE1_STAGE2_ADDRESS))
1194af28f636SEnrico Perla - Sun Microsystems = STAGE2_MEMADDR;
1195af28f636SEnrico Perla - Sun Microsystems *((ushort_t *)(install->stage1_buf + STAGE1_STAGE2_SEGMENT))
1196af28f636SEnrico Perla - Sun Microsystems = STAGE2_MEMADDR >> 4;
11977ce76caaSEnrico Perla - Sun Microsystems
1198af28f636SEnrico Perla - Sun Microsystems return (BC_SUCCESS);
11997ce76caaSEnrico Perla - Sun Microsystems }
12007ce76caaSEnrico Perla - Sun Microsystems
1201af28f636SEnrico Perla - Sun Microsystems /*
1202af28f636SEnrico Perla - Sun Microsystems * Grab stage1 from the specified device file descriptor.
1203af28f636SEnrico Perla - Sun Microsystems */
1204af28f636SEnrico Perla - Sun Microsystems static int
read_stage1_from_disk(int dev_fd,char * stage1_buf)1205af28f636SEnrico Perla - Sun Microsystems read_stage1_from_disk(int dev_fd, char *stage1_buf)
1206af28f636SEnrico Perla - Sun Microsystems {
1207af28f636SEnrico Perla - Sun Microsystems assert(stage1_buf != NULL);
12087ce76caaSEnrico Perla - Sun Microsystems
1209af28f636SEnrico Perla - Sun Microsystems if (read_in(dev_fd, stage1_buf, SECTOR_SIZE, 0) != BC_SUCCESS) {
1210af28f636SEnrico Perla - Sun Microsystems perror(gettext("Unable to read stage1 from disk"));
1211af28f636SEnrico Perla - Sun Microsystems return (BC_ERROR);
12127ce76caaSEnrico Perla - Sun Microsystems }
1213af28f636SEnrico Perla - Sun Microsystems return (BC_SUCCESS);
12147ce76caaSEnrico Perla - Sun Microsystems }
12157ce76caaSEnrico Perla - Sun Microsystems
12167ce76caaSEnrico Perla - Sun Microsystems static int
read_stage2_from_disk(int dev_fd,ig_stage2_t * stage2,int type)12171a902ef8SHans Rosenfeld read_stage2_from_disk(int dev_fd, ig_stage2_t *stage2, int type)
12187ce76caaSEnrico Perla - Sun Microsystems {
1219af28f636SEnrico Perla - Sun Microsystems uint32_t size;
1220af28f636SEnrico Perla - Sun Microsystems uint32_t buf_size;
1221af28f636SEnrico Perla - Sun Microsystems uint32_t mboot_off;
1222af28f636SEnrico Perla - Sun Microsystems multiboot_header_t *mboot;
12237ce76caaSEnrico Perla - Sun Microsystems
1224af28f636SEnrico Perla - Sun Microsystems assert(stage2 != NULL);
1225af28f636SEnrico Perla - Sun Microsystems assert(dev_fd != -1);
12267ce76caaSEnrico Perla - Sun Microsystems
1227af28f636SEnrico Perla - Sun Microsystems if (read_in(dev_fd, mboot_scan, sizeof (mboot_scan),
12281a902ef8SHans Rosenfeld STAGE2_BLKOFF(type) * SECTOR_SIZE) != BC_SUCCESS) {
1229af28f636SEnrico Perla - Sun Microsystems perror(gettext("Error reading stage2 sectors"));
1230af28f636SEnrico Perla - Sun Microsystems return (BC_ERROR);
1231af28f636SEnrico Perla - Sun Microsystems }
12327ce76caaSEnrico Perla - Sun Microsystems
1233af28f636SEnrico Perla - Sun Microsystems /* No multiboot means no chance of knowing stage2 size */
1234af28f636SEnrico Perla - Sun Microsystems if (find_multiboot(mboot_scan, sizeof (mboot_scan), &mboot_off)
1235af28f636SEnrico Perla - Sun Microsystems != BC_SUCCESS) {
1236af28f636SEnrico Perla - Sun Microsystems BOOT_DEBUG("Unable to find multiboot header\n");
1237af28f636SEnrico Perla - Sun Microsystems return (BC_NOEXTRA);
1238af28f636SEnrico Perla - Sun Microsystems }
1239af28f636SEnrico Perla - Sun Microsystems mboot = (multiboot_header_t *)(mboot_scan + mboot_off);
12407ce76caaSEnrico Perla - Sun Microsystems
1241af28f636SEnrico Perla - Sun Microsystems /*
1242af28f636SEnrico Perla - Sun Microsystems * Unfilled mboot values mean an older version of installgrub installed
1243af28f636SEnrico Perla - Sun Microsystems * the stage2. Again we have no chance of knowing stage2 size.
1244af28f636SEnrico Perla - Sun Microsystems */
1245af28f636SEnrico Perla - Sun Microsystems if (mboot->load_end_addr == 0 ||
1246af28f636SEnrico Perla - Sun Microsystems mboot->load_end_addr < mboot->load_addr)
1247af28f636SEnrico Perla - Sun Microsystems return (BC_NOEXTRA);
1248af28f636SEnrico Perla - Sun Microsystems
1249af28f636SEnrico Perla - Sun Microsystems /*
1250af28f636SEnrico Perla - Sun Microsystems * Currently, the amount of space reserved for extra information
1251af28f636SEnrico Perla - Sun Microsystems * is "fixed". We may have to scan for the terminating extra payload
1252af28f636SEnrico Perla - Sun Microsystems * in the future.
1253af28f636SEnrico Perla - Sun Microsystems */
1254af28f636SEnrico Perla - Sun Microsystems size = mboot->load_end_addr - mboot->load_addr;
1255af28f636SEnrico Perla - Sun Microsystems buf_size = P2ROUNDUP(size + SECTOR_SIZE, SECTOR_SIZE);
1256af28f636SEnrico Perla - Sun Microsystems
1257af28f636SEnrico Perla - Sun Microsystems stage2->buf = malloc(buf_size);
1258af28f636SEnrico Perla - Sun Microsystems if (stage2->buf == NULL) {
1259af28f636SEnrico Perla - Sun Microsystems perror(gettext("Memory allocation failed"));
1260af28f636SEnrico Perla - Sun Microsystems return (BC_ERROR);
1261af28f636SEnrico Perla - Sun Microsystems }
1262af28f636SEnrico Perla - Sun Microsystems stage2->buf_size = buf_size;
1263af28f636SEnrico Perla - Sun Microsystems
12641a902ef8SHans Rosenfeld if (read_in(dev_fd, stage2->buf, buf_size, STAGE2_BLKOFF(type) *
1265af28f636SEnrico Perla - Sun Microsystems SECTOR_SIZE) != BC_SUCCESS) {
1266af28f636SEnrico Perla - Sun Microsystems perror("read");
1267af28f636SEnrico Perla - Sun Microsystems free(stage2->buf);
1268af28f636SEnrico Perla - Sun Microsystems return (BC_ERROR);
1269af28f636SEnrico Perla - Sun Microsystems }
1270af28f636SEnrico Perla - Sun Microsystems
1271af28f636SEnrico Perla - Sun Microsystems /* Update pointers. */
1272af28f636SEnrico Perla - Sun Microsystems stage2->file = stage2->buf;
1273af28f636SEnrico Perla - Sun Microsystems stage2->file_size = size;
1274af28f636SEnrico Perla - Sun Microsystems stage2->mboot_off = mboot_off;
1275af28f636SEnrico Perla - Sun Microsystems stage2->mboot = (multiboot_header_t *)(stage2->buf + stage2->mboot_off);
1276af28f636SEnrico Perla - Sun Microsystems stage2->extra = stage2->buf + P2ROUNDUP(stage2->file_size, 8);
127714d44f22SHans Rosenfeld stage2->extra_size = stage2->buf_size - P2ROUNDUP(stage2->file_size, 8);
1278af28f636SEnrico Perla - Sun Microsystems
1279af28f636SEnrico Perla - Sun Microsystems return (BC_SUCCESS);
1280af28f636SEnrico Perla - Sun Microsystems }
1281af28f636SEnrico Perla - Sun Microsystems
1282af28f636SEnrico Perla - Sun Microsystems static boolean_t
is_update_necessary(ig_data_t * data,char * updt_str)1283af28f636SEnrico Perla - Sun Microsystems is_update_necessary(ig_data_t *data, char *updt_str)
1284af28f636SEnrico Perla - Sun Microsystems {
1285af28f636SEnrico Perla - Sun Microsystems bblk_einfo_t *einfo;
1286af28f636SEnrico Perla - Sun Microsystems bblk_hs_t stage2_hs;
1287af28f636SEnrico Perla - Sun Microsystems ig_stage2_t stage2_disk;
1288af28f636SEnrico Perla - Sun Microsystems ig_stage2_t *stage2_file = &data->stage2;
1289af28f636SEnrico Perla - Sun Microsystems ig_device_t *device = &data->device;
1290af28f636SEnrico Perla - Sun Microsystems int dev_fd = device->part_fd;
1291af28f636SEnrico Perla - Sun Microsystems
1292af28f636SEnrico Perla - Sun Microsystems assert(data != NULL);
1293af28f636SEnrico Perla - Sun Microsystems assert(device->part_fd != -1);
1294af28f636SEnrico Perla - Sun Microsystems
1295af28f636SEnrico Perla - Sun Microsystems bzero(&stage2_disk, sizeof (ig_stage2_t));
1296af28f636SEnrico Perla - Sun Microsystems
1297af28f636SEnrico Perla - Sun Microsystems /* Gather stage2 (if present) from the target device. */
12981a902ef8SHans Rosenfeld if (read_stage2_from_disk(dev_fd, &stage2_disk, device->type)
12991a902ef8SHans Rosenfeld != BC_SUCCESS) {
1300af28f636SEnrico Perla - Sun Microsystems BOOT_DEBUG("Unable to read stage2 from %s\n", device->path);
1301af28f636SEnrico Perla - Sun Microsystems BOOT_DEBUG("No multiboot wrapped stage2 on %s\n", device->path);
1302af28f636SEnrico Perla - Sun Microsystems return (B_TRUE);
1303af28f636SEnrico Perla - Sun Microsystems }
1304af28f636SEnrico Perla - Sun Microsystems
1305af28f636SEnrico Perla - Sun Microsystems /*
1306af28f636SEnrico Perla - Sun Microsystems * Look for the extended information structure in the extra payload
1307af28f636SEnrico Perla - Sun Microsystems * area.
1308af28f636SEnrico Perla - Sun Microsystems */
130914d44f22SHans Rosenfeld einfo = find_einfo(stage2_disk.extra, stage2_disk.extra_size);
1310af28f636SEnrico Perla - Sun Microsystems if (einfo == NULL) {
1311af28f636SEnrico Perla - Sun Microsystems BOOT_DEBUG("No extended information available\n");
1312af28f636SEnrico Perla - Sun Microsystems return (B_TRUE);
1313af28f636SEnrico Perla - Sun Microsystems }
1314af28f636SEnrico Perla - Sun Microsystems
1315af28f636SEnrico Perla - Sun Microsystems if (!do_version || updt_str == NULL) {
1316af28f636SEnrico Perla - Sun Microsystems (void) fprintf(stdout, "WARNING: target device %s has a "
1317af28f636SEnrico Perla - Sun Microsystems "versioned stage2 that is going to be overwritten by a non "
1318af28f636SEnrico Perla - Sun Microsystems "versioned one\n", device->path);
1319af28f636SEnrico Perla - Sun Microsystems return (B_TRUE);
1320af28f636SEnrico Perla - Sun Microsystems }
1321af28f636SEnrico Perla - Sun Microsystems
1322af28f636SEnrico Perla - Sun Microsystems if (force_update) {
1323af28f636SEnrico Perla - Sun Microsystems BOOT_DEBUG("Forcing update of %s bootblock\n", device->path);
1324af28f636SEnrico Perla - Sun Microsystems return (B_TRUE);
1325af28f636SEnrico Perla - Sun Microsystems }
1326af28f636SEnrico Perla - Sun Microsystems
1327af28f636SEnrico Perla - Sun Microsystems /* Compare the two extended information structures. */
1328af28f636SEnrico Perla - Sun Microsystems stage2_hs.src_buf = (unsigned char *)stage2_file->file;
1329af28f636SEnrico Perla - Sun Microsystems stage2_hs.src_size = stage2_file->file_size;
1330af28f636SEnrico Perla - Sun Microsystems
1331af28f636SEnrico Perla - Sun Microsystems return (einfo_should_update(einfo, &stage2_hs, updt_str));
13327ce76caaSEnrico Perla - Sun Microsystems }
13337ce76caaSEnrico Perla - Sun Microsystems
13347ce76caaSEnrico Perla - Sun Microsystems
13357c478bd9Sstevel@tonic-gate #define START_BLOCK(pos) (*(ulong_t *)(pos))
13367c478bd9Sstevel@tonic-gate #define NUM_BLOCK(pos) (*(ushort_t *)((pos) + 4))
13377c478bd9Sstevel@tonic-gate #define START_SEG(pos) (*(ushort_t *)((pos) + 6))
13387c478bd9Sstevel@tonic-gate
1339af28f636SEnrico Perla - Sun Microsystems static int
prepare_stage2(ig_data_t * install,char * updt_str)1340af28f636SEnrico Perla - Sun Microsystems prepare_stage2(ig_data_t *install, char *updt_str)
13417c478bd9Sstevel@tonic-gate {
1342af28f636SEnrico Perla - Sun Microsystems ig_device_t *device = &install->device;
1343af28f636SEnrico Perla - Sun Microsystems ig_stage2_t *stage2 = &install->stage2;
1344af28f636SEnrico Perla - Sun Microsystems uint32_t mboot_off = 0;
13457ce76caaSEnrico Perla - Sun Microsystems
1346af28f636SEnrico Perla - Sun Microsystems assert(install != NULL);
1347af28f636SEnrico Perla - Sun Microsystems assert(stage2->file != NULL);
1348af28f636SEnrico Perla - Sun Microsystems
1349af28f636SEnrico Perla - Sun Microsystems /* New stage2 files come with an embedded stage2. */
1350af28f636SEnrico Perla - Sun Microsystems if (find_multiboot(stage2->file, stage2->file_size, &mboot_off)
1351af28f636SEnrico Perla - Sun Microsystems != BC_SUCCESS) {
1352af28f636SEnrico Perla - Sun Microsystems BOOT_DEBUG("WARNING: no multiboot structure found in stage2, "
1353af28f636SEnrico Perla - Sun Microsystems "are you using an old GRUB stage2?\n");
1354af28f636SEnrico Perla - Sun Microsystems if (do_version == B_TRUE) {
1355af28f636SEnrico Perla - Sun Microsystems (void) fprintf(stderr, gettext("Versioning requested "
1356af28f636SEnrico Perla - Sun Microsystems "but stage2 does not support it.. skipping.\n"));
1357af28f636SEnrico Perla - Sun Microsystems do_version = B_FALSE;
1358af28f636SEnrico Perla - Sun Microsystems }
1359af28f636SEnrico Perla - Sun Microsystems } else {
1360af28f636SEnrico Perla - Sun Microsystems /* Keep track of where the multiboot header is. */
1361af28f636SEnrico Perla - Sun Microsystems stage2->mboot_off = mboot_off;
1362af28f636SEnrico Perla - Sun Microsystems stage2->mboot = (multiboot_header_t *)(stage2->file +
1363af28f636SEnrico Perla - Sun Microsystems mboot_off);
13647ce76caaSEnrico Perla - Sun Microsystems if (do_version) {
1365af28f636SEnrico Perla - Sun Microsystems /*
1366af28f636SEnrico Perla - Sun Microsystems * Adding stage2 information needs to happen before
1367af28f636SEnrico Perla - Sun Microsystems * we modify the copy of stage2 we have in memory, so
1368af28f636SEnrico Perla - Sun Microsystems * that the hashing reflects the one of the file.
1369af28f636SEnrico Perla - Sun Microsystems * An error here is not fatal.
1370af28f636SEnrico Perla - Sun Microsystems */
1371af28f636SEnrico Perla - Sun Microsystems add_stage2_einfo(stage2, updt_str);
1372af28f636SEnrico Perla - Sun Microsystems }
1373af28f636SEnrico Perla - Sun Microsystems /*
1374af28f636SEnrico Perla - Sun Microsystems * Fill multiboot information. We add them even without
1375af28f636SEnrico Perla - Sun Microsystems * versioning to support as much as possible mirroring.
1376af28f636SEnrico Perla - Sun Microsystems */
1377af28f636SEnrico Perla - Sun Microsystems prepare_fake_multiboot(stage2);
13787ce76caaSEnrico Perla - Sun Microsystems }
13797c478bd9Sstevel@tonic-gate
1380af28f636SEnrico Perla - Sun Microsystems if (is_bootpar(device->type)) {
1381af28f636SEnrico Perla - Sun Microsystems uint32_t blocklist[SECTOR_SIZE / sizeof (uint32_t)];
1382af28f636SEnrico Perla - Sun Microsystems uint32_t install_addr = STAGE2_MEMADDR + SECTOR_SIZE;
13837c478bd9Sstevel@tonic-gate int i = 0;
1384af28f636SEnrico Perla - Sun Microsystems uchar_t *pos;
13857c478bd9Sstevel@tonic-gate
1386af28f636SEnrico Perla - Sun Microsystems bzero(blocklist, sizeof (blocklist));
1387af28f636SEnrico Perla - Sun Microsystems if (read_stage2_blocklist(device->part_fd, blocklist) != 0) {
1388af28f636SEnrico Perla - Sun Microsystems (void) fprintf(stderr, gettext("Error reading pcfs "
1389af28f636SEnrico Perla - Sun Microsystems "stage2 blocklist\n"));
1390af28f636SEnrico Perla - Sun Microsystems return (BC_ERROR);
1391af28f636SEnrico Perla - Sun Microsystems }
13927c478bd9Sstevel@tonic-gate
1393af28f636SEnrico Perla - Sun Microsystems pos = (uchar_t *)stage2->file + STAGE2_BLOCKLIST;
1394af28f636SEnrico Perla - Sun Microsystems stage2->first_sector = device->start_sector + blocklist[0];
1395af28f636SEnrico Perla - Sun Microsystems stage2->pcfs_first_sectors[0] = blocklist[0];
1396af28f636SEnrico Perla - Sun Microsystems BOOT_DEBUG("stage2 first sector: %d\n", stage2->first_sector);
1397af28f636SEnrico Perla - Sun Microsystems
1398af28f636SEnrico Perla - Sun Microsystems
13997c478bd9Sstevel@tonic-gate if (blocklist[1] > 1) {
14007c478bd9Sstevel@tonic-gate blocklist[0]++;
14017c478bd9Sstevel@tonic-gate blocklist[1]--;
14027c478bd9Sstevel@tonic-gate } else {
14037c478bd9Sstevel@tonic-gate i += 2;
14047c478bd9Sstevel@tonic-gate }
14057c478bd9Sstevel@tonic-gate
1406af28f636SEnrico Perla - Sun Microsystems stage2->pcfs_first_sectors[1] = blocklist[i];
14077c478bd9Sstevel@tonic-gate
14087c478bd9Sstevel@tonic-gate while (blocklist[i]) {
14097c478bd9Sstevel@tonic-gate if (START_BLOCK(pos - 8) != 0 &&
14107c478bd9Sstevel@tonic-gate START_BLOCK(pos - 8) != blocklist[i + 2]) {
14117c478bd9Sstevel@tonic-gate (void) fprintf(stderr, PCFS_FRAGMENTED);
1412af28f636SEnrico Perla - Sun Microsystems return (BC_ERROR);
14137c478bd9Sstevel@tonic-gate }
1414af28f636SEnrico Perla - Sun Microsystems START_BLOCK(pos) = blocklist[i] + device->start_sector;
14157c478bd9Sstevel@tonic-gate START_SEG(pos) = (ushort_t)(install_addr >> 4);
14167c478bd9Sstevel@tonic-gate NUM_BLOCK(pos) = blocklist[i + 1];
14177c478bd9Sstevel@tonic-gate install_addr += blocklist[i + 1] * SECTOR_SIZE;
14187c478bd9Sstevel@tonic-gate pos -= 8;
14197c478bd9Sstevel@tonic-gate i += 2;
14207c478bd9Sstevel@tonic-gate }
14217c478bd9Sstevel@tonic-gate } else {
1422af28f636SEnrico Perla - Sun Microsystems /* Solaris VTOC */
14231a902ef8SHans Rosenfeld stage2->first_sector = device->start_sector +
14241a902ef8SHans Rosenfeld STAGE2_BLKOFF(device->type);
1425af28f636SEnrico Perla - Sun Microsystems BOOT_DEBUG("stage2 first sector: %d\n", stage2->first_sector);
14267c478bd9Sstevel@tonic-gate /*
14277c478bd9Sstevel@tonic-gate * In a solaris partition, stage2 is written to contiguous
14287c478bd9Sstevel@tonic-gate * blocks. So we update the starting block only.
14297c478bd9Sstevel@tonic-gate */
1430af28f636SEnrico Perla - Sun Microsystems *((ulong_t *)(stage2->file + STAGE2_BLOCKLIST)) =
1431af28f636SEnrico Perla - Sun Microsystems stage2->first_sector + 1;
14327c478bd9Sstevel@tonic-gate }
14337c478bd9Sstevel@tonic-gate
14347c478bd9Sstevel@tonic-gate /* force lba and set disk partition */
1435af28f636SEnrico Perla - Sun Microsystems *((unsigned char *) (stage2->file + STAGE2_FORCE_LBA)) = 1;
1436af28f636SEnrico Perla - Sun Microsystems *((long *)(stage2->file + STAGE2_INSTALLPART))
1437af28f636SEnrico Perla - Sun Microsystems = (device->partition << 16) | (device->slice << 8) | 0xff;
1438af28f636SEnrico Perla - Sun Microsystems
1439af28f636SEnrico Perla - Sun Microsystems return (BC_SUCCESS);
14407c478bd9Sstevel@tonic-gate }
14417c478bd9Sstevel@tonic-gate
1442af28f636SEnrico Perla - Sun Microsystems static int
find_x86_bootpar(struct mboot * mboot,int * part_num,uint32_t * start_sect)1443af28f636SEnrico Perla - Sun Microsystems find_x86_bootpar(struct mboot *mboot, int *part_num, uint32_t *start_sect)
1444af28f636SEnrico Perla - Sun Microsystems {
1445af28f636SEnrico Perla - Sun Microsystems int i;
14467c478bd9Sstevel@tonic-gate
1447af28f636SEnrico Perla - Sun Microsystems for (i = 0; i < FD_NUMPART; i++) {
1448af28f636SEnrico Perla - Sun Microsystems struct ipart *part;
14497c478bd9Sstevel@tonic-gate
1450af28f636SEnrico Perla - Sun Microsystems part = (struct ipart *)mboot->parts + i;
1451af28f636SEnrico Perla - Sun Microsystems if (part->systid == 0xbe) {
1452af28f636SEnrico Perla - Sun Microsystems if (start_sect)
1453af28f636SEnrico Perla - Sun Microsystems *start_sect = part->relsect;
1454af28f636SEnrico Perla - Sun Microsystems if (part_num)
1455af28f636SEnrico Perla - Sun Microsystems *part_num = i;
1456af28f636SEnrico Perla - Sun Microsystems /* solaris boot part */
1457af28f636SEnrico Perla - Sun Microsystems return (BC_SUCCESS);
14587c478bd9Sstevel@tonic-gate }
14597c478bd9Sstevel@tonic-gate }
1460af28f636SEnrico Perla - Sun Microsystems return (BC_ERROR);
14617c478bd9Sstevel@tonic-gate }
14627c478bd9Sstevel@tonic-gate
14637c478bd9Sstevel@tonic-gate static char *
get_raw_partition_path(ig_device_t * device)1464af28f636SEnrico Perla - Sun Microsystems get_raw_partition_path(ig_device_t *device)
14657c478bd9Sstevel@tonic-gate {
1466af28f636SEnrico Perla - Sun Microsystems char *raw;
14677c478bd9Sstevel@tonic-gate int len;
1468af28f636SEnrico Perla - Sun Microsystems
1469af28f636SEnrico Perla - Sun Microsystems if (is_bootpar(device->type)) {
1470af28f636SEnrico Perla - Sun Microsystems int part;
14717c478bd9Sstevel@tonic-gate struct mboot *mboot;
14727c478bd9Sstevel@tonic-gate
1473af28f636SEnrico Perla - Sun Microsystems mboot = (struct mboot *)device->boot_sector;
1474af28f636SEnrico Perla - Sun Microsystems if (find_x86_bootpar(mboot, &part, NULL) != BC_SUCCESS) {
1475af28f636SEnrico Perla - Sun Microsystems (void) fprintf(stderr, BOOTPAR_NOTFOUND,
1476af28f636SEnrico Perla - Sun Microsystems device->path_p0);
1477af28f636SEnrico Perla - Sun Microsystems return (NULL);
14787c478bd9Sstevel@tonic-gate }
14797c478bd9Sstevel@tonic-gate
1480af28f636SEnrico Perla - Sun Microsystems raw = strdup(device->path_p0);
1481af28f636SEnrico Perla - Sun Microsystems if (raw == NULL) {
1482af28f636SEnrico Perla - Sun Microsystems perror(gettext("Memory allocation failed"));
1483af28f636SEnrico Perla - Sun Microsystems return (NULL);
14847c478bd9Sstevel@tonic-gate }
1485af28f636SEnrico Perla - Sun Microsystems
1486af28f636SEnrico Perla - Sun Microsystems raw[strlen(raw) - 2] = '1' + part;
14877c478bd9Sstevel@tonic-gate return (raw);
14887c478bd9Sstevel@tonic-gate }
14897c478bd9Sstevel@tonic-gate
14907c478bd9Sstevel@tonic-gate /* For disk, remember slice and return whole fdisk partition */
1491af28f636SEnrico Perla - Sun Microsystems raw = strdup(device->path);
1492af28f636SEnrico Perla - Sun Microsystems if (raw == NULL) {
1493af28f636SEnrico Perla - Sun Microsystems perror(gettext("Memory allocation failed"));
1494af28f636SEnrico Perla - Sun Microsystems return (NULL);
1495af28f636SEnrico Perla - Sun Microsystems }
1496af28f636SEnrico Perla - Sun Microsystems
14977c478bd9Sstevel@tonic-gate len = strlen(raw);
14981a902ef8SHans Rosenfeld if (!is_efi(device->type) &&
14991a902ef8SHans Rosenfeld (raw[len - 2] != 's' || raw[len - 1] == '2')) {
15007c478bd9Sstevel@tonic-gate (void) fprintf(stderr, NOT_ROOT_SLICE);
1501af28f636SEnrico Perla - Sun Microsystems free(raw);
1502af28f636SEnrico Perla - Sun Microsystems return (NULL);
15037c478bd9Sstevel@tonic-gate }
1504af28f636SEnrico Perla - Sun Microsystems device->slice = atoi(&raw[len - 1]);
15057c478bd9Sstevel@tonic-gate
15061a902ef8SHans Rosenfeld if (!is_efi(device->type)) {
15077c478bd9Sstevel@tonic-gate raw[len - 2] = 's';
15087c478bd9Sstevel@tonic-gate raw[len - 1] = '2';
15091a902ef8SHans Rosenfeld }
1510af28f636SEnrico Perla - Sun Microsystems
15117c478bd9Sstevel@tonic-gate return (raw);
15127c478bd9Sstevel@tonic-gate }
15137c478bd9Sstevel@tonic-gate
1514af28f636SEnrico Perla - Sun Microsystems static int
get_raw_partition_fd(ig_device_t * device)1515af28f636SEnrico Perla - Sun Microsystems get_raw_partition_fd(ig_device_t *device)
1516af28f636SEnrico Perla - Sun Microsystems {
1517af28f636SEnrico Perla - Sun Microsystems struct stat stat = {0};
1518af28f636SEnrico Perla - Sun Microsystems char *raw;
1519af28f636SEnrico Perla - Sun Microsystems
1520af28f636SEnrico Perla - Sun Microsystems raw = get_raw_partition_path(device);
1521af28f636SEnrico Perla - Sun Microsystems if (raw == NULL)
1522af28f636SEnrico Perla - Sun Microsystems return (BC_ERROR);
1523af28f636SEnrico Perla - Sun Microsystems
1524af28f636SEnrico Perla - Sun Microsystems if (nowrite)
1525af28f636SEnrico Perla - Sun Microsystems device->part_fd = open(raw, O_RDONLY);
1526af28f636SEnrico Perla - Sun Microsystems else
1527af28f636SEnrico Perla - Sun Microsystems device->part_fd = open(raw, O_RDWR);
1528af28f636SEnrico Perla - Sun Microsystems
1529af28f636SEnrico Perla - Sun Microsystems if (device->part_fd < 0 || fstat(device->part_fd, &stat) != 0) {
1530af28f636SEnrico Perla - Sun Microsystems (void) fprintf(stderr, OPEN_FAIL, raw);
1531af28f636SEnrico Perla - Sun Microsystems free(raw);
1532af28f636SEnrico Perla - Sun Microsystems return (BC_ERROR);
1533af28f636SEnrico Perla - Sun Microsystems }
1534af28f636SEnrico Perla - Sun Microsystems
1535af28f636SEnrico Perla - Sun Microsystems if (S_ISCHR(stat.st_mode) == 0) {
1536af28f636SEnrico Perla - Sun Microsystems (void) fprintf(stderr, NOT_RAW_DEVICE, raw);
1537af28f636SEnrico Perla - Sun Microsystems (void) close(device->part_fd);
1538af28f636SEnrico Perla - Sun Microsystems device->part_fd = -1;
1539af28f636SEnrico Perla - Sun Microsystems free(raw);
1540af28f636SEnrico Perla - Sun Microsystems return (BC_ERROR);
1541af28f636SEnrico Perla - Sun Microsystems }
1542af28f636SEnrico Perla - Sun Microsystems
1543af28f636SEnrico Perla - Sun Microsystems free(raw);
1544af28f636SEnrico Perla - Sun Microsystems return (BC_SUCCESS);
1545af28f636SEnrico Perla - Sun Microsystems }
1546af28f636SEnrico Perla - Sun Microsystems
15477c478bd9Sstevel@tonic-gate #define TMP_MNTPT "/tmp/installgrub_pcfs"
1548af28f636SEnrico Perla - Sun Microsystems static int
copy_stage2_to_pcfs(ig_data_t * install)1549af28f636SEnrico Perla - Sun Microsystems copy_stage2_to_pcfs(ig_data_t *install)
15507c478bd9Sstevel@tonic-gate {
15517c478bd9Sstevel@tonic-gate FILE *mntfp;
1552af28f636SEnrico Perla - Sun Microsystems int pcfs_fp;
1553af28f636SEnrico Perla - Sun Microsystems int status = BC_ERROR;
15547c478bd9Sstevel@tonic-gate char buf[SECTOR_SIZE];
15557c478bd9Sstevel@tonic-gate char *cp;
15567c478bd9Sstevel@tonic-gate struct mnttab mp = {0}, mpref = {0};
1557af28f636SEnrico Perla - Sun Microsystems ig_device_t *device = &install->device;
1558af28f636SEnrico Perla - Sun Microsystems ig_stage2_t *stage2 = &install->stage2;
15597c478bd9Sstevel@tonic-gate
15607c478bd9Sstevel@tonic-gate /* convert raw to block device name by removing the first 'r' */
1561af28f636SEnrico Perla - Sun Microsystems (void) strncpy(buf, device->path, sizeof (buf));
15627c478bd9Sstevel@tonic-gate buf[sizeof (buf) - 1] = 0;
15637c478bd9Sstevel@tonic-gate cp = strchr(buf, 'r');
15647c478bd9Sstevel@tonic-gate if (cp == NULL) {
1565af28f636SEnrico Perla - Sun Microsystems (void) fprintf(stderr, CONVERT_FAIL, device->path);
1566af28f636SEnrico Perla - Sun Microsystems return (BC_ERROR);
15677c478bd9Sstevel@tonic-gate }
15687c478bd9Sstevel@tonic-gate do {
15697c478bd9Sstevel@tonic-gate *cp = *(cp + 1);
15707c478bd9Sstevel@tonic-gate } while (*(++cp));
15717c478bd9Sstevel@tonic-gate
15727c478bd9Sstevel@tonic-gate /* get the mount point, if any */
15737c478bd9Sstevel@tonic-gate mntfp = fopen("/etc/mnttab", "r");
15747c478bd9Sstevel@tonic-gate if (mntfp == NULL) {
15757c478bd9Sstevel@tonic-gate (void) fprintf(stderr, OPEN_FAIL_FILE, "/etc/mnttab");
1576af28f636SEnrico Perla - Sun Microsystems return (BC_ERROR);
15777c478bd9Sstevel@tonic-gate }
15787c478bd9Sstevel@tonic-gate
15797c478bd9Sstevel@tonic-gate mpref.mnt_special = buf;
15807c478bd9Sstevel@tonic-gate if (getmntany(mntfp, &mp, &mpref) != 0) {
15817c478bd9Sstevel@tonic-gate char cmd[128];
15827c478bd9Sstevel@tonic-gate
15837c478bd9Sstevel@tonic-gate /* not mounted, try remount */
15847c478bd9Sstevel@tonic-gate (void) mkdir(TMP_MNTPT, S_IRWXU);
15857c478bd9Sstevel@tonic-gate (void) snprintf(cmd, sizeof (cmd), "mount -F pcfs %s %s",
15867c478bd9Sstevel@tonic-gate buf, TMP_MNTPT);
15877c478bd9Sstevel@tonic-gate (void) system(cmd);
15887c478bd9Sstevel@tonic-gate rewind(mntfp);
15897c478bd9Sstevel@tonic-gate bzero(&mp, sizeof (mp));
15907c478bd9Sstevel@tonic-gate if (getmntany(mntfp, &mp, &mpref) != 0) {
15917c478bd9Sstevel@tonic-gate (void) fprintf(stderr, MOUNT_FAIL, buf);
1592af28f636SEnrico Perla - Sun Microsystems return (BC_ERROR);
15937c478bd9Sstevel@tonic-gate }
15947c478bd9Sstevel@tonic-gate }
15957c478bd9Sstevel@tonic-gate
15967c478bd9Sstevel@tonic-gate (void) snprintf(buf, sizeof (buf),
15977c478bd9Sstevel@tonic-gate "%s/boot", mp.mnt_mountp);
15987c478bd9Sstevel@tonic-gate (void) mkdir(buf, S_IRWXU);
15997c478bd9Sstevel@tonic-gate (void) strcat(buf, "/grub");
16007c478bd9Sstevel@tonic-gate (void) mkdir(buf, S_IRWXU);
16017c478bd9Sstevel@tonic-gate
16027c478bd9Sstevel@tonic-gate (void) strcat(buf, "/stage2");
16037c478bd9Sstevel@tonic-gate pcfs_fp = open(buf, O_WRONLY | O_CREAT, S_IRWXU);
16047c478bd9Sstevel@tonic-gate if (pcfs_fp == -1) {
16057c478bd9Sstevel@tonic-gate (void) fprintf(stderr, OPEN_FAIL_FILE, buf);
16067c478bd9Sstevel@tonic-gate perror("open:");
1607af28f636SEnrico Perla - Sun Microsystems goto out;
16087c478bd9Sstevel@tonic-gate }
16097c478bd9Sstevel@tonic-gate
1610af28f636SEnrico Perla - Sun Microsystems /* write stage2 to the pcfs mounted filesystem. */
1611af28f636SEnrico Perla - Sun Microsystems if (write(pcfs_fp, stage2->file, stage2->file_size)
1612af28f636SEnrico Perla - Sun Microsystems != stage2->file_size) {
1613af28f636SEnrico Perla - Sun Microsystems perror(gettext("Error writing stage2"));
1614af28f636SEnrico Perla - Sun Microsystems goto out;
16157c478bd9Sstevel@tonic-gate }
1616af28f636SEnrico Perla - Sun Microsystems
1617af28f636SEnrico Perla - Sun Microsystems status = BC_SUCCESS;
1618af28f636SEnrico Perla - Sun Microsystems out_fd:
16197c478bd9Sstevel@tonic-gate (void) close(pcfs_fp);
1620af28f636SEnrico Perla - Sun Microsystems out:
16217c478bd9Sstevel@tonic-gate (void) umount(TMP_MNTPT);
1622af28f636SEnrico Perla - Sun Microsystems (void) rmdir(TMP_MNTPT);
1623af28f636SEnrico Perla - Sun Microsystems return (status);
16247c478bd9Sstevel@tonic-gate }
1625