mirror of
https://github.com/gnif/LookingGlass.git
synced 2024-11-10 08:38:20 +00:00
[client] spice: add channel and mouse locking
This fixes a race condition which causes the mouse ringbuffer to overflow. It also corrects out of order message index IDs due to multiple threads sending messages asyncronously.
This commit is contained in:
parent
37ea662998
commit
c61d97b0ac
@ -18,6 +18,7 @@ Place, Suite 330, Boston, MA 02111-1307 USA
|
|||||||
*/
|
*/
|
||||||
|
|
||||||
#include "spice.h"
|
#include "spice.h"
|
||||||
|
#include "utils.h"
|
||||||
#include "debug.h"
|
#include "debug.h"
|
||||||
|
|
||||||
#include <string.h>
|
#include <string.h>
|
||||||
@ -67,6 +68,7 @@ struct SpiceChannel
|
|||||||
uint32_t ackFrequency;
|
uint32_t ackFrequency;
|
||||||
uint32_t ackCount;
|
uint32_t ackCount;
|
||||||
uint32_t serial;
|
uint32_t serial;
|
||||||
|
LG_Lock lock;
|
||||||
};
|
};
|
||||||
|
|
||||||
struct SpiceKeyboard
|
struct SpiceKeyboard
|
||||||
@ -74,7 +76,7 @@ struct SpiceKeyboard
|
|||||||
uint32_t modifiers;
|
uint32_t modifiers;
|
||||||
};
|
};
|
||||||
|
|
||||||
#define SPICE_MOUSE_QUEUE_SIZE 32
|
#define SPICE_MOUSE_QUEUE_SIZE 64
|
||||||
|
|
||||||
struct SpiceMouse
|
struct SpiceMouse
|
||||||
{
|
{
|
||||||
@ -84,6 +86,7 @@ struct SpiceMouse
|
|||||||
SpiceMsgcMouseMotion queue[SPICE_MOUSE_QUEUE_SIZE];
|
SpiceMsgcMouseMotion queue[SPICE_MOUSE_QUEUE_SIZE];
|
||||||
int rpos, wpos;
|
int rpos, wpos;
|
||||||
int queueLen;
|
int queueLen;
|
||||||
|
LG_Lock lock;
|
||||||
};
|
};
|
||||||
|
|
||||||
struct Spice
|
struct Spice
|
||||||
@ -136,6 +139,8 @@ bool spice_connect(const char * host, const short port, const char * password)
|
|||||||
spice.addr.sin_family = AF_INET;
|
spice.addr.sin_family = AF_INET;
|
||||||
spice.addr.sin_port = htons(port);
|
spice.addr.sin_port = htons(port);
|
||||||
|
|
||||||
|
LG_LOCK_INIT(spice.mouse.lock);
|
||||||
|
|
||||||
spice.channelID = 0;
|
spice.channelID = 0;
|
||||||
if (!spice_connect_channel(&spice.scMain))
|
if (!spice_connect_channel(&spice.scMain))
|
||||||
{
|
{
|
||||||
@ -153,6 +158,8 @@ void spice_disconnect()
|
|||||||
spice_disconnect_channel(&spice.scMain );
|
spice_disconnect_channel(&spice.scMain );
|
||||||
spice_disconnect_channel(&spice.scInputs);
|
spice_disconnect_channel(&spice.scInputs);
|
||||||
|
|
||||||
|
LG_LOCK_FREE(spice.mouse.lock);
|
||||||
|
|
||||||
spice.sessionID = 0;
|
spice.sessionID = 0;
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -206,16 +213,15 @@ bool spice_process()
|
|||||||
}
|
}
|
||||||
|
|
||||||
if (spice.scInputs.connected && i == spice.scInputs.socket)
|
if (spice.scInputs.connected && i == spice.scInputs.socket)
|
||||||
{
|
|
||||||
if (spice_on_inputs_channel_read())
|
|
||||||
{
|
{
|
||||||
if (!spice_process_ack(&spice.scInputs))
|
if (!spice_process_ack(&spice.scInputs))
|
||||||
{
|
{
|
||||||
DEBUG_ERROR("failed to process ack on inputs channel");
|
DEBUG_ERROR("failed to process ack on inputs channel");
|
||||||
return false;
|
return false;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
if (spice_on_inputs_channel_read())
|
||||||
continue;
|
continue;
|
||||||
}
|
|
||||||
else
|
else
|
||||||
{
|
{
|
||||||
DEBUG_ERROR("failed to perform read on inputs channel");
|
DEBUG_ERROR("failed to perform read on inputs channel");
|
||||||
@ -460,6 +466,7 @@ bool spice_on_inputs_channel_read()
|
|||||||
{
|
{
|
||||||
DEBUG_PROTO("SPICE_MSG_INPUTS_MOUSE_MOTION_ACK");
|
DEBUG_PROTO("SPICE_MSG_INPUTS_MOUSE_MOTION_ACK");
|
||||||
int sent = 0;
|
int sent = 0;
|
||||||
|
LG_LOCK(spice.mouse.lock);
|
||||||
while(spice.mouse.queueLen && sent < 4)
|
while(spice.mouse.queueLen && sent < 4)
|
||||||
{
|
{
|
||||||
SpiceMsgcMouseMotion *msg = &spice.mouse.queue[spice.mouse.rpos];
|
SpiceMsgcMouseMotion *msg = &spice.mouse.queue[spice.mouse.rpos];
|
||||||
@ -467,6 +474,7 @@ bool spice_on_inputs_channel_read()
|
|||||||
{
|
{
|
||||||
DEBUG_ERROR("failed to send post ack");
|
DEBUG_ERROR("failed to send post ack");
|
||||||
spice.mouse.sentCount = sent;
|
spice.mouse.sentCount = sent;
|
||||||
|
LG_UNLOCK(spice.mouse.lock);
|
||||||
return false;
|
return false;
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -478,6 +486,7 @@ bool spice_on_inputs_channel_read()
|
|||||||
}
|
}
|
||||||
|
|
||||||
spice.mouse.sentCount = sent;
|
spice.mouse.sentCount = sent;
|
||||||
|
LG_UNLOCK(spice.mouse.lock);
|
||||||
return true;
|
return true;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@ -496,6 +505,8 @@ bool spice_connect_channel(struct SpiceChannel * channel)
|
|||||||
channel->ackCount = 0;
|
channel->ackCount = 0;
|
||||||
channel->serial = 0;
|
channel->serial = 0;
|
||||||
|
|
||||||
|
LG_LOCK_INIT(channel->lock);
|
||||||
|
|
||||||
channel->socket = socket(AF_INET, SOCK_STREAM, 0);
|
channel->socket = socket(AF_INET, SOCK_STREAM, 0);
|
||||||
if (channel->socket == -1)
|
if (channel->socket == -1)
|
||||||
return false;
|
return false;
|
||||||
@ -637,6 +648,7 @@ void spice_disconnect_channel(struct SpiceChannel * channel)
|
|||||||
if (channel->connected)
|
if (channel->connected)
|
||||||
close(channel->socket);
|
close(channel->socket);
|
||||||
channel->connected = false;
|
channel->connected = false;
|
||||||
|
LG_LOCK_FREE(channel->lock);
|
||||||
}
|
}
|
||||||
|
|
||||||
// ============================================================================
|
// ============================================================================
|
||||||
@ -666,6 +678,8 @@ ssize_t spice_write(const struct SpiceChannel * channel, const void * buffer, co
|
|||||||
|
|
||||||
bool spice_write_msg(struct SpiceChannel * channel, uint32_t type, const void * buffer, const ssize_t size)
|
bool spice_write_msg(struct SpiceChannel * channel, uint32_t type, const void * buffer, const ssize_t size)
|
||||||
{
|
{
|
||||||
|
LG_LOCK(channel->lock);
|
||||||
|
|
||||||
SpiceDataHeader header;
|
SpiceDataHeader header;
|
||||||
header.serial = channel->serial++;
|
header.serial = channel->serial++;
|
||||||
header.type = type;
|
header.type = type;
|
||||||
@ -675,15 +689,18 @@ bool spice_write_msg(struct SpiceChannel * channel, uint32_t type, const void *
|
|||||||
if (spice_write(channel, &header, sizeof(header)) != sizeof(header))
|
if (spice_write(channel, &header, sizeof(header)) != sizeof(header))
|
||||||
{
|
{
|
||||||
DEBUG_ERROR("failed to write message header");
|
DEBUG_ERROR("failed to write message header");
|
||||||
|
LG_UNLOCK(channel->lock);
|
||||||
return false;
|
return false;
|
||||||
}
|
}
|
||||||
|
|
||||||
if (spice_write(channel, buffer, size) != size)
|
if (spice_write(channel, buffer, size) != size)
|
||||||
{
|
{
|
||||||
DEBUG_ERROR("failed to write message body");
|
DEBUG_ERROR("failed to write message body");
|
||||||
|
LG_UNLOCK(channel->lock);
|
||||||
return false;
|
return false;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
LG_UNLOCK(channel->lock);
|
||||||
return true;
|
return true;
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -819,11 +836,13 @@ bool spice_mouse_motion(int32_t x, int32_t y)
|
|||||||
return false;
|
return false;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
LG_LOCK(spice.mouse.lock);
|
||||||
if (spice.mouse.sentCount == 4)
|
if (spice.mouse.sentCount == 4)
|
||||||
{
|
{
|
||||||
if (spice.mouse.queueLen == SPICE_MOUSE_QUEUE_SIZE)
|
if (spice.mouse.queueLen == SPICE_MOUSE_QUEUE_SIZE)
|
||||||
{
|
{
|
||||||
DEBUG_ERROR("mouse motion ringbuffer full!");
|
DEBUG_ERROR("mouse motion ringbuffer full!");
|
||||||
|
LG_UNLOCK(spice.mouse.lock);
|
||||||
return false;
|
return false;
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -837,6 +856,7 @@ bool spice_mouse_motion(int32_t x, int32_t y)
|
|||||||
spice.mouse.wpos = 0;
|
spice.mouse.wpos = 0;
|
||||||
|
|
||||||
++spice.mouse.queueLen;
|
++spice.mouse.queueLen;
|
||||||
|
LG_UNLOCK(spice.mouse.lock);
|
||||||
return true;
|
return true;
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -846,6 +866,7 @@ bool spice_mouse_motion(int32_t x, int32_t y)
|
|||||||
msg.button_state = spice.mouse.buttonState;
|
msg.button_state = spice.mouse.buttonState;
|
||||||
|
|
||||||
++spice.mouse.sentCount;
|
++spice.mouse.sentCount;
|
||||||
|
LG_UNLOCK(spice.mouse.lock);
|
||||||
return spice_write_msg(&spice.scInputs, SPICE_MSGC_INPUTS_MOUSE_MOTION, &msg, sizeof(msg));
|
return spice_write_msg(&spice.scInputs, SPICE_MSGC_INPUTS_MOUSE_MOTION, &msg, sizeof(msg));
|
||||||
}
|
}
|
||||||
|
|
||||||
|
Loading…
Reference in New Issue
Block a user