summaryrefslogtreecommitdiff
path: root/source4/rpc_server/dcerpc_sock.c
diff options
context:
space:
mode:
Diffstat (limited to 'source4/rpc_server/dcerpc_sock.c')
-rw-r--r--source4/rpc_server/dcerpc_sock.c328
1 files changed, 328 insertions, 0 deletions
diff --git a/source4/rpc_server/dcerpc_sock.c b/source4/rpc_server/dcerpc_sock.c
new file mode 100644
index 0000000000..d5646a7c7a
--- /dev/null
+++ b/source4/rpc_server/dcerpc_sock.c
@@ -0,0 +1,328 @@
+/*
+ Unix SMB/CIFS implementation.
+
+ server side dcerpc using various kinds of sockets (tcp, unix domain)
+
+ Copyright (C) Andrew Tridgell 2003
+ Copyright (C) Stefan (metze) Metzmacher 2004
+ Copyright (C) Jelmer Vernooij 2004
+
+ This program is free software; you can redistribute it and/or modify
+ it under the terms of the GNU General Public License as published by
+ the Free Software Foundation; either version 2 of the License, or
+ (at your option) any later version.
+
+ This program is distributed in the hope that it will be useful,
+ but WITHOUT ANY WARRANTY; without even the implied warranty of
+ MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+ GNU General Public License for more details.
+
+ You should have received a copy of the GNU General Public License
+ along with this program; if not, write to the Free Software
+ Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
+*/
+
+#include "includes.h"
+
+struct dcesrv_socket_context {
+ const struct dcesrv_endpoint *endpoint;
+ struct dcesrv_context *dcesrv_ctx;
+};
+
+/*
+ write_fn callback for dcesrv_output()
+*/
+static ssize_t dcerpc_write_fn(void *private, DATA_BLOB *out)
+{
+ NTSTATUS status;
+ struct socket_context *sock = private;
+ size_t sendlen;
+
+ status = socket_send(sock, sock, out, &sendlen, 0);
+ if (!NT_STATUS_IS_OK(status)) {
+ return -1;
+ }
+
+ return sendlen;
+}
+
+void dcesrv_terminate_connection(struct dcesrv_connection *dce_conn, const char *reason)
+{
+ server_terminate_connection(dce_conn->srv_conn, reason);
+}
+
+static void add_socket_rpc_unix(struct server_service *service,
+ const struct model_ops *model_ops,
+ struct dcesrv_context *dce_ctx,
+ struct dcesrv_endpoint *e)
+{
+ struct server_socket *sock;
+ struct dcesrv_socket_context *dcesrv_sock;
+ uint16_t port = 1;
+
+ sock = service_setup_socket(service,model_ops, "unix", e->ep_description.options[0], &port);
+ if (!sock) {
+ DEBUG(0,("service_setup_socket(path=%s) failed\n",e->ep_description.options[0]));
+ return;
+ }
+
+ dcesrv_sock = talloc_p(sock, struct dcesrv_socket_context);
+ if (!dcesrv_sock) {
+ DEBUG(0,("talloc_p(sock->mem_ctx, struct dcesrv_socket_context) failed\n"));
+ return;
+ }
+
+ /* remember the endpoint of this socket */
+ dcesrv_sock->endpoint = e;
+ dcesrv_sock->dcesrv_ctx = dce_ctx;
+
+ sock->private_data = dcesrv_sock;
+}
+
+static void add_socket_rpc_ncalrpc(struct server_service *service,
+ const struct model_ops *model_ops,
+ struct dcesrv_context *dce_ctx,
+ struct dcesrv_endpoint *e)
+{
+ struct server_socket *sock;
+ struct dcesrv_socket_context *dcesrv_sock;
+ uint16_t port = 1;
+ char *full_path;
+
+ if (!e->ep_description.options) {
+ e->ep_description.options = talloc_array_p(dce_ctx, const char *, 2);
+ e->ep_description.options[0] = NULL;
+ }
+
+ if (!e->ep_description.options[0]) {
+ /* No identifier specified: generate one */
+ e->ep_description.options[0] = generate_random_str(dce_ctx, 10);
+ e->ep_description.options[1] = NULL;
+ }
+
+ full_path = talloc_asprintf(dce_ctx, "%s/%s", lp_ncalrpc_dir(), e->ep_description.options[0]);
+
+ sock = service_setup_socket(service,model_ops, "unix", full_path, &port);
+ if (!sock) {
+ DEBUG(0,("service_setup_socket(identifier=%s,path=%s) failed\n",e->ep_description.options[0], full_path));
+ return;
+ }
+
+ dcesrv_sock = talloc_p(sock, struct dcesrv_socket_context);
+ if (!dcesrv_sock) {
+ DEBUG(0,("talloc_p(sock->mem_ctx, struct dcesrv_socket_context) failed\n"));
+ return;
+ }
+
+ /* remember the endpoint of this socket */
+ dcesrv_sock->endpoint = e;
+ dcesrv_sock->dcesrv_ctx = dce_ctx;
+
+ sock->private_data = dcesrv_sock;
+
+ return;
+}
+
+/*
+ add a socket address to the list of events, one event per dcerpc endpoint
+*/
+static void add_socket_rpc_tcp_iface(struct server_service *service,
+ const struct model_ops *model_ops,
+ struct dcesrv_context *dce_ctx,
+ struct dcesrv_endpoint *e,
+ struct in_addr *ifip)
+{
+ struct server_socket *sock;
+ struct dcesrv_socket_context *dcesrv_sock;
+ uint16_t port = 0;
+ const char *ip_str = talloc_strdup(service, inet_ntoa(*ifip));
+
+ if (e->ep_description.options && e->ep_description.options[0])
+ port = atoi(e->ep_description.options[0]);
+
+ sock = service_setup_socket(service,model_ops, "ipv4", ip_str, &port);
+ if (!sock) {
+ DEBUG(0,("service_setup_socket(port=%u) failed\n",port));
+ return;
+ }
+
+ /* And put the settings back into the binding. This will
+ * go away once we store the 'encoded' endpoint instead of a
+ * string describing it */
+ if (e->ep_description.options == NULL) {
+ e->ep_description.options = talloc_array_p(dce_ctx, const char *, 2);
+ e->ep_description.options[0] = talloc_asprintf(dce_ctx, "%d", port);
+ e->ep_description.options[1] = NULL;
+ }
+
+ dcesrv_sock = talloc_p(sock, struct dcesrv_socket_context);
+ if (!dcesrv_sock) {
+ DEBUG(0,("talloc_p(sock->mem_ctx, struct dcesrv_socket_context) failed\n"));
+ return;
+ }
+
+ /* remember the endpoint of this socket */
+ dcesrv_sock->endpoint = e;
+ dcesrv_sock->dcesrv_ctx = dce_ctx;
+
+ sock->private_data = dcesrv_sock;
+
+ talloc_free(ip_str);
+
+ return;
+}
+
+static void add_socket_rpc_tcp(struct server_service *service,
+ const struct model_ops *model_ops,
+ struct dcesrv_context *dce_ctx,
+ struct dcesrv_endpoint *e)
+{
+ /* Add TCP/IP sockets */
+ if (lp_interfaces() && lp_bind_interfaces_only()) {
+ int num_interfaces = iface_count();
+ int i;
+ for(i = 0; i < num_interfaces; i++) {
+ struct in_addr *ifip = iface_n_ip(i);
+ if (ifip == NULL) {
+ continue;
+ }
+ add_socket_rpc_tcp_iface(service, model_ops, dce_ctx, e, ifip);
+ }
+ } else {
+ struct in_addr *ifip;
+ ifip = interpret_addr2(dce_ctx, lp_socket_address());
+ add_socket_rpc_tcp_iface(service, model_ops, dce_ctx, e, ifip);
+ talloc_free(ifip);
+ }
+
+ return;
+}
+
+/****************************************************************************
+ Open the listening sockets for RPC over NCACN_IP_TCP/NCALRPC/NCACN_UNIX_STREAM
+****************************************************************************/
+void dcesrv_sock_init(struct server_service *service, const struct model_ops *model_ops, struct dcesrv_context *dce_ctx)
+{
+ struct dcesrv_endpoint *e;
+
+ DEBUG(1,("dcesrv_sock_init\n"));
+
+ for (e=dce_ctx->endpoint_list;e;e=e->next) {
+ switch (e->ep_description.transport) {
+ case NCACN_UNIX_STREAM:
+ add_socket_rpc_unix(service, model_ops, dce_ctx, e);
+ break;
+
+ case NCALRPC:
+ add_socket_rpc_ncalrpc(service, model_ops, dce_ctx, e);
+ break;
+
+ case NCACN_IP_TCP:
+ add_socket_rpc_tcp(service, model_ops, dce_ctx, e);
+ break;
+
+ default:
+ break;
+ }
+ }
+
+ return;
+}
+
+void dcesrv_sock_accept(struct server_connection *conn)
+{
+ NTSTATUS status;
+ struct dcesrv_socket_context *dcesrv_sock = conn->server_socket->private_data;
+ struct dcesrv_connection *dcesrv_conn = NULL;
+
+ DEBUG(5,("dcesrv_sock_accept\n"));
+
+ status = dcesrv_endpoint_connect(dcesrv_sock->dcesrv_ctx, dcesrv_sock->endpoint, &dcesrv_conn);
+ if (!NT_STATUS_IS_OK(status)) {
+ DEBUG(0,("dcesrv_sock_accept: dcesrv_endpoint_connect failed: %s\n",
+ nt_errstr(status)));
+ return;
+ }
+
+ dcesrv_conn->srv_conn = conn;
+
+ conn->private_data = dcesrv_conn;
+
+ return;
+}
+
+void dcesrv_sock_recv(struct server_connection *conn, time_t t, uint16_t flags)
+{
+ NTSTATUS status;
+ struct dcesrv_connection *dce_conn = conn->private_data;
+ DATA_BLOB tmp_blob;
+
+ DEBUG(10,("dcesrv_sock_recv\n"));
+
+ status = socket_recv(conn->socket, conn->socket, &tmp_blob, 0x4000, 0);
+ if (!NT_STATUS_IS_OK(status)) {
+ if (NT_STATUS_IS_ERR(status)) {
+ dcesrv_terminate_connection(dce_conn, "eof on socket");
+ return;
+ }
+ return;
+ }
+
+ status = dcesrv_input(dce_conn, &tmp_blob);
+ talloc_free(tmp_blob.data);
+ if (!NT_STATUS_IS_OK(status)) {
+ dcesrv_terminate_connection(dce_conn, "eof on socket");
+ return;
+ }
+
+ if (dce_conn->call_list && dce_conn->call_list->replies) {
+ conn->event.fde->flags |= EVENT_FD_WRITE;
+ }
+
+ return;
+}
+
+void dcesrv_sock_send(struct server_connection *conn, time_t t, uint16_t flags)
+{
+ struct dcesrv_connection *dce_conn = conn->private_data;
+ NTSTATUS status;
+
+ DEBUG(10,("dcesrv_sock_send\n"));
+
+ status = dcesrv_output(dce_conn, conn->socket, dcerpc_write_fn);
+ if (!NT_STATUS_IS_OK(status)) {
+ dcesrv_terminate_connection(dce_conn, "eof on socket");
+ return;
+ }
+
+ if (!dce_conn->call_list || !dce_conn->call_list->replies) {
+ conn->event.fde->flags &= ~EVENT_FD_WRITE;
+ }
+
+ return;
+}
+
+void dcesrv_sock_idle(struct server_connection *conn, time_t t)
+{
+ DEBUG(10,("dcesrv_sock_idle\n"));
+ conn->event.idle->next_event = t + 5;
+
+ return;
+}
+
+void dcesrv_sock_close(struct server_connection *conn, const char *reason)
+{
+ struct dcesrv_connection *dce_conn = conn->private_data;
+
+ DEBUG(5,("dcesrv_sock_close: %s\n",reason));
+
+ talloc_free(dce_conn);
+
+ return;
+}
+
+void dcesrv_sock_exit(struct server_service *service, const char *reason)
+{
+ DEBUG(1,("dcesrv_sock_exit: %s\n",reason));
+ return;
+}