s3: only use netlogon/nbt header when needed.
[metze/samba/wip.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 "librpc/gen_ndr/messaging.h"
24 #include "libsmb/clidgram.h"
25
26 /*
27  * cli_send_mailslot, send a mailslot for client code ...
28  */
29
30 static bool cli_send_mailslot(struct messaging_context *msg_ctx,
31                        bool unique, const char *mailslot,
32                        uint16 priority,
33                        char *buf, int len,
34                        const char *srcname, int src_type,
35                        const char *dstname, int dest_type,
36                        const struct sockaddr_storage *dest_ss)
37 {
38         struct packet_struct p;
39         struct dgram_packet *dgram = &p.packet.dgram;
40         char *ptr, *p2;
41         char tmp[4];
42         pid_t nmbd_pid;
43         char addr[INET6_ADDRSTRLEN];
44
45         if ((nmbd_pid = pidfile_pid("nmbd")) == 0) {
46                 DEBUG(3, ("No nmbd found\n"));
47                 return False;
48         }
49
50         if (dest_ss->ss_family != AF_INET) {
51                 DEBUG(3, ("cli_send_mailslot: can't send to IPv6 address.\n"));
52                 return false;
53         }
54
55         memset((char *)&p, '\0', sizeof(p));
56
57         /*
58          * Next, build the DGRAM ...
59          */
60
61         /* DIRECT GROUP or UNIQUE datagram. */
62         dgram->header.msg_type = unique ? 0x10 : 0x11;
63         dgram->header.flags.node_type = M_NODE;
64         dgram->header.flags.first = True;
65         dgram->header.flags.more = False;
66         dgram->header.dgm_id = ((unsigned)time(NULL)%(unsigned)0x7FFF) +
67                 ((unsigned)sys_getpid()%(unsigned)100);
68         /* source ip is filled by nmbd */
69         dgram->header.dgm_length = 0; /* Let build_dgram() handle this. */
70         dgram->header.packet_offset = 0;
71
72         make_nmb_name(&dgram->source_name,srcname,src_type);
73         make_nmb_name(&dgram->dest_name,dstname,dest_type);
74
75         ptr = &dgram->data[0];
76
77         /* Setup the smb part. */
78         ptr -= 4; /* XXX Ugliness because of handling of tcp SMB length. */
79         memcpy(tmp,ptr,4);
80
81         if (smb_size + 17*2 + strlen(mailslot) + 1 + len > MAX_DGRAM_SIZE) {
82                 DEBUG(0, ("cli_send_mailslot: Cannot write beyond end of packet\n"));
83                 return False;
84         }
85
86         cli_set_message(ptr,17,strlen(mailslot) + 1 + len,True);
87         memcpy(ptr,tmp,4);
88
89         SCVAL(ptr,smb_com,SMBtrans);
90         SSVAL(ptr,smb_vwv1,len);
91         SSVAL(ptr,smb_vwv11,len);
92         SSVAL(ptr,smb_vwv12,70 + strlen(mailslot));
93         SSVAL(ptr,smb_vwv13,3);
94         SSVAL(ptr,smb_vwv14,1);
95         SSVAL(ptr,smb_vwv15,priority);
96         SSVAL(ptr,smb_vwv16,2);
97         p2 = smb_buf(ptr);
98         fstrcpy(p2,mailslot);
99         p2 = skip_string(ptr,MAX_DGRAM_SIZE,p2);
100         if (!p2) {
101                 return False;
102         }
103
104         memcpy(p2,buf,len);
105         p2 += len;
106
107         dgram->datasize = PTR_DIFF(p2,ptr+4); /* +4 for tcp length. */
108
109         p.packet_type = DGRAM_PACKET;
110         p.ip = ((const struct sockaddr_in *)dest_ss)->sin_addr;
111         p.timestamp = time(NULL);
112
113         DEBUG(4,("send_mailslot: Sending to mailslot %s from %s ",
114                  mailslot, nmb_namestr(&dgram->source_name)));
115         print_sockaddr(addr, sizeof(addr), dest_ss);
116
117         DEBUGADD(4,("to %s IP %s\n", nmb_namestr(&dgram->dest_name), addr));
118
119         return NT_STATUS_IS_OK(messaging_send_buf(msg_ctx,
120                                                   pid_to_procid(nmbd_pid),
121                                                   MSG_SEND_PACKET,
122                                                   (uint8 *)&p, sizeof(p)));
123 }
124
125 static const char *mailslot_name(TALLOC_CTX *mem_ctx, struct in_addr dc_ip)
126 {
127         return talloc_asprintf(mem_ctx, "%s%X",
128                                NBT_MAILSLOT_GETDC, dc_ip.s_addr);
129 }
130
131 bool send_getdc_request(TALLOC_CTX *mem_ctx,
132                         struct messaging_context *msg_ctx,
133                         struct sockaddr_storage *dc_ss,
134                         const char *domain_name,
135                         const struct dom_sid *sid,
136                         uint32_t nt_version)
137 {
138         struct in_addr dc_ip;
139         const char *my_acct_name = NULL;
140         const char *my_mailslot = NULL;
141         struct nbt_netlogon_packet packet;
142         struct NETLOGON_SAM_LOGON_REQUEST *s;
143         enum ndr_err_code ndr_err;
144         DATA_BLOB blob;
145         struct dom_sid my_sid;
146
147         ZERO_STRUCT(packet);
148         ZERO_STRUCT(my_sid);
149
150         if (dc_ss->ss_family != AF_INET) {
151                 return false;
152         }
153
154         if (sid) {
155                 my_sid = *sid;
156         }
157
158         dc_ip = ((struct sockaddr_in *)dc_ss)->sin_addr;
159         my_mailslot = mailslot_name(mem_ctx, dc_ip);
160         if (!my_mailslot) {
161                 return false;
162         }
163
164         my_acct_name = talloc_asprintf(mem_ctx, "%s$", global_myname());
165         if (!my_acct_name) {
166                 return false;
167         }
168
169         packet.command  = LOGON_SAM_LOGON_REQUEST;
170         s               = &packet.req.logon;
171
172         s->request_count        = 0;
173         s->computer_name        = global_myname();
174         s->user_name            = my_acct_name;
175         s->mailslot_name        = my_mailslot;
176         s->acct_control         = ACB_WSTRUST;
177         s->sid                  = my_sid;
178         s->nt_version           = nt_version;
179         s->lmnt_token           = 0xffff;
180         s->lm20_token           = 0xffff;
181
182         if (DEBUGLEVEL >= 10) {
183                 NDR_PRINT_DEBUG(nbt_netlogon_packet, &packet);
184         }
185
186         ndr_err = ndr_push_struct_blob(&blob, mem_ctx, &packet,
187                        (ndr_push_flags_fn_t)ndr_push_nbt_netlogon_packet);
188         if (!NDR_ERR_CODE_IS_SUCCESS(ndr_err)) {
189                 return false;
190         }
191
192         return cli_send_mailslot(msg_ctx,
193                                  false, NBT_MAILSLOT_NTLOGON, 0,
194                                  (char *)blob.data, blob.length,
195                                  global_myname(), 0, domain_name, 0x1c,
196                                  dc_ss);
197 }
198
199 bool receive_getdc_response(TALLOC_CTX *mem_ctx,
200                             struct sockaddr_storage *dc_ss,
201                             const char *domain_name,
202                             uint32_t *nt_version,
203                             const char **dc_name,
204                             struct netlogon_samlogon_response **_r)
205 {
206         struct packet_struct *packet;
207         const char *my_mailslot = NULL;
208         struct in_addr dc_ip;
209         DATA_BLOB blob;
210         struct netlogon_samlogon_response r;
211         union dgram_message_body p;
212         enum ndr_err_code ndr_err;
213         NTSTATUS status;
214
215         const char *returned_dc = NULL;
216         const char *returned_domain = NULL;
217
218         if (dc_ss->ss_family != AF_INET) {
219                 return false;
220         }
221
222         dc_ip = ((struct sockaddr_in *)dc_ss)->sin_addr;
223
224         my_mailslot = mailslot_name(mem_ctx, dc_ip);
225         if (!my_mailslot) {
226                 return false;
227         }
228
229         packet = receive_unexpected(DGRAM_PACKET, 0, my_mailslot);
230
231         if (packet == NULL) {
232                 DEBUG(5, ("Did not receive packet for %s\n", my_mailslot));
233                 return False;
234         }
235
236         DEBUG(5, ("Received packet for %s\n", my_mailslot));
237
238         blob = data_blob_const(packet->packet.dgram.data,
239                                packet->packet.dgram.datasize);
240
241         if (blob.length < 4) {
242                 DEBUG(0,("invalid length: %d\n", (int)blob.length));
243                 return false;
244         }
245
246         if (RIVAL(blob.data,0) != DGRAM_SMB) {
247                 DEBUG(0,("invalid packet\n"));
248                 return false;
249         }
250
251         blob.data += 4;
252         blob.length -= 4;
253
254         ndr_err = ndr_pull_union_blob_all(&blob, mem_ctx, &p, DGRAM_SMB,
255                        (ndr_pull_flags_fn_t)ndr_pull_dgram_smb_packet);
256         if (!NDR_ERR_CODE_IS_SUCCESS(ndr_err)) {
257                 DEBUG(0,("failed to parse packet\n"));
258                 return false;
259         }
260
261         if (p.smb.smb_command != SMB_TRANSACTION) {
262                 DEBUG(0,("invalid smb_command: %d\n", p.smb.smb_command));
263                 return false;
264         }
265
266         if (DEBUGLEVEL >= 10) {
267                 NDR_PRINT_DEBUG(dgram_smb_packet, &p);
268         }
269
270         blob = p.smb.body.trans.data;
271
272         ZERO_STRUCT(r);
273
274         status = pull_netlogon_samlogon_response(&blob, mem_ctx, &r);
275         if (!NT_STATUS_IS_OK(status)) {
276                 return false;
277         }
278
279         map_netlogon_samlogon_response(&r);
280
281         /* do we still need this ? */
282         *nt_version = r.ntver;
283
284         returned_domain = r.data.nt5_ex.domain_name;
285         returned_dc = r.data.nt5_ex.pdc_name;
286
287         if (!strequal(returned_domain, domain_name)) {
288                 DEBUG(3, ("GetDC: Expected domain %s, got %s\n",
289                           domain_name, returned_domain));
290                 return false;
291         }
292
293         *dc_name = talloc_strdup(mem_ctx, returned_dc);
294         if (!*dc_name) {
295                 return false;
296         }
297
298         if (**dc_name == '\\')  *dc_name += 1;
299         if (**dc_name == '\\')  *dc_name += 1;
300
301         if (_r) {
302                 *_r = (struct netlogon_samlogon_response *)talloc_memdup(
303                         mem_ctx, &r, sizeof(struct netlogon_samlogon_response));
304                 if (!*_r) {
305                         return false;
306                 }
307         }
308
309         DEBUG(10, ("GetDC gave name %s for domain %s\n",
310                    *dc_name, returned_domain));
311
312         return True;
313 }