2008-04-08 11:55:46 +03:00
|
|
|
/* See LICENSE file for license details. */
|
2006-10-12 10:33:38 +03:00
|
|
|
#define _XOPEN_SOURCE 500
|
2006-10-31 09:43:25 +02:00
|
|
|
#if HAVE_SHADOW_H
|
|
|
|
#include <shadow.h>
|
|
|
|
#endif
|
2006-10-12 09:11:08 +03:00
|
|
|
|
2006-10-12 10:33:38 +03:00
|
|
|
#include <ctype.h>
|
2009-11-26 14:53:26 +02:00
|
|
|
#include <errno.h>
|
2006-10-31 09:35:54 +02:00
|
|
|
#include <pwd.h>
|
2007-11-24 22:17:32 +02:00
|
|
|
#include <stdarg.h>
|
2006-10-11 13:35:21 +03: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 12:13:12 +02:00
|
|
|
|
|
|
|
#if HAVE_BSD_AUTH
|
|
|
|
#include <login_cap.h>
|
|
|
|
#include <bsd_auth.h>
|
|
|
|
#endif
|
2006-10-11 13:35:21 +03:00
|
|
|
|
2009-11-26 14:53:26 +02:00
|
|
|
struct st_lock {
|
|
|
|
int screen;
|
|
|
|
Window root, w;
|
|
|
|
Pixmap pmap;
|
|
|
|
};
|
|
|
|
|
|
|
|
extern const char *__progname;
|
|
|
|
|
2008-07-29 21:14:53 +03:00
|
|
|
static void
|
2008-07-29 21:08:18 +03:00
|
|
|
die(const char *errstr, ...) {
|
2007-11-24 22:17:32 +02:00
|
|
|
va_list ap;
|
|
|
|
|
2009-11-26 14:53:26 +02:00
|
|
|
fprintf(stderr, "%s: ", __progname);
|
2007-11-24 22:17:32 +02:00
|
|
|
va_start(ap, errstr);
|
|
|
|
vfprintf(stderr, errstr, ap);
|
|
|
|
va_end(ap);
|
2009-11-26 14:53:26 +02:00
|
|
|
fprintf(stderr, "\n");
|
|
|
|
fflush(stderr);
|
|
|
|
|
2007-11-24 22:17:32 +02:00
|
|
|
exit(EXIT_FAILURE);
|
|
|
|
}
|
|
|
|
|
2008-02-22 12:13:12 +02:00
|
|
|
#ifndef HAVE_BSD_AUTH
|
2008-07-29 21:14:53 +03:00
|
|
|
static const char *
|
2012-01-22 19:58:10 +02:00
|
|
|
getpw(void) { /* only run as root */
|
2006-10-31 09:35:54 +02:00
|
|
|
const char *rval;
|
|
|
|
struct passwd *pw;
|
|
|
|
|
|
|
|
pw = getpwuid(getuid());
|
2012-01-24 23:10:02 +02:00
|
|
|
if(!pw)
|
|
|
|
die("cannot retrieve password entry (make sure to suid or sgid slock)");
|
2006-10-31 09:35:54 +02:00
|
|
|
endpwent();
|
|
|
|
rval = pw->pw_passwd;
|
|
|
|
|
|
|
|
#if HAVE_SHADOW_H
|
|
|
|
{
|
|
|
|
struct spwd *sp;
|
|
|
|
sp = getspnam(getenv("USER"));
|
2012-01-24 23:10:02 +02:00
|
|
|
if(!sp)
|
|
|
|
die("slock: cannot retrieve shadow entry (make sure to suid or sgid slock)\n");
|
2006-10-31 09:35:54 +02:00
|
|
|
endspent();
|
|
|
|
rval = sp->sp_pwdp;
|
|
|
|
}
|
|
|
|
#endif
|
2008-02-22 12:13:12 +02:00
|
|
|
|
2006-10-31 09:35:54 +02:00
|
|
|
/* drop privileges */
|
2007-11-24 22:17:32 +02:00
|
|
|
if(setgid(pw->pw_gid) < 0 || setuid(pw->pw_uid) < 0)
|
2009-11-26 14:53:26 +02:00
|
|
|
die("cannot drop privileges");
|
2006-10-31 09:35:54 +02:00
|
|
|
return rval;
|
|
|
|
}
|
2008-02-22 12:13:12 +02:00
|
|
|
#endif
|
2006-10-31 09:35:54 +02:00
|
|
|
|
2009-11-26 14:53:26 +02:00
|
|
|
static void
|
|
|
|
#ifdef HAVE_BSD_AUTH
|
2012-01-22 19:58:10 +02:00
|
|
|
readpw(Display *dpy)
|
2009-11-26 14:53:26 +02:00
|
|
|
#else
|
2012-01-22 19:58:10 +02:00
|
|
|
readpw(Display *dpy, const char *pws)
|
2009-11-26 14:53:26 +02:00
|
|
|
#endif
|
|
|
|
{
|
2006-10-11 13:35:21 +03:00
|
|
|
char buf[32], passwd[256];
|
2009-11-26 14:53:26 +02:00
|
|
|
int num;
|
2008-02-22 12:13:12 +02:00
|
|
|
|
2006-10-12 10:33:38 +03:00
|
|
|
unsigned int len;
|
2006-10-11 13:35:21 +03:00
|
|
|
Bool running = True;
|
2006-10-11 18:04:04 +03:00
|
|
|
KeySym ksym;
|
2006-10-11 13:35:21 +03:00
|
|
|
XEvent ev;
|
|
|
|
|
2006-10-31 09:43:25 +02:00
|
|
|
len = 0;
|
2009-11-26 14:53:26 +02:00
|
|
|
running = True;
|
2006-10-11 14:33:04 +03:00
|
|
|
|
2009-11-26 14:53:26 +02: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-12-12 21:34:43 +02:00
|
|
|
|
2008-02-22 12:13:12 +02:00
|
|
|
while(running && !XNextEvent(dpy, &ev)) {
|
2006-10-11 13:35:21 +03:00
|
|
|
if(ev.type == KeyPress) {
|
|
|
|
buf[0] = 0;
|
2006-11-26 16:50:18 +02:00
|
|
|
num = XLookupString(&ev.xkey, buf, sizeof buf, &ksym, 0);
|
2008-07-29 21:14:53 +03:00
|
|
|
if(IsKeypadKey(ksym)) {
|
2008-04-08 11:55:46 +03: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 21:14:53 +03:00
|
|
|
}
|
2006-10-11 13:35:21 +03:00
|
|
|
if(IsFunctionKey(ksym) || IsKeypadKey(ksym)
|
|
|
|
|| IsMiscFunctionKey(ksym) || IsPFKey(ksym)
|
|
|
|
|| IsPrivateKeypadKey(ksym))
|
|
|
|
continue;
|
|
|
|
switch(ksym) {
|
|
|
|
case XK_Return:
|
2006-10-31 09:35:54 +02:00
|
|
|
passwd[len] = 0;
|
2008-02-22 12:13:12 +02:00
|
|
|
#ifdef HAVE_BSD_AUTH
|
|
|
|
running = !auth_userokay(getlogin(), NULL, "auth-xlock", passwd);
|
|
|
|
#else
|
|
|
|
running = strcmp(crypt(passwd, pws), pws);
|
|
|
|
#endif
|
|
|
|
if (running != 0)
|
2006-10-11 14:33:04 +03:00
|
|
|
XBell(dpy, 100);
|
2006-10-31 09:35:54 +02:00
|
|
|
len = 0;
|
2006-10-11 13:35:21 +03:00
|
|
|
break;
|
|
|
|
case XK_Escape:
|
2006-10-31 09:35:54 +02:00
|
|
|
len = 0;
|
2006-10-11 13:35:21 +03:00
|
|
|
break;
|
|
|
|
case XK_BackSpace:
|
|
|
|
if(len)
|
2006-10-31 09:43:25 +02:00
|
|
|
--len;
|
2006-10-11 13:35:21 +03:00
|
|
|
break;
|
|
|
|
default:
|
2007-01-13 15:09:41 +02:00
|
|
|
if(num && !iscntrl((int) buf[0]) && (len + num < sizeof passwd)) {
|
2006-10-31 09:43:25 +02:00
|
|
|
memcpy(passwd + len, buf, num);
|
2006-10-31 09:35:54 +02:00
|
|
|
len += num;
|
2006-10-11 13:35:21 +03:00
|
|
|
}
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
}
|
2008-02-22 12:13:12 +02:00
|
|
|
}
|
2009-11-26 14:53:26 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
static void
|
|
|
|
unlockscreen(Display *dpy, struct st_lock *lock) {
|
|
|
|
if (dpy == NULL || lock == NULL)
|
|
|
|
return;
|
|
|
|
|
2006-10-16 13:59:37 +03:00
|
|
|
XUngrabPointer(dpy, CurrentTime);
|
2009-11-26 14:53:26 +02:00
|
|
|
XFreePixmap(dpy, lock->pmap);
|
|
|
|
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);
|
|
|
|
}
|
|
|
|
|
2012-01-22 19:58:10 +02:00
|
|
|
static int
|
|
|
|
xerrordummy(Display *dpy, XErrorEvent *ee) {
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
2009-11-26 14:53:26 +02:00
|
|
|
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]))
|
2012-01-22 19:58:10 +02:00
|
|
|
die("slock-%s, © 2006-2012 Anselm R Garbe", VERSION);
|
2009-11-26 14:53:26 +02:00
|
|
|
else if(argc != 1)
|
|
|
|
usage();
|
|
|
|
|
2011-04-21 11:22:47 +03:00
|
|
|
if(!getpwuid(getuid()))
|
|
|
|
die("no passwd entry for you");
|
|
|
|
|
2009-11-26 14:53:26 +02:00
|
|
|
#ifndef HAVE_BSD_AUTH
|
2012-01-22 19:58:10 +02:00
|
|
|
pws = getpw();
|
2009-11-26 14:53:26 +02:00
|
|
|
#endif
|
|
|
|
|
|
|
|
if(!(dpy = XOpenDisplay(0)))
|
|
|
|
die("cannot open display");
|
2012-01-22 19:58:10 +02:00
|
|
|
/* prevent default error handler to take over */
|
|
|
|
XSetErrorHandler(xerrordummy);
|
2009-11-26 14:53:26 +02:00
|
|
|
/* 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
|
2012-01-22 19:58:10 +02:00
|
|
|
readpw(dpy);
|
2009-11-26 14:53:26 +02:00
|
|
|
#else
|
2012-01-22 19:58:10 +02:00
|
|
|
readpw(dpy, pws);
|
2009-11-26 14:53:26 +02:00
|
|
|
#endif
|
|
|
|
|
|
|
|
/* Password ok, unlock everything and quit. */
|
|
|
|
for (screen = 0; screen < nscreens; screen++)
|
|
|
|
unlockscreen(dpy, locks[screen]);
|
|
|
|
|
|
|
|
free(locks);
|
|
|
|
|
2006-10-11 13:35:21 +03:00
|
|
|
XCloseDisplay(dpy);
|
2009-11-26 14:53:26 +02:00
|
|
|
|
2006-10-11 13:35:21 +03:00
|
|
|
return 0;
|
|
|
|
}
|