Sweden-Number/dlls/winspool/tests/info.c

479 lines
17 KiB
C

/*
* Copyright (C) 2003, 2004 Stefan Leichter
* Copyright (C) 2005 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., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
*/
#include <stdarg.h>
#include "wine/test.h"
#include "windef.h"
#include "winbase.h"
#include "winerror.h"
#include "wingdi.h"
#include "winreg.h"
#include "winspool.h"
#define MAGIC_DEAD 0x00dead00
#define DEFAULT_PRINTER_SIZE 1000
static char env_x86[] = "Windows NT x86";
static char env_win9x_case[] = "windowS 4.0";
static HANDLE hwinspool;
static FARPROC pGetDefaultPrinterA;
static LPSTR find_default_printer(VOID)
{
static LPSTR default_printer = NULL;
static char buffer[DEFAULT_PRINTER_SIZE];
DWORD needed;
DWORD res;
LPSTR ptr;
if ((default_printer == NULL) && (pGetDefaultPrinterA))
{
/* w2k and above */
needed = sizeof(buffer);
res = pGetDefaultPrinterA(buffer, &needed);
if(res) default_printer = buffer;
trace("default_printer: '%s'\n", default_printer);
}
if (default_printer == NULL)
{
HKEY hwindows;
DWORD type;
/* NT 3.x and above */
if (RegOpenKeyEx(HKEY_CURRENT_USER,
"Software\\Microsoft\\Windows NT\\CurrentVersion\\Windows",
0, KEY_QUERY_VALUE, &hwindows) == NO_ERROR) {
needed = sizeof(buffer);
if (RegQueryValueEx(hwindows, "device", NULL,
&type, (LPBYTE)buffer, &needed) == NO_ERROR) {
ptr = strchr(buffer, ',');
if (ptr) {
ptr[0] = '\0';
default_printer = buffer;
}
}
RegCloseKey(hwindows);
}
trace("default_printer: '%s'\n", default_printer);
}
if (default_printer == NULL)
{
/* win9x */
needed = sizeof(buffer);
res = GetProfileStringA("windows", "device", "*", buffer, needed);
if(res) {
ptr = strchr(buffer, ',');
if (ptr) {
ptr[0] = '\0';
default_printer = buffer;
}
}
trace("default_printer: '%s'\n", default_printer);
}
return default_printer;
}
static void test_GetDefaultPrinter(void)
{
BOOL retval;
DWORD exact = DEFAULT_PRINTER_SIZE;
DWORD size;
char buffer[DEFAULT_PRINTER_SIZE];
if (!pGetDefaultPrinterA) return;
/* only supported on NT like OSes starting with win2k */
SetLastError(ERROR_SUCCESS);
retval = pGetDefaultPrinterA(buffer, &exact);
if (!retval || !exact || !strlen(buffer) ||
(ERROR_SUCCESS != GetLastError())) {
if ((ERROR_FILE_NOT_FOUND == GetLastError()) ||
(ERROR_INVALID_NAME == GetLastError()))
trace("this test requires a default printer to be set\n");
else {
ok( 0, "function call GetDefaultPrinterA failed unexpected!\n"
"function returned %s\n"
"last error 0x%08lx\n"
"returned buffer size 0x%08lx\n"
"returned buffer content %s\n",
retval ? "true" : "false", GetLastError(), exact, buffer);
}
return;
}
SetLastError(ERROR_SUCCESS);
retval = pGetDefaultPrinterA(NULL, NULL);
ok( !retval, "function result wrong! False expected\n");
ok( ERROR_INVALID_PARAMETER == GetLastError(),
"Last error wrong! ERROR_INVALID_PARAMETER expected, got 0x%08lx\n",
GetLastError());
SetLastError(ERROR_SUCCESS);
retval = pGetDefaultPrinterA(buffer, NULL);
ok( !retval, "function result wrong! False expected\n");
ok( ERROR_INVALID_PARAMETER == GetLastError(),
"Last error wrong! ERROR_INVALID_PARAMETER expected, got 0x%08lx\n",
GetLastError());
SetLastError(ERROR_SUCCESS);
size = 0;
retval = pGetDefaultPrinterA(NULL, &size);
ok( !retval, "function result wrong! False expected\n");
ok( ERROR_INSUFFICIENT_BUFFER == GetLastError(),
"Last error wrong! ERROR_INSUFFICIENT_BUFFER expected, got 0x%08lx\n",
GetLastError());
ok( size == exact, "Parameter size wrong! %ld expected got %ld\n",
exact, size);
SetLastError(ERROR_SUCCESS);
size = DEFAULT_PRINTER_SIZE;
retval = pGetDefaultPrinterA(NULL, &size);
ok( !retval, "function result wrong! False expected\n");
ok( ERROR_INSUFFICIENT_BUFFER == GetLastError(),
"Last error wrong! ERROR_INSUFFICIENT_BUFFER expected, got 0x%08lx\n",
GetLastError());
ok( size == exact, "Parameter size wrong! %ld expected got %ld\n",
exact, size);
size = 0;
retval = pGetDefaultPrinterA(buffer, &size);
ok( !retval, "function result wrong! False expected\n");
ok( ERROR_INSUFFICIENT_BUFFER == GetLastError(),
"Last error wrong! ERROR_INSUFFICIENT_BUFFER expected, got 0x%08lx\n",
GetLastError());
ok( size == exact, "Parameter size wrong! %ld expected got %ld\n",
exact, size);
size = exact;
retval = pGetDefaultPrinterA(buffer, &size);
ok( retval, "function result wrong! True expected\n");
ok( size == exact, "Parameter size wrong! %ld expected got %ld\n",
exact, size);
}
static void test_GetPrinterDriverDirectory(void)
{
LPBYTE buffer = NULL;
DWORD cbBuf = 0, pcbNeeded = 0;
BOOL res;
SetLastError(0x00dead00);
res = GetPrinterDriverDirectoryA( NULL, NULL, 1, NULL, 0, &cbBuf);
trace("GetPrinterDriverDirectoryA: first call returned 0x%04x, "
"buffer size 0x%08lx\n", res, cbBuf);
if((res == 0) && (GetLastError() == RPC_S_SERVER_UNAVAILABLE))
{
trace("The Service 'Spooler' is required for this test\n");
return;
}
ok((res == 0) && (GetLastError() == ERROR_INSUFFICIENT_BUFFER),
"returned %d with lasterror=%ld (expected '0' with " \
"ERROR_INSUFFICIENT_BUFFER)\n", res, GetLastError());
if (!cbBuf) {
trace("no valid buffer size returned, skipping tests\n");
return;
}
buffer = HeapAlloc( GetProcessHeap(), 0, cbBuf*2);
if (buffer == NULL) return ;
res = GetPrinterDriverDirectoryA(NULL, NULL, 1, buffer, cbBuf, &pcbNeeded);
ok( res, "expected result != 0, got %d\n", res);
ok( cbBuf == pcbNeeded, "pcbNeeded set to %ld instead of %ld\n",
pcbNeeded, cbBuf);
res = GetPrinterDriverDirectoryA(NULL, NULL, 1, buffer, cbBuf*2, &pcbNeeded);
ok( res, "expected result != 0, got %d\n", res);
ok( cbBuf == pcbNeeded, "pcbNeeded set to %ld instead of %ld\n",
pcbNeeded, cbBuf);
SetLastError(0x00dead00);
res = GetPrinterDriverDirectoryA( NULL, NULL, 1, buffer, cbBuf-1, &pcbNeeded);
ok( !res , "expected result == 0, got %d\n", res);
ok( cbBuf == pcbNeeded, "pcbNeeded set to %ld instead of %ld\n",
pcbNeeded, cbBuf);
ok( ERROR_INSUFFICIENT_BUFFER == GetLastError(),
"last error set to %ld instead of ERROR_INSUFFICIENT_BUFFER\n",
GetLastError());
SetLastError(0x00dead00);
res = GetPrinterDriverDirectoryA( NULL, NULL, 1, NULL, cbBuf, &pcbNeeded);
ok( (!res && ERROR_INVALID_USER_BUFFER == GetLastError()) ||
( res && ERROR_INVALID_PARAMETER == GetLastError()) ,
"expected either result == 0 and "
"last error == ERROR_INVALID_USER_BUFFER "
"or result != 0 and last error == ERROR_INVALID_PARAMETER "
"got result %d and last error == %ld\n", res, GetLastError());
SetLastError(0x00dead00);
res = GetPrinterDriverDirectoryA( NULL, NULL, 1, buffer, cbBuf, NULL);
ok( (!res && RPC_X_NULL_REF_POINTER == GetLastError()) || res,
"expected either result == 0 and "
"last error == RPC_X_NULL_REF_POINTER or result != 0 "
"got result %d and last error == %ld\n", res, GetLastError());
SetLastError(0x00dead00);
res = GetPrinterDriverDirectoryA( NULL, NULL, 1, NULL, cbBuf, NULL);
ok( (!res && RPC_X_NULL_REF_POINTER == GetLastError()) ||
( res && ERROR_INVALID_PARAMETER == GetLastError()) ,
"expected either result == 0 and "
"last error == RPC_X_NULL_REF_POINTER "
"or result != 0 and last error == ERROR_INVALID_PARAMETER "
"got result %d and last error == %ld\n", res, GetLastError());
/* with a valid buffer, but level is too large */
buffer[0] = '\0';
SetLastError(0x00dead00);
res = GetPrinterDriverDirectoryA(NULL, NULL, 2, buffer, cbBuf, &pcbNeeded);
/* Level not checked in win9x and wine:*/
if((res != FALSE) && buffer[0])
{
trace("invalid Level '2' not checked (valid Level is '1') => '%s'\n",
buffer);
}
else
{
ok( !res && (GetLastError() == ERROR_INVALID_LEVEL),
"returned %d with lasterror=%ld (expected '0' with " \
"ERROR_INVALID_LEVEL)\n", res, GetLastError());
}
/* printing environments are case insensitive */
/* "Windows 4.0" is valid for win9x and NT */
buffer[0] = '\0';
SetLastError(0x00dead00);
res = GetPrinterDriverDirectoryA(NULL, env_win9x_case, 1,
buffer, cbBuf*2, &pcbNeeded);
if(!res && (GetLastError() == ERROR_INSUFFICIENT_BUFFER)) {
cbBuf = pcbNeeded;
buffer = HeapReAlloc(GetProcessHeap(), 0, buffer, cbBuf*2);
if (buffer == NULL) return ;
SetLastError(0x00dead00);
res = GetPrinterDriverDirectoryA(NULL, env_win9x_case, 1,
buffer, cbBuf*2, &pcbNeeded);
}
ok(res && buffer[0], "returned %d with " \
"lasterror=%ld and len=%d (expected '1' with 'len > 0')\n",
res, GetLastError(), lstrlenA((char *)buffer));
buffer[0] = '\0';
SetLastError(0x00dead00);
res = GetPrinterDriverDirectoryA(NULL, env_x86, 1,
buffer, cbBuf*2, &pcbNeeded);
if(!res && (GetLastError() == ERROR_INSUFFICIENT_BUFFER)) {
cbBuf = pcbNeeded;
buffer = HeapReAlloc(GetProcessHeap(), 0, buffer, cbBuf*2);
if (buffer == NULL) return ;
buffer[0] = '\0';
SetLastError(0x00dead00);
res = GetPrinterDriverDirectoryA(NULL, env_x86, 1,
buffer, cbBuf*2, &pcbNeeded);
}
/* "Windows NT x86" is invalid for win9x */
ok( (res && buffer[0]) ||
(!res && (GetLastError() == ERROR_INVALID_ENVIRONMENT)),
"returned %d with lasterror=%ld and len=%d (expected '!= 0' with " \
"'len > 0' or '0' with ERROR_INVALID_ENVIRONMENT)\n",
res, GetLastError(), lstrlenA((char *)buffer));
/* A Setup-Programm (PDFCreator_0.8.0) use empty strings */
SetLastError(MAGIC_DEAD);
res = GetPrinterDriverDirectoryA("", "", 1, buffer, cbBuf*2, &pcbNeeded);
ok(res, "returned %d with %ld (expected '!=0')\n", res, GetLastError() );
SetLastError(MAGIC_DEAD);
res = GetPrinterDriverDirectoryA(NULL, "", 1, buffer, cbBuf*2, &pcbNeeded);
ok(res, "returned %d with %ld (expected '!=0')\n", res, GetLastError() );
SetLastError(MAGIC_DEAD);
res = GetPrinterDriverDirectoryA("", NULL, 1, buffer, cbBuf*2, &pcbNeeded);
ok(res, "returned %d with %ld (expected '!=0')\n", res, GetLastError() );
HeapFree( GetProcessHeap(), 0, buffer);
}
static void test_OpenPrinter(void)
{
PRINTER_DEFAULTSA defaults;
HANDLE hprinter;
LPSTR default_printer;
DWORD res;
DWORD size;
CHAR buffer[DEFAULT_PRINTER_SIZE];
LPSTR ptr;
SetLastError(MAGIC_DEAD);
res = OpenPrinter(NULL, NULL, NULL);
if((res == 0) && (GetLastError() == RPC_S_SERVER_UNAVAILABLE))
{
trace("The Service 'Spooler' is required for this test\n");
return;
}
ok(!res && (GetLastError() == ERROR_INVALID_PARAMETER),
"returned %ld with %ld (expected '0' with ERROR_INVALID_PARAMETER)\n",
res, GetLastError());
/* Get Handle for the local Printserver (NT only)*/
hprinter = (HANDLE) MAGIC_DEAD;
SetLastError(MAGIC_DEAD);
res = OpenPrinter(NULL, &hprinter, NULL);
ok(res || (!res && GetLastError() == ERROR_INVALID_PARAMETER),
"returned %ld with %ld (expected '!=0' or '0' with ERROR_INVALID_PARAMETER)\n",
res, GetLastError());
if(res) {
ClosePrinter(hprinter);
defaults.pDatatype=NULL;
defaults.pDevMode=NULL;
defaults.DesiredAccess=0;
hprinter = (HANDLE) MAGIC_DEAD;
SetLastError(MAGIC_DEAD);
res = OpenPrinter(NULL, &hprinter, &defaults);
ok(res, "returned %ld with %ld (expected '!=0')\n", res, GetLastError());
if (res) ClosePrinter(hprinter);
defaults.DesiredAccess=-1;
hprinter = (HANDLE) MAGIC_DEAD;
SetLastError(MAGIC_DEAD);
res = OpenPrinter(NULL, &hprinter, &defaults);
ok(!res && GetLastError() == ERROR_ACCESS_DENIED,
"returned %ld with %ld (expected '0' with ERROR_ACCESS_DENIED)\n",
res, GetLastError());
if (res) ClosePrinter(hprinter);
}
size = sizeof(buffer) - 3 ;
ptr = buffer;
ptr[0] = '\\';
ptr++;
ptr[0] = '\\';
ptr++;
if (GetComputerNameA(ptr, &size)) {
hprinter = (HANDLE) MAGIC_DEAD;
SetLastError(MAGIC_DEAD);
res = OpenPrinter(buffer, &hprinter, NULL);
todo_wine {
ok(res || (!res && GetLastError() == ERROR_INVALID_PARAMETER),
"returned %ld with %ld (expected '!=0' or '0' with ERROR_INVALID_PARAMETER)\n",
res, GetLastError());
}
if(res) ClosePrinter(hprinter);
}
/* Invalid Printername */
hprinter = (HANDLE) MAGIC_DEAD;
SetLastError(MAGIC_DEAD);
res = OpenPrinter("illegal,name", &hprinter, NULL);
ok(!res && ((GetLastError() == ERROR_INVALID_PRINTER_NAME) ||
(GetLastError() == ERROR_INVALID_PARAMETER) ),
"returned %ld with %ld (expected '0' with: ERROR_INVALID_PARAMETER or" \
"ERROR_INVALID_PRINTER_NAME)\n", res, GetLastError());
if(res) ClosePrinter(hprinter);
/* Get Handle for the default Printer */
if ((default_printer = find_default_printer()))
{
hprinter = (HANDLE) MAGIC_DEAD;
SetLastError(MAGIC_DEAD);
res = OpenPrinter(default_printer, &hprinter, NULL);
ok(res, "returned %ld with %ld (expected '!=0')\n", res, GetLastError());
if(res) ClosePrinter(hprinter);
defaults.pDatatype=NULL;
defaults.pDevMode=NULL;
defaults.DesiredAccess=0;
hprinter = (HANDLE) MAGIC_DEAD;
SetLastError(MAGIC_DEAD);
res = OpenPrinter(default_printer, &hprinter, &defaults);
ok(res, "returned %ld with %ld (expected '!=0')\n", res, GetLastError());
if(res) ClosePrinter(hprinter);
defaults.pDatatype="";
hprinter = (HANDLE) MAGIC_DEAD;
SetLastError(MAGIC_DEAD);
res = OpenPrinter(default_printer, &hprinter, &defaults);
ok(res || (GetLastError() == ERROR_INVALID_DATATYPE ||
GetLastError() == RPC_S_SERVER_UNAVAILABLE),
"returned %ld with %ld (expected '!=0' or '0' with: " \
"ERROR_INVALID_DATATYPE or RPC_S_SERVER_UNAVAILABLE)\n",
res, GetLastError());
if(res) ClosePrinter(hprinter);
defaults.pDatatype=NULL;
defaults.DesiredAccess=PRINTER_ACCESS_USE;
hprinter = (HANDLE) MAGIC_DEAD;
SetLastError(MAGIC_DEAD);
res = OpenPrinter(default_printer, &hprinter, &defaults);
ok(res || (GetLastError() == ERROR_ACCESS_DENIED ||
GetLastError() == RPC_S_SERVER_UNAVAILABLE),
"returned %ld with %ld (expected '!=0' or '0' with: " \
"ERROR_ACCESS_DENIED or RPC_S_SERVER_UNAVAILABLE)\n",
res, GetLastError());
if(res) ClosePrinter(hprinter);
defaults.DesiredAccess=PRINTER_ALL_ACCESS;
hprinter = (HANDLE) MAGIC_DEAD;
SetLastError(MAGIC_DEAD);
res = OpenPrinter(default_printer, &hprinter, &defaults);
ok(res || (GetLastError() == ERROR_ACCESS_DENIED ||
GetLastError() == RPC_S_SERVER_UNAVAILABLE),
"returned %ld with %ld (expected '!=0' or '0' with: " \
"ERROR_ACCESS_DENIED or RPC_S_SERVER_UNAVAILABLE)\n",
res, GetLastError());
if(res) ClosePrinter(hprinter);
}
}
START_TEST(info)
{
hwinspool = GetModuleHandleA("winspool.drv");
pGetDefaultPrinterA = (void *) GetProcAddress(hwinspool, "GetDefaultPrinterA");
find_default_printer();
test_GetDefaultPrinter();
test_GetPrinterDriverDirectory();
test_OpenPrinter();
}