windows-nt/Source/XPSP1/NT/admin/cys/win32/netbiospage.cpp
2020-09-26 16:20:57 +08:00

371 lines
7.6 KiB
C++

// Copyright (c) 1997-2001 Microsoft Corporation
//
// File: NetbiosPage.cpp
//
// Synopsis: Defines the new netbios name page used in the
// Express path for the CYS Wizard
//
// History: 02/08/2001 JeffJon Created
#include "pch.h"
#include "resource.h"
#include "InstallationUnitProvider.h"
#include "NetbiosPage.h"
#include "state.h"
static PCWSTR NETBIOS_PAGE_HELP = L"cys.chm::/cys_configuring_first_server.htm";
NetbiosDomainPage::NetbiosDomainPage()
:
CYSWizardPage(
IDD_NETBIOS_NAME,
IDS_NETBIOS_NAME_TITLE,
IDS_NETBIOS_NAME_SUBTITLE,
NETBIOS_PAGE_HELP)
{
LOG_CTOR(NetbiosDomainPage);
}
NetbiosDomainPage::~NetbiosDomainPage()
{
LOG_DTOR(NetbiosDomainPage);
}
void
NetbiosDomainPage::OnInit()
{
LOG_FUNCTION(NetbiosDomainPage::OnInit);
Win::Edit_LimitText(
Win::GetDlgItem(hwnd, IDC_NETBIOS),
MAX_NETBIOS_NAME_LENGTH);
}
static
void
enable(HWND dialog)
{
ASSERT(Win::IsWindow(dialog));
int next =
!Win::GetTrimmedDlgItemText(dialog, IDC_NETBIOS).empty()
? PSWIZB_NEXT : 0;
Win::PropSheet_SetWizButtons(
Win::GetParent(dialog),
PSWIZB_BACK | next);
}
HRESULT
MyDsRoleDnsNameToFlatName(
const String& domainDNSName,
String& result,
bool& nameWasTweaked)
{
LOG_FUNCTION(MyDsRoleDnsNameToFlatName);
ASSERT(!domainDNSName.empty());
nameWasTweaked = false;
result.erase();
LOG(L"Calling DsRoleDnsNameToFlatName");
LOG( L"lpServer : (null)");
LOG(String::format(L"lpDnsName : %1", domainDNSName.c_str()));
PWSTR flatName = 0;
ULONG flags = 0;
HRESULT hr =
Win32ToHresult(
::DsRoleDnsNameToFlatName(
0, // this server
domainDNSName.c_str(),
&flatName,
&flags));
LOG_HRESULT(hr);
if (SUCCEEDED(hr) && flatName)
{
LOG(String::format(L"lpFlatName : %1", flatName));
LOG(String::format(L"lpStatusFlag : %1!X!", flags));
result = flatName;
if (result.length() > DNLEN)
{
result.resize(DNLEN);
}
::DsRoleFreeMemory(flatName);
// the name was tweaked if it is not the default. 338443
nameWasTweaked = !(flags & DSROLE_FLATNAME_DEFAULT);
}
return hr;
}
// return true if the name generated has already been validated, false
// if not.
bool
GenerateDefaultNetbiosName(HWND parent)
{
LOG_FUNCTION(GenerateDefaultNetbiosName);
ASSERT(Win::IsWindow(parent));
Win::CursorSetting cursor(IDC_WAIT);
bool result = false;
String dnsDomainName =
InstallationUnitProvider::GetInstance().GetADInstallationUnit().GetNewDomainDNSName();
bool nameWasTweaked = false;
String generatedName;
HRESULT hr =
MyDsRoleDnsNameToFlatName(
dnsDomainName,
generatedName,
nameWasTweaked);
if (FAILED(hr))
{
// if the api call failed, the name could not have been validated.
result = false;
// fall back to just the first 15 characters of the first label
generatedName =
dnsDomainName.substr(0, min(DNLEN, dnsDomainName.find(L'.')));
LOG(String::format(L"falling back to %1", generatedName.c_str()));
}
else
{
// the api validated the name for us.
result = true;
}
generatedName.to_upper();
if (generatedName.is_numeric())
{
// the generated name is all-numeric. This is not allowed. So we
// toss it out. 368777 bis
generatedName.erase();
nameWasTweaked = false;
}
Win::SetDlgItemText(
parent,
IDC_NETBIOS,
generatedName);
// inform the user that the default NetBIOS name was adjusted due
// to name collision on the network
if (nameWasTweaked)
{
popup.Info(
parent,
String::format(
IDS_GENERATED_NAME_WAS_TWEAKED,
generatedName.c_str()));
}
return result;
}
bool
NetbiosDomainPage::OnSetActive()
{
LOG_FUNCTION(NetbiosDomainPage::OnSetActive);
Win::PropSheet_SetWizButtons(
Win::GetParent(hwnd),
PSWIZB_BACK);
// do this here instead of in init to regenerate a default name if the
// user has not annointed one already.
if (InstallationUnitProvider::GetInstance().GetADInstallationUnit().GetNewDomainNetbiosName().empty())
{
GenerateDefaultNetbiosName(hwnd);
}
enable(hwnd);
return true;
}
bool
NetbiosDomainPage::OnCommand(
HWND /* windowFrom */ ,
unsigned controlIDFrom,
unsigned code)
{
// LOG_FUNCTION(NetbiosDomainPage::OnCommand);
switch (controlIDFrom)
{
case IDC_NETBIOS:
{
if (code == EN_CHANGE)
{
SetChanged(controlIDFrom);
enable(hwnd);
}
break;
}
default:
{
// do nothing
break;
}
}
return false;
}
bool
ValidateDomainNetbiosName(HWND dialog, int editResID)
{
LOG_FUNCTION(ValidateDomainNetbiosName);
ASSERT(Win::IsWindow(dialog));
ASSERT(editResID > 0);
Win::CursorSetting cursor(IDC_WAIT);
String name = Win::GetTrimmedDlgItemText(dialog, editResID);
if (name.empty())
{
return false;
}
if (name.find(L".") != String::npos)
{
popup.Gripe(
dialog,
editResID,
IDS_NO_DOTS_IN_NETBIOS_NAME);
return false;
}
// Check that the name is not a number. 368777
if (name.is_numeric())
{
popup.Gripe(
dialog,
editResID,
String::format(IDS_NUMERIC_NETBIOS_NAME, name.c_str()));
return false;
}
// we pretend that the candidate name is a hostname, and attempt to
// generate a netbios name from it. If that can't be done, then the
// candidate name can't be a legal netbios name.
HRESULT hr = S_OK;
String s = Dns::HostnameToNetbiosName(name, &hr);
if (FAILED(hr))
{
popup.Gripe(
dialog,
editResID,
hr,
String::format(IDS_BAD_NETBIOS_NAME, name.c_str()));
return false;
}
if (s.length() < name.length())
{
// the name was truncated.
popup.Gripe(
dialog,
editResID,
String::format(
IDS_NETBIOS_NAME_TOO_LONG,
name.c_str(),
MAX_NETBIOS_NAME_LENGTH));
return false;
}
if (ValidateNetbiosDomainName(s) != VALID_NAME)
{
popup.Gripe(
dialog,
editResID,
String::format(
IDS_BAD_NETBIOS_CHARACTERS,
s.c_str()));
return false;
}
hr = MyNetValidateName(name, ::NetSetupNonExistentDomain);
if (hr == Win32ToHresult(ERROR_DUP_NAME))
{
popup.Gripe(
dialog,
editResID,
String::format(IDS_FLATNAME_IN_USE, name.c_str()));
return false;
}
if (hr == Win32ToHresult(ERROR_NETWORK_UNREACHABLE))
{
// 25968
if (
popup.MessageBox(
dialog,
String::format(
IDS_NET_NOT_REACHABLE,
name.c_str()),
MB_YESNO | MB_ICONWARNING) != IDYES)
{
HWND edit = Win::GetDlgItem(dialog, editResID);
Win::SendMessage(edit, EM_SETSEL, 0, -1);
Win::SetFocus(edit);
return false;
}
}
return true;
}
int
NetbiosDomainPage::Validate()
{
LOG_FUNCTION(NetbiosDomainPage::Validate);
int nextPage = -1;
if (!ValidateDomainNetbiosName(hwnd, IDC_NETBIOS))
{
nextPage = -1;
}
else
{
String netbiosName = Win::GetTrimmedDlgItemText(hwnd, IDC_NETBIOS);
InstallationUnitProvider::GetInstance().GetADInstallationUnit().SetNewDomainNetbiosName(netbiosName);
nextPage = IDD_RESTORE_PASSWORD_PAGE;
}
return nextPage;
}