source3/winbindd/winbindd_ping_dc.c

   1 /*
   2    Unix SMB/CIFS implementation.
   3    async implementation of WINBINDD_PING_DC
   4    Copyright (C) Volker Lendecke 2009
   5
   6    This program is free software; you can redistribute it and/or modify
   7    it under the terms of the GNU General Public License as published by
   8    the Free Software Foundation; either version 3 of the License, or
   9    (at your option) any later version.
  10
  11    This program is distributed in the hope that it will be useful,
  12    but WITHOUT ANY WARRANTY; without even the implied warranty of
  13    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14    GNU General Public License for more details.
  15
  16    You should have received a copy of the GNU General Public License
  17    along with this program.  If not, see <http://www.gnu.org/licenses/>.
  18 */
  19
  20 #include "includes.h"
  21 #include "winbindd.h"
  22 #include "librpc/gen_ndr/ndr_winbind_c.h"
  23
  24 struct winbindd_ping_dc_state {
  25         const char *dcname;
  26         NTSTATUS result;
  27 };
  28
  29 static void winbindd_ping_dc_done(struct tevent_req *subreq);
  30
  31 struct tevent_req *winbindd_ping_dc_send(TALLOC_CTX *mem_ctx,
  32                                          struct tevent_context *ev,
  33                                          struct winbindd_cli_state *cli,
  34                                          struct winbindd_request *request)
  35 {
  36         struct tevent_req *req, *subreq;
  37         struct winbindd_ping_dc_state *state;
  38         struct winbindd_domain *domain;
  39
  40         req = tevent_req_create(mem_ctx, &state,
  41                                 struct winbindd_ping_dc_state);
  42         if (req == NULL) {
  43                 return NULL;
  44         }
  45
  46         if (request->domain_name[0] == '\0') {
  47                 /* preserve old behavior, when no domain name is given */
  48                 domain = find_our_domain();
  49         } else {
  50                 domain = find_trust_from_name_noinit(request->domain_name);
  51         }
  52         if (domain == NULL) {
  53                 tevent_req_nterror(req, NT_STATUS_NO_SUCH_DOMAIN);
  54                 return tevent_req_post(req, ev);
  55         }
  56         if (domain->internal) {
  57                 const char *d = lp_dnsdomain();
  58                 const char *n = lp_netbios_name();
  59
  60                 /*
  61                  * Internal domains are passdb based, we can always
  62                  * contact them.
  63                  */
  64
  65                 if (d != NULL) {
  66                         char *h;
  67                         h = strlower_talloc(mem_ctx, n);
  68                         if (tevent_req_nomem(h, req)) {
  69                                 return tevent_req_post(req, ev);
  70                         }
  71
  72                         state->dcname = talloc_asprintf(state, "%s.%s", h, d);
  73                         TALLOC_FREE(h);
  74
  75                         if (tevent_req_nomem(state->dcname, req)) {
  76                                 return tevent_req_post(req, ev);
  77                         }
  78                 } else {
  79                         state->dcname = talloc_strdup(state, n);
  80                         if (tevent_req_nomem(state->dcname, req)) {
  81                                 return tevent_req_post(req, ev);
  82                         }
  83                 }
  84
  85                 tevent_req_done(req);
  86                 return tevent_req_post(req, ev);
  87         }
  88
  89         subreq = dcerpc_wbint_PingDc_send(state, ev, dom_child_handle(domain),
  90                                           &state->dcname);
  91         if (tevent_req_nomem(subreq, req)) {
  92                 return tevent_req_post(req, ev);
  93         }
  94         tevent_req_set_callback(subreq, winbindd_ping_dc_done, req);
  95         return req;
  96 }
  97
  98 static void winbindd_ping_dc_done(struct tevent_req *subreq)
  99 {
 100         struct tevent_req *req = tevent_req_callback_data(
 101                 subreq, struct tevent_req);
 102         struct winbindd_ping_dc_state *state = tevent_req_data(
 103                 req, struct winbindd_ping_dc_state);
 104         NTSTATUS status, result;
 105
 106         status = dcerpc_wbint_PingDc_recv(subreq, state, &result);
 107         state->result = result;
 108         if (any_nt_status_not_ok(status, result, &status)) {
 109                 tevent_req_nterror(req, status);
 110                 return;
 111         }
 112         tevent_req_done(req);
 113 }
 114
 115 NTSTATUS winbindd_ping_dc_recv(struct tevent_req *req,
 116                                struct winbindd_response *presp)
 117 {
 118         struct winbindd_ping_dc_state *state = tevent_req_data(
 119                 req, struct winbindd_ping_dc_state);
 120         NTSTATUS status;
 121
 122         if (tevent_req_is_nterror(req, &status)) {
 123                 return status;
 124         }
 125
 126         if (!NT_STATUS_IS_OK(state->result)) {
 127                 set_auth_errors(presp, state->result);
 128         }
 129
 130         if (state->dcname) {
 131                 presp->extra_data.data = talloc_strdup(presp, state->dcname);
 132                 if (presp->extra_data.data == NULL) {
 133                         return NT_STATUS_NO_MEMORY;
 134                 }
 135                 presp->length += strlen((char *)presp->extra_data.data) + 1;
 136         }
 137
 138         tevent_req_received(req);
 139         return NT_STATUS_OK;
 140 }