source3/libsmb/samlogon_cache.c

   1 /*
   2    Unix SMB/CIFS implementation.
   3    Net_sam_logon info3 helpers
   4    Copyright (C) Alexander Bokovoy              2002.
   5    Copyright (C) Andrew Bartlett                2002.
   6    Copyright (C) Gerald Carter                  2003.
   7    Copyright (C) Tim Potter                     2003.
   8    Copyright (C) Guenther Deschner              2008.
   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 "system/filesys.h"
  26 #include "librpc/gen_ndr/ndr_krb5pac.h"
  27 #include "../libcli/security/security.h"
  28 #include "util_tdb.h"
  29
  30 #define NETSAMLOGON_TDB "netsamlogon_cache.tdb"
  31
  32 static TDB_CONTEXT *netsamlogon_tdb = NULL;
  33
  34 /***********************************************************************
  35  open the tdb
  36  ***********************************************************************/
  37
  38 bool netsamlogon_cache_init(void)
  39 {
  40         bool first_try = true;
  41         char *path = NULL;
  42         int ret;
  43         struct tdb_context *tdb;
  44
  45         if (netsamlogon_tdb) {
  46                 return true;
  47         }
  48
  49         path = cache_path(NETSAMLOGON_TDB);
  50         if (path == NULL) {
  51                 return false;
  52         }
  53 again:
  54         tdb = tdb_open_log(path, 0, TDB_DEFAULT|TDB_INCOMPATIBLE_HASH,
  55                            O_RDWR | O_CREAT, 0600);
  56         if (tdb == NULL) {
  57                 DEBUG(0,("tdb_open_log('%s') - failed\n", path));
  58                 goto clear;
  59         }
  60
  61         ret = tdb_check(tdb, NULL, NULL);
  62         if (ret != 0) {
  63                 tdb_close(tdb);
  64                 DEBUG(0,("tdb_check('%s') - failed\n", path));
  65                 goto clear;
  66         }
  67
  68         netsamlogon_tdb = tdb;
  69         talloc_free(path);
  70         return true;
  71
  72 clear:
  73         if (!first_try) {
  74                 talloc_free(path);
  75                 return false;
  76         }
  77         first_try = false;
  78
  79         DEBUG(0,("retry after truncate for '%s'\n", path));
  80         ret = truncate(path, 0);
  81         if (ret == -1) {
  82                 DBG_ERR("truncate failed: %s\n", strerror(errno));
  83                 talloc_free(path);
  84                 return false;
  85         }
  86
  87         goto again;
  88 }
  89
  90 /***********************************************************************
  91  Clear cache getpwnam and getgroups entries from the winbindd cache
  92 ***********************************************************************/
  93
  94 void netsamlogon_clear_cached_user(const struct dom_sid *user_sid)
  95 {
  96         fstring keystr;
  97
  98         if (!netsamlogon_cache_init()) {
  99                 DEBUG(0,("netsamlogon_clear_cached_user: cannot open "
 100                         "%s for write!\n",
 101                         NETSAMLOGON_TDB));
 102                 return;
 103         }
 104
 105         /* Prepare key as DOMAIN-SID/USER-RID string */
 106         sid_to_fstring(keystr, user_sid);
 107
 108         DEBUG(10,("netsamlogon_clear_cached_user: SID [%s]\n", keystr));
 109
 110         tdb_delete_bystring(netsamlogon_tdb, keystr);
 111 }
 112
 113 /***********************************************************************
 114  Store a netr_SamInfo3 structure in a tdb for later user
 115  username should be in UTF-8 format
 116 ***********************************************************************/
 117
 118 bool netsamlogon_cache_store(const char *username, struct netr_SamInfo3 *info3)
 119 {
 120         TDB_DATA data;
 121         fstring keystr;
 122         bool result = false;
 123         struct dom_sid  user_sid;
 124         time_t t = time(NULL);
 125         TALLOC_CTX *tmp_ctx = talloc_stackframe();
 126         DATA_BLOB blob;
 127         enum ndr_err_code ndr_err;
 128         struct netsamlogoncache_entry r;
 129
 130         if (!info3) {
 131                 return false;
 132         }
 133
 134         if (!netsamlogon_cache_init()) {
 135                 DEBUG(0,("netsamlogon_cache_store: cannot open %s for write!\n",
 136                         NETSAMLOGON_TDB));
 137                 return false;
 138         }
 139
 140         sid_compose(&user_sid, info3->base.domain_sid, info3->base.rid);
 141
 142         /* Prepare key as DOMAIN-SID/USER-RID string */
 143         sid_to_fstring(keystr, &user_sid);
 144
 145         DEBUG(10,("netsamlogon_cache_store: SID [%s]\n", keystr));
 146
 147         /* Prepare data */
 148
 149         if (info3->base.full_name.string == NULL) {
 150                 struct netr_SamInfo3 *cached_info3;
 151                 const char *full_name = NULL;
 152
 153                 cached_info3 = netsamlogon_cache_get(tmp_ctx, &user_sid);
 154                 if (cached_info3 != NULL) {
 155                         full_name = cached_info3->base.full_name.string;
 156                 }
 157
 158                 if (full_name != NULL) {
 159                         info3->base.full_name.string = talloc_strdup(info3, full_name);
 160                 }
 161         }
 162
 163         /* only Samba fills in the username, not sure why NT doesn't */
 164         /* so we fill it in since winbindd_getpwnam() makes use of it */
 165
 166         if (!info3->base.account_name.string) {
 167                 info3->base.account_name.string = talloc_strdup(info3, username);
 168         }
 169
 170         r.timestamp = t;
 171         r.info3 = *info3;
 172
 173         if (DEBUGLEVEL >= 10) {
 174                 NDR_PRINT_DEBUG(netsamlogoncache_entry, &r);
 175         }
 176
 177         ndr_err = ndr_push_struct_blob(&blob, tmp_ctx, &r,
 178                                        (ndr_push_flags_fn_t)ndr_push_netsamlogoncache_entry);
 179         if (!NDR_ERR_CODE_IS_SUCCESS(ndr_err)) {
 180                 DEBUG(0,("netsamlogon_cache_store: failed to push entry to cache\n"));
 181                 TALLOC_FREE(tmp_ctx);
 182                 return false;
 183         }
 184
 185         data.dsize = blob.length;
 186         data.dptr = blob.data;
 187
 188         if (tdb_store_bystring(netsamlogon_tdb, keystr, data, TDB_REPLACE) == 0) {
 189                 result = true;
 190         }
 191
 192         TALLOC_FREE(tmp_ctx);
 193
 194         return result;
 195 }
 196
 197 /***********************************************************************
 198  Retrieves a netr_SamInfo3 structure from a tdb.  Caller must
 199  free the user_info struct (malloc()'d memory)
 200 ***********************************************************************/
 201
 202 struct netr_SamInfo3 *netsamlogon_cache_get(TALLOC_CTX *mem_ctx, const struct dom_sid *user_sid)
 203 {
 204         struct netr_SamInfo3 *info3 = NULL;
 205         TDB_DATA data;
 206         fstring keystr;
 207         enum ndr_err_code ndr_err;
 208         DATA_BLOB blob;
 209         struct netsamlogoncache_entry r;
 210
 211         if (!netsamlogon_cache_init()) {
 212                 DEBUG(0,("netsamlogon_cache_get: cannot open %s for write!\n",
 213                         NETSAMLOGON_TDB));
 214                 return NULL;
 215         }
 216
 217         /* Prepare key as DOMAIN-SID/USER-RID string */
 218         sid_to_fstring(keystr, user_sid);
 219         DEBUG(10,("netsamlogon_cache_get: SID [%s]\n", keystr));
 220         data = tdb_fetch_bystring( netsamlogon_tdb, keystr );
 221
 222         if (!data.dptr) {
 223                 return NULL;
 224         }
 225
 226         info3 = talloc_zero(mem_ctx, struct netr_SamInfo3);
 227         if (!info3) {
 228                 goto done;
 229         }
 230
 231         blob = data_blob_const(data.dptr, data.dsize);
 232
 233         ndr_err = ndr_pull_struct_blob(&blob, mem_ctx, &r,
 234                                       (ndr_pull_flags_fn_t)ndr_pull_netsamlogoncache_entry);
 235
 236         if (!NDR_ERR_CODE_IS_SUCCESS(ndr_err)) {
 237                 DEBUG(0,("netsamlogon_cache_get: failed to pull entry from cache\n"));
 238                 tdb_delete_bystring(netsamlogon_tdb, keystr);
 239                 TALLOC_FREE(info3);
 240                 goto done;
 241         }
 242
 243         if (DEBUGLEVEL >= 10) {
 244                 NDR_PRINT_DEBUG(netsamlogoncache_entry, &r);
 245         }
 246
 247         info3 = (struct netr_SamInfo3 *)talloc_memdup(mem_ctx, &r.info3,
 248                                                       sizeof(r.info3));
 249
 250  done:
 251         SAFE_FREE(data.dptr);
 252
 253         return info3;
 254 }
 255
 256 bool netsamlogon_cache_have(const struct dom_sid *user_sid)
 257 {
 258         TALLOC_CTX *mem_ctx = talloc_init("netsamlogon_cache_have");
 259         struct netr_SamInfo3 *info3 = NULL;
 260         bool result;
 261
 262         if (!mem_ctx)
 263                 return False;
 264
 265         info3 = netsamlogon_cache_get(mem_ctx, user_sid);
 266
 267         result = (info3 != NULL);
 268
 269         talloc_destroy(mem_ctx);
 270
 271         return result;
 272 }