Convert all uint32/16/8 to _t in source3/libsmb.
[bbaumbach/samba-autobuild/.git] / source3 / libsmb / clidgram.c
index baee95b9aad2994880f3165bdb69b734e768fa97..d8fa1c6b935a08aff7bd6efe4c42a33eadf9be50 100644 (file)
@@ -9,48 +9,44 @@
    it under the terms of the GNU General Public License as published by
    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,
    but WITHOUT ANY WARRANTY; without even the implied warranty of
    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
    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, see <http://www.gnu.org/licenses/>.
 */
 
 #include "includes.h"
+#include "libsmb/libsmb.h"
+#include "../lib/util/tevent_ntstatus.h"
+#include "libsmb/clidgram.h"
+#include "libsmb/nmblib.h"
+#include "messages.h"
+#include "librpc/gen_ndr/samr.h"
+#include "../lib/util/pidfile.h"
 
 /*
  * cli_send_mailslot, send a mailslot for client code ...
  */
 
-bool cli_send_mailslot(struct messaging_context *msg_ctx,
-                      bool unique, const char *mailslot,
-                      uint16 priority,
+static bool cli_prep_mailslot(bool unique, const char *mailslot,
+                      uint16_t priority,
                       char *buf, int len,
                       const char *srcname, int src_type,
                       const char *dstname, int dest_type,
-                      const struct sockaddr_storage *dest_ss)
+                      const struct sockaddr_storage *dest_ss,
+                      int dgm_id,
+                      struct packet_struct *p)
 {
-       struct packet_struct p;
-       struct dgram_packet *dgram = &p.packet.dgram;
+       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));
+       ZERO_STRUCTP(p);
 
        /*
         * Next, build the DGRAM ...
@@ -61,8 +57,7 @@ bool cli_send_mailslot(struct messaging_context *msg_ctx,
        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.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;
@@ -104,9 +99,9 @@ bool cli_send_mailslot(struct messaging_context *msg_ctx,
 
        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);
+       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)));
@@ -114,173 +109,369 @@ bool cli_send_mailslot(struct messaging_context *msg_ctx,
 
        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)));
+       return true;
 }
 
-static void mailslot_name(struct in_addr dc_ip, fstring name)
+static char *mailslot_name(TALLOC_CTX *mem_ctx, struct in_addr dc_ip)
 {
-       fstr_sprintf(name, "\\MAILSLOT\\NET\\GETDC%X", dc_ip.s_addr);
+       return talloc_asprintf(mem_ctx, "%s%X",
+                              NBT_MAILSLOT_GETDC, dc_ip.s_addr);
 }
 
-bool send_getdc_request(struct messaging_context *msg_ctx,
-                       struct sockaddr_storage *dc_ss,
-                       const char *domain_name,
-                       const DOM_SID *sid)
+static bool prep_getdc_request(const struct sockaddr_storage *dc_ss,
+                              const char *domain_name,
+                              const struct dom_sid *sid,
+                              uint32_t nt_version,
+                              const char *my_mailslot,
+                              int dgm_id,
+                              struct packet_struct *p)
 {
-       char outbuf[1024];
-       struct in_addr dc_ip;
-       char *p;
-       fstring my_acct_name;
-       fstring my_mailslot;
-       size_t sid_size;
-
-       if (dc_ss->ss_family != AF_INET) {
-               return false;
+       TALLOC_CTX *frame = talloc_stackframe();
+       const char *my_acct_name;
+       struct nbt_netlogon_packet packet;
+       struct NETLOGON_SAM_LOGON_REQUEST *s;
+       enum ndr_err_code ndr_err;
+       DATA_BLOB blob = data_blob_null;
+       struct dom_sid my_sid;
+       bool ret = false;
+
+       ZERO_STRUCT(packet);
+       ZERO_STRUCT(my_sid);
+
+       if (sid != NULL) {
+               my_sid = *sid;
        }
 
-       dc_ip = ((struct sockaddr_in *)dc_ss)->sin_addr;
-       mailslot_name(dc_ip, my_mailslot);
+       my_acct_name = talloc_asprintf(talloc_tos(), "%s$", lp_netbios_name());
+       if (my_acct_name == NULL) {
+               goto fail;
+       }
 
-       memset(outbuf, '\0', sizeof(outbuf));
+       packet.command  = LOGON_SAM_LOGON_REQUEST;
+       s               = &packet.req.logon;
+
+       s->request_count        = 0;
+       s->computer_name        = lp_netbios_name();
+       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);
+       }
 
-       p = outbuf;
+       ndr_err = ndr_push_struct_blob(&blob, talloc_tos(), &packet,
+                      (ndr_push_flags_fn_t)ndr_push_nbt_netlogon_packet);
+       if (!NDR_ERR_CODE_IS_SUCCESS(ndr_err)) {
+               goto fail;
+       }
 
-       SCVAL(p, 0, SAMLOGON);
-       p++;
+       ret = cli_prep_mailslot(false, NBT_MAILSLOT_NTLOGON, 0,
+                               (char *)blob.data, blob.length,
+                               lp_netbios_name(), 0, domain_name, 0x1c,
+                               dc_ss, dgm_id, p);
+fail:
+       TALLOC_FREE(frame);
+       return ret;
+}
 
-       SCVAL(p, 0, 0); /* Count pointer ... */
-       p++;
+static bool parse_getdc_response(
+       struct packet_struct *packet,
+       TALLOC_CTX *mem_ctx,
+       const char *domain_name,
+       uint32_t *nt_version,
+       const char **dc_name,
+       struct netlogon_samlogon_response **samlogon_response)
+{
+       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;
+
+       blob = data_blob_const(packet->packet.dgram.data,
+                              packet->packet.dgram.datasize);
+       if (blob.length < 4) {
+               DEBUG(1, ("invalid length: %d\n", (int)blob.length));
+               return false;
+       }
 
-       SIVAL(p, 0, 0); /* The sender's token ... */
-       p += 2;
+       if (RIVAL(blob.data,0) != DGRAM_SMB) {
+               DEBUG(1, ("invalid packet\n"));
+               return false;
+       }
 
-       p += dos_PutUniCode(p, global_myname(),
-                       sizeof(outbuf) - PTR_DIFF(p, outbuf), True);
-       fstr_sprintf(my_acct_name, "%s$", global_myname());
-       p += dos_PutUniCode(p, my_acct_name,
-                       sizeof(outbuf) - PTR_DIFF(p, outbuf), True);
+       blob.data += 4;
+       blob.length -= 4;
 
-       if (strlen(my_mailslot)+1 > sizeof(outbuf) - PTR_DIFF(p, outbuf)) {
+       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(1, ("failed to parse packet\n"));
                return false;
        }
 
-       memcpy(p, my_mailslot, strlen(my_mailslot)+1);
-       p += strlen(my_mailslot)+1;
-
-       if (sizeof(outbuf) - PTR_DIFF(p, outbuf) < 8) {
+       if (p.smb.smb_command != SMB_TRANSACTION) {
+               DEBUG(1, ("invalid smb_command: %d\n", p.smb.smb_command));
                return false;
        }
 
-       SIVAL(p, 0, 0x80);
-       p+=4;
+       if (DEBUGLEVEL >= 10) {
+               NDR_PRINT_DEBUG(dgram_smb_packet, &p);
+       }
 
-       sid_size = ndr_size_dom_sid(sid, 0);
+       blob = p.smb.body.trans.data;
 
-       SIVAL(p, 0, sid_size);
-       p+=4;
+       r = talloc_zero(mem_ctx, struct netlogon_samlogon_response);
+       if (!r) {
+               return false;
+       }
 
-       p = ALIGN4(p, outbuf);
-       if (PTR_DIFF(p, outbuf) > sizeof(outbuf)) {
+       status = pull_netlogon_samlogon_response(&blob, r, r);
+       if (!NT_STATUS_IS_OK(status)) {
+               TALLOC_FREE(r);
                return false;
        }
 
-       if (sid_size + 8 > sizeof(outbuf) - PTR_DIFF(p, outbuf)) {
+       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;
        }
-       if (sid) {
-               sid_linearize(p, sizeof(outbuf) - PTR_DIFF(p, outbuf), sid);
+
+       if (*returned_dc == '\\') returned_dc += 1;
+       if (*returned_dc == '\\') returned_dc += 1;
+
+       *dc_name = talloc_strdup(mem_ctx, returned_dc);
+       if (!*dc_name) {
+               TALLOC_FREE(r);
+               return false;
        }
 
-       p += sid_size;
+       if (samlogon_response) {
+               *samlogon_response = r;
+       } else {
+               TALLOC_FREE(r);
+       }
 
-       SIVAL(p, 0, 1);
-       SSVAL(p, 4, 0xffff);
-       SSVAL(p, 6, 0xffff);
-       p+=8;
+       DEBUG(10, ("GetDC gave name %s for domain %s\n",
+                  *dc_name, returned_domain));
 
-       return cli_send_mailslot(msg_ctx,
-                                False, "\\MAILSLOT\\NET\\NTLOGON", 0,
-                                outbuf, PTR_DIFF(p, outbuf),
-                                global_myname(), 0, domain_name, 0x1c,
-                                dc_ss);
+       return True;
 }
 
-bool receive_getdc_response(struct sockaddr_storage *dc_ss,
-                           const char *domain_name,
-                           fstring dc_name)
+struct nbt_getdc_state {
+       struct tevent_context *ev;
+       struct messaging_context *msg_ctx;
+       struct nb_packet_reader *reader;
+       const char *my_mailslot;
+       pid_t nmbd_pid;
+
+       const struct sockaddr_storage *dc_addr;
+       const char *domain_name;
+       const struct dom_sid *sid;
+       uint32_t nt_version;
+       const char *dc_name;
+       struct netlogon_samlogon_response *samlogon_response;
+
+       struct packet_struct p;
+};
+
+static void nbt_getdc_got_reader(struct tevent_req *subreq);
+static void nbt_getdc_got_response(struct tevent_req *subreq);
+
+struct tevent_req *nbt_getdc_send(TALLOC_CTX *mem_ctx,
+                                 struct tevent_context *ev,
+                                 struct messaging_context *msg_ctx,
+                                 const struct sockaddr_storage *dc_addr,
+                                 const char *domain_name,
+                                 const struct dom_sid *sid,
+                                 uint32_t nt_version)
 {
-       struct packet_struct *packet;
-       fstring my_mailslot;
-       char *buf, *p;
-       fstring dcname, user, domain;
-       int len;
-       struct in_addr dc_ip;
-
-       if (dc_ss->ss_family != AF_INET) {
-               return false;
+       struct tevent_req *req, *subreq;
+       struct nbt_getdc_state *state;
+       uint16_t dgm_id;
+
+       req = tevent_req_create(mem_ctx, &state, struct nbt_getdc_state);
+       if (req == NULL) {
+               return NULL;
+       }
+       state->ev = ev;
+       state->msg_ctx = msg_ctx;
+       state->dc_addr = dc_addr;
+       state->domain_name = domain_name;
+       state->sid = sid;
+       state->nt_version = nt_version;
+
+       if (dc_addr->ss_family != AF_INET) {
+               tevent_req_nterror(req, NT_STATUS_NOT_SUPPORTED);
+               return tevent_req_post(req, ev);
+       }
+       state->my_mailslot = mailslot_name(
+               state, ((const struct sockaddr_in *)dc_addr)->sin_addr);
+       if (tevent_req_nomem(state->my_mailslot, req)) {
+               return tevent_req_post(req, ev);
+       }
+       state->nmbd_pid = pidfile_pid(lp_pid_directory(), "nmbd");
+       if (state->nmbd_pid == 0) {
+               DEBUG(3, ("No nmbd found\n"));
+               tevent_req_nterror(req, NT_STATUS_NOT_SUPPORTED);
+               return tevent_req_post(req, ev);
        }
-       dc_ip = ((struct sockaddr_in *)dc_ss)->sin_addr;
-       mailslot_name(dc_ip, my_mailslot);
 
-       packet = receive_unexpected(DGRAM_PACKET, 0, my_mailslot);
+       generate_random_buffer((uint8_t *)(void *)&dgm_id, sizeof(dgm_id));
 
-       if (packet == NULL) {
-               DEBUG(5, ("Did not receive packet for %s\n", my_mailslot));
-               return False;
+       if (!prep_getdc_request(dc_addr, domain_name, sid, nt_version,
+                               state->my_mailslot, dgm_id & 0x7fff,
+                               &state->p)) {
+               DEBUG(3, ("prep_getdc_request failed\n"));
+               tevent_req_nterror(req, NT_STATUS_INVALID_PARAMETER);
+               return tevent_req_post(req, ev);
        }
 
-       DEBUG(5, ("Received packet for %s\n", my_mailslot));
+       subreq = nb_packet_reader_send(state, ev, DGRAM_PACKET, -1,
+                                      state->my_mailslot);
+       if (tevent_req_nomem(subreq, req)) {
+               return tevent_req_post(req, ev);
+       }
+       tevent_req_set_callback(subreq, nbt_getdc_got_reader, req);
+       return req;
+}
 
-       buf = packet->packet.dgram.data;
-       len = packet->packet.dgram.datasize;
+static void nbt_getdc_got_reader(struct tevent_req *subreq)
+{
+       struct tevent_req *req = tevent_req_callback_data(
+               subreq, struct tevent_req);
+       struct nbt_getdc_state *state = tevent_req_data(
+               req, struct nbt_getdc_state);
+       NTSTATUS status;
+
+       status = nb_packet_reader_recv(subreq, state, &state->reader);
+       TALLOC_FREE(subreq);
+       if (tevent_req_nterror(req, status)) {
+               DEBUG(10, ("nb_packet_reader_recv returned %s\n",
+                          nt_errstr(status)));
+               return;
+       }
 
-       if (len < 70) {
-               /* 70 is a completely arbitrary value to make sure
-                  the SVAL below does not read uninitialized memory */
-               DEBUG(3, ("GetDC got short response\n"));
-               return False;
+       status = messaging_send_buf(
+               state->msg_ctx, pid_to_procid(state->nmbd_pid),
+               MSG_SEND_PACKET, (uint8_t *)&state->p, sizeof(state->p));
+
+       if (tevent_req_nterror(req, status)) {
+               DEBUG(10, ("messaging_send_buf returned %s\n",
+                          nt_errstr(status)));
+               return;
        }
+       subreq = nb_packet_read_send(state, state->ev, state->reader);
+       if (tevent_req_nomem(subreq, req)) {
+               return;
+       }
+       tevent_req_set_callback(subreq, nbt_getdc_got_response, req);
+}
 
-       /* This should be (buf-4)+SVAL(buf-4, smb_vwv12)... */
-       p = buf+SVAL(buf, smb_vwv10);
-
-       switch (CVAL(p, 0)) {
-       case SAMLOGON_R:
-       case SAMLOGON_UNK_R:
-               p+=2;
-               pull_ucs2(buf, dcname, p, sizeof(dcname), PTR_DIFF(buf+len, p),
-                         STR_TERMINATE|STR_NOALIGN);
-               p = skip_unibuf(p, PTR_DIFF(buf+len, p));
-               pull_ucs2(buf, user, p, sizeof(user), PTR_DIFF(buf+len, p),
-                         STR_TERMINATE|STR_NOALIGN);
-               p = skip_unibuf(p, PTR_DIFF(buf+len, p));
-               pull_ucs2(buf, domain, p, sizeof(domain), PTR_DIFF(buf+len, p),
-                         STR_TERMINATE|STR_NOALIGN);
-               p = skip_unibuf(p, PTR_DIFF(buf+len, p));
-
-               if (!strequal(domain, domain_name)) {
-                       DEBUG(3, ("GetDC: Expected domain %s, got %s\n",
-                                 domain_name, domain));
-                       return False;
-               }
-               break;
-
-       default:
-               DEBUG(8, ("GetDC got invalid response type %d\n", CVAL(p, 0)));
-               return False;
+static void nbt_getdc_got_response(struct tevent_req *subreq)
+{
+       struct tevent_req *req = tevent_req_callback_data(
+               subreq, struct tevent_req);
+       struct nbt_getdc_state *state = tevent_req_data(
+               req, struct nbt_getdc_state);
+       struct packet_struct *p;
+       NTSTATUS status;
+       bool ret;
+
+       status = nb_packet_read_recv(subreq, &p);
+       TALLOC_FREE(subreq);
+       if (tevent_req_nterror(req, status)) {
+               return;
        }
-       p = dcname;
-       if (*p == '\\') p += 1;
-       if (*p == '\\') p += 1;
 
-       fstrcpy(dc_name, p);
+       ret = parse_getdc_response(p, state, state->domain_name,
+                                  &state->nt_version, &state->dc_name,
+                                  &state->samlogon_response);
+       free_packet(p);
+       if (!ret) {
+               tevent_req_nterror(req, NT_STATUS_INVALID_NETWORK_RESPONSE);
+               return;
+       }
+       tevent_req_done(req);
+}
 
-       DEBUG(10, ("GetDC gave name %s for domain %s\n",
-                  dc_name, domain));
+NTSTATUS nbt_getdc_recv(struct tevent_req *req, TALLOC_CTX *mem_ctx,
+                       uint32_t *nt_version, const char **dc_name,
+                       struct netlogon_samlogon_response **samlogon_response)
+{
+       struct nbt_getdc_state *state = tevent_req_data(
+               req, struct nbt_getdc_state);
+       NTSTATUS status;
 
-       return True;
+       if (tevent_req_is_nterror(req, &status)) {
+               return status;
+       }
+       if (nt_version != NULL) {
+               *nt_version = state->nt_version;
+       }
+       if (dc_name != NULL) {
+               *dc_name = talloc_move(mem_ctx, &state->dc_name);
+       }
+       if (samlogon_response != NULL) {
+               *samlogon_response = talloc_move(
+                       mem_ctx, &state->samlogon_response);
+       }
+       return NT_STATUS_OK;
 }
 
+NTSTATUS nbt_getdc(struct messaging_context *msg_ctx,
+                  uint32_t timeout_in_seconds,
+                  const struct sockaddr_storage *dc_addr,
+                  const char *domain_name,
+                  const struct dom_sid *sid,
+                  uint32_t nt_version,
+                  TALLOC_CTX *mem_ctx,
+                  uint32_t *pnt_version,
+                  const char **dc_name,
+                  struct netlogon_samlogon_response **samlogon_response)
+{
+       TALLOC_CTX *frame = talloc_stackframe();
+       struct tevent_context *ev;
+       struct tevent_req *req;
+       NTSTATUS status = NT_STATUS_NO_MEMORY;
+
+       ev = samba_tevent_context_init(frame);
+       if (ev == NULL) {
+               goto fail;
+       }
+       req = nbt_getdc_send(ev, ev, msg_ctx, dc_addr, domain_name,
+                            sid, nt_version);
+       if (req == NULL) {
+               goto fail;
+       }
+       if (!tevent_req_set_endtime(req, ev,
+                       timeval_current_ofs(timeout_in_seconds, 0))) {
+               goto fail;
+       }
+       if (!tevent_req_poll_ntstatus(req, ev, &status)) {
+               goto fail;
+       }
+       status = nbt_getdc_recv(req, mem_ctx, pnt_version, dc_name,
+                               samlogon_response);
+ fail:
+       TALLOC_FREE(frame);
+       return status;
+}