481 lines
10 KiB
C
481 lines
10 KiB
C
/*
|
|
* GRUB -- GRand Unified Bootloader
|
|
* Copyright (C) 1999,2000,2001,2002,2003,2004,2006,2007,2008,2009,2010,2011,2012,2013 Free Software Foundation, Inc.
|
|
*
|
|
* GRUB 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, either version 3 of the License, or
|
|
* (at your option) any later version.
|
|
*
|
|
* GRUB is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License
|
|
* along with GRUB. If not, see <http://www.gnu.org/licenses/>.
|
|
*/
|
|
|
|
#include <config-util.h>
|
|
|
|
#include <grub/disk.h>
|
|
#include <grub/partition.h>
|
|
#include <grub/msdos_partition.h>
|
|
#include <grub/types.h>
|
|
#include <grub/err.h>
|
|
#include <grub/emu/misc.h>
|
|
#include <grub/emu/hostdisk.h>
|
|
#include <grub/emu/getroot.h>
|
|
#include <grub/misc.h>
|
|
#include <grub/i18n.h>
|
|
#include <grub/list.h>
|
|
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
#include <ctype.h>
|
|
#include <assert.h>
|
|
#include <unistd.h>
|
|
#include <sys/types.h>
|
|
#include <sys/stat.h>
|
|
#include <fcntl.h>
|
|
#include <errno.h>
|
|
#include <limits.h>
|
|
|
|
#include <string.h>
|
|
#include <dos/dos.h>
|
|
#include <dos/filesystem.h>
|
|
#include <dos/exall.h>
|
|
#include <proto/dos.h>
|
|
#include <devices/hardblocks.h>
|
|
#include <devices/newstyle.h>
|
|
#include <proto/exec.h>
|
|
#include <proto/utility.h>
|
|
#include <proto/partition.h>
|
|
#include <devices/trackdisk.h>
|
|
#include <exec/errors.h>
|
|
|
|
#define BOUNCE_SIZE 1048576
|
|
|
|
static ULONG *bounce;
|
|
|
|
static grub_uint64_t
|
|
grub_util_get_fd_size_volume (grub_util_fd_t fd __attribute__ ((unused)),
|
|
const char *dev,
|
|
unsigned *log_secsize)
|
|
{
|
|
struct DriveGeometry *geo;
|
|
LONG err;
|
|
unsigned sector_size, log_sector_size;
|
|
|
|
if (!bounce)
|
|
bounce = AllocVec (BOUNCE_SIZE, MEMF_PUBLIC | MEMF_CLEAR);
|
|
if (!bounce)
|
|
grub_util_error ("out of memory");
|
|
|
|
fd->ioreq->iotd_Req.io_Command = TD_GETGEOMETRY;
|
|
fd->ioreq->iotd_Req.io_Length = sizeof (*geo);
|
|
fd->ioreq->iotd_Req.io_Data = bounce;
|
|
fd->ioreq->iotd_Req.io_Offset = 0;
|
|
fd->ioreq->iotd_Req.io_Actual = 0;
|
|
err = DoIO ((struct IORequest *) fd->ioreq);
|
|
if (err)
|
|
{
|
|
grub_util_info ("I/O failed with error %d, IoErr=%d", (int)err, (int) IoErr ());
|
|
return -1;
|
|
}
|
|
|
|
geo = (struct DriveGeometry *) bounce;
|
|
|
|
sector_size = geo->dg_SectorSize;
|
|
|
|
if (sector_size & (sector_size - 1) || !sector_size)
|
|
return -1;
|
|
|
|
for (log_sector_size = 0;
|
|
(1 << log_sector_size) < sector_size;
|
|
log_sector_size++);
|
|
|
|
if (log_secsize)
|
|
*log_secsize = log_sector_size;
|
|
|
|
return (grub_uint64_t) geo->dg_TotalSectors * (grub_uint64_t) geo->dg_SectorSize;
|
|
}
|
|
|
|
static grub_uint64_t
|
|
grub_util_get_fd_size_file (grub_util_fd_t fd,
|
|
const char *dev __attribute__ ((unused)),
|
|
unsigned *log_secsize)
|
|
{
|
|
off_t oo, ro;
|
|
*log_secsize = 9;
|
|
/* FIXME: support 64-bit offsets. */
|
|
oo = lseek (fd->fd, 0, SEEK_CUR);
|
|
ro = lseek (fd->fd, 0, SEEK_END);
|
|
lseek (fd->fd, oo, SEEK_SET);
|
|
return ro;
|
|
}
|
|
|
|
grub_err_t
|
|
grub_util_fd_seek (grub_util_fd_t fd, const char *name, grub_uint64_t off)
|
|
{
|
|
switch (fd->type)
|
|
{
|
|
case GRUB_UTIL_FD_FILE:
|
|
if (lseek (fd->fd, 0, SEEK_SET) == (off_t) -1)
|
|
return grub_error (GRUB_ERR_BAD_DEVICE, N_("cannot seek `%s': %s"),
|
|
name, strerror (errno));
|
|
fd->off = off;
|
|
return 0;
|
|
case GRUB_UTIL_FD_DISK:
|
|
fd->off = off;
|
|
return 0;
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
grub_util_fd_t
|
|
grub_util_fd_open (const char *dev, int flg)
|
|
{
|
|
grub_util_fd_t ret = xmalloc (sizeof (*ret));
|
|
const char *p, *p1, *p2;
|
|
char *tmp;
|
|
IPTR unit = 0;
|
|
ULONG flags = 0;
|
|
|
|
ret->off = 0;
|
|
|
|
if (dev[0] == '\0')
|
|
{
|
|
free (ret);
|
|
return NULL;
|
|
}
|
|
|
|
p = strchr (dev, ':');
|
|
|
|
if (!p || p[1])
|
|
{
|
|
ret->type = GRUB_UTIL_FD_FILE;
|
|
ret->fd = open (dev, flg);
|
|
if (ret->fd < 0)
|
|
{
|
|
free (ret);
|
|
return NULL;
|
|
}
|
|
return ret;
|
|
}
|
|
|
|
p1 = strchr (dev, '/');
|
|
if (!p1)
|
|
p1 = p;
|
|
else
|
|
{
|
|
unit = grub_strtoul (p1 + 1, (char **) &p2, 16);
|
|
if (p2 && *p2 == '/')
|
|
flags = grub_strtoul (p2 + 1, 0, 16);
|
|
}
|
|
|
|
ret->mp = CreateMsgPort();
|
|
if (!ret->mp)
|
|
{
|
|
free (ret);
|
|
return NULL;
|
|
}
|
|
ret->ioreq = (struct IOExtTD *) CreateIORequest(ret->mp,
|
|
sizeof(struct IOExtTD));
|
|
if (!ret->ioreq)
|
|
{
|
|
free (ret);
|
|
DeleteMsgPort (ret->mp);
|
|
return NULL;
|
|
}
|
|
|
|
tmp = xmalloc (p1 - dev + 1);
|
|
memcpy (tmp, dev, p1 - dev);
|
|
tmp[p1 - dev] = '\0';
|
|
ret->type = GRUB_UTIL_FD_DISK;
|
|
|
|
ret->is_floppy = (strcmp (tmp, TD_NAME) == 0);
|
|
ret->is_64 = 1;
|
|
|
|
if (OpenDevice ((unsigned char *) tmp, unit, (struct IORequest *) ret->ioreq, flags))
|
|
{
|
|
free (tmp);
|
|
free (ret);
|
|
DeleteMsgPort (ret->mp);
|
|
return NULL;
|
|
}
|
|
free (tmp);
|
|
return ret;
|
|
}
|
|
|
|
static ssize_t
|
|
grub_util_fd_read_file (grub_util_fd_t fd, char *buf, size_t len)
|
|
{
|
|
ssize_t size = len;
|
|
|
|
while (len)
|
|
{
|
|
ssize_t ret = read (fd->fd, buf, len);
|
|
|
|
if (ret <= 0)
|
|
{
|
|
if (errno == EINTR)
|
|
continue;
|
|
else
|
|
return ret;
|
|
}
|
|
|
|
fd->off += ret;
|
|
len -= ret;
|
|
buf += ret;
|
|
}
|
|
|
|
return size;
|
|
}
|
|
|
|
/* Write LEN bytes from BUF to FD. Return less than or equal to zero if an
|
|
error occurs, otherwise return LEN. */
|
|
static ssize_t
|
|
grub_util_fd_write_file (grub_util_fd_t fd, const char *buf, size_t len)
|
|
{
|
|
ssize_t size = len;
|
|
|
|
while (len)
|
|
{
|
|
ssize_t ret = write (fd->fd, buf, len);
|
|
|
|
if (ret <= 0)
|
|
{
|
|
if (errno == EINTR)
|
|
continue;
|
|
else
|
|
return ret;
|
|
}
|
|
|
|
fd->off += ret;
|
|
len -= ret;
|
|
buf += ret;
|
|
}
|
|
|
|
return size;
|
|
}
|
|
|
|
static void
|
|
stop_motor (grub_util_fd_t fd)
|
|
{
|
|
if (!fd->is_floppy)
|
|
return;
|
|
fd->ioreq->iotd_Req.io_Command = TD_MOTOR;
|
|
fd->ioreq->iotd_Req.io_Length = 0;
|
|
fd->ioreq->iotd_Req.io_Data = 0;
|
|
fd->ioreq->iotd_Req.io_Offset = 0;
|
|
fd->ioreq->iotd_Req.io_Actual = 0;
|
|
DoIO ((struct IORequest *) fd->ioreq);
|
|
}
|
|
|
|
static ssize_t
|
|
grub_util_fd_read_volume (grub_util_fd_t fd, char *buf, size_t len)
|
|
{
|
|
grub_uint64_t adj = 0;
|
|
|
|
if (!bounce)
|
|
bounce = AllocVec (BOUNCE_SIZE, MEMF_PUBLIC | MEMF_CLEAR);
|
|
if (!bounce)
|
|
grub_util_error ("out of memory");
|
|
|
|
while (len)
|
|
{
|
|
size_t cr = len;
|
|
LONG err;
|
|
if (cr > BOUNCE_SIZE)
|
|
cr = BOUNCE_SIZE;
|
|
retry:
|
|
if (fd->is_64)
|
|
fd->ioreq->iotd_Req.io_Command = NSCMD_TD_READ64;
|
|
else
|
|
fd->ioreq->iotd_Req.io_Command = CMD_READ;
|
|
fd->ioreq->iotd_Req.io_Length = cr;
|
|
fd->ioreq->iotd_Req.io_Data = bounce;
|
|
fd->ioreq->iotd_Req.io_Offset = fd->off & 0xFFFFFFFF;
|
|
fd->ioreq->iotd_Req.io_Actual = fd->off >> 32;
|
|
err = DoIO ((struct IORequest *) fd->ioreq);
|
|
if (err == IOERR_NOCMD && fd->is_64)
|
|
{
|
|
fd->is_64 = 0;
|
|
goto retry;
|
|
}
|
|
if (err)
|
|
{
|
|
grub_util_info ("I/O failed with error %d, IoErr=%d", (int)err, (int) IoErr ());
|
|
stop_motor (fd);
|
|
return -1;
|
|
}
|
|
memcpy (buf, bounce, cr);
|
|
adj += cr;
|
|
len -= cr;
|
|
buf += cr;
|
|
}
|
|
|
|
fd->off += adj;
|
|
stop_motor (fd);
|
|
return adj;
|
|
}
|
|
|
|
static ssize_t
|
|
grub_util_fd_write_volume (grub_util_fd_t fd, const char *buf, size_t len)
|
|
{
|
|
grub_uint64_t adj = 0;
|
|
if (!bounce)
|
|
bounce = AllocVec (BOUNCE_SIZE, MEMF_PUBLIC | MEMF_CLEAR);
|
|
if (!bounce)
|
|
grub_util_error ("out of memory");
|
|
|
|
while (len)
|
|
{
|
|
size_t cr = len;
|
|
LONG err;
|
|
if (cr > BOUNCE_SIZE)
|
|
cr = BOUNCE_SIZE;
|
|
retry:
|
|
if (fd->is_64)
|
|
fd->ioreq->iotd_Req.io_Command = NSCMD_TD_WRITE64;
|
|
else
|
|
fd->ioreq->iotd_Req.io_Command = CMD_WRITE;
|
|
fd->ioreq->iotd_Req.io_Length = cr;
|
|
fd->ioreq->iotd_Req.io_Data = bounce;
|
|
fd->ioreq->iotd_Req.io_Offset = fd->off & 0xFFFFFFFF;
|
|
fd->ioreq->iotd_Req.io_Actual = fd->off >> 32;
|
|
memcpy (bounce, buf, cr);
|
|
err = DoIO ((struct IORequest *) fd->ioreq);
|
|
if (err == IOERR_NOCMD && fd->is_64)
|
|
{
|
|
fd->is_64 = 0;
|
|
goto retry;
|
|
}
|
|
if (err)
|
|
{
|
|
grub_util_info ("I/O failed with error %d", err);
|
|
stop_motor (fd);
|
|
return -1;
|
|
}
|
|
|
|
adj += cr;
|
|
len -= cr;
|
|
buf += cr;
|
|
}
|
|
|
|
fd->off += adj;
|
|
stop_motor (fd);
|
|
return adj;
|
|
}
|
|
|
|
ssize_t
|
|
grub_util_fd_read (grub_util_fd_t fd, char *buf, size_t len)
|
|
{
|
|
switch (fd->type)
|
|
{
|
|
case GRUB_UTIL_FD_FILE:
|
|
return grub_util_fd_read_file (fd, buf, len);
|
|
case GRUB_UTIL_FD_DISK:
|
|
return grub_util_fd_read_volume (fd, buf, len);
|
|
}
|
|
return -1;
|
|
}
|
|
|
|
ssize_t
|
|
grub_util_fd_write (grub_util_fd_t fd, const char *buf, size_t len)
|
|
{
|
|
switch (fd->type)
|
|
{
|
|
case GRUB_UTIL_FD_FILE:
|
|
return grub_util_fd_write_file (fd, buf, len);
|
|
case GRUB_UTIL_FD_DISK:
|
|
return grub_util_fd_write_volume (fd, buf, len);
|
|
}
|
|
return -1;
|
|
}
|
|
|
|
grub_uint64_t
|
|
grub_util_get_fd_size (grub_util_fd_t fd,
|
|
const char *dev,
|
|
unsigned *log_secsize)
|
|
{
|
|
switch (fd->type)
|
|
{
|
|
case GRUB_UTIL_FD_FILE:
|
|
return grub_util_get_fd_size_file (fd, dev, log_secsize);
|
|
|
|
case GRUB_UTIL_FD_DISK:
|
|
return grub_util_get_fd_size_volume (fd, dev, log_secsize);
|
|
}
|
|
return -1;
|
|
}
|
|
|
|
void
|
|
grub_util_fd_close (grub_util_fd_t fd)
|
|
{
|
|
switch (fd->type)
|
|
{
|
|
case GRUB_UTIL_FD_FILE:
|
|
close (fd->fd);
|
|
return;
|
|
case GRUB_UTIL_FD_DISK:
|
|
CloseDevice ((struct IORequest *) fd->ioreq);
|
|
DeleteIORequest((struct IORequest *) fd->ioreq);
|
|
DeleteMsgPort (fd->mp);
|
|
return;
|
|
}
|
|
}
|
|
|
|
static void
|
|
grub_util_fd_sync_volume (grub_util_fd_t fd)
|
|
{
|
|
fd->ioreq->iotd_Req.io_Command = CMD_UPDATE;
|
|
fd->ioreq->iotd_Req.io_Length = 0;
|
|
fd->ioreq->iotd_Req.io_Data = 0;
|
|
fd->ioreq->iotd_Req.io_Offset = 0;
|
|
fd->ioreq->iotd_Req.io_Actual = 0;
|
|
DoIO ((struct IORequest *) fd->ioreq);
|
|
}
|
|
|
|
void
|
|
grub_util_fd_sync (grub_util_fd_t fd)
|
|
{
|
|
switch (fd->type)
|
|
{
|
|
case GRUB_UTIL_FD_FILE:
|
|
fsync (fd->fd);
|
|
return;
|
|
case GRUB_UTIL_FD_DISK:
|
|
grub_util_fd_sync_volume (fd);
|
|
return;
|
|
}
|
|
}
|
|
|
|
|
|
void
|
|
grub_hostdisk_configure_device_driver (grub_util_fd_t fd __attribute__ ((unused)))
|
|
{
|
|
}
|
|
|
|
void
|
|
grub_hostdisk_flush_initial_buffer (const char *os_dev __attribute__ ((unused)))
|
|
{
|
|
}
|
|
|
|
|
|
const char *
|
|
grub_util_fd_strerror (void)
|
|
{
|
|
static char buf[201];
|
|
LONG err = IoErr ();
|
|
if (!err)
|
|
return _("Success");
|
|
memset (buf, '\0', sizeof (buf));
|
|
Fault (err, (const unsigned char *) "", (STRPTR) buf, sizeof (buf));
|
|
if (buf[0] == ':')
|
|
return buf + 1;
|
|
return buf;
|
|
}
|