[client] tests: exercise mandatory LGMP streams

Replace legacy clipboard grant and queued input fixtures with bound SPSC
stream endpoints. Verify stream-only payload delivery, credit recovery,
release completion, and four 256 KiB records for a 1 MiB response.

Keep queue assertions for status, discovery, ownership, and control.
This commit is contained in:
Geoffrey McRae
2026-08-15 14:37:01 +10:00
parent be4bb19fa4
commit cecd4b6f48
3 changed files with 465 additions and 202 deletions

View File

@@ -119,7 +119,7 @@ set(LGMP_CLIPBOARD_CASES
claim claim
offer-request offer-request
inbound-stream inbound-stream
outbound-grant outbound-stream
blocked blocked
cancel cancel
restart restart

View File

@@ -26,6 +26,7 @@
#include <lgmp/client.h> #include <lgmp/client.h>
#include <lgmp/host.h> #include <lgmp/host.h>
#include <lgmp/stream.h>
#include <pthread.h> #include <pthread.h>
#include <stdatomic.h> #include <stdatomic.h>
@@ -38,14 +39,11 @@
#define TEST_WAIT_MS 2500U #define TEST_WAIT_MS 2500U
#define TEST_QUIET_MS 50U #define TEST_QUIET_MS 50U
#define TEST_MEMORY_MAX 128U #define TEST_MEMORY_MAX 128U
#define SLOT_BYTES \ #define SLOT_BYTES KVMFR_CLIPBOARD_STREAM_RECORD_BYTES
(sizeof(KVMFRClipboardSlotHeader) + KVMFR_CLIPBOARD_DATA_BYTES)
#define TEST_TRANSIENT_SLOTS 4U
#define TEST_SHM_OVERHEAD (1024U * 1024U) #define TEST_SHM_OVERHEAD (1024U * 1024U)
/* Match the production inbound and outbound slot pools, retain several /* Match the production duplex streams and leave room for LGMP metadata. */
* synthetic host DATA records at once, and leave room for LGMP metadata. */
#define TEST_SHM_SIZE \ #define TEST_SHM_SIZE \
((2U * KVMFR_CLIPBOARD_SLOT_COUNT + TEST_TRANSIENT_SLOTS) * SLOT_BYTES + \ (2U * KVMFR_CLIPBOARD_STREAM_SLOT_COUNT * SLOT_BYTES + \
TEST_SHM_OVERHEAD) TEST_SHM_OVERHEAD)
#define CHECK(x) \ #define CHECK(x) \
@@ -106,6 +104,11 @@ typedef struct TestState
PLGMPHost host; PLGMPHost host;
PLGMPHostQueue queue; PLGMPHostQueue queue;
PLGMPClient client; PLGMPClient client;
PLGMPHostStream hostToClientStream;
PLGMPHostStream clientToHostStream;
KVMFRStreamDescriptor hostToClientDescriptor;
KVMFRStreamDescriptor clientToHostDescriptor;
bool streamsBound;
LGMPClipboard * clipboard; LGMPClipboard * clipboard;
const LG_ClipboardOps * ops; const LG_ClipboardOps * ops;
uint32_t clientID; uint32_t clientID;
@@ -113,7 +116,6 @@ typedef struct TestState
uint64_t localClipboardGeneration; uint64_t localClipboardGeneration;
PLGMPMemory memories[TEST_MEMORY_MAX]; PLGMPMemory memories[TEST_MEMORY_MAX];
unsigned memoryCount; unsigned memoryCount;
PLGMPMemory grantMemory[KVMFR_CLIPBOARD_SLOT_COUNT];
atomic_uint statusCount; atomic_uint statusCount;
atomic_bool statusAvailable; atomic_bool statusAvailable;
atomic_uint statusGeneration; atomic_uint statusGeneration;
@@ -315,6 +317,23 @@ static bool hostProcess(TestState * state)
return true; return true;
} }
static KVMFRStreamDescriptor exportStreamDescriptor(
const struct LGMPStreamDescriptor * descriptor)
{
return (KVMFRStreamDescriptor)
{
.magic = descriptor->magic,
.version = descriptor->version,
.size = descriptor->size,
.offset = descriptor->offset,
.regionSize = descriptor->regionSize,
.direction = descriptor->direction,
.policy = descriptor->policy,
.slotCount = descriptor->slotCount,
.slotSize = descriptor->slotSize,
};
}
static bool waitAtomic(TestState * state, atomic_uint * value, static bool waitAtomic(TestState * state, atomic_uint * value,
unsigned expected) unsigned expected)
{ {
@@ -375,10 +394,31 @@ static uint32_t nextSerial(TestState * state)
return state->hostSerial; return state->hostSerial;
} }
static bool unbindStream(TestState * state, PLGMPHostStream stream)
{
for (unsigned i = 0; i < TEST_WAIT_MS; ++i)
{
const LGMP_STATUS status = lgmpHostStreamUnbind(stream);
if (status == LGMP_OK)
return true;
CHECK(status == LGMP_ERR_STREAM_BUSY);
CHECK(hostProcess(state));
usleep(1000);
}
return false;
}
static bool postStatus(TestState * state, uint32_t endpointGeneration, static bool postStatus(TestState * state, uint32_t endpointGeneration,
uint32_t ownerClientID, uint32_t ownerGeneration, uint32_t ownerClientID, uint32_t ownerGeneration,
bool available, KVMFRClipboardFormatFlags formats) bool available, KVMFRClipboardFormatFlags formats)
{ {
if (!ownerClientID && state->streamsBound)
{
CHECK(unbindStream(state, state->clientToHostStream));
CHECK(unbindStream(state, state->hostToClientStream));
state->streamsBound = false;
}
PLGMPMemory memory; PLGMPMemory memory;
CHECK(allocMemory(state, sizeof(KVMFRClipboardStatus), &memory)); CHECK(allocMemory(state, sizeof(KVMFRClipboardStatus), &memory));
KVMFRClipboardStatus * status = lgmpHostMemPtr(memory); KVMFRClipboardStatus * status = lgmpHostMemPtr(memory);
@@ -392,7 +432,11 @@ static bool postStatus(TestState * state, uint32_t endpointGeneration,
.ownerGeneration = ownerGeneration, .ownerGeneration = ownerGeneration,
.lease = 1000, .lease = 1000,
.formats = available ? formats : 0, .formats = available ? formats : 0,
.slotBytes = KVMFR_CLIPBOARD_DATA_BYTES, .slotBytes = KVMFR_CLIPBOARD_STREAM_SLOT_BYTES,
.streamVersion = KVMFR_CLIPBOARD_STREAM_VERSION,
.streamSlotCount = KVMFR_CLIPBOARD_STREAM_SLOT_COUNT,
.hostToClient = state->hostToClientDescriptor,
.clientToHost = state->clientToHostDescriptor,
}; };
CHECK(postMemory(state, memory, KVMFR_CLIPBOARD_QUEUE_STATUS, CHECK(postMemory(state, memory, KVMFR_CLIPBOARD_QUEUE_STATUS,
nextSerial(state))); nextSerial(state)));
@@ -404,37 +448,109 @@ static bool postRecord(TestState * state,
const KVMFRClipboardMessage * record, const KVMFRClipboardMessage * record,
KVMFRClipboardQueueType type, const void * data) KVMFRClipboardQueueType type, const void * data)
{ {
const size_t bytes = type == KVMFR_CLIPBOARD_QUEUE_DATA ? CHECK(type == KVMFR_CLIPBOARD_QUEUE_MESSAGE);
SLOT_BYTES : sizeof(*record); CHECK(record->length == 0);
CHECK(data == NULL);
PLGMPMemory memory; PLGMPMemory memory;
CHECK(allocMemory(state, bytes, &memory)); CHECK(allocMemory(state, sizeof(*record), &memory));
memcpy(lgmpHostMemPtr(memory), record, sizeof(*record)); memcpy(lgmpHostMemPtr(memory), record, sizeof(*record));
if (record->length && data)
memcpy((uint8_t *)lgmpHostMemPtr(memory) + sizeof(*record),
data, record->length);
CHECK(postMemory(state, memory, type, nextSerial(state))); CHECK(postMemory(state, memory, type, nextSerial(state)));
return true; return true;
} }
static bool postGrant(TestState * state, uint32_t generation, static bool postStreamRecord(TestState * state,
uint32_t token) const KVMFRClipboardMessage * record, const void * data)
{ {
CHECK(token >= 1 && token <= KVMFR_CLIPBOARD_SLOT_COUNT); CHECK(state->streamsBound);
PLGMPMemory * memory = &state->grantMemory[token - 1]; CHECK(record->length <= KVMFR_CLIPBOARD_STREAM_SLOT_BYTES);
if (!*memory) CHECK(!record->length || data);
CHECK(allocMemory(state, SLOT_BYTES, memory)); for (unsigned i = 0; i < TEST_WAIT_MS; ++i)
else {
CHECK(waitMemory(state, *memory)); CHECK(hostProcess(state));
LGMPStreamBuffer buffer = { 0 };
const LGMP_STATUS status = lgmpHostStreamWriteAcquire(
state->hostToClientStream, &buffer);
if (status == LGMP_ERR_STREAM_FULL)
{
usleep(1000);
continue;
}
CHECK(status == LGMP_OK);
const uint32_t bytes = sizeof(*record) + record->length;
CHECK(buffer.capacity >= bytes);
memcpy(buffer.data, record, sizeof(*record));
if (record->length)
memcpy((uint8_t *)buffer.data + sizeof(*record), data,
record->length);
CHECK(lgmpHostStreamWriteCommit(state->hostToClientStream,
&buffer, bytes) == LGMP_OK);
return true;
}
return false;
}
KVMFRClipboardSlotHeader * header = lgmpHostMemPtr(*memory); static bool settleHostStream(TestState * state)
memset(header, 0, sizeof(*header)); {
header->version = KVMFR_CLIPBOARD_VERSION; for (unsigned i = 0; i < TEST_QUIET_MS; ++i)
header->type = KVMFR_CLIPBOARD_MESSAGE_GRANT; {
header->generation = generation; CHECK(hostProcess(state));
header->size = KVMFR_CLIPBOARD_DATA_BYTES; usleep(1000);
header->token = token; }
return postMemory(state, *memory, return true;
KVMFR_CLIPBOARD_QUEUE_GRANT, token); }
static bool expectPayloadQueueClean(TestState * state)
{
for (;;)
{
KVMFRClipboardMessage record;
size_t size = sizeof(record);
uint32_t clientID = 0;
const LGMP_STATUS status = lgmpHostReadDataWithSource(
state->queue, &record, &size, &clientID);
if (status == LGMP_ERR_QUEUE_EMPTY)
return true;
CHECK(status == LGMP_OK);
CHECK(size == sizeof(record));
CHECK(clientID == state->clientID);
CHECK(lgmpHostAckData(state->queue) == LGMP_OK);
CHECK(record.type == KVMFR_CLIPBOARD_MESSAGE_KEEPALIVE);
}
}
static bool readClientStream(TestState * state,
KVMFRClipboardMessage * record, LGMPStreamBuffer * buffer,
const void ** data)
{
for (unsigned i = 0; i < TEST_WAIT_MS; ++i)
{
CHECK(hostProcess(state));
CHECK(expectPayloadQueueClean(state));
memset(buffer, 0, sizeof(*buffer));
const LGMP_STATUS status = lgmpHostStreamReadPeek(
state->clientToHostStream, buffer);
if (status == LGMP_ERR_STREAM_EMPTY)
{
usleep(1000);
continue;
}
CHECK(status == LGMP_OK);
CHECK(buffer->size >= sizeof(*record));
memcpy(record, buffer->data, sizeof(*record));
CHECK(buffer->size == sizeof(*record) + record->length);
*data = (const uint8_t *)buffer->data + sizeof(*record);
return true;
}
return false;
}
static bool releaseClientStream(TestState * state,
LGMPStreamBuffer * buffer)
{
CHECK(lgmpHostStreamReadRelease(
state->clientToHostStream, buffer) == LGMP_OK);
memset(buffer, 0, sizeof(*buffer));
return true;
} }
static bool readClient(TestState * state, KVMFRClipboardMessage * record) static bool readClient(TestState * state, KVMFRClipboardMessage * record)
@@ -454,6 +570,8 @@ static bool readClient(TestState * state, KVMFRClipboardMessage * record)
CHECK(status == LGMP_OK); CHECK(status == LGMP_OK);
CHECK(size == sizeof(*record)); CHECK(size == sizeof(*record));
CHECK(clientID == state->clientID); CHECK(clientID == state->clientID);
CHECK(record->type != KVMFR_CLIPBOARD_MESSAGE_DATA);
CHECK(record->type != KVMFR_CLIPBOARD_MESSAGE_FILE_DATA);
CHECK(lgmpHostAckData(state->queue) == LGMP_OK); CHECK(lgmpHostAckData(state->queue) == LGMP_OK);
return true; return true;
} }
@@ -484,7 +602,7 @@ static bool readNonKeepalive(TestState * state,
return false; return false;
} }
static bool noClientData(TestState * state) static bool noClientControl(TestState * state)
{ {
for (unsigned i = 0; i < TEST_QUIET_MS; ++i) for (unsigned i = 0; i < TEST_QUIET_MS; ++i)
{ {
@@ -531,12 +649,25 @@ static bool claim(TestState * state, uint32_t endpointGeneration,
CHECK(result->version == KVMFR_CLIPBOARD_VERSION); CHECK(result->version == KVMFR_CLIPBOARD_VERSION);
CHECK(result->generation != 0); CHECK(result->generation != 0);
CHECK(result->token == endpointGeneration); CHECK(result->token == endpointGeneration);
CHECK(result->flags == 0);
return true; return true;
} }
static bool own(TestState * state, uint32_t endpointGeneration, static bool own(TestState * state, uint32_t endpointGeneration,
uint32_t claimGeneration) uint32_t claimGeneration)
{ {
if (!state->streamsBound)
{
CHECK(lgmpHostStreamBind(state->hostToClientStream,
state->clientID, NULL) == LGMP_OK);
if (lgmpHostStreamBind(state->clientToHostStream,
state->clientID, NULL) != LGMP_OK)
{
lgmpHostStreamUnbind(state->hostToClientStream);
return false;
}
state->streamsBound = true;
}
CHECK(postStatus(state, endpointGeneration, state->clientID, CHECK(postStatus(state, endpointGeneration, state->clientID,
claimGeneration, true, 0)); claimGeneration, true, 0));
return true; return true;
@@ -625,7 +756,7 @@ static bool testInboundStream(TestState * state)
data.size = 5; data.size = 5;
data.format = KVMFR_CLIPBOARD_FORMAT_TEXT; data.format = KVMFR_CLIPBOARD_FORMAT_TEXT;
data.flags = KVMFR_CLIPBOARD_FLAG_BEGIN; data.flags = KVMFR_CLIPBOARD_FLAG_BEGIN;
CHECK(postRecord(state, &data, KVMFR_CLIPBOARD_QUEUE_DATA, NULL)); CHECK(postStreamRecord(state, &data, NULL));
CHECK(waitAtomic(state, &state->events.dataBegin, 1)); CHECK(waitAtomic(state, &state->events.dataBegin, 1));
CHECK(atomic_load(&state->events.dataBegin) == 1); CHECK(atomic_load(&state->events.dataBegin) == 1);
CHECK(atomic_load(&state->events.dataChunk) == 0); CHECK(atomic_load(&state->events.dataChunk) == 0);
@@ -636,7 +767,7 @@ static bool testInboundStream(TestState * state)
data.size = KVMFR_CLIPBOARD_SIZE_UNKNOWN; data.size = KVMFR_CLIPBOARD_SIZE_UNKNOWN;
data.flags = 0; data.flags = 0;
data.length = sizeof(bytes); data.length = sizeof(bytes);
CHECK(postRecord(state, &data, KVMFR_CLIPBOARD_QUEUE_DATA, bytes)); CHECK(postStreamRecord(state, &data, bytes));
CHECK(waitAtomic(state, &state->events.dataChunk, 1)); CHECK(waitAtomic(state, &state->events.dataChunk, 1));
CHECK(state->events.chunkOffset == 0); CHECK(state->events.chunkOffset == 0);
CHECK(state->events.chunkSize == sizeof(bytes)); CHECK(state->events.chunkSize == sizeof(bytes));
@@ -647,7 +778,7 @@ static bool testInboundStream(TestState * state)
data.size = sizeof(bytes); data.size = sizeof(bytes);
data.flags = KVMFR_CLIPBOARD_FLAG_END; data.flags = KVMFR_CLIPBOARD_FLAG_END;
data.length = 0; data.length = 0;
CHECK(postRecord(state, &data, KVMFR_CLIPBOARD_QUEUE_DATA, NULL)); CHECK(postStreamRecord(state, &data, NULL));
CHECK(waitAtomic(state, &state->events.dataEnd, 1)); CHECK(waitAtomic(state, &state->events.dataEnd, 1));
CHECK(atomic_load(&state->events.dataChunk) == 1); CHECK(atomic_load(&state->events.dataChunk) == 1);
CHECK(state->events.finalSize == 5); CHECK(state->events.finalSize == 5);
@@ -662,7 +793,7 @@ static bool testInboundStream(TestState * state)
data.format = KVMFR_CLIPBOARD_FORMAT_TEXT; data.format = KVMFR_CLIPBOARD_FORMAT_TEXT;
data.flags = KVMFR_CLIPBOARD_FLAG_BEGIN | data.flags = KVMFR_CLIPBOARD_FLAG_BEGIN |
KVMFR_CLIPBOARD_FLAG_END; KVMFR_CLIPBOARD_FLAG_END;
CHECK(postRecord(state, &data, KVMFR_CLIPBOARD_QUEUE_DATA, NULL)); CHECK(postStreamRecord(state, &data, NULL));
CHECK(waitAtomic(state, &state->events.dataEnd, 2)); CHECK(waitAtomic(state, &state->events.dataEnd, 2));
CHECK(atomic_load(&state->events.dataBegin) == 2); CHECK(atomic_load(&state->events.dataBegin) == 2);
CHECK(state->events.finalSize == 0); CHECK(state->events.finalSize == 0);
@@ -695,114 +826,84 @@ static bool prepareOutbound(TestState * state, uint32_t endpoint,
return true; return true;
} }
static bool checkCommit(TestState * state, uint32_t token, static bool checkDataRecord(TestState * state,
KVMFRClipboardMessage * commit, KVMFRClipboardMessage * slot, KVMFRClipboardMessage * record, LGMPStreamBuffer * buffer,
const void ** data) const void ** data)
{ {
CHECK(readType(state, KVMFR_CLIPBOARD_MESSAGE_COMMIT, commit)); CHECK(readClientStream(state, record, buffer, data));
CHECK(commit->token == token); CHECK(record->type == KVMFR_CLIPBOARD_MESSAGE_DATA);
*slot = *(KVMFRClipboardMessage *) CHECK(record->version == KVMFR_CLIPBOARD_VERSION);
lgmpHostMemPtr(state->grantMemory[token - 1]);
CHECK(slot->type == KVMFR_CLIPBOARD_MESSAGE_DATA);
CHECK(slot->generation == commit->generation);
CHECK(slot->clipboardGeneration == commit->clipboardGeneration);
CHECK(slot->transfer == commit->transfer);
CHECK(slot->offset == commit->offset);
CHECK(slot->size == commit->size);
CHECK(slot->format == commit->format);
CHECK(slot->flags == commit->flags);
CHECK(slot->length == commit->length);
CHECK(slot->sequence == commit->sequence);
*data = (const uint8_t *)lgmpHostMemPtr(
state->grantMemory[token - 1]) + sizeof(*slot);
return true; return true;
} }
static bool checkFileCommit(TestState * state, uint32_t grantToken, static bool checkFileDataRecord(TestState * state,
KVMFRClipboardFileOperation operation, KVMFRClipboardFileOperation operation,
KVMFRClipboardMessage * commit, KVMFRClipboardMessage * slot, KVMFRClipboardMessage * record, LGMPStreamBuffer * buffer,
const void ** data) const void ** data)
{ {
CHECK(readType(state, KVMFR_CLIPBOARD_MESSAGE_COMMIT, commit)); CHECK(readClientStream(state, record, buffer, data));
CHECK(commit->token == grantToken); CHECK(record->type == KVMFR_CLIPBOARD_MESSAGE_FILE_DATA);
*slot = *(KVMFRClipboardMessage *) CHECK(record->version == KVMFR_CLIPBOARD_VERSION);
lgmpHostMemPtr(state->grantMemory[grantToken - 1]); CHECK(record->token == operation);
CHECK(slot->type == KVMFR_CLIPBOARD_MESSAGE_FILE_DATA);
CHECK(slot->generation == commit->generation);
CHECK(slot->clipboardGeneration == commit->clipboardGeneration);
CHECK(slot->transfer == commit->transfer);
CHECK(slot->offset == commit->offset);
CHECK(slot->size == commit->size);
CHECK(slot->format == commit->format);
CHECK(slot->flags == commit->flags);
CHECK(slot->length == commit->length);
CHECK(slot->sequence == commit->sequence);
CHECK(slot->token == operation);
*data = (const uint8_t *)lgmpHostMemPtr(
state->grantMemory[grantToken - 1]) + sizeof(*slot);
return true; return true;
} }
static bool testOutboundGrant(TestState * state) static bool testOutboundStream(TestState * state)
{ {
KVMFRClipboardMessage claimRecord; KVMFRClipboardMessage claimRecord;
CHECK(prepareOutbound(state, 40, &claimRecord, 1)); CHECK(prepareOutbound(state, 40, &claimRecord, 1));
const LG_ClipboardRequest transfer = state->events.requestID; const LG_ClipboardRequest transfer = state->events.requestID;
CHECK(postGrant(state, claimRecord.generation, 1));
CHECK(waitMemory(state, state->grantMemory[0]));
CHECK(state->ops->dataBegin(state->clipboard, transfer, CHECK(state->ops->dataBegin(state->clipboard, transfer,
LG_CLIPBOARD_DATA_TEXT, 5) == LG_CLIPBOARD_RESULT_ACCEPTED); LG_CLIPBOARD_DATA_TEXT, 5) == LG_CLIPBOARD_RESULT_ACCEPTED);
CHECK(noClientData(state)); CHECK(noClientControl(state));
const uint8_t first[] = { 1, 2, 3 }; const uint8_t first[] = { 1, 2, 3 };
CHECK(state->ops->dataChunk(state->clipboard, transfer, CHECK(state->ops->dataChunk(state->clipboard, transfer,
0, first, sizeof(first)) == LG_CLIPBOARD_RESULT_ACCEPTED); 0, first, sizeof(first)) == LG_CLIPBOARD_RESULT_ACCEPTED);
KVMFRClipboardMessage commit; KVMFRClipboardMessage record;
KVMFRClipboardMessage slot; LGMPStreamBuffer buffer;
const void * data; const void * data = NULL;
CHECK(checkCommit(state, 1, &commit, &slot, &data)); CHECK(checkDataRecord(state, &record, &buffer, &data));
CHECK(slot.flags == KVMFR_CLIPBOARD_FLAG_BEGIN); CHECK(record.flags == KVMFR_CLIPBOARD_FLAG_BEGIN);
CHECK(slot.sequence == 0 && slot.offset == 0 && slot.size == 5); CHECK(record.sequence == 0 && record.offset == 0 && record.size == 5);
CHECK(memcmp(data, first, sizeof(first)) == 0); CHECK(memcmp(data, first, sizeof(first)) == 0);
CHECK(releaseClientStream(state, &buffer));
CHECK(postGrant(state, claimRecord.generation, 1));
CHECK(waitMemory(state, state->grantMemory[0]));
const uint8_t second[] = { 4, 5 }; const uint8_t second[] = { 4, 5 };
CHECK(state->ops->dataChunk(state->clipboard, transfer, CHECK(state->ops->dataChunk(state->clipboard, transfer,
3, second, sizeof(second)) == LG_CLIPBOARD_RESULT_ACCEPTED); 3, second, sizeof(second)) == LG_CLIPBOARD_RESULT_ACCEPTED);
CHECK(checkCommit(state, 1, &commit, &slot, &data)); CHECK(checkDataRecord(state, &record, &buffer, &data));
CHECK(slot.flags == 0 && slot.sequence == 1 && slot.offset == 3); CHECK(record.flags == 0 && record.sequence == 1 && record.offset == 3);
CHECK(slot.size == KVMFR_CLIPBOARD_SIZE_UNKNOWN); CHECK(record.size == KVMFR_CLIPBOARD_SIZE_UNKNOWN);
CHECK(memcmp(data, second, sizeof(second)) == 0); CHECK(memcmp(data, second, sizeof(second)) == 0);
CHECK(releaseClientStream(state, &buffer));
CHECK(postGrant(state, claimRecord.generation, 1));
CHECK(waitMemory(state, state->grantMemory[0]));
CHECK(state->ops->dataEnd(state->clipboard, transfer, 5) == CHECK(state->ops->dataEnd(state->clipboard, transfer, 5) ==
LG_CLIPBOARD_RESULT_ACCEPTED); LG_CLIPBOARD_RESULT_ACCEPTED);
CHECK(checkCommit(state, 1, &commit, &slot, &data)); CHECK(checkDataRecord(state, &record, &buffer, &data));
CHECK(slot.flags == KVMFR_CLIPBOARD_FLAG_END); CHECK(record.flags == KVMFR_CLIPBOARD_FLAG_END);
CHECK(slot.sequence == 2 && slot.offset == 5 && slot.size == 5); CHECK(record.sequence == 2 && record.offset == 5 && record.size == 5);
CHECK(slot.length == 0); CHECK(record.length == 0);
CHECK(releaseClientStream(state, &buffer));
KVMFRClipboardMessage requestRecord = KVMFRClipboardMessage requestRecord =
hostRecord(KVMFR_CLIPBOARD_MESSAGE_REQUEST, hostRecord(KVMFR_CLIPBOARD_MESSAGE_REQUEST,
claimRecord.generation); claimRecord.generation);
requestRecord.clipboardGeneration = slot.clipboardGeneration; requestRecord.clipboardGeneration = record.clipboardGeneration;
requestRecord.transfer = KVMFR_CLIPBOARD_TRANSFER_HELPER | 2; requestRecord.transfer = KVMFR_CLIPBOARD_TRANSFER_HELPER | 2;
requestRecord.format = KVMFR_CLIPBOARD_FORMAT_TEXT; requestRecord.format = KVMFR_CLIPBOARD_FORMAT_TEXT;
CHECK(postRecord(state, &requestRecord, CHECK(postRecord(state, &requestRecord,
KVMFR_CLIPBOARD_QUEUE_MESSAGE, NULL)); KVMFR_CLIPBOARD_QUEUE_MESSAGE, NULL));
CHECK(waitAtomic(state, &state->events.request, 2)); CHECK(waitAtomic(state, &state->events.request, 2));
CHECK(postGrant(state, claimRecord.generation, 1));
CHECK(waitMemory(state, state->grantMemory[0]));
CHECK(state->ops->dataBegin(state->clipboard, requestRecord.transfer, CHECK(state->ops->dataBegin(state->clipboard, requestRecord.transfer,
LG_CLIPBOARD_DATA_TEXT, 0) == LG_CLIPBOARD_RESULT_ACCEPTED); LG_CLIPBOARD_DATA_TEXT, 0) == LG_CLIPBOARD_RESULT_ACCEPTED);
CHECK(state->ops->dataEnd(state->clipboard, CHECK(state->ops->dataEnd(state->clipboard,
requestRecord.transfer, 0) == LG_CLIPBOARD_RESULT_ACCEPTED); requestRecord.transfer, 0) == LG_CLIPBOARD_RESULT_ACCEPTED);
CHECK(checkCommit(state, 1, &commit, &slot, &data)); CHECK(checkDataRecord(state, &record, &buffer, &data));
CHECK(slot.flags == (KVMFR_CLIPBOARD_FLAG_BEGIN | CHECK(record.flags == (KVMFR_CLIPBOARD_FLAG_BEGIN |
KVMFR_CLIPBOARD_FLAG_END)); KVMFR_CLIPBOARD_FLAG_END));
CHECK(slot.sequence == 0 && slot.offset == 0 && slot.size == 0); CHECK(record.sequence == 0 && record.offset == 0 && record.size == 0);
CHECK(releaseClientStream(state, &buffer));
return true; return true;
} }
@@ -812,19 +913,40 @@ static bool testBlocked(TestState * state)
CHECK(prepareOutbound(state, 50, &claimRecord, 3)); CHECK(prepareOutbound(state, 50, &claimRecord, 3));
const LG_ClipboardRequest transfer = state->events.requestID; const LG_ClipboardRequest transfer = state->events.requestID;
CHECK(state->ops->dataBegin(state->clipboard, transfer, CHECK(state->ops->dataBegin(state->clipboard, transfer,
LG_CLIPBOARD_DATA_TEXT, 2) == LG_CLIPBOARD_RESULT_ACCEPTED); LG_CLIPBOARD_DATA_TEXT, 5) == LG_CLIPBOARD_RESULT_ACCEPTED);
const uint8_t bytes[] = { 8, 9 }; const uint8_t bytes[] = { 8, 9, 10, 11, 12 };
for (unsigned i = 0; i < KVMFR_CLIPBOARD_STREAM_SLOT_COUNT; ++i)
CHECK(state->ops->dataChunk(state->clipboard, transfer, CHECK(state->ops->dataChunk(state->clipboard, transfer,
0, bytes, sizeof(bytes)) == LG_CLIPBOARD_RESULT_BLOCKED); i, &bytes[i], 1) == LG_CLIPBOARD_RESULT_ACCEPTED);
CHECK(postGrant(state, claimRecord.generation, 1)); CHECK(state->ops->dataChunk(state->clipboard, transfer,
4, &bytes[4], 1) == LG_CLIPBOARD_RESULT_BLOCKED);
KVMFRClipboardMessage record;
LGMPStreamBuffer buffer;
const void * data = NULL;
CHECK(checkDataRecord(state, &record, &buffer, &data));
CHECK(record.sequence == 0 && record.offset == 0 && record.length == 1);
CHECK(record.flags == KVMFR_CLIPBOARD_FLAG_BEGIN);
CHECK(*(const uint8_t *)data == bytes[0]);
CHECK(releaseClientStream(state, &buffer));
CHECK(waitAtomic(state, &state->events.dataReady, 1)); CHECK(waitAtomic(state, &state->events.dataReady, 1));
CHECK(state->events.streamID == transfer); CHECK(state->events.streamID == transfer);
CHECK(state->ops->dataChunk(state->clipboard, transfer, CHECK(state->ops->dataChunk(state->clipboard, transfer,
0, bytes, sizeof(bytes)) == LG_CLIPBOARD_RESULT_ACCEPTED); 4, &bytes[4], 1) == LG_CLIPBOARD_RESULT_ACCEPTED);
KVMFRClipboardMessage commit; for (unsigned i = 1; i < 5; ++i)
KVMFRClipboardMessage slot; {
const void * data; CHECK(checkDataRecord(state, &record, &buffer, &data));
CHECK(checkCommit(state, 1, &commit, &slot, &data)); CHECK(record.sequence == i && record.offset == i &&
record.length == 1);
CHECK(*(const uint8_t *)data == bytes[i]);
CHECK(releaseClientStream(state, &buffer));
}
CHECK(state->ops->dataEnd(state->clipboard, transfer, 5) ==
LG_CLIPBOARD_RESULT_ACCEPTED);
CHECK(checkDataRecord(state, &record, &buffer, &data));
CHECK(record.flags == KVMFR_CLIPBOARD_FLAG_END);
CHECK(record.sequence == 5 && record.offset == 5 && record.size == 5);
CHECK(releaseClientStream(state, &buffer));
KVMFRClipboardMessage requestRecord; KVMFRClipboardMessage requestRecord;
CHECK(offerAndRequest(state, claimRecord.generation, CHECK(offerAndRequest(state, claimRecord.generation,
@@ -839,7 +961,7 @@ static bool testBlocked(TestState * state)
wire.flags = KVMFR_CLIPBOARD_FLAG_BEGIN | wire.flags = KVMFR_CLIPBOARD_FLAG_BEGIN |
KVMFR_CLIPBOARD_FLAG_END; KVMFR_CLIPBOARD_FLAG_END;
wire.length = 1; wire.length = 1;
CHECK(postRecord(state, &wire, KVMFR_CLIPBOARD_QUEUE_DATA, bytes)); CHECK(postStreamRecord(state, &wire, bytes));
CHECK(waitAtomic(state, &state->events.dataChunk, 1)); CHECK(waitAtomic(state, &state->events.dataChunk, 1));
CHECK(atomic_load(&state->events.dataEnd) == 0); CHECK(atomic_load(&state->events.dataEnd) == 0);
CHECK(state->ops->dataReady(state->clipboard, 5001)); CHECK(state->ops->dataReady(state->clipboard, 5001));
@@ -1203,8 +1325,7 @@ static bool testMalformedFileData(TestState * state)
const uint8_t byte = 0; const uint8_t byte = 0;
const unsigned cancellations = const unsigned cancellations =
atomic_load(&state->events.fileCancel); atomic_load(&state->events.fileCancel);
CHECK(postRecord(state, &malformed, CHECK(postStreamRecord(state, &malformed, &byte));
KVMFR_CLIPBOARD_QUEUE_DATA, &byte));
CHECK(waitAtomic(state, &state->events.fileCancel, cancellations + 1)); CHECK(waitAtomic(state, &state->events.fileCancel, cancellations + 1));
CHECK(state->events.fileDataset == dataset); CHECK(state->events.fileDataset == dataset);
CHECK(state->events.fileAcquisition == request.request); CHECK(state->events.fileAcquisition == request.request);
@@ -1217,9 +1338,8 @@ static bool testMalformedFileData(TestState * state)
dataset, request.request, LG_CLIPBOARD_FILE_ERROR_CANCELLED)); dataset, request.request, LG_CLIPBOARD_FILE_ERROR_CANCELLED));
malformed.transfer = request.request + 1U; malformed.transfer = request.request + 1U;
CHECK(postRecord(state, &malformed, CHECK(postStreamRecord(state, &malformed, &byte));
KVMFR_CLIPBOARD_QUEUE_DATA, &byte)); CHECK(settleHostStream(state));
CHECK(waitMemory(state, state->memories[state->memoryCount - 1]));
CHECK(atomic_load(&state->events.fileCancel) == cancellations + 1); CHECK(atomic_load(&state->events.fileCancel) == cancellations + 1);
return true; return true;
} }
@@ -1278,26 +1398,50 @@ static bool testFileStream(TestState * state)
CHECK(state->ops->fileDataBegin(state->clipboard, CHECK(state->ops->fileDataBegin(state->clipboard,
&descriptor, sizeof(bytes)) == LG_CLIPBOARD_RESULT_ACCEPTED); &descriptor, sizeof(bytes)) == LG_CLIPBOARD_RESULT_ACCEPTED);
CHECK(postGrant(state, claimRecord.generation, 1)); for (unsigned i = 0; i < KVMFR_CLIPBOARD_STREAM_SLOT_COUNT; ++i)
CHECK(waitMemory(state, state->grantMemory[0])); {
const uint64_t offset =
(uint64_t)i * KVMFR_CLIPBOARD_STREAM_SLOT_BYTES;
CHECK(state->ops->fileDataChunk(state->clipboard, CHECK(state->ops->fileDataChunk(state->clipboard,
&descriptor, 0, bytes, sizeof(bytes), true) == &descriptor, offset, bytes + offset,
KVMFR_CLIPBOARD_STREAM_SLOT_BYTES,
i + 1 == KVMFR_CLIPBOARD_STREAM_SLOT_COUNT) ==
LG_CLIPBOARD_RESULT_ACCEPTED); LG_CLIPBOARD_RESULT_ACCEPTED);
KVMFRClipboardMessage commit; }
KVMFRClipboardMessage slot; KVMFRClipboardMessage record;
const void * data; LGMPStreamBuffer buffer;
CHECK(checkFileCommit(state, 1, KVMFR_CLIPBOARD_FILE_OP_READ, const void * data = NULL;
&commit, &slot, &data)); for (unsigned i = 0; i < KVMFR_CLIPBOARD_STREAM_SLOT_COUNT; ++i)
CHECK(slot.flags == (KVMFR_CLIPBOARD_FLAG_BEGIN | {
KVMFR_CLIPBOARD_FLAG_END)); const uint64_t offset =
CHECK(slot.offset == 0 && slot.sequence == 0 && (uint64_t)i * KVMFR_CLIPBOARD_STREAM_SLOT_BYTES;
slot.size == sizeof(bytes)); CHECK(checkFileDataRecord(state, KVMFR_CLIPBOARD_FILE_OP_READ,
CHECK(slot.length == sizeof(bytes)); &record, &buffer, &data));
CHECK(memcmp(data, bytes, sizeof(bytes)) == 0); CHECK(record.offset == offset && record.sequence == i);
CHECK(record.length == KVMFR_CLIPBOARD_STREAM_SLOT_BYTES);
if (i == 0)
{
CHECK(record.flags == KVMFR_CLIPBOARD_FLAG_BEGIN);
CHECK(record.size == sizeof(bytes));
}
else if (i + 1 == KVMFR_CLIPBOARD_STREAM_SLOT_COUNT)
{
CHECK(record.flags == KVMFR_CLIPBOARD_FLAG_END);
CHECK(record.size == sizeof(bytes));
}
else
{
CHECK(record.flags == 0);
CHECK(record.size == KVMFR_CLIPBOARD_SIZE_UNKNOWN);
}
CHECK(memcmp(data, bytes + offset,
KVMFR_CLIPBOARD_STREAM_SLOT_BYTES) == 0);
CHECK(releaseClientStream(state, &buffer));
}
CHECK(state->ops->fileDataEnd(state->clipboard, CHECK(state->ops->fileDataEnd(state->clipboard,
&descriptor, sizeof(bytes)) == LG_CLIPBOARD_RESULT_ACCEPTED); &descriptor, sizeof(bytes)) == LG_CLIPBOARD_RESULT_ACCEPTED);
CHECK(noClientData(state)); CHECK(noClientControl(state));
const unsigned requests = atomic_load(&state->events.fileRequest); const unsigned requests = atomic_load(&state->events.fileRequest);
request.transfer = KVMFR_CLIPBOARD_TRANSFER_HELPER | UINT64_C(0x7803); request.transfer = KVMFR_CLIPBOARD_TRANSFER_HELPER | UINT64_C(0x7803);
@@ -1313,16 +1457,15 @@ static bool testFileStream(TestState * state)
CHECK(state->ops->fileDataBegin(state->clipboard, CHECK(state->ops->fileDataBegin(state->clipboard,
&empty, KVMFR_CLIPBOARD_SIZE_UNKNOWN) == &empty, KVMFR_CLIPBOARD_SIZE_UNKNOWN) ==
LG_CLIPBOARD_RESULT_ACCEPTED); LG_CLIPBOARD_RESULT_ACCEPTED);
CHECK(postGrant(state, claimRecord.generation, 1));
CHECK(waitMemory(state, state->grantMemory[0]));
CHECK(state->ops->fileDataEnd(state->clipboard, CHECK(state->ops->fileDataEnd(state->clipboard,
&empty, 0) == LG_CLIPBOARD_RESULT_ACCEPTED); &empty, 0) == LG_CLIPBOARD_RESULT_ACCEPTED);
CHECK(checkFileCommit(state, 1, KVMFR_CLIPBOARD_FILE_OP_LIST, CHECK(checkFileDataRecord(state, KVMFR_CLIPBOARD_FILE_OP_LIST,
&commit, &slot, &data)); &record, &buffer, &data));
CHECK(slot.flags == (KVMFR_CLIPBOARD_FLAG_BEGIN | CHECK(record.flags == (KVMFR_CLIPBOARD_FLAG_BEGIN |
KVMFR_CLIPBOARD_FLAG_END)); KVMFR_CLIPBOARD_FLAG_END));
CHECK(slot.offset == 0 && slot.sequence == 0 && slot.size == 0); CHECK(record.offset == 0 && record.sequence == 0 && record.size == 0);
CHECK(slot.length == 0); CHECK(record.length == 0);
CHECK(releaseClientStream(state, &buffer));
return true; return true;
} }
@@ -1382,6 +1525,13 @@ static bool recreateClipboard(TestState * state)
{ {
state->ops->detach(state->clipboard); state->ops->detach(state->clipboard);
atomic_store(&state->events.attached, false); atomic_store(&state->events.attached, false);
KVMFRClipboardMessage release;
CHECK(readType(state, KVMFR_CLIPBOARD_MESSAGE_RELEASE, &release));
CHECK(release.generation != 0);
const uint32_t endpointGeneration =
atomic_load(&state->statusGeneration);
CHECK(endpointGeneration != 0);
CHECK(postStatus(state, endpointGeneration, 0, 0, true, 0));
} }
CHECK(disconnectAndDrain(state, NULL)); CHECK(disconnectAndDrain(state, NULL));
lgmpClipboard_destroy(&state->clipboard); lgmpClipboard_destroy(&state->clipboard);
@@ -1463,8 +1613,8 @@ static bool testProcessRestart(TestState * state)
staleData.format = KVMFR_CLIPBOARD_FORMAT_TEXT; staleData.format = KVMFR_CLIPBOARD_FORMAT_TEXT;
staleData.flags = KVMFR_CLIPBOARD_FLAG_BEGIN | staleData.flags = KVMFR_CLIPBOARD_FLAG_BEGIN |
KVMFR_CLIPBOARD_FLAG_END; KVMFR_CLIPBOARD_FLAG_END;
CHECK(postRecord(state, &staleData, KVMFR_CLIPBOARD_QUEUE_DATA, NULL)); CHECK(postStreamRecord(state, &staleData, NULL));
CHECK(waitMemory(state, state->memories[state->memoryCount - 1])); CHECK(settleHostStream(state));
CHECK(atomic_load(&state->events.dataBegin) == beginCount); CHECK(atomic_load(&state->events.dataBegin) == beginCount);
KVMFRClipboardMessage data = KVMFRClipboardMessage data =
@@ -1475,7 +1625,7 @@ static bool testProcessRestart(TestState * state)
data.flags = KVMFR_CLIPBOARD_FLAG_BEGIN | data.flags = KVMFR_CLIPBOARD_FLAG_BEGIN |
KVMFR_CLIPBOARD_FLAG_END; KVMFR_CLIPBOARD_FLAG_END;
const unsigned endCount = atomic_load(&state->events.dataEnd); const unsigned endCount = atomic_load(&state->events.dataEnd);
CHECK(postRecord(state, &data, KVMFR_CLIPBOARD_QUEUE_DATA, NULL)); CHECK(postStreamRecord(state, &data, NULL));
CHECK(waitAtomic(state, &state->events.dataEnd, endCount + 1)); CHECK(waitAtomic(state, &state->events.dataEnd, endCount + 1));
CHECK(atomic_load(&state->events.dataBegin) == beginCount + 1); CHECK(atomic_load(&state->events.dataBegin) == beginCount + 1);
@@ -1578,15 +1728,10 @@ static bool testMalformed(TestState * state)
CHECK(atomic_load(&state->events.notice) == 0); CHECK(atomic_load(&state->events.notice) == 0);
malformed.generation = claimRecord.generation; malformed.generation = claimRecord.generation;
CHECK(postRecord(state, &malformed, KVMFR_CLIPBOARD_QUEUE_DATA, NULL)); CHECK(postStreamRecord(state, &malformed, NULL));
CHECK(waitMemory(state, state->memories[state->memoryCount - 1])); CHECK(settleHostStream(state));
CHECK(atomic_load(&state->events.notice) == 0); CHECK(atomic_load(&state->events.notice) == 0);
CHECK(postGrant(state, claimRecord.generation, 1));
CHECK(waitMemory(state, state->grantMemory[0]));
CHECK(postGrant(state, claimRecord.generation, 1));
CHECK(waitMemory(state, state->grantMemory[0]));
KVMFRClipboardMessage requestRecord; KVMFRClipboardMessage requestRecord;
CHECK(offerAndRequest(state, claimRecord.generation, CHECK(offerAndRequest(state, claimRecord.generation,
901, 9001, &requestRecord)); 901, 9001, &requestRecord));
@@ -1599,14 +1744,14 @@ static bool testMalformed(TestState * state)
data.format = KVMFR_CLIPBOARD_FORMAT_TEXT; data.format = KVMFR_CLIPBOARD_FORMAT_TEXT;
data.flags = KVMFR_CLIPBOARD_FLAG_BEGIN | data.flags = KVMFR_CLIPBOARD_FLAG_BEGIN |
KVMFR_CLIPBOARD_FLAG_END; KVMFR_CLIPBOARD_FLAG_END;
CHECK(postRecord(state, &data, KVMFR_CLIPBOARD_QUEUE_DATA, NULL)); CHECK(postStreamRecord(state, &data, NULL));
CHECK(waitMemory(state, state->memories[state->memoryCount - 1])); CHECK(settleHostStream(state));
CHECK(atomic_load(&state->events.dataBegin) == 0); CHECK(atomic_load(&state->events.dataBegin) == 0);
const uint8_t byte = 42; const uint8_t byte = 42;
data.offset = 0; data.offset = 0;
data.length = 1; data.length = 1;
CHECK(postRecord(state, &data, KVMFR_CLIPBOARD_QUEUE_DATA, &byte)); CHECK(postStreamRecord(state, &data, &byte));
CHECK(waitAtomic(state, &state->events.dataEnd, 1)); CHECK(waitAtomic(state, &state->events.dataEnd, 1));
CHECK(atomic_load(&state->events.dataBegin) == 1); CHECK(atomic_load(&state->events.dataBegin) == 1);
CHECK(atomic_load(&state->events.dataChunk) == 1); CHECK(atomic_load(&state->events.dataChunk) == 1);
@@ -1623,7 +1768,7 @@ static bool testFileFormatIsolation(TestState * state)
const LG_ClipboardData files[] = { LG_CLIPBOARD_DATA_FILES }; const LG_ClipboardData files[] = { LG_CLIPBOARD_DATA_FILES };
CHECK(!state->ops->notifyTypes(state->clipboard, files, 1)); CHECK(!state->ops->notifyTypes(state->clipboard, files, 1));
CHECK(noClientData(state)); CHECK(noClientControl(state));
const uint64_t localDataset = UINT64_C(0x123456789abcdef); const uint64_t localDataset = UINT64_C(0x123456789abcdef);
CHECK(state->ops->offerFiles(state->clipboard, localDataset)); CHECK(state->ops->offerFiles(state->clipboard, localDataset));
@@ -1653,7 +1798,7 @@ static bool testFileFormatIsolation(TestState * state)
CHECK(state->events.noticeTypes[0] == LG_CLIPBOARD_DATA_FILES); CHECK(state->events.noticeTypes[0] == LG_CLIPBOARD_DATA_FILES);
CHECK(!state->ops->request(state->clipboard, CHECK(!state->ops->request(state->clipboard,
100, LG_CLIPBOARD_DATA_FILES)); 100, LG_CLIPBOARD_DATA_FILES));
CHECK(noClientData(state)); CHECK(noClientControl(state));
KVMFRClipboardMessage data = KVMFRClipboardMessage data =
hostRecord(KVMFR_CLIPBOARD_MESSAGE_DATA, claimRecord.generation); hostRecord(KVMFR_CLIPBOARD_MESSAGE_DATA, claimRecord.generation);
@@ -1662,8 +1807,8 @@ static bool testFileFormatIsolation(TestState * state)
data.format = KVMFR_CLIPBOARD_FORMAT_FILES; data.format = KVMFR_CLIPBOARD_FORMAT_FILES;
data.flags = KVMFR_CLIPBOARD_FLAG_BEGIN | data.flags = KVMFR_CLIPBOARD_FLAG_BEGIN |
KVMFR_CLIPBOARD_FLAG_END; KVMFR_CLIPBOARD_FLAG_END;
CHECK(postRecord(state, &data, KVMFR_CLIPBOARD_QUEUE_DATA, NULL)); CHECK(postStreamRecord(state, &data, NULL));
CHECK(waitMemory(state, state->memories[state->memoryCount - 1])); CHECK(settleHostStream(state));
CHECK(atomic_load(&state->events.dataBegin) == 0); CHECK(atomic_load(&state->events.dataBegin) == 0);
return true; return true;
} }
@@ -1721,6 +1866,31 @@ static bool stateInit(TestState * state)
CHECK(dataSize == sizeof(sessionData)); CHECK(dataSize == sizeof(sessionData));
CHECK(memcmp(data, &sessionData, sizeof(sessionData)) == 0); CHECK(memcmp(data, &sessionData, sizeof(sessionData)) == 0);
const struct LGMPStreamConfig hostToClientConfig =
{
.direction = LGMP_STREAM_HOST_TO_CLIENT,
.policy = LGMP_STREAM_RELIABLE_FIFO,
.slotCount = KVMFR_CLIPBOARD_STREAM_SLOT_COUNT,
.slotSize = SLOT_BYTES,
};
CHECK(lgmpHostStreamNew(state->host, hostToClientConfig,
&state->hostToClientStream) == LGMP_OK);
struct LGMPStreamDescriptor descriptor;
lgmpHostStreamGetDescriptor(state->hostToClientStream, &descriptor);
state->hostToClientDescriptor = exportStreamDescriptor(&descriptor);
const struct LGMPStreamConfig clientToHostConfig =
{
.direction = LGMP_STREAM_CLIENT_TO_HOST,
.policy = LGMP_STREAM_RELIABLE_FIFO,
.slotCount = KVMFR_CLIPBOARD_STREAM_SLOT_COUNT,
.slotSize = SLOT_BYTES,
};
CHECK(lgmpHostStreamNew(state->host, clientToHostConfig,
&state->clientToHostStream) == LGMP_OK);
lgmpHostStreamGetDescriptor(state->clientToHostStream, &descriptor);
state->clientToHostDescriptor = exportStreamDescriptor(&descriptor);
CHECK(lgmpClipboard_create(state->client, &state->clipboard)); CHECK(lgmpClipboard_create(state->client, &state->clipboard));
CHECK(lgmpClipboard_connect(state->clipboard, state->clientID)); CHECK(lgmpClipboard_connect(state->clipboard, state->clientID));
state->ops = lgmpClipboard_getOps(); state->ops = lgmpClipboard_getOps();
@@ -1753,6 +1923,8 @@ static void stateFree(TestState * state)
lgmpClientFree(&state->client); lgmpClientFree(&state->client);
for (unsigned i = 0; i < state->memoryCount; ++i) for (unsigned i = 0; i < state->memoryCount; ++i)
lgmpHostMemFree(&state->memories[i]); lgmpHostMemFree(&state->memories[i]);
lgmpHostStreamFree(&state->clientToHostStream);
lgmpHostStreamFree(&state->hostToClientStream);
lgmpHostFree(&state->host); lgmpHostFree(&state->host);
if (state->memory != MAP_FAILED) if (state->memory != MAP_FAILED)
munmap(state->memory, TEST_SHM_SIZE); munmap(state->memory, TEST_SHM_SIZE);
@@ -1770,7 +1942,7 @@ TESTS[] =
{ "claim" , testClaim }, { "claim" , testClaim },
{ "offer-request" , testOfferRequest }, { "offer-request" , testOfferRequest },
{ "inbound-stream" , testInboundStream }, { "inbound-stream" , testInboundStream },
{ "outbound-grant" , testOutboundGrant }, { "outbound-stream" , testOutboundStream },
{ "blocked" , testBlocked }, { "blocked" , testBlocked },
{ "cancel" , testCancel }, { "cancel" , testCancel },
{ "restart" , testRestart }, { "restart" , testRestart },

View File

@@ -26,6 +26,7 @@
#include <lgmp/client.h> #include <lgmp/client.h>
#include <lgmp/host.h> #include <lgmp/host.h>
#include <lgmp/stream.h>
#include <linux/input.h> #include <linux/input.h>
#include <pthread.h> #include <pthread.h>
@@ -58,6 +59,11 @@ typedef struct TestState
PLGMPHost host; PLGMPHost host;
PLGMPHostQueue queue; PLGMPHostQueue queue;
PLGMPClient client; PLGMPClient client;
PLGMPHostStream streams[KVMFR_INPUT_STREAM_ENDPOINT_COUNT];
KVMFRStreamDescriptor streamDescriptors[
KVMFR_INPUT_STREAM_ENDPOINT_COUNT];
uint32_t streamEpoch;
uint32_t streamGeneration;
LGMPInput * input; LGMPInput * input;
const LG_InputOps * ops; const LG_InputOps * ops;
uint32_t clientID; uint32_t clientID;
@@ -104,6 +110,23 @@ static bool hostProcess(TestState * state)
return true; return true;
} }
static KVMFRStreamDescriptor exportStreamDescriptor(
const struct LGMPStreamDescriptor * descriptor)
{
return (KVMFRStreamDescriptor)
{
.magic = descriptor->magic,
.version = descriptor->version,
.size = descriptor->size,
.offset = descriptor->offset,
.regionSize = descriptor->regionSize,
.direction = descriptor->direction,
.policy = descriptor->policy,
.slotCount = descriptor->slotCount,
.slotSize = descriptor->slotSize,
};
}
static bool waitStatus(TestState * state, unsigned count, static bool waitStatus(TestState * state, unsigned count,
bool available, uint32_t generation) bool available, uint32_t generation)
{ {
@@ -160,8 +183,22 @@ static bool postStatus(TestState * state, uint32_t endpointGeneration,
.ownerGeneration = ownerGeneration, .ownerGeneration = ownerGeneration,
.lease = 1000, .lease = 1000,
.maxButtons = KVMFR_INPUT_MOUSE_BUTTON_COUNT, .maxButtons = KVMFR_INPUT_MOUSE_BUTTON_COUNT,
.streamVersion = KVMFR_INPUT_STREAM_VERSION,
.streamEndpointCount = KVMFR_INPUT_STREAM_ENDPOINT_COUNT,
.streamGeneration = state->streamGeneration,
}; };
for (unsigned i = 0; i < KVMFR_INPUT_STREAM_ENDPOINT_COUNT; ++i)
{
status->streamEndpoint[i].stream = state->streamDescriptors[i];
status->streamEndpoint[i].flags =
KVMFR_INPUT_STREAM_ENDPOINT_AVAILABLE;
}
status->streamEndpoint[0].boundClientID = state->clientID;
status->streamEndpoint[0].bindingGeneration = state->streamEpoch;
status->streamEndpoint[0].flags |=
KVMFR_INPUT_STREAM_ENDPOINT_BOUND;
if (++state->statusSerial == 0) if (++state->statusSerial == 0)
++state->statusSerial; ++state->statusSerial;
CHECK(lgmpHostQueuePost(state->queue, state->statusSerial, memory) == CHECK(lgmpHostQueuePost(state->queue, state->statusSerial, memory) ==
@@ -186,28 +223,42 @@ static bool postOwner(TestState * state, uint32_t endpointGeneration,
return true; return true;
} }
static bool expectInputQueueEmpty(TestState * state)
{
KVMFRInputMessage message;
size_t size = sizeof(message);
uint32_t sourceClientID = 0;
const LGMP_STATUS status = lgmpHostReadDataWithSource(state->queue,
&message, &size, &sourceClientID);
if (status == LGMP_OK)
lgmpHostAckData(state->queue);
CHECK(status == LGMP_ERR_QUEUE_EMPTY);
return true;
}
static bool readInput(TestState * state, KVMFRInputMessage * result) static bool readInput(TestState * state, KVMFRInputMessage * result)
{ {
for (unsigned i = 0; i < TEST_WAIT_MS; ++i) for (unsigned i = 0; i < TEST_WAIT_MS; ++i)
{ {
CHECK(hostProcess(state)); CHECK(hostProcess(state));
size_t size = sizeof(*result); CHECK(expectInputQueueEmpty(state));
uint32_t sourceClientID = 0; LGMPStreamBuffer buffer = { 0 };
const LGMP_STATUS status = lgmpHostReadDataWithSource(state->queue, const LGMP_STATUS status = lgmpHostStreamReadPeek(
result, &size, &sourceClientID); state->streams[0], &buffer);
if (status == LGMP_ERR_QUEUE_EMPTY) if (status == LGMP_ERR_STREAM_EMPTY)
{ {
usleep(1000); usleep(1000);
continue; continue;
} }
CHECK(status == LGMP_OK); CHECK(status == LGMP_OK);
CHECK(size == sizeof(*result)); CHECK(buffer.size == sizeof(*result));
CHECK(sourceClientID == state->clientID); memcpy(result, buffer.data, sizeof(*result));
CHECK(result->reserved == 0); CHECK(result->reserved == 0);
CHECK(result->generation != 0); CHECK(result->generation != 0);
CHECK(result->sequence != 0); CHECK(result->sequence != 0);
CHECK(lgmpHostAckData(state->queue) == LGMP_OK); CHECK(lgmpHostStreamReadRelease(state->streams[0], &buffer) ==
LGMP_OK);
return true; return true;
} }
@@ -247,20 +298,19 @@ static bool expectNoInput(TestState * state)
for (unsigned i = 0; i < TEST_QUIET_MS; ++i) for (unsigned i = 0; i < TEST_QUIET_MS; ++i)
{ {
CHECK(hostProcess(state)); CHECK(hostProcess(state));
KVMFRInputMessage message; CHECK(expectInputQueueEmpty(state));
size_t size = sizeof(message); LGMPStreamBuffer buffer = { 0 };
uint32_t sourceClientID = 0; const LGMP_STATUS status = lgmpHostStreamReadPeek(
const LGMP_STATUS status = lgmpHostReadDataWithSource(state->queue, state->streams[0], &buffer);
&message, &size, &sourceClientID); if (status == LGMP_ERR_STREAM_EMPTY)
if (status == LGMP_ERR_QUEUE_EMPTY)
{ {
usleep(1000); usleep(1000);
continue; continue;
} }
if (status == LGMP_OK) if (status == LGMP_OK)
lgmpHostAckData(state->queue); lgmpHostStreamReadRelease(state->streams[0], &buffer);
CHECK(status == LGMP_ERR_QUEUE_EMPTY); CHECK(status == LGMP_ERR_STREAM_EMPTY);
} }
return true; return true;
@@ -297,6 +347,10 @@ static bool resetAndRelease(TestState * state, uint32_t generation)
CHECK(readUntilType(state, KVMFR_INPUT_MESSAGE_RELEASE, 8, CHECK(readUntilType(state, KVMFR_INPUT_MESSAGE_RELEASE, 8,
&message, &skipped)); &message, &skipped));
CHECK(message.generation == generation); CHECK(message.generation == generation);
const uint32_t endpointGeneration =
atomic_load(&state->status.generation);
CHECK(endpointGeneration != 0);
CHECK(postOwner(state, endpointGeneration, 0, 0));
return true; return true;
} }
@@ -512,6 +566,25 @@ static bool stateInit(TestState * state)
CHECK(dataSize == sizeof(sessionData)); CHECK(dataSize == sizeof(sessionData));
CHECK(memcmp(data, &sessionData, sizeof(sessionData)) == 0); CHECK(memcmp(data, &sessionData, sizeof(sessionData)) == 0);
const struct LGMPStreamConfig streamConfig =
{
.direction = LGMP_STREAM_CLIENT_TO_HOST,
.policy = LGMP_STREAM_RELIABLE_FIFO,
.slotCount = KVMFR_INPUT_STREAM_SLOT_COUNT,
.slotSize = KVMFR_INPUT_STREAM_SLOT_SIZE,
};
for (unsigned i = 0; i < KVMFR_INPUT_STREAM_ENDPOINT_COUNT; ++i)
{
CHECK(lgmpHostStreamNew(state->host, streamConfig,
&state->streams[i]) == LGMP_OK);
struct LGMPStreamDescriptor descriptor;
lgmpHostStreamGetDescriptor(state->streams[i], &descriptor);
state->streamDescriptors[i] = exportStreamDescriptor(&descriptor);
}
CHECK(lgmpHostStreamBind(state->streams[0], state->clientID,
&state->streamEpoch) == LGMP_OK);
state->streamGeneration = 1;
CHECK(lgmpInput_create(state->client, &state->input)); CHECK(lgmpInput_create(state->client, &state->input));
CHECK(lgmpInput_connect(state->input, state->clientID)); CHECK(lgmpInput_connect(state->input, state->clientID));
state->ops = lgmpInput_getOps(); state->ops = lgmpInput_getOps();
@@ -559,20 +632,21 @@ static bool disconnectAndDrain(TestState * state)
} }
bool valid = true; bool valid = true;
bool releaseAcknowledged = false;
for (unsigned i = 0; for (unsigned i = 0;
i < TEST_WAIT_MS && !atomic_load(&task.done); ++i) i < TEST_WAIT_MS && !atomic_load(&task.done); ++i)
{ {
if (lgmpHostProcess(state->host) != LGMP_OK) if (lgmpHostProcess(state->host) != LGMP_OK)
valid = false; valid = false;
if (!expectInputQueueEmpty(state))
valid = false;
for (;;) for (;;)
{ {
KVMFRInputMessage message; LGMPStreamBuffer buffer = { 0 };
size_t size = sizeof(message); const LGMP_STATUS status = lgmpHostStreamReadPeek(
uint32_t sourceClientID = 0; state->streams[0], &buffer);
const LGMP_STATUS status = lgmpHostReadDataWithSource(state->queue, if (status == LGMP_ERR_STREAM_EMPTY)
&message, &size, &sourceClientID);
if (status == LGMP_ERR_QUEUE_EMPTY)
break; break;
if (status != LGMP_OK) if (status != LGMP_OK)
{ {
@@ -580,12 +654,27 @@ static bool disconnectAndDrain(TestState * state)
break; break;
} }
valid &= size == sizeof(message); KVMFRInputMessage message;
valid &= sourceClientID == state->clientID; valid &= buffer.size == sizeof(message);
if (buffer.size == sizeof(message))
memcpy(&message, buffer.data, sizeof(message));
else
memset(&message, 0, sizeof(message));
valid &= message.type == KVMFR_INPUT_MESSAGE_RELEASE; valid &= message.type == KVMFR_INPUT_MESSAGE_RELEASE;
valid &= message.reserved == 0; valid &= message.reserved == 0;
if (lgmpHostAckData(state->queue) != LGMP_OK) if (lgmpHostStreamReadRelease(state->streams[0], &buffer) !=
LGMP_OK)
valid = false; valid = false;
if (!releaseAcknowledged &&
message.type == KVMFR_INPUT_MESSAGE_RELEASE)
{
const uint32_t endpointGeneration =
atomic_load(&state->status.generation);
if (endpointGeneration && !postStatus(
state, endpointGeneration, 0, 0))
valid = false;
releaseAcknowledged = true;
}
} }
usleep(1000); usleep(1000);
} }
@@ -604,6 +693,8 @@ static bool stateFree(TestState * state)
lgmpClientFree(&state->client); lgmpClientFree(&state->client);
for (unsigned i = 0; i < state->statusCount; ++i) for (unsigned i = 0; i < state->statusCount; ++i)
lgmpHostMemFree(&state->statuses[i]); lgmpHostMemFree(&state->statuses[i]);
for (unsigned i = 0; i < KVMFR_INPUT_STREAM_ENDPOINT_COUNT; ++i)
lgmpHostStreamFree(&state->streams[i]);
lgmpHostFree(&state->host); lgmpHostFree(&state->host);
if (state->memory != MAP_FAILED) if (state->memory != MAP_FAILED)
munmap(state->memory, TEST_SHM_SIZE); munmap(state->memory, TEST_SHM_SIZE);