X-Git-Url: https://scm.lunaixsky.com/lunaix-os.git/blobdiff_plain/df1e857ac4d1410ae2bd354e361210b842ab7bc8..1fe5f5eb5378a47bf0f3451762743c162e40faad:/lunaix-os/kernel/block/block.c?ds=sidebyside diff --git a/lunaix-os/kernel/block/block.c b/lunaix-os/kernel/block/block.c index bcbc60f..c4e29e1 100644 --- a/lunaix-os/kernel/block/block.c +++ b/lunaix-os/kernel/block/block.c @@ -1,16 +1,18 @@ -#include #include #include + +#include + #include + +#include #include #include #include +#include #include -#include - -#include - #include +#include #define BLOCK_EREAD 1 #define BLOCK_ESIG 2 @@ -24,6 +26,7 @@ LOG_MODULE("BLOCK") static struct cake_pile* lbd_pile; static struct block_dev** dev_registry; static struct twifs_node* blk_sysroot; +static struct device* blk_parent_dev; int free_slot = 0; @@ -41,6 +44,7 @@ block_init() dev_registry = vcalloc(sizeof(struct block_dev*), MAX_DEV); free_slot = 0; blk_sysroot = twifs_dir_node(NULL, "block"); + blk_parent_dev = device_addcat(NULL, "block"); } int @@ -74,9 +78,7 @@ __block_read(struct device* dev, void* buf, size_t offset, size_t len) } req = blkio_vrd(vbuf, rd_block, NULL, NULL, 0); - blkio_commit(bdev->blkio, req); - - pwait(&req->wait); + blkio_commit(bdev->blkio, req, BLKIO_WAIT); if (!(errno = req->errcode)) { memcpy(buf, head_buf + r, rd_size); @@ -98,10 +100,10 @@ int __block_write(struct device* dev, void* buf, size_t offset, size_t len) { struct block_dev* bdev = (struct block_dev*)dev->underlay; - size_t bsize = bdev->blk_size, rd_block = offset / bsize + bdev->start_lba, + size_t bsize = bdev->blk_size, wr_block = offset / bsize + bdev->start_lba, r = offset % bsize, wr_size = 0; - if (!(len = MIN(len, ((size_t)bdev->end_lba - rd_block + 1) * bsize))) { + if (!(len = MIN(len, ((size_t)bdev->end_lba - wr_block + 1) * bsize))) { return 0; } @@ -123,10 +125,8 @@ __block_write(struct device* dev, void* buf, size_t offset, size_t len) vbuf_alloc(&vbuf, buf + wr_size, llen); } - req = blkio_vwr(vbuf, rd_block, NULL, NULL, 0); - blkio_commit(bdev->blkio, req); - - pwait(&req->wait); + req = blkio_vwr(vbuf, wr_block, NULL, NULL, 0); + blkio_commit(bdev->blkio, req, BLKIO_WAIT); int errno = req->errcode; if (!errno) { @@ -144,6 +144,128 @@ __block_write(struct device* dev, void* buf, size_t offset, size_t len) return errno; } +int +__block_read_page(struct device* dev, void* buf, size_t offset) +{ + struct vecbuf* vbuf = NULL; + struct block_dev* bdev = (struct block_dev*)dev->underlay; + + u32_t lba = offset / bdev->blk_size + bdev->start_lba; + u32_t rd_lba = MIN(lba + PG_SIZE / bdev->blk_size, bdev->end_lba); + + if (rd_lba <= lba) { + return 0; + } + + rd_lba -= lba; + + vbuf_alloc(&vbuf, buf, rd_lba * bdev->blk_size); + + struct blkio_req* req = blkio_vrd(vbuf, lba, NULL, NULL, 0); + + blkio_commit(bdev->blkio, req, BLKIO_WAIT); + + int errno = req->errcode; + if (!errno) { + errno = rd_lba * bdev->blk_size; + } else { + errno = -errno; + } + + blkio_free_req(req); + vbuf_free(vbuf); + return errno; +} + +int +__block_write_page(struct device* dev, void* buf, size_t offset) +{ + struct vecbuf* vbuf = NULL; + struct block_dev* bdev = (struct block_dev*)dev->underlay; + + u32_t lba = offset / bdev->blk_size + bdev->start_lba; + u32_t rd_lba = MIN(lba + PG_SIZE / bdev->blk_size, bdev->end_lba); + + if (rd_lba <= lba) { + return 0; + } + + rd_lba -= lba; + + vbuf_alloc(&vbuf, buf, rd_lba * bdev->blk_size); + + struct blkio_req* req = blkio_vwr(vbuf, lba, NULL, NULL, 0); + + blkio_commit(bdev->blkio, req, BLKIO_WAIT); + + int errno = req->errcode; + if (!errno) { + errno = rd_lba * bdev->blk_size; + } else { + errno = -errno; + } + + blkio_free_req(req); + vbuf_free(vbuf); + return errno; +} + +int +__block_rd_lb(struct block_dev* bdev, void* buf, u64_t start, size_t count) +{ + struct vecbuf* vbuf = NULL; + vbuf_alloc(&vbuf, buf, bdev->blk_size * count); + + struct blkio_req* req = blkio_vrd(vbuf, start, NULL, NULL, 0); + blkio_commit(bdev->blkio, req, BLKIO_WAIT); + + int errno = req->errcode; + if (!errno) { + errno = count; + } else { + errno = -errno; + } + + blkio_free_req(req); + vbuf_free(vbuf); + + return errno; +} + +int +__block_wr_lb(struct block_dev* bdev, void* buf, u64_t start, size_t count) +{ + struct vecbuf* vbuf = NULL; + vbuf_alloc(&vbuf, buf, bdev->blk_size * count); + + struct blkio_req* req = blkio_vwr(vbuf, start, NULL, NULL, 0); + blkio_commit(bdev->blkio, req, BLKIO_WAIT); + + int errno = req->errcode; + if (!errno) { + errno = count; + } else { + errno = -errno; + } + + blkio_free_req(req); + vbuf_free(vbuf); + + return errno; +} + +void* +block_alloc_buf(struct block_dev* bdev) +{ + return valloc(bdev->blk_size); +} + +void +block_free_buf(struct block_dev* bdev, void* buf) +{ + return vfree(buf); +} + struct block_dev* block_alloc_dev(const char* blk_id, void* driver, req_handler ioreq_handler) { @@ -155,6 +277,8 @@ block_alloc_dev(const char* blk_id, void* driver, req_handler ioreq_handler) bdev->blkio = blkio_newctx(ioreq_handler); bdev->driver = driver; bdev->blkio->driver = driver; + bdev->ops = (struct block_dev_ops){ .block_read = __block_rd_lb, + .block_write = __block_wr_lb }; return bdev; } @@ -194,9 +318,12 @@ __block_register(struct block_dev* bdev) return 0; } - struct device* dev = device_addvol(NULL, bdev, "sd%c", 'a' + free_slot); + struct device* dev = + device_addvol(blk_parent_dev, bdev, "sd%c", 'a' + free_slot); dev->write = __block_write; + dev->write_page = __block_write_page; dev->read = __block_read; + dev->read_page = __block_read_page; bdev->dev = dev; strcpy(bdev->bdev_id, dev->name_val); @@ -217,7 +344,9 @@ blk_mount_part(struct block_dev* bdev, struct device* dev = device_addvol(NULL, pbdev, "%sp%d", bdev->bdev_id, index + 1); dev->write = __block_write; + dev->write_page = __block_write_page; dev->read = __block_read; + dev->read_page = __block_read_page; pbdev->start_lba = start_lba; pbdev->end_lba = end_lba;