path: root/src/kernel/blk.c

/*
 * This file is part of the Distributed Network Block Device 3
 *
 * Copyright(c) 2011-2012 Johann Latocha <johann@latocha.de>
 *
 * This file may be licensed under the terms of of the
 * GNU General Public License Version 2 (the ``GPL'').
 *
 * Software distributed under the License is distributed
 * on an ``AS IS'' basis, WITHOUT WARRANTY OF ANY KIND, either
 * express or implied. See the GPL for the specific language
 * governing rights and limitations.
 *
 * You should have received a copy of the GPL along with this
 * program. If not, go to http://www.gnu.org/licenses/gpl.html
 * or write to the Free Software Foundation, Inc.,
 * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
 *
 */

#include "blk.h"
#include "net.h"

int dnbd3_blk_add_device(dnbd3_device_t *dev, int minor)
{
    struct gendisk *disk;
    struct request_queue *blk_queue;

    init_waitqueue_head(&dev->process_queue_send);
    init_waitqueue_head(&dev->process_queue_receive);
    INIT_LIST_HEAD(&dev->request_queue_send);
    INIT_LIST_HEAD(&dev->request_queue_receive);

    if (!(disk = alloc_disk(1)))
    {
        printk("ERROR: dnbd3 alloc_disk failed.\n");
        return -EIO;
    }

    disk->major = major;
    disk->first_minor = minor;
    sprintf(disk->disk_name, "dnbd%d", minor);
    set_capacity(disk, 0);
    set_disk_ro(disk, 1);
    disk->fops = &dnbd3_blk_ops;

    spin_lock_init(&dev->blk_lock);
    if ((blk_queue = blk_init_queue(&dnbd3_blk_request, &dev->blk_lock)) == NULL)
    {
        printk("ERROR: dnbd3 blk_init_queue failed.\n");
        return -EIO;
    }

    blk_queue_logical_block_size(blk_queue, DNBD3_BLOCK_SIZE);
    disk->queue = blk_queue;
    disk->private_data = dev;
    queue_flag_set_unlocked(QUEUE_FLAG_NONROT, disk->queue);
    dev->disk = disk;

    dev->hb_request.cmd_type = REQ_TYPE_SPECIAL;

    add_disk(disk); // must be last
    return 0;
}

int dnbd3_blk_del_device(dnbd3_device_t *dev)
{
    if (dev->sock)
    {
        sock_release(dev->sock);
        dev->sock = NULL;
    }

    if (&dev->hb_timer)
        del_timer(&dev->hb_timer);

    del_gendisk(dev->disk);
    put_disk(dev->disk);
    blk_cleanup_queue(dev->disk->queue);
    return 0;
}

struct block_device_operations dnbd3_blk_ops =
{ .owner = THIS_MODULE, .ioctl = dnbd3_blk_ioctl, };

int dnbd3_blk_ioctl(struct block_device *bdev, fmode_t mode, unsigned int cmd, unsigned long arg)
{
    dnbd3_device_t *lo = bdev->bd_disk->private_data;

    switch (cmd)
    {
    case IOCTL_SET_HOST:
        strcpy(lo->host, (char *) arg);
        break;

    case IOCTL_SET_PORT:
        strcpy(lo->port, (char *) arg);
        break;
    case IOCTL_SET_IMAGE:
        strcpy(lo->image_id, (char *) arg);
        break;
    case IOCTL_CONNECT:
        if (lo->host && lo->port && lo->image_id)
            dnbd3_net_connect(lo);
        else
            return -1;
        break;
    case IOCTL_DISCONNECT:
        dnbd3_net_disconnect(lo);
        break;
    case BLKFLSBUF:
        break;

    default:
        return -1;

    }
    return 0;
}

void dnbd3_blk_request(struct request_queue *q)
{
    struct request *req;
    dnbd3_device_t *lo;

    while ((req = blk_fetch_request(q)) != NULL)
    {
        lo = req->rq_disk->private_data;

        if (req->cmd_type != REQ_TYPE_FS)
        {
            __blk_end_request_all(req, 0);
            continue;
        }

        if (rq_data_dir(req) == READ)
        {
            list_add_tail(&req->queuelist, &lo->request_queue_send);
            spin_unlock_irq(q->queue_lock);
            wake_up(&lo->process_queue_send);
            spin_lock_irq(q->queue_lock);
        }
    }
}