Module: wine Branch: master Commit: 178294a42305f1a3f7c67a58b081020037cb3f64 URL: http://source.winehq.org/git/wine.git/?a=commit;h=178294a42305f1a3f7c67a58b0...
Author: Detlef Riekenberg wine.dev@web.de Date: Wed Oct 13 00:25:17 2010 +0200
winspool: Implement SetDefaultPrinterA/W.
---
dlls/winspool.drv/info.c | 98 +++++++++++++++++++++++++++++++++++++++++++--- 1 files changed, 92 insertions(+), 6 deletions(-)
diff --git a/dlls/winspool.drv/info.c b/dlls/winspool.drv/info.c index 92098f1..2f3c347 100644 --- a/dlls/winspool.drv/info.c +++ b/dlls/winspool.drv/info.c @@ -5258,11 +5258,91 @@ end: */ BOOL WINAPI SetDefaultPrinterW(LPCWSTR pszPrinter) { + WCHAR default_printer[MAX_PATH]; + LPWSTR buffer = NULL; + HKEY hreg; + DWORD size; + DWORD namelen; + LONG lres;
TRACE("(%s)\n", debugstr_w(pszPrinter)); + if ((pszPrinter == NULL) || (pszPrinter[0] == '\0')) {
- SetLastError(ERROR_CALL_NOT_IMPLEMENTED); - return FALSE; + default_printer[0] = '\0'; + size = sizeof(default_printer)/sizeof(WCHAR); + + /* if we have a default Printer, do nothing. */ + if (GetDefaultPrinterW(default_printer, &size)) + return TRUE; + + pszPrinter = NULL; + /* we have no default Printer: search local Printers and use the first */ + if (!RegOpenKeyExW(HKEY_LOCAL_MACHINE, PrintersW, 0, KEY_READ, &hreg)) { + + default_printer[0] = '\0'; + size = sizeof(default_printer)/sizeof(WCHAR); + if (!RegEnumKeyExW(hreg, 0, default_printer, &size, NULL, NULL, NULL, NULL)) { + + pszPrinter = default_printer; + TRACE("using %s\n", debugstr_w(pszPrinter)); + } + RegCloseKey(hreg); + } + + if (pszPrinter == NULL) { + TRACE("no local printer found\n"); + SetLastError(ERROR_FILE_NOT_FOUND); + return FALSE; + } + } + + /* "pszPrinter" is never empty or NULL here. */ + namelen = lstrlenW(pszPrinter); + size = namelen + (MAX_PATH * 2) + 3; /* printer,driver,port and a 0 */ + buffer = HeapAlloc(GetProcessHeap(), 0, size * sizeof(WCHAR)); + if (!buffer || + (RegOpenKeyExW(HKEY_CURRENT_USER, user_printers_reg_key, 0, KEY_READ, &hreg) != ERROR_SUCCESS)) { + HeapFree(GetProcessHeap(), 0, buffer); + SetLastError(ERROR_FILE_NOT_FOUND); + return FALSE; + } + + /* read the devices entry for the printer (driver,port) to build the string for the + default device entry (printer,driver,port) */ + memcpy(buffer, pszPrinter, namelen * sizeof(WCHAR)); + buffer[namelen] = ','; + namelen++; /* move index to the start of the driver */ + + size = ((MAX_PATH * 2) + 2) * sizeof(WCHAR); /* driver,port and a 0 */ + lres = RegQueryValueExW(hreg, pszPrinter, NULL, NULL, (LPBYTE) (&buffer[namelen]), &size); + if (!lres) { + TRACE("set device to %s\n", debugstr_w(buffer)); + + if (!WriteProfileStringW(windowsW, deviceW, buffer)) { + TRACE("failed to set the device entry: %d\n", GetLastError()); + lres = ERROR_INVALID_PRINTER_NAME; + } + + /* remove the next section, when INIFileMapping is implemented */ + { + HKEY hdev; + if (!RegCreateKeyW(HKEY_CURRENT_USER, user_default_reg_key, &hdev)) { + RegSetValueExW(hdev, deviceW, 0, REG_SZ, (LPBYTE)buffer, (lstrlenW(buffer) + 1) * sizeof(WCHAR)); + RegCloseKey(hdev); + } + } + } + else + { + if (lres != ERROR_FILE_NOT_FOUND) + FIXME("RegQueryValueExW failed with %d for %s\n", lres, debugstr_w(pszPrinter)); + + SetLastError(ERROR_INVALID_PRINTER_NAME); + } + + RegCloseKey(hreg); + HeapFree(GetProcessHeap(), 0, buffer); + return (lres == ERROR_SUCCESS); }
/****************************************************************************** @@ -5273,14 +5353,20 @@ BOOL WINAPI SetDefaultPrinterW(LPCWSTR pszPrinter) */ BOOL WINAPI SetDefaultPrinterA(LPCSTR pszPrinter) { + LPWSTR bufferW = NULL; + BOOL res;
TRACE("(%s)\n", debugstr_a(pszPrinter)); - - SetLastError(ERROR_CALL_NOT_IMPLEMENTED); - return FALSE; + if(pszPrinter) { + INT len = MultiByteToWideChar(CP_ACP, 0, pszPrinter, -1, NULL, 0); + bufferW = HeapAlloc(GetProcessHeap(), 0, len * sizeof(WCHAR)); + if (bufferW) MultiByteToWideChar(CP_ACP, 0, pszPrinter, -1, bufferW, len); + } + res = SetDefaultPrinterW(bufferW); + HeapFree(GetProcessHeap(), 0, bufferW); + return res; }
- /****************************************************************************** * SetPrinterDataExA (WINSPOOL.@) */
-
Alexandre Julliard