blob: 2170c78962b3dedea680632fc692f6467d2ada01 (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
|
#include <stdio.h>
#include <unistd.h>
#include <stdlib.h>
#include <fcntl.h>
#include <string.h>
#define POWER_STATE_F "/sys/power/state"
void
die(const char *msg)
{
puts(msg);
exit(1);
}
void
err(const char *msg)
{
perror(msg);
exit(1);
}
void
suspend()
{
int fh = open(POWER_STATE_F, O_WRONLY);
int n;
const char *cmd = "mem";
if (fh == -1)
err("Opening "POWER_STATE_F);
n = write(fh, cmd, 3);
if (n == -1)
err("Writing to "POWER_STATE_F);
exit(0);
}
int
main(int argc, char **argv)
{
uid_t uid = getuid();
gid_t gid = getgid();
uid_t euid = geteuid();
char **a = argv+1;
char *lockProg = "slock";
pid_t pid;
int ret;
if (euid != 0)
die("Program must be run as root/setuid");
for (;*a;a++) {
if (strcmp("-l", *a) == 0) {
a++;
if (*a)
lockProg = *a;
else
die("-l must have an argument");
} else {
printf("Unknown argument: '%s'\n", *a);
exit(1);
}
}
pid = fork();
switch (pid) {
case -1:
err("fork");
case 0:
ret = setuid(uid);
if (ret == -1)
err("setuid");
ret = setgid(gid);
if (ret == -1)
err("setgid");
ret = execlp(lockProg, lockProg);
err("execlp");
default:
suspend();
}
}
|
