The WinSock library is now loaded at run-time, which means we can

[u/mdw/putty] / printing.c

diff --git a/printing.c b/printing.c
index cd4baa9..6248881 100644 (file)
--- a/printing.c
+++ b/printing.c
@@ -2,49 +2,97 @@
  * Printing interface for PuTTY.
  */
 
-#include <windows.h>
 #include "putty.h"
 
 struct printer_enum_tag {
     int nprinters;
-    LPPRINTER_INFO_5 info;
+    DWORD enum_level;
+    union {
+       LPPRINTER_INFO_4 i4;
+       LPPRINTER_INFO_5 i5;
+    } info;
 };
 
 struct printer_job_tag {
     HANDLE hprinter;
 };
 
-printer_enum *printer_start_enum(int *nprinters_ptr)
+static char *printer_add_enum(int param, DWORD level, char *buffer,
+                              int offset, int *nprinters_ptr)
 {
-    printer_enum *ret = smalloc(sizeof(printer_enum));
-    char *buffer = NULL;
     DWORD needed, nprinters;
 
-    *nprinters_ptr = 0;                       /* default return value */
-    buffer = smalloc(512);
-    if (EnumPrinters(PRINTER_ENUM_LOCAL, NULL, 5,
-                    buffer, 512, &needed, &nprinters) == 0)
-       goto error;
+    buffer = sresize(buffer, offset+512, char);
+
+    /*
+     * Exploratory call to EnumPrinters to determine how much space
+     * we'll need for the output. Discard the return value since it
+     * will almost certainly be a failure due to lack of space.
+     */
+    EnumPrinters(param, NULL, level, buffer+offset, 512,
+                &needed, &nprinters);
+
+    if (needed < 512)
+        needed = 512;
 
-    if (needed) {
-        buffer = srealloc(buffer, needed);
+    buffer = sresize(buffer, offset+needed, char);
+
+    if (EnumPrinters(param, NULL, level, buffer+offset,
+                     needed, &needed, &nprinters) == 0)
+        return NULL;
+
+    *nprinters_ptr += nprinters;
+
+    return buffer;
+}
 
-        if (EnumPrinters(PRINTER_ENUM_LOCAL, NULL, 5,
-                         (LPBYTE)buffer, needed, &needed, &nprinters) == 0)
-            goto error;
+printer_enum *printer_start_enum(int *nprinters_ptr)
+{
+    printer_enum *ret = snew(printer_enum);
+    char *buffer = NULL, *retval;
+
+    *nprinters_ptr = 0;                       /* default return value */
+    buffer = snewn(512, char);
+
+    /*
+     * Determine what enumeration level to use.
+     * When enumerating printers, we need to use PRINTER_INFO_4 on
+     * NT-class systems to avoid Windows looking too hard for them and
+     * slowing things down; and we need to avoid PRINTER_INFO_5 as
+     * we've seen network printers not show up.
+     * On 9x-class systems, PRINTER_INFO_4 isn't available and
+     * PRINTER_INFO_5 is recommended.
+     * Bletch.
+     */
+    if (osVersion.dwPlatformId != VER_PLATFORM_WIN32_NT) {
+       ret->enum_level = 5;
     } else {
-        nprinters = 0;
-        ret->info = NULL;
+       ret->enum_level = 4;
     }
 
-    ret->info = (LPPRINTER_INFO_5)buffer;
-    ret->nprinters = *nprinters_ptr = nprinters;
+    retval = printer_add_enum(PRINTER_ENUM_LOCAL | PRINTER_ENUM_CONNECTIONS,
+                             ret->enum_level, buffer, 0, nprinters_ptr);
+    if (!retval)
+        goto error;
+    else
+        buffer = retval;
+
+    switch (ret->enum_level) {
+      case 4:
+       ret->info.i4 = (LPPRINTER_INFO_4)buffer;
+       break;
+      case 5:
+       ret->info.i5 = (LPPRINTER_INFO_5)buffer;
+       break;
+    }
+    ret->nprinters = *nprinters_ptr;
     
     return ret;
 
     error:
     sfree(buffer);
     sfree(ret);
+    *nprinters_ptr = 0;
     return NULL;
 }
 
@@ -54,20 +102,34 @@ char *printer_get_name(printer_enum *pe, int i)
        return NULL;
     if (i < 0 || i >= pe->nprinters)
        return NULL;
-    return pe->info[i].pPrinterName;
+    switch (pe->enum_level) {
+      case 4:
+       return pe->info.i4[i].pPrinterName;
+      case 5:
+       return pe->info.i5[i].pPrinterName;
+      default:
+       return NULL;
+    }
 }
 
 void printer_finish_enum(printer_enum *pe)
 {
     if (!pe)
        return;
-    sfree(pe->info);
+    switch (pe->enum_level) {
+      case 4:
+       sfree(pe->info.i4);
+       break;
+      case 5:
+       sfree(pe->info.i5);
+       break;
+    }
     sfree(pe);
 }
 
 printer_job *printer_start_job(char *printer)
 {
-    printer_job *ret = smalloc(sizeof(printer_job));
+    printer_job *ret = snew(printer_job);
     DOC_INFO_1 docinfo;
     int jobstarted = 0, pagestarted = 0;