Leaked source code of windows server 2003
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.
 
 
 
 
 
 

892 lines
23 KiB

//================================================================================
// Copyright (C) 1997 Microsoft Corporation
// Author: RameshV
// Description: generic testing engine
//================================================================================
//===============================
// headers
//===============================
#include <hdrmacro.h>
#include <store.h>
#include <dhcpmsg.h>
#include <wchar.h>
#include <dhcpbas.h>
#include <mm\opt.h>
#include <mm\optl.h>
#include <mm\optdefl.h>
#include <mm\optclass.h>
#include <mm\classdefl.h>
#include <mm\bitmask.h>
#include <mm\reserve.h>
#include <mm\range.h>
#include <mm\subnet.h>
#include <mm\sscope.h>
#include <mm\oclassdl.h>
#include <mm\server.h>
#include <mm\address.h>
#include <mm\server2.h>
#include <mm\memfree.h>
#include <mmreg\regutil.h>
#include <mmreg\regread.h>
#include <mmreg\regsave.h>
#include <dhcpread.h>
#include <dhcpapi.h>
#include <rpcapi1.h>
#include <rpcapi2.h>
//BeginExport(typedefs)
// supported data types are listed. any new datatypes should be added here.
typedef enum /* anonymous */ {
Char,
wChar,
Word,
DWord,
String,
wString,
IpAddress,
xBytes,
Invalid
} NJIN_TYPES, *LPNJIN_TYPES;
typedef struct /* anonymous */ {
union /* anonymous */ {
CHAR *Char;
WCHAR *wChar;
WORD *Word;
DWORD *DWord;
LPSTR String;
LPWSTR wString;
DWORD *IpAddress;
struct /* anonymous */ {
DWORD *xBytesLen;
LPBYTE xBytesBuf;
};
};
} NJIN_DATA, *LPNJIN_DATA;
typedef struct /* anonymous */ {
LPWSTR VarName;
NJIN_TYPES VarType;
NJIN_DATA VarData;
VOID (*VarFunc)(LPWSTR VarName);
BOOL ReadOnly;
} NJIN_VAR, *LPNJIN_VAR;
typedef struct /* anonymous */ {
WCHAR cComment;
WCHAR cEquals;
WCHAR cReturn;
WCHAR cEndOfLine;
WCHAR cCharFmtActual;
WCHAR cCharFmtNum;
BOOL fEcho;
BOOL fVerbose;
} NJIN_ENV, *LPNJIN_ENV;
//EndExport(typedefs)
#define verbose if( Env->fVerbose) printf
#define echo if( Env->fEcho) printf
DWORD
ConvertToIntegral(
IN LPWSTR Str
)
{
return _wtol(Str);
}
DWORD
ConvertToAddress(
IN LPWSTR Str
)
{
CHAR Buf[1000];
LPSTR s;
s = Buf;
while(*s ++ = (CHAR) *Str++ )
/* skip */;
return inet_addr(Buf);
}
DWORD
CloneWString(
IN LPWSTR Buf,
IN LPWSTR Dest
)
{
wcscpy(Dest, Buf);
return ERROR_SUCCESS;
}
DWORD
CloneString(
IN LPWSTR Str,
IN LPSTR s
)
{
while(*s ++ = (CHAR) *Str ++ )
/* skip */;
return ERROR_SUCCESS;
}
BYTE
Hex(
IN WCHAR C
)
{
if( C >= L'0' && C <= L'9' ) return C - L'0';
if( C >= L'A' && C <= L'F' ) return C - L'A' + 10;
if( C >= L'a' && C <= L'f' ) return C - L'a' + 10;
return 0x55;
}
DWORD
ConvertToHexBytes(
IN LPWSTR Str,
IN OUT LPBYTE Buf,
IN OUT LPDWORD BufLen
)
{
BYTE ch, ch2;
*BufLen = 0;
while( (ch = Hex(*Str++)) <= 0xF) {
ch2 = Hex(*Str++);
if( ch2 > 0xF ) {
printf("illegal character found, ignored\n");
ch2 = 0;
}
Buf[(*BufLen)++] = (ch <<4) | ch2;
}
return ERROR_SUCCESS;
}
DWORD
ConvertDataType(
IN LPNJIN_ENV Env,
IN LPWSTR Buf,
IN DWORD VarType,
IN NJIN_DATA *VarData
)
{
switch(VarType) {
case Char:
if( Env->cCharFmtActual == Buf[0] ) {
*VarData->Char = (CHAR) Buf[1];
} else if( Env->cCharFmtNum == Buf[0] ) {
*VarData->Char = (CHAR) ConvertToIntegral(&Buf[1]);
} else return ERROR_INVALID_DATA;
return ERROR_SUCCESS;
case wChar:
if( Env->cCharFmtActual == Buf[0] ) {
*VarData->wChar = Buf[1];
} else if( Env->cCharFmtNum == Buf[0] ) {
*VarData->wChar = (WCHAR) ConvertToIntegral(&Buf[1]);
} else return ERROR_INVALID_DATA;
return ERROR_SUCCESS;
case Word:
*VarData->Word = (WORD) ConvertToIntegral(Buf);
return ERROR_SUCCESS;
case DWord:
*VarData->DWord = ConvertToIntegral(Buf);
return ERROR_SUCCESS;
case wString:
return CloneWString(Buf, VarData->wString);
case String:
return CloneString(Buf, VarData->String);
case IpAddress:
*VarData->IpAddress = ConvertToAddress(Buf);
return ERROR_SUCCESS;
case xBytes:
return ConvertToHexBytes(Buf, VarData->xBytesBuf, VarData->xBytesLen);
default:
verbose("unknown variable type\n");
return ERROR_INVALID_DATA;
}
}
//BeginExport(function)
DWORD
NjinExecuteLineA(
IN LPNJIN_ENV Env,
IN OUT LPNJIN_VAR Vars,
IN DWORD nVars,
IN LPWSTR Cmd
) //EndExport(function)
{
CHAR Ch, LastCh;
DWORD i, Result;
LPWSTR Var;
LPWSTR Value;
Var = NULL; Value = NULL;
for( i = 0; Cmd[i] != L'\0' ; i ++ ) {
if( Env->cEquals == Cmd[i] ) break;
}
if( Cmd[i] ) {
Cmd[i] = L'\0';
Var = Cmd;
Value = &Cmd[i+1];
} else {
verbose("No parameter passed\n");
return ERROR_INVALID_DATA;
}
for( i = 0; i < nVars; i ++ ) {
if( !Vars[i].ReadOnly && 0 == wcscmp(Vars[i].VarName, Var) )
break;
}
if( i >= nVars ) {
verbose("No such variable or operation\n");
return ERROR_FILE_NOT_FOUND;
}
Result = ConvertDataType(Env, Value, Vars[i].VarType, &Vars[i].VarData);
if( ERROR_SUCCESS != Result ) {
verbose("ConvertDataType: 0x%lx (%ld)\n", Result, Result );
return Result;
}
if( Vars[i].VarFunc ) Vars[i].VarFunc(Vars[i].VarName );
return ERROR_SUCCESS;
}
//BeginExport(function)
DWORD
NjinExecute(
IN LPNJIN_ENV Env,
IN OUT LPNJIN_VAR Vars,
IN DWORD nVars
) //EndExport(function)
{
char ch, lastch;
WCHAR buffer[1000];
unsigned int i;
DWORD Result;
verbose("NjinExecute called\n");
i = 0; lastch = (CHAR)Env->cEndOfLine;
while ((ch = getchar()) != EOF ) {
if( Env->cReturn == ch ) ch = (CHAR)Env->cEndOfLine;
if( Env->cEndOfLine == lastch && Env->cComment == ch ) {
while((ch = getchar()) != EOF && Env->cEndOfLine != ch )
;
lastch = ch;
continue;
}
if( Env->cEndOfLine == ch ) {
if( i ) {
buffer[i] = L'\0';
Result = NjinExecuteLineA(Env,Vars,nVars, buffer);
if( ERROR_SUCCESS != Result ) {
verbose("Error: %ld (0x%lx)\n", Result, Result);
return Result;
}
}
i = 0;
} else {
buffer[i++] = (WCHAR)ch;
}
lastch = ch;
}
return ERROR_SUCCESS;
}
//================================================================================
// DHCP SNAPIN DS WORK -- helper routines
//================================================================================
STORE_HANDLE hStore1, hStore2, hStore3, hStore4;
LPSTORE_HANDLE hRoot = NULL, hDhcpC = NULL, hObject = NULL, hDhcpRoot = NULL;
WCHAR gDsServer[100], gObject[100], gName[100], gComment[100];
WCHAR gClassName[100], gClassComment[100];
WCHAR Dummy[100];
WCHAR gOptName[100], gOptComment[100];
WCHAR gServerName[256];
DWORD gServerAddress, gServerState;
DWORD gOptId, gOptType, gOptLen, gIsVendor;
DWORD gDwordOptData;
BYTE gOptVal[255], gClassData[255];
DWORD gClassDataLen;
VOID
print_DHCP_OPTION_DATA(
IN DWORD nTabs,
IN LPDHCP_OPTION_DATA OptData
)
{
DWORD i;
i = nTabs;
while(i--) putchar('\t');
printf("NumElements=%ld, Type=%ld [0x%lx]\n",
OptData->NumElements, OptData->Elements[0].OptionType, OptData->Elements[0].Element.DWordOption
);
}
VOID
print_DHCP_OPTION(
IN DWORD nTabs,
IN LPDHCP_OPTION Option
)
{
DWORD i;
i = nTabs;
while(i--) putchar('\t');
printf("Option [x%03x] %ws %ws Type= %ld\n",
Option->OptionID, Option->OptionName, Option->OptionComment?Option->OptionComment:L"", Option->OptionType
);
print_DHCP_OPTION_DATA(nTabs+1, &Option->DefaultValue);
}
VOID
print_DHCP_OPTION_ARRAY(
IN DWORD nTabs,
IN LPDHCP_OPTION_ARRAY OptArray
)
{
DWORD i;
i = nTabs;
while(i--) putchar('\t');
printf("NumElements= %ld\n", OptArray->NumElements);
for( i = 0; i < OptArray->NumElements; i++ )
print_DHCP_OPTION(nTabs+1, &OptArray->Options[i]);
}
VOID
print_DHCP_OPTION_VALUE(
IN DWORD nTabs,
IN LPDHCP_OPTION_VALUE OptVal
)
{
DWORD i;
i = nTabs;
while(i--) putchar('\t');
printf("OptionId=%ld\n", OptVal->OptionID);
print_DHCP_OPTION_DATA(nTabs+1, &OptVal->Value);
}
VOID
print_DHCP_OPTION_VALUE_ARRAY(
IN DWORD nTabs,
IN LPDHCP_OPTION_VALUE_ARRAY OptValues
)
{
DWORD i;
i = nTabs;
while(i--) putchar('\t');
printf("NumElements=%ld\n", OptValues->NumElements);
for( i = 0; i < OptValues->NumElements ; i ++ )
print_DHCP_OPTION_VALUE(nTabs+1, &OptValues->Values[i]);
}
VOID
print_DHCP_CLASS_INFO(
IN DWORD nTabs,
IN LPDHCP_CLASS_INFO ClassInfo
)
{
DWORD i;
i = nTabs;
while(i--) putchar('\t');
printf("ClassName= %ws ClassComment= %ws DataLength = %ld\n",
ClassInfo->ClassName, ClassInfo->ClassComment, ClassInfo->ClassDataLength
);
}
VOID
print_DHCP_CLASS_INFO_ARRAY(
IN DWORD nTabs,
IN LPDHCP_CLASS_INFO_ARRAY ClassInfoArray
)
{
DWORD i;
i = nTabs;
while(i--) putchar('\t');
printf("NumElements=%ld\n", ClassInfoArray->NumElements);
for( i = 0; i < ClassInfoArray->NumElements ; i ++ )
print_DHCP_CLASS_INFO(nTabs+1, &ClassInfoArray->Classes[i]);
}
VOID
print_DHCPDS_SERVER(
IN DWORD nTabs,
IN LPDHCPDS_SERVER Server
)
{
DWORD i;
i = nTabs;
while(i--) putchar('\t');
printf("Name=[%ws] State=%ld Address=%s Loc=[%ws] Type=%ld\n",
Server->ServerName, Server->Flags, inet_ntoa(*(struct in_addr *)&Server->IpAddress),
Server->DsLocation? Server->DsLocation: L"(NULL)",
Server->DsLocType
);
}
VOID
print_DHCPDS_SERVERS(
IN DWORD nTabs,
IN LPDHCPDS_SERVERS Servers
)
{
DWORD i;
i = nTabs;
while(i--) putchar('\t');
printf("NumElements=%ld\n", Servers->NumElements);
for( i = 0; i < Servers->NumElements ; i ++ )
print_DHCPDS_SERVER(nTabs+1, &Servers->Servers[i]);
}
VOID
SetDsServer(
IN LPWSTR VarNameUnused
)
{
DWORD Err;
if( NULL != hRoot ) {
StoreCleanupHandle(hRoot, 0, TRUE);
}
hRoot = &hStore1;
Err = StoreInitHandle(
hRoot,
0,
wcslen(gDsServer)?gDsServer:NULL,
NULL,
NULL,
NULL,
0
);
if( ERROR_SUCCESS != Err ) {
printf("StoreInitHandle():%ld 0x%lx\n", Err, Err);
hRoot = NULL;
return ;
}
if( hDhcpC ) StoreCleanupHandle(hDhcpC, 0, TRUE);
hDhcpC = &hStore2;
Err = StoreGetHandle(
hRoot,
0,
StoreGetChildType,
DHCP_ROOT_OBJECT_PARENT_LOC,
hDhcpC
);
if( ERROR_SUCCESS != Err ) {
printf("StoreInitHandle():%ld 0x%lx\n", Err, Err);
hRoot = NULL;
return ;
}
if( hDhcpRoot ) StoreCleanupHandle(hDhcpRoot, 0, TRUE);
hDhcpRoot = &hStore4;
Err = DhcpDsGetRoot(
DDS_FLAGS_CREATE,
hRoot,
hDhcpRoot
);
if( ERROR_SUCCESS != Err) {
printf("DhcpDsGetRoot(): %ld 0x%lx", Err, Err);
hDhcpRoot = NULL;
}
}
VOID
SetObject(
IN LPWSTR VarNameUnused
)
{
}
VOID
CreateOptDef(
IN LPWSTR VarNameUnused
)
{
DWORD Result;
Result = DhcpDsCreateOptionDef(
hDhcpC,
hObject,
0,
gOptName,
wcslen(gOptComment)?gOptComment:NULL,
wcslen(gClassName)?gClassName: NULL,
gOptId,
gOptType,
gOptVal,
gOptLen
);
printf("DhcpDsCreateOptionDef: %ld (0x%lx)\n", Result, Result);
}
VOID
ModifyOptDef(
IN LPWSTR VarNameUnused
)
{
DWORD Result;
Result = DhcpDsModifyOptionDef(
hDhcpC,
hObject,
0,
gOptName,
wcslen(gOptComment)?gOptComment:NULL,
wcslen(gClassName)?gClassName: NULL,
gOptId,
gOptType,
gOptVal,
gOptLen
);
printf("DhcpDsModifyOptionDef: %ld (0x%lx)\n", Result, Result);
}
VOID
DeleteOptDef(
IN LPWSTR VarNameUnused
)
{
DWORD Result;
Result = DhcpDsDeleteOptionDef(
hDhcpC,
hObject,
0,
wcslen(gClassName)?gClassName: NULL,
gOptId
);
printf("DhcpDsDeleteOptionDef: %ld (0x%lx)\n", Result, Result);
}
VOID
EnumOptDefs(
IN LPWSTR VarNameUnused
)
{
DWORD Result;
LPDHCP_OPTION_ARRAY OptArray = NULL;
Result = DhcpDsEnumOptionDefs(
hDhcpC,
hObject,
0,
wcslen(gClassName)?gClassName: NULL,
gIsVendor,
&OptArray
);
printf("DhcpDsEnumOptionDefs: %ld (0x%lx)\n", Result, Result);
if( ERROR_SUCCESS == Result ) print_DHCP_OPTION_ARRAY(0, OptArray);
if( NULL != OptArray ) LocalFree(OptArray);
}
VOID
SetOptValue(
IN LPWSTR VarNameUnused
)
{
DWORD Result;
DHCP_OPTION_DATA OptData;
DHCP_OPTION_DATA_ELEMENT OptDataElement;
OptData.NumElements = 1;
OptData.Elements = &OptDataElement;
OptDataElement.OptionType = DhcpDWordOption;
OptDataElement.Element.DWordOption = gDwordOptData;
Result = DhcpDsSetOptionValue(
hDhcpC,
hObject,
0,
wcslen(gClassName)?gClassName: NULL,
gOptId,
&OptData
);
printf("DhcpDsSetOptionValue: %ld (0x%lx)\n", Result, Result);
}
VOID
RemoveOptValue(
IN LPWSTR VarNameUnused
)
{
DWORD Result;
Result = DhcpDsRemoveOptionValue(
hDhcpC,
hObject,
0,
wcslen(gClassName)? gClassName : NULL,
gOptId
);
printf("DhcpDsRemoveOptionValue: %ld (0x%lx)\n", Result, Result);
}
VOID
GetOptValue(
IN LPWSTR VarNameUnused
)
{
DWORD Result;
LPDHCP_OPTION_VALUE OptVal = NULL;
Result = DhcpDsGetOptionValue(
hDhcpC,
hObject,
0,
wcslen(gClassName)? gClassName : NULL,
gOptId,
&OptVal
);
printf("DhcpDsGetOptionValue: %ld (0x%lx)\n", Result, Result);
if( ERROR_SUCCESS == Result ) print_DHCP_OPTION_VALUE(0, OptVal);
if( OptVal ) LocalFree(OptVal);
}
VOID
EnumOptValues(
IN LPWSTR VarNameUnused
)
{
DWORD Result;
LPDHCP_OPTION_VALUE_ARRAY OptValues = NULL;
Result = DhcpDsEnumOptionValues(
hDhcpC,
hObject,
0,
wcslen(gClassName)? gClassName : NULL,
gIsVendor,
&OptValues
);
printf("DhcpDsEnumOptionValue: %ld (0x%lx)\n", Result, Result);
if( ERROR_SUCCESS == Result ) print_DHCP_OPTION_VALUE_ARRAY(0, OptValues);
if( OptValues ) LocalFree(OptValues);
}
VOID
CreateClass(
IN LPWSTR VarNameUnused
)
{
DWORD Result;
Result = DhcpDsCreateClass(
hDhcpC,
hObject,
0,
wcslen(gClassName)? gClassName : NULL,
wcslen(gClassComment)? gClassComment : NULL,
gClassData,
gClassDataLen
);
printf("DhcpDsCreateClass: %ld (0x%lx)\n", Result, Result);
}
VOID
DeleteClass(
IN LPWSTR VarNameUnused
)
{
DWORD Result;
Result = DhcpDsDeleteClass(
hDhcpC,
hObject,
0,
gClassName
);
printf("DhcpDsDeleteClass: %ld (0x%lx)\n", Result, Result);
}
VOID
EnumClasses(
IN LPWSTR VarNameUnused
)
{
DWORD Result;
LPDHCP_CLASS_INFO_ARRAY Classes = NULL;
Result = DhcpDsEnumClasses(
hDhcpC,
hObject,
0,
&Classes
);
printf("DhcpDsEnumClasses: %ld (0x%lx)\n", Result, Result);
if( ERROR_SUCCESS == Result ) print_DHCP_CLASS_INFO_ARRAY(0, Classes);
if( Classes ) LocalFree(Classes);
}
VOID
AddServer(
IN LPWSTR VarNameUnused
)
{
DWORD Result;
Result = DhcpDsAddServer(
hDhcpC,
hDhcpRoot,
0,
gServerName,
NULL,
gServerAddress,
gServerState
);
printf("DhcpDsAddServer: %ld (0x%lx)\n", Result, Result);
}
VOID
DelServer(
IN LPWSTR VarNameUnused
)
{
DWORD Result;
Result = DhcpDsDelServer(
hDhcpC,
hDhcpRoot,
0,
gServerName,
NULL,
gServerAddress
);
printf("DhcpDsDelServer: %ld (0x%lx)\n", Result, Result);
}
VOID
EnumServers(
IN LPWSTR VarNameUnused
)
{
DWORD Result;
LPDHCPDS_SERVERS Servers;
Servers = NULL;
Result = DhcpDsEnumServers(
hDhcpC,
hDhcpRoot,
0,
&Servers
);
printf("DhcpDsEnumServer: %ld (0x%lx)\n", Result, Result);
if( ERROR_SUCCESS == Result ) print_DHCPDS_SERVERS( 0, Servers);
if( Servers ) MemFree(Servers);
}
NJIN_VAR VarTable[] = {
L"DsServer", wString, { (LPVOID) &gDsServer }, SetDsServer, FALSE, //0
L"Object", wString, { (LPVOID) &gObject }, SetObject, FALSE, //1
L"OptName", wString, { (LPVOID) &gOptName }, NULL, FALSE, //2
L"OptComment", wString, { (LPVOID) &gOptComment }, NULL, FALSE, //3
L"ClassName", wString, { (LPVOID) &gClassName }, NULL, FALSE, //4
L"ClassComment",wString,{ (LPVOID) &gClassComment }, NULL, FALSE, //5
L"OptId", DWord, { (LPVOID) &gOptId }, NULL, FALSE, //6
L"OptType", DWord, { (LPVOID) &gOptType }, NULL, FALSE, //7
L"OptData", xBytes, { (LPVOID) NULL }, NULL, FALSE, //############ -> Need to fill pointer..
L"IsVendor", DWord, { (LPVOID) &gIsVendor }, NULL, FALSE, //9
L"DwordOptData", DWord, { (LPVOID) &gDwordOptData }, NULL, FALSE, //10
L"ClassData", xBytes, { (LPVOID) NULL }, NULL, FALSE, //############ -> need to fill pointer
L"ServerName", wString, { (LPVOID) &gServerName }, NULL, FALSE,
L"ServerAddress", IpAddress, { (LPVOID) &gServerAddress }, NULL, FALSE,
L"ServerState", DWord, { (LPVOID) &gServerState }, NULL, FALSE,
L"CreateOptionDef", wString, { (LPVOID) &Dummy }, CreateOptDef, FALSE, //12
L"ModifyOptionDef", wString, { (LPVOID) &Dummy }, ModifyOptDef, FALSE,
L"EnumOptionDefs", wString, { (LPVOID) &Dummy }, EnumOptDefs, FALSE,
L"SetOptionValue", wString, { (LPVOID) &Dummy }, SetOptValue, FALSE,
L"RemoveOptionValue", wString, { (LPVOID) &Dummy }, RemoveOptValue, FALSE,
L"GetOptionValue", wString, { (LPVOID) &Dummy }, GetOptValue, FALSE,
L"EnumOptionValues", wString, { (LPVOID) &Dummy }, EnumOptValues, FALSE,
L"CreateClass", wString, { (LPVOID) &Dummy }, CreateClass, FALSE,
L"DeleteClass", wString, { (LPVOID) &Dummy }, DeleteClass, FALSE,
L"EnumClasses", wString, { (LPVOID) &Dummy }, EnumClasses, FALSE,
L"AddServer", wString, { (LPVOID) &Dummy }, AddServer, FALSE,
L"DelServer", wString, { (LPVOID) &Dummy }, DelServer, FALSE,
L"EnumServers", wString, { (LPVOID) &Dummy }, EnumServers, FALSE,
};
NJIN_ENV Env[] = {
L'#',
L'=',
L'\r',
L'\n',
L' ',
L'#',
TRUE,
TRUE
};
#define VarTableSz (sizeof(VarTable)/sizeof(VarTable[0]))
void _cdecl main (void) {
DWORD Result;
DWORD i;
VarTable[8].VarData.xBytesLen = &gOptLen;
VarTable[8].VarData.xBytesBuf = gOptVal;
VarTable[11].VarData.xBytesLen = &gClassDataLen;
VarTable[11].VarData.xBytesBuf = gClassData;
Result = NjinExecute(
Env,
VarTable,
VarTableSz
);
if( ERROR_SUCCESS != Result ) {
printf("\nResult = %ld\n", Result);
}
}
//================================================================================
// end of file
//================================================================================