Mirrored_Repos/LookingGlass
1
0
Fork 0
mirror of https://github.com/gnif/LookingGlass.git synced 2024-11-22 05:27:20 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

[porthole] process all messages sent from qemu

Browse Source
This commit is contained in:
Geoffrey McRae 2019-11-05 14:56:07 +11:00
parent 4c424cdbdf
commit 4dde82c646
2 changed files with 114 additions and 89 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

189
porthole/src/linux/client.c
View File

@ -62,6 +62,9 @@ Mapping;
struct PortholeClient struct PortholeClient
{ {
int socket; int socket;
char rxbuf[1024];
int rxbuf_len;
PortholeMapEvent map_cb; PortholeMapEvent map_cb;
PortholeUnmapEvent unmap_cb; PortholeUnmapEvent unmap_cb;
PortholeDisconEvent discon_cb; PortholeDisconEvent discon_cb;
@ -168,24 +171,26 @@ static void * porthole_socket_thread(void * opaque)
while(handle->running) while(handle->running)
{ {
PHMsg msg;
struct iovec io = struct iovec io =
{ {
.iov_base = &msg, .iov_base = &handle->rxbuf[handle->rxbuf_len],
.iov_len = sizeof(msg) .iov_len = sizeof(handle->rxbuf) - handle->rxbuf_len
}; };
char buffer[256] = {0}; char cbuffer[256] = {0};
struct msghdr msghdr = struct msghdr msghdr =
{ {
.msg_iov = &io, .msg_iov = &io,
.msg_iovlen = 1, .msg_iovlen = 1,
.msg_control = &buffer, .msg_control = &cbuffer,
.msg_controllen = sizeof(buffer) .msg_controllen = sizeof(cbuffer)
}; };
if (recvmsg(handle->socket, &msghdr, 0) < 0) handle->rxbuf_len = recvmsg(handle->socket, &msghdr, 0);
if (handle->rxbuf_len < 0)
{ {
handle->rxbuf_len = 0;
if (errno == EAGAIN || errno == EWOULDBLOCK) if (errno == EAGAIN || errno == EWOULDBLOCK)
continue; continue;
@ -195,96 +200,114 @@ static void * porthole_socket_thread(void * opaque)
break; break;
} }
switch(msg.msg) int pos = 0;
while(pos != handle->rxbuf_len)
{ {
case PH_MSG_MAP: PHMsg *msg = (PHMsg *)&handle->rxbuf[pos];
if (handle->current)
{
DEBUG_WARN("Started a new map before finishing the last one");
porthole_free_map(handle->current);
}
handle->current = porthole_intmap_new(); // check for an incomplete message
if (pos + msg->len > handle->rxbuf_len)
break; break;
case PH_MSG_FD: assert(msg->len >= PH_MSG_BASE_SIZE);
pos += msg->len;
/* process the message */
switch(msg->msg)
{ {
struct cmsghdr * cmsg = CMSG_FIRSTHDR(&msghdr); case PH_MSG_MAP:
int * fds = (int *)CMSG_DATA(cmsg); if (handle->current)
porthole_sharedfd_new(handle, msg.u.fd.id, fds[0]);
break;
}
case PH_MSG_SEGMENT:
{
if (!handle->current)
DEBUG_FATAL("Segment sent before map, this is a bug in the guest porthole device or driver");
porthole_segment_new(
handle->fds,
handle->current,
msg.u.segment.fd_id,
msg.u.segment.addr,
msg.u.segment.size
);
break;
}
case PH_MSG_FINISH:
if (!handle->current)
DEBUG_FATAL("Finished map before starting one");
handle->current->id = msg.u.finish.id;
objectlist_push(handle->intmaps, handle->current);
porthole_do_map(handle, handle->current, msg.u.finish.type);
handle->current = NULL;
break;
case PH_MSG_UNMAP:
{
// notify the application of the unmap
handle->unmap_cb(msg.u.unmap.id);
// remove the PortholeMap object
unsigned int count = objectlist_count(handle->maps);
for(unsigned int i = 0; i < count; ++i)
{
PortholeMap *m = (PortholeMap *)objectlist_at(handle->maps, i);
if (m->id == msg.u.unmap.id)
{ {
objectlist_remove(handle->maps, i); DEBUG_WARN("Started a new map before finishing the last one");
break; porthole_free_map(handle->current);
} }
handle->current = porthole_intmap_new();
break;
case PH_MSG_FD:
{
struct cmsghdr * cmsg = CMSG_FIRSTHDR(&msghdr);
int * fds = (int *)CMSG_DATA(cmsg);
porthole_sharedfd_new(handle, msg->u.fd.id, fds[0]);
break;
} }
// remove the internal mapping object case PH_MSG_SEGMENT:
count = objectlist_count(handle->intmaps);
for(unsigned int i = 0; i < count; ++i)
{ {
Mapping *m = (Mapping *)objectlist_at(handle->intmaps, i); if (!handle->current)
if (m->id == msg.u.unmap.id) DEBUG_FATAL("Segment sent before map, this is a bug in the guest porthole device or driver");
porthole_segment_new(
handle->fds,
handle->current,
msg->u.segment.fd_id,
msg->u.segment.addr,
msg->u.segment.size
);
break;
}
case PH_MSG_FINISH:
if (!handle->current)
DEBUG_FATAL("Finished map before starting one");
handle->current->id = msg->u.finish.id;
objectlist_push(handle->intmaps, handle->current);
porthole_do_map(handle, handle->current, msg->u.finish.type);
handle->current = NULL;
break;
case PH_MSG_UNMAP:
{
// notify the application of the unmap
handle->unmap_cb(msg->u.unmap.id);
// remove the PortholeMap object
unsigned int count = objectlist_count(handle->maps);
for(unsigned int i = 0; i < count; ++i)
{ {
objectlist_remove(handle->intmaps, i); PortholeMap *m = (PortholeMap *)objectlist_at(handle->maps, i);
break; if (m->id == msg->u.unmap.id)
{
objectlist_remove(handle->maps, i);
break;
}
} }
}
// reply to the guest to allow it to continue // remove the internal mapping object
uint32_t reply = PH_MSG_UNMAP; count = objectlist_count(handle->intmaps);
msghdr.msg_controllen = 0; for(unsigned int i = 0; i < count; ++i)
io.iov_base = &reply; {
io.iov_len = sizeof(reply); Mapping *m = (Mapping *)objectlist_at(handle->intmaps, i);
if (sendmsg(handle->socket, &msghdr, 0) < 0) if (m->id == msg->u.unmap.id)
{ {
DEBUG_ERROR("Failed to respond to the guest"); objectlist_remove(handle->intmaps, i);
handle->running = false; break;
if (handle->discon_cb) }
handle->discon_cb(); }
}
break; // reply to the guest to allow it to continue
uint32_t reply = PH_MSG_UNMAP;
msghdr.msg_controllen = 0;
io.iov_base = &reply;
io.iov_len = sizeof(reply);
if (sendmsg(handle->socket, &msghdr, 0) < 0)
{
DEBUG_ERROR("Failed to respond to the guest");
handle->running = false;
if (handle->discon_cb)
handle->discon_cb();
}
break;
}
} }
} }
// save any remaining bytes for the next recv
handle->rxbuf_len -= pos;
memmove(handle->rxbuf, &handle->rxbuf[pos], handle->rxbuf_len);
} }
handle->running = false; handle->running = false;

14
porthole/src/phmsg.h
View File

@ -39,7 +39,8 @@ typedef struct {
} __attribute__ ((packed)) PHMsgUnmap; } __attribute__ ((packed)) PHMsgUnmap;
typedef struct { typedef struct {
uint32_t msg; uint8_t len; // the message length
uint8_t msg; // the message ID
union union
{ {
PHMsgFd fd; PHMsgFd fd;
@ -55,8 +56,9 @@ typedef struct {
#define PH_MSG_FINISH 0x4 // finish of map sequence #define PH_MSG_FINISH 0x4 // finish of map sequence
#define PH_MSG_UNMAP 0x5 // unmap a previous map #define PH_MSG_UNMAP 0x5 // unmap a previous map
#define PH_MSG_MAP_SIZE (sizeof(uint32_t)) #define PH_MSG_BASE_SIZE (sizeof(uint8_t) + sizeof(uint8_t))
#define PH_MSG_FD_SIZE (sizeof(uint32_t) + sizeof(PHMsgFd)) #define PH_MSG_MAP_SIZE (PH_MSG_BASE_SIZE)
#define PH_MSG_SEGMENT_SIZE (sizeof(uint32_t) + sizeof(PHMsgSegment)) #define PH_MSG_FD_SIZE (PH_MSG_BASE_SIZE + sizeof(PHMsgFd))
#define PH_MSG_FINISH_SIZE (sizeof(uint32_t) + sizeof(PHMsgFinish)) #define PH_MSG_SEGMENT_SIZE (PH_MSG_BASE_SIZE + sizeof(PHMsgSegment))
#define PH_MSG_UNMAP_SIZE (sizeof(uint32_t) + sizeof(PHMsgUnmap)) #define PH_MSG_FINISH_SIZE (PH_MSG_BASE_SIZE + sizeof(PHMsgFinish))
#define PH_MSG_UNMAP_SIZE (PH_MSG_BASE_SIZE + sizeof(PHMsgUnmap))