fix disconnect bug in libvncserver
[presencevnc] / libvnc / examples / example.c
1 /*
2  * 
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
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
66 static void clientgone(rfbClientPtr cl)
67 {
68   free(cl->clientData);
69 }
70
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
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
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
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
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
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 = (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
280 int main(int argc,char** argv)
281 {
282   rfbScreenInfoPtr rfbScreen =
283     rfbGetScreen(&argc,argv,maxx,maxy,8,3,bpp);
284   rfbScreen->desktopName = "LibVNCServer Example";
285   rfbScreen->frameBuffer = (char*)malloc(maxx*maxy*bpp);
286   rfbScreen->alwaysShared = TRUE;
287   rfbScreen->ptrAddEvent = doptr;
288   rfbScreen->kbdAddEvent = dokey;
289   rfbScreen->newClientHook = newclient;
290   rfbScreen->httpDir = "../classes";
291   rfbScreen->httpEnableProxyConnect = TRUE;
292
293   initBuffer((unsigned char*)rfbScreen->frameBuffer);
294   rfbDrawString(rfbScreen,&radonFont,20,100,"Hello, World!",0xffffff);
295
296   /* This call creates a mask and then a cursor: */
297   /* rfbScreen->defaultCursor =
298        rfbMakeXCursor(exampleCursorWidth,exampleCursorHeight,exampleCursor,0);
299   */
300
301   MakeRichCursor(rfbScreen);
302
303   /* initialize the server */
304   rfbInitServer(rfbScreen);
305
306 #ifndef BACKGROUND_LOOP_TEST
307 #ifdef USE_OWN_LOOP
308   {
309     int i;
310     for(i=0;rfbIsActive(rfbScreen);i++) {
311       fprintf(stderr,"%d\r",i);
312       rfbProcessEvents(rfbScreen,100000);
313     }
314   }
315 #else
316   /* this is the blocking event loop, i.e. it never returns */
317   /* 40000 are the microseconds to wait on select(), i.e. 0.04 seconds */
318   rfbRunEventLoop(rfbScreen,40000,FALSE);
319 #endif /* OWN LOOP */
320 #else
321 #if !defined(LIBVNCSERVER_HAVE_LIBPTHREAD)
322 #error "I need pthreads for that."
323 #endif
324
325   /* this is the non-blocking event loop; a background thread is started */
326   rfbRunEventLoop(rfbScreen,-1,TRUE);
327   fprintf(stderr, "Running background loop...\n");
328   /* now we could do some cool things like rendering in idle time */
329   while(1) sleep(5); /* render(); */
330 #endif /* BACKGROUND_LOOP */
331
332   free(rfbScreen->frameBuffer);
333   rfbScreenCleanup(rfbScreen);
334
335   return(0);
336 }