xref: /qemu/contrib/vhost-user-scsi/vhost-user-scsi.c (revision 8c36ed6e0275613e2c3191d251e5ff96eb4e8727)
149cc0340SFelipe Franciosi /*
249cc0340SFelipe Franciosi  * vhost-user-scsi sample application
349cc0340SFelipe Franciosi  *
449cc0340SFelipe Franciosi  * Copyright (c) 2016 Nutanix Inc. All rights reserved.
549cc0340SFelipe Franciosi  *
649cc0340SFelipe Franciosi  * Author:
749cc0340SFelipe Franciosi  *  Felipe Franciosi <felipe@nutanix.com>
849cc0340SFelipe Franciosi  *
949cc0340SFelipe Franciosi  * This work is licensed under the terms of the GNU GPL, version 2 only.
1049cc0340SFelipe Franciosi  * See the COPYING file in the top-level directory.
1149cc0340SFelipe Franciosi  */
1249cc0340SFelipe Franciosi 
1349cc0340SFelipe Franciosi #include "qemu/osdep.h"
1449cc0340SFelipe Franciosi #include "contrib/libvhost-user/libvhost-user.h"
1549cc0340SFelipe Franciosi #include "hw/virtio/virtio-scsi.h"
1649cc0340SFelipe Franciosi #include "iscsi/iscsi.h"
1749cc0340SFelipe Franciosi 
1849cc0340SFelipe Franciosi #include <glib.h>
1949cc0340SFelipe Franciosi 
2049cc0340SFelipe Franciosi /* #define VUS_DEBUG 1 */
2149cc0340SFelipe Franciosi 
2249cc0340SFelipe Franciosi /** Log helpers **/
2349cc0340SFelipe Franciosi 
2449cc0340SFelipe Franciosi #define PPRE                                                          \
2549cc0340SFelipe Franciosi     struct timespec ts;                                               \
2649cc0340SFelipe Franciosi     char   timebuf[64];                                               \
2749cc0340SFelipe Franciosi     struct tm tm;                                                     \
2849cc0340SFelipe Franciosi     (void)clock_gettime(CLOCK_REALTIME, &ts);                         \
2949cc0340SFelipe Franciosi     (void)strftime(timebuf, 64, "%Y%m%d %T", gmtime_r(&ts.tv_sec, &tm))
3049cc0340SFelipe Franciosi 
3149cc0340SFelipe Franciosi #define PEXT(lvl, msg, ...) do {                                      \
3249cc0340SFelipe Franciosi     PPRE;                                                             \
3349cc0340SFelipe Franciosi     fprintf(stderr, "%s.%06ld " lvl ": %s:%s():%d: " msg "\n",        \
3449cc0340SFelipe Franciosi             timebuf, ts.tv_nsec / 1000,                               \
3549cc0340SFelipe Franciosi             __FILE__, __func__, __LINE__, ## __VA_ARGS__);            \
3649cc0340SFelipe Franciosi } while (0)
3749cc0340SFelipe Franciosi 
3849cc0340SFelipe Franciosi #define PNOR(lvl, msg, ...) do {                                      \
3949cc0340SFelipe Franciosi     PPRE;                                                             \
4049cc0340SFelipe Franciosi     fprintf(stderr, "%s.%06ld " lvl ": " msg "\n",                    \
4149cc0340SFelipe Franciosi             timebuf, ts.tv_nsec / 1000, ## __VA_ARGS__);              \
4249cc0340SFelipe Franciosi } while (0)
4349cc0340SFelipe Franciosi 
4449cc0340SFelipe Franciosi #ifdef VUS_DEBUG
4549cc0340SFelipe Franciosi #define PDBG(msg, ...) PEXT("DBG", msg, ## __VA_ARGS__)
4649cc0340SFelipe Franciosi #define PERR(msg, ...) PEXT("ERR", msg, ## __VA_ARGS__)
4749cc0340SFelipe Franciosi #define PLOG(msg, ...) PEXT("LOG", msg, ## __VA_ARGS__)
4849cc0340SFelipe Franciosi #else
4949cc0340SFelipe Franciosi #define PDBG(msg, ...) { }
5049cc0340SFelipe Franciosi #define PERR(msg, ...) PNOR("ERR", msg, ## __VA_ARGS__)
5149cc0340SFelipe Franciosi #define PLOG(msg, ...) PNOR("LOG", msg, ## __VA_ARGS__)
5249cc0340SFelipe Franciosi #endif
5349cc0340SFelipe Franciosi 
5449cc0340SFelipe Franciosi /** vhost-user-scsi specific definitions **/
5549cc0340SFelipe Franciosi 
5649cc0340SFelipe Franciosi  /* Only 1 LUN and device supported today */
5749cc0340SFelipe Franciosi #define VUS_MAX_LUNS 1
5849cc0340SFelipe Franciosi #define VUS_MAX_DEVS 1
5949cc0340SFelipe Franciosi 
6049cc0340SFelipe Franciosi #define VUS_ISCSI_INITIATOR "iqn.2016-11.com.nutanix:vhost-user-scsi"
6149cc0340SFelipe Franciosi 
6249cc0340SFelipe Franciosi typedef struct iscsi_lun {
6349cc0340SFelipe Franciosi     struct iscsi_context *iscsi_ctx;
6449cc0340SFelipe Franciosi     int iscsi_lun;
6549cc0340SFelipe Franciosi } iscsi_lun_t;
6649cc0340SFelipe Franciosi 
6749cc0340SFelipe Franciosi typedef struct vhost_scsi_dev {
6849cc0340SFelipe Franciosi     VuDev vu_dev;
6949cc0340SFelipe Franciosi     int server_sock;
7049cc0340SFelipe Franciosi     GMainLoop *loop;
7149cc0340SFelipe Franciosi     GTree *fdmap;   /* fd -> gsource context id */
7249cc0340SFelipe Franciosi     iscsi_lun_t luns[VUS_MAX_LUNS];
7349cc0340SFelipe Franciosi } vhost_scsi_dev_t;
7449cc0340SFelipe Franciosi 
7549cc0340SFelipe Franciosi static vhost_scsi_dev_t *vhost_scsi_devs[VUS_MAX_DEVS];
7649cc0340SFelipe Franciosi 
7749cc0340SFelipe Franciosi /** glib event loop integration for libvhost-user and misc callbacks **/
7849cc0340SFelipe Franciosi 
7949cc0340SFelipe Franciosi QEMU_BUILD_BUG_ON((int)G_IO_IN != (int)VU_WATCH_IN);
8049cc0340SFelipe Franciosi QEMU_BUILD_BUG_ON((int)G_IO_OUT != (int)VU_WATCH_OUT);
8149cc0340SFelipe Franciosi QEMU_BUILD_BUG_ON((int)G_IO_PRI != (int)VU_WATCH_PRI);
8249cc0340SFelipe Franciosi QEMU_BUILD_BUG_ON((int)G_IO_ERR != (int)VU_WATCH_ERR);
8349cc0340SFelipe Franciosi QEMU_BUILD_BUG_ON((int)G_IO_HUP != (int)VU_WATCH_HUP);
8449cc0340SFelipe Franciosi 
8549cc0340SFelipe Franciosi typedef struct vus_gsrc {
8649cc0340SFelipe Franciosi     GSource parent;
8749cc0340SFelipe Franciosi     vhost_scsi_dev_t *vdev_scsi;
8849cc0340SFelipe Franciosi     GPollFD gfd;
8949cc0340SFelipe Franciosi     vu_watch_cb vu_cb;
9049cc0340SFelipe Franciosi } vus_gsrc_t;
9149cc0340SFelipe Franciosi 
9249cc0340SFelipe Franciosi static gint vus_fdmap_compare(gconstpointer a, gconstpointer b)
9349cc0340SFelipe Franciosi {
9449cc0340SFelipe Franciosi     return (b > a) - (b < a);
9549cc0340SFelipe Franciosi }
9649cc0340SFelipe Franciosi 
9749cc0340SFelipe Franciosi static gboolean vus_gsrc_prepare(GSource *src, gint *timeout)
9849cc0340SFelipe Franciosi {
9949cc0340SFelipe Franciosi     assert(timeout);
10049cc0340SFelipe Franciosi 
10149cc0340SFelipe Franciosi     *timeout = -1;
10249cc0340SFelipe Franciosi     return FALSE;
10349cc0340SFelipe Franciosi }
10449cc0340SFelipe Franciosi 
10549cc0340SFelipe Franciosi static gboolean vus_gsrc_check(GSource *src)
10649cc0340SFelipe Franciosi {
10749cc0340SFelipe Franciosi     vus_gsrc_t *vus_src = (vus_gsrc_t *)src;
10849cc0340SFelipe Franciosi 
10949cc0340SFelipe Franciosi     assert(vus_src);
11049cc0340SFelipe Franciosi 
11149cc0340SFelipe Franciosi     return vus_src->gfd.revents & vus_src->gfd.events;
11249cc0340SFelipe Franciosi }
11349cc0340SFelipe Franciosi 
11449cc0340SFelipe Franciosi static gboolean vus_gsrc_dispatch(GSource *src, GSourceFunc cb, gpointer data)
11549cc0340SFelipe Franciosi {
11649cc0340SFelipe Franciosi     vhost_scsi_dev_t *vdev_scsi;
11749cc0340SFelipe Franciosi     vus_gsrc_t *vus_src = (vus_gsrc_t *)src;
11849cc0340SFelipe Franciosi 
11949cc0340SFelipe Franciosi     assert(vus_src);
12049cc0340SFelipe Franciosi     assert(!(vus_src->vu_cb && cb));
12149cc0340SFelipe Franciosi 
12249cc0340SFelipe Franciosi     vdev_scsi = vus_src->vdev_scsi;
12349cc0340SFelipe Franciosi 
12449cc0340SFelipe Franciosi     assert(vdev_scsi);
12549cc0340SFelipe Franciosi 
12649cc0340SFelipe Franciosi     if (cb) {
12749cc0340SFelipe Franciosi         return cb(data);
12849cc0340SFelipe Franciosi     }
12949cc0340SFelipe Franciosi     if (vus_src->vu_cb) {
13049cc0340SFelipe Franciosi         vus_src->vu_cb(&vdev_scsi->vu_dev, vus_src->gfd.revents, data);
13149cc0340SFelipe Franciosi     }
13249cc0340SFelipe Franciosi     return G_SOURCE_CONTINUE;
13349cc0340SFelipe Franciosi }
13449cc0340SFelipe Franciosi 
13549cc0340SFelipe Franciosi static GSourceFuncs vus_gsrc_funcs = {
13649cc0340SFelipe Franciosi     vus_gsrc_prepare,
13749cc0340SFelipe Franciosi     vus_gsrc_check,
13849cc0340SFelipe Franciosi     vus_gsrc_dispatch,
13949cc0340SFelipe Franciosi     NULL
14049cc0340SFelipe Franciosi };
14149cc0340SFelipe Franciosi 
14249cc0340SFelipe Franciosi static int vus_gsrc_new(vhost_scsi_dev_t *vdev_scsi, int fd, GIOCondition cond,
14349cc0340SFelipe Franciosi                         vu_watch_cb vu_cb, GSourceFunc gsrc_cb, gpointer data)
14449cc0340SFelipe Franciosi {
14549cc0340SFelipe Franciosi     GSource *vus_gsrc;
14649cc0340SFelipe Franciosi     vus_gsrc_t *vus_src;
14749cc0340SFelipe Franciosi     guint id;
14849cc0340SFelipe Franciosi 
14949cc0340SFelipe Franciosi     assert(vdev_scsi);
15049cc0340SFelipe Franciosi     assert(fd >= 0);
15149cc0340SFelipe Franciosi     assert(vu_cb || gsrc_cb);
15249cc0340SFelipe Franciosi     assert(!(vu_cb && gsrc_cb));
15349cc0340SFelipe Franciosi 
15449cc0340SFelipe Franciosi     vus_gsrc = g_source_new(&vus_gsrc_funcs, sizeof(vus_gsrc_t));
15549cc0340SFelipe Franciosi     if (!vus_gsrc) {
15649cc0340SFelipe Franciosi         PERR("Error creating GSource for new watch");
15749cc0340SFelipe Franciosi         return -1;
15849cc0340SFelipe Franciosi     }
15949cc0340SFelipe Franciosi     vus_src = (vus_gsrc_t *)vus_gsrc;
16049cc0340SFelipe Franciosi 
16149cc0340SFelipe Franciosi     vus_src->vdev_scsi = vdev_scsi;
16249cc0340SFelipe Franciosi     vus_src->gfd.fd = fd;
16349cc0340SFelipe Franciosi     vus_src->gfd.events = cond;
16449cc0340SFelipe Franciosi     vus_src->vu_cb = vu_cb;
16549cc0340SFelipe Franciosi 
16649cc0340SFelipe Franciosi     g_source_add_poll(vus_gsrc, &vus_src->gfd);
16749cc0340SFelipe Franciosi     g_source_set_callback(vus_gsrc, gsrc_cb, data, NULL);
16849cc0340SFelipe Franciosi     id = g_source_attach(vus_gsrc, NULL);
16949cc0340SFelipe Franciosi     assert(id);
17049cc0340SFelipe Franciosi     g_source_unref(vus_gsrc);
17149cc0340SFelipe Franciosi 
17249cc0340SFelipe Franciosi     g_tree_insert(vdev_scsi->fdmap, (gpointer)(uintptr_t)fd,
17349cc0340SFelipe Franciosi                                     (gpointer)(uintptr_t)id);
17449cc0340SFelipe Franciosi 
17549cc0340SFelipe Franciosi     return 0;
17649cc0340SFelipe Franciosi }
17749cc0340SFelipe Franciosi 
17849cc0340SFelipe Franciosi /* from libiscsi's scsi-lowlevel.h **
17949cc0340SFelipe Franciosi  *
18049cc0340SFelipe Franciosi  * nb. We can't directly include scsi-lowlevel.h due to a namespace conflict:
18149cc0340SFelipe Franciosi  *     QEMU's scsi.h also defines "SCSI_XFER_NONE".
18249cc0340SFelipe Franciosi  */
18349cc0340SFelipe Franciosi 
18449cc0340SFelipe Franciosi #define SCSI_CDB_MAX_SIZE           16
18549cc0340SFelipe Franciosi 
18649cc0340SFelipe Franciosi struct scsi_iovector {
18749cc0340SFelipe Franciosi     struct scsi_iovec *iov;
18849cc0340SFelipe Franciosi     int niov;
18949cc0340SFelipe Franciosi     int nalloc;
19049cc0340SFelipe Franciosi     size_t offset;
19149cc0340SFelipe Franciosi     int consumed;
19249cc0340SFelipe Franciosi };
19349cc0340SFelipe Franciosi 
19449cc0340SFelipe Franciosi struct scsi_allocated_memory {
19549cc0340SFelipe Franciosi     struct scsi_allocated_memory *next;
19649cc0340SFelipe Franciosi     char buf[0];
19749cc0340SFelipe Franciosi };
19849cc0340SFelipe Franciosi 
19949cc0340SFelipe Franciosi struct scsi_data {
20049cc0340SFelipe Franciosi     int            size;
20149cc0340SFelipe Franciosi     unsigned char *data;
20249cc0340SFelipe Franciosi };
20349cc0340SFelipe Franciosi 
20449cc0340SFelipe Franciosi enum scsi_sense_key {
20549cc0340SFelipe Franciosi     SCSI_SENSE_NO_SENSE            = 0x00,
20649cc0340SFelipe Franciosi     SCSI_SENSE_RECOVERED_ERROR     = 0x01,
20749cc0340SFelipe Franciosi     SCSI_SENSE_NOT_READY           = 0x02,
20849cc0340SFelipe Franciosi     SCSI_SENSE_MEDIUM_ERROR        = 0x03,
20949cc0340SFelipe Franciosi     SCSI_SENSE_HARDWARE_ERROR      = 0x04,
21049cc0340SFelipe Franciosi     SCSI_SENSE_ILLEGAL_REQUEST     = 0x05,
21149cc0340SFelipe Franciosi     SCSI_SENSE_UNIT_ATTENTION      = 0x06,
21249cc0340SFelipe Franciosi     SCSI_SENSE_DATA_PROTECTION     = 0x07,
21349cc0340SFelipe Franciosi     SCSI_SENSE_BLANK_CHECK         = 0x08,
21449cc0340SFelipe Franciosi     SCSI_SENSE_VENDOR_SPECIFIC     = 0x09,
21549cc0340SFelipe Franciosi     SCSI_SENSE_COPY_ABORTED        = 0x0a,
21649cc0340SFelipe Franciosi     SCSI_SENSE_COMMAND_ABORTED     = 0x0b,
21749cc0340SFelipe Franciosi     SCSI_SENSE_OBSOLETE_ERROR_CODE = 0x0c,
21849cc0340SFelipe Franciosi     SCSI_SENSE_OVERFLOW_COMMAND    = 0x0d,
21949cc0340SFelipe Franciosi     SCSI_SENSE_MISCOMPARE          = 0x0e
22049cc0340SFelipe Franciosi };
22149cc0340SFelipe Franciosi 
22249cc0340SFelipe Franciosi struct scsi_sense {
22349cc0340SFelipe Franciosi     unsigned char       error_type;
22449cc0340SFelipe Franciosi     enum scsi_sense_key key;
22549cc0340SFelipe Franciosi     int                 ascq;
22649cc0340SFelipe Franciosi     unsigned            sense_specific:1;
22749cc0340SFelipe Franciosi     unsigned            ill_param_in_cdb:1;
22849cc0340SFelipe Franciosi     unsigned            bit_pointer_valid:1;
22949cc0340SFelipe Franciosi     unsigned char       bit_pointer;
23049cc0340SFelipe Franciosi     uint16_t            field_pointer;
23149cc0340SFelipe Franciosi };
23249cc0340SFelipe Franciosi 
23349cc0340SFelipe Franciosi enum scsi_residual {
23449cc0340SFelipe Franciosi     SCSI_RESIDUAL_NO_RESIDUAL = 0,
23549cc0340SFelipe Franciosi     SCSI_RESIDUAL_UNDERFLOW,
23649cc0340SFelipe Franciosi     SCSI_RESIDUAL_OVERFLOW
23749cc0340SFelipe Franciosi };
23849cc0340SFelipe Franciosi 
23949cc0340SFelipe Franciosi struct scsi_task {
24049cc0340SFelipe Franciosi     int status;
24149cc0340SFelipe Franciosi     int cdb_size;
24249cc0340SFelipe Franciosi     int xfer_dir;
24349cc0340SFelipe Franciosi     int expxferlen;
24449cc0340SFelipe Franciosi     unsigned char cdb[SCSI_CDB_MAX_SIZE];
24549cc0340SFelipe Franciosi     enum scsi_residual residual_status;
24649cc0340SFelipe Franciosi     size_t residual;
24749cc0340SFelipe Franciosi     struct scsi_sense sense;
24849cc0340SFelipe Franciosi     struct scsi_data datain;
24949cc0340SFelipe Franciosi     struct scsi_allocated_memory *mem;
25049cc0340SFelipe Franciosi     void *ptr;
25149cc0340SFelipe Franciosi 
25249cc0340SFelipe Franciosi     uint32_t itt;
25349cc0340SFelipe Franciosi     uint32_t cmdsn;
25449cc0340SFelipe Franciosi     uint32_t lun;
25549cc0340SFelipe Franciosi 
25649cc0340SFelipe Franciosi     struct scsi_iovector iovector_in;
25749cc0340SFelipe Franciosi     struct scsi_iovector iovector_out;
25849cc0340SFelipe Franciosi };
25949cc0340SFelipe Franciosi 
26049cc0340SFelipe Franciosi /** libiscsi integration **/
26149cc0340SFelipe Franciosi 
26249cc0340SFelipe Franciosi static int iscsi_add_lun(iscsi_lun_t *lun, char *iscsi_uri)
26349cc0340SFelipe Franciosi {
26449cc0340SFelipe Franciosi     struct iscsi_url *iscsi_url;
26549cc0340SFelipe Franciosi     struct iscsi_context *iscsi_ctx;
26649cc0340SFelipe Franciosi     int ret = 0;
26749cc0340SFelipe Franciosi 
26849cc0340SFelipe Franciosi     assert(lun);
26949cc0340SFelipe Franciosi     assert(iscsi_uri);
27049cc0340SFelipe Franciosi 
27149cc0340SFelipe Franciosi     iscsi_ctx = iscsi_create_context(VUS_ISCSI_INITIATOR);
27249cc0340SFelipe Franciosi     if (!iscsi_ctx) {
27349cc0340SFelipe Franciosi         PERR("Unable to create iSCSI context");
27449cc0340SFelipe Franciosi         return -1;
27549cc0340SFelipe Franciosi     }
27649cc0340SFelipe Franciosi 
27749cc0340SFelipe Franciosi     iscsi_url = iscsi_parse_full_url(iscsi_ctx, iscsi_uri);
27849cc0340SFelipe Franciosi     if (!iscsi_url) {
27949cc0340SFelipe Franciosi         PERR("Unable to parse iSCSI URL: %s", iscsi_get_error(iscsi_ctx));
28049cc0340SFelipe Franciosi         goto fail;
28149cc0340SFelipe Franciosi     }
28249cc0340SFelipe Franciosi 
28349cc0340SFelipe Franciosi     iscsi_set_session_type(iscsi_ctx, ISCSI_SESSION_NORMAL);
28449cc0340SFelipe Franciosi     iscsi_set_header_digest(iscsi_ctx, ISCSI_HEADER_DIGEST_NONE_CRC32C);
28549cc0340SFelipe Franciosi     if (iscsi_full_connect_sync(iscsi_ctx, iscsi_url->portal, iscsi_url->lun)) {
28649cc0340SFelipe Franciosi         PERR("Unable to login to iSCSI portal: %s", iscsi_get_error(iscsi_ctx));
28749cc0340SFelipe Franciosi         goto fail;
28849cc0340SFelipe Franciosi     }
28949cc0340SFelipe Franciosi 
29049cc0340SFelipe Franciosi     lun->iscsi_ctx = iscsi_ctx;
29149cc0340SFelipe Franciosi     lun->iscsi_lun = iscsi_url->lun;
29249cc0340SFelipe Franciosi 
29349cc0340SFelipe Franciosi     PDBG("Context %p created for lun 0: %s", iscsi_ctx, iscsi_uri);
29449cc0340SFelipe Franciosi 
29549cc0340SFelipe Franciosi out:
29649cc0340SFelipe Franciosi     if (iscsi_url) {
29749cc0340SFelipe Franciosi         iscsi_destroy_url(iscsi_url);
29849cc0340SFelipe Franciosi     }
29949cc0340SFelipe Franciosi     return ret;
30049cc0340SFelipe Franciosi 
30149cc0340SFelipe Franciosi fail:
30249cc0340SFelipe Franciosi     (void)iscsi_destroy_context(iscsi_ctx);
30349cc0340SFelipe Franciosi     ret = -1;
30449cc0340SFelipe Franciosi     goto out;
30549cc0340SFelipe Franciosi }
30649cc0340SFelipe Franciosi 
30749cc0340SFelipe Franciosi static struct scsi_task *scsi_task_new(int cdb_len, uint8_t *cdb, int dir,
308*8c36ed6eSMarc-André Lureau                                        int xfer_len)
309*8c36ed6eSMarc-André Lureau {
31049cc0340SFelipe Franciosi     struct scsi_task *task;
31149cc0340SFelipe Franciosi 
31249cc0340SFelipe Franciosi     assert(cdb_len > 0);
31349cc0340SFelipe Franciosi     assert(cdb);
31449cc0340SFelipe Franciosi 
31549cc0340SFelipe Franciosi     task = calloc(1, sizeof(struct scsi_task));
31649cc0340SFelipe Franciosi     if (!task) {
31749cc0340SFelipe Franciosi         PERR("Error allocating task: %s", strerror(errno));
31849cc0340SFelipe Franciosi         return NULL;
31949cc0340SFelipe Franciosi     }
32049cc0340SFelipe Franciosi 
32149cc0340SFelipe Franciosi     memcpy(task->cdb, cdb, cdb_len);
32249cc0340SFelipe Franciosi     task->cdb_size = cdb_len;
32349cc0340SFelipe Franciosi     task->xfer_dir = dir;
32449cc0340SFelipe Franciosi     task->expxferlen = xfer_len;
32549cc0340SFelipe Franciosi 
32649cc0340SFelipe Franciosi     return task;
32749cc0340SFelipe Franciosi }
32849cc0340SFelipe Franciosi 
32949cc0340SFelipe Franciosi static int get_cdb_len(uint8_t *cdb)
33049cc0340SFelipe Franciosi {
33149cc0340SFelipe Franciosi     assert(cdb);
33249cc0340SFelipe Franciosi 
33349cc0340SFelipe Franciosi     switch (cdb[0] >> 5) {
33449cc0340SFelipe Franciosi     case 0: return 6;
33549cc0340SFelipe Franciosi     case 1: /* fall through */
33649cc0340SFelipe Franciosi     case 2: return 10;
33749cc0340SFelipe Franciosi     case 4: return 16;
33849cc0340SFelipe Franciosi     case 5: return 12;
33949cc0340SFelipe Franciosi     }
34049cc0340SFelipe Franciosi     PERR("Unable to determine cdb len (0x%02hhX)", cdb[0] >> 5);
34149cc0340SFelipe Franciosi     return -1;
34249cc0340SFelipe Franciosi }
34349cc0340SFelipe Franciosi 
34449cc0340SFelipe Franciosi static int handle_cmd_sync(struct iscsi_context *ctx,
34549cc0340SFelipe Franciosi                            VirtIOSCSICmdReq *req,
34649cc0340SFelipe Franciosi                            struct iovec *out, unsigned int out_len,
34749cc0340SFelipe Franciosi                            VirtIOSCSICmdResp *rsp,
348*8c36ed6eSMarc-André Lureau                            struct iovec *in, unsigned int in_len)
349*8c36ed6eSMarc-André Lureau {
35049cc0340SFelipe Franciosi     struct scsi_task *task;
35149cc0340SFelipe Franciosi     uint32_t dir;
35249cc0340SFelipe Franciosi     uint32_t len;
35349cc0340SFelipe Franciosi     int cdb_len;
35449cc0340SFelipe Franciosi     int i;
35549cc0340SFelipe Franciosi 
35649cc0340SFelipe Franciosi     assert(ctx);
35749cc0340SFelipe Franciosi     assert(req);
35849cc0340SFelipe Franciosi     assert(rsp);
35949cc0340SFelipe Franciosi 
36049cc0340SFelipe Franciosi     if (!(!req->lun[1] && req->lun[2] == 0x40 && !req->lun[3])) {
36149cc0340SFelipe Franciosi         /* Ignore anything different than target=0, lun=0 */
36249cc0340SFelipe Franciosi         PDBG("Ignoring unconnected lun (0x%hhX, 0x%hhX)",
36349cc0340SFelipe Franciosi              req->lun[1], req->lun[3]);
36449cc0340SFelipe Franciosi         rsp->status = SCSI_STATUS_CHECK_CONDITION;
36549cc0340SFelipe Franciosi         memset(rsp->sense, 0, sizeof(rsp->sense));
36649cc0340SFelipe Franciosi         rsp->sense_len = 18;
36749cc0340SFelipe Franciosi         rsp->sense[0] = 0x70;
36849cc0340SFelipe Franciosi         rsp->sense[2] = SCSI_SENSE_ILLEGAL_REQUEST;
36949cc0340SFelipe Franciosi         rsp->sense[7] = 10;
37049cc0340SFelipe Franciosi         rsp->sense[12] = 0x24;
37149cc0340SFelipe Franciosi 
37249cc0340SFelipe Franciosi         return 0;
37349cc0340SFelipe Franciosi     }
37449cc0340SFelipe Franciosi 
37549cc0340SFelipe Franciosi     cdb_len = get_cdb_len(req->cdb);
37649cc0340SFelipe Franciosi     if (cdb_len == -1) {
37749cc0340SFelipe Franciosi         return -1;
37849cc0340SFelipe Franciosi     }
37949cc0340SFelipe Franciosi 
38049cc0340SFelipe Franciosi     len = 0;
38149cc0340SFelipe Franciosi     if (!out_len && !in_len) {
38249cc0340SFelipe Franciosi         dir = SCSI_XFER_NONE;
38349cc0340SFelipe Franciosi     } else if (out_len) {
38449cc0340SFelipe Franciosi         dir = SCSI_XFER_TO_DEV;
38549cc0340SFelipe Franciosi         for (i = 0; i < out_len; i++) {
38649cc0340SFelipe Franciosi             len += out[i].iov_len;
38749cc0340SFelipe Franciosi         }
38849cc0340SFelipe Franciosi     } else {
38949cc0340SFelipe Franciosi         dir = SCSI_XFER_FROM_DEV;
39049cc0340SFelipe Franciosi         for (i = 0; i < in_len; i++) {
39149cc0340SFelipe Franciosi             len += in[i].iov_len;
39249cc0340SFelipe Franciosi         }
39349cc0340SFelipe Franciosi     }
39449cc0340SFelipe Franciosi 
39549cc0340SFelipe Franciosi     task = scsi_task_new(cdb_len, req->cdb, dir, len);
39649cc0340SFelipe Franciosi     if (!task) {
39749cc0340SFelipe Franciosi         PERR("Unable to create iscsi task");
39849cc0340SFelipe Franciosi         return -1;
39949cc0340SFelipe Franciosi     }
40049cc0340SFelipe Franciosi 
40149cc0340SFelipe Franciosi     if (dir == SCSI_XFER_TO_DEV) {
40249cc0340SFelipe Franciosi         task->iovector_out.iov = (struct scsi_iovec *)out;
40349cc0340SFelipe Franciosi         task->iovector_out.niov = out_len;
40449cc0340SFelipe Franciosi     } else if (dir == SCSI_XFER_FROM_DEV) {
40549cc0340SFelipe Franciosi         task->iovector_in.iov = (struct scsi_iovec *)in;
40649cc0340SFelipe Franciosi         task->iovector_in.niov = in_len;
40749cc0340SFelipe Franciosi     }
40849cc0340SFelipe Franciosi 
40949cc0340SFelipe Franciosi     PDBG("Sending iscsi cmd (cdb_len=%d, dir=%d, task=%p)",
41049cc0340SFelipe Franciosi          cdb_len, dir, task);
41149cc0340SFelipe Franciosi     if (!iscsi_scsi_command_sync(ctx, 0, task, NULL)) {
41249cc0340SFelipe Franciosi         PERR("Error serving SCSI command");
41349cc0340SFelipe Franciosi         free(task);
41449cc0340SFelipe Franciosi         return -1;
41549cc0340SFelipe Franciosi     }
41649cc0340SFelipe Franciosi 
41749cc0340SFelipe Franciosi     memset(rsp, 0, sizeof(*rsp));
41849cc0340SFelipe Franciosi 
41949cc0340SFelipe Franciosi     rsp->status = task->status;
42049cc0340SFelipe Franciosi     rsp->resid  = task->residual;
42149cc0340SFelipe Franciosi 
42249cc0340SFelipe Franciosi     if (task->status == SCSI_STATUS_CHECK_CONDITION) {
42349cc0340SFelipe Franciosi         rsp->response = VIRTIO_SCSI_S_FAILURE;
42449cc0340SFelipe Franciosi         rsp->sense_len = task->datain.size - 2;
42549cc0340SFelipe Franciosi         memcpy(rsp->sense, &task->datain.data[2], rsp->sense_len);
42649cc0340SFelipe Franciosi     }
42749cc0340SFelipe Franciosi 
42849cc0340SFelipe Franciosi     free(task);
42949cc0340SFelipe Franciosi 
43049cc0340SFelipe Franciosi     PDBG("Filled in rsp: status=%hhX, resid=%u, response=%hhX, sense_len=%u",
43149cc0340SFelipe Franciosi          rsp->status, rsp->resid, rsp->response, rsp->sense_len);
43249cc0340SFelipe Franciosi 
43349cc0340SFelipe Franciosi     return 0;
43449cc0340SFelipe Franciosi }
43549cc0340SFelipe Franciosi 
43649cc0340SFelipe Franciosi /** libvhost-user callbacks **/
43749cc0340SFelipe Franciosi 
43849cc0340SFelipe Franciosi static vhost_scsi_dev_t *vdev_scsi_find_by_vu(VuDev *vu_dev);
43949cc0340SFelipe Franciosi 
44049cc0340SFelipe Franciosi static void vus_panic_cb(VuDev *vu_dev, const char *buf)
44149cc0340SFelipe Franciosi {
44249cc0340SFelipe Franciosi     vhost_scsi_dev_t *vdev_scsi;
44349cc0340SFelipe Franciosi 
44449cc0340SFelipe Franciosi     assert(vu_dev);
44549cc0340SFelipe Franciosi 
44649cc0340SFelipe Franciosi     vdev_scsi = vdev_scsi_find_by_vu(vu_dev);
44749cc0340SFelipe Franciosi 
44849cc0340SFelipe Franciosi     if (buf) {
44949cc0340SFelipe Franciosi         PERR("vu_panic: %s", buf);
45049cc0340SFelipe Franciosi     }
45149cc0340SFelipe Franciosi 
45249cc0340SFelipe Franciosi     if (vdev_scsi) {
45349cc0340SFelipe Franciosi         assert(vdev_scsi->loop);
45449cc0340SFelipe Franciosi         g_main_loop_quit(vdev_scsi->loop);
45549cc0340SFelipe Franciosi     }
45649cc0340SFelipe Franciosi }
45749cc0340SFelipe Franciosi 
45849cc0340SFelipe Franciosi static void vus_add_watch_cb(VuDev *vu_dev, int fd, int vu_evt, vu_watch_cb cb,
459*8c36ed6eSMarc-André Lureau                              void *pvt)
460*8c36ed6eSMarc-André Lureau {
46149cc0340SFelipe Franciosi     vhost_scsi_dev_t *vdev_scsi;
46249cc0340SFelipe Franciosi     guint id;
46349cc0340SFelipe Franciosi 
46449cc0340SFelipe Franciosi     assert(vu_dev);
46549cc0340SFelipe Franciosi     assert(fd >= 0);
46649cc0340SFelipe Franciosi     assert(cb);
46749cc0340SFelipe Franciosi 
46849cc0340SFelipe Franciosi     vdev_scsi = vdev_scsi_find_by_vu(vu_dev);
46949cc0340SFelipe Franciosi     if (!vdev_scsi) {
47049cc0340SFelipe Franciosi         vus_panic_cb(vu_dev, NULL);
47149cc0340SFelipe Franciosi         return;
47249cc0340SFelipe Franciosi     }
47349cc0340SFelipe Franciosi 
47449cc0340SFelipe Franciosi     id = (guint)(uintptr_t)g_tree_lookup(vdev_scsi->fdmap,
47549cc0340SFelipe Franciosi                                          (gpointer)(uintptr_t)fd);
47649cc0340SFelipe Franciosi     if (id) {
47749cc0340SFelipe Franciosi         GSource *vus_src = g_main_context_find_source_by_id(NULL, id);
47849cc0340SFelipe Franciosi         assert(vus_src);
47949cc0340SFelipe Franciosi         g_source_destroy(vus_src);
48049cc0340SFelipe Franciosi         (void)g_tree_remove(vdev_scsi->fdmap, (gpointer)(uintptr_t)fd);
48149cc0340SFelipe Franciosi     }
48249cc0340SFelipe Franciosi 
48349cc0340SFelipe Franciosi     if (vus_gsrc_new(vdev_scsi, fd, vu_evt, cb, NULL, pvt)) {
48449cc0340SFelipe Franciosi         vus_panic_cb(vu_dev, NULL);
48549cc0340SFelipe Franciosi     }
48649cc0340SFelipe Franciosi }
48749cc0340SFelipe Franciosi 
48849cc0340SFelipe Franciosi static void vus_del_watch_cb(VuDev *vu_dev, int fd)
48949cc0340SFelipe Franciosi {
49049cc0340SFelipe Franciosi     vhost_scsi_dev_t *vdev_scsi;
49149cc0340SFelipe Franciosi     guint id;
49249cc0340SFelipe Franciosi 
49349cc0340SFelipe Franciosi     assert(vu_dev);
49449cc0340SFelipe Franciosi     assert(fd >= 0);
49549cc0340SFelipe Franciosi 
49649cc0340SFelipe Franciosi     vdev_scsi = vdev_scsi_find_by_vu(vu_dev);
49749cc0340SFelipe Franciosi     if (!vdev_scsi) {
49849cc0340SFelipe Franciosi         vus_panic_cb(vu_dev, NULL);
49949cc0340SFelipe Franciosi         return;
50049cc0340SFelipe Franciosi     }
50149cc0340SFelipe Franciosi 
50249cc0340SFelipe Franciosi     id = (guint)(uintptr_t)g_tree_lookup(vdev_scsi->fdmap,
50349cc0340SFelipe Franciosi                                          (gpointer)(uintptr_t)fd);
50449cc0340SFelipe Franciosi     if (id) {
50549cc0340SFelipe Franciosi         GSource *vus_src = g_main_context_find_source_by_id(NULL, id);
50649cc0340SFelipe Franciosi         assert(vus_src);
50749cc0340SFelipe Franciosi         g_source_destroy(vus_src);
50849cc0340SFelipe Franciosi         (void)g_tree_remove(vdev_scsi->fdmap, (gpointer)(uintptr_t)fd);
50949cc0340SFelipe Franciosi     }
51049cc0340SFelipe Franciosi }
51149cc0340SFelipe Franciosi 
51249cc0340SFelipe Franciosi static void vus_proc_ctl(VuDev *vu_dev, int idx)
51349cc0340SFelipe Franciosi {
51449cc0340SFelipe Franciosi     /* Control VQ not implemented */
51549cc0340SFelipe Franciosi }
51649cc0340SFelipe Franciosi 
51749cc0340SFelipe Franciosi static void vus_proc_evt(VuDev *vu_dev, int idx)
51849cc0340SFelipe Franciosi {
51949cc0340SFelipe Franciosi     /* Event VQ not implemented */
52049cc0340SFelipe Franciosi }
52149cc0340SFelipe Franciosi 
52249cc0340SFelipe Franciosi static void vus_proc_req(VuDev *vu_dev, int idx)
52349cc0340SFelipe Franciosi {
52449cc0340SFelipe Franciosi     vhost_scsi_dev_t *vdev_scsi;
52549cc0340SFelipe Franciosi     VuVirtq *vq;
52649cc0340SFelipe Franciosi 
52749cc0340SFelipe Franciosi     assert(vu_dev);
52849cc0340SFelipe Franciosi 
52949cc0340SFelipe Franciosi     vdev_scsi = vdev_scsi_find_by_vu(vu_dev);
53049cc0340SFelipe Franciosi     if (!vdev_scsi) {
53149cc0340SFelipe Franciosi         vus_panic_cb(vu_dev, NULL);
53249cc0340SFelipe Franciosi         return;
53349cc0340SFelipe Franciosi     }
53449cc0340SFelipe Franciosi 
535*8c36ed6eSMarc-André Lureau     if (idx < 0 || idx >= VHOST_MAX_NR_VIRTQUEUE) {
53649cc0340SFelipe Franciosi         PERR("VQ Index out of range: %d", idx);
53749cc0340SFelipe Franciosi         vus_panic_cb(vu_dev, NULL);
53849cc0340SFelipe Franciosi         return;
53949cc0340SFelipe Franciosi     }
54049cc0340SFelipe Franciosi 
54149cc0340SFelipe Franciosi     vq = vu_get_queue(vu_dev, idx);
54249cc0340SFelipe Franciosi     if (!vq) {
54349cc0340SFelipe Franciosi         PERR("Error fetching VQ (dev=%p, idx=%d)", vu_dev, idx);
54449cc0340SFelipe Franciosi         vus_panic_cb(vu_dev, NULL);
54549cc0340SFelipe Franciosi         return;
54649cc0340SFelipe Franciosi     }
54749cc0340SFelipe Franciosi 
54849cc0340SFelipe Franciosi     PDBG("Got kicked on vq[%d]@%p", idx, vq);
54949cc0340SFelipe Franciosi 
55049cc0340SFelipe Franciosi     while (1) {
55149cc0340SFelipe Franciosi         VuVirtqElement *elem;
55249cc0340SFelipe Franciosi         VirtIOSCSICmdReq *req;
55349cc0340SFelipe Franciosi         VirtIOSCSICmdResp *rsp;
55449cc0340SFelipe Franciosi 
55549cc0340SFelipe Franciosi         elem = vu_queue_pop(vu_dev, vq, sizeof(VuVirtqElement));
55649cc0340SFelipe Franciosi         if (!elem) {
55749cc0340SFelipe Franciosi             PDBG("No more elements pending on vq[%d]@%p", idx, vq);
55849cc0340SFelipe Franciosi             break;
55949cc0340SFelipe Franciosi         }
56049cc0340SFelipe Franciosi         PDBG("Popped elem@%p", elem);
56149cc0340SFelipe Franciosi 
562*8c36ed6eSMarc-André Lureau         assert(!(elem->out_num > 1 && elem->in_num > 1));
563*8c36ed6eSMarc-André Lureau         assert(elem->out_num > 0 && elem->in_num > 0);
56449cc0340SFelipe Franciosi 
56549cc0340SFelipe Franciosi         if (elem->out_sg[0].iov_len < sizeof(VirtIOSCSICmdReq)) {
56649cc0340SFelipe Franciosi             PERR("Invalid virtio-scsi req header");
56749cc0340SFelipe Franciosi             vus_panic_cb(vu_dev, NULL);
56849cc0340SFelipe Franciosi             break;
56949cc0340SFelipe Franciosi         }
57049cc0340SFelipe Franciosi         req = (VirtIOSCSICmdReq *)elem->out_sg[0].iov_base;
57149cc0340SFelipe Franciosi 
57249cc0340SFelipe Franciosi         if (elem->in_sg[0].iov_len < sizeof(VirtIOSCSICmdResp)) {
57349cc0340SFelipe Franciosi             PERR("Invalid virtio-scsi rsp header");
57449cc0340SFelipe Franciosi             vus_panic_cb(vu_dev, NULL);
57549cc0340SFelipe Franciosi             break;
57649cc0340SFelipe Franciosi         }
57749cc0340SFelipe Franciosi         rsp = (VirtIOSCSICmdResp *)elem->in_sg[0].iov_base;
57849cc0340SFelipe Franciosi 
57949cc0340SFelipe Franciosi         if (handle_cmd_sync(vdev_scsi->luns[0].iscsi_ctx,
58049cc0340SFelipe Franciosi                             req, &elem->out_sg[1], elem->out_num - 1,
58149cc0340SFelipe Franciosi                             rsp, &elem->in_sg[1], elem->in_num - 1) != 0) {
58249cc0340SFelipe Franciosi             vus_panic_cb(vu_dev, NULL);
58349cc0340SFelipe Franciosi             break;
58449cc0340SFelipe Franciosi         }
58549cc0340SFelipe Franciosi 
58649cc0340SFelipe Franciosi         vu_queue_push(vu_dev, vq, elem, 0);
58749cc0340SFelipe Franciosi         vu_queue_notify(vu_dev, vq);
58849cc0340SFelipe Franciosi 
58949cc0340SFelipe Franciosi         free(elem);
59049cc0340SFelipe Franciosi     }
59149cc0340SFelipe Franciosi }
59249cc0340SFelipe Franciosi 
59349cc0340SFelipe Franciosi static void vus_queue_set_started(VuDev *vu_dev, int idx, bool started)
59449cc0340SFelipe Franciosi {
59549cc0340SFelipe Franciosi     VuVirtq *vq;
59649cc0340SFelipe Franciosi 
59749cc0340SFelipe Franciosi     assert(vu_dev);
59849cc0340SFelipe Franciosi 
599*8c36ed6eSMarc-André Lureau     if (idx < 0 || idx >= VHOST_MAX_NR_VIRTQUEUE) {
60049cc0340SFelipe Franciosi         PERR("VQ Index out of range: %d", idx);
60149cc0340SFelipe Franciosi         vus_panic_cb(vu_dev, NULL);
60249cc0340SFelipe Franciosi         return;
60349cc0340SFelipe Franciosi     }
60449cc0340SFelipe Franciosi 
60549cc0340SFelipe Franciosi     vq = vu_get_queue(vu_dev, idx);
60649cc0340SFelipe Franciosi 
60749cc0340SFelipe Franciosi     switch (idx) {
60849cc0340SFelipe Franciosi     case 0:
60949cc0340SFelipe Franciosi         vu_set_queue_handler(vu_dev, vq, started ? vus_proc_ctl : NULL);
61049cc0340SFelipe Franciosi         break;
61149cc0340SFelipe Franciosi     case 1:
61249cc0340SFelipe Franciosi         vu_set_queue_handler(vu_dev, vq, started ? vus_proc_evt : NULL);
61349cc0340SFelipe Franciosi         break;
61449cc0340SFelipe Franciosi     default:
61549cc0340SFelipe Franciosi         vu_set_queue_handler(vu_dev, vq, started ? vus_proc_req : NULL);
61649cc0340SFelipe Franciosi     }
61749cc0340SFelipe Franciosi }
61849cc0340SFelipe Franciosi 
61949cc0340SFelipe Franciosi static const VuDevIface vus_iface = {
62049cc0340SFelipe Franciosi     .queue_set_started = vus_queue_set_started,
62149cc0340SFelipe Franciosi };
62249cc0340SFelipe Franciosi 
62349cc0340SFelipe Franciosi static gboolean vus_vhost_cb(gpointer data)
62449cc0340SFelipe Franciosi {
62549cc0340SFelipe Franciosi     VuDev *vu_dev = (VuDev *)data;
62649cc0340SFelipe Franciosi 
62749cc0340SFelipe Franciosi     assert(vu_dev);
62849cc0340SFelipe Franciosi 
62949cc0340SFelipe Franciosi     if (!vu_dispatch(vu_dev) != 0) {
63049cc0340SFelipe Franciosi         PERR("Error processing vhost message");
63149cc0340SFelipe Franciosi         vus_panic_cb(vu_dev, NULL);
63249cc0340SFelipe Franciosi         return G_SOURCE_REMOVE;
63349cc0340SFelipe Franciosi     }
63449cc0340SFelipe Franciosi 
63549cc0340SFelipe Franciosi     return G_SOURCE_CONTINUE;
63649cc0340SFelipe Franciosi }
63749cc0340SFelipe Franciosi 
63849cc0340SFelipe Franciosi /** misc helpers **/
63949cc0340SFelipe Franciosi 
64049cc0340SFelipe Franciosi static int unix_sock_new(char *unix_fn)
64149cc0340SFelipe Franciosi {
64249cc0340SFelipe Franciosi     int sock;
64349cc0340SFelipe Franciosi     struct sockaddr_un un;
64449cc0340SFelipe Franciosi     size_t len;
64549cc0340SFelipe Franciosi 
64649cc0340SFelipe Franciosi     assert(unix_fn);
64749cc0340SFelipe Franciosi 
64849cc0340SFelipe Franciosi     sock = socket(AF_UNIX, SOCK_STREAM, 0);
64949cc0340SFelipe Franciosi     if (sock <= 0) {
65049cc0340SFelipe Franciosi         perror("socket");
65149cc0340SFelipe Franciosi         return -1;
65249cc0340SFelipe Franciosi     }
65349cc0340SFelipe Franciosi 
65449cc0340SFelipe Franciosi     un.sun_family = AF_UNIX;
65549cc0340SFelipe Franciosi     (void)snprintf(un.sun_path, sizeof(un.sun_path), "%s", unix_fn);
65649cc0340SFelipe Franciosi     len = sizeof(un.sun_family) + strlen(un.sun_path);
65749cc0340SFelipe Franciosi 
65849cc0340SFelipe Franciosi     (void)unlink(unix_fn);
65949cc0340SFelipe Franciosi     if (bind(sock, (struct sockaddr *)&un, len) < 0) {
66049cc0340SFelipe Franciosi         perror("bind");
66149cc0340SFelipe Franciosi         goto fail;
66249cc0340SFelipe Franciosi     }
66349cc0340SFelipe Franciosi 
66449cc0340SFelipe Franciosi     if (listen(sock, 1) < 0) {
66549cc0340SFelipe Franciosi         perror("listen");
66649cc0340SFelipe Franciosi         goto fail;
66749cc0340SFelipe Franciosi     }
66849cc0340SFelipe Franciosi 
66949cc0340SFelipe Franciosi     return sock;
67049cc0340SFelipe Franciosi 
67149cc0340SFelipe Franciosi fail:
67249cc0340SFelipe Franciosi     (void)close(sock);
67349cc0340SFelipe Franciosi 
67449cc0340SFelipe Franciosi     return -1;
67549cc0340SFelipe Franciosi }
67649cc0340SFelipe Franciosi 
67749cc0340SFelipe Franciosi /** vhost-user-scsi **/
67849cc0340SFelipe Franciosi 
67949cc0340SFelipe Franciosi static vhost_scsi_dev_t *vdev_scsi_find_by_vu(VuDev *vu_dev)
68049cc0340SFelipe Franciosi {
68149cc0340SFelipe Franciosi     int i;
68249cc0340SFelipe Franciosi 
68349cc0340SFelipe Franciosi     assert(vu_dev);
68449cc0340SFelipe Franciosi 
68549cc0340SFelipe Franciosi     for (i = 0; i < VUS_MAX_DEVS; i++) {
68649cc0340SFelipe Franciosi         if (&vhost_scsi_devs[i]->vu_dev == vu_dev) {
68749cc0340SFelipe Franciosi             return vhost_scsi_devs[i];
68849cc0340SFelipe Franciosi         }
68949cc0340SFelipe Franciosi     }
69049cc0340SFelipe Franciosi 
69149cc0340SFelipe Franciosi     PERR("Unknown VuDev %p", vu_dev);
69249cc0340SFelipe Franciosi     return NULL;
69349cc0340SFelipe Franciosi }
69449cc0340SFelipe Franciosi 
69549cc0340SFelipe Franciosi static void vdev_scsi_deinit(vhost_scsi_dev_t *vdev_scsi)
69649cc0340SFelipe Franciosi {
69749cc0340SFelipe Franciosi     if (!vdev_scsi) {
69849cc0340SFelipe Franciosi         return;
69949cc0340SFelipe Franciosi     }
70049cc0340SFelipe Franciosi 
70149cc0340SFelipe Franciosi     if (vdev_scsi->server_sock >= 0) {
70249cc0340SFelipe Franciosi         struct sockaddr_storage ss;
70349cc0340SFelipe Franciosi         socklen_t sslen = sizeof(ss);
70449cc0340SFelipe Franciosi 
70549cc0340SFelipe Franciosi         if (getsockname(vdev_scsi->server_sock, (struct sockaddr *)&ss,
70649cc0340SFelipe Franciosi                         &sslen) == 0) {
70749cc0340SFelipe Franciosi             struct sockaddr_un *su = (struct sockaddr_un *)&ss;
70849cc0340SFelipe Franciosi             (void)unlink(su->sun_path);
70949cc0340SFelipe Franciosi         }
71049cc0340SFelipe Franciosi 
71149cc0340SFelipe Franciosi         (void)close(vdev_scsi->server_sock);
71249cc0340SFelipe Franciosi         vdev_scsi->server_sock = -1;
71349cc0340SFelipe Franciosi     }
71449cc0340SFelipe Franciosi 
71549cc0340SFelipe Franciosi     if (vdev_scsi->loop) {
71649cc0340SFelipe Franciosi         g_main_loop_unref(vdev_scsi->loop);
71749cc0340SFelipe Franciosi         vdev_scsi->loop = NULL;
71849cc0340SFelipe Franciosi     }
71949cc0340SFelipe Franciosi }
72049cc0340SFelipe Franciosi 
7210644fe4aSMarc-André Lureau static vhost_scsi_dev_t *vdev_scsi_new(int server_sock)
72249cc0340SFelipe Franciosi {
72349cc0340SFelipe Franciosi     vhost_scsi_dev_t *vdev_scsi = NULL;
72449cc0340SFelipe Franciosi 
72549cc0340SFelipe Franciosi     vdev_scsi = calloc(1, sizeof(vhost_scsi_dev_t));
72649cc0340SFelipe Franciosi     if (!vdev_scsi) {
72749cc0340SFelipe Franciosi         PERR("calloc: %s", strerror(errno));
72849cc0340SFelipe Franciosi         return NULL;
72949cc0340SFelipe Franciosi     }
73049cc0340SFelipe Franciosi 
7310644fe4aSMarc-André Lureau     vdev_scsi->server_sock = server_sock;
73249cc0340SFelipe Franciosi     vdev_scsi->loop = g_main_loop_new(NULL, FALSE);
73349cc0340SFelipe Franciosi     if (!vdev_scsi->loop) {
73449cc0340SFelipe Franciosi         PERR("Error creating glib event loop");
73549cc0340SFelipe Franciosi         goto err;
73649cc0340SFelipe Franciosi     }
73749cc0340SFelipe Franciosi 
73849cc0340SFelipe Franciosi     vdev_scsi->fdmap = g_tree_new(vus_fdmap_compare);
73949cc0340SFelipe Franciosi     if (!vdev_scsi->fdmap) {
74049cc0340SFelipe Franciosi         PERR("Error creating glib tree for fdmap");
74149cc0340SFelipe Franciosi         goto err;
74249cc0340SFelipe Franciosi     }
74349cc0340SFelipe Franciosi 
74449cc0340SFelipe Franciosi     return vdev_scsi;
74549cc0340SFelipe Franciosi 
74649cc0340SFelipe Franciosi err:
74749cc0340SFelipe Franciosi     vdev_scsi_deinit(vdev_scsi);
74849cc0340SFelipe Franciosi     free(vdev_scsi);
74949cc0340SFelipe Franciosi 
75049cc0340SFelipe Franciosi     return NULL;
75149cc0340SFelipe Franciosi }
75249cc0340SFelipe Franciosi 
75349cc0340SFelipe Franciosi static int vdev_scsi_add_iscsi_lun(vhost_scsi_dev_t *vdev_scsi,
754*8c36ed6eSMarc-André Lureau                                    char *iscsi_uri, uint32_t lun)
755*8c36ed6eSMarc-André Lureau {
75649cc0340SFelipe Franciosi     assert(vdev_scsi);
75749cc0340SFelipe Franciosi     assert(iscsi_uri);
75849cc0340SFelipe Franciosi     assert(lun < VUS_MAX_LUNS);
75949cc0340SFelipe Franciosi 
76049cc0340SFelipe Franciosi     if (vdev_scsi->luns[lun].iscsi_ctx) {
76149cc0340SFelipe Franciosi         PERR("Lun %d already configured", lun);
76249cc0340SFelipe Franciosi         return -1;
76349cc0340SFelipe Franciosi     }
76449cc0340SFelipe Franciosi 
76549cc0340SFelipe Franciosi     if (iscsi_add_lun(&vdev_scsi->luns[lun], iscsi_uri) != 0) {
76649cc0340SFelipe Franciosi         return -1;
76749cc0340SFelipe Franciosi     }
76849cc0340SFelipe Franciosi 
76949cc0340SFelipe Franciosi     return 0;
77049cc0340SFelipe Franciosi }
77149cc0340SFelipe Franciosi 
77249cc0340SFelipe Franciosi static int vdev_scsi_run(vhost_scsi_dev_t *vdev_scsi)
77349cc0340SFelipe Franciosi {
77449cc0340SFelipe Franciosi     int cli_sock;
77549cc0340SFelipe Franciosi     int ret = 0;
77649cc0340SFelipe Franciosi 
77749cc0340SFelipe Franciosi     assert(vdev_scsi);
77849cc0340SFelipe Franciosi     assert(vdev_scsi->server_sock >= 0);
77949cc0340SFelipe Franciosi     assert(vdev_scsi->loop);
78049cc0340SFelipe Franciosi 
78149cc0340SFelipe Franciosi     cli_sock = accept(vdev_scsi->server_sock, (void *)0, (void *)0);
78249cc0340SFelipe Franciosi     if (cli_sock < 0) {
78349cc0340SFelipe Franciosi         perror("accept");
78449cc0340SFelipe Franciosi         return -1;
78549cc0340SFelipe Franciosi     }
78649cc0340SFelipe Franciosi 
78749cc0340SFelipe Franciosi     vu_init(&vdev_scsi->vu_dev,
78849cc0340SFelipe Franciosi             cli_sock,
78949cc0340SFelipe Franciosi             vus_panic_cb,
79049cc0340SFelipe Franciosi             vus_add_watch_cb,
79149cc0340SFelipe Franciosi             vus_del_watch_cb,
79249cc0340SFelipe Franciosi             &vus_iface);
79349cc0340SFelipe Franciosi 
79449cc0340SFelipe Franciosi     if (vus_gsrc_new(vdev_scsi, cli_sock, G_IO_IN, NULL, vus_vhost_cb,
79549cc0340SFelipe Franciosi                      &vdev_scsi->vu_dev)) {
79649cc0340SFelipe Franciosi         goto fail;
79749cc0340SFelipe Franciosi     }
79849cc0340SFelipe Franciosi 
79949cc0340SFelipe Franciosi     g_main_loop_run(vdev_scsi->loop);
80049cc0340SFelipe Franciosi 
80149cc0340SFelipe Franciosi out:
80249cc0340SFelipe Franciosi     vu_deinit(&vdev_scsi->vu_dev);
80349cc0340SFelipe Franciosi 
80449cc0340SFelipe Franciosi     return ret;
80549cc0340SFelipe Franciosi 
80649cc0340SFelipe Franciosi fail:
80749cc0340SFelipe Franciosi     ret = -1;
80849cc0340SFelipe Franciosi     goto out;
80949cc0340SFelipe Franciosi }
81049cc0340SFelipe Franciosi 
81149cc0340SFelipe Franciosi int main(int argc, char **argv)
81249cc0340SFelipe Franciosi {
81349cc0340SFelipe Franciosi     vhost_scsi_dev_t *vdev_scsi = NULL;
81449cc0340SFelipe Franciosi     char *unix_fn = NULL;
81549cc0340SFelipe Franciosi     char *iscsi_uri = NULL;
8160644fe4aSMarc-André Lureau     int sock, opt, err = EXIT_SUCCESS;
81749cc0340SFelipe Franciosi 
81849cc0340SFelipe Franciosi     while ((opt = getopt(argc, argv, "u:i:")) != -1) {
81949cc0340SFelipe Franciosi         switch (opt) {
82049cc0340SFelipe Franciosi         case 'h':
82149cc0340SFelipe Franciosi             goto help;
82249cc0340SFelipe Franciosi         case 'u':
823b085b050SMarc-André Lureau             unix_fn = g_strdup(optarg);
82449cc0340SFelipe Franciosi             break;
82549cc0340SFelipe Franciosi         case 'i':
826b085b050SMarc-André Lureau             iscsi_uri = g_strdup(optarg);
82749cc0340SFelipe Franciosi             break;
82849cc0340SFelipe Franciosi         default:
82949cc0340SFelipe Franciosi             goto help;
83049cc0340SFelipe Franciosi         }
83149cc0340SFelipe Franciosi     }
83249cc0340SFelipe Franciosi     if (!unix_fn || !iscsi_uri) {
83349cc0340SFelipe Franciosi         goto help;
83449cc0340SFelipe Franciosi     }
83549cc0340SFelipe Franciosi 
8360644fe4aSMarc-André Lureau     sock = unix_sock_new(unix_fn);
8370644fe4aSMarc-André Lureau     if (sock < 0) {
8380644fe4aSMarc-André Lureau         goto err;
8390644fe4aSMarc-André Lureau     }
8400644fe4aSMarc-André Lureau     vdev_scsi = vdev_scsi_new(sock);
84149cc0340SFelipe Franciosi     if (!vdev_scsi) {
84249cc0340SFelipe Franciosi         goto err;
84349cc0340SFelipe Franciosi     }
84449cc0340SFelipe Franciosi     vhost_scsi_devs[0] = vdev_scsi;
84549cc0340SFelipe Franciosi 
84649cc0340SFelipe Franciosi     if (vdev_scsi_add_iscsi_lun(vdev_scsi, iscsi_uri, 0) != 0) {
84749cc0340SFelipe Franciosi         goto err;
84849cc0340SFelipe Franciosi     }
84949cc0340SFelipe Franciosi 
85049cc0340SFelipe Franciosi     if (vdev_scsi_run(vdev_scsi) != 0) {
85149cc0340SFelipe Franciosi         goto err;
85249cc0340SFelipe Franciosi     }
85349cc0340SFelipe Franciosi 
85449cc0340SFelipe Franciosi out:
85549cc0340SFelipe Franciosi     if (vdev_scsi) {
85649cc0340SFelipe Franciosi         vdev_scsi_deinit(vdev_scsi);
85749cc0340SFelipe Franciosi         free(vdev_scsi);
85849cc0340SFelipe Franciosi     }
859b085b050SMarc-André Lureau     g_free(unix_fn);
860b085b050SMarc-André Lureau     g_free(iscsi_uri);
86149cc0340SFelipe Franciosi 
86249cc0340SFelipe Franciosi     return err;
86349cc0340SFelipe Franciosi 
86449cc0340SFelipe Franciosi err:
86549cc0340SFelipe Franciosi     err = EXIT_FAILURE;
86649cc0340SFelipe Franciosi     goto out;
86749cc0340SFelipe Franciosi 
86849cc0340SFelipe Franciosi help:
86949cc0340SFelipe Franciosi     fprintf(stderr, "Usage: %s [ -u unix_sock_path -i iscsi_uri ] | [ -h ]\n",
87049cc0340SFelipe Franciosi             argv[0]);
87149cc0340SFelipe Franciosi     fprintf(stderr, "          -u path to unix socket\n");
87249cc0340SFelipe Franciosi     fprintf(stderr, "          -i iscsi uri for lun 0\n");
87349cc0340SFelipe Franciosi     fprintf(stderr, "          -h print help and quit\n");
87449cc0340SFelipe Franciosi 
87549cc0340SFelipe Franciosi     goto err;
87649cc0340SFelipe Franciosi }
877