mirror of
https://github.com/Droid-Deck/DroidDeck.git
synced 2026-11-11 02:51:10 +00:00
81 lines
4.8 KiB
C++
81 lines
4.8 KiB
C++
#include "../src/xr_gpu_channel.h"
|
|
#include <cassert>
|
|
#include <cstdio>
|
|
#include <dirent.h>
|
|
|
|
static unsigned descriptors() {
|
|
DIR *directory = opendir("/proc/self/fd"); assert(directory);
|
|
unsigned count = 0; while (readdir(directory)) count++; closedir(directory); return count;
|
|
}
|
|
|
|
int main() {
|
|
assert(xr_gpu_dimensions(3360, 1760));
|
|
assert(!xr_gpu_dimensions(3, 4) && !xr_gpu_dimensions(8192, 4096));
|
|
int pair[2]; assert(!socketpair(AF_UNIX, SOCK_SEQPACKET | SOCK_CLOEXEC, 0, pair));
|
|
int fd = memfd_create("xr-gpu-test", MFD_CLOEXEC); assert(fd >= 0 && !ftruncate(fd, 4096));
|
|
XrGpuPoolInfo pool; int sent[3]{fd, fd, fd}; std::array<int, 3> received{{-1, -1, -1}};
|
|
assert(xr_gpu_send(pair[0], &pool, sizeof(pool), sent, 3));
|
|
assert(xr_gpu_receive(pair[1], &pool, sizeof(pool), received.data(), 3));
|
|
for (int handle : received) { assert(fcntl(handle, F_GETFD) & FD_CLOEXEC); close(handle); }
|
|
auto count = descriptors();
|
|
assert(xr_gpu_send(pair[0], &pool, sizeof(pool), sent, 3));
|
|
assert(!xr_gpu_receive(pair[1], &pool, sizeof(pool)) && descriptors() == count);
|
|
assert(xr_gpu_send(pair[0], &pool, sizeof(pool), sent, 2));
|
|
assert(!xr_gpu_receive(pair[1], &pool, sizeof(pool), received.data(), 3) && descriptors() == count);
|
|
assert(xr_gpu_send(pair[0], &pool, sizeof(pool), sent, 3));
|
|
assert(!xr_gpu_receive(pair[1], &pool, sizeof(pool) - 1, received.data(), 3) && descriptors() == count);
|
|
close(pair[0]); close(pair[1]);
|
|
|
|
char directory[] = "/tmp/dd-xr-gpu-XXXXXX"; assert(mkdtemp(directory));
|
|
std::string filename = std::string(directory) + "/gpu";
|
|
int listener = socket(AF_UNIX, SOCK_SEQPACKET | SOCK_CLOEXEC, 0); assert(listener >= 0);
|
|
sockaddr_un address{}; address.sun_family = AF_UNIX; strcpy(address.sun_path, filename.c_str());
|
|
assert(!bind(listener, reinterpret_cast<sockaddr *>(&address), sizeof(address)) && !listen(listener, 1));
|
|
for (unsigned attempt = 0; attempt < 3; attempt++) {
|
|
std::thread server([&] {
|
|
int peer = accept4(listener, nullptr, nullptr, SOCK_CLOEXEC); assert(peer >= 0 && xr_frame_peer(peer));
|
|
XrGpuRequest request; assert(xr_gpu_wait(peer, 1000) && xr_gpu_receive(peer, &request, sizeof(request)));
|
|
assert(request.width == 32 && request.height == 32);
|
|
XrGpuPoolInfo reply; reply.width = reply.height = 32;
|
|
for (auto &size : reply.sizes) size = attempt == 2 ? 4095 : 4096;
|
|
assert(xr_gpu_send(peer, &reply, sizeof(reply), sent, 3));
|
|
if (attempt != 2) {
|
|
XrGpuFrame frame; assert(xr_gpu_wait(peer, 1000) && xr_gpu_receive(peer, &frame, sizeof(frame)));
|
|
assert(frame.slot == 2 && frame.header.frame_id == 7);
|
|
auto now = xr_frame_now();
|
|
frame.header.captured_ns = now;
|
|
assert(xr_gpu_frame_state(frame, 32, 32, 0, now) == XrGpuFrameState::Ready);
|
|
assert(xr_gpu_frame_state(frame, 32, 32, 7, now) == XrGpuFrameState::Invalid);
|
|
frame.header.captured_ns = now - 1000000000LL;
|
|
assert(xr_gpu_frame_state(frame, 32, 32, 0, now) == XrGpuFrameState::Stale);
|
|
frame.slot = 3;
|
|
assert(xr_gpu_frame_state(frame, 32, 32, 0, now) == XrGpuFrameState::Invalid);
|
|
if (!attempt) std::this_thread::sleep_for(std::chrono::milliseconds(700));
|
|
uint64_t ack = attempt ? 8 : 7;
|
|
assert(xr_gpu_send(peer, &ack, sizeof(ack)));
|
|
}
|
|
close(peer);
|
|
});
|
|
XrGpuSender sender; received.fill(-1);
|
|
bool connected = sender.Connect(filename.c_str(), 32, 32, pool, received);
|
|
assert(connected == (attempt != 2));
|
|
if (connected) {
|
|
for (int handle : received) close(handle);
|
|
DroidDeckXrFrameHeader header{};
|
|
header.magic = DROIDDECK_XR_FRAME_MAGIC; header.version = DROIDDECK_XR_FRAME_VERSION;
|
|
header.width = header.height = 32; header.bytes = 4096; header.color_space = DROIDDECK_XR_COLOR_SRGB_RGBA8;
|
|
header.frame_id = 7; header.captured_ns = xr_frame_now();
|
|
for (int eye = 0; eye < 2; eye++) {
|
|
for (int axis = 0; axis < 3; axis++) header.eye_pose[eye][axis][axis] = 1;
|
|
header.projection[eye][0][0] = header.projection[eye][1][1] = 1;
|
|
header.projection[eye][3][2] = -1;
|
|
}
|
|
assert(sender.Send(header, 2) == (attempt == 0));
|
|
if (attempt) assert(!sender.Send(header, 2));
|
|
} else for (int handle : received) assert(handle == -1);
|
|
server.join();
|
|
}
|
|
close(fd); close(listener); unlink(filename.c_str()); rmdir(directory);
|
|
puts("XR GPU transport passed: atomic FD ownership, malformed pool rejection, truncation cleanup and exact completion acknowledgement");
|
|
}
|