mirror of https://github.com/tongzx/nt5src
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
426 lines
12 KiB
426 lines
12 KiB
/*++
|
|
|
|
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);
|
|
}
|
|
|