From 6ac4eeab2f70a26462eeef28301793bd94dd9f17 Mon Sep 17 00:00:00 2001 From: Yang Tse Date: Wed, 27 Jul 2011 19:04:55 +0200 Subject: NTLM single-sign on adjustments (II) Ensure test harness fake_ntlm main function can properly handle arguments. --- tests/server/fake_ntlm.c | 37 ++++++++++++++++++++++++++++++++++++- 1 file changed, 36 insertions(+), 1 deletion(-) (limited to 'tests/server/fake_ntlm.c') diff --git a/tests/server/fake_ntlm.c b/tests/server/fake_ntlm.c index f7abf621b..9ce8436ef 100644 --- a/tests/server/fake_ntlm.c +++ b/tests/server/fake_ntlm.c @@ -49,7 +49,7 @@ const char *serverlogfile = DEFAULT_LOGFILE; -int main(void) +int main(int argc, char *argv[]) { char buf[1024]; FILE *stream; @@ -60,6 +60,41 @@ int main(void) size_t size = 0; int testnum; const char *env; + int arg = 1; + char *helper_user = (char *)"unknown"; + char *helper_proto = (char *)"unknown"; + char *helper_domain = (char *)"unknown"; + bool use_cached_creds = FALSE; + + while(argc > arg) { + if(!strcmp("--use-cached-creds", argv[arg])) { + use_cached_creds = TRUE; + arg++; + } + else if(!strcmp("--helper-protocol", argv[arg])) { + arg++; + if(argc > arg) + helper_proto = argv[arg++]; + } + else if(!strcmp("--username", argv[arg])) { + arg++; + if(argc > arg) + helper_user = argv[arg++]; + } + else if(!strcmp("--domain", argv[arg])) { + arg++; + if(argc > arg) + helper_domain = argv[arg++]; + } + else { + puts("Usage: fake_ntlm [option]\n" + " --use-cached-creds\n" + " --helper-protocol [protocol]\n" + " --username [username]\n" + " --domain [domain]"); + exit(1); + } + } env = getenv("NTLM_AUTH_TESTNUM"); if (env) { -- cgit v1.2.3