545 lines
15 KiB
C
545 lines
15 KiB
C
/*
|
||
* VasEBoot -- GRand Unified Bootloader
|
||
* Copyright (C) 2002,2003,2004,2006,2007,2008,2009,2010 Free Software Foundation, Inc.
|
||
*
|
||
* VasEBoot 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.
|
||
*
|
||
* VasEBoot 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 VasEBoot. If not, see <http://www.gnu.org/licenses/>.
|
||
*/
|
||
|
||
#include <VasEBoot/disk.h>
|
||
#include <VasEBoot/err.h>
|
||
#include <VasEBoot/mm.h>
|
||
#include <VasEBoot/types.h>
|
||
#include <VasEBoot/partition.h>
|
||
#include <VasEBoot/misc.h>
|
||
#include <VasEBoot/time.h>
|
||
#include <VasEBoot/file.h>
|
||
#include <VasEBoot/i18n.h>
|
||
|
||
#define VasEBoot_CACHE_TIMEOUT 2
|
||
|
||
/* The last time the disk was used. */
|
||
static VasEBoot_uint64_t VasEBoot_last_time = 0;
|
||
|
||
struct VasEBoot_disk_cache VasEBoot_disk_cache_table[VasEBoot_DISK_CACHE_NUM];
|
||
|
||
void (*VasEBoot_disk_firmware_fini) (void);
|
||
int VasEBoot_disk_firmware_is_tainted;
|
||
|
||
#if DISK_CACHE_STATS
|
||
static unsigned long VasEBoot_disk_cache_hits;
|
||
static unsigned long VasEBoot_disk_cache_misses;
|
||
|
||
void
|
||
VasEBoot_disk_cache_get_performance (unsigned long *hits, unsigned long *misses)
|
||
{
|
||
*hits = VasEBoot_disk_cache_hits;
|
||
*misses = VasEBoot_disk_cache_misses;
|
||
}
|
||
#endif
|
||
|
||
VasEBoot_err_t (*VasEBoot_disk_write_weak) (VasEBoot_disk_t disk,
|
||
VasEBoot_disk_addr_t sector,
|
||
VasEBoot_off_t offset,
|
||
VasEBoot_size_t size,
|
||
const void *buf);
|
||
#include "disk_common.c"
|
||
|
||
void
|
||
VasEBoot_disk_cache_invalidate_all (void)
|
||
{
|
||
unsigned i;
|
||
|
||
for (i = 0; i < VasEBoot_DISK_CACHE_NUM; i++)
|
||
{
|
||
struct VasEBoot_disk_cache *cache = VasEBoot_disk_cache_table + i;
|
||
|
||
if (cache->data && ! cache->lock)
|
||
{
|
||
VasEBoot_free (cache->data);
|
||
cache->data = 0;
|
||
}
|
||
}
|
||
}
|
||
|
||
static char *
|
||
VasEBoot_disk_cache_fetch (unsigned long dev_id, unsigned long disk_id,
|
||
VasEBoot_disk_addr_t sector)
|
||
{
|
||
struct VasEBoot_disk_cache *cache;
|
||
unsigned cache_index;
|
||
|
||
cache_index = VasEBoot_disk_cache_get_index (dev_id, disk_id, sector);
|
||
cache = VasEBoot_disk_cache_table + cache_index;
|
||
|
||
if (cache->dev_id == dev_id && cache->disk_id == disk_id
|
||
&& cache->sector == sector)
|
||
{
|
||
cache->lock = 1;
|
||
#if DISK_CACHE_STATS
|
||
VasEBoot_disk_cache_hits++;
|
||
#endif
|
||
return cache->data;
|
||
}
|
||
|
||
#if DISK_CACHE_STATS
|
||
VasEBoot_disk_cache_misses++;
|
||
#endif
|
||
|
||
return 0;
|
||
}
|
||
|
||
static void
|
||
VasEBoot_disk_cache_unlock (unsigned long dev_id, unsigned long disk_id,
|
||
VasEBoot_disk_addr_t sector)
|
||
{
|
||
struct VasEBoot_disk_cache *cache;
|
||
unsigned cache_index;
|
||
|
||
cache_index = VasEBoot_disk_cache_get_index (dev_id, disk_id, sector);
|
||
cache = VasEBoot_disk_cache_table + cache_index;
|
||
|
||
if (cache->dev_id == dev_id && cache->disk_id == disk_id
|
||
&& cache->sector == sector)
|
||
cache->lock = 0;
|
||
}
|
||
|
||
static VasEBoot_err_t
|
||
VasEBoot_disk_cache_store (unsigned long dev_id, unsigned long disk_id,
|
||
VasEBoot_disk_addr_t sector, const char *data)
|
||
{
|
||
unsigned cache_index;
|
||
struct VasEBoot_disk_cache *cache;
|
||
|
||
cache_index = VasEBoot_disk_cache_get_index (dev_id, disk_id, sector);
|
||
cache = VasEBoot_disk_cache_table + cache_index;
|
||
|
||
cache->lock = 1;
|
||
VasEBoot_free (cache->data);
|
||
cache->data = 0;
|
||
cache->lock = 0;
|
||
|
||
cache->data = VasEBoot_malloc (VasEBoot_DISK_SECTOR_SIZE << VasEBoot_DISK_CACHE_BITS);
|
||
if (! cache->data)
|
||
return VasEBoot_errno;
|
||
|
||
VasEBoot_memcpy (cache->data, data,
|
||
VasEBoot_DISK_SECTOR_SIZE << VasEBoot_DISK_CACHE_BITS);
|
||
cache->dev_id = dev_id;
|
||
cache->disk_id = disk_id;
|
||
cache->sector = sector;
|
||
|
||
return VasEBoot_ERR_NONE;
|
||
}
|
||
|
||
|
||
|
||
VasEBoot_disk_dev_t VasEBoot_disk_dev_list;
|
||
|
||
void
|
||
VasEBoot_disk_dev_register (VasEBoot_disk_dev_t dev)
|
||
{
|
||
dev->next = VasEBoot_disk_dev_list;
|
||
VasEBoot_disk_dev_list = dev;
|
||
}
|
||
|
||
void
|
||
VasEBoot_disk_dev_unregister (VasEBoot_disk_dev_t dev)
|
||
{
|
||
VasEBoot_disk_dev_t *p, q;
|
||
|
||
for (p = &VasEBoot_disk_dev_list, q = *p; q; p = &(q->next), q = q->next)
|
||
if (q == dev)
|
||
{
|
||
*p = q->next;
|
||
break;
|
||
}
|
||
}
|
||
|
||
/* Return the location of the first ',', if any, which is not
|
||
escaped by a '\'. */
|
||
static const char *
|
||
find_part_sep (const char *name)
|
||
{
|
||
const char *p = name;
|
||
char c;
|
||
|
||
while ((c = *p++) != '\0')
|
||
{
|
||
if (c == '\\' && *p == ',')
|
||
p++;
|
||
else if (c == ',')
|
||
return p - 1;
|
||
}
|
||
return NULL;
|
||
}
|
||
|
||
VasEBoot_disk_t
|
||
VasEBoot_disk_open (const char *name)
|
||
{
|
||
const char *p;
|
||
VasEBoot_disk_t disk;
|
||
VasEBoot_disk_dev_t dev;
|
||
char *raw = (char *) name;
|
||
VasEBoot_uint64_t current_time;
|
||
|
||
VasEBoot_dprintf ("disk", "Opening `%s'...\n", name);
|
||
|
||
disk = (VasEBoot_disk_t) VasEBoot_zalloc (sizeof (*disk));
|
||
if (! disk)
|
||
return 0;
|
||
disk->log_sector_size = VasEBoot_DISK_SECTOR_BITS;
|
||
/* Default 1MiB of maximum agglomerate. */
|
||
disk->max_agglomerate = 1048576 >> (VasEBoot_DISK_SECTOR_BITS
|
||
+ VasEBoot_DISK_CACHE_BITS);
|
||
|
||
p = find_part_sep (name);
|
||
if (p)
|
||
{
|
||
VasEBoot_size_t len = p - name;
|
||
|
||
raw = VasEBoot_malloc (len + 1);
|
||
if (! raw)
|
||
goto fail;
|
||
|
||
VasEBoot_memcpy (raw, name, len);
|
||
raw[len] = '\0';
|
||
disk->name = VasEBoot_strdup (raw);
|
||
}
|
||
else
|
||
disk->name = VasEBoot_strdup (name);
|
||
if (! disk->name)
|
||
goto fail;
|
||
|
||
for (dev = VasEBoot_disk_dev_list; dev; dev = dev->next)
|
||
{
|
||
if ((dev->open) (raw, disk) == VasEBoot_ERR_NONE)
|
||
break;
|
||
else if (VasEBoot_errno == VasEBoot_ERR_UNKNOWN_DEVICE)
|
||
VasEBoot_errno = VasEBoot_ERR_NONE;
|
||
else
|
||
goto fail;
|
||
}
|
||
|
||
if (! dev)
|
||
{
|
||
VasEBoot_error (VasEBoot_ERR_UNKNOWN_DEVICE, N_("disk `%s' not found"),
|
||
name);
|
||
goto fail;
|
||
}
|
||
if (disk->log_sector_size > VasEBoot_DISK_CACHE_BITS + VasEBoot_DISK_SECTOR_BITS
|
||
|| disk->log_sector_size < VasEBoot_DISK_SECTOR_BITS)
|
||
{
|
||
VasEBoot_error (VasEBoot_ERR_NOT_IMPLEMENTED_YET,
|
||
"sector sizes of %d bytes aren't supported yet",
|
||
(1 << disk->log_sector_size));
|
||
goto fail;
|
||
}
|
||
|
||
disk->dev = dev;
|
||
|
||
if (p)
|
||
{
|
||
disk->partition = VasEBoot_partition_probe (disk, p + 1);
|
||
if (! disk->partition)
|
||
{
|
||
/* TRANSLATORS: It means that the specified partition e.g.
|
||
hd0,msdos1=/dev/sda1 doesn't exist. */
|
||
VasEBoot_error (VasEBoot_ERR_UNKNOWN_DEVICE, N_("no such partition"));
|
||
goto fail;
|
||
}
|
||
}
|
||
|
||
/* The cache will be invalidated about 2 seconds after a device was
|
||
closed. */
|
||
current_time = VasEBoot_get_time_ms ();
|
||
|
||
if (current_time > (VasEBoot_last_time
|
||
+ VasEBoot_CACHE_TIMEOUT * 1000))
|
||
VasEBoot_disk_cache_invalidate_all ();
|
||
|
||
VasEBoot_last_time = current_time;
|
||
|
||
fail:
|
||
|
||
if (raw && raw != name)
|
||
VasEBoot_free (raw);
|
||
|
||
if (VasEBoot_errno != VasEBoot_ERR_NONE)
|
||
{
|
||
VasEBoot_error_push ();
|
||
VasEBoot_dprintf ("disk", "Opening `%s' failed.\n", name);
|
||
VasEBoot_error_pop ();
|
||
|
||
VasEBoot_disk_close (disk);
|
||
return 0;
|
||
}
|
||
|
||
return disk;
|
||
}
|
||
|
||
void
|
||
VasEBoot_disk_close (VasEBoot_disk_t disk)
|
||
{
|
||
VasEBoot_partition_t part;
|
||
VasEBoot_dprintf ("disk", "Closing `%s'.\n", disk->name);
|
||
|
||
if (disk->dev && disk->dev->close)
|
||
(disk->dev->close) (disk);
|
||
|
||
/* Reset the timer. */
|
||
VasEBoot_last_time = VasEBoot_get_time_ms ();
|
||
|
||
while (disk->partition)
|
||
{
|
||
part = disk->partition->parent;
|
||
VasEBoot_free (disk->partition);
|
||
disk->partition = part;
|
||
}
|
||
VasEBoot_free ((void *) disk->name);
|
||
VasEBoot_free (disk);
|
||
}
|
||
|
||
/* Small read (less than cache size and not pass across cache unit boundaries).
|
||
sector is already adjusted and is divisible by cache unit size.
|
||
*/
|
||
static VasEBoot_err_t
|
||
VasEBoot_disk_read_small_real (VasEBoot_disk_t disk, VasEBoot_disk_addr_t sector,
|
||
VasEBoot_off_t offset, VasEBoot_size_t size, void *buf)
|
||
{
|
||
char *data;
|
||
char *tmp_buf;
|
||
|
||
/* Fetch the cache. */
|
||
data = VasEBoot_disk_cache_fetch (disk->dev->id, disk->id, sector);
|
||
if (data)
|
||
{
|
||
/* Just copy it! */
|
||
VasEBoot_memcpy (buf, data + offset, size);
|
||
VasEBoot_disk_cache_unlock (disk->dev->id, disk->id, sector);
|
||
return VasEBoot_ERR_NONE;
|
||
}
|
||
|
||
/* Allocate a temporary buffer. */
|
||
tmp_buf = VasEBoot_malloc (VasEBoot_DISK_SECTOR_SIZE << VasEBoot_DISK_CACHE_BITS);
|
||
if (! tmp_buf)
|
||
return VasEBoot_errno;
|
||
|
||
/* Otherwise read data from the disk actually. */
|
||
if (disk->total_sectors == VasEBoot_DISK_SIZE_UNKNOWN
|
||
|| sector + VasEBoot_DISK_CACHE_SIZE
|
||
< (disk->total_sectors << (disk->log_sector_size - VasEBoot_DISK_SECTOR_BITS)))
|
||
{
|
||
VasEBoot_err_t err;
|
||
err = (disk->dev->read) (disk, transform_sector (disk, sector),
|
||
1U << (VasEBoot_DISK_CACHE_BITS
|
||
+ VasEBoot_DISK_SECTOR_BITS
|
||
- disk->log_sector_size), tmp_buf);
|
||
if (!err)
|
||
{
|
||
/* Copy it and store it in the disk cache. */
|
||
VasEBoot_memcpy (buf, tmp_buf + offset, size);
|
||
VasEBoot_disk_cache_store (disk->dev->id, disk->id,
|
||
sector, tmp_buf);
|
||
VasEBoot_free (tmp_buf);
|
||
return VasEBoot_ERR_NONE;
|
||
}
|
||
}
|
||
|
||
VasEBoot_free (tmp_buf);
|
||
VasEBoot_errno = VasEBoot_ERR_NONE;
|
||
|
||
{
|
||
/* Uggh... Failed. Instead, just read necessary data. */
|
||
unsigned num;
|
||
VasEBoot_disk_addr_t aligned_sector;
|
||
|
||
sector += (offset >> VasEBoot_DISK_SECTOR_BITS);
|
||
offset &= ((1 << VasEBoot_DISK_SECTOR_BITS) - 1);
|
||
aligned_sector = (sector & ~((1ULL << (disk->log_sector_size
|
||
- VasEBoot_DISK_SECTOR_BITS))
|
||
- 1));
|
||
offset += ((sector - aligned_sector) << VasEBoot_DISK_SECTOR_BITS);
|
||
num = ((size + offset + (1ULL << (disk->log_sector_size))
|
||
- 1) >> (disk->log_sector_size));
|
||
|
||
tmp_buf = VasEBoot_malloc (num << disk->log_sector_size);
|
||
if (!tmp_buf)
|
||
return VasEBoot_errno;
|
||
|
||
if ((disk->dev->read) (disk, transform_sector (disk, aligned_sector),
|
||
num, tmp_buf))
|
||
{
|
||
VasEBoot_error_push ();
|
||
VasEBoot_dprintf ("disk", "%s read failed\n", disk->name);
|
||
VasEBoot_error_pop ();
|
||
VasEBoot_free (tmp_buf);
|
||
return VasEBoot_errno;
|
||
}
|
||
VasEBoot_memcpy (buf, tmp_buf + offset, size);
|
||
VasEBoot_free (tmp_buf);
|
||
return VasEBoot_ERR_NONE;
|
||
}
|
||
}
|
||
|
||
static VasEBoot_err_t
|
||
VasEBoot_disk_read_small (VasEBoot_disk_t disk, VasEBoot_disk_addr_t sector,
|
||
VasEBoot_off_t offset, VasEBoot_size_t size, void *buf)
|
||
{
|
||
VasEBoot_err_t err;
|
||
|
||
err = VasEBoot_disk_read_small_real (disk, sector, offset, size, buf);
|
||
if (err)
|
||
return err;
|
||
if (disk->read_hook)
|
||
(disk->read_hook) (sector + (offset >> VasEBoot_DISK_SECTOR_BITS),
|
||
offset & (VasEBoot_DISK_SECTOR_SIZE - 1),
|
||
size, disk->read_hook_data);
|
||
return VasEBoot_ERR_NONE;
|
||
}
|
||
|
||
/* Read data from the disk. */
|
||
VasEBoot_err_t
|
||
VasEBoot_disk_read (VasEBoot_disk_t disk, VasEBoot_disk_addr_t sector,
|
||
VasEBoot_off_t offset, VasEBoot_size_t size, void *buf)
|
||
{
|
||
/* First of all, check if the region is within the disk. */
|
||
if (VasEBoot_disk_adjust_range (disk, §or, &offset, size) != VasEBoot_ERR_NONE)
|
||
{
|
||
VasEBoot_error_push ();
|
||
VasEBoot_dprintf ("disk", "Read out of range: sector 0x%llx (%s).\n",
|
||
(unsigned long long) sector, VasEBoot_errmsg);
|
||
VasEBoot_error_pop ();
|
||
return VasEBoot_errno;
|
||
}
|
||
|
||
/* First read until first cache boundary. */
|
||
if (offset || (sector & (VasEBoot_DISK_CACHE_SIZE - 1)))
|
||
{
|
||
VasEBoot_disk_addr_t start_sector;
|
||
VasEBoot_size_t pos;
|
||
VasEBoot_err_t err;
|
||
VasEBoot_size_t len;
|
||
|
||
start_sector = sector & ~((VasEBoot_disk_addr_t) VasEBoot_DISK_CACHE_SIZE - 1);
|
||
pos = (sector - start_sector) << VasEBoot_DISK_SECTOR_BITS;
|
||
len = ((VasEBoot_DISK_SECTOR_SIZE << VasEBoot_DISK_CACHE_BITS)
|
||
- pos - offset);
|
||
if (len > size)
|
||
len = size;
|
||
err = VasEBoot_disk_read_small (disk, start_sector,
|
||
offset + pos, len, buf);
|
||
if (err)
|
||
return err;
|
||
buf = (char *) buf + len;
|
||
size -= len;
|
||
offset += len;
|
||
sector += (offset >> VasEBoot_DISK_SECTOR_BITS);
|
||
offset &= ((1 << VasEBoot_DISK_SECTOR_BITS) - 1);
|
||
}
|
||
|
||
/* Until SIZE is zero... */
|
||
while (size >= (VasEBoot_DISK_CACHE_SIZE << VasEBoot_DISK_SECTOR_BITS))
|
||
{
|
||
char *data = NULL;
|
||
VasEBoot_disk_addr_t agglomerate;
|
||
VasEBoot_err_t err;
|
||
|
||
/* agglomerate read until we find a first cached entry. */
|
||
for (agglomerate = 0; agglomerate
|
||
< (size >> (VasEBoot_DISK_SECTOR_BITS + VasEBoot_DISK_CACHE_BITS))
|
||
&& agglomerate < disk->max_agglomerate;
|
||
agglomerate++)
|
||
{
|
||
data = VasEBoot_disk_cache_fetch (disk->dev->id, disk->id,
|
||
sector + (agglomerate
|
||
<< VasEBoot_DISK_CACHE_BITS));
|
||
if (data)
|
||
break;
|
||
}
|
||
|
||
if (data)
|
||
{
|
||
VasEBoot_memcpy ((char *) buf
|
||
+ (agglomerate << (VasEBoot_DISK_CACHE_BITS
|
||
+ VasEBoot_DISK_SECTOR_BITS)),
|
||
data, VasEBoot_DISK_CACHE_SIZE << VasEBoot_DISK_SECTOR_BITS);
|
||
VasEBoot_disk_cache_unlock (disk->dev->id, disk->id,
|
||
sector + (agglomerate
|
||
<< VasEBoot_DISK_CACHE_BITS));
|
||
}
|
||
|
||
if (agglomerate)
|
||
{
|
||
VasEBoot_disk_addr_t i;
|
||
|
||
err = (disk->dev->read) (disk, transform_sector (disk, sector),
|
||
agglomerate << (VasEBoot_DISK_CACHE_BITS
|
||
+ VasEBoot_DISK_SECTOR_BITS
|
||
- disk->log_sector_size),
|
||
buf);
|
||
if (err)
|
||
return err;
|
||
|
||
for (i = 0; i < agglomerate; i ++)
|
||
VasEBoot_disk_cache_store (disk->dev->id, disk->id,
|
||
sector + (i << VasEBoot_DISK_CACHE_BITS),
|
||
(char *) buf
|
||
+ (i << (VasEBoot_DISK_CACHE_BITS
|
||
+ VasEBoot_DISK_SECTOR_BITS)));
|
||
|
||
|
||
if (disk->read_hook)
|
||
(disk->read_hook) (sector, 0, agglomerate << (VasEBoot_DISK_CACHE_BITS + VasEBoot_DISK_SECTOR_BITS),
|
||
disk->read_hook_data);
|
||
|
||
sector += agglomerate << VasEBoot_DISK_CACHE_BITS;
|
||
size -= agglomerate << (VasEBoot_DISK_CACHE_BITS + VasEBoot_DISK_SECTOR_BITS);
|
||
buf = (char *) buf
|
||
+ (agglomerate << (VasEBoot_DISK_CACHE_BITS + VasEBoot_DISK_SECTOR_BITS));
|
||
}
|
||
|
||
if (data)
|
||
{
|
||
if (disk->read_hook)
|
||
(disk->read_hook) (sector, 0, (VasEBoot_DISK_CACHE_SIZE << VasEBoot_DISK_SECTOR_BITS),
|
||
disk->read_hook_data);
|
||
sector += VasEBoot_DISK_CACHE_SIZE;
|
||
buf = (char *) buf + (VasEBoot_DISK_CACHE_SIZE << VasEBoot_DISK_SECTOR_BITS);
|
||
size -= (VasEBoot_DISK_CACHE_SIZE << VasEBoot_DISK_SECTOR_BITS);
|
||
}
|
||
}
|
||
|
||
/* And now read the last part. */
|
||
if (size)
|
||
{
|
||
VasEBoot_err_t err;
|
||
err = VasEBoot_disk_read_small (disk, sector, 0, size, buf);
|
||
if (err)
|
||
return err;
|
||
}
|
||
|
||
return VasEBoot_errno;
|
||
}
|
||
|
||
VasEBoot_uint64_t
|
||
VasEBoot_disk_get_size (VasEBoot_disk_t disk)
|
||
{
|
||
if (disk->partition)
|
||
return VasEBoot_partition_get_len (disk->partition);
|
||
else if (disk->total_sectors != VasEBoot_DISK_SIZE_UNKNOWN)
|
||
return disk->total_sectors << (disk->log_sector_size - VasEBoot_DISK_SECTOR_BITS);
|
||
else
|
||
return VasEBoot_DISK_SIZE_UNKNOWN;
|
||
}
|