]> Joshua Wise's Git repositories - netwatch.git/blame - net/rfb.c
We do not speak Pittsburgh here.
[netwatch.git] / net / rfb.c
CommitLineData
7da36fbd
JP
1#include <stdint.h>
2#include <minilib.h>
3#include <output.h>
4#include <fb.h>
5
6#include "lwip/tcp.h"
7
8#include "rfb.h"
9
10#define SET_PIXEL_FORMAT 0
11#define SET_ENCODINGS 2
12#define FB_UPDATE_REQUEST 3
13#define KEY_EVENT 4
14#define POINTER_EVENT 5
15#define CLIENT_CUT_TEXT 6
16
80726808 17#define RFB_BUF_SIZE 64
7da36fbd
JP
18
19struct pixel_format {
20 uint8_t bpp;
21 uint8_t depth;
22 uint8_t big_endian;
23 uint8_t true_color;
24 uint16_t red_max;
25 uint16_t green_max;
26 uint16_t blue_max;
27 uint8_t red_shift;
28 uint8_t green_shift;
29 uint8_t blue_shift;
30 uint8_t padding[3];
31};
32
33struct server_init_message {
34 uint16_t fb_width;
35 uint16_t fb_height;
36 struct pixel_format fmt;
37 uint32_t name_length;
38 char name_string[8];
39};
40
41struct fb_update_req {
42 uint8_t msgtype;
43 uint8_t incremental;
44 uint16_t xpos;
45 uint16_t ypos;
46 uint16_t width;
47 uint16_t height;
48};
49
50struct set_encs_req {
51 uint8_t msgtype;
52 uint8_t padding;
53 uint16_t num;
54 int32_t encodings[];
55};
56
57struct key_event_pkt {
58 uint8_t msgtype;
59 uint8_t downflag;
60 uint8_t pad[2];
61 uint32_t keysym;
62};
63
64struct pointer_event_pkt {
65 uint8_t msgtype;
66 uint8_t button_mask;
67 uint16_t x;
68 uint16_t y;
69};
70
71struct text_event_pkt {
72 uint8_t msgtype;
73 uint8_t padding[3];
74 uint32_t length;
75 char text[];
76};
77
78struct rfb_state {
79 enum {
80 ST_BEGIN,
81 ST_CLIENTINIT,
82 ST_MAIN
83 } state;
84 int version;
85 int encs_remaining;
86
87 char data[RFB_BUF_SIZE];
88 int readpos;
89 int writepos;
90
91 char next_update_incremental;
92 struct fb_update_req client_interest_area;
93
b80e06ba 94 uint8_t update_allowed;
80726808 95 uint8_t sending;
7da36fbd
JP
96};
97
98static struct server_init_message server_info;
99
100static void init_server_info() {
80726808 101 server_info.name_length = htonl(8);
7da36fbd
JP
102 memcpy(server_info.name_string, "NetWatch", 8);
103}
104
105static void update_server_info() {
106 if (fb != NULL) {
80726808
JP
107 outputf("RFB: setting fmt %d", fb->curmode.format);
108 server_info.fb_width = htons(fb->curmode.xres);
109 server_info.fb_height = htons(fb->curmode.yres);
7da36fbd
JP
110 switch (fb->curmode.format) {
111 case FB_RGB888:
112 server_info.fmt.bpp = 32;
113 server_info.fmt.depth = 24;
114 server_info.fmt.big_endian = 0;
115 server_info.fmt.true_color = 1;
80726808
JP
116 server_info.fmt.red_max = htons(255);
117 server_info.fmt.green_max = htons(255);
118 server_info.fmt.blue_max = htons(255);
7da36fbd
JP
119 server_info.fmt.red_shift = 0;
120 server_info.fmt.green_shift = 8;
121 server_info.fmt.blue_shift = 16;
122 break;
123 default:
124 outputf("RFB: unknown fb fmt %d", fb->curmode.format);
125 break;
126 }
80726808
JP
127 } else {
128 outputf("RFB: fb null");
7da36fbd
JP
129 }
130}
131
80726808
JP
132static void start_send(struct tcp_pcb *pcb, struct rfb_state *state) {
133 /* ... */
134}
135
7da36fbd
JP
136static void close_conn(struct tcp_pcb *pcb, struct rfb_state *state) {
137 tcp_arg(pcb, NULL);
138 tcp_sent(pcb, NULL);
139 tcp_recv(pcb, NULL);
140 mem_free(state);
141 tcp_close(pcb);
142}
143
144enum fsm_result {
145 NEEDMORE,
146 OK,
147 FAIL
148};
149
150static enum fsm_result recv_fsm(struct tcp_pcb *pcb, struct rfb_state *state) {
80726808
JP
151 int i;
152 int pktsize;
153
154 outputf("RFB FSM: st %d rp %d wp %d", state->state, state->readpos,
155 state->writepos);
7da36fbd
JP
156
157 switch(state->state) {
158 case ST_BEGIN:
159 if (state->writepos < 12) return NEEDMORE;
160
161 if (!strncmp(state->data, "RFB 003.003\n", 12)) {
162 state->version = 3;
163 } else if (!strncmp(state->data, "RFB 003.005\n", 12)) {
164 /* Spec states that "RFB 003.005", an incorrect value,
165 * should be treated by the server as 3.3. */
166 state->version = 3;
167 } else if (!strncmp(state->data, "RFB 003.007\n", 12)) {
168 state->version = 7;
169 } else if (!strncmp(state->data, "RFB 003.008\n", 12)) {
170 state->version = 8;
171 } else {
172 outputf("RFB: Negotiation fail");
173 return FAIL;
174 }
175
176 outputf("RFB: Negotiated v3.%d", state->version);
177
178 state->readpos += 12;
179 state->state = ST_CLIENTINIT;
180
80726808
JP
181 /* We support one security type, currently "none".
182 * Send that and SecurityResult. */
7da36fbd 183 if (state->version >= 7) {
80726808 184 tcp_write(pcb, "\x01\x01\x00\x00\x00\x00", 6, 0);
7da36fbd 185 } else {
80726808 186 tcp_write(pcb, "\x01\x00\x00\x00\x00", 5, 0);
7da36fbd
JP
187 }
188
7da36fbd
JP
189 tcp_output(pcb);
190
191 return OK;
192
193 case ST_CLIENTINIT:
80726808
JP
194 if (state->version >= 7) {
195 /* Ignore the security type and ClientInit */
196 if (state->writepos < 2) return NEEDMORE;
197 state->readpos += 2;
198 } else {
199 /* Just ClientInit */
200 if (state->writepos < 1) return NEEDMORE;
201 state->readpos += 1;
202 }
203
7da36fbd
JP
204 state->state = ST_MAIN;
205
80726808 206 outputf("RFB: Sending server info", state->version);
7da36fbd
JP
207 tcp_write(pcb, &server_info, sizeof(server_info), 0);
208 tcp_output(pcb);
209
210 return OK;
211
212 case ST_MAIN:
213 if (state->writepos < 1) return NEEDMORE;
214
80726808 215 outputf("RFB: cmd %d", state->data[0]);
7da36fbd
JP
216 switch (state->data[0]) {
217
218 case SET_PIXEL_FORMAT:
219 /* SetPixelFormat */
220 if (state->writepos < (sizeof(struct pixel_format) + 4))
221 return NEEDMORE;
80726808 222 outputf("RFB: SetPixelFormat");
7da36fbd
JP
223/*
224 struct pixel_format * new_fmt =
225 (struct pixel_format *)(&state->data[4]);
226*/
227 /* XXX ... */
228
229 state->readpos += sizeof(struct pixel_format) + 4;
230 return OK;
231
232 case SET_ENCODINGS:
233 if (state->writepos < 4) return NEEDMORE;
234
235 struct set_encs_req * req = (struct set_encs_req *)state->data;
236
80726808 237 pktsize = sizeof(struct set_encs_req) + (4 * ntohs(req->num));
7da36fbd 238
80726808
JP
239 outputf("RFB: SetEncodings [%d]", ntohs(req->num));
240 if (state->writepos < pktsize) return NEEDMORE;
241
242 for (i = 0; i < ntohs(req->num); i++) {
243 outputf("RFB: Encoding: %d", ntohl(req->encodings[i]));
244 /* XXX ... */
7da36fbd 245
80726808
JP
246 }
247
248 state->readpos += pktsize;
7da36fbd
JP
249 return OK;
250
251 case FB_UPDATE_REQUEST:
252 if (state->writepos < sizeof(struct fb_update_req))
253 return NEEDMORE;
80726808 254 outputf("RFB: UpdateRequest");
7da36fbd 255
b80e06ba 256 state->update_allowed = 1;
7da36fbd
JP
257 memcpy(&state->client_interest_area, state->data,
258 sizeof(struct fb_update_req));
259
260 state->readpos += sizeof(struct fb_update_req);
261 return OK;
262
263 case KEY_EVENT:
264 if (state->writepos < sizeof(struct key_event_pkt))
265 return NEEDMORE;
80726808 266 outputf("RFB: Key");
7da36fbd
JP
267
268 /* XXX stub */
269
270 state->readpos += sizeof(struct key_event_pkt);
271 return OK;
272
273 case POINTER_EVENT:
274 if (state->writepos < sizeof(struct pointer_event_pkt))
275 return NEEDMORE;
80726808 276 outputf("RFB: Pointer");
7da36fbd
JP
277
278 /* XXX stub */
279
280 state->readpos += sizeof(struct pointer_event_pkt);
281 return OK;
282
283 case CLIENT_CUT_TEXT:
284 if (state->writepos < sizeof(struct text_event_pkt))
285 return NEEDMORE;
80726808 286 outputf("RFB: Cut Text");
7da36fbd
JP
287
288 struct text_event_pkt * pkt =
289 (struct text_event_pkt *)state->data;
290
291 if (state->writepos < sizeof(struct text_event_pkt)
292 + pkt->length)
293 return NEEDMORE;
294
295 /* XXX stub */
296
297 state->readpos += sizeof(struct text_event_pkt)
298 + pkt->length;
299 return OK;
300
301 default:
302 outputf("RFB: Bad command: %d", state->data[0]);
303 }
304 default:
305 outputf("RFB: Bad state");
306 return FAIL;
307 }
308}
309
310static err_t rfb_recv(void *arg, struct tcp_pcb *pcb,
311 struct pbuf *p, err_t err) {
312 struct rfb_state *state = arg;
313
314 if (state == NULL)
315
316 if (err != ERR_OK) {
317 outputf("RFB: recv err %d", err);
318 /* FIXME do something better here? */
319 return ERR_OK;
320 }
321
322 if (p == NULL) {
323 outputf("RFB: Connection closed");
324 close_conn(pcb, state);
325 return ERR_OK;
326 }
327
328 if (p->tot_len > (RFB_BUF_SIZE - state->writepos)) {
329 /* Overflow! */
330 outputf("RFB: Overflow!");
331 close_conn(pcb, state);
332 return ERR_OK;
333 }
334
335 outputf("RFB: Processing %d", p->tot_len);
336 pbuf_copy_partial(p, state->data + state->writepos, p->tot_len, 0);
80726808
JP
337 state->writepos += p->tot_len;
338
7da36fbd
JP
339 tcp_recved(pcb, p->tot_len);
340 pbuf_free(p);
341
342 while (1) {
343 switch (recv_fsm(pcb, state)) {
344 case NEEDMORE:
80726808 345 outputf("RFB FSM: blocking");
7da36fbd
JP
346 /* Need more data */
347 return ERR_OK;
348
349 case OK:
80726808
JP
350 outputf("RFB FSM: ok");
351
352 /* Might as well send now... */
b80e06ba 353 if (state->update_allowed && !state->sending) {
80726808
JP
354 start_send(pcb, state);
355 }
356
7da36fbd
JP
357 if (state->readpos == state->writepos) {
358 state->readpos = 0;
359 state->writepos = 0;
360 return ERR_OK;
361 } else {
362 memmove(state->data,
363 state->data + state->readpos,
364 state->writepos - state->readpos);
365 }
366 break;
367 case FAIL:
368 /* Shit */
369 outputf("RFB: Protocol error");
370 close_conn(pcb, state);
371 return ERR_OK;
372 }
373 }
374}
375
376static err_t rfb_accept(void *arg, struct tcp_pcb *pcb, err_t err) {
377 struct rfb_state *state;
378
379 LWIP_UNUSED_ARG(arg);
380 LWIP_UNUSED_ARG(err);
381
382 state = (struct rfb_state *)mem_malloc(sizeof(struct rfb_state));
383
80726808
JP
384 state->state = ST_BEGIN;
385 state->readpos = 0;
386 state->writepos = 0;
b80e06ba 387 state->update_allowed = 0;
80726808
JP
388 state->sending = 0;
389
7da36fbd
JP
390 /* XXX: update_server_info() should be called from the 64ms timer, and deal
391 * with screen resizes appropriately. */
392 update_server_info();
393
394 if (!state)
395 {
396 outputf("rfb_accept: out of memory\n");
397 return ERR_MEM;
398 }
399
400 tcp_arg(pcb, state);
401 tcp_recv(pcb, rfb_recv);
402/*
403 tcp_err(pcb, rfb_err);
404 tcp_poll(pcb, rfb_poll, 2);
405*/
406 tcp_write(pcb, "RFB 003.008\n", 12, 0);
407 tcp_output(pcb);
408
409 return ERR_OK;
410}
411
412void rfb_init() {
413 struct tcp_pcb *pcb;
414
415 init_server_info();
416
417 pcb = tcp_new();
418 tcp_bind(pcb, IP_ADDR_ANY, RFB_PORT);
419 pcb = tcp_listen(pcb);
420 tcp_accept(pcb, rfb_accept);
421}
This page took 0.05551 seconds and 4 git commands to generate.