25239370fd
* include/grub/arc/arc.h (grub_arc_iterate_devs_hook_t): New type. (grub_arc_iterate_devs): Add hook_data argument. * include/grub/ata.h (grub_ata_dev_iterate_hook_t): New type. (struct grub_ata_dev.iterate): Add hook_data argument. * include/grub/device.h (grub_device_iterate_hook_t): New type. (grub_device_iterate): Add hook_data argument. * include/grub/disk.h (grub_disk_dev_iterate_hook_t): New type. (struct grub_disk_dev.iterate): Add hook_data argument. (grub_disk_dev_iterate): Likewise. * include/grub/gpt_partition.h (grub_gpt_partition_map_iterate): Likewise. * include/grub/msdos_partition.h (grub_partition_msdos_iterate): Likewise. * include/grub/partition.h (grub_partition_iterate_hook_t): New type. (struct grub_partition_map.iterate): Add hook_data argument. (grub_partition_iterate): Likewise. * include/grub/scsi.h (grub_scsi_dev_iterate_hook_t): New type. (struct grub_scsi_dev.iterate): Add hook_data argument. Update all callers.
351 lines
8.1 KiB
C
351 lines
8.1 KiB
C
/* ofdisk.c - Open Firmware disk access. */
|
|
/*
|
|
* GRUB -- GRand Unified Bootloader
|
|
* Copyright (C) 2004,2006,2007,2008,2009,2011 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 <grub/misc.h>
|
|
#include <grub/disk.h>
|
|
#include <grub/mm.h>
|
|
#include <grub/arc/arc.h>
|
|
#include <grub/i18n.h>
|
|
|
|
static grub_arc_fileno_t last_handle = 0;
|
|
static char *last_path = NULL;
|
|
static int handle_writable = 0;
|
|
|
|
static int lnum = 0;
|
|
|
|
struct arcdisk_hash_ent
|
|
{
|
|
char *devpath;
|
|
int num;
|
|
struct arcdisk_hash_ent *next;
|
|
};
|
|
|
|
#define ARCDISK_HASH_SZ 8
|
|
static struct arcdisk_hash_ent *arcdisk_hash[ARCDISK_HASH_SZ];
|
|
|
|
static int
|
|
arcdisk_hash_fn (const char *devpath)
|
|
{
|
|
int hash = 0;
|
|
while (*devpath)
|
|
hash ^= *devpath++;
|
|
return (hash & (ARCDISK_HASH_SZ - 1));
|
|
}
|
|
|
|
static struct arcdisk_hash_ent *
|
|
arcdisk_hash_find (const char *devpath)
|
|
{
|
|
struct arcdisk_hash_ent *p = arcdisk_hash[arcdisk_hash_fn (devpath)];
|
|
|
|
while (p)
|
|
{
|
|
if (!grub_strcmp (p->devpath, devpath))
|
|
break;
|
|
p = p->next;
|
|
}
|
|
return p;
|
|
}
|
|
|
|
static struct arcdisk_hash_ent *
|
|
arcdisk_hash_add (char *devpath)
|
|
{
|
|
struct arcdisk_hash_ent *p;
|
|
struct arcdisk_hash_ent **head = &arcdisk_hash[arcdisk_hash_fn(devpath)];
|
|
|
|
p = grub_malloc (sizeof (*p));
|
|
if (!p)
|
|
return NULL;
|
|
|
|
p->devpath = devpath;
|
|
p->next = *head;
|
|
p->num = lnum++;
|
|
*head = p;
|
|
return p;
|
|
}
|
|
|
|
|
|
/* Context for grub_arcdisk_iterate. */
|
|
struct grub_arcdisk_iterate_ctx
|
|
{
|
|
grub_disk_dev_iterate_hook_t hook;
|
|
void *hook_data;
|
|
};
|
|
|
|
/* Helper for grub_arcdisk_iterate. */
|
|
static int
|
|
grub_arcdisk_iterate_iter (const char *name,
|
|
const struct grub_arc_component *comp, void *data)
|
|
{
|
|
struct grub_arcdisk_iterate_ctx *ctx = data;
|
|
|
|
if (!(comp->type == GRUB_ARC_COMPONENT_TYPE_DISK
|
|
|| comp->type == GRUB_ARC_COMPONENT_TYPE_DISK
|
|
|| comp->type == GRUB_ARC_COMPONENT_TYPE_TAPE))
|
|
return 0;
|
|
return ctx->hook (name, ctx->hook_data);
|
|
}
|
|
|
|
static int
|
|
grub_arcdisk_iterate (int (*hook_in) (const char *name),
|
|
grub_disk_pull_t pull)
|
|
{
|
|
struct grub_arcdisk_iterate_ctx ctx = { hook, hook_data };
|
|
|
|
if (pull != GRUB_DISK_PULL_NONE)
|
|
return 0;
|
|
|
|
return grub_arc_iterate_devs (grub_arcdisk_iterate_iter, &ctx, 1);
|
|
}
|
|
|
|
#define RAW_SUFFIX "partition(10)"
|
|
|
|
static grub_err_t
|
|
reopen (const char *name, int writable)
|
|
{
|
|
grub_arc_fileno_t handle;
|
|
|
|
if (last_path && grub_strcmp (last_path, name) == 0
|
|
&& (!writable || handle_writable))
|
|
{
|
|
grub_dprintf ("arcdisk", "using already opened %s\n", name);
|
|
return GRUB_ERR_NONE;
|
|
}
|
|
if (last_path)
|
|
{
|
|
GRUB_ARC_FIRMWARE_VECTOR->close (last_handle);
|
|
grub_free (last_path);
|
|
last_path = NULL;
|
|
last_handle = 0;
|
|
handle_writable = 0;
|
|
}
|
|
if (GRUB_ARC_FIRMWARE_VECTOR->open (name,
|
|
writable ? GRUB_ARC_FILE_ACCESS_OPEN_RW
|
|
: GRUB_ARC_FILE_ACCESS_OPEN_RO, &handle))
|
|
{
|
|
grub_dprintf ("arcdisk", "couldn't open %s\n", name);
|
|
return grub_error (GRUB_ERR_IO, "couldn't open %s", name);
|
|
}
|
|
handle_writable = writable;
|
|
last_path = grub_strdup (name);
|
|
if (!last_path)
|
|
return grub_errno;
|
|
last_handle = handle;
|
|
grub_dprintf ("arcdisk", "opened %s\n", name);
|
|
return GRUB_ERR_NONE;
|
|
}
|
|
|
|
static grub_err_t
|
|
grub_arcdisk_open (const char *name, grub_disk_t disk)
|
|
{
|
|
char *fullname, *optr;
|
|
const char *iptr;
|
|
int state = 0;
|
|
grub_err_t err;
|
|
grub_arc_err_t r;
|
|
struct grub_arc_fileinfo info;
|
|
struct arcdisk_hash_ent *hash;
|
|
|
|
if (grub_memcmp (name, "arc/", 4) != 0)
|
|
return grub_error (GRUB_ERR_UNKNOWN_DEVICE, "not arc device");
|
|
fullname = grub_malloc (2 * grub_strlen (name) + sizeof (RAW_SUFFIX));
|
|
if (!fullname)
|
|
return grub_errno;
|
|
optr = fullname;
|
|
for (iptr = name + 4; *iptr; iptr++)
|
|
if (state == 0)
|
|
{
|
|
if (!grub_isdigit (*iptr))
|
|
*optr++ = *iptr;
|
|
else
|
|
{
|
|
*optr++ = '(';
|
|
*optr++ = *iptr;
|
|
state = 1;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
if (grub_isdigit (*iptr))
|
|
*optr++ = *iptr;
|
|
else
|
|
{
|
|
*optr++ = ')';
|
|
state = 0;
|
|
}
|
|
}
|
|
if (state)
|
|
*optr++ = ')';
|
|
grub_memcpy (optr, RAW_SUFFIX, sizeof (RAW_SUFFIX));
|
|
disk->data = fullname;
|
|
grub_dprintf ("arcdisk", "opening %s\n", fullname);
|
|
|
|
hash = arcdisk_hash_find (fullname);
|
|
if (!hash)
|
|
hash = arcdisk_hash_add (fullname);
|
|
if (!hash)
|
|
return grub_errno;
|
|
|
|
err = reopen (fullname, 0);
|
|
if (err)
|
|
return err;
|
|
|
|
r = GRUB_ARC_FIRMWARE_VECTOR->getfileinformation (last_handle, &info);
|
|
if (r)
|
|
{
|
|
grub_uint64_t res = 0;
|
|
int i;
|
|
|
|
grub_dprintf ("arcdisk", "couldn't retrieve size: %ld\n", r);
|
|
for (i = 40; i >= 9; i--)
|
|
{
|
|
grub_uint64_t pos = res | (1ULL << i);
|
|
char buf[512];
|
|
long unsigned count = 0;
|
|
grub_dprintf ("arcdisk",
|
|
"seek to 0x%" PRIxGRUB_UINT64_T "\n", pos);
|
|
if (GRUB_ARC_FIRMWARE_VECTOR->seek (last_handle, &pos, 0))
|
|
continue;
|
|
if (GRUB_ARC_FIRMWARE_VECTOR->read (last_handle, buf,
|
|
0x200, &count))
|
|
continue;
|
|
if (count == 0)
|
|
continue;
|
|
res |= (1ULL << i);
|
|
}
|
|
grub_dprintf ("arcdisk",
|
|
"determined disk size 0x%" PRIxGRUB_UINT64_T "\n", res);
|
|
disk->total_sectors = (res + 0x200) >> 9;
|
|
}
|
|
else
|
|
disk->total_sectors = (info.end >> 9);
|
|
|
|
disk->id = hash->num;
|
|
return GRUB_ERR_NONE;
|
|
}
|
|
|
|
static void
|
|
grub_arcdisk_close (grub_disk_t disk)
|
|
{
|
|
grub_free (disk->data);
|
|
}
|
|
|
|
static grub_err_t
|
|
grub_arcdisk_read (grub_disk_t disk, grub_disk_addr_t sector,
|
|
grub_size_t size, char *buf)
|
|
{
|
|
grub_err_t err;
|
|
grub_uint64_t pos = sector << 9;
|
|
unsigned long count;
|
|
grub_uint64_t totl = size << 9;
|
|
grub_arc_err_t r;
|
|
|
|
err = reopen (disk->data, 0);
|
|
if (err)
|
|
return err;
|
|
r = GRUB_ARC_FIRMWARE_VECTOR->seek (last_handle, &pos, 0);
|
|
if (r)
|
|
{
|
|
grub_dprintf ("arcdisk", "seek to 0x%" PRIxGRUB_UINT64_T " failed: %ld\n",
|
|
pos, r);
|
|
return grub_error (GRUB_ERR_IO, "couldn't seek");
|
|
}
|
|
|
|
while (totl)
|
|
{
|
|
if (GRUB_ARC_FIRMWARE_VECTOR->read (last_handle, buf,
|
|
totl, &count))
|
|
return grub_error (GRUB_ERR_READ_ERROR,
|
|
N_("failure reading sector 0x%llx "
|
|
"from `%s'"),
|
|
(unsigned long long) sector,
|
|
disk->name);
|
|
totl -= count;
|
|
buf += count;
|
|
}
|
|
|
|
return GRUB_ERR_NONE;
|
|
}
|
|
|
|
static grub_err_t
|
|
grub_arcdisk_write (grub_disk_t disk, grub_disk_addr_t sector,
|
|
grub_size_t size, const char *buf)
|
|
{
|
|
grub_err_t err;
|
|
grub_uint64_t pos = sector << 9;
|
|
unsigned long count;
|
|
grub_uint64_t totl = size << 9;
|
|
grub_arc_err_t r;
|
|
|
|
err = reopen (disk->data, 1);
|
|
if (err)
|
|
return err;
|
|
r = GRUB_ARC_FIRMWARE_VECTOR->seek (last_handle, &pos, 0);
|
|
if (r)
|
|
{
|
|
grub_dprintf ("arcdisk", "seek to 0x%" PRIxGRUB_UINT64_T " failed: %ld\n",
|
|
pos, r);
|
|
return grub_error (GRUB_ERR_IO, "couldn't seek");
|
|
}
|
|
|
|
while (totl)
|
|
{
|
|
if (GRUB_ARC_FIRMWARE_VECTOR->write (last_handle, buf,
|
|
totl, &count))
|
|
return grub_error (GRUB_ERR_WRITE_ERROR, N_("failure writing sector 0x%llx "
|
|
"to `%s'"),
|
|
(unsigned long long) sector,
|
|
disk->name);
|
|
totl -= count;
|
|
buf += count;
|
|
}
|
|
|
|
return GRUB_ERR_NONE;
|
|
}
|
|
|
|
static struct grub_disk_dev grub_arcdisk_dev =
|
|
{
|
|
.name = "arcdisk",
|
|
.id = GRUB_DISK_DEVICE_ARCDISK_ID,
|
|
.iterate = grub_arcdisk_iterate,
|
|
.open = grub_arcdisk_open,
|
|
.close = grub_arcdisk_close,
|
|
.read = grub_arcdisk_read,
|
|
.write = grub_arcdisk_write,
|
|
.next = 0
|
|
};
|
|
|
|
void
|
|
grub_arcdisk_init (void)
|
|
{
|
|
grub_disk_dev_register (&grub_arcdisk_dev);
|
|
}
|
|
|
|
void
|
|
grub_arcdisk_fini (void)
|
|
{
|
|
if (last_path)
|
|
{
|
|
GRUB_ARC_FIRMWARE_VECTOR->close (last_handle);
|
|
grub_free (last_path);
|
|
last_path = NULL;
|
|
last_handle = 0;
|
|
}
|
|
|
|
grub_disk_dev_unregister (&grub_arcdisk_dev);
|
|
}
|