ivshmem: Disentangle ivshmem_read()
Signed-off-by: Markus Armbruster <armbru@redhat.com> Reviewed-by: Marc-André Lureau <marcandre.lureau@redhat.com> Message-Id: <1458066895-20632-24-git-send-email-armbru@redhat.com>
This commit is contained in:
parent
cd9953f720
commit
ca0b7566cc
@ -564,116 +564,117 @@ static void setup_interrupt(IVShmemState *s, int vector)
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
static void ivshmem_read(void *opaque, const uint8_t *buf, int size)
|
static void process_msg_shmem(IVShmemState *s, int fd)
|
||||||
{
|
{
|
||||||
IVShmemState *s = opaque;
|
|
||||||
int incoming_fd;
|
|
||||||
int new_eventfd;
|
|
||||||
int64_t incoming_posn;
|
|
||||||
Error *err = NULL;
|
Error *err = NULL;
|
||||||
Peer *peer;
|
void *ptr;
|
||||||
|
|
||||||
if (!fifo_update_and_get_i64(s, buf, size, &incoming_posn)) {
|
if (memory_region_is_mapped(&s->ivshmem)) {
|
||||||
|
error_report("shm already initialized");
|
||||||
|
close(fd);
|
||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
|
|
||||||
incoming_fd = qemu_chr_fe_get_msgfd(s->server_chr);
|
if (check_shm_size(s, fd, &err) == -1) {
|
||||||
IVSHMEM_DPRINTF("posn is %" PRId64 ", fd is %d\n",
|
error_report_err(err);
|
||||||
incoming_posn, incoming_fd);
|
close(fd);
|
||||||
|
|
||||||
if (incoming_posn < -1 || incoming_posn > IVSHMEM_MAX_PEERS) {
|
|
||||||
error_report("server sent invalid message %" PRId64,
|
|
||||||
incoming_posn);
|
|
||||||
if (incoming_fd != -1) {
|
|
||||||
close(incoming_fd);
|
|
||||||
}
|
|
||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
|
|
||||||
if (incoming_posn >= s->nb_peers) {
|
/* mmap the region and map into the BAR2 */
|
||||||
resize_peers(s, incoming_posn + 1);
|
ptr = mmap(0, s->ivshmem_size, PROT_READ | PROT_WRITE, MAP_SHARED, fd, 0);
|
||||||
}
|
if (ptr == MAP_FAILED) {
|
||||||
|
error_report("Failed to mmap shared memory %s", strerror(errno));
|
||||||
peer = &s->peers[incoming_posn];
|
close(fd);
|
||||||
|
|
||||||
if (incoming_fd == -1) {
|
|
||||||
/* if posn is positive and unseen before then this is our posn*/
|
|
||||||
if (incoming_posn >= 0 && s->vm_id == -1) {
|
|
||||||
/* receive our posn */
|
|
||||||
s->vm_id = incoming_posn;
|
|
||||||
} else {
|
|
||||||
/* otherwise an fd == -1 means an existing peer has gone away */
|
|
||||||
IVSHMEM_DPRINTF("posn %" PRId64 " has gone away\n", incoming_posn);
|
|
||||||
close_peer_eventfds(s, incoming_posn);
|
|
||||||
}
|
|
||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
|
memory_region_init_ram_ptr(&s->ivshmem, OBJECT(s),
|
||||||
|
"ivshmem.bar2", s->ivshmem_size, ptr);
|
||||||
|
qemu_set_ram_fd(memory_region_get_ram_addr(&s->ivshmem), fd);
|
||||||
|
vmstate_register_ram(&s->ivshmem, DEVICE(s));
|
||||||
|
memory_region_add_subregion(&s->bar, 0, &s->ivshmem);
|
||||||
|
}
|
||||||
|
|
||||||
/* if the position is -1, then it's shared memory region fd */
|
static void process_msg_disconnect(IVShmemState *s, uint16_t posn)
|
||||||
if (incoming_posn == -1) {
|
{
|
||||||
void * map_ptr;
|
IVSHMEM_DPRINTF("posn %d has gone away\n", posn);
|
||||||
|
close_peer_eventfds(s, posn);
|
||||||
|
}
|
||||||
|
|
||||||
if (memory_region_is_mapped(&s->ivshmem)) {
|
static void process_msg_connect(IVShmemState *s, uint16_t posn, int fd)
|
||||||
error_report("shm already initialized");
|
{
|
||||||
close(incoming_fd);
|
Peer *peer = &s->peers[posn];
|
||||||
return;
|
int vector;
|
||||||
}
|
|
||||||
|
|
||||||
if (check_shm_size(s, incoming_fd, &err) == -1) {
|
/*
|
||||||
error_report_err(err);
|
* The N-th connect message for this peer comes with the file
|
||||||
close(incoming_fd);
|
* descriptor for vector N-1. Count messages to find the vector.
|
||||||
return;
|
*/
|
||||||
}
|
|
||||||
|
|
||||||
/* mmap the region and map into the BAR2 */
|
|
||||||
map_ptr = mmap(0, s->ivshmem_size, PROT_READ|PROT_WRITE, MAP_SHARED,
|
|
||||||
incoming_fd, 0);
|
|
||||||
if (map_ptr == MAP_FAILED) {
|
|
||||||
error_report("Failed to mmap shared memory %s", strerror(errno));
|
|
||||||
close(incoming_fd);
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
memory_region_init_ram_ptr(&s->ivshmem, OBJECT(s),
|
|
||||||
"ivshmem.bar2", s->ivshmem_size, map_ptr);
|
|
||||||
qemu_set_ram_fd(memory_region_get_ram_addr(&s->ivshmem),
|
|
||||||
incoming_fd);
|
|
||||||
vmstate_register_ram(&s->ivshmem, DEVICE(s));
|
|
||||||
|
|
||||||
IVSHMEM_DPRINTF("guest h/w addr = %p, size = %" PRIu64 "\n",
|
|
||||||
map_ptr, s->ivshmem_size);
|
|
||||||
|
|
||||||
memory_region_add_subregion(&s->bar, 0, &s->ivshmem);
|
|
||||||
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
/* each peer has an associated array of eventfds, and we keep
|
|
||||||
* track of how many eventfds received so far */
|
|
||||||
/* get a new eventfd: */
|
|
||||||
if (peer->nb_eventfds >= s->vectors) {
|
if (peer->nb_eventfds >= s->vectors) {
|
||||||
error_report("Too many eventfd received, device has %d vectors",
|
error_report("Too many eventfd received, device has %d vectors",
|
||||||
s->vectors);
|
s->vectors);
|
||||||
close(incoming_fd);
|
close(fd);
|
||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
|
vector = peer->nb_eventfds++;
|
||||||
|
|
||||||
new_eventfd = peer->nb_eventfds++;
|
IVSHMEM_DPRINTF("eventfds[%d][%d] = %d\n", posn, vector, fd);
|
||||||
|
event_notifier_init_fd(&peer->eventfds[vector], fd);
|
||||||
|
fcntl_setfl(fd, O_NONBLOCK); /* msix/irqfd poll non block */
|
||||||
|
|
||||||
/* this is an eventfd for a particular peer VM */
|
if (posn == s->vm_id) {
|
||||||
IVSHMEM_DPRINTF("eventfds[%" PRId64 "][%d] = %d\n", incoming_posn,
|
setup_interrupt(s, vector);
|
||||||
new_eventfd, incoming_fd);
|
|
||||||
event_notifier_init_fd(&peer->eventfds[new_eventfd], incoming_fd);
|
|
||||||
fcntl_setfl(incoming_fd, O_NONBLOCK); /* msix/irqfd poll non block */
|
|
||||||
|
|
||||||
if (incoming_posn == s->vm_id) {
|
|
||||||
setup_interrupt(s, new_eventfd);
|
|
||||||
}
|
}
|
||||||
|
|
||||||
if (ivshmem_has_feature(s, IVSHMEM_IOEVENTFD)) {
|
if (ivshmem_has_feature(s, IVSHMEM_IOEVENTFD)) {
|
||||||
ivshmem_add_eventfd(s, incoming_posn, new_eventfd);
|
ivshmem_add_eventfd(s, posn, vector);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
static void process_msg(IVShmemState *s, int64_t msg, int fd)
|
||||||
|
{
|
||||||
|
IVSHMEM_DPRINTF("posn is %" PRId64 ", fd is %d\n", msg, fd);
|
||||||
|
|
||||||
|
if (msg < -1 || msg > IVSHMEM_MAX_PEERS) {
|
||||||
|
error_report("server sent invalid message %" PRId64, msg);
|
||||||
|
close(fd);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (msg == -1) {
|
||||||
|
process_msg_shmem(s, fd);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (msg >= s->nb_peers) {
|
||||||
|
resize_peers(s, msg + 1);
|
||||||
|
}
|
||||||
|
|
||||||
|
if (fd >= 0) {
|
||||||
|
process_msg_connect(s, msg, fd);
|
||||||
|
} else if (s->vm_id == -1) {
|
||||||
|
s->vm_id = msg;
|
||||||
|
} else {
|
||||||
|
process_msg_disconnect(s, msg);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
static void ivshmem_read(void *opaque, const uint8_t *buf, int size)
|
||||||
|
{
|
||||||
|
IVShmemState *s = opaque;
|
||||||
|
int fd;
|
||||||
|
int64_t msg;
|
||||||
|
|
||||||
|
if (!fifo_update_and_get_i64(s, buf, size, &msg)) {
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
fd = qemu_chr_fe_get_msgfd(s->server_chr);
|
||||||
|
IVSHMEM_DPRINTF("posn is %" PRId64 ", fd is %d\n", msg, fd);
|
||||||
|
|
||||||
|
process_msg(s, msg, fd);
|
||||||
|
}
|
||||||
|
|
||||||
static void ivshmem_check_version(void *opaque, const uint8_t * buf, int size)
|
static void ivshmem_check_version(void *opaque, const uint8_t * buf, int size)
|
||||||
{
|
{
|
||||||
IVShmemState *s = opaque;
|
IVShmemState *s = opaque;
|
||||||
|
Loading…
Reference in New Issue
Block a user