auth/credentials/tests/simple.c

   1 /*
   2    Unix SMB/CIFS implementation.
   3    Samba utility functions
   4    Copyright (C) Jelmer Vernooij <jelmer@samba.org> 2007
   5
   6    This program is free software; you can redistribute it and/or modify
   7    it under the terms of the GNU General Public License as published by
   8    the Free Software Foundation; either version 3 of the License, or
   9    (at your option) any later version.
  10
  11    This program is distributed in the hope that it will be useful,
  12    but WITHOUT ANY WARRANTY; without even the implied warranty of
  13    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14    GNU General Public License for more details.
  15
  16    You should have received a copy of the GNU General Public License
  17    along with this program.  If not, see <http://www.gnu.org/licenses/>.
  18 */
  19
  20 #include "includes.h"
  21 #include "auth/credentials/credentials.h"
  22 #include "torture/torture.h"
  23 #include "torture/local/proto.h"
  24
  25 static bool test_init(struct torture_context *tctx)
  26 {
  27         struct cli_credentials *creds = cli_credentials_init(tctx);
  28
  29         cli_credentials_set_domain(creds, "bla", CRED_SPECIFIED);
  30
  31         torture_assert_str_equal(tctx, "BLA", cli_credentials_get_domain(creds),
  32                                  "domain");
  33
  34         cli_credentials_set_username(creds, "someuser", CRED_SPECIFIED);
  35
  36         torture_assert_str_equal(tctx, "someuser",
  37                                  cli_credentials_get_username(creds),
  38                                  "username");
  39
  40         cli_credentials_set_password(creds, "p4ssw0rd", CRED_SPECIFIED);
  41
  42         torture_assert_str_equal(tctx, "p4ssw0rd",
  43                                  cli_credentials_get_password(creds),
  44                                  "password");
  45
  46         return true;
  47 }
  48
  49 static bool test_init_anonymous(struct torture_context *tctx)
  50 {
  51         struct cli_credentials *creds = cli_credentials_init_anon(tctx);
  52
  53         torture_assert_str_equal(tctx, cli_credentials_get_domain(creds),
  54                                  "", "domain");
  55
  56         torture_assert_str_equal(tctx, cli_credentials_get_username(creds),
  57                                  "", "username");
  58
  59         torture_assert(tctx, cli_credentials_get_password(creds) == NULL,
  60                                  "password");
  61
  62         return true;
  63 }
  64
  65 static bool test_guess(struct torture_context *tctx)
  66 {
  67         struct cli_credentials *creds = cli_credentials_init_anon(tctx);
  68         enum credentials_use_kerberos old_kerb_state = \
  69                 cli_credentials_get_kerberos_state(creds);
  70         const char *logname = getenv("LOGNAME");
  71         const char *user = getenv("USER");
  72         const char *passwd = getenv("PASSWD");
  73         const char *passwd_fd = getenv("PASSWD_FD");
  74         const char *passwd_file = getenv("PASSWD_FILE");
  75
  76         cli_credentials_set_kerberos_state(creds,
  77                                            CRED_USE_KERBEROS_REQUIRED,
  78                                            CRED_SPECIFIED);
  79
  80         unsetenv("USER");
  81         unsetenv("PASSWD_FD");
  82         unsetenv("PASSWD_FILE");
  83
  84         setenv("LOGNAME", "xx", 1);
  85         setenv("PASSWD", "xx", 1);
  86
  87         cli_credentials_guess(creds, NULL);
  88         if (logname != NULL) {
  89                 setenv("LOGNAME", logname, 1);
  90         }
  91         if (user != NULL) {
  92                 setenv("USER", user, 1);
  93         }
  94         if (passwd != NULL) {
  95                 setenv("PASSWD", passwd, 1);
  96         }
  97         if (passwd_fd != NULL) {
  98                 setenv("PASSWD_FD", passwd_fd, 1);
  99         }
 100         if (passwd_file != NULL) {
 101                 setenv("PASSWD_FILE", passwd_file, 1);
 102         }
 103         cli_credentials_set_kerberos_state(creds,
 104                                            old_kerb_state,
 105                                            CRED_SPECIFIED);
 106
 107         return true;
 108 }
 109
 110
 111 static bool test_parse_string(struct torture_context *tctx)
 112 {
 113         struct cli_credentials *creds = cli_credentials_init_anon(tctx);
 114
 115         /* anonymous */
 116         cli_credentials_parse_string(creds, "%", CRED_SPECIFIED);
 117
 118         torture_assert_str_equal(tctx, cli_credentials_get_domain(creds),
 119                                  "", "domain");
 120
 121         torture_assert_str_equal(tctx, cli_credentials_get_username(creds),
 122                                  "", "username");
 123
 124         torture_assert(tctx, cli_credentials_get_password(creds) == NULL,
 125                                  "password");
 126
 127         /* username + password */
 128         cli_credentials_parse_string(creds, "somebody%secret",
 129                                      CRED_SPECIFIED);
 130
 131         torture_assert_str_equal(tctx, cli_credentials_get_domain(creds),
 132                                  "", "domain");
 133
 134         torture_assert_str_equal(tctx, cli_credentials_get_username(creds),
 135                                  "somebody", "username");
 136
 137         torture_assert_str_equal(tctx, cli_credentials_get_password(creds),
 138                                  "secret", "password");
 139
 140         /* principal */
 141         cli_credentials_parse_string(creds, "prin@styx",
 142                                      CRED_SPECIFIED);
 143
 144         torture_assert_str_equal(tctx, cli_credentials_get_realm(creds),
 145                                  "STYX", "realm");
 146
 147         torture_assert_str_equal(tctx,
 148                                  cli_credentials_get_principal(creds, tctx),
 149                                  "prin@styx", "principal");
 150
 151         return true;
 152 }
 153
 154 struct torture_suite *torture_local_credentials(TALLOC_CTX *mem_ctx)
 155 {
 156         struct torture_suite *suite = torture_suite_create(mem_ctx, "credentials");
 157
 158         torture_suite_add_simple_test(suite, "init", test_init);
 159         torture_suite_add_simple_test(suite, "init anonymous",
 160                                       test_init_anonymous);
 161         torture_suite_add_simple_test(suite, "guess",
 162                                       test_guess);
 163         torture_suite_add_simple_test(suite, "parse_string",
 164                                       test_parse_string);
 165
 166         return suite;
 167 }
 168