91 lines
1.8 KiB
C
91 lines
1.8 KiB
C
#include <pwd.h>
|
|
#include <security/pam_appl.h>
|
|
#include <stdbool.h>
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
#include <strings.h>
|
|
#include <sys/mman.h>
|
|
#include <unistd.h>
|
|
|
|
static char *input_buffer = NULL;
|
|
|
|
#define INPUT_BUFFER_SIZE 1024
|
|
|
|
static int
|
|
conversation_handler(int num_msg,
|
|
const struct pam_message **msg,
|
|
struct pam_response **resp,
|
|
void *data)
|
|
{
|
|
struct pam_response *pam_reply = calloc(num_msg, sizeof(struct pam_response));
|
|
|
|
if (pam_reply == NULL) {
|
|
return PAM_ABORT;
|
|
}
|
|
*resp = pam_reply;
|
|
for (int i = 0; i < num_msg; ++i) {
|
|
switch (msg[i]->msg_style) {
|
|
case PAM_PROMPT_ECHO_OFF:
|
|
case PAM_PROMPT_ECHO_ON:
|
|
pam_reply[i].resp = strdup(input_buffer);
|
|
if (pam_reply[i].resp == NULL) {
|
|
return PAM_ABORT;
|
|
}
|
|
break;
|
|
|
|
case PAM_ERROR_MSG:
|
|
case PAM_TEXT_INFO:
|
|
break;
|
|
}
|
|
}
|
|
return PAM_SUCCESS;
|
|
}
|
|
|
|
bool
|
|
check_password(const char *username)
|
|
{
|
|
const struct pam_conv conv = {
|
|
.conv = conversation_handler,
|
|
.appdata_ptr = NULL,
|
|
};
|
|
|
|
bool success = false;
|
|
pam_handle_t *auth_handle = NULL;
|
|
if (pam_start("hikari-unlocker", username, &conv, &auth_handle) !=
|
|
PAM_SUCCESS) {
|
|
return false;
|
|
}
|
|
|
|
read(0, input_buffer, INPUT_BUFFER_SIZE - 1);
|
|
int pam_status = pam_authenticate(auth_handle, 0);
|
|
memset(input_buffer, 0, INPUT_BUFFER_SIZE);
|
|
success = pam_status == PAM_SUCCESS;
|
|
write(1, &success, sizeof(bool));
|
|
|
|
pam_end(auth_handle, pam_status);
|
|
|
|
return success;
|
|
}
|
|
|
|
int
|
|
main(int argc, char **argv)
|
|
{
|
|
char input;
|
|
bool success = false;
|
|
struct passwd *passwd = getpwuid(getuid());
|
|
|
|
input_buffer = malloc(INPUT_BUFFER_SIZE);
|
|
memset(input_buffer, 0, INPUT_BUFFER_SIZE);
|
|
mlock(input_buffer, INPUT_BUFFER_SIZE);
|
|
|
|
while (!success) {
|
|
success = check_password(passwd->pw_name);
|
|
}
|
|
|
|
munlock(input_buffer, INPUT_BUFFER_SIZE);
|
|
free(input_buffer);
|
|
|
|
return 0;
|
|
}
|