@@ -350,6 +350,45 @@ const char *lkl_strerror(int err);
*/
void lkl_perror(char *msg, int err);
+/**
+ * struct lkl_dev_blk_ops - block device host operations, defined in lkl_host.h.
+ */
+struct lkl_dev_blk_ops;
+
+/**
+ * lkl_disk - host disk handle
+ *
+ * @dev - a pointer to 'virtio_blk_dev' structure for this disk
+ * @fd - a POSIX file descriptor that can be used by preadv/pwritev
+ * @handle - an NT file handle that can be used by ReadFile/WriteFile
+ */
+struct lkl_disk {
+ void *dev;
+ union {
+ int fd;
+ void *handle;
+ };
+ struct lkl_dev_blk_ops *ops;
+};
+
+/**
+ * lkl_disk_add - add a new disk
+ *
+ * @disk - the host disk handle
+ * @returns a disk id (0 is valid) or a strictly negative value in case of error
+ */
+int lkl_disk_add(struct lkl_disk *disk);
+
+/**
+ * lkl_disk_remove - remove a disk
+ *
+ * This function makes a cleanup of the @disk's virtio_dev structure
+ * that was initialized by lkl_disk_add before.
+ *
+ * @disk - the host disk handle
+ */
+int lkl_disk_remove(struct lkl_disk disk);
+
#ifdef __cplusplus
}
@@ -20,6 +20,63 @@ int lkl_printf(const char *fmt, ...);
extern char lkl_virtio_devs[4096];
+#ifdef LKL_HOST_CONFIG_POSIX
+#include <sys/uio.h>
+#else
+struct iovec {
+ void *iov_base;
+ size_t iov_len;
+};
+#endif
+
+extern struct lkl_dev_blk_ops lkl_dev_blk_ops;
+
+/**
+ * struct lkl_blk_req - block device request
+ *
+ * @type: type of request
+ * @prio: priority of request - currently unused
+ * @sector: offset in units 512 bytes for read / write requests
+ * @buf: an array of buffers to be used for read / write requests
+ * @count: the number of buffers
+ */
+struct lkl_blk_req {
+#define LKL_DEV_BLK_TYPE_READ 0
+#define LKL_DEV_BLK_TYPE_WRITE 1
+#define LKL_DEV_BLK_TYPE_FLUSH 4
+#define LKL_DEV_BLK_TYPE_FLUSH_OUT 5
+ unsigned int type;
+ unsigned int prio;
+ unsigned long long sector;
+ struct iovec *buf;
+ int count;
+};
+
+/**
+ * struct lkl_dev_blk_ops - block device host operations
+ */
+struct lkl_dev_blk_ops {
+ /**
+ * @get_capacity: returns the disk capacity in bytes
+ *
+ * @disk - the disk for which the capacity is requested;
+ * @res - pointer to receive the capacity, in bytes;
+ * @returns - 0 in case of success, negative value in case of error
+ */
+ int (*get_capacity)(struct lkl_disk disk, unsigned long long *res);
+#define LKL_DEV_BLK_STATUS_OK 0
+#define LKL_DEV_BLK_STATUS_IOERR 1
+#define LKL_DEV_BLK_STATUS_UNSUP 2
+ /**
+ * @request: issue a block request
+ *
+ * @disk - the disk the request is issued to;
+ * @req - a request described by &struct lkl_blk_req
+ */
+ int (*request)(struct lkl_disk disk, struct lkl_blk_req *req);
+};
+
+
#ifdef __cplusplus
}
#endif
@@ -3,6 +3,7 @@ CFLAGS_config.o += -I$(srctree)/tools/perf/pmu-events
liblkl-y += iomem.o
liblkl-y += jmp_buf.o
liblkl-y += utils.o
+liblkl-y += virtio_blk.o
liblkl-y += virtio.o
liblkl-y += dbg.o
liblkl-y += dbg_handler.o
new file mode 100644
@@ -0,0 +1,132 @@
+// SPDX-License-Identifier: GPL-2.0
+#include <lkl_host.h>
+#include "virtio.h"
+#include "endian.h"
+
+struct virtio_blk_dev {
+ struct virtio_dev dev;
+ struct lkl_virtio_blk_config config;
+ struct lkl_dev_blk_ops *ops;
+ struct lkl_disk disk;
+};
+
+struct virtio_blk_req_trailer {
+ uint8_t status;
+};
+
+static int blk_check_features(struct virtio_dev *dev)
+{
+ if (dev->driver_features == dev->device_features)
+ return 0;
+
+ return -LKL_EINVAL;
+}
+
+static int blk_enqueue(struct virtio_dev *dev, int q, struct virtio_req *req)
+{
+ struct virtio_blk_dev *blk_dev;
+ struct lkl_virtio_blk_outhdr *h;
+ struct virtio_blk_req_trailer *t;
+ struct lkl_blk_req lkl_req;
+
+ if (req->buf_count < 3) {
+ lkl_printf("virtio_blk: no status buf\n");
+ goto out;
+ }
+
+ h = req->buf[0].iov_base;
+ t = req->buf[req->buf_count - 1].iov_base;
+ blk_dev = container_of(dev, struct virtio_blk_dev, dev);
+
+ t->status = LKL_DEV_BLK_STATUS_IOERR;
+
+ if (req->buf[0].iov_len != sizeof(*h)) {
+ lkl_printf("virtio_blk: bad header buf\n");
+ goto out;
+ }
+
+ if (req->buf[req->buf_count - 1].iov_len != sizeof(*t)) {
+ lkl_printf("virtio_blk: bad status buf\n");
+ goto out;
+ }
+
+ lkl_req.type = le32toh(h->type);
+ lkl_req.prio = le32toh(h->ioprio);
+ lkl_req.sector = le32toh(h->sector);
+ lkl_req.buf = &req->buf[1];
+ lkl_req.count = req->buf_count - 2;
+
+ t->status = blk_dev->ops->request(blk_dev->disk, &lkl_req);
+
+out:
+ virtio_req_complete(req, 0);
+ return 0;
+}
+
+static struct virtio_dev_ops blk_ops = {
+ .check_features = blk_check_features,
+ .enqueue = blk_enqueue,
+};
+
+
+int lkl_disk_add(struct lkl_disk *disk)
+{
+ struct virtio_blk_dev *dev;
+ unsigned long long capacity;
+ int ret;
+
+ dev = lkl_host_ops.mem_alloc(sizeof(*dev));
+ if (!dev)
+ return -LKL_ENOMEM;
+
+ disk->dev = dev;
+
+ dev->dev.device_id = LKL_VIRTIO_ID_BLOCK;
+ dev->dev.vendor_id = 0;
+ dev->dev.device_features = 0;
+ dev->dev.config_gen = 0;
+ dev->dev.config_data = &dev->config;
+ dev->dev.config_len = sizeof(dev->config);
+ dev->dev.ops = &blk_ops;
+ if (disk->ops)
+ dev->ops = disk->ops;
+ else
+ dev->ops = &lkl_dev_blk_ops;
+ dev->disk = *disk;
+
+ ret = dev->ops->get_capacity(*disk, &capacity);
+ if (ret) {
+ ret = -LKL_ENOMEM;
+ goto out_free;
+ }
+ dev->config.capacity = capacity / 512;
+
+ ret = virtio_dev_setup(&dev->dev, 1, 32);
+ if (ret)
+ goto out_free;
+
+ return dev->dev.virtio_mmio_id;
+
+out_free:
+ lkl_host_ops.mem_free(dev);
+
+ return ret;
+}
+
+int lkl_disk_remove(struct lkl_disk disk)
+{
+ struct virtio_blk_dev *dev;
+ int ret;
+
+ dev = (struct virtio_blk_dev *)disk.dev;
+ if (!dev)
+ return -LKL_EINVAL;
+
+ ret = virtio_dev_cleanup(&dev->dev);
+ if (ret < 0)
+ return ret;
+
+ lkl_host_ops.mem_free(dev);
+
+ return 0;
+}