mirror of
https://github.com/gyurix1968/guacamole-client.git
synced 2025-09-06 21:27:40 +00:00
A bit more custom-client framework.
This commit is contained in:
@@ -1,152 +1,11 @@
|
||||
#include <stdlib.h>
|
||||
#include <stdio.h>
|
||||
#include <string.h>
|
||||
#include <png.h>
|
||||
|
||||
#include <rfb/rfbclient.h>
|
||||
|
||||
#include "guacio.h"
|
||||
#include "protocol.h"
|
||||
#include "client.h"
|
||||
|
||||
char __guac_password[] = "potato";
|
||||
|
||||
char* __GUAC_VNC_TAG_IO = "GUACIO";
|
||||
char* __GUAC_VNC_TAG_PNG_BUFFER = "PNG_BUFFER";
|
||||
char* __GUAC_VNC_TAG_PNG_BUFFER_ALPHA = "PNG_BUFFER_ALPHA";
|
||||
|
||||
void guac_vnc_cursor(rfbClient* client, int x, int y, int w, int h, int bpp) {
|
||||
|
||||
int dx, dy;
|
||||
|
||||
GUACIO* io = rfbClientGetClientData(client, __GUAC_VNC_TAG_IO);
|
||||
png_byte** png_buffer = rfbClientGetClientData(client, __GUAC_VNC_TAG_PNG_BUFFER_ALPHA);
|
||||
png_byte* row;
|
||||
|
||||
png_byte** png_row_current = png_buffer;
|
||||
|
||||
unsigned int bytesPerRow = bpp * w;
|
||||
unsigned char* fb_row_current = client->rcSource;
|
||||
unsigned char* fb_mask = client->rcMask;
|
||||
unsigned char* fb_row;
|
||||
unsigned int v;
|
||||
|
||||
/* Copy image data from VNC client to PNG */
|
||||
for (dy = 0; dy<h; dy++) {
|
||||
|
||||
row = *(png_row_current++);
|
||||
|
||||
fb_row = fb_row_current;
|
||||
fb_row_current += bytesPerRow;
|
||||
|
||||
for (dx = 0; dx<w; dx++) {
|
||||
|
||||
switch (bpp) {
|
||||
case 4:
|
||||
v = *((unsigned int*) fb_row);
|
||||
break;
|
||||
|
||||
case 2:
|
||||
v = *((unsigned short*) fb_row);
|
||||
break;
|
||||
|
||||
default:
|
||||
v = *((unsigned char*) fb_row);
|
||||
}
|
||||
|
||||
*(row++) = (v >> client->format.redShift) * 256 / (client->format.redMax+1);
|
||||
*(row++) = (v >> client->format.greenShift) * 256 / (client->format.greenMax+1);
|
||||
*(row++) = (v >> client->format.blueShift) * 256 / (client->format.blueMax+1);
|
||||
|
||||
/* Handle mask */
|
||||
if (*(fb_mask++))
|
||||
*(row++) = 255;
|
||||
else
|
||||
*(row++) = 0;
|
||||
|
||||
fb_row += bpp;
|
||||
|
||||
}
|
||||
}
|
||||
|
||||
/* SEND CURSOR */
|
||||
guac_send_cursor(io, x, y, png_buffer, w, h);
|
||||
guac_flush(io);
|
||||
|
||||
}
|
||||
void guac_vnc_update(rfbClient* client, int x, int y, int w, int h) {
|
||||
|
||||
int dx, dy;
|
||||
|
||||
GUACIO* io = rfbClientGetClientData(client, __GUAC_VNC_TAG_IO);
|
||||
png_byte** png_buffer = rfbClientGetClientData(client, __GUAC_VNC_TAG_PNG_BUFFER);
|
||||
png_byte* row;
|
||||
|
||||
png_byte** png_row_current = png_buffer;
|
||||
|
||||
unsigned int bpp = client->format.bitsPerPixel/8;
|
||||
unsigned int bytesPerRow = bpp * client->width;
|
||||
unsigned char* fb_row_current = client->frameBuffer + (y * bytesPerRow) + (x * bpp);
|
||||
unsigned char* fb_row;
|
||||
unsigned int v;
|
||||
|
||||
/* Copy image data from VNC client to PNG */
|
||||
for (dy = y; dy<y+h; dy++) {
|
||||
|
||||
row = *(png_row_current++);
|
||||
|
||||
fb_row = fb_row_current;
|
||||
fb_row_current += bytesPerRow;
|
||||
|
||||
for (dx = x; dx<x+w; dx++) {
|
||||
|
||||
switch (bpp) {
|
||||
case 4:
|
||||
v = *((unsigned int*) fb_row);
|
||||
break;
|
||||
|
||||
case 2:
|
||||
v = *((unsigned short*) fb_row);
|
||||
break;
|
||||
|
||||
default:
|
||||
v = *((unsigned char*) fb_row);
|
||||
}
|
||||
|
||||
*(row++) = (v >> client->format.redShift) * 256 / (client->format.redMax+1);
|
||||
*(row++) = (v >> client->format.greenShift) * 256 / (client->format.greenMax+1);
|
||||
*(row++) = (v >> client->format.blueShift) * 256 / (client->format.blueMax+1);
|
||||
|
||||
fb_row += bpp;
|
||||
|
||||
}
|
||||
}
|
||||
|
||||
guac_send_png(io, x, y, png_buffer, w, h);
|
||||
guac_flush(io);
|
||||
|
||||
}
|
||||
|
||||
void guac_vnc_copyrect(rfbClient* client, int src_x, int src_y, int w, int h, int dest_x, int dest_y) {
|
||||
|
||||
GUACIO* io = rfbClientGetClientData(client, __GUAC_VNC_TAG_IO);
|
||||
|
||||
guac_send_copy(io, src_x, src_y, w, h, dest_x, dest_y);
|
||||
guac_flush(io);
|
||||
|
||||
}
|
||||
|
||||
char* guac_vnc_get_password(rfbClient* client) {
|
||||
|
||||
/* Freed after use by libvncclient */
|
||||
char* password = malloc(64);
|
||||
strncpy(password, __guac_password, 63);
|
||||
|
||||
fprintf(stderr, "Sending password: %s\n", password);
|
||||
|
||||
return password;
|
||||
}
|
||||
|
||||
|
||||
png_byte** guac_alloc_png_buffer(int w, int h, int bpp) {
|
||||
|
||||
@@ -175,98 +34,64 @@ void guac_free_png_buffer(png_byte** png_buffer, int h) {
|
||||
|
||||
}
|
||||
|
||||
void client(int client_fd) {
|
||||
guac_client* guac_get_client(int client_fd, void (*client_init)(guac_client* client)) {
|
||||
|
||||
char* hostname;
|
||||
guac_client* client = malloc(sizeof(guac_client));
|
||||
|
||||
client->io = guac_open(client_fd);
|
||||
|
||||
client_init(client);
|
||||
|
||||
return client;
|
||||
|
||||
}
|
||||
|
||||
void guac_free_client(guac_client* client) {
|
||||
if (client->free_handler)
|
||||
client->free_handler(client);
|
||||
|
||||
guac_close(client->io);
|
||||
|
||||
free(client);
|
||||
}
|
||||
|
||||
|
||||
void guac_start_client(guac_client* client) {
|
||||
|
||||
GUACIO* io = client->io;
|
||||
int wait_result;
|
||||
rfbClient* rfb_client;
|
||||
|
||||
png_byte** png_buffer;
|
||||
png_byte** png_buffer_alpha;
|
||||
|
||||
GUACIO* io = guac_open(client_fd);
|
||||
|
||||
/*** INIT ***/
|
||||
|
||||
rfb_client = rfbGetClient(8, 3, 4); /* 32-bpp client */
|
||||
|
||||
/* Framebuffer update handler */
|
||||
rfb_client->GotFrameBufferUpdate = guac_vnc_update;
|
||||
/*rfb_client->GotCopyRect = guac_vnc_copyrect;*/
|
||||
|
||||
/* Enable client-side cursor */
|
||||
rfb_client->GotCursorShape = guac_vnc_cursor;
|
||||
rfb_client->appData.useRemoteCursor = TRUE;
|
||||
|
||||
/* Password */
|
||||
rfb_client->GetPassword = guac_vnc_get_password;
|
||||
|
||||
hostname = malloc(64);
|
||||
strcpy(hostname, "localhost");
|
||||
|
||||
rfb_client->serverHost = hostname;
|
||||
rfb_client->serverPort = 5902;
|
||||
|
||||
if (rfbInitClient(rfb_client, NULL, NULL)) {
|
||||
fprintf(stderr, "SUCCESS.\n");
|
||||
}
|
||||
|
||||
png_buffer = guac_alloc_png_buffer(rfb_client->width, rfb_client->height, 3); /* No-alpha */
|
||||
png_buffer_alpha = guac_alloc_png_buffer(rfb_client->width, rfb_client->height, 4); /* With alpha */
|
||||
|
||||
/* Store Guac data in client */
|
||||
rfbClientSetClientData(rfb_client, __GUAC_VNC_TAG_IO, io);
|
||||
rfbClientSetClientData(rfb_client, __GUAC_VNC_TAG_PNG_BUFFER, png_buffer);
|
||||
rfbClientSetClientData(rfb_client, __GUAC_VNC_TAG_PNG_BUFFER_ALPHA, png_buffer_alpha);
|
||||
|
||||
/* Send name */
|
||||
guac_send_name(io, rfb_client->desktopName);
|
||||
|
||||
/* Send size */
|
||||
guac_send_size(io, rfb_client->width, rfb_client->height);
|
||||
guac_flush(io);
|
||||
|
||||
/* VNC Client Loop */
|
||||
for (;;) {
|
||||
|
||||
/* TODO: Separate into THREADS ... no need to do this weird polling thing */
|
||||
|
||||
wait_result = WaitForMessage(rfb_client, 2000);
|
||||
if (wait_result < 0) {
|
||||
fprintf(stderr, "WAIT FAIL\n");
|
||||
break;
|
||||
}
|
||||
|
||||
if (wait_result > 0) {
|
||||
|
||||
if (!HandleRFBServerMessage(rfb_client)) {
|
||||
fprintf(stderr, "HANDLE FAIL\n");
|
||||
break;
|
||||
}
|
||||
|
||||
}
|
||||
/* Handle server messages */
|
||||
if (client->handle_messages)
|
||||
client->handle_messages(client);
|
||||
|
||||
wait_result = guac_instructions_waiting(io);
|
||||
if (wait_result > 0) {
|
||||
|
||||
guac_instruction* instruction;
|
||||
|
||||
while ((instruction = guac_read_instruction(io))) {
|
||||
|
||||
if (strcmp(instruction->opcode, "mouse") == 0) {
|
||||
SendPointerEvent(
|
||||
rfb_client,
|
||||
if (client->mouse_handler)
|
||||
client->mouse_handler(
|
||||
client,
|
||||
atoi(instruction->argv[0]), /* x */
|
||||
atoi(instruction->argv[1]), /* y */
|
||||
atoi(instruction->argv[2]) /* mask */
|
||||
);
|
||||
);
|
||||
}
|
||||
|
||||
else if (strcmp(instruction->opcode, "key") == 0) {
|
||||
SendKeyEvent(
|
||||
rfb_client,
|
||||
if (client->key_handler)
|
||||
client->key_handler(
|
||||
client,
|
||||
atoi(instruction->argv[0]), /* keysym */
|
||||
atoi(instruction->argv[1]) /* pressed */
|
||||
);
|
||||
);
|
||||
}
|
||||
|
||||
guac_free_instruction(instruction);
|
||||
@@ -275,15 +100,5 @@ void client(int client_fd) {
|
||||
|
||||
}
|
||||
|
||||
/* Free PNG data */
|
||||
guac_free_png_buffer(png_buffer, rfb_client->height);
|
||||
guac_free_png_buffer(png_buffer_alpha, rfb_client->height);
|
||||
|
||||
/* Clean up VNC client*/
|
||||
|
||||
rfbClientCleanup(rfb_client);
|
||||
|
||||
guac_write_string(io, "error:Test finished.;");
|
||||
guac_close(io);
|
||||
}
|
||||
|
||||
|
Reference in New Issue
Block a user