suckless/slock
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

applied Fernando Silveira's multiscreen patch for old style multihead setups

Browse source
This commit is contained in:
Anselm R Garbe 2009-11-26 12:53:26 +00:00
parent 0cb05bdb75
commit 732d2b3bf5
1 changed files with 152 additions and 70 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

220
slock.c
View file

@ -5,6 +5,7 @@
#endif #endif
#include <ctype.h> #include <ctype.h>
#include <errno.h>
#include <pwd.h> #include <pwd.h>
#include <stdarg.h> #include <stdarg.h>
#include <stdlib.h> #include <stdlib.h>
@ -15,31 +16,42 @@
#include <X11/keysym.h> #include <X11/keysym.h>
#include <X11/Xlib.h> #include <X11/Xlib.h>
#include <X11/Xutil.h> #include <X11/Xutil.h>
#include <X11/extensions/dpms.h>
#if HAVE_BSD_AUTH #if HAVE_BSD_AUTH
#include <login_cap.h> #include <login_cap.h>
#include <bsd_auth.h> #include <bsd_auth.h>
#endif #endif
struct st_lock {
int screen;
Window root, w;
Pixmap pmap;
};
extern const char *__progname;
static void static void
die(const char *errstr, ...) { die(const char *errstr, ...) {
va_list ap; va_list ap;
fprintf(stderr, "%s: ", __progname);
va_start(ap, errstr); va_start(ap, errstr);
vfprintf(stderr, errstr, ap); vfprintf(stderr, errstr, ap);
va_end(ap); va_end(ap);
fprintf(stderr, "\n");
fflush(stderr);
exit(EXIT_FAILURE); exit(EXIT_FAILURE);
} }
#ifndef HAVE_BSD_AUTH #ifndef HAVE_BSD_AUTH
static const char * static const char *
get_password() { /* only run as root */ get_password(void) { /* only run as root */
const char *rval; const char *rval;
struct passwd *pw; struct passwd *pw;
if(geteuid() != 0) if(geteuid() != 0)
die("slock: cannot retrieve password entry (make sure to suid slock)\n"); die("cannot retrieve password entry (make sure to suid slock)");
pw = getpwuid(getuid()); pw = getpwuid(getuid());
endpwent(); endpwent();
rval = pw->pw_passwd; rval = pw->pw_passwd;
@ -55,81 +67,34 @@ get_password() { /* only run as root */
/* drop privileges */ /* drop privileges */
if(setgid(pw->pw_gid) < 0 || setuid(pw->pw_uid) < 0) if(setgid(pw->pw_gid) < 0 || setuid(pw->pw_uid) < 0)
die("slock: cannot drop privileges\n"); die("cannot drop privileges");
return rval; return rval;
} }
#endif #endif
int static void
main(int argc, char **argv) { #ifdef HAVE_BSD_AUTH
char curs[] = {0, 0, 0, 0, 0, 0, 0, 0}; read_password(Display *dpy)
char buf[32], passwd[256]; #else
int num, screen; read_password(Display *dpy, const char *pws)
#ifndef HAVE_BSD_AUTH
const char *pws;
#endif #endif
{
char buf[32], passwd[256];
int num;
unsigned int len; unsigned int len;
Bool running = True; Bool running = True;
Cursor invisible;
Display *dpy;
KeySym ksym; KeySym ksym;
Pixmap pmap;
Window root, w;
XColor black, dummy;
XEvent ev; XEvent ev;
XSetWindowAttributes wa;
CARD16 standby, suspend, off;
if((argc == 2) && !strcmp("-v", argv[1]))
die("slock-"VERSION", © 2006-2008 Anselm R Garbe\n");
else if(argc != 1)
die("usage: slock [-v]\n");
#ifndef HAVE_BSD_AUTH
pws = get_password();
#endif
if(!(dpy = XOpenDisplay(0)))
die("slock: cannot open display\n");
screen = DefaultScreen(dpy);
root = RootWindow(dpy, screen);
/* init */
wa.override_redirect = 1;
wa.background_pixel = BlackPixel(dpy, screen);
w = XCreateWindow(dpy, root, 0, 0, DisplayWidth(dpy, screen), DisplayHeight(dpy, screen),
0, DefaultDepth(dpy, screen), CopyFromParent,
DefaultVisual(dpy, screen), CWOverrideRedirect | CWBackPixel, &wa);
XAllocNamedColor(dpy, DefaultColormap(dpy, screen), "black", &black, &dummy);
pmap = XCreateBitmapFromData(dpy, w, curs, 8, 8);
invisible = XCreatePixmapCursor(dpy, pmap, pmap, &black, &black, 0, 0);
XDefineCursor(dpy, w, invisible);
XMapRaised(dpy, w);
for(len = 1000; len; len--) {
if(XGrabPointer(dpy, root, False, ButtonPressMask | ButtonReleaseMask | PointerMotionMask,
GrabModeAsync, GrabModeAsync, None, invisible, CurrentTime) == GrabSuccess)
break;
usleep(1000);
}
if((running = running && (len > 0))) {
for(len = 1000; len; len--) {
if(XGrabKeyboard(dpy, root, True, GrabModeAsync, GrabModeAsync, CurrentTime)
== GrabSuccess)
break;
usleep(1000);
}
running = (len > 0);
}
len = 0; len = 0;
XSync(dpy, False); running = True;
if(DPMSCapable(dpy)) { /* save and customize DPMS settings */ /* As "slock" stands for "Simple X display locker", the DPMS settings
DPMSGetTimeouts(dpy, &standby, &suspend, &off); * had been removed and you can set it with "xset" or some other
DPMSSetTimeouts(dpy, 10, 30, 90); * utility. This way the user can easily set a customized DPMS
} * timeout. */
/* main event loop */
while(running && !XNextEvent(dpy, &ev)) { while(running && !XNextEvent(dpy, &ev)) {
if(ev.type == KeyPress) { if(ev.type == KeyPress) {
buf[0] = 0; buf[0] = 0;
@ -172,12 +137,129 @@ main(int argc, char **argv) {
} }
} }
} }
if(DPMSCapable(dpy)) { /* restore DPMS settings */
DPMSSetTimeouts(dpy, standby, suspend, off);
} }
static void
unlockscreen(Display *dpy, struct st_lock *lock) {
if (dpy == NULL || lock == NULL)
return;
XUngrabPointer(dpy, CurrentTime); XUngrabPointer(dpy, CurrentTime);
XFreePixmap(dpy, pmap); XFreePixmap(dpy, lock->pmap);
XDestroyWindow(dpy, w); XDestroyWindow(dpy, lock->w);
free(lock);
}
static struct st_lock *
lockscreen(Display *dpy, int screen) {
char curs[] = {0, 0, 0, 0, 0, 0, 0, 0};
unsigned int len;
struct st_lock *lock;
Bool running = True;
XColor black, dummy;
XSetWindowAttributes wa;
Cursor invisible;
if (dpy == NULL || screen < 0)
return NULL;
lock = malloc(sizeof(struct st_lock));
if (lock == NULL)
return NULL;
lock->screen = screen;
lock->root = RootWindow(dpy, lock->screen);
/* init */
wa.override_redirect = 1;
wa.background_pixel = BlackPixel(dpy, lock->screen);
lock->w = XCreateWindow(dpy, lock->root, 0, 0, DisplayWidth(dpy, lock->screen), DisplayHeight(dpy, lock->screen),
0, DefaultDepth(dpy, lock->screen), CopyFromParent,
DefaultVisual(dpy, lock->screen), CWOverrideRedirect | CWBackPixel, &wa);
XAllocNamedColor(dpy, DefaultColormap(dpy, lock->screen), "black", &black, &dummy);
lock->pmap = XCreateBitmapFromData(dpy, lock->w, curs, 8, 8);
invisible = XCreatePixmapCursor(dpy, lock->pmap, lock->pmap, &black, &black, 0, 0);
XDefineCursor(dpy, lock->w, invisible);
XMapRaised(dpy, lock->w);
for(len = 1000; len; len--) {
if(XGrabPointer(dpy, lock->root, False, ButtonPressMask | ButtonReleaseMask | PointerMotionMask,
GrabModeAsync, GrabModeAsync, None, invisible, CurrentTime) == GrabSuccess)
break;
usleep(1000);
}
if((running = running && (len > 0))) {
for(len = 1000; len; len--) {
if(XGrabKeyboard(dpy, lock->root, True, GrabModeAsync, GrabModeAsync, CurrentTime)
== GrabSuccess)
break;
usleep(1000);
}
running = (len > 0);
}
if (!running) {
unlockscreen(dpy, lock);
lock = NULL;
}
return lock;
}
static void
usage(void) {
fprintf(stderr, "usage: %s -v", __progname);
exit(EXIT_FAILURE);
}
int
main(int argc, char **argv) {
#ifndef HAVE_BSD_AUTH
const char *pws;
#endif
Display *dpy;
int nscreens, screen;
struct st_lock **locks;
if((argc == 2) && !strcmp("-v", argv[1]))
die("slock-%s, © 2006-2008 Anselm R Garbe", VERSION);
else if(argc != 1)
usage();
#ifndef HAVE_BSD_AUTH
pws = get_password();
#endif
if(!(dpy = XOpenDisplay(0)))
die("cannot open display");
/* Get the number of screens in display "dpy" and blank them all. */
nscreens = ScreenCount(dpy);
locks = malloc(sizeof(struct st_lock *) * nscreens);
if (locks == NULL)
die("malloc: %s", strerror(errno));
for (screen = 0; screen < nscreens; screen++)
locks[screen] = lockscreen(dpy, screen);
XSync(dpy, False);
/* Everything is now blank. Now wait for the correct password. */
#ifdef HAVE_BSD_AUTH
read_password(dpy);
#else
read_password(dpy, pws);
#endif
/* Password ok, unlock everything and quit. */
for (screen = 0; screen < nscreens; screen++)
unlockscreen(dpy, locks[screen]);
free(locks);
XCloseDisplay(dpy); XCloseDisplay(dpy);
return 0; return 0;
} }