1/*
2 *  Set printer capabilities in DsDriver Keys on remote printer
3 *  Copyright (C) Jim McDonough	<jmcd@us.ibm.com> 2002.
4 *
5 *  This program is free software; you can redistribute it and/or modify
6 *  it under the terms of the GNU General Public License as published by
7 *  the Free Software Foundation; either version 2 of the License, or
8 *  (at your option) any later version.
9 *
10 *  This program is distributed in the hope that it will be useful,
11 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
12 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
13 *  GNU General Public License for more details.
14 *
15 *  You should have received a copy of the GNU General Public License
16 *  along with this program; if not, write to the Free Software
17 *  Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
18 */
19
20/* This needs to be defined for certain compilers */
21#define WINVER 0x0500
22
23#include <tchar.h>
24#include <windows.h>
25#include <stdio.h>
26
27#define SAMBA_PORT _T("Samba")
28
29TCHAR *PrintLastError(void)
30{
31	static TCHAR msgtxt[1024*sizeof(TCHAR)];
32
33	FormatMessage(FORMAT_MESSAGE_FROM_SYSTEM,
34			NULL, GetLastError(),
35			0, msgtxt, 0, NULL);
36
37	return msgtxt;
38}
39
40void map_orientation(HANDLE ph, TCHAR *printer, TCHAR *port)
41{
42	DWORD rot;
43	TCHAR portrait_only[] = _T("PORTRAIT\0");
44	TCHAR both[] = _T("LANDSCAPE\0PORTRAIT\0");
45
46	/* orentation of 90 or 270 indicates landscape supported, 0 means it isn't */
47	rot = DeviceCapabilities(printer, port, DC_BINNAMES, NULL, NULL);
48
49	printf("printOrientationsSupported:\n");
50
51	if (rot) {
52		printf("\tPORTRAIT\n\tLANDSCAPE\n");
53		SetPrinterDataEx(ph, SPLDS_DRIVER_KEY, _T("printOrientationsSupported"), REG_MULTI_SZ,
54			both, sizeof(both));
55	} else {
56		printf("\tPORTRAIT\n");
57		SetPrinterDataEx(ph, SPLDS_DRIVER_KEY, _T("printOrientationsSupported"), REG_MULTI_SZ,
58			portrait_only, sizeof(portrait_only));
59	}
60}
61
62void map_resolution(HANDLE ph, TCHAR *printer, TCHAR *port)
63{
64	DWORD num, *res, maxres = 0, i;
65
66	num = DeviceCapabilities(printer, port, DC_ENUMRESOLUTIONS, NULL, NULL);
67	if ((DWORD) -1 == num)
68		return;
69	res = malloc(num*2*sizeof(DWORD));
70	num = DeviceCapabilities(printer, port, DC_ENUMRESOLUTIONS, (BYTE *) res, NULL);
71	for (i=0; i < num*2; i++) {
72		maxres = (res[i] > maxres) ? res[i] : maxres;
73	}
74	printf("printMaxResolutionSupported: %d\n", maxres);
75	SetPrinterDataEx(ph, SPLDS_DRIVER_KEY, _T("printMaxResolutionSupported"), REG_DWORD,
76		(BYTE *) &maxres, sizeof(maxres));
77}
78
79void map_extents(HANDLE ph, TCHAR *printer, TCHAR *port)
80{
81	DWORD extentval, xval, yval;
82
83	extentval = DeviceCapabilities(printer, port, DC_MINEXTENT, NULL, NULL);
84	xval = (DWORD) (LOWORD(extentval));
85	yval = (DWORD) (HIWORD(extentval));
86	printf("printMinXExtent: %d\n", xval);
87	printf("printMinYExtent: %d\n", yval);
88	SetPrinterDataEx(ph, SPLDS_DRIVER_KEY, _T("printMinXExtent"), REG_DWORD, (BYTE *) &xval, sizeof(xval));
89	SetPrinterDataEx(ph, SPLDS_DRIVER_KEY, _T("printMinYExtent"), REG_DWORD, (BYTE *) &yval, sizeof(yval));
90	extentval = DeviceCapabilities(printer, port, DC_MAXEXTENT, NULL, NULL);
91	xval = (DWORD) (LOWORD(extentval));
92	yval = (DWORD) (HIWORD(extentval));
93	printf("printMaxXExtent: %d\n", xval);
94	printf("printMaxYExtent: %d\n", yval);
95	SetPrinterDataEx(ph, SPLDS_DRIVER_KEY, _T("printMaxXExtent"), REG_DWORD, (BYTE *) &xval, sizeof(xval));
96	SetPrinterDataEx(ph, SPLDS_DRIVER_KEY, _T("printMaxYExtent"), REG_DWORD, (BYTE *) &yval, sizeof(yval));
97}
98
99void map_printrateunit(HANDLE ph, TCHAR *printer, TCHAR *port)
100{
101	DWORD unit;
102	TCHAR ppm[] = _T("PagesPerMinute");
103	TCHAR ipm[] = _T("InchesPerMinute");
104	TCHAR lpm[] = _T("LinesPerMinute");
105	TCHAR cps[] = _T("CharactersPerSecond");
106
107	unit = DeviceCapabilities(printer, port, DC_PRINTRATEUNIT, NULL, NULL);
108	switch(unit) {
109	case PRINTRATEUNIT_PPM:
110		printf("printRateUnit: %s\n", ppm);
111		SetPrinterDataEx(ph, SPLDS_DRIVER_KEY, _T("printRateUnit"), REG_SZ, ppm, sizeof(ppm));
112		break;
113	case PRINTRATEUNIT_IPM:
114		printf("printRateUnit: %s\n", ipm);
115		SetPrinterDataEx(ph, SPLDS_DRIVER_KEY, _T("printRateUnit"), REG_SZ, ipm, sizeof(ipm));
116		break;
117	case PRINTRATEUNIT_LPM:
118		printf("printRateUnit: %s\n", lpm);
119		SetPrinterDataEx(ph, SPLDS_DRIVER_KEY, _T("printRateUnit"), REG_SZ, lpm, sizeof(lpm));
120		break;
121	case PRINTRATEUNIT_CPS:
122		printf("printRateUnit: %s\n", cps);
123		SetPrinterDataEx(ph, SPLDS_DRIVER_KEY, _T("printRateUnit"), REG_SZ, cps, sizeof(cps));
124		break;
125	default:
126		printf("printRateUnit: unknown value %d\n", unit);
127	}
128}
129
130void map_generic_boolean(HANDLE ph, TCHAR *printer, TCHAR *port, WORD cap, TCHAR *key)
131{
132	BYTE boolval;
133	/* DeviceCapabilities doesn't always return 1 for true...just nonzero */
134	boolval = (BYTE) (DeviceCapabilities(printer, port, cap, NULL, NULL) ? 1 : 0);
135	printf("%s: %s\n", key, boolval ? "TRUE" : "FALSE");
136	SetPrinterDataEx(ph, SPLDS_DRIVER_KEY, key, REG_BINARY, &boolval, sizeof(boolval));
137}
138
139void map_generic_dword(HANDLE ph, TCHAR *printer, TCHAR *port, WORD cap, TCHAR *key)
140{
141	DWORD dword;
142
143	dword = DeviceCapabilities(printer, port, cap, NULL, NULL);
144	if ((DWORD) -1 == dword)
145		return;
146
147	printf("%s: %d\n", key, dword);
148	SetPrinterDataEx(ph, SPLDS_DRIVER_KEY, key, REG_DWORD, (BYTE *) &dword, sizeof(dword));
149}
150
151void map_generic_multi_sz(HANDLE ph, TCHAR *printer, TCHAR *port, WORD cap, TCHAR *key, int size)
152{
153	TCHAR *strings_in;
154	TCHAR *strings_out, *strings_cur;
155	DWORD num_items, i;
156
157	num_items = DeviceCapabilities(printer, port, cap, NULL, NULL);
158	if ((DWORD) -1 == num_items)
159		return;
160	strings_in = malloc(num_items * size);
161	strings_out = calloc(num_items, size);
162	num_items = DeviceCapabilities(printer, port, cap, strings_in, NULL);
163	printf("%s:\n", key);
164	for (i=0, strings_cur = strings_out; i < num_items; i++) {
165		_tcsncpy(strings_cur, &strings_in[i*size], size);
166		printf("\t%s\n", strings_cur);
167		strings_cur += _tcslen(strings_cur) + 1;
168	}
169
170	SetPrinterDataEx(ph, SPLDS_DRIVER_KEY, key, REG_MULTI_SZ, strings_out,
171		(strings_cur - strings_out + 1) * sizeof(TCHAR));
172
173	free(strings_in);
174	free(strings_out);
175}
176
177int main(int argc, char *argv[])
178{
179	HANDLE ph;
180	BYTE *driver_info;
181	DWORD needed;
182	TCHAR *printer;
183	TCHAR *port = SAMBA_PORT;
184	PRINTER_DEFAULTS admin_access = {NULL, NULL, PRINTER_ACCESS_ADMINISTER};
185	PRINTER_INFO_7 publish = {NULL, DSPRINT_PUBLISH};
186
187	if (argc < 2) {
188		printf("Usage: %s <printername>\n", argv[0]);
189		return -1;
190	}
191
192	printer = argv[1];
193
194	if (!(OpenPrinter(printer, &ph, &admin_access))) {
195		printf("OpenPrinter failed, error = %s\n", PrintLastError());
196		return -1;
197	}
198
199	GetPrinterDriver(ph, NULL, 1, NULL, 0, &needed);
200	if (!needed) {
201		printf("GetPrinterDriver failed, error = %s\n", PrintLastError());
202		ClosePrinter(ph);
203		return -1;
204	}
205	driver_info = malloc(needed);
206	if (!(GetPrinterDriver(ph, NULL, 1, driver_info, needed, &needed))) {
207		printf("GetPrinterDriver failed, error = %s\n", PrintLastError());
208		ClosePrinter(ph);
209		return -1;
210	}
211
212	map_generic_multi_sz(ph, printer, port, DC_BINNAMES, _T("printBinNames"), 24);
213	map_generic_boolean(ph, printer, port, DC_COLLATE, _T("printCollate"));
214	map_generic_dword(ph, printer, port, DC_COPIES, _T("printMaxCopies"));
215	map_generic_dword(ph, printer, port, DC_DRIVER, _T("driverVersion"));
216	map_generic_boolean(ph, printer, port, DC_DUPLEX, _T("printDuplexSupported"));
217	map_extents(ph, printer, port);
218	map_resolution(ph, printer, port);
219	map_orientation(ph, printer, port);
220	map_generic_multi_sz(ph, printer, port, DC_PAPERNAMES, _T("printMediaSupported"), 64);
221#if (WINVER >= 0x0500)
222	map_generic_boolean(ph, printer, port, DC_COLORDEVICE, _T("printColor"));
223	map_generic_multi_sz(ph, printer, port, DC_PERSONALITY, _T("printLanguage"), 64);
224	map_generic_multi_sz(ph, printer, port, DC_MEDIAREADY, _T("printMediaReady"),64);
225	map_generic_dword(ph, printer, port, DC_PRINTERMEM, _T("printMemory"));
226	map_generic_dword(ph, printer, port, DC_PRINTRATE, _T("printRate"));
227	map_printrateunit(ph, printer, port);
228#ifdef DC_STAPLE
229	map_generic_boolean(ph, printer, port, DC_STAPLE, _T("printStaplingSupported"));
230#endif
231#ifdef DC_PRINTRATEPPM
232	map_generic_dword(ph, printer, port, DC_PRINTRATEPPM, _T("printPagesPerMinute"));
233#endif
234#endif
235	SetPrinter(ph, 7, (BYTE *) &publish, 0);
236	ClosePrinter(ph);
237	return 0;
238}
239