lib/virtio: merge vhost_user.c and virtio_user.c into one source file
Similar with NVMe device driver, here `virtio` is a specification abstraction library, `pci` and `vhost_user` are transports layer, here we merge vhost_user.c and virtio_user.c into one new source file `virtio_vhost_user.c` so that to make code more clear. No logic change, just code movement in this commit. Change-Id: I8e3e5c477e7c45e6eeebad240b8cc3c9476b86d1 Signed-off-by: Changpeng Liu <changpeng.liu@intel.com> Reviewed-on: https://review.spdk.io/gerrit/c/spdk/spdk/+/13788 Reviewed-by: Aleksey Marchuk <alexeymar@nvidia.com> Reviewed-by: Jim Harris <james.r.harris@intel.com> Tested-by: SPDK CI Jenkins <sys_sgci@intel.com>
This commit is contained in:
parent
be1883d978
commit
4e6e7eafef
@ -10,7 +10,7 @@ SO_VER := 4
|
|||||||
SO_MINOR := 0
|
SO_MINOR := 0
|
||||||
|
|
||||||
CFLAGS += $(ENV_CFLAGS)
|
CFLAGS += $(ENV_CFLAGS)
|
||||||
C_SRCS = virtio.c virtio_user.c virtio_pci.c vhost_user.c
|
C_SRCS = virtio.c virtio_vhost_user.c virtio_pci.c
|
||||||
LIBNAME = virtio
|
LIBNAME = virtio
|
||||||
|
|
||||||
SPDK_MAP_FILE = $(abspath $(CURDIR)/spdk_virtio.map)
|
SPDK_MAP_FILE = $(abspath $(CURDIR)/spdk_virtio.map)
|
||||||
|
@ -1,461 +0,0 @@
|
|||||||
/* SPDX-License-Identifier: BSD-3-Clause
|
|
||||||
* Copyright(c) 2010-2016 Intel Corporation. All rights reserved.
|
|
||||||
* All rights reserved.
|
|
||||||
*/
|
|
||||||
|
|
||||||
#include "spdk/stdinc.h"
|
|
||||||
|
|
||||||
#include "vhost_user_internal.h"
|
|
||||||
|
|
||||||
#include "spdk/string.h"
|
|
||||||
#include "spdk_internal/vhost_user.h"
|
|
||||||
|
|
||||||
/* The version of the protocol we support */
|
|
||||||
#define VHOST_USER_VERSION 0x1
|
|
||||||
|
|
||||||
static int
|
|
||||||
vhost_user_write(int fd, void *buf, int len, int *fds, int fd_num)
|
|
||||||
{
|
|
||||||
int r;
|
|
||||||
struct msghdr msgh;
|
|
||||||
struct iovec iov;
|
|
||||||
size_t fd_size = fd_num * sizeof(int);
|
|
||||||
char control[CMSG_SPACE(fd_size)];
|
|
||||||
struct cmsghdr *cmsg;
|
|
||||||
|
|
||||||
memset(&msgh, 0, sizeof(msgh));
|
|
||||||
memset(control, 0, sizeof(control));
|
|
||||||
|
|
||||||
iov.iov_base = (uint8_t *)buf;
|
|
||||||
iov.iov_len = len;
|
|
||||||
|
|
||||||
msgh.msg_iov = &iov;
|
|
||||||
msgh.msg_iovlen = 1;
|
|
||||||
|
|
||||||
if (fds && fd_num > 0) {
|
|
||||||
msgh.msg_control = control;
|
|
||||||
msgh.msg_controllen = sizeof(control);
|
|
||||||
cmsg = CMSG_FIRSTHDR(&msgh);
|
|
||||||
cmsg->cmsg_len = CMSG_LEN(fd_size);
|
|
||||||
cmsg->cmsg_level = SOL_SOCKET;
|
|
||||||
cmsg->cmsg_type = SCM_RIGHTS;
|
|
||||||
memcpy(CMSG_DATA(cmsg), fds, fd_size);
|
|
||||||
} else {
|
|
||||||
msgh.msg_control = NULL;
|
|
||||||
msgh.msg_controllen = 0;
|
|
||||||
}
|
|
||||||
|
|
||||||
do {
|
|
||||||
r = sendmsg(fd, &msgh, 0);
|
|
||||||
} while (r < 0 && errno == EINTR);
|
|
||||||
|
|
||||||
if (r == -1) {
|
|
||||||
return -errno;
|
|
||||||
}
|
|
||||||
|
|
||||||
return 0;
|
|
||||||
}
|
|
||||||
|
|
||||||
static int
|
|
||||||
vhost_user_read(int fd, struct vhost_user_msg *msg)
|
|
||||||
{
|
|
||||||
uint32_t valid_flags = VHOST_USER_REPLY_MASK | VHOST_USER_VERSION;
|
|
||||||
ssize_t ret;
|
|
||||||
size_t sz_hdr = VHOST_USER_HDR_SIZE, sz_payload;
|
|
||||||
|
|
||||||
ret = recv(fd, (void *)msg, sz_hdr, 0);
|
|
||||||
if ((size_t)ret != sz_hdr) {
|
|
||||||
SPDK_WARNLOG("Failed to recv msg hdr: %zd instead of %zu.\n",
|
|
||||||
ret, sz_hdr);
|
|
||||||
if (ret == -1) {
|
|
||||||
return -errno;
|
|
||||||
} else {
|
|
||||||
return -EBUSY;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
/* validate msg flags */
|
|
||||||
if (msg->flags != (valid_flags)) {
|
|
||||||
SPDK_WARNLOG("Failed to recv msg: flags %"PRIx32" instead of %"PRIx32".\n",
|
|
||||||
msg->flags, valid_flags);
|
|
||||||
return -EIO;
|
|
||||||
}
|
|
||||||
|
|
||||||
sz_payload = msg->size;
|
|
||||||
|
|
||||||
if (sz_payload > VHOST_USER_PAYLOAD_SIZE) {
|
|
||||||
SPDK_WARNLOG("Received oversized msg: payload size %zu > available space %zu\n",
|
|
||||||
sz_payload, VHOST_USER_PAYLOAD_SIZE);
|
|
||||||
return -EIO;
|
|
||||||
}
|
|
||||||
|
|
||||||
if (sz_payload) {
|
|
||||||
ret = recv(fd, (void *)((char *)msg + sz_hdr), sz_payload, 0);
|
|
||||||
if ((size_t)ret != sz_payload) {
|
|
||||||
SPDK_WARNLOG("Failed to recv msg payload: %zd instead of %"PRIu32".\n",
|
|
||||||
ret, msg->size);
|
|
||||||
if (ret == -1) {
|
|
||||||
return -errno;
|
|
||||||
} else {
|
|
||||||
return -EBUSY;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
return 0;
|
|
||||||
}
|
|
||||||
|
|
||||||
struct hugepage_file_info {
|
|
||||||
uint64_t addr; /**< virtual addr */
|
|
||||||
size_t size; /**< the file size */
|
|
||||||
char path[PATH_MAX]; /**< path to backing file */
|
|
||||||
};
|
|
||||||
|
|
||||||
/* Two possible options:
|
|
||||||
* 1. Match HUGEPAGE_INFO_FMT to find the file storing struct hugepage_file
|
|
||||||
* array. This is simple but cannot be used in secondary process because
|
|
||||||
* secondary process will close and munmap that file.
|
|
||||||
* 2. Match HUGEFILE_FMT to find hugepage files directly.
|
|
||||||
*
|
|
||||||
* We choose option 2.
|
|
||||||
*/
|
|
||||||
static int
|
|
||||||
get_hugepage_file_info(struct hugepage_file_info hugepages[], int max)
|
|
||||||
{
|
|
||||||
int idx, rc;
|
|
||||||
FILE *f;
|
|
||||||
char buf[BUFSIZ], *tmp, *tail;
|
|
||||||
char *str_underline, *str_start;
|
|
||||||
int huge_index;
|
|
||||||
uint64_t v_start, v_end;
|
|
||||||
|
|
||||||
f = fopen("/proc/self/maps", "r");
|
|
||||||
if (!f) {
|
|
||||||
SPDK_ERRLOG("cannot open /proc/self/maps\n");
|
|
||||||
rc = -errno;
|
|
||||||
assert(rc < 0); /* scan-build hack */
|
|
||||||
return rc;
|
|
||||||
}
|
|
||||||
|
|
||||||
idx = 0;
|
|
||||||
while (fgets(buf, sizeof(buf), f) != NULL) {
|
|
||||||
if (sscanf(buf, "%" PRIx64 "-%" PRIx64, &v_start, &v_end) < 2) {
|
|
||||||
SPDK_ERRLOG("Failed to parse address\n");
|
|
||||||
rc = -EIO;
|
|
||||||
goto out;
|
|
||||||
}
|
|
||||||
|
|
||||||
tmp = strchr(buf, ' ') + 1; /** skip address */
|
|
||||||
tmp = strchr(tmp, ' ') + 1; /** skip perm */
|
|
||||||
tmp = strchr(tmp, ' ') + 1; /** skip offset */
|
|
||||||
tmp = strchr(tmp, ' ') + 1; /** skip dev */
|
|
||||||
tmp = strchr(tmp, ' ') + 1; /** skip inode */
|
|
||||||
while (*tmp == ' ') { /** skip spaces */
|
|
||||||
tmp++;
|
|
||||||
}
|
|
||||||
tail = strrchr(tmp, '\n'); /** remove newline if exists */
|
|
||||||
if (tail) {
|
|
||||||
*tail = '\0';
|
|
||||||
}
|
|
||||||
|
|
||||||
/* Match HUGEFILE_FMT, aka "%s/%smap_%d",
|
|
||||||
* which is defined in eal_filesystem.h
|
|
||||||
*/
|
|
||||||
str_underline = strrchr(tmp, '_');
|
|
||||||
if (!str_underline) {
|
|
||||||
continue;
|
|
||||||
}
|
|
||||||
|
|
||||||
str_start = str_underline - strlen("map");
|
|
||||||
if (str_start < tmp) {
|
|
||||||
continue;
|
|
||||||
}
|
|
||||||
|
|
||||||
if (sscanf(str_start, "map_%d", &huge_index) != 1) {
|
|
||||||
continue;
|
|
||||||
}
|
|
||||||
|
|
||||||
if (idx >= max) {
|
|
||||||
SPDK_ERRLOG("Exceed maximum of %d\n", max);
|
|
||||||
rc = -ENOSPC;
|
|
||||||
goto out;
|
|
||||||
}
|
|
||||||
|
|
||||||
if (idx > 0 &&
|
|
||||||
strncmp(tmp, hugepages[idx - 1].path, PATH_MAX) == 0 &&
|
|
||||||
v_start == hugepages[idx - 1].addr + hugepages[idx - 1].size) {
|
|
||||||
hugepages[idx - 1].size += (v_end - v_start);
|
|
||||||
continue;
|
|
||||||
}
|
|
||||||
|
|
||||||
hugepages[idx].addr = v_start;
|
|
||||||
hugepages[idx].size = v_end - v_start;
|
|
||||||
snprintf(hugepages[idx].path, PATH_MAX, "%s", tmp);
|
|
||||||
idx++;
|
|
||||||
}
|
|
||||||
|
|
||||||
rc = idx;
|
|
||||||
out:
|
|
||||||
fclose(f);
|
|
||||||
return rc;
|
|
||||||
}
|
|
||||||
|
|
||||||
static int
|
|
||||||
prepare_vhost_memory_user(struct vhost_user_msg *msg, int fds[])
|
|
||||||
{
|
|
||||||
int i, num;
|
|
||||||
struct hugepage_file_info hugepages[VHOST_USER_MEMORY_MAX_NREGIONS];
|
|
||||||
|
|
||||||
num = get_hugepage_file_info(hugepages, VHOST_USER_MEMORY_MAX_NREGIONS);
|
|
||||||
if (num < 0) {
|
|
||||||
SPDK_ERRLOG("Failed to prepare memory for vhost-user\n");
|
|
||||||
return num;
|
|
||||||
}
|
|
||||||
|
|
||||||
for (i = 0; i < num; ++i) {
|
|
||||||
/* the memory regions are unaligned */
|
|
||||||
msg->payload.memory.regions[i].guest_phys_addr = hugepages[i].addr; /* use vaddr! */
|
|
||||||
msg->payload.memory.regions[i].userspace_addr = hugepages[i].addr;
|
|
||||||
msg->payload.memory.regions[i].memory_size = hugepages[i].size;
|
|
||||||
msg->payload.memory.regions[i].flags_padding = 0;
|
|
||||||
fds[i] = open(hugepages[i].path, O_RDWR);
|
|
||||||
}
|
|
||||||
|
|
||||||
msg->payload.memory.nregions = num;
|
|
||||||
msg->payload.memory.padding = 0;
|
|
||||||
|
|
||||||
return 0;
|
|
||||||
}
|
|
||||||
|
|
||||||
static const char *const vhost_msg_strings[VHOST_USER_MAX] = {
|
|
||||||
[VHOST_USER_SET_OWNER] = "VHOST_SET_OWNER",
|
|
||||||
[VHOST_USER_RESET_OWNER] = "VHOST_RESET_OWNER",
|
|
||||||
[VHOST_USER_SET_FEATURES] = "VHOST_SET_FEATURES",
|
|
||||||
[VHOST_USER_GET_FEATURES] = "VHOST_GET_FEATURES",
|
|
||||||
[VHOST_USER_SET_VRING_CALL] = "VHOST_SET_VRING_CALL",
|
|
||||||
[VHOST_USER_GET_PROTOCOL_FEATURES] = "VHOST_USER_GET_PROTOCOL_FEATURES",
|
|
||||||
[VHOST_USER_SET_PROTOCOL_FEATURES] = "VHOST_USER_SET_PROTOCOL_FEATURES",
|
|
||||||
[VHOST_USER_SET_VRING_NUM] = "VHOST_SET_VRING_NUM",
|
|
||||||
[VHOST_USER_SET_VRING_BASE] = "VHOST_SET_VRING_BASE",
|
|
||||||
[VHOST_USER_GET_VRING_BASE] = "VHOST_GET_VRING_BASE",
|
|
||||||
[VHOST_USER_SET_VRING_ADDR] = "VHOST_SET_VRING_ADDR",
|
|
||||||
[VHOST_USER_SET_VRING_KICK] = "VHOST_SET_VRING_KICK",
|
|
||||||
[VHOST_USER_SET_MEM_TABLE] = "VHOST_SET_MEM_TABLE",
|
|
||||||
[VHOST_USER_SET_VRING_ENABLE] = "VHOST_SET_VRING_ENABLE",
|
|
||||||
[VHOST_USER_GET_QUEUE_NUM] = "VHOST_USER_GET_QUEUE_NUM",
|
|
||||||
[VHOST_USER_GET_CONFIG] = "VHOST_USER_GET_CONFIG",
|
|
||||||
[VHOST_USER_SET_CONFIG] = "VHOST_USER_SET_CONFIG",
|
|
||||||
};
|
|
||||||
|
|
||||||
static int
|
|
||||||
vhost_user_sock(struct virtio_user_dev *dev,
|
|
||||||
enum vhost_user_request req,
|
|
||||||
void *arg)
|
|
||||||
{
|
|
||||||
struct vhost_user_msg msg;
|
|
||||||
struct vhost_vring_file *file = 0;
|
|
||||||
int need_reply = 0;
|
|
||||||
int fds[VHOST_USER_MEMORY_MAX_NREGIONS];
|
|
||||||
int fd_num = 0;
|
|
||||||
int i, len, rc;
|
|
||||||
int vhostfd = dev->vhostfd;
|
|
||||||
|
|
||||||
SPDK_DEBUGLOG(virtio_user, "sent message %d = %s\n", req, vhost_msg_strings[req]);
|
|
||||||
|
|
||||||
msg.request = req;
|
|
||||||
msg.flags = VHOST_USER_VERSION;
|
|
||||||
msg.size = 0;
|
|
||||||
|
|
||||||
switch (req) {
|
|
||||||
case VHOST_USER_GET_FEATURES:
|
|
||||||
case VHOST_USER_GET_PROTOCOL_FEATURES:
|
|
||||||
case VHOST_USER_GET_QUEUE_NUM:
|
|
||||||
need_reply = 1;
|
|
||||||
break;
|
|
||||||
|
|
||||||
case VHOST_USER_SET_FEATURES:
|
|
||||||
case VHOST_USER_SET_LOG_BASE:
|
|
||||||
case VHOST_USER_SET_PROTOCOL_FEATURES:
|
|
||||||
msg.payload.u64 = *((__u64 *)arg);
|
|
||||||
msg.size = sizeof(msg.payload.u64);
|
|
||||||
break;
|
|
||||||
|
|
||||||
case VHOST_USER_SET_OWNER:
|
|
||||||
case VHOST_USER_RESET_OWNER:
|
|
||||||
break;
|
|
||||||
|
|
||||||
case VHOST_USER_SET_MEM_TABLE:
|
|
||||||
rc = prepare_vhost_memory_user(&msg, fds);
|
|
||||||
if (rc < 0) {
|
|
||||||
return rc;
|
|
||||||
}
|
|
||||||
fd_num = msg.payload.memory.nregions;
|
|
||||||
msg.size = sizeof(msg.payload.memory.nregions);
|
|
||||||
msg.size += sizeof(msg.payload.memory.padding);
|
|
||||||
msg.size += fd_num * sizeof(struct vhost_memory_region);
|
|
||||||
break;
|
|
||||||
|
|
||||||
case VHOST_USER_SET_LOG_FD:
|
|
||||||
fds[fd_num++] = *((int *)arg);
|
|
||||||
break;
|
|
||||||
|
|
||||||
case VHOST_USER_SET_VRING_NUM:
|
|
||||||
case VHOST_USER_SET_VRING_BASE:
|
|
||||||
case VHOST_USER_SET_VRING_ENABLE:
|
|
||||||
memcpy(&msg.payload.state, arg, sizeof(msg.payload.state));
|
|
||||||
msg.size = sizeof(msg.payload.state);
|
|
||||||
break;
|
|
||||||
|
|
||||||
case VHOST_USER_GET_VRING_BASE:
|
|
||||||
memcpy(&msg.payload.state, arg, sizeof(msg.payload.state));
|
|
||||||
msg.size = sizeof(msg.payload.state);
|
|
||||||
need_reply = 1;
|
|
||||||
break;
|
|
||||||
|
|
||||||
case VHOST_USER_SET_VRING_ADDR:
|
|
||||||
memcpy(&msg.payload.addr, arg, sizeof(msg.payload.addr));
|
|
||||||
msg.size = sizeof(msg.payload.addr);
|
|
||||||
break;
|
|
||||||
|
|
||||||
case VHOST_USER_SET_VRING_KICK:
|
|
||||||
case VHOST_USER_SET_VRING_CALL:
|
|
||||||
case VHOST_USER_SET_VRING_ERR:
|
|
||||||
file = arg;
|
|
||||||
msg.payload.u64 = file->index & VHOST_USER_VRING_IDX_MASK;
|
|
||||||
msg.size = sizeof(msg.payload.u64);
|
|
||||||
if (file->fd > 0) {
|
|
||||||
fds[fd_num++] = file->fd;
|
|
||||||
} else {
|
|
||||||
msg.payload.u64 |= VHOST_USER_VRING_NOFD_MASK;
|
|
||||||
}
|
|
||||||
break;
|
|
||||||
|
|
||||||
case VHOST_USER_GET_CONFIG:
|
|
||||||
memcpy(&msg.payload.cfg, arg, sizeof(msg.payload.cfg));
|
|
||||||
msg.size = sizeof(msg.payload.cfg);
|
|
||||||
need_reply = 1;
|
|
||||||
break;
|
|
||||||
|
|
||||||
case VHOST_USER_SET_CONFIG:
|
|
||||||
memcpy(&msg.payload.cfg, arg, sizeof(msg.payload.cfg));
|
|
||||||
msg.size = sizeof(msg.payload.cfg);
|
|
||||||
break;
|
|
||||||
|
|
||||||
default:
|
|
||||||
SPDK_ERRLOG("trying to send unknown msg\n");
|
|
||||||
return -EINVAL;
|
|
||||||
}
|
|
||||||
|
|
||||||
len = VHOST_USER_HDR_SIZE + msg.size;
|
|
||||||
rc = vhost_user_write(vhostfd, &msg, len, fds, fd_num);
|
|
||||||
if (rc < 0) {
|
|
||||||
SPDK_ERRLOG("%s failed: %s\n",
|
|
||||||
vhost_msg_strings[req], spdk_strerror(-rc));
|
|
||||||
return rc;
|
|
||||||
}
|
|
||||||
|
|
||||||
if (req == VHOST_USER_SET_MEM_TABLE)
|
|
||||||
for (i = 0; i < fd_num; ++i) {
|
|
||||||
close(fds[i]);
|
|
||||||
}
|
|
||||||
|
|
||||||
if (need_reply) {
|
|
||||||
rc = vhost_user_read(vhostfd, &msg);
|
|
||||||
if (rc < 0) {
|
|
||||||
SPDK_WARNLOG("Received msg failed: %s\n", spdk_strerror(-rc));
|
|
||||||
return rc;
|
|
||||||
}
|
|
||||||
|
|
||||||
if (req != msg.request) {
|
|
||||||
SPDK_WARNLOG("Received unexpected msg type\n");
|
|
||||||
return -EIO;
|
|
||||||
}
|
|
||||||
|
|
||||||
switch (req) {
|
|
||||||
case VHOST_USER_GET_FEATURES:
|
|
||||||
case VHOST_USER_GET_PROTOCOL_FEATURES:
|
|
||||||
case VHOST_USER_GET_QUEUE_NUM:
|
|
||||||
if (msg.size != sizeof(msg.payload.u64)) {
|
|
||||||
SPDK_WARNLOG("Received bad msg size\n");
|
|
||||||
return -EIO;
|
|
||||||
}
|
|
||||||
*((__u64 *)arg) = msg.payload.u64;
|
|
||||||
break;
|
|
||||||
case VHOST_USER_GET_VRING_BASE:
|
|
||||||
if (msg.size != sizeof(msg.payload.state)) {
|
|
||||||
SPDK_WARNLOG("Received bad msg size\n");
|
|
||||||
return -EIO;
|
|
||||||
}
|
|
||||||
memcpy(arg, &msg.payload.state,
|
|
||||||
sizeof(struct vhost_vring_state));
|
|
||||||
break;
|
|
||||||
case VHOST_USER_GET_CONFIG:
|
|
||||||
if (msg.size != sizeof(msg.payload.cfg)) {
|
|
||||||
SPDK_WARNLOG("Received bad msg size\n");
|
|
||||||
return -EIO;
|
|
||||||
}
|
|
||||||
memcpy(arg, &msg.payload.cfg, sizeof(msg.payload.cfg));
|
|
||||||
break;
|
|
||||||
default:
|
|
||||||
SPDK_WARNLOG("Received unexpected msg type\n");
|
|
||||||
return -EBADMSG;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
return 0;
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Set up environment to talk with a vhost user backend.
|
|
||||||
*
|
|
||||||
* @return
|
|
||||||
* - (-1) if fail;
|
|
||||||
* - (0) if succeed.
|
|
||||||
*/
|
|
||||||
static int
|
|
||||||
vhost_user_setup(struct virtio_user_dev *dev)
|
|
||||||
{
|
|
||||||
int fd;
|
|
||||||
int flag;
|
|
||||||
struct sockaddr_un un;
|
|
||||||
ssize_t rc;
|
|
||||||
|
|
||||||
fd = socket(AF_UNIX, SOCK_STREAM, 0);
|
|
||||||
if (fd < 0) {
|
|
||||||
SPDK_ERRLOG("socket() error, %s\n", spdk_strerror(errno));
|
|
||||||
return -errno;
|
|
||||||
}
|
|
||||||
|
|
||||||
flag = fcntl(fd, F_GETFD);
|
|
||||||
if (fcntl(fd, F_SETFD, flag | FD_CLOEXEC) < 0) {
|
|
||||||
SPDK_ERRLOG("fcntl failed, %s\n", spdk_strerror(errno));
|
|
||||||
}
|
|
||||||
|
|
||||||
memset(&un, 0, sizeof(un));
|
|
||||||
un.sun_family = AF_UNIX;
|
|
||||||
rc = snprintf(un.sun_path, sizeof(un.sun_path), "%s", dev->path);
|
|
||||||
if (rc < 0 || (size_t)rc >= sizeof(un.sun_path)) {
|
|
||||||
SPDK_ERRLOG("socket path too long\n");
|
|
||||||
close(fd);
|
|
||||||
if (rc < 0) {
|
|
||||||
return -errno;
|
|
||||||
} else {
|
|
||||||
return -EINVAL;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
if (connect(fd, (struct sockaddr *)&un, sizeof(un)) < 0) {
|
|
||||||
SPDK_ERRLOG("connect error, %s\n", spdk_strerror(errno));
|
|
||||||
close(fd);
|
|
||||||
return -errno;
|
|
||||||
}
|
|
||||||
|
|
||||||
dev->vhostfd = fd;
|
|
||||||
return 0;
|
|
||||||
}
|
|
||||||
|
|
||||||
struct virtio_user_backend_ops ops_user = {
|
|
||||||
.setup = vhost_user_setup,
|
|
||||||
.send_request = vhost_user_sock,
|
|
||||||
};
|
|
||||||
|
|
||||||
SPDK_LOG_REGISTER_COMPONENT(virtio_user)
|
|
@ -1,41 +0,0 @@
|
|||||||
/* SPDX-License-Identifier: BSD-3-Clause
|
|
||||||
* Copyright(c) 2010-2016 Intel Corporation. All rights reserved.
|
|
||||||
* All rights reserved.
|
|
||||||
*/
|
|
||||||
|
|
||||||
#ifndef _VHOST_H
|
|
||||||
#define _VHOST_H
|
|
||||||
|
|
||||||
#include "spdk/stdinc.h"
|
|
||||||
|
|
||||||
#include "spdk/log.h"
|
|
||||||
#include "spdk_internal/virtio.h"
|
|
||||||
#include "spdk_internal/vhost_user.h"
|
|
||||||
|
|
||||||
struct virtio_user_backend_ops;
|
|
||||||
|
|
||||||
struct virtio_user_dev {
|
|
||||||
int vhostfd;
|
|
||||||
|
|
||||||
int callfds[SPDK_VIRTIO_MAX_VIRTQUEUES];
|
|
||||||
int kickfds[SPDK_VIRTIO_MAX_VIRTQUEUES];
|
|
||||||
uint32_t queue_size;
|
|
||||||
|
|
||||||
uint8_t status;
|
|
||||||
char path[PATH_MAX];
|
|
||||||
uint64_t protocol_features;
|
|
||||||
struct vring vrings[SPDK_VIRTIO_MAX_VIRTQUEUES];
|
|
||||||
struct virtio_user_backend_ops *ops;
|
|
||||||
struct spdk_mem_map *mem_map;
|
|
||||||
};
|
|
||||||
|
|
||||||
struct virtio_user_backend_ops {
|
|
||||||
int (*setup)(struct virtio_user_dev *dev);
|
|
||||||
int (*send_request)(struct virtio_user_dev *dev,
|
|
||||||
enum vhost_user_request req,
|
|
||||||
void *arg);
|
|
||||||
};
|
|
||||||
|
|
||||||
extern struct virtio_user_backend_ops ops_user;
|
|
||||||
|
|
||||||
#endif
|
|
@ -7,17 +7,489 @@
|
|||||||
|
|
||||||
#include <sys/eventfd.h>
|
#include <sys/eventfd.h>
|
||||||
|
|
||||||
#include "vhost_user_internal.h"
|
|
||||||
#include "spdk/string.h"
|
#include "spdk/string.h"
|
||||||
#include "spdk/config.h"
|
#include "spdk/config.h"
|
||||||
#include "spdk/util.h"
|
#include "spdk/util.h"
|
||||||
|
|
||||||
#include "spdk_internal/virtio.h"
|
#include "spdk_internal/virtio.h"
|
||||||
|
#include "spdk_internal/vhost_user.h"
|
||||||
|
|
||||||
|
/* The version of the protocol we support */
|
||||||
|
#define VHOST_USER_VERSION 0x1
|
||||||
|
|
||||||
#define VIRTIO_USER_SUPPORTED_PROTOCOL_FEATURES \
|
#define VIRTIO_USER_SUPPORTED_PROTOCOL_FEATURES \
|
||||||
((1ULL << VHOST_USER_PROTOCOL_F_MQ) | \
|
((1ULL << VHOST_USER_PROTOCOL_F_MQ) | \
|
||||||
(1ULL << VHOST_USER_PROTOCOL_F_CONFIG))
|
(1ULL << VHOST_USER_PROTOCOL_F_CONFIG))
|
||||||
|
|
||||||
|
struct virtio_user_backend_ops;
|
||||||
|
|
||||||
|
struct virtio_user_dev {
|
||||||
|
int vhostfd;
|
||||||
|
|
||||||
|
int callfds[SPDK_VIRTIO_MAX_VIRTQUEUES];
|
||||||
|
int kickfds[SPDK_VIRTIO_MAX_VIRTQUEUES];
|
||||||
|
uint32_t queue_size;
|
||||||
|
|
||||||
|
uint8_t status;
|
||||||
|
char path[PATH_MAX];
|
||||||
|
uint64_t protocol_features;
|
||||||
|
struct vring vrings[SPDK_VIRTIO_MAX_VIRTQUEUES];
|
||||||
|
struct virtio_user_backend_ops *ops;
|
||||||
|
struct spdk_mem_map *mem_map;
|
||||||
|
};
|
||||||
|
|
||||||
|
struct virtio_user_backend_ops {
|
||||||
|
int (*setup)(struct virtio_user_dev *dev);
|
||||||
|
int (*send_request)(struct virtio_user_dev *dev,
|
||||||
|
enum vhost_user_request req,
|
||||||
|
void *arg);
|
||||||
|
};
|
||||||
|
|
||||||
|
static int
|
||||||
|
vhost_user_write(int fd, void *buf, int len, int *fds, int fd_num)
|
||||||
|
{
|
||||||
|
int r;
|
||||||
|
struct msghdr msgh;
|
||||||
|
struct iovec iov;
|
||||||
|
size_t fd_size = fd_num * sizeof(int);
|
||||||
|
char control[CMSG_SPACE(fd_size)];
|
||||||
|
struct cmsghdr *cmsg;
|
||||||
|
|
||||||
|
memset(&msgh, 0, sizeof(msgh));
|
||||||
|
memset(control, 0, sizeof(control));
|
||||||
|
|
||||||
|
iov.iov_base = (uint8_t *)buf;
|
||||||
|
iov.iov_len = len;
|
||||||
|
|
||||||
|
msgh.msg_iov = &iov;
|
||||||
|
msgh.msg_iovlen = 1;
|
||||||
|
|
||||||
|
if (fds && fd_num > 0) {
|
||||||
|
msgh.msg_control = control;
|
||||||
|
msgh.msg_controllen = sizeof(control);
|
||||||
|
cmsg = CMSG_FIRSTHDR(&msgh);
|
||||||
|
cmsg->cmsg_len = CMSG_LEN(fd_size);
|
||||||
|
cmsg->cmsg_level = SOL_SOCKET;
|
||||||
|
cmsg->cmsg_type = SCM_RIGHTS;
|
||||||
|
memcpy(CMSG_DATA(cmsg), fds, fd_size);
|
||||||
|
} else {
|
||||||
|
msgh.msg_control = NULL;
|
||||||
|
msgh.msg_controllen = 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
do {
|
||||||
|
r = sendmsg(fd, &msgh, 0);
|
||||||
|
} while (r < 0 && errno == EINTR);
|
||||||
|
|
||||||
|
if (r == -1) {
|
||||||
|
return -errno;
|
||||||
|
}
|
||||||
|
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
static int
|
||||||
|
vhost_user_read(int fd, struct vhost_user_msg *msg)
|
||||||
|
{
|
||||||
|
uint32_t valid_flags = VHOST_USER_REPLY_MASK | VHOST_USER_VERSION;
|
||||||
|
ssize_t ret;
|
||||||
|
size_t sz_hdr = VHOST_USER_HDR_SIZE, sz_payload;
|
||||||
|
|
||||||
|
ret = recv(fd, (void *)msg, sz_hdr, 0);
|
||||||
|
if ((size_t)ret != sz_hdr) {
|
||||||
|
SPDK_WARNLOG("Failed to recv msg hdr: %zd instead of %zu.\n",
|
||||||
|
ret, sz_hdr);
|
||||||
|
if (ret == -1) {
|
||||||
|
return -errno;
|
||||||
|
} else {
|
||||||
|
return -EBUSY;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/* validate msg flags */
|
||||||
|
if (msg->flags != (valid_flags)) {
|
||||||
|
SPDK_WARNLOG("Failed to recv msg: flags %"PRIx32" instead of %"PRIx32".\n",
|
||||||
|
msg->flags, valid_flags);
|
||||||
|
return -EIO;
|
||||||
|
}
|
||||||
|
|
||||||
|
sz_payload = msg->size;
|
||||||
|
|
||||||
|
if (sz_payload > VHOST_USER_PAYLOAD_SIZE) {
|
||||||
|
SPDK_WARNLOG("Received oversized msg: payload size %zu > available space %zu\n",
|
||||||
|
sz_payload, VHOST_USER_PAYLOAD_SIZE);
|
||||||
|
return -EIO;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (sz_payload) {
|
||||||
|
ret = recv(fd, (void *)((char *)msg + sz_hdr), sz_payload, 0);
|
||||||
|
if ((size_t)ret != sz_payload) {
|
||||||
|
SPDK_WARNLOG("Failed to recv msg payload: %zd instead of %"PRIu32".\n",
|
||||||
|
ret, msg->size);
|
||||||
|
if (ret == -1) {
|
||||||
|
return -errno;
|
||||||
|
} else {
|
||||||
|
return -EBUSY;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
struct hugepage_file_info {
|
||||||
|
uint64_t addr; /**< virtual addr */
|
||||||
|
size_t size; /**< the file size */
|
||||||
|
char path[PATH_MAX]; /**< path to backing file */
|
||||||
|
};
|
||||||
|
|
||||||
|
/* Two possible options:
|
||||||
|
* 1. Match HUGEPAGE_INFO_FMT to find the file storing struct hugepage_file
|
||||||
|
* array. This is simple but cannot be used in secondary process because
|
||||||
|
* secondary process will close and munmap that file.
|
||||||
|
* 2. Match HUGEFILE_FMT to find hugepage files directly.
|
||||||
|
*
|
||||||
|
* We choose option 2.
|
||||||
|
*/
|
||||||
|
static int
|
||||||
|
get_hugepage_file_info(struct hugepage_file_info hugepages[], int max)
|
||||||
|
{
|
||||||
|
int idx, rc;
|
||||||
|
FILE *f;
|
||||||
|
char buf[BUFSIZ], *tmp, *tail;
|
||||||
|
char *str_underline, *str_start;
|
||||||
|
int huge_index;
|
||||||
|
uint64_t v_start, v_end;
|
||||||
|
|
||||||
|
f = fopen("/proc/self/maps", "r");
|
||||||
|
if (!f) {
|
||||||
|
SPDK_ERRLOG("cannot open /proc/self/maps\n");
|
||||||
|
rc = -errno;
|
||||||
|
assert(rc < 0); /* scan-build hack */
|
||||||
|
return rc;
|
||||||
|
}
|
||||||
|
|
||||||
|
idx = 0;
|
||||||
|
while (fgets(buf, sizeof(buf), f) != NULL) {
|
||||||
|
if (sscanf(buf, "%" PRIx64 "-%" PRIx64, &v_start, &v_end) < 2) {
|
||||||
|
SPDK_ERRLOG("Failed to parse address\n");
|
||||||
|
rc = -EIO;
|
||||||
|
goto out;
|
||||||
|
}
|
||||||
|
|
||||||
|
tmp = strchr(buf, ' ') + 1; /** skip address */
|
||||||
|
tmp = strchr(tmp, ' ') + 1; /** skip perm */
|
||||||
|
tmp = strchr(tmp, ' ') + 1; /** skip offset */
|
||||||
|
tmp = strchr(tmp, ' ') + 1; /** skip dev */
|
||||||
|
tmp = strchr(tmp, ' ') + 1; /** skip inode */
|
||||||
|
while (*tmp == ' ') { /** skip spaces */
|
||||||
|
tmp++;
|
||||||
|
}
|
||||||
|
tail = strrchr(tmp, '\n'); /** remove newline if exists */
|
||||||
|
if (tail) {
|
||||||
|
*tail = '\0';
|
||||||
|
}
|
||||||
|
|
||||||
|
/* Match HUGEFILE_FMT, aka "%s/%smap_%d",
|
||||||
|
* which is defined in eal_filesystem.h
|
||||||
|
*/
|
||||||
|
str_underline = strrchr(tmp, '_');
|
||||||
|
if (!str_underline) {
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
|
||||||
|
str_start = str_underline - strlen("map");
|
||||||
|
if (str_start < tmp) {
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (sscanf(str_start, "map_%d", &huge_index) != 1) {
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (idx >= max) {
|
||||||
|
SPDK_ERRLOG("Exceed maximum of %d\n", max);
|
||||||
|
rc = -ENOSPC;
|
||||||
|
goto out;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (idx > 0 &&
|
||||||
|
strncmp(tmp, hugepages[idx - 1].path, PATH_MAX) == 0 &&
|
||||||
|
v_start == hugepages[idx - 1].addr + hugepages[idx - 1].size) {
|
||||||
|
hugepages[idx - 1].size += (v_end - v_start);
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
|
||||||
|
hugepages[idx].addr = v_start;
|
||||||
|
hugepages[idx].size = v_end - v_start;
|
||||||
|
snprintf(hugepages[idx].path, PATH_MAX, "%s", tmp);
|
||||||
|
idx++;
|
||||||
|
}
|
||||||
|
|
||||||
|
rc = idx;
|
||||||
|
out:
|
||||||
|
fclose(f);
|
||||||
|
return rc;
|
||||||
|
}
|
||||||
|
|
||||||
|
static int
|
||||||
|
prepare_vhost_memory_user(struct vhost_user_msg *msg, int fds[])
|
||||||
|
{
|
||||||
|
int i, num;
|
||||||
|
struct hugepage_file_info hugepages[VHOST_USER_MEMORY_MAX_NREGIONS];
|
||||||
|
|
||||||
|
num = get_hugepage_file_info(hugepages, VHOST_USER_MEMORY_MAX_NREGIONS);
|
||||||
|
if (num < 0) {
|
||||||
|
SPDK_ERRLOG("Failed to prepare memory for vhost-user\n");
|
||||||
|
return num;
|
||||||
|
}
|
||||||
|
|
||||||
|
for (i = 0; i < num; ++i) {
|
||||||
|
/* the memory regions are unaligned */
|
||||||
|
msg->payload.memory.regions[i].guest_phys_addr = hugepages[i].addr; /* use vaddr! */
|
||||||
|
msg->payload.memory.regions[i].userspace_addr = hugepages[i].addr;
|
||||||
|
msg->payload.memory.regions[i].memory_size = hugepages[i].size;
|
||||||
|
msg->payload.memory.regions[i].flags_padding = 0;
|
||||||
|
fds[i] = open(hugepages[i].path, O_RDWR);
|
||||||
|
}
|
||||||
|
|
||||||
|
msg->payload.memory.nregions = num;
|
||||||
|
msg->payload.memory.padding = 0;
|
||||||
|
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
static const char *const vhost_msg_strings[VHOST_USER_MAX] = {
|
||||||
|
[VHOST_USER_SET_OWNER] = "VHOST_SET_OWNER",
|
||||||
|
[VHOST_USER_RESET_OWNER] = "VHOST_RESET_OWNER",
|
||||||
|
[VHOST_USER_SET_FEATURES] = "VHOST_SET_FEATURES",
|
||||||
|
[VHOST_USER_GET_FEATURES] = "VHOST_GET_FEATURES",
|
||||||
|
[VHOST_USER_SET_VRING_CALL] = "VHOST_SET_VRING_CALL",
|
||||||
|
[VHOST_USER_GET_PROTOCOL_FEATURES] = "VHOST_USER_GET_PROTOCOL_FEATURES",
|
||||||
|
[VHOST_USER_SET_PROTOCOL_FEATURES] = "VHOST_USER_SET_PROTOCOL_FEATURES",
|
||||||
|
[VHOST_USER_SET_VRING_NUM] = "VHOST_SET_VRING_NUM",
|
||||||
|
[VHOST_USER_SET_VRING_BASE] = "VHOST_SET_VRING_BASE",
|
||||||
|
[VHOST_USER_GET_VRING_BASE] = "VHOST_GET_VRING_BASE",
|
||||||
|
[VHOST_USER_SET_VRING_ADDR] = "VHOST_SET_VRING_ADDR",
|
||||||
|
[VHOST_USER_SET_VRING_KICK] = "VHOST_SET_VRING_KICK",
|
||||||
|
[VHOST_USER_SET_MEM_TABLE] = "VHOST_SET_MEM_TABLE",
|
||||||
|
[VHOST_USER_SET_VRING_ENABLE] = "VHOST_SET_VRING_ENABLE",
|
||||||
|
[VHOST_USER_GET_QUEUE_NUM] = "VHOST_USER_GET_QUEUE_NUM",
|
||||||
|
[VHOST_USER_GET_CONFIG] = "VHOST_USER_GET_CONFIG",
|
||||||
|
[VHOST_USER_SET_CONFIG] = "VHOST_USER_SET_CONFIG",
|
||||||
|
};
|
||||||
|
|
||||||
|
static int
|
||||||
|
vhost_user_sock(struct virtio_user_dev *dev,
|
||||||
|
enum vhost_user_request req,
|
||||||
|
void *arg)
|
||||||
|
{
|
||||||
|
struct vhost_user_msg msg;
|
||||||
|
struct vhost_vring_file *file = 0;
|
||||||
|
int need_reply = 0;
|
||||||
|
int fds[VHOST_USER_MEMORY_MAX_NREGIONS];
|
||||||
|
int fd_num = 0;
|
||||||
|
int i, len, rc;
|
||||||
|
int vhostfd = dev->vhostfd;
|
||||||
|
|
||||||
|
SPDK_DEBUGLOG(virtio_user, "sent message %d = %s\n", req, vhost_msg_strings[req]);
|
||||||
|
|
||||||
|
msg.request = req;
|
||||||
|
msg.flags = VHOST_USER_VERSION;
|
||||||
|
msg.size = 0;
|
||||||
|
|
||||||
|
switch (req) {
|
||||||
|
case VHOST_USER_GET_FEATURES:
|
||||||
|
case VHOST_USER_GET_PROTOCOL_FEATURES:
|
||||||
|
case VHOST_USER_GET_QUEUE_NUM:
|
||||||
|
need_reply = 1;
|
||||||
|
break;
|
||||||
|
|
||||||
|
case VHOST_USER_SET_FEATURES:
|
||||||
|
case VHOST_USER_SET_LOG_BASE:
|
||||||
|
case VHOST_USER_SET_PROTOCOL_FEATURES:
|
||||||
|
msg.payload.u64 = *((__u64 *)arg);
|
||||||
|
msg.size = sizeof(msg.payload.u64);
|
||||||
|
break;
|
||||||
|
|
||||||
|
case VHOST_USER_SET_OWNER:
|
||||||
|
case VHOST_USER_RESET_OWNER:
|
||||||
|
break;
|
||||||
|
|
||||||
|
case VHOST_USER_SET_MEM_TABLE:
|
||||||
|
rc = prepare_vhost_memory_user(&msg, fds);
|
||||||
|
if (rc < 0) {
|
||||||
|
return rc;
|
||||||
|
}
|
||||||
|
fd_num = msg.payload.memory.nregions;
|
||||||
|
msg.size = sizeof(msg.payload.memory.nregions);
|
||||||
|
msg.size += sizeof(msg.payload.memory.padding);
|
||||||
|
msg.size += fd_num * sizeof(struct vhost_memory_region);
|
||||||
|
break;
|
||||||
|
|
||||||
|
case VHOST_USER_SET_LOG_FD:
|
||||||
|
fds[fd_num++] = *((int *)arg);
|
||||||
|
break;
|
||||||
|
|
||||||
|
case VHOST_USER_SET_VRING_NUM:
|
||||||
|
case VHOST_USER_SET_VRING_BASE:
|
||||||
|
case VHOST_USER_SET_VRING_ENABLE:
|
||||||
|
memcpy(&msg.payload.state, arg, sizeof(msg.payload.state));
|
||||||
|
msg.size = sizeof(msg.payload.state);
|
||||||
|
break;
|
||||||
|
|
||||||
|
case VHOST_USER_GET_VRING_BASE:
|
||||||
|
memcpy(&msg.payload.state, arg, sizeof(msg.payload.state));
|
||||||
|
msg.size = sizeof(msg.payload.state);
|
||||||
|
need_reply = 1;
|
||||||
|
break;
|
||||||
|
|
||||||
|
case VHOST_USER_SET_VRING_ADDR:
|
||||||
|
memcpy(&msg.payload.addr, arg, sizeof(msg.payload.addr));
|
||||||
|
msg.size = sizeof(msg.payload.addr);
|
||||||
|
break;
|
||||||
|
|
||||||
|
case VHOST_USER_SET_VRING_KICK:
|
||||||
|
case VHOST_USER_SET_VRING_CALL:
|
||||||
|
case VHOST_USER_SET_VRING_ERR:
|
||||||
|
file = arg;
|
||||||
|
msg.payload.u64 = file->index & VHOST_USER_VRING_IDX_MASK;
|
||||||
|
msg.size = sizeof(msg.payload.u64);
|
||||||
|
if (file->fd > 0) {
|
||||||
|
fds[fd_num++] = file->fd;
|
||||||
|
} else {
|
||||||
|
msg.payload.u64 |= VHOST_USER_VRING_NOFD_MASK;
|
||||||
|
}
|
||||||
|
break;
|
||||||
|
|
||||||
|
case VHOST_USER_GET_CONFIG:
|
||||||
|
memcpy(&msg.payload.cfg, arg, sizeof(msg.payload.cfg));
|
||||||
|
msg.size = sizeof(msg.payload.cfg);
|
||||||
|
need_reply = 1;
|
||||||
|
break;
|
||||||
|
|
||||||
|
case VHOST_USER_SET_CONFIG:
|
||||||
|
memcpy(&msg.payload.cfg, arg, sizeof(msg.payload.cfg));
|
||||||
|
msg.size = sizeof(msg.payload.cfg);
|
||||||
|
break;
|
||||||
|
|
||||||
|
default:
|
||||||
|
SPDK_ERRLOG("trying to send unknown msg\n");
|
||||||
|
return -EINVAL;
|
||||||
|
}
|
||||||
|
|
||||||
|
len = VHOST_USER_HDR_SIZE + msg.size;
|
||||||
|
rc = vhost_user_write(vhostfd, &msg, len, fds, fd_num);
|
||||||
|
if (rc < 0) {
|
||||||
|
SPDK_ERRLOG("%s failed: %s\n",
|
||||||
|
vhost_msg_strings[req], spdk_strerror(-rc));
|
||||||
|
return rc;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (req == VHOST_USER_SET_MEM_TABLE)
|
||||||
|
for (i = 0; i < fd_num; ++i) {
|
||||||
|
close(fds[i]);
|
||||||
|
}
|
||||||
|
|
||||||
|
if (need_reply) {
|
||||||
|
rc = vhost_user_read(vhostfd, &msg);
|
||||||
|
if (rc < 0) {
|
||||||
|
SPDK_WARNLOG("Received msg failed: %s\n", spdk_strerror(-rc));
|
||||||
|
return rc;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (req != msg.request) {
|
||||||
|
SPDK_WARNLOG("Received unexpected msg type\n");
|
||||||
|
return -EIO;
|
||||||
|
}
|
||||||
|
|
||||||
|
switch (req) {
|
||||||
|
case VHOST_USER_GET_FEATURES:
|
||||||
|
case VHOST_USER_GET_PROTOCOL_FEATURES:
|
||||||
|
case VHOST_USER_GET_QUEUE_NUM:
|
||||||
|
if (msg.size != sizeof(msg.payload.u64)) {
|
||||||
|
SPDK_WARNLOG("Received bad msg size\n");
|
||||||
|
return -EIO;
|
||||||
|
}
|
||||||
|
*((__u64 *)arg) = msg.payload.u64;
|
||||||
|
break;
|
||||||
|
case VHOST_USER_GET_VRING_BASE:
|
||||||
|
if (msg.size != sizeof(msg.payload.state)) {
|
||||||
|
SPDK_WARNLOG("Received bad msg size\n");
|
||||||
|
return -EIO;
|
||||||
|
}
|
||||||
|
memcpy(arg, &msg.payload.state,
|
||||||
|
sizeof(struct vhost_vring_state));
|
||||||
|
break;
|
||||||
|
case VHOST_USER_GET_CONFIG:
|
||||||
|
if (msg.size != sizeof(msg.payload.cfg)) {
|
||||||
|
SPDK_WARNLOG("Received bad msg size\n");
|
||||||
|
return -EIO;
|
||||||
|
}
|
||||||
|
memcpy(arg, &msg.payload.cfg, sizeof(msg.payload.cfg));
|
||||||
|
break;
|
||||||
|
default:
|
||||||
|
SPDK_WARNLOG("Received unexpected msg type\n");
|
||||||
|
return -EBADMSG;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Set up environment to talk with a vhost user backend.
|
||||||
|
*
|
||||||
|
* @return
|
||||||
|
* - (-1) if fail;
|
||||||
|
* - (0) if succeed.
|
||||||
|
*/
|
||||||
|
static int
|
||||||
|
vhost_user_setup(struct virtio_user_dev *dev)
|
||||||
|
{
|
||||||
|
int fd;
|
||||||
|
int flag;
|
||||||
|
struct sockaddr_un un;
|
||||||
|
ssize_t rc;
|
||||||
|
|
||||||
|
fd = socket(AF_UNIX, SOCK_STREAM, 0);
|
||||||
|
if (fd < 0) {
|
||||||
|
SPDK_ERRLOG("socket() error, %s\n", spdk_strerror(errno));
|
||||||
|
return -errno;
|
||||||
|
}
|
||||||
|
|
||||||
|
flag = fcntl(fd, F_GETFD);
|
||||||
|
if (fcntl(fd, F_SETFD, flag | FD_CLOEXEC) < 0) {
|
||||||
|
SPDK_ERRLOG("fcntl failed, %s\n", spdk_strerror(errno));
|
||||||
|
}
|
||||||
|
|
||||||
|
memset(&un, 0, sizeof(un));
|
||||||
|
un.sun_family = AF_UNIX;
|
||||||
|
rc = snprintf(un.sun_path, sizeof(un.sun_path), "%s", dev->path);
|
||||||
|
if (rc < 0 || (size_t)rc >= sizeof(un.sun_path)) {
|
||||||
|
SPDK_ERRLOG("socket path too long\n");
|
||||||
|
close(fd);
|
||||||
|
if (rc < 0) {
|
||||||
|
return -errno;
|
||||||
|
} else {
|
||||||
|
return -EINVAL;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if (connect(fd, (struct sockaddr *)&un, sizeof(un)) < 0) {
|
||||||
|
SPDK_ERRLOG("connect error, %s\n", spdk_strerror(errno));
|
||||||
|
close(fd);
|
||||||
|
return -errno;
|
||||||
|
}
|
||||||
|
|
||||||
|
dev->vhostfd = fd;
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
struct virtio_user_backend_ops ops_user = {
|
||||||
|
.setup = vhost_user_setup,
|
||||||
|
.send_request = vhost_user_sock,
|
||||||
|
};
|
||||||
|
|
||||||
static int
|
static int
|
||||||
virtio_user_create_queue(struct virtio_dev *vdev, uint32_t queue_sel)
|
virtio_user_create_queue(struct virtio_dev *vdev, uint32_t queue_sel)
|
||||||
{
|
{
|
||||||
@ -585,3 +1057,4 @@ err:
|
|||||||
virtio_dev_destruct(vdev);
|
virtio_dev_destruct(vdev);
|
||||||
return rc;
|
return rc;
|
||||||
}
|
}
|
||||||
|
SPDK_LOG_REGISTER_COMPONENT(virtio_user)
|
Loading…
Reference in New Issue
Block a user