2 Unix SMB/CIFS implementation.
4 server side dcerpc using various kinds of sockets (tcp, unix domain)
6 Copyright (C) Andrew Tridgell 2003
7 Copyright (C) Stefan (metze) Metzmacher 2004-2005
8 Copyright (C) Jelmer Vernooij 2004
10 This program is free software; you can redistribute it and/or modify
11 it under the terms of the GNU General Public License as published by
12 the Free Software Foundation; either version 2 of the License, or
13 (at your option) any later version.
15 This program is distributed in the hope that it will be useful,
16 but WITHOUT ANY WARRANTY; without even the implied warranty of
17 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
18 GNU General Public License for more details.
20 You should have received a copy of the GNU General Public License
21 along with this program; if not, write to the Free Software
22 Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
27 #include "rpc_server/dcerpc_server.h"
28 #include "smbd/service_stream.h"
30 struct dcesrv_socket_context {
31 const struct dcesrv_endpoint *endpoint;
32 struct dcesrv_context *dcesrv_ctx;
36 write_fn callback for dcesrv_output()
38 static ssize_t dcerpc_write_fn(void *private, DATA_BLOB *out)
41 struct socket_context *sock = private;
44 status = socket_send(sock, out, &sendlen, 0);
45 if (NT_STATUS_IS_ERR(status)) {
52 static void dcesrv_terminate_connection(struct dcesrv_connection *dce_conn, const char *reason)
54 stream_terminate_connection(dce_conn->srv_conn, reason);
58 void dcesrv_sock_accept(struct stream_connection *srv_conn)
61 struct dcesrv_socket_context *dcesrv_sock =
62 talloc_get_type(srv_conn->private, struct dcesrv_socket_context);
63 struct dcesrv_connection *dcesrv_conn = NULL;
65 status = dcesrv_endpoint_connect(dcesrv_sock->dcesrv_ctx,
67 dcesrv_sock->endpoint,
70 if (!NT_STATUS_IS_OK(status)) {
71 DEBUG(0,("dcesrv_sock_accept: dcesrv_endpoint_connect failed: %s\n",
76 srv_conn->private = dcesrv_conn;
81 void dcesrv_sock_recv(struct stream_connection *conn, uint16_t flags)
84 struct dcesrv_connection *dce_conn = talloc_get_type(conn->private, struct dcesrv_connection);
88 tmp_blob = data_blob_talloc(conn->socket, NULL, 0x1000);
89 if (tmp_blob.data == NULL) {
90 dcesrv_terminate_connection(dce_conn, "out of memory");
94 status = socket_recv(conn->socket, tmp_blob.data, tmp_blob.length, &nread, 0);
95 if (NT_STATUS_IS_ERR(status)) {
96 dcesrv_terminate_connection(dce_conn, nt_errstr(status));
100 talloc_free(tmp_blob.data);
104 tmp_blob.length = nread;
106 status = dcesrv_input(dce_conn, &tmp_blob);
107 talloc_free(tmp_blob.data);
109 if (!NT_STATUS_IS_OK(status)) {
110 dcesrv_terminate_connection(dce_conn, nt_errstr(status));
114 if (dce_conn->call_list && dce_conn->call_list->replies) {
115 EVENT_FD_WRITEABLE(conn->event.fde);
119 void dcesrv_sock_send(struct stream_connection *conn, uint16_t flags)
121 struct dcesrv_connection *dce_conn = talloc_get_type(conn->private, struct dcesrv_connection);
124 status = dcesrv_output(dce_conn, conn->socket, dcerpc_write_fn);
125 if (!NT_STATUS_IS_OK(status)) {
126 dcesrv_terminate_connection(dce_conn, "eof on socket");
130 if (!dce_conn->call_list || !dce_conn->call_list->replies) {
131 EVENT_FD_NOT_WRITEABLE(conn->event.fde);
136 static const struct stream_server_ops dcesrv_stream_ops = {
138 .accept_connection = dcesrv_sock_accept,
139 .recv_handler = dcesrv_sock_recv,
140 .send_handler = dcesrv_sock_send,
145 static NTSTATUS add_socket_rpc_unix(struct dcesrv_context *dce_ctx, struct dcesrv_endpoint *e,
146 struct event_context *event_ctx, const struct model_ops *model_ops)
148 struct dcesrv_socket_context *dcesrv_sock;
152 dcesrv_sock = talloc(dce_ctx, struct dcesrv_socket_context);
153 NT_STATUS_HAVE_NO_MEMORY(dcesrv_sock);
155 /* remember the endpoint of this socket */
156 dcesrv_sock->endpoint = e;
157 dcesrv_sock->dcesrv_ctx = talloc_reference(dcesrv_sock, dce_ctx);
159 status = stream_setup_socket(event_ctx, model_ops, &dcesrv_stream_ops,
160 "unix", e->ep_description.endpoint, &port,
162 if (!NT_STATUS_IS_OK(status)) {
163 DEBUG(0,("service_setup_stream_socket(path=%s) failed - %s\n",
164 e->ep_description.endpoint, nt_errstr(status)));
170 static NTSTATUS add_socket_rpc_ncalrpc(struct dcesrv_context *dce_ctx, struct dcesrv_endpoint *e,
171 struct event_context *event_ctx, const struct model_ops *model_ops)
173 struct dcesrv_socket_context *dcesrv_sock;
178 if (!e->ep_description.endpoint) {
179 /* No identifier specified: use DEFAULT.
180 * DO NOT hardcode this value anywhere else. Rather, specify
181 * no endpoint and let the epmapper worry about it. */
182 e->ep_description.endpoint = talloc_strdup(dce_ctx, "DEFAULT");
185 full_path = talloc_asprintf(dce_ctx, "%s/%s", lp_ncalrpc_dir(), e->ep_description.endpoint);
187 dcesrv_sock = talloc(dce_ctx, struct dcesrv_socket_context);
188 NT_STATUS_HAVE_NO_MEMORY(dcesrv_sock);
190 /* remember the endpoint of this socket */
191 dcesrv_sock->endpoint = e;
192 dcesrv_sock->dcesrv_ctx = dce_ctx;
194 status = stream_setup_socket(event_ctx, model_ops, &dcesrv_stream_ops,
195 "unix", full_path, &port, dcesrv_sock);
196 if (!NT_STATUS_IS_OK(status)) {
197 DEBUG(0,("service_setup_stream_socket(identifier=%s,path=%s) failed - %s\n",
198 e->ep_description.endpoint, full_path, nt_errstr(status)));
204 add a socket address to the list of events, one event per dcerpc endpoint
206 static NTSTATUS add_socket_rpc_tcp_iface(struct dcesrv_context *dce_ctx, struct dcesrv_endpoint *e,
207 struct event_context *event_ctx, const struct model_ops *model_ops,
210 struct dcesrv_socket_context *dcesrv_sock;
214 if (e->ep_description.endpoint) {
215 port = atoi(e->ep_description.endpoint);
218 dcesrv_sock = talloc(dce_ctx, struct dcesrv_socket_context);
219 NT_STATUS_HAVE_NO_MEMORY(dcesrv_sock);
221 /* remember the endpoint of this socket */
222 dcesrv_sock->endpoint = e;
223 dcesrv_sock->dcesrv_ctx = dce_ctx;
225 status = stream_setup_socket(event_ctx, model_ops, &dcesrv_stream_ops,
226 "ipv4", address, &port, dcesrv_sock);
227 if (!NT_STATUS_IS_OK(status)) {
228 DEBUG(0,("service_setup_stream_socket(address=%s,port=%u) failed - %s\n",
229 address, port, nt_errstr(status)));
232 if (e->ep_description.endpoint == NULL) {
233 e->ep_description.endpoint = talloc_asprintf(dce_ctx, "%d", port);
239 static NTSTATUS add_socket_rpc_tcp(struct dcesrv_context *dce_ctx, struct dcesrv_endpoint *e,
240 struct event_context *event_ctx, const struct model_ops *model_ops)
244 /* Add TCP/IP sockets */
245 if (lp_interfaces() && lp_bind_interfaces_only()) {
246 int num_interfaces = iface_count();
248 for(i = 0; i < num_interfaces; i++) {
249 const char *address = sys_inet_ntoa(*iface_n_ip(i));
250 status = add_socket_rpc_tcp_iface(dce_ctx, e, event_ctx, model_ops, address);
251 NT_STATUS_NOT_OK_RETURN(status);
254 status = add_socket_rpc_tcp_iface(dce_ctx, e, event_ctx, model_ops, lp_socket_address());
255 NT_STATUS_NOT_OK_RETURN(status);
261 /****************************************************************************
262 Open the listening sockets for RPC over NCACN_IP_TCP/NCALRPC/NCACN_UNIX_STREAM
263 ****************************************************************************/
264 NTSTATUS dcesrv_sock_init(struct dcesrv_context *dce_ctx,
265 struct event_context *event_ctx, const struct model_ops *model_ops)
267 struct dcesrv_endpoint *e;
270 /* Make sure the directory for NCALRPC exists */
271 if (!directory_exist(lp_ncalrpc_dir(), NULL)) {
272 mkdir(lp_ncalrpc_dir(), 0755);
275 for (e=dce_ctx->endpoint_list;e;e=e->next) {
276 switch (e->ep_description.transport) {
277 case NCACN_UNIX_STREAM:
278 status = add_socket_rpc_unix(dce_ctx, e, event_ctx, model_ops);
279 NT_STATUS_NOT_OK_RETURN(status);
283 status = add_socket_rpc_ncalrpc(dce_ctx, e, event_ctx, model_ops);
284 NT_STATUS_NOT_OK_RETURN(status);
288 status = add_socket_rpc_tcp(dce_ctx, e, event_ctx, model_ops);
289 NT_STATUS_NOT_OK_RETURN(status);