09ad523a |
1 | /* |
2 | * Printing interface for PuTTY. |
3 | */ |
4 | |
09ad523a |
5 | #include "putty.h" |
4e95095a |
6 | #include <winspool.h> |
11773fc1 |
7 | |
09ad523a |
8 | struct printer_enum_tag { |
9 | int nprinters; |
4c48c989 |
10 | DWORD enum_level; |
11 | union { |
12 | LPPRINTER_INFO_4 i4; |
13 | LPPRINTER_INFO_5 i5; |
14 | } info; |
09ad523a |
15 | }; |
16 | |
17 | struct printer_job_tag { |
18 | HANDLE hprinter; |
19 | }; |
20 | |
0c33d3a6 |
21 | static int printer_add_enum(int param, DWORD level, char **buffer, |
22 | int offset, int *nprinters_ptr) |
11773fc1 |
23 | { |
bec8c0c1 |
24 | DWORD needed = 0, nprinters = 0; |
11773fc1 |
25 | |
0c33d3a6 |
26 | *buffer = sresize(*buffer, offset+512, char); |
11773fc1 |
27 | |
01f7687f |
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 | */ |
0c33d3a6 |
33 | EnumPrinters(param, NULL, level, (*buffer)+offset, 512, |
01f7687f |
34 | &needed, &nprinters); |
11773fc1 |
35 | |
36 | if (needed < 512) |
37 | needed = 512; |
38 | |
0c33d3a6 |
39 | *buffer = sresize(*buffer, offset+needed, char); |
11773fc1 |
40 | |
0c33d3a6 |
41 | if (EnumPrinters(param, NULL, level, (*buffer)+offset, |
11773fc1 |
42 | needed, &needed, &nprinters) == 0) |
0c33d3a6 |
43 | return FALSE; |
11773fc1 |
44 | |
45 | *nprinters_ptr += nprinters; |
46 | |
0c33d3a6 |
47 | return TRUE; |
11773fc1 |
48 | } |
49 | |
09ad523a |
50 | printer_enum *printer_start_enum(int *nprinters_ptr) |
51 | { |
3d88e64d |
52 | printer_enum *ret = snew(printer_enum); |
0c33d3a6 |
53 | char *buffer = NULL; |
09ad523a |
54 | |
55 | *nprinters_ptr = 0; /* default return value */ |
3d88e64d |
56 | buffer = snewn(512, char); |
11773fc1 |
57 | |
4c48c989 |
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 | |
0c33d3a6 |
74 | if (!printer_add_enum(PRINTER_ENUM_LOCAL | PRINTER_ENUM_CONNECTIONS, |
75 | ret->enum_level, &buffer, 0, nprinters_ptr)) |
11773fc1 |
76 | goto error; |
11773fc1 |
77 | |
4c48c989 |
78 | switch (ret->enum_level) { |
79 | case 4: |
80 | ret->info.i4 = (LPPRINTER_INFO_4)buffer; |
81 | break; |
82 | case 5: |
83 | ret->info.i5 = (LPPRINTER_INFO_5)buffer; |
84 | break; |
85 | } |
11773fc1 |
86 | ret->nprinters = *nprinters_ptr; |
09ad523a |
87 | |
88 | return ret; |
89 | |
90 | error: |
91 | sfree(buffer); |
92 | sfree(ret); |
11773fc1 |
93 | *nprinters_ptr = 0; |
09ad523a |
94 | return NULL; |
95 | } |
96 | |
97 | char *printer_get_name(printer_enum *pe, int i) |
98 | { |
99 | if (!pe) |
100 | return NULL; |
101 | if (i < 0 || i >= pe->nprinters) |
102 | return NULL; |
4c48c989 |
103 | switch (pe->enum_level) { |
104 | case 4: |
105 | return pe->info.i4[i].pPrinterName; |
106 | case 5: |
107 | return pe->info.i5[i].pPrinterName; |
108 | default: |
109 | return NULL; |
110 | } |
09ad523a |
111 | } |
112 | |
113 | void printer_finish_enum(printer_enum *pe) |
114 | { |
115 | if (!pe) |
116 | return; |
4c48c989 |
117 | switch (pe->enum_level) { |
118 | case 4: |
119 | sfree(pe->info.i4); |
120 | break; |
121 | case 5: |
122 | sfree(pe->info.i5); |
123 | break; |
124 | } |
09ad523a |
125 | sfree(pe); |
126 | } |
127 | |
128 | printer_job *printer_start_job(char *printer) |
129 | { |
3d88e64d |
130 | printer_job *ret = snew(printer_job); |
09ad523a |
131 | DOC_INFO_1 docinfo; |
132 | int jobstarted = 0, pagestarted = 0; |
133 | |
134 | ret->hprinter = NULL; |
135 | if (!OpenPrinter(printer, &ret->hprinter, NULL)) |
136 | goto error; |
137 | |
138 | docinfo.pDocName = "PuTTY remote printer output"; |
139 | docinfo.pOutputFile = NULL; |
140 | docinfo.pDatatype = "RAW"; |
141 | |
142 | if (!StartDocPrinter(ret->hprinter, 1, (LPSTR)&docinfo)) |
143 | goto error; |
144 | jobstarted = 1; |
145 | |
146 | if (!StartPagePrinter(ret->hprinter)) |
147 | goto error; |
148 | pagestarted = 1; |
149 | |
150 | return ret; |
151 | |
152 | error: |
153 | if (pagestarted) |
154 | EndPagePrinter(ret->hprinter); |
155 | if (jobstarted) |
156 | EndDocPrinter(ret->hprinter); |
157 | if (ret->hprinter) |
158 | ClosePrinter(ret->hprinter); |
159 | sfree(ret); |
160 | return NULL; |
161 | } |
162 | |
163 | void printer_job_data(printer_job *pj, void *data, int len) |
164 | { |
165 | DWORD written; |
166 | |
167 | if (!pj) |
168 | return; |
169 | |
170 | WritePrinter(pj->hprinter, data, len, &written); |
171 | } |
172 | |
173 | void printer_finish_job(printer_job *pj) |
174 | { |
175 | if (!pj) |
176 | return; |
177 | |
178 | EndPagePrinter(pj->hprinter); |
179 | EndDocPrinter(pj->hprinter); |
180 | ClosePrinter(pj->hprinter); |
181 | sfree(pj); |
182 | } |