svn commit: samba r24233 - in branches/SAMBA_3_2/source/smbd: .

[vlendec at samba.org]

Author: vlendec
Date: 2007-08-05 09:18:14 +0000 (Sun, 05 Aug 2007)
New Revision: 24233

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

Log:
Push reply_prep_legacy into handle_trans
Modified:
   branches/SAMBA_3_2/source/smbd/ipc.c


Changeset:
Modified: branches/SAMBA_3_2/source/smbd/ipc.c
===================================================================
--- branches/SAMBA_3_2/source/smbd/ipc.c	2007-08-05 09:01:15 UTC (rev 24232)
+++ branches/SAMBA_3_2/source/smbd/ipc.c	2007-08-05 09:18:14 UTC (rev 24233)
@@ -426,18 +426,23 @@
 }
 
 static NTSTATUS handle_trans(connection_struct *conn,
-				struct trans_state *state,
-				const char *inbuf,
-				char *outbuf,
-				int *outsize)
+			     struct smb_request *req,
+			     struct trans_state *state,
+			     int *outsize)
 {
 	char *local_machine_name;
 	int name_offset = 0;
+	char *inbuf, *outbuf;
+	int size, bufsize;
 
 	DEBUG(3,("trans <%s> data=%u params=%u setup=%u\n",
 		 state->name,(unsigned int)state->total_data,(unsigned int)state->total_param,
 		 (unsigned int)state->setup_count));
 
+	if (!reply_prep_legacy(req, &inbuf, &outbuf, &size, &bufsize)) {
+		return NT_STATUS_NO_MEMORY;
+	}
+
 	/*
 	 * WinCE wierdness....
 	 */
@@ -488,6 +493,8 @@
 	if (state->close_on_completion)
 		close_cnum(conn,state->vuid);
 
+	reply_post_legacy(req, *outsize);
+
 	return NT_STATUS_OK;
 }
 
@@ -504,8 +511,7 @@
 	unsigned int pscnt;
 	struct trans_state *state;
 	NTSTATUS result;
-	char *inbuf, *outbuf;
-	int size, bufsize;
+	int size;
 
 	START_PROFILE(SMBtrans);
 
@@ -654,21 +660,8 @@
 		return;
 	}
 
-	if (!(reply_prep_legacy(req, &inbuf, &outbuf, &size, &bufsize))) {
-		SAFE_FREE(state->data);
-		SAFE_FREE(state->param);
-		TALLOC_FREE(state);
-		reply_nterror(req, NT_STATUS_NO_MEMORY);
-		END_PROFILE(SMBtrans);
-		return;
-	}
+	result = handle_trans(conn, req, state, &outsize);
 
-	result = handle_trans(conn, state, inbuf, outbuf, &outsize);
-
-	if (NT_STATUS_IS_OK(result)) {
-		reply_post_legacy(req, outsize);
-	}
-
 	SAFE_FREE(state->data);
 	SAFE_FREE(state->param);
 	TALLOC_FREE(state);
@@ -709,8 +702,7 @@
 	unsigned int pcnt,poff,dcnt,doff,pdisp,ddisp;
 	struct trans_state *state;
 	NTSTATUS result;
-	char *inbuf, *outbuf;
-	int size, bufsize;
+	int size;
 
 	START_PROFILE(SMBtranss);
 
@@ -787,7 +779,7 @@
 		if (ddisp > state->total_data)
 			goto bad_param;
 		if ((smb_base(req->inbuf) + doff + dcnt
-		     > (char *)inbuf + size) ||
+		     > (char *)req->inbuf + size) ||
 		    (smb_base(req->inbuf) + doff + dcnt
 		     < smb_base(req->inbuf)))
 			goto bad_param;
@@ -804,23 +796,13 @@
 		return;
 	}
 
-	if (!reply_prep_legacy(req, &inbuf, &outbuf, &size, &bufsize)) {
-		reply_nterror(req, NT_STATUS_NO_MEMORY);
-		END_PROFILE(SMBtranss);
-		return;
-	}
-
 	/* construct_reply_common has done us the favor to pre-fill the
 	 * command field with SMBtranss which is wrong :-)
 	 */
 	SCVAL(req->outbuf,smb_com,SMBtrans);
 
-	result = handle_trans(conn, state, inbuf, outbuf, &outsize);
+	result = handle_trans(conn, req, state, &outsize);
 
-	if (NT_STATUS_IS_OK(result)) {
-		reply_post_legacy(req, outsize);
-	}
-
 	DLIST_REMOVE(conn->pending_trans, state);
 	SAFE_FREE(state->data);
 	SAFE_FREE(state->param);