screenlock - Sort functions alphabetically; add declarations; and fix warnings
This commit is contained in:
Miles Alan
parent
b51644baca
commit
a35ca3f39f
1 changed files with 188 additions and 174 deletions
|
|
@ -3,10 +3,13 @@
|
||||||
#include <stdio.h>
|
#include <stdio.h>
|
||||||
#include <stdlib.h>
|
#include <stdlib.h>
|
||||||
#include <string.h>
|
#include <string.h>
|
||||||
|
#include <unistd.h>
|
||||||
#include <X11/keysym.h>
|
#include <X11/keysym.h>
|
||||||
#include <X11/XF86keysym.h>
|
#include <X11/XF86keysym.h>
|
||||||
#include <X11/Xlib.h>
|
#include <X11/Xlib.h>
|
||||||
|
#include <X11/Xutil.h>
|
||||||
|
|
||||||
|
// Types
|
||||||
enum State {
|
enum State {
|
||||||
StateNoInput, // Screen on / input lock
|
StateNoInput, // Screen on / input lock
|
||||||
StateNoInputNoScreen, // Screen off / input lock
|
StateNoInputNoScreen, // Screen off / input lock
|
||||||
|
|
@ -14,7 +17,6 @@ enum State {
|
||||||
StateSuspendPending, // Suspend 'woken up', must leave state in <10s, or kicks to StateSuspend
|
StateSuspendPending, // Suspend 'woken up', must leave state in <10s, or kicks to StateSuspend
|
||||||
StateDead // Exit the appliation
|
StateDead // Exit the appliation
|
||||||
};
|
};
|
||||||
|
|
||||||
enum Color {
|
enum Color {
|
||||||
Red,
|
Red,
|
||||||
Blue,
|
Blue,
|
||||||
|
|
@ -22,179 +24,24 @@ enum Color {
|
||||||
Off
|
Off
|
||||||
};
|
};
|
||||||
|
|
||||||
static Display *dpy;
|
// Fn declarations
|
||||||
static enum State state = StateNoInput;
|
void configuresuspendsettingsandwakeupsources();
|
||||||
static int lastkeysym = NULL;
|
void die(const char *err, ...);
|
||||||
static int lastkeyn = 0;
|
int getoldbrightness();
|
||||||
static char oldbrightness[10] = "200";
|
void lockscreen(Display *dpy, int screen);
|
||||||
static char * brightnessfile = "/sys/devices/platform/backlight/backlight/backlight/brightness";
|
void readinputloop(Display *dpy, int screen);
|
||||||
static char * powerstatefile = "/sys/power/state";
|
void setpineled(enum Color c);
|
||||||
|
void syncstate();
|
||||||
|
void writefile(char *filepath, char *str);
|
||||||
|
|
||||||
void
|
// Variables
|
||||||
writefile(char *filepath, char *str)
|
Display *dpy;
|
||||||
{
|
enum State state = StateNoInput;
|
||||||
int f;
|
KeySym lastkeysym = XK_Cancel;
|
||||||
f = open(filepath, O_WRONLY);
|
int lastkeyn = 0;
|
||||||
if (f != NULL) {
|
char oldbrightness[10] = "200";
|
||||||
write(f, str, strlen(str));
|
char * brightnessfile = "/sys/devices/platform/backlight/backlight/backlight/brightness";
|
||||||
close(f);
|
char * powerstatefile = "/sys/power/state";
|
||||||
} else {
|
|
||||||
fprintf(stderr, "Couldn't open filepath <%s>\n", filepath);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
void
|
|
||||||
setpineled(enum Color c)
|
|
||||||
{
|
|
||||||
if (c == Red) {
|
|
||||||
writefile("/sys/class/leds/red:indicator/brightness", "1");
|
|
||||||
writefile("/sys/class/leds/blue:indicator/brightness", "0");
|
|
||||||
} else if (c == Blue) {
|
|
||||||
writefile("/sys/class/leds/red:indicator/brightness", "0");
|
|
||||||
writefile("/sys/class/leds/blue:indicator/brightness", "1");
|
|
||||||
} else if (c == Purple) {
|
|
||||||
writefile("/sys/class/leds/red:indicator/brightness", "1");
|
|
||||||
writefile("/sys/class/leds/blue:indicator/brightness", "1");
|
|
||||||
} else if (c == Off) {
|
|
||||||
writefile("/sys/class/leds/red:indicator/brightness", "0");
|
|
||||||
writefile("/sys/class/leds/blue:indicator/brightness", "0");
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
void
|
|
||||||
syncstate()
|
|
||||||
{
|
|
||||||
if (state == StateSuspend) {
|
|
||||||
setpineled(Red);
|
|
||||||
configuresuspendsettingsandwakeupsources();
|
|
||||||
writefile(powerstatefile, "mem");
|
|
||||||
state = StateSuspendPending;
|
|
||||||
syncstate();
|
|
||||||
} else if (state == StateNoInput) {
|
|
||||||
setpineled(Blue);
|
|
||||||
writefile(brightnessfile, oldbrightness);
|
|
||||||
} else if (state == StateNoInputNoScreen || state == StateSuspendPending) {
|
|
||||||
setpineled(Purple);
|
|
||||||
writefile(brightnessfile, "0");
|
|
||||||
} else if (state == StateDead) {
|
|
||||||
writefile(brightnessfile, oldbrightness);
|
|
||||||
setpineled(Off);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
static void
|
|
||||||
die(const char *err, ...)
|
|
||||||
{
|
|
||||||
fprintf(stderr, "Error: %s", err);
|
|
||||||
state = StateDead;
|
|
||||||
syncstate();
|
|
||||||
exit(1);
|
|
||||||
}
|
|
||||||
|
|
||||||
// Loosely derived from suckless' slock's lockscreen binding logic but
|
|
||||||
// alot more coarse, intentionally so can be triggered while grab_key
|
|
||||||
// for dwm multikey path already holding..
|
|
||||||
void
|
|
||||||
lockscreen(Display *dpy, int screen)
|
|
||||||
{
|
|
||||||
int i, ptgrab, kbgrab;
|
|
||||||
Window root;
|
|
||||||
root = RootWindow(dpy, screen);
|
|
||||||
for (i = 0, ptgrab = kbgrab = -1; i < 9999999; i++) {
|
|
||||||
if (ptgrab != GrabSuccess) {
|
|
||||||
ptgrab = XGrabPointer(dpy, root, False,
|
|
||||||
ButtonPressMask | ButtonReleaseMask |
|
|
||||||
PointerMotionMask, GrabModeAsync,
|
|
||||||
GrabModeAsync, None, None, CurrentTime);
|
|
||||||
}
|
|
||||||
if (kbgrab != GrabSuccess) {
|
|
||||||
kbgrab = XGrabKeyboard(dpy, root, True,
|
|
||||||
GrabModeAsync, GrabModeAsync, CurrentTime);
|
|
||||||
}
|
|
||||||
if (ptgrab == GrabSuccess && kbgrab == GrabSuccess) {
|
|
||||||
XSelectInput(dpy, root, SubstructureNotifyMask);
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
usleep(100000);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
void
|
|
||||||
readinputloop(Display *dpy, int screen) {
|
|
||||||
KeySym keysym;
|
|
||||||
XEvent ev;
|
|
||||||
char buf[32];
|
|
||||||
fd_set fdset;
|
|
||||||
int xfd;
|
|
||||||
int selectresult;
|
|
||||||
struct timeval xeventtimeout = {10, 0};
|
|
||||||
xfd = ConnectionNumber(dpy);
|
|
||||||
|
|
||||||
for (;;) {
|
|
||||||
FD_ZERO(&fdset);
|
|
||||||
FD_SET(xfd, &fdset);
|
|
||||||
if (state == StateSuspendPending)
|
|
||||||
selectresult = select(FD_SETSIZE, &fdset, NULL, NULL, &xeventtimeout);
|
|
||||||
else
|
|
||||||
selectresult = select(FD_SETSIZE, &fdset, NULL, NULL, NULL);
|
|
||||||
|
|
||||||
if (FD_ISSET(xfd, &fdset) && XPending(dpy)) {
|
|
||||||
XNextEvent(dpy, &ev);
|
|
||||||
if (ev.type == KeyPress) {
|
|
||||||
XLookupString(&ev.xkey, buf, sizeof(buf), &keysym, 0);
|
|
||||||
if (lastkeysym == keysym) {
|
|
||||||
lastkeyn++;
|
|
||||||
} else {
|
|
||||||
lastkeysym = keysym;
|
|
||||||
lastkeyn = 1;
|
|
||||||
}
|
|
||||||
|
|
||||||
if (lastkeyn < 3)
|
|
||||||
continue;
|
|
||||||
|
|
||||||
lastkeyn = 0;
|
|
||||||
lastkeysym = NULL;
|
|
||||||
switch (keysym) {
|
|
||||||
case XF86XK_AudioRaiseVolume:
|
|
||||||
state = StateSuspend;
|
|
||||||
break;
|
|
||||||
case XF86XK_AudioLowerVolume:
|
|
||||||
state = (state == StateNoInput ? StateNoInputNoScreen : StateNoInput);
|
|
||||||
break;
|
|
||||||
case XF86XK_PowerOff:
|
|
||||||
state = StateDead;
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
syncstate();
|
|
||||||
}
|
|
||||||
} else if (state == StateSuspendPending) {
|
|
||||||
state = StateSuspend;
|
|
||||||
syncstate();
|
|
||||||
}
|
|
||||||
|
|
||||||
if (state == StateDead) break;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
int
|
|
||||||
getoldbrightness() {
|
|
||||||
char * buffer = 0;
|
|
||||||
long length;
|
|
||||||
FILE * f = fopen(brightnessfile, "r");
|
|
||||||
if (f) {
|
|
||||||
fseek(f, 0, SEEK_END);
|
|
||||||
length = ftell(f);
|
|
||||||
fseek(f, 0, SEEK_SET);
|
|
||||||
buffer = malloc(length);
|
|
||||||
if (buffer) {
|
|
||||||
fread(buffer, 1, length, f);
|
|
||||||
}
|
|
||||||
fclose(f);
|
|
||||||
}
|
|
||||||
if (buffer) {
|
|
||||||
sprintf(oldbrightness, "%d", atoi(buffer));
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
void
|
void
|
||||||
configuresuspendsettingsandwakeupsources()
|
configuresuspendsettingsandwakeupsources()
|
||||||
|
|
@ -250,9 +97,176 @@ configuresuspendsettingsandwakeupsources()
|
||||||
writefile("/sys/power/mem_sleep", "deep");
|
writefile("/sys/power/mem_sleep", "deep");
|
||||||
}
|
}
|
||||||
|
|
||||||
|
void
|
||||||
|
die(const char *err, ...)
|
||||||
|
{
|
||||||
|
fprintf(stderr, "Error: %s", err);
|
||||||
|
state = StateDead;
|
||||||
|
syncstate();
|
||||||
|
exit(1);
|
||||||
|
}
|
||||||
|
|
||||||
|
int
|
||||||
|
getoldbrightness() {
|
||||||
|
char * buffer = 0;
|
||||||
|
long length;
|
||||||
|
FILE * f = fopen(brightnessfile, "r");
|
||||||
|
if (f) {
|
||||||
|
fseek(f, 0, SEEK_END);
|
||||||
|
length = ftell(f);
|
||||||
|
fseek(f, 0, SEEK_SET);
|
||||||
|
buffer = malloc(length);
|
||||||
|
if (buffer) {
|
||||||
|
fread(buffer, 1, length, f);
|
||||||
|
}
|
||||||
|
fclose(f);
|
||||||
|
}
|
||||||
|
if (buffer) {
|
||||||
|
sprintf(oldbrightness, "%d", atoi(buffer));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
void
|
||||||
|
lockscreen(Display *dpy, int screen)
|
||||||
|
{
|
||||||
|
// Loosely derived from suckless' slock's lockscreen binding logic but
|
||||||
|
// alot more coarse, intentionally so can be triggered while grab_key
|
||||||
|
// for dwm multikey path already holding..
|
||||||
|
int i, ptgrab, kbgrab;
|
||||||
|
Window root;
|
||||||
|
root = RootWindow(dpy, screen);
|
||||||
|
for (i = 0, ptgrab = kbgrab = -1; i < 9999999; i++) {
|
||||||
|
if (ptgrab != GrabSuccess) {
|
||||||
|
ptgrab = XGrabPointer(dpy, root, False,
|
||||||
|
ButtonPressMask | ButtonReleaseMask |
|
||||||
|
PointerMotionMask, GrabModeAsync,
|
||||||
|
GrabModeAsync, None, None, CurrentTime);
|
||||||
|
}
|
||||||
|
if (kbgrab != GrabSuccess) {
|
||||||
|
kbgrab = XGrabKeyboard(dpy, root, True,
|
||||||
|
GrabModeAsync, GrabModeAsync, CurrentTime);
|
||||||
|
}
|
||||||
|
if (ptgrab == GrabSuccess && kbgrab == GrabSuccess) {
|
||||||
|
XSelectInput(dpy, root, SubstructureNotifyMask);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
usleep(100000);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
void
|
||||||
|
readinputloop(Display *dpy, int screen) {
|
||||||
|
KeySym keysym;
|
||||||
|
XEvent ev;
|
||||||
|
char buf[32];
|
||||||
|
fd_set fdset;
|
||||||
|
int xfd;
|
||||||
|
int selectresult;
|
||||||
|
struct timeval xeventtimeout = {10, 0};
|
||||||
|
xfd = ConnectionNumber(dpy);
|
||||||
|
|
||||||
|
for (;;) {
|
||||||
|
FD_ZERO(&fdset);
|
||||||
|
FD_SET(xfd, &fdset);
|
||||||
|
if (state == StateSuspendPending)
|
||||||
|
selectresult = select(FD_SETSIZE, &fdset, NULL, NULL, &xeventtimeout);
|
||||||
|
else
|
||||||
|
selectresult = select(FD_SETSIZE, &fdset, NULL, NULL, NULL);
|
||||||
|
|
||||||
|
if (FD_ISSET(xfd, &fdset) && XPending(dpy)) {
|
||||||
|
XNextEvent(dpy, &ev);
|
||||||
|
if (ev.type == KeyPress) {
|
||||||
|
XLookupString(&ev.xkey, buf, sizeof(buf), &keysym, 0);
|
||||||
|
if (lastkeysym == keysym) {
|
||||||
|
lastkeyn++;
|
||||||
|
} else {
|
||||||
|
lastkeysym = keysym;
|
||||||
|
lastkeyn = 1;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (lastkeyn < 3)
|
||||||
|
continue;
|
||||||
|
|
||||||
|
lastkeyn = 0;
|
||||||
|
lastkeysym = XK_Cancel;
|
||||||
|
switch (keysym) {
|
||||||
|
case XF86XK_AudioRaiseVolume:
|
||||||
|
state = StateSuspend;
|
||||||
|
break;
|
||||||
|
case XF86XK_AudioLowerVolume:
|
||||||
|
state = (state == StateNoInput ? StateNoInputNoScreen : StateNoInput);
|
||||||
|
break;
|
||||||
|
case XF86XK_PowerOff:
|
||||||
|
state = StateDead;
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
syncstate();
|
||||||
|
}
|
||||||
|
} else if (state == StateSuspendPending) {
|
||||||
|
state = StateSuspend;
|
||||||
|
syncstate();
|
||||||
|
}
|
||||||
|
|
||||||
|
if (state == StateDead) break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
void
|
||||||
|
setpineled(enum Color c)
|
||||||
|
{
|
||||||
|
if (c == Red) {
|
||||||
|
writefile("/sys/class/leds/red:indicator/brightness", "1");
|
||||||
|
writefile("/sys/class/leds/blue:indicator/brightness", "0");
|
||||||
|
} else if (c == Blue) {
|
||||||
|
writefile("/sys/class/leds/red:indicator/brightness", "0");
|
||||||
|
writefile("/sys/class/leds/blue:indicator/brightness", "1");
|
||||||
|
} else if (c == Purple) {
|
||||||
|
writefile("/sys/class/leds/red:indicator/brightness", "1");
|
||||||
|
writefile("/sys/class/leds/blue:indicator/brightness", "1");
|
||||||
|
} else if (c == Off) {
|
||||||
|
writefile("/sys/class/leds/red:indicator/brightness", "0");
|
||||||
|
writefile("/sys/class/leds/blue:indicator/brightness", "0");
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
void
|
||||||
|
syncstate()
|
||||||
|
{
|
||||||
|
if (state == StateSuspend) {
|
||||||
|
setpineled(Red);
|
||||||
|
configuresuspendsettingsandwakeupsources();
|
||||||
|
writefile(powerstatefile, "mem");
|
||||||
|
state = StateSuspendPending;
|
||||||
|
syncstate();
|
||||||
|
} else if (state == StateNoInput) {
|
||||||
|
setpineled(Blue);
|
||||||
|
writefile(brightnessfile, oldbrightness);
|
||||||
|
} else if (state == StateNoInputNoScreen || state == StateSuspendPending) {
|
||||||
|
setpineled(Purple);
|
||||||
|
writefile(brightnessfile, "0");
|
||||||
|
} else if (state == StateDead) {
|
||||||
|
writefile(brightnessfile, oldbrightness);
|
||||||
|
setpineled(Off);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
void
|
||||||
|
writefile(char *filepath, char *str)
|
||||||
|
{
|
||||||
|
int f;
|
||||||
|
f = open(filepath, O_WRONLY);
|
||||||
|
if (f != -1) {
|
||||||
|
write(f, str, strlen(str));
|
||||||
|
close(f);
|
||||||
|
} else {
|
||||||
|
fprintf(stderr, "Couldn't open filepath <%s>\n", filepath);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
int
|
int
|
||||||
main(int argc, char **argv) {
|
main(int argc, char **argv) {
|
||||||
Screen *screen;
|
int screen;
|
||||||
|
|
||||||
if (setuid(0))
|
if (setuid(0))
|
||||||
die("setuid(0) failed\n");
|
die("setuid(0) failed\n");
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue