source: mainline/uspace/drv/bus/usb/usbmast/main.c@ e5291e0

/*
 * Copyright (c) 2011 Vojtech Horky
 * Copyright (c) 2011 Jiri Svoboda
 * All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 *
 * - Redistributions of source code must retain the above copyright
 *   notice, this list of conditions and the following disclaimer.
 * - Redistributions in binary form must reproduce the above copyright
 *   notice, this list of conditions and the following disclaimer in the
 *   documentation and/or other materials provided with the distribution.
 * - The name of the author may not be used to endorse or promote products
 *   derived from this software without specific prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
 * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
 * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
 * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
 * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
 * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 */

/** @addtogroup drvusbmast
 * @{
 */
/**
 * @file
 * Main routines of USB mass storage driver.
 */
#include <as.h>
#include <async.h>
#include <ipc/bd.h>
#include <macros.h>
#include <usb/dev/driver.h>
#include <usb/debug.h>
#include <usb/classes/classes.h>
#include <usb/classes/massstor.h>
#include <errno.h>
#include <str_error.h>
#include "cmdw.h"
#include "bo_trans.h"
#include "scsi_ms.h"
#include "usbmast.h"

#define NAME "usbmast"

#define GET_BULK_IN(dev) ((dev)->pipes[BULK_IN_EP].pipe)
#define GET_BULK_OUT(dev) ((dev)->pipes[BULK_OUT_EP].pipe)

static usb_endpoint_description_t bulk_in_ep = {
        .transfer_type = USB_TRANSFER_BULK,
        .direction = USB_DIRECTION_IN,
        .interface_class = USB_CLASS_MASS_STORAGE,
        .interface_subclass = USB_MASSSTOR_SUBCLASS_SCSI,
        .interface_protocol = USB_MASSSTOR_PROTOCOL_BBB,
        .flags = 0
};
static usb_endpoint_description_t bulk_out_ep = {
        .transfer_type = USB_TRANSFER_BULK,
        .direction = USB_DIRECTION_OUT,
        .interface_class = USB_CLASS_MASS_STORAGE,
        .interface_subclass = USB_MASSSTOR_SUBCLASS_SCSI,
        .interface_protocol = USB_MASSSTOR_PROTOCOL_BBB,
        .flags = 0
};

static const usb_endpoint_description_t *mast_endpoints[] = {
        &bulk_in_ep,
        &bulk_out_ep,
        NULL
};

static int usbmast_fun_create(usbmast_dev_t *mdev, unsigned lun);
static void usbmast_bd_connection(ipc_callid_t iid, ipc_call_t *icall,
    void *arg);

/** Callback when a device is removed from the system.
 *
 * @param dev Representation of USB device.
 * @return Error code.
 */
static int usbmast_device_gone(usb_device_t *dev)
{
        usbmast_dev_t *mdev = dev->driver_data;
        assert(mdev);

        for (size_t i = 0; i < mdev->lun_count; ++i) {
                const int rc = ddf_fun_unbind(mdev->luns[i]);
                if (rc != EOK) {
                        usb_log_error("Failed to unbind LUN function %zu: "
                            "%s\n", i, str_error(rc));
                        return rc;
                }
                ddf_fun_destroy(mdev->luns[i]);
                mdev->luns[i] = NULL;
        }
        free(mdev->luns);
        return EOK;
}

/** Callback when new device is attached and recognized as a mass storage.
 *
 * @param dev Representation of a the USB device.
 * @return Error code.
 */
static int usbmast_device_add(usb_device_t *dev)
{
        int rc;
        usbmast_dev_t *mdev = NULL;
        unsigned i;

        /* Allocate softstate */
        mdev = usb_device_data_alloc(dev, sizeof(usbmast_dev_t));
        if (mdev == NULL) {
                usb_log_error("Failed allocating softstate.\n");
                return ENOMEM;
        }

        mdev->ddf_dev = dev->ddf_dev;
        mdev->usb_dev = dev;

        usb_log_info("Initializing mass storage `%s'.\n", dev->ddf_dev->name);
        usb_log_debug(" Bulk in endpoint: %d [%zuB].\n",
            dev->pipes[BULK_IN_EP].pipe->endpoint_no,
            (size_t) dev->pipes[BULK_IN_EP].descriptor->max_packet_size);
        usb_log_debug("Bulk out endpoint: %d [%zuB].\n",
            dev->pipes[BULK_OUT_EP].pipe->endpoint_no,
            (size_t) dev->pipes[BULK_OUT_EP].descriptor->max_packet_size);

        usb_log_debug("Get LUN count...\n");
        mdev->lun_count = usb_masstor_get_lun_count(mdev);
        mdev->luns = calloc(mdev->lun_count, sizeof(ddf_fun_t*));
        if (mdev->luns == NULL) {
                rc = ENOMEM;
                usb_log_error("Failed allocating luns table.\n");
                goto error;
        }

        for (i = 0; i < mdev->lun_count; i++) {
                rc = usbmast_fun_create(mdev, i);
                if (rc != EOK)
                        goto error;
        }

        return EOK;
error:
        /* Destroy functions */
        for (size_t i = 0; i < mdev->lun_count; ++i) {
                if (mdev->luns[i] == NULL)
                        continue;
                const int rc = ddf_fun_unbind(mdev->luns[i]);
                if (rc != EOK) {
                        usb_log_warning("Failed to unbind LUN function %zu: "
                            "%s.\n", i, str_error(rc));
                }
                ddf_fun_destroy(mdev->luns[i]);
        }
        free(mdev->luns);
        return rc;
}

/** Create mass storage function.
 *
 * Called once for each LUN.
 *
 * @param mdev          Mass storage device
 * @param lun           LUN
 * @return              EOK on success or negative error code.
 */
static int usbmast_fun_create(usbmast_dev_t *mdev, unsigned lun)
{
        int rc;
        char *fun_name = NULL;
        ddf_fun_t *fun = NULL;
        usbmast_fun_t *mfun = NULL;

        if (asprintf(&fun_name, "l%u", lun) < 0) {
                usb_log_error("Out of memory.\n");
                rc = ENOMEM;
                goto error;
        }

        fun = ddf_fun_create(mdev->ddf_dev, fun_exposed, fun_name);
        if (fun == NULL) {
                usb_log_error("Failed to create DDF function %s.\n", fun_name);
                rc = ENOMEM;
                goto error;
        }

        /* Allocate soft state */
        mfun = ddf_fun_data_alloc(fun, sizeof(usbmast_fun_t));
        if (mfun == NULL) {
                usb_log_error("Failed allocating softstate.\n");
                rc = ENOMEM;
                goto error;
        }

        mfun->ddf_fun = fun;
        mfun->mdev = mdev;
        mfun->lun = lun;

        /* Set up a connection handler. */
        fun->conn_handler = usbmast_bd_connection;

        usb_log_debug("Inquire...\n");
        usbmast_inquiry_data_t inquiry;
        rc = usbmast_inquiry(mfun, &inquiry);
        if (rc != EOK) {
                usb_log_warning("Failed to inquire device `%s': %s.\n",
                    mdev->ddf_dev->name, str_error(rc));
                rc = EIO;
                goto error;
        }

        usb_log_info("Mass storage `%s' LUN %u: " \
            "%s by %s rev. %s is %s (%s).\n",
            mdev->ddf_dev->name,
            lun,
            inquiry.product,
            inquiry.vendor,
            inquiry.revision,
            usbmast_scsi_dev_type_str(inquiry.device_type),
            inquiry.removable ? "removable" : "non-removable");

        uint32_t nblocks, block_size;

        rc = usbmast_read_capacity(mfun, &nblocks, &block_size);
        if (rc != EOK) {
                usb_log_warning("Failed to read capacity, device `%s': %s.\n",
                    mdev->ddf_dev->name, str_error(rc));
                rc = EIO;
                goto error;
        }

        usb_log_info("Read Capacity: nblocks=%" PRIu32 ", "
            "block_size=%" PRIu32 "\n", nblocks, block_size);

        mfun->nblocks = nblocks;
        mfun->block_size = block_size;

        rc = ddf_fun_bind(fun);
        if (rc != EOK) {
                usb_log_error("Failed to bind DDF function %s: %s.\n",
                    fun_name, str_error(rc));
                goto error;
        }

        free(fun_name);
        mdev->luns[lun] = fun;

        return EOK;

        /* Error cleanup */
error:
        if (fun != NULL)
                ddf_fun_destroy(fun);
        if (fun_name != NULL)
                free(fun_name);
        return rc;
}

/** Blockdev client connection handler. */
static void usbmast_bd_connection(ipc_callid_t iid, ipc_call_t *icall,
    void *arg)
{
        usbmast_fun_t *mfun;
        void *comm_buf = NULL;
        size_t comm_size;
        ipc_callid_t callid;
        ipc_call_t call;
        unsigned int flags;
        sysarg_t method;
        uint64_t ba;
        size_t cnt;
        int retval;

        async_answer_0(iid, EOK);

        if (!async_share_out_receive(&callid, &comm_size, &flags)) {
                async_answer_0(callid, EHANGUP);
                return;
        }

        comm_buf = as_get_mappable_page(comm_size);
        if (comm_buf == NULL) {
                async_answer_0(callid, EHANGUP);
                return;
        }

        (void) async_share_out_finalize(callid, comm_buf);

        mfun = (usbmast_fun_t *) ((ddf_fun_t *)arg)->driver_data;

        while (true) {
                callid = async_get_call(&call);
                method = IPC_GET_IMETHOD(call);

                if (!method) {
                        /* The other side hung up. */
                        async_answer_0(callid, EOK);
                        return;
                }

                switch (method) {
                case BD_GET_BLOCK_SIZE:
                        async_answer_1(callid, EOK, mfun->block_size);
                        break;
                case BD_GET_NUM_BLOCKS:
                        async_answer_2(callid, EOK, LOWER32(mfun->nblocks),
                            UPPER32(mfun->nblocks));
                        break;
                case BD_READ_BLOCKS:
                        ba = MERGE_LOUP32(IPC_GET_ARG1(call), IPC_GET_ARG2(call));
                        cnt = IPC_GET_ARG3(call);
                        retval = usbmast_read(mfun, ba, cnt, comm_buf);
                        async_answer_0(callid, retval);
                        break;
                case BD_WRITE_BLOCKS:
                        ba = MERGE_LOUP32(IPC_GET_ARG1(call), IPC_GET_ARG2(call));
                        cnt = IPC_GET_ARG3(call);
                        retval = usbmast_write(mfun, ba, cnt, comm_buf);
                        async_answer_0(callid, retval);
                        break;
                default:
                        async_answer_0(callid, EINVAL);
                }
        }
}

/** USB mass storage driver ops. */
static usb_driver_ops_t usbmast_driver_ops = {
        .device_add = usbmast_device_add,
        .device_gone = usbmast_device_gone,
};

/** USB mass storage driver. */
static usb_driver_t usbmast_driver = {
        .name = NAME,
        .ops = &usbmast_driver_ops,
        .endpoints = mast_endpoints
};

int main(int argc, char *argv[])
{
        usb_log_enable(USB_LOG_LEVEL_DEFAULT, NAME);

        return usb_driver_main(&usbmast_driver);
}

/**
 * @}
 */