* OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
* THE SOFTWARE.
*/
-#include <stdlib.h>
-#include <stdio.h>
-#include <stdarg.h>
-#include <string.h>
-#include <getopt.h>
-#include <inttypes.h>
-#include <unistd.h>
-#include <sys/mman.h>
-#include <fcntl.h>
-#include <signal.h>
-#include <time.h>
-#include <sys/time.h>
-#include <malloc.h>
-#include <termios.h>
-#include <sys/poll.h>
-#include <errno.h>
-#include <sys/wait.h>
-#include <netinet/in.h>
-
#include "vl.h"
-#define NO_THUNK_TYPE_SIZE
-#include "thunk.h"
+#ifndef _WIN32
+#include <sys/mman.h>
+#endif
+
+#include "cow.h"
struct BlockDriverState {
int fd; /* if -1, only COW mappings */
int64_t total_sectors;
- int read_only;
+ int read_only; /* if true, the media is read only */
+ int inserted; /* if true, the media is present */
+ int removable; /* if true, the media can be removed */
+ int locked; /* if true, the media cannot temporarily be ejected */
+ /* event callback when inserting/removing */
+ void (*change_cb)(void *opaque);
+ void *change_opaque;
uint8_t *cow_bitmap; /* if non NULL, COW mappings are used first */
uint8_t *cow_bitmap_addr; /* mmap address of cow_bitmap */
int cow_bitmap_size;
int cow_fd;
int64_t cow_sectors_offset;
+ int boot_sector_enabled;
+ uint8_t boot_sector_data[512];
+
char filename[1024];
+
+ /* NOTE: the following infos are only hints for real hardware
+ drivers. They are not used by the block driver */
+ int cyls, heads, secs;
+ int type;
+ char device_name[32];
+ BlockDriverState *next;
};
-BlockDriverState *bdrv_open(const char *filename, int snapshot)
+static BlockDriverState *bdrv_first;
+
+/* create a new block device (by default it is empty) */
+BlockDriverState *bdrv_new(const char *device_name)
{
- BlockDriverState *bs;
- int fd, cow_fd;
+ BlockDriverState **pbs, *bs;
+
+ bs = qemu_mallocz(sizeof(BlockDriverState));
+ if(!bs)
+ return NULL;
+ pstrcpy(bs->device_name, sizeof(bs->device_name), device_name);
+ /* insert at the end */
+ pbs = &bdrv_first;
+ while (*pbs != NULL)
+ pbs = &(*pbs)->next;
+ *pbs = bs;
+ return bs;
+}
+
+int bdrv_open(BlockDriverState *bs, const char *filename, int snapshot)
+{
+ int fd;
int64_t size;
- char template[] = "/tmp/vl.XXXXXX";
struct cow_header_v2 cow_header;
+#ifndef _WIN32
+ char template[] = "/tmp/vl.XXXXXX";
+ int cow_fd;
struct stat st;
+#endif
- bs = malloc(sizeof(BlockDriverState));
- if(!bs)
- return NULL;
bs->read_only = 0;
bs->fd = -1;
bs->cow_fd = -1;
strcpy(bs->filename, filename);
/* open standard HD image */
+#ifdef _WIN32
+ fd = open(filename, O_RDWR | O_BINARY);
+#else
fd = open(filename, O_RDWR | O_LARGEFILE);
+#endif
if (fd < 0) {
/* read only image on disk */
+#ifdef _WIN32
+ fd = open(filename, O_RDONLY | O_BINARY);
+#else
fd = open(filename, O_RDONLY | O_LARGEFILE);
+#endif
if (fd < 0) {
perror(filename);
goto fail;
fprintf(stderr, "%s: could not read header\n", filename);
goto fail;
}
- if (cow_header.magic == htonl(COW_MAGIC) &&
- cow_header.version == htonl(COW_VERSION)) {
+#ifndef _WIN32
+ if (be32_to_cpu(cow_header.magic) == COW_MAGIC &&
+ be32_to_cpu(cow_header.version) == COW_VERSION) {
/* cow image found */
size = cow_header.size;
#ifndef WORDS_BIGENDIAN
fprintf(stderr, "%s: could not find original disk image '%s'\n", filename, cow_header.backing_file);
goto fail;
}
- if (st.st_mtime != htonl(cow_header.mtime)) {
+ if (st.st_mtime != be32_to_cpu(cow_header.mtime)) {
fprintf(stderr, "%s: original raw disk image '%s' does not match saved timestamp\n", filename, cow_header.backing_file);
goto fail;
}
bs->cow_bitmap = bs->cow_bitmap_addr + sizeof(cow_header);
bs->cow_sectors_offset = (bs->cow_bitmap_size + 511) & ~511;
snapshot = 0;
- } else {
+ } else
+#endif
+ {
/* standard raw image */
size = lseek64(fd, 0, SEEK_END);
bs->total_sectors = size / 512;
bs->fd = fd;
}
+#ifndef _WIN32
if (snapshot) {
/* create a temporary COW file */
- cow_fd = mkstemp(template);
+ cow_fd = mkstemp64(template);
if (cow_fd < 0)
goto fail;
bs->cow_fd = cow_fd;
bs->cow_bitmap = bs->cow_bitmap_addr;
bs->cow_sectors_offset = 0;
}
+#endif
- return bs;
+ bs->inserted = 1;
+
+ /* call the change callback */
+ if (bs->change_cb)
+ bs->change_cb(bs->change_opaque);
+
+ return 0;
fail:
bdrv_close(bs);
- return NULL;
+ return -1;
}
void bdrv_close(BlockDriverState *bs)
{
- /* we unmap the mapping so that it is written to the COW file */
- if (bs->cow_bitmap_addr)
- munmap(bs->cow_bitmap_addr, bs->cow_bitmap_size);
- if (bs->cow_fd >= 0)
- close(bs->cow_fd);
- if (bs->fd >= 0)
- close(bs->fd);
- free(bs);
+ if (bs->inserted) {
+#ifndef _WIN32
+ /* we unmap the mapping so that it is written to the COW file */
+ if (bs->cow_bitmap_addr)
+ munmap(bs->cow_bitmap_addr, bs->cow_bitmap_size);
+#endif
+ if (bs->cow_fd >= 0)
+ close(bs->cow_fd);
+ if (bs->fd >= 0)
+ close(bs->fd);
+ bs->inserted = 0;
+
+ /* call the change callback */
+ if (bs->change_cb)
+ bs->change_cb(bs->change_opaque);
+ }
+}
+
+void bdrv_delete(BlockDriverState *bs)
+{
+ bdrv_close(bs);
+ qemu_free(bs);
}
static inline void set_bit(uint8_t *bitmap, int64_t bitnum)
int64_t i;
uint8_t *cow_bitmap;
+ if (!bs->inserted)
+ return -1;
+
if (!bs->cow_bitmap) {
fprintf(stderr, "Already committed to %s\n", bs->filename);
return 0;
int ret, n, fd;
int64_t offset;
+ if (!bs->inserted)
+ return -1;
+
while (nb_sectors > 0) {
if (is_changed(bs->cow_bitmap, sector_num, nb_sectors, &n)) {
fd = bs->cow_fd;
offset = bs->cow_sectors_offset;
+ } else if (sector_num == 0 && bs->boot_sector_enabled) {
+ memcpy(buf, bs->boot_sector_data, 512);
+ n = 1;
+ goto next;
} else {
fd = bs->fd;
offset = 0;
return -1;
}
}
+ next:
nb_sectors -= n;
sector_num += n;
buf += n * 512;
{
int ret, fd, i;
int64_t offset, retl;
-
+
+ if (!bs->inserted)
+ return -1;
if (bs->read_only)
return -1;
{
*nb_sectors_ptr = bs->total_sectors;
}
+
+/* force a given boot sector. */
+void bdrv_set_boot_sector(BlockDriverState *bs, const uint8_t *data, int size)
+{
+ bs->boot_sector_enabled = 1;
+ if (size > 512)
+ size = 512;
+ memcpy(bs->boot_sector_data, data, size);
+ memset(bs->boot_sector_data + size, 0, 512 - size);
+}
+
+void bdrv_set_geometry_hint(BlockDriverState *bs,
+ int cyls, int heads, int secs)
+{
+ bs->cyls = cyls;
+ bs->heads = heads;
+ bs->secs = secs;
+}
+
+void bdrv_set_type_hint(BlockDriverState *bs, int type)
+{
+ bs->type = type;
+ bs->removable = ((type == BDRV_TYPE_CDROM ||
+ type == BDRV_TYPE_FLOPPY));
+}
+
+void bdrv_get_geometry_hint(BlockDriverState *bs,
+ int *pcyls, int *pheads, int *psecs)
+{
+ *pcyls = bs->cyls;
+ *pheads = bs->heads;
+ *psecs = bs->secs;
+}
+
+int bdrv_get_type_hint(BlockDriverState *bs)
+{
+ return bs->type;
+}
+
+int bdrv_is_removable(BlockDriverState *bs)
+{
+ return bs->removable;
+}
+
+int bdrv_is_read_only(BlockDriverState *bs)
+{
+ return bs->read_only;
+}
+
+int bdrv_is_inserted(BlockDriverState *bs)
+{
+ return bs->inserted;
+}
+
+int bdrv_is_locked(BlockDriverState *bs)
+{
+ return bs->locked;
+}
+
+void bdrv_set_locked(BlockDriverState *bs, int locked)
+{
+ bs->locked = locked;
+}
+
+void bdrv_set_change_cb(BlockDriverState *bs,
+ void (*change_cb)(void *opaque), void *opaque)
+{
+ bs->change_cb = change_cb;
+ bs->change_opaque = opaque;
+}
+
+BlockDriverState *bdrv_find(const char *name)
+{
+ BlockDriverState *bs;
+
+ for (bs = bdrv_first; bs != NULL; bs = bs->next) {
+ if (!strcmp(name, bs->device_name))
+ return bs;
+ }
+ return NULL;
+}
+
+void bdrv_info(void)
+{
+ BlockDriverState *bs;
+
+ for (bs = bdrv_first; bs != NULL; bs = bs->next) {
+ term_printf("%s:", bs->device_name);
+ term_printf(" type=");
+ switch(bs->type) {
+ case BDRV_TYPE_HD:
+ term_printf("hd");
+ break;
+ case BDRV_TYPE_CDROM:
+ term_printf("cdrom");
+ break;
+ case BDRV_TYPE_FLOPPY:
+ term_printf("floppy");
+ break;
+ }
+ term_printf(" removable=%d", bs->removable);
+ if (bs->removable) {
+ term_printf(" locked=%d", bs->locked);
+ }
+ if (bs->inserted) {
+ term_printf(" file=%s", bs->filename);
+ term_printf(" ro=%d", bs->read_only);
+ } else {
+ term_printf(" [not inserted]");
+ }
+ term_printf("\n");
+ }
+}