1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
|
/*++
Copyright (c) 1987-1993 Microsoft Corporation
Module Name:
querydir.c
Abstract:
Contains I_NetRpl_QueryDirectory
Author:
Jon Newman 13 - January - 1994
Environment:
User mode
Revision History :
--*/
#define RPLDATA_ALLOCATE
#include "local.h"
#undef RPLDATA_ALLOCATE
#define RPL_REGISTRY_PARAMETERS L"System\\CurrentControlSet\\Services\\RemoteBoot\\Parameters"
#define RPL_REGISTRY_DIRECTORY L"Directory"
#define RPL_DEFAULT_DIRECTORY L"%SystemRoot%\\rpl\\"
// BUGBUG server\rplmain.c should call this routine
DWORD I_NetRpl_QueryDirectory( OUT LPWSTR pchFilePath,
OUT LPDWORD pcchFilePathLength)
{
DWORD Error;
BYTE Buffer[ (PATHLEN + 1) * sizeof(WCHAR)];
PWCHAR Value;
HKEY Key;
DWORD Length;
DWORD Type;
DWORD Size;
// RPL_ASSERT( pchFilePath != NULL);
//
// First open the RPL service parameters registry tree.
//
Error = RegOpenKeyEx(
HKEY_LOCAL_MACHINE,
RPL_REGISTRY_PARAMETERS,
0,
KEY_READ,
&Key
);
if (Error != ERROR_SUCCESS){
// RplDump( ++RG_Assert, ("Error = %d", Error));
return( Error);
}
// Length = sizeof( Buffer); This is totally wrong! Has Rl Server been
// using the default value this whole time?
// Length = sizeof( Buffer);
Size = sizeof( Buffer);
Error = RegQueryValueEx(
Key,
RPL_REGISTRY_DIRECTORY,
NULL,
&Type,
Buffer,
&Size
);
(VOID)RegCloseKey( Key);
if ( Error == NO_ERROR) {
//
// We have the data. First validate the data, then append the
// backslash if needed.
//
Value = (PWCHAR)Buffer;
if ( Type != REG_EXPAND_SZ || (Size&1) != 0 || Size < sizeof(L"c:")) {
// RplDump( ++RG_Assert, ("Type = 0x%x, Value = 0x%x, Size=%d",
// Type, Value, Size));
return( RPL_BUGBUG_ERROR);
}
Length = Size / sizeof(WCHAR) - 1;
if ( Value[ Length] != 0 || Length != wcslen(Value)) {
// RplDump( ++RG_Assert, ("Length = %d, Value = 0x%x, Size=%d",
// Length, Value, Size));
return( RPL_BUGBUG_ERROR);
}
if ( Value[ Length - 1] != L'\\') {
//
// Need to append the backslash.
//
if ( Length >= PATHLEN) {
// RplDump( ++RG_Assert, ("Length = %d, Value = 0x%x, Size=%d",
// Length, Value, Size));
return( RPL_BUGBUG_ERROR); // not enough space
}
Value[ Length] = L'\\';
Value[ ++Length] = 0;
}
} else {
Value = RPL_DEFAULT_DIRECTORY;
}
//
// Length returned by ExpandEnvironmentalStrings includes terminating
// NULL byte.
//
// This code was wrong!! The third parameter to ExpandEnvironmentStrings
// should be a character count, not a byte count.
//
// Length = ExpandEnvironmentStrings( Value, RG_Directory, sizeof( RG_Directory));
Length = ExpandEnvironmentStrings( Value, pchFilePath, PATHLEN+1);
if ( Length == 0) {
Error = GetLastError();
// RplDump( ++RG_Assert, ("Error = %d", Error));
return( Error);
}
if ( Length > PATHLEN+1) {
// RplDump( ++RG_Assert, ("Length = %d", Length));
return( RPL_BUGBUG_ERROR);
}
if ( Length != wcslen( pchFilePath) + 1) {
// RplDump( ++RG_Assert, ("Length = %d", Length));
return( RPL_BUGBUG_ERROR);
}
if (pcchFilePathLength != NULL) {
*pcchFilePathLength = (DWORD)(Length - 1);
}
return( NO_ERROR);
}
|
