block/iscsi.c

   1 /*
   2  * QEMU Block driver for iSCSI images
   3  *
   4  * Copyright (c) 2010-2011 Ronnie Sahlberg <ronniesahlberg@gmail.com>
   5  *
   6  * Permission is hereby granted, free of charge, to any person obtaining a copy
   7  * of this software and associated documentation files (the "Software"), to deal
   8  * in the Software without restriction, including without limitation the rights
   9  * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
  10  * copies of the Software, and to permit persons to whom the Software is
  11  * furnished to do so, subject to the following conditions:
  12  *
  13  * The above copyright notice and this permission notice shall be included in
  14  * all copies or substantial portions of the Software.
  15  *
  16  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  17  * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  18  * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL
  19  * THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  20  * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
  21  * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
  22  * THE SOFTWARE.
  23  */
  24
  25 #include "config-host.h"
  26
  27 #include <poll.h>
  28 #include <arpa/inet.h>
  29 #include "qemu-common.h"
  30 #include "qemu-error.h"
  31 #include "block_int.h"
  32 #include "trace.h"
  33 #include "hw/scsi-defs.h"
  34
  35 #include <iscsi/iscsi.h>
  36 #include <iscsi/scsi-lowlevel.h>
  37
  38 #ifdef __linux__
  39 #include <scsi/sg.h>
  40 #include <hw/scsi-defs.h>
  41 #endif
  42
  43 typedef struct IscsiLun {
  44     struct iscsi_context *iscsi;
  45     int lun;
  46     enum scsi_inquiry_peripheral_device_type type;
  47     int block_size;
  48     uint64_t num_blocks;
  49     int events;
  50 } IscsiLun;
  51
  52 typedef struct IscsiAIOCB {
  53     BlockDriverAIOCB common;
  54     QEMUIOVector *qiov;
  55     QEMUBH *bh;
  56     IscsiLun *iscsilun;
  57     struct scsi_task *task;
  58     uint8_t *buf;
  59     int status;
  60     int canceled;
  61     size_t read_size;
  62     size_t read_offset;
  63 #ifdef __linux__
  64     sg_io_hdr_t *ioh;
  65 #endif
  66 } IscsiAIOCB;
  67
  68 struct IscsiTask {
  69     IscsiLun *iscsilun;
  70     BlockDriverState *bs;
  71     int status;
  72     int complete;
  73 };
  74
  75 static void
  76 iscsi_bh_cb(void *p)
  77 {
  78     IscsiAIOCB *acb = p;
  79
  80     qemu_bh_delete(acb->bh);
  81
  82     if (acb->canceled == 0) {
  83         acb->common.cb(acb->common.opaque, acb->status);
  84     }
  85
  86     if (acb->task != NULL) {
  87         scsi_free_scsi_task(acb->task);
  88         acb->task = NULL;
  89     }
  90
  91     qemu_aio_release(acb);
  92 }
  93
  94 static void
  95 iscsi_schedule_bh(IscsiAIOCB *acb)
  96 {
  97     if (acb->bh) {
  98         return;
  99     }
 100     acb->bh = qemu_bh_new(iscsi_bh_cb, acb);
 101     qemu_bh_schedule(acb->bh);
 102 }
 103
 104
 105 static void
 106 iscsi_abort_task_cb(struct iscsi_context *iscsi, int status, void *command_data,
 107                     void *private_data)
 108 {
 109     IscsiAIOCB *acb = private_data;
 110
 111     acb->status = -ECANCELED;
 112     iscsi_schedule_bh(acb);
 113 }
 114
 115 static void
 116 iscsi_aio_cancel(BlockDriverAIOCB *blockacb)
 117 {
 118     IscsiAIOCB *acb = (IscsiAIOCB *)blockacb;
 119     IscsiLun *iscsilun = acb->iscsilun;
 120
 121     if (acb->status != -EINPROGRESS) {
 122         return;
 123     }
 124
 125     acb->canceled = 1;
 126
 127     /* send a task mgmt call to the target to cancel the task on the target */
 128     iscsi_task_mgmt_abort_task_async(iscsilun->iscsi, acb->task,
 129                                      iscsi_abort_task_cb, acb);
 130
 131     while (acb->status == -EINPROGRESS) {
 132         qemu_aio_wait();
 133     }
 134 }
 135
 136 static AIOPool iscsi_aio_pool = {
 137     .aiocb_size         = sizeof(IscsiAIOCB),
 138     .cancel             = iscsi_aio_cancel,
 139 };
 140
 141
 142 static void iscsi_process_read(void *arg);
 143 static void iscsi_process_write(void *arg);
 144
 145 static int iscsi_process_flush(void *arg)
 146 {
 147     IscsiLun *iscsilun = arg;
 148
 149     return iscsi_queue_length(iscsilun->iscsi) > 0;
 150 }
 151
 152 static void
 153 iscsi_set_events(IscsiLun *iscsilun)
 154 {
 155     struct iscsi_context *iscsi = iscsilun->iscsi;
 156     int ev;
 157
 158     /* We always register a read handler.  */
 159     ev = POLLIN;
 160     ev |= iscsi_which_events(iscsi);
 161     if (ev != iscsilun->events) {
 162         qemu_aio_set_fd_handler(iscsi_get_fd(iscsi),
 163                       iscsi_process_read,
 164                       (ev & POLLOUT) ? iscsi_process_write : NULL,
 165                       iscsi_process_flush,
 166                       iscsilun);
 167
 168     }
 169
 170     /* If we just added an event, the callback might be delayed
 171      * unless we call qemu_notify_event().
 172      */
 173     if (ev & ~iscsilun->events) {
 174         qemu_notify_event();
 175     }
 176     iscsilun->events = ev;
 177 }
 178
 179 static void
 180 iscsi_process_read(void *arg)
 181 {
 182     IscsiLun *iscsilun = arg;
 183     struct iscsi_context *iscsi = iscsilun->iscsi;
 184
 185     iscsi_service(iscsi, POLLIN);
 186     iscsi_set_events(iscsilun);
 187 }
 188
 189 static void
 190 iscsi_process_write(void *arg)
 191 {
 192     IscsiLun *iscsilun = arg;
 193     struct iscsi_context *iscsi = iscsilun->iscsi;
 194
 195     iscsi_service(iscsi, POLLOUT);
 196     iscsi_set_events(iscsilun);
 197 }
 198
 199
 200 static void
 201 iscsi_aio_write16_cb(struct iscsi_context *iscsi, int status,
 202                      void *command_data, void *opaque)
 203 {
 204     IscsiAIOCB *acb = opaque;
 205
 206     trace_iscsi_aio_write16_cb(iscsi, status, acb, acb->canceled);
 207
 208     g_free(acb->buf);
 209
 210     if (acb->canceled != 0) {
 211         return;
 212     }
 213
 214     acb->status = 0;
 215     if (status < 0) {
 216         error_report("Failed to write16 data to iSCSI lun. %s",
 217                      iscsi_get_error(iscsi));
 218         acb->status = -EIO;
 219     }
 220
 221     iscsi_schedule_bh(acb);
 222 }
 223
 224 static int64_t sector_qemu2lun(int64_t sector, IscsiLun *iscsilun)
 225 {
 226     return sector * BDRV_SECTOR_SIZE / iscsilun->block_size;
 227 }
 228
 229 static BlockDriverAIOCB *
 230 iscsi_aio_writev(BlockDriverState *bs, int64_t sector_num,
 231                  QEMUIOVector *qiov, int nb_sectors,
 232                  BlockDriverCompletionFunc *cb,
 233                  void *opaque)
 234 {
 235     IscsiLun *iscsilun = bs->opaque;
 236     struct iscsi_context *iscsi = iscsilun->iscsi;
 237     IscsiAIOCB *acb;
 238     size_t size;
 239     uint32_t num_sectors;
 240     uint64_t lba;
 241     struct iscsi_data data;
 242
 243     acb = qemu_aio_get(&iscsi_aio_pool, bs, cb, opaque);
 244     trace_iscsi_aio_writev(iscsi, sector_num, nb_sectors, opaque, acb);
 245
 246     acb->iscsilun = iscsilun;
 247     acb->qiov     = qiov;
 248
 249     acb->canceled   = 0;
 250     acb->bh         = NULL;
 251     acb->status     = -EINPROGRESS;
 252
 253     /* XXX we should pass the iovec to write16 to avoid the extra copy */
 254     /* this will allow us to get rid of 'buf' completely */
 255     size = nb_sectors * BDRV_SECTOR_SIZE;
 256     acb->buf = g_malloc(size);
 257     qemu_iovec_to_buf(acb->qiov, 0, acb->buf, size);
 258
 259     acb->task = malloc(sizeof(struct scsi_task));
 260     if (acb->task == NULL) {
 261         error_report("iSCSI: Failed to allocate task for scsi WRITE16 "
 262                      "command. %s", iscsi_get_error(iscsi));
 263         qemu_aio_release(acb);
 264         return NULL;
 265     }
 266     memset(acb->task, 0, sizeof(struct scsi_task));
 267
 268     acb->task->xfer_dir = SCSI_XFER_WRITE;
 269     acb->task->cdb_size = 16;
 270     acb->task->cdb[0] = 0x8a;
 271     if (!(bs->open_flags & BDRV_O_CACHE_WB)) {
 272         /* set FUA on writes when cache mode is write through */
 273         acb->task->cdb[1] |= 0x04;
 274     }
 275     lba = sector_qemu2lun(sector_num, iscsilun);
 276     *(uint32_t *)&acb->task->cdb[2]  = htonl(lba >> 32);
 277     *(uint32_t *)&acb->task->cdb[6]  = htonl(lba & 0xffffffff);
 278     num_sectors = size / iscsilun->block_size;
 279     *(uint32_t *)&acb->task->cdb[10] = htonl(num_sectors);
 280     acb->task->expxferlen = size;
 281
 282     data.data = acb->buf;
 283     data.size = size;
 284
 285     if (iscsi_scsi_command_async(iscsi, iscsilun->lun, acb->task,
 286                                  iscsi_aio_write16_cb,
 287                                  &data,
 288                                  acb) != 0) {
 289         scsi_free_scsi_task(acb->task);
 290         g_free(acb->buf);
 291         qemu_aio_release(acb);
 292         return NULL;
 293     }
 294
 295     iscsi_set_events(iscsilun);
 296
 297     return &acb->common;
 298 }
 299
 300 static void
 301 iscsi_aio_read16_cb(struct iscsi_context *iscsi, int status,
 302                     void *command_data, void *opaque)
 303 {
 304     IscsiAIOCB *acb = opaque;
 305
 306     trace_iscsi_aio_read16_cb(iscsi, status, acb, acb->canceled);
 307
 308     if (acb->canceled != 0) {
 309         return;
 310     }
 311
 312     acb->status = 0;
 313     if (status != 0) {
 314         error_report("Failed to read16 data from iSCSI lun. %s",
 315                      iscsi_get_error(iscsi));
 316         acb->status = -EIO;
 317     }
 318
 319     iscsi_schedule_bh(acb);
 320 }
 321
 322 static BlockDriverAIOCB *
 323 iscsi_aio_readv(BlockDriverState *bs, int64_t sector_num,
 324                 QEMUIOVector *qiov, int nb_sectors,
 325                 BlockDriverCompletionFunc *cb,
 326                 void *opaque)
 327 {
 328     IscsiLun *iscsilun = bs->opaque;
 329     struct iscsi_context *iscsi = iscsilun->iscsi;
 330     IscsiAIOCB *acb;
 331     size_t qemu_read_size;
 332     int i;
 333     uint64_t lba;
 334     uint32_t num_sectors;
 335
 336     qemu_read_size = BDRV_SECTOR_SIZE * (size_t)nb_sectors;
 337
 338     acb = qemu_aio_get(&iscsi_aio_pool, bs, cb, opaque);
 339     trace_iscsi_aio_readv(iscsi, sector_num, nb_sectors, opaque, acb);
 340
 341     acb->iscsilun = iscsilun;
 342     acb->qiov     = qiov;
 343
 344     acb->canceled    = 0;
 345     acb->bh          = NULL;
 346     acb->status      = -EINPROGRESS;
 347     acb->read_size   = qemu_read_size;
 348     acb->buf         = NULL;
 349
 350     /* If LUN blocksize is bigger than BDRV_BLOCK_SIZE a read from QEMU
 351      * may be misaligned to the LUN, so we may need to read some extra
 352      * data.
 353      */
 354     acb->read_offset = 0;
 355     if (iscsilun->block_size > BDRV_SECTOR_SIZE) {
 356         uint64_t bdrv_offset = BDRV_SECTOR_SIZE * sector_num;
 357
 358         acb->read_offset  = bdrv_offset % iscsilun->block_size;
 359     }
 360
 361     num_sectors  = (qemu_read_size + iscsilun->block_size
 362                     + acb->read_offset - 1)
 363                     / iscsilun->block_size;
 364
 365     acb->task = malloc(sizeof(struct scsi_task));
 366     if (acb->task == NULL) {
 367         error_report("iSCSI: Failed to allocate task for scsi READ16 "
 368                      "command. %s", iscsi_get_error(iscsi));
 369         qemu_aio_release(acb);
 370         return NULL;
 371     }
 372     memset(acb->task, 0, sizeof(struct scsi_task));
 373
 374     acb->task->xfer_dir = SCSI_XFER_READ;
 375     lba = sector_qemu2lun(sector_num, iscsilun);
 376     acb->task->expxferlen = qemu_read_size;
 377
 378     switch (iscsilun->type) {
 379     case TYPE_DISK:
 380         acb->task->cdb_size = 16;
 381         acb->task->cdb[0]  = 0x88;
 382         *(uint32_t *)&acb->task->cdb[2]  = htonl(lba >> 32);
 383         *(uint32_t *)&acb->task->cdb[6]  = htonl(lba & 0xffffffff);
 384         *(uint32_t *)&acb->task->cdb[10] = htonl(num_sectors);
 385         break;
 386     default:
 387         acb->task->cdb_size = 10;
 388         acb->task->cdb[0]  = 0x28;
 389         *(uint32_t *)&acb->task->cdb[2] = htonl(lba);
 390         *(uint16_t *)&acb->task->cdb[7] = htons(num_sectors);
 391         break;
 392     }
 393
 394     if (iscsi_scsi_command_async(iscsi, iscsilun->lun, acb->task,
 395                                  iscsi_aio_read16_cb,
 396                                  NULL,
 397                                  acb) != 0) {
 398         scsi_free_scsi_task(acb->task);
 399         qemu_aio_release(acb);
 400         return NULL;
 401     }
 402
 403     for (i = 0; i < acb->qiov->niov; i++) {
 404         scsi_task_add_data_in_buffer(acb->task,
 405                 acb->qiov->iov[i].iov_len,
 406                 acb->qiov->iov[i].iov_base);
 407     }
 408
 409     iscsi_set_events(iscsilun);
 410
 411     return &acb->common;
 412 }
 413
 414
 415 static void
 416 iscsi_synccache10_cb(struct iscsi_context *iscsi, int status,
 417                      void *command_data, void *opaque)
 418 {
 419     IscsiAIOCB *acb = opaque;
 420
 421     if (acb->canceled != 0) {
 422         return;
 423     }
 424
 425     acb->status = 0;
 426     if (status < 0) {
 427         error_report("Failed to sync10 data on iSCSI lun. %s",
 428                      iscsi_get_error(iscsi));
 429         acb->status = -EIO;
 430     }
 431
 432     iscsi_schedule_bh(acb);
 433 }
 434
 435 static BlockDriverAIOCB *
 436 iscsi_aio_flush(BlockDriverState *bs,
 437                 BlockDriverCompletionFunc *cb, void *opaque)
 438 {
 439     IscsiLun *iscsilun = bs->opaque;
 440     struct iscsi_context *iscsi = iscsilun->iscsi;
 441     IscsiAIOCB *acb;
 442
 443     acb = qemu_aio_get(&iscsi_aio_pool, bs, cb, opaque);
 444
 445     acb->iscsilun = iscsilun;
 446     acb->canceled   = 0;
 447     acb->bh         = NULL;
 448     acb->status     = -EINPROGRESS;
 449
 450     acb->task = iscsi_synchronizecache10_task(iscsi, iscsilun->lun,
 451                                          0, 0, 0, 0,
 452                                          iscsi_synccache10_cb,
 453                                          acb);
 454     if (acb->task == NULL) {
 455         error_report("iSCSI: Failed to send synchronizecache10 command. %s",
 456                      iscsi_get_error(iscsi));
 457         qemu_aio_release(acb);
 458         return NULL;
 459     }
 460
 461     iscsi_set_events(iscsilun);
 462
 463     return &acb->common;
 464 }
 465
 466 static void
 467 iscsi_unmap_cb(struct iscsi_context *iscsi, int status,
 468                      void *command_data, void *opaque)
 469 {
 470     IscsiAIOCB *acb = opaque;
 471
 472     if (acb->canceled != 0) {
 473         return;
 474     }
 475
 476     acb->status = 0;
 477     if (status < 0) {
 478         error_report("Failed to unmap data on iSCSI lun. %s",
 479                      iscsi_get_error(iscsi));
 480         acb->status = -EIO;
 481     }
 482
 483     iscsi_schedule_bh(acb);
 484 }
 485
 486 static BlockDriverAIOCB *
 487 iscsi_aio_discard(BlockDriverState *bs,
 488                   int64_t sector_num, int nb_sectors,
 489                   BlockDriverCompletionFunc *cb, void *opaque)
 490 {
 491     IscsiLun *iscsilun = bs->opaque;
 492     struct iscsi_context *iscsi = iscsilun->iscsi;
 493     IscsiAIOCB *acb;
 494     struct unmap_list list[1];
 495
 496     acb = qemu_aio_get(&iscsi_aio_pool, bs, cb, opaque);
 497
 498     acb->iscsilun = iscsilun;
 499     acb->canceled   = 0;
 500     acb->bh         = NULL;
 501     acb->status     = -EINPROGRESS;
 502
 503     list[0].lba = sector_qemu2lun(sector_num, iscsilun);
 504     list[0].num = nb_sectors * BDRV_SECTOR_SIZE / iscsilun->block_size;
 505
 506     acb->task = iscsi_unmap_task(iscsi, iscsilun->lun,
 507                                  0, 0, &list[0], 1,
 508                                  iscsi_unmap_cb,
 509                                  acb);
 510     if (acb->task == NULL) {
 511         error_report("iSCSI: Failed to send unmap command. %s",
 512                      iscsi_get_error(iscsi));
 513         qemu_aio_release(acb);
 514         return NULL;
 515     }
 516
 517     iscsi_set_events(iscsilun);
 518
 519     return &acb->common;
 520 }
 521
 522 #ifdef __linux__
 523 static void
 524 iscsi_aio_ioctl_cb(struct iscsi_context *iscsi, int status,
 525                      void *command_data, void *opaque)
 526 {
 527     IscsiAIOCB *acb = opaque;
 528
 529     if (acb->canceled != 0) {
 530         return;
 531     }
 532
 533     acb->status = 0;
 534     if (status < 0) {
 535         error_report("Failed to ioctl(SG_IO) to iSCSI lun. %s",
 536                      iscsi_get_error(iscsi));
 537         acb->status = -EIO;
 538     }
 539
 540     acb->ioh->driver_status = 0;
 541     acb->ioh->host_status   = 0;
 542     acb->ioh->resid         = 0;
 543
 544 #define SG_ERR_DRIVER_SENSE    0x08
 545
 546     if (status == SCSI_STATUS_CHECK_CONDITION && acb->task->datain.size >= 2) {
 547         int ss;
 548
 549         acb->ioh->driver_status |= SG_ERR_DRIVER_SENSE;
 550
 551         acb->ioh->sb_len_wr = acb->task->datain.size - 2;
 552         ss = (acb->ioh->mx_sb_len >= acb->ioh->sb_len_wr) ?
 553              acb->ioh->mx_sb_len : acb->ioh->sb_len_wr;
 554         memcpy(acb->ioh->sbp, &acb->task->datain.data[2], ss);
 555     }
 556
 557     iscsi_schedule_bh(acb);
 558 }
 559
 560 static BlockDriverAIOCB *iscsi_aio_ioctl(BlockDriverState *bs,
 561         unsigned long int req, void *buf,
 562         BlockDriverCompletionFunc *cb, void *opaque)
 563 {
 564     IscsiLun *iscsilun = bs->opaque;
 565     struct iscsi_context *iscsi = iscsilun->iscsi;
 566     struct iscsi_data data;
 567     IscsiAIOCB *acb;
 568
 569     assert(req == SG_IO);
 570
 571     acb = qemu_aio_get(&iscsi_aio_pool, bs, cb, opaque);
 572
 573     acb->iscsilun = iscsilun;
 574     acb->canceled    = 0;
 575     acb->bh          = NULL;
 576     acb->status      = -EINPROGRESS;
 577     acb->buf         = NULL;
 578     acb->ioh         = buf;
 579
 580     acb->task = malloc(sizeof(struct scsi_task));
 581     if (acb->task == NULL) {
 582         error_report("iSCSI: Failed to allocate task for scsi command. %s",
 583                      iscsi_get_error(iscsi));
 584         qemu_aio_release(acb);
 585         return NULL;
 586     }
 587     memset(acb->task, 0, sizeof(struct scsi_task));
 588
 589     switch (acb->ioh->dxfer_direction) {
 590     case SG_DXFER_TO_DEV:
 591         acb->task->xfer_dir = SCSI_XFER_WRITE;
 592         break;
 593     case SG_DXFER_FROM_DEV:
 594         acb->task->xfer_dir = SCSI_XFER_READ;
 595         break;
 596     default:
 597         acb->task->xfer_dir = SCSI_XFER_NONE;
 598         break;
 599     }
 600
 601     acb->task->cdb_size = acb->ioh->cmd_len;
 602     memcpy(&acb->task->cdb[0], acb->ioh->cmdp, acb->ioh->cmd_len);
 603     acb->task->expxferlen = acb->ioh->dxfer_len;
 604
 605     if (acb->task->xfer_dir == SCSI_XFER_WRITE) {
 606         data.data = acb->ioh->dxferp;
 607         data.size = acb->ioh->dxfer_len;
 608     }
 609     if (iscsi_scsi_command_async(iscsi, iscsilun->lun, acb->task,
 610                                  iscsi_aio_ioctl_cb,
 611                                  (acb->task->xfer_dir == SCSI_XFER_WRITE) ?
 612                                      &data : NULL,
 613                                  acb) != 0) {
 614         scsi_free_scsi_task(acb->task);
 615         qemu_aio_release(acb);
 616         return NULL;
 617     }
 618
 619     /* tell libiscsi to read straight into the buffer we got from ioctl */
 620     if (acb->task->xfer_dir == SCSI_XFER_READ) {
 621         scsi_task_add_data_in_buffer(acb->task,
 622                                      acb->ioh->dxfer_len,
 623                                      acb->ioh->dxferp);
 624     }
 625
 626     iscsi_set_events(iscsilun);
 627
 628     return &acb->common;
 629 }
 630
 631
 632 static void ioctl_cb(void *opaque, int status)
 633 {
 634     int *p_status = opaque;
 635     *p_status = status;
 636 }
 637
 638 static int iscsi_ioctl(BlockDriverState *bs, unsigned long int req, void *buf)
 639 {
 640     IscsiLun *iscsilun = bs->opaque;
 641     int status;
 642
 643     switch (req) {
 644     case SG_GET_VERSION_NUM:
 645         *(int *)buf = 30000;
 646         break;
 647     case SG_GET_SCSI_ID:
 648         ((struct sg_scsi_id *)buf)->scsi_type = iscsilun->type;
 649         break;
 650     case SG_IO:
 651         status = -EINPROGRESS;
 652         iscsi_aio_ioctl(bs, req, buf, ioctl_cb, &status);
 653
 654         while (status == -EINPROGRESS) {
 655             qemu_aio_wait();
 656         }
 657
 658         return 0;
 659     default:
 660         return -1;
 661     }
 662     return 0;
 663 }
 664 #endif
 665
 666 static int64_t
 667 iscsi_getlength(BlockDriverState *bs)
 668 {
 669     IscsiLun *iscsilun = bs->opaque;
 670     int64_t len;
 671
 672     len  = iscsilun->num_blocks;
 673     len *= iscsilun->block_size;
 674
 675     return len;
 676 }
 677
 678 static void
 679 iscsi_readcapacity16_cb(struct iscsi_context *iscsi, int status,
 680                         void *command_data, void *opaque)
 681 {
 682     struct IscsiTask *itask = opaque;
 683     struct scsi_readcapacity16 *rc16;
 684     struct scsi_task *task = command_data;
 685
 686     if (status != 0) {
 687         error_report("iSCSI: Failed to read capacity of iSCSI lun. %s",
 688                      iscsi_get_error(iscsi));
 689         itask->status   = 1;
 690         itask->complete = 1;
 691         scsi_free_scsi_task(task);
 692         return;
 693     }
 694
 695     rc16 = scsi_datain_unmarshall(task);
 696     if (rc16 == NULL) {
 697         error_report("iSCSI: Failed to unmarshall readcapacity16 data.");
 698         itask->status   = 1;
 699         itask->complete = 1;
 700         scsi_free_scsi_task(task);
 701         return;
 702     }
 703
 704     itask->iscsilun->block_size = rc16->block_length;
 705     itask->iscsilun->num_blocks = rc16->returned_lba + 1;
 706     itask->bs->total_sectors    = itask->iscsilun->num_blocks *
 707                                itask->iscsilun->block_size / BDRV_SECTOR_SIZE ;
 708
 709     itask->status   = 0;
 710     itask->complete = 1;
 711     scsi_free_scsi_task(task);
 712 }
 713
 714 static void
 715 iscsi_readcapacity10_cb(struct iscsi_context *iscsi, int status,
 716                         void *command_data, void *opaque)
 717 {
 718     struct IscsiTask *itask = opaque;
 719     struct scsi_readcapacity10 *rc10;
 720     struct scsi_task *task = command_data;
 721
 722     if (status != 0) {
 723         error_report("iSCSI: Failed to read capacity of iSCSI lun. %s",
 724                      iscsi_get_error(iscsi));
 725         itask->status   = 1;
 726         itask->complete = 1;
 727         scsi_free_scsi_task(task);
 728         return;
 729     }
 730
 731     rc10 = scsi_datain_unmarshall(task);
 732     if (rc10 == NULL) {
 733         error_report("iSCSI: Failed to unmarshall readcapacity10 data.");
 734         itask->status   = 1;
 735         itask->complete = 1;
 736         scsi_free_scsi_task(task);
 737         return;
 738     }
 739
 740     itask->iscsilun->block_size = rc10->block_size;
 741     if (rc10->lba == 0) {
 742         /* blank disk loaded */
 743         itask->iscsilun->num_blocks = 0;
 744     } else {
 745         itask->iscsilun->num_blocks = rc10->lba + 1;
 746     }
 747     itask->bs->total_sectors    = itask->iscsilun->num_blocks *
 748                                itask->iscsilun->block_size / BDRV_SECTOR_SIZE ;
 749
 750     itask->status   = 0;
 751     itask->complete = 1;
 752     scsi_free_scsi_task(task);
 753 }
 754
 755 static void
 756 iscsi_inquiry_cb(struct iscsi_context *iscsi, int status, void *command_data,
 757                  void *opaque)
 758 {
 759     struct IscsiTask *itask = opaque;
 760     struct scsi_task *task = command_data;
 761     struct scsi_inquiry_standard *inq;
 762
 763     if (status != 0) {
 764         itask->status   = 1;
 765         itask->complete = 1;
 766         scsi_free_scsi_task(task);
 767         return;
 768     }
 769
 770     inq = scsi_datain_unmarshall(task);
 771     if (inq == NULL) {
 772         error_report("iSCSI: Failed to unmarshall inquiry data.");
 773         itask->status   = 1;
 774         itask->complete = 1;
 775         scsi_free_scsi_task(task);
 776         return;
 777     }
 778
 779     itask->iscsilun->type = inq->periperal_device_type;
 780
 781     scsi_free_scsi_task(task);
 782
 783     switch (itask->iscsilun->type) {
 784     case TYPE_DISK:
 785         task = iscsi_readcapacity16_task(iscsi, itask->iscsilun->lun,
 786                                    iscsi_readcapacity16_cb, opaque);
 787         if (task == NULL) {
 788             error_report("iSCSI: failed to send readcapacity16 command.");
 789             itask->status   = 1;
 790             itask->complete = 1;
 791             return;
 792         }
 793         break;
 794     case TYPE_ROM:
 795         task = iscsi_readcapacity10_task(iscsi, itask->iscsilun->lun,
 796                                    0, 0,
 797                                    iscsi_readcapacity10_cb, opaque);
 798         if (task == NULL) {
 799             error_report("iSCSI: failed to send readcapacity16 command.");
 800             itask->status   = 1;
 801             itask->complete = 1;
 802             return;
 803         }
 804         break;
 805     default:
 806         itask->status   = 0;
 807         itask->complete = 1;
 808     }
 809 }
 810
 811 static void
 812 iscsi_connect_cb(struct iscsi_context *iscsi, int status, void *command_data,
 813                  void *opaque)
 814 {
 815     struct IscsiTask *itask = opaque;
 816     struct scsi_task *task;
 817
 818     if (status != 0) {
 819         itask->status   = 1;
 820         itask->complete = 1;
 821         return;
 822     }
 823
 824     task = iscsi_inquiry_task(iscsi, itask->iscsilun->lun,
 825                               0, 0, 36,
 826                               iscsi_inquiry_cb, opaque);
 827     if (task == NULL) {
 828         error_report("iSCSI: failed to send inquiry command.");
 829         itask->status   = 1;
 830         itask->complete = 1;
 831         return;
 832     }
 833 }
 834
 835 static int parse_chap(struct iscsi_context *iscsi, const char *target)
 836 {
 837     QemuOptsList *list;
 838     QemuOpts *opts;
 839     const char *user = NULL;
 840     const char *password = NULL;
 841
 842     list = qemu_find_opts("iscsi");
 843     if (!list) {
 844         return 0;
 845     }
 846
 847     opts = qemu_opts_find(list, target);
 848     if (opts == NULL) {
 849         opts = QTAILQ_FIRST(&list->head);
 850         if (!opts) {
 851             return 0;
 852         }
 853     }
 854
 855     user = qemu_opt_get(opts, "user");
 856     if (!user) {
 857         return 0;
 858     }
 859
 860     password = qemu_opt_get(opts, "password");
 861     if (!password) {
 862         error_report("CHAP username specified but no password was given");
 863         return -1;
 864     }
 865
 866     if (iscsi_set_initiator_username_pwd(iscsi, user, password)) {
 867         error_report("Failed to set initiator username and password");
 868         return -1;
 869     }
 870
 871     return 0;
 872 }
 873
 874 static void parse_header_digest(struct iscsi_context *iscsi, const char *target)
 875 {
 876     QemuOptsList *list;
 877     QemuOpts *opts;
 878     const char *digest = NULL;
 879
 880     list = qemu_find_opts("iscsi");
 881     if (!list) {
 882         return;
 883     }
 884
 885     opts = qemu_opts_find(list, target);
 886     if (opts == NULL) {
 887         opts = QTAILQ_FIRST(&list->head);
 888         if (!opts) {
 889             return;
 890         }
 891     }
 892
 893     digest = qemu_opt_get(opts, "header-digest");
 894     if (!digest) {
 895         return;
 896     }
 897
 898     if (!strcmp(digest, "CRC32C")) {
 899         iscsi_set_header_digest(iscsi, ISCSI_HEADER_DIGEST_CRC32C);
 900     } else if (!strcmp(digest, "NONE")) {
 901         iscsi_set_header_digest(iscsi, ISCSI_HEADER_DIGEST_NONE);
 902     } else if (!strcmp(digest, "CRC32C-NONE")) {
 903         iscsi_set_header_digest(iscsi, ISCSI_HEADER_DIGEST_CRC32C_NONE);
 904     } else if (!strcmp(digest, "NONE-CRC32C")) {
 905         iscsi_set_header_digest(iscsi, ISCSI_HEADER_DIGEST_NONE_CRC32C);
 906     } else {
 907         error_report("Invalid header-digest setting : %s", digest);
 908     }
 909 }
 910
 911 static char *parse_initiator_name(const char *target)
 912 {
 913     QemuOptsList *list;
 914     QemuOpts *opts;
 915     const char *name = NULL;
 916     const char *iscsi_name = qemu_get_vm_name();
 917
 918     list = qemu_find_opts("iscsi");
 919     if (list) {
 920         opts = qemu_opts_find(list, target);
 921         if (!opts) {
 922             opts = QTAILQ_FIRST(&list->head);
 923         }
 924         if (opts) {
 925             name = qemu_opt_get(opts, "initiator-name");
 926         }
 927     }
 928
 929     if (name) {
 930         return g_strdup(name);
 931     } else {
 932         return g_strdup_printf("iqn.2008-11.org.linux-kvm%s%s",
 933                                iscsi_name ? ":" : "",
 934                                iscsi_name ? iscsi_name : "");
 935     }
 936 }
 937
 938 /*
 939  * We support iscsi url's on the form
 940  * iscsi://[<username>%<password>@]<host>[:<port>]/<targetname>/<lun>
 941  */
 942 static int iscsi_open(BlockDriverState *bs, const char *filename, int flags)
 943 {
 944     IscsiLun *iscsilun = bs->opaque;
 945     struct iscsi_context *iscsi = NULL;
 946     struct iscsi_url *iscsi_url = NULL;
 947     struct IscsiTask task;
 948     char *initiator_name = NULL;
 949     int ret;
 950
 951     if ((BDRV_SECTOR_SIZE % 512) != 0) {
 952         error_report("iSCSI: Invalid BDRV_SECTOR_SIZE. "
 953                      "BDRV_SECTOR_SIZE(%lld) is not a multiple "
 954                      "of 512", BDRV_SECTOR_SIZE);
 955         return -EINVAL;
 956     }
 957
 958     iscsi_url = iscsi_parse_full_url(iscsi, filename);
 959     if (iscsi_url == NULL) {
 960         error_report("Failed to parse URL : %s %s", filename,
 961                      iscsi_get_error(iscsi));
 962         ret = -EINVAL;
 963         goto out;
 964     }
 965
 966     memset(iscsilun, 0, sizeof(IscsiLun));
 967
 968     initiator_name = parse_initiator_name(iscsi_url->target);
 969
 970     iscsi = iscsi_create_context(initiator_name);
 971     if (iscsi == NULL) {
 972         error_report("iSCSI: Failed to create iSCSI context.");
 973         ret = -ENOMEM;
 974         goto out;
 975     }
 976
 977     if (iscsi_set_targetname(iscsi, iscsi_url->target)) {
 978         error_report("iSCSI: Failed to set target name.");
 979         ret = -EINVAL;
 980         goto out;
 981     }
 982
 983     if (iscsi_url->user != NULL) {
 984         ret = iscsi_set_initiator_username_pwd(iscsi, iscsi_url->user,
 985                                               iscsi_url->passwd);
 986         if (ret != 0) {
 987             error_report("Failed to set initiator username and password");
 988             ret = -EINVAL;
 989             goto out;
 990         }
 991     }
 992
 993     /* check if we got CHAP username/password via the options */
 994     if (parse_chap(iscsi, iscsi_url->target) != 0) {
 995         error_report("iSCSI: Failed to set CHAP user/password");
 996         ret = -EINVAL;
 997         goto out;
 998     }
 999
1000     if (iscsi_set_session_type(iscsi, ISCSI_SESSION_NORMAL) != 0) {
1001         error_report("iSCSI: Failed to set session type to normal.");
1002         ret = -EINVAL;
1003         goto out;
1004     }
1005
1006     iscsi_set_header_digest(iscsi, ISCSI_HEADER_DIGEST_NONE_CRC32C);
1007
1008     /* check if we got HEADER_DIGEST via the options */
1009     parse_header_digest(iscsi, iscsi_url->target);
1010
1011     task.iscsilun = iscsilun;
1012     task.status = 0;
1013     task.complete = 0;
1014     task.bs = bs;
1015
1016     iscsilun->iscsi = iscsi;
1017     iscsilun->lun   = iscsi_url->lun;
1018
1019     if (iscsi_full_connect_async(iscsi, iscsi_url->portal, iscsi_url->lun,
1020                                  iscsi_connect_cb, &task)
1021         != 0) {
1022         error_report("iSCSI: Failed to start async connect.");
1023         ret = -EINVAL;
1024         goto out;
1025     }
1026
1027     while (!task.complete) {
1028         iscsi_set_events(iscsilun);
1029         qemu_aio_wait();
1030     }
1031     if (task.status != 0) {
1032         error_report("iSCSI: Failed to connect to LUN : %s",
1033                      iscsi_get_error(iscsi));
1034         ret = -EINVAL;
1035         goto out;
1036     }
1037
1038     /* Medium changer or tape. We dont have any emulation for this so this must
1039      * be sg ioctl compatible. We force it to be sg, otherwise qemu will try
1040      * to read from the device to guess the image format.
1041      */
1042     if (iscsilun->type == TYPE_MEDIUM_CHANGER ||
1043         iscsilun->type == TYPE_TAPE) {
1044         bs->sg = 1;
1045     }
1046
1047     ret = 0;
1048
1049 out:
1050     if (initiator_name != NULL) {
1051         g_free(initiator_name);
1052     }
1053     if (iscsi_url != NULL) {
1054         iscsi_destroy_url(iscsi_url);
1055     }
1056
1057     if (ret) {
1058         if (iscsi != NULL) {
1059             iscsi_destroy_context(iscsi);
1060         }
1061         memset(iscsilun, 0, sizeof(IscsiLun));
1062     }
1063     return ret;
1064 }
1065
1066 static void iscsi_close(BlockDriverState *bs)
1067 {
1068     IscsiLun *iscsilun = bs->opaque;
1069     struct iscsi_context *iscsi = iscsilun->iscsi;
1070
1071     qemu_aio_set_fd_handler(iscsi_get_fd(iscsi), NULL, NULL, NULL, NULL);
1072     iscsi_destroy_context(iscsi);
1073     memset(iscsilun, 0, sizeof(IscsiLun));
1074 }
1075
1076 static BlockDriver bdrv_iscsi = {
1077     .format_name     = "iscsi",
1078     .protocol_name   = "iscsi",
1079
1080     .instance_size   = sizeof(IscsiLun),
1081     .bdrv_file_open  = iscsi_open,
1082     .bdrv_close      = iscsi_close,
1083
1084     .bdrv_getlength  = iscsi_getlength,
1085
1086     .bdrv_aio_readv  = iscsi_aio_readv,
1087     .bdrv_aio_writev = iscsi_aio_writev,
1088     .bdrv_aio_flush  = iscsi_aio_flush,
1089
1090     .bdrv_aio_discard = iscsi_aio_discard,
1091
1092 #ifdef __linux__
1093     .bdrv_ioctl       = iscsi_ioctl,
1094     .bdrv_aio_ioctl   = iscsi_aio_ioctl,
1095 #endif
1096 };
1097
1098 static void iscsi_block_init(void)
1099 {
1100     bdrv_register(&bdrv_iscsi);
1101 }
1102
1103 block_init(iscsi_block_init);