← Back to branch summary

~zulcss/samba/server-dailies-3.4

« back to all changes in this revision

Viewing changes to source3/winbindd/idmap_hash/idmap_hash.c

  • Committer: Chuck Short
  • Date: 2010-09-28 20:38:39 UTC
  • Revision ID: zulcss@ubuntu.com-20100928203839-pgjulytsi9ue63x1
Initial version

Show diffs side-by-side

added added

removed removed

Lines of Context:
source3/winbindd/idmap_hash/idmap_hash.c
 
1
/*
 
2
 *  idmap_hash.c
 
3
 *
 
4
 * Copyright (C) Gerald Carter  <jerry@samba.org>      2007 - 2008
 
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
 
 
21
#include "includes.h"
 
22
#include "winbindd/winbindd.h"
 
23
#include "idmap_hash.h"
 
24
 
 
25
#undef DBGC_CLASS
 
26
#define DBGC_CLASS DBGC_IDMAP
 
27
 
 
28
struct sid_hash_table {
 
29
        DOM_SID *sid;
 
30
};
 
31
 
 
32
struct sid_hash_table *hashed_domains = NULL;
 
33
 
 
34
/*********************************************************************
 
35
 Hash a domain SID (S-1-5-12-aaa-bbb-ccc) to a 12bit number
 
36
 ********************************************************************/
 
37
 
 
38
static uint32_t hash_domain_sid(const DOM_SID *sid)
 
39
{
 
40
        uint32_t hash;
 
41
 
 
42
        if (sid->num_auths != 4)
 
43
                return 0;
 
44
 
 
45
        /* XOR the last three subauths */
 
46
 
 
47
        hash = ((sid->sub_auths[1] ^ sid->sub_auths[2]) ^ sid->sub_auths[3]);
 
48
 
 
49
        /* Take all 32-bits into account when generating the 12-bit
 
50
           hash value */
 
51
        hash = (((hash & 0xFFF00000) >> 20)
 
52
                + ((hash & 0x000FFF00) >> 8)
 
53
                + (hash & 0x000000FF)) & 0x0000FFF;
 
54
 
 
55
        /* return a 12-bit hash value */
 
56
 
 
57
        return hash;
 
58
}
 
59
 
 
60
/*********************************************************************
 
61
 Hash a Relative ID to a 20 bit number
 
62
 ********************************************************************/
 
63
 
 
64
static uint32_t hash_rid(uint32_t rid)
 
65
{
 
66
        /* 20 bits for the rid which allows us to support
 
67
           the first 100K users/groups in a domain */
 
68
 
 
69
        return (rid & 0x0007FFFF);
 
70
}
 
71
 
 
72
/*********************************************************************
 
73
 ********************************************************************/
 
74
 
 
75
static uint32_t combine_hashes(uint32_t h_domain,
 
76
                               uint32_t h_rid)
 
77
{
 
78
        uint32_t return_id = 0;
 
79
 
 
80
        /* shift the hash_domain 19 bits to the left and OR with the
 
81
           hash_rid */
 
82
 
 
83
        return_id = ((h_domain<<19) | h_rid);
 
84
 
 
85
        return return_id;
 
86
}
 
87
 
 
88
/*********************************************************************
 
89
 ********************************************************************/
 
90
 
 
91
static void separate_hashes(uint32_t id,
 
92
                            uint32_t *h_domain,
 
93
                            uint32_t *h_rid)
 
94
{
 
95
        *h_rid = id & 0x0007FFFF;
 
96
        *h_domain = (id & 0x7FF80000) >> 19;
 
97
 
 
98
        return;
 
99
}
 
100
 
 
101
 
 
102
/*********************************************************************
 
103
 ********************************************************************/
 
104
 
 
105
static NTSTATUS be_init(struct idmap_domain *dom,
 
106
                        const char *params)
 
107
{
 
108
        NTSTATUS nt_status = NT_STATUS_UNSUCCESSFUL;
 
109
        struct winbindd_tdc_domain *dom_list = NULL;
 
110
        size_t num_domains = 0;
 
111
        int i;
 
112
 
 
113
        /* If the domain SID hash talbe has been initialized, assume
 
114
           that we completed this function previously */
 
115
 
 
116
        if ( hashed_domains ) {
 
117
                nt_status = NT_STATUS_OK;
 
118
                goto done;
 
119
        }
 
120
 
 
121
        if (!wcache_tdc_fetch_list(&dom_list, &num_domains)) {
 
122
                nt_status = NT_STATUS_TRUSTED_DOMAIN_FAILURE;
 
123
                BAIL_ON_NTSTATUS_ERROR(nt_status);
 
124
        }
 
125
 
 
126
        /* Create the hash table of domain SIDs */
 
127
 
 
128
        hashed_domains = TALLOC_ZERO_ARRAY(NULL, struct sid_hash_table, 4096);
 
129
        BAIL_ON_PTR_NT_ERROR(hashed_domains, nt_status);
 
130
 
 
131
        /* create the hash table of domain SIDs */
 
132
 
 
133
        for (i=0; i<num_domains; i++) {
 
134
                uint32_t hash;
 
135
 
 
136
                if (is_null_sid(&dom_list[i].sid))
 
137
                        continue;
 
138
                if ((hash = hash_domain_sid(&dom_list[i].sid)) == 0)
 
139
                        continue;
 
140
 
 
141
                DEBUG(5,("hash:be_init() Adding %s (%s) -> %d\n",
 
142
                         dom_list[i].domain_name,
 
143
                         sid_string_dbg(&dom_list[i].sid),
 
144
                         hash));
 
145
 
 
146
                hashed_domains[hash].sid = talloc(hashed_domains, DOM_SID);
 
147
                sid_copy(hashed_domains[hash].sid, &dom_list[i].sid);
 
148
        }
 
149
 
 
150
done:
 
151
        return nt_status;
 
152
}
 
153
 
 
154
/*********************************************************************
 
155
 ********************************************************************/
 
156
 
 
157
static NTSTATUS unixids_to_sids(struct idmap_domain *dom,
 
158
                                struct id_map **ids)
 
159
{
 
160
        NTSTATUS nt_status = NT_STATUS_UNSUCCESSFUL;
 
161
        int i;
 
162
 
 
163
        /* initialize the status to avoid suprise */
 
164
        for (i = 0; ids[i]; i++) {
 
165
                ids[i]->status = ID_UNKNOWN;
 
166
        }
 
167
        
 
168
        nt_status = be_init(dom, NULL);
 
169
        BAIL_ON_NTSTATUS_ERROR(nt_status);
 
170
 
 
171
        if (!ids) {
 
172
                nt_status = NT_STATUS_INVALID_PARAMETER;
 
173
                BAIL_ON_NTSTATUS_ERROR(nt_status);
 
174
        }
 
175
 
 
176
        for (i=0; ids[i]; i++) {
 
177
                uint32_t h_domain, h_rid;
 
178
 
 
179
                ids[i]->status = ID_UNMAPPED;
 
180
 
 
181
                separate_hashes(ids[i]->xid.id, &h_domain, &h_rid);
 
182
 
 
183
                /* Make sure the caller allocated memor for us */
 
184
 
 
185
                if (!ids[i]->sid) {
 
186
                        nt_status = NT_STATUS_INVALID_PARAMETER;
 
187
                        BAIL_ON_NTSTATUS_ERROR(nt_status);
 
188
                }
 
189
 
 
190
                /* If the domain hash doesn't find a SID in the table,
 
191
                   skip it */
 
192
 
 
193
                if (!hashed_domains[h_domain].sid)
 
194
                        continue;
 
195
 
 
196
                sid_copy(ids[i]->sid, hashed_domains[h_domain].sid);
 
197
                sid_append_rid(ids[i]->sid, h_rid);
 
198
                ids[i]->status = ID_MAPPED;
 
199
        }
 
200
 
 
201
done:
 
202
        return nt_status;
 
203
}
 
204
 
 
205
/*********************************************************************
 
206
 ********************************************************************/
 
207
 
 
208
static NTSTATUS sids_to_unixids(struct idmap_domain *dom,
 
209
                                struct id_map **ids)
 
210
{
 
211
        NTSTATUS nt_status = NT_STATUS_UNSUCCESSFUL;
 
212
        int i;
 
213
 
 
214
        /* initialize the status to avoid suprise */
 
215
        for (i = 0; ids[i]; i++) {
 
216
                ids[i]->status = ID_UNKNOWN;
 
217
        }
 
218
        
 
219
        nt_status = be_init(dom, NULL);
 
220
        BAIL_ON_NTSTATUS_ERROR(nt_status);
 
221
 
 
222
        if (!ids) {
 
223
                nt_status = NT_STATUS_INVALID_PARAMETER;
 
224
                BAIL_ON_NTSTATUS_ERROR(nt_status);
 
225
        }
 
226
 
 
227
        for (i=0; ids[i]; i++) {
 
228
                DOM_SID sid;
 
229
                uint32_t rid;
 
230
                uint32_t h_domain, h_rid;
 
231
 
 
232
                ids[i]->status = ID_UNMAPPED;
 
233
 
 
234
                sid_copy(&sid, ids[i]->sid);
 
235
                sid_split_rid(&sid, &rid);
 
236
 
 
237
                h_domain = hash_domain_sid(&sid);
 
238
                h_rid = hash_rid(rid);
 
239
 
 
240
                /* Check that both hashes are non-zero*/
 
241
 
 
242
                if (h_domain && h_rid) {
 
243
                        ids[i]->xid.id = combine_hashes(h_domain, h_rid);
 
244
                        ids[i]->status = ID_MAPPED;
 
245
                }
 
246
        }
 
247
 
 
248
done:
 
249
        return nt_status;
 
250
}
 
251
 
 
252
/*********************************************************************
 
253
 ********************************************************************/
 
254
 
 
255
static NTSTATUS be_close(struct idmap_domain *dom)
 
256
{
 
257
        if (hashed_domains)
 
258
                talloc_free(hashed_domains);
 
259
 
 
260
        return NT_STATUS_OK;
 
261
}
 
262
 
 
263
/*********************************************************************
 
264
 ********************************************************************/
 
265
 
 
266
static NTSTATUS nss_hash_init(struct nss_domain_entry *e )
 
267
{
 
268
        return be_init(NULL, NULL);
 
269
}
 
270
 
 
271
/**********************************************************************
 
272
 *********************************************************************/
 
273
 
 
274
static NTSTATUS nss_hash_get_info(struct nss_domain_entry *e,
 
275
                                    const DOM_SID *sid,
 
276
                                    TALLOC_CTX *ctx,
 
277
                                    ADS_STRUCT *ads,
 
278
                                    LDAPMessage *msg,
 
279
                                    char **homedir,
 
280
                                    char **shell,
 
281
                                    char **gecos,
 
282
                                    gid_t *p_gid )
 
283
{
 
284
        NTSTATUS nt_status = NT_STATUS_UNSUCCESSFUL;
 
285
 
 
286
        nt_status = nss_hash_init(e);
 
287
        BAIL_ON_NTSTATUS_ERROR(nt_status);
 
288
 
 
289
        if (!homedir || !shell || !gecos) {
 
290
                nt_status = NT_STATUS_INVALID_PARAMETER;
 
291
                BAIL_ON_NTSTATUS_ERROR(nt_status);
 
292
        }
 
293
 
 
294
        *homedir = talloc_strdup(ctx, lp_template_homedir());
 
295
        BAIL_ON_PTR_NT_ERROR(*homedir, nt_status);
 
296
 
 
297
        *shell   = talloc_strdup(ctx, lp_template_shell());
 
298
        BAIL_ON_PTR_NT_ERROR(*shell, nt_status);
 
299
 
 
300
        *gecos   = NULL;
 
301
 
 
302
        /* Initialize the gid so that the upper layer fills
 
303
           in the proper Windows primary group */
 
304
 
 
305
        if (*p_gid) {
 
306
                *p_gid = (gid_t)-1;
 
307
        }
 
308
 
 
309
done:
 
310
        return nt_status;
 
311
}
 
312
 
 
313
/**********************************************************************
 
314
 *********************************************************************/
 
315
 
 
316
static NTSTATUS nss_hash_map_to_alias(TALLOC_CTX *mem_ctx,
 
317
                                        struct nss_domain_entry *e,
 
318
                                        const char *name,
 
319
                                        char **alias)
 
320
{
 
321
        NTSTATUS nt_status = NT_STATUS_UNSUCCESSFUL;
 
322
        const char *value;
 
323
 
 
324
        value = talloc_asprintf(mem_ctx, "%s\\%s", e->domain, name);
 
325
        BAIL_ON_PTR_NT_ERROR(value, nt_status);
 
326
 
 
327
        nt_status = mapfile_lookup_key(mem_ctx, value, alias);
 
328
        BAIL_ON_NTSTATUS_ERROR(nt_status);
 
329
 
 
330
done:
 
331
        return nt_status;
 
332
}
 
333
 
 
334
/**********************************************************************
 
335
 *********************************************************************/
 
336
 
 
337
static NTSTATUS nss_hash_map_from_alias(TALLOC_CTX *mem_ctx,
 
338
                                          struct nss_domain_entry *e,
 
339
                                          const char *alias,
 
340
                                          char **name)
 
341
{
 
342
        return mapfile_lookup_value(mem_ctx, alias, name);
 
343
}
 
344
 
 
345
/**********************************************************************
 
346
 *********************************************************************/
 
347
 
 
348
static NTSTATUS nss_hash_close(void)
 
349
{
 
350
        return NT_STATUS_OK;
 
351
}
 
352
 
 
353
/*********************************************************************
 
354
 Dispatch Tables for IDMap and NssInfo Methods
 
355
********************************************************************/
 
356
 
 
357
static struct idmap_methods hash_idmap_methods = {
 
358
        .init            = be_init,
 
359
        .unixids_to_sids = unixids_to_sids,
 
360
        .sids_to_unixids = sids_to_unixids,
 
361
        .close_fn        = be_close
 
362
};
 
363
 
 
364
static struct nss_info_methods hash_nss_methods = {
 
365
        .init           = nss_hash_init,
 
366
        .get_nss_info   = nss_hash_get_info,
 
367
        .map_to_alias   = nss_hash_map_to_alias,
 
368
        .map_from_alias = nss_hash_map_from_alias,
 
369
        .close_fn       = nss_hash_close
 
370
};
 
371
 
 
372
/**********************************************************************
 
373
 Register with the idmap and idmap_nss subsystems. We have to protect
 
374
 against the idmap and nss_info interfaces being in a half-registered
 
375
 state.
 
376
 **********************************************************************/
 
377
 
 
378
NTSTATUS idmap_hash_init(void)
 
379
{
 
380
        static NTSTATUS idmap_status = NT_STATUS_UNSUCCESSFUL;
 
381
        static NTSTATUS nss_status = NT_STATUS_UNSUCCESSFUL;
 
382
 
 
383
        if ( !NT_STATUS_IS_OK(idmap_status) ) {
 
384
                idmap_status =  smb_register_idmap(SMB_IDMAP_INTERFACE_VERSION,
 
385
                                                   "hash", &hash_idmap_methods);
 
386
 
 
387
                if ( !NT_STATUS_IS_OK(idmap_status) ) {
 
388
                        DEBUG(0,("Failed to register hash idmap plugin.\n"));
 
389
                        return idmap_status;
 
390
                }
 
391
        }
 
392
 
 
393
        if ( !NT_STATUS_IS_OK(nss_status) ) {
 
394
                nss_status = smb_register_idmap_nss(SMB_NSS_INFO_INTERFACE_VERSION,
 
395
                                                    "hash", &hash_nss_methods);
 
396
                if ( !NT_STATUS_IS_OK(nss_status) ) {
 
397
                        DEBUG(0,("Failed to register hash idmap nss plugin.\n"));
 
398
                        return nss_status;
 
399
                }
 
400
        }
 
401
 
 
402
        return NT_STATUS_OK;
 
403
}