09ad523a |
1 | /* |
2 | * Printing interface for PuTTY. |
3 | */ |
4 | |
5 | #include <windows.h> |
6 | #include "putty.h" |
7 | |
11773fc1 |
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 | */ |
01f7687f |
14 | #if 0 |
11773fc1 |
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 | |
09ad523a |
26 | struct printer_enum_tag { |
27 | int nprinters; |
11773fc1 |
28 | ENUM_PTR info; |
09ad523a |
29 | }; |
30 | |
31 | struct printer_job_tag { |
32 | HANDLE hprinter; |
33 | }; |
34 | |
11773fc1 |
35 | static char *printer_add_enum(int param, char *buffer, |
36 | int offset, int *nprinters_ptr) |
37 | { |
38 | DWORD needed, nprinters; |
39 | |
3d88e64d |
40 | buffer = sresize(buffer, offset+512, char); |
11773fc1 |
41 | |
01f7687f |
42 | /* |
43 | * Exploratory call to EnumPrinters to determine how much space |
44 | * we'll need for the output. Discard the return value since it |
45 | * will almost certainly be a failure due to lack of space. |
46 | */ |
47 | EnumPrinters(param, NULL, ENUM_LEVEL, buffer+offset, 512, |
48 | &needed, &nprinters); |
11773fc1 |
49 | |
50 | if (needed < 512) |
51 | needed = 512; |
52 | |
3d88e64d |
53 | buffer = sresize(buffer, offset+needed, char); |
11773fc1 |
54 | |
55 | if (EnumPrinters(param, NULL, ENUM_LEVEL, buffer+offset, |
56 | needed, &needed, &nprinters) == 0) |
57 | return NULL; |
58 | |
59 | *nprinters_ptr += nprinters; |
60 | |
61 | return buffer; |
62 | } |
63 | |
09ad523a |
64 | printer_enum *printer_start_enum(int *nprinters_ptr) |
65 | { |
3d88e64d |
66 | printer_enum *ret = snew(printer_enum); |
11773fc1 |
67 | char *buffer = NULL, *retval; |
09ad523a |
68 | |
69 | *nprinters_ptr = 0; /* default return value */ |
3d88e64d |
70 | buffer = snewn(512, char); |
11773fc1 |
71 | |
a5751e2d |
72 | retval = printer_add_enum(PRINTER_ENUM_LOCAL | PRINTER_ENUM_CONNECTIONS, |
73 | buffer, 0, nprinters_ptr); |
11773fc1 |
74 | if (!retval) |
75 | goto error; |
76 | else |
77 | buffer = retval; |
78 | |
79 | ret->info = (ENUM_PTR)buffer; |
80 | ret->nprinters = *nprinters_ptr; |
09ad523a |
81 | |
82 | return ret; |
83 | |
84 | error: |
85 | sfree(buffer); |
86 | sfree(ret); |
11773fc1 |
87 | *nprinters_ptr = 0; |
09ad523a |
88 | return NULL; |
89 | } |
90 | |
91 | char *printer_get_name(printer_enum *pe, int i) |
92 | { |
93 | if (!pe) |
94 | return NULL; |
95 | if (i < 0 || i >= pe->nprinters) |
96 | return NULL; |
11773fc1 |
97 | return pe->info[i].ENUM_MEMBER; |
09ad523a |
98 | } |
99 | |
100 | void printer_finish_enum(printer_enum *pe) |
101 | { |
102 | if (!pe) |
103 | return; |
104 | sfree(pe->info); |
105 | sfree(pe); |
106 | } |
107 | |
108 | printer_job *printer_start_job(char *printer) |
109 | { |
3d88e64d |
110 | printer_job *ret = snew(printer_job); |
09ad523a |
111 | DOC_INFO_1 docinfo; |
112 | int jobstarted = 0, pagestarted = 0; |
113 | |
114 | ret->hprinter = NULL; |
115 | if (!OpenPrinter(printer, &ret->hprinter, NULL)) |
116 | goto error; |
117 | |
118 | docinfo.pDocName = "PuTTY remote printer output"; |
119 | docinfo.pOutputFile = NULL; |
120 | docinfo.pDatatype = "RAW"; |
121 | |
122 | if (!StartDocPrinter(ret->hprinter, 1, (LPSTR)&docinfo)) |
123 | goto error; |
124 | jobstarted = 1; |
125 | |
126 | if (!StartPagePrinter(ret->hprinter)) |
127 | goto error; |
128 | pagestarted = 1; |
129 | |
130 | return ret; |
131 | |
132 | error: |
133 | if (pagestarted) |
134 | EndPagePrinter(ret->hprinter); |
135 | if (jobstarted) |
136 | EndDocPrinter(ret->hprinter); |
137 | if (ret->hprinter) |
138 | ClosePrinter(ret->hprinter); |
139 | sfree(ret); |
140 | return NULL; |
141 | } |
142 | |
143 | void printer_job_data(printer_job *pj, void *data, int len) |
144 | { |
145 | DWORD written; |
146 | |
147 | if (!pj) |
148 | return; |
149 | |
150 | WritePrinter(pj->hprinter, data, len, &written); |
151 | } |
152 | |
153 | void printer_finish_job(printer_job *pj) |
154 | { |
155 | if (!pj) |
156 | return; |
157 | |
158 | EndPagePrinter(pj->hprinter); |
159 | EndDocPrinter(pj->hprinter); |
160 | ClosePrinter(pj->hprinter); |
161 | sfree(pj); |
162 | } |