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