source3/lib/util_cmdline.c

   1 /*
   2    Unix SMB/CIFS implementation.
   3    Samba utility functions
   4    Copyright (C) Andrew Tridgell 1992-1998
   5    Copyright (C) Jeremy Allison 2001-2007
   6    Copyright (C) Simo Sorce 2001
   7    Copyright (C) Jim McDonough <jmcd@us.ibm.com> 2003
   8    Copyright (C) James Peach 2006
   9
  10    This program is free software; you can redistribute it and/or modify
  11    it under the terms of the GNU General Public License as published by
  12    the Free Software Foundation; either version 3 of the License, or
  13    (at your option) any later version.
  14
  15    This program is distributed in the hope that it will be useful,
  16    but WITHOUT ANY WARRANTY; without even the implied warranty of
  17    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  18    GNU General Public License for more details.
  19
  20    You should have received a copy of the GNU General Public License
  21    along with this program.  If not, see <http://www.gnu.org/licenses/>.
  22 */
  23
  24 #include "includes.h"
  25 #include "popt_common.h"
  26 #include "secrets.h"
  27
  28 /**************************************************************************n
  29   Code to cope with username/password auth options from the commandline.
  30   Used mainly in client tools.
  31 ****************************************************************************/
  32
  33 struct user_auth_info *user_auth_info_init(TALLOC_CTX *mem_ctx)
  34 {
  35         struct user_auth_info *result;
  36
  37         result = talloc_zero(mem_ctx, struct user_auth_info);
  38         if (result == NULL) {
  39                 return NULL;
  40         }
  41
  42         result->signing_state = Undefined;
  43         return result;
  44 }
  45
  46 const char *get_cmdline_auth_info_username(const struct user_auth_info *auth_info)
  47 {
  48         if (!auth_info->username) {
  49                 return "";
  50         }
  51         return auth_info->username;
  52 }
  53
  54 void set_cmdline_auth_info_username(struct user_auth_info *auth_info,
  55                                     const char *username)
  56 {
  57         TALLOC_FREE(auth_info->username);
  58         auth_info->username = talloc_strdup(auth_info, username);
  59         if (!auth_info->username) {
  60                 exit(ENOMEM);
  61         }
  62 }
  63
  64 const char *get_cmdline_auth_info_domain(const struct user_auth_info *auth_info)
  65 {
  66         if (!auth_info->domain) {
  67                 return "";
  68         }
  69         return auth_info->domain;
  70 }
  71
  72 void set_cmdline_auth_info_domain(struct user_auth_info *auth_info,
  73                                   const char *domain)
  74 {
  75         TALLOC_FREE(auth_info->domain);
  76         auth_info->domain = talloc_strdup(auth_info, domain);
  77         if (!auth_info->domain) {
  78                 exit(ENOMEM);
  79         }
  80 }
  81
  82 const char *get_cmdline_auth_info_password(const struct user_auth_info *auth_info)
  83 {
  84         if (!auth_info->password) {
  85                 return "";
  86         }
  87         return auth_info->password;
  88 }
  89
  90 void set_cmdline_auth_info_password(struct user_auth_info *auth_info,
  91                                     const char *password)
  92 {
  93         TALLOC_FREE(auth_info->password);
  94         if (password == NULL) {
  95                 password = "";
  96         }
  97         auth_info->password = talloc_strdup(auth_info, password);
  98         if (!auth_info->password) {
  99                 exit(ENOMEM);
 100         }
 101         auth_info->got_pass = true;
 102 }
 103
 104 bool set_cmdline_auth_info_signing_state(struct user_auth_info *auth_info,
 105                                          const char *arg)
 106 {
 107         auth_info->signing_state = -1;
 108         if (strequal(arg, "off") || strequal(arg, "no") ||
 109                         strequal(arg, "false")) {
 110                 auth_info->signing_state = false;
 111         } else if (strequal(arg, "on") || strequal(arg, "yes") ||
 112                         strequal(arg, "true") || strequal(arg, "auto")) {
 113                 auth_info->signing_state = true;
 114         } else if (strequal(arg, "force") || strequal(arg, "required") ||
 115                         strequal(arg, "forced")) {
 116                 auth_info->signing_state = Required;
 117         } else {
 118                 return false;
 119         }
 120         return true;
 121 }
 122
 123 int get_cmdline_auth_info_signing_state(const struct user_auth_info *auth_info)
 124 {
 125         return auth_info->signing_state;
 126 }
 127
 128 void set_cmdline_auth_info_use_ccache(struct user_auth_info *auth_info, bool b)
 129 {
 130         auth_info->use_ccache = b;
 131 }
 132
 133 bool get_cmdline_auth_info_use_ccache(const struct user_auth_info *auth_info)
 134 {
 135         return auth_info->use_ccache;
 136 }
 137
 138 void set_cmdline_auth_info_use_kerberos(struct user_auth_info *auth_info,
 139                                         bool b)
 140 {
 141         auth_info->use_kerberos = b;
 142 }
 143
 144 bool get_cmdline_auth_info_use_kerberos(const struct user_auth_info *auth_info)
 145 {
 146         return auth_info->use_kerberos;
 147 }
 148
 149 void set_cmdline_auth_info_fallback_after_kerberos(struct user_auth_info *auth_info,
 150                                         bool b)
 151 {
 152         auth_info->fallback_after_kerberos = b;
 153 }
 154
 155 bool get_cmdline_auth_info_fallback_after_kerberos(const struct user_auth_info *auth_info)
 156 {
 157         return auth_info->fallback_after_kerberos;
 158 }
 159
 160 /* This should only be used by lib/popt_common.c JRA */
 161 void set_cmdline_auth_info_use_krb5_ticket(struct user_auth_info *auth_info)
 162 {
 163         auth_info->use_kerberos = true;
 164         auth_info->got_pass = true;
 165 }
 166
 167 /* This should only be used by lib/popt_common.c JRA */
 168 void set_cmdline_auth_info_smb_encrypt(struct user_auth_info *auth_info)
 169 {
 170         auth_info->smb_encrypt = true;
 171 }
 172
 173 void set_cmdline_auth_info_use_machine_account(struct user_auth_info *auth_info)
 174 {
 175         auth_info->use_machine_account = true;
 176 }
 177
 178 bool get_cmdline_auth_info_got_pass(const struct user_auth_info *auth_info)
 179 {
 180         return auth_info->got_pass;
 181 }
 182
 183 bool get_cmdline_auth_info_smb_encrypt(const struct user_auth_info *auth_info)
 184 {
 185         return auth_info->smb_encrypt;
 186 }
 187
 188 bool get_cmdline_auth_info_use_machine_account(const struct user_auth_info *auth_info)
 189 {
 190         return auth_info->use_machine_account;
 191 }
 192
 193 bool set_cmdline_auth_info_machine_account_creds(struct user_auth_info *auth_info)
 194 {
 195         char *pass = NULL;
 196         char *account = NULL;
 197
 198         if (!get_cmdline_auth_info_use_machine_account(auth_info)) {
 199                 return false;
 200         }
 201
 202         if (!secrets_init()) {
 203                 d_printf("ERROR: Unable to open secrets database\n");
 204                 return false;
 205         }
 206
 207         if (asprintf(&account, "%s$@%s", lp_netbios_name(), lp_realm()) < 0) {
 208                 return false;
 209         }
 210
 211         pass = secrets_fetch_machine_password(lp_workgroup(), NULL, NULL);
 212         if (!pass) {
 213                 d_printf("ERROR: Unable to fetch machine password for "
 214                         "%s in domain %s\n",
 215                         account, lp_workgroup());
 216                 SAFE_FREE(account);
 217                 return false;
 218         }
 219
 220         set_cmdline_auth_info_username(auth_info, account);
 221         set_cmdline_auth_info_password(auth_info, pass);
 222
 223         SAFE_FREE(account);
 224         SAFE_FREE(pass);
 225
 226         return true;
 227 }
 228
 229 /****************************************************************************
 230  Ensure we have a password if one not given.
 231 ****************************************************************************/
 232
 233 void set_cmdline_auth_info_getpass(struct user_auth_info *auth_info)
 234 {
 235         char *label = NULL;
 236         char *pass;
 237         TALLOC_CTX *frame;
 238
 239         if (get_cmdline_auth_info_got_pass(auth_info) ||
 240                         get_cmdline_auth_info_use_kerberos(auth_info)) {
 241                 /* Already got one... */
 242                 return;
 243         }
 244
 245         frame = talloc_stackframe();
 246         label = talloc_asprintf(frame, "Enter %s's password: ",
 247                         get_cmdline_auth_info_username(auth_info));
 248         pass = getpass(label);
 249         if (pass) {
 250                 set_cmdline_auth_info_password(auth_info, pass);
 251         }
 252         TALLOC_FREE(frame);
 253 }