Added a min UID that can be checked and added all output to a verbose mode (-v).

This commit is contained in:
Johannes Findeisen 2024-08-10 13:42:04 +02:00
commit 5b13ee99b3

View file

@ -14,11 +14,16 @@
#include <stdio.h> #include <stdio.h>
#include <stdlib.h> #include <stdlib.h>
#include <string.h> #include <string.h>
#include <unistd.h> #include <unistd.h> // For getopt and access to user info
#include <pwd.h> // For struct passwd and getpwuid
#define MAX_USERNAME_LEN 32 #define MAX_USERNAME_LEN 32
#define MAX_PASSWORD_LEN 256 #define MAX_PASSWORD_LEN 256
#ifndef MIN_UID
#define MIN_UID 1000
#endif
// Custom data structure to hold user-entered password // Custom data structure to hold user-entered password
struct pam_credentials { struct pam_credentials {
const char *password; const char *password;
@ -27,7 +32,6 @@ struct pam_credentials {
// PAM conversation function to supply the password // PAM conversation function to supply the password
int pam_conversation(int num_msg, const struct pam_message **msg, int pam_conversation(int num_msg, const struct pam_message **msg,
struct pam_response **resp, void *appdata_ptr) { struct pam_response **resp, void *appdata_ptr) {
struct pam_response *response = NULL; struct pam_response *response = NULL;
struct pam_credentials *credentials = (struct pam_credentials *)appdata_ptr; struct pam_credentials *credentials = (struct pam_credentials *)appdata_ptr;
int i; int i;
@ -48,43 +52,58 @@ int pam_conversation(int num_msg, const struct pam_message **msg,
return PAM_SUCCESS; return PAM_SUCCESS;
} }
int authenticate(const char *username, const char *password) { int authenticate(const char *username, const char *password, int verbose) {
pam_handle_t *pamh = NULL; pam_handle_t *pamh = NULL;
int retval; int retval;
struct pam_credentials credentials = { password }; struct pam_credentials credentials = { password };
struct pam_conv conv = { pam_conversation, &credentials }; struct pam_conv conv = { pam_conversation, &credentials };
if (verbose) {
printf("Starting PAM authentication for user '%s'.\n", username);
}
retval = pam_start("login", username, &conv, &pamh); retval = pam_start("login", username, &conv, &pamh);
if (retval == PAM_SUCCESS) { if (retval == PAM_SUCCESS) {
if (verbose) {
printf("PAM authentication initialized.\n");
}
retval = pam_authenticate(pamh, 0); // Attempt to authenticate retval = pam_authenticate(pamh, 0); // Attempt to authenticate
} }
if (retval == PAM_SUCCESS) { if (retval == PAM_SUCCESS) {
if (verbose) {
printf("User '%s' authenticated successfully.\n", username);
}
retval = pam_acct_mgmt(pamh, 0); // Check account validity retval = pam_acct_mgmt(pamh, 0); // Check account validity
} }
if (pam_end(pamh, retval) != PAM_SUCCESS) { if (pam_end(pamh, retval) != PAM_SUCCESS) {
pamh = NULL; pamh = NULL;
fprintf(stderr, "Failed to release PAM authenticator\n"); if (verbose) {
fprintf(stderr, "Failed to release PAM authenticator\n");
}
exit(1); exit(1);
} }
if (retval != PAM_SUCCESS && verbose) {
printf("Authentication failed for user '%s'.\n", username);
}
return (retval == PAM_SUCCESS ? 0 : 1); // 0 for success, 1 for failure return (retval == PAM_SUCCESS ? 0 : 1); // 0 for success, 1 for failure
} }
int main(int argc, char *argv[]) { int main(int argc, char *argv[]) {
char username[MAX_USERNAME_LEN]; char username[MAX_USERNAME_LEN];
char password[MAX_PASSWORD_LEN]; char password[MAX_PASSWORD_LEN];
int verbose = 0; // Verbose mode flag
int opt; int opt;
memset(username, 0, sizeof(username)); memset(username, 0, sizeof(username));
memset(password, 0, sizeof(password)); memset(password, 0, sizeof(password));
// Parse command-line arguments // Parse command-line arguments
while ((opt = getopt(argc, argv, "u:p:")) != -1) { while ((opt = getopt(argc, argv, "u:p:v")) != -1) {
switch (opt) { switch (opt) {
case 'u': case 'u':
if (strlen(optarg) >= MAX_USERNAME_LEN) { if (strlen(optarg) >= MAX_USERNAME_LEN) {
@ -100,24 +119,54 @@ int main(int argc, char *argv[]) {
} }
strncpy(password, optarg, MAX_PASSWORD_LEN - 1); strncpy(password, optarg, MAX_PASSWORD_LEN - 1);
break; break;
case 'v':
verbose = 1; // Enable verbose mode
break;
default: default:
fprintf(stderr, "Usage: %s -u <username> -p <password>\n", argv[0]); fprintf(stderr, "Usage: %s -u <username> -p <password> [-v]\n", argv[0]);
exit(1); exit(1);
} }
} }
// Check if both username and password are provided // Check if both username and password are provided
if (username[0] == '\0' || password[0] == '\0') { if (username[0] == '\0' || password[0] == '\0') {
fprintf(stderr, "Usage: %s -u <username> -p <password>\n", argv[0]); if (verbose) {
fprintf(stderr, "Usage: %s -u <username> -p <password> [-v]\n", argv[0]);
}
exit(1); exit(1);
} }
// Retrieve user information from the username
struct passwd *pwd = getpwnam(username);
if (pwd == NULL) {
if (verbose) {
fprintf(stderr, "Error: User '%s' not found.\n", username);
}
exit(1);
}
// Check if the user's UID is below the minimum allowed UID
if (pwd->pw_uid < MIN_UID) {
if (verbose) {
fprintf(stderr, "Error: User '%s' has a UID less than %d and is not allowed to authenticate.\n", username, MIN_UID);
}
exit(1);
}
if (verbose) {
printf("User '%s' passed UID check (UID: %d).\n", username, pwd->pw_uid);
}
// Authenticate the user // Authenticate the user
if (authenticate(username, password) == 0) { if (authenticate(username, password, verbose) == 0) {
printf("Authenticated successfully.\n"); if (verbose) {
printf("Authenticated successfully.\n");
}
return 0; return 0;
} else { } else {
printf("Authentication failed.\n"); if (verbose) {
printf("Authentication failed.\n");
}
return 1; return 1;
} }
} }