1 /**
2 * @example example.c
3 * This is an example of how to use libvncserver.
4 *
5 * libvncserver example
6 * Copyright (C) 2001 Johannes E. Schindelin <Johannes.Schindelin@gmx.de>
7 *
8 * This is free software; you can redistribute it and/or modify
9 * it under the terms of the GNU General Public License as published by
10 * the Free Software Foundation; either version 2 of the License, or
11 * (at your option) any later version.
12 *
13 * This software is distributed in the hope that it will be useful,
14 * but WITHOUT ANY WARRANTY; without even the implied warranty of
15 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
16 * GNU General Public License for more details.
17 *
18 * You should have received a copy of the GNU General Public License
19 * along with this software; if not, write to the Free Software
20 * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307,
21 * USA.
22 */
23
24 #ifdef WIN32
25 #define sleep Sleep
26 #else
27 #include <unistd.h>
28 #endif
29
30 #ifdef __IRIX__
31 #include <netdb.h>
32 #endif
33
34 #include <rfb/rfb.h>
35 #include <rfb/keysym.h>
36
37 static const int bpp=4;
38 static int maxx=800, maxy=600;
39 /* TODO: odd maxx doesn't work (vncviewer bug) */
40
41 /* This initializes a nice (?) background */
42
initBuffer(unsigned char * buffer)43 static void initBuffer(unsigned char* buffer)
44 {
45 int i,j;
46 for(j=0;j<maxy;++j) {
47 for(i=0;i<maxx;++i) {
48 buffer[(j*maxx+i)*bpp+0]=(i+j)*128/(maxx+maxy); /* red */
49 buffer[(j*maxx+i)*bpp+1]=i*128/maxx; /* green */
50 buffer[(j*maxx+i)*bpp+2]=j*256/maxy; /* blue */
51 }
52 buffer[j*maxx*bpp+0]=0xff;
53 buffer[j*maxx*bpp+1]=0xff;
54 buffer[j*maxx*bpp+2]=0xff;
55 buffer[j*maxx*bpp+3]=0xff;
56 }
57 }
58
59 /* Here we create a structure so that every client has it's own pointer */
60
61 typedef struct ClientData {
62 rfbBool oldButton;
63 int oldx,oldy;
64 } ClientData;
65
clientgone(rfbClientPtr cl)66 static void clientgone(rfbClientPtr cl)
67 {
68 free(cl->clientData);
69 }
70
newclient(rfbClientPtr cl)71 static enum rfbNewClientAction newclient(rfbClientPtr cl)
72 {
73 cl->clientData = (void*)calloc(sizeof(ClientData),1);
74 cl->clientGoneHook = clientgone;
75 return RFB_CLIENT_ACCEPT;
76 }
77
78 /* switch to new framebuffer contents */
79
newframebuffer(rfbScreenInfoPtr screen,int width,int height)80 static void newframebuffer(rfbScreenInfoPtr screen, int width, int height)
81 {
82 unsigned char *oldfb, *newfb;
83
84 maxx = width;
85 maxy = height;
86 oldfb = (unsigned char*)screen->frameBuffer;
87 newfb = (unsigned char*)malloc(maxx * maxy * bpp);
88 initBuffer(newfb);
89 rfbNewFramebuffer(screen, (char*)newfb, maxx, maxy, 8, 3, bpp);
90 free(oldfb);
91
92 /*** FIXME: Re-install cursor. ***/
93 }
94
95 /* aux function to draw a line */
96
drawline(unsigned char * buffer,int rowstride,int bpp,int x1,int y1,int x2,int y2)97 static void drawline(unsigned char* buffer,int rowstride,int bpp,int x1,int y1,int x2,int y2)
98 {
99 int i,j;
100 i=x1-x2; j=y1-y2;
101 if(i==0 && j==0) {
102 for(i=0;i<bpp;i++)
103 buffer[y1*rowstride+x1*bpp+i]=0xff;
104 return;
105 }
106 if(i<0) i=-i;
107 if(j<0) j=-j;
108 if(i<j) {
109 if(y1>y2) { i=y2; y2=y1; y1=i; i=x2; x2=x1; x1=i; }
110 for(j=y1;j<=y2;j++)
111 for(i=0;i<bpp;i++)
112 buffer[j*rowstride+(x1+(j-y1)*(x2-x1)/(y2-y1))*bpp+i]=0xff;
113 } else {
114 if(x1>x2) { i=y2; y2=y1; y1=i; i=x2; x2=x1; x1=i; }
115 for(i=x1;i<=x2;i++)
116 for(j=0;j<bpp;j++)
117 buffer[(y1+(i-x1)*(y2-y1)/(x2-x1))*rowstride+i*bpp+j]=0xff;
118 }
119 }
120
121 /* Here the pointer events are handled */
122
doptr(int buttonMask,int x,int y,rfbClientPtr cl)123 static void doptr(int buttonMask,int x,int y,rfbClientPtr cl)
124 {
125 ClientData* cd=cl->clientData;
126
127 if(x>=0 && y>=0 && x<maxx && y<maxy) {
128 if(buttonMask) {
129 int i,j,x1,x2,y1,y2;
130
131 if(cd->oldButton==buttonMask) { /* draw a line */
132 drawline((unsigned char*)cl->screen->frameBuffer,cl->screen->paddedWidthInBytes,bpp,
133 x,y,cd->oldx,cd->oldy);
134 x1=x; y1=y;
135 if(x1>cd->oldx) x1++; else cd->oldx++;
136 if(y1>cd->oldy) y1++; else cd->oldy++;
137 rfbMarkRectAsModified(cl->screen,x,y,cd->oldx,cd->oldy);
138 } else { /* draw a point (diameter depends on button) */
139 int w=cl->screen->paddedWidthInBytes;
140 x1=x-buttonMask; if(x1<0) x1=0;
141 x2=x+buttonMask; if(x2>maxx) x2=maxx;
142 y1=y-buttonMask; if(y1<0) y1=0;
143 y2=y+buttonMask; if(y2>maxy) y2=maxy;
144
145 for(i=x1*bpp;i<x2*bpp;i++)
146 for(j=y1;j<y2;j++)
147 cl->screen->frameBuffer[j*w+i]=(char)0xff;
148 rfbMarkRectAsModified(cl->screen,x1,y1,x2,y2);
149 }
150
151 /* we could get a selection like that:
152 rfbGotXCutText(cl->screen,"Hallo",5);
153 */
154 } else
155 cd->oldButton=0;
156
157 cd->oldx=x; cd->oldy=y; cd->oldButton=buttonMask;
158 }
159 rfbDefaultPtrAddEvent(buttonMask,x,y,cl);
160 }
161
162 /* aux function to draw a character to x, y */
163
164 #include "radon.h"
165
166 /* Here the key events are handled */
167
dokey(rfbBool down,rfbKeySym key,rfbClientPtr cl)168 static void dokey(rfbBool down,rfbKeySym key,rfbClientPtr cl)
169 {
170 if(down) {
171 if(key==XK_Escape)
172 rfbCloseClient(cl);
173 else if(key==XK_F12)
174 /* close down server, disconnecting clients */
175 rfbShutdownServer(cl->screen,TRUE);
176 else if(key==XK_F11)
177 /* close down server, but wait for all clients to disconnect */
178 rfbShutdownServer(cl->screen,FALSE);
179 else if(key==XK_Page_Up) {
180 initBuffer((unsigned char*)cl->screen->frameBuffer);
181 rfbMarkRectAsModified(cl->screen,0,0,maxx,maxy);
182 } else if (key == XK_Up) {
183 if (maxx < 1024) {
184 if (maxx < 800) {
185 newframebuffer(cl->screen, 800, 600);
186 } else {
187 newframebuffer(cl->screen, 1024, 768);
188 }
189 }
190 } else if(key==XK_Down) {
191 if (maxx > 640) {
192 if (maxx > 800) {
193 newframebuffer(cl->screen, 800, 600);
194 } else {
195 newframebuffer(cl->screen, 640, 480);
196 }
197 }
198 } else if(key>=' ' && key<0x100) {
199 ClientData* cd=cl->clientData;
200 int x1=cd->oldx,y1=cd->oldy,x2,y2;
201 cd->oldx+=rfbDrawCharWithClip(cl->screen,&radonFont,cd->oldx,cd->oldy,(char)key,0,0,cl->screen->width,cl->screen->height,0x00ffffff,0x00ffffff);
202 rfbFontBBox(&radonFont,(char)key,&x1,&y1,&x2,&y2);
203 rfbMarkRectAsModified(cl->screen,x1,y1,x2-1,y2-1);
204 }
205 }
206 }
207
208 /* Example for an XCursor (foreground/background only) */
209
210 #ifdef JUST_AN_EXAMPLE
211
212 static int exampleXCursorWidth=9,exampleXCursorHeight=7;
213 static char exampleXCursor[]=
214 " "
215 " xx xx "
216 " xx xx "
217 " xxx "
218 " xx xx "
219 " xx xx "
220 " ";
221
222 #endif
223
224 /* Example for a rich cursor (full-colour) */
225
MakeRichCursor(rfbScreenInfoPtr rfbScreen)226 static void MakeRichCursor(rfbScreenInfoPtr rfbScreen)
227 {
228 int i,j,w=32,h=32;
229 rfbCursorPtr c = rfbScreen->cursor;
230 char bitmap[]=
231 " "
232 " xxxxxx "
233 " xxxxxxxxxxxxxxxxx "
234 " xxxxxxxxxxxxxxxxxxxxxx "
235 " xxxxx xxxxxxxx xxxxxxxx "
236 " xxxxxxxxxxxxxxxxxxxxxxxxxxx "
237 " xxxxxxxxxxxxxxxxxxxxxxxxxxxxx "
238 " xxxxx xxxxxxxxxxx xxxxxxx "
239 " xxxx xxxxxxxxx xxxxxx "
240 " xxxxx xxxxxxxxxxx xxxxxxx "
241 " xxxxxxxxxxxxxxxxxxxxxxxxxxxxxx "
242 " xxxxxxxxxxxxxxxxxxxxxxxxxxxxxx "
243 " xxxxxxxxxxxx xxxxxxxxxxxxxxx "
244 " xxxxxxxxxxxxxxxxxxxxxxxxxxxx "
245 " xxxxxxxxxxxxxxxxxxxxxxxxxxxx "
246 " xxxxxxxxxxx xxxxxxxxxxxxxx "
247 " xxxxxxxxxx xxxxxxxxxxxx "
248 " xxxxxxxxx xxxxxxxxx "
249 " xxxxxxxxxx xxxxxxxxx "
250 " xxxxxxxxxxxxxxxxxxx "
251 " xxxxxxxxxxxxxxxxxxx "
252 " xxxxxxxxxxxxxxxxxxx "
253 " xxxxxxxxxxxxxxxxx "
254 " xxxxxxxxxxxxxxx "
255 " xxxx xxxxxxxxxxxxx "
256 " xx x xxxxxxxxxxx "
257 " xxx xxxxxxxxxxx "
258 " xxxx xxxxxxxxxxx "
259 " xxxxxx xxxxxxxxxxxx "
260 " xxxxxxxxxxxxxxxxxxxxxx "
261 " xxxxxxxxxxxxxxxx "
262 " ";
263 c=rfbScreen->cursor = rfbMakeXCursor(w,h,bitmap,bitmap);
264 c->xhot = 16; c->yhot = 24;
265
266 c->richSource = (unsigned char*)malloc(w*h*bpp);
267 c->cleanupRichSource = TRUE;
268 for(j=0;j<h;j++) {
269 for(i=0;i<w;i++) {
270 c->richSource[j*w*bpp+i*bpp+0]=i*0xff/w;
271 c->richSource[j*w*bpp+i*bpp+1]=(i+j)*0xff/(w+h);
272 c->richSource[j*w*bpp+i*bpp+2]=j*0xff/h;
273 c->richSource[j*w*bpp+i*bpp+3]=0;
274 }
275 }
276 }
277
278 /* Initialization */
279
main(int argc,char ** argv)280 int main(int argc,char** argv)
281 {
282 rfbScreenInfoPtr rfbScreen = rfbGetScreen(&argc,argv,maxx,maxy,8,3,bpp);
283 if(!rfbScreen)
284 return 0;
285 rfbScreen->desktopName = "LibVNCServer Example";
286 rfbScreen->frameBuffer = (char*)malloc(maxx*maxy*bpp);
287 rfbScreen->alwaysShared = TRUE;
288 rfbScreen->ptrAddEvent = doptr;
289 rfbScreen->kbdAddEvent = dokey;
290 rfbScreen->newClientHook = newclient;
291 rfbScreen->httpDir = "../webclients";
292 rfbScreen->httpEnableProxyConnect = TRUE;
293
294 initBuffer((unsigned char*)rfbScreen->frameBuffer);
295 rfbDrawString(rfbScreen,&radonFont,20,100,"Hello, World!",0xffffff);
296
297 /* This call creates a mask and then a cursor: */
298 /* rfbScreen->defaultCursor =
299 rfbMakeXCursor(exampleCursorWidth,exampleCursorHeight,exampleCursor,0);
300 */
301
302 MakeRichCursor(rfbScreen);
303
304 /* initialize the server */
305 rfbInitServer(rfbScreen);
306
307 #ifndef BACKGROUND_LOOP_TEST
308 #ifdef USE_OWN_LOOP
309 {
310 int i;
311 for(i=0;rfbIsActive(rfbScreen);i++) {
312 fprintf(stderr,"%d\r",i);
313 rfbProcessEvents(rfbScreen,100000);
314 }
315 }
316 #else
317 /* this is the blocking event loop, i.e. it never returns */
318 /* 40000 are the microseconds to wait on select(), i.e. 0.04 seconds */
319 rfbRunEventLoop(rfbScreen,40000,FALSE);
320 #endif /* OWN LOOP */
321 #else
322 #if !defined(LIBVNCSERVER_HAVE_LIBPTHREAD)
323 #error "I need pthreads for that."
324 #endif
325
326 /* this is the non-blocking event loop; a background thread is started */
327 rfbRunEventLoop(rfbScreen,-1,TRUE);
328 fprintf(stderr, "Running background loop...\n");
329 /* now we could do some cool things like rendering in idle time */
330 while(1) sleep(5); /* render(); */
331 #endif /* BACKGROUND_LOOP */
332
333 free(rfbScreen->frameBuffer);
334 rfbScreenCleanup(rfbScreen);
335
336 return(0);
337 }
338