409 lines
9.8 KiB
C
Executable File
409 lines
9.8 KiB
C
Executable File
/* ----------------------------------------------------------------------- *
|
|
*
|
|
* Copyright 1998-2008 H. Peter Anvin - All Rights Reserved
|
|
* Copyright 2010 Intel Corporation; author: H. Peter Anvin
|
|
*
|
|
* This program is free software; you can redistribute it and/or modify
|
|
* it under the terms of the GNU General Public License as published by
|
|
* the Free Software Foundation, Inc., 53 Temple Place Ste 330,
|
|
* Boston MA 02111-1307, USA; either version 2 of the License, or
|
|
* (at your option) any later version; incorporated herein by reference.
|
|
*
|
|
* ----------------------------------------------------------------------- */
|
|
|
|
/*
|
|
* syslinux.c - Linux installer program for SYSLINUX
|
|
*
|
|
* This program now requires mtools. It turned out to be a lot
|
|
* easier to deal with than dealing with needing mount privileges.
|
|
* We need device write permission anyway.
|
|
*/
|
|
|
|
#define _GNU_SOURCE
|
|
#include <alloca.h>
|
|
#include <errno.h>
|
|
#include <fcntl.h>
|
|
#include <getopt.h>
|
|
#include <inttypes.h>
|
|
#include <mntent.h>
|
|
#include <paths.h>
|
|
#include <stdio.h>
|
|
#include <string.h>
|
|
#include <stdlib.h>
|
|
#include <sysexits.h>
|
|
#include <syslog.h>
|
|
#include <unistd.h>
|
|
#include <sys/types.h>
|
|
#include <sys/stat.h>
|
|
#include <sys/wait.h>
|
|
|
|
#include "syslinux.h"
|
|
#include "libfat.h"
|
|
#include "setadv.h"
|
|
#include "syslxopt.h"
|
|
#include "syslxfs.h"
|
|
|
|
char *program; /* Name of program */
|
|
pid_t mypid;
|
|
|
|
void __attribute__ ((noreturn)) die(const char *msg)
|
|
{
|
|
fprintf(stderr, "%s: %s\n", program, msg);
|
|
exit(1);
|
|
}
|
|
|
|
void __attribute__ ((noreturn)) die_err(const char *msg)
|
|
{
|
|
fprintf(stderr, "%s: %s: %s\n", program, msg, strerror(errno));
|
|
exit(1);
|
|
}
|
|
|
|
/*
|
|
* read/write wrapper functions
|
|
*/
|
|
ssize_t xpread(int fd, void *buf, size_t count, off_t offset)
|
|
{
|
|
char *bufp = (char *)buf;
|
|
ssize_t rv;
|
|
ssize_t done = 0;
|
|
|
|
while (count) {
|
|
rv = pread(fd, bufp, count, offset);
|
|
if (rv == 0) {
|
|
die("short read");
|
|
} else if (rv == -1) {
|
|
if (errno == EINTR) {
|
|
continue;
|
|
} else {
|
|
die(strerror(errno));
|
|
}
|
|
} else {
|
|
bufp += rv;
|
|
offset += rv;
|
|
done += rv;
|
|
count -= rv;
|
|
}
|
|
}
|
|
|
|
return done;
|
|
}
|
|
|
|
ssize_t xpwrite(int fd, const void *buf, size_t count, off_t offset)
|
|
{
|
|
const char *bufp = (const char *)buf;
|
|
ssize_t rv;
|
|
ssize_t done = 0;
|
|
|
|
while (count) {
|
|
rv = pwrite(fd, bufp, count, offset);
|
|
if (rv == 0) {
|
|
die("short write");
|
|
} else if (rv == -1) {
|
|
if (errno == EINTR) {
|
|
continue;
|
|
} else {
|
|
die(strerror(errno));
|
|
}
|
|
} else {
|
|
bufp += rv;
|
|
offset += rv;
|
|
done += rv;
|
|
count -= rv;
|
|
}
|
|
}
|
|
|
|
return done;
|
|
}
|
|
|
|
/*
|
|
* Version of the read function suitable for libfat
|
|
*/
|
|
int libfat_xpread(intptr_t pp, void *buf, size_t secsize,
|
|
libfat_sector_t sector)
|
|
{
|
|
off_t offset = (off_t) sector * secsize + opt.offset;
|
|
return xpread(pp, buf, secsize, offset);
|
|
}
|
|
|
|
static int move_file(char *filename)
|
|
{
|
|
char target_file[4096], command[5120];
|
|
char *cp = target_file, *ep = target_file + sizeof target_file - 16;
|
|
const char *sd;
|
|
int slash = 1;
|
|
int status;
|
|
|
|
cp += sprintf(cp, "'s:/");
|
|
for (sd = opt.directory; *sd; sd++) {
|
|
if (*sd == '/' || *sd == '\\') {
|
|
if (slash)
|
|
continue; /* Remove duplicated slashes */
|
|
slash = 1;
|
|
} else if (*sd == '\'' || *sd == '!') {
|
|
slash = 0;
|
|
if (cp < ep)
|
|
*cp++ = '\'';
|
|
if (cp < ep)
|
|
*cp++ = '\\';
|
|
if (cp < ep)
|
|
*cp++ = *sd;
|
|
if (cp < ep)
|
|
*cp++ = '\'';
|
|
continue;
|
|
} else {
|
|
slash = 0;
|
|
}
|
|
|
|
if (cp < ep)
|
|
*cp++ = *sd;
|
|
}
|
|
if (!slash)
|
|
*cp++ = '/';
|
|
sprintf(cp, "%s'", filename);
|
|
|
|
/* This command may fail legitimately */
|
|
sprintf(command, "mattrib -h -r -s %s 2>/dev/null", target_file);
|
|
status = system(command);
|
|
(void)status; /* Keep _FORTIFY_SOURCE happy */
|
|
|
|
sprintf(command, "mmove -D o -D O s:/%s %s", filename, target_file);
|
|
status = system(command);
|
|
|
|
if (!WIFEXITED(status) || WEXITSTATUS(status)) {
|
|
fprintf(stderr,
|
|
"%s: warning: unable to move %s\n", program, filename);
|
|
|
|
sprintf(command, "mattrib +r +h +s s:/%s", filename);
|
|
status = system(command);
|
|
} else {
|
|
sprintf(command, "mattrib +r +h +s %s", target_file);
|
|
status = system(command);
|
|
}
|
|
|
|
return status;
|
|
}
|
|
|
|
int main(int argc, char *argv[])
|
|
{
|
|
static unsigned char sectbuf[SECTOR_SIZE];
|
|
int dev_fd;
|
|
struct stat st;
|
|
int status;
|
|
const char *tmpdir;
|
|
char *mtools_conf;
|
|
int mtc_fd;
|
|
FILE *mtc, *mtp;
|
|
struct libfat_filesystem *fs;
|
|
libfat_sector_t s, *secp;
|
|
libfat_sector_t *sectors;
|
|
int32_t ldlinux_cluster;
|
|
int nsectors;
|
|
const char *errmsg;
|
|
int ldlinux_sectors, patch_sectors;
|
|
int i;
|
|
|
|
(void)argc; /* Unused */
|
|
|
|
mypid = getpid();
|
|
program = argv[0];
|
|
|
|
parse_options(argc, argv, MODE_SYSLINUX);
|
|
|
|
if (!opt.device)
|
|
usage(EX_USAGE, MODE_SYSLINUX);
|
|
|
|
if (opt.sectors || opt.heads || opt.reset_adv || opt.set_once
|
|
|| (opt.update_only > 0) || opt.menu_save) {
|
|
fprintf(stderr,
|
|
"At least one specified option not yet implemented"
|
|
" for this installer.\n");
|
|
exit(1);
|
|
}
|
|
|
|
/*
|
|
* Temp directory of choice...
|
|
*/
|
|
tmpdir = getenv("TMPDIR");
|
|
if (!tmpdir) {
|
|
#ifdef P_tmpdir
|
|
tmpdir = P_tmpdir;
|
|
#elif defined(_PATH_TMP)
|
|
tmpdir = _PATH_TMP;
|
|
#else
|
|
tmpdir = "/tmp";
|
|
#endif
|
|
}
|
|
|
|
/*
|
|
* First make sure we can open the device at all, and that we have
|
|
* read/write permission.
|
|
*/
|
|
dev_fd = open(opt.device, O_RDWR);
|
|
if (dev_fd < 0 || fstat(dev_fd, &st) < 0) {
|
|
die_err(opt.device);
|
|
exit(1);
|
|
}
|
|
|
|
if (!opt.force && !S_ISBLK(st.st_mode) && !S_ISREG(st.st_mode)) {
|
|
fprintf(stderr,
|
|
"%s: not a block device or regular file (use -f to override)\n",
|
|
opt.device);
|
|
exit(1);
|
|
}
|
|
|
|
xpread(dev_fd, sectbuf, SECTOR_SIZE, opt.offset);
|
|
|
|
/*
|
|
* Check to see that what we got was indeed an MS-DOS boot sector/superblock
|
|
*/
|
|
if ((errmsg = syslinux_check_bootsect(sectbuf, NULL))) {
|
|
die(errmsg);
|
|
}
|
|
|
|
/*
|
|
* Create an mtools configuration file
|
|
*/
|
|
if (asprintf(&mtools_conf, "%s//syslinux-mtools-XXXXXX", tmpdir) < 0 ||
|
|
!mtools_conf)
|
|
die_err(tmpdir);
|
|
|
|
mtc_fd = mkstemp(mtools_conf);
|
|
if (mtc_fd < 0 || !(mtc = fdopen(mtc_fd, "w")))
|
|
die_err(mtools_conf);
|
|
|
|
fprintf(mtc,
|
|
/* These are needed for some flash memories */
|
|
"MTOOLS_SKIP_CHECK=1\n"
|
|
"MTOOLS_FAT_COMPATIBILITY=1\n"
|
|
"drive s:\n"
|
|
" file=\"/proc/%lu/fd/%d\"\n"
|
|
" offset=%llu\n",
|
|
(unsigned long)mypid,
|
|
dev_fd, (unsigned long long)opt.offset);
|
|
|
|
if (ferror(mtc) || fclose(mtc))
|
|
die_err(mtools_conf);
|
|
|
|
/*
|
|
* Run mtools to create the LDLINUX.SYS file
|
|
*/
|
|
if (setenv("MTOOLSRC", mtools_conf, 1)) {
|
|
perror(program);
|
|
exit(1);
|
|
}
|
|
|
|
/*
|
|
* Create a vacuous ADV in memory. This should be smarter.
|
|
*/
|
|
syslinux_reset_adv(syslinux_adv);
|
|
|
|
/* This command may fail legitimately */
|
|
status = system("mattrib -h -r -s s:/ldlinux.sys 2>/dev/null");
|
|
(void)status; /* Keep _FORTIFY_SOURCE happy */
|
|
|
|
mtp = popen("mcopy -D o -D O -o - s:/ldlinux.sys", "w");
|
|
if (!mtp ||
|
|
fwrite((const void _force *)syslinux_ldlinux,
|
|
1, syslinux_ldlinux_len, mtp)
|
|
!= syslinux_ldlinux_len ||
|
|
fwrite((const void _force *)syslinux_adv,
|
|
1, 2 * ADV_SIZE, mtp)
|
|
!= 2 * ADV_SIZE ||
|
|
(status = pclose(mtp), !WIFEXITED(status) || WEXITSTATUS(status))) {
|
|
die("failed to create ldlinux.sys");
|
|
}
|
|
|
|
/*
|
|
* Now, use libfat to create a block map
|
|
*/
|
|
ldlinux_sectors = (syslinux_ldlinux_len + 2 * ADV_SIZE
|
|
+ SECTOR_SIZE - 1) >> SECTOR_SHIFT;
|
|
sectors = calloc(ldlinux_sectors, sizeof *sectors);
|
|
fs = libfat_open(libfat_xpread, dev_fd);
|
|
ldlinux_cluster = libfat_searchdir(fs, 0, "LDLINUX SYS", NULL);
|
|
secp = sectors;
|
|
nsectors = 0;
|
|
s = libfat_clustertosector(fs, ldlinux_cluster);
|
|
while (s && nsectors < ldlinux_sectors) {
|
|
*secp++ = s;
|
|
nsectors++;
|
|
s = libfat_nextsector(fs, s);
|
|
}
|
|
libfat_close(fs);
|
|
|
|
/* Patch ldlinux.sys and the boot sector */
|
|
i = syslinux_patch(sectors, nsectors, opt.stupid_mode, opt.raid_mode,
|
|
opt.directory, NULL);
|
|
patch_sectors = (i + SECTOR_SIZE - 1) >> SECTOR_SHIFT;
|
|
|
|
/* Write the now-patched first sectors of ldlinux.sys */
|
|
for (i = 0; i < patch_sectors; i++) {
|
|
xpwrite(dev_fd, (const char _force *)syslinux_ldlinux
|
|
+ i * SECTOR_SIZE, SECTOR_SIZE,
|
|
opt.offset + ((off_t) sectors[i] << SECTOR_SHIFT));
|
|
}
|
|
|
|
/* Move ldlinux.sys to the desired location */
|
|
if (opt.directory) {
|
|
status = move_file("ldlinux.sys");
|
|
} else {
|
|
status = system("mattrib +r +h +s s:/ldlinux.sys");
|
|
}
|
|
|
|
if (!WIFEXITED(status) || WEXITSTATUS(status)) {
|
|
fprintf(stderr,
|
|
"%s: warning: failed to set system bit on ldlinux.sys\n",
|
|
program);
|
|
}
|
|
|
|
/* This command may fail legitimately */
|
|
status = system("mattrib -h -r -s s:/ldlinux.c32 2>/dev/null");
|
|
(void)status; /* Keep _FORTIFY_SOURCE happy */
|
|
|
|
mtp = popen("mcopy -D o -D O -o - s:/ldlinux.c32", "w");
|
|
if (!mtp || fwrite((const char _force *)syslinux_ldlinuxc32,
|
|
1, syslinux_ldlinuxc32_len, mtp)
|
|
!= syslinux_ldlinuxc32_len ||
|
|
(status = pclose(mtp), !WIFEXITED(status) || WEXITSTATUS(status))) {
|
|
die("failed to create ldlinux.c32");
|
|
}
|
|
|
|
/* Move ldlinux.c32 to the desired location */
|
|
if (opt.directory) {
|
|
status = move_file("ldlinux.c32");
|
|
} else {
|
|
status = system("mattrib +r +h +s s:/ldlinux.c32");
|
|
}
|
|
|
|
if (!WIFEXITED(status) || WEXITSTATUS(status)) {
|
|
fprintf(stderr,
|
|
"%s: warning: failed to set system bit on ldlinux.c32\n",
|
|
program);
|
|
}
|
|
|
|
/*
|
|
* Cleanup
|
|
*/
|
|
unlink(mtools_conf);
|
|
|
|
/*
|
|
* To finish up, write the boot sector
|
|
*/
|
|
|
|
/* Read the superblock again since it might have changed while mounted */
|
|
xpread(dev_fd, sectbuf, SECTOR_SIZE, opt.offset);
|
|
|
|
/* Copy the syslinux code into the boot sector */
|
|
syslinux_make_bootsect(sectbuf, VFAT);
|
|
|
|
/* Write new boot sector */
|
|
xpwrite(dev_fd, sectbuf, SECTOR_SIZE, opt.offset);
|
|
|
|
close(dev_fd);
|
|
sync();
|
|
|
|
/* Done! */
|
|
|
|
return 0;
|
|
}
|