Annotation of qemu/block/qed.c, revision 1.1.1.1

1.1       root        1: /*
                      2:  * QEMU Enhanced Disk Format
                      3:  *
                      4:  * Copyright IBM, Corp. 2010
                      5:  *
                      6:  * Authors:
                      7:  *  Stefan Hajnoczi   <[email protected]>
                      8:  *  Anthony Liguori   <[email protected]>
                      9:  *
                     10:  * This work is licensed under the terms of the GNU LGPL, version 2 or later.
                     11:  * See the COPYING.LIB file in the top-level directory.
                     12:  *
                     13:  */
                     14: 
                     15: #include "trace.h"
                     16: #include "qed.h"
                     17: #include "qerror.h"
                     18: 
                     19: static void qed_aio_cancel(BlockDriverAIOCB *blockacb)
                     20: {
                     21:     QEDAIOCB *acb = (QEDAIOCB *)blockacb;
                     22:     bool finished = false;
                     23: 
                     24:     /* Wait for the request to finish */
                     25:     acb->finished = &finished;
                     26:     while (!finished) {
                     27:         qemu_aio_wait();
                     28:     }
                     29: }
                     30: 
                     31: static AIOPool qed_aio_pool = {
                     32:     .aiocb_size         = sizeof(QEDAIOCB),
                     33:     .cancel             = qed_aio_cancel,
                     34: };
                     35: 
                     36: static int bdrv_qed_probe(const uint8_t *buf, int buf_size,
                     37:                           const char *filename)
                     38: {
                     39:     const QEDHeader *header = (const QEDHeader *)buf;
                     40: 
                     41:     if (buf_size < sizeof(*header)) {
                     42:         return 0;
                     43:     }
                     44:     if (le32_to_cpu(header->magic) != QED_MAGIC) {
                     45:         return 0;
                     46:     }
                     47:     return 100;
                     48: }
                     49: 
                     50: /**
                     51:  * Check whether an image format is raw
                     52:  *
                     53:  * @fmt:    Backing file format, may be NULL
                     54:  */
                     55: static bool qed_fmt_is_raw(const char *fmt)
                     56: {
                     57:     return fmt && strcmp(fmt, "raw") == 0;
                     58: }
                     59: 
                     60: static void qed_header_le_to_cpu(const QEDHeader *le, QEDHeader *cpu)
                     61: {
                     62:     cpu->magic = le32_to_cpu(le->magic);
                     63:     cpu->cluster_size = le32_to_cpu(le->cluster_size);
                     64:     cpu->table_size = le32_to_cpu(le->table_size);
                     65:     cpu->header_size = le32_to_cpu(le->header_size);
                     66:     cpu->features = le64_to_cpu(le->features);
                     67:     cpu->compat_features = le64_to_cpu(le->compat_features);
                     68:     cpu->autoclear_features = le64_to_cpu(le->autoclear_features);
                     69:     cpu->l1_table_offset = le64_to_cpu(le->l1_table_offset);
                     70:     cpu->image_size = le64_to_cpu(le->image_size);
                     71:     cpu->backing_filename_offset = le32_to_cpu(le->backing_filename_offset);
                     72:     cpu->backing_filename_size = le32_to_cpu(le->backing_filename_size);
                     73: }
                     74: 
                     75: static void qed_header_cpu_to_le(const QEDHeader *cpu, QEDHeader *le)
                     76: {
                     77:     le->magic = cpu_to_le32(cpu->magic);
                     78:     le->cluster_size = cpu_to_le32(cpu->cluster_size);
                     79:     le->table_size = cpu_to_le32(cpu->table_size);
                     80:     le->header_size = cpu_to_le32(cpu->header_size);
                     81:     le->features = cpu_to_le64(cpu->features);
                     82:     le->compat_features = cpu_to_le64(cpu->compat_features);
                     83:     le->autoclear_features = cpu_to_le64(cpu->autoclear_features);
                     84:     le->l1_table_offset = cpu_to_le64(cpu->l1_table_offset);
                     85:     le->image_size = cpu_to_le64(cpu->image_size);
                     86:     le->backing_filename_offset = cpu_to_le32(cpu->backing_filename_offset);
                     87:     le->backing_filename_size = cpu_to_le32(cpu->backing_filename_size);
                     88: }
                     89: 
                     90: static int qed_write_header_sync(BDRVQEDState *s)
                     91: {
                     92:     QEDHeader le;
                     93:     int ret;
                     94: 
                     95:     qed_header_cpu_to_le(&s->header, &le);
                     96:     ret = bdrv_pwrite(s->bs->file, 0, &le, sizeof(le));
                     97:     if (ret != sizeof(le)) {
                     98:         return ret;
                     99:     }
                    100:     return 0;
                    101: }
                    102: 
                    103: typedef struct {
                    104:     GenericCB gencb;
                    105:     BDRVQEDState *s;
                    106:     struct iovec iov;
                    107:     QEMUIOVector qiov;
                    108:     int nsectors;
                    109:     uint8_t *buf;
                    110: } QEDWriteHeaderCB;
                    111: 
                    112: static void qed_write_header_cb(void *opaque, int ret)
                    113: {
                    114:     QEDWriteHeaderCB *write_header_cb = opaque;
                    115: 
                    116:     qemu_vfree(write_header_cb->buf);
                    117:     gencb_complete(write_header_cb, ret);
                    118: }
                    119: 
                    120: static void qed_write_header_read_cb(void *opaque, int ret)
                    121: {
                    122:     QEDWriteHeaderCB *write_header_cb = opaque;
                    123:     BDRVQEDState *s = write_header_cb->s;
                    124:     BlockDriverAIOCB *acb;
                    125: 
                    126:     if (ret) {
                    127:         qed_write_header_cb(write_header_cb, ret);
                    128:         return;
                    129:     }
                    130: 
                    131:     /* Update header */
                    132:     qed_header_cpu_to_le(&s->header, (QEDHeader *)write_header_cb->buf);
                    133: 
                    134:     acb = bdrv_aio_writev(s->bs->file, 0, &write_header_cb->qiov,
                    135:                           write_header_cb->nsectors, qed_write_header_cb,
                    136:                           write_header_cb);
                    137:     if (!acb) {
                    138:         qed_write_header_cb(write_header_cb, -EIO);
                    139:     }
                    140: }
                    141: 
                    142: /**
                    143:  * Update header in-place (does not rewrite backing filename or other strings)
                    144:  *
                    145:  * This function only updates known header fields in-place and does not affect
                    146:  * extra data after the QED header.
                    147:  */
                    148: static void qed_write_header(BDRVQEDState *s, BlockDriverCompletionFunc cb,
                    149:                              void *opaque)
                    150: {
                    151:     /* We must write full sectors for O_DIRECT but cannot necessarily generate
                    152:      * the data following the header if an unrecognized compat feature is
                    153:      * active.  Therefore, first read the sectors containing the header, update
                    154:      * them, and write back.
                    155:      */
                    156: 
                    157:     BlockDriverAIOCB *acb;
                    158:     int nsectors = (sizeof(QEDHeader) + BDRV_SECTOR_SIZE - 1) /
                    159:                    BDRV_SECTOR_SIZE;
                    160:     size_t len = nsectors * BDRV_SECTOR_SIZE;
                    161:     QEDWriteHeaderCB *write_header_cb = gencb_alloc(sizeof(*write_header_cb),
                    162:                                                     cb, opaque);
                    163: 
                    164:     write_header_cb->s = s;
                    165:     write_header_cb->nsectors = nsectors;
                    166:     write_header_cb->buf = qemu_blockalign(s->bs, len);
                    167:     write_header_cb->iov.iov_base = write_header_cb->buf;
                    168:     write_header_cb->iov.iov_len = len;
                    169:     qemu_iovec_init_external(&write_header_cb->qiov, &write_header_cb->iov, 1);
                    170: 
                    171:     acb = bdrv_aio_readv(s->bs->file, 0, &write_header_cb->qiov, nsectors,
                    172:                          qed_write_header_read_cb, write_header_cb);
                    173:     if (!acb) {
                    174:         qed_write_header_cb(write_header_cb, -EIO);
                    175:     }
                    176: }
                    177: 
                    178: static uint64_t qed_max_image_size(uint32_t cluster_size, uint32_t table_size)
                    179: {
                    180:     uint64_t table_entries;
                    181:     uint64_t l2_size;
                    182: 
                    183:     table_entries = (table_size * cluster_size) / sizeof(uint64_t);
                    184:     l2_size = table_entries * cluster_size;
                    185: 
                    186:     return l2_size * table_entries;
                    187: }
                    188: 
                    189: static bool qed_is_cluster_size_valid(uint32_t cluster_size)
                    190: {
                    191:     if (cluster_size < QED_MIN_CLUSTER_SIZE ||
                    192:         cluster_size > QED_MAX_CLUSTER_SIZE) {
                    193:         return false;
                    194:     }
                    195:     if (cluster_size & (cluster_size - 1)) {
                    196:         return false; /* not power of 2 */
                    197:     }
                    198:     return true;
                    199: }
                    200: 
                    201: static bool qed_is_table_size_valid(uint32_t table_size)
                    202: {
                    203:     if (table_size < QED_MIN_TABLE_SIZE ||
                    204:         table_size > QED_MAX_TABLE_SIZE) {
                    205:         return false;
                    206:     }
                    207:     if (table_size & (table_size - 1)) {
                    208:         return false; /* not power of 2 */
                    209:     }
                    210:     return true;
                    211: }
                    212: 
                    213: static bool qed_is_image_size_valid(uint64_t image_size, uint32_t cluster_size,
                    214:                                     uint32_t table_size)
                    215: {
                    216:     if (image_size % BDRV_SECTOR_SIZE != 0) {
                    217:         return false; /* not multiple of sector size */
                    218:     }
                    219:     if (image_size > qed_max_image_size(cluster_size, table_size)) {
                    220:         return false; /* image is too large */
                    221:     }
                    222:     return true;
                    223: }
                    224: 
                    225: /**
                    226:  * Read a string of known length from the image file
                    227:  *
                    228:  * @file:       Image file
                    229:  * @offset:     File offset to start of string, in bytes
                    230:  * @n:          String length in bytes
                    231:  * @buf:        Destination buffer
                    232:  * @buflen:     Destination buffer length in bytes
                    233:  * @ret:        0 on success, -errno on failure
                    234:  *
                    235:  * The string is NUL-terminated.
                    236:  */
                    237: static int qed_read_string(BlockDriverState *file, uint64_t offset, size_t n,
                    238:                            char *buf, size_t buflen)
                    239: {
                    240:     int ret;
                    241:     if (n >= buflen) {
                    242:         return -EINVAL;
                    243:     }
                    244:     ret = bdrv_pread(file, offset, buf, n);
                    245:     if (ret < 0) {
                    246:         return ret;
                    247:     }
                    248:     buf[n] = '\0';
                    249:     return 0;
                    250: }
                    251: 
                    252: /**
                    253:  * Allocate new clusters
                    254:  *
                    255:  * @s:          QED state
                    256:  * @n:          Number of contiguous clusters to allocate
                    257:  * @ret:        Offset of first allocated cluster
                    258:  *
                    259:  * This function only produces the offset where the new clusters should be
                    260:  * written.  It updates BDRVQEDState but does not make any changes to the image
                    261:  * file.
                    262:  */
                    263: static uint64_t qed_alloc_clusters(BDRVQEDState *s, unsigned int n)
                    264: {
                    265:     uint64_t offset = s->file_size;
                    266:     s->file_size += n * s->header.cluster_size;
                    267:     return offset;
                    268: }
                    269: 
                    270: QEDTable *qed_alloc_table(BDRVQEDState *s)
                    271: {
                    272:     /* Honor O_DIRECT memory alignment requirements */
                    273:     return qemu_blockalign(s->bs,
                    274:                            s->header.cluster_size * s->header.table_size);
                    275: }
                    276: 
                    277: /**
                    278:  * Allocate a new zeroed L2 table
                    279:  */
                    280: static CachedL2Table *qed_new_l2_table(BDRVQEDState *s)
                    281: {
                    282:     CachedL2Table *l2_table = qed_alloc_l2_cache_entry(&s->l2_cache);
                    283: 
                    284:     l2_table->table = qed_alloc_table(s);
                    285:     l2_table->offset = qed_alloc_clusters(s, s->header.table_size);
                    286: 
                    287:     memset(l2_table->table->offsets, 0,
                    288:            s->header.cluster_size * s->header.table_size);
                    289:     return l2_table;
                    290: }
                    291: 
                    292: static void qed_aio_next_io(void *opaque, int ret);
                    293: 
                    294: static int bdrv_qed_open(BlockDriverState *bs, int flags)
                    295: {
                    296:     BDRVQEDState *s = bs->opaque;
                    297:     QEDHeader le_header;
                    298:     int64_t file_size;
                    299:     int ret;
                    300: 
                    301:     s->bs = bs;
                    302:     QSIMPLEQ_INIT(&s->allocating_write_reqs);
                    303: 
                    304:     ret = bdrv_pread(bs->file, 0, &le_header, sizeof(le_header));
                    305:     if (ret < 0) {
                    306:         return ret;
                    307:     }
                    308:     ret = 0; /* ret should always be 0 or -errno */
                    309:     qed_header_le_to_cpu(&le_header, &s->header);
                    310: 
                    311:     if (s->header.magic != QED_MAGIC) {
                    312:         return -EINVAL;
                    313:     }
                    314:     if (s->header.features & ~QED_FEATURE_MASK) {
                    315:         /* image uses unsupported feature bits */
                    316:         char buf[64];
                    317:         snprintf(buf, sizeof(buf), "%" PRIx64,
                    318:             s->header.features & ~QED_FEATURE_MASK);
                    319:         qerror_report(QERR_UNKNOWN_BLOCK_FORMAT_FEATURE,
                    320:             bs->device_name, "QED", buf);
                    321:         return -ENOTSUP;
                    322:     }
                    323:     if (!qed_is_cluster_size_valid(s->header.cluster_size)) {
                    324:         return -EINVAL;
                    325:     }
                    326: 
                    327:     /* Round down file size to the last cluster */
                    328:     file_size = bdrv_getlength(bs->file);
                    329:     if (file_size < 0) {
                    330:         return file_size;
                    331:     }
                    332:     s->file_size = qed_start_of_cluster(s, file_size);
                    333: 
                    334:     if (!qed_is_table_size_valid(s->header.table_size)) {
                    335:         return -EINVAL;
                    336:     }
                    337:     if (!qed_is_image_size_valid(s->header.image_size,
                    338:                                  s->header.cluster_size,
                    339:                                  s->header.table_size)) {
                    340:         return -EINVAL;
                    341:     }
                    342:     if (!qed_check_table_offset(s, s->header.l1_table_offset)) {
                    343:         return -EINVAL;
                    344:     }
                    345: 
                    346:     s->table_nelems = (s->header.cluster_size * s->header.table_size) /
                    347:                       sizeof(uint64_t);
                    348:     s->l2_shift = ffs(s->header.cluster_size) - 1;
                    349:     s->l2_mask = s->table_nelems - 1;
                    350:     s->l1_shift = s->l2_shift + ffs(s->table_nelems) - 1;
                    351: 
                    352:     if ((s->header.features & QED_F_BACKING_FILE)) {
                    353:         if ((uint64_t)s->header.backing_filename_offset +
                    354:             s->header.backing_filename_size >
                    355:             s->header.cluster_size * s->header.header_size) {
                    356:             return -EINVAL;
                    357:         }
                    358: 
                    359:         ret = qed_read_string(bs->file, s->header.backing_filename_offset,
                    360:                               s->header.backing_filename_size, bs->backing_file,
                    361:                               sizeof(bs->backing_file));
                    362:         if (ret < 0) {
                    363:             return ret;
                    364:         }
                    365: 
                    366:         if (s->header.features & QED_F_BACKING_FORMAT_NO_PROBE) {
                    367:             pstrcpy(bs->backing_format, sizeof(bs->backing_format), "raw");
                    368:         }
                    369:     }
                    370: 
                    371:     /* Reset unknown autoclear feature bits.  This is a backwards
                    372:      * compatibility mechanism that allows images to be opened by older
                    373:      * programs, which "knock out" unknown feature bits.  When an image is
                    374:      * opened by a newer program again it can detect that the autoclear
                    375:      * feature is no longer valid.
                    376:      */
                    377:     if ((s->header.autoclear_features & ~QED_AUTOCLEAR_FEATURE_MASK) != 0 &&
                    378:         !bdrv_is_read_only(bs->file)) {
                    379:         s->header.autoclear_features &= QED_AUTOCLEAR_FEATURE_MASK;
                    380: 
                    381:         ret = qed_write_header_sync(s);
                    382:         if (ret) {
                    383:             return ret;
                    384:         }
                    385: 
                    386:         /* From here on only known autoclear feature bits are valid */
                    387:         bdrv_flush(bs->file);
                    388:     }
                    389: 
                    390:     s->l1_table = qed_alloc_table(s);
                    391:     qed_init_l2_cache(&s->l2_cache);
                    392: 
                    393:     ret = qed_read_l1_table_sync(s);
                    394:     if (ret) {
                    395:         goto out;
                    396:     }
                    397: 
                    398:     /* If image was not closed cleanly, check consistency */
                    399:     if (s->header.features & QED_F_NEED_CHECK) {
                    400:         /* Read-only images cannot be fixed.  There is no risk of corruption
                    401:          * since write operations are not possible.  Therefore, allow
                    402:          * potentially inconsistent images to be opened read-only.  This can
                    403:          * aid data recovery from an otherwise inconsistent image.
                    404:          */
                    405:         if (!bdrv_is_read_only(bs->file)) {
                    406:             BdrvCheckResult result = {0};
                    407: 
                    408:             ret = qed_check(s, &result, true);
                    409:             if (!ret && !result.corruptions && !result.check_errors) {
                    410:                 /* Ensure fixes reach storage before clearing check bit */
                    411:                 bdrv_flush(s->bs);
                    412: 
                    413:                 s->header.features &= ~QED_F_NEED_CHECK;
                    414:                 qed_write_header_sync(s);
                    415:             }
                    416:         }
                    417:     }
                    418: 
                    419: out:
                    420:     if (ret) {
                    421:         qed_free_l2_cache(&s->l2_cache);
                    422:         qemu_vfree(s->l1_table);
                    423:     }
                    424:     return ret;
                    425: }
                    426: 
                    427: static void bdrv_qed_close(BlockDriverState *bs)
                    428: {
                    429:     BDRVQEDState *s = bs->opaque;
                    430: 
                    431:     /* Ensure writes reach stable storage */
                    432:     bdrv_flush(bs->file);
                    433: 
                    434:     /* Clean shutdown, no check required on next open */
                    435:     if (s->header.features & QED_F_NEED_CHECK) {
                    436:         s->header.features &= ~QED_F_NEED_CHECK;
                    437:         qed_write_header_sync(s);
                    438:     }
                    439: 
                    440:     qed_free_l2_cache(&s->l2_cache);
                    441:     qemu_vfree(s->l1_table);
                    442: }
                    443: 
                    444: static int bdrv_qed_flush(BlockDriverState *bs)
                    445: {
                    446:     return bdrv_flush(bs->file);
                    447: }
                    448: 
                    449: static int qed_create(const char *filename, uint32_t cluster_size,
                    450:                       uint64_t image_size, uint32_t table_size,
                    451:                       const char *backing_file, const char *backing_fmt)
                    452: {
                    453:     QEDHeader header = {
                    454:         .magic = QED_MAGIC,
                    455:         .cluster_size = cluster_size,
                    456:         .table_size = table_size,
                    457:         .header_size = 1,
                    458:         .features = 0,
                    459:         .compat_features = 0,
                    460:         .l1_table_offset = cluster_size,
                    461:         .image_size = image_size,
                    462:     };
                    463:     QEDHeader le_header;
                    464:     uint8_t *l1_table = NULL;
                    465:     size_t l1_size = header.cluster_size * header.table_size;
                    466:     int ret = 0;
                    467:     BlockDriverState *bs = NULL;
                    468: 
                    469:     ret = bdrv_create_file(filename, NULL);
                    470:     if (ret < 0) {
                    471:         return ret;
                    472:     }
                    473: 
                    474:     ret = bdrv_file_open(&bs, filename, BDRV_O_RDWR | BDRV_O_CACHE_WB);
                    475:     if (ret < 0) {
                    476:         return ret;
                    477:     }
                    478: 
                    479:     /* File must start empty and grow, check truncate is supported */
                    480:     ret = bdrv_truncate(bs, 0);
                    481:     if (ret < 0) {
                    482:         goto out;
                    483:     }
                    484: 
                    485:     if (backing_file) {
                    486:         header.features |= QED_F_BACKING_FILE;
                    487:         header.backing_filename_offset = sizeof(le_header);
                    488:         header.backing_filename_size = strlen(backing_file);
                    489: 
                    490:         if (qed_fmt_is_raw(backing_fmt)) {
                    491:             header.features |= QED_F_BACKING_FORMAT_NO_PROBE;
                    492:         }
                    493:     }
                    494: 
                    495:     qed_header_cpu_to_le(&header, &le_header);
                    496:     ret = bdrv_pwrite(bs, 0, &le_header, sizeof(le_header));
                    497:     if (ret < 0) {
                    498:         goto out;
                    499:     }
                    500:     ret = bdrv_pwrite(bs, sizeof(le_header), backing_file,
                    501:                       header.backing_filename_size);
                    502:     if (ret < 0) {
                    503:         goto out;
                    504:     }
                    505: 
                    506:     l1_table = qemu_mallocz(l1_size);
                    507:     ret = bdrv_pwrite(bs, header.l1_table_offset, l1_table, l1_size);
                    508:     if (ret < 0) {
                    509:         goto out;
                    510:     }
                    511: 
                    512:     ret = 0; /* success */
                    513: out:
                    514:     qemu_free(l1_table);
                    515:     bdrv_delete(bs);
                    516:     return ret;
                    517: }
                    518: 
                    519: static int bdrv_qed_create(const char *filename, QEMUOptionParameter *options)
                    520: {
                    521:     uint64_t image_size = 0;
                    522:     uint32_t cluster_size = QED_DEFAULT_CLUSTER_SIZE;
                    523:     uint32_t table_size = QED_DEFAULT_TABLE_SIZE;
                    524:     const char *backing_file = NULL;
                    525:     const char *backing_fmt = NULL;
                    526: 
                    527:     while (options && options->name) {
                    528:         if (!strcmp(options->name, BLOCK_OPT_SIZE)) {
                    529:             image_size = options->value.n;
                    530:         } else if (!strcmp(options->name, BLOCK_OPT_BACKING_FILE)) {
                    531:             backing_file = options->value.s;
                    532:         } else if (!strcmp(options->name, BLOCK_OPT_BACKING_FMT)) {
                    533:             backing_fmt = options->value.s;
                    534:         } else if (!strcmp(options->name, BLOCK_OPT_CLUSTER_SIZE)) {
                    535:             if (options->value.n) {
                    536:                 cluster_size = options->value.n;
                    537:             }
                    538:         } else if (!strcmp(options->name, BLOCK_OPT_TABLE_SIZE)) {
                    539:             if (options->value.n) {
                    540:                 table_size = options->value.n;
                    541:             }
                    542:         }
                    543:         options++;
                    544:     }
                    545: 
                    546:     if (!qed_is_cluster_size_valid(cluster_size)) {
                    547:         fprintf(stderr, "QED cluster size must be within range [%u, %u] and power of 2\n",
                    548:                 QED_MIN_CLUSTER_SIZE, QED_MAX_CLUSTER_SIZE);
                    549:         return -EINVAL;
                    550:     }
                    551:     if (!qed_is_table_size_valid(table_size)) {
                    552:         fprintf(stderr, "QED table size must be within range [%u, %u] and power of 2\n",
                    553:                 QED_MIN_TABLE_SIZE, QED_MAX_TABLE_SIZE);
                    554:         return -EINVAL;
                    555:     }
                    556:     if (!qed_is_image_size_valid(image_size, cluster_size, table_size)) {
                    557:         fprintf(stderr, "QED image size must be a non-zero multiple of "
                    558:                         "cluster size and less than %" PRIu64 " bytes\n",
                    559:                 qed_max_image_size(cluster_size, table_size));
                    560:         return -EINVAL;
                    561:     }
                    562: 
                    563:     return qed_create(filename, cluster_size, image_size, table_size,
                    564:                       backing_file, backing_fmt);
                    565: }
                    566: 
                    567: typedef struct {
                    568:     int is_allocated;
                    569:     int *pnum;
                    570: } QEDIsAllocatedCB;
                    571: 
                    572: static void qed_is_allocated_cb(void *opaque, int ret, uint64_t offset, size_t len)
                    573: {
                    574:     QEDIsAllocatedCB *cb = opaque;
                    575:     *cb->pnum = len / BDRV_SECTOR_SIZE;
                    576:     cb->is_allocated = ret == QED_CLUSTER_FOUND;
                    577: }
                    578: 
                    579: static int bdrv_qed_is_allocated(BlockDriverState *bs, int64_t sector_num,
                    580:                                   int nb_sectors, int *pnum)
                    581: {
                    582:     BDRVQEDState *s = bs->opaque;
                    583:     uint64_t pos = (uint64_t)sector_num * BDRV_SECTOR_SIZE;
                    584:     size_t len = (size_t)nb_sectors * BDRV_SECTOR_SIZE;
                    585:     QEDIsAllocatedCB cb = {
                    586:         .is_allocated = -1,
                    587:         .pnum = pnum,
                    588:     };
                    589:     QEDRequest request = { .l2_table = NULL };
                    590: 
                    591:     async_context_push();
                    592: 
                    593:     qed_find_cluster(s, &request, pos, len, qed_is_allocated_cb, &cb);
                    594: 
                    595:     while (cb.is_allocated == -1) {
                    596:         qemu_aio_wait();
                    597:     }
                    598: 
                    599:     async_context_pop();
                    600: 
                    601:     qed_unref_l2_cache_entry(request.l2_table);
                    602: 
                    603:     return cb.is_allocated;
                    604: }
                    605: 
                    606: static int bdrv_qed_make_empty(BlockDriverState *bs)
                    607: {
                    608:     return -ENOTSUP;
                    609: }
                    610: 
                    611: static BDRVQEDState *acb_to_s(QEDAIOCB *acb)
                    612: {
                    613:     return acb->common.bs->opaque;
                    614: }
                    615: 
                    616: /**
                    617:  * Read from the backing file or zero-fill if no backing file
                    618:  *
                    619:  * @s:          QED state
                    620:  * @pos:        Byte position in device
                    621:  * @qiov:       Destination I/O vector
                    622:  * @cb:         Completion function
                    623:  * @opaque:     User data for completion function
                    624:  *
                    625:  * This function reads qiov->size bytes starting at pos from the backing file.
                    626:  * If there is no backing file then zeroes are read.
                    627:  */
                    628: static void qed_read_backing_file(BDRVQEDState *s, uint64_t pos,
                    629:                                   QEMUIOVector *qiov,
                    630:                                   BlockDriverCompletionFunc *cb, void *opaque)
                    631: {
                    632:     BlockDriverAIOCB *aiocb;
                    633:     uint64_t backing_length = 0;
                    634:     size_t size;
                    635: 
                    636:     /* If there is a backing file, get its length.  Treat the absence of a
                    637:      * backing file like a zero length backing file.
                    638:      */
                    639:     if (s->bs->backing_hd) {
                    640:         int64_t l = bdrv_getlength(s->bs->backing_hd);
                    641:         if (l < 0) {
                    642:             cb(opaque, l);
                    643:             return;
                    644:         }
                    645:         backing_length = l;
                    646:     }
                    647: 
                    648:     /* Zero all sectors if reading beyond the end of the backing file */
                    649:     if (pos >= backing_length ||
                    650:         pos + qiov->size > backing_length) {
                    651:         qemu_iovec_memset(qiov, 0, qiov->size);
                    652:     }
                    653: 
                    654:     /* Complete now if there are no backing file sectors to read */
                    655:     if (pos >= backing_length) {
                    656:         cb(opaque, 0);
                    657:         return;
                    658:     }
                    659: 
                    660:     /* If the read straddles the end of the backing file, shorten it */
                    661:     size = MIN((uint64_t)backing_length - pos, qiov->size);
                    662: 
                    663:     BLKDBG_EVENT(s->bs->file, BLKDBG_READ_BACKING);
                    664:     aiocb = bdrv_aio_readv(s->bs->backing_hd, pos / BDRV_SECTOR_SIZE,
                    665:                            qiov, size / BDRV_SECTOR_SIZE, cb, opaque);
                    666:     if (!aiocb) {
                    667:         cb(opaque, -EIO);
                    668:     }
                    669: }
                    670: 
                    671: typedef struct {
                    672:     GenericCB gencb;
                    673:     BDRVQEDState *s;
                    674:     QEMUIOVector qiov;
                    675:     struct iovec iov;
                    676:     uint64_t offset;
                    677: } CopyFromBackingFileCB;
                    678: 
                    679: static void qed_copy_from_backing_file_cb(void *opaque, int ret)
                    680: {
                    681:     CopyFromBackingFileCB *copy_cb = opaque;
                    682:     qemu_vfree(copy_cb->iov.iov_base);
                    683:     gencb_complete(&copy_cb->gencb, ret);
                    684: }
                    685: 
                    686: static void qed_copy_from_backing_file_write(void *opaque, int ret)
                    687: {
                    688:     CopyFromBackingFileCB *copy_cb = opaque;
                    689:     BDRVQEDState *s = copy_cb->s;
                    690:     BlockDriverAIOCB *aiocb;
                    691: 
                    692:     if (ret) {
                    693:         qed_copy_from_backing_file_cb(copy_cb, ret);
                    694:         return;
                    695:     }
                    696: 
                    697:     BLKDBG_EVENT(s->bs->file, BLKDBG_COW_WRITE);
                    698:     aiocb = bdrv_aio_writev(s->bs->file, copy_cb->offset / BDRV_SECTOR_SIZE,
                    699:                             &copy_cb->qiov,
                    700:                             copy_cb->qiov.size / BDRV_SECTOR_SIZE,
                    701:                             qed_copy_from_backing_file_cb, copy_cb);
                    702:     if (!aiocb) {
                    703:         qed_copy_from_backing_file_cb(copy_cb, -EIO);
                    704:     }
                    705: }
                    706: 
                    707: /**
                    708:  * Copy data from backing file into the image
                    709:  *
                    710:  * @s:          QED state
                    711:  * @pos:        Byte position in device
                    712:  * @len:        Number of bytes
                    713:  * @offset:     Byte offset in image file
                    714:  * @cb:         Completion function
                    715:  * @opaque:     User data for completion function
                    716:  */
                    717: static void qed_copy_from_backing_file(BDRVQEDState *s, uint64_t pos,
                    718:                                        uint64_t len, uint64_t offset,
                    719:                                        BlockDriverCompletionFunc *cb,
                    720:                                        void *opaque)
                    721: {
                    722:     CopyFromBackingFileCB *copy_cb;
                    723: 
                    724:     /* Skip copy entirely if there is no work to do */
                    725:     if (len == 0) {
                    726:         cb(opaque, 0);
                    727:         return;
                    728:     }
                    729: 
                    730:     copy_cb = gencb_alloc(sizeof(*copy_cb), cb, opaque);
                    731:     copy_cb->s = s;
                    732:     copy_cb->offset = offset;
                    733:     copy_cb->iov.iov_base = qemu_blockalign(s->bs, len);
                    734:     copy_cb->iov.iov_len = len;
                    735:     qemu_iovec_init_external(&copy_cb->qiov, &copy_cb->iov, 1);
                    736: 
                    737:     qed_read_backing_file(s, pos, &copy_cb->qiov,
                    738:                           qed_copy_from_backing_file_write, copy_cb);
                    739: }
                    740: 
                    741: /**
                    742:  * Link one or more contiguous clusters into a table
                    743:  *
                    744:  * @s:              QED state
                    745:  * @table:          L2 table
                    746:  * @index:          First cluster index
                    747:  * @n:              Number of contiguous clusters
                    748:  * @cluster:        First cluster byte offset in image file
                    749:  */
                    750: static void qed_update_l2_table(BDRVQEDState *s, QEDTable *table, int index,
                    751:                                 unsigned int n, uint64_t cluster)
                    752: {
                    753:     int i;
                    754:     for (i = index; i < index + n; i++) {
                    755:         table->offsets[i] = cluster;
                    756:         cluster += s->header.cluster_size;
                    757:     }
                    758: }
                    759: 
                    760: static void qed_aio_complete_bh(void *opaque)
                    761: {
                    762:     QEDAIOCB *acb = opaque;
                    763:     BlockDriverCompletionFunc *cb = acb->common.cb;
                    764:     void *user_opaque = acb->common.opaque;
                    765:     int ret = acb->bh_ret;
                    766:     bool *finished = acb->finished;
                    767: 
                    768:     qemu_bh_delete(acb->bh);
                    769:     qemu_aio_release(acb);
                    770: 
                    771:     /* Invoke callback */
                    772:     cb(user_opaque, ret);
                    773: 
                    774:     /* Signal cancel completion */
                    775:     if (finished) {
                    776:         *finished = true;
                    777:     }
                    778: }
                    779: 
                    780: static void qed_aio_complete(QEDAIOCB *acb, int ret)
                    781: {
                    782:     BDRVQEDState *s = acb_to_s(acb);
                    783: 
                    784:     trace_qed_aio_complete(s, acb, ret);
                    785: 
                    786:     /* Free resources */
                    787:     qemu_iovec_destroy(&acb->cur_qiov);
                    788:     qed_unref_l2_cache_entry(acb->request.l2_table);
                    789: 
                    790:     /* Arrange for a bh to invoke the completion function */
                    791:     acb->bh_ret = ret;
                    792:     acb->bh = qemu_bh_new(qed_aio_complete_bh, acb);
                    793:     qemu_bh_schedule(acb->bh);
                    794: 
                    795:     /* Start next allocating write request waiting behind this one.  Note that
                    796:      * requests enqueue themselves when they first hit an unallocated cluster
                    797:      * but they wait until the entire request is finished before waking up the
                    798:      * next request in the queue.  This ensures that we don't cycle through
                    799:      * requests multiple times but rather finish one at a time completely.
                    800:      */
                    801:     if (acb == QSIMPLEQ_FIRST(&s->allocating_write_reqs)) {
                    802:         QSIMPLEQ_REMOVE_HEAD(&s->allocating_write_reqs, next);
                    803:         acb = QSIMPLEQ_FIRST(&s->allocating_write_reqs);
                    804:         if (acb) {
                    805:             qed_aio_next_io(acb, 0);
                    806:         }
                    807:     }
                    808: }
                    809: 
                    810: /**
                    811:  * Commit the current L2 table to the cache
                    812:  */
                    813: static void qed_commit_l2_update(void *opaque, int ret)
                    814: {
                    815:     QEDAIOCB *acb = opaque;
                    816:     BDRVQEDState *s = acb_to_s(acb);
                    817:     CachedL2Table *l2_table = acb->request.l2_table;
                    818: 
                    819:     qed_commit_l2_cache_entry(&s->l2_cache, l2_table);
                    820: 
                    821:     /* This is guaranteed to succeed because we just committed the entry to the
                    822:      * cache.
                    823:      */
                    824:     acb->request.l2_table = qed_find_l2_cache_entry(&s->l2_cache,
                    825:                                                     l2_table->offset);
                    826:     assert(acb->request.l2_table != NULL);
                    827: 
                    828:     qed_aio_next_io(opaque, ret);
                    829: }
                    830: 
                    831: /**
                    832:  * Update L1 table with new L2 table offset and write it out
                    833:  */
                    834: static void qed_aio_write_l1_update(void *opaque, int ret)
                    835: {
                    836:     QEDAIOCB *acb = opaque;
                    837:     BDRVQEDState *s = acb_to_s(acb);
                    838:     int index;
                    839: 
                    840:     if (ret) {
                    841:         qed_aio_complete(acb, ret);
                    842:         return;
                    843:     }
                    844: 
                    845:     index = qed_l1_index(s, acb->cur_pos);
                    846:     s->l1_table->offsets[index] = acb->request.l2_table->offset;
                    847: 
                    848:     qed_write_l1_table(s, index, 1, qed_commit_l2_update, acb);
                    849: }
                    850: 
                    851: /**
                    852:  * Update L2 table with new cluster offsets and write them out
                    853:  */
                    854: static void qed_aio_write_l2_update(void *opaque, int ret)
                    855: {
                    856:     QEDAIOCB *acb = opaque;
                    857:     BDRVQEDState *s = acb_to_s(acb);
                    858:     bool need_alloc = acb->find_cluster_ret == QED_CLUSTER_L1;
                    859:     int index;
                    860: 
                    861:     if (ret) {
                    862:         goto err;
                    863:     }
                    864: 
                    865:     if (need_alloc) {
                    866:         qed_unref_l2_cache_entry(acb->request.l2_table);
                    867:         acb->request.l2_table = qed_new_l2_table(s);
                    868:     }
                    869: 
                    870:     index = qed_l2_index(s, acb->cur_pos);
                    871:     qed_update_l2_table(s, acb->request.l2_table->table, index, acb->cur_nclusters,
                    872:                          acb->cur_cluster);
                    873: 
                    874:     if (need_alloc) {
                    875:         /* Write out the whole new L2 table */
                    876:         qed_write_l2_table(s, &acb->request, 0, s->table_nelems, true,
                    877:                             qed_aio_write_l1_update, acb);
                    878:     } else {
                    879:         /* Write out only the updated part of the L2 table */
                    880:         qed_write_l2_table(s, &acb->request, index, acb->cur_nclusters, false,
                    881:                             qed_aio_next_io, acb);
                    882:     }
                    883:     return;
                    884: 
                    885: err:
                    886:     qed_aio_complete(acb, ret);
                    887: }
                    888: 
                    889: /**
                    890:  * Flush new data clusters before updating the L2 table
                    891:  *
                    892:  * This flush is necessary when a backing file is in use.  A crash during an
                    893:  * allocating write could result in empty clusters in the image.  If the write
                    894:  * only touched a subregion of the cluster, then backing image sectors have
                    895:  * been lost in the untouched region.  The solution is to flush after writing a
                    896:  * new data cluster and before updating the L2 table.
                    897:  */
                    898: static void qed_aio_write_flush_before_l2_update(void *opaque, int ret)
                    899: {
                    900:     QEDAIOCB *acb = opaque;
                    901:     BDRVQEDState *s = acb_to_s(acb);
                    902: 
                    903:     if (!bdrv_aio_flush(s->bs->file, qed_aio_write_l2_update, opaque)) {
                    904:         qed_aio_complete(acb, -EIO);
                    905:     }
                    906: }
                    907: 
                    908: /**
                    909:  * Write data to the image file
                    910:  */
                    911: static void qed_aio_write_main(void *opaque, int ret)
                    912: {
                    913:     QEDAIOCB *acb = opaque;
                    914:     BDRVQEDState *s = acb_to_s(acb);
                    915:     uint64_t offset = acb->cur_cluster +
                    916:                       qed_offset_into_cluster(s, acb->cur_pos);
                    917:     BlockDriverCompletionFunc *next_fn;
                    918:     BlockDriverAIOCB *file_acb;
                    919: 
                    920:     trace_qed_aio_write_main(s, acb, ret, offset, acb->cur_qiov.size);
                    921: 
                    922:     if (ret) {
                    923:         qed_aio_complete(acb, ret);
                    924:         return;
                    925:     }
                    926: 
                    927:     if (acb->find_cluster_ret == QED_CLUSTER_FOUND) {
                    928:         next_fn = qed_aio_next_io;
                    929:     } else {
                    930:         if (s->bs->backing_hd) {
                    931:             next_fn = qed_aio_write_flush_before_l2_update;
                    932:         } else {
                    933:             next_fn = qed_aio_write_l2_update;
                    934:         }
                    935:     }
                    936: 
                    937:     BLKDBG_EVENT(s->bs->file, BLKDBG_WRITE_AIO);
                    938:     file_acb = bdrv_aio_writev(s->bs->file, offset / BDRV_SECTOR_SIZE,
                    939:                                &acb->cur_qiov,
                    940:                                acb->cur_qiov.size / BDRV_SECTOR_SIZE,
                    941:                                next_fn, acb);
                    942:     if (!file_acb) {
                    943:         qed_aio_complete(acb, -EIO);
                    944:     }
                    945: }
                    946: 
                    947: /**
                    948:  * Populate back untouched region of new data cluster
                    949:  */
                    950: static void qed_aio_write_postfill(void *opaque, int ret)
                    951: {
                    952:     QEDAIOCB *acb = opaque;
                    953:     BDRVQEDState *s = acb_to_s(acb);
                    954:     uint64_t start = acb->cur_pos + acb->cur_qiov.size;
                    955:     uint64_t len =
                    956:         qed_start_of_cluster(s, start + s->header.cluster_size - 1) - start;
                    957:     uint64_t offset = acb->cur_cluster +
                    958:                       qed_offset_into_cluster(s, acb->cur_pos) +
                    959:                       acb->cur_qiov.size;
                    960: 
                    961:     if (ret) {
                    962:         qed_aio_complete(acb, ret);
                    963:         return;
                    964:     }
                    965: 
                    966:     trace_qed_aio_write_postfill(s, acb, start, len, offset);
                    967:     qed_copy_from_backing_file(s, start, len, offset,
                    968:                                 qed_aio_write_main, acb);
                    969: }
                    970: 
                    971: /**
                    972:  * Populate front untouched region of new data cluster
                    973:  */
                    974: static void qed_aio_write_prefill(void *opaque, int ret)
                    975: {
                    976:     QEDAIOCB *acb = opaque;
                    977:     BDRVQEDState *s = acb_to_s(acb);
                    978:     uint64_t start = qed_start_of_cluster(s, acb->cur_pos);
                    979:     uint64_t len = qed_offset_into_cluster(s, acb->cur_pos);
                    980: 
                    981:     trace_qed_aio_write_prefill(s, acb, start, len, acb->cur_cluster);
                    982:     qed_copy_from_backing_file(s, start, len, acb->cur_cluster,
                    983:                                 qed_aio_write_postfill, acb);
                    984: }
                    985: 
                    986: /**
                    987:  * Check if the QED_F_NEED_CHECK bit should be set during allocating write
                    988:  */
                    989: static bool qed_should_set_need_check(BDRVQEDState *s)
                    990: {
                    991:     /* The flush before L2 update path ensures consistency */
                    992:     if (s->bs->backing_hd) {
                    993:         return false;
                    994:     }
                    995: 
                    996:     return !(s->header.features & QED_F_NEED_CHECK);
                    997: }
                    998: 
                    999: /**
                   1000:  * Write new data cluster
                   1001:  *
                   1002:  * @acb:        Write request
                   1003:  * @len:        Length in bytes
                   1004:  *
                   1005:  * This path is taken when writing to previously unallocated clusters.
                   1006:  */
                   1007: static void qed_aio_write_alloc(QEDAIOCB *acb, size_t len)
                   1008: {
                   1009:     BDRVQEDState *s = acb_to_s(acb);
                   1010: 
                   1011:     /* Freeze this request if another allocating write is in progress */
                   1012:     if (acb != QSIMPLEQ_FIRST(&s->allocating_write_reqs)) {
                   1013:         QSIMPLEQ_INSERT_TAIL(&s->allocating_write_reqs, acb, next);
                   1014:     }
                   1015:     if (acb != QSIMPLEQ_FIRST(&s->allocating_write_reqs)) {
                   1016:         return; /* wait for existing request to finish */
                   1017:     }
                   1018: 
                   1019:     acb->cur_nclusters = qed_bytes_to_clusters(s,
                   1020:             qed_offset_into_cluster(s, acb->cur_pos) + len);
                   1021:     acb->cur_cluster = qed_alloc_clusters(s, acb->cur_nclusters);
                   1022:     qemu_iovec_copy(&acb->cur_qiov, acb->qiov, acb->qiov_offset, len);
                   1023: 
                   1024:     if (qed_should_set_need_check(s)) {
                   1025:         s->header.features |= QED_F_NEED_CHECK;
                   1026:         qed_write_header(s, qed_aio_write_prefill, acb);
                   1027:     } else {
                   1028:         qed_aio_write_prefill(acb, 0);
                   1029:     }
                   1030: }
                   1031: 
                   1032: /**
                   1033:  * Write data cluster in place
                   1034:  *
                   1035:  * @acb:        Write request
                   1036:  * @offset:     Cluster offset in bytes
                   1037:  * @len:        Length in bytes
                   1038:  *
                   1039:  * This path is taken when writing to already allocated clusters.
                   1040:  */
                   1041: static void qed_aio_write_inplace(QEDAIOCB *acb, uint64_t offset, size_t len)
                   1042: {
                   1043:     /* Calculate the I/O vector */
                   1044:     acb->cur_cluster = offset;
                   1045:     qemu_iovec_copy(&acb->cur_qiov, acb->qiov, acb->qiov_offset, len);
                   1046: 
                   1047:     /* Do the actual write */
                   1048:     qed_aio_write_main(acb, 0);
                   1049: }
                   1050: 
                   1051: /**
                   1052:  * Write data cluster
                   1053:  *
                   1054:  * @opaque:     Write request
                   1055:  * @ret:        QED_CLUSTER_FOUND, QED_CLUSTER_L2, QED_CLUSTER_L1,
                   1056:  *              or -errno
                   1057:  * @offset:     Cluster offset in bytes
                   1058:  * @len:        Length in bytes
                   1059:  *
                   1060:  * Callback from qed_find_cluster().
                   1061:  */
                   1062: static void qed_aio_write_data(void *opaque, int ret,
                   1063:                                uint64_t offset, size_t len)
                   1064: {
                   1065:     QEDAIOCB *acb = opaque;
                   1066: 
                   1067:     trace_qed_aio_write_data(acb_to_s(acb), acb, ret, offset, len);
                   1068: 
                   1069:     acb->find_cluster_ret = ret;
                   1070: 
                   1071:     switch (ret) {
                   1072:     case QED_CLUSTER_FOUND:
                   1073:         qed_aio_write_inplace(acb, offset, len);
                   1074:         break;
                   1075: 
                   1076:     case QED_CLUSTER_L2:
                   1077:     case QED_CLUSTER_L1:
                   1078:         qed_aio_write_alloc(acb, len);
                   1079:         break;
                   1080: 
                   1081:     default:
                   1082:         qed_aio_complete(acb, ret);
                   1083:         break;
                   1084:     }
                   1085: }
                   1086: 
                   1087: /**
                   1088:  * Read data cluster
                   1089:  *
                   1090:  * @opaque:     Read request
                   1091:  * @ret:        QED_CLUSTER_FOUND, QED_CLUSTER_L2, QED_CLUSTER_L1,
                   1092:  *              or -errno
                   1093:  * @offset:     Cluster offset in bytes
                   1094:  * @len:        Length in bytes
                   1095:  *
                   1096:  * Callback from qed_find_cluster().
                   1097:  */
                   1098: static void qed_aio_read_data(void *opaque, int ret,
                   1099:                               uint64_t offset, size_t len)
                   1100: {
                   1101:     QEDAIOCB *acb = opaque;
                   1102:     BDRVQEDState *s = acb_to_s(acb);
                   1103:     BlockDriverState *bs = acb->common.bs;
                   1104:     BlockDriverAIOCB *file_acb;
                   1105: 
                   1106:     /* Adjust offset into cluster */
                   1107:     offset += qed_offset_into_cluster(s, acb->cur_pos);
                   1108: 
                   1109:     trace_qed_aio_read_data(s, acb, ret, offset, len);
                   1110: 
                   1111:     if (ret < 0) {
                   1112:         goto err;
                   1113:     }
                   1114: 
                   1115:     qemu_iovec_copy(&acb->cur_qiov, acb->qiov, acb->qiov_offset, len);
                   1116: 
                   1117:     /* Handle backing file and unallocated sparse hole reads */
                   1118:     if (ret != QED_CLUSTER_FOUND) {
                   1119:         qed_read_backing_file(s, acb->cur_pos, &acb->cur_qiov,
                   1120:                               qed_aio_next_io, acb);
                   1121:         return;
                   1122:     }
                   1123: 
                   1124:     BLKDBG_EVENT(bs->file, BLKDBG_READ_AIO);
                   1125:     file_acb = bdrv_aio_readv(bs->file, offset / BDRV_SECTOR_SIZE,
                   1126:                               &acb->cur_qiov,
                   1127:                               acb->cur_qiov.size / BDRV_SECTOR_SIZE,
                   1128:                               qed_aio_next_io, acb);
                   1129:     if (!file_acb) {
                   1130:         ret = -EIO;
                   1131:         goto err;
                   1132:     }
                   1133:     return;
                   1134: 
                   1135: err:
                   1136:     qed_aio_complete(acb, ret);
                   1137: }
                   1138: 
                   1139: /**
                   1140:  * Begin next I/O or complete the request
                   1141:  */
                   1142: static void qed_aio_next_io(void *opaque, int ret)
                   1143: {
                   1144:     QEDAIOCB *acb = opaque;
                   1145:     BDRVQEDState *s = acb_to_s(acb);
                   1146:     QEDFindClusterFunc *io_fn =
                   1147:         acb->is_write ? qed_aio_write_data : qed_aio_read_data;
                   1148: 
                   1149:     trace_qed_aio_next_io(s, acb, ret, acb->cur_pos + acb->cur_qiov.size);
                   1150: 
                   1151:     /* Handle I/O error */
                   1152:     if (ret) {
                   1153:         qed_aio_complete(acb, ret);
                   1154:         return;
                   1155:     }
                   1156: 
                   1157:     acb->qiov_offset += acb->cur_qiov.size;
                   1158:     acb->cur_pos += acb->cur_qiov.size;
                   1159:     qemu_iovec_reset(&acb->cur_qiov);
                   1160: 
                   1161:     /* Complete request */
                   1162:     if (acb->cur_pos >= acb->end_pos) {
                   1163:         qed_aio_complete(acb, 0);
                   1164:         return;
                   1165:     }
                   1166: 
                   1167:     /* Find next cluster and start I/O */
                   1168:     qed_find_cluster(s, &acb->request,
                   1169:                       acb->cur_pos, acb->end_pos - acb->cur_pos,
                   1170:                       io_fn, acb);
                   1171: }
                   1172: 
                   1173: static BlockDriverAIOCB *qed_aio_setup(BlockDriverState *bs,
                   1174:                                        int64_t sector_num,
                   1175:                                        QEMUIOVector *qiov, int nb_sectors,
                   1176:                                        BlockDriverCompletionFunc *cb,
                   1177:                                        void *opaque, bool is_write)
                   1178: {
                   1179:     QEDAIOCB *acb = qemu_aio_get(&qed_aio_pool, bs, cb, opaque);
                   1180: 
                   1181:     trace_qed_aio_setup(bs->opaque, acb, sector_num, nb_sectors,
                   1182:                          opaque, is_write);
                   1183: 
                   1184:     acb->is_write = is_write;
                   1185:     acb->finished = NULL;
                   1186:     acb->qiov = qiov;
                   1187:     acb->qiov_offset = 0;
                   1188:     acb->cur_pos = (uint64_t)sector_num * BDRV_SECTOR_SIZE;
                   1189:     acb->end_pos = acb->cur_pos + nb_sectors * BDRV_SECTOR_SIZE;
                   1190:     acb->request.l2_table = NULL;
                   1191:     qemu_iovec_init(&acb->cur_qiov, qiov->niov);
                   1192: 
                   1193:     /* Start request */
                   1194:     qed_aio_next_io(acb, 0);
                   1195:     return &acb->common;
                   1196: }
                   1197: 
                   1198: static BlockDriverAIOCB *bdrv_qed_aio_readv(BlockDriverState *bs,
                   1199:                                             int64_t sector_num,
                   1200:                                             QEMUIOVector *qiov, int nb_sectors,
                   1201:                                             BlockDriverCompletionFunc *cb,
                   1202:                                             void *opaque)
                   1203: {
                   1204:     return qed_aio_setup(bs, sector_num, qiov, nb_sectors, cb, opaque, false);
                   1205: }
                   1206: 
                   1207: static BlockDriverAIOCB *bdrv_qed_aio_writev(BlockDriverState *bs,
                   1208:                                              int64_t sector_num,
                   1209:                                              QEMUIOVector *qiov, int nb_sectors,
                   1210:                                              BlockDriverCompletionFunc *cb,
                   1211:                                              void *opaque)
                   1212: {
                   1213:     return qed_aio_setup(bs, sector_num, qiov, nb_sectors, cb, opaque, true);
                   1214: }
                   1215: 
                   1216: static BlockDriverAIOCB *bdrv_qed_aio_flush(BlockDriverState *bs,
                   1217:                                             BlockDriverCompletionFunc *cb,
                   1218:                                             void *opaque)
                   1219: {
                   1220:     return bdrv_aio_flush(bs->file, cb, opaque);
                   1221: }
                   1222: 
                   1223: static int bdrv_qed_truncate(BlockDriverState *bs, int64_t offset)
                   1224: {
                   1225:     return -ENOTSUP;
                   1226: }
                   1227: 
                   1228: static int64_t bdrv_qed_getlength(BlockDriverState *bs)
                   1229: {
                   1230:     BDRVQEDState *s = bs->opaque;
                   1231:     return s->header.image_size;
                   1232: }
                   1233: 
                   1234: static int bdrv_qed_get_info(BlockDriverState *bs, BlockDriverInfo *bdi)
                   1235: {
                   1236:     BDRVQEDState *s = bs->opaque;
                   1237: 
                   1238:     memset(bdi, 0, sizeof(*bdi));
                   1239:     bdi->cluster_size = s->header.cluster_size;
                   1240:     return 0;
                   1241: }
                   1242: 
                   1243: static int bdrv_qed_change_backing_file(BlockDriverState *bs,
                   1244:                                         const char *backing_file,
                   1245:                                         const char *backing_fmt)
                   1246: {
                   1247:     BDRVQEDState *s = bs->opaque;
                   1248:     QEDHeader new_header, le_header;
                   1249:     void *buffer;
                   1250:     size_t buffer_len, backing_file_len;
                   1251:     int ret;
                   1252: 
                   1253:     /* Refuse to set backing filename if unknown compat feature bits are
                   1254:      * active.  If the image uses an unknown compat feature then we may not
                   1255:      * know the layout of data following the header structure and cannot safely
                   1256:      * add a new string.
                   1257:      */
                   1258:     if (backing_file && (s->header.compat_features &
                   1259:                          ~QED_COMPAT_FEATURE_MASK)) {
                   1260:         return -ENOTSUP;
                   1261:     }
                   1262: 
                   1263:     memcpy(&new_header, &s->header, sizeof(new_header));
                   1264: 
                   1265:     new_header.features &= ~(QED_F_BACKING_FILE |
                   1266:                              QED_F_BACKING_FORMAT_NO_PROBE);
                   1267: 
                   1268:     /* Adjust feature flags */
                   1269:     if (backing_file) {
                   1270:         new_header.features |= QED_F_BACKING_FILE;
                   1271: 
                   1272:         if (qed_fmt_is_raw(backing_fmt)) {
                   1273:             new_header.features |= QED_F_BACKING_FORMAT_NO_PROBE;
                   1274:         }
                   1275:     }
                   1276: 
                   1277:     /* Calculate new header size */
                   1278:     backing_file_len = 0;
                   1279: 
                   1280:     if (backing_file) {
                   1281:         backing_file_len = strlen(backing_file);
                   1282:     }
                   1283: 
                   1284:     buffer_len = sizeof(new_header);
                   1285:     new_header.backing_filename_offset = buffer_len;
                   1286:     new_header.backing_filename_size = backing_file_len;
                   1287:     buffer_len += backing_file_len;
                   1288: 
                   1289:     /* Make sure we can rewrite header without failing */
                   1290:     if (buffer_len > new_header.header_size * new_header.cluster_size) {
                   1291:         return -ENOSPC;
                   1292:     }
                   1293: 
                   1294:     /* Prepare new header */
                   1295:     buffer = qemu_malloc(buffer_len);
                   1296: 
                   1297:     qed_header_cpu_to_le(&new_header, &le_header);
                   1298:     memcpy(buffer, &le_header, sizeof(le_header));
                   1299:     buffer_len = sizeof(le_header);
                   1300: 
                   1301:     memcpy(buffer + buffer_len, backing_file, backing_file_len);
                   1302:     buffer_len += backing_file_len;
                   1303: 
                   1304:     /* Write new header */
                   1305:     ret = bdrv_pwrite_sync(bs->file, 0, buffer, buffer_len);
                   1306:     qemu_free(buffer);
                   1307:     if (ret == 0) {
                   1308:         memcpy(&s->header, &new_header, sizeof(new_header));
                   1309:     }
                   1310:     return ret;
                   1311: }
                   1312: 
                   1313: static int bdrv_qed_check(BlockDriverState *bs, BdrvCheckResult *result)
                   1314: {
                   1315:     BDRVQEDState *s = bs->opaque;
                   1316: 
                   1317:     return qed_check(s, result, false);
                   1318: }
                   1319: 
                   1320: static QEMUOptionParameter qed_create_options[] = {
                   1321:     {
                   1322:         .name = BLOCK_OPT_SIZE,
                   1323:         .type = OPT_SIZE,
                   1324:         .help = "Virtual disk size (in bytes)"
                   1325:     }, {
                   1326:         .name = BLOCK_OPT_BACKING_FILE,
                   1327:         .type = OPT_STRING,
                   1328:         .help = "File name of a base image"
                   1329:     }, {
                   1330:         .name = BLOCK_OPT_BACKING_FMT,
                   1331:         .type = OPT_STRING,
                   1332:         .help = "Image format of the base image"
                   1333:     }, {
                   1334:         .name = BLOCK_OPT_CLUSTER_SIZE,
                   1335:         .type = OPT_SIZE,
                   1336:         .help = "Cluster size (in bytes)"
                   1337:     }, {
                   1338:         .name = BLOCK_OPT_TABLE_SIZE,
                   1339:         .type = OPT_SIZE,
                   1340:         .help = "L1/L2 table size (in clusters)"
                   1341:     },
                   1342:     { /* end of list */ }
                   1343: };
                   1344: 
                   1345: static BlockDriver bdrv_qed = {
                   1346:     .format_name              = "qed",
                   1347:     .instance_size            = sizeof(BDRVQEDState),
                   1348:     .create_options           = qed_create_options,
                   1349: 
                   1350:     .bdrv_probe               = bdrv_qed_probe,
                   1351:     .bdrv_open                = bdrv_qed_open,
                   1352:     .bdrv_close               = bdrv_qed_close,
                   1353:     .bdrv_create              = bdrv_qed_create,
                   1354:     .bdrv_flush               = bdrv_qed_flush,
                   1355:     .bdrv_is_allocated        = bdrv_qed_is_allocated,
                   1356:     .bdrv_make_empty          = bdrv_qed_make_empty,
                   1357:     .bdrv_aio_readv           = bdrv_qed_aio_readv,
                   1358:     .bdrv_aio_writev          = bdrv_qed_aio_writev,
                   1359:     .bdrv_aio_flush           = bdrv_qed_aio_flush,
                   1360:     .bdrv_truncate            = bdrv_qed_truncate,
                   1361:     .bdrv_getlength           = bdrv_qed_getlength,
                   1362:     .bdrv_get_info            = bdrv_qed_get_info,
                   1363:     .bdrv_change_backing_file = bdrv_qed_change_backing_file,
                   1364:     .bdrv_check               = bdrv_qed_check,
                   1365: };
                   1366: 
                   1367: static void bdrv_qed_init(void)
                   1368: {
                   1369:     bdrv_register(&bdrv_qed);
                   1370: }
                   1371: 
                   1372: block_init(bdrv_qed_init);

unix.superglobalmegacorp.com

This archive runs on limited infrastructure. Preserving old code on modern bandwidth. Automated agents are requested to crawl responsibly.