442 lines
6.8 KiB
C
442 lines
6.8 KiB
C
|
/*++
|
||
|
|
||
|
Copyright (C) 2000 Microsoft Corporation
|
||
|
|
||
|
Module Name:
|
||
|
|
||
|
precomp.h
|
||
|
|
||
|
Abstract:
|
||
|
|
||
|
precompiled header
|
||
|
|
||
|
--*/
|
||
|
|
||
|
|
||
|
//#pragma warning(disable : 4115 )
|
||
|
//#pragma warning(disable : 4214 )
|
||
|
//#pragma warning(disable : 4200 )
|
||
|
//#pragma warning(disable : 4213 )
|
||
|
//#pragma warning(disable : 4211 )
|
||
|
//#pragma warning(disable : 4310 )
|
||
|
|
||
|
//
|
||
|
// NT public header files
|
||
|
//
|
||
|
|
||
|
|
||
|
|
||
|
#include <nt.h>
|
||
|
#include <ntrtl.h>
|
||
|
#include <nturtl.h>
|
||
|
#include <ntseapi.h>
|
||
|
#include <windows.h>
|
||
|
#include <shellapi.h>
|
||
|
#include <align.h>
|
||
|
#include <time.h>
|
||
|
|
||
|
//#pragma warning(disable : 4115 )
|
||
|
//#pragma warning(disable : 4214 )
|
||
|
//#pragma warning(disable : 4200 )
|
||
|
//#pragma warning(disable : 4213 )
|
||
|
//#pragma warning(disable : 4211 )
|
||
|
//#pragma warning(disable : 4310 )
|
||
|
|
||
|
#include <lmcons.h>
|
||
|
#include <netlib.h>
|
||
|
#include <lmapibuf.h>
|
||
|
|
||
|
#include <winsock2.h>
|
||
|
#include <excpt.h>
|
||
|
#include <accctrl.h>
|
||
|
|
||
|
|
||
|
//
|
||
|
// C Runtime library includes.
|
||
|
//
|
||
|
|
||
|
#include <stdlib.h>
|
||
|
#include <stdio.h>
|
||
|
#include <string.h>
|
||
|
|
||
|
|
||
|
//
|
||
|
// MM related files
|
||
|
//
|
||
|
|
||
|
#include <mm\mm.h>
|
||
|
#include <mm\array.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 <stdio.h>
|
||
|
#include <stdlib.h>
|
||
|
#include <winsock2.h>
|
||
|
#include <dhcpupg.h>
|
||
|
#include <esent.h>
|
||
|
#include <dhcpmsg.h>
|
||
|
#include <dhcplib.h>
|
||
|
#include <dhcpexim.h>
|
||
|
#include <dhcpapi.h>
|
||
|
|
||
|
//
|
||
|
// Constants
|
||
|
//
|
||
|
|
||
|
#define DHCPEXIM_REG_CFG_LOC5 TEXT("Software\\Microsoft\\DHCPServer\\Configuration")
|
||
|
#define DHCPEXIM_REG_CFG_LOC4 TEXT("System\\CurrentControlSet\\Services\\DHCPServer\\Configuration")
|
||
|
|
||
|
#define SAVE_BUF_SIZE (1004096L)
|
||
|
|
||
|
extern CHAR DhcpEximOemDatabaseName[2048];
|
||
|
extern CHAR DhcpEximOemDatabasePath[2048];
|
||
|
extern HANDLE hTextFile; // defined in dbfile.c
|
||
|
extern PUCHAR SaveBuf; // defined in dbfile.c
|
||
|
extern ULONG SaveBufSize; //defined in dbfile.c
|
||
|
|
||
|
enum {
|
||
|
LoadJet200,
|
||
|
LoadJet500,
|
||
|
LoadJet97, /* Win2K, ESENT */
|
||
|
LoadJet2001 /* Whistler, ESENT */
|
||
|
};
|
||
|
|
||
|
|
||
|
typedef struct _MM_ITERATE_CTXT {
|
||
|
//
|
||
|
// This is filled in for all Iterate* routines
|
||
|
//
|
||
|
|
||
|
PM_SERVER Server;
|
||
|
PVOID ExtraCtxt;
|
||
|
|
||
|
//
|
||
|
// This is filled in for IterateClasses
|
||
|
//
|
||
|
|
||
|
PM_CLASSDEF ClassDef;
|
||
|
|
||
|
//
|
||
|
// This is filled in for IterateOptDefs
|
||
|
//
|
||
|
|
||
|
PM_OPTDEF OptDef;
|
||
|
|
||
|
|
||
|
//
|
||
|
// This is filled in for IterateOptions
|
||
|
//
|
||
|
|
||
|
PM_OPTION Option;
|
||
|
|
||
|
//
|
||
|
// These two are filled in by both IterateOptDefs and IterateOptions
|
||
|
//
|
||
|
|
||
|
PM_CLASSDEF UserClass;
|
||
|
PM_CLASSDEF VendorClass;
|
||
|
|
||
|
//
|
||
|
// This is used by IterateScopes
|
||
|
//
|
||
|
|
||
|
PM_SUBNET Scope;
|
||
|
PM_SSCOPE SScope;
|
||
|
|
||
|
//
|
||
|
// This is used by IterateScopeRanges
|
||
|
//
|
||
|
|
||
|
PM_RANGE Range;
|
||
|
|
||
|
//
|
||
|
// This is used by IterateScopeExclusions
|
||
|
//
|
||
|
|
||
|
PM_EXCL Excl;
|
||
|
|
||
|
//
|
||
|
// This is used by IterateScopeReservations
|
||
|
//
|
||
|
|
||
|
PM_RESERVATION Res;
|
||
|
|
||
|
}MM_ITERATE_CTXT, *PMM_ITERATE_CTXT;
|
||
|
|
||
|
DWORD
|
||
|
IterateClasses(
|
||
|
IN PM_SERVER Server,
|
||
|
IN PVOID ExtraCtxt,
|
||
|
IN DWORD (*Callback)( IN OUT PMM_ITERATE_CTXT )
|
||
|
);
|
||
|
|
||
|
DWORD
|
||
|
IterateOptDefs(
|
||
|
IN PM_SERVER Server,
|
||
|
IN PVOID ExtraCtxt,
|
||
|
IN DWORD (*Callback)( IN OUT PMM_ITERATE_CTXT )
|
||
|
);
|
||
|
|
||
|
DWORD
|
||
|
IterateOptionsOnOptClass(
|
||
|
IN PM_SERVER Server,
|
||
|
IN PM_OPTCLASS OptClass,
|
||
|
IN PVOID ExtraCtxt,
|
||
|
IN DWORD (*Callback)( IN OUT PMM_ITERATE_CTXT )
|
||
|
);
|
||
|
|
||
|
DWORD
|
||
|
IterateServerOptions(
|
||
|
IN PM_SERVER Server,
|
||
|
IN PVOID ExtraCtxt,
|
||
|
IN DWORD (*Callback)( IN OUT PMM_ITERATE_CTXT )
|
||
|
);
|
||
|
|
||
|
DWORD
|
||
|
IterateScopeOptions(
|
||
|
IN PM_SUBNET Subnet,
|
||
|
IN PVOID ExtraCtxt,
|
||
|
IN DWORD (*Callback)( IN OUT PMM_ITERATE_CTXT )
|
||
|
);
|
||
|
|
||
|
DWORD
|
||
|
IterateReservationOptions(
|
||
|
IN PM_SERVER Server,
|
||
|
IN PM_RESERVATION Res,
|
||
|
IN PVOID ExtraCtxt,
|
||
|
IN DWORD (*Callback)( IN OUT PMM_ITERATE_CTXT )
|
||
|
);
|
||
|
|
||
|
DWORD
|
||
|
IterateScopes(
|
||
|
IN PM_SERVER Server,
|
||
|
IN PVOID ExtraCtxt,
|
||
|
IN DWORD (*Callback)( IN OUT PMM_ITERATE_CTXT )
|
||
|
);
|
||
|
|
||
|
DWORD
|
||
|
IterateScopeRanges(
|
||
|
IN PM_SUBNET Scope,
|
||
|
IN PVOID ExtraCtxt,
|
||
|
IN DWORD (*Callback)( IN OUT PMM_ITERATE_CTXT )
|
||
|
);
|
||
|
|
||
|
DWORD
|
||
|
IterateScopeExclusions(
|
||
|
IN PM_SUBNET Scope,
|
||
|
IN PVOID ExtraCtxt,
|
||
|
IN DWORD (*Callback)( IN OUT PMM_ITERATE_CTXT )
|
||
|
);
|
||
|
|
||
|
DWORD
|
||
|
IterateScopeReservations(
|
||
|
IN PM_SUBNET Scope,
|
||
|
IN PVOID ExtraCtxt,
|
||
|
IN DWORD (*Callback)( IN OUT PMM_ITERATE_CTXT )
|
||
|
);
|
||
|
|
||
|
|
||
|
//
|
||
|
// readreg.c
|
||
|
//
|
||
|
|
||
|
DWORD
|
||
|
DhcpeximReadRegistryConfiguration(
|
||
|
IN OUT PM_SERVER *Server
|
||
|
);
|
||
|
|
||
|
|
||
|
DWORD
|
||
|
DhcpeximReadRegistryParameters(
|
||
|
VOID
|
||
|
);
|
||
|
|
||
|
//
|
||
|
// writereg.c
|
||
|
//
|
||
|
|
||
|
DWORD
|
||
|
DhcpeximWriteRegistryConfiguration(
|
||
|
IN PM_SERVER Server
|
||
|
);
|
||
|
|
||
|
//
|
||
|
// readdb.c
|
||
|
//
|
||
|
|
||
|
DWORD
|
||
|
DhcpeximReadDatabaseConfiguration(
|
||
|
IN OUT PM_SERVER *Server
|
||
|
);
|
||
|
|
||
|
//
|
||
|
// writedb.c
|
||
|
//
|
||
|
|
||
|
DWORD
|
||
|
DhcpeximWriteDatabaseConfiguration(
|
||
|
IN PM_SERVER Server
|
||
|
);
|
||
|
|
||
|
//
|
||
|
// dbfile.c
|
||
|
//
|
||
|
|
||
|
DWORD
|
||
|
AddRecordNoSize(
|
||
|
IN LPSTR Buffer,
|
||
|
IN ULONG BufSize
|
||
|
);
|
||
|
|
||
|
DWORD
|
||
|
InitializeDatabaseParameters(
|
||
|
VOID
|
||
|
);
|
||
|
|
||
|
DWORD
|
||
|
CleanupDatabaseParameters(
|
||
|
VOID
|
||
|
);
|
||
|
|
||
|
DWORD
|
||
|
SaveDatabaseEntriesToFile(
|
||
|
IN PULONG Subnets,
|
||
|
IN ULONG nSubnets
|
||
|
);
|
||
|
|
||
|
DWORD
|
||
|
SaveFileEntriesToDatabase(
|
||
|
IN LPBYTE Mem,
|
||
|
IN ULONG MemSize,
|
||
|
IN PULONG Subnets,
|
||
|
IN ULONG nSubnets
|
||
|
);
|
||
|
|
||
|
DWORD
|
||
|
OpenTextFile(
|
||
|
IN LPWSTR FileName,
|
||
|
IN BOOL fRead,
|
||
|
OUT HANDLE *hFile,
|
||
|
OUT LPBYTE *Mem,
|
||
|
OUT ULONG *MemSize
|
||
|
);
|
||
|
|
||
|
VOID
|
||
|
CloseTextFile(
|
||
|
IN OUT HANDLE hFile,
|
||
|
IN OUT LPBYTE Mem
|
||
|
);
|
||
|
|
||
|
//
|
||
|
// mmfile.c
|
||
|
//
|
||
|
|
||
|
DWORD
|
||
|
SaveConfigurationToFile(
|
||
|
IN PM_SERVER Server
|
||
|
);
|
||
|
|
||
|
DWORD
|
||
|
ReadDbEntries(
|
||
|
IN OUT LPBYTE *Mem,
|
||
|
IN OUT ULONG *MemSize,
|
||
|
IN OUT PM_SERVER *Server
|
||
|
);
|
||
|
|
||
|
//
|
||
|
// merge.c
|
||
|
//
|
||
|
|
||
|
DWORD
|
||
|
MergeConfigurations(
|
||
|
IN OUT PM_SERVER DestServer,
|
||
|
IN OUT PM_SERVER Server
|
||
|
);
|
||
|
|
||
|
//
|
||
|
// main.c
|
||
|
//
|
||
|
|
||
|
DWORD
|
||
|
Tr(
|
||
|
IN LPSTR Format, ...
|
||
|
);
|
||
|
|
||
|
BOOL IsNT4();
|
||
|
BOOL IsNT5();
|
||
|
|
||
|
VOID
|
||
|
IpAddressToStringW(
|
||
|
IN DWORD IpAddress,
|
||
|
IN LPWSTR String // must have enough space preallocated
|
||
|
);
|
||
|
|
||
|
DWORD
|
||
|
CmdLineDoExport(
|
||
|
IN LPWSTR *Args,
|
||
|
IN ULONG nArgs
|
||
|
);
|
||
|
|
||
|
DWORD
|
||
|
CmdLineDoImport(
|
||
|
IN LPWSTR *Args,
|
||
|
IN ULONG nArgs
|
||
|
);
|
||
|
|
||
|
DWORD
|
||
|
ImportConfiguration(
|
||
|
IN OUT PM_SERVER SvcConfig,
|
||
|
IN ULONG *Subnets,
|
||
|
IN ULONG nSubnets,
|
||
|
IN LPBYTE Mem, // import file : shared mem
|
||
|
IN ULONG MemSize // shared mem size
|
||
|
);
|
||
|
|
||
|
DWORD
|
||
|
ExportConfiguration(
|
||
|
IN OUT PM_SERVER SvcConfig,
|
||
|
IN ULONG *Subnets,
|
||
|
IN ULONG nSubnets,
|
||
|
IN HANDLE hFile
|
||
|
);
|
||
|
|
||
|
DWORD
|
||
|
CleanupServiceConfig(
|
||
|
IN OUT PM_SERVER Server
|
||
|
);
|
||
|
|
||
|
DWORD
|
||
|
InitializeAndGetServiceConfig(
|
||
|
OUT PM_SERVER *pServer
|
||
|
);
|
||
|
|
||
|
//
|
||
|
// select.c
|
||
|
//
|
||
|
|
||
|
DWORD
|
||
|
SelectConfiguration(
|
||
|
IN OUT PM_SERVER Server,
|
||
|
IN ULONG *Subnets,
|
||
|
IN ULONG nSubnets
|
||
|
);
|
||
|
|
||
|
#pragma hdrstop
|
||
|
|
||
|
|