[PATCH] s4:kdc/smbd - Merged kdc_tcp_accept() and kpasswdd_tcp_accept().

[Endi S. Dewata]

---
 source4/kdc/kdc.c |   48 ++++++++++++++++--------------------------------
 1 files changed, 16 insertions(+), 32 deletions(-)

diff --git a/source4/kdc/kdc.c b/source4/kdc/kdc.c
index 5700e5b..b210707 100644
--- a/source4/kdc/kdc.c
+++ b/source4/kdc/kdc.c
@@ -355,9 +355,9 @@ static bool kdc_process(struct kdc_server *kdc,
 /*
   called when we get a new connection
 */
-static void kdc_tcp_generic_accept(struct stream_connection *conn, kdc_process_fn_t process_fn)
+static void kdc_tcp_accept(struct stream_connection *conn)
 {
-	struct kdc_server *kdc = talloc_get_type(conn->private_data, struct kdc_server);
+ 	struct kdc_socket *kdc_socket = talloc_get_type(conn->private_data, struct kdc_socket);
 	struct kdc_tcp_connection *kdcconn;
 
 	kdcconn = talloc_zero(conn, struct kdc_tcp_connection);
@@ -366,8 +366,8 @@ static void kdc_tcp_generic_accept(struct stream_connection *conn, kdc_process_f
 		return;
 	}
 	kdcconn->conn	 = conn;
-	kdcconn->kdc	 = kdc;
-	kdcconn->process = process_fn;
+	kdcconn->kdc	 = kdc_socket->kdc;
+	kdcconn->process = kdc_socket->process;
 	conn->private_data    = kdcconn;
 
 	kdcconn->packet = packet_init(kdcconn);
@@ -385,30 +385,6 @@ static void kdc_tcp_generic_accept(struct stream_connection *conn, kdc_process_f
 	packet_set_serialise(kdcconn->packet);
 }
 
-static void kdc_tcp_accept(struct stream_connection *conn)
-{
-	kdc_tcp_generic_accept(conn, kdc_process);
-}
-
-static const struct stream_server_ops kdc_tcp_stream_ops = {
-	.name			= "kdc_tcp",
-	.accept_connection	= kdc_tcp_accept,
-	.recv_handler		= kdc_tcp_recv_handler,
-	.send_handler		= kdc_tcp_send
-};
-
-static void kpasswdd_tcp_accept(struct stream_connection *conn)
-{
-	kdc_tcp_generic_accept(conn, kpasswdd_process);
-}
-
-static const struct stream_server_ops kpasswdd_tcp_stream_ops = {
-	.name			= "kpasswdd_tcp",
-	.accept_connection	= kpasswdd_tcp_accept,
-	.recv_handler		= kdc_tcp_recv_handler,
-	.send_handler		= kdc_tcp_send
-};
-
 /*
   start listening on the given address
 */
@@ -417,11 +393,11 @@ static NTSTATUS kdc_add_socket(struct kdc_server *kdc,
 			       const char *name,
 			       const char *address,
 			       uint16_t port,
-			       const struct stream_server_ops *tcp_stream_ops,
 			       kdc_process_fn_t process)
 {
  	struct kdc_socket *kdc_socket;
 	struct socket_address *socket_address;
+	struct stream_server_ops *tcp_stream_ops;
 	NTSTATUS status;
 
 	kdc_socket = talloc(kdc, struct kdc_socket);
@@ -455,13 +431,23 @@ static NTSTATUS kdc_add_socket(struct kdc_server *kdc,
 		return status;
 	}
 
+	tcp_stream_ops = talloc(kdc_socket, struct stream_server_ops);
+	NT_STATUS_HAVE_NO_MEMORY_AND_FREE(tcp_stream_ops, kdc_socket);
+
+	tcp_stream_ops->name = talloc_asprintf(kdc_socket, "%s_tcp", name);
+	NT_STATUS_HAVE_NO_MEMORY_AND_FREE(tcp_stream_ops->name, kdc_socket);
+
+	tcp_stream_ops->accept_connection = kdc_tcp_accept;
+	tcp_stream_ops->recv_handler = kdc_tcp_recv_handler;
+	tcp_stream_ops->send_handler = kdc_tcp_send;
+
 	status = stream_setup_socket(kdc->task->event_ctx, 
 				     kdc->task->lp_ctx,
 				     model_ops, 
 				     tcp_stream_ops, 
 				     "ip", address, &port, 
 				     lp_socket_options(kdc->task->lp_ctx), 
-				     kdc);
+				     kdc_socket);
 	if (!NT_STATUS_IS_OK(status)) {
 		DEBUG(0,("Failed to bind to %s:%u TCP - %s\n",
 			 address, port, nt_errstr(status)));
@@ -504,7 +490,6 @@ static NTSTATUS kdc_startup_interfaces(struct kdc_server *kdc, struct loadparm_c
 		if (kdc_port) {
 			status = kdc_add_socket(kdc, model_ops,
 					"kdc", address, kdc_port,
-					&kdc_tcp_stream_ops, 
 					kdc_process);
 			NT_STATUS_NOT_OK_RETURN(status);
 		}
@@ -512,7 +497,6 @@ static NTSTATUS kdc_startup_interfaces(struct kdc_server *kdc, struct loadparm_c
 		if (kpasswd_port) {
 			status = kdc_add_socket(kdc, model_ops,
 					"kpasswd", address, kpasswd_port,
-					&kpasswdd_tcp_stream_ops,
 					kpasswdd_process);
 			NT_STATUS_NOT_OK_RETURN(status);
 		}
-- 
1.6.0.6


------=_Part_22990_2109022267.1259086651927--