r15356: Remove unused 'flags' argument from socket_send() and friends.
[bbaumbach/samba-autobuild/.git] / source4 / libcli / dgram / dgramsocket.c
1 /* 
2    Unix SMB/CIFS implementation.
3
4    low level socket handling for nbt dgram requests (UDP138)
5
6    Copyright (C) Andrew Tridgell 2005
7    
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 2 of the License, or
11    (at your option) any later version.
12    
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.
17    
18    You should have received a copy of the GNU General Public License
19    along with this program; if not, write to the Free Software
20    Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
21 */
22
23 #include "includes.h"
24 #include "lib/events/events.h"
25 #include "dlinklist.h"
26 #include "libcli/dgram/libdgram.h"
27 #include "lib/socket/socket.h"
28 #include "librpc/gen_ndr/ndr_nbt.h"
29
30
31 /*
32   handle recv events on a nbt dgram socket
33 */
34 static void dgm_socket_recv(struct nbt_dgram_socket *dgmsock)
35 {
36         TALLOC_CTX *tmp_ctx = talloc_new(dgmsock);
37         NTSTATUS status;
38         struct socket_address *src;
39         DATA_BLOB blob;
40         size_t nread, dsize;
41         struct nbt_dgram_packet *packet;
42         const char *mailslot_name;
43
44         status = socket_pending(dgmsock->sock, &dsize);
45         if (!NT_STATUS_IS_OK(status)) {
46                 talloc_free(tmp_ctx);
47                 return;
48         }
49
50         blob = data_blob_talloc(tmp_ctx, NULL, dsize);
51         if (blob.data == NULL) {
52                 talloc_free(tmp_ctx);
53                 return;
54         }
55
56         status = socket_recvfrom(dgmsock->sock, blob.data, blob.length, &nread,
57                                  tmp_ctx, &src);
58         if (!NT_STATUS_IS_OK(status)) {
59                 talloc_free(tmp_ctx);
60                 return;
61         }
62         blob.length = nread;
63
64         DEBUG(2,("Received dgram packet of length %d from %s:%d\n", 
65                  (int)blob.length, src->addr, src->port));
66
67         packet = talloc(tmp_ctx, struct nbt_dgram_packet);
68         if (packet == NULL) {
69                 talloc_free(tmp_ctx);
70                 return;
71         }
72
73         /* parse the request */
74         status = ndr_pull_struct_blob(&blob, packet, packet, 
75                                       (ndr_pull_flags_fn_t)ndr_pull_nbt_dgram_packet);
76         if (!NT_STATUS_IS_OK(status)) {
77                 DEBUG(2,("Failed to parse incoming NBT DGRAM packet - %s\n",
78                          nt_errstr(status)));
79                 talloc_free(tmp_ctx);
80                 return;
81         }
82
83         /* if this is a mailslot message, then see if we can dispatch it to a handler */
84         mailslot_name = dgram_mailslot_name(packet);
85         if (mailslot_name) {
86                 struct dgram_mailslot_handler *dgmslot;
87                 dgmslot = dgram_mailslot_find(dgmsock, mailslot_name);
88                 if (dgmslot) {
89                         dgmslot->handler(dgmslot, packet, src);
90                 } else {
91                         DEBUG(2,("No mailslot handler for '%s'\n", mailslot_name));
92                 }
93         } else {
94                 /* dispatch if there is a general handler */
95                 if (dgmsock->incoming.handler) {
96                         dgmsock->incoming.handler(dgmsock, packet, src);
97                 }
98         }
99
100         talloc_free(tmp_ctx);
101 }
102
103
104 /*
105   handle send events on a nbt dgram socket
106 */
107 static void dgm_socket_send(struct nbt_dgram_socket *dgmsock)
108 {
109         struct nbt_dgram_request *req;
110         NTSTATUS status;
111
112         while ((req = dgmsock->send_queue)) {
113                 size_t len;
114                 
115                 len = req->encoded.length;
116                 status = socket_sendto(dgmsock->sock, &req->encoded, &len,
117                                        req->dest);
118                 if (NT_STATUS_IS_ERR(status)) {
119                         DEBUG(3,("Failed to send datagram of length %u to %s:%d: %s\n",
120                                  (unsigned)req->encoded.length, req->dest->addr, req->dest->port, 
121                                  nt_errstr(status)));
122                         DLIST_REMOVE(dgmsock->send_queue, req);
123                         talloc_free(req);
124                         continue;
125                 }
126
127                 if (!NT_STATUS_IS_OK(status)) return;
128
129                 DLIST_REMOVE(dgmsock->send_queue, req);
130                 talloc_free(req);
131         }
132
133         EVENT_FD_NOT_WRITEABLE(dgmsock->fde);
134         return;
135 }
136
137
138 /*
139   handle fd events on a nbt_dgram_socket
140 */
141 static void dgm_socket_handler(struct event_context *ev, struct fd_event *fde,
142                                uint16_t flags, void *private)
143 {
144         struct nbt_dgram_socket *dgmsock = talloc_get_type(private, 
145                                                            struct nbt_dgram_socket);
146         if (flags & EVENT_FD_WRITE) {
147                 dgm_socket_send(dgmsock);
148         } 
149         if (flags & EVENT_FD_READ) {
150                 dgm_socket_recv(dgmsock);
151         }
152 }
153
154 /*
155   initialise a nbt_dgram_socket. The event_ctx is optional, if provided
156   then operations will use that event context
157 */
158 struct nbt_dgram_socket *nbt_dgram_socket_init(TALLOC_CTX *mem_ctx, 
159                                               struct event_context *event_ctx)
160 {
161         struct nbt_dgram_socket *dgmsock;
162         NTSTATUS status;
163
164         dgmsock = talloc(mem_ctx, struct nbt_dgram_socket);
165         if (dgmsock == NULL) goto failed;
166
167         if (event_ctx == NULL) {
168                 dgmsock->event_ctx = event_context_init(dgmsock);
169         } else {
170                 dgmsock->event_ctx = talloc_reference(dgmsock, event_ctx);
171         }
172         if (dgmsock->event_ctx == NULL) goto failed;
173
174         status = socket_create("ip", SOCKET_TYPE_DGRAM, &dgmsock->sock, 0);
175         if (!NT_STATUS_IS_OK(status)) goto failed;
176
177         socket_set_option(dgmsock->sock, "SO_BROADCAST", "1");
178
179         talloc_steal(dgmsock, dgmsock->sock);
180
181         dgmsock->fde = event_add_fd(dgmsock->event_ctx, dgmsock, 
182                                     socket_get_fd(dgmsock->sock), 0,
183                                     dgm_socket_handler, dgmsock);
184
185         dgmsock->send_queue = NULL;
186         dgmsock->incoming.handler = NULL;
187         dgmsock->mailslot_handlers = NULL;
188         
189         return dgmsock;
190
191 failed:
192         talloc_free(dgmsock);
193         return NULL;
194 }
195
196
197 /*
198   setup a handler for generic incoming requests
199 */
200 NTSTATUS dgram_set_incoming_handler(struct nbt_dgram_socket *dgmsock,
201                                     void (*handler)(struct nbt_dgram_socket *, 
202                                                     struct nbt_dgram_packet *, 
203                                                     struct socket_address *),
204                                     void *private)
205 {
206         dgmsock->incoming.handler = handler;
207         dgmsock->incoming.private = private;
208         EVENT_FD_READABLE(dgmsock->fde);
209         return NT_STATUS_OK;
210 }
211
212
213 /*
214   queue a datagram for send
215 */
216 NTSTATUS nbt_dgram_send(struct nbt_dgram_socket *dgmsock,
217                         struct nbt_dgram_packet *packet,
218                         struct socket_address *dest)
219 {
220         struct nbt_dgram_request *req;
221         NTSTATUS status = NT_STATUS_NO_MEMORY;
222
223         req = talloc(dgmsock, struct nbt_dgram_request);
224         if (req == NULL) goto failed;
225
226         req->dest = dest;
227         if (talloc_reference(req, dest) == NULL) goto failed;
228
229         status = ndr_push_struct_blob(&req->encoded, req, packet, 
230                                       (ndr_push_flags_fn_t)ndr_push_nbt_dgram_packet);
231         if (!NT_STATUS_IS_OK(status)) goto failed;
232
233         DLIST_ADD_END(dgmsock->send_queue, req, struct nbt_dgram_request *);
234
235         EVENT_FD_WRITEABLE(dgmsock->fde);
236
237         return NT_STATUS_OK;
238
239 failed:
240         talloc_free(req);
241         return status;
242 }