rooty/yubico-pam
1
0
Fork 0
mirror of https://github.com/Yubico/yubico-pam.git synced 2025-01-20 10:52:16 +01:00
Code Issues Projects Releases Wiki Activity

add check_user_challenge_file tests

Browse Source
This commit is contained in:
Larhard 2017-06-24 08:30:40 +00:00
parent d3f50c43ac
commit 42e8a06bfe
1 changed files with 118 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

118
tests/util_test.c
View File

@ -35,6 +35,7 @@
#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>
#include <unistd.h>
#include <pwd.h>
@ -144,6 +145,122 @@ static void test_load_chalresp_state(void) {
fclose(file);
}
static void test_check_user_challenge_file(void) {
int ret;
char * tmpdir_path;
char * buf;
FILE * file;
struct passwd user;
buf = malloc(256);
#define create_tmpdir_dir(path) \
strcpy(buf, tmpdir_path); \
strcat(buf, "/"); \
strcat(buf, path); \
mkdir(buf, 0755);
#define remove_tmpdir_dir(path) \
strcpy(buf, tmpdir_path); \
strcat(buf, "/"); \
strcat(buf, path); \
rmdir(buf);
#define create_tmpdir_file(path) \
strcpy(buf, tmpdir_path); \
strcat(buf, "/"); \
strcat(buf, path); \
file = fopen(buf, "w"); \
fclose(file);
#define remove_tmpdir_file(path) \
strcpy(buf, tmpdir_path); \
strcat(buf, "/"); \
strcat(buf, path); \
unlink(buf);
/* create temporary directory */
tmpdir_path = tempnam(NULL, "pamtest");
assert(tmpdir_path != NULL);
ret = mkdir(tmpdir_path, 0755);
assert(ret == 0);
/* set user data */
user.pw_name = "tester";
user.pw_dir = tmpdir_path;
/* execute tests */
/* no asserts here as we have directory to remove */
int case_001_empty_chalresp_dir;
case_001_empty_chalresp_dir = check_user_challenge_file(tmpdir_path, &user, stdout);
int case_002_one_challenge_file;
create_tmpdir_file("tester");
case_002_one_challenge_file = check_user_challenge_file(tmpdir_path, &user, stdout);
remove_tmpdir_file("tester");
int case_003_multiple_challenge_files;
create_tmpdir_file("tester-001");
create_tmpdir_file("tester-002");
case_003_multiple_challenge_files = check_user_challenge_file(tmpdir_path, &user, stdout);
remove_tmpdir_file("tester-002");
remove_tmpdir_file("tester-001");
int case_004_other_users_files;
create_tmpdir_file("tester1");
create_tmpdir_file("tester1-001");
case_004_other_users_files = check_user_challenge_file(tmpdir_path, &user, stdout);
remove_tmpdir_file("tester1-001");
remove_tmpdir_file("tester1");
int case_005_no_chalresp_no_yubico;
case_005_no_chalresp_no_yubico = check_user_challenge_file(NULL, &user, stdout);
int case_006_no_chalresp_empty_yubico;
create_tmpdir_dir(".yubico");
case_006_no_chalresp_empty_yubico = check_user_challenge_file(NULL, &user, stdout);
remove_tmpdir_dir(".yubico");
int case_007_no_chalresp_one_challenge_file;
create_tmpdir_dir(".yubico");
create_tmpdir_file(".yubico/challenge");
case_007_no_chalresp_one_challenge_file = check_user_challenge_file(NULL, &user, stdout);
remove_tmpdir_file(".yubico/challenge");
remove_tmpdir_dir(".yubico");
int case_008_no_chalresp_multiple_challenge_files;
create_tmpdir_dir(".yubico");
create_tmpdir_file(".yubico/challenge-001");
create_tmpdir_file(".yubico/challenge-002");
case_008_no_chalresp_multiple_challenge_files = check_user_challenge_file(NULL, &user, stdout);
remove_tmpdir_file(".yubico/challenge-002");
remove_tmpdir_file(".yubico/challenge-001");
remove_tmpdir_dir(".yubico");
/* remove temporary directory */
ret = rmdir(tmpdir_path);
assert(ret == 0);
free(tmpdir_path);
free(buf);
/* check test results */
assert(case_001_empty_chalresp_dir == AUTH_NOT_FOUND);
assert(case_002_one_challenge_file == AUTH_FOUND);
assert(case_003_multiple_challenge_files == AUTH_FOUND);
assert(case_004_other_users_files == AUTH_NOT_FOUND);
assert(case_005_no_chalresp_no_yubico == AUTH_NOT_FOUND);
assert(case_006_no_chalresp_empty_yubico == AUTH_NOT_FOUND);
assert(case_007_no_chalresp_one_challenge_file == AUTH_FOUND);
assert(case_008_no_chalresp_multiple_challenge_files == AUTH_FOUND);
#undef create_tmpdir_dir
#undef remove_tmpdir_dir
#undef create_tmpdir_file
#undef remove_tmpdir_file
}
#endif /* HAVE_CR */
static void test_filter_printf(void) {
@ -171,6 +288,7 @@ int main (void) {
test_check_user_token();
#if HAVE_CR
test_load_chalresp_state();
test_check_user_challenge_file();
#endif /* HAVE_CR */
return 0;
}