From d3fc6f8f7788215b7c495f9442b4782982a8daf3 Mon Sep 17 00:00:00 2001
From: Drew DeVault <sir@cmpwn.com>
Date: Thu, 17 Dec 2015 08:44:30 -0500
Subject: [PATCH] Implement PAM password verification in swaylock

---
 CMakeLists.txt |  2 ++
 main.c         | 34 ++++++++++++++++++++++++++++++++++
 2 files changed, 36 insertions(+)

diff --git a/CMakeLists.txt b/CMakeLists.txt
index 977cc2f..306be80 100644
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -4,6 +4,7 @@ include_directories(
 	${CAIRO_INCLUDE_DIRS}
 	${GDK_PIXBUF_INCLUDE_DIRS}
 	${PANGO_INCLUDE_DIRS}
+	${PAM_INCLUDE_DIRS}
 )
 
 add_executable(swaylock
@@ -18,6 +19,7 @@ target_link_libraries(swaylock
 	${CAIRO_LIBRARIES}
 	${GDK_PIXBUF_LIBRARIES}
 	${PANGO_LIBRARIES}
+	${PAM_LIBRARIES}
 	m
 )
 
diff --git a/main.c b/main.c
index 95921d5..19993ce 100644
--- a/main.c
+++ b/main.c
@@ -1,4 +1,5 @@
 #include "wayland-swaylock-client-protocol.h"
+#include <security/pam_appl.h>
 #include <stdio.h>
 #include <stdlib.h>
 #include <string.h>
@@ -29,6 +30,39 @@ void sway_terminate(void) {
 	exit(EXIT_FAILURE);
 }
 
+struct pam_response *pam_reply;
+
+int function_conversation(int num_msg, const struct pam_message **msg,
+		struct pam_response **resp, void *appdata_ptr) {  
+	*resp = pam_reply;
+	return PAM_SUCCESS;  
+}  
+
+/**
+ * password will be zeroed out.
+ */
+bool verify_password(const char *username, char *password) {
+	const struct pam_conv local_conversation = { function_conversation, NULL };
+	pam_handle_t *local_auth_handle = NULL;
+	int pam_err;
+	if ((pam_err = pam_start("swaylock", username, &local_conversation, &local_auth_handle)) != PAM_SUCCESS) {
+		sway_abort("PAM returned %d\n", pam_err);
+	}
+	pam_reply = (struct pam_response *)malloc(sizeof(struct pam_response));
+	pam_reply[0].resp = password;
+	pam_reply[0].resp_retcode = 0;
+	if ((pam_err = pam_authenticate(local_auth_handle, 0)) != PAM_SUCCESS) {
+		memset(password, 0, strlen(password));
+		return false;
+	}
+	if ((pam_err = pam_end(local_auth_handle, pam_err)) != PAM_SUCCESS) {
+		memset(password, 0, strlen(password));
+		return false;
+	}
+	memset(password, 0, strlen(password));
+	return true;
+}
+
 void notify_key(enum wl_keyboard_key_state state, xkb_keysym_t sym, uint32_t code, uint32_t codepoint) {
 	sway_log(L_INFO, "notified of key %c", (char)codepoint);
 }