Changeset 745 for trunk/server/source3/libsmb/clidgram.c

Timestamp:

Nov 27, 2012, 4:43:17 PM (13 years ago)

Author:

Silvan Scherrer

Message:

Samba Server: updated trunk to 3.6.0

Location:

trunk/server

Files:

• . (modified) (1 prop)
• source3/libsmb/clidgram.c (modified) (12 diffs)

trunk/server/source3/libsmb/clidgram.c

@@ -10 +10 @@
    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/>.
@@ -21 +21 @@
 
 #include "includes.h"
+#include "libsmb/libsmb.h"
+#include "../lib/util/tevent_ntstatus.h"
+#include "libsmb/clidgram.h"
+#include "libsmb/nmblib.h"
+#include "messages.h"
 
 /*
@@ -26 +31 @@
  */
 
-static bool cli_send_mailslot(struct messaging_context *msg_ctx,
-                       bool unique, const char *mailslot,
+static bool cli_prep_mailslot(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)
-{
-        struct packet_struct p;
-        struct dgram_packet *dgram = &p.packet.dgram;
+                       const struct sockaddr_storage *dest_ss,
+                       int dgm_id,
+                       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));
+        ZERO_STRUCTP(p);
 
         /*
@@ -62 +56 @@
         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. */
@@ -105 +98 @@
         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 ",
@@ -115 +108 @@
         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)));
-}
-
-static const char *mailslot_name(TALLOC_CTX *mem_ctx, struct in_addr dc_ip)
+        return true;
+}
+
+static char *mailslot_name(TALLOC_CTX *mem_ctx, struct in_addr dc_ip)
 {
         return talloc_asprintf(mem_ctx, "%s%X",
@@ -127 +117 @@
 }
 
-bool send_getdc_request(TALLOC_CTX *mem_ctx,
-                        struct messaging_context *msg_ctx,
-                        struct sockaddr_storage *dc_ss,
-                        const char *domain_name,
-                        const DOM_SID *sid,
-                        uint32_t nt_version)
-{
-        struct in_addr dc_ip;
-        const char *my_acct_name = NULL;
-        const char *my_mailslot = NULL;
+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)
+{
+        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 blob = data_blob_null;
         struct dom_sid my_sid;
+        bool ret = false;
 
         ZERO_STRUCT(packet);
         ZERO_STRUCT(my_sid);
 
-        if (dc_ss->ss_family != AF_INET) {
-                return false;
-        }
-
-        if (sid) {
+        if (sid != NULL) {
                 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;
+        my_acct_name = talloc_asprintf(talloc_tos(), "%s$", global_myname());
+        if (my_acct_name == NULL) {
+                goto fail;
         }
 
@@ -182 +163 @@
         }
 
-        ndr_err = ndr_push_struct_blob(&blob, mem_ctx, NULL, &packet,
+        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)) {
-                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);
-}
-
-bool receive_getdc_response(TALLOC_CTX *mem_ctx,
-                            struct sockaddr_storage *dc_ss,
-                            const char *domain_name,
-                            uint32_t *nt_version,
-                            const char **dc_name,
-                            struct netlogon_samlogon_response **_r)
-{
-        struct packet_struct *packet;
-        const char *my_mailslot = NULL;
-        struct in_addr dc_ip;
+                goto fail;
+        }
+
+        ret = cli_prep_mailslot(false, NBT_MAILSLOT_NTLOGON, 0,
+                                (char *)blob.data, blob.length,
+                                global_myname(), 0, domain_name, 0x1c,
+                                dc_ss, dgm_id, p);
+fail:
+        TALLOC_FREE(frame);
+        return ret;
+}
+
+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;
+        struct netlogon_samlogon_response *r;
         union dgram_message_body p;
         enum ndr_err_code ndr_err;
@@ -214 +195 @@
         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, 0, 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));
-                free_packet(packet);
+                DEBUG(1, ("invalid length: %d\n", (int)blob.length));
                 return false;
         }
 
         if (RIVAL(blob.data,0) != DGRAM_SMB) {
-                DEBUG(0,("invalid packet\n"));
-                free_packet(packet);
+                DEBUG(1, ("invalid packet\n"));
                 return false;
         }
@@ -252 +210 @@
         blob.length -= 4;
 
-        ndr_err = ndr_pull_union_blob_all(&blob, mem_ctx, NULL, &p, DGRAM_SMB,
+        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"));
-                free_packet(packet);
+                DEBUG(1, ("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));
-                free_packet(packet);
+                DEBUG(1, ("invalid smb_command: %d\n", p.smb.smb_command));
                 return false;
         }
@@ -272 +228 @@
         blob = p.smb.body.trans.data;
 
-        ZERO_STRUCT(r);
-
-        status = pull_netlogon_samlogon_response(&blob, mem_ctx, NULL, &r);
+        r = TALLOC_ZERO_P(mem_ctx, struct netlogon_samlogon_response);
+        if (!r) {
+                return false;
+        }
+
+        status = pull_netlogon_samlogon_response(&blob, r, r);
         if (!NT_STATUS_IS_OK(status)) {
-                free_packet(packet);
-                return false;
-        }
-
-        map_netlogon_samlogon_response(&r);
+                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;
-        returned_dc = r.data.nt5_ex.pdc_name;
+        *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));
-                free_packet(packet);
-                return false;
-        }
+                TALLOC_FREE(r);
+                return false;
+        }
+
+        if (*returned_dc == '\\') returned_dc += 1;
+        if (*returned_dc == '\\') returned_dc += 1;
 
         *dc_name = talloc_strdup(mem_ctx, returned_dc);
         if (!*dc_name) {
-                free_packet(packet);
-                return false;
-        }
-
-        if (**dc_name == '\\')  *dc_name += 1;
-        if (**dc_name == '\\')  *dc_name += 1;
-
-        if (_r) {
-                *_r = (struct netlogon_samlogon_response *)talloc_memdup(
-                        mem_ctx, &r, sizeof(struct netlogon_samlogon_response));
-                if (!*_r) {
-                free_packet(packet);
-                        return false;
-                }
+                TALLOC_FREE(r);
+                return false;
+        }
+
+        if (samlogon_response) {
+                *samlogon_response = r;
+        } else {
+                TALLOC_FREE(r);
         }
 
@@ -316 +272 @@
                    *dc_name, returned_domain));
 
-        free_packet(packet);
         return True;
 }
+
+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 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, ((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("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);
+        }
+
+        generate_random_buffer((uint8_t *)(void *)&dgm_id, sizeof(dgm_id));
+
+        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);
+        }
+
+        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;
+}
+
+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;
+        }
+
+        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);
+}
+
+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;
+        }
+
+        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);
+}
+
+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;
+
+        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,
+                   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 = 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_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;
+}