2008-04-08 04:55:46 -04:00
|
|
|
/* See LICENSE file for license details. */
|
2006-10-12 03:33:38 -04:00
|
|
|
#define _XOPEN_SOURCE 500
|
2006-10-31 02:43:25 -05:00
|
|
|
#if HAVE_SHADOW_H
|
|
|
|
#include <shadow.h>
|
|
|
|
#endif
|
2006-10-12 02:11:08 -04:00
|
|
|
|
2006-10-12 03:33:38 -04:00
|
|
|
#include <ctype.h>
|
2009-11-26 07:53:26 -05:00
|
|
|
#include <errno.h>
|
2006-10-31 02:35:54 -05:00
|
|
|
#include <pwd.h>
|
2007-11-24 15:17:32 -05:00
|
|
|
#include <stdarg.h>
|
2006-10-11 06:35:21 -04:00
|
|
|
#include <stdlib.h>
|
|
|
|
#include <stdio.h>
|
|
|
|
#include <string.h>
|
|
|
|
#include <unistd.h>
|
|
|
|
#include <sys/types.h>
|
|
|
|
#include <X11/keysym.h>
|
|
|
|
#include <X11/Xlib.h>
|
|
|
|
#include <X11/Xutil.h>
|
2008-02-22 05:13:12 -05:00
|
|
|
|
|
|
|
#if HAVE_BSD_AUTH
|
|
|
|
#include <login_cap.h>
|
|
|
|
#include <bsd_auth.h>
|
|
|
|
#endif
|
2006-10-11 06:35:21 -04:00
|
|
|
|
2012-02-05 10:38:58 -05:00
|
|
|
typedef struct {
|
2009-11-26 07:53:26 -05:00
|
|
|
int screen;
|
2012-02-05 10:38:58 -05:00
|
|
|
Window root, win;
|
2009-11-26 07:53:26 -05:00
|
|
|
Pixmap pmap;
|
2012-02-05 10:38:58 -05:00
|
|
|
} Lock;
|
2009-11-26 07:53:26 -05:00
|
|
|
|
2012-02-05 10:38:58 -05:00
|
|
|
static Lock **locks;
|
|
|
|
static int nscreens;
|
|
|
|
static Bool running = True;
|
2009-11-26 07:53:26 -05:00
|
|
|
|
2008-07-29 14:14:53 -04:00
|
|
|
static void
|
2008-07-29 14:08:18 -04:00
|
|
|
die(const char *errstr, ...) {
|
2007-11-24 15:17:32 -05:00
|
|
|
va_list ap;
|
|
|
|
|
|
|
|
va_start(ap, errstr);
|
|
|
|
vfprintf(stderr, errstr, ap);
|
|
|
|
va_end(ap);
|
|
|
|
exit(EXIT_FAILURE);
|
|
|
|
}
|
|
|
|
|
2008-02-22 05:13:12 -05:00
|
|
|
#ifndef HAVE_BSD_AUTH
|
2008-07-29 14:14:53 -04:00
|
|
|
static const char *
|
2012-01-22 12:58:10 -05:00
|
|
|
getpw(void) { /* only run as root */
|
2006-10-31 02:35:54 -05:00
|
|
|
const char *rval;
|
|
|
|
struct passwd *pw;
|
|
|
|
|
|
|
|
pw = getpwuid(getuid());
|
2012-01-24 16:10:02 -05:00
|
|
|
if(!pw)
|
2012-02-05 10:38:58 -05:00
|
|
|
die("slock: cannot retrieve password entry (make sure to suid or sgid slock)");
|
2006-10-31 02:35:54 -05:00
|
|
|
endpwent();
|
|
|
|
rval = pw->pw_passwd;
|
|
|
|
|
|
|
|
#if HAVE_SHADOW_H
|
|
|
|
{
|
|
|
|
struct spwd *sp;
|
|
|
|
sp = getspnam(getenv("USER"));
|
2012-01-24 16:10:02 -05:00
|
|
|
if(!sp)
|
|
|
|
die("slock: cannot retrieve shadow entry (make sure to suid or sgid slock)\n");
|
2006-10-31 02:35:54 -05:00
|
|
|
endspent();
|
|
|
|
rval = sp->sp_pwdp;
|
|
|
|
}
|
|
|
|
#endif
|
2008-02-22 05:13:12 -05:00
|
|
|
|
2006-10-31 02:35:54 -05:00
|
|
|
/* drop privileges */
|
2007-11-24 15:17:32 -05:00
|
|
|
if(setgid(pw->pw_gid) < 0 || setuid(pw->pw_uid) < 0)
|
2012-02-05 10:38:58 -05:00
|
|
|
die("slock: cannot drop privileges");
|
2006-10-31 02:35:54 -05:00
|
|
|
return rval;
|
|
|
|
}
|
2008-02-22 05:13:12 -05:00
|
|
|
#endif
|
2006-10-31 02:35:54 -05:00
|
|
|
|
2009-11-26 07:53:26 -05:00
|
|
|
static void
|
|
|
|
#ifdef HAVE_BSD_AUTH
|
2012-01-22 12:58:10 -05:00
|
|
|
readpw(Display *dpy)
|
2009-11-26 07:53:26 -05:00
|
|
|
#else
|
2012-01-22 12:58:10 -05:00
|
|
|
readpw(Display *dpy, const char *pws)
|
2009-11-26 07:53:26 -05:00
|
|
|
#endif
|
|
|
|
{
|
2006-10-11 06:35:21 -04:00
|
|
|
char buf[32], passwd[256];
|
2012-02-05 10:38:58 -05:00
|
|
|
int num, screen;
|
2006-10-12 03:33:38 -04:00
|
|
|
unsigned int len;
|
2006-10-11 11:04:04 -04:00
|
|
|
KeySym ksym;
|
2006-10-11 06:35:21 -04:00
|
|
|
XEvent ev;
|
|
|
|
|
2006-10-31 02:43:25 -05:00
|
|
|
len = 0;
|
2009-11-26 07:53:26 -05:00
|
|
|
running = True;
|
2006-10-11 07:33:04 -04:00
|
|
|
|
2009-11-26 07:53:26 -05:00
|
|
|
/* As "slock" stands for "Simple X display locker", the DPMS settings
|
|
|
|
* had been removed and you can set it with "xset" or some other
|
|
|
|
* utility. This way the user can easily set a customized DPMS
|
|
|
|
* timeout. */
|
2008-02-22 05:13:12 -05:00
|
|
|
while(running && !XNextEvent(dpy, &ev)) {
|
2006-10-11 06:35:21 -04:00
|
|
|
if(ev.type == KeyPress) {
|
|
|
|
buf[0] = 0;
|
2006-11-26 09:50:18 -05:00
|
|
|
num = XLookupString(&ev.xkey, buf, sizeof buf, &ksym, 0);
|
2008-07-29 14:14:53 -04:00
|
|
|
if(IsKeypadKey(ksym)) {
|
2008-04-08 04:55:46 -04:00
|
|
|
if(ksym == XK_KP_Enter)
|
|
|
|
ksym = XK_Return;
|
|
|
|
else if(ksym >= XK_KP_0 && ksym <= XK_KP_9)
|
|
|
|
ksym = (ksym - XK_KP_0) + XK_0;
|
2008-07-29 14:14:53 -04:00
|
|
|
}
|
2006-10-11 06:35:21 -04:00
|
|
|
if(IsFunctionKey(ksym) || IsKeypadKey(ksym)
|
|
|
|
|| IsMiscFunctionKey(ksym) || IsPFKey(ksym)
|
|
|
|
|| IsPrivateKeypadKey(ksym))
|
|
|
|
continue;
|
|
|
|
switch(ksym) {
|
|
|
|
case XK_Return:
|
2006-10-31 02:35:54 -05:00
|
|
|
passwd[len] = 0;
|
2008-02-22 05:13:12 -05:00
|
|
|
#ifdef HAVE_BSD_AUTH
|
|
|
|
running = !auth_userokay(getlogin(), NULL, "auth-xlock", passwd);
|
|
|
|
#else
|
|
|
|
running = strcmp(crypt(passwd, pws), pws);
|
|
|
|
#endif
|
2012-02-05 10:38:58 -05:00
|
|
|
if(running != False)
|
2006-10-11 07:33:04 -04:00
|
|
|
XBell(dpy, 100);
|
2006-10-31 02:35:54 -05:00
|
|
|
len = 0;
|
2006-10-11 06:35:21 -04:00
|
|
|
break;
|
|
|
|
case XK_Escape:
|
2006-10-31 02:35:54 -05:00
|
|
|
len = 0;
|
2006-10-11 06:35:21 -04:00
|
|
|
break;
|
|
|
|
case XK_BackSpace:
|
|
|
|
if(len)
|
2006-10-31 02:43:25 -05:00
|
|
|
--len;
|
2006-10-11 06:35:21 -04:00
|
|
|
break;
|
|
|
|
default:
|
2007-01-13 08:09:41 -05:00
|
|
|
if(num && !iscntrl((int) buf[0]) && (len + num < sizeof passwd)) {
|
2006-10-31 02:43:25 -05:00
|
|
|
memcpy(passwd + len, buf, num);
|
2006-10-31 02:35:54 -05:00
|
|
|
len += num;
|
2006-10-11 06:35:21 -04:00
|
|
|
}
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
}
|
2012-02-05 10:38:58 -05:00
|
|
|
else for(screen = 0; screen < nscreens; screen++)
|
|
|
|
XMapRaised(dpy, locks[screen]->win);
|
2008-02-22 05:13:12 -05:00
|
|
|
}
|
2009-11-26 07:53:26 -05:00
|
|
|
}
|
|
|
|
|
|
|
|
static void
|
2012-02-05 10:38:58 -05:00
|
|
|
unlockscreen(Display *dpy, Lock *lock) {
|
|
|
|
if(dpy == NULL || lock == NULL)
|
2009-11-26 07:53:26 -05:00
|
|
|
return;
|
|
|
|
|
2006-10-16 06:59:37 -04:00
|
|
|
XUngrabPointer(dpy, CurrentTime);
|
2009-11-26 07:53:26 -05:00
|
|
|
XFreePixmap(dpy, lock->pmap);
|
2012-02-05 10:38:58 -05:00
|
|
|
XDestroyWindow(dpy, lock->win);
|
2009-11-26 07:53:26 -05:00
|
|
|
|
|
|
|
free(lock);
|
|
|
|
}
|
|
|
|
|
2012-02-05 10:38:58 -05:00
|
|
|
static Lock *
|
2009-11-26 07:53:26 -05:00
|
|
|
lockscreen(Display *dpy, int screen) {
|
|
|
|
char curs[] = {0, 0, 0, 0, 0, 0, 0, 0};
|
|
|
|
unsigned int len;
|
2012-02-05 10:38:58 -05:00
|
|
|
Lock *lock;
|
2009-11-26 07:53:26 -05:00
|
|
|
XColor black, dummy;
|
|
|
|
XSetWindowAttributes wa;
|
|
|
|
Cursor invisible;
|
|
|
|
|
2012-02-05 10:38:58 -05:00
|
|
|
if(dpy == NULL || screen < 0)
|
2009-11-26 07:53:26 -05:00
|
|
|
return NULL;
|
|
|
|
|
2012-02-05 10:38:58 -05:00
|
|
|
lock = malloc(sizeof(Lock));
|
|
|
|
if(lock == NULL)
|
2009-11-26 07:53:26 -05:00
|
|
|
return NULL;
|
|
|
|
|
|
|
|
lock->screen = screen;
|
|
|
|
|
|
|
|
lock->root = RootWindow(dpy, lock->screen);
|
|
|
|
|
|
|
|
/* init */
|
|
|
|
wa.override_redirect = 1;
|
|
|
|
wa.background_pixel = BlackPixel(dpy, lock->screen);
|
2012-02-05 10:38:58 -05:00
|
|
|
lock->win = XCreateWindow(dpy, lock->root, 0, 0, DisplayWidth(dpy, lock->screen), DisplayHeight(dpy, lock->screen),
|
2009-11-26 07:53:26 -05:00
|
|
|
0, DefaultDepth(dpy, lock->screen), CopyFromParent,
|
|
|
|
DefaultVisual(dpy, lock->screen), CWOverrideRedirect | CWBackPixel, &wa);
|
|
|
|
XAllocNamedColor(dpy, DefaultColormap(dpy, lock->screen), "black", &black, &dummy);
|
2012-02-05 10:38:58 -05:00
|
|
|
lock->pmap = XCreateBitmapFromData(dpy, lock->win, curs, 8, 8);
|
2009-11-26 07:53:26 -05:00
|
|
|
invisible = XCreatePixmapCursor(dpy, lock->pmap, lock->pmap, &black, &black, 0, 0);
|
2012-02-05 10:38:58 -05:00
|
|
|
XDefineCursor(dpy, lock->win, invisible);
|
|
|
|
XMapRaised(dpy, lock->win);
|
2009-11-26 07:53:26 -05:00
|
|
|
for(len = 1000; len; len--) {
|
|
|
|
if(XGrabPointer(dpy, lock->root, False, ButtonPressMask | ButtonReleaseMask | PointerMotionMask,
|
|
|
|
GrabModeAsync, GrabModeAsync, None, invisible, CurrentTime) == GrabSuccess)
|
|
|
|
break;
|
|
|
|
usleep(1000);
|
|
|
|
}
|
2012-02-05 10:38:58 -05:00
|
|
|
if(running && (len > 0)) {
|
2009-11-26 07:53:26 -05:00
|
|
|
for(len = 1000; len; len--) {
|
|
|
|
if(XGrabKeyboard(dpy, lock->root, True, GrabModeAsync, GrabModeAsync, CurrentTime)
|
|
|
|
== GrabSuccess)
|
|
|
|
break;
|
|
|
|
usleep(1000);
|
|
|
|
}
|
|
|
|
running = (len > 0);
|
|
|
|
}
|
|
|
|
|
2012-02-05 10:38:58 -05:00
|
|
|
if(!running) {
|
2009-11-26 07:53:26 -05:00
|
|
|
unlockscreen(dpy, lock);
|
|
|
|
lock = NULL;
|
|
|
|
}
|
|
|
|
|
|
|
|
return lock;
|
|
|
|
}
|
|
|
|
|
|
|
|
static void
|
|
|
|
usage(void) {
|
2012-02-05 10:38:58 -05:00
|
|
|
fprintf(stderr, "usage: slock [-v]");
|
2009-11-26 07:53:26 -05:00
|
|
|
exit(EXIT_FAILURE);
|
|
|
|
}
|
|
|
|
|
|
|
|
int
|
|
|
|
main(int argc, char **argv) {
|
|
|
|
#ifndef HAVE_BSD_AUTH
|
|
|
|
const char *pws;
|
|
|
|
#endif
|
|
|
|
Display *dpy;
|
2012-02-05 10:38:58 -05:00
|
|
|
int screen;
|
2009-11-26 07:53:26 -05:00
|
|
|
|
|
|
|
if((argc == 2) && !strcmp("-v", argv[1]))
|
2012-01-22 12:58:10 -05:00
|
|
|
die("slock-%s, © 2006-2012 Anselm R Garbe", VERSION);
|
2009-11-26 07:53:26 -05:00
|
|
|
else if(argc != 1)
|
|
|
|
usage();
|
|
|
|
|
2011-04-21 04:22:47 -04:00
|
|
|
if(!getpwuid(getuid()))
|
2012-02-05 10:38:58 -05:00
|
|
|
die("slock: no passwd entry for you");
|
2011-04-21 04:22:47 -04:00
|
|
|
|
2009-11-26 07:53:26 -05:00
|
|
|
#ifndef HAVE_BSD_AUTH
|
2012-01-22 12:58:10 -05:00
|
|
|
pws = getpw();
|
2009-11-26 07:53:26 -05:00
|
|
|
#endif
|
|
|
|
|
|
|
|
if(!(dpy = XOpenDisplay(0)))
|
2012-02-05 10:38:58 -05:00
|
|
|
die("slock: cannot open display");
|
2009-11-26 07:53:26 -05:00
|
|
|
/* Get the number of screens in display "dpy" and blank them all. */
|
|
|
|
nscreens = ScreenCount(dpy);
|
2012-02-05 10:38:58 -05:00
|
|
|
locks = malloc(sizeof(Lock *) * nscreens);
|
|
|
|
if(locks == NULL)
|
|
|
|
die("slock: malloc: %s", strerror(errno));
|
|
|
|
for(screen = 0; screen < nscreens; screen++)
|
2009-11-26 07:53:26 -05:00
|
|
|
locks[screen] = lockscreen(dpy, screen);
|
|
|
|
XSync(dpy, False);
|
|
|
|
|
|
|
|
/* Everything is now blank. Now wait for the correct password. */
|
|
|
|
#ifdef HAVE_BSD_AUTH
|
2012-01-22 12:58:10 -05:00
|
|
|
readpw(dpy);
|
2009-11-26 07:53:26 -05:00
|
|
|
#else
|
2012-01-22 12:58:10 -05:00
|
|
|
readpw(dpy, pws);
|
2009-11-26 07:53:26 -05:00
|
|
|
#endif
|
|
|
|
|
|
|
|
/* Password ok, unlock everything and quit. */
|
2012-02-05 10:38:58 -05:00
|
|
|
for(screen = 0; screen < nscreens; screen++)
|
2009-11-26 07:53:26 -05:00
|
|
|
unlockscreen(dpy, locks[screen]);
|
|
|
|
|
|
|
|
free(locks);
|
2006-10-11 06:35:21 -04:00
|
|
|
XCloseDisplay(dpy);
|
2009-11-26 07:53:26 -05:00
|
|
|
|
2006-10-11 06:35:21 -04:00
|
|
|
return 0;
|
|
|
|
}
|