Skip to content

Commit

Permalink
applied BSD support patch
Browse files Browse the repository at this point in the history
  • Loading branch information
Anselm R. Garbe committed Oct 12, 2006
1 parent 2fa1221 commit 0f1157d
Show file tree
Hide file tree
Showing 2 changed files with 25 additions and 2 deletions.
5 changes: 4 additions & 1 deletion config.mk
Original file line number Diff line number Diff line change
Expand Up @@ -15,11 +15,14 @@ INCS = -I. -I/usr/include -I${X11INC}
LIBS = -L/usr/lib -lc -lcrypt -L${X11LIB} -lX11

# flags
CFLAGS = -Os ${INCS} -DVERSION=\"${VERSION}\"
CFLAGS = -Os ${INCS} -DVERSION=\"${VERSION}\" -DHAVE_SHADOW_H
LDFLAGS = ${LIBS}
#CFLAGS = -g -Wall -O2 ${INCS} -DVERSION=\"${VERSION}\"
#LDFLAGS = -g ${LIBS}

# On *BSD remove -DHAVE_SHADOW_H from CFLAGS
# On OpenBSD remove -lcrypt from LIBS

# compiler and linker
CC = cc
LD = ${CC}
22 changes: 21 additions & 1 deletion slock.c
Original file line number Diff line number Diff line change
Expand Up @@ -2,7 +2,13 @@
* See LICENSE file for license details.
*/
#define _XOPEN_SOURCE

#if HAVE_SHADOW_H
#include <shadow.h>
#else
#include <pwd.h>
#endif

#include <stdlib.h>
#include <stdio.h>
#include <string.h>
Expand All @@ -17,7 +23,11 @@ main(int argc, char **argv) {
char curs[] = {0, 0, 0, 0, 0, 0, 0, 0};
char buf[32], passwd[256];
int num, prev_nitem, screen;
#if HAVE_SHADOW_H
struct spwd *sp;
#else
struct passwd *pw;
#endif
unsigned int i, len;
Bool running = True;
Cursor invisible;
Expand All @@ -33,11 +43,17 @@ main(int argc, char **argv) {
fputs("slock-"VERSION", (C)opyright MMVI Anselm R. Garbe\n", stdout);
exit(EXIT_SUCCESS);
}
if(!(sp = getspnam(getenv("USER")))) {
if(geteuid() != 0) {
fputs("slock: cannot retrieve password entry (make sure to suid slock)\n", stderr);
exit(EXIT_FAILURE);
}
#if HAVE_SHADOW_H
sp = getspnam(getenv("USER"));
endspent();
#else
pw = getpwuid(getuid());
endpwent();
#endif
if(!(dpy = XOpenDisplay(0))) {
fputs("slock: cannot open display\n", stderr);
exit(EXIT_FAILURE);
Expand Down Expand Up @@ -87,7 +103,11 @@ main(int argc, char **argv) {
}
switch(ksym) {
case XK_Return:
#if HAVE_SHADOW_H
if((running = strncmp(crypt(passwd, sp->sp_pwdp), sp->sp_pwdp, sizeof(passwd))))
#else
if((running = strncmp(crypt(passwd, pw->pw_passwd), pw->pw_passwd, sizeof(passwd))))
#endif
XBell(dpy, 100);
passwd[0] = 0;
break;
Expand Down

0 comments on commit 0f1157d

Please sign in to comment.