2 Unix SMB/CIFS implementation.
6 Copyright (C) Andrew Tridgell 2005
8 This program is free software; you can redistribute it and/or modify
9 it under the terms of the GNU General Public License as published by
10 the Free Software Foundation; either version 3 of the License, or
11 (at your option) any later version.
13 This program is distributed in the hope that it will be useful,
14 but WITHOUT ANY WARRANTY; without even the implied warranty of
15 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
16 GNU General Public License for more details.
18 You should have received a copy of the GNU General Public License
19 along with this program. If not, see <http://www.gnu.org/licenses/>.
23 #include "nbt_server/nbt_server.h"
24 #include "smbd/service_task.h"
25 #include "lib/socket/socket.h"
26 #include "libcli/resolve/resolve.h"
27 #include "nbt_server/dgram/proto.h"
28 #include "librpc/gen_ndr/ndr_nbt.h"
29 #include "param/param.h"
32 a list of mailslots that we have static handlers for
35 const char *mailslot_name;
36 dgram_mailslot_handler_t handler;
37 } mailslot_handlers[] = {
38 { NBT_MAILSLOT_NETLOGON, nbtd_mailslot_netlogon_handler },
39 { NBT_MAILSLOT_NTLOGON, nbtd_mailslot_ntlogon_handler },
40 { NBT_MAILSLOT_BROWSE, nbtd_mailslot_browse_handler }
44 receive an incoming dgram request. This is used for general datagram
45 requests. Mailslot requests for our listening mailslots
46 are handled in the specific mailslot handlers
48 void dgram_request_handler(struct nbt_dgram_socket *dgmsock,
49 struct nbt_dgram_packet *packet,
50 struct socket_address *src)
52 DEBUG(0,("General datagram request from %s:%d\n", src->addr, src->port));
53 NDR_PRINT_DEBUG(nbt_dgram_packet, packet);
58 setup the port 138 datagram listener for a given interface
60 NTSTATUS nbtd_dgram_setup(struct nbtd_interface *iface, const char *bind_address)
62 struct nbt_dgram_socket *bcast_dgmsock = NULL;
63 struct nbtd_server *nbtsrv = iface->nbtsrv;
64 struct socket_address *bcast_addr, *bind_addr;
66 TALLOC_CTX *tmp_ctx = talloc_new(iface);
67 /* the list of mailslots that we are interested in */
71 return NT_STATUS_NO_MEMORY;
74 if (strcmp("0.0.0.0", iface->netmask) != 0) {
75 /* listen for broadcasts on port 138 */
76 bcast_dgmsock = nbt_dgram_socket_init(iface,
77 nbtsrv->task->event_ctx,
78 lp_iconv_convenience(nbtsrv->task->lp_ctx));
81 return NT_STATUS_NO_MEMORY;
84 bcast_addr = socket_address_from_strings(tmp_ctx, bcast_dgmsock->sock->backend_name,
86 lp_dgram_port(iface->nbtsrv->task->lp_ctx));
89 return NT_STATUS_NO_MEMORY;
92 status = socket_listen(bcast_dgmsock->sock, bcast_addr, 0, 0);
93 if (!NT_STATUS_IS_OK(status)) {
95 DEBUG(0,("Failed to bind to %s:%d - %s\n",
96 iface->bcast_address, lp_dgram_port(iface->nbtsrv->task->lp_ctx),
101 dgram_set_incoming_handler(bcast_dgmsock, dgram_request_handler, iface);
104 /* listen for unicasts on port 138 */
105 iface->dgmsock = nbt_dgram_socket_init(iface, nbtsrv->task->event_ctx,
106 lp_iconv_convenience(nbtsrv->task->lp_ctx));
107 if (!iface->dgmsock) {
108 talloc_free(tmp_ctx);
109 return NT_STATUS_NO_MEMORY;
112 bind_addr = socket_address_from_strings(tmp_ctx, iface->dgmsock->sock->backend_name,
113 bind_address, lp_dgram_port(iface->nbtsrv->task->lp_ctx));
115 talloc_free(tmp_ctx);
116 return NT_STATUS_NO_MEMORY;
119 status = socket_listen(iface->dgmsock->sock, bind_addr, 0, 0);
120 if (!NT_STATUS_IS_OK(status)) {
121 talloc_free(tmp_ctx);
122 DEBUG(0,("Failed to bind to %s:%d - %s\n",
123 bind_address, lp_dgram_port(iface->nbtsrv->task->lp_ctx), nt_errstr(status)));
127 dgram_set_incoming_handler(iface->dgmsock, dgram_request_handler, iface);
129 talloc_free(tmp_ctx);
131 for (i=0;i<ARRAY_SIZE(mailslot_handlers);i++) {
132 /* note that we don't need to keep the pointer
133 to the dgmslot around - the callback is all
135 struct dgram_mailslot_handler *dgmslot;
138 dgmslot = dgram_mailslot_listen(bcast_dgmsock,
139 mailslot_handlers[i].mailslot_name,
140 mailslot_handlers[i].handler, iface);
141 NT_STATUS_HAVE_NO_MEMORY(dgmslot);
144 dgmslot = dgram_mailslot_listen(iface->dgmsock,
145 mailslot_handlers[i].mailslot_name,
146 mailslot_handlers[i].handler, iface);
147 NT_STATUS_HAVE_NO_MEMORY(dgmslot);