r12608: Remove some unused #include lines.
[jelmer/samba4-debian.git] / source / ntptr / simple_ldb / ntptr_simple_ldb.c
index dc79ba8a1adac109192f2cb9ba569d73a7d938f5..8135625734c2c8a938caa92e381d61c3e35c2852 100644 (file)
@@ -32,7 +32,8 @@
 #include "ntptr/ntptr.h"
 #include "librpc/gen_ndr/ndr_spoolss.h"
 #include "lib/ldb/include/ldb.h"
-#include "db_wrap.h"
+#include "auth/auth.h"
+#include "dsdb/samdb/samdb.h"
 
 /*
   connect to the SPOOLSS database
  */
 static struct ldb_context *sptr_db_connect(TALLOC_CTX *mem_ctx)
 {
-       return ldb_wrap_connect(mem_ctx, lp_spoolss_url(), 0, NULL);
+       return ldb_wrap_connect(mem_ctx, lp_spoolss_url(), system_session(mem_ctx), 
+                               NULL, 0, NULL);
 }
 
 static int sptr_db_search(struct ldb_context *ldb,
                          TALLOC_CTX *mem_ctx,
-                         const char *basedn,
+                         const struct ldb_dn *basedn,
                          struct ldb_message ***res,
                          const char * const *attrs,
                          const char *format, ...) PRINTF_ATTRIBUTE(6,7);
 
 static int sptr_db_search(struct ldb_context *ldb,
                          TALLOC_CTX *mem_ctx,
-                         const char *basedn,
+                         const struct ldb_dn *basedn,
                          struct ldb_message ***res,
                          const char * const *attrs,
                          const char *format, ...)
@@ -105,7 +107,9 @@ static WERROR sptr_OpenPrintServer(struct ntptr_context *ntptr, TALLOC_CTX *mem_
 
        server->type            = NTPTR_HANDLE_SERVER;
        server->ntptr           = ntptr;
-       server->access_mask     = r->in.access_mask;
+       server->object_name     = talloc_strdup(server, server_name);
+       W_ERROR_HAVE_NO_MEMORY(server->object_name);
+       server->access_mask     = 0;
        server->private_data    = NULL;
 
        *_server = server;
@@ -158,6 +162,53 @@ static WERROR sptr_GetPrintServerData(struct ntptr_GenericHandle *server, TALLOC
                r->out.type             = SPOOLSS_PRINTER_DATA_TYPE_UINT32;
                r->out.data.value       = 1;
                return WERR_OK;
+       } else if (strcmp("OSVersion", r->in.value_name) == 0) {
+               DATA_BLOB blob;
+               NTSTATUS status;
+               struct spoolss_OSVersion os;
+
+               os.major                = dcesrv_common_get_version_major(mem_ctx, NULL);
+               os.minor                = dcesrv_common_get_version_minor(mem_ctx, NULL);
+               os.build                = dcesrv_common_get_version_build(mem_ctx, NULL);
+               os.extra_string         = "";
+
+               status = ndr_push_struct_blob(&blob, mem_ctx, &os, (ndr_push_flags_fn_t)ndr_push_spoolss_OSVersion);
+               if (!NT_STATUS_IS_OK(status)) {
+                       return WERR_GENERAL_FAILURE;
+               }
+
+               r->out.type             = SPOOLSS_PRINTER_DATA_TYPE_BINARY;
+               r->out.data.binary      = blob;
+               return WERR_OK;
+       } else if (strcmp("OSVersionEx", r->in.value_name) == 0) {
+               DATA_BLOB blob;
+               NTSTATUS status;
+               struct spoolss_OSVersionEx os_ex;
+
+               os_ex.major             = dcesrv_common_get_version_major(mem_ctx, NULL);
+               os_ex.minor             = dcesrv_common_get_version_minor(mem_ctx, NULL);
+               os_ex.build             = dcesrv_common_get_version_build(mem_ctx, NULL);
+               os_ex.extra_string              = "";
+               os_ex.unknown2          = 0;
+               os_ex.unknown3          = 0;
+
+               status = ndr_push_struct_blob(&blob, mem_ctx, &os_ex, (ndr_push_flags_fn_t)ndr_push_spoolss_OSVersionEx);
+               if (!NT_STATUS_IS_OK(status)) {
+                       return WERR_GENERAL_FAILURE;
+               }
+
+               r->out.type             = SPOOLSS_PRINTER_DATA_TYPE_BINARY;
+               r->out.data.binary      = blob;
+               return WERR_OK;
+       } else if (strcmp("DNSMachineName", r->in.value_name) == 0) {
+               if (!lp_realm()) return WERR_INVALID_PARAM;
+
+               r->out.type             = SPOOLSS_PRINTER_DATA_TYPE_STRING;
+               r->out.data.string      = talloc_asprintf(mem_ctx, "%s.%s",
+                                                                  lp_netbios_name(),
+                                                                  lp_realm());
+               W_ERROR_HAVE_NO_MEMORY(r->out.data.string);
+               return WERR_OK;
        }
 
        return WERR_INVALID_PARAM;
@@ -173,8 +224,9 @@ static WERROR sptr_EnumPrintServerForms(struct ntptr_GenericHandle *server, TALL
        int i;
        union spoolss_FormInfo *info;
 
-       count = sptr_db_search(sptr_db, mem_ctx, "CN=Forms,CN=PrintServer", &msgs, NULL,
-                              "(&(objectclass=form))");
+       count = sptr_db_search(sptr_db, mem_ctx,
+                               ldb_dn_explode(mem_ctx, "CN=Forms,CN=PrintServer"),
+                               &msgs, NULL, "(&(objectClass=form))");
 
        if (count == 0) return WERR_OK;
        if (count < 0) return WERR_GENERAL_FAILURE;
@@ -187,16 +239,16 @@ static WERROR sptr_EnumPrintServerForms(struct ntptr_GenericHandle *server, TALL
                for (i=0; i < count; i++) {
                        info[i].info1.flags             = samdb_result_uint(msgs[i], "flags", SPOOLSS_FORM_BUILTIN);
 
-                       info[i].info1.form_name         = samdb_result_string(msgs[i], "form_name", NULL);
+                       info[i].info1.form_name         = samdb_result_string(msgs[i], "form-name", NULL);
                        W_ERROR_HAVE_NO_MEMORY(info[i].info1.form_name);
 
-                       info[i].info1.size.width        = samdb_result_uint(msgs[i], "size_width", 0);
-                       info[i].info1.size.height       = samdb_result_uint(msgs[i], "size_height", 0);
+                       info[i].info1.size.width        = samdb_result_uint(msgs[i], "size-width", 0);
+                       info[i].info1.size.height       = samdb_result_uint(msgs[i], "size-height", 0);
 
-                       info[i].info1.area.left         = samdb_result_uint(msgs[i], "area_left", 0);
-                       info[i].info1.area.top          = samdb_result_uint(msgs[i], "area_top", 0);
-                       info[i].info1.area.right        = samdb_result_uint(msgs[i], "area_right", 0);
-                       info[i].info1.area.bottom       = samdb_result_uint(msgs[i], "area_bottom", 0);
+                       info[i].info1.area.left         = samdb_result_uint(msgs[i], "area-left", 0);
+                       info[i].info1.area.top          = samdb_result_uint(msgs[i], "area-top", 0);
+                       info[i].info1.area.right        = samdb_result_uint(msgs[i], "area-right", 0);
+                       info[i].info1.area.bottom       = samdb_result_uint(msgs[i], "area-bottom", 0);
                }
                break;
        default:
@@ -227,8 +279,9 @@ static WERROR sptr_AddPrintServerForm(struct ntptr_GenericHandle *server, TALLOC
                if (!r->in.info.info1) {
                        return WERR_FOOBAR;
                }
-               count = sptr_db_search(sptr_db, mem_ctx, "CN=Forms,CN=PrintServer", &msgs, attrs,
-                                      "(&(form_name=%s)(objectClass=form))",
+               count = sptr_db_search(sptr_db, mem_ctx,
+                                      ldb_dn_explode(mem_ctx, "CN=Forms,CN=PrintServer"),
+                                      &msgs, attrs, "(&(form-name=%s)(objectClass=form))",
                                       r->in.info.info1->form_name);
 
                if (count == 1) return WERR_FOOBAR;
@@ -243,21 +296,22 @@ static WERROR sptr_AddPrintServerForm(struct ntptr_GenericHandle *server, TALLOC
                W_ERROR_HAVE_NO_MEMORY(msg);
 
                /* add core elements to the ldb_message for the Form */
-               msg->dn = talloc_asprintf(msg, "form_name=%s,CN=Forms,CN=PrintServer",
-                                               r->in.info.info1->form_name);
+               msg->dn = ldb_dn_build_child(msg,
+                                            "form-name", r->in.info.info1->form_name,
+                                            ldb_dn_explode(msg, "CN=Forms,CN=PrintServer"));
                SET_STRING(sptr_db, msg, "objectClass", "form");
 
                SET_UINT(sptr_db, msg, "flags", r->in.info.info1->flags);
 
-               SET_STRING(sptr_db, msg, "form_name", r->in.info.info1->form_name);
+               SET_STRING(sptr_db, msg, "form-name", r->in.info.info1->form_name);
 
-               SET_UINT(sptr_db, msg, "size_width", r->in.info.info1->size.width);
-               SET_UINT(sptr_db, msg, "size_height", r->in.info.info1->size.height);
+               SET_UINT(sptr_db, msg, "size-width", r->in.info.info1->size.width);
+               SET_UINT(sptr_db, msg, "size-height", r->in.info.info1->size.height);
 
-               SET_UINT(sptr_db, msg, "area_left", r->in.info.info1->area.left);
-               SET_UINT(sptr_db, msg, "area_top", r->in.info.info1->area.top);
-               SET_UINT(sptr_db, msg, "area_right", r->in.info.info1->area.right);
-               SET_UINT(sptr_db, msg, "area_bottom", r->in.info.info1->area.bottom);
+               SET_UINT(sptr_db, msg, "area-left", r->in.info.info1->area.left);
+               SET_UINT(sptr_db, msg, "area-top", r->in.info.info1->area.top);
+               SET_UINT(sptr_db, msg, "area-right", r->in.info.info1->area.right);
+               SET_UINT(sptr_db, msg, "area-bottom", r->in.info.info1->area.bottom);
                break;
        default:
                return WERR_UNKNOWN_LEVEL;
@@ -271,54 +325,6 @@ static WERROR sptr_AddPrintServerForm(struct ntptr_GenericHandle *server, TALLOC
        return WERR_OK;
 }
 
-static WERROR sptr_GetPrintServerForm(struct ntptr_GenericHandle *server, TALLOC_CTX *mem_ctx,
-                                     struct spoolss_GetForm *r)
-{
-       struct ldb_context *sptr_db = talloc_get_type(server->ntptr->private_data, struct ldb_context);
-       struct ldb_message **msgs;
-       int count;
-       union spoolss_FormInfo *info;
-
-       /* TODO: do checks access here
-        * if (!(server->access_mask & desired_access)) {
-        *      return WERR_FOOBAR;
-        * }
-        */
-
-       count = sptr_db_search(sptr_db, mem_ctx, "CN=Forms,CN=PrintServer", &msgs, NULL,
-                              "(&(form_name=%s)(objectClass=form))",
-                              r->in.form_name);
-
-       if (count == 0) return WERR_FOOBAR;
-       if (count > 1) return WERR_FOOBAR;
-       if (count < 0) return WERR_GENERAL_FAILURE;
-
-       info = talloc(mem_ctx, union spoolss_FormInfo);
-       W_ERROR_HAVE_NO_MEMORY(info);
-
-       switch (r->in.level) {
-       case 1:
-               info->info1.flags       = samdb_result_uint(msgs[0], "flags", SPOOLSS_FORM_BUILTIN);
-
-               info->info1.form_name   = samdb_result_string(msgs[0], "form_name", NULL);
-               W_ERROR_HAVE_NO_MEMORY(info->info1.form_name);
-
-               info->info1.size.width  = samdb_result_uint(msgs[0], "size_width", 0);
-               info->info1.size.height = samdb_result_uint(msgs[0], "size_height", 0);
-
-               info->info1.area.left   = samdb_result_uint(msgs[0], "area_left", 0);
-               info->info1.area.top    = samdb_result_uint(msgs[0], "area_top", 0);
-               info->info1.area.right  = samdb_result_uint(msgs[0], "area_right", 0);
-               info->info1.area.bottom = samdb_result_uint(msgs[0], "area_bottom", 0);
-               break;
-       default:
-               return WERR_UNKNOWN_LEVEL;
-       }
-
-       r->out.info     = info;
-       return WERR_OK;
-}
-
 static WERROR sptr_SetPrintServerForm(struct ntptr_GenericHandle *server, TALLOC_CTX *mem_ctx,
                                      struct spoolss_SetForm *r)
 {
@@ -340,8 +346,9 @@ static WERROR sptr_SetPrintServerForm(struct ntptr_GenericHandle *server, TALLOC
                        return WERR_FOOBAR;
                }
 
-               count = sptr_db_search(sptr_db, mem_ctx, "CN=Forms,CN=PrintServer", &msgs, attrs,
-                                      "(&(form_name=%s)(objectClass=form))",
+               count = sptr_db_search(sptr_db, mem_ctx,
+                                      ldb_dn_explode(mem_ctx, "CN=Forms,CN=PrintServer"),
+                                      &msgs, attrs, "(&(form-name=%s)(objectClass=form))",
                                       r->in.info.info1->form_name);
 
                if (count == 0) return WERR_FOOBAR;
@@ -361,15 +368,15 @@ static WERROR sptr_SetPrintServerForm(struct ntptr_GenericHandle *server, TALLOC
 
                SET_UINT(sptr_db, msg, "flags", r->in.info.info1->flags);
 
-               SET_STRING(sptr_db, msg, "form_name", r->in.info.info1->form_name);
+               SET_STRING(sptr_db, msg, "form-name", r->in.info.info1->form_name);
 
-               SET_UINT(sptr_db, msg, "size_width", r->in.info.info1->size.width);
-               SET_UINT(sptr_db, msg, "size_height", r->in.info.info1->size.height);
+               SET_UINT(sptr_db, msg, "size-width", r->in.info.info1->size.width);
+               SET_UINT(sptr_db, msg, "size-height", r->in.info.info1->size.height);
 
-               SET_UINT(sptr_db, msg, "area_left", r->in.info.info1->area.left);
-               SET_UINT(sptr_db, msg, "area_top", r->in.info.info1->area.top);
-               SET_UINT(sptr_db, msg, "area_right", r->in.info.info1->area.right);
-               SET_UINT(sptr_db, msg, "area_bottom", r->in.info.info1->area.bottom);
+               SET_UINT(sptr_db, msg, "area-left", r->in.info.info1->area.left);
+               SET_UINT(sptr_db, msg, "area-top", r->in.info.info1->area.top);
+               SET_UINT(sptr_db, msg, "area-right", r->in.info.info1->area.right);
+               SET_UINT(sptr_db, msg, "area-bottom", r->in.info.info1->area.bottom);
                break;
        default:
                return WERR_UNKNOWN_LEVEL;
@@ -402,8 +409,9 @@ static WERROR sptr_DeletePrintServerForm(struct ntptr_GenericHandle *server, TAL
                return WERR_FOOBAR;
        }
 
-       count = sptr_db_search(sptr_db, mem_ctx, "CN=Forms,CN=PrintServer", &msgs, attrs,
-                              "(&(form_name=%s)(objectclass=form))",
+       count = sptr_db_search(sptr_db, mem_ctx,
+                              ldb_dn_explode(mem_ctx, "CN=Forms,CN=PrintServer"),
+                              &msgs, attrs, "(&(form-name=%s)(objectclass=form))",
                               r->in.form_name);
 
        if (count == 0) return WERR_FOOBAR;
@@ -612,22 +620,22 @@ static WERROR sptr_EnumPorts(struct ntptr_context *ntptr, TALLOC_CTX *mem_ctx,
        switch (r->in.level) {
        case 1:
                for (i = 0; i < count; i++) {
-                       info[i].info1.port_name         = samdb_result_string(msgs[i], "port_name", "");
+                       info[i].info1.port_name         = samdb_result_string(msgs[i], "port-name", "");
                        W_ERROR_HAVE_NO_MEMORY(info[i].info1.port_name);
                }
                break;
        case 2:
                for (i=0; i < count; i++) {
-                       info[i].info2.port_name         = samdb_result_string(msgs[i], "port_name", "");
+                       info[i].info2.port_name         = samdb_result_string(msgs[i], "port-name", "");
                        W_ERROR_HAVE_NO_MEMORY(info[i].info2.port_name);
 
-                       info[i].info2.monitor_name      = samdb_result_string(msgs[i], "monitor_name", "");
+                       info[i].info2.monitor_name      = samdb_result_string(msgs[i], "monitor-name", "");
                        W_ERROR_HAVE_NO_MEMORY(info[i].info2.monitor_name);
 
                        info[i].info2.description       = samdb_result_string(msgs[i], "description", "");
                        W_ERROR_HAVE_NO_MEMORY(info[i].info2.description);
 
-                       info[i].info2.port_type         = samdb_result_uint(msgs[i], "port_type", SPOOLSS_PORT_TYPE_WRITE);
+                       info[i].info2.port_type         = samdb_result_uint(msgs[i], "port-type", SPOOLSS_PORT_TYPE_WRITE);
                        info[i].info2.reserved          = samdb_result_uint(msgs[i], "reserved", 0);
                }
                break;
@@ -662,19 +670,19 @@ static WERROR sptr_EnumMonitors(struct ntptr_context *ntptr, TALLOC_CTX *mem_ctx
        switch (r->in.level) {
        case 1:
                for (i = 0; i < count; i++) {
-                       info[i].info1.monitor_name      = samdb_result_string(msgs[i], "monitor_name", "");
+                       info[i].info1.monitor_name      = samdb_result_string(msgs[i], "monitor-name", "");
                        W_ERROR_HAVE_NO_MEMORY(info[i].info1.monitor_name);
                }
                break;
        case 2:
                for (i=0; i < count; i++) {
-                       info[i].info2.monitor_name      = samdb_result_string(msgs[i], "monitor_name", "");
+                       info[i].info2.monitor_name      = samdb_result_string(msgs[i], "monitor-name", "");
                        W_ERROR_HAVE_NO_MEMORY(info[i].info2.monitor_name);
 
                        info[i].info2.environment       = samdb_result_string(msgs[i], "environment", "");
                        W_ERROR_HAVE_NO_MEMORY(info[i].info2.environment);
 
-                       info[i].info2.dll_name          = samdb_result_string(msgs[i], "dll_name", "");
+                       info[i].info2.dll_name          = samdb_result_string(msgs[i], "dll-name", "");
                        W_ERROR_HAVE_NO_MEMORY(info[i].info2.dll_name);
                }
                break;
@@ -688,6 +696,60 @@ static WERROR sptr_EnumMonitors(struct ntptr_context *ntptr, TALLOC_CTX *mem_ctx
        return WERR_OK;
 }
 
+/* Printer Form functions */
+static WERROR sptr_GetPrinterForm(struct ntptr_GenericHandle *printer, TALLOC_CTX *mem_ctx,
+                                 struct spoolss_GetForm *r)
+{
+       struct ldb_context *sptr_db = talloc_get_type(printer->ntptr->private_data, struct ldb_context);
+       struct ldb_message **msgs;
+       const struct ldb_dn *base_dn;
+       int count;
+       union spoolss_FormInfo *info;
+
+       /* TODO: do checks access here
+        * if (!(printer->access_mask & desired_access)) {
+        *      return WERR_FOOBAR;
+        * }
+        */
+
+       base_dn = ldb_dn_string_compose(mem_ctx, NULL, "CN=Forms, CN=%s, CN=Printers", printer->object_name);
+       W_ERROR_HAVE_NO_MEMORY(base_dn);
+
+       count = sptr_db_search(sptr_db, mem_ctx, base_dn, &msgs, NULL,
+                              "(&(form-name=%s)(objectClass=form))",
+                              r->in.form_name);
+
+       if (count == 0) return WERR_FOOBAR;
+       if (count > 1) return WERR_FOOBAR;
+       if (count < 0) return WERR_GENERAL_FAILURE;
+
+       info = talloc(mem_ctx, union spoolss_FormInfo);
+       W_ERROR_HAVE_NO_MEMORY(info);
+
+       switch (r->in.level) {
+       case 1:
+               info->info1.flags       = samdb_result_uint(msgs[0], "flags", SPOOLSS_FORM_BUILTIN);
+
+               info->info1.form_name   = samdb_result_string(msgs[0], "form-name", NULL);
+               W_ERROR_HAVE_NO_MEMORY(info->info1.form_name);
+
+               info->info1.size.width  = samdb_result_uint(msgs[0], "size-width", 0);
+               info->info1.size.height = samdb_result_uint(msgs[0], "size-height", 0);
+
+               info->info1.area.left   = samdb_result_uint(msgs[0], "area-left", 0);
+               info->info1.area.top    = samdb_result_uint(msgs[0], "area-top", 0);
+               info->info1.area.right  = samdb_result_uint(msgs[0], "area-right", 0);
+               info->info1.area.bottom = samdb_result_uint(msgs[0], "area-bottom", 0);
+               break;
+       default:
+               return WERR_UNKNOWN_LEVEL;
+       }
+
+       r->out.info     = info;
+       return WERR_OK;
+}
+
+
 /*
   initialialise the simble ldb backend, registering ourselves with the ntptr subsystem
  */
@@ -707,7 +769,6 @@ static const struct ntptr_ops ntptr_simple_ldb_ops = {
        /* PrintServer Form functions */
        .EnumPrintServerForms           = sptr_EnumPrintServerForms,
        .AddPrintServerForm             = sptr_AddPrintServerForm,
-       .GetPrintServerForm             = sptr_GetPrintServerForm,
        .SetPrintServerForm             = sptr_SetPrintServerForm,
        .DeletePrintServerForm          = sptr_DeletePrintServerForm,
 
@@ -748,8 +809,8 @@ static const struct ntptr_ops ntptr_simple_ldb_ops = {
        /* Printer Form functions */
 /*     .EnumPrinterForms               = sptr_EnumPrinterForms,
        .AddPrinterForm                 = sptr_AddPrinterForm,
-       .GetPrinterForm                 = sptr_GetPrinterForm,
-       .SetPrinterForm                 = sptr_SetPrinterForm,
+*/     .GetPrinterForm                 = sptr_GetPrinterForm,
+/*     .SetPrinterForm                 = sptr_SetPrinterForm,
        .DeletePrinterForm              = sptr_DeletePrinterForm,
 */
        /* Printer Job functions */