wksta.c 7.45 KB
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17
/*
 * Copyright 2002 Andriy Palamarchuk
 *
 * Conformance test of the workstation functions.
 *
 * 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
18
 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
19 20
 */

21 22
#include <stdarg.h>

23
#include "wine/test.h"
24
#include "windef.h"
25
#include "winbase.h"
26 27
#include "wingdi.h"
#include "winnls.h"
28
#include "winresrc.h" /* Ensure we use Unicode defns with native headers */
29 30 31 32 33 34
#include "nb30.h"
#include "lmcons.h"
#include "lmerr.h"
#include "lmwksta.h"
#include "lmapibuf.h"

35 36 37 38
static NET_API_STATUS (WINAPI *pNetApiBufferFree)(LPVOID)=NULL;
static NET_API_STATUS (WINAPI *pNetApiBufferSize)(LPVOID,LPDWORD)=NULL;
static NET_API_STATUS (WINAPI *pNetpGetComputerName)(LPWSTR*)=NULL;
static NET_API_STATUS (WINAPI *pNetWkstaUserGetInfo)(LPWSTR,DWORD,PBYTE*)=NULL;
39 40
static NET_API_STATUS (WINAPI *pNetWkstaTransportEnum)(LPWSTR,DWORD,LPBYTE*,
 DWORD,LPDWORD,LPDWORD,LPDWORD)=NULL;
41

42 43 44
WCHAR user_name[UNLEN + 1];
WCHAR computer_name[MAX_COMPUTERNAME_LENGTH + 1];

45
static int init_wksta_tests(void)
46 47
{
    DWORD dwSize;
48
    BOOL rc;
49 50

    user_name[0] = 0;
51
    dwSize = sizeof(user_name)/sizeof(user_name[0]);
52
    rc=GetUserNameW(user_name, &dwSize);
53 54
    if (rc==FALSE && GetLastError()==ERROR_CALL_NOT_IMPLEMENTED) {
        skip("GetUserNameW is not implemented\n");
55
        return 0;
56
    }
57
    ok(rc, "User Name Retrieved\n");
58 59

    computer_name[0] = 0;
60
    dwSize = sizeof(computer_name)/sizeof(computer_name[0]);
61
    ok(GetComputerNameW(computer_name, &dwSize), "Computer Name Retrieved\n");
62
    return 1;
63 64
}

65
static void run_get_comp_name_tests(void)
66
{
67
    LPWSTR ws = NULL;
68

69 70
    ok(pNetpGetComputerName(&ws) == NERR_Success, "Computer name is retrieved\n");
    ok(!lstrcmpW(computer_name, ws), "This is really computer name\n");
71
    pNetApiBufferFree(ws);
72 73
}

74
static void run_wkstausergetinfo_tests(void)
75 76 77 78 79 80 81
{
    LPWKSTA_USER_INFO_0 ui0 = NULL;
    LPWKSTA_USER_INFO_1 ui1 = NULL;
    LPWKSTA_USER_INFO_1101 ui1101 = NULL;
    DWORD dwSize;

    /* Level 0 */
82
    ok(pNetWkstaUserGetInfo(NULL, 0, (LPBYTE *)&ui0) == NERR_Success,
83 84 85
       "NetWkstaUserGetInfo is unsuccessful\n");

    ok(ui0 != NULL, "ui0 is NULL\n");
86
    /* This failure occurred when I ran sshd as service and didn't authenticate
87 88 89 90 91 92 93
     * Since the test dereferences ui0, the rest of this test is worthless
     */
    if (!ui0)
    {
        return;
    }

94
    ok(!lstrcmpW(user_name, ui0->wkui0_username), "This is really user name\n");
95
    pNetApiBufferSize(ui0, &dwSize);
96 97
    ok(dwSize >= (sizeof(WKSTA_USER_INFO_0) +
                 lstrlenW(ui0->wkui0_username) * sizeof(WCHAR)),
98
       "Is allocated with NetApiBufferAllocate\n");
99 100

    /* Level 1 */
101
    ok(pNetWkstaUserGetInfo(NULL, 1, (LPBYTE *)&ui1) == NERR_Success,
102
       "NetWkstaUserGetInfo is successful\n");
103
    ok(lstrcmpW(ui1->wkui1_username, ui0->wkui0_username) == 0,
104
       "the same name as returned for level 0\n");
105
    pNetApiBufferSize(ui1, &dwSize);
106 107 108 109 110
    ok(dwSize >= (sizeof(WKSTA_USER_INFO_1) +
                  (lstrlenW(ui1->wkui1_username) +
                   lstrlenW(ui1->wkui1_logon_domain) +
                   lstrlenW(ui1->wkui1_oth_domains) +
                   lstrlenW(ui1->wkui1_logon_server)) * sizeof(WCHAR)),
111
       "Is allocated with NetApiBufferAllocate\n");
112 113

    /* Level 1101 */
114
    ok(pNetWkstaUserGetInfo(NULL, 1101, (LPBYTE *)&ui1101) == NERR_Success,
115
       "NetWkstaUserGetInfo is successful\n");
116
    ok(lstrcmpW(ui1101->wkui1101_oth_domains, ui1->wkui1_oth_domains) == 0,
117
       "the same oth_domains as returned for level 1\n");
118
    pNetApiBufferSize(ui1101, &dwSize);
119 120
    ok(dwSize >= (sizeof(WKSTA_USER_INFO_1101) +
                 lstrlenW(ui1101->wkui1101_oth_domains) * sizeof(WCHAR)),
121
       "Is allocated with NetApiBufferAllocate\n");
122

123 124 125
    pNetApiBufferFree(ui0);
    pNetApiBufferFree(ui1);
    pNetApiBufferFree(ui1101);
126 127

    /* errors handling */
128
    ok(pNetWkstaUserGetInfo(NULL, 10000, (LPBYTE *)&ui0) == ERROR_INVALID_LEVEL,
129
       "Invalid level\n");
130 131
}

132 133 134 135 136 137 138 139 140
static void run_wkstatransportenum_tests(void)
{
    LPBYTE bufPtr;
    NET_API_STATUS apiReturn;
    DWORD entriesRead, totalEntries;

    /* 1st check: is param 2 (level) correct? (only if param 5 passed?) */
    apiReturn = pNetWkstaTransportEnum(NULL, 1, NULL, MAX_PREFERRED_LENGTH,
        NULL, &totalEntries, NULL);
141
    ok(apiReturn == ERROR_INVALID_LEVEL || apiReturn == ERROR_INVALID_PARAMETER,
142
       "NetWkstaTransportEnum returned %d\n", apiReturn);
143 144 145 146 147 148 149 150 151

    /* 2nd check: is param 5 passed? (only if level passes?) */
    apiReturn = pNetWkstaTransportEnum(NULL, 0, NULL, MAX_PREFERRED_LENGTH,
        NULL, &totalEntries, NULL);

    /* if no network adapter present, bail, the rest of the test will fail */
    if (apiReturn == ERROR_NETWORK_UNREACHABLE)
        return;

152
    ok(apiReturn == STATUS_ACCESS_VIOLATION || apiReturn == ERROR_INVALID_PARAMETER,
153
       "NetWkstaTransportEnum returned %d\n", apiReturn);
154 155 156 157

    /* 3rd check: is param 3 passed? */
    apiReturn = pNetWkstaTransportEnum(NULL, 0, NULL, MAX_PREFERRED_LENGTH,
        NULL, NULL, NULL);
Juan Lang's avatar
Juan Lang committed
158
    ok(apiReturn == STATUS_ACCESS_VIOLATION || apiReturn == RPC_X_NULL_REF_POINTER || apiReturn == ERROR_INVALID_PARAMETER,
159
       "NetWkstaTransportEnum returned %d\n", apiReturn);
160 161 162 163

    /* 4th check: is param 6 passed? */
    apiReturn = pNetWkstaTransportEnum(NULL, 0, &bufPtr, MAX_PREFERRED_LENGTH,
        &entriesRead, NULL, NULL);
164
    ok(apiReturn == RPC_X_NULL_REF_POINTER, "null pointer\n");
165 166 167 168

    /* final check: valid return, actually get data back */
    apiReturn = pNetWkstaTransportEnum(NULL, 0, &bufPtr, MAX_PREFERRED_LENGTH,
        &entriesRead, &totalEntries, NULL);
169
    ok(apiReturn == NERR_Success || apiReturn == ERROR_NETWORK_UNREACHABLE,
170
       "NetWkstaTransportEnum returned %d\n", apiReturn);
171 172 173
    if (apiReturn == NERR_Success) {
        /* WKSTA_TRANSPORT_INFO_0 *transports = (WKSTA_TRANSPORT_INFO_0 *)bufPtr; */

174 175 176
        ok(bufPtr != NULL, "got data back\n");
        ok(entriesRead > 0, "read at least one transport\n");
        ok(totalEntries > 0, "at least one transport\n");
177
        pNetApiBufferFree(bufPtr);
178 179 180
    }
}

181 182
START_TEST(wksta)
{
183
    HMODULE hnetapi32=LoadLibraryA("netapi32.dll");
184

185 186 187 188
    pNetApiBufferFree=(void*)GetProcAddress(hnetapi32,"NetApiBufferFree");
    pNetApiBufferSize=(void*)GetProcAddress(hnetapi32,"NetApiBufferSize");
    pNetpGetComputerName=(void*)GetProcAddress(hnetapi32,"NetpGetComputerName");
    pNetWkstaUserGetInfo=(void*)GetProcAddress(hnetapi32,"NetWkstaUserGetInfo");
189
    pNetWkstaTransportEnum=(void*)GetProcAddress(hnetapi32,"NetWkstaTransportEnum");
190 191 192 193 194 195 196 197 198

    /* These functions were introduced with NT. It's safe to assume that
     * if one is not available, none are.
     */
    if (!pNetApiBufferFree) {
        skip("Needed functions are not available\n");
        FreeLibrary(hnetapi32);
        return;
    }
199 200 201 202

    if (init_wksta_tests()) {
        run_get_comp_name_tests();
        run_wkstausergetinfo_tests();
203
        run_wkstatransportenum_tests();
204
    }
205 206

    FreeLibrary(hnetapi32);
207
}