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 1
  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 | PRINTER_ENUM_CONNECTIONS,
  69                               buffer, 0, nprinters_ptr);
  70     if (!retval)
  71         goto error;
  72     else
  73         buffer = retval;
  74
  75     ret->info = (ENUM_PTR)buffer;
  76     ret->nprinters = *nprinters_ptr;
  77
  78     return ret;
  79
  80     error:
  81     sfree(buffer);
  82     sfree(ret);
  83     *nprinters_ptr = 0;
  84     return NULL;
  85 }
  86
  87 char *printer_get_name(printer_enum *pe, int i)
  88 {
  89     if (!pe)
  90         return NULL;
  91     if (i < 0 || i >= pe->nprinters)
  92         return NULL;
  93     return pe->info[i].ENUM_MEMBER;
  94 }
  95
  96 void printer_finish_enum(printer_enum *pe)
  97 {
  98     if (!pe)
  99         return;
 100     sfree(pe->info);
 101     sfree(pe);
 102 }
 103
 104 printer_job *printer_start_job(char *printer)
 105 {
 106     printer_job *ret = smalloc(sizeof(printer_job));
 107     DOC_INFO_1 docinfo;
 108     int jobstarted = 0, pagestarted = 0;
 109
 110     ret->hprinter = NULL;
 111     if (!OpenPrinter(printer, &ret->hprinter, NULL))
 112         goto error;
 113
 114     docinfo.pDocName = "PuTTY remote printer output";
 115     docinfo.pOutputFile = NULL;
 116     docinfo.pDatatype = "RAW";
 117
 118     if (!StartDocPrinter(ret->hprinter, 1, (LPSTR)&docinfo))
 119         goto error;
 120     jobstarted = 1;
 121
 122     if (!StartPagePrinter(ret->hprinter))
 123         goto error;
 124     pagestarted = 1;
 125
 126     return ret;
 127
 128     error:
 129     if (pagestarted)
 130         EndPagePrinter(ret->hprinter);
 131     if (jobstarted)
 132         EndDocPrinter(ret->hprinter);
 133     if (ret->hprinter)
 134         ClosePrinter(ret->hprinter);
 135     sfree(ret);
 136     return NULL;
 137 }
 138
 139 void printer_job_data(printer_job *pj, void *data, int len)
 140 {
 141     DWORD written;
 142
 143     if (!pj)
 144         return;
 145
 146     WritePrinter(pj->hprinter, data, len, &written);
 147 }
 148
 149 void printer_finish_job(printer_job *pj)
 150 {
 151     if (!pj)
 152         return;
 153
 154     EndPagePrinter(pj->hprinter);
 155     EndDocPrinter(pj->hprinter);
 156     ClosePrinter(pj->hprinter);
 157     sfree(pj);
 158 }