userboot/userboot/userboot_disk.c

199767f8919635c4928607450d9e0abb932109ceToomas Soome/*-
199767f8919635c4928607450d9e0abb932109ceToomas Soome * Copyright (c) 2011 Google, Inc.
199767f8919635c4928607450d9e0abb932109ceToomas Soome * All rights reserved.
199767f8919635c4928607450d9e0abb932109ceToomas Soome *
199767f8919635c4928607450d9e0abb932109ceToomas Soome * Redistribution and use in source and binary forms, with or without
199767f8919635c4928607450d9e0abb932109ceToomas Soome * modification, are permitted provided that the following conditions
199767f8919635c4928607450d9e0abb932109ceToomas Soome * are met:
199767f8919635c4928607450d9e0abb932109ceToomas Soome * 1. Redistributions of source code must retain the above copyright
199767f8919635c4928607450d9e0abb932109ceToomas Soome *    notice, this list of conditions and the following disclaimer.
199767f8919635c4928607450d9e0abb932109ceToomas Soome * 2. Redistributions in binary form must reproduce the above copyright
199767f8919635c4928607450d9e0abb932109ceToomas Soome *    notice, this list of conditions and the following disclaimer in the
199767f8919635c4928607450d9e0abb932109ceToomas Soome *    documentation and/or other materials provided with the distribution.
199767f8919635c4928607450d9e0abb932109ceToomas Soome *
199767f8919635c4928607450d9e0abb932109ceToomas Soome * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
199767f8919635c4928607450d9e0abb932109ceToomas Soome * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
199767f8919635c4928607450d9e0abb932109ceToomas Soome * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
199767f8919635c4928607450d9e0abb932109ceToomas Soome * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
199767f8919635c4928607450d9e0abb932109ceToomas Soome * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
199767f8919635c4928607450d9e0abb932109ceToomas Soome * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
199767f8919635c4928607450d9e0abb932109ceToomas Soome * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
199767f8919635c4928607450d9e0abb932109ceToomas Soome * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
199767f8919635c4928607450d9e0abb932109ceToomas Soome * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
199767f8919635c4928607450d9e0abb932109ceToomas Soome * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
199767f8919635c4928607450d9e0abb932109ceToomas Soome * SUCH DAMAGE.
199767f8919635c4928607450d9e0abb932109ceToomas Soome */
199767f8919635c4928607450d9e0abb932109ceToomas Soome
199767f8919635c4928607450d9e0abb932109ceToomas Soome#include <sys/cdefs.h>
199767f8919635c4928607450d9e0abb932109ceToomas Soome__FBSDID("$FreeBSD$");
199767f8919635c4928607450d9e0abb932109ceToomas Soome
199767f8919635c4928607450d9e0abb932109ceToomas Soome/*
199767f8919635c4928607450d9e0abb932109ceToomas Soome * Userboot disk image handling.
199767f8919635c4928607450d9e0abb932109ceToomas Soome */
199767f8919635c4928607450d9e0abb932109ceToomas Soome
199767f8919635c4928607450d9e0abb932109ceToomas Soome#include <sys/disk.h>
199767f8919635c4928607450d9e0abb932109ceToomas Soome#include <stand.h>
199767f8919635c4928607450d9e0abb932109ceToomas Soome#include <stdarg.h>
199767f8919635c4928607450d9e0abb932109ceToomas Soome#include <bootstrap.h>
199767f8919635c4928607450d9e0abb932109ceToomas Soome
199767f8919635c4928607450d9e0abb932109ceToomas Soome#include "disk.h"
199767f8919635c4928607450d9e0abb932109ceToomas Soome#include "libuserboot.h"
199767f8919635c4928607450d9e0abb932109ceToomas Soome
199767f8919635c4928607450d9e0abb932109ceToomas Soomestruct userdisk_info {
199767f8919635c4928607450d9e0abb932109ceToomas Soome    uint64_t    mediasize;
199767f8919635c4928607450d9e0abb932109ceToomas Soome    uint16_t    sectorsize;
199767f8919635c4928607450d9e0abb932109ceToomas Soome};
199767f8919635c4928607450d9e0abb932109ceToomas Soome
199767f8919635c4928607450d9e0abb932109ceToomas Soomeint userboot_disk_maxunit = 0;
199767f8919635c4928607450d9e0abb932109ceToomas Soome
199767f8919635c4928607450d9e0abb932109ceToomas Soomestatic int userdisk_maxunit = 0;
199767f8919635c4928607450d9e0abb932109ceToomas Soomestatic struct userdisk_info *ud_info;
199767f8919635c4928607450d9e0abb932109ceToomas Soome
199767f8919635c4928607450d9e0abb932109ceToomas Soomestatic int  userdisk_init(void);
199767f8919635c4928607450d9e0abb932109ceToomas Soomestatic void userdisk_cleanup(void);
199767f8919635c4928607450d9e0abb932109ceToomas Soomestatic int  userdisk_strategy(void *devdata, int flag, daddr_t dblk,
199767f8919635c4928607450d9e0abb932109ceToomas Soome            size_t size, char *buf, size_t *rsize);
199767f8919635c4928607450d9e0abb932109ceToomas Soomestatic int  userdisk_open(struct open_file *f, ...);
199767f8919635c4928607450d9e0abb932109ceToomas Soomestatic int  userdisk_close(struct open_file *f);
199767f8919635c4928607450d9e0abb932109ceToomas Soomestatic int  userdisk_ioctl(struct open_file *f, u_long cmd, void *data);
199767f8919635c4928607450d9e0abb932109ceToomas Soomestatic void userdisk_print(int verbose);
199767f8919635c4928607450d9e0abb932109ceToomas Soome
199767f8919635c4928607450d9e0abb932109ceToomas Soomestruct devsw userboot_disk = {
199767f8919635c4928607450d9e0abb932109ceToomas Soome    "disk",
199767f8919635c4928607450d9e0abb932109ceToomas Soome    DEVT_DISK,
199767f8919635c4928607450d9e0abb932109ceToomas Soome    userdisk_init,
199767f8919635c4928607450d9e0abb932109ceToomas Soome    userdisk_strategy,
199767f8919635c4928607450d9e0abb932109ceToomas Soome    userdisk_open,
199767f8919635c4928607450d9e0abb932109ceToomas Soome    userdisk_close,
199767f8919635c4928607450d9e0abb932109ceToomas Soome    userdisk_ioctl,
199767f8919635c4928607450d9e0abb932109ceToomas Soome    userdisk_print,
199767f8919635c4928607450d9e0abb932109ceToomas Soome    userdisk_cleanup
199767f8919635c4928607450d9e0abb932109ceToomas Soome};
199767f8919635c4928607450d9e0abb932109ceToomas Soome
199767f8919635c4928607450d9e0abb932109ceToomas Soome/*
199767f8919635c4928607450d9e0abb932109ceToomas Soome * Initialize userdisk_info structure for each disk.
199767f8919635c4928607450d9e0abb932109ceToomas Soome */
199767f8919635c4928607450d9e0abb932109ceToomas Soomestatic int
199767f8919635c4928607450d9e0abb932109ceToomas Soomeuserdisk_init(void)
199767f8919635c4928607450d9e0abb932109ceToomas Soome{
199767f8919635c4928607450d9e0abb932109ceToomas Soome    off_t mediasize;
199767f8919635c4928607450d9e0abb932109ceToomas Soome    u_int sectorsize;
199767f8919635c4928607450d9e0abb932109ceToomas Soome    int i;
199767f8919635c4928607450d9e0abb932109ceToomas Soome
199767f8919635c4928607450d9e0abb932109ceToomas Soome    userdisk_maxunit = userboot_disk_maxunit;
199767f8919635c4928607450d9e0abb932109ceToomas Soome    if (userdisk_maxunit > 0) {
199767f8919635c4928607450d9e0abb932109ceToomas Soome        ud_info = malloc(sizeof(*ud_info) * userdisk_maxunit);
199767f8919635c4928607450d9e0abb932109ceToomas Soome        if (ud_info == NULL)
199767f8919635c4928607450d9e0abb932109ceToomas Soome            return (ENOMEM);
199767f8919635c4928607450d9e0abb932109ceToomas Soome        for (i = 0; i < userdisk_maxunit; i++) {
199767f8919635c4928607450d9e0abb932109ceToomas Soome            if (CALLBACK(diskioctl, i, DIOCGSECTORSIZE,
199767f8919635c4928607450d9e0abb932109ceToomas Soome                &sectorsize) != 0 || CALLBACK(diskioctl, i,
199767f8919635c4928607450d9e0abb932109ceToomas Soome                DIOCGMEDIASIZE, &mediasize) != 0)
199767f8919635c4928607450d9e0abb932109ceToomas Soome                return (ENXIO);
199767f8919635c4928607450d9e0abb932109ceToomas Soome            ud_info[i].mediasize = mediasize;
199767f8919635c4928607450d9e0abb932109ceToomas Soome            ud_info[i].sectorsize = sectorsize;
199767f8919635c4928607450d9e0abb932109ceToomas Soome        }
199767f8919635c4928607450d9e0abb932109ceToomas Soome    }
199767f8919635c4928607450d9e0abb932109ceToomas Soome
199767f8919635c4928607450d9e0abb932109ceToomas Soome    return(0);
199767f8919635c4928607450d9e0abb932109ceToomas Soome}
199767f8919635c4928607450d9e0abb932109ceToomas Soome
199767f8919635c4928607450d9e0abb932109ceToomas Soomestatic void
199767f8919635c4928607450d9e0abb932109ceToomas Soomeuserdisk_cleanup(void)
199767f8919635c4928607450d9e0abb932109ceToomas Soome{
199767f8919635c4928607450d9e0abb932109ceToomas Soome
199767f8919635c4928607450d9e0abb932109ceToomas Soome    if (userdisk_maxunit > 0)
199767f8919635c4928607450d9e0abb932109ceToomas Soome        free(ud_info);
199767f8919635c4928607450d9e0abb932109ceToomas Soome    disk_cleanup(&userboot_disk);
199767f8919635c4928607450d9e0abb932109ceToomas Soome}
199767f8919635c4928607450d9e0abb932109ceToomas Soome
199767f8919635c4928607450d9e0abb932109ceToomas Soome/*
199767f8919635c4928607450d9e0abb932109ceToomas Soome * Print information about disks
199767f8919635c4928607450d9e0abb932109ceToomas Soome */
199767f8919635c4928607450d9e0abb932109ceToomas Soomestatic void
199767f8919635c4928607450d9e0abb932109ceToomas Soomeuserdisk_print(int verbose)
199767f8919635c4928607450d9e0abb932109ceToomas Soome{
199767f8919635c4928607450d9e0abb932109ceToomas Soome    struct disk_devdesc dev;
199767f8919635c4928607450d9e0abb932109ceToomas Soome    char line[80];
199767f8919635c4928607450d9e0abb932109ceToomas Soome    int i;
199767f8919635c4928607450d9e0abb932109ceToomas Soome
199767f8919635c4928607450d9e0abb932109ceToomas Soome    for (i = 0; i < userdisk_maxunit; i++) {
199767f8919635c4928607450d9e0abb932109ceToomas Soome        sprintf(line, "    disk%d:   Guest drive image\n", i);
199767f8919635c4928607450d9e0abb932109ceToomas Soome        pager_output(line);
199767f8919635c4928607450d9e0abb932109ceToomas Soome        dev.d_dev = &userboot_disk;
199767f8919635c4928607450d9e0abb932109ceToomas Soome        dev.d_unit = i;
199767f8919635c4928607450d9e0abb932109ceToomas Soome        dev.d_slice = -1;
199767f8919635c4928607450d9e0abb932109ceToomas Soome        dev.d_partition = -1;
199767f8919635c4928607450d9e0abb932109ceToomas Soome        if (disk_open(&dev, ud_info[i].mediasize,
199767f8919635c4928607450d9e0abb932109ceToomas Soome            ud_info[i].sectorsize, 0) == 0) {
199767f8919635c4928607450d9e0abb932109ceToomas Soome            sprintf(line, "    disk%d", i);
199767f8919635c4928607450d9e0abb932109ceToomas Soome            disk_print(&dev, line, verbose);
199767f8919635c4928607450d9e0abb932109ceToomas Soome            disk_close(&dev);
199767f8919635c4928607450d9e0abb932109ceToomas Soome        }
199767f8919635c4928607450d9e0abb932109ceToomas Soome    }
199767f8919635c4928607450d9e0abb932109ceToomas Soome}
199767f8919635c4928607450d9e0abb932109ceToomas Soome
199767f8919635c4928607450d9e0abb932109ceToomas Soome/*
199767f8919635c4928607450d9e0abb932109ceToomas Soome * Attempt to open the disk described by (dev) for use by (f).
199767f8919635c4928607450d9e0abb932109ceToomas Soome */
199767f8919635c4928607450d9e0abb932109ceToomas Soomestatic int
199767f8919635c4928607450d9e0abb932109ceToomas Soomeuserdisk_open(struct open_file *f, ...)
199767f8919635c4928607450d9e0abb932109ceToomas Soome{
199767f8919635c4928607450d9e0abb932109ceToomas Soome    va_list         ap;
199767f8919635c4928607450d9e0abb932109ceToomas Soome    struct disk_devdesc *dev;
199767f8919635c4928607450d9e0abb932109ceToomas Soome
199767f8919635c4928607450d9e0abb932109ceToomas Soome    va_start(ap, f);
199767f8919635c4928607450d9e0abb932109ceToomas Soome    dev = va_arg(ap, struct disk_devdesc *);
199767f8919635c4928607450d9e0abb932109ceToomas Soome    va_end(ap);
199767f8919635c4928607450d9e0abb932109ceToomas Soome
199767f8919635c4928607450d9e0abb932109ceToomas Soome    if (dev->d_unit < 0 || dev->d_unit >= userdisk_maxunit)
199767f8919635c4928607450d9e0abb932109ceToomas Soome        return (EIO);
199767f8919635c4928607450d9e0abb932109ceToomas Soome
199767f8919635c4928607450d9e0abb932109ceToomas Soome    return (disk_open(dev, ud_info[dev->d_unit].mediasize,
199767f8919635c4928607450d9e0abb932109ceToomas Soome        ud_info[dev->d_unit].sectorsize, 0));
199767f8919635c4928607450d9e0abb932109ceToomas Soome}
199767f8919635c4928607450d9e0abb932109ceToomas Soome
199767f8919635c4928607450d9e0abb932109ceToomas Soomestatic int
199767f8919635c4928607450d9e0abb932109ceToomas Soomeuserdisk_close(struct open_file *f)
199767f8919635c4928607450d9e0abb932109ceToomas Soome{
199767f8919635c4928607450d9e0abb932109ceToomas Soome    struct disk_devdesc *dev;
199767f8919635c4928607450d9e0abb932109ceToomas Soome
199767f8919635c4928607450d9e0abb932109ceToomas Soome    dev = (struct disk_devdesc *)f->f_devdata;
199767f8919635c4928607450d9e0abb932109ceToomas Soome    return (disk_close(dev));
199767f8919635c4928607450d9e0abb932109ceToomas Soome}
199767f8919635c4928607450d9e0abb932109ceToomas Soome
199767f8919635c4928607450d9e0abb932109ceToomas Soomestatic int
199767f8919635c4928607450d9e0abb932109ceToomas Soomeuserdisk_strategy(void *devdata, int rw, daddr_t dblk, size_t size,
199767f8919635c4928607450d9e0abb932109ceToomas Soome    char *buf, size_t *rsize)
199767f8919635c4928607450d9e0abb932109ceToomas Soome{
199767f8919635c4928607450d9e0abb932109ceToomas Soome    struct disk_devdesc *dev = devdata;
199767f8919635c4928607450d9e0abb932109ceToomas Soome    uint64_t    off;
199767f8919635c4928607450d9e0abb932109ceToomas Soome    size_t      resid;
199767f8919635c4928607450d9e0abb932109ceToomas Soome    int     rc;
199767f8919635c4928607450d9e0abb932109ceToomas Soome
199767f8919635c4928607450d9e0abb932109ceToomas Soome    if (rw == F_WRITE)
199767f8919635c4928607450d9e0abb932109ceToomas Soome        return (EROFS);
199767f8919635c4928607450d9e0abb932109ceToomas Soome    if (rw != F_READ)
199767f8919635c4928607450d9e0abb932109ceToomas Soome        return (EINVAL);
199767f8919635c4928607450d9e0abb932109ceToomas Soome    if (rsize)
199767f8919635c4928607450d9e0abb932109ceToomas Soome        *rsize = 0;
199767f8919635c4928607450d9e0abb932109ceToomas Soome    off = (dblk + dev->d_offset) * ud_info[dev->d_unit].sectorsize;
199767f8919635c4928607450d9e0abb932109ceToomas Soome    rc = CALLBACK(diskread, dev->d_unit, off, buf, size, &resid);
199767f8919635c4928607450d9e0abb932109ceToomas Soome    if (rc)
199767f8919635c4928607450d9e0abb932109ceToomas Soome        return (rc);
199767f8919635c4928607450d9e0abb932109ceToomas Soome    if (rsize)
199767f8919635c4928607450d9e0abb932109ceToomas Soome        *rsize = size - resid;
199767f8919635c4928607450d9e0abb932109ceToomas Soome    return (0);
199767f8919635c4928607450d9e0abb932109ceToomas Soome}
199767f8919635c4928607450d9e0abb932109ceToomas Soome
199767f8919635c4928607450d9e0abb932109ceToomas Soomestatic int
199767f8919635c4928607450d9e0abb932109ceToomas Soomeuserdisk_ioctl(struct open_file *f, u_long cmd, void *data)
199767f8919635c4928607450d9e0abb932109ceToomas Soome{
199767f8919635c4928607450d9e0abb932109ceToomas Soome    struct disk_devdesc *dev;
199767f8919635c4928607450d9e0abb932109ceToomas Soome
199767f8919635c4928607450d9e0abb932109ceToomas Soome    dev = (struct disk_devdesc *)f->f_devdata;
199767f8919635c4928607450d9e0abb932109ceToomas Soome    return (CALLBACK(diskioctl, dev->d_unit, cmd, data));
199767f8919635c4928607450d9e0abb932109ceToomas Soome}