← Back to branch summary

~zulcss/samba/server-dailies-3.4

« back to all changes in this revision

Viewing changes to source4/libnet/libnet_user.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:
source4/libnet/libnet_user.c
 
1
/* 
 
2
   Unix SMB/CIFS implementation.
 
3
   
 
4
   Copyright (C) Rafal Szczesniak  2005
 
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 "libnet/libnet.h"
 
23
#include "libcli/composite/composite.h"
 
24
#include "auth/credentials/credentials.h"
 
25
#include "librpc/ndr/libndr.h"
 
26
#include "librpc/gen_ndr/samr.h"
 
27
#include "librpc/gen_ndr/ndr_samr_c.h"
 
28
#include "librpc/gen_ndr/lsa.h"
 
29
#include "librpc/gen_ndr/ndr_lsa_c.h"
 
30
#include "libcli/security/security.h"
 
31
 
 
32
 
 
33
struct create_user_state {
 
34
        struct libnet_CreateUser r;
 
35
        struct libnet_DomainOpen domain_open;
 
36
        struct libnet_rpc_useradd user_add;
 
37
        struct libnet_context *ctx;
 
38
 
 
39
        /* information about the progress */
 
40
        void (*monitor_fn)(struct monitor_msg *);
 
41
};
 
42
 
 
43
 
 
44
static void continue_rpc_useradd(struct composite_context *ctx);
 
45
static void continue_domain_open_create(struct composite_context *ctx);
 
46
 
 
47
 
 
48
/**
 
49
 * Sends request to create user account
 
50
 *
 
51
 * @param ctx initialised libnet context
 
52
 * @param mem_ctx memory context of this call
 
53
 * @param r pointer to a structure containing arguments and results of this call
 
54
 * @param monitor function pointer for receiving monitor messages
 
55
 * @return compostite context of this request
 
56
 */
 
57
struct composite_context* libnet_CreateUser_send(struct libnet_context *ctx,
 
58
                                                 TALLOC_CTX *mem_ctx,
 
59
                                                 struct libnet_CreateUser *r,
 
60
                                                 void (*monitor)(struct monitor_msg*))
 
61
{
 
62
        struct composite_context *c;
 
63
        struct create_user_state *s;
 
64
        struct composite_context *create_req;
 
65
        bool prereq_met = false;
 
66
 
 
67
        /* composite context allocation and setup */
 
68
        c = composite_create(mem_ctx, ctx->event_ctx);
 
69
        if (c == NULL) return NULL;
 
70
 
 
71
        s = talloc_zero(c, struct create_user_state);
 
72
        if (composite_nomem(s, c)) return c;
 
73
 
 
74
        c->private_data = s;
 
75
 
 
76
        /* store arguments in the state structure */
 
77
        s->ctx = ctx;
 
78
        s->r   = *r;
 
79
        ZERO_STRUCT(s->r.out);
 
80
 
 
81
        /* prerequisite: make sure the domain is opened */
 
82
        prereq_met = samr_domain_opened(ctx, s->r.in.domain_name, &c, &s->domain_open,
 
83
                                        continue_domain_open_create, monitor);
 
84
        if (!prereq_met) return c;
 
85
 
 
86
        /* prepare arguments for useradd call */
 
87
        s->user_add.in.username       = r->in.user_name;
 
88
        s->user_add.in.domain_handle  = ctx->samr.handle;
 
89
 
 
90
        /* send the request */
 
91
        create_req = libnet_rpc_useradd_send(ctx->samr.pipe, &s->user_add, monitor);
 
92
        if (composite_nomem(create_req, c)) return c;
 
93
 
 
94
        /* set the next stage */
 
95
        composite_continue(c, create_req, continue_rpc_useradd, c);
 
96
        return c;
 
97
}
 
98
 
 
99
 
 
100
/*
 
101
 * Stage 0.5 (optional): receive result of domain open request
 
102
 * and send useradd request
 
103
 */
 
104
static void continue_domain_open_create(struct composite_context *ctx)
 
105
{
 
106
        struct composite_context *c;
 
107
        struct create_user_state *s;
 
108
        struct composite_context *create_req;
 
109
        struct monitor_msg msg;
 
110
 
 
111
        c = talloc_get_type(ctx->async.private_data, struct composite_context);
 
112
        s = talloc_get_type(c->private_data, struct create_user_state);
 
113
 
 
114
        /* receive result of DomainOpen call */
 
115
        c->status = libnet_DomainOpen_recv(ctx, s->ctx, c, &s->domain_open);
 
116
        if (!composite_is_ok(c)) return;
 
117
 
 
118
        /* send monitor message */
 
119
        if (s->monitor_fn) s->monitor_fn(&msg);
 
120
        
 
121
        /* prepare arguments for useradd call */
 
122
        s->user_add.in.username       = s->r.in.user_name;
 
123
        s->user_add.in.domain_handle  = s->ctx->samr.handle;
 
124
 
 
125
        /* send the request */
 
126
        create_req = libnet_rpc_useradd_send(s->ctx->samr.pipe, &s->user_add, s->monitor_fn);
 
127
        if (composite_nomem(create_req, c)) return;
 
128
 
 
129
        /* set the next stage */
 
130
        composite_continue(c, create_req, continue_rpc_useradd, c);
 
131
}
 
132
 
 
133
 
 
134
/*
 
135
 * Stage 1: receive result of useradd call
 
136
 */
 
137
static void continue_rpc_useradd(struct composite_context *ctx)
 
138
{
 
139
        struct composite_context *c;
 
140
        struct create_user_state *s;
 
141
        struct monitor_msg msg;
 
142
 
 
143
        c = talloc_get_type(ctx->async.private_data, struct composite_context);
 
144
        s = talloc_get_type(c->private_data, struct create_user_state);
 
145
        
 
146
        /* receive result of the call */
 
147
        c->status = libnet_rpc_useradd_recv(ctx, c, &s->user_add);
 
148
        if (!composite_is_ok(c)) return;
 
149
 
 
150
        /* send monitor message */
 
151
        if (s->monitor_fn) s->monitor_fn(&msg);
 
152
 
 
153
        /* we're done */
 
154
        composite_done(c);
 
155
}
 
156
 
 
157
 
 
158
/**
 
159
 * Receive result of CreateUser call
 
160
 *
 
161
 * @param c composite context returned by send request routine
 
162
 * @param mem_ctx memory context of this call
 
163
 * @param r pointer to a structure containing arguments and result of this call
 
164
 * @return nt status
 
165
 */
 
166
NTSTATUS libnet_CreateUser_recv(struct composite_context *c, TALLOC_CTX *mem_ctx,
 
167
                                struct libnet_CreateUser *r)
 
168
{
 
169
        NTSTATUS status;
 
170
        struct create_user_state *s;
 
171
 
 
172
        r->out.error_string = NULL;
 
173
 
 
174
        /* wait for result of async request and check status code */
 
175
        status = composite_wait(c);
 
176
        if (!NT_STATUS_IS_OK(status)) {
 
177
                s = talloc_get_type(c->private_data, struct create_user_state);
 
178
                r->out.error_string = talloc_strdup(mem_ctx, nt_errstr(status));
 
179
        }
 
180
 
 
181
        return status;
 
182
}
 
183
 
 
184
 
 
185
/**
 
186
 * Synchronous version of CreateUser call
 
187
 *
 
188
 * @param ctx initialised libnet context
 
189
 * @param mem_ctx memory context of this call
 
190
 * @param r pointer to a structure containing arguments and result of this call
 
191
 * @return nt status
 
192
 */
 
193
NTSTATUS libnet_CreateUser(struct libnet_context *ctx, TALLOC_CTX *mem_ctx,
 
194
                           struct libnet_CreateUser *r)
 
195
{
 
196
        struct composite_context *c;
 
197
 
 
198
        c = libnet_CreateUser_send(ctx, mem_ctx, r, NULL);
 
199
        return libnet_CreateUser_recv(c, mem_ctx, r);
 
200
}
 
201
 
 
202
 
 
203
struct delete_user_state {
 
204
        struct libnet_DeleteUser r;
 
205
        struct libnet_context *ctx;
 
206
        struct libnet_DomainOpen domain_open;
 
207
        struct libnet_rpc_userdel user_del;
 
208
 
 
209
        /* information about the progress */
 
210
        void (*monitor_fn)(struct monitor_msg *);
 
211
};
 
212
 
 
213
 
 
214
static void continue_rpc_userdel(struct composite_context *ctx);
 
215
static void continue_domain_open_delete(struct composite_context *ctx);
 
216
 
 
217
 
 
218
/**
 
219
 * Sends request to delete user account
 
220
 *
 
221
 * @param ctx initialised libnet context
 
222
 * @param mem_ctx memory context of this call
 
223
 * @param r pointer to structure containing arguments and result of this call
 
224
 * @param monitor function pointer for receiving monitor messages
 
225
 */
 
226
struct composite_context *libnet_DeleteUser_send(struct libnet_context *ctx,
 
227
                                                 TALLOC_CTX *mem_ctx,
 
228
                                                 struct libnet_DeleteUser *r,
 
229
                                                 void (*monitor)(struct monitor_msg*))
 
230
{
 
231
        struct composite_context *c;
 
232
        struct delete_user_state *s;
 
233
        struct composite_context *delete_req;
 
234
        bool prereq_met = false;
 
235
 
 
236
        /* composite context allocation and setup */
 
237
        c = composite_create(mem_ctx, ctx->event_ctx);
 
238
        if (c == NULL) return NULL;
 
239
 
 
240
        s = talloc_zero(c, struct delete_user_state);
 
241
        if (composite_nomem(s, c)) return c;
 
242
 
 
243
        c->private_data = s;
 
244
 
 
245
        /* store arguments in state structure */
 
246
        s->ctx = ctx;
 
247
        s->r   = *r;
 
248
        ZERO_STRUCT(s->r.out);
 
249
        
 
250
        /* prerequisite: make sure the domain is opened before proceeding */
 
251
        prereq_met = samr_domain_opened(ctx, s->r.in.domain_name, &c, &s->domain_open,
 
252
                                        continue_domain_open_delete, monitor);
 
253
        if (!prereq_met) return c;
 
254
 
 
255
        /* prepare arguments for userdel call */
 
256
        s->user_del.in.username       = r->in.user_name;
 
257
        s->user_del.in.domain_handle  = ctx->samr.handle;
 
258
 
 
259
        /* send request */
 
260
        delete_req = libnet_rpc_userdel_send(ctx->samr.pipe, &s->user_del, monitor);
 
261
        if (composite_nomem(delete_req, c)) return c;
 
262
        
 
263
        /* set the next stage */
 
264
        composite_continue(c, delete_req, continue_rpc_userdel, c);
 
265
        return c;
 
266
}
 
267
 
 
268
 
 
269
/*
 
270
 * Stage 0.5 (optional): receive result of domain open request
 
271
 * and send useradd request
 
272
 */
 
273
static void continue_domain_open_delete(struct composite_context *ctx)
 
274
{
 
275
        struct composite_context *c;
 
276
        struct delete_user_state *s;
 
277
        struct composite_context *delete_req;
 
278
        struct monitor_msg msg;
 
279
 
 
280
        c = talloc_get_type(ctx->async.private_data, struct composite_context);
 
281
        s = talloc_get_type(c->private_data, struct delete_user_state);
 
282
 
 
283
        /* receive result of DomainOpen call */
 
284
        c->status = libnet_DomainOpen_recv(ctx, s->ctx, c, &s->domain_open);
 
285
        if (!composite_is_ok(c)) return;
 
286
        
 
287
        /* send monitor message */
 
288
        if (s->monitor_fn) s->monitor_fn(&msg);
 
289
 
 
290
        /* prepare arguments for userdel call */
 
291
        s->user_del.in.username       = s->r.in.user_name;
 
292
        s->user_del.in.domain_handle  = s->ctx->samr.handle;
 
293
 
 
294
        /* send request */
 
295
        delete_req = libnet_rpc_userdel_send(s->ctx->samr.pipe, &s->user_del, s->monitor_fn);
 
296
        if (composite_nomem(delete_req, c)) return;
 
297
 
 
298
        /* set the next stage */
 
299
        composite_continue(c, delete_req, continue_rpc_userdel, c);
 
300
}
 
301
 
 
302
 
 
303
/*
 
304
 * Stage 1: receive result of userdel call and finish the composite function
 
305
 */
 
306
static void continue_rpc_userdel(struct composite_context *ctx)
 
307
{
 
308
        struct composite_context *c;
 
309
        struct delete_user_state *s;
 
310
        struct monitor_msg msg;
 
311
 
 
312
        c = talloc_get_type(ctx->async.private_data, struct composite_context);
 
313
        s = talloc_get_type(c->private_data, struct delete_user_state);
 
314
 
 
315
        /* receive result of userdel call */
 
316
        c->status = libnet_rpc_userdel_recv(ctx, c, &s->user_del);
 
317
        if (!composite_is_ok(c)) return;
 
318
 
 
319
        /* send monitor message */
 
320
        if (s->monitor_fn) s->monitor_fn(&msg);
 
321
 
 
322
        /* we're done */
 
323
        composite_done(c);
 
324
}
 
325
 
 
326
 
 
327
/**
 
328
 * Receives result of asynchronous DeleteUser call
 
329
 *
 
330
 * @param c composite context returned by async DeleteUser call
 
331
 * @param mem_ctx memory context of this call
 
332
 * @param r pointer to structure containing arguments and result
 
333
 */
 
334
NTSTATUS libnet_DeleteUser_recv(struct composite_context *c, TALLOC_CTX *mem_ctx,
 
335
                                struct libnet_DeleteUser *r)
 
336
{
 
337
        NTSTATUS status;
 
338
        struct delete_user_state *s;
 
339
 
 
340
        r->out.error_string = NULL;
 
341
 
 
342
        /* wait for result of async request and check status code */
 
343
        status = composite_wait(c);
 
344
        if (!NT_STATUS_IS_OK(status)) {
 
345
                s = talloc_get_type(c->private_data, struct delete_user_state);
 
346
                r->out.error_string = talloc_steal(mem_ctx, s->r.out.error_string);
 
347
        }
 
348
        
 
349
        return status;
 
350
}
 
351
 
 
352
 
 
353
/**
 
354
 * Synchronous version of DeleteUser call
 
355
 *
 
356
 * @param ctx initialised libnet context
 
357
 * @param mem_ctx memory context of this call
 
358
 * @param r pointer to structure containing arguments and result
 
359
 */
 
360
NTSTATUS libnet_DeleteUser(struct libnet_context *ctx, TALLOC_CTX *mem_ctx,
 
361
                           struct libnet_DeleteUser *r)
 
362
{
 
363
        struct composite_context *c;
 
364
        
 
365
        c = libnet_DeleteUser_send(ctx, mem_ctx, r, NULL);
 
366
        return libnet_DeleteUser_recv(c, mem_ctx, r);
 
367
}
 
368
 
 
369
 
 
370
struct modify_user_state {
 
371
        struct libnet_ModifyUser r;
 
372
        struct libnet_context *ctx;
 
373
        struct libnet_DomainOpen domain_open;
 
374
        struct libnet_rpc_userinfo user_info;
 
375
        struct libnet_rpc_usermod user_mod;
 
376
 
 
377
        void (*monitor_fn)(struct monitor_msg *);
 
378
};
 
379
 
 
380
 
 
381
static void continue_rpc_usermod(struct composite_context *ctx);
 
382
static void continue_domain_open_modify(struct composite_context *ctx);
 
383
static NTSTATUS set_user_changes(TALLOC_CTX *mem_ctx, struct usermod_change *mod,
 
384
                                 struct libnet_rpc_userinfo *info, struct libnet_ModifyUser *r);
 
385
static void continue_rpc_userinfo(struct composite_context *ctx);
 
386
 
 
387
 
 
388
/**
 
389
 * Sends request to modify user account
 
390
 *
 
391
 * @param ctx initialised libnet context
 
392
 * @param mem_ctx memory context of this call
 
393
 * @param r pointer to structure containing arguments and result of this call
 
394
 * @param monitor function pointer for receiving monitor messages
 
395
 */
 
396
struct composite_context *libnet_ModifyUser_send(struct libnet_context *ctx,
 
397
                                                 TALLOC_CTX *mem_ctx,
 
398
                                                 struct libnet_ModifyUser *r,
 
399
                                                 void (*monitor)(struct monitor_msg*))
 
400
{
 
401
        const uint16_t level = 21;
 
402
        struct composite_context *c;
 
403
        struct modify_user_state *s;
 
404
        struct composite_context *userinfo_req;
 
405
        bool prereq_met = false;
 
406
 
 
407
        c = composite_create(mem_ctx, ctx->event_ctx);
 
408
        if (c == NULL) return NULL;
 
409
 
 
410
        s = talloc_zero(c, struct modify_user_state);
 
411
        if (composite_nomem(s, c)) return c;
 
412
 
 
413
        c->private_data = s;
 
414
 
 
415
        s->ctx = ctx;
 
416
        s->r = *r;
 
417
 
 
418
        prereq_met = samr_domain_opened(ctx, s->r.in.domain_name, &c, &s->domain_open,
 
419
                                        continue_domain_open_modify, monitor);
 
420
        if (!prereq_met) return c;
 
421
 
 
422
        s->user_info.in.username      = r->in.user_name;
 
423
        s->user_info.in.domain_handle = ctx->samr.handle;
 
424
        s->user_info.in.level         = level;
 
425
 
 
426
        userinfo_req = libnet_rpc_userinfo_send(ctx->samr.pipe, &s->user_info, monitor);
 
427
        if (composite_nomem(userinfo_req, c)) return c;
 
428
 
 
429
        composite_continue(c, userinfo_req, continue_rpc_userinfo, c);
 
430
        return c;
 
431
}
 
432
 
 
433
 
 
434
/*
 
435
 * Stage 0.5 (optional): receive result of domain open request
 
436
 * and send userinfo request
 
437
 */
 
438
static void continue_domain_open_modify(struct composite_context *ctx)
 
439
{
 
440
        const uint16_t level = 21;
 
441
        struct composite_context *c;
 
442
        struct modify_user_state *s;
 
443
        struct composite_context *userinfo_req;
 
444
        struct monitor_msg msg;
 
445
 
 
446
        c = talloc_get_type(ctx->async.private_data, struct composite_context);
 
447
        s = talloc_get_type(c->private_data, struct modify_user_state);
 
448
 
 
449
        c->status = libnet_DomainOpen_recv(ctx, s->ctx, c, &s->domain_open);
 
450
        if (!composite_is_ok(c)) return;
 
451
 
 
452
        if (s->monitor_fn) s->monitor_fn(&msg);
 
453
        
 
454
        s->user_info.in.domain_handle  = s->ctx->samr.handle;
 
455
        s->user_info.in.username       = s->r.in.user_name;
 
456
        s->user_info.in.level          = level;
 
457
 
 
458
        userinfo_req = libnet_rpc_userinfo_send(s->ctx->samr.pipe, &s->user_info, s->monitor_fn);
 
459
        if (composite_nomem(userinfo_req, c)) return;
 
460
        
 
461
        composite_continue(c, userinfo_req, continue_rpc_userinfo, c);
 
462
}
 
463
 
 
464
 
 
465
/*
 
466
 * Stage 1: receive result of userinfo call, prepare user changes
 
467
 * (set the fields a caller required to change) and send usermod request
 
468
 */
 
469
static void continue_rpc_userinfo(struct composite_context *ctx)
 
470
{
 
471
        struct composite_context *c;
 
472
        struct modify_user_state *s;
 
473
        struct composite_context *usermod_req;
 
474
 
 
475
        c = talloc_get_type(ctx->async.private_data, struct composite_context);
 
476
        s = talloc_get_type(c->private_data, struct modify_user_state);
 
477
 
 
478
        c->status = libnet_rpc_userinfo_recv(ctx, c, &s->user_info);
 
479
        if (!composite_is_ok(c)) return;
 
480
 
 
481
        s->user_mod.in.domain_handle = s->ctx->samr.handle;
 
482
        s->user_mod.in.username      = s->r.in.user_name;
 
483
 
 
484
        c->status = set_user_changes(c, &s->user_mod.in.change, &s->user_info, &s->r);
 
485
 
 
486
        usermod_req = libnet_rpc_usermod_send(s->ctx->samr.pipe, &s->user_mod, s->monitor_fn);
 
487
        if (composite_nomem(usermod_req, c)) return;
 
488
 
 
489
        composite_continue(c, usermod_req, continue_rpc_usermod, c);
 
490
}
 
491
 
 
492
 
 
493
/*
 
494
 * Prepare user changes: compare userinfo result to requested changes and
 
495
 * set the field values and flags accordingly for user modify call
 
496
 */
 
497
static NTSTATUS set_user_changes(TALLOC_CTX *mem_ctx, struct usermod_change *mod,
 
498
                                 struct libnet_rpc_userinfo *info, struct libnet_ModifyUser *r)
 
499
{
 
500
        struct samr_UserInfo21 *user;
 
501
 
 
502
        if (mod == NULL || info == NULL || r == NULL || info->in.level != 21) {
 
503
                return NT_STATUS_INVALID_PARAMETER;
 
504
        }
 
505
 
 
506
        user = &info->out.info.info21;
 
507
        mod->fields = 0;        /* reset flag field before setting individual flags */
 
508
 
 
509
        /* account name change */
 
510
        SET_FIELD_LSA_STRING(r->in, user, mod, account_name, USERMOD_FIELD_ACCOUNT_NAME);
 
511
 
 
512
        /* full name change */
 
513
        SET_FIELD_LSA_STRING(r->in, user, mod, full_name, USERMOD_FIELD_FULL_NAME);
 
514
 
 
515
        /* description change */
 
516
        SET_FIELD_LSA_STRING(r->in, user, mod, description, USERMOD_FIELD_DESCRIPTION);
 
517
 
 
518
        /* comment change */
 
519
        SET_FIELD_LSA_STRING(r->in, user, mod, comment, USERMOD_FIELD_COMMENT);
 
520
 
 
521
        /* home directory change */
 
522
        SET_FIELD_LSA_STRING(r->in, user, mod, home_directory, USERMOD_FIELD_HOME_DIRECTORY);
 
523
 
 
524
        /* home drive change */
 
525
        SET_FIELD_LSA_STRING(r->in, user, mod, home_drive, USERMOD_FIELD_HOME_DRIVE);
 
526
 
 
527
        /* logon script change */
 
528
        SET_FIELD_LSA_STRING(r->in, user, mod, logon_script, USERMOD_FIELD_LOGON_SCRIPT);
 
529
 
 
530
        /* profile path change */
 
531
        SET_FIELD_LSA_STRING(r->in, user, mod, profile_path, USERMOD_FIELD_PROFILE_PATH);
 
532
 
 
533
        /* account expiry change */
 
534
        SET_FIELD_NTTIME(r->in, user, mod, acct_expiry, USERMOD_FIELD_ACCT_EXPIRY);
 
535
 
 
536
        /* account flags change */
 
537
        SET_FIELD_ACCT_FLAGS(r->in, user, mod, acct_flags, USERMOD_FIELD_ACCT_FLAGS);
 
538
 
 
539
        return NT_STATUS_OK;
 
540
}
 
541
 
 
542
 
 
543
/*
 
544
 * Stage 2: receive result of usermod request and finish the composite function
 
545
 */
 
546
static void continue_rpc_usermod(struct composite_context *ctx)
 
547
{
 
548
        struct composite_context *c;
 
549
        struct modify_user_state *s;
 
550
        struct monitor_msg msg;
 
551
 
 
552
        c = talloc_get_type(ctx->async.private_data, struct composite_context);
 
553
        s = talloc_get_type(c->private_data, struct modify_user_state);
 
554
        
 
555
        c->status = libnet_rpc_usermod_recv(ctx, c, &s->user_mod);
 
556
        if (!composite_is_ok(c)) return;
 
557
        
 
558
        if (s->monitor_fn) s->monitor_fn(&msg);
 
559
        composite_done(c);
 
560
}
 
561
 
 
562
 
 
563
/**
 
564
 * Receive result of ModifyUser call
 
565
 *
 
566
 * @param c composite context returned by send request routine
 
567
 * @param mem_ctx memory context of this call
 
568
 * @param r pointer to a structure containing arguments and result of this call
 
569
 * @return nt status
 
570
 */
 
571
NTSTATUS libnet_ModifyUser_recv(struct composite_context *c, TALLOC_CTX *mem_ctx,
 
572
                                struct libnet_ModifyUser *r)
 
573
{
 
574
        NTSTATUS status = composite_wait(c);
 
575
        return status;
 
576
}
 
577
 
 
578
 
 
579
/**
 
580
 * Synchronous version of ModifyUser call
 
581
 *
 
582
 * @param ctx initialised libnet context
 
583
 * @param mem_ctx memory context of this call
 
584
 * @param r pointer to a structure containing arguments and result of this call
 
585
 * @return nt status
 
586
 */
 
587
NTSTATUS libnet_ModifyUser(struct libnet_context *ctx, TALLOC_CTX *mem_ctx,
 
588
                           struct libnet_ModifyUser *r)
 
589
{
 
590
        struct composite_context *c;
 
591
 
 
592
        c = libnet_ModifyUser_send(ctx, mem_ctx, r, NULL);
 
593
        return libnet_ModifyUser_recv(c, mem_ctx, r);
 
594
}
 
595
 
 
596
 
 
597
struct user_info_state {
 
598
        struct libnet_context *ctx;
 
599
        const char *domain_name;
 
600
        enum libnet_UserInfo_level level;
 
601
        const char *user_name;
 
602
        const char *sid_string;
 
603
        struct libnet_LookupName lookup;
 
604
        struct libnet_DomainOpen domopen;
 
605
        struct libnet_rpc_userinfo userinfo;
 
606
 
 
607
        /* information about the progress */
 
608
        void (*monitor_fn)(struct monitor_msg *);
 
609
};
 
610
 
 
611
 
 
612
static void continue_name_found(struct composite_context *ctx);
 
613
static void continue_domain_open_info(struct composite_context *ctx);
 
614
static void continue_info_received(struct composite_context *ctx);
 
615
 
 
616
 
 
617
/**
 
618
 * Sends request to get user account information
 
619
 *
 
620
 * @param ctx initialised libnet context
 
621
 * @param mem_ctx memory context of this call
 
622
 * @param r pointer to a structure containing arguments and results of this call
 
623
 * @param monitor function pointer for receiving monitor messages
 
624
 * @return compostite context of this request
 
625
 */
 
626
struct composite_context* libnet_UserInfo_send(struct libnet_context *ctx,
 
627
                                               TALLOC_CTX *mem_ctx,
 
628
                                               struct libnet_UserInfo *r,
 
629
                                               void (*monitor)(struct monitor_msg*))
 
630
{
 
631
        struct composite_context *c;
 
632
        struct user_info_state *s;
 
633
        struct composite_context *lookup_req, *info_req;
 
634
        bool prereq_met = false;
 
635
 
 
636
        /* composite context allocation and setup */
 
637
        c = composite_create(mem_ctx, ctx->event_ctx);
 
638
        if (c == NULL) return NULL;
 
639
 
 
640
        s = talloc_zero(c, struct user_info_state);
 
641
        if (composite_nomem(s, c)) return c;
 
642
 
 
643
        c->private_data = s;
 
644
 
 
645
        /* store arguments in the state structure */
 
646
        s->monitor_fn = monitor;
 
647
        s->ctx = ctx;
 
648
        s->domain_name = talloc_strdup(c, r->in.domain_name);
 
649
        s->level = r->in.level;
 
650
        switch (s->level) {
 
651
        case USER_INFO_BY_NAME:
 
652
                s->user_name = talloc_strdup(c, r->in.data.user_name);
 
653
                s->sid_string = NULL;
 
654
                break;
 
655
        case USER_INFO_BY_SID:
 
656
                s->user_name = NULL;
 
657
                s->sid_string = dom_sid_string(c, r->in.data.user_sid);
 
658
                break;
 
659
        }
 
660
 
 
661
        /* prerequisite: make sure the domain is opened */
 
662
        prereq_met = samr_domain_opened(ctx, s->domain_name, &c, &s->domopen,
 
663
                                        continue_domain_open_info, monitor);
 
664
        if (!prereq_met) return c;
 
665
 
 
666
        switch (s->level) {
 
667
        case USER_INFO_BY_NAME:
 
668
                /* prepare arguments for LookupName call */
 
669
                s->lookup.in.domain_name = s->domain_name;
 
670
                s->lookup.in.name        = s->user_name;
 
671
 
 
672
                /* send the request */
 
673
                lookup_req = libnet_LookupName_send(ctx, c, &s->lookup,
 
674
                                                    s->monitor_fn);
 
675
                if (composite_nomem(lookup_req, c)) return c;
 
676
 
 
677
                /* set the next stage */
 
678
                composite_continue(c, lookup_req, continue_name_found, c);
 
679
                break;
 
680
        case USER_INFO_BY_SID:
 
681
                /* prepare arguments for UserInfo call */
 
682
                s->userinfo.in.domain_handle = s->ctx->samr.handle;
 
683
                s->userinfo.in.sid = s->sid_string;
 
684
                s->userinfo.in.level = 21;
 
685
 
 
686
                /* send the request */
 
687
                info_req = libnet_rpc_userinfo_send(s->ctx->samr.pipe,
 
688
                                                    &s->userinfo,
 
689
                                                    s->monitor_fn);
 
690
                if (composite_nomem(info_req, c)) return c;
 
691
 
 
692
                /* set the next stage */
 
693
                composite_continue(c, info_req, continue_info_received, c);
 
694
                break;
 
695
        }
 
696
 
 
697
        return c;
 
698
}
 
699
 
 
700
 
 
701
/*
 
702
 * Stage 0.5 (optional): receive result of domain open request
 
703
 * and send LookupName request
 
704
 */
 
705
static void continue_domain_open_info(struct composite_context *ctx)
 
706
{
 
707
        struct composite_context *c;
 
708
        struct user_info_state *s;
 
709
        struct composite_context *lookup_req, *info_req;
 
710
        struct monitor_msg msg;
 
711
 
 
712
        c = talloc_get_type(ctx->async.private_data, struct composite_context);
 
713
        s = talloc_get_type(c->private_data, struct user_info_state);
 
714
 
 
715
        /* receive result of DomainOpen call */
 
716
        c->status = libnet_DomainOpen_recv(ctx, s->ctx, c, &s->domopen);
 
717
        if (!composite_is_ok(c)) return;
 
718
 
 
719
        /* send monitor message */
 
720
        if (s->monitor_fn) s->monitor_fn(&msg);
 
721
 
 
722
        switch (s->level) {
 
723
        case USER_INFO_BY_NAME:
 
724
                /* prepare arguments for LookupName call */
 
725
                s->lookup.in.domain_name = s->domain_name;
 
726
                s->lookup.in.name        = s->user_name;
 
727
 
 
728
                /* send the request */
 
729
                lookup_req = libnet_LookupName_send(s->ctx, c, &s->lookup, s->monitor_fn);
 
730
                if (composite_nomem(lookup_req, c)) return;
 
731
 
 
732
                /* set the next stage */
 
733
                composite_continue(c, lookup_req, continue_name_found, c);
 
734
                break;
 
735
 
 
736
        case USER_INFO_BY_SID:
 
737
                /* prepare arguments for UserInfo call */
 
738
                s->userinfo.in.domain_handle = s->ctx->samr.handle;
 
739
                s->userinfo.in.sid = s->sid_string;
 
740
                s->userinfo.in.level = 21;
 
741
 
 
742
                /* send the request */
 
743
                info_req = libnet_rpc_userinfo_send(s->ctx->samr.pipe,
 
744
                                                    &s->userinfo,
 
745
                                                    s->monitor_fn);
 
746
                if (composite_nomem(info_req, c)) return;
 
747
 
 
748
                /* set the next stage */
 
749
                composite_continue(c, info_req, continue_info_received, c);
 
750
                break;
 
751
        }
 
752
}
 
753
 
 
754
 
 
755
/*
 
756
 * Stage 1: receive the name (if found) and send userinfo request
 
757
 */
 
758
static void continue_name_found(struct composite_context *ctx)
 
759
{
 
760
        struct composite_context *c;
 
761
        struct user_info_state *s;
 
762
        struct composite_context *info_req;
 
763
 
 
764
        c = talloc_get_type(ctx->async.private_data, struct composite_context);
 
765
        s = talloc_get_type(c->private_data, struct user_info_state);
 
766
 
 
767
        /* receive result of LookupName call */
 
768
        c->status = libnet_LookupName_recv(ctx, c, &s->lookup);
 
769
        if (!composite_is_ok(c)) return;
 
770
 
 
771
        /* we're only interested in user accounts this time */
 
772
        if (s->lookup.out.sid_type != SID_NAME_USER) {
 
773
                composite_error(c, NT_STATUS_NO_SUCH_USER);
 
774
                return;
 
775
        }
 
776
 
 
777
        /* prepare arguments for UserInfo call */
 
778
        s->userinfo.in.domain_handle = s->ctx->samr.handle;
 
779
        s->userinfo.in.sid = s->lookup.out.sidstr;
 
780
        s->userinfo.in.level = 21;
 
781
 
 
782
        /* send the request */
 
783
        info_req = libnet_rpc_userinfo_send(s->ctx->samr.pipe, &s->userinfo, s->monitor_fn);
 
784
        if (composite_nomem(info_req, c)) return;
 
785
 
 
786
        /* set the next stage */
 
787
        composite_continue(c, info_req, continue_info_received, c);
 
788
}
 
789
 
 
790
 
 
791
/*
 
792
 * Stage 2: receive user account information and finish the composite function
 
793
 */
 
794
static void continue_info_received(struct composite_context *ctx)
 
795
{
 
796
        struct composite_context *c;
 
797
        struct user_info_state *s;
 
798
 
 
799
        c = talloc_get_type(ctx->async.private_data, struct composite_context);
 
800
        s = talloc_get_type(c->private_data, struct user_info_state);
 
801
        
 
802
        /* receive result of userinfo call */
 
803
        c->status = libnet_rpc_userinfo_recv(ctx, c, &s->userinfo);
 
804
        if (!composite_is_ok(c)) return;
 
805
 
 
806
        composite_done(c);
 
807
}
 
808
 
 
809
 
 
810
/**
 
811
 * Receive result of UserInfo call
 
812
 *
 
813
 * @param c composite context returned by send request routine
 
814
 * @param mem_ctx memory context of this call
 
815
 * @param r pointer to a structure containing arguments and result of this call
 
816
 * @return nt status
 
817
 */
 
818
NTSTATUS libnet_UserInfo_recv(struct composite_context *c, TALLOC_CTX *mem_ctx,
 
819
                              struct libnet_UserInfo *r)
 
820
{
 
821
        NTSTATUS status;
 
822
        struct user_info_state *s;
 
823
 
 
824
        status = composite_wait(c);
 
825
 
 
826
        if (NT_STATUS_IS_OK(status) && r != NULL) {
 
827
                struct samr_UserInfo21 *info;
 
828
 
 
829
                s = talloc_get_type(c->private_data, struct user_info_state);
 
830
                info = &s->userinfo.out.info.info21;
 
831
 
 
832
                r->out.user_sid = dom_sid_add_rid(mem_ctx, s->ctx->samr.sid, info->rid);
 
833
                r->out.primary_group_sid = dom_sid_add_rid(mem_ctx, s->ctx->samr.sid, info->primary_gid);
 
834
 
 
835
                /* string fields */
 
836
                r->out.account_name   = talloc_steal(mem_ctx, info->account_name.string);
 
837
                r->out.full_name      = talloc_steal(mem_ctx, info->full_name.string);
 
838
                r->out.description    = talloc_steal(mem_ctx, info->description.string);
 
839
                r->out.home_directory = talloc_steal(mem_ctx, info->home_directory.string);
 
840
                r->out.home_drive     = talloc_steal(mem_ctx, info->home_drive.string);
 
841
                r->out.comment        = talloc_steal(mem_ctx, info->comment.string);
 
842
                r->out.logon_script   = talloc_steal(mem_ctx, info->logon_script.string);
 
843
                r->out.profile_path   = talloc_steal(mem_ctx, info->profile_path.string);
 
844
 
 
845
                /* time fields (allocation) */
 
846
                r->out.acct_expiry           = talloc(mem_ctx, struct timeval);
 
847
                r->out.allow_password_change = talloc(mem_ctx, struct timeval);
 
848
                r->out.force_password_change = talloc(mem_ctx, struct timeval);
 
849
                r->out.last_logon            = talloc(mem_ctx, struct timeval);
 
850
                r->out.last_logoff           = talloc(mem_ctx, struct timeval);
 
851
                r->out.last_password_change  = talloc(mem_ctx, struct timeval);
 
852
                
 
853
                /* time fields (converting) */
 
854
                nttime_to_timeval(r->out.acct_expiry, info->acct_expiry);
 
855
                nttime_to_timeval(r->out.allow_password_change, info->allow_password_change);
 
856
                nttime_to_timeval(r->out.force_password_change, info->force_password_change);
 
857
                nttime_to_timeval(r->out.last_logon, info->last_logon);
 
858
                nttime_to_timeval(r->out.last_logoff, info->last_logoff);
 
859
                nttime_to_timeval(r->out.last_password_change, info->last_password_change);
 
860
 
 
861
                /* flag and number fields */
 
862
                r->out.acct_flags = info->acct_flags;
 
863
 
 
864
                r->out.error_string = talloc_strdup(mem_ctx, "Success");
 
865
 
 
866
        } else {
 
867
                r->out.error_string = talloc_asprintf(mem_ctx, "Error: %s", nt_errstr(status));
 
868
        }
 
869
 
 
870
        talloc_free(c);
 
871
        
 
872
        return status;
 
873
}
 
874
 
 
875
 
 
876
/**
 
877
 * Synchronous version of UserInfo call
 
878
 *
 
879
 * @param ctx initialised libnet context
 
880
 * @param mem_ctx memory context of this call
 
881
 * @param r pointer to a structure containing arguments and result of this call
 
882
 * @return nt status
 
883
 */
 
884
NTSTATUS libnet_UserInfo(struct libnet_context *ctx, TALLOC_CTX *mem_ctx,
 
885
                         struct libnet_UserInfo *r)
 
886
{
 
887
        struct composite_context *c;
 
888
        
 
889
        c = libnet_UserInfo_send(ctx, mem_ctx, r, NULL);
 
890
        return libnet_UserInfo_recv(c, mem_ctx, r);
 
891
}
 
892
 
 
893
 
 
894
struct userlist_state {
 
895
        struct libnet_context *ctx;
 
896
        const char *domain_name;
 
897
        struct lsa_DomainInfo dominfo;
 
898
        int page_size;
 
899
        uint32_t resume_index;
 
900
        struct userlist *users;
 
901
        uint32_t count;
 
902
 
 
903
        struct libnet_DomainOpen domain_open;
 
904
        struct lsa_QueryInfoPolicy query_domain;
 
905
        struct samr_EnumDomainUsers user_list;
 
906
 
 
907
        void (*monitor_fn)(struct monitor_msg*);
 
908
};
 
909
 
 
910
 
 
911
static void continue_lsa_domain_opened(struct composite_context *ctx);
 
912
static void continue_domain_queried(struct rpc_request *req);
 
913
static void continue_samr_domain_opened(struct composite_context *ctx);
 
914
static void continue_users_enumerated(struct rpc_request *req);
 
915
 
 
916
 
 
917
/**
 
918
 * Sends request to list (enumerate) user accounts
 
919
 *
 
920
 * @param ctx initialised libnet context
 
921
 * @param mem_ctx memory context of this call
 
922
 * @param r pointer to structure containing arguments and results of this call
 
923
 * @param monitor function pointer for receiving monitor messages
 
924
 * @return compostite context of this request
 
925
 */
 
926
struct composite_context* libnet_UserList_send(struct libnet_context *ctx,
 
927
                                               TALLOC_CTX *mem_ctx,
 
928
                                               struct libnet_UserList *r,
 
929
                                               void (*monitor)(struct monitor_msg*))
 
930
{
 
931
        struct composite_context *c;
 
932
        struct userlist_state *s;
 
933
        struct rpc_request *query_req;
 
934
        bool prereq_met = false;
 
935
 
 
936
        /* composite context allocation and setup */
 
937
        c = composite_create(mem_ctx, ctx->event_ctx);
 
938
        if (c == NULL) return NULL;
 
939
 
 
940
        s = talloc_zero(c, struct userlist_state);
 
941
        if (composite_nomem(s, c)) return c;
 
942
 
 
943
        c->private_data = s;
 
944
 
 
945
        /* store the arguments in the state structure */
 
946
        s->ctx          = ctx;
 
947
        s->page_size    = r->in.page_size;
 
948
        s->resume_index = (uint32_t)r->in.resume_index;
 
949
        s->domain_name  = talloc_strdup(c, r->in.domain_name);
 
950
        s->monitor_fn   = monitor;
 
951
 
 
952
        /* make sure we have lsa domain handle before doing anything */
 
953
        prereq_met = lsa_domain_opened(ctx, s->domain_name, &c, &s->domain_open,
 
954
                                       continue_lsa_domain_opened, monitor);
 
955
        if (!prereq_met) return c;
 
956
 
 
957
        /* prepare arguments of QueryDomainInfo call */
 
958
        s->query_domain.in.handle = &ctx->lsa.handle;
 
959
        s->query_domain.in.level  = LSA_POLICY_INFO_DOMAIN;
 
960
        s->query_domain.out.info  = talloc_zero(c, union lsa_PolicyInformation *);
 
961
        if (composite_nomem(s->query_domain.out.info, c)) return c;
 
962
 
 
963
        /* send the request */
 
964
        query_req = dcerpc_lsa_QueryInfoPolicy_send(ctx->lsa.pipe, c, &s->query_domain);
 
965
        if (composite_nomem(query_req, c)) return c;
 
966
 
 
967
        composite_continue_rpc(c, query_req, continue_domain_queried, c);
 
968
        return c;
 
969
}
 
970
 
 
971
 
 
972
/*
 
973
 * Stage 0.5 (optional): receive lsa domain handle and send
 
974
 * request to query domain info
 
975
 */
 
976
static void continue_lsa_domain_opened(struct composite_context *ctx)
 
977
{
 
978
        struct composite_context *c;
 
979
        struct userlist_state *s;
 
980
        struct rpc_request *query_req;
 
981
        
 
982
        c = talloc_get_type(ctx->async.private_data, struct composite_context);
 
983
        s = talloc_get_type(c->private_data, struct userlist_state);
 
984
        
 
985
        /* receive lsa domain handle */
 
986
        c->status = libnet_DomainOpen_recv(ctx, s->ctx, c, &s->domain_open);
 
987
        if (!composite_is_ok(c)) return;
 
988
 
 
989
        /* prepare arguments of QueryDomainInfo call */
 
990
        s->query_domain.in.handle = &s->ctx->lsa.handle;
 
991
        s->query_domain.in.level  = LSA_POLICY_INFO_DOMAIN;
 
992
        s->query_domain.out.info  = talloc_zero(c, union lsa_PolicyInformation *);
 
993
        if (composite_nomem(s->query_domain.out.info, c)) return;
 
994
 
 
995
        /* send the request */
 
996
        query_req = dcerpc_lsa_QueryInfoPolicy_send(s->ctx->lsa.pipe, c, &s->query_domain);
 
997
        if (composite_nomem(query_req, c)) return;
 
998
 
 
999
        composite_continue_rpc(c, query_req, continue_domain_queried, c);
 
1000
}
 
1001
 
 
1002
 
 
1003
/*
 
1004
 * Stage 1: receive domain info and request to enum users,
 
1005
 * provided a valid samr handle is opened
 
1006
 */
 
1007
static void continue_domain_queried(struct rpc_request *req)
 
1008
{
 
1009
        struct composite_context *c;
 
1010
        struct userlist_state *s;
 
1011
        struct rpc_request *enum_req;
 
1012
        bool prereq_met = false;
 
1013
        
 
1014
        c = talloc_get_type(req->async.private_data, struct composite_context);
 
1015
        s = talloc_get_type(c->private_data, struct userlist_state);
 
1016
 
 
1017
        /* receive result of rpc request */
 
1018
        c->status = dcerpc_ndr_request_recv(req);
 
1019
        if (!composite_is_ok(c)) return;
 
1020
 
 
1021
        /* get the returned domain info */
 
1022
        s->dominfo = (*s->query_domain.out.info)->domain;
 
1023
 
 
1024
        /* make sure we have samr domain handle before continuing */
 
1025
        prereq_met = samr_domain_opened(s->ctx, s->domain_name, &c, &s->domain_open,
 
1026
                                        continue_samr_domain_opened, s->monitor_fn);
 
1027
        if (!prereq_met) return;
 
1028
 
 
1029
        /* prepare arguments of EnumDomainUsers call */
 
1030
        s->user_list.in.domain_handle = &s->ctx->samr.handle;
 
1031
        s->user_list.in.max_size = s->page_size;
 
1032
        s->user_list.in.resume_handle = &s->resume_index;
 
1033
        s->user_list.in.acct_flags = ACB_NORMAL;
 
1034
        s->user_list.out.resume_handle = &s->resume_index;
 
1035
        s->user_list.out.num_entries = talloc(s, uint32_t);
 
1036
        if (composite_nomem(s->user_list.out.num_entries, c)) return;
 
1037
        s->user_list.out.sam = talloc(s, struct samr_SamArray *);
 
1038
        if (composite_nomem(s->user_list.out.sam, c)) return;
 
1039
 
 
1040
        /* send the request */
 
1041
        enum_req = dcerpc_samr_EnumDomainUsers_send(s->ctx->samr.pipe, c, &s->user_list);
 
1042
        if (composite_nomem(enum_req, c)) return;
 
1043
 
 
1044
        composite_continue_rpc(c, enum_req, continue_users_enumerated, c);
 
1045
}
 
1046
 
 
1047
 
 
1048
/*
 
1049
 * Stage 1.5 (optional): receive samr domain handle
 
1050
 * and request to enumerate accounts
 
1051
 */
 
1052
static void continue_samr_domain_opened(struct composite_context *ctx)
 
1053
{
 
1054
        struct composite_context *c;
 
1055
        struct userlist_state *s;
 
1056
        struct rpc_request *enum_req;
 
1057
 
 
1058
        c = talloc_get_type(ctx->async.private_data, struct composite_context);
 
1059
        s = talloc_get_type(c->private_data, struct userlist_state);
 
1060
 
 
1061
        /* receive samr domain handle */
 
1062
        c->status = libnet_DomainOpen_recv(ctx, s->ctx, c, &s->domain_open);
 
1063
        if (!composite_is_ok(c)) return;
 
1064
 
 
1065
        /* prepare arguments of EnumDomainUsers call */
 
1066
        s->user_list.in.domain_handle = &s->ctx->samr.handle;
 
1067
        s->user_list.in.max_size = s->page_size;
 
1068
        s->user_list.in.resume_handle = &s->resume_index;
 
1069
        s->user_list.in.acct_flags = ACB_NORMAL;
 
1070
        s->user_list.out.resume_handle = &s->resume_index;
 
1071
        s->user_list.out.sam = talloc(s, struct samr_SamArray *);
 
1072
        if (composite_nomem(s->user_list.out.sam, c)) return;
 
1073
        s->user_list.out.num_entries = talloc(s, uint32_t);
 
1074
        if (composite_nomem(s->user_list.out.num_entries, c)) return;
 
1075
        
 
1076
        /* send the request */
 
1077
        enum_req = dcerpc_samr_EnumDomainUsers_send(s->ctx->samr.pipe, c, &s->user_list);
 
1078
        if (composite_nomem(enum_req, c)) return;
 
1079
 
 
1080
        composite_continue_rpc(c, enum_req, continue_users_enumerated, c);
 
1081
}
 
1082
 
 
1083
 
 
1084
/*
 
1085
 * Stage 2: receive enumerated users and their rids
 
1086
 */
 
1087
static void continue_users_enumerated(struct rpc_request *req)
 
1088
{
 
1089
        struct composite_context *c;
 
1090
        struct userlist_state *s;
 
1091
        int i;
 
1092
 
 
1093
        c = talloc_get_type(req->async.private_data, struct composite_context);
 
1094
        s = talloc_get_type(c->private_data, struct userlist_state);
 
1095
 
 
1096
        /* receive result of rpc request */
 
1097
        c->status = dcerpc_ndr_request_recv(req);
 
1098
        if (!composite_is_ok(c)) return;
 
1099
 
 
1100
        /* get the actual status of the rpc call result
 
1101
           (instead of rpc layer status) */
 
1102
        c->status = s->user_list.out.result;
 
1103
 
 
1104
        /* we're interested in status "ok" as well as two
 
1105
           enum-specific status codes */
 
1106
        if (NT_STATUS_IS_OK(c->status) ||
 
1107
            NT_STATUS_EQUAL(c->status, STATUS_MORE_ENTRIES) ||
 
1108
            NT_STATUS_EQUAL(c->status, NT_STATUS_NO_MORE_ENTRIES)) {
 
1109
 
 
1110
                /* get enumerated accounts counter and resume handle (the latter allows
 
1111
                   making subsequent call to continue enumeration) */
 
1112
                s->resume_index = *s->user_list.out.resume_handle;
 
1113
                s->count        = *s->user_list.out.num_entries;
 
1114
                
 
1115
                /* prepare returned user accounts array */
 
1116
                s->users        = talloc_array(c, struct userlist, (*s->user_list.out.sam)->count);
 
1117
                if (composite_nomem(s->users, c)) return;
 
1118
 
 
1119
                for (i = 0; i < (*s->user_list.out.sam)->count; i++) {
 
1120
                        struct dom_sid *user_sid;
 
1121
                        struct samr_SamEntry *entry = &(*s->user_list.out.sam)->entries[i];
 
1122
                        struct dom_sid *domain_sid = (*s->query_domain.out.info)->domain.sid;
 
1123
                        
 
1124
                        /* construct user sid from returned rid and queried domain sid */
 
1125
                        user_sid = dom_sid_add_rid(c, domain_sid, entry->idx);
 
1126
                        if (composite_nomem(user_sid, c)) return;
 
1127
                        
 
1128
                        /* username */
 
1129
                        s->users[i].username = talloc_strdup(c, entry->name.string);
 
1130
                        if (composite_nomem(s->users[i].username, c)) return;
 
1131
 
 
1132
                        /* sid string */
 
1133
                        s->users[i].sid = dom_sid_string(c, user_sid);
 
1134
                        if (composite_nomem(s->users[i].sid, c)) return;
 
1135
                }
 
1136
                
 
1137
                /* that's it */
 
1138
                composite_done(c);
 
1139
 
 
1140
        } else {
 
1141
                /* something went wrong */
 
1142
                composite_error(c, c->status);
 
1143
        }
 
1144
}
 
1145
 
 
1146
 
 
1147
/**
 
1148
 * Receive result of UserList call
 
1149
 *
 
1150
 * @param c composite context returned by send request routine
 
1151
 * @param mem_ctx memory context of this call
 
1152
 * @param r pointer to structure containing arguments and result of this call
 
1153
 * @return nt status
 
1154
 */
 
1155
NTSTATUS libnet_UserList_recv(struct composite_context* c, TALLOC_CTX *mem_ctx,
 
1156
                              struct libnet_UserList *r)
 
1157
{
 
1158
        NTSTATUS status;
 
1159
        struct userlist_state *s;
 
1160
 
 
1161
        if (c == NULL || mem_ctx == NULL || r == NULL) {
 
1162
                return NT_STATUS_INVALID_PARAMETER;
 
1163
        }
 
1164
        
 
1165
        status = composite_wait(c);
 
1166
        if (NT_STATUS_IS_OK(status) ||
 
1167
            NT_STATUS_EQUAL(status, STATUS_MORE_ENTRIES) ||
 
1168
            NT_STATUS_EQUAL(status, NT_STATUS_NO_MORE_ENTRIES)) {
 
1169
                
 
1170
                s = talloc_get_type(c->private_data, struct userlist_state);
 
1171
                
 
1172
                /* get results from composite context */
 
1173
                r->out.count = s->count;
 
1174
                r->out.resume_index = s->resume_index;
 
1175
                r->out.users = talloc_steal(mem_ctx, s->users);
 
1176
                
 
1177
                if (NT_STATUS_IS_OK(status)) {
 
1178
                        r->out.error_string = talloc_strdup(mem_ctx, "Success");
 
1179
                } else {
 
1180
                        /* success, but we're not done yet */
 
1181
                        r->out.error_string = talloc_asprintf(mem_ctx, "Success (status: %s)",
 
1182
                                                              nt_errstr(status));
 
1183
                }
 
1184
 
 
1185
        } else {
 
1186
                r->out.error_string = talloc_asprintf(mem_ctx, "Error: %s", nt_errstr(status));
 
1187
        }
 
1188
 
 
1189
        return status;
 
1190
}
 
1191
 
 
1192
 
 
1193
/**
 
1194
 * Synchronous version of UserList call
 
1195
 *
 
1196
 * @param ctx initialised libnet context
 
1197
 * @param mem_ctx memory context of this call
 
1198
 * @param r pointer to structure containing arguments and result of this call
 
1199
 * @return nt status
 
1200
 */
 
1201
NTSTATUS libnet_UserList(struct libnet_context *ctx,
 
1202
                         TALLOC_CTX *mem_ctx,
 
1203
                         struct libnet_UserList *r)
 
1204
{
 
1205
        struct composite_context *c;
 
1206
 
 
1207
        c = libnet_UserList_send(ctx, mem_ctx, r, NULL);
 
1208
        return libnet_UserList_recv(c, mem_ctx, r);
 
1209
}