427 lines
12 KiB
C
427 lines
12 KiB
C
/*++
|
||
|
||
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);
|
||
}
|
||
|