s3-tevent: only include ../lib/util/tevent wrappers where needed.
[bbaumbach/samba-autobuild/.git] / source3 / libsmb / clidgram.c
1 /* 
2    Unix SMB/CIFS implementation.
3    client dgram calls
4    Copyright (C) Andrew Tridgell 1994-1998
5    Copyright (C) Richard Sharpe 2001
6    Copyright (C) John Terpstra 2001
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 3 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, see <http://www.gnu.org/licenses/>.
20 */
21
22 #include "includes.h"
23 #include "../lib/util/tevent_ntstatus.h"
24 #include "libsmb/clidgram.h"
25 #include "libsmb/nmblib.h"
26 #include "messages.h"
27
28 /*
29  * cli_send_mailslot, send a mailslot for client code ...
30  */
31
32 static bool cli_prep_mailslot(bool unique, const char *mailslot,
33                        uint16 priority,
34                        char *buf, int len,
35                        const char *srcname, int src_type,
36                        const char *dstname, int dest_type,
37                        const struct sockaddr_storage *dest_ss,
38                        int dgm_id,
39                        struct packet_struct *p)
40 {
41         struct dgram_packet *dgram = &p->packet.dgram;
42         char *ptr, *p2;
43         char tmp[4];
44         char addr[INET6_ADDRSTRLEN];
45
46         ZERO_STRUCTP(p);
47
48         /*
49          * Next, build the DGRAM ...
50          */
51
52         /* DIRECT GROUP or UNIQUE datagram. */
53         dgram->header.msg_type = unique ? 0x10 : 0x11;
54         dgram->header.flags.node_type = M_NODE;
55         dgram->header.flags.first = True;
56         dgram->header.flags.more = False;
57         dgram->header.dgm_id = dgm_id;
58         /* source ip is filled by nmbd */
59         dgram->header.dgm_length = 0; /* Let build_dgram() handle this. */
60         dgram->header.packet_offset = 0;
61
62         make_nmb_name(&dgram->source_name,srcname,src_type);
63         make_nmb_name(&dgram->dest_name,dstname,dest_type);
64
65         ptr = &dgram->data[0];
66
67         /* Setup the smb part. */
68         ptr -= 4; /* XXX Ugliness because of handling of tcp SMB length. */
69         memcpy(tmp,ptr,4);
70
71         if (smb_size + 17*2 + strlen(mailslot) + 1 + len > MAX_DGRAM_SIZE) {
72                 DEBUG(0, ("cli_send_mailslot: Cannot write beyond end of packet\n"));
73                 return False;
74         }
75
76         cli_set_message(ptr,17,strlen(mailslot) + 1 + len,True);
77         memcpy(ptr,tmp,4);
78
79         SCVAL(ptr,smb_com,SMBtrans);
80         SSVAL(ptr,smb_vwv1,len);
81         SSVAL(ptr,smb_vwv11,len);
82         SSVAL(ptr,smb_vwv12,70 + strlen(mailslot));
83         SSVAL(ptr,smb_vwv13,3);
84         SSVAL(ptr,smb_vwv14,1);
85         SSVAL(ptr,smb_vwv15,priority);
86         SSVAL(ptr,smb_vwv16,2);
87         p2 = smb_buf(ptr);
88         fstrcpy(p2,mailslot);
89         p2 = skip_string(ptr,MAX_DGRAM_SIZE,p2);
90         if (!p2) {
91                 return False;
92         }
93
94         memcpy(p2,buf,len);
95         p2 += len;
96
97         dgram->datasize = PTR_DIFF(p2,ptr+4); /* +4 for tcp length. */
98
99         p->packet_type = DGRAM_PACKET;
100         p->ip = ((const struct sockaddr_in *)dest_ss)->sin_addr;
101         p->timestamp = time(NULL);
102
103         DEBUG(4,("send_mailslot: Sending to mailslot %s from %s ",
104                  mailslot, nmb_namestr(&dgram->source_name)));
105         print_sockaddr(addr, sizeof(addr), dest_ss);
106
107         DEBUGADD(4,("to %s IP %s\n", nmb_namestr(&dgram->dest_name), addr));
108
109         return true;
110 }
111
112 static char *mailslot_name(TALLOC_CTX *mem_ctx, struct in_addr dc_ip)
113 {
114         return talloc_asprintf(mem_ctx, "%s%X",
115                                NBT_MAILSLOT_GETDC, dc_ip.s_addr);
116 }
117
118 static bool prep_getdc_request(const struct sockaddr_storage *dc_ss,
119                                const char *domain_name,
120                                const struct dom_sid *sid,
121                                uint32_t nt_version,
122                                const char *my_mailslot,
123                                int dgm_id,
124                                struct packet_struct *p)
125 {
126         TALLOC_CTX *frame = talloc_stackframe();
127         const char *my_acct_name;
128         struct nbt_netlogon_packet packet;
129         struct NETLOGON_SAM_LOGON_REQUEST *s;
130         enum ndr_err_code ndr_err;
131         DATA_BLOB blob = data_blob_null;
132         struct dom_sid my_sid;
133         bool ret = false;
134
135         ZERO_STRUCT(packet);
136         ZERO_STRUCT(my_sid);
137
138         if (sid != NULL) {
139                 my_sid = *sid;
140         }
141
142         my_acct_name = talloc_asprintf(talloc_tos(), "%s$", global_myname());
143         if (my_acct_name == NULL) {
144                 goto fail;
145         }
146
147         packet.command  = LOGON_SAM_LOGON_REQUEST;
148         s               = &packet.req.logon;
149
150         s->request_count        = 0;
151         s->computer_name        = global_myname();
152         s->user_name            = my_acct_name;
153         s->mailslot_name        = my_mailslot;
154         s->acct_control         = ACB_WSTRUST;
155         s->sid                  = my_sid;
156         s->nt_version           = nt_version;
157         s->lmnt_token           = 0xffff;
158         s->lm20_token           = 0xffff;
159
160         if (DEBUGLEVEL >= 10) {
161                 NDR_PRINT_DEBUG(nbt_netlogon_packet, &packet);
162         }
163
164         ndr_err = ndr_push_struct_blob(&blob, talloc_tos(), &packet,
165                        (ndr_push_flags_fn_t)ndr_push_nbt_netlogon_packet);
166         if (!NDR_ERR_CODE_IS_SUCCESS(ndr_err)) {
167                 goto fail;
168         }
169
170         ret = cli_prep_mailslot(false, NBT_MAILSLOT_NTLOGON, 0,
171                                 (char *)blob.data, blob.length,
172                                 global_myname(), 0, domain_name, 0x1c,
173                                 dc_ss, dgm_id, p);
174 fail:
175         TALLOC_FREE(frame);
176         return ret;
177 }
178
179 static bool parse_getdc_response(
180         struct packet_struct *packet,
181         TALLOC_CTX *mem_ctx,
182         const char *domain_name,
183         uint32_t *nt_version,
184         const char **dc_name,
185         struct netlogon_samlogon_response **samlogon_response)
186 {
187         DATA_BLOB blob;
188         struct netlogon_samlogon_response *r;
189         union dgram_message_body p;
190         enum ndr_err_code ndr_err;
191         NTSTATUS status;
192
193         const char *returned_dc = NULL;
194         const char *returned_domain = NULL;
195
196         blob = data_blob_const(packet->packet.dgram.data,
197                                packet->packet.dgram.datasize);
198         if (blob.length < 4) {
199                 DEBUG(1, ("invalid length: %d\n", (int)blob.length));
200                 return false;
201         }
202
203         if (RIVAL(blob.data,0) != DGRAM_SMB) {
204                 DEBUG(1, ("invalid packet\n"));
205                 return false;
206         }
207
208         blob.data += 4;
209         blob.length -= 4;
210
211         ndr_err = ndr_pull_union_blob_all(&blob, mem_ctx, &p, DGRAM_SMB,
212                        (ndr_pull_flags_fn_t)ndr_pull_dgram_smb_packet);
213         if (!NDR_ERR_CODE_IS_SUCCESS(ndr_err)) {
214                 DEBUG(1, ("failed to parse packet\n"));
215                 return false;
216         }
217
218         if (p.smb.smb_command != SMB_TRANSACTION) {
219                 DEBUG(1, ("invalid smb_command: %d\n", p.smb.smb_command));
220                 return false;
221         }
222
223         if (DEBUGLEVEL >= 10) {
224                 NDR_PRINT_DEBUG(dgram_smb_packet, &p);
225         }
226
227         blob = p.smb.body.trans.data;
228
229         r = TALLOC_ZERO_P(mem_ctx, struct netlogon_samlogon_response);
230         if (!r) {
231                 return false;
232         }
233
234         status = pull_netlogon_samlogon_response(&blob, r, r);
235         if (!NT_STATUS_IS_OK(status)) {
236                 TALLOC_FREE(r);
237                 return false;
238         }
239
240         map_netlogon_samlogon_response(r);
241
242         /* do we still need this ? */
243         *nt_version = r->ntver;
244
245         returned_domain = r->data.nt5_ex.domain_name;
246         returned_dc = r->data.nt5_ex.pdc_name;
247
248         if (!strequal(returned_domain, domain_name)) {
249                 DEBUG(3, ("GetDC: Expected domain %s, got %s\n",
250                           domain_name, returned_domain));
251                 TALLOC_FREE(r);
252                 return false;
253         }
254
255         if (*returned_dc == '\\') returned_dc += 1;
256         if (*returned_dc == '\\') returned_dc += 1;
257
258         *dc_name = talloc_strdup(mem_ctx, returned_dc);
259         if (!*dc_name) {
260                 TALLOC_FREE(r);
261                 return false;
262         }
263
264         if (samlogon_response) {
265                 *samlogon_response = r;
266         } else {
267                 TALLOC_FREE(r);
268         }
269
270         DEBUG(10, ("GetDC gave name %s for domain %s\n",
271                    *dc_name, returned_domain));
272
273         return True;
274 }
275
276 struct nbt_getdc_state {
277         struct tevent_context *ev;
278         struct messaging_context *msg_ctx;
279         struct nb_packet_reader *reader;
280         const char *my_mailslot;
281         pid_t nmbd_pid;
282
283         const struct sockaddr_storage *dc_addr;
284         const char *domain_name;
285         const struct dom_sid *sid;
286         uint32_t nt_version;
287         const char *dc_name;
288         struct netlogon_samlogon_response *samlogon_response;
289
290         struct packet_struct p;
291 };
292
293 static void nbt_getdc_got_reader(struct tevent_req *subreq);
294 static void nbt_getdc_got_response(struct tevent_req *subreq);
295
296 struct tevent_req *nbt_getdc_send(TALLOC_CTX *mem_ctx,
297                                   struct tevent_context *ev,
298                                   struct messaging_context *msg_ctx,
299                                   const struct sockaddr_storage *dc_addr,
300                                   const char *domain_name,
301                                   const struct dom_sid *sid,
302                                   uint32_t nt_version)
303 {
304         struct tevent_req *req, *subreq;
305         struct nbt_getdc_state *state;
306         uint16_t dgm_id;
307
308         req = tevent_req_create(mem_ctx, &state, struct nbt_getdc_state);
309         if (req == NULL) {
310                 return NULL;
311         }
312         state->ev = ev;
313         state->msg_ctx = msg_ctx;
314         state->dc_addr = dc_addr;
315         state->domain_name = domain_name;
316         state->sid = sid;
317         state->nt_version = nt_version;
318
319         if (dc_addr->ss_family != AF_INET) {
320                 tevent_req_nterror(req, NT_STATUS_NOT_SUPPORTED);
321                 return tevent_req_post(req, ev);
322         }
323         state->my_mailslot = mailslot_name(
324                 state, ((struct sockaddr_in *)dc_addr)->sin_addr);
325         if (tevent_req_nomem(state->my_mailslot, req)) {
326                 return tevent_req_post(req, ev);
327         }
328         state->nmbd_pid = pidfile_pid("nmbd");
329         if (state->nmbd_pid == 0) {
330                 DEBUG(3, ("No nmbd found\n"));
331                 tevent_req_nterror(req, NT_STATUS_NOT_SUPPORTED);
332                 return tevent_req_post(req, ev);
333         }
334
335         generate_random_buffer((uint8_t *)(void *)&dgm_id, sizeof(dgm_id));
336
337         if (!prep_getdc_request(dc_addr, domain_name, sid, nt_version,
338                                 state->my_mailslot, dgm_id & 0x7fff,
339                                 &state->p)) {
340                 DEBUG(3, ("prep_getdc_request failed\n"));
341                 tevent_req_nterror(req, NT_STATUS_INVALID_PARAMETER);
342                 return tevent_req_post(req, ev);
343         }
344
345         subreq = nb_packet_reader_send(state, ev, DGRAM_PACKET, -1,
346                                        state->my_mailslot);
347         if (tevent_req_nomem(subreq, req)) {
348                 return tevent_req_post(req, ev);
349         }
350         tevent_req_set_callback(subreq, nbt_getdc_got_reader, req);
351         return req;
352 }
353
354 static void nbt_getdc_got_reader(struct tevent_req *subreq)
355 {
356         struct tevent_req *req = tevent_req_callback_data(
357                 subreq, struct tevent_req);
358         struct nbt_getdc_state *state = tevent_req_data(
359                 req, struct nbt_getdc_state);
360         NTSTATUS status;
361
362         status = nb_packet_reader_recv(subreq, state, &state->reader);
363         TALLOC_FREE(subreq);
364         if (tevent_req_nterror(req, status)) {
365                 DEBUG(10, ("nb_packet_reader_recv returned %s\n",
366                            nt_errstr(status)));
367                 return;
368         }
369
370         status = messaging_send_buf(
371                 state->msg_ctx, pid_to_procid(state->nmbd_pid),
372                 MSG_SEND_PACKET, (uint8_t *)&state->p, sizeof(state->p));
373
374         if (tevent_req_nterror(req, status)) {
375                 DEBUG(10, ("messaging_send_buf returned %s\n",
376                            nt_errstr(status)));
377                 return;
378         }
379         subreq = nb_packet_read_send(state, state->ev, state->reader);
380         if (tevent_req_nomem(subreq, req)) {
381                 return;
382         }
383         tevent_req_set_callback(subreq, nbt_getdc_got_response, req);
384 }
385
386 static void nbt_getdc_got_response(struct tevent_req *subreq)
387 {
388         struct tevent_req *req = tevent_req_callback_data(
389                 subreq, struct tevent_req);
390         struct nbt_getdc_state *state = tevent_req_data(
391                 req, struct nbt_getdc_state);
392         struct packet_struct *p;
393         NTSTATUS status;
394         bool ret;
395
396         status = nb_packet_read_recv(subreq, &p);
397         TALLOC_FREE(subreq);
398         if (tevent_req_nterror(req, status)) {
399                 return;
400         }
401
402         ret = parse_getdc_response(p, state, state->domain_name,
403                                    &state->nt_version, &state->dc_name,
404                                    &state->samlogon_response);
405         free_packet(p);
406         if (!ret) {
407                 tevent_req_nterror(req, NT_STATUS_INVALID_NETWORK_RESPONSE);
408                 return;
409         }
410         tevent_req_done(req);
411 }
412
413 NTSTATUS nbt_getdc_recv(struct tevent_req *req, TALLOC_CTX *mem_ctx,
414                         uint32_t *nt_version, const char **dc_name,
415                         struct netlogon_samlogon_response **samlogon_response)
416 {
417         struct nbt_getdc_state *state = tevent_req_data(
418                 req, struct nbt_getdc_state);
419         NTSTATUS status;
420
421         if (tevent_req_is_nterror(req, &status)) {
422                 return status;
423         }
424         if (nt_version != NULL) {
425                 *nt_version = state->nt_version;
426         }
427         if (dc_name != NULL) {
428                 *dc_name = talloc_move(mem_ctx, &state->dc_name);
429         }
430         if (samlogon_response != NULL) {
431                 *samlogon_response = talloc_move(
432                         mem_ctx, &state->samlogon_response);
433         }
434         return NT_STATUS_OK;
435 }
436
437 NTSTATUS nbt_getdc(struct messaging_context *msg_ctx,
438                    const struct sockaddr_storage *dc_addr,
439                    const char *domain_name,
440                    const struct dom_sid *sid,
441                    uint32_t nt_version,
442                    TALLOC_CTX *mem_ctx,
443                    uint32_t *pnt_version,
444                    const char **dc_name,
445                    struct netlogon_samlogon_response **samlogon_response)
446 {
447         TALLOC_CTX *frame = talloc_stackframe();
448         struct tevent_context *ev;
449         struct tevent_req *req;
450         NTSTATUS status = NT_STATUS_NO_MEMORY;
451
452         ev = tevent_context_init(frame);
453         if (ev == NULL) {
454                 goto fail;
455         }
456         req = nbt_getdc_send(ev, ev, msg_ctx, dc_addr, domain_name,
457                              sid, nt_version);
458         if (req == NULL) {
459                 goto fail;
460         }
461         if (!tevent_req_poll_ntstatus(req, ev, &status)) {
462                 goto fail;
463         }
464         status = nbt_getdc_recv(req, mem_ctx, pnt_version, dc_name,
465                                 samlogon_response);
466  fail:
467         TALLOC_FREE(frame);
468         return status;
469 }