Archipelago » qemu

 * Copyright (C) 2010 Christian Brunner <chb@muc.de>

#include "rbd_types.h"

#include <rados/librados.h>

    int aiocnt;

    int64_t segsize;

    rados_pool_t pool;

    rados_pool_t header_pool;

    char name[RBD_MAX_OBJ_NAME_SIZE];

    char block_name[RBD_MAX_BLOCK_NAME_SIZE];

    uint64_t size;

    uint64_t objsize;

typedef struct rbd_obj_header_ondisk RbdHeader1;

static int rbd_next_tok(char *dst, int dst_len,

                        char *src, char delim,

                        const char *name,

                        char **p)

static int rbd_parsename(const char *filename,

                         char *pool, int pool_len,

                         char *snap, int snap_len,

                         char *name, int name_len)

    ret = rbd_next_tok(pool, pool_len, p, '/', "pool name", &p);

    ret = rbd_next_tok(name, name_len, p, '@', "object name", &p);

    ret = rbd_next_tok(snap, snap_len, p, '\0', "snap name", &p);

static int create_tmap_op(uint8_t op, const char *name, char **tmap_desc)

{

    uint32_t len = strlen(name);

    uint32_t len_le = cpu_to_le32(len);

    /* total_len = encoding op + name + empty buffer */

    uint32_t total_len = 1 + (sizeof(uint32_t) + len) + sizeof(uint32_t);

    uint8_t *desc = NULL;

    desc = qemu_malloc(total_len);

    *tmap_desc = (char *)desc;

    *desc = op;

    desc++;

    memcpy(desc, &len_le, sizeof(len_le));

    desc += sizeof(len_le);

    memcpy(desc, name, len);

    desc += len;

    len = 0; /* no need for endian conversion for 0 */

    memcpy(desc, &len, sizeof(len));

    desc += sizeof(len);

    return (char *)desc - *tmap_desc;

}

static void free_tmap_op(char *tmap_desc)

{

    qemu_free(tmap_desc);

}

static int rbd_register_image(rados_pool_t pool, const char *name)

{

    char *tmap_desc;

    const char *dir = RBD_DIRECTORY;

    int ret;

    ret = create_tmap_op(CEPH_OSD_TMAP_SET, name, &tmap_desc);

    if (ret < 0) {

        return ret;

    }

    ret = rados_tmap_update(pool, dir, tmap_desc, ret);

    free_tmap_op(tmap_desc);

    return ret;

}

static int touch_rbd_info(rados_pool_t pool, const char *info_oid)

{

    int r = rados_write(pool, info_oid, 0, NULL, 0);

    if (r < 0) {

        return r;

    }

    return 0;

}

static int rbd_assign_bid(rados_pool_t pool, uint64_t *id)

{

    uint64_t out[1];

    const char *info_oid = RBD_INFO;

    *id = 0;

    int r = touch_rbd_info(pool, info_oid);

    if (r < 0) {

        return r;

    }

    r = rados_exec(pool, info_oid, "rbd", "assign_bid", NULL,

                   0, (char *)out, sizeof(out));

    if (r < 0) {

        return r;

    }

    le64_to_cpus(out);

    *id = out[0];

    return 0;

}

static int rbd_create(const char *filename, QEMUOptionParameter *options)

    uint64_t size;

    time_t mtime;

    uint8_t obj_order = RBD_DEFAULT_OBJ_ORDER;

    char pool[RBD_MAX_SEG_NAME_SIZE];

    char n[RBD_MAX_SEG_NAME_SIZE];

    char name[RBD_MAX_OBJ_NAME_SIZE];

    char snap_buf[RBD_MAX_SEG_NAME_SIZE];

    RbdHeader1 header;

    rados_pool_t p;

    uint64_t bid;

    uint32_t hi, lo;

    if (rbd_parsename(filename,

                      pool, sizeof(pool),

                      snap_buf, sizeof(snap_buf),

                      name, sizeof(name)) < 0) {

    snprintf(n, sizeof(n), "%s%s", name, RBD_SUFFIX);

		obj_order = ffs(objsize) - 1;

    memset(&header, 0, sizeof(header));

    pstrcpy(header.text, sizeof(header.text), RBD_HEADER_TEXT);

    pstrcpy(header.signature, sizeof(header.signature), RBD_HEADER_SIGNATURE);

    pstrcpy(header.version, sizeof(header.version), RBD_HEADER_VERSION);

    header.image_size = cpu_to_le64(bytes);

    header.options.order = obj_order;

    header.options.crypt_type = RBD_CRYPT_NONE;

    header.options.comp_type = RBD_COMP_NONE;

    header.snap_seq = 0;

    header.snap_count = 0;

    if (rados_initialize(0, NULL) < 0) {

    if (rados_open_pool(pool, &p)) {

        error_report("error opening pool %s", pool);

        rados_deinitialize();

    /* check for existing rbd header file */

    ret = rados_stat(p, n, &size, &mtime);

    if (ret == 0) {

        ret=-EEXIST;

        goto done;

    }

    ret = rbd_assign_bid(p, &bid);

    if (ret < 0) {

        error_report("failed assigning block id");

        rados_deinitialize();

    hi = bid >> 32;

    lo = bid & 0xFFFFFFFF;

    snprintf(header.block_name, sizeof(header.block_name), "rb.%x.%x", hi, lo);

    /* create header file */

    ret = rados_write(p, n, 0, (const char *)&header, sizeof(header));

    if (ret < 0) {

        goto done;

    ret = rbd_register_image(p, name);

done:

    rados_close_pool(p);

    rados_deinitialize();

 * This aio completion is being called from rbd_aio_event_reader() and

 * runs in qemu context. It schedules a bh, but just in case the aio

static void rbd_complete_aio(RADOSCB *rcb)

    acb->aiocnt--;

        if (!acb->aiocnt) {

            qemu_vfree(acb->bounce);

            qemu_aio_release(acb);

        }

            acb->ret += rcb->segsize;

        if (r == -ENOENT) {

            memset(rcb->buf, 0, rcb->segsize);

            if (!acb->error) {

                acb->ret += rcb->segsize;

            }

        } else if (r < 0) {

	    memset(rcb->buf, 0, rcb->segsize);

        } else if (r < rcb->segsize) {

            memset(rcb->buf + r, 0, rcb->segsize - r);

                acb->ret += rcb->segsize;

            acb->ret += r;

    if (!acb->aiocnt) {

        acb->bh = qemu_bh_new(rbd_aio_bh_cb, acb);

        qemu_bh_schedule(acb->bh);

    }

static void rbd_aio_event_reader(void *opaque)

                    rbd_complete_aio(s->event_rcb);

                    s->qemu_aio_count --;

static int rbd_aio_flush_cb(void *opaque)

static int rbd_set_snapc(rados_pool_t pool, const char *snap, RbdHeader1 *header)

{

    uint32_t snap_count = le32_to_cpu(header->snap_count);

    rados_snap_t *snaps = NULL;

    rados_snap_t seq;

    uint32_t i;

    uint64_t snap_names_len = le64_to_cpu(header->snap_names_len);

    int r;

    rados_snap_t snapid = 0;

    if (snap_count) {

        const char *header_snap = (const char *)&header->snaps[snap_count];

        const char *end = header_snap + snap_names_len;

        snaps = qemu_malloc(sizeof(rados_snap_t) * header->snap_count);

        for (i=0; i < snap_count; i++) {

            snaps[i] = le64_to_cpu(header->snaps[i].id);

            if (snap && strcmp(snap, header_snap) == 0) {

                snapid = snaps[i];

            }

            header_snap += strlen(header_snap) + 1;

            if (header_snap > end) {

                error_report("bad header, snapshot list broken");

            }

        }

    }

    if (snap && !snapid) {

        error_report("snapshot not found");

        qemu_free(snaps);

        return -ENOENT;

    }

    seq = le32_to_cpu(header->snap_seq);

    r = rados_set_snap_context(pool, seq, snaps, snap_count);

    rados_set_snap(pool, snapid);

    qemu_free(snaps);

    return r;

}

#define BUF_READ_START_LEN    4096

static int rbd_read_header(BDRVRBDState *s, char **hbuf)

{

    char *buf = NULL;

    char n[RBD_MAX_SEG_NAME_SIZE];

    uint64_t len = BUF_READ_START_LEN;

    int r;

    snprintf(n, sizeof(n), "%s%s", s->name, RBD_SUFFIX);

    buf = qemu_malloc(len);

    r = rados_read(s->header_pool, n, 0, buf, len);

    if (r < 0) {

        goto failed;

    }

    if (r < len) {

        goto done;

    }

    qemu_free(buf);

    buf = qemu_malloc(len);

    r = rados_stat(s->header_pool, n, &len, NULL);

    if (r < 0) {

        goto failed;

    }

    r = rados_read(s->header_pool, n, 0, buf, len);

    if (r < 0) {

        goto failed;

    }

done:

    *hbuf = buf;

    return 0;

failed:

    qemu_free(buf);

    return r;

}

static int rbd_open(BlockDriverState *bs, const char *filename, int flags)

    RbdHeader1 *header;

    char pool[RBD_MAX_SEG_NAME_SIZE];

    char snap_buf[RBD_MAX_SEG_NAME_SIZE];

    char *snap = NULL;

    char *hbuf = NULL;

    if (rbd_parsename(filename, pool, sizeof(pool),

                      snap_buf, sizeof(snap_buf),

                      s->name, sizeof(s->name)) < 0) {

        snap = snap_buf;

    if ((r = rados_initialize(0, NULL)) < 0) {

    if ((r = rados_open_pool(pool, &s->pool))) {

        error_report("error opening pool %s", pool);

        rados_deinitialize();

    if ((r = rados_open_pool(pool, &s->header_pool))) {

        error_report("error opening pool %s", pool);

        rados_deinitialize();

    if ((r = rbd_read_header(s, &hbuf)) < 0) {

        error_report("error reading header from %s", s->name);

        goto failed;

    }

    if (memcmp(hbuf + 64, RBD_HEADER_SIGNATURE, 4)) {

        error_report("Invalid header signature");

        r = -EMEDIUMTYPE;

        goto failed;

    }

    if (memcmp(hbuf + 68, RBD_HEADER_VERSION, 8)) {

        error_report("Unknown image version");

        r = -EMEDIUMTYPE;

        goto failed;

    header = (RbdHeader1 *) hbuf;

    s->size = le64_to_cpu(header->image_size);

    s->objsize = 1ULL << header->options.order;

    memcpy(s->block_name, header->block_name, sizeof(header->block_name));

    r = rbd_set_snapc(s->pool, snap, header);

        error_report("failed setting snap context: %s", strerror(-r));

        goto failed;

    bs->read_only = (snap != NULL);

    qemu_aio_set_fd_handler(s->fds[RBD_FD_READ], rbd_aio_event_reader, NULL,

        rbd_aio_flush_cb, NULL, s);

    qemu_free(hbuf);

    qemu_free(hbuf);

    rados_close_pool(s->header_pool);

    rados_close_pool(s->pool);

    rados_deinitialize();

static void rbd_close(BlockDriverState *bs)

    rados_close_pool(s->header_pool);

    rados_close_pool(s->pool);

    rados_deinitialize();

static void rbd_aio_cancel(BlockDriverAIOCB *blockacb)

    .cancel = rbd_aio_cancel,

/*

 * This is the callback function for rados_aio_read and _write

 *

 * Note: this function is being called from a non qemu thread so

 * we need to be careful about what we do here. Generally we only

 * write to the block notification pipe, and do the rest of the

 * io completion handling from rbd_aio_event_reader() which

 * runs in a qemu context.

 */

static void rbd_finish_aiocb(rados_completion_t c, RADOSCB *rcb)

    int ret;

    rcb->ret = rados_aio_get_return_value(c);

    rados_aio_release(c);

        int fd = rcb->s->fds[RBD_FD_WRITE];

        /* send the rcb pointer to the qemu thread that is responsible

           for the aio completion. Must do it in a qemu thread context */

	}

	}

        error_report("failed writing to acb->s->fds\n");

/* Callback when all queued rados_aio requests are complete */

    rados_completion_t c;

    char n[RBD_MAX_SEG_NAME_SIZE];

    int64_t segnr, segoffs, segsize, last_segnr;

    acb->aiocnt = 0;

    segnr = off / s->objsize;

    segoffs = off % s->objsize;

    segsize = s->objsize - segoffs;

    last_segnr = ((off + size - 1) / s->objsize);

    acb->aiocnt = (last_segnr - segnr) + 1;

    s->qemu_aio_count += acb->aiocnt; /* All the RADOSCB */

    while (size > 0) {

        if (size < segsize) {

            segsize = size;

        }

        snprintf(n, sizeof(n), "%s.%012" PRIx64, s->block_name,

                 segnr);

        rcb = qemu_malloc(sizeof(RADOSCB));

        rcb->done = 0;

        rcb->acb = acb;

        rcb->segsize = segsize;

        rcb->buf = buf;

        rcb->s = acb->s;

        if (write) {

            rados_aio_create_completion(rcb, NULL,

                                        (rados_callback_t) rbd_finish_aiocb,

                                        &c);

            rados_aio_write(s->pool, n, segoffs, buf, segsize, c);

        } else {

            rados_aio_create_completion(rcb,

                                        (rados_callback_t) rbd_finish_aiocb,

                                        NULL, &c);

            rados_aio_read(s->pool, n, segoffs, buf, segsize, c);

        }

        buf += segsize;

        size -= segsize;

        segoffs = 0;

        segsize = s->objsize;

        segnr++;

static BlockDriverAIOCB *rbd_aio_readv(BlockDriverState * bs,

                                       int64_t sector_num, QEMUIOVector * qiov,

                                       int nb_sectors,

                                       BlockDriverCompletionFunc * cb,

                                       void *opaque)

static BlockDriverAIOCB *rbd_aio_writev(BlockDriverState * bs,

                                        int64_t sector_num, QEMUIOVector * qiov,

                                        int nb_sectors,

                                        BlockDriverCompletionFunc * cb,

                                        void *opaque)

static int rbd_getinfo(BlockDriverState * bs, BlockDriverInfo * bdi)

    bdi->cluster_size = s->objsize;

static int64_t rbd_getlength(BlockDriverState * bs)

    return s->size;

static int rbd_snap_create(BlockDriverState *bs, QEMUSnapshotInfo *sn_info)

    char inbuf[512], outbuf[128];

    uint64_t snap_id;

    char *p = inbuf;

    char *end = inbuf + sizeof(inbuf);

    char n[RBD_MAX_SEG_NAME_SIZE];

    char *hbuf = NULL;

    RbdHeader1 *header;

    r = rados_selfmanaged_snap_create(s->header_pool, &snap_id);

        error_report("failed to create snap id: %s", strerror(-r));

    *(uint32_t *)p = strlen(sn_info->name);

    cpu_to_le32s((uint32_t *)p);

    p += sizeof(uint32_t);

    strncpy(p, sn_info->name, end - p);

    p += strlen(p);

    if (p + sizeof(snap_id) > end) {

        error_report("invalid input parameter");

        return -EINVAL;

    }

    *(uint64_t *)p = snap_id;

    cpu_to_le64s((uint64_t *)p);

    snprintf(n, sizeof(n), "%s%s", s->name, RBD_SUFFIX);

    r = rados_exec(s->header_pool, n, "rbd", "snap_add", inbuf,

                   sizeof(inbuf), outbuf, sizeof(outbuf));

    if (r < 0) {

        error_report("rbd.snap_add execution failed failed: %s", strerror(-r));

        return r;

    }

    sprintf(sn_info->id_str, "%s", sn_info->name);

    r = rbd_read_header(s, &hbuf);

    if (r < 0) {

        error_report("failed reading header: %s", strerror(-r));

        return r;

    }

    header = (RbdHeader1 *) hbuf;

    r = rbd_set_snapc(s->pool, sn_info->name, header);

    if (r < 0) {

        error_report("failed setting snap context: %s", strerror(-r));

        goto failed;

    }

    return 0;

failed:

    qemu_free(header);

    return r;

}

static int decode32(char **p, const char *end, uint32_t *v)

{

    if (*p + 4 > end) {

	return -ERANGE;

    }

    *v = *(uint32_t *)(*p);

    le32_to_cpus(v);

    *p += 4;

static int decode64(char **p, const char *end, uint64_t *v)

{

    if (*p + 8 > end) {

        return -ERANGE;

    }

    *v = *(uint64_t *)(*p);

    le64_to_cpus(v);

    *p += 8;

    return 0;

}

static int decode_str(char **p, const char *end, char **s)

{

    uint32_t len;

    int r;

    if ((r = decode32(p, end, &len)) < 0) {

        return r;

    }

    *s = qemu_malloc(len + 1);

    memcpy(*s, *p, len);

    *p += len;

    (*s)[len] = '\0';

    return len;

}

static int rbd_snap_list(BlockDriverState *bs, QEMUSnapshotInfo **psn_tab)

    char n[RBD_MAX_SEG_NAME_SIZE];

    RbdHeader1 *header;

    char *hbuf = NULL;

    char *outbuf = NULL, *end, *buf;

    uint64_t len;

    uint64_t snap_seq;

    uint32_t snap_count;

    int r, i;

    /* read header to estimate how much space we need to read the snap

     * list */

    if ((r = rbd_read_header(s, &hbuf)) < 0) {

        goto done_err;

    }

    header = (RbdHeader1 *)hbuf;

    len = le64_to_cpu(header->snap_names_len);

    len += 1024; /* should have already been enough, but new snapshots might

                    already been created since we read the header. just allocate

                    a bit more, so that in most cases it'll suffice anyway */

    qemu_free(hbuf);

    snprintf(n, sizeof(n), "%s%s", s->name, RBD_SUFFIX);

    while (1) {

        qemu_free(outbuf);

        outbuf = qemu_malloc(len);

        r = rados_exec(s->header_pool, n, "rbd", "snap_list", NULL, 0,

                       outbuf, len);

        if (r < 0) {

            error_report("rbd.snap_list execution failed failed: %s", strerror(-r));

            goto done_err;

        if (r != len) {

            break;

	}

        /* if we're here, we probably raced with some snaps creation */

        len *= 2;

    }

    buf = outbuf;

    end = buf + len;

    if ((r = decode64(&buf, end, &snap_seq)) < 0) {

        goto done_err;

    }

    if ((r = decode32(&buf, end, &snap_count)) < 0) {

        goto done_err;

    for (i = 0; i < snap_count; i++) {

        uint64_t id, image_size;

        char *snap_name;

        if ((r = decode64(&buf, end, &id)) < 0) {

            goto done_err;

        }

        if ((r = decode64(&buf, end, &image_size)) < 0) {

            goto done_err;

        }

        if ((r = decode_str(&buf, end, &snap_name)) < 0) {

            goto done_err;

        }

        qemu_free(snap_name);

        sn_info->vm_state_size = image_size;

    qemu_free(outbuf);

done_err:

    qemu_free(sn_tab);

    qemu_free(outbuf);

    return r;

static QEMUOptionParameter rbd_create_options[] = {

    .bdrv_file_open     = rbd_open,

    .bdrv_close         = rbd_close,

    .bdrv_create        = rbd_create,

    .bdrv_get_info      = rbd_getinfo,

    .create_options     = rbd_create_options,

    .bdrv_getlength     = rbd_getlength,

    .bdrv_aio_readv     = rbd_aio_readv,

    .bdrv_aio_writev    = rbd_aio_writev,

    .bdrv_snapshot_create = rbd_snap_create,

    .bdrv_snapshot_list = rbd_snap_list,