windows-nt/Source/XPSP1/NT/termsrv/admtools/chglogon/chglogon.c
2020-09-26 16:20:57 +08:00

194 lines
4.5 KiB
C

// Copyright (c) 1998-1999 Microsoft Corporation
/******************************************************************************
*
* CHGLOGON.C
*
* This module contains code for the CHGLOGON utility.
*
*
*******************************************************************************/
#include <nt.h>
#include <ntrtl.h>
#include <nturtl.h>
#include <windows.h>
#include <winstaw.h>
#include <assert.h>
#include <stdio.h>
#include <stdlib.h>
#include <utilsub.h>
#include <string.h>
#include <locale.h>
#include <printfoa.h>
#include "chglogon.h"
/*
* Global Data
*/
USHORT help_flag = FALSE; // User wants help
USHORT fQuery = FALSE; // query winstations
USHORT fEnable = FALSE; // enable winstations
USHORT fDisable = FALSE; // disable winstations
TOKMAP ptm[] = {
{L"/q", TMFLAG_OPTIONAL, TMFORM_BOOLEAN, sizeof(USHORT), &fQuery},
{L"/query", TMFLAG_OPTIONAL, TMFORM_BOOLEAN, sizeof(USHORT), &fQuery},
{L"/enable", TMFLAG_OPTIONAL, TMFORM_BOOLEAN, sizeof(USHORT), &fEnable},
{L"/disable", TMFLAG_OPTIONAL, TMFORM_BOOLEAN, sizeof(USHORT), &fDisable},
{L"/?", TMFLAG_OPTIONAL, TMFORM_BOOLEAN, sizeof(USHORT), &help_flag},
{0, 0, 0, 0, 0}
};
/*
* Local function prototypes.
*/
void Usage(BOOLEAN bError);
/*******************************************************************************
*
* main
*
******************************************************************************/
int __cdecl
main(INT argc, CHAR **argv)
{
WCHAR **argvW;
ULONG rc;
INT i;
PPOLICY_TS_MACHINE Ppolicy;
setlocale(LC_ALL, ".OCP");
/*
* Massage the command line.
*/
argvW = MassageCommandLine((DWORD)argc);
if (argvW == NULL) {
ErrorPrintf(IDS_ERROR_MALLOC);
return(FAILURE);
}
/*
* parse the cmd line without parsing the program name (argc-1, argv+1)
*/
rc = ParseCommandLine(argc-1, argvW+1, ptm, 0);
/*
* Check for error from ParseCommandLine
*/
if ( help_flag || rc ) {
if ( !help_flag && !(rc & PARSE_FLAG_NO_PARMS) ) {
Usage(TRUE);
return(FAILURE);
} else {
Usage(FALSE);
return(SUCCESS);
}
}
//Check if we are running under Terminal Server
if(!AreWeRunningTerminalServices())
{
ErrorPrintf(IDS_ERROR_NOT_TS);
return(FAILURE);
}
/*
* Check if Group policy has thrown the big switch, if so, inform and refuse any changes
*/
Ppolicy = LocalAlloc( LPTR, sizeof(POLICY_TS_MACHINE) );
if (Ppolicy == NULL) {
ErrorPrintf(IDS_ERROR_MALLOC);
return(FAILURE);
}
RegGetMachinePolicy( Ppolicy );
if ( Ppolicy->fPolicyDenyTSConnections )
{
if (Ppolicy->fDenyTSConnections)
{
ErrorPrintf(IDS_ERROR_WINSTATIONS_GP_DENY_CONNECTIONS_1 );
}
else
{
ErrorPrintf(IDS_ERROR_WINSTATIONS_GP_DENY_CONNECTIONS_0 );
}
LocalFree( Ppolicy );
Ppolicy = NULL;
return( FAILURE );
}
if (Ppolicy != NULL) {
LocalFree( Ppolicy );
Ppolicy = NULL;
}
/*
* Enable or disable
*/
if ( fDisable ) {
rc = WriteProfileString( APPLICATION_NAME, WINSTATIONS_DISABLED, TEXT("1") );
}
else if ( fEnable ) {
rc = WriteProfileString( APPLICATION_NAME, WINSTATIONS_DISABLED, TEXT("0") );
}
/*
* Query or error ?
*/
if ( !fQuery && (rc != 1) ) {
ErrorPrintf(IDS_ACCESS_DENIED);
}
else if ( GetProfileInt( APPLICATION_NAME, WINSTATIONS_DISABLED, 0 ) == 0 ) {
ErrorPrintf(IDS_WINSTATIONS_ENABLED);
}
else {
ErrorPrintf(IDS_WINSTATIONS_DISABLED);
}
return(SUCCESS);
}
/*******************************************************************************
*
* Usage
*
* Output the usage message for this utility.
*
* ENTRY:
* bError (input)
* TRUE if the 'invalid parameter(s)' message should preceed the usage
* message and the output go to stderr; FALSE for no such error
* string and output goes to stdout.
*
* EXIT:
*
*
******************************************************************************/
void
Usage( BOOLEAN bError )
{
if ( bError ) {
ErrorPrintf(IDS_ERROR_INVALID_PARAMETERS);
}
ErrorPrintf(IDS_HELP_USAGE1);
ErrorPrintf(IDS_HELP_USAGE2);
ErrorPrintf(IDS_HELP_USAGE3);
ErrorPrintf(IDS_HELP_USAGE4);
ErrorPrintf(IDS_HELP_USAGE5);
} /* Usage() */