git.distorted.org.uk Git - u/mdw/putty/blob - printing.c

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