windows-nt/Source/XPSP1/NT/base/cluster/clusapi/tgin.c

427 lines
12 KiB
C
Raw Normal View History

2020-09-26 03:20:57 -05:00
/*++
Copyright (c) 1996 Microsoft Corporation
Module Name:
tcontrol.c
Abstract:
Test for cluster group controls
Author:
Rod Gamache (rodga) 30-Dec-1996
Revision History:
--*/
#include "windows.h"
#include "cluster.h"
#include "stdio.h"
#include "stdlib.h"
#include "resapi.h"
LPWSTR ClusterName=NULL;
LPWSTR GroupName=NULL;
LPWSTR NodeName=NULL;
DWORD ControlCode=0xffffffff;
DWORD Access=CLUS_ACCESS_READ;
#if 1
typedef struct _PROP_ITEM_16 {
DWORD NameType;
DWORD NameLength;
WCHAR PI[16];
} PROP_ITEM_16;
typedef struct _PROP_ITEM_18 {
DWORD NameType;
DWORD NameLength;
WCHAR PI[18];
} PROP_ITEM_18;
typedef struct _PROP_ITEM_20 {
DWORD NameType;
DWORD NameLength;
WCHAR PI[20];
} PROP_ITEM_20;
typedef struct _PROP_ITEM_ALIGN_15 {
DWORD NameType;
DWORD NameLength;
WCHAR PI[15];
WCHAR _align;
} PROP_ITEM_ALIGN_15;
typedef struct _PROP_ITEM_ALIGN_17 {
DWORD NameType;
DWORD NameLength;
WCHAR PI[17];
WCHAR _align;
} PROP_ITEM_ALIGN_17;
typedef struct _PROP_DWORD_VALUE {
DWORD ValueType;
DWORD ValueLength;
DWORD Value;
DWORD align;
} PROP_DWORD_VALUE;
typedef struct _PROP_LIST {
DWORD ItemCount;
PROP_ITEM_16 PropItem1; // PersistentState (16 chars)
PROP_DWORD_VALUE PropValue1;
PROP_ITEM_18 PropItem2; // FailoverThreshold (18 chars)
PROP_DWORD_VALUE PropValue2;
PROP_ITEM_ALIGN_15 PropItem3; // FailoverPeriod (15 chars)
PROP_DWORD_VALUE PropValue3;
PROP_ITEM_ALIGN_17 PropItem4; // AutoFailbackType (17 chars)
PROP_DWORD_VALUE PropValue4;
PROP_ITEM_20 PropItem5; // FailbackWindowStart (20 chars)
PROP_DWORD_VALUE PropValue5;
PROP_ITEM_18 PropItem6; // FailbackWindowEnd (18 chars)
PROP_DWORD_VALUE PropValue6;
DWORD EndValue2;
} PROP_LIST;
#pragma warning( default : 4200 )
#endif
#if 1
PROP_LIST PropList = {
0x00000006, // # of parameters
0x00040003,
0x00000020,
L"PersistentState",
0x00010002,
0x00000004,
0x00000001, // State is on
0x00000000,
0x00040003, // Name
0x00000024, // Name Length
L"FailoverThreshold",
0x00010002,
0x00000004,
0x0000000a, // Failover count is 10
0x00000000,
0x00040003, // Name
0x0000001e, // Name Length
L"FailoverPeriod", // Not a multiple of 4 bytes
0x0, // alignment needed
0x00010002,
0x00000004,
0x00000006,
0x00000000,
0x00040003,
0x00000022,
L"AutoFailbackType", // Not a multiple of 4 bytes
0x0, // alignment needed
0x00010002,
0x00000004,
0x00000000,
0x00000000,
0x00040003, // Name
0x00000028, // Name Length
L"FailbackWindowStart",
0x00010002,
0x00000004,
0x2,
0x00000000,
0x00040003, // Name
0x00000024, // Name Length
L"FailbackWindowEnd",
0x00010002,
0x00000004,
0x3,
0x00000000,
0x00000000 };
#else
DWORD PropList[] = {
0x00000006, 0x00040003, 0x00000020, 0x00650050,
0x00730072, 0x00730069, 0x00650074, 0x0074006e,
0x00740053, 0x00740061, 0x00000065, 0x00010002,
0x00000004, 0x00000001, 0x00000000, 0x00040003,
0x00000024, 0x00610046, 0x006c0069, 0x0076006f,
0x00720065, 0x00680054, 0x00650072, 0x00680073,
0x006c006f, 0x00000064, 0x00010002, 0x00000004,
0x0000000a, 0x00000000, 0x00040003, 0x0000001e,
0x00610046, 0x006c0069, 0x0076006f, 0x00720065,
0x00650050, 0x00690072, 0x0064006f, 0x00000000, // Alignment needed 4th dword
0x00010002, 0x00000004, 0x00000006, 0x00000000,
0x00040003, 0x00000022, 0x00750041, 0x006f0074,
0x00610046, 0x006c0069, 0x00610062, 0x006b0063,
0x00790054, 0x00650070, 0x00000000, 0x00010002, // Alignment needed 3rd dword
0x00000004, 0x00000000, 0x00000000, 0x00040003,
0x00000028, 0x00610046, 0x006c0069, 0x00610062,
0x006b0063, 0x00690057, 0x0064006e, 0x0077006f,
0x00740053, 0x00720061, 0x00000074, 0x00010002,
0x00000004, 0x2, 0x00000000, 0x00040003,
0x00000024, 0x00610046, 0x006c0069, 0x00610062,
0x006b0063, 0x00690057, 0x0064006e, 0x0077006f,
0x006e0045, 0x00000064, 0x00010002, 0x00000004,
0x3, 0x00000000, 0x00000000 };
#endif
CHAR UsageText[] =
"TCONTROL [-c cluster] -n node -g group -a access ControlCode\n"
" cluster\tspecifies the name of the cluster to connect to\n"
" node\tspecifies the node to direct the request to\n"
" group\tspecifies the name of the group to control\n"
" access\tspecifies the access to the group (read, write or any)\n"
" ControlCode\ta number between 1 and 99\n";
void
Usage(
void
)
{
fprintf(stderr, UsageText);
exit(1);
}
LPWSTR
GetString(
IN LPSTR String
)
{
LPWSTR wString;
DWORD Length;
Length = strlen(String)+1;
wString = malloc(Length*sizeof(WCHAR));
if (wString == NULL) {
fprintf(stderr, "GetString couldn't malloc %d bytes\n",Length*sizeof(WCHAR));
}
mbstowcs(wString, String, Length);
return(wString);
}
void
ParseArgs(
int argc,
char *argv[]
)
{
int i;
DWORD IntCount;
DWORD Value;
CHAR TestValue[16];
PUCHAR ControlData;
LPWSTR access;
for (i=1;i<argc;i++) {
if ((argv[i][0] == '-') ||
(argv[i][0] == '/')) {
switch (argv[i][1]) {
case 'c':
if (++i == argc) {
Usage();
}
ClusterName = GetString(argv[i]);
break;
case 'n':
if ( ++i == argc ) {
Usage();
}
NodeName = GetString(argv[i]);
break;
case 'g':
if ( ++i == argc ) {
Usage();
}
GroupName = GetString(argv[i]);
fprintf(stdout, "Group = %ws\n", GroupName);
break;
case 'a':
if ( ++i == argc ) {
Usage();
}
access = GetString(argv[i]);
if ( lstrcmpiW( access, L"read" ) ) {
Access = CLUS_ACCESS_READ;
} else if ( lstrcmpiW( access, L"write" ) ) {
Access = CLUS_ACCESS_WRITE;
} else if ( lstrcmpiW( access, L"any" ) ) {
Access = CLUS_ACCESS_ANY;
} else {
Usage();
}
break;
default:
Usage();
break;
}
} else {
ControlData = argv[i];
IntCount = sscanf( ControlData, "%d", &Value );
if ( IntCount == 1 ) {
sprintf( TestValue, "%d\0", Value );
if ( strcmp( TestValue, ControlData ) == 0 ) {
ControlCode = Value;
fprintf(stdout, "ControlCode = %d\n", ControlCode);
}
}
}
}
if ( ControlCode == 0xffffffff ) {
Usage();
}
}
_cdecl
main (argc, argv)
int argc;
char *argv[];
{
HCLUSTER hClus;
HGROUP hGroup;
HNODE hNode = NULL;
DWORD status;
DWORD ReturnSize;
DWORD bufSize;
CHAR InBuffer[64];
DWORD i,j;
LPDWORD Data;
LPDWORD PrintData;
CHAR PrintBuffer[32];
PUCHAR buffer;
DWORD controlCode;
BOOL retry = TRUE;
DWORD inputSize;
ParseArgs(argc, argv);
hClus = OpenCluster(ClusterName);
if (hClus == NULL) {
fprintf(stderr,
"OpenCluster %ws failed %d\n",
(ClusterName == NULL) ? L"(NULL)" : ClusterName,
GetLastError());
return(0);
}
hGroup = OpenClusterGroup( hClus, GroupName );
if ( hGroup == NULL ) {
fprintf(stderr,
"OpenGroup %ws failed %d\n", GroupName, GetLastError());
return(0);
}
if ( NodeName != NULL ) {
hNode = OpenClusterNode( hClus, NodeName );
if ( hNode == NULL ) {
fprintf(stderr,
"OpenNode %ws failed %d\n", NodeName, GetLastError());
return(0);
}
}
if ( Access == CLUS_ACCESS_WRITE ) {
controlCode = CLCTL_EXTERNAL_CODE( ControlCode, Access, CLUS_MODIFY );
} else {
controlCode = CLCTL_EXTERNAL_CODE( ControlCode, Access, CLUS_NO_MODIFY );
}
try_again:
controlCode = CLUSCTL_GROUP_CODE( controlCode );
ReturnSize = 0;
status = ClusterGroupControl( hGroup,
hNode,
controlCode,
NULL,
0,
NULL,
0,
&ReturnSize );
if ( retry &&
(status == ERROR_INVALID_FUNCTION) ) {
controlCode = CLCTL_EXTERNAL_CODE( ControlCode, CLUS_ACCESS_WRITE, CLUS_MODIFY );
retry = FALSE;
goto try_again;
}
fprintf(stdout, "Status of initial request is %d, size is %d.\n",
status, ReturnSize);
if ( (status != ERROR_SUCCESS) ||
(ReturnSize == 0) ) {
return(0);
}
bufSize = ReturnSize;
buffer = LocalAlloc( LMEM_FIXED, bufSize );
if ( buffer == NULL ) {
fprintf(stdout, "Failed to allocate a return buffer.\n");
return(0);
}
status = ClusterGroupControl( hGroup,
hNode,
controlCode,
NULL,
0,
buffer,
bufSize,
&ReturnSize );
fprintf(stdout, "Status of Control request = %d, size = %d\n",
status, ReturnSize);
if ( status == ERROR_SUCCESS ) {
inputSize = ReturnSize;
Data = (LPDWORD)buffer;
PrintData = (LPDWORD)PrintBuffer;
while ( ReturnSize ) {
j = ReturnSize;
if ( j > 16 ) j = 16;
ZeroMemory(PrintBuffer, 18);
MoveMemory(PrintBuffer, Data, j);
ReturnSize -= j;
for ( i = 0; i < 4; i++ ) {
fprintf(stdout,
" %08lx", PrintData[i]);
}
fprintf(stdout, " ");
for ( i = 0; i < 16; i++ ) {
fprintf(stdout, "%c",
isprint(PrintBuffer[i])?PrintBuffer[i]:'.');
}
Data += 4;
fprintf(stdout, "\n");
}
switch ( controlCode ) {
case CLUSCTL_GROUP_GET_COMMON_PROPERTIES:
controlCode = CLUSCTL_GROUP_SET_COMMON_PROPERTIES;
break;
case CLUSCTL_GROUP_GET_PRIVATE_PROPERTIES:
controlCode = CLUSCTL_GROUP_SET_PRIVATE_PROPERTIES;
break;
default:
controlCode = 0;
break;
}
if ( controlCode != 0 ) {
status = ClusterGroupControl( hGroup,
hNode,
controlCode,
&PropList,
sizeof(PropList),
NULL,
0,
&ReturnSize );
}
fprintf(stdout, "Status of *INPUT* Control request = %d, insize = %d, retsize = %d\n",
status, sizeof(PropList), ReturnSize);
}
LocalFree(buffer);
return(0);
}