/*
Copyright (C) 2002-2010 Karl J. Runge <runge@karlrunge.com>
All rights reserved.
This file is part of x11vnc.
x11vnc is free software; you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation; either version 2 of the License, or (at
your option) any later version.
x11vnc is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU General Public License for more details.
You should have received a copy of the GNU General Public License
along with x11vnc; if not, write to the Free Software
Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA
or see <http://www.gnu.org/licenses/>.
In addition, as a special exception, Karl J. Runge
gives permission to link the code of its release of x11vnc with the
OpenSSL project's "OpenSSL" library (or with modified versions of it
that use the same license as the "OpenSSL" library), and distribute
the linked executables. You must obey the GNU General Public License
in all respects for all of the code used other than "OpenSSL". If you
modify this file, you may extend this exception to your version of the
file, but you are not obligated to do so. If you do not wish to do
so, delete this exception statement from your version.
*/
/* -- screen.c -- */
#include "x11vnc.h"
#include "xevents.h"
#include "xwrappers.h"
#include "xinerama.h"
#include "xdamage.h"
#include "win_utils.h"
#include "cleanup.h"
#include "userinput.h"
#include "scan.h"
#include "user.h"
#include "rates.h"
#include "pointer.h"
#include "keyboard.h"
#include "cursor.h"
#include "connections.h"
#include "remote.h"
#include "unixpw.h"
#include "sslcmds.h"
#include "sslhelper.h"
#include "v4l.h"
#include "linuxfb.h"
#include "macosx.h"
#include "macosxCG.h"
#include "avahi.h"
#include "solid.h"
#include "inet.h"
#include "xrandr.h"
#include "xrecord.h"
#include "pm.h"
#include <rfb/rfbclient.h>
void set_greyscale_colormap(void);
void set_hi240_colormap(void);
void unset_colormap(void);
void set_colormap(int reset);
void set_nofb_params(int restore);
void set_raw_fb_params(int restore);
void do_new_fb(int reset_mem);
void free_old_fb(void);
void check_padded_fb(void);
void install_padded_fb(char *geom);
XImage *initialize_xdisplay_fb(void);
void parse_scale_string(char *str, double *factor_x, double *factor_y, int *scaling, int *blend,
int *nomult4, int *pad, int *interpolate, int *numer, int *denom, int w_in, int h_in);
int parse_rotate_string(char *str, int *mode);
int scale_round(int len, double fac);
void initialize_screen(int *argc, char **argv, XImage *fb);
void set_vnc_desktop_name(void);
void announce(int lport, int ssl, char *iface);
char *vnc_reflect_guess(char *str, char **raw_fb_addr);
void vnc_reflect_process_client(void);
rfbBool vnc_reflect_send_pointer(int x, int y, int mask);
rfbBool vnc_reflect_send_key(uint32_t key, rfbBool down);
rfbBool vnc_reflect_send_cuttext(char *str, int len);
static void debug_colormap(XImage *fb);
static void set_visual(char *str);
static void nofb_hook(rfbClientPtr cl);
static void remove_fake_fb(void);
static void install_fake_fb(int w, int h, int bpp);
static void initialize_snap_fb(void);
XImage *initialize_raw_fb(int);
static void initialize_clipshift(void);
static int wait_until_mapped(Window win);
static void setup_scaling(int *width_in, int *height_in);
static void check_filexfer(void);
static void record_last_fb_update(void);
static void check_cursor_changes(void);
static int choose_delay(double dt);
int rawfb_reset = -1;
int rawfb_dev_video = 0;
int rawfb_vnc_reflect = 0;
/*
* X11 and rfb display/screen related routines
*/
/*
* Some handling of 8bpp PseudoColor colormaps. Called for initializing
* the clients and dynamically if -flashcmap is specified.
*/
#define NCOLOR 256
/* this is only for rawfb */
void set_greyscale_colormap(void) {
int i;
if (! screen) {
return;
}
/* mutex */
if (screen->colourMap.data.shorts) {
free(screen->colourMap.data.shorts);
screen->colourMap.data.shorts = NULL;
}
if (0) fprintf(stderr, "set_greyscale_colormap: %s\n", raw_fb_pixfmt);
screen->colourMap.count = NCOLOR;
screen->serverFormat.trueColour = FALSE;
screen->colourMap.is16 = TRUE;
screen->colourMap.data.shorts = (unsigned short *)
malloc(3*sizeof(unsigned short) * NCOLOR);
for(i = 0; i < NCOLOR; i++) {
unsigned short lvl = i * 256;
screen->colourMap.data.shorts[i*3+0] = lvl;
screen->colourMap.data.shorts[i*3+1] = lvl;
screen->colourMap.data.shorts[i*3+2] = lvl;
}
rfbSetClientColourMaps(screen, 0, NCOLOR);
}
/* this is specific to bttv rf tuner card */
void set_hi240_colormap(void) {
int i;
if (! screen) {
return;
}
/* mutex */
if (0) fprintf(stderr, "set_hi240_colormap: %s\n", raw_fb_pixfmt);
if (screen->colourMap.data.shorts) {
free(screen->colourMap.data.shorts);
screen->colourMap.data.shorts = NULL;
}
screen->colourMap.count = 256;
screen->serverFormat.trueColour = FALSE;
screen->colourMap.is16 = TRUE;
screen->colourMap.data.shorts = (unsigned short *)
calloc(3*sizeof(unsigned short) * 256, 1);
for(i = 0; i < 225; i++) {
int r, g, b;
r = ( (i/5) % 5 ) * 255.0 / 4 + 0.5;
g = ( (i/25) ) * 255.0 / 8 + 0.5;
b = ( i % 5 ) * 255.0 / 4 + 0.5;
screen->colourMap.data.shorts[(i+16)*3+0] = (unsigned short) (r * 256);
screen->colourMap.data.shorts[(i+16)*3+1] = (unsigned short) (g * 256);
screen->colourMap.data.shorts[(i+16)*3+2] = (unsigned short) (b * 256);
}
rfbSetClientColourMaps(screen, 0, 256);
}
/* this is only for rawfb */
void unset_colormap(void) {
if (! screen) {
return;
}
if (screen->colourMap.data.shorts) {
free(screen->colourMap.data.shorts);
screen->colourMap.data.shorts = NULL;
}
screen->serverFormat.trueColour = TRUE;
if (0) fprintf(stderr, "unset_colormap: %s\n", raw_fb_pixfmt);
}
/* this is X11 case */
void set_colormap(int reset) {
#if NO_X11
if (!reset) {}
return;
#else
static int init = 1;
static XColor *color = NULL, *prev = NULL;
static int ncolor = 0;
Colormap cmap;
Visual *vis;
int i, ncells, diffs = 0;
if (reset) {
init = 1;
ncolor = 0;
/* mutex */
if (screen->colourMap.data.shorts) {
free(screen->colourMap.data.shorts);
screen->colourMap.data.shorts = NULL;
}
if (color) {
free(color);
color = NULL;
}
if (prev) {
free(prev);
prev = NULL;
}
}
if (init) {
if (depth > 16) {
ncolor = NCOLOR;
} else if (depth > 8) {
ncolor = 1 << depth;
} else {
ncolor = NCOLOR;
}
/* mutex */
screen->colourMap.count = ncolor;
screen->serverFormat.trueColour = FALSE;
screen->colourMap.is16 = TRUE;
screen->colourMap.data.shorts = (unsigned short *)
malloc(3*sizeof(unsigned short) * ncolor);
}
if (color == NULL) {
color = (XColor *) calloc(ncolor * sizeof(XColor), 1);
prev = (XColor *) calloc(ncolor * sizeof(XColor), 1);
}
for (i=0; i < ncolor; i++) {
prev[i].red = color[i].red;
prev[i].green = color[i].green;
prev[i].blue = color[i].blue;
}
RAWFB_RET_VOID
X_LOCK;
cmap = DefaultColormap(dpy, scr);
ncells = CellsOfScreen(ScreenOfDisplay(dpy, scr));
vis = default_visual;
if (subwin) {
XWindowAttributes attr;
if (XGetWindowAttributes(dpy, window, &attr)) {
cmap = attr.colormap;
vis = attr.visual;
ncells = vis->map_entries;
}
}
if (ncells != ncolor) {
if (! shift_cmap) {
screen->colourMap.count = ncells;
}
}
if (init && ! quiet) {
rfbLog("set_colormap: number of cells: %d, "
"ncolor(%d) is %d.\n", ncells, depth, ncolor);
}
if (flash_cmap && ! init) {
XWindowAttributes attr;
Window c;
int tries = 0;
c = window;
while (c && tries++ < 16) {
c = query_pointer(c);
if (valid_window(c, &attr, 0)) {
if (attr.colormap && attr.map_installed) {
cmap = attr.colormap;
vis = attr.visual;
ncells = vis->map_entries;
break;
}
} else {
break;
}
}
}
if (ncells > ncolor && ! quiet) {
rfbLog("set_colormap: big problem: ncells=%d > %d\n",
ncells, ncolor);
}
if (vis->class == TrueColor || vis->class == DirectColor) {
/*
* Kludge to make 8bpp TrueColor & DirectColor be like
* the StaticColor map. The ncells = 8 is "8 per subfield"
* mentioned in xdpyinfo. Looks OK... perhaps fortuitously.
*/
if (ncells == 8 && ! shift_cmap) {
ncells = ncolor;
}
}
for (i=0; i < ncells; i++) {
color[i].pixel = i;
color[i].pad = 0;
}
XQueryColors(dpy, cmap, color, ncells);
X_UNLOCK;
for(i = ncells - 1; i >= 0; i--) {
int k = i + shift_cmap;
screen->colourMap.data.shorts[i*3+0] = color[i].red;
screen->colourMap.data.shorts[i*3+1] = color[i].green;
screen->colourMap.data.shorts[i*3+2] = color[i].blue;
if (prev[i].red != color[i].red ||
prev[i].green != color[i].green ||
prev[i].blue != color[i].blue ) {
diffs++;
}
if (shift_cmap && k >= 0 && k < ncolor) {
/* kludge to copy the colors to higher pixel values */
screen->colourMap.data.shorts[k*3+0] = color[i].red;
screen->colourMap.data.shorts[k*3+1] = color[i].green;
screen->colourMap.data.shorts[k*3+2] = color[i].blue;
}
}
if (diffs && ! init) {
if (! all_clients_initialized()) {
rfbLog("set_colormap: warning: sending cmap "
"with uninitialized clients.\n");
}
if (shift_cmap) {
rfbSetClientColourMaps(screen, 0, ncolor);
} else {
rfbSetClientColourMaps(screen, 0, ncells);
}
}
init = 0;
#endif /* NO_X11 */
}
static void debug_colormap(XImage *fb) {
static int debug_cmap = -1;
int i, k, *histo;
int ncolor;
if (debug_cmap < 0) {
if (getenv("DEBUG_CMAP") != NULL) {
debug_cmap = 1;
} else {
debug_cmap = 0;
}
}
if (! debug_cmap) {
return;
}
if (! fb) {
return;
}
if (fb->bits_per_pixel > 16) {
return;
}
ncolor = screen->colourMap.count;
histo = (int *) calloc(ncolor * sizeof(int), 1);
for (i=0; i < ncolor; i++) {
histo[i] = 0;
}
for (k = 0; k < fb->width * fb->height; k++) {
unsigned char n;
char c = *(fb->data + k);
n = (unsigned char) c;
histo[n]++;
}
fprintf(stderr, "\nColormap histogram for current screen contents:\n");
for (i=0; i < ncolor; i++) {
unsigned short r = screen->colourMap.data.shorts[i*3+0];
unsigned short g = screen->colourMap.data.shorts[i*3+1];
unsigned short b = screen->colourMap.data.shorts[i*3+2];
fprintf(stderr, " %03d: %7d %04x/%04x/%04x", i, histo[i],
r, g, b);
if ((i+1) % 2 == 0) {
fprintf(stderr, "\n");
}
}
free(histo);
fprintf(stderr, "\n");
}
/*
* Experimental mode to force the visual of the window instead of querying
* it. Used for testing, overriding some rare cases (win2vnc), and for
* -overlay . Input string can be a decimal or 0x hex or something like
* TrueColor or TrueColor:24 to force a depth as well.
*
* visual_id and possibly visual_depth are set.
*/
static void set_visual(char *str) {
#if NO_X11
RAWFB_RET_VOID
if (!str) {}
return;
#else
int vis, vdepth, defdepth;
XVisualInfo vinfo;
char *p, *vstring = strdup(str);
RAWFB_RET_VOID
defdepth = DefaultDepth(dpy, scr);
visual_id = (VisualID) 0;
visual_depth = 0;
if (!strcmp(vstring, "ignore") || !strcmp(vstring, "default")
|| !strcmp(vstring, "")) {
free(vstring);
return;
}
/* set visual depth */
if ((p = strchr(vstring, ':')) != NULL) {
visual_depth = atoi(p+1);
*p = '\0';
vdepth = visual_depth;
} else {
vdepth = defdepth;
}
if (! quiet) {
fprintf(stderr, "\nVisual Info:\n");
fprintf(stderr, " set_visual(\"%s\")\n", str);
fprintf(stderr, " visual_depth: %d\n", vdepth);
}
/* set visual id number */
if (strcmp(vstring, "StaticGray") == 0) {
vis = StaticGray;
} else if (strcmp(vstring, "GrayScale") == 0) {
vis = GrayScale;
} else if (strcmp(vstring, "StaticColor") == 0) {
vis = StaticColor;
} else if (strcmp(vstring, "PseudoColor") == 0) {
vis = PseudoColor;
} else if (strcmp(vstring, "TrueColor") == 0) {
vis = TrueColor;
} else if (strcmp(vstring, "DirectColor") == 0) {
vis = DirectColor;
} else {
unsigned int v_in;
if (sscanf(vstring, "0x%x", &v_in) != 1) {
if (sscanf(vstring, "%u", &v_in) == 1) {
visual_id = (VisualID) v_in;
return;
}
rfbLogEnable(1);
rfbLog("invalid -visual arg: %s\n", vstring);
X_UNLOCK;
clean_up_exit(1);
}
visual_id = (VisualID) v_in;
free(vstring);
return;
}
if (! quiet) fprintf(stderr, " visual: %d\n", vis);
if (XMatchVisualInfo(dpy, scr, visual_depth, vis, &vinfo)) {
;
} else if (XMatchVisualInfo(dpy, scr, defdepth, vis, &vinfo)) {
;
} else {
rfbLogEnable(1);
rfbLog("could not find visual: %s\n", vstring);
X_UNLOCK;
clean_up_exit(1);
}
free(vstring);
/* set numerical visual id. */
visual_id = vinfo.visualid;
#endif /* NO_X11 */
}
void set_nofb_params(int restore) {
static int first = 1;
static int save[100];
static char *scroll = NULL;
int i = 0;
if (first) {
first = 0;
save[i++] = use_xfixes;
save[i++] = use_xdamage;
save[i++] = use_xrecord;
save[i++] = wireframe;
save[i++] = use_solid_bg;
save[i++] = overlay;
save[i++] = overlay_cursor;
save[i++] = using_shm;
save[i++] = single_copytile;
save[i++] = take_naps;
save[i++] = measure_speeds;
save[i++] = grab_buster;
save[i++] = show_cursor;
save[i++] = cursor_shape_updates;
save[i++] = cursor_pos_updates;
save[i++] = ncache;
scroll = scroll_copyrect;
}
if (restore) {
i = 0;
use_xfixes = save[i++];
use_xdamage = save[i++];
use_xrecord = save[i++];
wireframe = save[i++];
use_solid_bg = save[i++];
overlay = save[i++];
overlay_cursor = save[i++];
using_shm = save[i++];
single_copytile = save[i++];
take_naps = save[i++];
measure_speeds = save[i++];
grab_buster = save[i++];
show_cursor = save[i++];
cursor_shape_updates = save[i++];
cursor_pos_updates = save[i++];
ncache = save[i++];
scroll_copyrect = scroll;
if (cursor_shape_updates) {
restore_cursor_shape_updates(screen);
}
initialize_cursors_mode();
return;
}
use_xfixes = 0;
use_xdamage = 0;
use_xrecord = 0;
wireframe = 0;
use_solid_bg = 0;
overlay = 0;
overlay_cursor = 0;
using_shm = 0;
single_copytile = 1;
take_naps = 0;
measure_speeds = 0;
/* got_grab_buster? */
grab_buster = 0;
show_cursor = 0;
show_multiple_cursors = 0;
cursor_shape_updates = 0;
if (! got_cursorpos) {
cursor_pos_updates = 0;
}
ncache = 0;
scroll_copyrect = "never";
if (! quiet) {
rfbLog("disabling: xfixes, xdamage, solid, overlay, shm,\n");
rfbLog(" wireframe, scrollcopyrect, ncache,\n");
rfbLog(" noonetile, nap, cursor, %scursorshape\n",
got_cursorpos ? "" : "cursorpos, " );
rfbLog(" in -nofb mode.\n");
}
}
static char *raw_fb_orig_dpy = NULL;
void set_raw_fb_params(int restore) {
static int first = 1;
static int vo0, us0, sm0, ws0, wp0, wc0, wb0, na0, tn0;
static int xr0, xrm0, sb0, re0;
static char *mc0;
/*
* set turn off a bunch of parameters not compatible with
* -rawfb mode: 1) ignoring the X server 2) ignoring user input.
*/
if (first) {
/* at least save the initial settings... */
vo0 = view_only;
ws0 = watch_selection;
wp0 = watch_primary;
wc0 = watch_clipboard;
wb0 = watch_bell;
na0 = no_autorepeat;
sb0 = use_solid_bg;
us0 = use_snapfb;
sm0 = using_shm;
tn0 = take_naps;
xr0 = xrandr;
xrm0 = xrandr_maybe;
re0 = noxrecord;
mc0 = multiple_cursors_mode;
first = 0;
}
if (restore) {
view_only = vo0;
watch_selection = ws0;
watch_primary = wp0;
watch_clipboard = wc0;
watch_bell = wb0;
no_autorepeat = na0;
use_solid_bg = sb0;
use_snapfb = us0;
using_shm = sm0;
take_naps = tn0;
xrandr = xr0;
xrandr_maybe = xrm0;
noxrecord = re0;
multiple_cursors_mode = mc0;
if (! dpy && raw_fb_orig_dpy) {
dpy = XOpenDisplay_wr(raw_fb_orig_dpy);
last_open_xdisplay = time(NULL);
if (dpy) {
if (! quiet) rfbLog("reopened DISPLAY: %s\n",
raw_fb_orig_dpy);
scr = DefaultScreen(dpy);
rootwin = RootWindow(dpy, scr);
check_xevents(1);
} else {
if (! quiet) rfbLog("WARNING: failed to reopen "
"DISPLAY: %s\n", raw_fb_orig_dpy);
}
}
return;
}
if (verbose) {
rfbLog("set_raw_fb_params: modifying settings for "
"-rawfb mode.\n");
}
if (got_noviewonly) {
/*
* The user input parameters are not unset under
* -noviewonly... this usage should be very rare
* (i.e. rawfb but also send user input to the X
* display, most likely using /dev/fb0 for some reason...)
*/
if (verbose) {
rfbLog("rawfb: -noviewonly mode: still sending mouse and\n");
rfbLog("rawfb: keyboard input to the X DISPLAY!!\n");
}
} else {
/* Normal case: */
#if 0
if (! view_only && ! pipeinput_str) {
if (! quiet) rfbLog(" rawfb: setting view_only\n");
view_only = 1;
}
#endif
if (raw_fb_str && strstr(raw_fb_str, "vnc") == raw_fb_str) {
;
} else if (watch_selection) {
if (verbose) rfbLog(" rawfb: turning off "
"watch_selection\n");
watch_selection = 0;
}
if (watch_primary) {
if (verbose) rfbLog(" rawfb: turning off "
"watch_primary\n");
watch_primary = 0;
}
if (watch_clipboard) {
if (verbose) rfbLog(" rawfb: turning off "
"watch_clipboard\n");
watch_clipboard = 0;
}
if (watch_bell) {
if (verbose) rfbLog(" rawfb: turning off watch_bell\n");
watch_bell = 0;
}
if (no_autorepeat) {
if (verbose) rfbLog(" rawfb: turning off "
"no_autorepeat\n");
no_autorepeat = 0;
}
if (use_solid_bg) {
if (verbose) rfbLog(" rawfb: turning off "
"use_solid_bg\n");
use_solid_bg = 0;
}
#ifndef MACOSX
if (raw_fb_str && strstr(raw_fb_str, "vnc") == raw_fb_str) {
;
} else {
multiple_cursors_mode = strdup("arrow");
}
#endif
}
if (using_shm) {
if (verbose) rfbLog(" rawfb: turning off using_shm\n");
using_shm = 0;
}
if (take_naps) {
if (verbose) rfbLog(" rawfb: turning off take_naps\n");
take_naps = 0;
}
if (xrandr) {
if (verbose) rfbLog(" rawfb: turning off xrandr\n");
xrandr = 0;
}
if (xrandr_maybe) {
if (verbose) rfbLog(" rawfb: turning off xrandr_maybe\n");
xrandr_maybe = 0;
}
if (! noxrecord) {
if (verbose) rfbLog(" rawfb: turning off xrecord\n");
noxrecord = 1;
}
}
/*
* Presumably under -nofb the clients will never request the framebuffer.
* However, we have gotten such a request... so let's just give them
* the current view on the display. n.b. x2vnc and perhaps win2vnc
* requests a 1x1 pixel for some workaround so sadly this evidently
* nearly always happens.
*/
static void nofb_hook(rfbClientPtr cl) {
XImage *fb;
XImage raw;
rfbLog("framebuffer requested in -nofb mode by client %s\n", cl->host);
/* ignore xrandr */
if (raw_fb && ! dpy) {
fb = &raw;
fb->data = (char *)malloc(32);
} else {
int use_real_ximage = 0;
if (use_real_ximage) {
fb = XGetImage_wr(dpy, window, 0, 0, dpy_x, dpy_y, AllPlanes, ZPixmap);
} else {
fb = &raw;
fb->data = (char *) calloc(dpy_x*dpy_y*bpp/8, 1);
}
}
main_fb = fb->data;
rfb_fb = main_fb;
/* mutex */
screen->frameBuffer = rfb_fb;
screen->displayHook = NULL;
}
void free_old_fb(void) {
char *fbs[16];
int i, j, nfb = 0, db = 0;
fbs[nfb++] = main_fb; main_fb = NULL;
fbs[nfb++] = rfb_fb; rfb_fb = NULL;
fbs[nfb++] = cmap8to24_fb; cmap8to24_fb = NULL;
fbs[nfb++] = snap_fb; snap_fb = NULL;
fbs[nfb++] = rot_fb; rot_fb = NULL;
fbs[nfb++] = raw_fb; raw_fb = NULL;
for (i=0; i < nfb; i++) {
char *fb = fbs[i];
int freeit = 1;
if (! fb || fb < (char *) 0x10) {
continue;
}
for (j=0; j < i; j++) {
if (fb == fbs[j]) {
freeit = 0;
break;
}
}
if (freeit) {
if (db) fprintf(stderr, "free: %i %p\n", i, fb);
free(fb);
} else {
if (db) fprintf(stderr, "skip: %i %p\n", i, fb);
}
}
}
static char _lcs_tmp[128];
static int _bytes0_size = 128, _bytes0[128];
static char *lcs(rfbClientPtr cl) {
sprintf(_lcs_tmp, "%d/%d/%d/%d/%d-%d/%d/%d",
!!(cl->newFBSizePending),
!!(cl->cursorWasChanged),
!!(cl->cursorWasMoved),
!!(cl->reverseConnection),
cl->state,
cl->modifiedRegion ? !!(sraRgnEmpty(cl->modifiedRegion)) : 2,
cl->requestedRegion ? !!(sraRgnEmpty(cl->requestedRegion)) : 2,
cl->copyRegion ? !!(sraRgnEmpty(cl->copyRegion)) : 2
);
return _lcs_tmp;
}
static int lock_client_sends(int lock) {
static rfbClientPtr *cls = NULL;
static int cls_len = 0;
static int blocked = 0;
static int state = 0;
rfbClientIteratorPtr iter;
rfbClientPtr cl;
char *s;
if (!use_threads || !screen) {
return 0;
}
if (lock < 0) {
return state;
}
state = lock;
if (lock) {
if (cls_len < client_count + 128) {
if (cls != NULL) {
free(cls);
}
cls_len = client_count + 256;
cls = (rfbClientPtr *) calloc(cls_len * sizeof(rfbClientPtr), 1);
}
iter = rfbGetClientIterator(screen);
blocked = 0;
while ((cl = rfbClientIteratorNext(iter)) != NULL) {
s = lcs(cl);
SEND_LOCK(cl);
rfbLog("locked client: %p %.6f %s\n", cl, dnowx(), s);
cls[blocked++] = cl;
}
rfbReleaseClientIterator(iter);
} else {
int i;
for (i=0; i < blocked; i++) {
cl = cls[i];
if (cl != NULL) {
s = lcs(cl);
SEND_UNLOCK(cl)
rfbLog("unlocked client: %p %.6f %s\n", cl, dnowx(), s);
}
cls[i] = NULL;
}
blocked = 0;
}
return state;
}
static void settle_clients(int init) {
rfbClientIteratorPtr iter;
rfbClientPtr cl;
int fb_pend, i, ms = 1000;
char *s;
if (!use_threads || !screen) {
return;
}
if (init) {
iter = rfbGetClientIterator(screen);
i = 0;
while ((cl = rfbClientIteratorNext(iter)) != NULL) {
if (i < _bytes0_size) {
_bytes0[i] = rfbStatGetSentBytesIfRaw(cl);
}
i++;
}
rfbReleaseClientIterator(iter);
if (getenv("X11VNC_THREADS_NEW_FB_SLEEP")) {
ms = atoi(getenv("X11VNC_THREADS_NEW_FB_SLEEP"));
} else if (subwin) {
ms = 250;
} else {
ms = 500;
}
usleep(ms * 1000);
return;
}
if (getenv("X11VNC_THREADS_NEW_FB_SLEEP")) {
ms = atoi(getenv("X11VNC_THREADS_NEW_FB_SLEEP"));
} else if (subwin) {
ms = 500;
} else {
ms = 1000;
}
usleep(ms * 1000);
for (i=0; i < 5; i++) {
fb_pend = 0;
iter = rfbGetClientIterator(screen);
while ((cl = rfbClientIteratorNext(iter)) != NULL) {
s = lcs(cl);
if (cl->newFBSizePending) {
fb_pend++;
rfbLog("pending fb size: %p %.6f %s\n", cl, dnowx(), s);
}
}
rfbReleaseClientIterator(iter);
if (fb_pend > 0) {
rfbLog("do_new_fb: newFBSizePending extra -threads sleep (%d)\n", i+1);
usleep(ms * 1000);
} else {
break;
}
}
for (i=0; i < 5; i++) {
int stuck = 0, tot = 0, j = 0;
iter = rfbGetClientIterator(screen);
while ((cl = rfbClientIteratorNext(iter)) != NULL) {
if (j < _bytes0_size) {
int db = rfbStatGetSentBytesIfRaw(cl) - _bytes0[j];
int Bpp = cl->format.bitsPerPixel / 8;
s = lcs(cl);
rfbLog("addl bytes sent: %p %.6f %s %d %d\n",
cl, dnowx(), s, db, _bytes0[j]);
if (i==0) {
if (db < Bpp * dpy_x * dpy_y) {
stuck++;
}
} else if (i==1) {
if (db < 0.5 * Bpp * dpy_x * dpy_y) {
stuck++;
}
} else {
if (db <= 0) {
stuck++;
}
}
}
tot++;
j++;
}
rfbReleaseClientIterator(iter);
if (stuck > 0) {
rfbLog("clients stuck: %d/%d sleep(%d)\n", stuck, tot, i);
usleep(2 * ms * 1000);
} else {
break;
}
}
}
static void prep_clients_for_new_fb(void) {
rfbClientIteratorPtr iter;
rfbClientPtr cl;
if (!use_threads || !screen) {
return;
}
iter = rfbGetClientIterator(screen);
while ((cl = rfbClientIteratorNext(iter)) != NULL) {
if (!cl->newFBSizePending) {
rfbLog("** set_new_fb_size_pending client: %p\n", cl);
cl->newFBSizePending = TRUE;
}
cl->cursorWasChanged = FALSE;
cl->cursorWasMoved = FALSE;
}
rfbReleaseClientIterator(iter);
}
void do_new_fb(int reset_mem) {
XImage *fb;
/* for threaded we really should lock libvncserver out. */
if (use_threads) {
int ms = 1000;
if (getenv("X11VNC_THREADS_NEW_FB_SLEEP")) {
ms = atoi(getenv("X11VNC_THREADS_NEW_FB_SLEEP"));
} else if (subwin) {
ms = 500;
} else {
ms = 1000;
}
rfbLog("Warning: changing framebuffers in threaded mode may be unstable.\n");
threads_drop_input = 1;
usleep(ms * 1000);
}
INPUT_LOCK;
lock_client_sends(1);
if (use_threads) {
settle_clients(1);
}
#ifdef MACOSX
if (macosx_console) {
macosxCG_fini();
}
#endif
if (reset_mem == 1) {
/* reset_mem == 2 is a hack for changing users... */
clean_shm(0);
free_tiles();
}
free_old_fb();
fb = initialize_xdisplay_fb();
initialize_screen(NULL, NULL, fb);
if (reset_mem) {
initialize_tiles();
initialize_blackouts_and_xinerama();
initialize_polling_images();
}
if (ncache) {
check_ncache(1, 0);
}
prep_clients_for_new_fb();
lock_client_sends(0);
INPUT_UNLOCK;
if (use_threads) {
/* need to let things settle... */
settle_clients(0);
threads_drop_input = 0;
}
}
static void remove_fake_fb(void) {
if (! screen) {
return;
}
rfbLog("removing fake fb: 0x%x\n", fake_fb);
do_new_fb(1);
/*
* fake_fb is freed in do_new_fb(), but we set to NULL here to
* indicate it is gone.
*/
fake_fb = NULL;
}
static void rfb_new_framebuffer(rfbScreenInfoPtr rfbScreen, char *framebuffer,
int width,int height, int bitsPerSample,int samplesPerPixel,
int bytesPerPixel) {
rfbNewFramebuffer(rfbScreen, framebuffer, width, height, bitsPerSample,
samplesPerPixel, bytesPerPixel);
}
static void install_fake_fb(int w, int h, int bpp) {
int bpc;
if (! screen) {
return;
}
lock_client_sends(1);
if (fake_fb) {
free(fake_fb);
}
fake_fb = (char *) calloc(w*h*bpp/8, 1);
if (! fake_fb) {
rfbLog("could not create fake fb: %dx%d %d\n", w, h, bpp);
lock_client_sends(0);
return;
}
bpc = guess_bits_per_color(bpp);
rfbLog("installing fake fb: %dx%d %d\n", w, h, bpp);
rfbLog("rfbNewFramebuffer(0x%x, 0x%x, %d, %d, %d, %d, %d)\n",
screen, fake_fb, w, h, bpc, 1, bpp/8);
rfb_new_framebuffer(screen, fake_fb, w, h, bpc, 1, bpp/8);
lock_client_sends(0);
}
void check_padded_fb(void) {
if (! fake_fb) {
return;
}
if (unixpw_in_progress) return;
if (time(NULL) > pad_geometry_time+1 && all_clients_initialized()) {
remove_fake_fb();
}
}
void install_padded_fb(char *geom) {
int w, h;
int ok = 1;
if (! geom || *geom == '\0') {
ok = 0;
} else if (sscanf(geom, "%dx%d", &w, &h) != 2) {
ok = 0;
}
w = nabs(w);
h = nabs(h);
if (w < 5) w = 5;
if (h < 5) h = 5;
if (!ok) {
rfbLog("skipping invalid pad geometry: '%s'\n", NONUL(geom));
return;
}
install_fake_fb(w, h, bpp);
pad_geometry_time = time(NULL);
}
static void initialize_snap_fb(void) {
RAWFB_RET_VOID
if (snap_fb) {
free(snap_fb);
}
snap = XGetImage_wr(dpy, window, 0, 0, dpy_x, dpy_y, AllPlanes,
ZPixmap);
snap_fb = snap->data;
}
static rfbClient* client = NULL;
void vnc_reflect_bell(rfbClient *cl) {
if (cl) {}
if (sound_bell) {
if (unixpw_in_progress) {
return;
}
if (! all_clients_initialized()) {
rfbLog("vnc_reflect_bell: not sending bell: "
"uninitialized clients\n");
} else {
if (screen && client_count) {
rfbSendBell(screen);
}
}
}
}
void vnc_reflect_recv_cuttext(rfbClient *cl, const char *str, int len) {
if (cl) {}
if (unixpw_in_progress) {
return;
}
if (! watch_selection) {
return;
}
if (! all_clients_initialized()) {
rfbLog("vnc_reflect_recv_cuttext: no send: uninitialized clients\n");
return; /* some clients initializing, cannot send */
}
rfbSendServerCutText(screen, (char *)str, len);
}
void vnc_reflect_got_update(rfbClient *cl, int x, int y, int w, int h) {
if (cl) {}
if (use_xdamage) {
static int first = 1;
if (first) {
collect_non_X_xdamage(-1, -1, -1, -1, 0);
first = 0;
}
collect_non_X_xdamage(x, y, w, h, 1);
}
}
void vnc_reflect_got_cursorshape(rfbClient *cl, int xhot, int yhot, int width, int height, int bytesPerPixel) {
static int serial = 1;
int i, j;
char *pixels = NULL;
unsigned long r, g, b;
unsigned int ui = 0;
unsigned long red_mask, green_mask, blue_mask;
if (cl) {}
if (unixpw_in_progress) {
return;
}
if (! all_clients_initialized()) {
rfbLog("vnc_reflect_got_copyshape: no send: uninitialized clients\n");
return; /* some clients initializing, cannot send */
}
if (! client->rcSource) {
return;
}
if (bytesPerPixel != 1 && bytesPerPixel != 2 && bytesPerPixel != 4) {
return;
}
red_mask = (client->format.redMax << client->format.redShift);
green_mask = (client->format.greenMax << client->format.greenShift);
blue_mask = (client->format.blueMax << client->format.blueShift);
pixels = (char *)malloc(4*width*height);
for (j=0; j<height; j++) {
for (i=0; i<width; i++) {
unsigned int* uip;
unsigned char* uic;
int m;
if (bytesPerPixel == 1) {
unsigned char* p = (unsigned char *) client->rcSource;
ui = (unsigned long) *(p + j * width + i);
} else if (bytesPerPixel == 2) {
unsigned short* p = (unsigned short *) client->rcSource;
ui = (unsigned long) *(p + j * width + i);
} else if (bytesPerPixel == 4) {
unsigned int* p = (unsigned int *) client->rcSource;
ui = (unsigned long) *(p + j * width + i);
}
r = (red_mask & ui) >> client->format.redShift;
g = (green_mask & ui) >> client->format.greenShift;
b = (blue_mask & ui) >> client->format.blueShift;
r = (255 * r) / client->format.redMax;
g = (255 * g) / client->format.greenMax;
b = (255 * b) / client->format.blueMax;
ui = (r << 16 | g << 8 | b << 0) ;
uic = (unsigned char *)client->rcMask;
m = (int) *(uic + j * width + i);
if (m) {
ui |= 0xff000000;
}
uip = (unsigned int *)pixels;
*(uip + j * width + i) = ui;
}
}
store_cursor(serial++, (unsigned long*) pixels, width, height, 32, xhot, yhot);
free(pixels);
set_cursor(cursor_x, cursor_y, get_which_cursor());
}
rfbBool vnc_reflect_cursor_pos(rfbClient *cl, int x, int y) {
if (cl) {}
if (debug_pointer) {
rfbLog("vnc_reflect_cursor_pos: %d %d\n", x, y);
}
if (unixpw_in_progress) {
if (debug_pointer) {
rfbLog("vnc_reflect_cursor_pos: unixpw_in_progress%d\n", unixpw_in_progress);
}
return TRUE;
}
if (! all_clients_initialized()) {
rfbLog("vnc_reflect_cursor_pos: no send: uninitialized clients\n");
return TRUE; /* some clients initializing, cannot send */
}
cursor_position(x, y);
set_cursor(x, y, get_which_cursor());
return TRUE;
}
static void from_libvncclient_CopyRectangleFromRectangle(rfbClient* client, int src_x, int src_y, int w, int h, int dest_x, int dest_y) {
int i,j;
#define COPY_RECT_FROM_RECT(BPP) \
{ \
uint##BPP##_t* _buffer=((uint##BPP##_t*)client->frameBuffer)+(src_y-dest_y)*client->width+src_x-dest_x; \
if (dest_y < src_y) { \
for(j = dest_y*client->width; j < (dest_y+h)*client->width; j += client->width) { \
if (dest_x < src_x) { \
for(i = dest_x; i < dest_x+w; i++) { \
((uint##BPP##_t*)client->frameBuffer)[j+i]=_buffer[j+i]; \
} \
} else { \
for(i = dest_x+w-1; i >= dest_x; i--) { \
((uint##BPP##_t*)client->frameBuffer)[j+i]=_buffer[j+i]; \
} \
} \
} \
} else { \
for(j = (dest_y+h-1)*client->width; j >= dest_y*client->width; j-=client->width) { \
if (dest_x < src_x) { \
for(i = dest_x; i < dest_x+w; i++) { \
((uint##BPP##_t*)client->frameBuffer)[j+i]=_buffer[j+i]; \
} \
} else { \
for(i = dest_x+w-1; i >= dest_x; i--) { \
((uint##BPP##_t*)client->frameBuffer)[j+i]=_buffer[j+i]; \
} \
} \
} \
} \
}
switch(client->format.bitsPerPixel) {
case 8: COPY_RECT_FROM_RECT(8); break;
case 16: COPY_RECT_FROM_RECT(16); break;
case 32: COPY_RECT_FROM_RECT(32); break;
default:
rfbClientLog("Unsupported bitsPerPixel: %d\n",client->format.bitsPerPixel);
}
}
void vnc_reflect_got_copyrect(rfbClient *cl, int src_x, int src_y, int w, int h, int dest_x, int dest_y) {
sraRegionPtr reg;
int dx, dy, rc = -1;
static int last_dx = 0, last_dy = 0;
if (cl) {}
if (unixpw_in_progress) {
return;
}
if (! all_clients_initialized()) {
rfbLog("vnc_reflect_got_copyrect: no send: uninitialized clients\n");
return; /* some clients initializing, cannot send */
}
dx = dest_x - src_x;
dy = dest_y - src_y;
if (dx != last_dx || dy != last_dy) {
rc = fb_push_wait(0.05, FB_COPY|FB_MOD);
}
if (0) fprintf(stderr, "vnc_reflect_got_copyrect: %03dx%03d+%03d+%03d %3d %3d rc=%d\n", dest_x, dest_y, w, h, dx, dy, rc);
reg = sraRgnCreateRect(dest_x, dest_y, dest_x + w, dest_y + h);
do_copyregion(reg, dx, dy, 0);
sraRgnDestroy(reg);
last_dx = dx;
last_dy = dy;
from_libvncclient_CopyRectangleFromRectangle(cl, src_x, src_y, w, h, dest_x, dest_y);
}
rfbBool vnc_reflect_resize(rfbClient *cl) {
static int first = 1;
if(cl->frameBuffer) {
free(cl->frameBuffer);
}
cl->frameBuffer= malloc(cl->width * cl->height * cl->format.bitsPerPixel/8);
rfbLog("vnc_reflect_resize: %dx%dx%d first=%d\n", cl->width, cl->height,
cl->format.bitsPerPixel, first);
if (!first) {
do_new_fb(1);
}
first = 0;
return cl->frameBuffer ? TRUE : FALSE;
}
#ifdef rfbCredentialTypeX509
static rfbCredential* vnc_reflect_get_credential(rfbClient* client, int type) {
char *pass = getenv("X11VNC_REFLECT_PASSWORD");
char *user = getenv("X11VNC_REFLECT_USER");
char *cert = getenv("X11VNC_REFLECT_CACERT");
char *ccrl = getenv("X11VNC_REFLECT_CACRL");
char *clic = getenv("X11VNC_REFLECT_CLIENTCERT");
char *clik = getenv("X11VNC_REFLECT_CLIENTKEY");
int db = 0;
if (client) {}
if (db) fprintf(stderr, "type: %d\n", type);
#ifdef rfbCredentialTypeUser
if (type == rfbCredentialTypeUser) {
if (!pass && !user) {
return NULL;
} else {
rfbCredential *rc = (rfbCredential *) calloc(sizeof(rfbCredential), 1);
rc->userCredential.username = (user ? strdup(user) : NULL);
rc->userCredential.password = (pass ? strdup(pass) : NULL);
return rc;
}
}
#endif
if (type == rfbCredentialTypeX509) {
if (db) fprintf(stderr, "cert: %s\n", cert);
if (db) fprintf(stderr, "ccrl: %s\n", ccrl);
if (db) fprintf(stderr, "clic: %s\n", clic);
if (db) fprintf(stderr, "clik: %s\n", clik);
if (!cert && !ccrl && !clic && !clik) {
return NULL;
} else {
rfbCredential *rc = (rfbCredential *) calloc(sizeof(rfbCredential), 1);
rc->x509Credential.x509CACertFile = (cert ? strdup(cert) : NULL);
rc->x509Credential.x509CACrlFile = (ccrl ? strdup(ccrl) : NULL);
rc->x509Credential.x509ClientCertFile = (clic ? strdup(clic) : NULL);
rc->x509Credential.x509ClientKeyFile = (clik ? strdup(clik) : NULL);
return rc;
}
}
return NULL;
}
#endif
static char* vnc_reflect_get_password(rfbClient* client) {
char *q, *p, *str = getenv("X11VNC_REFLECT_PASSWORD");
int len = 110;
if (client) {}
if (str) {
len += 2*strlen(str);
}
p = (char *) calloc(len, 1);
if (!str || strlen(str) == 0) {
fprintf(stderr, "VNC Reflect Password: ");
fgets(p, 100, stdin);
} else {
if (strstr(str, "file:") == str) {
FILE *f = fopen(str + strlen("file:"), "r");
if (f) {
fgets(p, 100, f);
fclose(f);
}
}
if (p[0] == '\0') {
strncpy(p, str, 100);
}
}
q = p;
while (*q != '\0') {
if (*q == '\n') {
*q = '\0';
}
q++;
}
return p;
}
char *vnc_reflect_guess(char *str, char **raw_fb_addr) {
static int first = 1;
char *hp = str + strlen("vnc:");
char *at = NULL;
int argc = 0, i;
char *argv[16];
char str2[256];
char *str0 = strdup(str);
if (client == NULL) {
int bitsPerSample = 8;
int samplesPerPixel = 3;
int bytesPerPixel = 4;
char *s;
s = getenv("X11VNC_REFLECT_bitsPerSample");
if (s) bitsPerSample = atoi(s);
s = getenv("X11VNC_REFLECT_samplesPerPixel");
if (s) samplesPerPixel = atoi(s);
s = getenv("X11VNC_REFLECT_bytesPerPixel");
if (s) bytesPerPixel = atoi(s);
rfbLog("rfbGetClient(bitsPerSample=%d, samplesPerPixel=%d, bytesPerPixel=%d)\n",
bitsPerSample, samplesPerPixel, bytesPerPixel);
client = rfbGetClient(bitsPerSample, samplesPerPixel, bytesPerPixel);
}
rfbLog("rawfb: %s\n", str);
at = strchr(hp, '@');
if (at) {
*at = '\0';
at++;
}
client->appData.useRemoteCursor = TRUE;
client->canHandleNewFBSize = TRUE;
client->HandleCursorPos = vnc_reflect_cursor_pos;
client->GotFrameBufferUpdate = vnc_reflect_got_update;
client->MallocFrameBuffer = vnc_reflect_resize;
client->Bell = vnc_reflect_bell;
#if 0
client->SoftCursorLockArea = NULL;
client->SoftCursorUnlockScreen = NULL;
client->FinishedFrameBufferUpdate = NULL;
client->HandleKeyboardLedState = NULL;
client->HandleTextChat = NULL;
#endif
client->GotXCutText = vnc_reflect_recv_cuttext;
client->GotCursorShape = vnc_reflect_got_cursorshape;
client->GotCopyRect = vnc_reflect_got_copyrect;
if (getenv("X11VNC_REFLECT_PASSWORD")) {
client->GetPassword = vnc_reflect_get_password;
}
#ifdef rfbCredentialTypeX509
client->GetCredential = NULL;
if (0 || getenv("LIBVNCCLIENT_GET_CREDENTIAL")) {
client->GetCredential = vnc_reflect_get_credential;
}
#endif
if (first) {
argv[argc++] = "x11vnc_rawfb_vnc";
if (strstr(hp, "listen") == hp) {
char *q = strrchr(hp, ':');
argv[argc++] = strdup("-listen");
if (q) {
client->listenPort = atoi(q+1);
} else {
client->listenPort = LISTEN_PORT_OFFSET;
}
} else {
argv[argc++] = strdup(hp);
}
if (! rfbInitClient(client, &argc, argv)) {
rfbLog("vnc_reflector failed for: %s\n", str0);
clean_up_exit(1);
}
}
if (at) {
sprintf(str2, "map:/dev/null@%s", at);
} else {
unsigned long red_mask, green_mask, blue_mask;
red_mask = (client->format.redMax << client->format.redShift);
green_mask = (client->format.greenMax << client->format.greenShift);
blue_mask = (client->format.blueMax << client->format.blueShift);
sprintf(str2, "map:/dev/null@%dx%dx%d:0x%lx/0x%lx/0x%lx",
client->width, client->height, client->format.bitsPerPixel,
red_mask, green_mask, blue_mask);
}
*raw_fb_addr = (char *) client->frameBuffer;
free(str0);
if (first) {
setup_cursors_and_push();
for (i=0; i<10; i++) {
vnc_reflect_process_client();
}
}
first = 0;
return strdup(str2);
}
rfbBool vnc_reflect_send_pointer(int x, int y, int mask) {
int rc;
if (mask >= 0) {
got_user_input++;
got_pointer_input++;
last_pointer_time = time(NULL);
}
if (clipshift) {
x += coff_x;
y += coff_y;
}
if (cursor_x != x || cursor_y != y) {
last_pointer_motion_time = dnow();
}
cursor_x = x;
cursor_y = y;
/* record the x, y position for the rfb screen as well. */
cursor_position(x, y);
/* change the cursor shape if necessary */
rc = set_cursor(x, y, get_which_cursor());
cursor_changes += rc;
return SendPointerEvent(client, x, y, mask);
}
rfbBool vnc_reflect_send_key(uint32_t key, rfbBool down) {
return SendKeyEvent(client, key, down);
}
rfbBool vnc_reflect_send_cuttext(char *str, int len) {
return SendClientCutText(client, str, len);
}
void vnc_reflect_process_client(void) {
int num;
if (client == NULL) {
return;
}
num = WaitForMessage(client, 1000);
if (num > 0) {
if (!HandleRFBServerMessage(client)) {
rfbLog("vnc_reflect_process_client: read failure to server\n");
shut_down = 1;
}
}
}
void linux_dev_fb_msg(char* q) {
if (strstr(q, "/dev/fb") && strstr(UT.sysname, "Linux")) {
rfbLog("\n");
rfbLog("On Linux you may need to load a kernel module to enable\n");
rfbLog("the framebuffer device /dev/fb*; e.g.:\n");
rfbLog(" vga=0x303 (and others) kernel boot parameter\n");
rfbLog(" modprobe uvesafb\n");
rfbLog(" modprobe radeonfb (card specific)\n");
rfbLog(" modprobe nvidiafb (card specific, others)\n");
rfbLog(" modprobe vesafb (?)\n");
rfbLog(" modprobe vga16fb\n");
rfbLog("\n");
rfbLog("You may also need root permission to open /dev/fb*\n");
rfbLog("and/or /dev/tty*.\n");
rfbLog("\n");
}
}
#define RAWFB_MMAP 1
#define RAWFB_FILE 2
#define RAWFB_SHM 3
XImage *initialize_raw_fb(int reset) {
char *str, *rstr, *q;
int w, h, b, shmid = 0;
unsigned long rm = 0, gm = 0, bm = 0, tm;
static XImage ximage_struct; /* n.b.: not (XImage *) */
static XImage ximage_struct_snap;
int closedpy = 1, i, m, db = 0;
int do_macosx = 0;
int do_reflect = 0;
char *unlink_me = NULL;
static char *last_file = NULL;
static int last_mode = 0;
if (reset && last_mode) {
int fd;
if (last_mode != RAWFB_MMAP && last_mode != RAWFB_FILE) {
return NULL;
}
if (last_mode == RAWFB_MMAP) {
munmap(raw_fb_addr, raw_fb_mmap);
}
if (raw_fb_fd >= 0) {
close(raw_fb_fd);
}
raw_fb_fd = -1;
if (db) fprintf(stderr, "initialize_raw_fb reset\n");
fd = -1;
if (rawfb_dev_video) {
fd = open(last_file, O_RDWR);
}
if (fd < 0) {
fd = open(last_file, O_RDONLY);
}
if (fd < 0) {
rfbLogEnable(1);
rfbLog("failed to rawfb file: %s\n", last_file);
rfbLogPerror("open");
clean_up_exit(1);
}
raw_fb_fd = fd;
if (last_mode == RAWFB_MMAP) {
raw_fb_addr = mmap(0, raw_fb_mmap, PROT_READ,
MAP_SHARED, fd, 0);
if (raw_fb_addr == MAP_FAILED || raw_fb_addr == NULL) {
rfbLogEnable(1);
rfbLog("failed to mmap file: %s\n", last_file);
rfbLog(" raw_fb_addr: %p\n", raw_fb_addr);
rfbLogPerror("mmap");
clean_up_exit(1);
}
}
return NULL;
}
#ifdef MACOSX
if (raw_fb_addr != NULL && macosx_console && raw_fb_addr == macosx_get_fb_addr()) {
raw_fb_addr = NULL;
}
#endif
if (raw_fb_addr || raw_fb_seek) {
if (raw_fb_shm) {
shmdt(raw_fb_addr);
#if LIBVNCSERVER_HAVE_MMAP
} else if (raw_fb_mmap) {
munmap(raw_fb_addr, raw_fb_mmap);
if (raw_fb_fd >= 0) {
close(raw_fb_fd);
}
raw_fb_fd = -1;
#endif
} else if (raw_fb_seek) {
if (raw_fb_fd >= 0) {
close(raw_fb_fd);
}
raw_fb_fd = -1;
}
raw_fb_addr = NULL;
raw_fb_mmap = 0;
raw_fb_seek = 0;
}
if (! raw_fb_str) {
return NULL;
}
if (raw_fb_str[0] == '+') {
rstr = strdup(raw_fb_str+1);
closedpy = 0;
if (! window) {
window = rootwin;
}
} else {
rstr = strdup(raw_fb_str);
}
/* testing aliases */
if (!strcasecmp(rstr, "NULL") || !strcasecmp(rstr, "ZERO")
|| !strcasecmp(rstr, "NONE")) {
rstr = strdup("map:/dev/zero@640x480x32");
} else if (!strcasecmp(rstr, "NULLBIG") || !strcasecmp(rstr, "NONEBIG")) {
rstr = strdup("map:/dev/zero@1024x768x32");
}
if (!strcasecmp(rstr, "RAND")) {
rstr = strdup("file:/dev/urandom@128x128x16");
} else if (!strcasecmp(rstr, "RANDBIG")) {
rstr = strdup("file:/dev/urandom@640x480x16");
} else if (!strcasecmp(rstr, "RANDHUGE")) {
rstr = strdup("file:/dev/urandom@1024x768x16");
}
if (strstr(rstr, "solid=") == rstr) {
char *n = rstr + strlen("solid=");
char tmp[] = "/tmp/rawfb_solid.XXXXXX";
char str[100];
unsigned int vals[1024], val;
int x, y, fd, w = 1024, h = 768;
if (strstr(n, "0x")) {
if (sscanf(n, "0x%x", &val) != 1) {
val = 0;
}
}
if (val == 0) {
val = get_pixel(n);
}
if (val == 0) {
val = 0xFF00FF;
}
fd = mkstemp(tmp);
for (y = 0; y < h; y++) {
for (x = 0; x < w; x++) {
vals[x] = val;
}
write(fd, (char *)vals, 4 * w);
}
close(fd);
fd = open(tmp, O_WRONLY);
unlink_me = strdup(tmp);
sprintf(str, "map:%s@%dx%dx32", tmp, w, h);
rstr = strdup(str);
} else if (strstr(rstr, "swirl") == rstr) {
char tmp[] = "/tmp/rawfb_swirl.XXXXXX";
char str[100];
unsigned int val[1024];
unsigned int c1, c2, c3, c4;
int x, y, fd, w = 1024, h = 768;
fd = mkstemp(tmp);
for (y = 0; y < h; y++) {
for (x = 0; x < w; x++) {
c1 = 0;
c2 = ((x+y)*128)/(w+h);
c3 = (x*128)/w;
c4 = (y*256)/h;
val[x] = (c1 << 24) | (c2 << 16) | (c3 << 8) | (c4 << 0);
}
write(fd, (char *)val, 4 * w);
}
close(fd);
fd = open(tmp, O_WRONLY);
unlink_me = strdup(tmp);
sprintf(str, "map:%s@%dx%dx32", tmp, w, h);
rstr = strdup(str);
}
if ( (q = strstr(rstr, "setup:")) == rstr) {
FILE *pipe;
char line[1024], *t;
set_child_info();
q += strlen("setup:");
/* rawfb-setup */
if (no_external_cmds || !cmd_ok("rawfb-setup")) {
rfbLogEnable(1);
rfbLog("cannot run external commands in -nocmds "
"mode:\n");
rfbLog(" \"%s\"\n", q);
rfbLog(" exiting.\n");
clean_up_exit(1);
}
rfbLog("running command to setup rawfb: %s\n", q);
close_exec_fds();
pipe = popen(q, "r");
if (! pipe) {
rfbLogEnable(1);
rfbLog("popen of setup command failed.\n");
rfbLogPerror("popen");
clean_up_exit(1);
}
line[0] = '\0';
if (fgets(line, 1024, pipe) == NULL) {
rfbLogEnable(1);
rfbLog("read of setup command failed.\n");
clean_up_exit(1);
}
pclose(pipe);
str = strdup(line);
t = str;
while (*t != '\0') {
if (*t == '\n') {
*t = '\0';
}
t++;
}
rfbLog("setup command returned: %s\n", str);
} else {
str = strdup(rstr);
}
raw_fb_shm = 0;
raw_fb_mmap = 0;
raw_fb_seek = 0;
raw_fb_fd = -1;
raw_fb_addr = NULL;
raw_fb_offset = 0;
raw_fb_bytes_per_line = 0;
rawfb_vnc_reflect = 0;
last_mode = 0;
if (last_file) {
free(last_file);
last_file = NULL;
}
if (strstr(str, "Video") == str) {
if (pipeinput_str != NULL) {
free(pipeinput_str);
}
pipeinput_str = strdup("VID");
initialize_pipeinput();
str[0] = 'v';
}
if (strstr(str, "video") == str || strstr(str, "/dev/video") == str) {
char *str2 = v4l_guess(str, &raw_fb_fd);
if (str2 == NULL) {
rfbLog("v4l_guess failed for: %s\n", str);
clean_up_exit(1);
}
str = str2;
rfbLog("v4l_guess returned: %s\n", str);
rawfb_dev_video = 1;
} else if (strstr(str, "dev/video")) {
rawfb_dev_video = 1;
} else if (strstr(str, "console") == str || strstr(str, "fb") == str ||
strstr(str, "/dev/fb") == str || strstr(str, "vt") == str) {
char *str2 = console_guess(str, &raw_fb_fd);
if (str2 == NULL) {
rfbLog("console_guess failed for: %s\n", str);
clean_up_exit(1);
}
str = str2;
rfbLog("console_guess returned: %s\n", str);
} else if (strstr(str, "vnc:") == str) {
char *str2 = vnc_reflect_guess(str, &raw_fb_addr);
rawfb_vnc_reflect = 1;
do_reflect = 1;
str = str2;
rfbLog("vnc_reflector set rawfb str to: %s\n", str);
if (pipeinput_str == NULL) {
pipeinput_str = strdup("VNC");
}
initialize_pipeinput();
}
if (closedpy && !view_only && got_noviewonly) {
rfbLog("not closing X DISPLAY under -noviewonly option.\n");
closedpy = 0;
if (! window) {
window = rootwin;
}
}
if (! raw_fb_orig_dpy && dpy) {
raw_fb_orig_dpy = strdup(DisplayString(dpy));
}
#ifndef BOLDLY_CLOSE_DISPLAY
#define BOLDLY_CLOSE_DISPLAY 1
#endif
#if BOLDLY_CLOSE_DISPLAY
if (closedpy) {
if (dpy) {
rfbLog("closing X DISPLAY: %s in rawfb mode.\n",
DisplayString(dpy));
XCloseDisplay_wr(dpy); /* yow! */
}
dpy = NULL;
}
#endif
/*
* -rawfb shm:163938442@640x480x32:ff/ff00/ff0000+3000
* -rawfb map:/path/to/file@640x480x32:ff/ff00/ff0000
* -rawfb file:/path/to/file@640x480x32:ff/ff00/ff0000
*/
if (raw_fb_full_str) {
free(raw_fb_full_str);
}
raw_fb_full_str = strdup(str);
/* +O offset */
if ((q = strrchr(str, '+')) != NULL) {
if (sscanf(q, "+%d", &raw_fb_offset) == 1) {
*q = '\0';
} else {
raw_fb_offset = 0;
}
}
/* :R/G/B masks */
if ((q = strrchr(str, ':')) != NULL) {
if (sscanf(q, ":%lx/%lx/%lx", &rm, &gm, &bm) == 3) {
*q = '\0';
} else if (sscanf(q, ":0x%lx/0x%lx/0x%lx", &rm, &gm, &bm)== 3) {
*q = '\0';
} else if (sscanf(q, ":%lu/%lu/%lu", &rm, &gm, &bm) == 3) {
*q = '\0';
} else {
rm = 0;
gm = 0;
bm = 0;
}
}
if ((q = strrchr(str, '@')) == NULL) {
rfbLogEnable(1);
rfbLog("invalid rawfb str: %s\n", str);
clean_up_exit(1);
}
if (strrchr(q, '-')) {
char *q2 = strrchr(q, '-');
raw_fb_bytes_per_line = atoi(q2+1);
*q2 = '\0';
}
/* @WxHxB */
if (sscanf(q, "@%dx%dx%d", &w, &h, &b) != 3) {
rfbLogEnable(1);
rfbLog("invalid rawfb str: %s\n", str);
clean_up_exit(1);
}
*q = '\0';
if (rm == 0 && gm == 0 && bm == 0) {
/* guess masks... */
if (b == 24 || b == 32) {
rm = 0xff0000;
gm = 0x00ff00;
bm = 0x0000ff;
} else if (b == 16) {
rm = 0xf800;
gm = 0x07e0;
bm = 0x001f;
} else if (b == 8) {
rm = 0x07;
gm = 0x38;
bm = 0xc0;
}
}
/* we can fake -flipbyteorder to some degree... */
if (flip_byte_order) {
if (b == 24 || b == 32) {
tm = rm;
rm = bm;
bm = tm;
} else if (b == 16) {
unsigned short s1, s2;
s1 = (unsigned short) rm;
s2 = ((0xff & s1) << 8) | ((0xff00 & s1) >> 8);
rm = (unsigned long) s2;
s1 = (unsigned short) gm;
s2 = ((0xff & s1) << 8) | ((0xff00 & s1) >> 8);
gm = (unsigned long) s2;
s1 = (unsigned short) bm;
s2 = ((0xff & s1) << 8) | ((0xff00 & s1) >> 8);
bm = (unsigned long) s2;
}
}
/* native fb stuff for bpp < 8 only */
raw_fb_native_bpp = b;
raw_fb_native_red_mask = rm;
raw_fb_native_green_mask = gm;
raw_fb_native_blue_mask = bm;
raw_fb_native_red_shift = 100;
raw_fb_native_green_shift = 100;
raw_fb_native_blue_shift = 100;
raw_fb_native_red_max = 1;
raw_fb_native_green_max = 1;
raw_fb_native_blue_max = 1;
m = 1;
for (i=0; i<32; i++) {
if (raw_fb_native_red_mask & m) {
if (raw_fb_native_red_shift == 100) {
raw_fb_native_red_shift = i;
}
raw_fb_native_red_max *= 2;
}
if (raw_fb_native_green_mask & m) {
if (raw_fb_native_green_shift == 100) {
raw_fb_native_green_shift = i;
}
raw_fb_native_green_max *= 2;
}
if (raw_fb_native_blue_mask & m) {
if (raw_fb_native_blue_shift == 100) {
raw_fb_native_blue_shift = i;
}
raw_fb_native_blue_max *= 2;
}
m = m << 1;
}
raw_fb_native_red_max -= 1;
raw_fb_native_green_max -= 1;
raw_fb_native_blue_max -= 1;
if (b < 8) {
/* e.g. VGA16 */
rfbLog("raw_fb_native_bpp: %d 0x%02lx 0x%02lx 0x%02lx %d/%d/%d %d/%d/%d\n", raw_fb_native_bpp,
raw_fb_native_red_mask, raw_fb_native_green_mask, raw_fb_native_blue_mask,
raw_fb_native_red_max, raw_fb_native_green_max, raw_fb_native_blue_max,
raw_fb_native_red_shift, raw_fb_native_green_shift, raw_fb_native_blue_shift);
raw_fb_expand_bytes = 1;
b = 8;
rm = 0x07;
gm = 0x38;
bm = 0xc0;
}
/* end of stuff for bpp < 8 */
dpy_x = wdpy_x = w;
dpy_y = wdpy_y = h;
off_x = 0;
off_y = 0;
if (rawfb_dev_video) {
if (b == 24) {
rfbLog("enabling -24to32 for 24bpp video\n");
xform24to32 = 1;
} else {
if (xform24to32) {
rfbLog("disabling -24to32 for 24bpp video\n");
}
xform24to32 = 0;
}
}
if (xform24to32) {
if (b != 24) {
rfbLog("warning: -24to32 mode and bpp=%d\n", b);
}
b = 32;
}
if (strstr(str, "snap:") == str) {
use_snapfb = 1;
str[0] = 'f'; str[1] = 'i'; str[2] = 'l'; str[3] = 'e';
}
if (strstr(str, "shm:") != str && strstr(str, "mmap:") != str &&
strstr(str, "map:") != str && strstr(str, "file:") != str) {
/* hmmm, not following directions, see if map: applies */
struct stat sbuf;
if (stat(str, &sbuf) == 0) {
char *newstr;
int len = strlen("map:") + strlen(str) + 1;
rfbLog("no type prefix: %s\n", raw_fb_str);
rfbLog(" but file exists, so assuming: map:%s\n",
raw_fb_str);
newstr = (char *) malloc(len);
strcpy(newstr, "map:");
strcat(newstr, str);
free(str);
str = newstr;
}
}
if (sscanf(str, "shm:%d", &shmid) == 1) {
/* shm:N */
#if LIBVNCSERVER_HAVE_XSHM || LIBVNCSERVER_HAVE_SHMAT
raw_fb_addr = (char *) shmat(shmid, 0, SHM_RDONLY);
if (! raw_fb_addr) {
rfbLogEnable(1);
rfbLog("failed to attach to shm: %d, %s\n", shmid, str);
rfbLogPerror("shmat");
clean_up_exit(1);
}
raw_fb_shm = 1;
rfbLog("rawfb: shm: %d W: %d H: %d B: %d addr: %p\n",
shmid, w, h, b, raw_fb_addr);
last_mode = RAWFB_SHM;
#else
rfbLogEnable(1);
rfbLog("x11vnc was compiled without shm support.\n");
rfbLogPerror("shmat");
clean_up_exit(1);
#endif
} else if (strstr(str, "map:") == str || strstr(str, "mmap:") == str
|| strstr(str, "file:") == str) {
/* map:/path/... or file:/path */
int fd, do_mmap = 1, size;
struct stat sbuf;
if (*str == 'f') {
do_mmap = 0;
}
q = strchr(str, ':');
q++;
macosx_console = 0;
if (strstr(q, "macosx:") == q) {
/* mmap:macosx:/dev/null@... */
q += strlen("macosx:");
do_macosx = 1;
do_mmap = 0;
macosx_console = 1;
}
last_file = strdup(q);
fd = raw_fb_fd;
if (fd < 0 && rawfb_dev_video) {
fd = open(q, O_RDWR);
}
if (fd < 0) {
fd = open(q, O_RDONLY);
}
if (fd < 0) {
rfbLogEnable(1);
rfbLog("failed to open file: %s, %s\n", q, str);
rfbLogPerror("open");
linux_dev_fb_msg(q);
clean_up_exit(1);
}
raw_fb_fd = fd;
if (raw_fb_native_bpp < 8) {
size = w*h*raw_fb_native_bpp/8 + raw_fb_offset;
} else if (xform24to32) {
size = w*h*24/8 + raw_fb_offset;
} else {
size = w*h*b/8 + raw_fb_offset;
}
if (fstat(fd, &sbuf) == 0) {
if (S_ISREG(sbuf.st_mode)) {
if (0) size = sbuf.st_size;
} else {
rfbLog("raw fb is non-regular file: %s\n", q);
}
}
if (do_macosx) {
raw_fb_addr = macosx_get_fb_addr();
raw_fb_mmap = size;
rfbLog("rawfb: macosx fb: %s\n", q);
rfbLog(" w: %d h: %d b: %d addr: %p sz: %d\n", w, h,
b, raw_fb_addr, size);
last_mode = 0;
} else if (do_reflect) {
raw_fb_mmap = size;
rfbLog("rawfb: vnc fb: %s\n", q);
rfbLog(" w: %d h: %d b: %d addr: %p sz: %d\n", w, h,
b, raw_fb_addr, size);
last_mode = 0;
} else if (do_mmap) {
#if LIBVNCSERVER_HAVE_MMAP
raw_fb_addr = mmap(0, size, PROT_READ, MAP_SHARED,
fd, 0);
if (raw_fb_addr == MAP_FAILED || raw_fb_addr == NULL) {
rfbLogEnable(1);
rfbLog("failed to mmap file: %s, %s\n", q, str);
rfbLog(" raw_fb_addr: %p\n", raw_fb_addr);
rfbLogPerror("mmap");
raw_fb_addr = NULL;
rfbLog("mmap(2) failed, trying slower lseek(2)\n");
raw_fb_seek = size;
last_mode = RAWFB_FILE;
} else {
raw_fb_mmap = size;
rfbLog("rawfb: mmap file: %s\n", q);
rfbLog(" w: %d h: %d b: %d addr: %p sz: %d\n", w, h,
b, raw_fb_addr, size);
last_mode = RAWFB_MMAP;
}
#else
rfbLog("mmap(2) not supported on system, using"
" slower lseek(2)\n");
raw_fb_seek = size;
last_mode = RAWFB_FILE;
#endif
} else {
raw_fb_seek = size;
last_mode = RAWFB_FILE;
rfbLog("rawfb: seek file: %s\n", q);
rfbLog(" W: %d H: %d B: %d sz: %d\n", w, h, b, size);
}
} else {
rfbLogEnable(1);
rfbLog("invalid rawfb str: %s\n", str);
clean_up_exit(1);
}
if (unlink_me) {
unlink(unlink_me);
}
if (! raw_fb_image) {
raw_fb_image = &ximage_struct;
}
initialize_clipshift();
if (raw_fb_bytes_per_line == 0) {
raw_fb_bytes_per_line = dpy_x*b/8;
/*
* Put cases here were we can determine that
* raw_bytes_per_line != dpy_x*b/8
*/
#ifdef MACOSX
if (do_macosx) {
raw_fb_bytes_per_line = macosxCG_CGDisplayBytesPerRow();
}
#endif
}
raw_fb_image->bytes_per_line = dpy_x * b/8;
raw_fb = (char *) malloc(dpy_y * dpy_x * b/8);
raw_fb_image->data = raw_fb;
raw_fb_image->format = ZPixmap;
raw_fb_image->width = dpy_x;
raw_fb_image->height = dpy_y;
raw_fb_image->bits_per_pixel = b;
raw_fb_image->bitmap_unit = -1;
if (use_snapfb && (raw_fb_seek || raw_fb_mmap)) {
int b_use = b;
if (snap_fb) {
free(snap_fb);
}
if (b_use == 32 && xform24to32) {
/*
* The actual framebuffer (e.g. mapped addr) and
* snap fb must be the same bpp. E.g. both 24bpp.
* Reading FROM snap to utility image will be
* transformed 24->32 in copy_raw_fb_24_to_32.
*
* addr -> snap -> (scanline, fullscreen, ...)
*/
b_use = 24;
raw_fb_bytes_per_line = dpy_x * b_use/8;
}
snap_fb = (char *) malloc(dpy_y * dpy_x * b_use/8);
snap = &ximage_struct_snap;
snap->data = snap_fb;
snap->format = ZPixmap;
snap->width = dpy_x;
snap->height = dpy_y;
snap->bits_per_pixel = b_use;
snap->bytes_per_line = dpy_x * b_use/8;
snap->bitmap_unit = -1;
}
raw_fb_image->red_mask = rm;
raw_fb_image->green_mask = gm;
raw_fb_image->blue_mask = bm;
raw_fb_image->depth = 0;
m = 1;
for (i=0; i<32; i++) {
if (rm & m) {
raw_fb_image->depth++;
}
if (gm & m) {
raw_fb_image->depth++;
}
if (bm & m) {
raw_fb_image->depth++;
}
m = m << 1;
}
if (raw_fb_native_bpp < 8) {
raw_fb_image->depth = raw_fb_expand_bytes * 8;
}
if (! raw_fb_image->depth) {
raw_fb_image->depth = (b == 32) ? 24 : b;
}
depth = raw_fb_image->depth;
if (raw_fb_image->depth == 15) {
/* unresolved bug with RGB555... */
depth++;
}
if (clipshift || raw_fb_native_bpp < 8) {
memset(raw_fb, 0xff, dpy_y * raw_fb_image->bytes_per_line);
} else if (raw_fb_addr && ! xform24to32) {
memcpy(raw_fb, raw_fb_addr + raw_fb_offset, dpy_y * raw_fb_image->bytes_per_line);
} else {
memset(raw_fb, 0xff, dpy_y * raw_fb_image->bytes_per_line);
}
if (verbose) {
rfbLog("\n");
rfbLog("rawfb: raw_fb %p\n", raw_fb);
rfbLog(" format %d\n", raw_fb_image->format);
rfbLog(" width %d\n", raw_fb_image->width);
rfbLog(" height %d\n", raw_fb_image->height);
rfbLog(" bpp %d\n", raw_fb_image->bits_per_pixel);
rfbLog(" depth %d\n", raw_fb_image->depth);
rfbLog(" bpl %d\n", raw_fb_image->bytes_per_line);
if (use_snapfb && snap_fb) {
rfbLog(" snap_fb %p\n", snap_fb);
}
}
free(str);
return raw_fb_image;
}
static void initialize_clipshift(void) {
clipshift = 0;
cdpy_x = cdpy_y = coff_x = coff_y = 0;
if (clip_str) {
int w, h, x, y, bad = 0;
if (parse_geom(clip_str, &w, &h, &x, &y, wdpy_x, wdpy_y)) {
if (x < 0) {
x = 0;
}
if (y < 0) {
y = 0;
}
if (x + w > wdpy_x) {
w = wdpy_x - x;
}
if (y + h > wdpy_y) {
h = wdpy_y - y;
}
if (w <= 0 || h <= 0) {
bad = 1;
}
} else {
bad = 1;
}
if (bad) {
rfbLog("*** ignoring invalid -clip WxH+X+Y: %s\n",
clip_str);
} else {
/* OK, change geom behind everyone's back... */
cdpy_x = w;
cdpy_y = h;
coff_x = x;
coff_y = y;
clipshift = 1;
dpy_x = cdpy_x;
dpy_y = cdpy_y;
}
}
}
static int wait_until_mapped(Window win) {
#if NO_X11
if (!win) {}
return 0;
#else
int ms = 50, waittime = 30;
time_t start = time(NULL);
XWindowAttributes attr;
while (1) {
if (! valid_window(win, NULL, 0)) {
if (time(NULL) > start + waittime) {
break;
}
usleep(ms * 1000);
continue;
}
if (dpy && ! XGetWindowAttributes(dpy, win, &attr)) {
return 0;
}
if (attr.map_state == IsViewable) {
return 1;
}
usleep(ms * 1000);
}
return 0;
#endif /* NO_X11 */
}
/*
* initialize a fb for the X display
*/
XImage *initialize_xdisplay_fb(void) {
#if NO_X11
if (raw_fb_str) {
return initialize_raw_fb(0);
}
return NULL;
#else
XImage *fb;
char *vis_str = visual_str;
int try = 0, subwin_tries = 3;
XErrorHandler old_handler = NULL;
int subwin_bs;
if (raw_fb_str) {
return initialize_raw_fb(0);
}
X_LOCK;
if (subwin) {
if (subwin_wait_mapped) {
wait_until_mapped(subwin);
}
if (!valid_window((Window) subwin, NULL, 0)) {
rfbLogEnable(1);
rfbLog("invalid sub-window: 0x%lx\n", subwin);
X_UNLOCK;
clean_up_exit(1);
}
}
if (overlay) {
/*
* ideally we'd like to not have to cook up the
* visual variables but rather let it all come out
* of XReadScreen(), however there is no way to get
* a default visual out of it, so we pretend -visual
* TrueColor:NN was supplied with NN usually 24.
*/
char str[32];
Window twin = subwin ? subwin : rootwin;
XImage *xi;
xi = xreadscreen(dpy, twin, 0, 0, 8, 8, False);
sprintf(str, "TrueColor:%d", xi->depth);
if (xi->depth != 24 && ! quiet) {
rfbLog("warning: overlay image has depth %d "
"instead of 24.\n", xi->depth);
}
XDestroyImage(xi);
if (visual_str != NULL && ! quiet) {
rfbLog("warning: replacing '-visual %s' by '%s' "
"for use with -overlay\n", visual_str, str);
}
vis_str = strdup(str);
}
if (xform24to32) {
if (DefaultDepth(dpy, scr) == 24) {
vis_str = strdup("TrueColor:32");
rfbLog("initialize_xdisplay_fb: vis_str set to: %s\n",
vis_str);
visual_id = (VisualID) 0;
visual_depth = 0;
set_visual_str_to_something = 1;
}
} else if (DefaultDepth(dpy, scr) < 8) {
/* check very low bpp case, e.g. mono or vga16 */
Screen *s = DefaultScreenOfDisplay(dpy);
XImage *xi = XGetImage_wr(dpy, DefaultRootWindow(dpy), 0, 0, 2, 2, AllPlanes,
ZPixmap);
if (xi && xi->bits_per_pixel < 8) {
int lowbpp = xi->bits_per_pixel;
if (!vis_str) {
char tmp[32];
sprintf(tmp, "0x%x:8", (int) s->root_visual->visualid);
vis_str = strdup(tmp);
rfbLog("initialize_xdisplay_fb: low bpp[%d], vis_str "
"set to: %s\n", lowbpp, vis_str);
}
if (using_shm) {
using_shm = 0;
rfbLog("initialize_xdisplay_fb: low bpp[%d], "
"disabling shm\n", lowbpp);
}
visual_id = (VisualID) 0;
visual_depth = 0;
set_visual_str_to_something = 1;
}
if (xi) {
XDestroyImage(xi);
}
}
if (vis_str != NULL) {
set_visual(vis_str);
if (vis_str != visual_str) {
free(vis_str);
}
}
if (0) fprintf(stderr, "vis_str %s\n", vis_str ? vis_str : "notset");
/* set up parameters for subwin or non-subwin cases: */
again:
if (! subwin) {
/* full screen */
window = rootwin;
dpy_x = wdpy_x = DisplayWidth(dpy, scr);
dpy_y = wdpy_y = DisplayHeight(dpy, scr);
off_x = 0;
off_y = 0;
/* this may be overridden via visual_id below */
default_visual = DefaultVisual(dpy, scr);
} else {
/* single window */
XWindowAttributes attr;
window = (Window) subwin;
if (! XGetWindowAttributes(dpy, window, &attr)) {
rfbLogEnable(1);
rfbLog("invalid window: 0x%lx\n", window);
X_UNLOCK;
clean_up_exit(1);
}
dpy_x = wdpy_x = attr.width;
dpy_y = wdpy_y = attr.height;
subwin_bs = attr.backing_store;
/* this may be overridden via visual_id below */
default_visual = attr.visual;
X_UNLOCK;
set_offset();
X_LOCK;
}
initialize_clipshift();
/* initialize depth to reasonable value, visual_id may override */
depth = DefaultDepth(dpy, scr);
if (0) fprintf(stderr, "DefaultDepth: %d visial_id: %d\n", depth, (int) visual_id);
if (visual_id) {
int n;
XVisualInfo vinfo_tmpl, *vinfo;
/*
* we are in here from -visual or -overlay options
* visual_id and visual_depth were set in set_visual().
*/
vinfo_tmpl.visualid = visual_id;
vinfo = XGetVisualInfo(dpy, VisualIDMask, &vinfo_tmpl, &n);
if (vinfo == NULL || n == 0) {
rfbLogEnable(1);
rfbLog("could not match visual_id: 0x%x\n",
(int) visual_id);
X_UNLOCK;
clean_up_exit(1);
}
default_visual = vinfo->visual;
depth = vinfo->depth;
if (visual_depth) {
/* force it from -visual MooColor:NN */
depth = visual_depth;
}
if (! quiet) {
fprintf(stderr, " initialize_xdisplay_fb()\n");
fprintf(stderr, " Visual*: %p\n",
(void *) vinfo->visual);
fprintf(stderr, " visualid: 0x%x\n",
(int) vinfo->visualid);
fprintf(stderr, " screen: %d\n", vinfo->screen);
fprintf(stderr, " depth: %d\n", vinfo->depth);
fprintf(stderr, " class: %d\n", vinfo->class);
fprintf(stderr, " red_mask: 0x%08lx %s\n",
vinfo->red_mask, bitprint(vinfo->red_mask, 32));
fprintf(stderr, " green_mask: 0x%08lx %s\n",
vinfo->green_mask, bitprint(vinfo->green_mask, 32));
fprintf(stderr, " blue_mask: 0x%08lx %s\n",
vinfo->blue_mask, bitprint(vinfo->blue_mask, 32));
fprintf(stderr, " cmap_size: %d\n",
vinfo->colormap_size);
fprintf(stderr, " bits b/rgb: %d\n",
vinfo->bits_per_rgb);
fprintf(stderr, "\n");
}
XFree_wr(vinfo);
}
if (! quiet) {
rfbLog("Default visual ID: 0x%x\n",
(int) XVisualIDFromVisual(default_visual));
}
if (subwin) {
int shift = 0, resize = 0;
int subwin_x, subwin_y;
int disp_x = DisplayWidth(dpy, scr);
int disp_y = DisplayHeight(dpy, scr);
Window twin;
/* subwins can be a dicey if they are changing size... */
trapped_xerror = 0;
old_handler = XSetErrorHandler(trap_xerror); /* reset in if(subwin) block below */
XTranslateCoordinates(dpy, window, rootwin, 0, 0, &subwin_x,
&subwin_y, &twin);
if (wdpy_x > disp_x) {
resize = 1;
dpy_x = wdpy_x = disp_x - 4;
}
if (wdpy_y > disp_y) {
resize = 1;
dpy_y = wdpy_y = disp_y - 4;
}
if (subwin_x + wdpy_x > disp_x) {
shift = 1;
subwin_x = disp_x - wdpy_x - 3;
}
if (subwin_y + wdpy_y > disp_y) {
shift = 1;
subwin_y = disp_y - wdpy_y - 3;
}
if (subwin_x < 0) {
shift = 1;
subwin_x = 1;
}
if (subwin_y < 0) {
shift = 1;
subwin_y = 1;
}
if (resize) {
XResizeWindow(dpy, window, wdpy_x, wdpy_y);
}
if (shift) {
XMoveWindow(dpy, window, subwin_x, subwin_y);
off_x = subwin_x;
off_y = subwin_y;
}
XMapRaised(dpy, window);
XRaiseWindow(dpy, window);
XSync(dpy, False);
}
try++;
if (nofb) {
/*
* For -nofb we do not allocate the framebuffer, so we
* can save a few MB of memory.
*/
fb = XCreateImage_wr(dpy, default_visual, depth, ZPixmap,
0, NULL, dpy_x, dpy_y, BitmapPad(dpy), 0);
} else if (visual_id) {
/*
* we need to call XCreateImage to supply the visual
*/
fb = XCreateImage_wr(dpy, default_visual, depth, ZPixmap,
0, NULL, dpy_x, dpy_y, BitmapPad(dpy), 0);
if (fb) {
fb->data = (char *) malloc(fb->bytes_per_line * fb->height);
}
} else {
fb = XGetImage_wr(dpy, window, 0, 0, dpy_x, dpy_y, AllPlanes,
ZPixmap);
if (! quiet) {
rfbLog("Read initial data from X display into"
" framebuffer.\n");
}
}
if (subwin) {
XSetErrorHandler(old_handler);
if (trapped_xerror || fb == NULL) {
rfbLog("trapped GetImage at SUBWIN creation.\n");
if (try < subwin_tries) {
usleep(250 * 1000);
if (!get_window_size(window, &wdpy_x, &wdpy_y)) {
rfbLogEnable(1);
rfbLog("could not get size of subwin "
"0x%lx\n", subwin);
X_UNLOCK;
clean_up_exit(1);
}
goto again;
}
}
trapped_xerror = 0;
} else if (fb == NULL) {
XEvent xev;
rfbLog("initialize_xdisplay_fb: *** fb creation failed: 0x%x try: %d\n", fb, try);
#if LIBVNCSERVER_HAVE_LIBXRANDR
if (xrandr_present && xrandr_base_event_type) {
int cnt = 0;
while (XCheckTypedEvent(dpy, xrandr_base_event_type + RRScreenChangeNotify, &xev)) {
XRRScreenChangeNotifyEvent *rev;
rev = (XRRScreenChangeNotifyEvent *) &xev;
rfbLog("initialize_xdisplay_fb: XRANDR event while redoing fb[%d]:\n", cnt++);
rfbLog(" serial: %d\n", (int) rev->serial);
rfbLog(" timestamp: %d\n", (int) rev->timestamp);
rfbLog(" cfg_timestamp: %d\n", (int) rev->config_timestamp);
rfbLog(" size_id: %d\n", (int) rev->size_index);
rfbLog(" sub_pixel: %d\n", (int) rev->subpixel_order);
rfbLog(" rotation: %d\n", (int) rev->rotation);
rfbLog(" width: %d\n", (int) rev->width);
rfbLog(" height: %d\n", (int) rev->height);
rfbLog(" mwidth: %d mm\n", (int) rev->mwidth);
rfbLog(" mheight: %d mm\n", (int) rev->mheight);
rfbLog("\n");
rfbLog("previous WxH: %dx%d\n", wdpy_x, wdpy_y);
xrandr_width = rev->width;
xrandr_height = rev->height;
xrandr_timestamp = rev->timestamp;
xrandr_cfg_time = rev->config_timestamp;
xrandr_rotation = (int) rev->rotation;
rfbLog("initialize_xdisplay_fb: updating XRANDR config...\n");
XRRUpdateConfiguration(&xev);
}
}
#endif
if (try < 5) {
XFlush_wr(dpy);
usleep(250 * 1000);
if (try < 3) {
XSync(dpy, False);
} else if (try >= 3) {
XSync(dpy, True);
}
goto again;
}
}
if (use_snapfb) {
initialize_snap_fb();
}
X_UNLOCK;
if (fb->bits_per_pixel == 24 && ! quiet) {
rfbLog("warning: 24 bpp may have poor performance.\n");
}
return fb;
#endif /* NO_X11 */
}
void parse_scale_string(char *str, double *factor_x, double *factor_y, int *scaling, int *blend,
int *nomult4, int *pad, int *interpolate, int *numer, int *denom, int w_in, int h_in) {
int m, n;
char *p, *tstr;
double f, f2;
*factor_x = 1.0;
*factor_y = 1.0;
*scaling = 0;
*blend = 1;
*nomult4 = 0;
*pad = 0;
*interpolate = 0;
*numer = 0, *denom = 0;
if (str == NULL || str[0] == '\0') {
return;
}
tstr = strdup(str);
if ( (p = strchr(tstr, ':')) != NULL) {
/* options */
if (strstr(p+1, "nb") != NULL) {
*blend = 0;
}
if (strstr(p+1, "fb") != NULL) {
*blend = 2;
}
if (strstr(p+1, "n4") != NULL) {
*nomult4 = 1;
}
if (strstr(p+1, "in") != NULL) {
*interpolate = 1;
}
if (strstr(p+1, "pad") != NULL) {
*pad = 1;
}
if (strstr(p+1, "nocr") != NULL) {
/* global */
scaling_copyrect = 0;
} else if (strstr(p+1, "cr") != NULL) {
/* global */
scaling_copyrect = 1;
}
*p = '\0';
}
if (strchr(tstr, '.') != NULL) {
double test, diff, eps = 1.0e-7;
if (sscanf(tstr, "%lfx%lf", &f, &f2) == 2) {
*factor_x = (double) f;
*factor_y = (double) f2;
} else if (sscanf(tstr, "%lf", &f) != 1) {
rfbLogEnable(1);
rfbLog("invalid -scale arg: %s\n", tstr);
clean_up_exit(1);
} else {
*factor_x = (double) f;
*factor_y = (double) f;
}
/* look for common fractions from small ints: */
if (*factor_x == *factor_y) {
for (n=2; n<=10; n++) {
for (m=1; m<n; m++) {
test = ((double) m)/ n;
diff = *factor_x - test;
if (-eps < diff && diff < eps) {
*numer = m;
*denom = n;
break;
}
}
if (*denom) {
break;
}
}
if (*factor_x < 0.01) {
rfbLogEnable(1);
rfbLog("-scale factor too small: %f\n", *factor_x);
clean_up_exit(1);
}
}
} else if (sscanf(tstr, "%dx%d", &m, &n) == 2 && w_in > 0 && h_in > 0) {
*factor_x = ((double) m) / ((double) w_in);
*factor_y = ((double) n) / ((double) h_in);
} else {
if (sscanf(tstr, "%d/%d", &m, &n) != 2) {
if (sscanf(tstr, "%d", &m) != 1) {
rfbLogEnable(1);
rfbLog("invalid -scale arg: %s\n", tstr);
clean_up_exit(1);
} else {
/* e.g. -scale 1 or -scale 2 */
n = 1;
}
}
if (n <= 0 || m <=0) {
rfbLogEnable(1);
rfbLog("invalid -scale arg: %s\n", tstr);
clean_up_exit(1);
}
*factor_x = ((double) m)/ n;
*factor_y = ((double) m)/ n;
if (*factor_x < 0.01) {
rfbLogEnable(1);
rfbLog("-scale factor too small: %f\n", *factor_x);
clean_up_exit(1);
}
*numer = m;
*denom = n;
}
if (*factor_x == 1.0 && *factor_y == 1.0) {
if (! quiet) {
rfbLog("scaling disabled for factor %f %f\n", *factor_x, *factor_y);
}
} else {
*scaling = 1;
}
free(tstr);
}
int parse_rotate_string(char *str, int *mode) {
int m = ROTATE_NONE;
if (str == NULL || !strcmp(str, "") || !strcmp(str, "0")) {
m = ROTATE_NONE;
} else if (!strcmp(str, "x")) {
m = ROTATE_X;
} else if (!strcmp(str, "y")) {
m = ROTATE_Y;
} else if (!strcmp(str, "xy") || !strcmp(str, "yx") ||
!strcmp(str,"+180") || !strcmp(str,"-180") || !strcmp(str,"180")) {
m = ROTATE_XY;
} else if (!strcmp(str, "+90") || !strcmp(str, "90")) {
m = ROTATE_90;
} else if (!strcmp(str, "+90x") || !strcmp(str, "90x")) {
m = ROTATE_90X;
} else if (!strcmp(str, "+90y") || !strcmp(str, "90y")) {
m = ROTATE_90Y;
} else if (!strcmp(str, "-90") || !strcmp(str, "270") ||
!strcmp(str, "+270")) {
m = ROTATE_270;
} else {
rfbLog("invalid -rotate mode: %s\n", str);
}
if (mode) {
*mode = m;
}
return m;
}
int scale_round(int len, double fac) {
double eps = 0.000001;
len = (int) (len * fac + eps);
if (len < 1) {
len = 1;
}
return len;
}
static void setup_scaling(int *width_in, int *height_in) {
int width = *width_in;
int height = *height_in;
parse_scale_string(scale_str, &scale_fac_x, &scale_fac_y, &scaling, &scaling_blend,
&scaling_nomult4, &scaling_pad, &scaling_interpolate,
&scale_numer, &scale_denom, *width_in, *height_in);
if (scaling) {
width = scale_round(width, scale_fac_x);
height = scale_round(height, scale_fac_y);
if (scale_denom && scaling_pad) {
/* it is not clear this padding is useful anymore */
rfbLog("width %% denom: %d %% %d = %d\n", width,
scale_denom, width % scale_denom);
rfbLog("height %% denom: %d %% %d = %d\n", height,
scale_denom, height % scale_denom);
if (width % scale_denom != 0) {
int w = width;
w += scale_denom - (w % scale_denom);
if (!scaling_nomult4 && w % 4 != 0) {
/* need to make mult of 4 as well */
int c = 0;
while (w % 4 != 0 && c++ <= 5) {
w += scale_denom;
}
}
width = w;
rfbLog("padded width to: %d (mult of %d%s\n",
width, scale_denom, !scaling_nomult4 ?
" and 4)" : ")");
}
if (height % scale_denom != 0) {
height += scale_denom - (height % scale_denom);
rfbLog("padded height to: %d (mult of %d)\n",
height, scale_denom);
}
}
if (!scaling_nomult4 && width % 4 != 0 && width > 2) {
/* reset width to be multiple of 4 */
int width0 = width;
if ((width+1) % 4 == 0) {
width = width+1;
} else if ((width-1) % 4 == 0) {
width = width-1;
} else if ((width+2) % 4 == 0) {
width = width+2;
}
rfbLog("reset scaled width %d -> %d to be a multiple of"
" 4 (to\n", width0, width);
rfbLog("make vncviewers happy). use -scale m/n:n4 to "
"disable.\n");
}
scaled_x = width;
scaled_y = height;
*width_in = width;
*height_in = height;
}
}
static void setup_rotating(void) {
char *rs = rotating_str;
rotating_cursors = 1;
if (rs && strstr(rs, "nc:") == rs) {
rs += strlen("nc:");
rotating_cursors = 0;
}
rotating = parse_rotate_string(rs, NULL);
if (! rotating) {
rotating_cursors = 0;
}
if (rotating == ROTATE_90 || rotating == ROTATE_90X ||
rotating == ROTATE_90Y || rotating == ROTATE_270) {
rotating_same = 0;
} else {
rotating_same = 1;
}
}
static rfbBool set_xlate_wrapper(rfbClientPtr cl) {
static int first = 1;
if (first) {
first = 0;
} else if (ncache) {
int save = ncache_xrootpmap;
rfbLog("set_xlate_wrapper: clearing -ncache for new pixel format.\n");
INPUT_LOCK;
ncache_xrootpmap = 0;
check_ncache(1, 0);
ncache_xrootpmap = save;
INPUT_UNLOCK;
}
return rfbSetTranslateFunction(cl);
}
/*
* initialize the rfb framebuffer/screen
*/
void initialize_screen(int *argc, char **argv, XImage *fb) {
int have_masks = 0;
int width = fb->width;
int height = fb->height;
int create_screen = screen ? 0 : 1;
int bits_per_color;
int fb_bpp, fb_Bpl, fb_depth;
int locked_sends = 0;
bpp = fb->bits_per_pixel;
fb_bpp = (int) fb->bits_per_pixel;
fb_Bpl = (int) fb->bytes_per_line;
fb_depth = (int) fb->depth;
rfbLog("initialize_screen: fb_depth/fb_bpp/fb_Bpl %d/%d/%d\n", fb_depth,
fb_bpp, fb_Bpl);
main_bytes_per_line = fb->bytes_per_line;
if (cmap8to24) {
if (raw_fb) {
if (!quiet) rfbLog("disabling -8to24 mode"
" in raw_fb mode.\n");
cmap8to24 = 0;
} else if (depth == 24) {
if (fb_bpp != 32) {
if (!quiet) rfbLog("disabling -8to24 mode:"
" bpp != 32 with depth == 24\n");
cmap8to24 = 0;
}
} else if (depth <= 16) {
/* need to cook up the screen fb to be depth 24 */
fb_bpp = 32;
fb_depth = 24;
} else {
if (!quiet) rfbLog("disabling -8to24 mode:"
" default depth not 16 or less\n");
cmap8to24 = 0;
}
}
setup_scaling(&width, &height);
if (scaling) {
rfbLog("scaling screen: %dx%d -> %dx%d\n", fb->width, fb->height, scaled_x, scaled_y);
rfbLog("scaling screen: scale_fac_x=%.5f scale_fac_y=%.5f\n", scale_fac_x, scale_fac_y);
rfb_bytes_per_line = (main_bytes_per_line / fb->width) * width;
} else {
rfb_bytes_per_line = main_bytes_per_line;
}
setup_rotating();
if (rotating) {
if (! rotating_same) {
int t, b = main_bytes_per_line / fb->width;
if (scaling) {
rot_bytes_per_line = b * height;
} else {
rot_bytes_per_line = b * fb->height;
}
t = width;
width = height; /* The big swap... */
height = t;
} else {
rot_bytes_per_line = rfb_bytes_per_line;
}
}
#ifndef NO_NCACHE
if (ncache > 0 && !nofb) {
# ifdef MACOSX
if (! raw_fb_str || macosx_console) {
# else
if (! raw_fb_str) {
# endif
char *new_fb;
int sz = fb->height * fb->bytes_per_line;
int ns = 1+ncache;
if (ncache_xrootpmap) {
ns++;
}
new_fb = (char *) calloc((size_t) (sz * ns), 1);
if (fb->data) {
memcpy(new_fb, fb->data, sz);
free(fb->data);
}
fb->data = new_fb;
fb->height *= (ns);
height *= (ns);
ncache0 = ncache;
}
}
#endif
if (cmap8to24) {
if (depth <= 8) {
rfb_bytes_per_line *= 4;
rot_bytes_per_line *= 4;
} else if (depth <= 16) {
rfb_bytes_per_line *= 2;
rot_bytes_per_line *= 2;
}
}
/*
* These are just hints wrt pixel format just to let
* rfbGetScreen/rfbNewFramebuffer proceed with reasonable
* defaults. We manually set them in painful detail below.
*/
bits_per_color = guess_bits_per_color(fb_bpp);
if (lock_client_sends(-1) == 0) {
lock_client_sends(1);
locked_sends = 1;
}
/* n.b. samplesPerPixel (set = 1 here) seems to be unused. */
if (create_screen) {
if (use_stunnel) {
setup_stunnel(0, argc, argv);
}
if (use_openssl) {
if (use_stunnel && enc_str && !strcmp(enc_str, "none")) {
/* emulating HTTPS oneport */
;
} else {
openssl_init(0);
}
}
screen = rfbGetScreen(argc, argv, width, height,
bits_per_color, 1, fb_bpp/8);
if (screen && http_dir) {
http_connections(1);
}
if (unix_sock) {
unix_sock_fd = listen_unix(unix_sock);
}
} else {
/* set set frameBuffer member below. */
rfbLog("rfbNewFramebuffer(0x%x, 0x%x, %d, %d, %d, %d, %d)\n",
screen, NULL, width, height, bits_per_color, 1, fb_bpp/8);
/* these are probably overwritten, but just to be safe: */
screen->bitsPerPixel = fb_bpp;
screen->depth = fb_depth;
rfb_new_framebuffer(screen, NULL, width, height,
bits_per_color, 1, (int) fb_bpp/8);
}
if (! screen) {
int i;
rfbLogEnable(1);
rfbLog("\n");
rfbLog("failed to create rfb screen.\n");
for (i=0; i< *argc; i++) {
rfbLog("\t[%d] %s\n", i, argv[i]);
}
clean_up_exit(1);
}
if (create_screen && *argc != 1) {
int i;
rfbLogEnable(1);
rfbLog("*** unrecognized option(s) ***\n");
for (i=1; i< *argc; i++) {
rfbLog("\t[%d] %s\n", i, argv[i]);
}
rfbLog("For a list of options run: x11vnc -opts\n");
rfbLog("or for the full help: x11vnc -help\n");
rfbLog("\n");
rfbLog("Here is a list of removed or obsolete"
" options:\n");
rfbLog("\n");
rfbLog("removed: -hints, -nohints\n");
rfbLog("removed: -cursorposall\n");
rfbLog("removed: -nofilexfer, now the default.\n");
rfbLog("\n");
rfbLog("renamed: -old_copytile, use -onetile\n");
rfbLog("renamed: -mouse, use -cursor\n");
rfbLog("renamed: -mouseX, use -cursor X\n");
rfbLog("renamed: -X, use -cursor X\n");
rfbLog("renamed: -nomouse, use -nocursor\n");
rfbLog("renamed: -old_pointer, use -pointer_mode 1\n");
clean_up_exit(1);
}
/* set up format from scratch: */
if (rotating && ! rotating_same) {
screen->paddedWidthInBytes = rot_bytes_per_line;
} else {
screen->paddedWidthInBytes = rfb_bytes_per_line;
}
screen->serverFormat.bitsPerPixel = fb_bpp;
screen->serverFormat.depth = fb_depth;
screen->serverFormat.trueColour = TRUE;
screen->serverFormat.redShift = 0;
screen->serverFormat.greenShift = 0;
screen->serverFormat.blueShift = 0;
screen->serverFormat.redMax = 0;
screen->serverFormat.greenMax = 0;
screen->serverFormat.blueMax = 0;
/* these main_* formats are used generally. */
main_red_shift = 0;
main_green_shift = 0;
main_blue_shift = 0;
main_red_max = 0;
main_green_max = 0;
main_blue_max = 0;
main_red_mask = fb->red_mask;
main_green_mask = fb->green_mask;
main_blue_mask = fb->blue_mask;
have_masks = ((fb->red_mask|fb->green_mask|fb->blue_mask) != 0);
if (force_indexed_color) {
have_masks = 0;
}
if (cmap8to24 && depth <= 16 && dpy) {
XVisualInfo vinfo;
vinfo.red_mask = 0;
vinfo.green_mask = 0;
vinfo.blue_mask = 0;
/* more cooking up... */
have_masks = 2;
/* need to fetch TrueColor visual */
X_LOCK;
if (dpy
#if !NO_X11
&& XMatchVisualInfo(dpy, scr, 24, TrueColor, &vinfo)
#endif
) {
main_red_mask = vinfo.red_mask;
main_green_mask = vinfo.green_mask;
main_blue_mask = vinfo.blue_mask;
} else if (fb->byte_order == LSBFirst) {
main_red_mask = 0x00ff0000;
main_green_mask = 0x0000ff00;
main_blue_mask = 0x000000ff;
} else {
main_red_mask = 0x000000ff;
main_green_mask = 0x0000ff00;
main_blue_mask = 0x00ff0000;
}
X_UNLOCK;
}
if (raw_fb_str && raw_fb_pixfmt) {
char *fmt = strdup(raw_fb_pixfmt);
uppercase(fmt);
if (strstr(fmt, "GREY")) {
set_greyscale_colormap();
} else if (strstr(fmt, "HI240")) {
set_hi240_colormap();
} else {
unset_colormap();
}
free(fmt);
}
if (! have_masks && screen->serverFormat.bitsPerPixel <= 16
&& dpy && CellsOfScreen(ScreenOfDisplay(dpy, scr))) {
/* indexed color. we let 1 <= bpp <= 16, but it is normally 8 */
if (!quiet) {
rfbLog("\n");
rfbLog("X display %s is %dbpp indexed color, depth=%d\n",
DisplayString(dpy), screen->serverFormat.bitsPerPixel,
screen->serverFormat.depth);
if (! flash_cmap && ! overlay) {
rfbLog("\n");
rfbLog("In 8bpp PseudoColor mode if you "
"experience color\n");
rfbLog("problems you may want to enable "
"following the\n");
rfbLog("changing colormap by using the "
"-flashcmap option.\n");
rfbLog("\n");
}
}
if (screen->serverFormat.depth < 8) {
rfbLog("resetting serverFormat.depth %d -> 8.\n",
screen->serverFormat.depth);
rfbLog("resetting serverFormat.bpp %d -> 8.\n",
screen->serverFormat.bitsPerPixel);
screen->serverFormat.depth = 8;
screen->serverFormat.bitsPerPixel = 8;
}
if (screen->serverFormat.depth > 8) {
rfbLog("WARNING: indexed color depth > 8, Tight encoding will crash.\n");
}
screen->serverFormat.trueColour = FALSE;
indexed_color = 1;
set_colormap(1);
debug_colormap(fb);
} else {
/*
* general case, we call it truecolor, but could be direct
* color, static color, etc....
*/
if (! quiet) {
if (raw_fb) {
rfbLog("\n");
rfbLog("Raw fb at addr %p is %dbpp depth=%d "
"true color\n", raw_fb_addr,
fb_bpp, fb_depth);
} else if (! dpy) {
;
} else if (have_masks == 2) {
rfbLog("\n");
rfbLog("X display %s is %dbpp depth=%d indexed "
"color (-8to24 mode)\n", DisplayString(dpy),
fb->bits_per_pixel, fb->depth);
} else {
rfbLog("\n");
rfbLog("X display %s is %dbpp depth=%d true "
"color\n", DisplayString(dpy),
fb_bpp, fb_depth);
}
}
indexed_color = 0;
if (!have_masks) {
int M, B = bits_per_color;
if (3*B > fb_bpp) B--;
if (B < 1) B = 1;
M = (1 << B) - 1;
rfbLog("WARNING: all TrueColor RGB masks are zero!\n");
rfbLog("WARNING: cooking something up for VNC fb... B=%d M=%d\n", B, M);
main_red_mask = M;
main_green_mask = M;
main_blue_mask = M;
main_red_mask = main_red_mask << (2*B);
main_green_mask = main_green_mask << (1*B);
main_blue_mask = main_blue_mask << (0*B);
fprintf(stderr, " red_mask: 0x%08lx %s\n", main_red_mask,
bitprint(main_red_mask, 32));
fprintf(stderr, " green_mask: 0x%08lx %s\n", main_green_mask,
bitprint(main_green_mask, 32));
fprintf(stderr, " blue_mask: 0x%08lx %s\n", main_blue_mask,
bitprint(main_blue_mask, 32));
}
/* convert masks to bit shifts and max # colors */
if (main_red_mask) {
while (! (main_red_mask
& (1 << screen->serverFormat.redShift))) {
screen->serverFormat.redShift++;
}
}
if (main_green_mask) {
while (! (main_green_mask
& (1 << screen->serverFormat.greenShift))) {
screen->serverFormat.greenShift++;
}
}
if (main_blue_mask) {
while (! (main_blue_mask
& (1 << screen->serverFormat.blueShift))) {
screen->serverFormat.blueShift++;
}
}
screen->serverFormat.redMax
= main_red_mask >> screen->serverFormat.redShift;
screen->serverFormat.greenMax
= main_green_mask >> screen->serverFormat.greenShift;
screen->serverFormat.blueMax
= main_blue_mask >> screen->serverFormat.blueShift;
main_red_max = screen->serverFormat.redMax;
main_green_max = screen->serverFormat.greenMax;
main_blue_max = screen->serverFormat.blueMax;
main_red_shift = screen->serverFormat.redShift;
main_green_shift = screen->serverFormat.greenShift;
main_blue_shift = screen->serverFormat.blueShift;
}
#if !SMALL_FOOTPRINT
if (verbose) {
fprintf(stderr, "\n");
fprintf(stderr, "FrameBuffer Info:\n");
fprintf(stderr, " width: %d\n", fb->width);
fprintf(stderr, " height: %d\n", fb->height);
fprintf(stderr, " scaled_width: %d\n", width);
fprintf(stderr, " scaled_height: %d\n", height);
fprintf(stderr, " indexed_color: %d\n", indexed_color);
fprintf(stderr, " bits_per_pixel: %d\n", fb->bits_per_pixel);
fprintf(stderr, " depth: %d\n", fb->depth);
fprintf(stderr, " red_mask: 0x%08lx %s\n", fb->red_mask,
bitprint(fb->red_mask, 32));
fprintf(stderr, " green_mask: 0x%08lx %s\n", fb->green_mask,
bitprint(fb->green_mask, 32));
fprintf(stderr, " blue_mask: 0x%08lx %s\n", fb->blue_mask,
bitprint(fb->blue_mask, 32));
if (cmap8to24) {
fprintf(stderr, " 8to24 info:\n");
fprintf(stderr, " fb_bpp: %d\n", fb_bpp);
fprintf(stderr, " fb_depth: %d\n", fb_depth);
fprintf(stderr, " red_mask: 0x%08lx %s\n", main_red_mask,
bitprint(main_red_mask, 32));
fprintf(stderr, " green_mask: 0x%08lx %s\n", main_green_mask,
bitprint(main_green_mask, 32));
fprintf(stderr, " blue_mask: 0x%08lx %s\n", main_blue_mask,
bitprint(main_blue_mask, 32));
}
fprintf(stderr, " red: max: %3d shift: %2d\n",
main_red_max, main_red_shift);
fprintf(stderr, " green: max: %3d shift: %2d\n",
main_green_max, main_green_shift);
fprintf(stderr, " blue: max: %3d shift: %2d\n",
main_blue_max, main_blue_shift);
fprintf(stderr, " mainfb_bytes_per_line: %d\n",
main_bytes_per_line);
fprintf(stderr, " rfb_fb_bytes_per_line: %d\n",
rfb_bytes_per_line);
fprintf(stderr, " rot_fb_bytes_per_line: %d\n",
rot_bytes_per_line);
fprintf(stderr, " raw_fb_bytes_per_line: %d\n",
raw_fb_bytes_per_line);
switch(fb->format) {
case XYBitmap:
fprintf(stderr, " format: XYBitmap\n"); break;
case XYPixmap:
fprintf(stderr, " format: XYPixmap\n"); break;
case ZPixmap:
fprintf(stderr, " format: ZPixmap\n"); break;
default:
fprintf(stderr, " format: %d\n", fb->format); break;
}
switch(fb->byte_order) {
case LSBFirst:
fprintf(stderr, " byte_order: LSBFirst\n"); break;
case MSBFirst:
fprintf(stderr, " byte_order: MSBFirst\n"); break;
default:
fprintf(stderr, " byte_order: %d\n", fb->byte_order);
break;
}
fprintf(stderr, " bitmap_pad: %d\n", fb->bitmap_pad);
fprintf(stderr, " bitmap_unit: %d\n", fb->bitmap_unit);
switch(fb->bitmap_bit_order) {
case LSBFirst:
fprintf(stderr, " bitmap_bit_order: LSBFirst\n"); break;
case MSBFirst:
fprintf(stderr, " bitmap_bit_order: MSBFirst\n"); break;
default:
fprintf(stderr, " bitmap_bit_order: %d\n",
fb->bitmap_bit_order); break;
}
}
if (overlay && ! quiet) {
rfbLog("\n");
rfbLog("Overlay mode enabled: If you experience color\n");
rfbLog("problems when popup menus are on the screen, try\n");
rfbLog("disabling SaveUnders in your X server, one way is\n");
rfbLog("to start the X server with the '-su' option, e.g.:\n");
rfbLog("Xsun -su ... see Xserver(1), xinit(1) for more info.\n");
rfbLog("\n");
}
#endif
/* nofb is for pointer/keyboard only handling. */
if (nofb) {
main_fb = NULL;
rfb_fb = main_fb;
cmap8to24_fb = NULL;
rot_fb = NULL;
screen->displayHook = nofb_hook;
} else {
main_fb = fb->data;
rfb_fb = NULL;
cmap8to24_fb = NULL;
rot_fb = NULL;
if (cmap8to24) {
int n = main_bytes_per_line * fb->height;
if (depth <= 8) {
n *= 4;
} else if (depth <= 16) {
n *= 2;
}
cmap8to24_fb = (char *) malloc(n);
memset(cmap8to24_fb, 0, n);
}
if (rotating) {
int n = rot_bytes_per_line * height;
rot_fb = (char *) malloc(n);
memset(rot_fb, 0, n);
}
if (scaling) {
int n = rfb_bytes_per_line * height;
if (rotating && ! rotating_same) {
n = rot_bytes_per_line * height;
}
rfb_fb = (char *) malloc(n);
memset(rfb_fb, 0, n);
} else if (cmap8to24) {
rfb_fb = cmap8to24_fb;
} else {
rfb_fb = main_fb;
}
}
if (rot_fb) {
screen->frameBuffer = rot_fb;
} else {
screen->frameBuffer = rfb_fb;
}
if (verbose) {
fprintf(stderr, " rfb_fb: %p\n", rfb_fb);
fprintf(stderr, " main_fb: %p\n", main_fb);
fprintf(stderr, " 8to24_fb: %p\n", cmap8to24_fb);
fprintf(stderr, " rot_fb: %p\n", rot_fb);
fprintf(stderr, " snap_fb: %p\n", snap_fb);
fprintf(stderr, " raw_fb: %p\n", raw_fb);
fprintf(stderr, " fake_fb: %p\n", fake_fb);
fprintf(stderr, "\n");
}
/* may need, bpp, main_red_max, etc. */
parse_wireframe();
parse_scroll_copyrect();
setup_cursors_and_push();
if (scaling || rotating || cmap8to24) {
mark_rect_as_modified(0, 0, dpy_x, dpy_y, 0);
}
if (! create_screen) {
rfbClientIteratorPtr iter;
rfbClientPtr cl;
/*
* since bits_per_color above may have been approximate,
* try to reset the individual translation tables...
* we do not seem to need this with rfbGetScreen()...
*/
if (!quiet) rfbLog("calling setTranslateFunction()...\n");
iter = rfbGetClientIterator(screen);
while ((cl = rfbClientIteratorNext(iter)) != NULL) {
screen->setTranslateFunction(cl);
}
rfbReleaseClientIterator(iter);
if (!quiet) rfbLog(" done.\n");
/* done for framebuffer change case */
if (locked_sends) {
lock_client_sends(0);
}
do_copy_screen = 1;
return;
}
/*
* the rest is screen server initialization, etc, only needed
* at screen creation time.
*/
/* event callbacks: */
screen->newClientHook = new_client;
screen->kbdAddEvent = keyboard;
screen->ptrAddEvent = pointer_event;
screen->setXCutText = xcut_receive;
screen->setTranslateFunction = set_xlate_wrapper;
screen->kbdReleaseAllKeys = kbd_release_all_keys;
screen->setSingleWindow = set_single_window;
screen->setServerInput = set_server_input;
screen->setTextChat = set_text_chat;
screen->getFileTransferPermission = get_file_transfer_permitted;
/* called from inetd, we need to treat stdio as our socket */
if (inetd && use_openssl) {
/* accept_openssl() called later */
screen->port = 0;
} else if (inetd) {
int fd = dup(0);
if (fd < 0) {
rfbLogEnable(1);
rfbErr("dup(0) = %d failed.\n", fd);
rfbLogPerror("dup");
clean_up_exit(1);
}
fclose(stdin);
fclose(stdout);
/* we keep stderr for logging */
screen->inetdSock = fd;
screen->port = 0;
} else if (! got_rfbport && auto_port > 0) {
int lport = find_free_port(auto_port, auto_port+200);
screen->autoPort = FALSE;
screen->port = lport;
} else if (! got_rfbport) {
screen->autoPort = TRUE;
} else if (got_rfbport && got_rfbport_val == 0) {
screen->autoPort = FALSE;
screen->port = 0;
}
if (! got_nevershared && ! got_alwaysshared) {
if (shared) {
screen->alwaysShared = TRUE;
} else {
screen->neverShared = TRUE;
}
screen->dontDisconnect = TRUE;
}
if (! got_deferupdate) {
screen->deferUpdateTime = defer_update;
} else {
defer_update = screen->deferUpdateTime;
}
if (noipv4 || getenv("IPV4_FAILS")) {
rfbBool ap = screen->autoPort;
int port = screen->port;
if (getenv("IPV4_FAILS")) {
rfbLog("TESTING: IPV4_FAILS for rfbInitServer()\n");
}
screen->autoPort = FALSE;
screen->port = 0;
rfbInitServer(screen);
screen->autoPort = ap;
screen->port = port;
} else {
rfbInitServer(screen);
}
if (use_openssl) {
openssl_port(0);
if (https_port_num >= 0) {
https_port(0);
}
} else {
if (ipv6_listen) {
int fd = -1;
if (screen->port <= 0) {
if (got_rfbport) {
screen->port = got_rfbport_val;
} else {
int ap = 5900;
if (auto_port > 0) {
ap = auto_port;
}
screen->port = find_free_port6(ap, ap+200);
}
}
fd = listen6(screen->port);
if (fd < 0) {
ipv6_listen = 0;
rfbLog("Not listening on IPv6 interface.\n");
} else {
rfbLog("Listening %s on IPv6 port %d (socket %d)\n",
screen->listenSock < 0 ? "only" : "also",
screen->port, fd);
ipv6_listen_fd = fd;
}
}
}
install_passwds();
if (locked_sends) {
lock_client_sends(0);
}
return;
}
#define DO_AVAHI \
if (avahi) { \
avahi_initialise(); \
avahi_advertise(vnc_desktop_name, host, lport); \
usleep(1000*1000); \
}
void announce(int lport, int ssl, char *iface) {
char *host = this_host();
char *tvdt;
if (remote_direct) {
return;
}
if (! ssl) {
tvdt = "The VNC desktop is: ";
} else {
if (enc_str && !strcmp(enc_str, "none")) {
tvdt = "The VNC desktop is: ";
} else if (enc_str) {
tvdt = "The ENC VNC desktop is: ";
} else {
tvdt = "The SSL VNC desktop is: ";
}
}
if (iface != NULL && *iface != '\0' && strcmp(iface, "any")) {
host = iface;
}
if (host != NULL) {
/* note that vncviewer special cases 5900-5999 */
int sz = 256;
if (inetd) {
; /* should not occur (port) */
} else if (quiet) {
if (lport >= 5900) {
snprintf(vnc_desktop_name, sz, "%s:%d",
host, lport - 5900);
DO_AVAHI
fprintf(stderr, "\n%s %s\n", tvdt,
vnc_desktop_name);
} else {
snprintf(vnc_desktop_name, sz, "%s:%d",
host, lport);
DO_AVAHI
fprintf(stderr, "\n%s %s\n", tvdt,
vnc_desktop_name);
}
} else if (lport >= 5900) {
snprintf(vnc_desktop_name, sz, "%s:%d",
host, lport - 5900);
DO_AVAHI
fprintf(stderr, "\n%s %s\n", tvdt, vnc_desktop_name);
if (lport >= 6000) {
rfbLog("possible aliases: %s:%d, "
"%s::%d\n", host, lport,
host, lport);
}
} else {
snprintf(vnc_desktop_name, sz, "%s:%d",
host, lport);
DO_AVAHI
fprintf(stderr, "\n%s %s\n", tvdt, vnc_desktop_name);
rfbLog("possible alias: %s::%d\n",
host, lport);
}
}
}
static void announce_http(int lport, int ssl, char *iface, char *extra) {
char *host = this_host();
char *jvu;
int http = 0;
if (enc_str && !strcmp(enc_str, "none") && !use_stunnel) {
jvu = "Java viewer URL: http";
http = 1;
} else if (ssl == 1) {
jvu = "Java SSL viewer URL: https";
} else if (ssl == 2) {
jvu = "Java SSL viewer URL: http";
http = 1;
} else {
jvu = "Java viewer URL: http";
http = 1;
}
if (iface != NULL && *iface != '\0' && strcmp(iface, "any")) {
host = iface;
}
if (http && getenv("X11VNC_HTTP_LISTEN_LOCALHOST")) {
host = "localhost";
}
if (host != NULL) {
if (! inetd) {
fprintf(stderr, "%s://%s:%d/%s\n", jvu, host, lport, extra);
}
}
}
void do_announce_http(void) {
if (!screen) {
return;
}
if (remote_direct) {
return;
}
/* XXX ipv6? */
if ((screen->httpListenSock > -1 || ipv6_http_fd > -1) && screen->httpPort) {
int enc_none = (enc_str && !strcmp(enc_str, "none"));
char *SPORT = " (single port)";
if (use_openssl && ! enc_none) {
announce_http(screen->port, 1, listen_str, SPORT);
if (https_port_num >= 0) {
announce_http(https_port_num, 1,
listen_str, "");
}
announce_http(screen->httpPort, 2, listen_str, "");
} else if (use_stunnel) {
char pmsg[100];
pmsg[0] = '\0';
if (stunnel_port) {
sprintf(pmsg, "?PORT=%d", stunnel_port);
}
announce_http(screen->httpPort, 2, listen_str, pmsg);
if (stunnel_http_port > 0) {
announce_http(stunnel_http_port, 1, NULL, pmsg);
}
if (enc_none) {
strcat(pmsg, SPORT);
announce_http(stunnel_port, 1, NULL, pmsg);
}
} else {
announce_http(screen->httpPort, 0, listen_str, "");
if (enc_none) {
announce_http(screen->port, 1, NULL, SPORT);
}
}
}
}
void do_mention_java_urls(void) {
if (! quiet && screen) {
/* XXX ipv6? */
if (screen->httpListenSock > -1 && screen->httpPort) {
rfbLog("\n");
rfbLog("The URLs printed out below ('Java ... viewer URL') can\n");
rfbLog("be used for Java enabled Web browser connections.\n");
if (!stunnel_port && enc_str && !strcmp(enc_str, "none")) {
;
} else if (use_openssl || stunnel_port) {
rfbLog("Here are some additional possibilities:\n");
rfbLog("\n");
rfbLog("https://host:port/proxy.vnc (MUST be used if Web Proxy used)\n");
rfbLog("\n");
rfbLog("https://host:port/ultra.vnc (Use UltraVNC Java Viewer)\n");
rfbLog("https://host:port/ultraproxy.vnc (Web Proxy with UltraVNC)\n");
rfbLog("https://host:port/ultrasigned.vnc (Signed UltraVNC Filexfer)\n");
rfbLog("\n");
rfbLog("Where you replace \"host:port\" with that printed below, or\n");
rfbLog("whatever is needed to reach the host e.g. Internet IP number\n");
rfbLog("\n");
rfbLog("Append ?GET=1 to a URL for faster loading or supply:\n");
rfbLog("-env X11VNC_EXTRA_HTTPS_PARAMS='?GET=1' to cmdline.\n");
}
}
rfbLog("\n");
}
}
void set_vnc_desktop_name(void) {
sprintf(vnc_desktop_name, "unknown");
if (inetd) {
sprintf(vnc_desktop_name, "%s/inetd-no-further-clients",
this_host());
}
if (remote_direct) {
return;
}
if (screen->port) {
do_mention_java_urls();
if (use_openssl) {
announce(screen->port, 1, listen_str);
} else {
announce(screen->port, 0, listen_str);
}
if (stunnel_port) {
announce(stunnel_port, 1, NULL);
}
do_announce_http();
fflush(stderr);
if (inetd) {
; /* should not occur (port != 0) */
} else {
fprintf(stdout, "PORT=%d\n", screen->port);
if (stunnel_port) {
fprintf(stdout, "SSLPORT=%d\n", stunnel_port);
} else if (use_openssl) {
if (enc_str && !strcmp(enc_str, "none")) {
;
} else if (enc_str) {
fprintf(stdout, "ENCPORT=%d\n", screen->port);
} else {
fprintf(stdout, "SSLPORT=%d\n", screen->port);
}
}
fflush(stdout);
if (flagfile) {
FILE *flag = fopen(flagfile, "w");
if (flag) {
fprintf(flag, "PORT=%d\n",screen->port);
if (stunnel_port) {
fprintf(flag, "SSL_PORT=%d\n",
stunnel_port);
}
fflush(flag);
fclose(flag);
} else {
rfbLog("could not open flag file: %s\n",
flagfile);
}
}
if (rm_flagfile) {
int create = 0;
struct stat sb;
if (strstr(rm_flagfile, "create:") == rm_flagfile) {
char *s = rm_flagfile;
create = 1;
rm_flagfile = strdup(rm_flagfile + strlen("create:"));
free(s);
}
if (strstr(rm_flagfile, "nocreate:") == rm_flagfile) {
char *s = rm_flagfile;
create = 0;
rm_flagfile = strdup(rm_flagfile + strlen("nocreate:"));
free(s);
} else if (stat(rm_flagfile, &sb) != 0) {
create = 1;
}
if (create) {
FILE *flag = fopen(rm_flagfile, "w");
if (flag) {
fprintf(flag, "%d\n", getpid());
fclose(flag);
}
}
}
}
fflush(stdout);
}
}
static void check_cursor_changes(void) {
static double last_push = 0.0;
if (unixpw_in_progress) return;
cursor_changes += check_x11_pointer();
if (cursor_changes) {
double tm, max_push = 0.125, multi_push = 0.01, wait = 0.02;
int cursor_shape, dopush = 0, link, latency, netrate;
if (! all_clients_initialized()) {
/* play it safe */
return;
}
if (0) cursor_shape = cursor_shape_updates_clients(screen);
dtime0(&tm);
link = link_rate(&latency, &netrate);
if (link == LR_DIALUP) {
max_push = 0.2;
wait = 0.05;
} else if (link == LR_BROADBAND) {
max_push = 0.075;
wait = 0.05;
} else if (link == LR_LAN) {
max_push = 0.01;
} else if (latency < 5 && netrate > 200) {
max_push = 0.01;
}
if (tm > last_push + max_push) {
dopush = 1;
} else if (cursor_changes > 1 && tm > last_push + multi_push) {
dopush = 1;
}
if (dopush) {
mark_rect_as_modified(0, 0, 1, 1, 1);
fb_push_wait(wait, FB_MOD);
last_push = tm;
} else {
rfbPE(0);
}
}
cursor_changes = 0;
}
/*
* These watch_loop() releated were moved from x11vnc.c so we could try
* to remove -O2 from its compilation. TDB new file, e.g. watch.c.
*/
static void check_filexfer(void) {
static time_t last_check = 0;
rfbClientIteratorPtr iter;
rfbClientPtr cl;
int transferring = 0;
if (time(NULL) <= last_check) {
return;
}
#if 0
if (getenv("NOFT")) {
return;
}
#endif
iter = rfbGetClientIterator(screen);
while( (cl = rfbClientIteratorNext(iter)) ) {
if (cl->fileTransfer.receiving) {
transferring = 1;
break;
}
if (cl->fileTransfer.sending) {
transferring = 1;
break;
}
}
rfbReleaseClientIterator(iter);
if (transferring) {
double start = dnow();
while (dnow() < start + 0.5) {
rfbCFD(5000);
rfbCFD(1000);
rfbCFD(0);
}
} else {
last_check = time(NULL);
}
}
static void record_last_fb_update(void) {
static int rbs0 = -1;
static time_t last_call = 0;
time_t now = time(NULL);
int rbs = -1;
rfbClientIteratorPtr iter;
rfbClientPtr cl;
if (last_fb_bytes_sent == 0) {
last_fb_bytes_sent = now;
last_call = now;
}
if (now <= last_call + 1) {
/* check every second or so */
return;
}
if (unixpw_in_progress) return;
last_call = now;
if (! screen) {
return;
}
iter = rfbGetClientIterator(screen);
while( (cl = rfbClientIteratorNext(iter)) ) {
#if 0
rbs += cl->rawBytesEquivalent;
#else
#if LIBVNCSERVER_HAS_STATS
rbs += rfbStatGetSentBytesIfRaw(cl);
#endif
#endif
}
rfbReleaseClientIterator(iter);
if (rbs != rbs0) {
rbs0 = rbs;
if (debug_tiles > 1) {
fprintf(stderr, "record_last_fb_update: %d %d\n",
(int) now, (int) last_fb_bytes_sent);
}
last_fb_bytes_sent = now;
}
}
static int choose_delay(double dt) {
static double t0 = 0.0, t1 = 0.0, t2 = 0.0, now;
static int x0, y0, x1, y1, x2, y2, first = 1;
int dx0, dy0, dx1, dy1, dm, i, msec = waitms;
double cut1 = 0.15, cut2 = 0.075, cut3 = 0.25;
double bogdown_time = 0.25, bave = 0.0;
int bogdown = 1, bcnt = 0;
int ndt = 8, nave = 3;
double fac = 1.0;
static int db = 0, did_set_defer = 0;
static double dts[8];
static int link = LR_UNSET, latency = -1, netrate = -1;
static double last_link = 0.0;
if (screen && did_set_defer) {
/* reset defer in case we changed it */
screen->deferUpdateTime = defer_update;
}
if (waitms == 0) {
return waitms;
}
if (nofb) {
return waitms;
}
if (first) {
for(i=0; i<ndt; i++) {
dts[i] = 0.0;
}
if (getenv("DEBUG_DELAY")) {
db = atoi(getenv("DEBUG_DELAY"));
}
if (getenv("SET_DEFER")) {
set_defer = atoi(getenv("SET_DEFER"));
}
first = 0;
}
now = dnow();
if (now > last_link + 30.0 || link == LR_UNSET) {
link = link_rate(&latency, &netrate);
last_link = now;
}
/*
* first check for bogdown, e.g. lots of activity, scrolling text
* from command output, etc.
*/
if (nap_ok) {
dt = 0.0;
}
if (! wait_bog) {
bogdown = 0;
} else if (button_mask || now < last_keyboard_time + 2*bogdown_time) {
/*
* let scrolls & keyboard input through the normal way
* otherwise, it will likely just annoy them.
*/
bogdown = 0;
} else if (dt > 0.0) {
/*
* inspect recent dt's:
* 0 1 2 3 4 5 6 7 dt
* ^ ^ ^
*/
for (i = ndt - (nave - 1); i < ndt; i++) {
bave += dts[i];
bcnt++;
if (dts[i] < bogdown_time) {
bogdown = 0;
break;
}
}
bave += dt;
bcnt++;
bave = bave / bcnt;
if (dt < bogdown_time) {
bogdown = 0;
}
} else {
bogdown = 0;
}
/* shift for next time */
for (i = 0; i < ndt-1; i++) {
dts[i] = dts[i+1];
}
dts[ndt-1] = dt;
if (0 && dt > 0.0) fprintf(stderr, "dt: %.5f %.4f\n", dt, dnowx());
if (bogdown) {
if (use_xdamage) {
/* DAMAGE can queue ~1000 rectangles for a scroll */
clear_xdamage_mark_region(NULL, 0);
}
msec = (int) (1000 * 1.75 * bave);
if (dts[ndt - nave - 1] > 0.75 * bave) {
msec = 1.5 * msec;
set_xdamage_mark(0, 0, dpy_x, dpy_y);
}
if (msec > 1500) {
msec = 1500;
}
if (msec < waitms) {
msec = waitms;
}
db = (db || debug_tiles);
if (db) fprintf(stderr, "bogg[%d] %.3f %.3f %.3f %.3f\n",
msec, dts[ndt-4], dts[ndt-3], dts[ndt-2], dts[ndt-1]);
return msec;
}
/* next check for pointer motion, keystrokes, to speed up */
t2 = dnow();
x2 = cursor_x;
y2 = cursor_y;
dx0 = nabs(x1 - x0);
dy0 = nabs(y1 - y0);
dx1 = nabs(x2 - x1);
dy1 = nabs(y2 - y1);
/* bigger displacement for most recent dt: */
if (dx1 > dy1) {
dm = dx1;
} else {
dm = dy1;
}
if ((dx0 || dy0) && (dx1 || dy1)) {
/* if mouse moved the previous two times: */
if (t2 < t0 + cut1 || t2 < t1 + cut2 || dm > 20) {
/*
* if within 0.15s(0) or 0.075s(1) or mouse
* moved > 20pixels, set and bump up the cut
* down factor.
*/
fac = wait_ui * 1.5;
} else if ((dx1 || dy1) && dm > 40) {
fac = wait_ui;
} else {
/* still 1.0? */
if (db > 1) fprintf(stderr, "wait_ui: still 1.0\n");
}
} else if ((dx1 || dy1) && dm > 40) {
/* if mouse moved > 40 last time: */
fac = wait_ui;
}
if (fac == 1.0 && t2 < last_keyboard_time + cut3) {
/* if typed in last 0.25s set wait_ui */
fac = wait_ui;
}
if (fac != 1.0) {
if (link == LR_LAN || latency <= 3) {
fac *= 1.5;
}
}
msec = (int) (((double) waitms) / fac);
if (msec == 0) {
msec = 1;
}
if (set_defer && fac != 1.0 && screen) {
/* this is wait_ui mode, set defer to match wait: */
if (set_defer >= 1) {
screen->deferUpdateTime = msec;
} else if (set_defer <= -1) {
screen->deferUpdateTime = 0;
}
if (nabs(set_defer) == 2) {
urgent_update = 1;
}
did_set_defer = 1;
}
x0 = x1;
y0 = y1;
t0 = t1;
x1 = x2;
y1 = y2;
t1 = t2;
if (db > 1) fprintf(stderr, "wait: %2d defer[%02d]: %2d\n", msec, defer_update, screen->deferUpdateTime);
return msec;
}
/*
* main x11vnc loop: polls, checks for events, iterate libvncserver, etc.
*/
void watch_loop(void) {
int cnt = 0, tile_diffs = 0, skip_pe = 0, wait;
double tm, dtr, dt = 0.0;
time_t start = time(NULL);
if (use_threads && !started_rfbRunEventLoop) {
started_rfbRunEventLoop = 1;
rfbRunEventLoop(screen, -1, TRUE);
}
while (1) {
char msg[] = "new client: %s taking unixpw client off hold.\n";
int skip_scan_for_updates = 0;
got_user_input = 0;
got_pointer_input = 0;
got_local_pointer_input = 0;
got_pointer_calls = 0;
got_keyboard_input = 0;
got_keyboard_calls = 0;
urgent_update = 0;
x11vnc_current = dnow();
if (! use_threads) {
dtime0(&tm);
if (! skip_pe) {
if (unixpw_in_progress) {
rfbClientPtr cl = unixpw_client;
if (cl && cl->onHold) {
rfbLog(msg, cl->host);
unixpw_client->onHold = FALSE;
}
} else {
measure_send_rates(1);
}
unixpw_in_rfbPE = 1;
/*
* do a few more since a key press may
* have induced a small change we want to
* see quickly (just 1 rfbPE will likely
* only process the subsequent "up" event)
*/
if (tm < last_keyboard_time + 0.20) {
rfbPE(0);
rfbPE(0);
rfbPE(-1);
rfbPE(0);
rfbPE(0);
} else {
if (extra_fbur > 0) {
int i;
for (i=0; i < extra_fbur; i++) {
rfbPE(0);
}
}
rfbPE(-1);
}
if (x11vnc_current < last_new_client + 0.5) {
urgent_update = 1;
}
unixpw_in_rfbPE = 0;
if (unixpw_in_progress) {
/* rfbPE loop until logged in. */
skip_pe = 0;
check_new_clients();
continue;
} else {
measure_send_rates(0);
fb_update_sent(NULL);
}
} else {
if (unixpw_in_progress) {
skip_pe = 0;
check_new_clients();
continue;
}
}
dtr = dtime(&tm);
if (! cursor_shape_updates) {
/* undo any cursor shape requests */
disable_cursor_shape_updates(screen);
}
if (screen && screen->clientHead) {
int ret = check_user_input(dt, dtr, tile_diffs, &cnt);
/* true: loop back for more input */
if (ret == 2) {
skip_pe = 1;
}
if (ret) {
if (debug_scroll) fprintf(stderr, "watch_loop: LOOP-BACK: %d\n", ret);
continue;
}
}
/* watch for viewonly input piling up: */
if ((got_pointer_calls > got_pointer_input) ||
(got_keyboard_calls > got_keyboard_input)) {
eat_viewonly_input(10, 3);
}
} else {
/* -threads here. */
if (unixpw_in_progress) {
rfbClientPtr cl = unixpw_client;
if (cl && cl->onHold) {
rfbLog(msg, cl->host);
unixpw_client->onHold = FALSE;
}
}
if (use_xrecord) {
check_xrecord();
}
if (wireframe && button_mask) {
check_wireframe();
}
}
skip_pe = 0;
if (shut_down) {
clean_up_exit(0);
}
if (unixpw_in_progress) {
check_new_clients();
continue;
}
if (! urgent_update) {
if (do_copy_screen) {
do_copy_screen = 0;
copy_screen();
}
check_new_clients();
check_ncache(0, 0);
check_xevents(0);
check_autorepeat();
check_pm();
check_filexfer();
check_keycode_state();
check_connect_inputs();
check_gui_inputs();
check_stunnel();
check_openssl();
check_https();
record_last_fb_update();
check_padded_fb();
check_fixscreen();
check_xdamage_state();
check_xrecord_reset(0);
check_add_keysyms();
check_new_passwds(0);
#ifdef ENABLE_GRABLOCAL
if (grab_local) {
check_local_grab();
}
#endif
if (started_as_root) {
check_switched_user();
}
if (first_conn_timeout < 0) {
start = time(NULL);
first_conn_timeout = -first_conn_timeout;
}
}
if (rawfb_vnc_reflect) {
static time_t lastone = 0;
if (time(NULL) > lastone + 10) {
lastone = time(NULL);
vnc_reflect_process_client();
}
}
if (first_conn_timeout) {
int t = first_conn_timeout;
if (!clients_served) {
if (time(NULL) - start > first_conn_timeout) {
rfbLog("No client after %d secs.\n", t);
shut_down = 1;
}
} else {
if (!client_normal_count) {
if (time(NULL) - start > t + 3) {
rfbLog("No valid client after %d secs.\n", t + 3);
shut_down = 1;
}
}
}
}
if (! screen || ! screen->clientHead) {
/* waiting for a client */
usleep(200 * 1000);
continue;
}
if (first_conn_timeout && all_clients_initialized()) {
first_conn_timeout = 0;
}
if (nofb) {
/* no framebuffer polling needed */
if (cursor_pos_updates) {
check_x11_pointer();
}
#ifdef MACOSX
else check_x11_pointer();
#endif
continue;
}
if (x11vnc_current < last_new_client + 0.5 && !all_clients_initialized()) {
continue;
}
if (subwin && freeze_when_obscured) {
/* XXX not working */
X_LOCK;
XFlush_wr(dpy);
X_UNLOCK;
check_xevents(0);
if (subwin_obscured) {
skip_scan_for_updates = 1;
}
}
if (skip_scan_for_updates) {
;
} else if (button_mask && (!show_dragging || pointer_mode == 0)) {
/*
* if any button is pressed in this mode do
* not update rfb screen, but do flush the
* X11 display.
*/
X_LOCK;
XFlush_wr(dpy);
X_UNLOCK;
dt = 0.0;
} else {
static double last_dt = 0.0;
double xdamage_thrash = 0.4;
static int tilecut = -1;
check_cursor_changes();
/* for timing the scan to try to detect thrashing */
if (use_xdamage && last_dt > xdamage_thrash) {
clear_xdamage_mark_region(NULL, 0);
}
if (unixpw_in_progress) continue;
if (rawfb_vnc_reflect) {
vnc_reflect_process_client();
}
dtime0(&tm);
#if !NO_X11
if (xrandr_present && !xrandr && xrandr_maybe) {
int delay = 180;
/* there may be xrandr right after xsession start */
if (tm < x11vnc_start + delay || tm < last_client + delay) {
int tw = 20;
if (auth_file != NULL) {
tw = 120;
}
X_LOCK;
if (tm < x11vnc_start + tw || tm < last_client + tw) {
XSync(dpy, False);
} else {
XFlush_wr(dpy);
}
X_UNLOCK;
}
X_LOCK;
check_xrandr_event("before-scan");
X_UNLOCK;
}
#endif
if (use_snapfb) {
int t, tries = 3;
copy_snap();
for (t=0; t < tries; t++) {
tile_diffs = scan_for_updates(0);
}
} else {
tile_diffs = scan_for_updates(0);
}
dt = dtime(&tm);
if (! nap_ok) {
last_dt = dt;
}
if (tilecut < 0) {
if (getenv("TILECUT")) {
tilecut = atoi(getenv("TILECUT"));
}
if (tilecut < 0) tilecut = 4;
}
if ((debug_tiles || debug_scroll > 1 || debug_wireframe > 1)
&& (tile_diffs > tilecut || debug_tiles > 1)) {
double rate = (tile_x * tile_y * bpp/8 * tile_diffs) / dt;
fprintf(stderr, "============================= TILES: %d dt: %.4f"
" t: %.4f %.2f MB/s nap_ok: %d\n", tile_diffs, dt,
tm - x11vnc_start, rate/1000000.0, nap_ok);
}
}
/* sleep a bit to lessen load */
wait = choose_delay(dt);
if (urgent_update) {
;
} else if (wait > 2*waitms) {
/* bog case, break it up */
nap_sleep(wait, 10);
} else {
double t1, t2;
int idt;
if (extra_fbur > 0) {
int i;
for (i=0; i <= extra_fbur; i++) {
int r = rfbPE(0);
if (!r) break;
}
}
/* sometimes the sleep is too short, so measure it: */
t1 = dnow();
usleep(wait * 1000);
t2 = dnow();
idt = (int) (1000. * (t2 - t1));
if (idt > 0 && idt < wait) {
/* try to sleep the remainder */
usleep((wait - idt) * 1000);
}
}
cnt++;
}
}