lib/swm_hack.c

   1 /*
   2  * Copyright (c) 2009 Marco Peereboom <marco@peereboom.us>
   3  * Copyright (c) 2009 Ryan McBride <mcbride@countersiege.com>
   4  *
   5  * Permission to use, copy, modify, and distribute this software for any
   6  * purpose with or without fee is hereby granted, provided that the above
   7  * copyright notice and this permission notice appear in all copies.
   8  *
   9  * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
  10  * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
  11  * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
  12  * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
  13  * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
  14  * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
  15  * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
  16  */
  17 /*
  18  * Copyright (C) 2005-2007 Carsten Haitzler
  19  * Copyright (C) 2006-2007 Kim Woelders
  20  *
  21  * Permission is hereby granted, free of charge, to any person obtaining a copy
  22  * of this software and associated documentation files (the "Software"), to
  23  * deal in the Software without restriction, including without limitation the
  24  * rights to use, copy, modify, merge, publish, distribute, sublicense, and/or
  25  * sell copies of the Software, and to permit persons to whom the Software is
  26  * furnished to do so, subject to the following conditions:
  27  *
  28  * The above copyright notice and this permission notice shall be included in
  29  * all copies of the Software, its documentation and marketing & publicity
  30  * materials, and acknowledgment shall be given in the documentation, materials
  31  * and software packages that this Software was used.
  32  *
  33  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  34  * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  35  * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL
  36  * THE AUTHORS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER
  37  * IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
  38  * CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
  39  */
  40 /*
  41  * Basic hack mechanism (dlopen etc.) taken from e_hack.c in e17.
  42  */
  43 #include <string.h>
  44 #include <stdio.h>
  45 #include <stdlib.h>
  46 #include <dlfcn.h>
  47 #include <X11/Xlib.h>
  48 #include <X11/X.h>
  49 #include <X11/Xatom.h>
  50 #include <X11/Intrinsic.h>
  51
  52 /* dlopened libs so we can find the symbols in the real one to call them */
  53 static void             *lib_xlib = NULL;
  54 static void             *lib_xtlib = NULL;
  55
  56 static Window           root = None;
  57 static int              xterm = 0;
  58 static Display          *dpy = NULL;
  59
  60 /* Find our root window */
  61 static              Window
  62 MyRoot(Display * dpy)
  63 {
  64         char               *s;
  65
  66         if (root != None)
  67                 return root;
  68
  69         root = DefaultRootWindow(dpy);
  70
  71         s = getenv("ENL_WM_ROOT");
  72         if (!s)
  73                 return root;
  74
  75         sscanf(s, "%lx", &root);
  76         return root;
  77 }
  78
  79
  80 typedef Atom    (XIA) (Display *display, char *atom_name, Bool
  81                     only_if_exists);
  82
  83 typedef int     (XCP) (Display *display, Window w, Atom property,
  84                     Atom type, int format, int mode, unsigned char *data,
  85                     int nelements);
  86
  87 #define SWM_PROPLEN     (16)
  88 void
  89 set_property(Display *dpy, Window id, char *name, char *val)
  90 {
  91         Atom                    atom = 0;
  92         char                    prop[SWM_PROPLEN];
  93         static XIA              *xia = NULL;
  94         static XCP              *xcp = NULL;
  95
  96         /* find the real Xlib and the real X function */
  97         if (!lib_xlib)
  98                 lib_xlib = dlopen("libX11.so", RTLD_GLOBAL | RTLD_LAZY);
  99         if (!xia)
 100                 xia = (XIA *) dlsym(lib_xlib, "XInternAtom");
 101         if (!xcp)
 102                 xcp = (XCP *) dlsym(lib_xlib, "XChangeProperty");
 103
 104         /* Try to update the window's workspace property */
 105         atom = (*xia)(dpy, name, False);
 106         if (atom)
 107                 if (snprintf(prop, SWM_PROPLEN, "%s", val) < SWM_PROPLEN)
 108                         (*xcp)(dpy, id, atom, XA_STRING,
 109                             8, PropModeReplace, (unsigned char *)prop,
 110                             strlen((char *)prop));
 111 }
 112
 113 typedef             Window(CWF) (Display * _display, Window _parent, int _x,
 114                                  int _y, unsigned int _width,
 115                                  unsigned int _height,
 116                                  unsigned int _border_width, int _depth,
 117                                  unsigned int _class, Visual * _visual,
 118                                  unsigned long _valuemask,
 119                                  XSetWindowAttributes * _attributes);
 120
 121 /* XCreateWindow intercept hack */
 122 Window
 123 XCreateWindow(Display * display, Window parent, int x, int y,
 124    unsigned int width, unsigned int height,
 125    unsigned int border_width,
 126    int depth, unsigned int clss, Visual * visual,
 127    unsigned long valuemask, XSetWindowAttributes * attributes)
 128 {
 129         static CWF      *func = NULL;
 130         char            *env;
 131         Window          id;
 132
 133         /* find the real Xlib and the real X function */
 134         if (!lib_xlib)
 135                 lib_xlib = dlopen("libX11.so", RTLD_GLOBAL | RTLD_LAZY);
 136         if (!func) {
 137                 func = (CWF *) dlsym(lib_xlib, "XCreateWindow");
 138                 dpy = display;
 139         }
 140
 141         if (parent == DefaultRootWindow(display))
 142                 parent = MyRoot(display);
 143
 144         id = (*func) (display, parent, x, y, width, height, border_width,
 145             depth, clss, visual, valuemask, attributes);
 146
 147         if (id) {
 148                 if ((env = getenv("_SWM_WS")) != NULL)
 149                         set_property(display, id, "_SWM_WS", env);
 150                 if ((env = getenv("_SWM_PID")) != NULL)
 151                         set_property(display, id, "_SWM_PID", env);
 152                 if ((env = getenv("_SWM_XTERM_FONTADJ")) != NULL) {
 153                         unsetenv("_SWM_XTERM_FONTADJ");
 154                         xterm = 1;
 155                 }
 156         }
 157         return (id);
 158 }
 159
 160 typedef             Window(CSWF) (Display * _display, Window _parent, int _x,
 161                                   int _y, unsigned int _width,
 162                                   unsigned int _height,
 163                                   unsigned int _border_width,
 164                                   unsigned long _border,
 165                                   unsigned long _background);
 166
 167 /* XCreateSimpleWindow intercept hack */
 168 Window
 169 XCreateSimpleWindow(Display * display, Window parent, int x, int y,
 170     unsigned int width, unsigned int height,
 171     unsigned int border_width,
 172     unsigned long border, unsigned long background)
 173 {
 174         static CSWF     *func = NULL;
 175         char            *env;
 176         Window          id;
 177
 178         /* find the real Xlib and the real X function */
 179         if (!lib_xlib)
 180                 lib_xlib = dlopen("libX11.so", RTLD_GLOBAL | RTLD_LAZY);
 181         if (!func)
 182                 func = (CSWF *) dlsym(lib_xlib, "XCreateSimpleWindow");
 183
 184         if (parent == DefaultRootWindow(display))
 185                 parent = MyRoot(display);
 186
 187         id = (*func) (display, parent, x, y, width, height,
 188             border_width, border, background);
 189
 190         if (id) {
 191                 if ((env = getenv("_SWM_WS")) != NULL)
 192                         set_property(display, id, "_SWM_WS", env);
 193                 if ((env = getenv("_SWM_PID")) != NULL)
 194                         set_property(display, id, "_SWM_PID", env);
 195                 if ((env = getenv("_SWM_XTERM_FONTADJ")) != NULL) {
 196                         unsetenv("_SWM_XTERM_FONTADJ");
 197                         xterm = 1;
 198                 }
 199         }
 200         return (id);
 201 }
 202
 203 typedef int         (RWF) (Display * _display, Window _window, Window _parent,
 204                            int x, int y);
 205
 206 /* XReparentWindow intercept hack */
 207 int
 208 XReparentWindow(Display * display, Window window, Window parent, int x, int y)
 209 {
 210         static RWF         *func = NULL;
 211
 212         /* find the real Xlib and the real X function */
 213         if (!lib_xlib)
 214                 lib_xlib = dlopen("libX11.so", RTLD_GLOBAL | RTLD_LAZY);
 215         if (!func)
 216                 func = (RWF *) dlsym(lib_xlib, "XReparentWindow");
 217
 218         if (parent == DefaultRootWindow(display))
 219                 parent = MyRoot(display);
 220
 221         return (*func) (display, window, parent, x, y);
 222 }
 223
 224 typedef         void (ANEF) (XtAppContext app_context, XEvent *event_return);
 225 int             evcount = 0;
 226
 227 /*
 228  * XtAppNextEvent Intercept Hack
 229  * Normally xterm rejects "synthetic" (XSendEvent) events to prevent spoofing.
 230  * We don't want to disable this completely, it's insecure. But hook here
 231  * and allow these mostly harmless ones that we use to adjust fonts.
 232  */
 233 void
 234 XtAppNextEvent(XtAppContext app_context, XEvent *event_return)
 235 {
 236         static ANEF     *func = NULL;
 237         static int      kp_add = 0, kp_subtract = 0;
 238
 239         /* find the real Xlib and the real X function */
 240         if (!lib_xtlib)
 241                 lib_xtlib = dlopen("libXt.so", RTLD_GLOBAL | RTLD_LAZY);
 242         if (!func) {
 243                 func = (ANEF *) dlsym(lib_xtlib, "XtAppNextEvent");
 244                 if (dpy != NULL) {
 245                         kp_add = XKeysymToKeycode(dpy, XK_KP_Add);
 246                         kp_subtract = XKeysymToKeycode(dpy, XK_KP_Subtract);
 247                 }
 248         }
 249
 250         (*func) (app_context, event_return);
 251
 252         /* Return here if it's not an Xterm. */
 253         if (!xterm)
 254                 return;
 255
 256         /* Allow spoofing of font change keystrokes. */
 257         if ((event_return->type == KeyPress ||
 258            event_return->type == KeyRelease) &&
 259            event_return->xkey.state == ShiftMask &&
 260            (event_return->xkey.keycode == kp_add ||
 261            event_return->xkey.keycode == kp_subtract))
 262                 event_return->xkey.send_event = 0;
 263 }