r4088: Get medieval on our ass about malloc.... :-). Take control of all our allocation
[samba.git] / source3 / libads / ldap_printer.c
index 0185bf78113cd92d62081e2dc9b4e42b18491750..68e6735891038a1de49a390e0cc6e8abb8189605 100644 (file)
@@ -1,7 +1,7 @@
 /* 
    Unix SMB/CIFS implementation.
    ads (active directory) printer utility library
-   Copyright (C) Jim McDonough 2002
+   Copyright (C) Jim McDonough <jmcd@us.ibm.com> 2002
    
    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
@@ -31,12 +31,12 @@ ADS_STATUS ads_find_printer_on_server(ADS_STRUCT *ads, void **res,
                                      const char *printer, const char *servername)
 {
        ADS_STATUS status;
-       char *srv_dn, **srv_cn, *exp;
+       char *srv_dn, **srv_cn, *s;
        const char *attrs[] = {"*", "nTSecurityDescriptor", NULL};
 
        status = ads_find_machine_acct(ads, res, servername);
        if (!ADS_ERR_OK(status)) {
-               DEBUG(1, ("ads_add_printer: cannot find host %s in ads\n",
+               DEBUG(1, ("ads_find_printer_on_server: cannot find host %s in ads\n",
                          servername));
                return status;
        }
@@ -44,152 +44,310 @@ ADS_STATUS ads_find_printer_on_server(ADS_STRUCT *ads, void **res,
        srv_cn = ldap_explode_dn(srv_dn, 1);
        ads_msgfree(ads, *res);
 
-       asprintf(&exp, "(cn=%s-%s)", srv_cn[0], printer);
-       status = ads_search(ads, res, exp, attrs);
+       asprintf(&s, "(cn=%s-%s)", srv_cn[0], printer);
+       status = ads_search(ads, res, s, attrs);
 
        ldap_memfree(srv_dn);
        ldap_value_free(srv_cn);
-       free(exp);
+       free(s);
        return status;  
 }
 
+ADS_STATUS ads_find_printers(ADS_STRUCT *ads, void **res)
+{
+       char *ldap_expr;
+       const char *attrs[] = { "objectClass", "printerName", "location", "driverName",
+                               "serverName", "description", NULL };
+
+       /* For the moment only display all printers */
+
+       ldap_expr = "(&(!(showInAdvancedViewOnly=TRUE))(uncName=*)"
+               "(objectCategory=printQueue))";
+
+       return ads_search(ads, res, ldap_expr, attrs);
+}
+
 /*
-  modify an entire printer entry in the directory
+  modify a printer entry in the directory
 */
 ADS_STATUS ads_mod_printer_entry(ADS_STRUCT *ads, char *prt_dn,
-                                const ADS_PRINTER_ENTRY *prt)
+                                TALLOC_CTX *ctx, const ADS_MODLIST *mods)
 {
-       ADS_MODLIST mods;
-       ADS_STATUS status;
-       TALLOC_CTX *ctx;
-
-       if (!(ctx = talloc_init_named("mod_printer_entry")))
-               return ADS_ERROR(LDAP_NO_MEMORY);
-
-       /* allocate the list */
-       mods = ads_init_mods(ctx);
-
-       /* add the attributes to the list - required ones first */
-       ads_mod_str(ctx, &mods, "printerName", prt->printerName);
-       ads_mod_str(ctx, &mods, "serverName", prt->serverName);
-       ads_mod_str(ctx, &mods, "shortServerName", prt->shortServerName);
-       ads_mod_str(ctx, &mods, "uNCName", prt->uNCName);
-       ads_mod_str(ctx, &mods, "versionNumber", prt->versionNumber);
-
-       /* now the optional ones */
-       ads_mod_strlist(ctx, &mods, "description", (const char **)prt->description);
-       ads_mod_str(ctx, &mods, "assetNumber",prt->assetNumber);
-       ads_mod_str(ctx, &mods, "bytesPerMinute",prt->bytesPerMinute);
-       ads_mod_str(ctx, &mods, "defaultPriority",prt->defaultPriority);
-       ads_mod_str(ctx, &mods, "driverName", prt->driverName);
-       ads_mod_str(ctx, &mods, "driverVersion",prt->driverVersion);
-       ads_mod_str(ctx, &mods, "location", prt->location);
-       ads_mod_str(ctx, &mods, "operatingSystem",prt->operatingSystem);
-       ads_mod_str(ctx, &mods, "operatingSystemHotfix",
-                    prt->operatingSystemHotfix);
-       ads_mod_str(ctx, &mods, "operatingSystemServicePack",
-                    prt->operatingSystemServicePack);
-       ads_mod_str(ctx, &mods, "operatingSystemVersion",
-                    prt->operatingSystemVersion);
-       ads_mod_str(ctx, &mods, "physicalLocationObject",
-                    prt->physicalLocationObject);
-       ads_mod_strlist(ctx, &mods, "portName", (const char **)prt->portName);
-       ads_mod_str(ctx, &mods, "printStartTime", prt->printStartTime);
-       ads_mod_str(ctx, &mods, "printEndTime", prt->printEndTime);
-       ads_mod_strlist(ctx, &mods, "printBinNames", (const char **)prt->printBinNames);
-       /*... and many others */
-
-       /* do the ldap modify */
-       status = ads_gen_mod(ads, prt_dn, mods);
-
-       /* free mod list, mods, and values */
-       talloc_destroy(ctx); 
-
-       return status;
+       return ads_gen_mod(ads, prt_dn, *mods);
 }
 
 /*
   add a printer to the directory
 */
-static ADS_STATUS ads_add_printer_entry(ADS_STRUCT *ads, char *prt_dn,
-                                       const ADS_PRINTER_ENTRY *prt)
+ADS_STATUS ads_add_printer_entry(ADS_STRUCT *ads, char *prt_dn,
+                                       TALLOC_CTX *ctx, ADS_MODLIST *mods)
 {
-        ADS_STATUS status;
-       TALLOC_CTX *ctx;
-       ADS_MODLIST mods;
+       ads_mod_str(ctx, mods, "objectClass", "printQueue");
+       return ads_gen_add(ads, prt_dn, *mods);
+}
 
-       if (!(ctx = talloc_init_named("add_printer_entry")))
-               return ADS_ERROR(LDAP_NO_MEMORY);
+/*
+  map a REG_SZ to an ldap mod
+*/
+static BOOL map_sz(TALLOC_CTX *ctx, ADS_MODLIST *mods, 
+                           const REGISTRY_VALUE *value)
+{
+       char *str_value = NULL;
+       ADS_STATUS status;
 
-       if (!(mods = ads_init_mods(ctx)))
-               return ADS_ERROR(LDAP_NO_MEMORY);
+       if (value->type != REG_SZ)
+               return False;
 
-       /* These are the fields a printQueue must contain */
-       ads_mod_str(ctx, &mods, "uNCName", prt->uNCName);
-       ads_mod_str(ctx, &mods, "versionNumber", prt->versionNumber);
-       ads_mod_str(ctx, &mods, "serverName", prt->serverName);
-       ads_mod_str(ctx, &mods, "shortServerName", prt->shortServerName);
-       ads_mod_str(ctx, &mods, "printerName", prt->printerName);
-       ads_mod_str(ctx, &mods, "objectClass", "printQueue");
+       if (value->size && *((smb_ucs2_t *) value->data_p)) {
+               pull_ucs2_talloc(ctx, &str_value, (const smb_ucs2_t *) value->data_p);
+               status = ads_mod_str(ctx, mods, value->valuename, str_value);
+               return ADS_ERR_OK(status);
+       }
+       return True;
+               
+}
 
+/*
+  map a REG_DWORD to an ldap mod
+*/
+static BOOL map_dword(TALLOC_CTX *ctx, ADS_MODLIST *mods, 
+                     const REGISTRY_VALUE *value)
+{
+       char *str_value = NULL;
+       ADS_STATUS status;
 
-       status = ads_gen_add(ads, prt_dn, mods);
+       if (value->type != REG_DWORD)
+               return False;
+       str_value = talloc_asprintf(ctx, "%d", *((uint32 *) value->data_p));
+       status = ads_mod_str(ctx, mods, value->valuename, str_value);
+       return ADS_ERR_OK(status);
+}
 
-       talloc_destroy(ctx);
+/*
+  map a boolean REG_BINARY to an ldap mod
+*/
+static BOOL map_bool(TALLOC_CTX *ctx, ADS_MODLIST *mods,
+                    const REGISTRY_VALUE *value)
+{
+       char *str_value;
+       ADS_STATUS status;
 
-        return status;
+       if ((value->type != REG_BINARY) || (value->size != 1))
+               return False;
+       str_value =  talloc_asprintf(ctx, "%s", 
+                                    *(value->data_p) ? "TRUE" : "FALSE");
+       status = ads_mod_str(ctx, mods, value->valuename, str_value);
+       return ADS_ERR_OK(status);
 }
 
 /*
-  publish a printer in the ADS
+  map a REG_MULTI_SZ to an ldap mod
 */
-
-ADS_STATUS ads_add_printer(ADS_STRUCT *ads, const ADS_PRINTER_ENTRY *prt)
+static BOOL map_multi_sz(TALLOC_CTX *ctx, ADS_MODLIST *mods,
+                        const REGISTRY_VALUE *value)
 {
+       char **str_values = NULL;
+       smb_ucs2_t *cur_str = (smb_ucs2_t *) value->data_p;
+        uint32 size = 0, num_vals = 0, i=0;
        ADS_STATUS status;
-       void *res;
-       char *host_dn, *prt_dn;
-       const char *attrs[] = {"*", "nTSecurityDescriptor", NULL};
 
-       status = ads_find_machine_acct(ads, (void **)&res,
-                                      prt->shortServerName);
-       if (!ADS_ERR_OK(status)) {
-               DEBUG(1, ("ads_add_printer: cannot find host %s in ads\n",
-                         prt->shortServerName));
-               return status;
-       }
-       host_dn = ads_get_dn(ads, res);
-       ads_msgfree(ads, res);
+       if (value->type != REG_MULTI_SZ)
+               return False;
+
+       while(cur_str && *cur_str && (size < value->size)) {            
+               size += 2 * (strlen_w(cur_str) + 1);
+               cur_str += strlen_w(cur_str) + 1;
+               num_vals++;
+       };
+
+       if (num_vals) {
+               str_values = TALLOC_ARRAY(ctx, char *, num_vals + 1);
+               memset(str_values, '\0', 
+                      (num_vals + 1) * sizeof(char *));
+
+               cur_str = (smb_ucs2_t *) value->data_p;
+               for (i=0; i < num_vals; i++)
+                       cur_str += pull_ucs2_talloc(ctx, &str_values[i],
+                                                   cur_str);
+
+               status = ads_mod_strlist(ctx, mods, value->valuename, 
+                                        (const char **) str_values);
+               return ADS_ERR_OK(status);
+       } 
+       return True;
+}
 
-       ads_find_printer_on_server(ads, &res, prt->printerName, 
-                                  prt->shortServerName);
+struct valmap_to_ads {
+       const char *valname;
+       BOOL (*fn)(TALLOC_CTX *, ADS_MODLIST *, const REGISTRY_VALUE *);
+};
 
-       if (ADS_ERR_OK(status) && ads_count_replies(ads, res)) {
-               DEBUG(1, ("ads_add_printer: printer %s already exists\n",
-                         prt->printerName));
-               /* nothing to do, just free results */
-               ads_msgfree(ads, res);
-       } else {
-               ads_msgfree(ads, res);
-               status = ads_add_printer_entry(ads, prt_dn, prt);
-               if (!ADS_ERR_OK(status)) {
-                       DEBUG(0, ("ads_add_printer: ads_add_printer_entry failed\n"));
-                       return status;
+/*
+  map a REG_SZ to an ldap mod
+*/
+static void map_regval_to_ads(TALLOC_CTX *ctx, ADS_MODLIST *mods, 
+                             REGISTRY_VALUE *value)
+{
+       const struct valmap_to_ads map[] = {
+               {SPOOL_REG_ASSETNUMBER, map_sz},
+               {SPOOL_REG_BYTESPERMINUTE, map_dword},
+               {SPOOL_REG_DEFAULTPRIORITY, map_dword},
+               {SPOOL_REG_DESCRIPTION, map_sz},
+               {SPOOL_REG_DRIVERNAME, map_sz},
+               {SPOOL_REG_DRIVERVERSION, map_dword},
+               {SPOOL_REG_FLAGS, map_dword},
+               {SPOOL_REG_LOCATION, map_sz},
+               {SPOOL_REG_OPERATINGSYSTEM, map_sz},
+               {SPOOL_REG_OPERATINGSYSTEMHOTFIX, map_sz},
+               {SPOOL_REG_OPERATINGSYSTEMSERVICEPACK, map_sz},
+               {SPOOL_REG_OPERATINGSYSTEMVERSION, map_sz},
+               {SPOOL_REG_PORTNAME, map_multi_sz},
+               {SPOOL_REG_PRINTATTRIBUTES, map_dword},
+               {SPOOL_REG_PRINTBINNAMES, map_multi_sz},
+               {SPOOL_REG_PRINTCOLLATE, map_bool},
+               {SPOOL_REG_PRINTCOLOR, map_bool},
+               {SPOOL_REG_PRINTDUPLEXSUPPORTED, map_bool},
+               {SPOOL_REG_PRINTENDTIME, map_dword},
+               {SPOOL_REG_PRINTFORMNAME, map_sz},
+               {SPOOL_REG_PRINTKEEPPRINTEDJOBS, map_bool},
+               {SPOOL_REG_PRINTLANGUAGE, map_multi_sz},
+               {SPOOL_REG_PRINTMACADDRESS, map_sz},
+               {SPOOL_REG_PRINTMAXCOPIES, map_sz},
+               {SPOOL_REG_PRINTMAXRESOLUTIONSUPPORTED, map_dword},
+               {SPOOL_REG_PRINTMAXXEXTENT, map_dword},
+               {SPOOL_REG_PRINTMAXYEXTENT, map_dword},
+               {SPOOL_REG_PRINTMEDIAREADY, map_multi_sz},
+               {SPOOL_REG_PRINTMEDIASUPPORTED, map_multi_sz},
+               {SPOOL_REG_PRINTMEMORY, map_dword},
+               {SPOOL_REG_PRINTMINXEXTENT, map_dword},
+               {SPOOL_REG_PRINTMINYEXTENT, map_dword},
+               {SPOOL_REG_PRINTNETWORKADDRESS, map_sz},
+               {SPOOL_REG_PRINTNOTIFY, map_sz},
+               {SPOOL_REG_PRINTNUMBERUP, map_dword},
+               {SPOOL_REG_PRINTORIENTATIONSSUPPORTED, map_multi_sz},
+               {SPOOL_REG_PRINTOWNER, map_sz},
+               {SPOOL_REG_PRINTPAGESPERMINUTE, map_dword},
+               {SPOOL_REG_PRINTRATE, map_dword},
+               {SPOOL_REG_PRINTRATEUNIT, map_sz},
+               {SPOOL_REG_PRINTSEPARATORFILE, map_sz},
+               {SPOOL_REG_PRINTSHARENAME, map_sz},
+               {SPOOL_REG_PRINTSPOOLING, map_sz},
+               {SPOOL_REG_PRINTSTAPLINGSUPPORTED, map_bool},
+               {SPOOL_REG_PRINTSTARTTIME, map_dword},
+               {SPOOL_REG_PRINTSTATUS, map_sz},
+               {SPOOL_REG_PRIORITY, map_dword},
+               {SPOOL_REG_SERVERNAME, map_sz},
+               {SPOOL_REG_SHORTSERVERNAME, map_sz},
+               {SPOOL_REG_UNCNAME, map_sz},
+               {SPOOL_REG_URL, map_sz},
+               {SPOOL_REG_VERSIONNUMBER, map_dword},
+               {NULL, NULL}
+       };
+       int i;
+
+       for (i=0; map[i].valname; i++) {
+               if (StrCaseCmp(map[i].valname, value->valuename) == 0) {
+                       if (!map[i].fn(ctx, mods, value)) {
+                               DEBUG(5, ("Add of value %s to modlist failed\n", value->valuename));
+                       } else {
+                               DEBUG(7, ("Mapped value %s\n", value->valuename));
+                       }
+                       
                }
        }
+}
 
-       status = ads_search_dn(ads, &res, prt_dn, attrs);
 
-       if (ADS_ERR_OK(status) && ads_count_replies(ads, res)) {
-               /* need to retrieve GUID from results
-                  prt->GUID */
-               status = ads_mod_printer_entry(ads, prt_dn, prt);
+WERROR get_remote_printer_publishing_data(struct cli_state *cli, 
+                                         TALLOC_CTX *mem_ctx,
+                                         ADS_MODLIST *mods,
+                                         const char *printer)
+{
+       WERROR result;
+       char *printername, *servername;
+       REGVAL_CTR dsdriver_ctr, dsspooler_ctr;
+       BOOL got_dsdriver = False, got_dsspooler = False;
+       uint32 needed, i;
+       POLICY_HND pol;
+
+       asprintf(&servername, "\\\\%s", cli->desthost);
+       asprintf(&printername, "%s\\%s", servername, printer);
+       if (!servername || !printername) {
+               DEBUG(3, ("Insufficient memory\n"));
+               return WERR_NOMEM;
+       }
+       
+       result = cli_spoolss_open_printer_ex(cli, mem_ctx, printername, 
+                                            "", MAXIMUM_ALLOWED_ACCESS, 
+                                            servername, cli->user_name, &pol);
+       if (!W_ERROR_IS_OK(result)) {
+               DEBUG(3, ("Unable to open printer %s, error is %s.\n",
+                         printername, dos_errstr(result)));
+               return result;
        }
+       
+       result = cli_spoolss_enumprinterdataex(cli, mem_ctx, 0, &needed, 
+                                              &pol, SPOOL_DSDRIVER_KEY, NULL);
+
+       if (W_ERROR_V(result) == ERRmoredata)
+               result = cli_spoolss_enumprinterdataex(cli, mem_ctx, needed, 
+                                                      NULL, &pol, 
+                                                      SPOOL_DSDRIVER_KEY,
+                                                      &dsdriver_ctr);
+
+       if (!W_ERROR_IS_OK(result)) {
+               DEBUG(3, ("Unable to do enumdataex on %s, error is %s.\n",
+                         printername, dos_errstr(result)));
+       } else {
 
-       ads_msgfree(ads, res);
+               /* Have the data we need now, so start building */
+               got_dsdriver = True;
+               for (i=0; i < dsdriver_ctr.num_values; i++)
+                       map_regval_to_ads(mem_ctx, mods, 
+                                         dsdriver_ctr.values[i]);
+       }
+       
+       result = cli_spoolss_enumprinterdataex(cli, mem_ctx, 0, &needed, 
+                                              &pol, SPOOL_DSSPOOLER_KEY, 
+                                              NULL);
+
+       if (W_ERROR_V(result) == ERRmoredata)
+               result = cli_spoolss_enumprinterdataex(cli, mem_ctx, needed, 
+                                                      NULL, &pol, 
+                                                      SPOOL_DSSPOOLER_KEY,
+                                                      &dsspooler_ctr);
+
+       if (!W_ERROR_IS_OK(result)) {
+               DEBUG(3, ("Unable to do enumdataex on %s, error is %s.\n",
+                         printername, dos_errstr(result)));
+       } else {
+               got_dsspooler = True;
+               for (i=0; i < dsspooler_ctr.num_values; i++)
+                       map_regval_to_ads(mem_ctx, mods, 
+                                         dsspooler_ctr.values[i]);
+       }
+       
+       ads_mod_str(mem_ctx, mods, SPOOL_REG_PRINTERNAME, printer);
 
+       if (got_dsdriver) regval_ctr_destroy(&dsdriver_ctr);
+       if (got_dsspooler) regval_ctr_destroy(&dsspooler_ctr);
+       cli_spoolss_close_printer(cli, mem_ctx, &pol);
 
-       return status;
+       return result;
+}
+
+BOOL get_local_printer_publishing_data(TALLOC_CTX *mem_ctx,
+                                      ADS_MODLIST *mods,
+                                      NT_PRINTER_DATA *data)
+{
+       uint32 key,val;
+
+       for (key=0; key < data->num_keys; key++) {
+               REGVAL_CTR ctr = data->keys[key].values;
+               for (val=0; val < ctr.num_values; val++)
+                       map_regval_to_ads(mem_ctx, mods, ctr.values[val]);
+       }
+       return True;
 }
 
 #endif