/*++

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
    );

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 );
}