mirror of
https://github.com/gwenhael-le-moine/x49gp.git
synced 2024-12-26 21:58:41 +01:00
631 lines
14 KiB
C
631 lines
14 KiB
C
/* $Id: block.c,v 1.1 2008/12/11 12:18:17 ecd Exp $
|
|
*/
|
|
|
|
#include <unistd.h>
|
|
#include <stdlib.h>
|
|
#include <stdio.h>
|
|
#include <string.h>
|
|
#include <errno.h>
|
|
#include <sys/stat.h>
|
|
|
|
#ifndef QEMU_OLD
|
|
#include "qemu-common.h"
|
|
#endif
|
|
#include <block.h>
|
|
#ifndef QEMU_OLD
|
|
#include "block_int.h"
|
|
#endif
|
|
|
|
#define SECTOR_BITS 9
|
|
#define SECTOR_SIZE (1 << SECTOR_BITS)
|
|
|
|
static BlockDriverState *bdrv_first;
|
|
static BlockDriver *first_drv;
|
|
|
|
#ifndef ENOMEDIUM
|
|
#define ENOMEDIUM ENODEV
|
|
#endif
|
|
|
|
static void bdrv_close(BlockDriverState *bs);
|
|
|
|
static int
|
|
path_is_absolute(const char *path)
|
|
{
|
|
const char *p;
|
|
#ifdef _WIN32
|
|
/* specific case for names like: "\\.\d:" */
|
|
if (*path == '/' || *path == '\\')
|
|
return 1;
|
|
#endif
|
|
p = strchr(path, ':');
|
|
if (p)
|
|
p++;
|
|
else
|
|
p = path;
|
|
#ifdef _WIN32
|
|
return (*p == '/' || *p == '\\');
|
|
#else
|
|
return (*p == '/');
|
|
#endif
|
|
}
|
|
|
|
/* if filename is absolute, just copy it to dest. Otherwise, build a
|
|
path to it by considering it is relative to base_path. URL are
|
|
supported. */
|
|
static void
|
|
path_combine(char *dest, int dest_size,
|
|
const char *base_path,
|
|
const char *filename)
|
|
{
|
|
const char *p, *p1;
|
|
int len;
|
|
|
|
if (dest_size <= 0)
|
|
return;
|
|
if (path_is_absolute(filename)) {
|
|
pstrcpy(dest, dest_size, filename);
|
|
} else {
|
|
p = strchr(base_path, ':');
|
|
if (p)
|
|
p++;
|
|
else
|
|
p = base_path;
|
|
p1 = strrchr(base_path, '/');
|
|
#ifdef _WIN32
|
|
{
|
|
const char *p2;
|
|
p2 = strrchr(base_path, '\\');
|
|
if (!p1 || p2 > p1)
|
|
p1 = p2;
|
|
}
|
|
#endif
|
|
if (p1)
|
|
p1++;
|
|
else
|
|
p1 = base_path;
|
|
if (p1 > p)
|
|
p = p1;
|
|
len = p - base_path;
|
|
if (len > dest_size - 1)
|
|
len = dest_size - 1;
|
|
memcpy(dest, base_path, len);
|
|
dest[len] = '\0';
|
|
pstrcat(dest, dest_size, filename);
|
|
}
|
|
}
|
|
|
|
#ifdef _WIN32
|
|
void get_tmp_filename(char *filename, int size)
|
|
{
|
|
char temp_dir[MAX_PATH];
|
|
|
|
GetTempPath(MAX_PATH, temp_dir);
|
|
GetTempFileName(temp_dir, "x49", 0, filename);
|
|
}
|
|
#else
|
|
void get_tmp_filename(char *filename, int size)
|
|
{
|
|
int fd;
|
|
|
|
/* XXX: race condition possible */
|
|
pstrcpy(filename, size, "/tmp/x49gp.XXXXXX");
|
|
fd = mkstemp(filename);
|
|
close(fd);
|
|
}
|
|
#endif
|
|
|
|
#ifdef _WIN32
|
|
static int is_windows_drive_prefix(const char *filename)
|
|
{
|
|
return (((filename[0] >= 'a' && filename[0] <= 'z') ||
|
|
(filename[0] >= 'A' && filename[0] <= 'Z')) &&
|
|
filename[1] == ':');
|
|
}
|
|
|
|
static int is_windows_drive(const char *filename)
|
|
{
|
|
if (is_windows_drive_prefix(filename) &&
|
|
filename[2] == '\0')
|
|
return 1;
|
|
if (strstart(filename, "\\\\.\\", NULL) ||
|
|
strstart(filename, "//./", NULL))
|
|
return 1;
|
|
return 0;
|
|
}
|
|
#endif
|
|
|
|
static BlockDriver *
|
|
find_protocol(const char *filename)
|
|
{
|
|
BlockDriver *drv1;
|
|
char protocol[128];
|
|
int len;
|
|
const char *p;
|
|
|
|
#ifdef _WIN32
|
|
if (is_windows_drive(filename) ||
|
|
is_windows_drive_prefix(filename))
|
|
return &bdrv_raw;
|
|
#endif
|
|
p = strchr(filename, ':');
|
|
if (!p)
|
|
return &bdrv_raw;
|
|
len = p - filename;
|
|
if (len > sizeof(protocol) - 1)
|
|
len = sizeof(protocol) - 1;
|
|
memcpy(protocol, filename, len);
|
|
protocol[len] = '\0';
|
|
for(drv1 = first_drv; drv1 != NULL; drv1 = drv1->next) {
|
|
fprintf(stderr, "%s:%u: protocol '%s', drv->protocol_name '%s'\n", __FUNCTION__, __LINE__, protocol, drv1->protocol_name);
|
|
if (drv1->protocol_name &&
|
|
!strcmp(drv1->protocol_name, protocol)) {
|
|
fprintf(stderr, "%s:%u: protocol '%s', drv %p\n", __FUNCTION__, __LINE__, protocol, drv1);
|
|
return drv1;
|
|
}
|
|
}
|
|
fprintf(stderr, "%s:%u: protocol '%s', NULL\n", __FUNCTION__, __LINE__, protocol);
|
|
return NULL;
|
|
}
|
|
|
|
/* XXX: force raw format if block or character device ? It would
|
|
simplify the BSD case */
|
|
static BlockDriver *
|
|
find_image_format(const char *filename)
|
|
{
|
|
int ret, score, score_max;
|
|
BlockDriver *drv1, *drv;
|
|
uint8_t buf[2048];
|
|
BlockDriverState *bs;
|
|
|
|
drv = find_protocol(filename);
|
|
/* no need to test disk image formats for vvfat */
|
|
if (drv == &bdrv_vvfat)
|
|
return drv;
|
|
|
|
ret = bdrv_file_open(&bs, filename, BDRV_O_RDONLY);
|
|
if (ret < 0)
|
|
return NULL;
|
|
ret = bdrv_pread(bs, 0, buf, sizeof(buf));
|
|
bdrv_delete(bs);
|
|
if (ret < 0) {
|
|
return NULL;
|
|
}
|
|
|
|
score_max = 0;
|
|
for(drv1 = first_drv; drv1 != NULL; drv1 = drv1->next) {
|
|
if (drv1->bdrv_probe) {
|
|
score = drv1->bdrv_probe(buf, ret, filename);
|
|
if (score > score_max) {
|
|
score_max = score;
|
|
drv = drv1;
|
|
}
|
|
}
|
|
}
|
|
return drv;
|
|
}
|
|
|
|
int
|
|
bdrv_create(BlockDriver *drv,
|
|
const char *filename, int64_t size_in_sectors,
|
|
const char *backing_file, int flags)
|
|
{
|
|
if (!drv->bdrv_create)
|
|
return -ENOTSUP;
|
|
return drv->bdrv_create(filename, size_in_sectors, backing_file, flags);
|
|
}
|
|
|
|
static void
|
|
bdrv_register(BlockDriver *bdrv)
|
|
{
|
|
bdrv->next = first_drv;
|
|
first_drv = bdrv;
|
|
}
|
|
|
|
/* create a new block device (by default it is empty) */
|
|
BlockDriverState *
|
|
bdrv_new(const char *device_name)
|
|
{
|
|
BlockDriverState **pbs, *bs;
|
|
|
|
bs = qemu_mallocz(sizeof(BlockDriverState));
|
|
if(!bs)
|
|
return NULL;
|
|
pstrcpy(bs->device_name, sizeof(bs->device_name), device_name);
|
|
if (device_name[0] != '\0') {
|
|
/* insert at the end */
|
|
pbs = &bdrv_first;
|
|
while (*pbs != NULL)
|
|
pbs = &(*pbs)->next;
|
|
*pbs = bs;
|
|
}
|
|
return bs;
|
|
}
|
|
|
|
/**
|
|
* Truncate file to 'offset' bytes (needed only for file protocols)
|
|
*/
|
|
int
|
|
bdrv_truncate(BlockDriverState *bs, int64_t offset)
|
|
{
|
|
BlockDriver *drv = bs->drv;
|
|
if (!drv)
|
|
return -ENOMEDIUM;
|
|
if (!drv->bdrv_truncate)
|
|
return -ENOTSUP;
|
|
return drv->bdrv_truncate(bs, offset);
|
|
}
|
|
|
|
/**
|
|
* Length of a file in bytes. Return < 0 if error or unknown.
|
|
*/
|
|
int64_t
|
|
bdrv_getlength(BlockDriverState *bs)
|
|
{
|
|
BlockDriver *drv = bs->drv;
|
|
if (!drv)
|
|
return -ENOMEDIUM;
|
|
if (!drv->bdrv_getlength) {
|
|
/* legacy mode */
|
|
return bs->total_sectors * SECTOR_SIZE;
|
|
}
|
|
return drv->bdrv_getlength(bs);
|
|
}
|
|
|
|
int
|
|
bdrv_file_open(BlockDriverState **pbs, const char *filename, int flags)
|
|
{
|
|
BlockDriverState *bs;
|
|
int ret;
|
|
|
|
fprintf(stderr, "%s:%u: filename '%s'\n", __FUNCTION__, __LINE__, filename);
|
|
|
|
bs = bdrv_new("");
|
|
if (!bs)
|
|
return -ENOMEM;
|
|
ret = bdrv_open(bs, filename, flags | BDRV_O_FILE);
|
|
if (ret < 0) {
|
|
bdrv_delete(bs);
|
|
fprintf(stderr, "%s:%u: '%s': %d\n", __FUNCTION__, __LINE__, filename, ret);
|
|
return ret;
|
|
}
|
|
*pbs = bs;
|
|
fprintf(stderr, "%s:%u: return 0\n", __FUNCTION__, __LINE__);
|
|
return 0;
|
|
}
|
|
|
|
int
|
|
bdrv_open(BlockDriverState *bs, const char *filename, int flags)
|
|
{
|
|
int ret, open_flags;
|
|
char backing_filename[1024];
|
|
BlockDriver *drv = NULL;
|
|
|
|
fprintf(stderr, "%s:%u: filename '%s'\n", __FUNCTION__, __LINE__, filename);
|
|
|
|
bs->read_only = 0;
|
|
bs->is_temporary = 0;
|
|
bs->encrypted = 0;
|
|
|
|
pstrcpy(bs->filename, sizeof(bs->filename), filename);
|
|
if (flags & BDRV_O_FILE) {
|
|
drv = find_protocol(filename);
|
|
if (!drv) {
|
|
fprintf(stderr, "%s:%u: drv: %p\n", __FUNCTION__, __LINE__, drv);
|
|
return -ENOENT;
|
|
}
|
|
} else {
|
|
if (!drv) {
|
|
drv = find_image_format(filename);
|
|
if (!drv) {
|
|
fprintf(stderr, "%s:%u: drv: %p\n", __FUNCTION__, __LINE__, drv);
|
|
return -1;
|
|
}
|
|
}
|
|
}
|
|
|
|
fprintf(stderr, "%s:%u: drv: %p\n", __FUNCTION__, __LINE__, drv);
|
|
bs->drv = drv;
|
|
bs->opaque = qemu_mallocz(drv->instance_size);
|
|
if (bs->opaque == NULL && drv->instance_size > 0) {
|
|
fprintf(stderr, "%s:%u: no opaque\n", __FUNCTION__, __LINE__);
|
|
return -1;
|
|
}
|
|
/* Note: for compatibility, we open disk image files as RDWR, and
|
|
RDONLY as fallback */
|
|
if (!(flags & BDRV_O_FILE))
|
|
open_flags = BDRV_O_RDWR;
|
|
else
|
|
open_flags = flags & ~(BDRV_O_FILE);
|
|
ret = drv->bdrv_open(bs, filename, open_flags);
|
|
fprintf(stderr, "%s:%u: drv->bdrv_open: %d\n", __FUNCTION__, __LINE__, ret);
|
|
if (ret == -EACCES && !(flags & BDRV_O_FILE)) {
|
|
ret = drv->bdrv_open(bs, filename, BDRV_O_RDONLY);
|
|
bs->read_only = 1;
|
|
}
|
|
if (ret < 0) {
|
|
qemu_free(bs->opaque);
|
|
bs->opaque = NULL;
|
|
bs->drv = NULL;
|
|
fprintf(stderr, "%s:%u: return %d\n", __FUNCTION__, __LINE__, ret);
|
|
return ret;
|
|
}
|
|
if (drv->bdrv_getlength) {
|
|
bs->total_sectors = bdrv_getlength(bs) >> SECTOR_BITS;
|
|
}
|
|
#ifndef _WIN32
|
|
if (bs->is_temporary) {
|
|
unlink(filename);
|
|
}
|
|
#endif
|
|
if (bs->backing_file[0] != '\0') {
|
|
/* if there is a backing file, use it */
|
|
bs->backing_hd = bdrv_new("");
|
|
if (!bs->backing_hd) {
|
|
fail:
|
|
bdrv_close(bs);
|
|
fprintf(stderr, "%s:%u: return -ENOMEM\n", __FUNCTION__, __LINE__);
|
|
return -ENOMEM;
|
|
}
|
|
fprintf(stderr, "%s:%u: combine '%s' '%s'\n", __FUNCTION__, __LINE__, filename, bs->backing_file);
|
|
path_combine(backing_filename, sizeof(backing_filename),
|
|
filename, bs->backing_file);
|
|
fprintf(stderr, "%s:%u: combine: '%s'\n", __FUNCTION__, __LINE__, backing_filename);
|
|
if (bdrv_open(bs->backing_hd, backing_filename, 0) < 0) {
|
|
fprintf(stderr, "%s:%u: backing fail\n", __FUNCTION__, __LINE__);
|
|
goto fail;
|
|
}
|
|
}
|
|
|
|
/* call the change callback */
|
|
bs->media_changed = 1;
|
|
if (bs->change_cb)
|
|
bs->change_cb(bs->change_opaque);
|
|
|
|
fprintf(stderr, "%s:%u: return 0\n", __FUNCTION__, __LINE__);
|
|
return 0;
|
|
}
|
|
|
|
static void
|
|
bdrv_close(BlockDriverState *bs)
|
|
{
|
|
if (NULL == bs->drv)
|
|
return;
|
|
|
|
/* call the change callback */
|
|
bs->media_changed = 1;
|
|
if (bs->change_cb)
|
|
bs->change_cb(bs->change_opaque);
|
|
|
|
if (bs->backing_hd)
|
|
bdrv_delete(bs->backing_hd);
|
|
|
|
bs->drv->bdrv_close(bs);
|
|
|
|
#ifdef _WIN32
|
|
if (bs->is_temporary) {
|
|
unlink(bs->filename);
|
|
}
|
|
#endif
|
|
|
|
qemu_free(bs->opaque);
|
|
bs->opaque = NULL;
|
|
bs->drv = NULL;
|
|
}
|
|
|
|
void
|
|
bdrv_delete(BlockDriverState *bs)
|
|
{
|
|
/* XXX: remove the driver list */
|
|
bdrv_close(bs);
|
|
qemu_free(bs);
|
|
}
|
|
|
|
/* return < 0 if error. See bdrv_write() for the return codes */
|
|
int
|
|
bdrv_read(BlockDriverState * bs, int64_t sector_num,
|
|
uint8_t * buf, int nb_sectors)
|
|
{
|
|
BlockDriver *drv = bs->drv;
|
|
|
|
if (!drv)
|
|
return -ENOMEDIUM;
|
|
|
|
if (sector_num == 0 && bs->boot_sector_enabled && nb_sectors > 0) {
|
|
memcpy(buf, bs->boot_sector_data, 512);
|
|
sector_num++;
|
|
nb_sectors--;
|
|
buf += 512;
|
|
if (nb_sectors == 0)
|
|
return 0;
|
|
}
|
|
if (drv->bdrv_pread) {
|
|
int ret, len;
|
|
|
|
len = nb_sectors * 512;
|
|
ret = drv->bdrv_pread(bs, sector_num * 512, buf, len);
|
|
if (ret < 0)
|
|
return ret;
|
|
else if (ret != len)
|
|
return -EINVAL;
|
|
else
|
|
return 0;
|
|
} else {
|
|
return drv->bdrv_read(bs, sector_num, buf, nb_sectors);
|
|
}
|
|
}
|
|
|
|
/* Return < 0 if error. Important errors are:
|
|
-EIO generic I/O error (may happen for all errors)
|
|
-ENOMEDIUM No media inserted.
|
|
-EINVAL Invalid sector number or nb_sectors
|
|
-EACCES Trying to write a read-only device
|
|
*/
|
|
static int
|
|
bdrv_write(BlockDriverState * bs, int64_t sector_num,
|
|
const uint8_t * buf, int nb_sectors)
|
|
{
|
|
BlockDriver *drv = bs->drv;
|
|
|
|
if (!bs->drv)
|
|
return -ENOMEDIUM;
|
|
if (bs->read_only)
|
|
return -EACCES;
|
|
if (sector_num == 0 && bs->boot_sector_enabled && nb_sectors > 0) {
|
|
memcpy(bs->boot_sector_data, buf, 512);
|
|
}
|
|
if (drv->bdrv_pwrite) {
|
|
int ret, len;
|
|
|
|
len = nb_sectors * 512;
|
|
ret = drv->bdrv_pwrite(bs, sector_num * 512, buf, len);
|
|
if (ret < 0)
|
|
return ret;
|
|
else if (ret != len)
|
|
return -EIO;
|
|
else
|
|
return 0;
|
|
} else {
|
|
return drv->bdrv_write(bs, sector_num, buf, nb_sectors);
|
|
}
|
|
}
|
|
|
|
static int
|
|
bdrv_pread_em(BlockDriverState * bs, int64_t offset, uint8_t * buf, int count1)
|
|
{
|
|
uint8_t tmp_buf[SECTOR_SIZE];
|
|
int len, nb_sectors, count;
|
|
int64_t sector_num;
|
|
|
|
count = count1;
|
|
/* first read to align to sector start */
|
|
len = (SECTOR_SIZE - offset) & (SECTOR_SIZE - 1);
|
|
if (len > count)
|
|
len = count;
|
|
sector_num = offset >> SECTOR_BITS;
|
|
if (len > 0) {
|
|
if (bdrv_read(bs, sector_num, tmp_buf, 1) < 0)
|
|
return -EIO;
|
|
memcpy(buf, tmp_buf + (offset & (SECTOR_SIZE - 1)), len);
|
|
count -= len;
|
|
if (count == 0)
|
|
return count1;
|
|
sector_num++;
|
|
buf += len;
|
|
}
|
|
|
|
/* read the sectors "in place" */
|
|
nb_sectors = count >> SECTOR_BITS;
|
|
if (nb_sectors > 0) {
|
|
if (bdrv_read(bs, sector_num, buf, nb_sectors) < 0)
|
|
return -EIO;
|
|
sector_num += nb_sectors;
|
|
len = nb_sectors << SECTOR_BITS;
|
|
buf += len;
|
|
count -= len;
|
|
}
|
|
|
|
/* add data from the last sector */
|
|
if (count > 0) {
|
|
if (bdrv_read(bs, sector_num, tmp_buf, 1) < 0)
|
|
return -EIO;
|
|
memcpy(buf, tmp_buf, count);
|
|
}
|
|
return count1;
|
|
}
|
|
|
|
static int
|
|
bdrv_pwrite_em(BlockDriverState * bs, int64_t offset,
|
|
const uint8_t * buf, int count1)
|
|
{
|
|
uint8_t tmp_buf[SECTOR_SIZE];
|
|
int len, nb_sectors, count;
|
|
int64_t sector_num;
|
|
|
|
count = count1;
|
|
/* first write to align to sector start */
|
|
len = (SECTOR_SIZE - offset) & (SECTOR_SIZE - 1);
|
|
if (len > count)
|
|
len = count;
|
|
sector_num = offset >> SECTOR_BITS;
|
|
if (len > 0) {
|
|
if (bdrv_read(bs, sector_num, tmp_buf, 1) < 0)
|
|
return -EIO;
|
|
memcpy(tmp_buf + (offset & (SECTOR_SIZE - 1)), buf, len);
|
|
if (bdrv_write(bs, sector_num, tmp_buf, 1) < 0)
|
|
return -EIO;
|
|
count -= len;
|
|
if (count == 0)
|
|
return count1;
|
|
sector_num++;
|
|
buf += len;
|
|
}
|
|
|
|
/* write the sectors "in place" */
|
|
nb_sectors = count >> SECTOR_BITS;
|
|
if (nb_sectors > 0) {
|
|
if (bdrv_write(bs, sector_num, buf, nb_sectors) < 0)
|
|
return -EIO;
|
|
sector_num += nb_sectors;
|
|
len = nb_sectors << SECTOR_BITS;
|
|
buf += len;
|
|
count -= len;
|
|
}
|
|
|
|
/* add data from the last sector */
|
|
if (count > 0) {
|
|
if (bdrv_read(bs, sector_num, tmp_buf, 1) < 0)
|
|
return -EIO;
|
|
memcpy(tmp_buf, buf, count);
|
|
if (bdrv_write(bs, sector_num, tmp_buf, 1) < 0)
|
|
return -EIO;
|
|
}
|
|
return count1;
|
|
}
|
|
|
|
/**
|
|
* Read with byte offsets (needed only for file protocols)
|
|
*/
|
|
int
|
|
bdrv_pread(BlockDriverState * bs, int64_t offset, void *buf1, int count1)
|
|
{
|
|
BlockDriver *drv = bs->drv;
|
|
|
|
if (!drv)
|
|
return -ENOMEDIUM;
|
|
if (!drv->bdrv_pread)
|
|
return bdrv_pread_em(bs, offset, buf1, count1);
|
|
return drv->bdrv_pread(bs, offset, buf1, count1);
|
|
}
|
|
|
|
/**
|
|
* Write with byte offsets (needed only for file protocols)
|
|
*/
|
|
int
|
|
bdrv_pwrite(BlockDriverState * bs, int64_t offset, const void *buf1, int count1)
|
|
{
|
|
BlockDriver *drv = bs->drv;
|
|
|
|
if (!drv)
|
|
return -ENOMEDIUM;
|
|
if (!drv->bdrv_pwrite)
|
|
return bdrv_pwrite_em(bs, offset, buf1, count1);
|
|
return drv->bdrv_pwrite(bs, offset, buf1, count1);
|
|
}
|
|
|
|
void
|
|
bdrv_flush(BlockDriverState *bs)
|
|
{
|
|
if (bs->drv->bdrv_flush)
|
|
bs->drv->bdrv_flush(bs);
|
|
if (bs->backing_hd)
|
|
bdrv_flush(bs->backing_hd);
|
|
}
|
|
|
|
void
|
|
bdrv_init(void)
|
|
{
|
|
/* bdrv_register(&bdrv_raw); */
|
|
/* bdrv_register(&bdrv_host_device); */
|
|
bdrv_register(&bdrv_qcow);
|
|
bdrv_register(&bdrv_vvfat);
|
|
}
|