printing.c

   1 /*
   2  * Printing interface for PuTTY.
   3  */
   4
   5 #include <windows.h>
   6 #include "putty.h"
   7
   8 /*
   9  * Boggle. Flipping between the two branches of this #if appears to
  10  * make all the difference as to whether network printers show up
  11  * under PRINTER_ENUM_CONNECTIONS on NT 4. I don't pretend to
  12  * understand this...
  13  */
  14 #if 0
  15 #define ENUM_LEVEL 5
  16 #define ENUM_PTR LPPRINTER_INFO_5
  17 #define ENUM_TYPE PRINTER_INFO_5
  18 #define ENUM_MEMBER pPrinterName
  19 #else
  20 #define ENUM_LEVEL 1
  21 #define ENUM_PTR LPPRINTER_INFO_1
  22 #define ENUM_TYPE PRINTER_INFO_1
  23 #define ENUM_MEMBER pName
  24 #endif
  25
  26 struct printer_enum_tag {
  27     int nprinters;
  28     ENUM_PTR info;
  29 };
  30
  31 struct printer_job_tag {
  32     HANDLE hprinter;
  33 };
  34
  35 static char *printer_add_enum(int param, char *buffer,
  36                               int offset, int *nprinters_ptr)
  37 {
  38     DWORD needed, nprinters;
  39
  40     buffer = srealloc(buffer, offset+512);
  41
  42     if (EnumPrinters(param, NULL, ENUM_LEVEL, buffer+offset,
  43                      512, &needed, &nprinters) == 0)
  44         return NULL;
  45
  46     if (needed < 512)
  47         needed = 512;
  48
  49     buffer = srealloc(buffer, offset+needed);
  50
  51     if (EnumPrinters(param, NULL, ENUM_LEVEL, buffer+offset,
  52                      needed, &needed, &nprinters) == 0)
  53         return NULL;
  54
  55     *nprinters_ptr += nprinters;
  56
  57     return buffer;
  58 }
  59
  60 printer_enum *printer_start_enum(int *nprinters_ptr)
  61 {
  62     printer_enum *ret = smalloc(sizeof(printer_enum));
  63     char *buffer = NULL, *retval;
  64
  65     *nprinters_ptr = 0;                /* default return value */
  66     buffer = smalloc(512);
  67
  68     retval = printer_add_enum(PRINTER_ENUM_LOCAL, buffer, 0, nprinters_ptr);
  69     if (!retval)
  70         goto error;
  71     else
  72         buffer = retval;
  73     retval = printer_add_enum(PRINTER_ENUM_CONNECTIONS, buffer,
  74                               sizeof(ENUM_TYPE) * *nprinters_ptr,
  75                               nprinters_ptr);
  76     if (!retval)
  77         goto error;
  78     else
  79         buffer = retval;
  80
  81     ret->info = (ENUM_PTR)buffer;
  82     ret->nprinters = *nprinters_ptr;
  83
  84     return ret;
  85
  86     error:
  87     sfree(buffer);
  88     sfree(ret);
  89     *nprinters_ptr = 0;
  90     return NULL;
  91 }
  92
  93 char *printer_get_name(printer_enum *pe, int i)
  94 {
  95     if (!pe)
  96         return NULL;
  97     if (i < 0 || i >= pe->nprinters)
  98         return NULL;
  99     return pe->info[i].ENUM_MEMBER;
 100 }
 101
 102 void printer_finish_enum(printer_enum *pe)
 103 {
 104     if (!pe)
 105         return;
 106     sfree(pe->info);
 107     sfree(pe);
 108 }
 109
 110 printer_job *printer_start_job(char *printer)
 111 {
 112     printer_job *ret = smalloc(sizeof(printer_job));
 113     DOC_INFO_1 docinfo;
 114     int jobstarted = 0, pagestarted = 0;
 115
 116     ret->hprinter = NULL;
 117     if (!OpenPrinter(printer, &ret->hprinter, NULL))
 118         goto error;
 119
 120     docinfo.pDocName = "PuTTY remote printer output";
 121     docinfo.pOutputFile = NULL;
 122     docinfo.pDatatype = "RAW";
 123
 124     if (!StartDocPrinter(ret->hprinter, 1, (LPSTR)&docinfo))
 125         goto error;
 126     jobstarted = 1;
 127
 128     if (!StartPagePrinter(ret->hprinter))
 129         goto error;
 130     pagestarted = 1;
 131
 132     return ret;
 133
 134     error:
 135     if (pagestarted)
 136         EndPagePrinter(ret->hprinter);
 137     if (jobstarted)
 138         EndDocPrinter(ret->hprinter);
 139     if (ret->hprinter)
 140         ClosePrinter(ret->hprinter);
 141     sfree(ret);
 142     return NULL;
 143 }
 144
 145 void printer_job_data(printer_job *pj, void *data, int len)
 146 {
 147     DWORD written;
 148
 149     if (!pj)
 150         return;
 151
 152     WritePrinter(pj->hprinter, data, len, &written);
 153 }
 154
 155 void printer_finish_job(printer_job *pj)
 156 {
 157     if (!pj)
 158         return;
 159
 160     EndPagePrinter(pj->hprinter);
 161     EndDocPrinter(pj->hprinter);
 162     ClosePrinter(pj->hprinter);
 163     sfree(pj);
 164 }