source4/dsdb/samdb/samdb_privilege.c

   1 /*
   2    Unix SMB/CIFS implementation.
   3
   4    manipulate privilege records in samdb
   5
   6    Copyright (C) Andrew Tridgell 2004
   7
   8    This program is free software; you can redistribute it and/or modify
   9    it under the terms of the GNU General Public License as published by
  10    the Free Software Foundation; either version 3 of the License, or
  11    (at your option) any later version.
  12
  13    This program is distributed in the hope that it will be useful,
  14    but WITHOUT ANY WARRANTY; without even the implied warranty of
  15    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  16    GNU General Public License for more details.
  17
  18    You should have received a copy of the GNU General Public License
  19    along with this program.  If not, see <http://www.gnu.org/licenses/>.
  20 */
  21
  22 #include "includes.h"
  23 #include "libcli/ldap/ldap_ndr.h"
  24 #include "dsdb/samdb/samdb.h"
  25 #include "auth/auth.h"
  26 #include "libcli/security/security.h"
  27 #include "../lib/util/util_ldb.h"
  28 #include "param/param.h"
  29 #include "ldb_wrap.h"
  30
  31 /* connect to the privilege database */
  32 struct ldb_context *privilege_connect(TALLOC_CTX *mem_ctx,
  33                                       struct tevent_context *ev_ctx,
  34                                       struct loadparm_context *lp_ctx)
  35 {
  36         char *path;
  37         struct ldb_context *pdb;
  38
  39         path = private_path(mem_ctx, lp_ctx, "privilege.ldb");
  40         if (!path) return NULL;
  41
  42         pdb = ldb_wrap_connect(mem_ctx, ev_ctx, lp_ctx, path,
  43                                NULL, NULL, 0);
  44         talloc_free(path);
  45
  46         return pdb;
  47 }
  48
  49 /*
  50   add privilege bits for one sid to a security_token
  51 */
  52 static NTSTATUS samdb_privilege_setup_sid(struct ldb_context *pdb, TALLOC_CTX *mem_ctx,
  53                                           struct security_token *token,
  54                                           const struct dom_sid *sid)
  55 {
  56         const char * const attrs[] = { "privilege", NULL };
  57         struct ldb_message **res = NULL;
  58         struct ldb_message_element *el;
  59         int ret, i;
  60         char *sidstr;
  61
  62         sidstr = ldap_encode_ndr_dom_sid(mem_ctx, sid);
  63         NT_STATUS_HAVE_NO_MEMORY(sidstr);
  64
  65         ret = gendb_search(pdb, mem_ctx, NULL, &res, attrs, "objectSid=%s", sidstr);
  66         talloc_free(sidstr);
  67         if (ret != 1) {
  68                 /* not an error to not match */
  69                 return NT_STATUS_OK;
  70         }
  71
  72         el = ldb_msg_find_element(res[0], "privilege");
  73         if (el == NULL) {
  74                 return NT_STATUS_OK;
  75         }
  76
  77         for (i=0;i<el->num_values;i++) {
  78                 const char *priv_str = (const char *)el->values[i].data;
  79                 enum sec_privilege privilege = sec_privilege_id(priv_str);
  80                 if (privilege == -1) {
  81                         DEBUG(1,("Unknown privilege '%s' in samdb\n",
  82                                  priv_str));
  83                         continue;
  84                 }
  85                 security_token_set_privilege(token, privilege);
  86         }
  87
  88         return NT_STATUS_OK;
  89 }
  90
  91 /*
  92   setup the privilege mask for this security token based on our
  93   local SAM
  94 */
  95 NTSTATUS samdb_privilege_setup(struct tevent_context *ev_ctx,
  96                                struct loadparm_context *lp_ctx, struct security_token *token)
  97 {
  98         struct ldb_context *pdb;
  99         TALLOC_CTX *mem_ctx;
 100         int i;
 101         NTSTATUS status;
 102
 103         /* Shortcuts to prevent recursion and avoid lookups */
 104         if (token->user_sid == NULL) {
 105                 token->privilege_mask = 0;
 106                 return NT_STATUS_OK;
 107         }
 108
 109         if (security_token_is_system(token)) {
 110                 token->privilege_mask = ~0;
 111                 return NT_STATUS_OK;
 112         }
 113
 114         if (security_token_is_anonymous(token)) {
 115                 token->privilege_mask = 0;
 116                 return NT_STATUS_OK;
 117         }
 118
 119         mem_ctx = talloc_new(token);
 120         pdb = privilege_connect(mem_ctx, ev_ctx, lp_ctx);
 121         if (pdb == NULL) {
 122                 talloc_free(mem_ctx);
 123                 return NT_STATUS_INTERNAL_DB_CORRUPTION;
 124         }
 125
 126         token->privilege_mask = 0;
 127
 128         for (i=0;i<token->num_sids;i++) {
 129                 status = samdb_privilege_setup_sid(pdb, mem_ctx,
 130                                                    token, token->sids[i]);
 131                 if (!NT_STATUS_IS_OK(status)) {
 132                         talloc_free(mem_ctx);
 133                         return status;
 134                 }
 135         }
 136
 137         talloc_free(mem_ctx);
 138
 139         return NT_STATUS_OK;
 140 }