toaruos/userspace/lib/yutani.c

#include <string.h>
#include <stdlib.h>
#include <syscall.h>

#include "yutani.h"
#include "pex.h"
#include "graphics.h"

yutani_msg_t * yutani_wait_for(yutani_t * y, uint32_t type) {
	do {
		yutani_msg_t * out;
		size_t size;
		{
			char tmp[MAX_PACKET_SIZE];
			size = pex_recv(y->sock, tmp);
			out = malloc(size);
			memcpy(out, tmp, size);
		}

		if (out->type == type) {
			return out;
		} else {
			/* XXX: Add other messages to a queue to be handled later. */
			free(out);
		}
	} while (1); /* XXX: (!y->abort) */
}

yutani_msg_t * yutani_msg_build_hello(void) {
	size_t s = sizeof(struct yutani_message);
	yutani_msg_t * msg = malloc(s); /* No extra data for a hello. */

	msg->magic = YUTANI_MSG__MAGIC;
	msg->type  = YUTANI_MSG_HELLO;
	msg->size  = s;

	return msg;
}

yutani_msg_t * yutani_msg_build_flip(void) {
	size_t s = sizeof(struct yutani_message);
	yutani_msg_t * msg = malloc(s); /* No extra data for a hello. */

	msg->magic = YUTANI_MSG__MAGIC;
	msg->type  = YUTANI_MSG_FLIP;
	msg->size  = s;

	return msg;
}

yutani_msg_t * yutani_msg_build_welcome(uint32_t width, uint32_t height) {
	size_t s = sizeof(struct yutani_message) + sizeof(struct yutani_msg_welcome);
	yutani_msg_t * msg = malloc(s); /* No extra data for a hello. */

	msg->magic = YUTANI_MSG__MAGIC;
	msg->type  = YUTANI_MSG_WELCOME;
	msg->size  = s;

	struct yutani_msg_welcome * mw = (void *)msg->data;

	mw->display_width = width;
	mw->display_height = height;

	return msg;
}

yutani_msg_t * yutani_msg_build_window_new(uint32_t width, uint32_t height) {
	size_t s = sizeof(struct yutani_message) + sizeof(struct yutani_msg_window_new);
	yutani_msg_t * msg = malloc(s); /* No extra data for a hello. */

	msg->magic = YUTANI_MSG__MAGIC;
	msg->type  = YUTANI_MSG_WINDOW_NEW;
	msg->size  = s;

	struct yutani_msg_window_new * mw = (void *)msg->data;

	mw->width = width;
	mw->height = height;

	return msg;
}

yutani_msg_t * yutani_msg_build_window_init(uint32_t width, uint32_t height, uint32_t bufid) {
	size_t s = sizeof(struct yutani_message) + sizeof(struct yutani_msg_window_init);
	yutani_msg_t * msg = malloc(s); /* No extra data for a hello. */

	msg->magic = YUTANI_MSG__MAGIC;
	msg->type  = YUTANI_MSG_WINDOW_INIT;
	msg->size  = s;

	struct yutani_msg_window_init * mw = (void *)msg->data;

	mw->width = width;
	mw->height = height;
	mw->bufid = bufid;

	return msg;
}

int yutani_msg_send(yutani_t * y, yutani_msg_t * msg) {
	return pex_reply(y->sock, msg->size, (char *)msg);
}

yutani_t * yutani_context_create(FILE * socket) {
	yutani_t * out = malloc(sizeof(yutani_t));

	out->sock = socket;
	out->display_width  = 0;
	out->display_height = 0;
	return out;
}

yutani_t * yutani_init(void) {
	/* XXX: Display, etc? */
	FILE * c = pex_connect("compositor");

	if (!c) return NULL; /* Connection failed. */

	fprintf(stderr, "[yutani-client] Sending hello...\n");
	yutani_t * y = yutani_context_create(c);
	yutani_msg_t * m = yutani_msg_build_hello();
	int result = yutani_msg_send(y, m);
	free(m);

	fprintf(stderr, "[yutani-client] Waiting for welcome...\n");
	m = yutani_wait_for(y, YUTANI_MSG_WELCOME);
	struct yutani_msg_welcome * mw = (void *)&m->data;
	y->display_width = mw->display_width;
	y->display_height = mw->display_height;
	free(m);

	fprintf(stderr, "[yutani-client] Received welcome with server size of: %dx%d\n", y->display_width, y->display_height);

	return y;
}

yutani_window_t * yutani_window_create(yutani_t * y, int width, int height) {
	yutani_window_t * win = malloc(sizeof(yutani_window_t));

	yutani_msg_t * m = yutani_msg_build_window_new(width, height);
	int result = yutani_msg_send(y, m);

	fprintf(stderr, "[yutani-client] Waiting for window response...\n");
	m = yutani_wait_for(y, YUTANI_MSG_WINDOW_INIT);
	struct yutani_msg_window_init * mw = (void *)&m->data;
	fprintf(stderr, "[yutani-client] Received response, buffer ID is %d\n", mw->bufid);

	win->width = mw->width;
	win->height = mw->height;
	win->bufid = mw->bufid;

	char key[1024];
	YUTANI_SHMKEY(key, 1024, win);

	size_t size = (width * height * 4);
	win->buffer = (uint8_t *)syscall_shm_obtain(key, &size);
	return win;
}

void yutani_flip(yutani_t * y) {
	yutani_msg_t * m = yutani_msg_build_flip();
	int result = yutani_msg_send(y, m);
}

gfx_context_t * init_graphics_yutani(yutani_window_t * window) {
	gfx_context_t * out = malloc(sizeof(gfx_context_t));
	out->width  = window->width;
	out->height = window->height;
	out->depth  = 32;
	out->size   = GFX_H(out) * GFX_W(out) * GFX_B(out);
	out->buffer = window->buffer;
	out->backbuffer = out->buffer;
	return out;
}

gfx_context_t *  init_graphics_yutani_double_buffer(yutani_window_t * window) {
	gfx_context_t * out = init_graphics_yutani(window);
	out->backbuffer = malloc(sizeof(uint32_t) * GFX_W(out) * GFX_H(out));
	return out;
}