331 lines
10 KiB
C
331 lines
10 KiB
C
/*
|
|
* 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 WCHAR does_not_existW[] = {'d','o','e','s','_','n','o','t','_','e','x','i','s','t',0};
|
|
static WCHAR emptyW[] = {0};
|
|
static CHAR fmt_comA[] = {'C','O','M','%','u',':',0};
|
|
static CHAR fmt_lptA[] = {'L','P','T','%','u',':',0};
|
|
static WCHAR localportW[] = {'L','o','c','a','l',' ','P','o','r','t',0};
|
|
static 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(LPWSTR pPort)
|
|
{
|
|
PORT_INFO_2W * pi;
|
|
DWORD res;
|
|
|
|
if (!pi_buffer) {
|
|
res = EnumPortsW(NULL, 2, NULL, 0, &pi_needed, &pi_numports);
|
|
ok(!res, "EnumPorts failed: 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 dialogs should be displayed, when hWnd
|
|
* is NULL, but native localui does not care
|
|
* - when the new port already exist,
|
|
* TRUE is returned, but new_portname is NULL
|
|
* - when the new port starts with "COM" or "LPT",
|
|
* FALSE is returned with ERROR_NOT_SUPPORTED in 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 */
|
|
|
|
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);
|
|
}
|
|
|
|
/* "FILE:" */
|
|
file_present = find_portinfo2(portname_fileW);
|
|
|
|
test_AddPortUI();
|
|
test_ConfigurePortUI();
|
|
|
|
/* cleanup */
|
|
HeapFree(GetProcessHeap(), 0, lpt_absent);
|
|
HeapFree(GetProcessHeap(), 0, com_absent);
|
|
HeapFree(GetProcessHeap(), 0, pi_buffer);
|
|
}
|