torture: Fix copy and paste error in debug message.
[kai/samba.git] / source4 / winbind / wb_cmd_getdcname.c
index 3ddcbc896202f7c712cbd2aed6c02de5227ab71b..f3c24825799470ff55d7497203a62d007df21655 100644 (file)
@@ -7,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/>.
 */
 
 #include "includes.h"
 #include "libcli/composite/composite.h"
 #include "winbind/wb_server.h"
-#include "smbd/service_stream.h"
+#include "smbd/service_task.h"
 
-#include "librpc/gen_ndr/ndr_netlogon.h"
-
-static void composite_netr_GetAnyDCName_recv_rpc(struct rpc_request *req);
-
-static struct composite_context *composite_netr_GetAnyDCName_send(struct dcerpc_pipe *p,
-                                                                 TALLOC_CTX *mem_ctx,
-                                                                 struct netr_GetAnyDCName *r)
-{
-       struct composite_context *result;
-       struct rpc_request *req;
-
-       result = talloc(mem_ctx, struct composite_context);
-       if (result == NULL) goto failed;
-       result->state = COMPOSITE_STATE_IN_PROGRESS;
-       result->async.fn = NULL;
-       result->event_ctx = p->conn->event_ctx;
-
-       req = dcerpc_netr_GetAnyDCName_send(p, mem_ctx, r);
-       if (req == NULL) goto failed;
-       req->async.callback = composite_netr_GetAnyDCName_recv_rpc;
-       req->async.private = result;
-       return result;
-
- failed:
-       talloc_free(result);
-       return NULL;
-}
-
-static void composite_netr_GetAnyDCName_recv_rpc(struct rpc_request *req)
-{
-       struct composite_context *ctx =
-               talloc_get_type(req->async.private, struct composite_context);
-
-       ctx->status = dcerpc_ndr_request_recv(req);
-       if (!composite_is_ok(ctx)) return;
-       composite_done(ctx);
-}
-
-NTSTATUS composite_netr_GetAnyDCName_recv(struct composite_context *ctx)
-{
-       NTSTATUS status = composite_wait(ctx);
-       talloc_free(ctx);
-       return status;
-}
+#include "librpc/gen_ndr/ndr_netlogon_c.h"
 
 struct cmd_getdcname_state {
        struct composite_context *ctx;
-       struct wbsrv_domain *domain;
        const char *domain_name;
 
        struct netr_GetAnyDCName g;
 };
 
-static struct composite_context *getdcname_send_req(void *p);
-static NTSTATUS getdcname_recv_req(struct composite_context *ctx, void *p);
+static void getdcname_recv_domain(struct composite_context *ctx);
+static void getdcname_recv_dcname(struct tevent_req *subreq);
 
-struct composite_context *wb_cmd_getdcname_send(struct wbsrv_call *call,
-                                               const char *domain)
+struct composite_context *wb_cmd_getdcname_send(TALLOC_CTX *mem_ctx,
+                                               struct wbsrv_service *service,
+                                               const char *domain_name)
 {
+       struct composite_context *result, *ctx;
        struct cmd_getdcname_state *state;
-       struct wbsrv_service *service = call->wbconn->listen_socket->service;
-
-       state = talloc(NULL, struct cmd_getdcname_state);
-       state->domain = service->domains;
-       state->domain_name = talloc_strdup(state, domain);
-       state->ctx = wb_queue_domain_send(state, state->domain,
-                                         call->event_ctx,
-                                         call->wbconn->conn->msg_ctx,
-                                         getdcname_send_req,
-                                         getdcname_recv_req,
-                                         state);
-       if (state->ctx == NULL) {
-               talloc_free(state);
-               return NULL;
-       }
-       state->ctx->private_data = state;
-       return state->ctx;
+
+       result = composite_create(mem_ctx, service->task->event_ctx);
+       if (result == NULL) goto failed;
+
+       state = talloc(result, struct cmd_getdcname_state);
+       if (state == NULL) goto failed;
+       state->ctx = result;
+       result->private_data = state;
+
+       state->domain_name = talloc_strdup(state, domain_name);
+       if (state->domain_name == NULL) goto failed;
+
+       ctx = wb_sid2domain_send(state, service, service->primary_sid);
+       if (ctx == NULL) goto failed;
+
+       ctx->async.fn = getdcname_recv_domain;
+       ctx->async.private_data = state;
+       return result;
+
+ failed:
+       talloc_free(result);
+       return NULL;
 }
 
-static struct composite_context *getdcname_send_req(void *p)
+static void getdcname_recv_domain(struct composite_context *ctx)
 {
        struct cmd_getdcname_state *state =
-               talloc_get_type(p, struct cmd_getdcname_state);
+               talloc_get_type(ctx->async.private_data,
+                               struct cmd_getdcname_state);
+       struct wbsrv_domain *domain;
+       struct tevent_req *subreq;
+
+       state->ctx->status = wb_sid2domain_recv(ctx, &domain);
+       if (!composite_is_ok(state->ctx)) return;
 
        state->g.in.logon_server = talloc_asprintf(
                state, "\\\\%s",
-               dcerpc_server_name(state->domain->netlogon_pipe));
+               dcerpc_server_name(domain->netlogon_pipe));
        state->g.in.domainname = state->domain_name;
+       state->g.out.dcname = talloc(state, const char *);
 
-       return composite_netr_GetAnyDCName_send(state->domain->netlogon_pipe,
-                                               state, &state->g);
+       subreq = dcerpc_netr_GetAnyDCName_r_send(state,
+                                                state->ctx->event_ctx,
+                                                domain->netlogon_pipe->binding_handle,
+                                                &state->g);
+       if (composite_nomem(subreq, state->ctx)) return;
+
+       tevent_req_set_callback(subreq, getdcname_recv_dcname, state);
 }
 
-static NTSTATUS getdcname_recv_req(struct composite_context *ctx, void *p)
+static void getdcname_recv_dcname(struct tevent_req *subreq)
 {
        struct cmd_getdcname_state *state =
-               talloc_get_type(p, struct cmd_getdcname_state);
-       NTSTATUS status;
-
-       status = composite_netr_GetAnyDCName_recv(ctx);
-       NT_STATUS_NOT_OK_RETURN(status);
+               tevent_req_callback_data(subreq,
+               struct cmd_getdcname_state);
 
-       if (!W_ERROR_IS_OK(state->g.out.result)) {
-               return werror_to_ntstatus(state->g.out.result);
-       }
+       state->ctx->status = dcerpc_netr_GetAnyDCName_r_recv(subreq, state);
+       TALLOC_FREE(subreq);
+       if (!composite_is_ok(state->ctx)) return;
+       state->ctx->status = werror_to_ntstatus(state->g.out.result);
+       if (!composite_is_ok(state->ctx)) return;
 
-       return NT_STATUS_OK;
+       composite_done(state->ctx);
 }
 
 NTSTATUS wb_cmd_getdcname_recv(struct composite_context *c,
@@ -141,8 +114,13 @@ NTSTATUS wb_cmd_getdcname_recv(struct composite_context *c,
        struct cmd_getdcname_state *state =
                talloc_get_type(c->private_data, struct cmd_getdcname_state);
        NTSTATUS status = composite_wait(c);
+       if (NT_STATUS_EQUAL(status, NT_STATUS_NO_SUCH_DOMAIN)) {
+               /* special case: queried DC is PDC */
+               state->g.out.dcname = &state->g.in.logon_server;
+               status = NT_STATUS_OK;
+       }
        if (NT_STATUS_IS_OK(status)) {
-               const char *p = state->g.out.dcname;
+               const char *p = *(state->g.out.dcname);
                if (*p == '\\') p += 1;
                if (*p == '\\') p += 1;
                *dcname = talloc_strdup(mem_ctx, p);