s4/drs: Refactor to be more SAMBA.Coding style compliant
[Samba.git] / source3 / libsmb / clidgram.c
blobf5dbd72f22103842c83ba66edba75974bf5bbc5a
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
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/>.
22 #include "includes.h"
25 * cli_send_mailslot, send a mailslot for client code ...
28 static bool cli_send_mailslot(struct messaging_context *msg_ctx,
29 bool unique, const char *mailslot,
30 uint16 priority,
31 char *buf, int len,
32 const char *srcname, int src_type,
33 const char *dstname, int dest_type,
34 const struct sockaddr_storage *dest_ss)
36 struct packet_struct p;
37 struct dgram_packet *dgram = &p.packet.dgram;
38 char *ptr, *p2;
39 char tmp[4];
40 pid_t nmbd_pid;
41 char addr[INET6_ADDRSTRLEN];
43 if ((nmbd_pid = pidfile_pid("nmbd")) == 0) {
44 DEBUG(3, ("No nmbd found\n"));
45 return False;
48 if (dest_ss->ss_family != AF_INET) {
49 DEBUG(3, ("cli_send_mailslot: can't send to IPv6 address.\n"));
50 return false;
53 memset((char *)&p, '\0', sizeof(p));
56 * Next, build the DGRAM ...
59 /* DIRECT GROUP or UNIQUE datagram. */
60 dgram->header.msg_type = unique ? 0x10 : 0x11;
61 dgram->header.flags.node_type = M_NODE;
62 dgram->header.flags.first = True;
63 dgram->header.flags.more = False;
64 dgram->header.dgm_id = ((unsigned)time(NULL)%(unsigned)0x7FFF) +
65 ((unsigned)sys_getpid()%(unsigned)100);
66 /* source ip is filled by nmbd */
67 dgram->header.dgm_length = 0; /* Let build_dgram() handle this. */
68 dgram->header.packet_offset = 0;
70 make_nmb_name(&dgram->source_name,srcname,src_type);
71 make_nmb_name(&dgram->dest_name,dstname,dest_type);
73 ptr = &dgram->data[0];
75 /* Setup the smb part. */
76 ptr -= 4; /* XXX Ugliness because of handling of tcp SMB length. */
77 memcpy(tmp,ptr,4);
79 if (smb_size + 17*2 + strlen(mailslot) + 1 + len > MAX_DGRAM_SIZE) {
80 DEBUG(0, ("cli_send_mailslot: Cannot write beyond end of packet\n"));
81 return False;
84 cli_set_message(ptr,17,strlen(mailslot) + 1 + len,True);
85 memcpy(ptr,tmp,4);
87 SCVAL(ptr,smb_com,SMBtrans);
88 SSVAL(ptr,smb_vwv1,len);
89 SSVAL(ptr,smb_vwv11,len);
90 SSVAL(ptr,smb_vwv12,70 + strlen(mailslot));
91 SSVAL(ptr,smb_vwv13,3);
92 SSVAL(ptr,smb_vwv14,1);
93 SSVAL(ptr,smb_vwv15,priority);
94 SSVAL(ptr,smb_vwv16,2);
95 p2 = smb_buf(ptr);
96 fstrcpy(p2,mailslot);
97 p2 = skip_string(ptr,MAX_DGRAM_SIZE,p2);
98 if (!p2) {
99 return False;
102 memcpy(p2,buf,len);
103 p2 += len;
105 dgram->datasize = PTR_DIFF(p2,ptr+4); /* +4 for tcp length. */
107 p.packet_type = DGRAM_PACKET;
108 p.ip = ((const struct sockaddr_in *)dest_ss)->sin_addr;
109 p.timestamp = time(NULL);
111 DEBUG(4,("send_mailslot: Sending to mailslot %s from %s ",
112 mailslot, nmb_namestr(&dgram->source_name)));
113 print_sockaddr(addr, sizeof(addr), dest_ss);
115 DEBUGADD(4,("to %s IP %s\n", nmb_namestr(&dgram->dest_name), addr));
117 return NT_STATUS_IS_OK(messaging_send_buf(msg_ctx,
118 pid_to_procid(nmbd_pid),
119 MSG_SEND_PACKET,
120 (uint8 *)&p, sizeof(p)));
123 static const char *mailslot_name(TALLOC_CTX *mem_ctx, struct in_addr dc_ip)
125 return talloc_asprintf(mem_ctx, "%s%X",
126 NBT_MAILSLOT_GETDC, dc_ip.s_addr);
129 bool send_getdc_request(TALLOC_CTX *mem_ctx,
130 struct messaging_context *msg_ctx,
131 struct sockaddr_storage *dc_ss,
132 const char *domain_name,
133 const DOM_SID *sid,
134 uint32_t nt_version)
136 struct in_addr dc_ip;
137 const char *my_acct_name = NULL;
138 const char *my_mailslot = NULL;
139 struct nbt_netlogon_packet packet;
140 struct NETLOGON_SAM_LOGON_REQUEST *s;
141 enum ndr_err_code ndr_err;
142 DATA_BLOB blob;
143 struct dom_sid my_sid;
145 ZERO_STRUCT(packet);
146 ZERO_STRUCT(my_sid);
148 if (dc_ss->ss_family != AF_INET) {
149 return false;
152 if (sid) {
153 my_sid = *sid;
156 dc_ip = ((struct sockaddr_in *)dc_ss)->sin_addr;
157 my_mailslot = mailslot_name(mem_ctx, dc_ip);
158 if (!my_mailslot) {
159 return false;
162 my_acct_name = talloc_asprintf(mem_ctx, "%s$", global_myname());
163 if (!my_acct_name) {
164 return false;
167 packet.command = LOGON_SAM_LOGON_REQUEST;
168 s = &packet.req.logon;
170 s->request_count = 0;
171 s->computer_name = global_myname();
172 s->user_name = my_acct_name;
173 s->mailslot_name = my_mailslot;
174 s->acct_control = ACB_WSTRUST;
175 s->sid = my_sid;
176 s->nt_version = nt_version;
177 s->lmnt_token = 0xffff;
178 s->lm20_token = 0xffff;
180 if (DEBUGLEVEL >= 10) {
181 NDR_PRINT_DEBUG(nbt_netlogon_packet, &packet);
184 ndr_err = ndr_push_struct_blob(&blob, mem_ctx, NULL, &packet,
185 (ndr_push_flags_fn_t)ndr_push_nbt_netlogon_packet);
186 if (!NDR_ERR_CODE_IS_SUCCESS(ndr_err)) {
187 return false;
190 return cli_send_mailslot(msg_ctx,
191 false, NBT_MAILSLOT_NTLOGON, 0,
192 (char *)blob.data, blob.length,
193 global_myname(), 0, domain_name, 0x1c,
194 dc_ss);
197 bool receive_getdc_response(TALLOC_CTX *mem_ctx,
198 struct sockaddr_storage *dc_ss,
199 const char *domain_name,
200 uint32_t *nt_version,
201 const char **dc_name,
202 struct netlogon_samlogon_response **_r)
204 struct packet_struct *packet;
205 const char *my_mailslot = NULL;
206 struct in_addr dc_ip;
207 DATA_BLOB blob;
208 struct netlogon_samlogon_response r;
209 union dgram_message_body p;
210 enum ndr_err_code ndr_err;
211 NTSTATUS status;
213 const char *returned_dc = NULL;
214 const char *returned_domain = NULL;
216 if (dc_ss->ss_family != AF_INET) {
217 return false;
220 dc_ip = ((struct sockaddr_in *)dc_ss)->sin_addr;
222 my_mailslot = mailslot_name(mem_ctx, dc_ip);
223 if (!my_mailslot) {
224 return false;
227 packet = receive_unexpected(DGRAM_PACKET, 0, my_mailslot);
229 if (packet == NULL) {
230 DEBUG(5, ("Did not receive packet for %s\n", my_mailslot));
231 return False;
234 DEBUG(5, ("Received packet for %s\n", my_mailslot));
236 blob = data_blob_const(packet->packet.dgram.data,
237 packet->packet.dgram.datasize);
239 if (blob.length < 4) {
240 DEBUG(0,("invalid length: %d\n", (int)blob.length));
241 return false;
244 if (RIVAL(blob.data,0) != DGRAM_SMB) {
245 DEBUG(0,("invalid packet\n"));
246 return false;
249 blob.data += 4;
250 blob.length -= 4;
252 ndr_err = ndr_pull_union_blob_all(&blob, mem_ctx, NULL, &p, DGRAM_SMB,
253 (ndr_pull_flags_fn_t)ndr_pull_dgram_smb_packet);
254 if (!NDR_ERR_CODE_IS_SUCCESS(ndr_err)) {
255 DEBUG(0,("failed to parse packet\n"));
256 return false;
259 if (p.smb.smb_command != SMB_TRANSACTION) {
260 DEBUG(0,("invalid smb_command: %d\n", p.smb.smb_command));
261 return false;
264 if (DEBUGLEVEL >= 10) {
265 NDR_PRINT_DEBUG(dgram_smb_packet, &p);
268 blob = p.smb.body.trans.data;
270 ZERO_STRUCT(r);
272 status = pull_netlogon_samlogon_response(&blob, mem_ctx, NULL, &r);
273 if (!NT_STATUS_IS_OK(status)) {
274 return false;
277 map_netlogon_samlogon_response(&r);
279 /* do we still need this ? */
280 *nt_version = r.ntver;
282 returned_domain = r.data.nt5_ex.domain;
283 returned_dc = r.data.nt5_ex.pdc_name;
285 if (!strequal(returned_domain, domain_name)) {
286 DEBUG(3, ("GetDC: Expected domain %s, got %s\n",
287 domain_name, returned_domain));
288 return false;
291 *dc_name = talloc_strdup(mem_ctx, returned_dc);
292 if (!*dc_name) {
293 return false;
296 if (**dc_name == '\\') *dc_name += 1;
297 if (**dc_name == '\\') *dc_name += 1;
299 if (_r) {
300 *_r = (struct netlogon_samlogon_response *)talloc_memdup(
301 mem_ctx, &r, sizeof(struct netlogon_samlogon_response));
302 if (!*_r) {
303 return false;
307 DEBUG(10, ("GetDC gave name %s for domain %s\n",
308 *dc_name, returned_domain));
310 return True;