static
DWORD
TestNetlogonEnumDomainTrusts(
PTEST pTest,
PCWSTR pwszHostname,
PCWSTR pwszBindingString,
PCREDENTIALS pCreds,
PPARAMETER pOptions,
DWORD dwOptcount
)
{
const DWORD dwDefTrustFlags = NETR_TRUST_FLAG_IN_FOREST |
NETR_TRUST_FLAG_OUTBOUND |
NETR_TRUST_FLAG_TREEROOT |
NETR_TRUST_FLAG_PRIMARY |
NETR_TRUST_FLAG_NATIVE |
NETR_TRUST_FLAG_INBOUND;
BOOLEAN bRet = TRUE;
NTSTATUS ntStatus = STATUS_SUCCESS;
WINERROR err = ERROR_SUCCESS;
NETR_BINDING hNetr = NULL;
enum param_err perr = perr_success;
DWORD dwTrustFlags = 0;
DWORD dwCount = 0;
NetrDomainTrust *pTrusts = NULL;
DWORD iTrust = 0;
TESTINFO(pTest, pwszHostname);
perr = fetch_value(pOptions, dwOptcount, "trustflags", pt_uint32, &dwTrustFlags,
&dwDefTrustFlags);
if (!perr_is_ok(perr)) perr_fail(perr);
PARAM_INFO("trustflags", pt_uint32, &dwTrustFlags);
bRet &= CreateRpcBinding(OUT_PPVOID(&hNetr),
RPC_NETLOGON_BINDING,
pwszHostname,
pwszBindingString,
pCreds);
CALL_NETAPI(err, DsrEnumerateDomainTrusts(hNetr, pwszHostname, dwTrustFlags,
&pTrusts, &dwCount));
BAIL_ON_WIN_ERROR(err);
bRet &= TestValidateDomainTrusts(pTrusts, dwCount);
for (iTrust = 0; iTrust < dwCount; iTrust++)
{
OUTPUT_ARG_WSTR(pTrusts[iTrust].netbios_name);
OUTPUT_ARG_WSTR(pTrusts[iTrust].dns_name);
OUTPUT_ARG_UINT(pTrusts[iTrust].trust_flags);
}
error:
if (pTrusts)
{
NetrFreeMemory(pTrusts);
}
NetrFreeBinding(&hNetr);
if (ntStatus != STATUS_SUCCESS ||
err != ERROR_SUCCESS)
{
bRet = FALSE;
}
return bRet;
}
static
DWORD
TestDsrRoleGetPrimaryDomainInformation(
PTEST pTest,
PCWSTR pwszHostname,
PCWSTR pwszBindingString,
PCREDENTIALS pCreds,
PPARAMETER pOptions,
DWORD dwOptcount
)
{
const DWORD dwDefLevel = (WORD)(-1);
BOOLEAN bRet = TRUE;
WINERROR err = ERROR_SUCCESS;
enum param_err perr = perr_success;
DSR_BINDING hDsr = NULL;
DWORD dwSelectedLevels[] = {0};
DWORD dwAvailableLevels[] = { DS_ROLE_BASIC_INFORMATION,
DS_ROLE_UPGRADE_STATUS,
DS_ROLE_OP_STATUS };
PDWORD pdwLevels = NULL;
DWORD dwNumLevels = 0;
DWORD dwLevel = 0;
DWORD i = 0;
PDSR_ROLE_INFO pInfo = NULL;
perr = fetch_value(pOptions, dwOptcount, "level", pt_uint32, &dwLevel,
&dwDefLevel);
if (!perr_is_ok(perr)) perr_fail(perr);
PARAM_INFO("level", pt_uint32, &dwLevel);
TESTINFO(pTest, pwszHostname);
if (dwLevel == (WORD)(-1))
{
pdwLevels = dwAvailableLevels;
dwNumLevels = sizeof(dwAvailableLevels)/sizeof(dwAvailableLevels[0]);
}
else
{
dwSelectedLevels[0] = dwLevel;
pdwLevels = dwSelectedLevels;
dwNumLevels = sizeof(dwSelectedLevels)/sizeof(dwSelectedLevels[0]);
}
bRet &= CreateRpcBinding(OUT_PPVOID(&hDsr),
RPC_DSSETUP_BINDING,
pwszHostname,
pwszBindingString,
pCreds);
if (!bRet)
{
goto error;
}
for (i = 0; i < dwNumLevels; i++)
{
dwLevel = pdwLevels[i];
INPUT_ARG_UINT(dwLevel);
CALL_NETAPI(err, DsrRoleGetPrimaryDomainInformation(
hDsr,
dwLevel,
&pInfo));
BAIL_ON_WIN_ERROR(err);
switch (dwLevel)
{
case DS_ROLE_BASIC_INFORMATION:
ASSERT_TEST((pInfo->Basic.dwRole >= DS_ROLE_STANDALONE_WORKSTATION &&
pInfo->Basic.dwRole <= DS_ROLE_PRIMARY_DC));
ASSERT_TEST(pInfo->Basic.pwszDomain != NULL);
ASSERT_TEST(pInfo->Basic.pwszDnsDomain != NULL);
ASSERT_TEST(pInfo->Basic.pwszForest != NULL);
break;
case DS_ROLE_UPGRADE_STATUS:
ASSERT_TEST((pInfo->Upgrade.swUpgradeStatus == DS_ROLE_NOT_UPGRADING ||
pInfo->Upgrade.swUpgradeStatus == DS_ROLE_UPGRADING));
ASSERT_TEST((pInfo->Upgrade.dwPrevious >= DS_ROLE_PREVIOUS_UNKNOWN &&
pInfo->Upgrade.dwPrevious <= DS_ROLE_PREVIOUS_BACKUP));
break;
case DS_ROLE_OP_STATUS:
ASSERT_TEST(pInfo->OpStatus.swStatus <= DS_ROLE_NEEDS_REBOOT);
break;
}
if (pInfo)
{
DsrFreeMemory(pInfo);
pInfo = NULL;
}
}
DsrFreeBinding(&hDsr);
error:
if (pInfo)
{
DsrFreeMemory(pInfo);
}
if (err != ERROR_SUCCESS)
{
bRet = FALSE;
}
return bRet;
}
static
DWORD
TestNetlogonGetDcName(
PTEST pTest,
PCWSTR pwszHostname,
PCWSTR pwszBindingString,
PCREDENTIALS pCreds,
PPARAMETER pOptions,
DWORD dwOptcount
)
{
const DWORD dwDefGetDcFlags = DS_FORCE_REDISCOVERY;
const PSTR pszDefDomainName = "DOMAIN";
BOOLEAN bRet = TRUE;
NTSTATUS ntStatus = STATUS_SUCCESS;
WINERROR err = ERROR_SUCCESS;
NETR_BINDING hNetr = NULL;
enum param_err perr = perr_success;
DWORD dwGetDcFlags = 0;
PWSTR pwszDomainName = NULL;
DsrDcNameInfo *pInfo = NULL;
TESTINFO(pTest, pwszHostname);
perr = fetch_value(pOptions, dwOptcount, "getdcflags", pt_uint32, &dwGetDcFlags,
&dwDefGetDcFlags);
if (!perr_is_ok(perr)) perr_fail(perr);
perr = fetch_value(pOptions, dwOptcount, "domainname", pt_w16string, &pwszDomainName,
&pszDefDomainName);
if (!perr_is_ok(perr)) perr_fail(perr);
PARAM_INFO("getdcflags", pt_uint32, &dwGetDcFlags);
PARAM_INFO("domainname", pt_w16string, pwszDomainName);
bRet &= CreateRpcBinding(OUT_PPVOID(&hNetr),
RPC_NETLOGON_BINDING,
pwszHostname,
pwszBindingString,
pCreds);
CALL_NETAPI(err, DsrGetDcName(hNetr, pwszHostname, pwszDomainName,
NULL, NULL, dwGetDcFlags, &pInfo));
BAIL_ON_WIN_ERROR(err);
error:
if (pInfo)
{
NetrFreeMemory(pInfo);
}
NetrFreeBinding(&hNetr);
if (ntStatus != STATUS_SUCCESS ||
err != ERROR_SUCCESS)
{
bRet = FALSE;
}
return bRet;
}
