| /* |
| * Unit test suite for the Local Printmonitor User Interface |
| * |
| * Copyright 2007 Detlef Riekenberg |
| * |
| * This library is free software; you can redistribute it and/or |
| * modify it under the terms of the GNU Lesser General Public |
| * License as published by the Free Software Foundation; either |
| * version 2.1 of the License, or (at your option) any later version. |
| * |
| * This library is distributed in the hope that it will be useful, |
| * but WITHOUT ANY WARRANTY; without even the implied warranty of |
| * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU |
| * Lesser General Public License for more details. |
| * |
| * You should have received a copy of the GNU Lesser General Public |
| * License along with this library; if not, write to the Free Software |
| * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA |
| * |
| */ |
| |
| #include <stdarg.h> |
| #include <stdio.h> |
| |
| #include "windef.h" |
| #include "winbase.h" |
| #include "winerror.h" |
| #include "wingdi.h" |
| #include "winnls.h" |
| #include "winreg.h" |
| |
| #include "winspool.h" |
| #include "ddk/winsplp.h" |
| |
| #include "wine/test.h" |
| |
| |
| /* ##### */ |
| |
| static HMODULE hdll; |
| static PMONITORUI (WINAPI *pInitializePrintMonitorUI)(VOID); |
| static PMONITORUI pui; |
| static BOOL (WINAPI *pAddPortUI)(PCWSTR, HWND, PCWSTR, PWSTR *); |
| static BOOL (WINAPI *pConfigurePortUI)(PCWSTR, HWND, PCWSTR); |
| static BOOL (WINAPI *pDeletePortUI)(PCWSTR, HWND, PCWSTR); |
| |
| static const WCHAR does_not_existW[] = {'d','o','e','s','_','n','o','t','_','e','x','i','s','t',0}; |
| static const WCHAR emptyW[] = {0}; |
| static const CHAR fmt_comA[] = {'C','O','M','%','u',':',0}; |
| static const CHAR fmt_lptA[] = {'L','P','T','%','u',':',0}; |
| static const WCHAR localportW[] = {'L','o','c','a','l',' ','P','o','r','t',0}; |
| static const WCHAR portname_fileW[] = {'F','I','L','E',':',0}; |
| |
| static LPBYTE pi_buffer; |
| static DWORD pi_numports; |
| static DWORD pi_needed; |
| |
| static PORT_INFO_2W * lpt_present; |
| static PORT_INFO_2W * com_present; |
| static PORT_INFO_2W * file_present; |
| |
| static LPWSTR lpt_absent; |
| static LPWSTR com_absent; |
| |
| /* ########################### */ |
| |
| static PORT_INFO_2W * find_portinfo2(LPCWSTR pPort) |
| { |
| PORT_INFO_2W * pi; |
| DWORD res; |
| |
| if (!pi_buffer) { |
| res = EnumPortsW(NULL, 2, NULL, 0, &pi_needed, &pi_numports); |
| if (!res && (GetLastError() == RPC_S_SERVER_UNAVAILABLE)) { |
| win_skip("The service 'Spooler' is required for many tests\n"); |
| return NULL; |
| } |
| ok(!res, "EnumPorts succeeded: got %d\n", res); |
| pi_buffer = HeapAlloc(GetProcessHeap(), 0, pi_needed); |
| res = EnumPortsW(NULL, 2, pi_buffer, pi_needed, &pi_needed, &pi_numports); |
| ok(res == 1, "EnumPorts failed: got %d\n", res); |
| } |
| if (pi_buffer) { |
| pi = (PORT_INFO_2W *) pi_buffer; |
| res = 0; |
| while (pi_numports > res) { |
| if (lstrcmpiW(pi->pPortName, pPort) == 0) { |
| return pi; |
| } |
| pi++; |
| res++; |
| } |
| } |
| return NULL; |
| } |
| |
| |
| /* ########################### */ |
| |
| static LPCSTR load_functions(void) |
| { |
| LPCSTR ptr; |
| |
| ptr = "localui.dll"; |
| hdll = LoadLibraryA(ptr); |
| if (!hdll) return ptr; |
| |
| ptr = "InitializePrintMonitorUI"; |
| pInitializePrintMonitorUI = (VOID *) GetProcAddress(hdll, ptr); |
| if (!pInitializePrintMonitorUI) return ptr; |
| |
| return NULL; |
| } |
| |
| /* ########################### |
| * strdupW [internal] |
| */ |
| |
| static LPWSTR strdupW(LPCWSTR strW) |
| { |
| LPWSTR ptr; |
| |
| ptr = HeapAlloc(GetProcessHeap(), 0, (lstrlenW(strW) + 1) * sizeof(WCHAR)); |
| if (ptr) { |
| lstrcpyW(ptr, strW); |
| } |
| return ptr; |
| } |
| |
| /* ########################### */ |
| |
| static void test_AddPortUI(void) |
| { |
| DWORD res; |
| LPWSTR new_portname; |
| |
| /* not present before w2k */ |
| if (!pAddPortUI) { |
| skip("AddPortUI not found\n"); |
| return; |
| } |
| |
| SetLastError(0xdeadbeef); |
| res = pAddPortUI(NULL, NULL, NULL, NULL); |
| ok( !res && |
| ((GetLastError() == ERROR_UNKNOWN_PORT) || (GetLastError() == ERROR_INVALID_PRINTER_NAME)), |
| "got %d with %u (expected '0' with: ERROR_UNKNOWN_PORT or " |
| "ERROR_INVALID_PRINTER_NAME)\n", res, GetLastError()); |
| |
| SetLastError(0xdeadbeef); |
| res = pAddPortUI(NULL, NULL, emptyW, NULL); |
| ok( !res && |
| ((GetLastError() == ERROR_UNKNOWN_PORT) || (GetLastError() == ERROR_INVALID_PRINTER_NAME)), |
| "got %d with %u (expected '0' with: ERROR_UNKNOWN_PORT or " |
| "ERROR_INVALID_PRINTER_NAME)\n", res, GetLastError()); |
| |
| SetLastError(0xdeadbeef); |
| res = pAddPortUI(NULL, NULL, does_not_existW, NULL); |
| ok( !res && |
| ((GetLastError() == ERROR_UNKNOWN_PORT) || (GetLastError() == ERROR_INVALID_PRINTER_NAME)), |
| "got %d with %u (expected '0' with: ERROR_UNKNOWN_PORT or " |
| "ERROR_INVALID_PRINTER_NAME)\n", res, GetLastError()); |
| |
| if (winetest_interactive) { |
| SetLastError(0xdeadbeef); |
| new_portname = NULL; |
| /* |
| * - On MSDN, you can read that no dialog should be displayed when hWnd |
| * is NULL, but native localui does not care |
| * - When the new port already exists, |
| * TRUE is returned, but new_portname is NULL |
| * - When the new port starts with "COM" or "LPT", |
| * FALSE is returned with ERROR_NOT_SUPPORTED on windows |
| */ |
| res = pAddPortUI(NULL, NULL, localportW, &new_portname); |
| ok( res || |
| (GetLastError() == ERROR_CANCELLED) || |
| (GetLastError() == ERROR_ACCESS_DENIED) || |
| (GetLastError() == ERROR_NOT_SUPPORTED), |
| "got %d with %u and %p (expected '!= 0' or '0' with: " |
| "ERROR_CANCELLED, ERROR_ACCESS_DENIED or ERROR_NOT_SUPPORTED)\n", |
| res, GetLastError(), new_portname); |
| |
| GlobalFree(new_portname); |
| } |
| } |
| |
| /* ########################### */ |
| |
| static void test_ConfigurePortUI(void) |
| { |
| DWORD res; |
| |
| /* not present before w2k */ |
| if (!pConfigurePortUI) { |
| skip("ConfigurePortUI not found\n"); |
| return; |
| } |
| |
| SetLastError(0xdeadbeef); |
| res = pConfigurePortUI(NULL, NULL, NULL); |
| ok( !res && |
| ((GetLastError() == ERROR_UNKNOWN_PORT) || (GetLastError() == ERROR_INVALID_PRINTER_NAME)), |
| "got %d with %u (expected '0' with: ERROR_UNKNOWN_PORT or " |
| "ERROR_INVALID_PRINTER_NAME)\n", res, GetLastError()); |
| |
| SetLastError(0xdeadbeef); |
| res = pConfigurePortUI(NULL, NULL, emptyW); |
| ok( !res && |
| ((GetLastError() == ERROR_UNKNOWN_PORT) || (GetLastError() == ERROR_INVALID_PRINTER_NAME)), |
| "got %d with %u (expected '0' with: ERROR_UNKNOWN_PORT or " |
| "ERROR_INVALID_PRINTER_NAME)\n", res, GetLastError()); |
| |
| |
| SetLastError(0xdeadbeef); |
| res = pConfigurePortUI(NULL, NULL, does_not_existW); |
| ok( !res && |
| ((GetLastError() == ERROR_UNKNOWN_PORT) || (GetLastError() == ERROR_INVALID_PRINTER_NAME)), |
| "got %d with %u (expected '0' with: ERROR_UNKNOWN_PORT or " |
| "ERROR_INVALID_PRINTER_NAME)\n", res, GetLastError()); |
| |
| if (winetest_interactive && lpt_present) { |
| SetLastError(0xdeadbeef); |
| res = pConfigurePortUI(NULL, NULL, lpt_present->pPortName); |
| ok( res || |
| (GetLastError() == ERROR_CANCELLED) || (GetLastError() == ERROR_ACCESS_DENIED), |
| "got %d with %u (expected '!= 0' or '0' with: ERROR_CANCELLED or " |
| "ERROR_ACCESS_DENIED)\n", res, GetLastError()); |
| } |
| |
| if (lpt_absent) { |
| SetLastError(0xdeadbeef); |
| res = pConfigurePortUI(NULL, NULL, lpt_absent); |
| ok( !res && |
| ((GetLastError() == ERROR_UNKNOWN_PORT) || (GetLastError() == ERROR_INVALID_PRINTER_NAME)), |
| "got %d with %u (expected '0' with: ERROR_UNKNOWN_PORT or " |
| "ERROR_INVALID_PRINTER_NAME)\n", res, GetLastError()); |
| } |
| |
| if (winetest_interactive && com_present) { |
| SetLastError(0xdeadbeef); |
| res = pConfigurePortUI(NULL, NULL, com_present->pPortName); |
| ok( res || |
| (GetLastError() == ERROR_CANCELLED) || (GetLastError() == ERROR_ACCESS_DENIED), |
| "got %d with %u (expected '!= 0' or '0' with: ERROR_CANCELLED or " |
| "ERROR_ACCESS_DENIED)\n", res, GetLastError()); |
| } |
| |
| if (com_absent) { |
| SetLastError(0xdeadbeef); |
| res = pConfigurePortUI(NULL, NULL, com_absent); |
| ok( !res && |
| ((GetLastError() == ERROR_UNKNOWN_PORT) || (GetLastError() == ERROR_INVALID_PRINTER_NAME)), |
| "got %d with %u (expected '0' with: ERROR_UNKNOWN_PORT or " |
| "ERROR_INVALID_PRINTER_NAME)\n", res, GetLastError()); |
| |
| } |
| |
| if (winetest_interactive && file_present) { |
| SetLastError(0xdeadbeef); |
| res = pConfigurePortUI(NULL, NULL, portname_fileW); |
| ok( !res && |
| ((GetLastError() == ERROR_CANCELLED) || (GetLastError() == ERROR_ACCESS_DENIED)), |
| "got %d with %u (expected '0' with: ERROR_CANCELLED or " |
| "ERROR_ACCESS_DENIED)\n", res, GetLastError()); |
| } |
| } |
| |
| /* ########################### */ |
| |
| START_TEST(localui) |
| { |
| LPCSTR ptr; |
| DWORD numentries; |
| PORT_INFO_2W * pi2; |
| WCHAR bufferW[16]; |
| CHAR bufferA[16]; |
| DWORD id; |
| |
| /* localui.dll does not exist before w2k */ |
| ptr = load_functions(); |
| if (ptr) { |
| skip("%s not found\n", ptr); |
| return; |
| } |
| |
| pui = pInitializePrintMonitorUI(); |
| if (pui) { |
| numentries = (pui->dwMonitorUISize - sizeof(DWORD)) / sizeof(VOID *); |
| ok( numentries == 3, |
| "dwMonitorUISize (%d) => %d Functions\n", pui->dwMonitorUISize, numentries); |
| |
| if (numentries > 2) { |
| pAddPortUI = pui->pfnAddPortUI; |
| pConfigurePortUI = pui->pfnConfigurePortUI; |
| pDeletePortUI = pui->pfnDeletePortUI; |
| } |
| } |
| |
| /* find installed ports */ |
| |
| /* "FILE:" */ |
| file_present = find_portinfo2(portname_fileW); |
| |
| if (!pi_numports) /* Nothing to test without a port */ |
| return; |
| |
| id = 0; |
| /* "LPT1:" - "LPT9:" */ |
| while (((lpt_present == NULL) || (lpt_absent == NULL)) && id < 9) { |
| id++; |
| sprintf(bufferA, fmt_lptA, id); |
| MultiByteToWideChar( CP_ACP, 0, bufferA, -1, bufferW, sizeof(bufferW)/sizeof(WCHAR) ); |
| pi2 = find_portinfo2(bufferW); |
| if (pi2 && (lpt_present == NULL)) lpt_present = pi2; |
| if (!pi2 && (lpt_absent == NULL)) lpt_absent = strdupW(bufferW); |
| } |
| |
| id = 0; |
| /* "COM1:" - "COM9:" */ |
| while (((com_present == NULL) || (com_absent == NULL)) && id < 9) { |
| id++; |
| sprintf(bufferA, fmt_comA, id); |
| MultiByteToWideChar( CP_ACP, 0, bufferA, -1, bufferW, sizeof(bufferW)/sizeof(WCHAR) ); |
| pi2 = find_portinfo2(bufferW); |
| if (pi2 && (com_present == NULL)) com_present = pi2; |
| if (!pi2 && (com_absent == NULL)) com_absent = strdupW(bufferW); |
| } |
| |
| test_AddPortUI(); |
| test_ConfigurePortUI(); |
| |
| /* cleanup */ |
| HeapFree(GetProcessHeap(), 0, lpt_absent); |
| HeapFree(GetProcessHeap(), 0, com_absent); |
| HeapFree(GetProcessHeap(), 0, pi_buffer); |
| } |