windows/winprint.c

   1 /*
   2  * Printing interface for PuTTY.
   3  */
   4
   5 #include "putty.h"
   6 #include <winspool.h>
   7
   8 struct printer_enum_tag {
   9     int nprinters;
  10     DWORD enum_level;
  11     union {
  12         LPPRINTER_INFO_4 i4;
  13         LPPRINTER_INFO_5 i5;
  14     } info;
  15 };
  16
  17 struct printer_job_tag {
  18     HANDLE hprinter;
  19 };
  20
  21 static char *printer_add_enum(int param, DWORD level, char *buffer,
  22                               int offset, int *nprinters_ptr)
  23 {
  24     DWORD needed, nprinters;
  25
  26     buffer = sresize(buffer, offset+512, char);
  27
  28     /*
  29      * Exploratory call to EnumPrinters to determine how much space
  30      * we'll need for the output. Discard the return value since it
  31      * will almost certainly be a failure due to lack of space.
  32      */
  33     EnumPrinters(param, NULL, level, buffer+offset, 512,
  34                  &needed, &nprinters);
  35
  36     if (needed < 512)
  37         needed = 512;
  38
  39     buffer = sresize(buffer, offset+needed, char);
  40
  41     if (EnumPrinters(param, NULL, level, buffer+offset,
  42                      needed, &needed, &nprinters) == 0)
  43         return NULL;
  44
  45     *nprinters_ptr += nprinters;
  46
  47     return buffer;
  48 }
  49
  50 printer_enum *printer_start_enum(int *nprinters_ptr)
  51 {
  52     printer_enum *ret = snew(printer_enum);
  53     char *buffer = NULL, *retval;
  54
  55     *nprinters_ptr = 0;                /* default return value */
  56     buffer = snewn(512, char);
  57
  58     /*
  59      * Determine what enumeration level to use.
  60      * When enumerating printers, we need to use PRINTER_INFO_4 on
  61      * NT-class systems to avoid Windows looking too hard for them and
  62      * slowing things down; and we need to avoid PRINTER_INFO_5 as
  63      * we've seen network printers not show up.
  64      * On 9x-class systems, PRINTER_INFO_4 isn't available and
  65      * PRINTER_INFO_5 is recommended.
  66      * Bletch.
  67      */
  68     if (osVersion.dwPlatformId != VER_PLATFORM_WIN32_NT) {
  69         ret->enum_level = 5;
  70     } else {
  71         ret->enum_level = 4;
  72     }
  73
  74     retval = printer_add_enum(PRINTER_ENUM_LOCAL | PRINTER_ENUM_CONNECTIONS,
  75                               ret->enum_level, buffer, 0, nprinters_ptr);
  76     if (!retval)
  77         goto error;
  78     else
  79         buffer = retval;
  80
  81     switch (ret->enum_level) {
  82       case 4:
  83         ret->info.i4 = (LPPRINTER_INFO_4)buffer;
  84         break;
  85       case 5:
  86         ret->info.i5 = (LPPRINTER_INFO_5)buffer;
  87         break;
  88     }
  89     ret->nprinters = *nprinters_ptr;
  90
  91     return ret;
  92
  93     error:
  94     sfree(buffer);
  95     sfree(ret);
  96     *nprinters_ptr = 0;
  97     return NULL;
  98 }
  99
 100 char *printer_get_name(printer_enum *pe, int i)
 101 {
 102     if (!pe)
 103         return NULL;
 104     if (i < 0 || i >= pe->nprinters)
 105         return NULL;
 106     switch (pe->enum_level) {
 107       case 4:
 108         return pe->info.i4[i].pPrinterName;
 109       case 5:
 110         return pe->info.i5[i].pPrinterName;
 111       default:
 112         return NULL;
 113     }
 114 }
 115
 116 void printer_finish_enum(printer_enum *pe)
 117 {
 118     if (!pe)
 119         return;
 120     switch (pe->enum_level) {
 121       case 4:
 122         sfree(pe->info.i4);
 123         break;
 124       case 5:
 125         sfree(pe->info.i5);
 126         break;
 127     }
 128     sfree(pe);
 129 }
 130
 131 printer_job *printer_start_job(char *printer)
 132 {
 133     printer_job *ret = snew(printer_job);
 134     DOC_INFO_1 docinfo;
 135     int jobstarted = 0, pagestarted = 0;
 136
 137     ret->hprinter = NULL;
 138     if (!OpenPrinter(printer, &ret->hprinter, NULL))
 139         goto error;
 140
 141     docinfo.pDocName = "PuTTY remote printer output";
 142     docinfo.pOutputFile = NULL;
 143     docinfo.pDatatype = "RAW";
 144
 145     if (!StartDocPrinter(ret->hprinter, 1, (LPSTR)&docinfo))
 146         goto error;
 147     jobstarted = 1;
 148
 149     if (!StartPagePrinter(ret->hprinter))
 150         goto error;
 151     pagestarted = 1;
 152
 153     return ret;
 154
 155     error:
 156     if (pagestarted)
 157         EndPagePrinter(ret->hprinter);
 158     if (jobstarted)
 159         EndDocPrinter(ret->hprinter);
 160     if (ret->hprinter)
 161         ClosePrinter(ret->hprinter);
 162     sfree(ret);
 163     return NULL;
 164 }
 165
 166 void printer_job_data(printer_job *pj, void *data, int len)
 167 {
 168     DWORD written;
 169
 170     if (!pj)
 171         return;
 172
 173     WritePrinter(pj->hprinter, data, len, &written);
 174 }
 175
 176 void printer_finish_job(printer_job *pj)
 177 {
 178     if (!pj)
 179         return;
 180
 181     EndPagePrinter(pj->hprinter);
 182     EndDocPrinter(pj->hprinter);
 183     ClosePrinter(pj->hprinter);
 184     sfree(pj);
 185 }