svn commit: samba r8904 - in branches/SAMBA_4_0/source/libnet: .

[mimir]

Author: mimir
Date: 2005-08-02 01:12:58 +0000 (Tue, 02 Aug 2005)
New Revision: 8904

WebSVN: http://websvn.samba.org/cgi-bin/viewcvs.cgi?view=rev&root=samba&rev=8904

Log:
Split off the query stage as a separate function.


rafal


Modified:
   branches/SAMBA_4_0/source/libnet/userman.c


Changeset:
Modified: branches/SAMBA_4_0/source/libnet/userman.c
===================================================================
--- branches/SAMBA_4_0/source/libnet/userman.c  2005-08-01 22:19:28 UTC (rev 
8903)
+++ branches/SAMBA_4_0/source/libnet/userman.c  2005-08-02 01:12:58 UTC (rev 
8904)
@@ -591,10 +591,6 @@
        c->status = dcerpc_ndr_request_recv(s->req);
        NT_STATUS_NOT_OK_RETURN(c->status);
 
-       if (s->stage == USERMOD_QUERY) {
-               s->info = *s->queryuser.out.info;
-       }
-
        /* Prepare UserInfo level and data based on bitmask field */
        s->change.fields = usermod_setfields(s, &level, i);
 
@@ -620,6 +616,35 @@
 
 
 /**
+ * Stage 2a (optional): Query the user information
+ */
+static NTSTATUS usermod_query(struct composite_context *c,
+                             struct usermod_state *s)
+{
+       union samr_UserInfo *i = &s->info;
+       uint16_t level;
+
+       c->status = dcerpc_ndr_request_recv(s->req);
+       NT_STATUS_NOT_OK_RETURN(c->status);
+
+       s->info = *s->queryuser.out.info;
+
+       s->change.fields = usermod_setfields(s, &level, i);
+
+       s->setuser.in.user_handle  = &s->user_handle;
+       s->setuser.in.level        = level;
+       s->setuser.in.info         = i;
+       
+       s->req = dcerpc_samr_SetUserInfo_send(s->pipe, c, &s->setuser);
+
+       s->req->async.callback = usermod_handler;
+       s->req->async.private  = c;
+
+       return NT_STATUS_OK;
+}
+
+
+/**
  * Stage 3: Set new user account data
  */
 static NTSTATUS usermod_modify(struct composite_context *c,
@@ -653,10 +678,13 @@
                break;
 
        case USERMOD_OPEN:
-       case USERMOD_QUERY:
                c->status = usermod_open(c, s);
                break;
 
+       case USERMOD_QUERY:
+               c->status = usermod_query(c, s);
+               break;
+
        case USERMOD_MODIFY:
                c->status = usermod_modify(c, s);
                break;