windows-nt/Source/XPSP1/NT/ds/security/tools/joindom/joindom.c
2020-09-26 16:20:57 +08:00

554 lines
17 KiB
C
Raw Permalink Blame History

This file contains invisible Unicode characters

This file contains invisible Unicode characters that are indistinguishable to humans but may be processed differently by a computer. If you think that this is intentional, you can safely ignore this warning. Use the Escape button to reveal them.

/*++
Copyright (c) 1997 - 1997 Microsoft Corporation
Module Name:
joindom.c
Abstract:
Unit test for NetJoinDomain and NetUnjoinDomain APIs
Author:
Mac McLain (MacM) 19-Feb-1997
Environment:
User mode only.
Revision History:
--*/
#include <nt.h>
#include <ntrtl.h>
#include <nturtl.h>
#include <windows.h>
#include <lmcons.h>
#include <lmerr.h>
#include <lmjoin.h>
#include <lmapibuf.h>
#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#define JOIN_TAG "join"
#define UNJOIN_TAG "unjoin"
#define NAME_TAG "name"
#define RENAME_TAG "rename"
#define INFO_TAG "info"
#define OU_TAG "ous"
#define ADD_ALT_NAME "addaltname"
#define DEL_ALT_NAME "delaltname"
#define SET_PRI_NAME "setpriname"
#define GET_PRI_NAME "getpriname"
#define GET_ALT_NAMES "getaltnames"
#define GET_ALL_NAMES "getallnames"
NET_API_STATUS
NET_API_FUNCTION
NetpGetListOfJoinableOUs(
IN LPWSTR Domain,
IN LPWSTR Account,
IN LPWSTR Password,
OUT PULONG Count,
OUT PWSTR **OUs
);
NET_API_STATUS
NET_API_FUNCTION
NetpCreateComputerObjectInOU(
IN LPWSTR DC,
IN LPWSTR OU,
IN LPWSTR ComputerName,
IN LPWSTR Account,
IN LPWSTR Password,
IN LPWSTR MachinePassword
);
void
Usage(
VOID
)
/*++
Routine Description:
Displays the expected usage
Arguments:
None
Return Value:
VOID
--*/
{
printf("joindom -operation options domain/server [account] [password] [new/alt name] [OU]\n");
printf(" where:\n");
printf(" operation indicates what joindom should do:\n");
printf(" -%s Joins a workstation to a domain/workgroup\n", JOIN_TAG);
printf(" -%s Unjoins a workstation from a domain\n", UNJOIN_TAG);
printf(" -%s Renames the machine to a new name\n", RENAME_TAG);
printf(" -%s Validates the give name for validity\n", NAME_TAG);
printf(" -%s Returns information about the state of the join\n", INFO_TAG );
printf(" -%s Returns information about the joinable ous\n", OU_TAG );
printf(" -%s Add alternative computer name\n", ADD_ALT_NAME );
printf(" -%s Delete alternative computer name\n", DEL_ALT_NAME );
printf(" -%s Set primary computer name\n", SET_PRI_NAME );
printf(" -%s Get primary computer name\n", GET_PRI_NAME );
printf(" -%s Get altername computer names\n", GET_ALT_NAMES );
printf(" -%s Get all computer names\n", GET_ALL_NAMES );
printf(" options specifies the flags for the command\n");
printf(" 0x000000001 Joins a domain instead of workgroup\n");
printf(" 0x000000002 Creates the account on the domain\n");
printf(" 0x000000004 Deletes the account when leaving a domain\n");
printf(" domain/server specifies the domain (or server for alt name APIs) path to operate on\n" );
printf(" [account] is the name of the account to use for accessing domain objects\n");
printf(" [password] is the password of the account to use while "
"accessing domain objects\n");
printf(" [newname] is the new computer name \n");
printf(" [OU] is the name of the ou to create the machine account in\n");
}
int __cdecl
main(
IN int argc,
IN char ** argv
)
/*++
Routine Description:
See main comment.
Arguments:
argc - the number of command-line arguments.
argv - an array of pointers to the arguments.
Return Value:
Exit status
--*/
{
NET_API_STATUS NetStatus = NERR_Success;
WCHAR wszDomain[MAX_PATH + 1];
WCHAR rgwszOpts[3][MAX_PATH + 1];
PWSTR ppwszOpts[3] = { NULL, NULL, NULL };
PWSTR ppwszJoinTypes[ ] = { L"NetSetupUnknownStatus",
L"NetSetupUnjoined",
L"NetSetupWorkgroupName",
L"NetSetupDomainName" };
WCHAR wszMachine[MAX_COMPUTERNAME_LENGTH + 1];
PWSTR pwszJoinInfo;
WCHAR wszInfo[256];
ULONG i, Options = 0, cLen, Count;
PWSTR *OUs;
PSTR pszNext;
NETSETUP_JOIN_STATUS JoinType;
BOOLEAN EnumerateComputerName = FALSE;
NET_COMPUTER_NAME_TYPE ComputerNameType;
LPSTR ComputerNameTypeString = NULL;
BOOLEAN PrintUsage = FALSE;
if (argc < 4) {
PrintUsage = TRUE;
goto Cleanup;
} else {
cLen = MAX_COMPUTERNAME_LENGTH + 1;
if ( GetComputerNameW( wszMachine, &cLen ) == FALSE ) {
NetStatus = GetLastError();
printf("Failed to get the local computer name: %lu\n", NetStatus );
}
}
if ( NetStatus == NERR_Success) {
//
// Process the command line
//
Options = strtoul( argv[2], &pszNext, 0 );
mbstowcs( wszDomain, argv[3], strlen(argv[3]) + 1 );
for ( i = 4 ; i < (ULONG)argc && i < 7 ; i++ ) {
mbstowcs(rgwszOpts[i - 4], argv[i], strlen( argv[i] ) + 1 );
ppwszOpts[i - 4] = rgwszOpts[i - 4];
}
}
//
// Now, do the operations
//
if ( NetStatus == NERR_Success && (*argv[1] == '/' || *argv[1] == '-' ) ) {
if ( _stricmp( argv[1] + 1, JOIN_TAG ) == 0 ) {
//
// Do the call
//
NetStatus = NetJoinDomain( wszMachine, wszDomain, ppwszOpts[2],
ppwszOpts[0], ppwszOpts[1],
Options );
if ( NetStatus != NERR_Success ) {
printf( "NetJoinDomain [%ws] [%ws] [%ws] [0x%lx] failed with %lu\n",
wszDomain,
ppwszOpts[0] == NULL ? L"NULL" : ppwszOpts[0],
ppwszOpts[1] == NULL ? L"NULL" : ppwszOpts[1],
Options,
NetStatus );
}
} else if ( _stricmp( argv[1] + 1, UNJOIN_TAG ) == 0 ) {
NetStatus = NetUnjoinDomain( NULL, ppwszOpts[0], ppwszOpts[1], Options );
if ( NetStatus != NERR_Success ) {
printf( "NetUnjoinDomain [%ws] [%ws] [0x%lx] failed with %lu\n",
ppwszOpts[0] == NULL ? L"NULL" : ppwszOpts[0],
ppwszOpts[1] == NULL ? L"NULL" : ppwszOpts[1],
Options,
NetStatus );
}
} else if ( _stricmp( argv[1] + 1, RENAME_TAG ) == 0 ) {
//
// First, set the computer name
//
if ( ppwszOpts[2] == NULL ) {
PrintUsage = TRUE;
goto Cleanup;
} else {
if( SetComputerName( ppwszOpts[2] ) == FALSE ) {
NetStatus = GetLastError();
} else {
NetStatus = NetRenameMachineInDomain( NULL, ppwszOpts[ 2 ],
ppwszOpts[0], ppwszOpts[1],
Options );
if ( NetStatus != NERR_Success ) {
printf( "NetRenameMachineInDomain [%ws] [%ws] [%ws] [%ws] [0x%lx] failed with %lu\n",
ppwszOpts[2],
wszDomain,
ppwszOpts[0] == NULL ? L"NULL" : ppwszOpts[0],
ppwszOpts[1] == NULL ? L"NULL" : ppwszOpts[1],
Options,
NetStatus );
}
}
}
} else if ( _stricmp( argv[1] + 1, NAME_TAG ) == 0 ) {
//
// Validate name as all four types
//
if ( NetStatus == NERR_Success ) {
NetStatus = NetValidateName( wszMachine, wszDomain, ppwszOpts[0], ppwszOpts[1],
NetSetupMachine );
if( NetStatus == NERR_Success ) {
printf( "Name %ws valid as machine name\n", wszDomain );
} else {
printf( "Name %ws invalid as machine name: %lu\n",
wszDomain, NetStatus );
}
//
// workgroup name
//
NetStatus = NetValidateName( wszMachine, wszDomain, ppwszOpts[0], ppwszOpts[1],
NetSetupWorkgroup );
if( NetStatus == NERR_Success ) {
printf( "Name %ws valid as workgroup name\n", wszDomain );
} else {
printf( "Name %ws invalid as workgroup name: %lu\n",
wszDomain, NetStatus );
}
//
// Domain name
//
NetStatus = NetValidateName( wszMachine, wszDomain, ppwszOpts[0], ppwszOpts[1],
NetSetupDomain );
if( NetStatus == NERR_Success ) {
printf( "Name %ws valid as domain name\n", wszDomain );
} else {
printf( "Name %ws invalid as domain name: %lu\n",
wszDomain, NetStatus );
}
//
// Domain name
//
NetStatus = NetValidateName( wszMachine, wszDomain, ppwszOpts[0], ppwszOpts[1],
NetSetupNonExistentDomain );
if( NetStatus == NERR_Success ) {
printf( "Name %ws valid as non existent domain name\n", wszDomain );
} else {
printf( "Name %ws invalid as nonexistent domain name: %lu\n",
wszDomain, NetStatus );
}
}
} else if ( _stricmp( argv[1] + 1, INFO_TAG ) == 0 ) {
//
// Call it 3 times, 1: NULL buffer, 2: valid ptr, 3: remotely
//
#if 0
NetStatus = NetGetJoinInformation( NULL, NULL, &JoinType );
if ( NetStatus == ERROR_INVALID_PARAMETER ) {
NetStatus = NERR_Success;
printf( "JoinInfo with NULL buffer returned correctly\n" );
} else {
printf( "JoinInfo with NULL buffer incorrectly returned %lu\n", NetStatus );
NetStatus = ERROR_INVALID_DATA;
}
#endif
NetStatus = NetGetJoinInformation( NULL, &pwszJoinInfo, &JoinType );
if ( NetStatus == NERR_Success ) {
printf( "Join type: %ws\n", ppwszJoinTypes[ JoinType ] );
printf( "Joined to %ws\n", pwszJoinInfo );
NetApiBufferFree( pwszJoinInfo );
} else {
printf(" NetGetJoinInformation, failed with %lu\n", NetStatus );
}
#if 0
NetStatus = NetGetJoinInformation( wszMachine, &pwszJoinInfo, &JoinType );
if ( NetStatus == NERR_Success ) {
printf( "Joined to %ws\n", pwszJoinInfo );
NetApiBufferFree( pwszJoinInfo );
} else {
printf(" NetGetJoinInformation, remotely, failed with %lu\n", NetStatus );
}
#endif
} else if ( _stricmp( argv[1] + 1, OU_TAG ) == 0 ) {
NetStatus = NetGetJoinableOUs( NULL, ( LPCWSTR )wszDomain, ( LPCWSTR )ppwszOpts[0],
( LPCWSTR )ppwszOpts[1], &Count, &OUs );
if ( NetStatus != NERR_Success ) {
printf( "NetGetJoinableOUs returned %lu\n", NetStatus );
} else {
printf( "NetGetJoinableOUs returned %lu ous\n", Count );
for ( i = 0; i < Count; i++ ) {
printf(" %ws\n", OUs[ i ] );
}
NetApiBufferFree( OUs );
}
} else if ( _stricmp( argv[1] + 1, ADD_ALT_NAME ) == 0 ) {
printf( "\nAdding alternate computer name with following parameters:\n" );
printf( " Server: %ws\n", wszDomain );
printf( " Alternate name: %ws\n", ppwszOpts[2] );
printf( " Options: 0x%lx\n", Options );
printf( " Account: %ws\n", ppwszOpts[0] );
printf( " Password: %ws\n\n", ppwszOpts[1] );
NetStatus = NetAddAlternateComputerName(
wszDomain, // treated as server name
ppwszOpts[2], // alternate name
ppwszOpts[0], // domain account
ppwszOpts[1], // domain account pwd
Options );
if ( NetStatus != NERR_Success ) {
printf( "NetAddAlternateComputerName failed: 0x%lx\n", NetStatus );
}
} else if ( _stricmp( argv[1] + 1, DEL_ALT_NAME ) == 0 ) {
printf( "\nDeleting alternate computer name with following parameters:\n" );
printf( " Server: %ws\n", wszDomain );
printf( " Alternate name: %ws\n", ppwszOpts[2] );
printf( " Options: 0x%lx\n", Options );
printf( " Account: %ws\n", ppwszOpts[0] );
printf( " Password: %ws\n\n", ppwszOpts[1] );
NetStatus = NetRemoveAlternateComputerName(
wszDomain, // treated as server name
ppwszOpts[2], // alternate name
ppwszOpts[0], // domain account
ppwszOpts[1], // domain account pwd
Options );
if ( NetStatus != NERR_Success ) {
printf( "NetRemoveAlternateComputerName failed: 0x%lx\n", NetStatus );
}
} else if ( _stricmp( argv[1] + 1, SET_PRI_NAME ) == 0 ) {
printf( "\nSetting primary computer name with following parameters:\n" );
printf( " Server: %ws\n", wszDomain );
printf( " Primary name: %ws\n", ppwszOpts[2] );
printf( " Options: 0x%lx\n", Options );
printf( " Account: %ws\n", ppwszOpts[0] );
printf( " Password: %ws\n\n", ppwszOpts[1] );
NetStatus = NetSetPrimaryComputerName(
wszDomain, // treated as server name
ppwszOpts[2], // alternate name
ppwszOpts[0], // domain account
ppwszOpts[1], // domain account pwd
Options );
if ( NetStatus != NERR_Success ) {
printf( "NetSetPrimaryComputerName failed: 0x%lx\n", NetStatus );
}
} else if ( _stricmp( argv[1] + 1, GET_PRI_NAME ) == 0 ) {
EnumerateComputerName = TRUE;
ComputerNameType = NetPrimaryComputerName;
ComputerNameTypeString = "primary";
} else if ( _stricmp( argv[1] + 1, GET_ALT_NAMES ) == 0 ) {
EnumerateComputerName = TRUE;
ComputerNameType = NetAlternateComputerNames;
ComputerNameTypeString = "alternate";
} else if ( _stricmp( argv[1] + 1, GET_ALL_NAMES ) == 0 ) {
EnumerateComputerName = TRUE;
ComputerNameType = NetAllComputerNames;
ComputerNameTypeString = "all";
} else {
PrintUsage = TRUE;
goto Cleanup;
}
} else {
PrintUsage = TRUE;
goto Cleanup;
}
if ( EnumerateComputerName ) {
ULONG EntryCount = 0;
LPWSTR *ComputerNames = NULL;
printf( "\nGetting %s computer name(s) for server %ws with options 0x%lx:\n",
ComputerNameTypeString,
wszDomain, // treated as server name
Options );
NetStatus = NetEnumerateComputerNames(
wszDomain, // treated as server name
ComputerNameType, // name type
Options,
&EntryCount,
&ComputerNames );
if ( NetStatus != NERR_Success ) {
printf( "NetEnumerateComputerNames failed: 0x%lx\n", NetStatus );
} else {
ULONG i;
if ( EntryCount > 0 ) {
for ( i = 0; i < EntryCount; i++ ) {
printf( "%ws\n", ComputerNames[i] );
}
} else {
printf( "No names returned.\n" );
}
NetApiBufferFree( ComputerNames );
}
}
Cleanup:
if ( PrintUsage ) {
Usage();
}
if ( NetStatus == NERR_Success ) {
printf( "The command completed successfully\n" );
} else {
printf( "The command failed with error %lu\n", NetStatus );
}
return( NetStatus == NERR_Success );
}