s4-idmap: Add mapping using uidNumber and gidNumber like idmap_ad

[samba.git] / source4 / winbind / wb_irpc.c

diff --git a/source4/winbind/wb_irpc.c b/source4/winbind/wb_irpc.c
index efbc310b68180d6f50665c15ecec42d1f3e2baa4..2f2b0780c7b1a18fcd0b58b7fdc9ee2f81be9f41 100644 (file)
--- a/source4/winbind/wb_irpc.c
+++ b/source4/winbind/wb_irpc.c
@@ -22,7 +22,6 @@
 #include "winbind/wb_server.h"
 #include "lib/messaging/irpc.h"
 #include "libcli/composite/composite.h"
-#include "libcli/security/security.h"
 #include "librpc/gen_ndr/ndr_winbind.h"
 #include "smbd/service_task.h"
 
@@ -72,6 +71,52 @@ static void wb_irpc_SamLogon_callback(struct composite_context *ctx)
        irpc_send_reply(s->msg, status);
 }
 
+struct wb_irpc_DsrUpdateReadOnlyServerDnsRecords_state {
+       struct irpc_message *msg;
+       struct winbind_DsrUpdateReadOnlyServerDnsRecords *req;
+};
+
+static void wb_irpc_DsrUpdateReadOnlyServerDnsRecords_callback(struct composite_context *ctx);
+
+static NTSTATUS wb_irpc_DsrUpdateReadOnlyServerDnsRecords(struct irpc_message *msg,
+                                struct winbind_DsrUpdateReadOnlyServerDnsRecords *req)
+{
+       struct wbsrv_service *service = talloc_get_type(msg->private_data,
+                                       struct wbsrv_service);
+       struct wb_irpc_DsrUpdateReadOnlyServerDnsRecords_state *s;
+       struct composite_context *ctx;
+
+       DEBUG(5, ("wb_irpc_DsrUpdateReadOnlyServerDnsRecords called\n"));
+
+       s = talloc(msg, struct wb_irpc_DsrUpdateReadOnlyServerDnsRecords_state);
+       NT_STATUS_HAVE_NO_MEMORY(s);
+
+       s->msg = msg;
+       s->req = req;
+
+       ctx = wb_update_rodc_dns_send(msg, service, req);
+       NT_STATUS_HAVE_NO_MEMORY(ctx);
+
+       ctx->async.fn = wb_irpc_DsrUpdateReadOnlyServerDnsRecords_callback;
+       ctx->async.private_data = s;
+
+       msg->defer_reply = true;
+       return NT_STATUS_OK;
+}
+
+static void wb_irpc_DsrUpdateReadOnlyServerDnsRecords_callback(struct composite_context *ctx)
+{
+       struct wb_irpc_DsrUpdateReadOnlyServerDnsRecords_state *s = talloc_get_type(ctx->async.private_data,
+                                          struct wb_irpc_DsrUpdateReadOnlyServerDnsRecords_state);
+       NTSTATUS status;
+
+       DEBUG(5, ("wb_irpc_DsrUpdateReadOnlyServerDnsRecords_callback called\n"));
+
+       status = wb_update_rodc_dns_recv(ctx, s, s->req);
+
+       irpc_send_reply(s->msg, status);
+}
+
 struct wb_irpc_get_idmap_state {
        struct irpc_message *msg;
        struct winbind_get_idmap *req;
@@ -127,7 +172,7 @@ static void wb_irpc_get_idmap_callback(struct composite_context *ctx)
 
        switch(s->level) {
                case WINBIND_IDMAP_LEVEL_SIDS_TO_XIDS:
-                       status = wb_sids2xids_recv(ctx, &s->req->out.ids);
+                       status = wb_sids2xids_recv(ctx, &s->req->out.ids, NULL);
                        break;
                case WINBIND_IDMAP_LEVEL_XIDS_TO_SIDS:
                        status = wb_xids2sids_recv(ctx, &s->req->out.ids);
@@ -150,6 +195,10 @@ NTSTATUS wbsrv_init_irpc(struct wbsrv_service *service)
                               wb_irpc_SamLogon, service);
        NT_STATUS_NOT_OK_RETURN(status);
 
+       status = IRPC_REGISTER(service->task->msg_ctx, winbind, WINBIND_DSRUPDATEREADONLYSERVERDNSRECORDS,
+                              wb_irpc_DsrUpdateReadOnlyServerDnsRecords, service);
+       NT_STATUS_NOT_OK_RETURN(status);
+
        status = IRPC_REGISTER(service->task->msg_ctx, winbind, WINBIND_GET_IDMAP,
                               wb_irpc_get_idmap, service);
        NT_STATUS_NOT_OK_RETURN(status);