Apply dpms patch

[slock.git] / slock.c

diff --git a/slock.c b/slock.c
index 6dbf966f69b0ec45aad6d0e4b930ac82de255b3c..2b06ca2a9c3b79c6ef7b9fe6fa2686edab91be7f 100644 (file)
--- a/slock.c
+++ b/slock.c
@@ -13,8 +13,10 @@
 #include <stdio.h>
 #include <string.h>
 #include <unistd.h>
+#include <time.h>
 #include <sys/types.h>
 #include <X11/extensions/Xrandr.h>
+#include <X11/extensions/dpms.h>
 #include <X11/keysym.h>
 #include <X11/Xlib.h>
 #include <X11/Xutil.h>
@@ -24,6 +26,8 @@
 
 char *argv0;
 
+static time_t locktime;
+
 enum {
        INIT,
        INPUT,
@@ -31,8 +35,6 @@ enum {
        NUMCOLS
 };
 
-#include "config.h"
-
 struct lock {
        int screen;
        Window root, win;
@@ -46,6 +48,8 @@ struct xrandr {
        int errbase;
 };
 
+#include "config.h"
+
 static void
 die(const char *errstr, ...)
 {
@@ -100,7 +104,7 @@ gethash(void)
        hash = pw->pw_passwd;
 
 #if HAVE_SHADOW_H
-       if (hash[0] == 'x' && hash[1] == '\0') {
+       if (!strcmp(hash, "x")) {
                struct spwd *sp;
                if (!(sp = getspnam(pw->pw_name)))
                        die("slock: getspnam: cannot retrieve shadow entry. "
@@ -108,7 +112,7 @@ gethash(void)
                hash = sp->sp_pwdp;
        }
 #else
-       if (hash[0] == '*' && hash[1] == '\0') {
+       if (!strcmp(hash, "*")) {
 #ifdef __OpenBSD__
                if (!(pw = getpwuid_shadow(getuid())))
                        die("slock: getpwnam_shadow: cannot retrieve shadow entry. "
@@ -141,6 +145,7 @@ readpw(Display *dpy, struct xrandr *rr, struct lock **locks, int nscreens,
        oldc = INIT;
 
        while (running && !XNextEvent(dpy, &ev)) {
+               running = !((time(NULL) - locktime < timetocancel) && (ev.type == MotionNotify));
                if (ev.type == KeyPress) {
                        explicit_bzero(&buf, sizeof(buf));
                        num = XLookupString(&ev.xkey, buf, sizeof(buf), &ksym, 0);
@@ -156,6 +161,22 @@ readpw(Display *dpy, struct xrandr *rr, struct lock **locks, int nscreens,
                            IsPFKey(ksym) ||
                            IsPrivateKeypadKey(ksym))
                                continue;
+                       if (ev.xkey.state & ControlMask) {
+                               switch (ksym) {
+                               case XK_u:
+                                       ksym = XK_Escape;
+                                       break;
+                               case XK_m:
+                                       ksym = XK_Return;
+                                       break;
+                               case XK_j:
+                                       ksym = XK_Return;
+                                       break;
+                               case XK_h:
+                                       ksym = XK_BackSpace;
+                                       break;
+                               }
+                       }
                        switch (ksym) {
                        case XK_Return:
                                passwd[len] = '\0';
@@ -166,7 +187,7 @@ readpw(Display *dpy, struct xrandr *rr, struct lock **locks, int nscreens,
                                        running = !!strcmp(inputhash, hash);
                                if (running) {
                                        XBell(dpy, 100);
-                                       failure = True;
+                                       failure = 1;
                                }
                                explicit_bzero(&passwd, sizeof(passwd));
                                len = 0;
@@ -177,7 +198,7 @@ readpw(Display *dpy, struct xrandr *rr, struct lock **locks, int nscreens,
                                break;
                        case XK_BackSpace:
                                if (len)
-                                       passwd[len--] = '\0';
+                                       passwd[--len] = '\0';
                                break;
                        default:
                                if (num && !iscntrl((int)buf[0]) &&
@@ -201,13 +222,21 @@ readpw(Display *dpy, struct xrandr *rr, struct lock **locks, int nscreens,
                        rre = (XRRScreenChangeNotifyEvent*)&ev;
                        for (screen = 0; screen < nscreens; screen++) {
                                if (locks[screen]->win == rre->window) {
-                                       XResizeWindow(dpy, locks[screen]->win,
-                                                     rre->width, rre->height);
+                                       if (rre->rotation == RR_Rotate_90 ||
+                                           rre->rotation == RR_Rotate_270)
+                                               XResizeWindow(dpy, locks[screen]->win,
+                                                             rre->height, rre->width);
+                                       else
+                                               XResizeWindow(dpy, locks[screen]->win,
+                                                             rre->width, rre->height);
                                        XClearWindow(dpy, locks[screen]->win);
+                                       break;
                                }
                        }
-               } else for (screen = 0; screen < nscreens; screen++)
-                       XRaiseWindow(dpy, locks[screen]->win);
+               } else {
+                       for (screen = 0; screen < nscreens; screen++)
+                               XRaiseWindow(dpy, locks[screen]->win);
+               }
        }
 }
 
@@ -268,6 +297,7 @@ lockscreen(Display *dpy, struct xrandr *rr, int screen)
                                XRRSelectInput(dpy, lock->win, RRScreenChangeNotifyMask);
 
                        XSelectInput(dpy, lock->root, SubstructureNotifyMask);
+                       locktime = time(NULL);
                        return lock;
                }
 
@@ -306,6 +336,7 @@ main(int argc, char **argv) {
        const char *hash;
        Display *dpy;
        int s, nlocks, nscreens;
+       CARD16 standby, suspend, off;
 
        ARGBEGIN {
        case 'v':
@@ -366,6 +397,20 @@ main(int argc, char **argv) {
        if (nlocks != nscreens)
                return 1;
 
+       /* DPMS magic to disable the monitor */
+       if (!DPMSCapable(dpy))
+               die("slock: DPMSCapable failed\n");
+       if (!DPMSEnable(dpy))
+               die("slock: DPMSEnable failed\n");
+       if (!DPMSGetTimeouts(dpy, &standby, &suspend, &off))
+               die("slock: DPMSGetTimeouts failed\n");
+       if (!standby || !suspend || !off)
+               die("slock: at least one DPMS variable is zero\n");
+       if (!DPMSSetTimeouts(dpy, monitortime, monitortime, monitortime))
+               die("slock: DPMSSetTimeouts failed\n");
+
+       XSync(dpy, 0);
+
        /* run post-lock command */
        if (argc > 0) {
                switch (fork()) {
@@ -383,5 +428,9 @@ main(int argc, char **argv) {
        /* everything is now blank. Wait for the correct password */
        readpw(dpy, &rr, locks, nscreens, hash);
 
+       /* reset DPMS values to inital ones */
+       DPMSSetTimeouts(dpy, standby, suspend, off);
+       XSync(dpy, 0);
+
        return 0;
 }