s3: Add some const to send_getdc_request
[metze/samba/wip.git] / source3 / libsmb / clidgram.c
index a45cc009ac88f5826558a27ccc90445d064eca7f..c1b2fd19b0c989ab53ff7c23be6d4fb2c2f29037 100644 (file)
@@ -1,6 +1,5 @@
 /* 
-   Unix SMB/Netbios implementation.
-   Version 3.0
+   Unix SMB/CIFS implementation.
    client dgram calls
    Copyright (C) Andrew Tridgell 1994-1998
    Copyright (C) Richard Sharpe 2001
@@ -8,7 +7,7 @@
 
    This program is free software; you can redistribute it and/or modify
    it under the terms of the GNU General Public License as published by
-   the Free Software Foundation; either version 2 of the License, or
+   the Free Software Foundation; either version 3 of the License, or
    (at your option) any later version.
    
    This program is distributed in the hope that it will be useful,
    GNU General Public License for more details.
    
    You should have received a copy of the GNU General Public License
-   along with this program; if not, write to the Free Software
-   Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
+   along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */
 
-#define NO_SYSLOG
-
 #include "includes.h"
+#include "librpc/gen_ndr/messaging.h"
+#include "libsmb/clidgram.h"
 
 /*
  * cli_send_mailslot, send a mailslot for client code ...
  */
 
-static int dgram_sock = -1;
-
-int cli_send_mailslot(BOOL unique, char *mailslot, char *buf, int len,
-                     const char *srcname, int src_type, 
-                     const char *dstname, int dest_type,
-                     struct in_addr dest_ip, struct in_addr src_ip,
-                     int dest_port)
+static bool cli_send_mailslot(struct messaging_context *msg_ctx,
+                      bool unique, const char *mailslot,
+                      uint16 priority,
+                      char *buf, int len,
+                      const char *srcname, int src_type,
+                      const char *dstname, int dest_type,
+                      const struct sockaddr_storage *dest_ss,
+                      int dgm_id)
 {
-  struct packet_struct p;
-  struct dgram_packet *dgram = &p.packet.dgram;
-  struct sockaddr_in sock_out;
-  char *ptr, *p2;
-  char tmp[4];
-
-  bzero((char *)&p, sizeof(p));
-
-  /* 
-   * First, check if we have an open socket to the dest IP 
-   */
-
-  if (dgram_sock < 1) {
-
-    int name_size;
-
-    if ((dgram_sock = open_socket_out(SOCK_DGRAM, &dest_ip, 138, LONG_CONNECT_TIMEOUT)) < 0) {
-
-      DEBUG(4, ("open_sock_out failed ..."));
-      return False;
-
-    }
-
-    /* Make it a broadcast socket ... */
-
-    set_socket_options(dgram_sock, "SO_BROADCAST");
-
-    /* Now, bind my addr to it ... */
-
-    bzero((char *)&sock_out, sizeof(sock_out));
-    putip((char *)&sock_out.sin_addr, (char *)&src_ip);
-    sock_out.sin_port = INADDR_ANY;
-    sock_out.sin_family = AF_INET;
-
-    bind(dgram_sock, (struct sockaddr_in *)&sock_out, sizeof(sock_out));
-
-    /* Now, figure out what socket name we were bound to. We want the port */
-
-    name_size = sizeof(sock_out);
-
-    getsockname(dgram_sock, (struct sockaddr_in *)&sock_out, &name_size);
-
-    fprintf(stderr, "Socket bound to IP:%s, port: %d\n", inet_ntoa(sock_out.sin_addr), ntohs(sock_out.sin_port));
-
-  }
-
-  /*
-   * Next, build the DGRAM ...
-   */
-
-  /* DIRECT GROUP or UNIQUE datagram. */
-  dgram->header.msg_type = unique ? 0x10 : 0x11; 
-  dgram->header.flags.node_type = M_NODE;
-  dgram->header.flags.first = True;
-  dgram->header.flags.more = False;
-  dgram->header.dgm_id = ((unsigned)time(NULL)%(unsigned)0x7FFF) + ((unsigned)sys_getpid()%(unsigned)100);
-  dgram->header.source_ip = src_ip;
-  dgram->header.source_port = ntohs(sock_out.sin_port);
-  dgram->header.dgm_length = 0; /* Let build_dgram() handle this. */
-  dgram->header.packet_offset = 0;
-  
-  make_nmb_name(&dgram->source_name,srcname,src_type);
-  make_nmb_name(&dgram->dest_name,dstname,dest_type);
-
-  ptr = &dgram->data[0];
-
-  /* Setup the smb part. */
-  ptr -= 4; /* XXX Ugliness because of handling of tcp SMB length. */
-  memcpy(tmp,ptr,4);
-  set_message(ptr,17,17 + len,True);
-  memcpy(ptr,tmp,4);
-
-  CVAL(ptr,smb_com) = SMBtrans;
-  SSVAL(ptr,smb_vwv1,len);
-  SSVAL(ptr,smb_vwv11,len);
-  SSVAL(ptr,smb_vwv12,70 + strlen(mailslot));
-  SSVAL(ptr,smb_vwv13,3);
-  SSVAL(ptr,smb_vwv14,1);
-  SSVAL(ptr,smb_vwv15,1);
-  SSVAL(ptr,smb_vwv16,2);
-  p2 = smb_buf(ptr);
-  pstrcpy(p2,mailslot);
-  p2 = skip_string(p2,1);
-
-  memcpy(p2,buf,len);
-  p2 += len;
-
-  dgram->datasize = PTR_DIFF(p2,ptr+4); /* +4 for tcp length. */
-
-  p.ip = dest_ip;
-  p.port = dest_port;
-  p.fd = dgram_sock;
-  p.timestamp = time(NULL);
-  p.packet_type = DGRAM_PACKET;
-
-  DEBUG(4,("send_mailslot: Sending to mailslot %s from %s IP %s ", mailslot,
-                    nmb_namestr(&dgram->source_name), inet_ntoa(src_ip)));
-  DEBUG(4,("to %s IP %s\n", nmb_namestr(&dgram->dest_name), inet_ntoa(dest_ip)));
-
-  return send_packet(&p);
-
+       struct packet_struct p;
+       struct dgram_packet *dgram = &p.packet.dgram;
+       char *ptr, *p2;
+       char tmp[4];
+       pid_t nmbd_pid;
+       char addr[INET6_ADDRSTRLEN];
+
+       if ((nmbd_pid = pidfile_pid("nmbd")) == 0) {
+               DEBUG(3, ("No nmbd found\n"));
+               return False;
+       }
+
+       if (dest_ss->ss_family != AF_INET) {
+               DEBUG(3, ("cli_send_mailslot: can't send to IPv6 address.\n"));
+               return false;
+       }
+
+       memset((char *)&p, '\0', sizeof(p));
+
+       /*
+        * Next, build the DGRAM ...
+        */
+
+       /* DIRECT GROUP or UNIQUE datagram. */
+       dgram->header.msg_type = unique ? 0x10 : 0x11;
+       dgram->header.flags.node_type = M_NODE;
+       dgram->header.flags.first = True;
+       dgram->header.flags.more = False;
+       dgram->header.dgm_id = dgm_id;
+       /* source ip is filled by nmbd */
+       dgram->header.dgm_length = 0; /* Let build_dgram() handle this. */
+       dgram->header.packet_offset = 0;
+
+       make_nmb_name(&dgram->source_name,srcname,src_type);
+       make_nmb_name(&dgram->dest_name,dstname,dest_type);
+
+       ptr = &dgram->data[0];
+
+       /* Setup the smb part. */
+       ptr -= 4; /* XXX Ugliness because of handling of tcp SMB length. */
+       memcpy(tmp,ptr,4);
+
+       if (smb_size + 17*2 + strlen(mailslot) + 1 + len > MAX_DGRAM_SIZE) {
+               DEBUG(0, ("cli_send_mailslot: Cannot write beyond end of packet\n"));
+               return False;
+       }
+
+       cli_set_message(ptr,17,strlen(mailslot) + 1 + len,True);
+       memcpy(ptr,tmp,4);
+
+       SCVAL(ptr,smb_com,SMBtrans);
+       SSVAL(ptr,smb_vwv1,len);
+       SSVAL(ptr,smb_vwv11,len);
+       SSVAL(ptr,smb_vwv12,70 + strlen(mailslot));
+       SSVAL(ptr,smb_vwv13,3);
+       SSVAL(ptr,smb_vwv14,1);
+       SSVAL(ptr,smb_vwv15,priority);
+       SSVAL(ptr,smb_vwv16,2);
+       p2 = smb_buf(ptr);
+       fstrcpy(p2,mailslot);
+       p2 = skip_string(ptr,MAX_DGRAM_SIZE,p2);
+       if (!p2) {
+               return False;
+       }
+
+       memcpy(p2,buf,len);
+       p2 += len;
+
+       dgram->datasize = PTR_DIFF(p2,ptr+4); /* +4 for tcp length. */
+
+       p.packet_type = DGRAM_PACKET;
+       p.ip = ((const struct sockaddr_in *)dest_ss)->sin_addr;
+       p.timestamp = time(NULL);
+
+       DEBUG(4,("send_mailslot: Sending to mailslot %s from %s ",
+                mailslot, nmb_namestr(&dgram->source_name)));
+       print_sockaddr(addr, sizeof(addr), dest_ss);
+
+       DEBUGADD(4,("to %s IP %s\n", nmb_namestr(&dgram->dest_name), addr));
+
+       return NT_STATUS_IS_OK(messaging_send_buf(msg_ctx,
+                                                 pid_to_procid(nmbd_pid),
+                                                 MSG_SEND_PACKET,
+                                                 (uint8 *)&p, sizeof(p)));
 }
 
-/*
- * cli_get_response: Get a response ...
- */
-int cli_get_response(BOOL unique, char *mailslot, char *buf, int bufsiz)
+static const char *mailslot_name(TALLOC_CTX *mem_ctx, struct in_addr dc_ip)
 {
-  struct packet_struct *packet;
-
-  packet = read_packet(dgram_sock, DGRAM_PACKET);
-
-  if (packet) { /* We got one, pull what we want out of the SMB data ... */
-
-    struct dgram_packet *dgram = &packet->packet.dgram;
-
-    /*
-     * We should probably parse the SMB, but for now, we will pull what
-     * from fixed, known locations ...
-     */
-
-    /* Copy the data to buffer, respecting sizes ... */
-
-    bcopy(&dgram->data[92], buf, MIN(bufsiz, (dgram->datasize - 92)));
-
-  }
-  else 
-    return -1;
-
+       return talloc_asprintf(mem_ctx, "%s%X",
+                              NBT_MAILSLOT_GETDC, dc_ip.s_addr);
 }
 
-/*
- * cli_get_backup_list: Send a get backup list request ...
- */
-
-static char cli_backup_list[1024];
-
-int cli_get_backup_list(const char *myname, const char *send_to_name)
+bool send_getdc_request(TALLOC_CTX *mem_ctx,
+                       struct messaging_context *msg_ctx,
+                       const struct sockaddr_storage *dc_ss,
+                       const char *domain_name,
+                       const struct dom_sid *sid,
+                       uint32_t nt_version,
+                       int dgm_id)
 {
-  char outbuf[15];
-  char *p;
-  struct in_addr sendto_ip, my_ip;
-
-  if (!resolve_name(send_to_name, &sendto_ip, 0x1d)) {
-
-    fprintf(stderr, "Could not resolve name: %s<1D>\n", send_to_name);
-
-  }
-
-  inet_aton("0.0.0.0", &my_ip);
-
-  if (!resolve_name(myname, &my_ip, 0x00)) { /* FIXME: Call others here */
-
-    fprintf(stderr, "Could not resolve name: %s<00>\n", myname);
-
-  }
-
-  bzero(cli_backup_list, sizeof(cli_backup_list));
-  bzero(outbuf, sizeof(outbuf));
-
-  p = outbuf;
-
-  SCVAL(p, 0, ANN_GetBackupListReq);
-  p++;
-
-  SCVAL(p, 0, 1); /* Count pointer ... */
-  p++;
-
-  SIVAL(p, 0, 1); /* The sender's token ... */
-  p += 4;
-
-  cli_send_mailslot(True, "\\MAILSLOT\\BROWSE", outbuf, PTR_DIFF(p, outbuf),
-                   myname, 0, send_to_name, 0x1d, sendto_ip, my_ip, 138);
-
-  /* We should check the error and return if we got one */
-
-  /* Now, get the response ... */
-
-  cli_get_response(True, "\\MAILSLOT\\BROWSE", cli_backup_list, sizeof(cli_backup_list));
-
-  /* Should check the response here ... FIXME */
-
+       struct in_addr dc_ip;
+       const char *my_acct_name = NULL;
+       const char *my_mailslot = NULL;
+       struct nbt_netlogon_packet packet;
+       struct NETLOGON_SAM_LOGON_REQUEST *s;
+       enum ndr_err_code ndr_err;
+       DATA_BLOB blob;
+       struct dom_sid my_sid;
+
+       ZERO_STRUCT(packet);
+       ZERO_STRUCT(my_sid);
+
+       if (dc_ss->ss_family != AF_INET) {
+               return false;
+       }
+
+       if (sid) {
+               my_sid = *sid;
+       }
+
+       dc_ip = ((struct sockaddr_in *)dc_ss)->sin_addr;
+       my_mailslot = mailslot_name(mem_ctx, dc_ip);
+       if (!my_mailslot) {
+               return false;
+       }
+
+       my_acct_name = talloc_asprintf(mem_ctx, "%s$", global_myname());
+       if (!my_acct_name) {
+               return false;
+       }
+
+       packet.command  = LOGON_SAM_LOGON_REQUEST;
+       s               = &packet.req.logon;
+
+       s->request_count        = 0;
+       s->computer_name        = global_myname();
+       s->user_name            = my_acct_name;
+       s->mailslot_name        = my_mailslot;
+       s->acct_control         = ACB_WSTRUST;
+       s->sid                  = my_sid;
+       s->nt_version           = nt_version;
+       s->lmnt_token           = 0xffff;
+       s->lm20_token           = 0xffff;
+
+       if (DEBUGLEVEL >= 10) {
+               NDR_PRINT_DEBUG(nbt_netlogon_packet, &packet);
+       }
+
+       ndr_err = ndr_push_struct_blob(&blob, mem_ctx, &packet,
+                      (ndr_push_flags_fn_t)ndr_push_nbt_netlogon_packet);
+       if (!NDR_ERR_CODE_IS_SUCCESS(ndr_err)) {
+               return false;
+       }
+
+       return cli_send_mailslot(msg_ctx,
+                                false, NBT_MAILSLOT_NTLOGON, 0,
+                                (char *)blob.data, blob.length,
+                                global_myname(), 0, domain_name, 0x1c,
+                                dc_ss, dgm_id);
 }
 
-/*
- * cli_get_backup_server: Get the backup list and retrieve a server from it
- */
-
-int cli_get_backup_server(char *my_name, char *target, char *servername, int namesize)
+bool receive_getdc_response(TALLOC_CTX *mem_ctx,
+                           struct sockaddr_storage *dc_ss,
+                           const char *domain_name,
+                           int dgm_id,
+                           uint32_t *nt_version,
+                           const char **dc_name,
+                           struct netlogon_samlogon_response **samlogon_response)
 {
+       struct packet_struct *packet;
+       const char *my_mailslot = NULL;
+       struct in_addr dc_ip;
+       DATA_BLOB blob;
+       struct netlogon_samlogon_response *r;
+       union dgram_message_body p;
+       enum ndr_err_code ndr_err;
+       NTSTATUS status;
+
+       const char *returned_dc = NULL;
+       const char *returned_domain = NULL;
+
+       if (dc_ss->ss_family != AF_INET) {
+               return false;
+       }
+
+       dc_ip = ((struct sockaddr_in *)dc_ss)->sin_addr;
+
+       my_mailslot = mailslot_name(mem_ctx, dc_ip);
+       if (!my_mailslot) {
+               return false;
+       }
+
+       packet = receive_unexpected(DGRAM_PACKET, dgm_id, my_mailslot);
+
+       if (packet == NULL) {
+               DEBUG(5, ("Did not receive packet for %s\n", my_mailslot));
+               return False;
+       }
+
+       DEBUG(5, ("Received packet for %s\n", my_mailslot));
+
+       blob = data_blob_const(packet->packet.dgram.data,
+                              packet->packet.dgram.datasize);
+
+       if (blob.length < 4) {
+               DEBUG(0,("invalid length: %d\n", (int)blob.length));
+               return false;
+       }
+
+       if (RIVAL(blob.data,0) != DGRAM_SMB) {
+               DEBUG(0,("invalid packet\n"));
+               return false;
+       }
+
+       blob.data += 4;
+       blob.length -= 4;
+
+       ndr_err = ndr_pull_union_blob_all(&blob, mem_ctx, &p, DGRAM_SMB,
+                      (ndr_pull_flags_fn_t)ndr_pull_dgram_smb_packet);
+       if (!NDR_ERR_CODE_IS_SUCCESS(ndr_err)) {
+               DEBUG(0,("failed to parse packet\n"));
+               return false;
+       }
+
+       if (p.smb.smb_command != SMB_TRANSACTION) {
+               DEBUG(0,("invalid smb_command: %d\n", p.smb.smb_command));
+               return false;
+       }
+
+       if (DEBUGLEVEL >= 10) {
+               NDR_PRINT_DEBUG(dgram_smb_packet, &p);
+       }
+
+       blob = p.smb.body.trans.data;
+
+       r = TALLOC_ZERO_P(mem_ctx, struct netlogon_samlogon_response);
+       if (!r) {
+               return false;
+       }
+
+       status = pull_netlogon_samlogon_response(&blob, mem_ctx, r);
+       if (!NT_STATUS_IS_OK(status)) {
+               TALLOC_FREE(r);
+               return false;
+       }
+
+       map_netlogon_samlogon_response(r);
+
+       /* do we still need this ? */
+       *nt_version = r->ntver;
+
+       returned_domain = r->data.nt5_ex.domain_name;
+       returned_dc = r->data.nt5_ex.pdc_name;
+
+       if (!strequal(returned_domain, domain_name)) {
+               DEBUG(3, ("GetDC: Expected domain %s, got %s\n",
+                         domain_name, returned_domain));
+               TALLOC_FREE(r);
+               return false;
+       }
+
+       *dc_name = talloc_strdup(mem_ctx, returned_dc);
+       if (!*dc_name) {
+               TALLOC_FREE(r);
+               return false;
+       }
+
+       if (**dc_name == '\\')  *dc_name += 1;
+       if (**dc_name == '\\')  *dc_name += 1;
+
+       if (samlogon_response) {
+               *samlogon_response = r;
+       } else {
+               TALLOC_FREE(r);
+       }
 
-  /* Get the backup list first. We could pull this from the cache later */
-
-  cli_get_backup_list(my_name, target);  /* FIXME: Check the response */
-
-  strncpy(servername, cli_backup_list, MIN(16, namesize));
+       DEBUG(10, ("GetDC gave name %s for domain %s\n",
+                  *dc_name, returned_domain));
 
+       return True;
 }