Adds ANCM to IISIntegration (#492)
This commit is contained in:
commit
4b53f4bc53
|
|
@ -42,6 +42,7 @@
|
|||
*.fs text=auto
|
||||
*.fsx text=auto
|
||||
*.hs text=auto
|
||||
*.rc text=auto
|
||||
|
||||
*.csproj text=auto
|
||||
*.vbproj text=auto
|
||||
|
|
|
|||
|
|
@ -25,10 +25,34 @@ project.lock.json
|
|||
*.ncrunchsolution
|
||||
*.*sdf
|
||||
*.ipch
|
||||
*.vs/
|
||||
.vscode/
|
||||
.testPublish/
|
||||
.build/
|
||||
*.nuget.props
|
||||
*.nuget.targets
|
||||
*.bin
|
||||
*.vs/
|
||||
.testPublish/
|
||||
|
||||
*.obj
|
||||
*.tlog
|
||||
*.CppClean.log
|
||||
*msbuild.log
|
||||
|
||||
src/*/Debug/
|
||||
src/*/x64/Debug/
|
||||
src/*/Release/
|
||||
src/*/x64/Release/
|
||||
x64/
|
||||
|
||||
*vcxproj.filters
|
||||
*.aps
|
||||
*.pdb
|
||||
*.lib
|
||||
*.idb
|
||||
|
||||
src/AspNetCore/aspnetcore_msg.h
|
||||
src/AspNetCore/aspnetcore_msg.rc
|
||||
src/AspNetCore/version.h
|
||||
.build
|
||||
|
||||
*.VC.*db
|
||||
global.json
|
||||
|
|
|
|||
|
|
@ -1,4 +1,4 @@
|
|||
<Project>
|
||||
<Project>
|
||||
<Import
|
||||
Project="$([MSBuild]::GetDirectoryNameOfFileAbove($(MSBuildThisFileDirectory), AspNetCoreSettings.props))\AspNetCoreSettings.props"
|
||||
Condition=" '$(CI)' != 'true' AND '$([MSBuild]::GetDirectoryNameOfFileAbove($(MSBuildThisFileDirectory), AspNetCoreSettings.props))' != '' " />
|
||||
|
|
|
|||
|
|
@ -1,6 +1,6 @@
|
|||
Microsoft Visual Studio Solution File, Format Version 12.00
|
||||
# Visual Studio 15
|
||||
VisualStudioVersion = 15.0.26730.10
|
||||
VisualStudioVersion = 15.0.27130.2010
|
||||
MinimumVisualStudioVersion = 15.0.26730.03
|
||||
Project("{2150E333-8FDC-42A3-9474-1A3956D46DE8}") = "src", "src", "{04B1EDB6-E967-4D25-89B9-E6F8304038CD}"
|
||||
ProjectSection(SolutionItems) = preProject
|
||||
|
|
@ -18,6 +18,7 @@ EndProject
|
|||
Project("{2150E333-8FDC-42A3-9474-1A3956D46DE8}") = "test", "test", "{EF30B533-D715-421A-92B7-92FEF460AC9C}"
|
||||
ProjectSection(SolutionItems) = preProject
|
||||
test\Directory.Build.props = test\Directory.Build.props
|
||||
test\WebSocketClientEXE\WebSocketClientEXE.csproj = test\WebSocketClientEXE\WebSocketClientEXE.csproj
|
||||
EndProjectSection
|
||||
EndProject
|
||||
Project("{2150E333-8FDC-42A3-9474-1A3956D46DE8}") = "samples", "samples", "{C74B8F36-FD2F-45C9-9B8A-00E7CF0126A9}"
|
||||
|
|
@ -45,6 +46,18 @@ Project("{9A19103F-16F7-4668-BE54-9A1E7A4F7556}") = "IISTestSite", "test\IISTest
|
|||
EndProject
|
||||
Project("{9A19103F-16F7-4668-BE54-9A1E7A4F7556}") = "IISIntegration.IISServerFunctionalTests", "test\IISIntegration.IISServerFunctionalTests\IISIntegration.IISServerFunctionalTests.csproj", "{C745BBFD-7888-4038-B41B-6D1832D13878}"
|
||||
EndProject
|
||||
Project("{8BC9CEB8-8B4A-11D0-8D11-00A0C91BC942}") = "AspNetCore", "src\AspNetCore\AspNetCore.vcxproj", "{439824F9-1455-4CC4-BD79-B44FA0A16552}"
|
||||
EndProject
|
||||
Project("{8BC9CEB8-8B4A-11D0-8D11-00A0C91BC942}") = "IISLib", "src\IISLib\IISLib.vcxproj", "{4787A64F-9A3E-4867-A55A-70CB4B2B2FFE}"
|
||||
EndProject
|
||||
Project("{8BC9CEB8-8B4A-11D0-8D11-00A0C91BC942}") = "CommonLib", "src\CommonLib\CommonLib.vcxproj", "{55494E58-E061-4C4C-A0A8-837008E72F85}"
|
||||
EndProject
|
||||
Project("{8BC9CEB8-8B4A-11D0-8D11-00A0C91BC942}") = "RequestHandler", "src\RequestHandler\RequestHandler.vcxproj", "{D57EA297-6DC2-4BC0-8C91-334863327863}"
|
||||
EndProject
|
||||
Project("{9A19103F-16F7-4668-BE54-9A1E7A4F7556}") = "AspNetCoreModule.Test", "test\AspNetCoreModule.Test\AspNetCoreModule.Test.csproj", "{5F31656B-4990-44FE-9090-00E32D933376}"
|
||||
EndProject
|
||||
Project("{9A19103F-16F7-4668-BE54-9A1E7A4F7556}") = "AspNetCoreModule.TestSites.Standard", "test\AspNetCoreModule.TestSites.Standard\AspNetCoreModule.TestSites.Standard.csproj", "{93ECA06C-767E-4A4D-AC59-21F250381297}"
|
||||
EndProject
|
||||
Global
|
||||
GlobalSection(SolutionConfigurationPlatforms) = preSolution
|
||||
Debug|Any CPU = Debug|Any CPU
|
||||
|
|
@ -150,6 +163,70 @@ Global
|
|||
{C745BBFD-7888-4038-B41B-6D1832D13878}.Release|x64.ActiveCfg = Release|Any CPU
|
||||
{C745BBFD-7888-4038-B41B-6D1832D13878}.Release|x64.Build.0 = Release|Any CPU
|
||||
{C745BBFD-7888-4038-B41B-6D1832D13878}.Release|x86.ActiveCfg = Release|Any CPU
|
||||
{439824F9-1455-4CC4-BD79-B44FA0A16552}.Debug|Any CPU.ActiveCfg = Debug|Win32
|
||||
{439824F9-1455-4CC4-BD79-B44FA0A16552}.Debug|x64.ActiveCfg = Debug|x64
|
||||
{439824F9-1455-4CC4-BD79-B44FA0A16552}.Debug|x64.Build.0 = Debug|x64
|
||||
{439824F9-1455-4CC4-BD79-B44FA0A16552}.Debug|x86.ActiveCfg = Debug|Win32
|
||||
{439824F9-1455-4CC4-BD79-B44FA0A16552}.Debug|x86.Build.0 = Debug|Win32
|
||||
{439824F9-1455-4CC4-BD79-B44FA0A16552}.Release|Any CPU.ActiveCfg = Release|Win32
|
||||
{439824F9-1455-4CC4-BD79-B44FA0A16552}.Release|x64.ActiveCfg = Release|x64
|
||||
{439824F9-1455-4CC4-BD79-B44FA0A16552}.Release|x64.Build.0 = Release|x64
|
||||
{439824F9-1455-4CC4-BD79-B44FA0A16552}.Release|x86.ActiveCfg = Release|Win32
|
||||
{439824F9-1455-4CC4-BD79-B44FA0A16552}.Release|x86.Build.0 = Release|Win32
|
||||
{4787A64F-9A3E-4867-A55A-70CB4B2B2FFE}.Debug|Any CPU.ActiveCfg = Debug|Win32
|
||||
{4787A64F-9A3E-4867-A55A-70CB4B2B2FFE}.Debug|x64.ActiveCfg = Debug|x64
|
||||
{4787A64F-9A3E-4867-A55A-70CB4B2B2FFE}.Debug|x64.Build.0 = Debug|x64
|
||||
{4787A64F-9A3E-4867-A55A-70CB4B2B2FFE}.Debug|x86.ActiveCfg = Debug|Win32
|
||||
{4787A64F-9A3E-4867-A55A-70CB4B2B2FFE}.Debug|x86.Build.0 = Debug|Win32
|
||||
{4787A64F-9A3E-4867-A55A-70CB4B2B2FFE}.Release|Any CPU.ActiveCfg = Release|Win32
|
||||
{4787A64F-9A3E-4867-A55A-70CB4B2B2FFE}.Release|x64.ActiveCfg = Release|x64
|
||||
{4787A64F-9A3E-4867-A55A-70CB4B2B2FFE}.Release|x64.Build.0 = Release|x64
|
||||
{4787A64F-9A3E-4867-A55A-70CB4B2B2FFE}.Release|x86.ActiveCfg = Release|Win32
|
||||
{4787A64F-9A3E-4867-A55A-70CB4B2B2FFE}.Release|x86.Build.0 = Release|Win32
|
||||
{55494E58-E061-4C4C-A0A8-837008E72F85}.Debug|Any CPU.ActiveCfg = Debug|Win32
|
||||
{55494E58-E061-4C4C-A0A8-837008E72F85}.Debug|x64.ActiveCfg = Debug|x64
|
||||
{55494E58-E061-4C4C-A0A8-837008E72F85}.Debug|x64.Build.0 = Debug|x64
|
||||
{55494E58-E061-4C4C-A0A8-837008E72F85}.Debug|x86.ActiveCfg = Debug|Win32
|
||||
{55494E58-E061-4C4C-A0A8-837008E72F85}.Debug|x86.Build.0 = Debug|Win32
|
||||
{55494E58-E061-4C4C-A0A8-837008E72F85}.Release|Any CPU.ActiveCfg = Release|Win32
|
||||
{55494E58-E061-4C4C-A0A8-837008E72F85}.Release|x64.ActiveCfg = Release|x64
|
||||
{55494E58-E061-4C4C-A0A8-837008E72F85}.Release|x64.Build.0 = Release|x64
|
||||
{55494E58-E061-4C4C-A0A8-837008E72F85}.Release|x86.ActiveCfg = Release|Win32
|
||||
{55494E58-E061-4C4C-A0A8-837008E72F85}.Release|x86.Build.0 = Release|Win32
|
||||
{D57EA297-6DC2-4BC0-8C91-334863327863}.Debug|Any CPU.ActiveCfg = Debug|Win32
|
||||
{D57EA297-6DC2-4BC0-8C91-334863327863}.Debug|x64.ActiveCfg = Debug|x64
|
||||
{D57EA297-6DC2-4BC0-8C91-334863327863}.Debug|x64.Build.0 = Debug|x64
|
||||
{D57EA297-6DC2-4BC0-8C91-334863327863}.Debug|x86.ActiveCfg = Debug|Win32
|
||||
{D57EA297-6DC2-4BC0-8C91-334863327863}.Debug|x86.Build.0 = Debug|Win32
|
||||
{D57EA297-6DC2-4BC0-8C91-334863327863}.Release|Any CPU.ActiveCfg = Release|Win32
|
||||
{D57EA297-6DC2-4BC0-8C91-334863327863}.Release|x64.ActiveCfg = Release|x64
|
||||
{D57EA297-6DC2-4BC0-8C91-334863327863}.Release|x64.Build.0 = Release|x64
|
||||
{D57EA297-6DC2-4BC0-8C91-334863327863}.Release|x86.ActiveCfg = Release|Win32
|
||||
{D57EA297-6DC2-4BC0-8C91-334863327863}.Release|x86.Build.0 = Release|Win32
|
||||
{5F31656B-4990-44FE-9090-00E32D933376}.Debug|Any CPU.ActiveCfg = Debug|Any CPU
|
||||
{5F31656B-4990-44FE-9090-00E32D933376}.Debug|Any CPU.Build.0 = Debug|Any CPU
|
||||
{5F31656B-4990-44FE-9090-00E32D933376}.Debug|x64.ActiveCfg = Debug|Any CPU
|
||||
{5F31656B-4990-44FE-9090-00E32D933376}.Debug|x64.Build.0 = Debug|Any CPU
|
||||
{5F31656B-4990-44FE-9090-00E32D933376}.Debug|x86.ActiveCfg = Debug|Any CPU
|
||||
{5F31656B-4990-44FE-9090-00E32D933376}.Debug|x86.Build.0 = Debug|Any CPU
|
||||
{5F31656B-4990-44FE-9090-00E32D933376}.Release|Any CPU.ActiveCfg = Release|Any CPU
|
||||
{5F31656B-4990-44FE-9090-00E32D933376}.Release|Any CPU.Build.0 = Release|Any CPU
|
||||
{5F31656B-4990-44FE-9090-00E32D933376}.Release|x64.ActiveCfg = Release|Any CPU
|
||||
{5F31656B-4990-44FE-9090-00E32D933376}.Release|x64.Build.0 = Release|Any CPU
|
||||
{5F31656B-4990-44FE-9090-00E32D933376}.Release|x86.ActiveCfg = Release|Any CPU
|
||||
{5F31656B-4990-44FE-9090-00E32D933376}.Release|x86.Build.0 = Release|Any CPU
|
||||
{93ECA06C-767E-4A4D-AC59-21F250381297}.Debug|Any CPU.ActiveCfg = Debug|Any CPU
|
||||
{93ECA06C-767E-4A4D-AC59-21F250381297}.Debug|Any CPU.Build.0 = Debug|Any CPU
|
||||
{93ECA06C-767E-4A4D-AC59-21F250381297}.Debug|x64.ActiveCfg = Debug|Any CPU
|
||||
{93ECA06C-767E-4A4D-AC59-21F250381297}.Debug|x64.Build.0 = Debug|Any CPU
|
||||
{93ECA06C-767E-4A4D-AC59-21F250381297}.Debug|x86.ActiveCfg = Debug|Any CPU
|
||||
{93ECA06C-767E-4A4D-AC59-21F250381297}.Debug|x86.Build.0 = Debug|Any CPU
|
||||
{93ECA06C-767E-4A4D-AC59-21F250381297}.Release|Any CPU.ActiveCfg = Release|Any CPU
|
||||
{93ECA06C-767E-4A4D-AC59-21F250381297}.Release|Any CPU.Build.0 = Release|Any CPU
|
||||
{93ECA06C-767E-4A4D-AC59-21F250381297}.Release|x64.ActiveCfg = Release|Any CPU
|
||||
{93ECA06C-767E-4A4D-AC59-21F250381297}.Release|x64.Build.0 = Release|Any CPU
|
||||
{93ECA06C-767E-4A4D-AC59-21F250381297}.Release|x86.ActiveCfg = Release|Any CPU
|
||||
{93ECA06C-767E-4A4D-AC59-21F250381297}.Release|x86.Build.0 = Release|Any CPU
|
||||
EndGlobalSection
|
||||
GlobalSection(SolutionProperties) = preSolution
|
||||
HideSolutionNode = FALSE
|
||||
|
|
@ -163,6 +240,12 @@ Global
|
|||
{9BC4AFCB-325D-4C81-8228-8CF301CE2F97} = {C74B8F36-FD2F-45C9-9B8A-00E7CF0126A9}
|
||||
{679FA2A2-898B-4320-884E-C2D294A97CE1} = {EF30B533-D715-421A-92B7-92FEF460AC9C}
|
||||
{C745BBFD-7888-4038-B41B-6D1832D13878} = {EF30B533-D715-421A-92B7-92FEF460AC9C}
|
||||
{439824F9-1455-4CC4-BD79-B44FA0A16552} = {04B1EDB6-E967-4D25-89B9-E6F8304038CD}
|
||||
{4787A64F-9A3E-4867-A55A-70CB4B2B2FFE} = {04B1EDB6-E967-4D25-89B9-E6F8304038CD}
|
||||
{55494E58-E061-4C4C-A0A8-837008E72F85} = {04B1EDB6-E967-4D25-89B9-E6F8304038CD}
|
||||
{D57EA297-6DC2-4BC0-8C91-334863327863} = {04B1EDB6-E967-4D25-89B9-E6F8304038CD}
|
||||
{5F31656B-4990-44FE-9090-00E32D933376} = {EF30B533-D715-421A-92B7-92FEF460AC9C}
|
||||
{93ECA06C-767E-4A4D-AC59-21F250381297} = {EF30B533-D715-421A-92B7-92FEF460AC9C}
|
||||
EndGlobalSection
|
||||
GlobalSection(ExtensibilityGlobals) = postSolution
|
||||
SolutionGuid = {DB4F868D-E1AE-4FD7-9333-69FA15B268C5}
|
||||
|
|
|
|||
24
LICENSE.txt
24
LICENSE.txt
|
|
@ -12,3 +12,27 @@ Unless required by applicable law or agreed to in writing, software distributed
|
|||
under the License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR
|
||||
CONDITIONS OF ANY KIND, either express or implied. See the License for the
|
||||
specific language governing permissions and limitations under the License.
|
||||
|
||||
ASP.NET Core Module
|
||||
|
||||
Copyright (c) .NET Foundation
|
||||
All rights reserved.
|
||||
|
||||
MIT License
|
||||
|
||||
Permission is hereby granted, free of charge, to any person obtaining a copy of
|
||||
this software and associated documentation files (the ""Software""), to deal in
|
||||
the Software without restriction, including without limitation the rights to
|
||||
use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of
|
||||
the Software, and to permit persons to whom the Software is furnished to do so,
|
||||
subject to the following conditions:
|
||||
|
||||
The above copyright notice and this permission notice shall be included in all
|
||||
copies or substantial portions of the Software.
|
||||
|
||||
THE SOFTWARE IS PROVIDED *AS IS*, WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
||||
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS
|
||||
FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR
|
||||
COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER
|
||||
IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
|
||||
CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
|
||||
|
|
@ -1,4 +1,10 @@
|
|||
{
|
||||
"adx-nonshipping": {
|
||||
"rules": [],
|
||||
"packages": {
|
||||
"Microsoft.AspNetCore.AspNetCoreModule": {}
|
||||
}
|
||||
},
|
||||
"Default": {
|
||||
"rules": [
|
||||
"DefaultCompositeRule"
|
||||
|
|
|
|||
|
|
@ -1,6 +1,3 @@
|
|||
ASP.NET Core IIS Integration
|
||||
========
|
||||
|
||||
This repo hosts the ASP.NET Core middleware for IIS integration.
|
||||
This repo hosts the ASP.NET Core middleware for IIS integration and the ASP.NET Core Module.
|
||||
|
||||
This project is part of ASP.NET Core. You can find samples, documentation and getting started instructions for ASP.NET Core at the [Home](https://github.com/aspnet/home) repo.
|
||||
|
|
|
|||
|
|
@ -0,0 +1,87 @@
|
|||
<?xml version="1.0" encoding="utf-8"?>
|
||||
<Project ToolsVersion="12.0" xmlns="http://schemas.microsoft.com/developer/msbuild/2003">
|
||||
<PropertyGroup>
|
||||
<SolutionDir Condition="$(SolutionDir) == '' Or $(SolutionDir) == '*Undefined*'">$(MSBuildThisFileDirectory)..\</SolutionDir>
|
||||
<Configuration Condition="'$(Configuration)'==''">Debug</Configuration>
|
||||
<Platform Condition="'$(Platform)' == ''">Win32</Platform>
|
||||
<PlatformToolset Condition=" '$(VisualStudioVersion)' == '12.0'">v120</PlatformToolset>
|
||||
<PlatformToolset Condition=" '$(VisualStudioVersion)' == '14.0'">v140</PlatformToolset>
|
||||
<PlatformToolset Condition=" '$(PlatformToolset)' == ''">v120</PlatformToolset>
|
||||
<OutputPath Condition="'$(OutputPath)' == ''">$(MSBuildProjectDirectory)\bin\$(Configuration)\$(Platform)\</OutputPath>
|
||||
<OutDir>$(OutputPath)</OutDir>
|
||||
<AspNetCoreModuleTargetName>aspnetcore</AspNetCoreModuleTargetName>
|
||||
</PropertyGroup>
|
||||
|
||||
<PropertyGroup Condition="'$(Configuration)' == 'Debug'">
|
||||
<UseDebugLibraries>true</UseDebugLibraries>
|
||||
<WholeProgramOptimization>false</WholeProgramOptimization>
|
||||
</PropertyGroup>
|
||||
|
||||
<PropertyGroup Condition="'$(Configuration)' == 'Release'">
|
||||
<UseDebugLibraries>false</UseDebugLibraries>
|
||||
<WholeProgramOptimization>true</WholeProgramOptimization>
|
||||
</PropertyGroup>
|
||||
|
||||
<ItemDefinitionGroup>
|
||||
<ClCompile>
|
||||
<PrecompiledHeader>Use</PrecompiledHeader>
|
||||
<TreatWarningAsError Condition="'$(TreatWarningsAsErrors)' != ''">true</TreatWarningAsError>
|
||||
<SDLCheck>true</SDLCheck>
|
||||
<StringPooling>true</StringPooling>
|
||||
</ClCompile>
|
||||
<Link>
|
||||
<SubSystem>Windows</SubSystem>
|
||||
<GenerateDebugInformation>true</GenerateDebugInformation>
|
||||
<ProgramDatabaseFile>$(OutDir)$(TargetName).pdb</ProgramDatabaseFile>
|
||||
<StripPrivateSymbols>$(OutDir)$(TargetName).pub.pdb</StripPrivateSymbols>
|
||||
<OptimizeReferences>true</OptimizeReferences>
|
||||
<EnableCOMDATFolding>true</EnableCOMDATFolding>
|
||||
<Profile>true</Profile>
|
||||
</Link>
|
||||
</ItemDefinitionGroup>
|
||||
|
||||
<ItemDefinitionGroup Condition="'$(Configuration)'=='Debug'">
|
||||
<ClCompile>
|
||||
<Optimization>Disabled</Optimization>
|
||||
</ClCompile>
|
||||
</ItemDefinitionGroup>
|
||||
|
||||
<ItemDefinitionGroup Condition="'$(Configuration)'=='Release'">
|
||||
<ClCompile>
|
||||
<Optimization>MaxSpeed</Optimization>
|
||||
<FunctionLevelLinking>true</FunctionLevelLinking>
|
||||
<IntrinsicFunctions>true</IntrinsicFunctions>
|
||||
</ClCompile>
|
||||
</ItemDefinitionGroup>
|
||||
|
||||
<ItemDefinitionGroup Condition="'$(Configuration)|$(Platform)'=='Debug|Win32'">
|
||||
<ClCompile>
|
||||
<PreprocessorDefinitions>WIN32;_DEBUG;%(PreprocessorDefinitions)</PreprocessorDefinitions>
|
||||
</ClCompile>
|
||||
</ItemDefinitionGroup>
|
||||
|
||||
<ItemDefinitionGroup Condition="'$(Configuration)|$(Platform)'=='Debug|x64'">
|
||||
<ClCompile>
|
||||
<PreprocessorDefinitions>_WIN64;_DEBUG;%(PreprocessorDefinitions)</PreprocessorDefinitions>
|
||||
</ClCompile>
|
||||
</ItemDefinitionGroup>
|
||||
|
||||
<ItemDefinitionGroup Condition="'$(Configuration)|$(Platform)'=='Release|Win32'">
|
||||
<ClCompile>
|
||||
<PreprocessorDefinitions>WIN32;NDEBUG;%(PreprocessorDefinitions)</PreprocessorDefinitions>
|
||||
</ClCompile>
|
||||
<Link>
|
||||
<GenerateDebugInformation>true</GenerateDebugInformation>
|
||||
</Link>
|
||||
</ItemDefinitionGroup>
|
||||
|
||||
<ItemDefinitionGroup Condition="'$(Configuration)|$(Platform)'=='Release|x64'">
|
||||
<ClCompile>
|
||||
<PreprocessorDefinitions>_WIN64;NDEBUG;%(PreprocessorDefinitions)</PreprocessorDefinitions>
|
||||
</ClCompile>
|
||||
<Link>
|
||||
<GenerateDebugInformation>true</GenerateDebugInformation>
|
||||
</Link>
|
||||
</ItemDefinitionGroup>
|
||||
|
||||
</Project>
|
||||
|
|
@ -0,0 +1,21 @@
|
|||
<?xml version="1.0" encoding="utf-8"?>
|
||||
<Project DefaultTargets="Build" ToolsVersion="12.0" xmlns="http://schemas.microsoft.com/developer/msbuild/2003">
|
||||
<ItemGroup Label="ProjectConfigurations">
|
||||
<ProjectConfiguration Include="Debug|Win32">
|
||||
<Configuration>Debug</Configuration>
|
||||
<Platform>Win32</Platform>
|
||||
</ProjectConfiguration>
|
||||
<ProjectConfiguration Include="Debug|x64">
|
||||
<Configuration>Debug</Configuration>
|
||||
<Platform>x64</Platform>
|
||||
</ProjectConfiguration>
|
||||
<ProjectConfiguration Include="Release|Win32">
|
||||
<Configuration>Release</Configuration>
|
||||
<Platform>Win32</Platform>
|
||||
</ProjectConfiguration>
|
||||
<ProjectConfiguration Include="Release|x64">
|
||||
<Configuration>Release</Configuration>
|
||||
<Platform>x64</Platform>
|
||||
</ProjectConfiguration>
|
||||
</ItemGroup>
|
||||
</Project>
|
||||
|
|
@ -0,0 +1,30 @@
|
|||
<?xml version="1.0" encoding="utf-8"?>
|
||||
<Project ToolsVersion="12.0" DefaultTargets="Test" xmlns="http://schemas.microsoft.com/developer/msbuild/2003">
|
||||
<Import Project="$(MSBuildThisFileDirectory)\Build.Settings" />
|
||||
<ItemGroup>
|
||||
<Projects Include="$(SolutionDir)\src\AspNetCore\AspNetCore.vcxproj" />
|
||||
<Projects Include="$(SolutionDir)\src\RequestHandler\RequestHandler.vcxproj" />
|
||||
</ItemGroup>
|
||||
|
||||
<Target Name="Build">
|
||||
<MSBuild Targets="$(BuildTargets)"
|
||||
Projects="@(Projects)"
|
||||
Properties="Configuration=$(Configuration);Platform=$(Platform);PlatformToolset=$(PlatformToolset)" />
|
||||
</Target>
|
||||
|
||||
<Target Name="Clean">
|
||||
<MSBuild Targets="Clean"
|
||||
Projects="@(Projects)" />
|
||||
</Target>
|
||||
|
||||
<Target Name="Rebuild">
|
||||
<MSBuild Targets="Clean;Build"
|
||||
Projects="$(MSBuildProjectFile)"
|
||||
Properties="BuildTargets=Rebuild;Configuration=$(Configuration);Platform=$(Platform);PlatformToolset=$(PlatformToolset)"/>
|
||||
</Target>
|
||||
|
||||
<Target Name="Test" DependsOnTargets="Build">
|
||||
<!-- once we have test project ready, we should add executions to run the test post build-->
|
||||
</Target>
|
||||
<Import Project="Config.Definitions.Props" />
|
||||
</Project>
|
||||
|
|
@ -1,35 +1,41 @@
|
|||
<Project>
|
||||
<Project>
|
||||
<PropertyGroup>
|
||||
<MSBuildAllProjects>$(MSBuildAllProjects);$(MSBuildThisFileFullPath)</MSBuildAllProjects>
|
||||
</PropertyGroup>
|
||||
<PropertyGroup Label="Package Versions">
|
||||
<InternalAspNetCoreSdkPackageVersion>2.1.0-preview1-15626</InternalAspNetCoreSdkPackageVersion>
|
||||
<InternalAspNetCoreSdkPackageVersion>2.1.0-preview1-15638</InternalAspNetCoreSdkPackageVersion>
|
||||
<MicrosoftAspNetCoreAllPackageVersion>2.1.0-preview1-27867</MicrosoftAspNetCoreAllPackageVersion>
|
||||
<MicrosoftAspNetCoreAspNetCoreModulePackageVersion>1.0.0-pre-10223</MicrosoftAspNetCoreAspNetCoreModulePackageVersion>
|
||||
<MicrosoftAspNetCoreAuthenticationCorePackageVersion>2.1.0-preview1-27849</MicrosoftAspNetCoreAuthenticationCorePackageVersion>
|
||||
<MicrosoftAspNetCoreHostingAbstractionsPackageVersion>2.1.0-preview1-27849</MicrosoftAspNetCoreHostingAbstractionsPackageVersion>
|
||||
<MicrosoftAspNetCoreHostingPackageVersion>2.1.0-preview1-27849</MicrosoftAspNetCoreHostingPackageVersion>
|
||||
<MicrosoftAspNetCoreHttpExtensionsPackageVersion>2.1.0-preview1-27849</MicrosoftAspNetCoreHttpExtensionsPackageVersion>
|
||||
<MicrosoftAspNetCoreHttpOverridesPackageVersion>2.1.0-preview1-27849</MicrosoftAspNetCoreHttpOverridesPackageVersion>
|
||||
<MicrosoftAspNetCoreHttpPackageVersion>2.1.0-preview1-27849</MicrosoftAspNetCoreHttpPackageVersion>
|
||||
<MicrosoftAspNetCoreHttpSysSourcesPackageVersion>2.1.0-preview1-27849</MicrosoftAspNetCoreHttpSysSourcesPackageVersion>
|
||||
<MicrosoftAspNetCoreServerIntegrationTestingPackageVersion>0.5.0-preview1-27849</MicrosoftAspNetCoreServerIntegrationTestingPackageVersion>
|
||||
<MicrosoftAspNetCoreServerKestrelPackageVersion>2.1.0-preview1-27849</MicrosoftAspNetCoreServerKestrelPackageVersion>
|
||||
<MicrosoftAspNetCoreTestHostPackageVersion>2.1.0-preview1-27849</MicrosoftAspNetCoreTestHostPackageVersion>
|
||||
<MicrosoftAspNetCoreWebUtilitiesPackageVersion>2.1.0-preview1-27849</MicrosoftAspNetCoreWebUtilitiesPackageVersion>
|
||||
<MicrosoftExtensionsConfigurationEnvironmentVariablesPackageVersion>2.1.0-preview1-27849</MicrosoftExtensionsConfigurationEnvironmentVariablesPackageVersion>
|
||||
<MicrosoftExtensionsConfigurationJsonPackageVersion>2.1.0-preview1-27849</MicrosoftExtensionsConfigurationJsonPackageVersion>
|
||||
<MicrosoftExtensionsLoggingAbstractionsPackageVersion>2.1.0-preview1-27849</MicrosoftExtensionsLoggingAbstractionsPackageVersion>
|
||||
<MicrosoftExtensionsLoggingConsolePackageVersion>2.1.0-preview1-27849</MicrosoftExtensionsLoggingConsolePackageVersion>
|
||||
<MicrosoftExtensionsLoggingDebugPackageVersion>2.1.0-preview1-27849</MicrosoftExtensionsLoggingDebugPackageVersion>
|
||||
<MicrosoftExtensionsLoggingPackageVersion>2.1.0-preview1-27849</MicrosoftExtensionsLoggingPackageVersion>
|
||||
<MicrosoftExtensionsLoggingTestingPackageVersion>2.1.0-preview1-27849</MicrosoftExtensionsLoggingTestingPackageVersion>
|
||||
<MicrosoftExtensionsOptionsPackageVersion>2.1.0-preview1-27849</MicrosoftExtensionsOptionsPackageVersion>
|
||||
<MicrosoftExtensionsSecurityHelperSourcesPackageVersion>2.1.0-preview1-27849</MicrosoftExtensionsSecurityHelperSourcesPackageVersion>
|
||||
<MicrosoftAspNetCoreAuthenticationCorePackageVersion>2.1.0-preview1-27867</MicrosoftAspNetCoreAuthenticationCorePackageVersion>
|
||||
<MicrosoftAspNetCoreHostingAbstractionsPackageVersion>2.1.0-preview1-27867</MicrosoftAspNetCoreHostingAbstractionsPackageVersion>
|
||||
<MicrosoftAspNetCoreHostingPackageVersion>2.1.0-preview1-27867</MicrosoftAspNetCoreHostingPackageVersion>
|
||||
<MicrosoftAspNetCoreHttpExtensionsPackageVersion>2.1.0-preview1-27867</MicrosoftAspNetCoreHttpExtensionsPackageVersion>
|
||||
<MicrosoftAspNetCoreHttpOverridesPackageVersion>2.1.0-preview1-27867</MicrosoftAspNetCoreHttpOverridesPackageVersion>
|
||||
<MicrosoftAspNetCoreHttpPackageVersion>2.1.0-preview1-27867</MicrosoftAspNetCoreHttpPackageVersion>
|
||||
<MicrosoftAspNetCoreHttpSysSourcesPackageVersion>2.1.0-preview1-27867</MicrosoftAspNetCoreHttpSysSourcesPackageVersion>
|
||||
<MicrosoftAspNetCoreServerIntegrationTestingPackageVersion>0.5.0-preview1-27867</MicrosoftAspNetCoreServerIntegrationTestingPackageVersion>
|
||||
<MicrosoftAspNetCoreServerKestrelPackageVersion>2.1.0-preview1-27867</MicrosoftAspNetCoreServerKestrelPackageVersion>
|
||||
<MicrosoftAspNetCoreTestHostPackageVersion>2.1.0-preview1-27867</MicrosoftAspNetCoreTestHostPackageVersion>
|
||||
<MicrosoftAspNetCoreTestingPackageVersion>2.1.0-preview1-27867</MicrosoftAspNetCoreTestingPackageVersion>
|
||||
<MicrosoftAspNetCoreWebUtilitiesPackageVersion>2.1.0-preview1-27867</MicrosoftAspNetCoreWebUtilitiesPackageVersion>
|
||||
<MicrosoftExtensionsConfigurationEnvironmentVariablesPackageVersion>2.1.0-preview1-27867</MicrosoftExtensionsConfigurationEnvironmentVariablesPackageVersion>
|
||||
<MicrosoftExtensionsConfigurationJsonPackageVersion>2.1.0-preview1-27867</MicrosoftExtensionsConfigurationJsonPackageVersion>
|
||||
<MicrosoftExtensionsLoggingAbstractionsPackageVersion>2.1.0-preview1-27867</MicrosoftExtensionsLoggingAbstractionsPackageVersion>
|
||||
<MicrosoftExtensionsLoggingConsolePackageVersion>2.1.0-preview1-27867</MicrosoftExtensionsLoggingConsolePackageVersion>
|
||||
<MicrosoftExtensionsLoggingDebugPackageVersion>2.1.0-preview1-27867</MicrosoftExtensionsLoggingDebugPackageVersion>
|
||||
<MicrosoftExtensionsLoggingPackageVersion>2.1.0-preview1-27867</MicrosoftExtensionsLoggingPackageVersion>
|
||||
<MicrosoftExtensionsLoggingTestingPackageVersion>2.1.0-preview1-27867</MicrosoftExtensionsLoggingTestingPackageVersion>
|
||||
<MicrosoftExtensionsOptionsPackageVersion>2.1.0-preview1-27867</MicrosoftExtensionsOptionsPackageVersion>
|
||||
<MicrosoftExtensionsPlatformAbstractionsPackageVersion>1.1.0</MicrosoftExtensionsPlatformAbstractionsPackageVersion>
|
||||
<MicrosoftExtensionsSecurityHelperSourcesPackageVersion>2.1.0-preview1-27867</MicrosoftExtensionsSecurityHelperSourcesPackageVersion>
|
||||
<MicrosoftNETCoreApp20PackageVersion>2.0.0</MicrosoftNETCoreApp20PackageVersion>
|
||||
<MicrosoftNETCoreApp21PackageVersion>2.1.0-preview1-26016-05</MicrosoftNETCoreApp21PackageVersion>
|
||||
<MicrosoftNetHttpHeadersPackageVersion>2.1.0-preview1-27867</MicrosoftNetHttpHeadersPackageVersion>
|
||||
<MicrosoftNETTestSdkPackageVersion>15.3.0</MicrosoftNETTestSdkPackageVersion>
|
||||
<MicrosoftWebAdministrationPackageVersion>11.0.0</MicrosoftWebAdministrationPackageVersion>
|
||||
<SystemBuffersPackageVersion>4.5.0-preview1-26016-05</SystemBuffersPackageVersion>
|
||||
<SystemIOPipelinesPackageVersion>0.1.0-e171215-1</SystemIOPipelinesPackageVersion>
|
||||
<SystemManagementAutomationPackageVersion>6.1.7601.17515</SystemManagementAutomationPackageVersion>
|
||||
<SystemMemoryPackageVersion>4.5.0-preview1-26016-05</SystemMemoryPackageVersion>
|
||||
<SystemNumericsVectorsPackageVersion>4.5.0-preview1-26016-05</SystemNumericsVectorsPackageVersion>
|
||||
<SystemRuntimeCompilerServicesUnsafePackageVersion>4.5.0-preview1-26016-05</SystemRuntimeCompilerServicesUnsafePackageVersion>
|
||||
|
|
|
|||
|
|
@ -1,4 +1,4 @@
|
|||
<Project>
|
||||
<Project>
|
||||
<Import Project="dependencies.props" />
|
||||
|
||||
<ItemGroup>
|
||||
|
|
@ -6,7 +6,11 @@
|
|||
<ExcludeFromTest Include="$(RepositoryRoot)test\IISTestSite\*.csproj" />
|
||||
<ExcludeFromTest Include="$(RepositoryRoot)test\IISIntegration.FunctionalTests\*.csproj" Condition="'$(OS)' != 'Windows_NT'" />
|
||||
<ExcludeFromTest Include="$(RepositoryRoot)test\IISIntegration.IISServerFunctionalTests\*.csproj" Condition="'$(OS)' != 'Windows_NT'" />
|
||||
<ExcludeFromTest Include="$(RepositoryRoot)test\AspNetCoreModule.TestSites.Standard\*.csproj" />
|
||||
<ExcludeFromTest Include="$(RepositoryRoot)test\WebSocketClientEXE\*.csproj" />
|
||||
<ExcludeFromTest Include="$(RepositoryRoot)test\AspNetCoreModule.Test\*.csproj" Condition="'$(OS)' != 'Windows_NT'" />
|
||||
</ItemGroup>
|
||||
|
||||
<PropertyGroup>
|
||||
<!-- These properties are use by the automation that updates dependencies.props -->
|
||||
<LineupPackageId>Internal.AspNetCore.Universe.Lineup</LineupPackageId>
|
||||
|
|
|
|||
|
|
@ -0,0 +1,26 @@
|
|||
<Project>
|
||||
|
||||
<Target Name="BuildNativeAssets" DependsOnTargets="GetToolsets" BeforeTargets="Compile" >
|
||||
<ItemGroup>
|
||||
<BuildConfigurations Include="/p:Configuration=Release /p:platform=Win32" />
|
||||
<BuildConfigurations Include="/p:Configuration=Release /p:platform=x64" />
|
||||
</ItemGroup>
|
||||
|
||||
<Error
|
||||
Text="Could not find an installation of Visual Studio with the C++ development tools."
|
||||
Condition="'$(VisualStudioMSBuildx86Path)' == ''" />
|
||||
|
||||
<Exec Command=""$(VisualStudioMSBuildx86Path)" "$(RepositoryRoot)src\AspNetCore\AspNetCore.vcxproj" %(BuildConfigurations.Identity)"
|
||||
Condition="'$(VisualStudioMSBuildx86Path)' != ''" />
|
||||
<Exec Command=""$(VisualStudioMSBuildx86Path)" "$(RepositoryRoot)src\RequestHandler\RequestHandler.vcxproj" %(BuildConfigurations.Identity)"
|
||||
Condition="'$(VisualStudioMSBuildx86Path)' != ''" />
|
||||
</Target>
|
||||
|
||||
<Target Name="PackageProjects">
|
||||
<MSBuild
|
||||
Projects="$(MSBuildThisFileDirectory)../nuget/AspNetCore.csproj"
|
||||
Targets="Restore;Pack"
|
||||
Properties="PackageVersion=$(PackageVersion);PackageOutputPath=$(BuildDir)" />
|
||||
</Target>
|
||||
|
||||
</Project>
|
||||
|
|
@ -1,4 +1,4 @@
|
|||
<Project>
|
||||
<Project>
|
||||
<Import Project="$(DotNetRestoreSourcePropsPath)" Condition="'$(DotNetRestoreSourcePropsPath)' != ''"/>
|
||||
|
||||
<PropertyGroup Label="RestoreSources">
|
||||
|
|
|
|||
|
|
@ -1,2 +1,2 @@
|
|||
version:2.1.0-preview1-15626
|
||||
commithash:fd6410e9c90c428bc01238372303ad09cb9ec889
|
||||
version:2.1.0-preview1-15638
|
||||
commithash:1d3a0c725dc6b8ae6b0e47800fd6b4d8f8b8d545
|
||||
|
|
|
|||
|
|
@ -1,4 +1,14 @@
|
|||
{
|
||||
"$schema": "https://raw.githubusercontent.com/aspnet/BuildTools/dev/tools/korebuild.schema.json",
|
||||
"channel": "dev"
|
||||
}
|
||||
"$schema": "https://raw.githubusercontent.com/aspnet/BuildTools/dev/tools/korebuild.schema.json",
|
||||
"channel": "dev",
|
||||
"toolsets": {
|
||||
"visualstudio": {
|
||||
"required": ["Windows"],
|
||||
"includePrerelease": true,
|
||||
"minVersion": "15.0.26730.03",
|
||||
"requiredWorkloads": [
|
||||
"Microsoft.VisualStudio.Component.VC.Tools.x86.x64"
|
||||
]
|
||||
}
|
||||
}
|
||||
}
|
||||
|
|
@ -0,0 +1,9 @@
|
|||
<Project Sdk="Microsoft.NET.Sdk">
|
||||
|
||||
<PropertyGroup>
|
||||
<TargetFramework>netstandard1.0</TargetFramework>
|
||||
<NuspecFile>$(MSBuildThisFileDirectory)AspNetCore.nuspec</NuspecFile>
|
||||
<NuspecProperties>version=$(PackageVersion)</NuspecProperties>
|
||||
</PropertyGroup>
|
||||
|
||||
</Project>
|
||||
|
|
@ -0,0 +1,30 @@
|
|||
<?xml version="1.0"?>
|
||||
<package xmlns="http://schemas.microsoft.com/packaging/2013/05/nuspec.xsd">
|
||||
<metadata>
|
||||
<id>Microsoft.AspNetCore.AspNetCoreModule</id>
|
||||
<title>Microsoft ASP.NET Core Module</title>
|
||||
<version>$VERSION$</version>
|
||||
<authors>Microsoft</authors>
|
||||
<owners>Microsoft</owners>
|
||||
<licenseUrl>http://www.microsoft.com/web/webpi/eula/net_library_eula_ENU.htm</licenseUrl>
|
||||
<copyright>© .NET Foundation. All rights reserved.</copyright>
|
||||
<projectUrl>http://www.asp.net/</projectUrl>
|
||||
<requireLicenseAcceptance>true</requireLicenseAcceptance>
|
||||
<description>ASP.NET Core Module</description>
|
||||
<language>en-US</language>
|
||||
<tags>Microsoft.AspNetCore.AspNetCoreModule</tags>
|
||||
<contentFiles>
|
||||
<files include="any/any/*/*.dll" buildAction="None" copyToOutput="true" flatten="false" />
|
||||
</contentFiles>
|
||||
</metadata>
|
||||
<files>
|
||||
<file src="..\artifacts\build\AspNetCore\bin\Release\Win32\aspnetcore.dll" target="contentFiles\any\any\x86\aspnetcore.dll" />
|
||||
<file src="..\artifacts\build\AspNetCore\bin\Release\x64\aspnetcore.dll" target="contentFiles\any\any\x64\aspnetcore.dll" />
|
||||
<file src="..\artifacts\build\AspNetCore\bin\Release\Win32\aspnetcorerh.dll" target="contentFiles\any\any\x86\aspnetcorerh.dll" />
|
||||
<file src="..\artifacts\build\AspNetCore\bin\Release\x64\aspnetcorerh.dll" target="contentFiles\any\any\x64\aspnetcorerh.dll" />
|
||||
<file src="..\artifacts\build\AspNetCore\bin\Release\x64\*.xml"/>
|
||||
<file src="..\tools\installancm.ps1"/>
|
||||
<file src="..\LICENSE.txt"/>
|
||||
<file src="Microsoft.AspNetCore.AspNetCoreModule.props" target="build\" />
|
||||
</files>
|
||||
</package>
|
||||
|
|
@ -0,0 +1,10 @@
|
|||
<Project>
|
||||
|
||||
<PropertyGroup>
|
||||
<AspNetCoreModuleX64Location>$(MSBuildThisFileDirectory)..\contentFiles\any\any\x64\aspnetcore.dll</AspNetCoreModuleX64Location>
|
||||
<AspNetCoreModuleX86Location>$(MSBuildThisFileDirectory)..\contentFiles\any\any\x86\aspnetcore.dll</AspNetCoreModuleX86Location>
|
||||
<RequestHandlerX64Location>$(MSBuildThisFileDirectory)..\contentFiles\any\any\x64\aspnetcorerh.dll</RequestHandlerX64Location>
|
||||
<RequestHandlerX86Location>$(MSBuildThisFileDirectory)..\contentFiles\any\any\x86\aspnetcorerh.dll</RequestHandlerX86Location>
|
||||
</PropertyGroup>
|
||||
|
||||
</Project>
|
||||
2
run.ps1
2
run.ps1
|
|
@ -193,4 +193,4 @@ try {
|
|||
}
|
||||
finally {
|
||||
Remove-Module 'KoreBuild' -ErrorAction Ignore
|
||||
}
|
||||
}
|
||||
|
|
@ -0,0 +1,267 @@
|
|||
<?xml version="1.0" encoding="utf-8"?>
|
||||
<Project DefaultTargets="Build" ToolsVersion="15.0" xmlns="http://schemas.microsoft.com/developer/msbuild/2003">
|
||||
<Import Project="..\..\Build\Build.Settings" />
|
||||
<ItemGroup Label="ProjectConfigurations">
|
||||
<ProjectConfiguration Include="Debug|Win32">
|
||||
<Configuration>Debug</Configuration>
|
||||
<Platform>Win32</Platform>
|
||||
</ProjectConfiguration>
|
||||
<ProjectConfiguration Include="Debug|x64">
|
||||
<Configuration>Debug</Configuration>
|
||||
<Platform>x64</Platform>
|
||||
</ProjectConfiguration>
|
||||
<ProjectConfiguration Include="Release|Win32">
|
||||
<Configuration>Release</Configuration>
|
||||
<Platform>Win32</Platform>
|
||||
</ProjectConfiguration>
|
||||
<ProjectConfiguration Include="Release|x64">
|
||||
<Configuration>Release</Configuration>
|
||||
<Platform>x64</Platform>
|
||||
</ProjectConfiguration>
|
||||
</ItemGroup>
|
||||
<PropertyGroup Label="Globals">
|
||||
<ProjectGuid>{439824F9-1455-4CC4-BD79-B44FA0A16552}</ProjectGuid>
|
||||
<Keyword>Win32Proj</Keyword>
|
||||
<RootNamespace>AspNetCoreModule</RootNamespace>
|
||||
<ProjectName>AspNetCore</ProjectName>
|
||||
<TargetName>aspnetcore</TargetName>
|
||||
<LinkIncremental>false</LinkIncremental>
|
||||
<WindowsTargetPlatformVersion>10.0.15063.0</WindowsTargetPlatformVersion>
|
||||
</PropertyGroup>
|
||||
<Import Project="$(VCTargetsPath)\Microsoft.Cpp.Default.props" />
|
||||
<PropertyGroup Condition="'$(Configuration)|$(Platform)'=='Debug|Win32'" Label="Configuration">
|
||||
<ConfigurationType>DynamicLibrary</ConfigurationType>
|
||||
<UseDebugLibraries>true</UseDebugLibraries>
|
||||
<PlatformToolset>v141</PlatformToolset>
|
||||
<CharacterSet>Unicode</CharacterSet>
|
||||
</PropertyGroup>
|
||||
<PropertyGroup Condition="'$(Configuration)|$(Platform)'=='Debug|x64'" Label="Configuration">
|
||||
<ConfigurationType>DynamicLibrary</ConfigurationType>
|
||||
<UseDebugLibraries>true</UseDebugLibraries>
|
||||
<PlatformToolset>v141</PlatformToolset>
|
||||
<CharacterSet>Unicode</CharacterSet>
|
||||
</PropertyGroup>
|
||||
<PropertyGroup Condition="'$(Configuration)|$(Platform)'=='Release|Win32'" Label="Configuration">
|
||||
<ConfigurationType>DynamicLibrary</ConfigurationType>
|
||||
<UseDebugLibraries>false</UseDebugLibraries>
|
||||
<PlatformToolset>v141</PlatformToolset>
|
||||
<WholeProgramOptimization>true</WholeProgramOptimization>
|
||||
<CharacterSet>Unicode</CharacterSet>
|
||||
</PropertyGroup>
|
||||
<PropertyGroup Condition="'$(Configuration)|$(Platform)'=='Release|x64'" Label="Configuration">
|
||||
<ConfigurationType>DynamicLibrary</ConfigurationType>
|
||||
<UseDebugLibraries>false</UseDebugLibraries>
|
||||
<PlatformToolset>v141</PlatformToolset>
|
||||
<WholeProgramOptimization>true</WholeProgramOptimization>
|
||||
<CharacterSet>Unicode</CharacterSet>
|
||||
</PropertyGroup>
|
||||
<Import Project="$(VCTargetsPath)\Microsoft.Cpp.props" />
|
||||
<ImportGroup Label="ExtensionSettings">
|
||||
</ImportGroup>
|
||||
<ImportGroup Label="PropertySheets" Condition="'$(Configuration)|$(Platform)'=='Debug|Win32'">
|
||||
<Import Project="$(UserRootDir)\Microsoft.Cpp.$(Platform).user.props" Condition="exists('$(UserRootDir)\Microsoft.Cpp.$(Platform).user.props')" Label="LocalAppDataPlatform" />
|
||||
</ImportGroup>
|
||||
<ImportGroup Condition="'$(Configuration)|$(Platform)'=='Debug|x64'" Label="PropertySheets">
|
||||
<Import Project="$(UserRootDir)\Microsoft.Cpp.$(Platform).user.props" Condition="exists('$(UserRootDir)\Microsoft.Cpp.$(Platform).user.props')" Label="LocalAppDataPlatform" />
|
||||
</ImportGroup>
|
||||
<ImportGroup Label="PropertySheets" Condition="'$(Configuration)|$(Platform)'=='Release|Win32'">
|
||||
<Import Project="$(UserRootDir)\Microsoft.Cpp.$(Platform).user.props" Condition="exists('$(UserRootDir)\Microsoft.Cpp.$(Platform).user.props')" Label="LocalAppDataPlatform" />
|
||||
</ImportGroup>
|
||||
<ImportGroup Condition="'$(Configuration)|$(Platform)'=='Release|x64'" Label="PropertySheets">
|
||||
<Import Project="$(UserRootDir)\Microsoft.Cpp.$(Platform).user.props" Condition="exists('$(UserRootDir)\Microsoft.Cpp.$(Platform).user.props')" Label="LocalAppDataPlatform" />
|
||||
</ImportGroup>
|
||||
<PropertyGroup Label="UserMacros" />
|
||||
<PropertyGroup Condition="'$(Configuration)'=='Release'">
|
||||
<OutDir>$(SolutionDir)artifacts\build\$(ProjectName)\bin\$(Configuration)\$(Platform)\</OutDir>
|
||||
</PropertyGroup>
|
||||
<PropertyGroup Condition="'$(Configuration)|$(Platform)'=='Debug|Win32'">
|
||||
<OutDir>$(SolutionDir)artifacts\build\$(ProjectName)\bin\$(Configuration)\$(Platform)\</OutDir>
|
||||
</PropertyGroup>
|
||||
<PropertyGroup Condition="'$(Configuration)|$(Platform)'=='Debug|x64'">
|
||||
<OutDir>$(SolutionDir)artifacts\build\$(ProjectName)\bin\$(Configuration)\$(Platform)\</OutDir>
|
||||
</PropertyGroup>
|
||||
<ItemDefinitionGroup Condition="'$(Configuration)|$(Platform)'=='Debug|Win32'">
|
||||
<ClCompile>
|
||||
<PrecompiledHeader>NotUsing</PrecompiledHeader>
|
||||
<WarningLevel>Level4</WarningLevel>
|
||||
<Optimization>Disabled</Optimization>
|
||||
<PreprocessorDefinitions>WIN32;_DEBUG;_WINDOWS;_USRDLL;ASPNETCOREMODULE_EXPORTS;%(PreprocessorDefinitions)</PreprocessorDefinitions>
|
||||
<PrecompiledHeaderFile>precomp.hxx</PrecompiledHeaderFile>
|
||||
<PrecompiledHeaderOutputFile>$(IntDir)$(TargetName).pch</PrecompiledHeaderOutputFile>
|
||||
<AdditionalIncludeDirectories>..\IISLib;.\Inc</AdditionalIncludeDirectories>
|
||||
<DebugInformationFormat>ProgramDatabase</DebugInformationFormat>
|
||||
<RuntimeLibrary>MultiThreadedDebug</RuntimeLibrary>
|
||||
<TreatWarningAsError>true</TreatWarningAsError>
|
||||
<SDLCheck>true</SDLCheck>
|
||||
<WholeProgramOptimization>true</WholeProgramOptimization>
|
||||
<PreprocessKeepComments>false</PreprocessKeepComments>
|
||||
<ExceptionHandling>SyncCThrow</ExceptionHandling>
|
||||
<StructMemberAlignment>8Bytes</StructMemberAlignment>
|
||||
<FunctionLevelLinking>true</FunctionLevelLinking>
|
||||
<RuntimeTypeInfo>false</RuntimeTypeInfo>
|
||||
<OmitDefaultLibName>true</OmitDefaultLibName>
|
||||
<CompileAs>CompileAsCpp</CompileAs>
|
||||
<IntrinsicFunctions>true</IntrinsicFunctions>
|
||||
</ClCompile>
|
||||
<Link>
|
||||
<SubSystem>Windows</SubSystem>
|
||||
<GenerateDebugInformation>true</GenerateDebugInformation>
|
||||
<AdditionalDependencies>kernel32.lib;user32.lib;advapi32.lib;ole32.lib;oleaut32.lib;uuid.lib;odbc32.lib;odbccp32.lib;ahadmin.lib;rpcrt4.lib;winhttp.lib;pdh.lib;ws2_32.lib;wbemuuid.lib;iphlpapi.lib;%(AdditionalDependencies)</AdditionalDependencies>
|
||||
<ModuleDefinitionFile>Source.def</ModuleDefinitionFile>
|
||||
</Link>
|
||||
</ItemDefinitionGroup>
|
||||
<ItemDefinitionGroup Condition="'$(Configuration)|$(Platform)'=='Debug|x64'">
|
||||
<ClCompile>
|
||||
<PrecompiledHeader>NotUsing</PrecompiledHeader>
|
||||
<WarningLevel>Level4</WarningLevel>
|
||||
<Optimization>Disabled</Optimization>
|
||||
<PreprocessorDefinitions>WIN32;_DEBUG;_WINDOWS;_USRDLL;ASPNETCOREMODULE_EXPORTS;%(PreprocessorDefinitions)</PreprocessorDefinitions>
|
||||
<PrecompiledHeaderFile>precomp.hxx</PrecompiledHeaderFile>
|
||||
<PrecompiledHeaderOutputFile>$(IntDir)$(TargetName).pch</PrecompiledHeaderOutputFile>
|
||||
<AdditionalIncludeDirectories>..\IISLib;.\Inc</AdditionalIncludeDirectories>
|
||||
<DebugInformationFormat>ProgramDatabase</DebugInformationFormat>
|
||||
<RuntimeLibrary>MultiThreadedDebug</RuntimeLibrary>
|
||||
<TreatWarningAsError>true</TreatWarningAsError>
|
||||
<SDLCheck>true</SDLCheck>
|
||||
<WholeProgramOptimization>true</WholeProgramOptimization>
|
||||
<PreprocessKeepComments>false</PreprocessKeepComments>
|
||||
<ExceptionHandling>SyncCThrow</ExceptionHandling>
|
||||
<StructMemberAlignment>8Bytes</StructMemberAlignment>
|
||||
<FunctionLevelLinking>true</FunctionLevelLinking>
|
||||
<RuntimeTypeInfo>false</RuntimeTypeInfo>
|
||||
<OmitDefaultLibName>true</OmitDefaultLibName>
|
||||
<CompileAs>CompileAsCpp</CompileAs>
|
||||
<IntrinsicFunctions>true</IntrinsicFunctions>
|
||||
</ClCompile>
|
||||
<Link>
|
||||
<SubSystem>Windows</SubSystem>
|
||||
<GenerateDebugInformation>true</GenerateDebugInformation>
|
||||
<AdditionalDependencies>kernel32.lib;user32.lib;advapi32.lib;ole32.lib;oleaut32.lib;uuid.lib;odbc32.lib;odbccp32.lib;ahadmin.lib;rpcrt4.lib;winhttp.lib;pdh.lib;ws2_32.lib;wbemuuid.lib;iphlpapi.lib;%(AdditionalDependencies)</AdditionalDependencies>
|
||||
<ModuleDefinitionFile>Source.def</ModuleDefinitionFile>
|
||||
</Link>
|
||||
</ItemDefinitionGroup>
|
||||
<ItemDefinitionGroup Condition="'$(Configuration)|$(Platform)'=='Release|Win32'">
|
||||
<ClCompile>
|
||||
<WarningLevel>Level4</WarningLevel>
|
||||
<PrecompiledHeader>NotUsing</PrecompiledHeader>
|
||||
<Optimization>MaxSpeed</Optimization>
|
||||
<FunctionLevelLinking>true</FunctionLevelLinking>
|
||||
<IntrinsicFunctions>true</IntrinsicFunctions>
|
||||
<PreprocessorDefinitions>WIN32;NDEBUG;_WINDOWS;_USRDLL;ASPNETCOREMODULE_EXPORTS;%(PreprocessorDefinitions)</PreprocessorDefinitions>
|
||||
<AdditionalIncludeDirectories>..\IISLib;inc</AdditionalIncludeDirectories>
|
||||
<PrecompiledHeaderFile>precomp.hxx</PrecompiledHeaderFile>
|
||||
<RuntimeLibrary>MultiThreaded</RuntimeLibrary>
|
||||
<TreatWarningAsError>true</TreatWarningAsError>
|
||||
<SDLCheck>true</SDLCheck>
|
||||
<WholeProgramOptimization>true</WholeProgramOptimization>
|
||||
<PreprocessKeepComments>false</PreprocessKeepComments>
|
||||
<ExceptionHandling>SyncCThrow</ExceptionHandling>
|
||||
<StructMemberAlignment>8Bytes</StructMemberAlignment>
|
||||
<FunctionLevelLinking>true</FunctionLevelLinking>
|
||||
<RuntimeTypeInfo>false</RuntimeTypeInfo>
|
||||
<OmitDefaultLibName>true</OmitDefaultLibName>
|
||||
<CompileAs>CompileAsCpp</CompileAs>
|
||||
<IntrinsicFunctions>true</IntrinsicFunctions>
|
||||
</ClCompile>
|
||||
<Link>
|
||||
<SubSystem>Windows</SubSystem>
|
||||
<GenerateDebugInformation>false</GenerateDebugInformation>
|
||||
<EnableCOMDATFolding>true</EnableCOMDATFolding>
|
||||
<OptimizeReferences>true</OptimizeReferences>
|
||||
<ModuleDefinitionFile>Source.def</ModuleDefinitionFile>
|
||||
<AdditionalDependencies>kernel32.lib;user32.lib;gdi32.lib;winspool.lib;comdlg32.lib;advapi32.lib;shell32.lib;ole32.lib;oleaut32.lib;uuid.lib;ahadmin.lib;winhttp.lib;odbc32.lib;ws2_32.lib;odbccp32.lib;wbemuuid.lib;iphlpapi.lib;pdh.lib;rpcrt4.lib;%(AdditionalDependencies)</AdditionalDependencies>
|
||||
</Link>
|
||||
</ItemDefinitionGroup>
|
||||
<ItemDefinitionGroup Condition="'$(Configuration)|$(Platform)'=='Release|x64'">
|
||||
<ClCompile>
|
||||
<WarningLevel>Level4</WarningLevel>
|
||||
<PrecompiledHeader>NotUsing</PrecompiledHeader>
|
||||
<Optimization>MaxSpeed</Optimization>
|
||||
<FunctionLevelLinking>true</FunctionLevelLinking>
|
||||
<IntrinsicFunctions>true</IntrinsicFunctions>
|
||||
<PreprocessorDefinitions>WIN32;NDEBUG;_WINDOWS;_USRDLL;ASPNETCOREMODULE_EXPORTS;%(PreprocessorDefinitions)</PreprocessorDefinitions>
|
||||
<PrecompiledHeaderFile>precomp.hxx</PrecompiledHeaderFile>
|
||||
<AdditionalIncludeDirectories>..\IISLib;inc</AdditionalIncludeDirectories>
|
||||
<RuntimeLibrary>MultiThreaded</RuntimeLibrary>
|
||||
<TreatWarningAsError>true</TreatWarningAsError>
|
||||
<SDLCheck>true</SDLCheck>
|
||||
<WholeProgramOptimization>true</WholeProgramOptimization>
|
||||
<PreprocessKeepComments>false</PreprocessKeepComments>
|
||||
<ExceptionHandling>SyncCThrow</ExceptionHandling>
|
||||
<StructMemberAlignment>8Bytes</StructMemberAlignment>
|
||||
<FunctionLevelLinking>true</FunctionLevelLinking>
|
||||
<RuntimeTypeInfo>false</RuntimeTypeInfo>
|
||||
<OmitDefaultLibName>true</OmitDefaultLibName>
|
||||
<CompileAs>CompileAsCpp</CompileAs>
|
||||
<IntrinsicFunctions>true</IntrinsicFunctions>
|
||||
</ClCompile>
|
||||
<Link>
|
||||
<SubSystem>Windows</SubSystem>
|
||||
<GenerateDebugInformation>false</GenerateDebugInformation>
|
||||
<EnableCOMDATFolding>true</EnableCOMDATFolding>
|
||||
<OptimizeReferences>true</OptimizeReferences>
|
||||
<ModuleDefinitionFile>Source.def</ModuleDefinitionFile>
|
||||
<AdditionalDependencies>kernel32.lib;user32.lib;gdi32.lib;winspool.lib;comdlg32.lib;advapi32.lib;shell32.lib;ole32.lib;oleaut32.lib;uuid.lib;ahadmin.lib;rpcrt4.lib;winhttp.lib;pdh.lib;ws2_32.lib;wbemuuid.lib;iphlpapi.lib;%(AdditionalDependencies)</AdditionalDependencies>
|
||||
</Link>
|
||||
</ItemDefinitionGroup>
|
||||
<ItemGroup>
|
||||
<ClInclude Include="Inc\applicationinfo.h" />
|
||||
<ClInclude Include="Inc\appoffline.h" />
|
||||
<ClInclude Include="inc\globalmodule.h" />
|
||||
<ClInclude Include="Inc\resource.h" />
|
||||
<ClInclude Include="Inc\applicationmanager.h" />
|
||||
<ClInclude Include="Inc\filewatcher.h" />
|
||||
<ClInclude Include="Inc\proxymodule.h" />
|
||||
<ClInclude Include="Src\precomp.hxx" />
|
||||
</ItemGroup>
|
||||
<ItemGroup>
|
||||
<ClCompile Include="Src\applicationinfo.cpp" />
|
||||
<ClCompile Include="Src\applicationmanager.cxx" />
|
||||
<ClCompile Include="Src\dllmain.cpp" />
|
||||
<ClCompile Include="Src\filewatcher.cxx" />
|
||||
<ClCompile Include="src\globalmodule.cpp" />
|
||||
<ClCompile Include="Src\proxymodule.cxx" />
|
||||
</ItemGroup>
|
||||
<Target Name="CreateVersionHeader" BeforeTargets="PrepareForBuild">
|
||||
<PropertyGroup>
|
||||
<Revision>$(build_number)</Revision>
|
||||
<Revision Condition="'$(Revision)' == ''">0</Revision>
|
||||
</PropertyGroup>
|
||||
<ItemGroup>
|
||||
<VersionHeaderContents Include="// Copyright (c) .NET Foundation. All rights reserved." />
|
||||
<VersionHeaderContents Include="// Licensed under the MIT License. See LICENSE.txt in the project root for license information." />
|
||||
<VersionHeaderContents Include="%0a" />
|
||||
<VersionHeaderContents Include="// This file is auto-generated" />
|
||||
<VersionHeaderContents Include="%0a" />
|
||||
<VersionHeaderContents Include="#define FileVersion $(AspNetCoreModuleVersionMajor),$(AspNetCoreModuleVersionMinor),$(AspNetCoreModuleVersionPatch),$(Revision)" />
|
||||
<VersionHeaderContents Include="#define FileVersionStr "$(AspNetCoreModuleVersionMajor).$(AspNetCoreModuleVersionMinor).$(AspNetCoreModuleVersionPatch).$(Revision)\0"" />
|
||||
<VersionHeaderContents Include="#define ProductVersion $(AspNetCoreModuleVersionMajor),$(AspNetCoreModuleVersionMinor),$(AspNetCoreModuleVersionPatch),$(Revision)" />
|
||||
<VersionHeaderContents Include="#define ProductVersionStr "$(AspNetCoreModuleVersionMajor).$(AspNetCoreModuleVersionMinor).$(AspNetCoreModuleVersionPatch).$(Revision)\0"" />
|
||||
<VersionHeaderContents Include="#define PlatformToolset "$(PlatformToolset)\0"" />
|
||||
</ItemGroup>
|
||||
<WriteLinesToFile File="version.h" Lines="@(VersionHeaderContents)" OverWrite="true" />
|
||||
</Target>
|
||||
<ItemGroup>
|
||||
<ProjectReference Include="..\CommonLib\CommonLib.vcxproj">
|
||||
<Project>{55494e58-e061-4c4c-a0a8-837008e72f85}</Project>
|
||||
</ProjectReference>
|
||||
<ProjectReference Include="..\IISLib\IISLib.vcxproj">
|
||||
<Project>{4787a64f-9a3e-4867-a55a-70cb4b2b2ffe}</Project>
|
||||
<LinkLibraryDependencies>false</LinkLibraryDependencies>
|
||||
</ProjectReference>
|
||||
</ItemGroup>
|
||||
<ItemGroup>
|
||||
<None Include="Source.def" />
|
||||
</ItemGroup>
|
||||
<ItemGroup>
|
||||
<Content Include="aspnetcore_schema.xml">
|
||||
<CopyToOutputDirectory>PreserveNewest</CopyToOutputDirectory>
|
||||
</Content>
|
||||
</ItemGroup>
|
||||
<ItemGroup>
|
||||
<Xml Include="aspnetcore_schema.xml" />
|
||||
</ItemGroup>
|
||||
<Import Project="$(VCTargetsPath)\Microsoft.Cpp.targets" />
|
||||
<ImportGroup Label="ExtensionTargets">
|
||||
</ImportGroup>
|
||||
</Project>
|
||||
|
|
@ -0,0 +1,218 @@
|
|||
// Copyright (c) .NET Foundation. All rights reserved.
|
||||
// Licensed under the MIT License. See License.txt in the project root for license information.
|
||||
|
||||
#pragma once
|
||||
|
||||
typedef
|
||||
HRESULT
|
||||
(WINAPI * PFN_ASPNETCORE_CREATE_APPLICATION)(
|
||||
_In_ IHttpServer *pServer,
|
||||
_In_ ASPNETCORE_CONFIG *pConfig,
|
||||
_Out_ APPLICATION **pApplication
|
||||
);
|
||||
|
||||
typedef
|
||||
HRESULT
|
||||
(WINAPI * PFN_ASPNETCORE_CREATE_REQUEST_HANDLER)(
|
||||
_In_ IHttpContext *pHttpContext,
|
||||
_In_ HTTP_MODULE_ID *pModuleId,
|
||||
_In_ APPLICATION *pApplication,
|
||||
_Out_ REQUEST_HANDLER **pRequestHandler
|
||||
);
|
||||
//
|
||||
// The key used for hash-table lookups, consists of the port on which the http process is created.
|
||||
//
|
||||
class APPLICATION_INFO_KEY
|
||||
{
|
||||
public:
|
||||
|
||||
APPLICATION_INFO_KEY(
|
||||
VOID
|
||||
) : INLINE_STRU_INIT(m_struKey)
|
||||
{
|
||||
}
|
||||
|
||||
HRESULT
|
||||
Initialize(
|
||||
_In_ LPCWSTR pszKey
|
||||
)
|
||||
{
|
||||
return m_struKey.Copy(pszKey);
|
||||
}
|
||||
|
||||
BOOL
|
||||
GetIsEqual(
|
||||
const APPLICATION_INFO_KEY * key2
|
||||
) const
|
||||
{
|
||||
return m_struKey.Equals(key2->m_struKey);
|
||||
}
|
||||
|
||||
DWORD CalcKeyHash() const
|
||||
{
|
||||
return Hash(m_struKey.QueryStr());
|
||||
}
|
||||
|
||||
private:
|
||||
|
||||
INLINE_STRU(m_struKey, 1024);
|
||||
};
|
||||
|
||||
|
||||
class APPLICATION_INFO
|
||||
{
|
||||
public:
|
||||
|
||||
APPLICATION_INFO(IHttpServer *pServer) :
|
||||
m_pServer(pServer),
|
||||
m_cRefs(1), m_fAppOfflineFound(FALSE),
|
||||
m_pAppOfflineHtm(NULL), m_pFileWatcherEntry(NULL),
|
||||
m_pConfiguration(NULL),
|
||||
m_pfnAspNetCoreCreateApplication(NULL),
|
||||
m_pfnAspNetCoreCreateRequestHandler(NULL)
|
||||
{
|
||||
InitializeSRWLock(&m_srwLock);
|
||||
}
|
||||
|
||||
APPLICATION_INFO_KEY *
|
||||
QueryApplicationInfoKey()
|
||||
{
|
||||
return &m_applicationInfoKey;
|
||||
}
|
||||
|
||||
virtual
|
||||
~APPLICATION_INFO();
|
||||
|
||||
HRESULT
|
||||
Initialize(
|
||||
_In_ ASPNETCORE_CONFIG *pConfiguration,
|
||||
_In_ FILE_WATCHER *pFileWatcher
|
||||
);
|
||||
|
||||
VOID
|
||||
ReferenceApplicationInfo() const
|
||||
{
|
||||
InterlockedIncrement(&m_cRefs);
|
||||
}
|
||||
|
||||
VOID
|
||||
DereferenceApplicationInfo() const
|
||||
{
|
||||
if (InterlockedDecrement(&m_cRefs) == 0)
|
||||
{
|
||||
delete this;
|
||||
}
|
||||
}
|
||||
|
||||
APP_OFFLINE_HTM* QueryAppOfflineHtm()
|
||||
{
|
||||
return m_pAppOfflineHtm;
|
||||
}
|
||||
|
||||
BOOL
|
||||
AppOfflineFound()
|
||||
{
|
||||
return m_fAppOfflineFound;
|
||||
}
|
||||
|
||||
VOID
|
||||
UpdateAppOfflineFileHandle();
|
||||
|
||||
HRESULT
|
||||
StartMonitoringAppOffline();
|
||||
|
||||
ASPNETCORE_CONFIG*
|
||||
QueryConfig()
|
||||
{
|
||||
return m_pConfiguration;
|
||||
}
|
||||
|
||||
APPLICATION*
|
||||
QueryApplication()
|
||||
{
|
||||
return m_pApplication;
|
||||
}
|
||||
|
||||
HRESULT
|
||||
EnsureApplicationCreated();
|
||||
|
||||
PFN_ASPNETCORE_CREATE_REQUEST_HANDLER
|
||||
QueryCreateRequestHandler()
|
||||
{
|
||||
return m_pfnAspNetCoreCreateRequestHandler;
|
||||
}
|
||||
|
||||
private:
|
||||
HRESULT FindRequestHandlerAssembly();
|
||||
HRESULT FindNativeAssemblyFromGlobalLocation(STRU* struFilename);
|
||||
HRESULT FindNativeAssemblyFromLocalBin(STRU* struFilename);
|
||||
HRESULT GetRequestHandlerFromRuntimeStore(STRU* struFilename);
|
||||
|
||||
mutable LONG m_cRefs;
|
||||
APPLICATION_INFO_KEY m_applicationInfoKey;
|
||||
BOOL m_fAppOfflineFound;
|
||||
APP_OFFLINE_HTM *m_pAppOfflineHtm;
|
||||
FILE_WATCHER_ENTRY *m_pFileWatcherEntry;
|
||||
ASPNETCORE_CONFIG *m_pConfiguration;
|
||||
APPLICATION *m_pApplication;
|
||||
SRWLOCK m_srwLock;
|
||||
IHttpServer *m_pServer;
|
||||
PFN_ASPNETCORE_CREATE_APPLICATION m_pfnAspNetCoreCreateApplication;
|
||||
PFN_ASPNETCORE_CREATE_REQUEST_HANDLER m_pfnAspNetCoreCreateRequestHandler;
|
||||
};
|
||||
|
||||
class APPLICATION_INFO_HASH :
|
||||
public HASH_TABLE<APPLICATION_INFO, APPLICATION_INFO_KEY *>
|
||||
{
|
||||
|
||||
public:
|
||||
|
||||
APPLICATION_INFO_HASH()
|
||||
{}
|
||||
|
||||
APPLICATION_INFO_KEY *
|
||||
ExtractKey(
|
||||
APPLICATION_INFO *pApplicationInfo
|
||||
)
|
||||
{
|
||||
return pApplicationInfo->QueryApplicationInfoKey();
|
||||
}
|
||||
|
||||
DWORD
|
||||
CalcKeyHash(
|
||||
APPLICATION_INFO_KEY *key
|
||||
)
|
||||
{
|
||||
return key->CalcKeyHash();
|
||||
}
|
||||
|
||||
BOOL
|
||||
EqualKeys(
|
||||
APPLICATION_INFO_KEY *key1,
|
||||
APPLICATION_INFO_KEY *key2
|
||||
)
|
||||
{
|
||||
return key1->GetIsEqual(key2);
|
||||
}
|
||||
|
||||
VOID
|
||||
ReferenceRecord(
|
||||
APPLICATION_INFO *pApplicationInfo
|
||||
)
|
||||
{
|
||||
pApplicationInfo->ReferenceApplicationInfo();
|
||||
}
|
||||
|
||||
VOID
|
||||
DereferenceRecord(
|
||||
APPLICATION_INFO *pApplicationInfo
|
||||
)
|
||||
{
|
||||
pApplicationInfo->DereferenceApplicationInfo();
|
||||
}
|
||||
|
||||
private:
|
||||
|
||||
APPLICATION_INFO_HASH(const APPLICATION_INFO_HASH &);
|
||||
void operator=(const APPLICATION_INFO_HASH &);
|
||||
};
|
||||
|
|
@ -0,0 +1,135 @@
|
|||
// Copyright (c) .NET Foundation. All rights reserved.
|
||||
// Licensed under the MIT License. See License.txt in the project root for license information.
|
||||
|
||||
#pragma once
|
||||
|
||||
#define DEFAULT_HASH_BUCKETS 293
|
||||
|
||||
//
|
||||
// This class will manage the lifecycle of all Asp.Net Core applciation
|
||||
// It should be global singleton.
|
||||
// Should always call GetInstance to get the object instance
|
||||
//
|
||||
class APPLICATION_MANAGER
|
||||
{
|
||||
public:
|
||||
|
||||
static
|
||||
APPLICATION_MANAGER*
|
||||
GetInstance(
|
||||
VOID
|
||||
)
|
||||
{
|
||||
if ( sm_pApplicationManager == NULL )
|
||||
{
|
||||
sm_pApplicationManager = new APPLICATION_MANAGER();
|
||||
}
|
||||
|
||||
return sm_pApplicationManager;
|
||||
}
|
||||
|
||||
static
|
||||
VOID
|
||||
Cleanup(
|
||||
VOID
|
||||
)
|
||||
{
|
||||
if(sm_pApplicationManager != NULL)
|
||||
{
|
||||
delete sm_pApplicationManager;
|
||||
sm_pApplicationManager = NULL;
|
||||
}
|
||||
}
|
||||
|
||||
HRESULT
|
||||
GetApplicationInfo(
|
||||
_In_ IHttpServer* pServer,
|
||||
_In_ ASPNETCORE_CONFIG* pConfig,
|
||||
_Out_ APPLICATION_INFO ** ppApplicationInfo
|
||||
);
|
||||
|
||||
HRESULT
|
||||
RecycleApplication(
|
||||
_In_ LPCWSTR pszApplicationId
|
||||
);
|
||||
|
||||
VOID
|
||||
ShutDown();
|
||||
|
||||
~APPLICATION_MANAGER()
|
||||
{
|
||||
if(m_pApplicationInfoHash != NULL)
|
||||
{
|
||||
m_pApplicationInfoHash->Clear();
|
||||
delete m_pApplicationInfoHash;
|
||||
m_pApplicationInfoHash = NULL;
|
||||
}
|
||||
|
||||
if( m_pFileWatcher!= NULL )
|
||||
{
|
||||
delete m_pFileWatcher;
|
||||
m_pFileWatcher = NULL;
|
||||
}
|
||||
}
|
||||
|
||||
FILE_WATCHER*
|
||||
GetFileWatcher()
|
||||
{
|
||||
return m_pFileWatcher;
|
||||
}
|
||||
|
||||
HRESULT Initialize()
|
||||
{
|
||||
HRESULT hr = S_OK;
|
||||
|
||||
if(m_pApplicationInfoHash == NULL)
|
||||
{
|
||||
m_pApplicationInfoHash = new APPLICATION_INFO_HASH();
|
||||
if(m_pApplicationInfoHash == NULL)
|
||||
{
|
||||
hr = E_OUTOFMEMORY;
|
||||
goto Finished;
|
||||
}
|
||||
|
||||
hr = m_pApplicationInfoHash->Initialize(DEFAULT_HASH_BUCKETS);
|
||||
if(FAILED(hr))
|
||||
{
|
||||
goto Finished;
|
||||
}
|
||||
}
|
||||
|
||||
if( m_pFileWatcher == NULL )
|
||||
{
|
||||
m_pFileWatcher = new FILE_WATCHER;
|
||||
if(m_pFileWatcher == NULL)
|
||||
{
|
||||
hr = HRESULT_FROM_WIN32( ERROR_NOT_ENOUGH_MEMORY );
|
||||
goto Finished;
|
||||
}
|
||||
|
||||
m_pFileWatcher->Create();
|
||||
}
|
||||
|
||||
Finished:
|
||||
return hr;
|
||||
}
|
||||
|
||||
private:
|
||||
//
|
||||
// we currently limit the size of m_pstrErrorInfo to 5000, be careful if you want to change its payload
|
||||
//
|
||||
APPLICATION_MANAGER() : m_pApplicationInfoHash(NULL),
|
||||
m_pFileWatcher(NULL),
|
||||
m_hostingModel(HOSTING_UNKNOWN),
|
||||
m_fInShutdown(FALSE)
|
||||
{
|
||||
InitializeSRWLock(&m_srwLock);
|
||||
}
|
||||
|
||||
FILE_WATCHER *m_pFileWatcher;
|
||||
APPLICATION_INFO_HASH *m_pApplicationInfoHash;
|
||||
static APPLICATION_MANAGER *sm_pApplicationManager;
|
||||
SRWLOCK m_srwLock;
|
||||
APP_HOSTING_MODEL m_hostingModel;
|
||||
bool m_fInShutdown;
|
||||
};
|
||||
|
|
@ -0,0 +1,101 @@
|
|||
// Copyright (c) .NET Foundation. All rights reserved.
|
||||
// Licensed under the MIT License. See License.txt in the project root for license information.
|
||||
|
||||
#pragma once
|
||||
class APP_OFFLINE_HTM
|
||||
{
|
||||
public:
|
||||
APP_OFFLINE_HTM(LPCWSTR pszPath) : m_cRefs(1)
|
||||
{
|
||||
m_Path.Copy(pszPath);
|
||||
}
|
||||
|
||||
VOID
|
||||
ReferenceAppOfflineHtm() const
|
||||
{
|
||||
InterlockedIncrement(&m_cRefs);
|
||||
}
|
||||
|
||||
VOID
|
||||
DereferenceAppOfflineHtm() const
|
||||
{
|
||||
if (InterlockedDecrement(&m_cRefs) == 0)
|
||||
{
|
||||
delete this;
|
||||
}
|
||||
}
|
||||
|
||||
BOOL
|
||||
Load(
|
||||
VOID
|
||||
)
|
||||
{
|
||||
BOOL fResult = TRUE;
|
||||
LARGE_INTEGER li = { 0 };
|
||||
CHAR *pszBuff = NULL;
|
||||
HANDLE handle = INVALID_HANDLE_VALUE;
|
||||
|
||||
handle = CreateFile(m_Path.QueryStr(),
|
||||
GENERIC_READ,
|
||||
FILE_SHARE_READ | FILE_SHARE_WRITE | FILE_SHARE_DELETE,
|
||||
NULL,
|
||||
OPEN_EXISTING,
|
||||
FILE_ATTRIBUTE_NORMAL,
|
||||
NULL);
|
||||
|
||||
if (handle == INVALID_HANDLE_VALUE)
|
||||
{
|
||||
if (GetLastError() == ERROR_FILE_NOT_FOUND)
|
||||
{
|
||||
fResult = FALSE;
|
||||
}
|
||||
|
||||
// This Load() member function is supposed be called only when the change notification event of file creation or file modification happens.
|
||||
// If file is currenlty locked exclusively by other processes, we might get INVALID_HANDLE_VALUE even though the file exists. In that case, we should return TRUE here.
|
||||
goto Finished;
|
||||
}
|
||||
|
||||
if (!GetFileSizeEx(handle, &li))
|
||||
{
|
||||
goto Finished;
|
||||
}
|
||||
|
||||
if (li.HighPart != 0)
|
||||
{
|
||||
// > 4gb file size not supported
|
||||
// todo: log a warning at event log
|
||||
goto Finished;
|
||||
}
|
||||
|
||||
DWORD bytesRead = 0;
|
||||
|
||||
if (li.LowPart > 0)
|
||||
{
|
||||
pszBuff = new CHAR[li.LowPart + 1];
|
||||
|
||||
if (ReadFile(handle, pszBuff, li.LowPart, &bytesRead, NULL))
|
||||
{
|
||||
m_Contents.Copy(pszBuff, bytesRead);
|
||||
}
|
||||
}
|
||||
|
||||
Finished:
|
||||
if (handle != INVALID_HANDLE_VALUE)
|
||||
{
|
||||
CloseHandle(handle);
|
||||
handle = INVALID_HANDLE_VALUE;
|
||||
}
|
||||
|
||||
if (pszBuff != NULL)
|
||||
{
|
||||
delete[] pszBuff;
|
||||
pszBuff = NULL;
|
||||
}
|
||||
|
||||
return fResult;
|
||||
}
|
||||
|
||||
mutable LONG m_cRefs;
|
||||
STRA m_Contents;
|
||||
STRU m_Path;
|
||||
};
|
||||
|
|
@ -0,0 +1,61 @@
|
|||
//
|
||||
// this file is automatically generated
|
||||
// by beaver.exe 1.11.2003.0
|
||||
//
|
||||
|
||||
//
|
||||
// if you want to use a private version file and customize this, see
|
||||
// file://samsndrop02/CoreXT-Latest/docs/corext/corext/version.htm
|
||||
//
|
||||
|
||||
#ifndef _BLDVER_H_
|
||||
#define _BLDVER_H_
|
||||
|
||||
#define BUILD_NUMBER "1965.0"
|
||||
#define BUILD_NUM 1965,0
|
||||
#define PRODUCT_NUMBER "7.1"
|
||||
#define PRODUCT_NUM 7,1
|
||||
#define INET_VERSION "7.1.1965.0"
|
||||
#define INET_VERSION_L L"7.1.1965.0"
|
||||
#define INET_VER 7,1,1965,0
|
||||
|
||||
#define PRODUCT_MAJOR 7
|
||||
#define PRODUCT_MAJOR_STRING "7"
|
||||
#define PRODUCT_MAJOR_NUMBER 7
|
||||
|
||||
#define PRODUCT_MINOR 1
|
||||
#define PRODUCT_MINOR_STRING "1"
|
||||
#define PRODUCT_MINOR_NUMBER 1
|
||||
|
||||
#define BUILD_MAJOR 1965
|
||||
#define BUILD_MAJOR_STRING "1965"
|
||||
#define BUILD_MAJOR_NUMBER 1965
|
||||
|
||||
#define BUILD_MINOR 0
|
||||
#define BUILD_MINOR_STRING "0"
|
||||
#define BUILD_MINOR_NUMBER 0
|
||||
|
||||
#define BUILD_PRIVATE "Built by panwang on IIS-OOB.\0"
|
||||
|
||||
// beaver.exe can't handle a pragma to disable redefinition
|
||||
#undef VER_PRODUCTVERSION
|
||||
#undef VER_PRODUCTVERSION_STR
|
||||
#undef VER_PRODUCTMAJORVERSION
|
||||
#undef VER_PRODUCTMINORVERSION
|
||||
#undef VER_PRODUCTBUILD
|
||||
#undef VER_PRODUCTBUILD_QFE
|
||||
#undef VER_PRODUCTNAME_STR
|
||||
#undef VER_COMPANYNAME_STR
|
||||
|
||||
#define VER_PRODUCTVERSION 7,1,1965,0
|
||||
#define VER_PRODUCTVERSION_STR 7.1.1965.0
|
||||
#define VER_PRODUCTMAJORVERSION 7
|
||||
#define VER_PRODUCTMINORVERSION 1
|
||||
#define VER_PRODUCTBUILD 1965
|
||||
#define VER_PRODUCTBUILD_QFE 0
|
||||
#define VER_PRODUCTNAME_STR "Microsoft Web Platform Extensions"
|
||||
#define VER_COMPANYNAME_STR "Microsoft Corporation"
|
||||
|
||||
|
||||
|
||||
#endif
|
||||
|
|
@ -0,0 +1,126 @@
|
|||
// Copyright (c) .NET Foundation. All rights reserved.
|
||||
// Licensed under the MIT License. See License.txt in the project root for license information.
|
||||
|
||||
#pragma once
|
||||
|
||||
#define FILE_WATCHER_SHUTDOWN_KEY (ULONG_PTR)(-1)
|
||||
#define FILE_WATCHER_ENTRY_BUFFER_SIZE 4096
|
||||
#ifndef CONTAINING_RECORD
|
||||
//
|
||||
// Calculate the address of the base of the structure given its type, and an
|
||||
// address of a field within the structure.
|
||||
//
|
||||
|
||||
#define CONTAINING_RECORD(address, type, field) \
|
||||
((type *)((PCHAR)(address)-(ULONG_PTR)(&((type *)0)->field)))
|
||||
|
||||
#endif // !CONTAINING_RECORD
|
||||
#define FILE_NOTIFY_VALID_MASK 0x00000fff
|
||||
#define FILE_WATCHER_ENTRY_SIGNATURE ((DWORD) 'FWES')
|
||||
#define FILE_WATCHER_ENTRY_SIGNATURE_FREE ((DWORD) 'sewf')
|
||||
|
||||
class APPLICATION_INFO;
|
||||
|
||||
class FILE_WATCHER{
|
||||
public:
|
||||
|
||||
FILE_WATCHER();
|
||||
|
||||
~FILE_WATCHER();
|
||||
|
||||
HRESULT Create();
|
||||
|
||||
HANDLE
|
||||
QueryCompletionPort(
|
||||
VOID
|
||||
) const
|
||||
{
|
||||
return m_hCompletionPort;
|
||||
}
|
||||
|
||||
static
|
||||
DWORD
|
||||
WINAPI ChangeNotificationThread(LPVOID);
|
||||
|
||||
static
|
||||
void
|
||||
WINAPI FileWatcherCompletionRoutine
|
||||
(
|
||||
DWORD dwCompletionStatus,
|
||||
DWORD cbCompletion,
|
||||
OVERLAPPED * pOverlapped
|
||||
);
|
||||
|
||||
private:
|
||||
HANDLE m_hCompletionPort;
|
||||
HANDLE m_hChangeNotificationThread;
|
||||
};
|
||||
|
||||
class FILE_WATCHER_ENTRY
|
||||
{
|
||||
public:
|
||||
FILE_WATCHER_ENTRY(FILE_WATCHER * pFileMonitor);
|
||||
|
||||
OVERLAPPED _overlapped;
|
||||
|
||||
HRESULT
|
||||
Create(
|
||||
_In_ PCWSTR pszDirectoryToMonitor,
|
||||
_In_ PCWSTR pszFileNameToMonitor,
|
||||
_In_ APPLICATION_INFO* pApplicationInfo,
|
||||
_In_ HANDLE hImpersonationToken
|
||||
);
|
||||
|
||||
VOID
|
||||
ReferenceFileWatcherEntry() const
|
||||
{
|
||||
InterlockedIncrement(&_cRefs);
|
||||
}
|
||||
|
||||
VOID
|
||||
DereferenceFileWatcherEntry() const
|
||||
{
|
||||
if (InterlockedDecrement(&_cRefs) == 0)
|
||||
{
|
||||
delete this;
|
||||
}
|
||||
}
|
||||
|
||||
BOOL
|
||||
QueryIsValid() const
|
||||
{
|
||||
return _fIsValid;
|
||||
}
|
||||
|
||||
VOID
|
||||
MarkEntryInValid()
|
||||
{
|
||||
_fIsValid = FALSE;
|
||||
}
|
||||
|
||||
HRESULT Monitor();
|
||||
|
||||
VOID StopMonitor();
|
||||
|
||||
HRESULT
|
||||
HandleChangeCompletion(
|
||||
_In_ DWORD dwCompletionStatus,
|
||||
_In_ DWORD cbCompletion
|
||||
);
|
||||
|
||||
private:
|
||||
virtual ~FILE_WATCHER_ENTRY();
|
||||
|
||||
DWORD _dwSignature;
|
||||
BUFFER _buffDirectoryChanges;
|
||||
HANDLE _hImpersonationToken;
|
||||
HANDLE _hDirectory;
|
||||
FILE_WATCHER* _pFileMonitor;
|
||||
APPLICATION_INFO* _pApplicationInfo;
|
||||
STRU _strFileName;
|
||||
STRU _strDirectoryName;
|
||||
LONG _lStopMonitorCalled;
|
||||
mutable LONG _cRefs;
|
||||
BOOL _fIsValid;
|
||||
SRWLOCK _srwLock;
|
||||
};
|
||||
|
|
@ -0,0 +1,49 @@
|
|||
// Copyright (c) .NET Foundation and contributors. All rights reserved.
|
||||
// Licensed under the MIT license. See LICENSE file in the project root for full license information.
|
||||
|
||||
#ifndef __FX_VER_H__
|
||||
#define __FX_VER_H__
|
||||
#include <string>
|
||||
|
||||
// Note: This is not SemVer (esp., in comparing pre-release part, fx_ver_t does not
|
||||
// compare multiple dot separated identifiers individually.) ex: 1.0.0-beta.2 vs. 1.0.0-beta.11
|
||||
struct fx_ver_t
|
||||
{
|
||||
fx_ver_t(int major, int minor, int patch);
|
||||
fx_ver_t(int major, int minor, int patch, const std::wstring& pre);
|
||||
fx_ver_t(int major, int minor, int patch, const std::wstring& pre, const std::wstring& build);
|
||||
|
||||
int get_major() const { return m_major; }
|
||||
int get_minor() const { return m_minor; }
|
||||
int get_patch() const { return m_patch; }
|
||||
|
||||
void set_major(int m) { m_major = m; }
|
||||
void set_minor(int m) { m_minor = m; }
|
||||
void set_patch(int p) { m_patch = p; }
|
||||
|
||||
bool is_prerelease() const { return !m_pre.empty(); }
|
||||
|
||||
std::wstring as_str() const;
|
||||
std::wstring prerelease_glob() const;
|
||||
std::wstring patch_glob() const;
|
||||
|
||||
bool operator ==(const fx_ver_t& b) const;
|
||||
bool operator !=(const fx_ver_t& b) const;
|
||||
bool operator <(const fx_ver_t& b) const;
|
||||
bool operator >(const fx_ver_t& b) const;
|
||||
bool operator <=(const fx_ver_t& b) const;
|
||||
bool operator >=(const fx_ver_t& b) const;
|
||||
|
||||
static bool parse(const std::wstring& ver, fx_ver_t* fx_ver, bool parse_only_production = false);
|
||||
|
||||
private:
|
||||
int m_major;
|
||||
int m_minor;
|
||||
int m_patch;
|
||||
std::wstring m_pre;
|
||||
std::wstring m_build;
|
||||
|
||||
static int compare(const fx_ver_t&a, const fx_ver_t& b);
|
||||
};
|
||||
|
||||
#endif // __FX_VER_H__
|
||||
|
|
@ -0,0 +1,37 @@
|
|||
// Copyright (c) .NET Foundation. All rights reserved.
|
||||
// Licensed under the MIT License. See License.txt in the project root for license information.
|
||||
|
||||
#pragma once
|
||||
|
||||
class ASPNET_CORE_GLOBAL_MODULE : public CGlobalModule
|
||||
{
|
||||
|
||||
public:
|
||||
|
||||
ASPNET_CORE_GLOBAL_MODULE(
|
||||
APPLICATION_MANAGER* pApplicationManager
|
||||
);
|
||||
|
||||
~ASPNET_CORE_GLOBAL_MODULE()
|
||||
{
|
||||
}
|
||||
|
||||
VOID Terminate()
|
||||
{
|
||||
// Remove the class from memory.
|
||||
delete this;
|
||||
}
|
||||
|
||||
GLOBAL_NOTIFICATION_STATUS
|
||||
OnGlobalStopListening(
|
||||
_In_ IGlobalStopListeningProvider * pProvider
|
||||
);
|
||||
|
||||
GLOBAL_NOTIFICATION_STATUS
|
||||
OnGlobalConfigurationChange(
|
||||
_In_ IGlobalConfigurationChangeProvider * pProvider
|
||||
);
|
||||
|
||||
private:
|
||||
APPLICATION_MANAGER * m_pApplicationManager;
|
||||
};
|
||||
|
|
@ -0,0 +1,64 @@
|
|||
// Copyright (c) .NET Foundation. All rights reserved.
|
||||
// Licensed under the MIT License. See License.txt in the project root for license information.
|
||||
|
||||
#pragma once
|
||||
|
||||
extern HTTP_MODULE_ID g_pModuleId;
|
||||
extern IHttpServer *g_pHttpServer;
|
||||
extern HMODULE g_hAspnetCoreRH;
|
||||
|
||||
class ASPNET_CORE_PROXY_MODULE : public CHttpModule
|
||||
{
|
||||
public:
|
||||
|
||||
ASPNET_CORE_PROXY_MODULE();
|
||||
|
||||
~ASPNET_CORE_PROXY_MODULE();
|
||||
|
||||
void * operator new(size_t size, IModuleAllocator * pPlacement)
|
||||
{
|
||||
return pPlacement->AllocateMemory(static_cast<DWORD>(size));
|
||||
}
|
||||
|
||||
VOID
|
||||
operator delete(
|
||||
void *
|
||||
)
|
||||
{}
|
||||
|
||||
__override
|
||||
REQUEST_NOTIFICATION_STATUS
|
||||
OnExecuteRequestHandler(
|
||||
IHttpContext * pHttpContext,
|
||||
IHttpEventProvider * pProvider
|
||||
);
|
||||
|
||||
__override
|
||||
REQUEST_NOTIFICATION_STATUS
|
||||
OnAsyncCompletion(
|
||||
IHttpContext * pHttpContext,
|
||||
DWORD dwNotification,
|
||||
BOOL fPostNotification,
|
||||
IHttpEventProvider * pProvider,
|
||||
IHttpCompletionInfo * pCompletionInfo
|
||||
);
|
||||
|
||||
private:
|
||||
|
||||
APPLICATION_INFO *m_pApplicationInfo;
|
||||
APPLICATION *m_pApplication;
|
||||
REQUEST_HANDLER *m_pHandler;
|
||||
};
|
||||
|
||||
class ASPNET_CORE_PROXY_MODULE_FACTORY : public IHttpModuleFactory
|
||||
{
|
||||
public:
|
||||
HRESULT
|
||||
GetHttpModule(
|
||||
CHttpModule ** ppModule,
|
||||
IModuleAllocator * pAllocator
|
||||
);
|
||||
|
||||
VOID
|
||||
Terminate();
|
||||
};
|
||||
|
|
@ -0,0 +1,25 @@
|
|||
// Copyright (c) .NET Foundation. All rights reserved.
|
||||
// Licensed under the MIT License. See License.txt in the project root for license information.
|
||||
|
||||
#pragma once
|
||||
|
||||
#define IDS_INVALID_PROPERTY 1000
|
||||
#define IDS_SERVER_ERROR 1001
|
||||
|
||||
// TODO remove this file?
|
||||
#define ASPNETCORE_EVENT_MSG_BUFFER_SIZE 256
|
||||
#define ASPNETCORE_EVENT_PROCESS_START_SUCCESS_MSG L"Application '%s' started process '%d' successfully and is listening on port '%d'."
|
||||
#define ASPNETCORE_EVENT_RAPID_FAIL_COUNT_EXCEEDED_MSG L"Maximum rapid fail count per minute of '%d' exceeded."
|
||||
#define ASPNETCORE_EVENT_PROCESS_START_INTERNAL_ERROR_MSG L"Application '%s' failed to parse processPath and arguments due to internal error, ErrorCode = '0x%x'."
|
||||
#define ASPNETCORE_EVENT_PROCESS_START_POSTCREATE_ERROR_MSG L"Application '%s' with physical root '%s' created process with commandline '%s'but failed to get its status, ErrorCode = '0x%x'."
|
||||
#define ASPNETCORE_EVENT_PROCESS_START_ERROR_MSG L"Application '%s' with physical root '%s' failed to start process with commandline '%s', ErrorCode = '0x%x' processStatus code '%x'."
|
||||
#define ASPNETCORE_EVENT_PROCESS_START_WRONGPORT_ERROR_MSG L"Application '%s' with physical root '%s' created process with commandline '%s' but failed to listen on the given port '%d'"
|
||||
#define ASPNETCORE_EVENT_PROCESS_START_NOTREADY_ERROR_MSG L"Application '%s' with physical root '%s' created process with commandline '%s' but either crashed or did not reponse or did not listen on the given port '%d', ErrorCode = '0x%x'"
|
||||
#define ASPNETCORE_EVENT_INVALID_STDOUT_LOG_FILE_MSG L"Warning: Could not create stdoutLogFile %s, ErrorCode = %d."
|
||||
#define ASPNETCORE_EVENT_GRACEFUL_SHUTDOWN_FAILURE_MSG L"Failed to gracefully shutdown process '%d'."
|
||||
#define ASPNETCORE_EVENT_SENT_SHUTDOWN_HTTP_REQUEST_MSG L"Sent shutdown HTTP message to process '%d' and received http status '%d'."
|
||||
#define ASPNETCORE_EVENT_LOAD_CLR_FALIURE_MSG L"Application '%s' with physical root '%s' failed to load clr and managed application, ErrorCode = '0x%x."
|
||||
#define ASPNETCORE_EVENT_DUPLICATED_INPROCESS_APP_MSG L"Only one inprocess application is allowed per IIS application pool. Please assign the application '%s' to a different IIS application pool."
|
||||
#define ASPNETCORE_EVENT_MIXED_HOSTING_MODEL_ERROR_MSG L"Mixed hosting model is not supported. Application '%s' configured with different hostingModel value '%s' other than the one of running application(s)."
|
||||
#define ASPNETCORE_EVENT_ADD_APPLICATION_ERROR_MSG L"Failed to start application '%s', ErrorCode '0x%x'."
|
||||
#define ASPNETCORE_EVENT_INPROCESS_THREAD_EXIT_MSG L"Application '%s' with physical root '%s' hit unexpected managed background thread eixt, ErrorCode = '0x%x."
|
||||
|
|
@ -0,0 +1,4 @@
|
|||
LIBRARY aspnetcore
|
||||
|
||||
EXPORTS
|
||||
RegisterModule
|
||||
|
|
@ -0,0 +1,332 @@
|
|||
// Copyright (c) .NET Foundation. All rights reserved.
|
||||
// Licensed under the MIT License. See License.txt in the project root for license information.
|
||||
|
||||
#include "precomp.hxx"
|
||||
|
||||
APPLICATION_INFO::~APPLICATION_INFO()
|
||||
{
|
||||
if (m_pAppOfflineHtm != NULL)
|
||||
{
|
||||
m_pAppOfflineHtm->DereferenceAppOfflineHtm();
|
||||
m_pAppOfflineHtm = NULL;
|
||||
}
|
||||
|
||||
if (m_pFileWatcherEntry != NULL)
|
||||
{
|
||||
// Mark the entry as invalid,
|
||||
// StopMonitor will close the file handle and trigger a FCN
|
||||
// the entry will delete itself when processing this FCN
|
||||
m_pFileWatcherEntry->MarkEntryInValid();
|
||||
m_pFileWatcherEntry->StopMonitor();
|
||||
m_pFileWatcherEntry = NULL;
|
||||
}
|
||||
|
||||
if (m_pApplication != NULL)
|
||||
{
|
||||
// shutdown the application
|
||||
m_pApplication->ShutDown();
|
||||
m_pApplication->DereferenceApplication();
|
||||
m_pApplication = NULL;
|
||||
}
|
||||
|
||||
// configuration should be dereferenced after application shutdown
|
||||
// since the former will use it during shutdown
|
||||
if (m_pConfiguration != NULL)
|
||||
{
|
||||
// Need to dereference the configuration instance
|
||||
m_pConfiguration->DereferenceConfiguration();
|
||||
m_pConfiguration = NULL;
|
||||
}
|
||||
}
|
||||
|
||||
HRESULT
|
||||
APPLICATION_INFO::Initialize(
|
||||
_In_ ASPNETCORE_CONFIG *pConfiguration,
|
||||
_In_ FILE_WATCHER *pFileWatcher
|
||||
)
|
||||
{
|
||||
HRESULT hr = S_OK;
|
||||
|
||||
DBG_ASSERT(pConfiguration);
|
||||
DBG_ASSERT(pFileWatcher);
|
||||
|
||||
m_pConfiguration = pConfiguration;
|
||||
|
||||
// reference the configuration instance to prevent it will be not release
|
||||
// earlier in case of configuration change and shutdown
|
||||
m_pConfiguration->ReferenceConfiguration();
|
||||
|
||||
hr = m_applicationInfoKey.Initialize(pConfiguration->QueryConfigPath()->QueryStr());
|
||||
if (FAILED(hr))
|
||||
{
|
||||
goto Finished;
|
||||
}
|
||||
|
||||
if (m_pFileWatcherEntry == NULL)
|
||||
{
|
||||
m_pFileWatcherEntry = new FILE_WATCHER_ENTRY(pFileWatcher);
|
||||
if (m_pFileWatcherEntry == NULL)
|
||||
{
|
||||
hr = E_OUTOFMEMORY;
|
||||
goto Finished;
|
||||
}
|
||||
}
|
||||
|
||||
UpdateAppOfflineFileHandle();
|
||||
|
||||
Finished:
|
||||
return hr;
|
||||
}
|
||||
|
||||
HRESULT
|
||||
APPLICATION_INFO::StartMonitoringAppOffline()
|
||||
{
|
||||
HRESULT hr = S_OK;
|
||||
if (m_pFileWatcherEntry != NULL)
|
||||
{
|
||||
hr = m_pFileWatcherEntry->Create(m_pConfiguration->QueryApplicationPhysicalPath()->QueryStr(), L"app_offline.htm", this, NULL);
|
||||
}
|
||||
return hr;
|
||||
}
|
||||
|
||||
VOID
|
||||
APPLICATION_INFO::UpdateAppOfflineFileHandle()
|
||||
{
|
||||
STRU strFilePath;
|
||||
UTILITY::ConvertPathToFullPath(L".\\app_offline.htm",
|
||||
m_pConfiguration->QueryApplicationPhysicalPath()->QueryStr(),
|
||||
&strFilePath);
|
||||
APP_OFFLINE_HTM *pOldAppOfflineHtm = NULL;
|
||||
APP_OFFLINE_HTM *pNewAppOfflineHtm = NULL;
|
||||
|
||||
if (INVALID_FILE_ATTRIBUTES == GetFileAttributes(strFilePath.QueryStr()) &&
|
||||
GetLastError() == ERROR_FILE_NOT_FOUND)
|
||||
{
|
||||
m_fAppOfflineFound = FALSE;
|
||||
}
|
||||
else
|
||||
{
|
||||
m_fAppOfflineFound = TRUE;
|
||||
pNewAppOfflineHtm = new APP_OFFLINE_HTM(strFilePath.QueryStr());
|
||||
|
||||
if (pNewAppOfflineHtm != NULL)
|
||||
{
|
||||
if (pNewAppOfflineHtm->Load())
|
||||
{
|
||||
//
|
||||
// loaded the new app_offline.htm
|
||||
//
|
||||
pOldAppOfflineHtm = (APP_OFFLINE_HTM *)InterlockedExchangePointer((VOID**)&m_pAppOfflineHtm, pNewAppOfflineHtm);
|
||||
|
||||
if (pOldAppOfflineHtm != NULL)
|
||||
{
|
||||
pOldAppOfflineHtm->DereferenceAppOfflineHtm();
|
||||
pOldAppOfflineHtm = NULL;
|
||||
}
|
||||
}
|
||||
else
|
||||
{
|
||||
// ignored the new app_offline file because the file does not exist.
|
||||
pNewAppOfflineHtm->DereferenceAppOfflineHtm();
|
||||
pNewAppOfflineHtm = NULL;
|
||||
}
|
||||
}
|
||||
|
||||
// recycle the application
|
||||
if (m_pApplication != NULL)
|
||||
{
|
||||
m_pApplication->ShutDown();
|
||||
m_pApplication->DereferenceApplication();
|
||||
m_pApplication = NULL;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
HRESULT
|
||||
APPLICATION_INFO::EnsureApplicationCreated()
|
||||
{
|
||||
HRESULT hr = S_OK;
|
||||
BOOL fLocked = FALSE;
|
||||
APPLICATION* pApplication = NULL;
|
||||
STACK_STRU(struFileName, 300); // >MAX_PATH
|
||||
STRU hostFxrDllLocation;
|
||||
|
||||
if (m_pApplication != NULL)
|
||||
{
|
||||
goto Finished;
|
||||
}
|
||||
|
||||
hr = FindRequestHandlerAssembly();
|
||||
if (FAILED(hr))
|
||||
{
|
||||
goto Finished;
|
||||
}
|
||||
|
||||
if (m_pApplication == NULL)
|
||||
{
|
||||
AcquireSRWLockExclusive(&m_srwLock);
|
||||
fLocked = TRUE;
|
||||
if (m_pApplication != NULL)
|
||||
{
|
||||
goto Finished;
|
||||
}
|
||||
|
||||
if (m_pfnAspNetCoreCreateApplication == NULL)
|
||||
{
|
||||
hr = HRESULT_FROM_WIN32(ERROR_INVALID_FUNCTION);
|
||||
goto Finished;
|
||||
}
|
||||
|
||||
hr = m_pfnAspNetCoreCreateApplication(m_pServer, m_pConfiguration, &pApplication);
|
||||
if (FAILED(hr))
|
||||
{
|
||||
goto Finished;
|
||||
}
|
||||
m_pApplication = pApplication;
|
||||
}
|
||||
|
||||
Finished:
|
||||
if (fLocked)
|
||||
{
|
||||
ReleaseSRWLockExclusive(&m_srwLock);
|
||||
}
|
||||
return hr;
|
||||
}
|
||||
|
||||
HRESULT
|
||||
APPLICATION_INFO::FindRequestHandlerAssembly()
|
||||
{
|
||||
HRESULT hr = S_OK;
|
||||
BOOL fLocked = FALSE;
|
||||
STACK_STRU(struFileName, 256);
|
||||
|
||||
if (g_fAspnetcoreRHLoadedError)
|
||||
{
|
||||
hr = E_APPLICATION_ACTIVATION_EXEC_FAILURE;
|
||||
goto Finished;
|
||||
}
|
||||
else if (!g_fAspnetcoreRHAssemblyLoaded)
|
||||
{
|
||||
AcquireSRWLockExclusive(&g_srwLock);
|
||||
fLocked = TRUE;
|
||||
if (g_fAspnetcoreRHLoadedError)
|
||||
{
|
||||
hr = E_APPLICATION_ACTIVATION_EXEC_FAILURE;
|
||||
goto Finished;
|
||||
}
|
||||
if (g_fAspnetcoreRHAssemblyLoaded)
|
||||
{
|
||||
goto Finished;
|
||||
}
|
||||
|
||||
// load assembly and create the application
|
||||
if (m_pConfiguration->QueryHostingModel() == APP_HOSTING_MODEL::HOSTING_IN_PROCESS)
|
||||
{
|
||||
// Look at inetsvr only for now. TODO add in functionality
|
||||
hr = FindNativeAssemblyFromGlobalLocation(&struFileName);
|
||||
|
||||
if (FAILED(hr))
|
||||
{
|
||||
goto Finished;
|
||||
}
|
||||
}
|
||||
else
|
||||
{
|
||||
hr = FindNativeAssemblyFromGlobalLocation(&struFileName);
|
||||
if (FAILED(hr))
|
||||
{
|
||||
goto Finished;
|
||||
}
|
||||
}
|
||||
|
||||
g_hAspnetCoreRH = LoadLibraryW(struFileName.QueryStr());
|
||||
if (g_hAspnetCoreRH == NULL)
|
||||
{
|
||||
hr = HRESULT_FROM_WIN32(GetLastError());
|
||||
goto Finished;
|
||||
}
|
||||
|
||||
g_pfnAspNetCoreCreateApplication = (PFN_ASPNETCORE_CREATE_APPLICATION)
|
||||
GetProcAddress(g_hAspnetCoreRH, "CreateApplication");
|
||||
if (g_pfnAspNetCoreCreateApplication == NULL)
|
||||
{
|
||||
hr = HRESULT_FROM_WIN32(GetLastError());
|
||||
goto Finished;
|
||||
}
|
||||
|
||||
g_pfnAspNetCoreCreateRequestHandler = (PFN_ASPNETCORE_CREATE_REQUEST_HANDLER)
|
||||
GetProcAddress(g_hAspnetCoreRH, "CreateRequestHandler");
|
||||
if (g_pfnAspNetCoreCreateRequestHandler == NULL)
|
||||
{
|
||||
hr = HRESULT_FROM_WIN32(GetLastError());
|
||||
goto Finished;
|
||||
}
|
||||
g_fAspnetcoreRHAssemblyLoaded = TRUE;
|
||||
}
|
||||
|
||||
Finished:
|
||||
//
|
||||
// Question: we remember the load failure so that we will not try again.
|
||||
// User needs to check whether the fuction pointer is NULL
|
||||
//
|
||||
m_pfnAspNetCoreCreateApplication = g_pfnAspNetCoreCreateApplication;
|
||||
m_pfnAspNetCoreCreateRequestHandler = g_pfnAspNetCoreCreateRequestHandler;
|
||||
if (!g_fAspnetcoreRHLoadedError && FAILED(hr))
|
||||
{
|
||||
g_fAspnetcoreRHLoadedError = TRUE;
|
||||
}
|
||||
|
||||
if (fLocked)
|
||||
{
|
||||
ReleaseSRWLockExclusive(&g_srwLock);
|
||||
}
|
||||
return hr;
|
||||
}
|
||||
|
||||
HRESULT
|
||||
APPLICATION_INFO::FindNativeAssemblyFromGlobalLocation(STRU* struFilename)
|
||||
{
|
||||
HRESULT hr = S_OK;
|
||||
DWORD dwSize = MAX_PATH;
|
||||
BOOL fDone = FALSE;
|
||||
DWORD dwPosition = 0;
|
||||
|
||||
// Though we could call LoadLibrary(L"aspnetcorerh.dll") relying the OS to solve
|
||||
// the path (the targeted dll is the same folder of w3wp.exe/iisexpress)
|
||||
// let's still load with full path to avoid security issue
|
||||
while (!fDone)
|
||||
{
|
||||
DWORD dwReturnedSize = GetModuleFileName(NULL, struFilename->QueryStr(), dwSize);
|
||||
if (dwReturnedSize == 0)
|
||||
{
|
||||
hr = HRESULT_FROM_WIN32(GetLastError());
|
||||
fDone = TRUE;
|
||||
goto Finished;
|
||||
}
|
||||
else if ((dwReturnedSize == dwSize) && (GetLastError() == ERROR_INSUFFICIENT_BUFFER))
|
||||
{
|
||||
dwSize *= 2; // smaller buffer. increase the buffer and retry
|
||||
struFilename->Resize(dwSize + 20); // aspnetcorerh.dll
|
||||
}
|
||||
else
|
||||
{
|
||||
fDone = TRUE;
|
||||
}
|
||||
}
|
||||
|
||||
if (FAILED(hr = struFilename->SyncWithBuffer()))
|
||||
{
|
||||
goto Finished;
|
||||
}
|
||||
dwPosition = struFilename->LastIndexOf(L'\\', 0);
|
||||
struFilename->QueryStr()[dwPosition] = L'\0';
|
||||
|
||||
if (FAILED(hr = struFilename->SyncWithBuffer()) ||
|
||||
FAILED(hr = struFilename->Append(g_pwzAspnetcoreRequestHandlerName)))
|
||||
{
|
||||
goto Finished;
|
||||
}
|
||||
|
||||
Finished:
|
||||
return hr;
|
||||
}
|
||||
|
|
@ -0,0 +1,269 @@
|
|||
// Copyright (c) .NET Foundation. All rights reserved.
|
||||
// Licensed under the MIT License. See License.txt in the project root for license information.
|
||||
|
||||
#include "precomp.hxx"
|
||||
|
||||
APPLICATION_MANAGER* APPLICATION_MANAGER::sm_pApplicationManager = NULL;
|
||||
|
||||
HRESULT
|
||||
APPLICATION_MANAGER::GetApplicationInfo(
|
||||
_In_ IHttpServer* pServer,
|
||||
_In_ ASPNETCORE_CONFIG* pConfig,
|
||||
_Out_ APPLICATION_INFO ** ppApplicationInfo
|
||||
)
|
||||
{
|
||||
HRESULT hr = S_OK;
|
||||
APPLICATION_INFO *pApplicationInfo = NULL;
|
||||
APPLICATION_INFO_KEY key;
|
||||
BOOL fExclusiveLock = FALSE;
|
||||
BOOL fMixedHostingModelError = FALSE;
|
||||
BOOL fDuplicatedInProcessApp = FALSE;
|
||||
PCWSTR pszApplicationId = NULL;
|
||||
LPCWSTR apsz[1];
|
||||
STACK_STRU ( strEventMsg, 256 );
|
||||
|
||||
*ppApplicationInfo = NULL;
|
||||
|
||||
DBG_ASSERT(pServer != NULL);
|
||||
DBG_ASSERT(pConfig != NULL);
|
||||
|
||||
pszApplicationId = pConfig->QueryConfigPath()->QueryStr();
|
||||
|
||||
hr = key.Initialize(pszApplicationId);
|
||||
if (FAILED(hr))
|
||||
{
|
||||
goto Finished;
|
||||
}
|
||||
|
||||
AcquireSRWLockShared(&m_srwLock);
|
||||
if (m_fInShutdown)
|
||||
{
|
||||
ReleaseSRWLockShared(&m_srwLock);
|
||||
hr = HRESULT_FROM_WIN32(ERROR_SERVER_SHUTDOWN_IN_PROGRESS);
|
||||
goto Finished;
|
||||
}
|
||||
m_pApplicationInfoHash->FindKey(&key, ppApplicationInfo);
|
||||
ReleaseSRWLockShared(&m_srwLock);
|
||||
|
||||
if (*ppApplicationInfo == NULL)
|
||||
{
|
||||
switch (pConfig->QueryHostingModel())
|
||||
{
|
||||
case HOSTING_IN_PROCESS:
|
||||
if (m_pApplicationInfoHash->Count() > 0)
|
||||
{
|
||||
// Only one inprocess app is allowed per IIS worker process
|
||||
fDuplicatedInProcessApp = TRUE;
|
||||
hr = HRESULT_FROM_WIN32(ERROR_APP_INIT_FAILURE);
|
||||
goto Finished;
|
||||
}
|
||||
break;
|
||||
|
||||
case HOSTING_OUT_PROCESS:
|
||||
break;
|
||||
|
||||
default:
|
||||
hr = E_UNEXPECTED;
|
||||
goto Finished;
|
||||
}
|
||||
pApplicationInfo = new APPLICATION_INFO(pServer);
|
||||
if (pApplicationInfo == NULL)
|
||||
{
|
||||
hr = E_OUTOFMEMORY;
|
||||
goto Finished;
|
||||
}
|
||||
|
||||
AcquireSRWLockExclusive(&m_srwLock);
|
||||
fExclusiveLock = TRUE;
|
||||
if (m_fInShutdown)
|
||||
{
|
||||
// Already in shuting down. No need to create the application
|
||||
hr = HRESULT_FROM_WIN32(ERROR_SERVER_SHUTDOWN_IN_PROGRESS);
|
||||
goto Finished;
|
||||
}
|
||||
m_pApplicationInfoHash->FindKey(&key, ppApplicationInfo);
|
||||
|
||||
if (*ppApplicationInfo != NULL)
|
||||
{
|
||||
// someone else created the application
|
||||
delete pApplicationInfo;
|
||||
pApplicationInfo = NULL;
|
||||
goto Finished;
|
||||
}
|
||||
|
||||
// hosting model check. We do not allow mixed scenario for now
|
||||
// could be changed in the future
|
||||
if (m_hostingModel != HOSTING_UNKNOWN)
|
||||
{
|
||||
if (m_hostingModel != pConfig->QueryHostingModel())
|
||||
{
|
||||
// hosting model does not match, error out
|
||||
fMixedHostingModelError = TRUE;
|
||||
hr = HRESULT_FROM_WIN32(ERROR_APP_INIT_FAILURE);
|
||||
goto Finished;
|
||||
}
|
||||
}
|
||||
|
||||
hr = pApplicationInfo->Initialize(pConfig, m_pFileWatcher);
|
||||
if (FAILED(hr))
|
||||
{
|
||||
goto Finished;
|
||||
}
|
||||
|
||||
hr = m_pApplicationInfoHash->InsertRecord( pApplicationInfo );
|
||||
if (FAILED(hr))
|
||||
{
|
||||
goto Finished;
|
||||
}
|
||||
|
||||
//
|
||||
// first application will decide which hosting model allowed by this process
|
||||
//
|
||||
if (m_hostingModel == HOSTING_UNKNOWN)
|
||||
{
|
||||
m_hostingModel = pConfig->QueryHostingModel();
|
||||
}
|
||||
|
||||
*ppApplicationInfo = pApplicationInfo;
|
||||
ReleaseSRWLockExclusive(&m_srwLock);
|
||||
fExclusiveLock = FALSE;
|
||||
|
||||
pApplicationInfo->StartMonitoringAppOffline();
|
||||
pApplicationInfo = NULL;
|
||||
}
|
||||
|
||||
Finished:
|
||||
|
||||
if (fExclusiveLock)
|
||||
{
|
||||
ReleaseSRWLockExclusive(&m_srwLock);
|
||||
}
|
||||
|
||||
if (pApplicationInfo != NULL)
|
||||
{
|
||||
pApplicationInfo->DereferenceApplicationInfo();
|
||||
pApplicationInfo = NULL;
|
||||
}
|
||||
|
||||
if (FAILED(hr))
|
||||
{
|
||||
if (fDuplicatedInProcessApp)
|
||||
{
|
||||
if (SUCCEEDED(strEventMsg.SafeSnwprintf(
|
||||
ASPNETCORE_EVENT_DUPLICATED_INPROCESS_APP_MSG,
|
||||
pszApplicationId)))
|
||||
{
|
||||
/*apsz[0] = strEventMsg.QueryStr();
|
||||
if (FORWARDING_HANDLER::QueryEventLog() != NULL)
|
||||
{
|
||||
ReportEventW(FORWARDING_HANDLER::QueryEventLog(),
|
||||
EVENTLOG_ERROR_TYPE,
|
||||
0,
|
||||
ASPNETCORE_EVENT_DUPLICATED_INPROCESS_APP,
|
||||
NULL,
|
||||
1,
|
||||
0,
|
||||
apsz,
|
||||
NULL);
|
||||
}*/
|
||||
}
|
||||
}
|
||||
else if (fMixedHostingModelError)
|
||||
{
|
||||
//if (SUCCEEDED(strEventMsg.SafeSnwprintf(
|
||||
// ASPNETCORE_EVENT_MIXED_HOSTING_MODEL_ERROR_MSG,
|
||||
// pszApplicationId,
|
||||
// pConfig->QueryHostingModelStr())))
|
||||
//{
|
||||
// apsz[0] = strEventMsg.QueryStr();
|
||||
// /*if (FORWARDING_HANDLER::QueryEventLog() != NULL)
|
||||
// {
|
||||
// ReportEventW(FORWARDING_HANDLER::QueryEventLog(),
|
||||
// EVENTLOG_ERROR_TYPE,
|
||||
// 0,
|
||||
// ASPNETCORE_EVENT_MIXED_HOSTING_MODEL_ERROR,
|
||||
// NULL,
|
||||
// 1,
|
||||
// 0,
|
||||
// apsz,
|
||||
// NULL);
|
||||
// }*/
|
||||
//}
|
||||
}
|
||||
else
|
||||
{
|
||||
if (SUCCEEDED(strEventMsg.SafeSnwprintf(
|
||||
ASPNETCORE_EVENT_ADD_APPLICATION_ERROR_MSG,
|
||||
pszApplicationId,
|
||||
hr)))
|
||||
{
|
||||
apsz[0] = strEventMsg.QueryStr();
|
||||
/*if (FORWARDING_HANDLER::QueryEventLog() != NULL)
|
||||
{
|
||||
ReportEventW(FORWARDING_HANDLER::QueryEventLog(),
|
||||
EVENTLOG_ERROR_TYPE,
|
||||
0,
|
||||
ASPNETCORE_EVENT_ADD_APPLICATION_ERROR,
|
||||
NULL,
|
||||
1,
|
||||
0,
|
||||
apsz,
|
||||
NULL);
|
||||
}*/
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
return hr;
|
||||
}
|
||||
|
||||
HRESULT
|
||||
APPLICATION_MANAGER::RecycleApplication(
|
||||
_In_ LPCWSTR pszApplicationId
|
||||
)
|
||||
{
|
||||
HRESULT hr = S_OK;
|
||||
APPLICATION_INFO_KEY key;
|
||||
|
||||
hr = key.Initialize(pszApplicationId);
|
||||
if (FAILED(hr))
|
||||
{
|
||||
goto Finished;
|
||||
}
|
||||
AcquireSRWLockExclusive(&m_srwLock);
|
||||
m_pApplicationInfoHash->DeleteKey(&key);
|
||||
|
||||
if (m_pApplicationInfoHash->Count() == 0)
|
||||
{
|
||||
m_hostingModel = HOSTING_UNKNOWN;
|
||||
}
|
||||
|
||||
if (g_fAspnetcoreRHLoadedError)
|
||||
{
|
||||
// We had assembly loading failure
|
||||
// this error blocked the start of all applications
|
||||
// Let's recycle the worker process if user redeployed any application
|
||||
g_pHttpServer->RecycleProcess(L"AspNetCore Recycle Process on Demand due to assembly loading failure");
|
||||
}
|
||||
|
||||
ReleaseSRWLockExclusive(&m_srwLock);
|
||||
|
||||
Finished:
|
||||
|
||||
return hr;
|
||||
}
|
||||
|
||||
VOID
|
||||
APPLICATION_MANAGER::ShutDown()
|
||||
{
|
||||
m_fInShutdown = TRUE;
|
||||
if (m_pApplicationInfoHash != NULL)
|
||||
{
|
||||
AcquireSRWLockExclusive(&m_srwLock);
|
||||
|
||||
// clean up the hash table so that the application will be informed on shutdown
|
||||
m_pApplicationInfoHash->Clear();
|
||||
ReleaseSRWLockExclusive(&m_srwLock);
|
||||
}
|
||||
|
||||
}
|
||||
|
|
@ -0,0 +1,208 @@
|
|||
// Copyright (c) .NET Foundation. All rights reserved.
|
||||
// Licensed under the MIT License. See License.txt in the project root for license information.
|
||||
|
||||
#include "precomp.hxx"
|
||||
#include <IPHlpApi.h>
|
||||
|
||||
HTTP_MODULE_ID g_pModuleId = NULL;
|
||||
IHttpServer * g_pHttpServer = NULL;
|
||||
BOOL g_fRecycleProcessCalled = FALSE;
|
||||
PCWSTR g_pszModuleName = NULL;
|
||||
HINSTANCE g_hModule;
|
||||
HMODULE g_hAspnetCoreRH = NULL;
|
||||
BOOL g_fAspnetcoreRHAssemblyLoaded = FALSE;
|
||||
BOOL g_fAspnetcoreRHLoadedError = FALSE;
|
||||
DWORD g_dwAspNetCoreDebugFlags = 0;
|
||||
DWORD g_dwActiveServerProcesses = 0;
|
||||
SRWLOCK g_srwLock;
|
||||
DWORD g_dwDebugFlags = 0;
|
||||
PCSTR g_szDebugLabel = "ASPNET_CORE_MODULE";
|
||||
PCWSTR g_pwzAspnetcoreRequestHandlerName = L"\\aspnetcorerh.dll";
|
||||
PFN_ASPNETCORE_CREATE_APPLICATION g_pfnAspNetCoreCreateApplication;
|
||||
PFN_ASPNETCORE_CREATE_REQUEST_HANDLER g_pfnAspNetCoreCreateRequestHandler;
|
||||
|
||||
VOID
|
||||
StaticCleanup()
|
||||
{
|
||||
APPLICATION_MANAGER::Cleanup();
|
||||
}
|
||||
|
||||
BOOL WINAPI DllMain(HMODULE hModule,
|
||||
DWORD ul_reason_for_call,
|
||||
LPVOID lpReserved
|
||||
)
|
||||
{
|
||||
UNREFERENCED_PARAMETER(lpReserved);
|
||||
|
||||
switch (ul_reason_for_call)
|
||||
{
|
||||
case DLL_PROCESS_ATTACH:
|
||||
g_hModule = hModule;
|
||||
DisableThreadLibraryCalls(hModule);
|
||||
break;
|
||||
case DLL_PROCESS_DETACH:
|
||||
StaticCleanup();
|
||||
default:
|
||||
break;
|
||||
}
|
||||
|
||||
return TRUE;
|
||||
}
|
||||
|
||||
HRESULT
|
||||
__stdcall
|
||||
RegisterModule(
|
||||
DWORD dwServerVersion,
|
||||
IHttpModuleRegistrationInfo * pModuleInfo,
|
||||
IHttpServer * pHttpServer
|
||||
)
|
||||
/*++
|
||||
|
||||
Routine description:
|
||||
|
||||
Function called by IIS immediately after loading the module, used to let
|
||||
IIS know what notifications the module is interested in
|
||||
|
||||
Arguments:
|
||||
|
||||
dwServerVersion - IIS version the module is being loaded on
|
||||
pModuleInfo - info regarding this module
|
||||
pHttpServer - callback functions which can be used by the module at
|
||||
any point
|
||||
|
||||
Return value:
|
||||
|
||||
HRESULT
|
||||
|
||||
--*/
|
||||
{
|
||||
HRESULT hr = S_OK;
|
||||
HKEY hKey;
|
||||
BOOL fDisableANCM = FALSE;
|
||||
ASPNET_CORE_PROXY_MODULE_FACTORY * pFactory = NULL;
|
||||
ASPNET_CORE_GLOBAL_MODULE * pGlobalModule = NULL;
|
||||
APPLICATION_MANAGER * pApplicationManager = NULL;
|
||||
|
||||
UNREFERENCED_PARAMETER(dwServerVersion);
|
||||
|
||||
#ifdef DEBUG
|
||||
CREATE_DEBUG_PRINT_OBJECT("Asp.Net Core Module");
|
||||
g_dwDebugFlags = DEBUG_FLAGS_ANY;
|
||||
#endif // DEBUG
|
||||
|
||||
CREATE_DEBUG_PRINT_OBJECT;
|
||||
|
||||
//LoadGlobalConfiguration();
|
||||
|
||||
InitializeSRWLock(&g_srwLock);
|
||||
|
||||
g_pModuleId = pModuleInfo->GetId();
|
||||
g_pszModuleName = pModuleInfo->GetName();
|
||||
g_pHttpServer = pHttpServer;
|
||||
|
||||
// check whether the feature is disabled due to security reason
|
||||
if (RegOpenKeyEx(HKEY_LOCAL_MACHINE,
|
||||
L"SOFTWARE\\Microsoft\\IIS Extensions\\IIS AspNetCore Module\\Parameters",
|
||||
0,
|
||||
KEY_READ,
|
||||
&hKey) == NO_ERROR)
|
||||
{
|
||||
DWORD dwType;
|
||||
DWORD dwData;
|
||||
DWORD cbData;
|
||||
|
||||
cbData = sizeof(dwData);
|
||||
if ((RegQueryValueEx(hKey,
|
||||
L"DisableANCM",
|
||||
NULL,
|
||||
&dwType,
|
||||
(LPBYTE)&dwData,
|
||||
&cbData) == NO_ERROR) &&
|
||||
(dwType == REG_DWORD))
|
||||
{
|
||||
fDisableANCM = (dwData != 0);
|
||||
}
|
||||
}
|
||||
|
||||
if (fDisableANCM)
|
||||
{
|
||||
// Logging
|
||||
goto Finished;
|
||||
}
|
||||
|
||||
//
|
||||
// Create the factory before any static initialization.
|
||||
// The ASPNET_CORE_PROXY_MODULE_FACTORY::Terminate method will clean any
|
||||
// static object initialized.
|
||||
//
|
||||
pFactory = new ASPNET_CORE_PROXY_MODULE_FACTORY;
|
||||
|
||||
if (pFactory == NULL)
|
||||
{
|
||||
hr = E_OUTOFMEMORY;
|
||||
goto Finished;
|
||||
}
|
||||
|
||||
hr = pModuleInfo->SetRequestNotifications(
|
||||
pFactory,
|
||||
RQ_EXECUTE_REQUEST_HANDLER,
|
||||
0);
|
||||
if (FAILED(hr))
|
||||
{
|
||||
goto Finished;
|
||||
}
|
||||
|
||||
pFactory = NULL;
|
||||
pApplicationManager = APPLICATION_MANAGER::GetInstance();
|
||||
if(pApplicationManager == NULL)
|
||||
{
|
||||
hr = E_OUTOFMEMORY;
|
||||
goto Finished;
|
||||
}
|
||||
|
||||
hr = pApplicationManager->Initialize();
|
||||
if(FAILED(hr))
|
||||
{
|
||||
goto Finished;
|
||||
}
|
||||
pGlobalModule = NULL;
|
||||
|
||||
pGlobalModule = new ASPNET_CORE_GLOBAL_MODULE(pApplicationManager);
|
||||
if (pGlobalModule == NULL)
|
||||
{
|
||||
hr = E_OUTOFMEMORY;
|
||||
goto Finished;
|
||||
}
|
||||
|
||||
hr = pModuleInfo->SetGlobalNotifications(
|
||||
pGlobalModule,
|
||||
GL_CONFIGURATION_CHANGE | GL_STOP_LISTENING);
|
||||
|
||||
if (FAILED(hr))
|
||||
{
|
||||
goto Finished;
|
||||
}
|
||||
pGlobalModule = NULL;
|
||||
|
||||
hr = ALLOC_CACHE_HANDLER::StaticInitialize();
|
||||
if (FAILED(hr))
|
||||
{
|
||||
goto Finished;
|
||||
}
|
||||
|
||||
Finished:
|
||||
if (pGlobalModule != NULL)
|
||||
{
|
||||
delete pGlobalModule;
|
||||
pGlobalModule = NULL;
|
||||
}
|
||||
|
||||
if (pFactory != NULL)
|
||||
{
|
||||
pFactory->Terminate();
|
||||
pFactory = NULL;
|
||||
}
|
||||
|
||||
return hr;
|
||||
}
|
||||
|
||||
|
|
@ -0,0 +1,466 @@
|
|||
// Copyright (c) .NET Foundation. All rights reserved.
|
||||
// Licensed under the MIT License. See License.txt in the project root for license information.
|
||||
|
||||
#include "precomp.hxx"
|
||||
|
||||
FILE_WATCHER::FILE_WATCHER() :
|
||||
m_hCompletionPort(NULL),
|
||||
m_hChangeNotificationThread(NULL)
|
||||
{
|
||||
}
|
||||
|
||||
FILE_WATCHER::~FILE_WATCHER()
|
||||
{
|
||||
if (m_hChangeNotificationThread != NULL)
|
||||
{
|
||||
PostQueuedCompletionStatus(m_hCompletionPort, 0, FILE_WATCHER_SHUTDOWN_KEY, NULL);
|
||||
WaitForSingleObject(m_hChangeNotificationThread, INFINITE);
|
||||
CloseHandle(m_hChangeNotificationThread);
|
||||
m_hChangeNotificationThread = NULL;
|
||||
}
|
||||
|
||||
if (NULL != m_hCompletionPort)
|
||||
{
|
||||
CloseHandle(m_hCompletionPort);
|
||||
m_hCompletionPort = NULL;
|
||||
}
|
||||
}
|
||||
|
||||
HRESULT
|
||||
FILE_WATCHER::Create(
|
||||
VOID
|
||||
)
|
||||
{
|
||||
HRESULT hr = S_OK;
|
||||
|
||||
m_hCompletionPort = CreateIoCompletionPort(INVALID_HANDLE_VALUE,
|
||||
NULL,
|
||||
0,
|
||||
0);
|
||||
|
||||
if (m_hCompletionPort == NULL)
|
||||
{
|
||||
hr = HRESULT_FROM_WIN32(GetLastError());
|
||||
goto Finished;
|
||||
}
|
||||
|
||||
m_hChangeNotificationThread = CreateThread(NULL,
|
||||
0,
|
||||
ChangeNotificationThread,
|
||||
this,
|
||||
0,
|
||||
NULL);
|
||||
|
||||
if (m_hChangeNotificationThread == NULL)
|
||||
{
|
||||
hr = HRESULT_FROM_WIN32(GetLastError());
|
||||
|
||||
CloseHandle(m_hCompletionPort);
|
||||
m_hCompletionPort = NULL;
|
||||
|
||||
goto Finished;
|
||||
}
|
||||
|
||||
Finished:
|
||||
return hr;
|
||||
}
|
||||
|
||||
DWORD
|
||||
WINAPI
|
||||
FILE_WATCHER::ChangeNotificationThread(
|
||||
LPVOID pvArg
|
||||
)
|
||||
/*++
|
||||
|
||||
Routine Description:
|
||||
|
||||
IO completion thread
|
||||
|
||||
Arguments:
|
||||
|
||||
None
|
||||
|
||||
Return Value:
|
||||
|
||||
Win32 error
|
||||
|
||||
--*/
|
||||
{
|
||||
FILE_WATCHER * pFileMonitor;
|
||||
BOOL fSuccess = FALSE;
|
||||
DWORD cbCompletion = 0;
|
||||
OVERLAPPED * pOverlapped = NULL;
|
||||
DWORD dwErrorStatus;
|
||||
ULONG_PTR completionKey;
|
||||
|
||||
pFileMonitor = (FILE_WATCHER*)pvArg;
|
||||
DBG_ASSERT(pFileMonitor != NULL);
|
||||
|
||||
while (TRUE)
|
||||
{
|
||||
fSuccess = GetQueuedCompletionStatus(
|
||||
pFileMonitor->m_hCompletionPort,
|
||||
&cbCompletion,
|
||||
&completionKey,
|
||||
&pOverlapped,
|
||||
INFINITE);
|
||||
|
||||
DBG_ASSERT(fSuccess);
|
||||
DebugPrint(1, "FILE_WATCHER::ChangeNotificationThread");
|
||||
dwErrorStatus = fSuccess ? ERROR_SUCCESS : GetLastError();
|
||||
|
||||
if (completionKey == FILE_WATCHER_SHUTDOWN_KEY)
|
||||
{
|
||||
break;
|
||||
}
|
||||
|
||||
DBG_ASSERT(pOverlapped != NULL);
|
||||
if (pOverlapped != NULL)
|
||||
{
|
||||
FileWatcherCompletionRoutine(
|
||||
dwErrorStatus,
|
||||
cbCompletion,
|
||||
pOverlapped);
|
||||
}
|
||||
pOverlapped = NULL;
|
||||
cbCompletion = 0;
|
||||
}
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
VOID
|
||||
WINAPI
|
||||
FILE_WATCHER::FileWatcherCompletionRoutine(
|
||||
DWORD dwCompletionStatus,
|
||||
DWORD cbCompletion,
|
||||
OVERLAPPED * pOverlapped
|
||||
)
|
||||
/*++
|
||||
|
||||
Routine Description:
|
||||
|
||||
Called when ReadDirectoryChangesW() completes
|
||||
|
||||
Arguments:
|
||||
|
||||
dwCompletionStatus - Error of completion
|
||||
cbCompletion - Bytes of completion
|
||||
pOverlapped - State of completion
|
||||
|
||||
Return Value:
|
||||
|
||||
None
|
||||
|
||||
--*/
|
||||
{
|
||||
FILE_WATCHER_ENTRY * pMonitorEntry;
|
||||
pMonitorEntry = CONTAINING_RECORD(pOverlapped, FILE_WATCHER_ENTRY, _overlapped);
|
||||
pMonitorEntry->DereferenceFileWatcherEntry();
|
||||
DBG_ASSERT(pMonitorEntry != NULL);
|
||||
|
||||
pMonitorEntry->HandleChangeCompletion(dwCompletionStatus, cbCompletion);
|
||||
|
||||
if (pMonitorEntry->QueryIsValid())
|
||||
{
|
||||
//
|
||||
// Continue monitoring
|
||||
//
|
||||
pMonitorEntry->Monitor();
|
||||
}
|
||||
else
|
||||
{
|
||||
//
|
||||
// Marked by application distructor
|
||||
// Deference the entry to delete it
|
||||
//
|
||||
pMonitorEntry->DereferenceFileWatcherEntry();
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
FILE_WATCHER_ENTRY::FILE_WATCHER_ENTRY(FILE_WATCHER * pFileMonitor) :
|
||||
_pFileMonitor(pFileMonitor),
|
||||
_hDirectory(INVALID_HANDLE_VALUE),
|
||||
_hImpersonationToken(NULL),
|
||||
_pApplicationInfo(NULL),
|
||||
_lStopMonitorCalled(0),
|
||||
_cRefs(1),
|
||||
_fIsValid(TRUE)
|
||||
{
|
||||
_dwSignature = FILE_WATCHER_ENTRY_SIGNATURE;
|
||||
InitializeSRWLock(&_srwLock);
|
||||
}
|
||||
|
||||
FILE_WATCHER_ENTRY::~FILE_WATCHER_ENTRY()
|
||||
{
|
||||
_dwSignature = FILE_WATCHER_ENTRY_SIGNATURE_FREE;
|
||||
|
||||
if (_hDirectory != INVALID_HANDLE_VALUE)
|
||||
{
|
||||
CloseHandle(_hDirectory);
|
||||
_hDirectory = INVALID_HANDLE_VALUE;
|
||||
}
|
||||
|
||||
if (_hImpersonationToken != NULL)
|
||||
{
|
||||
CloseHandle(_hImpersonationToken);
|
||||
_hImpersonationToken = NULL;
|
||||
}
|
||||
}
|
||||
|
||||
#pragma warning(disable:4100)
|
||||
|
||||
HRESULT
|
||||
FILE_WATCHER_ENTRY::HandleChangeCompletion(
|
||||
_In_ DWORD dwCompletionStatus,
|
||||
_In_ DWORD cbCompletion
|
||||
)
|
||||
/*++
|
||||
|
||||
Routine Description:
|
||||
|
||||
Handle change notification (see if any of associated config files
|
||||
need to be flushed)
|
||||
|
||||
Arguments:
|
||||
|
||||
dwCompletionStatus - Completion status
|
||||
cbCompletion - Bytes of completion
|
||||
|
||||
Return Value:
|
||||
|
||||
HRESULT
|
||||
|
||||
--*/
|
||||
{
|
||||
HRESULT hr = S_OK;
|
||||
FILE_NOTIFY_INFORMATION * pNotificationInfo;
|
||||
BOOL fFileChanged = FALSE;
|
||||
|
||||
AcquireSRWLockExclusive(&_srwLock);
|
||||
if (!_fIsValid)
|
||||
{
|
||||
goto Finished;
|
||||
}
|
||||
|
||||
// When directory handle is closed then HandleChangeCompletion
|
||||
// happens with cbCompletion = 0 and dwCompletionStatus = 0
|
||||
// From documentation it is not clear if that combination
|
||||
// of return values is specific to closing handles or whether
|
||||
// it could also mean an error condition. Hence we will maintain
|
||||
// explicit flag that will help us determine if entry
|
||||
// is being shutdown (StopMonitor() called)
|
||||
//
|
||||
if (_lStopMonitorCalled)
|
||||
{
|
||||
goto Finished;
|
||||
}
|
||||
|
||||
//
|
||||
// There could be a FCN overflow
|
||||
// Let assume the file got changed instead of checking files
|
||||
// Othersie we have to cache the file info
|
||||
//
|
||||
if (cbCompletion == 0)
|
||||
{
|
||||
fFileChanged = TRUE;
|
||||
}
|
||||
else
|
||||
{
|
||||
pNotificationInfo = (FILE_NOTIFY_INFORMATION*)_buffDirectoryChanges.QueryPtr();
|
||||
DBG_ASSERT(pNotificationInfo != NULL);
|
||||
|
||||
while (pNotificationInfo != NULL)
|
||||
{
|
||||
//
|
||||
// check whether the monitored file got changed
|
||||
//
|
||||
if (_wcsnicmp(pNotificationInfo->FileName,
|
||||
_strFileName.QueryStr(),
|
||||
pNotificationInfo->FileNameLength / sizeof(WCHAR)) == 0)
|
||||
{
|
||||
fFileChanged = TRUE;
|
||||
break;
|
||||
}
|
||||
//
|
||||
// Advance to next notification
|
||||
//
|
||||
if (pNotificationInfo->NextEntryOffset == 0)
|
||||
{
|
||||
pNotificationInfo = NULL;
|
||||
}
|
||||
else
|
||||
{
|
||||
pNotificationInfo = (FILE_NOTIFY_INFORMATION*)
|
||||
((PBYTE)pNotificationInfo +
|
||||
pNotificationInfo->NextEntryOffset);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
if (fFileChanged)
|
||||
{
|
||||
//
|
||||
// so far we only monitoring app_offline
|
||||
//
|
||||
_pApplicationInfo->UpdateAppOfflineFileHandle();
|
||||
}
|
||||
|
||||
Finished:
|
||||
ReleaseSRWLockExclusive(&_srwLock);
|
||||
return hr;
|
||||
}
|
||||
|
||||
#pragma warning( error : 4100 )
|
||||
|
||||
HRESULT
|
||||
FILE_WATCHER_ENTRY::Monitor(VOID)
|
||||
{
|
||||
HRESULT hr = S_OK;
|
||||
DWORD cbRead;
|
||||
|
||||
AcquireSRWLockExclusive(&_srwLock);
|
||||
ReferenceFileWatcherEntry();
|
||||
ZeroMemory(&_overlapped, sizeof(_overlapped));
|
||||
|
||||
if (!ReadDirectoryChangesW(_hDirectory,
|
||||
_buffDirectoryChanges.QueryPtr(),
|
||||
_buffDirectoryChanges.QuerySize(),
|
||||
FALSE, // Watching sub dirs. Set to False now as only monitoring app_offline
|
||||
FILE_NOTIFY_VALID_MASK & ~FILE_NOTIFY_CHANGE_LAST_ACCESS,
|
||||
&cbRead,
|
||||
&_overlapped,
|
||||
NULL))
|
||||
{
|
||||
hr = HRESULT_FROM_WIN32(GetLastError());
|
||||
}
|
||||
|
||||
ReleaseSRWLockExclusive(&_srwLock);
|
||||
return hr;
|
||||
}
|
||||
|
||||
VOID
|
||||
FILE_WATCHER_ENTRY::StopMonitor(VOID)
|
||||
{
|
||||
//
|
||||
// Flag that monitoring is being stopped so that
|
||||
// we know that HandleChangeCompletion() call
|
||||
// can be ignored
|
||||
//
|
||||
InterlockedExchange(&_lStopMonitorCalled, 1);
|
||||
|
||||
AcquireSRWLockExclusive(&_srwLock);
|
||||
|
||||
if (_hDirectory != INVALID_HANDLE_VALUE)
|
||||
{
|
||||
CloseHandle(_hDirectory);
|
||||
_hDirectory = INVALID_HANDLE_VALUE;
|
||||
}
|
||||
|
||||
ReleaseSRWLockExclusive(&_srwLock);
|
||||
}
|
||||
|
||||
HRESULT
|
||||
FILE_WATCHER_ENTRY::Create(
|
||||
_In_ PCWSTR pszDirectoryToMonitor,
|
||||
_In_ PCWSTR pszFileNameToMonitor,
|
||||
_In_ APPLICATION_INFO* pApplicationInfo,
|
||||
_In_ HANDLE hImpersonationToken
|
||||
)
|
||||
{
|
||||
HRESULT hr = S_OK;
|
||||
BOOL fRet = FALSE;
|
||||
|
||||
if (pszDirectoryToMonitor == NULL ||
|
||||
pszFileNameToMonitor == NULL ||
|
||||
pApplicationInfo == NULL)
|
||||
{
|
||||
DBG_ASSERT(FALSE);
|
||||
hr = HRESULT_FROM_WIN32(ERROR_INVALID_PARAMETER);
|
||||
goto Finished;
|
||||
}
|
||||
|
||||
//
|
||||
//remember the application
|
||||
//
|
||||
_pApplicationInfo = pApplicationInfo;
|
||||
|
||||
if (FAILED(hr = _strFileName.Copy(pszFileNameToMonitor)))
|
||||
{
|
||||
goto Finished;
|
||||
}
|
||||
|
||||
if (FAILED(hr = _strDirectoryName.Copy(pszDirectoryToMonitor)))
|
||||
{
|
||||
goto Finished;
|
||||
}
|
||||
|
||||
//
|
||||
// Resize change buffer to something "reasonable"
|
||||
//
|
||||
if (!_buffDirectoryChanges.Resize(FILE_WATCHER_ENTRY_BUFFER_SIZE))
|
||||
{
|
||||
hr = HRESULT_FROM_WIN32(ERROR_NOT_ENOUGH_MEMORY);
|
||||
goto Finished;
|
||||
}
|
||||
|
||||
if (hImpersonationToken != NULL)
|
||||
{
|
||||
fRet = DuplicateHandle(GetCurrentProcess(),
|
||||
hImpersonationToken,
|
||||
GetCurrentProcess(),
|
||||
&_hImpersonationToken,
|
||||
0,
|
||||
FALSE,
|
||||
DUPLICATE_SAME_ACCESS);
|
||||
|
||||
if (!fRet)
|
||||
{
|
||||
hr = HRESULT_FROM_WIN32(GetLastError());
|
||||
goto Finished;
|
||||
}
|
||||
}
|
||||
else
|
||||
{
|
||||
if (_hImpersonationToken != NULL)
|
||||
{
|
||||
CloseHandle(_hImpersonationToken);
|
||||
_hImpersonationToken = NULL;
|
||||
}
|
||||
}
|
||||
|
||||
_hDirectory = CreateFileW(
|
||||
_strDirectoryName.QueryStr(),
|
||||
FILE_LIST_DIRECTORY,
|
||||
FILE_SHARE_READ | FILE_SHARE_WRITE | FILE_SHARE_DELETE,
|
||||
NULL,
|
||||
OPEN_EXISTING,
|
||||
FILE_FLAG_BACKUP_SEMANTICS | FILE_FLAG_OVERLAPPED,
|
||||
NULL);
|
||||
|
||||
if (_hDirectory == INVALID_HANDLE_VALUE)
|
||||
{
|
||||
hr = HRESULT_FROM_WIN32(GetLastError());
|
||||
goto Finished;
|
||||
}
|
||||
|
||||
if (CreateIoCompletionPort(
|
||||
_hDirectory,
|
||||
_pFileMonitor->QueryCompletionPort(),
|
||||
NULL,
|
||||
0) == NULL)
|
||||
{
|
||||
hr = HRESULT_FROM_WIN32(GetLastError());
|
||||
goto Finished;
|
||||
}
|
||||
|
||||
//
|
||||
// Start monitoring
|
||||
//
|
||||
hr = Monitor();
|
||||
|
||||
Finished:
|
||||
|
||||
return hr;
|
||||
}
|
||||
|
|
@ -0,0 +1,59 @@
|
|||
#include "precomp.hxx"
|
||||
|
||||
ASPNET_CORE_GLOBAL_MODULE::ASPNET_CORE_GLOBAL_MODULE(
|
||||
APPLICATION_MANAGER* pApplicationManager)
|
||||
{
|
||||
m_pApplicationManager = pApplicationManager;
|
||||
}
|
||||
|
||||
//
|
||||
// Is called when IIS decided to terminate worker process
|
||||
// Shut down all core apps
|
||||
//
|
||||
GLOBAL_NOTIFICATION_STATUS
|
||||
ASPNET_CORE_GLOBAL_MODULE::OnGlobalStopListening(
|
||||
_In_ IGlobalStopListeningProvider * pProvider
|
||||
)
|
||||
{
|
||||
UNREFERENCED_PARAMETER(pProvider);
|
||||
|
||||
if (m_pApplicationManager != NULL)
|
||||
{
|
||||
// we should let application manager to shudown all allication
|
||||
// and dereference it as some requests may still reference to application manager
|
||||
m_pApplicationManager->ShutDown();
|
||||
m_pApplicationManager = NULL;
|
||||
}
|
||||
|
||||
// Return processing to the pipeline.
|
||||
return GL_NOTIFICATION_CONTINUE;
|
||||
}
|
||||
|
||||
//
|
||||
// Is called when configuration changed
|
||||
// Recycled the corresponding core app if its configuration changed
|
||||
//
|
||||
GLOBAL_NOTIFICATION_STATUS
|
||||
ASPNET_CORE_GLOBAL_MODULE::OnGlobalConfigurationChange(
|
||||
_In_ IGlobalConfigurationChangeProvider * pProvider
|
||||
)
|
||||
{
|
||||
UNREFERENCED_PARAMETER(pProvider);
|
||||
|
||||
// Retrieve the path that has changed.
|
||||
PCWSTR pwszChangePath = pProvider->GetChangePath();
|
||||
|
||||
// Test for an error.
|
||||
if (NULL != pwszChangePath &&
|
||||
_wcsicmp(pwszChangePath, L"MACHINE") != 0 &&
|
||||
_wcsicmp(pwszChangePath, L"MACHINE/WEBROOT") != 0)
|
||||
{
|
||||
if (m_pApplicationManager != NULL)
|
||||
{
|
||||
m_pApplicationManager->RecycleApplication(pwszChangePath);
|
||||
}
|
||||
}
|
||||
|
||||
// Return processing to the pipeline.
|
||||
return GL_NOTIFICATION_CONTINUE;
|
||||
}
|
||||
|
|
@ -0,0 +1,155 @@
|
|||
// Copyright (c) .NET Foundation. All rights reserved.
|
||||
// Licensed under the MIT License. See License.txt in the project root for license information.
|
||||
|
||||
#pragma once
|
||||
#pragma warning( disable : 4091)
|
||||
|
||||
//
|
||||
// System related headers
|
||||
//
|
||||
#define _WINSOCKAPI_
|
||||
|
||||
#define NTDDI_VERSION 0x06010000
|
||||
#define WINVER 0x0601
|
||||
#define _WIN32_WINNT 0x0601
|
||||
|
||||
#include <windows.h>
|
||||
#include <atlbase.h>
|
||||
#include <pdh.h>
|
||||
|
||||
//#include <ntassert.h>
|
||||
#include <Shlobj.h>
|
||||
#include <httpserv.h>
|
||||
|
||||
// This should remove our issue of compiling for win7 without header files.
|
||||
// We force the Windows 8 version check logic in iiswebsocket.h to succeed even though we're compiling for Windows 7.
|
||||
// Then, we set the version defines back to Windows 7 to for the remainder of the compilation.
|
||||
#undef NTDDI_VERSION
|
||||
#undef WINVER
|
||||
#undef _WIN32_WINNT
|
||||
#define NTDDI_VERSION 0x06020000
|
||||
#define WINVER 0x0602
|
||||
#define _WIN32_WINNT 0x0602
|
||||
#include <iiswebsocket.h>
|
||||
#undef NTDDI_VERSION
|
||||
#undef WINVER
|
||||
#undef _WIN32_WINNT
|
||||
|
||||
#define NTDDI_VERSION 0x06010000
|
||||
#define WINVER 0x0601
|
||||
#define _WIN32_WINNT 0x0601
|
||||
|
||||
#include <httptrace.h>
|
||||
#include <winhttp.h>
|
||||
|
||||
#include <cstdlib>
|
||||
#include <vector>
|
||||
#include <wchar.h>
|
||||
|
||||
//
|
||||
// Option available starting Windows 8.
|
||||
// 111 is the value in SDK on May 15, 2012.
|
||||
//
|
||||
#ifndef WINHTTP_OPTION_ASSURED_NON_BLOCKING_CALLBACKS
|
||||
#define WINHTTP_OPTION_ASSURED_NON_BLOCKING_CALLBACKS 111
|
||||
#endif
|
||||
|
||||
#ifdef max
|
||||
#undef max
|
||||
template<typename T> inline T max(T a, T b)
|
||||
{
|
||||
return a > b ? a : b;
|
||||
}
|
||||
#endif
|
||||
|
||||
#ifdef min
|
||||
#undef min
|
||||
template<typename T> inline T min(T a, T b)
|
||||
{
|
||||
return a < b ? a : b;
|
||||
}
|
||||
#endif
|
||||
|
||||
inline bool IsSpace(char ch)
|
||||
{
|
||||
switch(ch)
|
||||
{
|
||||
case 32: // ' '
|
||||
case 9: // '\t'
|
||||
case 10: // '\n'
|
||||
case 13: // '\r'
|
||||
case 11: // '\v'
|
||||
case 12: // '\f'
|
||||
return true;
|
||||
default:
|
||||
return false;
|
||||
}
|
||||
}
|
||||
|
||||
#include <hashfn.h>
|
||||
#include <hashtable.h>
|
||||
#include "stringa.h"
|
||||
#include "stringu.h"
|
||||
#include "dbgutil.h"
|
||||
#include "ahutil.h"
|
||||
#include "multisz.h"
|
||||
#include "multisza.h"
|
||||
#include "base64.h"
|
||||
#include <listentry.h>
|
||||
#include <datetime.h>
|
||||
#include <reftrace.h>
|
||||
#include <acache.h>
|
||||
#include <time.h>
|
||||
|
||||
#include "..\..\CommonLib\environmentvariablehash.h"
|
||||
#include "..\..\CommonLib\aspnetcoreconfig.h"
|
||||
#include "..\..\CommonLib\application.h"
|
||||
#include "..\..\CommonLib\utility.h"
|
||||
#include "..\..\CommonLib\debugutil.h"
|
||||
#include "..\..\CommonLib\requesthandler.h"
|
||||
//#include "..\aspnetcore_msg.h"
|
||||
//#include "aspnetcore_event.h"
|
||||
#include "appoffline.h"
|
||||
#include "filewatcher.h"
|
||||
#include "applicationinfo.h"
|
||||
#include "applicationmanager.h"
|
||||
#include "globalmodule.h"
|
||||
#include "resource.h"
|
||||
#include "proxymodule.h"
|
||||
|
||||
|
||||
FORCEINLINE
|
||||
DWORD
|
||||
WIN32_FROM_HRESULT(
|
||||
HRESULT hr
|
||||
)
|
||||
{
|
||||
if ((FAILED(hr)) &&
|
||||
(HRESULT_FACILITY(hr) == FACILITY_WIN32))
|
||||
{
|
||||
return HRESULT_CODE(hr);
|
||||
}
|
||||
return hr;
|
||||
}
|
||||
|
||||
FORCEINLINE
|
||||
HRESULT
|
||||
HRESULT_FROM_GETLASTERROR()
|
||||
{
|
||||
return ( GetLastError() != NO_ERROR )
|
||||
? HRESULT_FROM_WIN32( GetLastError() )
|
||||
: E_FAIL;
|
||||
}
|
||||
|
||||
extern PVOID g_pModuleId;
|
||||
extern BOOL g_fAspnetcoreRHAssemblyLoaded;
|
||||
extern BOOL g_fAspnetcoreRHLoadedError;
|
||||
extern BOOL g_fEnableReferenceCountTracing;
|
||||
extern DWORD g_dwActiveServerProcesses;
|
||||
extern HMODULE g_hAspnetCoreRH;
|
||||
extern SRWLOCK g_srwLock;
|
||||
extern PCWSTR g_pwzAspnetcoreRequestHandlerName;
|
||||
|
||||
extern PFN_ASPNETCORE_CREATE_APPLICATION g_pfnAspNetCoreCreateApplication;
|
||||
extern PFN_ASPNETCORE_CREATE_REQUEST_HANDLER g_pfnAspNetCoreCreateRequestHandler;
|
||||
#pragma warning( error : 4091)
|
||||
|
|
@ -0,0 +1,192 @@
|
|||
// Copyright (c) .NET Foundation. All rights reserved.
|
||||
// Licensed under the MIT License. See License.txt in the project root for license information.
|
||||
|
||||
#include "precomp.hxx"
|
||||
|
||||
__override
|
||||
HRESULT
|
||||
ASPNET_CORE_PROXY_MODULE_FACTORY::GetHttpModule(
|
||||
CHttpModule ** ppModule,
|
||||
IModuleAllocator * pAllocator
|
||||
)
|
||||
{
|
||||
ASPNET_CORE_PROXY_MODULE *pModule = new (pAllocator) ASPNET_CORE_PROXY_MODULE();
|
||||
if (pModule == NULL)
|
||||
{
|
||||
return E_OUTOFMEMORY;
|
||||
}
|
||||
|
||||
*ppModule = pModule;
|
||||
return S_OK;
|
||||
}
|
||||
|
||||
__override
|
||||
VOID
|
||||
ASPNET_CORE_PROXY_MODULE_FACTORY::Terminate(
|
||||
VOID
|
||||
)
|
||||
/*++
|
||||
|
||||
Routine description:
|
||||
|
||||
Function called by IIS for global (non-request-specific) notifications
|
||||
|
||||
Arguments:
|
||||
|
||||
None.
|
||||
|
||||
Return value:
|
||||
|
||||
None
|
||||
|
||||
--*/
|
||||
{
|
||||
/* FORWARDING_HANDLER::StaticTerminate();
|
||||
|
||||
WEBSOCKET_HANDLER::StaticTerminate();*/
|
||||
|
||||
ALLOC_CACHE_HANDLER::StaticTerminate();
|
||||
|
||||
delete this;
|
||||
}
|
||||
|
||||
ASPNET_CORE_PROXY_MODULE::ASPNET_CORE_PROXY_MODULE(
|
||||
) : m_pApplicationInfo(NULL), m_pHandler(NULL)
|
||||
{
|
||||
}
|
||||
|
||||
ASPNET_CORE_PROXY_MODULE::~ASPNET_CORE_PROXY_MODULE()
|
||||
{
|
||||
if (m_pApplicationInfo != NULL)
|
||||
{
|
||||
m_pApplicationInfo->DereferenceApplicationInfo();
|
||||
m_pApplicationInfo = NULL;
|
||||
}
|
||||
|
||||
if (m_pHandler != NULL)
|
||||
{
|
||||
m_pHandler->DereferenceRequestHandler();
|
||||
m_pHandler = NULL;
|
||||
}
|
||||
}
|
||||
|
||||
__override
|
||||
REQUEST_NOTIFICATION_STATUS
|
||||
ASPNET_CORE_PROXY_MODULE::OnExecuteRequestHandler(
|
||||
IHttpContext * pHttpContext,
|
||||
IHttpEventProvider *
|
||||
)
|
||||
{
|
||||
HRESULT hr = S_OK;
|
||||
ASPNETCORE_CONFIG *pConfig = NULL;
|
||||
APPLICATION_MANAGER *pApplicationManager = NULL;
|
||||
REQUEST_NOTIFICATION_STATUS retVal = RQ_NOTIFICATION_CONTINUE;
|
||||
APPLICATION* pApplication = NULL;
|
||||
STACK_STRU(struFileName, 256);
|
||||
|
||||
hr = ASPNETCORE_CONFIG::GetConfig(g_pHttpServer, g_pModuleId, pHttpContext, &pConfig);
|
||||
if (FAILED(hr))
|
||||
{
|
||||
goto Finished;
|
||||
}
|
||||
|
||||
pApplicationManager = APPLICATION_MANAGER::GetInstance();
|
||||
if (pApplicationManager == NULL)
|
||||
{
|
||||
hr = E_OUTOFMEMORY;
|
||||
goto Finished;
|
||||
}
|
||||
|
||||
hr = pApplicationManager->GetApplicationInfo(
|
||||
g_pHttpServer,
|
||||
pConfig,
|
||||
&m_pApplicationInfo);
|
||||
if (FAILED(hr))
|
||||
{
|
||||
goto Finished;
|
||||
}
|
||||
|
||||
// app_offline check to avoid loading aspnetcorerh.dll unnecessarily
|
||||
if (m_pApplicationInfo->AppOfflineFound())
|
||||
{
|
||||
// servicing app_offline
|
||||
HTTP_DATA_CHUNK DataChunk;
|
||||
IHttpResponse *pResponse = NULL;
|
||||
APP_OFFLINE_HTM *pAppOfflineHtm = NULL;
|
||||
|
||||
pResponse = pHttpContext->GetResponse();
|
||||
pAppOfflineHtm = m_pApplicationInfo->QueryAppOfflineHtm();
|
||||
DBG_ASSERT(pAppOfflineHtm);
|
||||
DBG_ASSERT(pResponse);
|
||||
|
||||
// Ignore failure hresults as nothing we can do
|
||||
// Set fTrySkipCustomErrors to true as we want client see the offline content
|
||||
pResponse->SetStatus(503, "Service Unavailable", 0, hr, NULL, TRUE);
|
||||
pResponse->SetHeader("Content-Type",
|
||||
"text/html",
|
||||
(USHORT)strlen("text/html"),
|
||||
FALSE
|
||||
);
|
||||
|
||||
DataChunk.DataChunkType = HttpDataChunkFromMemory;
|
||||
DataChunk.FromMemory.pBuffer = (PVOID)pAppOfflineHtm->m_Contents.QueryStr();
|
||||
DataChunk.FromMemory.BufferLength = pAppOfflineHtm->m_Contents.QueryCB();
|
||||
pResponse->WriteEntityChunkByReference(&DataChunk);
|
||||
|
||||
retVal = RQ_NOTIFICATION_FINISH_REQUEST;
|
||||
goto Finished;
|
||||
}
|
||||
|
||||
// make sure assmebly is loaded and application is created
|
||||
|
||||
hr = m_pApplicationInfo->EnsureApplicationCreated();
|
||||
if (FAILED(hr))
|
||||
{
|
||||
goto Finished;
|
||||
}
|
||||
pApplication = m_pApplicationInfo->QueryApplication();
|
||||
DBG_ASSERT(pApplication);
|
||||
|
||||
// make sure application is in running state
|
||||
// cannot recreate the application as we cannot reload clr for inprocess
|
||||
if (pApplication->QueryStatus() != APPLICATION_STATUS::RUNNING)
|
||||
{
|
||||
hr = HRESULT_FROM_WIN32(ERROR_SERVER_DISABLED);
|
||||
goto Finished;
|
||||
}
|
||||
|
||||
// Create RequestHandler and process the request
|
||||
hr = m_pApplicationInfo->QueryCreateRequestHandler()(pHttpContext,
|
||||
(HTTP_MODULE_ID*) &g_pModuleId,
|
||||
pApplication,
|
||||
&m_pHandler);
|
||||
|
||||
if (FAILED(hr))
|
||||
{
|
||||
goto Finished;
|
||||
}
|
||||
retVal = m_pHandler->OnExecuteRequestHandler();
|
||||
|
||||
Finished:
|
||||
if (FAILED(hr))
|
||||
{
|
||||
pHttpContext->GetResponse()->SetStatus(500, "Internal Server Error", 0, hr);
|
||||
retVal = RQ_NOTIFICATION_FINISH_REQUEST;
|
||||
}
|
||||
return retVal;
|
||||
}
|
||||
|
||||
__override
|
||||
REQUEST_NOTIFICATION_STATUS
|
||||
ASPNET_CORE_PROXY_MODULE::OnAsyncCompletion(
|
||||
IHttpContext *,
|
||||
DWORD,
|
||||
BOOL,
|
||||
IHttpEventProvider *,
|
||||
IHttpCompletionInfo * pCompletionInfo
|
||||
)
|
||||
{
|
||||
return m_pHandler->OnAsyncCompletion(
|
||||
pCompletionInfo->GetCompletionBytes(),
|
||||
pCompletionInfo->GetCompletionStatus());
|
||||
}
|
||||
|
|
@ -0,0 +1,42 @@
|
|||
<!--
|
||||
|
||||
IIS Asp.Net Core Extension Schema
|
||||
|
||||
** Please DO NOT edit this file yourself. **
|
||||
|
||||
If you want to add configuration sections to the schema, you may place
|
||||
them in .xml files similar to this one, in this directory. They will be
|
||||
picked up automatically on startup.
|
||||
|
||||
-->
|
||||
|
||||
<configSchema>
|
||||
<sectionSchema name="system.webServer/aspNetCore">
|
||||
<attribute name="processPath" type="string" expanded="true"/>
|
||||
<attribute name="arguments" type="string" expanded="true" defaultValue=""/>
|
||||
<attribute name="startupTimeLimit" type="uint" defaultValue="120" validationType="integerRange" validationParameter="0,3600"/>
|
||||
<!-- in seconds -->
|
||||
<attribute name="shutdownTimeLimit" type="uint" defaultValue="10" validationType="integerRange" validationParameter="0,600"/>
|
||||
<!-- in seconds -->
|
||||
<attribute name="rapidFailsPerMinute" type="uint" defaultValue="10" validationType="integerRange" validationParameter="0,100"/>
|
||||
<attribute name="requestTimeout" type="timeSpan" defaultValue="00:02:00" validationType="timeSpanRange" validationParameter="0,1296000,60"/>
|
||||
<attribute name="stdoutLogEnabled" type="bool" defaultValue="false" />
|
||||
<attribute name="stdoutLogFile" type="string" defaultValue=".\aspnetcore-stdout" expanded="true"/>
|
||||
<attribute name="processesPerApplication" type="uint" defaultValue="1" validationType="integerRange" validationParameter="1,100"/>
|
||||
<attribute name="forwardWindowsAuthToken" type="bool" defaultValue="true" />
|
||||
<attribute name="disableStartUpErrorPage" type="bool" defaultValue="false" />
|
||||
<attribute name="hostingModel" type="string" />
|
||||
<element name="recycleOnFileChange">
|
||||
|
||||
<collection addElement="file" clearElement="clear">
|
||||
<attribute name="path" type="string" required="true" validationType="nonEmptyString" expanded="true"/>
|
||||
</collection>
|
||||
</element>
|
||||
<element name="environmentVariables">
|
||||
<collection addElement="environmentVariable" clearElement="clear" >
|
||||
<attribute name="name" type="string" required="true" validationType="nonEmptyString"/>
|
||||
<attribute name="value" type="string" required="true"/>
|
||||
</collection>
|
||||
</element>
|
||||
</sectionSchema>
|
||||
</configSchema>
|
||||
|
|
@ -0,0 +1,117 @@
|
|||
// Microsoft Visual C++ generated resource script.
|
||||
//
|
||||
#include <windows.h>
|
||||
#include "version.h"
|
||||
#include "resource.h"
|
||||
/////////////////////////////////////////////////////////////////////////////
|
||||
// English (United States) resources
|
||||
|
||||
#if !defined(AFX_RESOURCE_DLL) || defined(AFX_TARG_ENU)
|
||||
LANGUAGE LANG_ENGLISH, SUBLANG_ENGLISH_US
|
||||
|
||||
/////////////////////////////////////////////////////////////////////////////
|
||||
//
|
||||
// 11
|
||||
//
|
||||
|
||||
//1 11
|
||||
//BEGIN
|
||||
// 0x0001, 0x0000, 0x03e8, 0x0000, 0x03ed, 0x0000, 0x0010, 0x0000, 0x0010,
|
||||
// 0x0001, 0x0025, 0x0031, 0x000d, 0x000a, 0x0000, 0x0000, 0x0010, 0x0001,
|
||||
// 0x0025, 0x0031, 0x000d, 0x000a, 0x0000, 0x0000, 0x0010, 0x0001, 0x0025,
|
||||
// 0x0031, 0x000d, 0x000a, 0x0000, 0x0000, 0x0010, 0x0001, 0x0025, 0x0031,
|
||||
// 0x000d, 0x000a, 0x0000, 0x0000, 0x0010, 0x0001, 0x0025, 0x0031, 0x000d,
|
||||
// 0x000a, 0x0000, 0x0000, 0x0010, 0x0001, 0x0025, 0x0031, 0x000d, 0x000a,
|
||||
// 0x0000, 0x0000
|
||||
//END
|
||||
|
||||
|
||||
#ifdef APSTUDIO_INVOKED
|
||||
/////////////////////////////////////////////////////////////////////////////
|
||||
//
|
||||
// TEXTINCLUDE
|
||||
//
|
||||
|
||||
1 TEXTINCLUDE
|
||||
BEGIN
|
||||
"resource.h\0"
|
||||
END
|
||||
|
||||
2 TEXTINCLUDE
|
||||
BEGIN
|
||||
"\0"
|
||||
END
|
||||
|
||||
3 TEXTINCLUDE
|
||||
BEGIN
|
||||
"\r\n"
|
||||
"\0"
|
||||
END
|
||||
|
||||
#endif // APSTUDIO_INVOKED
|
||||
|
||||
|
||||
/////////////////////////////////////////////////////////////////////////////
|
||||
//
|
||||
// Version
|
||||
//
|
||||
|
||||
VS_VERSION_INFO VERSIONINFO
|
||||
FILEVERSION FileVersion
|
||||
PRODUCTVERSION ProductVersion
|
||||
FILEFLAGSMASK 0x3fL
|
||||
#ifdef _DEBUG
|
||||
FILEFLAGS 0x1L
|
||||
#else
|
||||
FILEFLAGS 0x0L
|
||||
#endif
|
||||
FILEOS 0x40004L
|
||||
FILETYPE 0x2L
|
||||
FILESUBTYPE 0x0L
|
||||
BEGIN
|
||||
BLOCK "StringFileInfo"
|
||||
BEGIN
|
||||
BLOCK "040904b0"
|
||||
BEGIN
|
||||
VALUE "CompanyName", "Microsoft"
|
||||
VALUE "FileDescription", "IIS ASP.NET Core Module"
|
||||
VALUE "FileVersion", FileVersionStr
|
||||
VALUE "InternalName", "aspnetcore.dll"
|
||||
VALUE "LegalCopyright", "Copyright (C) 2016"
|
||||
VALUE "OriginalFilename", "aspnetcore.dll"
|
||||
VALUE "ProductName", "ASP.NET Core Module"
|
||||
VALUE "ProductVersion", ProductVersionStr
|
||||
END
|
||||
END
|
||||
BLOCK "VarFileInfo"
|
||||
BEGIN
|
||||
VALUE "Translation", 0x409, 1200
|
||||
END
|
||||
END
|
||||
|
||||
|
||||
/////////////////////////////////////////////////////////////////////////////
|
||||
//
|
||||
// String Table
|
||||
//
|
||||
|
||||
STRINGTABLE
|
||||
BEGIN
|
||||
IDS_INVALID_PROPERTY "Property name '%s' in system.webServer/aspNetCore section has invalid value '%s' which does not conform to the prescribed format"
|
||||
IDS_SERVER_ERROR "There was a connection error while trying to route the request."
|
||||
END
|
||||
|
||||
#endif // English (United States) resources
|
||||
/////////////////////////////////////////////////////////////////////////////
|
||||
|
||||
|
||||
|
||||
#ifndef APSTUDIO_INVOKED
|
||||
/////////////////////////////////////////////////////////////////////////////
|
||||
//
|
||||
// Generated from the TEXTINCLUDE 3 resource.
|
||||
//
|
||||
|
||||
|
||||
/////////////////////////////////////////////////////////////////////////////
|
||||
#endif // not APSTUDIO_INVOKED
|
||||
|
|
@ -0,0 +1,18 @@
|
|||
//{{NO_DEPENDENCIES}}
|
||||
// Microsoft Visual C++ generated include file.
|
||||
// Used by aspnetcoremodule.rc
|
||||
//
|
||||
#define ASPNETCORE_EVENT_MSG_BUFFER_SIZE 256
|
||||
#define IDS_INVALID_PROPERTY 1000
|
||||
#define IDS_SERVER_ERROR 1001
|
||||
|
||||
// Next default values for new objects
|
||||
//
|
||||
#ifdef APSTUDIO_INVOKED
|
||||
#ifndef APSTUDIO_READONLY_SYMBOLS
|
||||
#define _APS_NEXT_RESOURCE_VALUE 101
|
||||
#define _APS_NEXT_COMMAND_VALUE 40001
|
||||
#define _APS_NEXT_CONTROL_VALUE 1000
|
||||
#define _APS_NEXT_SYMED_VALUE 101
|
||||
#endif
|
||||
#endif
|
||||
|
|
@ -0,0 +1,204 @@
|
|||
<?xml version="1.0" encoding="utf-8"?>
|
||||
<Project DefaultTargets="Build" ToolsVersion="15.0" xmlns="http://schemas.microsoft.com/developer/msbuild/2003">
|
||||
<ItemGroup Label="ProjectConfigurations">
|
||||
<ProjectConfiguration Include="Debug|Win32">
|
||||
<Configuration>Debug</Configuration>
|
||||
<Platform>Win32</Platform>
|
||||
</ProjectConfiguration>
|
||||
<ProjectConfiguration Include="Release|Win32">
|
||||
<Configuration>Release</Configuration>
|
||||
<Platform>Win32</Platform>
|
||||
</ProjectConfiguration>
|
||||
<ProjectConfiguration Include="Debug|x64">
|
||||
<Configuration>Debug</Configuration>
|
||||
<Platform>x64</Platform>
|
||||
</ProjectConfiguration>
|
||||
<ProjectConfiguration Include="Release|x64">
|
||||
<Configuration>Release</Configuration>
|
||||
<Platform>x64</Platform>
|
||||
</ProjectConfiguration>
|
||||
</ItemGroup>
|
||||
<PropertyGroup Label="Globals">
|
||||
<VCProjectVersion>15.0</VCProjectVersion>
|
||||
<ProjectGuid>{55494E58-E061-4C4C-A0A8-837008E72F85}</ProjectGuid>
|
||||
<Keyword>Win32Proj</Keyword>
|
||||
<RootNamespace>NewCommon</RootNamespace>
|
||||
<WindowsTargetPlatformVersion>10.0.15063.0</WindowsTargetPlatformVersion>
|
||||
</PropertyGroup>
|
||||
<Import Project="$(VCTargetsPath)\Microsoft.Cpp.Default.props" />
|
||||
<PropertyGroup Condition="'$(Configuration)|$(Platform)'=='Debug|Win32'" Label="Configuration">
|
||||
<ConfigurationType>StaticLibrary</ConfigurationType>
|
||||
<UseDebugLibraries>true</UseDebugLibraries>
|
||||
<PlatformToolset>v141</PlatformToolset>
|
||||
<CharacterSet>Unicode</CharacterSet>
|
||||
</PropertyGroup>
|
||||
<PropertyGroup Condition="'$(Configuration)|$(Platform)'=='Release|Win32'" Label="Configuration">
|
||||
<ConfigurationType>StaticLibrary</ConfigurationType>
|
||||
<UseDebugLibraries>false</UseDebugLibraries>
|
||||
<PlatformToolset>v141</PlatformToolset>
|
||||
<WholeProgramOptimization>true</WholeProgramOptimization>
|
||||
<CharacterSet>Unicode</CharacterSet>
|
||||
</PropertyGroup>
|
||||
<PropertyGroup Condition="'$(Configuration)|$(Platform)'=='Debug|x64'" Label="Configuration">
|
||||
<ConfigurationType>StaticLibrary</ConfigurationType>
|
||||
<UseDebugLibraries>true</UseDebugLibraries>
|
||||
<PlatformToolset>v141</PlatformToolset>
|
||||
<CharacterSet>Unicode</CharacterSet>
|
||||
</PropertyGroup>
|
||||
<PropertyGroup Condition="'$(Configuration)|$(Platform)'=='Release|x64'" Label="Configuration">
|
||||
<ConfigurationType>StaticLibrary</ConfigurationType>
|
||||
<UseDebugLibraries>false</UseDebugLibraries>
|
||||
<PlatformToolset>v141</PlatformToolset>
|
||||
<WholeProgramOptimization>false</WholeProgramOptimization>
|
||||
<CharacterSet>Unicode</CharacterSet>
|
||||
</PropertyGroup>
|
||||
<Import Project="$(VCTargetsPath)\Microsoft.Cpp.props" />
|
||||
<ImportGroup Label="ExtensionSettings">
|
||||
</ImportGroup>
|
||||
<ImportGroup Label="Shared">
|
||||
</ImportGroup>
|
||||
<ImportGroup Label="PropertySheets" Condition="'$(Configuration)|$(Platform)'=='Debug|Win32'">
|
||||
<Import Project="$(UserRootDir)\Microsoft.Cpp.$(Platform).user.props" Condition="exists('$(UserRootDir)\Microsoft.Cpp.$(Platform).user.props')" Label="LocalAppDataPlatform" />
|
||||
</ImportGroup>
|
||||
<ImportGroup Label="PropertySheets" Condition="'$(Configuration)|$(Platform)'=='Release|Win32'">
|
||||
<Import Project="$(UserRootDir)\Microsoft.Cpp.$(Platform).user.props" Condition="exists('$(UserRootDir)\Microsoft.Cpp.$(Platform).user.props')" Label="LocalAppDataPlatform" />
|
||||
</ImportGroup>
|
||||
<ImportGroup Label="PropertySheets" Condition="'$(Configuration)|$(Platform)'=='Debug|x64'">
|
||||
<Import Project="$(UserRootDir)\Microsoft.Cpp.$(Platform).user.props" Condition="exists('$(UserRootDir)\Microsoft.Cpp.$(Platform).user.props')" Label="LocalAppDataPlatform" />
|
||||
</ImportGroup>
|
||||
<ImportGroup Label="PropertySheets" Condition="'$(Configuration)|$(Platform)'=='Release|x64'">
|
||||
<Import Project="$(UserRootDir)\Microsoft.Cpp.$(Platform).user.props" Condition="exists('$(UserRootDir)\Microsoft.Cpp.$(Platform).user.props')" Label="LocalAppDataPlatform" />
|
||||
</ImportGroup>
|
||||
<PropertyGroup Label="UserMacros" />
|
||||
<PropertyGroup Condition="'$(Configuration)|$(Platform)'=='Debug|Win32'">
|
||||
<LinkIncremental>true</LinkIncremental>
|
||||
</PropertyGroup>
|
||||
<PropertyGroup Condition="'$(Configuration)|$(Platform)'=='Debug|x64'">
|
||||
<LinkIncremental>true</LinkIncremental>
|
||||
</PropertyGroup>
|
||||
<PropertyGroup Condition="'$(Configuration)|$(Platform)'=='Release|Win32'">
|
||||
<LinkIncremental>false</LinkIncremental>
|
||||
</PropertyGroup>
|
||||
<PropertyGroup Condition="'$(Configuration)|$(Platform)'=='Release|x64'">
|
||||
<LinkIncremental>false</LinkIncremental>
|
||||
<IncludePath>C:\AspNetCoreModule\src\IISLib;$(IncludePath)</IncludePath>
|
||||
</PropertyGroup>
|
||||
<ItemDefinitionGroup Condition="'$(Configuration)|$(Platform)'=='Debug|Win32'">
|
||||
<ClCompile>
|
||||
<PrecompiledHeader>Use</PrecompiledHeader>
|
||||
<WarningLevel>Level4</WarningLevel>
|
||||
<TreatWarningAsError>true</TreatWarningAsError>
|
||||
<Optimization>Disabled</Optimization>
|
||||
<SDLCheck>false</SDLCheck>
|
||||
<PreprocessorDefinitions>WIN32;_DEBUG;_LIB;%(PreprocessorDefinitions)</PreprocessorDefinitions>
|
||||
<ConformanceMode>true</ConformanceMode>
|
||||
<RuntimeLibrary>MultiThreadedDebug</RuntimeLibrary>
|
||||
<ShowIncludes>false</ShowIncludes>
|
||||
</ClCompile>
|
||||
<Link>
|
||||
<SubSystem>Windows</SubSystem>
|
||||
<GenerateDebugInformation>true</GenerateDebugInformation>
|
||||
</Link>
|
||||
</ItemDefinitionGroup>
|
||||
<ItemDefinitionGroup Condition="'$(Configuration)|$(Platform)'=='Debug|x64'">
|
||||
<ClCompile>
|
||||
<PrecompiledHeader>Use</PrecompiledHeader>
|
||||
<WarningLevel>Level4</WarningLevel>
|
||||
<TreatWarningAsError>true</TreatWarningAsError>
|
||||
<Optimization>Disabled</Optimization>
|
||||
<SDLCheck>false</SDLCheck>
|
||||
<PreprocessorDefinitions>_DEBUG;_LIB;%(PreprocessorDefinitions)</PreprocessorDefinitions>
|
||||
<ConformanceMode>true</ConformanceMode>
|
||||
<DebugInformationFormat>ProgramDatabase</DebugInformationFormat>
|
||||
<MinimalRebuild>false</MinimalRebuild>
|
||||
<RuntimeLibrary>MultiThreadedDebug</RuntimeLibrary>
|
||||
<ShowIncludes>false</ShowIncludes>
|
||||
</ClCompile>
|
||||
<Link>
|
||||
<SubSystem>Windows</SubSystem>
|
||||
<GenerateDebugInformation>true</GenerateDebugInformation>
|
||||
</Link>
|
||||
</ItemDefinitionGroup>
|
||||
<ItemDefinitionGroup Condition="'$(Configuration)|$(Platform)'=='Release|Win32'">
|
||||
<ClCompile>
|
||||
<PrecompiledHeader>Use</PrecompiledHeader>
|
||||
<WarningLevel>Level4</WarningLevel>
|
||||
<TreatWarningAsError>true</TreatWarningAsError>
|
||||
<Optimization>MaxSpeed</Optimization>
|
||||
<FunctionLevelLinking>true</FunctionLevelLinking>
|
||||
<IntrinsicFunctions>true</IntrinsicFunctions>
|
||||
<SDLCheck>false</SDLCheck>
|
||||
<PreprocessorDefinitions>WIN32;NDEBUG;_LIB;%(PreprocessorDefinitions)</PreprocessorDefinitions>
|
||||
<ConformanceMode>true</ConformanceMode>
|
||||
<RuntimeLibrary>MultiThreaded</RuntimeLibrary>
|
||||
<ShowIncludes>false</ShowIncludes>
|
||||
</ClCompile>
|
||||
<Link>
|
||||
<SubSystem>Windows</SubSystem>
|
||||
<EnableCOMDATFolding>true</EnableCOMDATFolding>
|
||||
<OptimizeReferences>true</OptimizeReferences>
|
||||
<GenerateDebugInformation>true</GenerateDebugInformation>
|
||||
</Link>
|
||||
</ItemDefinitionGroup>
|
||||
<ItemDefinitionGroup Condition="'$(Configuration)|$(Platform)'=='Release|x64'">
|
||||
<ClCompile>
|
||||
<PrecompiledHeader>Use</PrecompiledHeader>
|
||||
<WarningLevel>Level4</WarningLevel>
|
||||
<TreatWarningAsError>true</TreatWarningAsError>
|
||||
<Optimization>MaxSpeed</Optimization>
|
||||
<FunctionLevelLinking>true</FunctionLevelLinking>
|
||||
<IntrinsicFunctions>true</IntrinsicFunctions>
|
||||
<SDLCheck>false</SDLCheck>
|
||||
<PreprocessorDefinitions>NDEBUG;_LIB;%(PreprocessorDefinitions)</PreprocessorDefinitions>
|
||||
<ConformanceMode>true</ConformanceMode>
|
||||
<AdditionalIncludeDirectories>
|
||||
</AdditionalIncludeDirectories>
|
||||
<AdditionalUsingDirectories>
|
||||
</AdditionalUsingDirectories>
|
||||
<RuntimeLibrary>MultiThreaded</RuntimeLibrary>
|
||||
<ShowIncludes>false</ShowIncludes>
|
||||
</ClCompile>
|
||||
<Link>
|
||||
<SubSystem>Windows</SubSystem>
|
||||
<EnableCOMDATFolding>true</EnableCOMDATFolding>
|
||||
<OptimizeReferences>true</OptimizeReferences>
|
||||
<GenerateDebugInformation>true</GenerateDebugInformation>
|
||||
</Link>
|
||||
<Lib>
|
||||
<AdditionalLibraryDirectories>..\iislib</AdditionalLibraryDirectories>
|
||||
</Lib>
|
||||
</ItemDefinitionGroup>
|
||||
<ItemGroup>
|
||||
<ClInclude Include="application.h" />
|
||||
<ClInclude Include="aspnetcoreconfig.h" />
|
||||
<ClInclude Include="debugutil.h" />
|
||||
<ClInclude Include="environmentvariablehash.h" />
|
||||
<ClInclude Include="fx_ver.h" />
|
||||
<ClInclude Include="hostfxr_utility.h" />
|
||||
<ClInclude Include="requesthandler.h" />
|
||||
<ClInclude Include="stdafx.h" />
|
||||
<ClInclude Include="utility.h" />
|
||||
</ItemGroup>
|
||||
<ItemGroup>
|
||||
<ClCompile Include="application.cpp" />
|
||||
<ClCompile Include="aspnetcoreconfig.cxx" />
|
||||
<ClCompile Include="fx_ver.cxx" />
|
||||
<ClCompile Include="hostfxr_utility.cpp" />
|
||||
<ClCompile Include="requesthandler.cxx" />
|
||||
<ClCompile Include="stdafx.cpp">
|
||||
<PrecompiledHeader Condition="'$(Configuration)|$(Platform)'=='Debug|Win32'">Create</PrecompiledHeader>
|
||||
<PrecompiledHeader Condition="'$(Configuration)|$(Platform)'=='Debug|x64'">Create</PrecompiledHeader>
|
||||
<PrecompiledHeader Condition="'$(Configuration)|$(Platform)'=='Release|Win32'">Create</PrecompiledHeader>
|
||||
<PrecompiledHeader Condition="'$(Configuration)|$(Platform)'=='Release|x64'">Create</PrecompiledHeader>
|
||||
</ClCompile>
|
||||
<ClCompile Include="utility.cxx" />
|
||||
</ItemGroup>
|
||||
<ItemGroup>
|
||||
<ProjectReference Include="..\IISLib\IISLib.vcxproj">
|
||||
<Project>{4787a64f-9a3e-4867-a55a-70cb4b2b2ffe}</Project>
|
||||
</ProjectReference>
|
||||
</ItemGroup>
|
||||
<Import Project="$(VCTargetsPath)\Microsoft.Cpp.targets" />
|
||||
<ImportGroup Label="ExtensionTargets">
|
||||
</ImportGroup>
|
||||
</Project>
|
||||
|
|
@ -0,0 +1,50 @@
|
|||
// Copyright (c) .NET Foundation. All rights reserved.
|
||||
// Licensed under the MIT License. See License.txt in the project root for license information.
|
||||
|
||||
#include "stdafx.h"
|
||||
|
||||
APPLICATION::APPLICATION(
|
||||
_In_ IHttpServer* pHttpServer,
|
||||
_In_ ASPNETCORE_CONFIG* pConfig) :
|
||||
m_cRefs(1),
|
||||
m_pHttpServer(pHttpServer),
|
||||
m_pConfig(pConfig),
|
||||
m_status(APPLICATION_STATUS::UNKNOWN)
|
||||
{
|
||||
}
|
||||
|
||||
APPLICATION::~APPLICATION()
|
||||
{
|
||||
}
|
||||
|
||||
APPLICATION_STATUS
|
||||
APPLICATION::QueryStatus()
|
||||
{
|
||||
return m_status;
|
||||
}
|
||||
|
||||
ASPNETCORE_CONFIG*
|
||||
APPLICATION::QueryConfig()
|
||||
{
|
||||
return m_pConfig;
|
||||
}
|
||||
|
||||
VOID
|
||||
APPLICATION::ReferenceApplication()
|
||||
const
|
||||
{
|
||||
InterlockedIncrement(&m_cRefs);
|
||||
}
|
||||
|
||||
VOID
|
||||
APPLICATION::DereferenceApplication()
|
||||
const
|
||||
{
|
||||
DBG_ASSERT(m_cRefs != 0);
|
||||
|
||||
LONG cRefs = 0;
|
||||
if ((cRefs = InterlockedDecrement(&m_cRefs)) == 0)
|
||||
{
|
||||
delete this;
|
||||
}
|
||||
}
|
||||
|
|
@ -0,0 +1,48 @@
|
|||
// Copyright (c) .NET Foundation. All rights reserved.
|
||||
// Licensed under the MIT License. See License.txt in the project root for license information.
|
||||
|
||||
#pragma once
|
||||
|
||||
enum APPLICATION_STATUS
|
||||
{
|
||||
UNKNOWN = 0,
|
||||
RUNNING,
|
||||
FAUL
|
||||
};
|
||||
|
||||
class ASPNETCORE_CONFIG;
|
||||
|
||||
class APPLICATION
|
||||
{
|
||||
public:
|
||||
APPLICATION(
|
||||
_In_ IHttpServer* pHttpServer,
|
||||
_In_ ASPNETCORE_CONFIG* pConfig);
|
||||
|
||||
virtual
|
||||
VOID
|
||||
ShutDown() = 0;
|
||||
|
||||
virtual
|
||||
~APPLICATION();
|
||||
|
||||
APPLICATION_STATUS
|
||||
QueryStatus();
|
||||
|
||||
ASPNETCORE_CONFIG*
|
||||
QueryConfig();
|
||||
|
||||
VOID
|
||||
ReferenceApplication()
|
||||
const;
|
||||
|
||||
VOID
|
||||
DereferenceApplication()
|
||||
const;
|
||||
|
||||
protected:
|
||||
mutable LONG m_cRefs;
|
||||
APPLICATION_STATUS m_status;
|
||||
IHttpServer* m_pHttpServer;
|
||||
ASPNETCORE_CONFIG* m_pConfig;
|
||||
};
|
||||
|
|
@ -0,0 +1,522 @@
|
|||
// Copyright (c) .NET Foundation. All rights reserved.
|
||||
// Licensed under the MIT License. See License.txt in the project root for license information.
|
||||
|
||||
#include "stdafx.h"
|
||||
#include "aspnetcoreconfig.h"
|
||||
|
||||
ASPNETCORE_CONFIG::~ASPNETCORE_CONFIG()
|
||||
{
|
||||
if (m_pEnvironmentVariables != NULL)
|
||||
{
|
||||
m_pEnvironmentVariables->Clear();
|
||||
delete m_pEnvironmentVariables;
|
||||
m_pEnvironmentVariables = NULL;
|
||||
}
|
||||
}
|
||||
|
||||
VOID
|
||||
ASPNETCORE_CONFIG::ReferenceConfiguration(
|
||||
VOID
|
||||
) const
|
||||
{
|
||||
InterlockedIncrement(&m_cRefs);
|
||||
}
|
||||
|
||||
|
||||
VOID
|
||||
ASPNETCORE_CONFIG::DereferenceConfiguration(
|
||||
VOID
|
||||
) const
|
||||
{
|
||||
DBG_ASSERT(m_cRefs != 0);
|
||||
LONG cRefs = 0;
|
||||
if ((cRefs = InterlockedDecrement(&m_cRefs)) == 0)
|
||||
{
|
||||
delete this;
|
||||
}
|
||||
}
|
||||
|
||||
HRESULT
|
||||
ASPNETCORE_CONFIG::GetConfig(
|
||||
_In_ IHttpServer *pHttpServer,
|
||||
_In_ HTTP_MODULE_ID pModuleId,
|
||||
_In_ IHttpContext *pHttpContext,
|
||||
_Out_ ASPNETCORE_CONFIG **ppAspNetCoreConfig
|
||||
)
|
||||
{
|
||||
HRESULT hr = S_OK;
|
||||
IHttpApplication *pHttpApplication = pHttpContext->GetApplication();
|
||||
ASPNETCORE_CONFIG *pAspNetCoreConfig = NULL;
|
||||
|
||||
if (ppAspNetCoreConfig == NULL)
|
||||
{
|
||||
hr = E_INVALIDARG;
|
||||
goto Finished;
|
||||
}
|
||||
|
||||
*ppAspNetCoreConfig = NULL;
|
||||
|
||||
// potential bug if user sepcific config at virtual dir level
|
||||
pAspNetCoreConfig = (ASPNETCORE_CONFIG*)
|
||||
pHttpApplication->GetModuleContextContainer()->GetModuleContext(pModuleId);
|
||||
|
||||
if (pAspNetCoreConfig != NULL)
|
||||
{
|
||||
*ppAspNetCoreConfig = pAspNetCoreConfig;
|
||||
pAspNetCoreConfig = NULL;
|
||||
goto Finished;
|
||||
}
|
||||
|
||||
pAspNetCoreConfig = new ASPNETCORE_CONFIG;
|
||||
if (pAspNetCoreConfig == NULL)
|
||||
{
|
||||
hr = E_OUTOFMEMORY;
|
||||
goto Finished;
|
||||
}
|
||||
|
||||
hr = pAspNetCoreConfig->Populate(pHttpServer, pHttpContext);
|
||||
if (FAILED(hr))
|
||||
{
|
||||
goto Finished;
|
||||
}
|
||||
|
||||
hr = pHttpApplication->GetModuleContextContainer()->
|
||||
SetModuleContext(pAspNetCoreConfig, pModuleId);
|
||||
if (FAILED(hr))
|
||||
{
|
||||
if (hr == HRESULT_FROM_WIN32(ERROR_ALREADY_ASSIGNED))
|
||||
{
|
||||
delete pAspNetCoreConfig;
|
||||
|
||||
pAspNetCoreConfig = (ASPNETCORE_CONFIG*)pHttpApplication->
|
||||
GetModuleContextContainer()->
|
||||
GetModuleContext(pModuleId);
|
||||
|
||||
_ASSERT(pAspNetCoreConfig != NULL);
|
||||
|
||||
hr = S_OK;
|
||||
}
|
||||
else
|
||||
{
|
||||
goto Finished;
|
||||
}
|
||||
}
|
||||
else
|
||||
{
|
||||
// set appliction info here instead of inside Populate()
|
||||
// as the destructor will delete the backend process
|
||||
hr = pAspNetCoreConfig->QueryApplicationPath()->Copy(pHttpApplication->GetApplicationId());
|
||||
if (FAILED(hr))
|
||||
{
|
||||
goto Finished;
|
||||
}
|
||||
}
|
||||
|
||||
*ppAspNetCoreConfig = pAspNetCoreConfig;
|
||||
pAspNetCoreConfig = NULL;
|
||||
|
||||
Finished:
|
||||
|
||||
if (pAspNetCoreConfig != NULL)
|
||||
{
|
||||
delete pAspNetCoreConfig;
|
||||
pAspNetCoreConfig = NULL;
|
||||
}
|
||||
|
||||
return hr;
|
||||
}
|
||||
|
||||
HRESULT
|
||||
ASPNETCORE_CONFIG::Populate(
|
||||
IHttpServer *pHttpServer,
|
||||
IHttpContext *pHttpContext
|
||||
)
|
||||
{
|
||||
HRESULT hr = S_OK;
|
||||
STRU strEnvName;
|
||||
STRU strEnvValue;
|
||||
STRU strExpandedEnvValue;
|
||||
STRU strApplicationFullPath;
|
||||
IAppHostAdminManager *pAdminManager = NULL;
|
||||
IAppHostElement *pAspNetCoreElement = NULL;
|
||||
IAppHostElement *pWindowsAuthenticationElement = NULL;
|
||||
IAppHostElement *pBasicAuthenticationElement = NULL;
|
||||
IAppHostElement *pAnonymousAuthenticationElement = NULL;
|
||||
IAppHostElement *pEnvVarList = NULL;
|
||||
IAppHostElement *pEnvVar = NULL;
|
||||
IAppHostElementCollection *pEnvVarCollection = NULL;
|
||||
ULONGLONG ullRawTimeSpan = 0;
|
||||
ENUM_INDEX index;
|
||||
ENVIRONMENT_VAR_ENTRY* pEntry = NULL;
|
||||
DWORD dwCounter = 0;
|
||||
DWORD dwPosition = 0;
|
||||
WCHAR* pszPath = NULL;
|
||||
BSTR bstrWindowAuthSection = NULL;
|
||||
BSTR bstrBasicAuthSection = NULL;
|
||||
BSTR bstrAnonymousAuthSection = NULL;
|
||||
BSTR bstrAspNetCoreSection = NULL;
|
||||
|
||||
m_pEnvironmentVariables = new ENVIRONMENT_VAR_HASH();
|
||||
if (m_pEnvironmentVariables == NULL)
|
||||
{
|
||||
hr = E_OUTOFMEMORY;
|
||||
goto Finished;
|
||||
}
|
||||
if (FAILED(hr = m_pEnvironmentVariables->Initialize(37 /*prime*/)))
|
||||
{
|
||||
delete m_pEnvironmentVariables;
|
||||
m_pEnvironmentVariables = NULL;
|
||||
goto Finished;
|
||||
}
|
||||
|
||||
pAdminManager = pHttpServer->GetAdminManager();
|
||||
hr = m_struConfigPath.Copy(pHttpContext->GetApplication()->GetAppConfigPath());
|
||||
if (FAILED(hr))
|
||||
{
|
||||
goto Finished;
|
||||
}
|
||||
|
||||
hr = m_struApplicationPhysicalPath.Copy(pHttpContext->GetApplication()->GetApplicationPhysicalPath());
|
||||
if (FAILED(hr))
|
||||
{
|
||||
goto Finished;
|
||||
}
|
||||
|
||||
pszPath = m_struConfigPath.QueryStr();
|
||||
while (pszPath[dwPosition] != NULL)
|
||||
{
|
||||
if (pszPath[dwPosition] == '/')
|
||||
{
|
||||
dwCounter++;
|
||||
if (dwCounter == 4)
|
||||
break;
|
||||
}
|
||||
dwPosition++;
|
||||
}
|
||||
|
||||
if (dwCounter == 4)
|
||||
{
|
||||
hr = m_struApplicationVirtualPath.Copy(pszPath + dwPosition);
|
||||
}
|
||||
else
|
||||
{
|
||||
hr = m_struApplicationVirtualPath.Copy(L"/");
|
||||
}
|
||||
|
||||
// Will setup the application virtual path.
|
||||
if (FAILED(hr))
|
||||
{
|
||||
goto Finished;
|
||||
}
|
||||
|
||||
bstrWindowAuthSection = SysAllocString(CS_WINDOWS_AUTHENTICATION_SECTION);
|
||||
if (bstrWindowAuthSection == NULL)
|
||||
{
|
||||
hr = E_OUTOFMEMORY;
|
||||
goto Finished;
|
||||
}
|
||||
|
||||
hr = pAdminManager->GetAdminSection(bstrWindowAuthSection,
|
||||
m_struConfigPath.QueryStr(),
|
||||
&pWindowsAuthenticationElement);
|
||||
if (FAILED(hr))
|
||||
{
|
||||
// assume the corresponding authen was not enabled
|
||||
// as the section may get deleted by user in some HWC case
|
||||
// ToDo: log a warning to event log
|
||||
m_fWindowsAuthEnabled = FALSE;
|
||||
}
|
||||
else
|
||||
{
|
||||
hr = GetElementBoolProperty(pWindowsAuthenticationElement,
|
||||
CS_AUTHENTICATION_ENABLED,
|
||||
&m_fWindowsAuthEnabled);
|
||||
if (FAILED(hr))
|
||||
{
|
||||
goto Finished;
|
||||
}
|
||||
}
|
||||
|
||||
bstrBasicAuthSection = SysAllocString(CS_BASIC_AUTHENTICATION_SECTION);
|
||||
if (bstrBasicAuthSection == NULL)
|
||||
{
|
||||
hr = E_OUTOFMEMORY;
|
||||
goto Finished;
|
||||
}
|
||||
hr = pAdminManager->GetAdminSection(bstrBasicAuthSection,
|
||||
m_struConfigPath.QueryStr(),
|
||||
&pBasicAuthenticationElement);
|
||||
if (FAILED(hr))
|
||||
{
|
||||
m_fBasicAuthEnabled = FALSE;
|
||||
}
|
||||
else
|
||||
{
|
||||
hr = GetElementBoolProperty(pBasicAuthenticationElement,
|
||||
CS_AUTHENTICATION_ENABLED,
|
||||
&m_fBasicAuthEnabled);
|
||||
if (FAILED(hr))
|
||||
{
|
||||
goto Finished;
|
||||
}
|
||||
}
|
||||
bstrAnonymousAuthSection = SysAllocString(CS_ANONYMOUS_AUTHENTICATION_SECTION);
|
||||
if (bstrAnonymousAuthSection == NULL)
|
||||
{
|
||||
hr = E_OUTOFMEMORY;
|
||||
goto Finished;
|
||||
}
|
||||
hr = pAdminManager->GetAdminSection(bstrAnonymousAuthSection,
|
||||
m_struConfigPath.QueryStr(),
|
||||
&pAnonymousAuthenticationElement);
|
||||
if (FAILED(hr))
|
||||
{
|
||||
m_fAnonymousAuthEnabled = FALSE;
|
||||
}
|
||||
else
|
||||
{
|
||||
hr = GetElementBoolProperty(pAnonymousAuthenticationElement,
|
||||
CS_AUTHENTICATION_ENABLED,
|
||||
&m_fAnonymousAuthEnabled);
|
||||
if (FAILED(hr))
|
||||
{
|
||||
goto Finished;
|
||||
}
|
||||
}
|
||||
|
||||
bstrAspNetCoreSection = SysAllocString(CS_ASPNETCORE_SECTION);
|
||||
if (bstrAspNetCoreSection == NULL)
|
||||
{
|
||||
hr = E_OUTOFMEMORY;
|
||||
goto Finished;
|
||||
}
|
||||
hr = pAdminManager->GetAdminSection(bstrAspNetCoreSection,
|
||||
m_struConfigPath.QueryStr(),
|
||||
&pAspNetCoreElement);
|
||||
if (FAILED(hr))
|
||||
{
|
||||
goto Finished;
|
||||
}
|
||||
|
||||
hr = GetElementStringProperty(pAspNetCoreElement,
|
||||
CS_ASPNETCORE_PROCESS_EXE_PATH,
|
||||
&m_struProcessPath);
|
||||
if (FAILED(hr))
|
||||
{
|
||||
goto Finished;
|
||||
}
|
||||
|
||||
hr = GetElementStringProperty(pAspNetCoreElement,
|
||||
CS_ASPNETCORE_HOSTING_MODEL,
|
||||
&m_strHostingModel);
|
||||
if (FAILED(hr))
|
||||
{
|
||||
// Swallow this error for backward compatability
|
||||
// Use default behavior for empty string
|
||||
hr = S_OK;
|
||||
}
|
||||
|
||||
if (m_strHostingModel.IsEmpty() || m_strHostingModel.Equals(L"outofprocess", TRUE))
|
||||
{
|
||||
m_hostingModel = HOSTING_OUT_PROCESS;
|
||||
}
|
||||
else if (m_strHostingModel.Equals(L"inprocess", TRUE))
|
||||
{
|
||||
m_hostingModel = HOSTING_IN_PROCESS;
|
||||
}
|
||||
else
|
||||
{
|
||||
// block unknown hosting value
|
||||
hr = HRESULT_FROM_WIN32(ERROR_NOT_SUPPORTED);
|
||||
goto Finished;
|
||||
}
|
||||
|
||||
hr = GetElementStringProperty(pAspNetCoreElement,
|
||||
CS_ASPNETCORE_PROCESS_ARGUMENTS,
|
||||
&m_struArguments);
|
||||
if (FAILED(hr))
|
||||
{
|
||||
goto Finished;
|
||||
}
|
||||
|
||||
hr = GetElementDWORDProperty(pAspNetCoreElement,
|
||||
CS_ASPNETCORE_RAPID_FAILS_PER_MINUTE,
|
||||
&m_dwRapidFailsPerMinute);
|
||||
if (FAILED(hr))
|
||||
{
|
||||
goto Finished;
|
||||
}
|
||||
|
||||
//
|
||||
// rapidFailsPerMinute cannot be greater than 100.
|
||||
//
|
||||
if (m_dwRapidFailsPerMinute > MAX_RAPID_FAILS_PER_MINUTE)
|
||||
{
|
||||
m_dwRapidFailsPerMinute = MAX_RAPID_FAILS_PER_MINUTE;
|
||||
}
|
||||
|
||||
hr = GetElementDWORDProperty(pAspNetCoreElement,
|
||||
CS_ASPNETCORE_PROCESSES_PER_APPLICATION,
|
||||
&m_dwProcessesPerApplication);
|
||||
if (FAILED(hr))
|
||||
{
|
||||
goto Finished;
|
||||
}
|
||||
|
||||
hr = GetElementDWORDProperty(
|
||||
pAspNetCoreElement,
|
||||
CS_ASPNETCORE_PROCESS_STARTUP_TIME_LIMIT,
|
||||
&m_dwStartupTimeLimitInMS
|
||||
);
|
||||
if (FAILED(hr))
|
||||
{
|
||||
goto Finished;
|
||||
}
|
||||
|
||||
m_dwStartupTimeLimitInMS *= MILLISECONDS_IN_ONE_SECOND;
|
||||
|
||||
hr = GetElementDWORDProperty(
|
||||
pAspNetCoreElement,
|
||||
CS_ASPNETCORE_PROCESS_SHUTDOWN_TIME_LIMIT,
|
||||
&m_dwShutdownTimeLimitInMS
|
||||
);
|
||||
if (FAILED(hr))
|
||||
{
|
||||
goto Finished;
|
||||
}
|
||||
m_dwShutdownTimeLimitInMS *= MILLISECONDS_IN_ONE_SECOND;
|
||||
|
||||
hr = GetElementBoolProperty(pAspNetCoreElement,
|
||||
CS_ASPNETCORE_FORWARD_WINDOWS_AUTH_TOKEN,
|
||||
&m_fForwardWindowsAuthToken);
|
||||
if (FAILED(hr))
|
||||
{
|
||||
goto Finished;
|
||||
}
|
||||
|
||||
hr = GetElementBoolProperty(pAspNetCoreElement,
|
||||
CS_ASPNETCORE_DISABLE_START_UP_ERROR_PAGE,
|
||||
&m_fDisableStartUpErrorPage);
|
||||
if (FAILED(hr))
|
||||
{
|
||||
goto Finished;
|
||||
}
|
||||
|
||||
hr = GetElementRawTimeSpanProperty(
|
||||
pAspNetCoreElement,
|
||||
CS_ASPNETCORE_WINHTTP_REQUEST_TIMEOUT,
|
||||
&ullRawTimeSpan
|
||||
);
|
||||
if (FAILED(hr))
|
||||
{
|
||||
goto Finished;
|
||||
}
|
||||
|
||||
m_dwRequestTimeoutInMS = (DWORD)TIMESPAN_IN_MILLISECONDS(ullRawTimeSpan);
|
||||
|
||||
hr = GetElementBoolProperty(pAspNetCoreElement,
|
||||
CS_ASPNETCORE_STDOUT_LOG_ENABLED,
|
||||
&m_fStdoutLogEnabled);
|
||||
if (FAILED(hr))
|
||||
{
|
||||
goto Finished;
|
||||
}
|
||||
hr = GetElementStringProperty(pAspNetCoreElement,
|
||||
CS_ASPNETCORE_STDOUT_LOG_FILE,
|
||||
&m_struStdoutLogFile);
|
||||
if (FAILED(hr))
|
||||
{
|
||||
goto Finished;
|
||||
}
|
||||
|
||||
hr = GetElementChildByName(pAspNetCoreElement,
|
||||
CS_ASPNETCORE_ENVIRONMENT_VARIABLES,
|
||||
&pEnvVarList);
|
||||
if (FAILED(hr))
|
||||
{
|
||||
goto Finished;
|
||||
}
|
||||
|
||||
hr = pEnvVarList->get_Collection(&pEnvVarCollection);
|
||||
if (FAILED(hr))
|
||||
{
|
||||
goto Finished;
|
||||
}
|
||||
|
||||
for (hr = FindFirstElement(pEnvVarCollection, &index, &pEnvVar);
|
||||
SUCCEEDED(hr);
|
||||
hr = FindNextElement(pEnvVarCollection, &index, &pEnvVar))
|
||||
{
|
||||
if (hr == S_FALSE)
|
||||
{
|
||||
hr = S_OK;
|
||||
break;
|
||||
}
|
||||
|
||||
if (FAILED(hr = GetElementStringProperty(pEnvVar,
|
||||
CS_ASPNETCORE_ENVIRONMENT_VARIABLE_NAME,
|
||||
&strEnvName)) ||
|
||||
FAILED(hr = GetElementStringProperty(pEnvVar,
|
||||
CS_ASPNETCORE_ENVIRONMENT_VARIABLE_VALUE,
|
||||
&strEnvValue)) ||
|
||||
FAILED(hr = strEnvName.Append(L"=")) ||
|
||||
FAILED(hr = STRU::ExpandEnvironmentVariables(strEnvValue.QueryStr(), &strExpandedEnvValue)))
|
||||
{
|
||||
goto Finished;
|
||||
}
|
||||
|
||||
pEntry = new ENVIRONMENT_VAR_ENTRY();
|
||||
if (pEntry == NULL)
|
||||
{
|
||||
hr = E_OUTOFMEMORY;
|
||||
goto Finished;
|
||||
}
|
||||
|
||||
if (FAILED(hr = pEntry->Initialize(strEnvName.QueryStr(), strExpandedEnvValue.QueryStr())) ||
|
||||
FAILED(hr = m_pEnvironmentVariables->InsertRecord(pEntry)))
|
||||
{
|
||||
goto Finished;
|
||||
}
|
||||
strEnvName.Reset();
|
||||
strEnvValue.Reset();
|
||||
strExpandedEnvValue.Reset();
|
||||
pEnvVar->Release();
|
||||
pEnvVar = NULL;
|
||||
pEntry->Dereference();
|
||||
pEntry = NULL;
|
||||
}
|
||||
|
||||
Finished:
|
||||
|
||||
if (pAspNetCoreElement != NULL)
|
||||
{
|
||||
pAspNetCoreElement->Release();
|
||||
pAspNetCoreElement = NULL;
|
||||
}
|
||||
|
||||
if (pEnvVarList != NULL)
|
||||
{
|
||||
pEnvVarList->Release();
|
||||
pEnvVarList = NULL;
|
||||
}
|
||||
|
||||
if (pEnvVar != NULL)
|
||||
{
|
||||
pEnvVar->Release();
|
||||
pEnvVar = NULL;
|
||||
}
|
||||
|
||||
if (pEnvVarCollection != NULL)
|
||||
{
|
||||
pEnvVarCollection->Release();
|
||||
pEnvVarCollection = NULL;
|
||||
}
|
||||
|
||||
if (pEntry != NULL)
|
||||
{
|
||||
pEntry->Dereference();
|
||||
pEntry = NULL;
|
||||
}
|
||||
|
||||
return hr;
|
||||
}
|
||||
|
|
@ -0,0 +1,286 @@
|
|||
// Copyright (c) .NET Foundation. All rights reserved.
|
||||
// Licensed under the MIT License. See License.txt in the project root for license information.
|
||||
|
||||
#pragma once
|
||||
#define CS_ROOTWEB_CONFIG L"MACHINE/WEBROOT/APPHOST/"
|
||||
#define CS_ROOTWEB_CONFIG_LEN _countof(CS_ROOTWEB_CONFIG)-1
|
||||
#define CS_ASPNETCORE_SECTION L"system.webServer/aspNetCore"
|
||||
#define CS_WINDOWS_AUTHENTICATION_SECTION L"system.webServer/security/authentication/windowsAuthentication"
|
||||
#define CS_BASIC_AUTHENTICATION_SECTION L"system.webServer/security/authentication/basicAuthentication"
|
||||
#define CS_ANONYMOUS_AUTHENTICATION_SECTION L"system.webServer/security/authentication/anonymousAuthentication"
|
||||
#define CS_AUTHENTICATION_ENABLED L"enabled"
|
||||
#define CS_ASPNETCORE_PROCESS_EXE_PATH L"processPath"
|
||||
#define CS_ASPNETCORE_PROCESS_ARGUMENTS L"arguments"
|
||||
#define CS_ASPNETCORE_PROCESS_STARTUP_TIME_LIMIT L"startupTimeLimit"
|
||||
#define CS_ASPNETCORE_PROCESS_SHUTDOWN_TIME_LIMIT L"shutdownTimeLimit"
|
||||
#define CS_ASPNETCORE_WINHTTP_REQUEST_TIMEOUT L"requestTimeout"
|
||||
#define CS_ASPNETCORE_RAPID_FAILS_PER_MINUTE L"rapidFailsPerMinute"
|
||||
#define CS_ASPNETCORE_STDOUT_LOG_ENABLED L"stdoutLogEnabled"
|
||||
#define CS_ASPNETCORE_STDOUT_LOG_FILE L"stdoutLogFile"
|
||||
#define CS_ASPNETCORE_ENVIRONMENT_VARIABLES L"environmentVariables"
|
||||
#define CS_ASPNETCORE_ENVIRONMENT_VARIABLE L"environmentVariable"
|
||||
#define CS_ASPNETCORE_ENVIRONMENT_VARIABLE_NAME L"name"
|
||||
#define CS_ASPNETCORE_ENVIRONMENT_VARIABLE_VALUE L"value"
|
||||
#define CS_ASPNETCORE_PROCESSES_PER_APPLICATION L"processesPerApplication"
|
||||
#define CS_ASPNETCORE_FORWARD_WINDOWS_AUTH_TOKEN L"forwardWindowsAuthToken"
|
||||
#define CS_ASPNETCORE_DISABLE_START_UP_ERROR_PAGE L"disableStartUpErrorPage"
|
||||
#define CS_ASPNETCORE_RECYCLE_ON_FILE_CHANGE L"recycleOnFileChange"
|
||||
#define CS_ASPNETCORE_RECYCLE_ON_FILE_CHANGE_FILE L"file"
|
||||
#define CS_ASPNETCORE_RECYCLE_ON_FILE_CHANGE_FILE_PATH L"path"
|
||||
#define CS_ASPNETCORE_HOSTING_MODEL L"hostingModel"
|
||||
|
||||
#define MAX_RAPID_FAILS_PER_MINUTE 100
|
||||
#define MILLISECONDS_IN_ONE_SECOND 1000
|
||||
#define MIN_PORT 1025
|
||||
#define MAX_PORT 48000
|
||||
|
||||
#define TIMESPAN_IN_MILLISECONDS(x) ((x)/((LONGLONG)(10000)))
|
||||
#define TIMESPAN_IN_SECONDS(x) ((TIMESPAN_IN_MILLISECONDS(x))/((LONGLONG)(1000)))
|
||||
#define TIMESPAN_IN_MINUTES(x) ((TIMESPAN_IN_SECONDS(x))/((LONGLONG)(60)))
|
||||
|
||||
//#define HEX_TO_ASCII(c) ((CHAR)(((c) < 10) ? ((c) + '0') : ((c) + 'a' - 10)))
|
||||
|
||||
#include "stdafx.h"
|
||||
|
||||
enum APP_HOSTING_MODEL
|
||||
{
|
||||
HOSTING_UNKNOWN = 0,
|
||||
HOSTING_IN_PROCESS,
|
||||
HOSTING_OUT_PROCESS
|
||||
};
|
||||
|
||||
class ASPNETCORE_CONFIG : IHttpStoredContext
|
||||
{
|
||||
public:
|
||||
|
||||
virtual
|
||||
~ASPNETCORE_CONFIG();
|
||||
|
||||
VOID
|
||||
CleanupStoredContext()
|
||||
{
|
||||
DereferenceConfiguration();
|
||||
}
|
||||
|
||||
static
|
||||
HRESULT
|
||||
GetConfig(
|
||||
_In_ IHttpServer *pHttpServer,
|
||||
_In_ HTTP_MODULE_ID pModuleId,
|
||||
_In_ IHttpContext *pHttpContext,
|
||||
_Out_ ASPNETCORE_CONFIG **ppAspNetCoreConfig
|
||||
);
|
||||
|
||||
ENVIRONMENT_VAR_HASH*
|
||||
QueryEnvironmentVariables(
|
||||
VOID
|
||||
)
|
||||
{
|
||||
return m_pEnvironmentVariables;
|
||||
}
|
||||
|
||||
DWORD
|
||||
QueryRapidFailsPerMinute(
|
||||
VOID
|
||||
)
|
||||
{
|
||||
return m_dwRapidFailsPerMinute;
|
||||
}
|
||||
|
||||
DWORD
|
||||
QueryStartupTimeLimitInMS(
|
||||
VOID
|
||||
)
|
||||
{
|
||||
return m_dwStartupTimeLimitInMS;
|
||||
}
|
||||
|
||||
DWORD
|
||||
QueryShutdownTimeLimitInMS(
|
||||
VOID
|
||||
)
|
||||
{
|
||||
return m_dwShutdownTimeLimitInMS;
|
||||
}
|
||||
|
||||
DWORD
|
||||
QueryProcessesPerApplication(
|
||||
VOID
|
||||
)
|
||||
{
|
||||
return m_dwProcessesPerApplication;
|
||||
}
|
||||
|
||||
DWORD
|
||||
QueryRequestTimeoutInMS(
|
||||
VOID
|
||||
)
|
||||
{
|
||||
return m_dwRequestTimeoutInMS;
|
||||
}
|
||||
|
||||
STRU*
|
||||
QueryArguments(
|
||||
VOID
|
||||
)
|
||||
{
|
||||
return &m_struArguments;
|
||||
}
|
||||
|
||||
STRU*
|
||||
QueryApplicationPath(
|
||||
VOID
|
||||
)
|
||||
{
|
||||
return &m_struApplication;
|
||||
}
|
||||
|
||||
STRU*
|
||||
QueryApplicationPhysicalPath(
|
||||
VOID
|
||||
)
|
||||
{
|
||||
return &m_struApplicationPhysicalPath;
|
||||
}
|
||||
|
||||
STRU*
|
||||
QueryApplicationVirtualPath(
|
||||
VOID
|
||||
)
|
||||
{
|
||||
return &m_struApplicationVirtualPath;
|
||||
}
|
||||
|
||||
STRU*
|
||||
QueryProcessPath(
|
||||
VOID
|
||||
)
|
||||
{
|
||||
return &m_struProcessPath;
|
||||
}
|
||||
|
||||
APP_HOSTING_MODEL
|
||||
QueryHostingModel(
|
||||
VOID
|
||||
)
|
||||
{
|
||||
return m_hostingModel;
|
||||
}
|
||||
|
||||
BOOL
|
||||
QueryStdoutLogEnabled()
|
||||
{
|
||||
return m_fStdoutLogEnabled;
|
||||
}
|
||||
|
||||
BOOL
|
||||
QueryForwardWindowsAuthToken()
|
||||
{
|
||||
return m_fForwardWindowsAuthToken;
|
||||
}
|
||||
|
||||
BOOL
|
||||
QueryWindowsAuthEnabled()
|
||||
{
|
||||
return m_fWindowsAuthEnabled;
|
||||
}
|
||||
|
||||
BOOL
|
||||
QueryBasicAuthEnabled()
|
||||
{
|
||||
return m_fBasicAuthEnabled;
|
||||
}
|
||||
|
||||
BOOL
|
||||
QueryAnonymousAuthEnabled()
|
||||
{
|
||||
return m_fAnonymousAuthEnabled;
|
||||
}
|
||||
|
||||
BOOL
|
||||
QueryDisableStartUpErrorPage()
|
||||
{
|
||||
return m_fDisableStartUpErrorPage;
|
||||
}
|
||||
|
||||
STRU*
|
||||
QueryStdoutLogFile()
|
||||
{
|
||||
return &m_struStdoutLogFile;
|
||||
}
|
||||
|
||||
STRU*
|
||||
QueryConfigPath()
|
||||
{
|
||||
return &m_struConfigPath;
|
||||
}
|
||||
|
||||
STRU*
|
||||
QueryHostfxrPath()
|
||||
{
|
||||
return &m_struHostFxrPath;
|
||||
}
|
||||
|
||||
BOOL
|
||||
QueryIsStandAloneApplication(
|
||||
VOID
|
||||
)
|
||||
{
|
||||
return m_fIsStandAloneApplication;
|
||||
}
|
||||
|
||||
VOID
|
||||
ReferenceConfiguration(
|
||||
VOID
|
||||
) const;
|
||||
|
||||
VOID
|
||||
DereferenceConfiguration(
|
||||
VOID
|
||||
) const;
|
||||
|
||||
private:
|
||||
|
||||
//
|
||||
// private constructor
|
||||
//
|
||||
ASPNETCORE_CONFIG():
|
||||
m_fStdoutLogEnabled( FALSE ),
|
||||
m_pEnvironmentVariables( NULL ),
|
||||
m_cRefs( 1 ),
|
||||
m_hostingModel( HOSTING_UNKNOWN )
|
||||
{
|
||||
}
|
||||
|
||||
HRESULT
|
||||
Populate(
|
||||
IHttpServer *pHttpServer,
|
||||
IHttpContext *pHttpContext
|
||||
);
|
||||
|
||||
mutable LONG m_cRefs;
|
||||
|
||||
DWORD m_dwRequestTimeoutInMS;
|
||||
DWORD m_dwStartupTimeLimitInMS;
|
||||
DWORD m_dwShutdownTimeLimitInMS;
|
||||
DWORD m_dwRapidFailsPerMinute;
|
||||
DWORD m_dwProcessesPerApplication;
|
||||
STRU m_struArguments;
|
||||
STRU m_struProcessPath;
|
||||
STRU m_struStdoutLogFile;
|
||||
STRU m_struApplication;
|
||||
STRU m_struApplicationPhysicalPath;
|
||||
STRU m_struApplicationVirtualPath;
|
||||
STRU m_struConfigPath;
|
||||
STRU m_strHostingModel;
|
||||
STRU m_struHostFxrPath;
|
||||
BOOL m_fStdoutLogEnabled;
|
||||
BOOL m_fForwardWindowsAuthToken;
|
||||
BOOL m_fDisableStartUpErrorPage;
|
||||
BOOL m_fWindowsAuthEnabled;
|
||||
BOOL m_fBasicAuthEnabled;
|
||||
BOOL m_fAnonymousAuthEnabled;
|
||||
BOOL m_fIsStandAloneApplication;
|
||||
APP_HOSTING_MODEL m_hostingModel;
|
||||
ENVIRONMENT_VAR_HASH* m_pEnvironmentVariables;
|
||||
};
|
||||
|
|
@ -0,0 +1,81 @@
|
|||
// Copyright (c) .NET Foundation. All rights reserved.
|
||||
// Licensed under the MIT License. See License.txt in the project root for license information.
|
||||
|
||||
#pragma once
|
||||
#define ASPNETCORE_DEBUG_FLAG_INFO 0x00000001
|
||||
#define ASPNETCORE_DEBUG_FLAG_WARNING 0x00000002
|
||||
#define ASPNETCORE_DEBUG_FLAG_ERROR 0x00000004
|
||||
|
||||
extern DWORD g_dwAspNetCoreDebugFlags;
|
||||
|
||||
static
|
||||
BOOL
|
||||
IfDebug(
|
||||
DWORD dwFlag
|
||||
)
|
||||
{
|
||||
return ( dwFlag & g_dwAspNetCoreDebugFlags );
|
||||
}
|
||||
|
||||
static
|
||||
VOID
|
||||
DebugPrint(
|
||||
DWORD dwFlag,
|
||||
LPCSTR szString
|
||||
)
|
||||
{
|
||||
STACK_STRA (strOutput, 256);
|
||||
HRESULT hr = S_OK;
|
||||
|
||||
if ( IfDebug( dwFlag ) )
|
||||
{
|
||||
hr = strOutput.SafeSnprintf(
|
||||
"[aspnetcore.dll] %s\r\n",
|
||||
szString );
|
||||
|
||||
if (FAILED (hr))
|
||||
{
|
||||
goto Finished;
|
||||
}
|
||||
|
||||
OutputDebugStringA( strOutput.QueryStr() );
|
||||
}
|
||||
|
||||
Finished:
|
||||
|
||||
return;
|
||||
}
|
||||
|
||||
static
|
||||
VOID
|
||||
DebugPrintf(
|
||||
DWORD dwFlag,
|
||||
LPCSTR szFormat,
|
||||
...
|
||||
)
|
||||
{
|
||||
STACK_STRA (strCooked,256);
|
||||
|
||||
va_list args;
|
||||
HRESULT hr = S_OK;
|
||||
|
||||
if ( IfDebug( dwFlag ) )
|
||||
{
|
||||
va_start( args, szFormat );
|
||||
|
||||
hr = strCooked.SafeVsnprintf(szFormat, args );
|
||||
|
||||
va_end( args );
|
||||
|
||||
if (FAILED (hr))
|
||||
{
|
||||
goto Finished;
|
||||
}
|
||||
|
||||
DebugPrint( dwFlag, strCooked.QueryStr() );
|
||||
}
|
||||
|
||||
Finished:
|
||||
return;
|
||||
}
|
||||
|
||||
|
|
@ -0,0 +1,155 @@
|
|||
// Copyright (c) .NET Foundation. All rights reserved.
|
||||
// Licensed under the MIT License. See License.txt in the project root for license information.
|
||||
|
||||
#pragma once
|
||||
|
||||
//
|
||||
// The key used for hash-table lookups, consists of the port on which the http process is created.
|
||||
//
|
||||
|
||||
class ENVIRONMENT_VAR_ENTRY
|
||||
{
|
||||
public:
|
||||
ENVIRONMENT_VAR_ENTRY():
|
||||
_cRefs(1)
|
||||
{
|
||||
}
|
||||
|
||||
HRESULT
|
||||
Initialize(
|
||||
PCWSTR pszName,
|
||||
PCWSTR pszValue
|
||||
)
|
||||
{
|
||||
HRESULT hr = S_OK;
|
||||
if (FAILED(hr = _strName.Copy(pszName)) ||
|
||||
FAILED(hr = _strValue.Copy(pszValue)))
|
||||
{
|
||||
}
|
||||
return hr;
|
||||
}
|
||||
|
||||
VOID
|
||||
Reference() const
|
||||
{
|
||||
InterlockedIncrement(&_cRefs);
|
||||
}
|
||||
|
||||
VOID
|
||||
Dereference() const
|
||||
{
|
||||
if (InterlockedDecrement(&_cRefs) == 0)
|
||||
{
|
||||
delete this;
|
||||
}
|
||||
}
|
||||
|
||||
PWSTR const
|
||||
QueryName()
|
||||
{
|
||||
return _strName.QueryStr();
|
||||
}
|
||||
|
||||
PWSTR const
|
||||
QueryValue()
|
||||
{
|
||||
return _strValue.QueryStr();
|
||||
}
|
||||
|
||||
private:
|
||||
~ENVIRONMENT_VAR_ENTRY()
|
||||
{
|
||||
}
|
||||
|
||||
STRU _strName;
|
||||
STRU _strValue;
|
||||
mutable LONG _cRefs;
|
||||
};
|
||||
|
||||
class ENVIRONMENT_VAR_HASH : public HASH_TABLE<ENVIRONMENT_VAR_ENTRY, PWSTR>
|
||||
{
|
||||
public:
|
||||
ENVIRONMENT_VAR_HASH()
|
||||
{}
|
||||
|
||||
PWSTR
|
||||
ExtractKey(
|
||||
ENVIRONMENT_VAR_ENTRY * pEntry
|
||||
)
|
||||
{
|
||||
return pEntry->QueryName();
|
||||
}
|
||||
|
||||
DWORD
|
||||
CalcKeyHash(
|
||||
PWSTR pszName
|
||||
)
|
||||
{
|
||||
return HashStringNoCase(pszName);
|
||||
}
|
||||
|
||||
BOOL
|
||||
EqualKeys(
|
||||
PWSTR pszName1,
|
||||
PWSTR pszName2
|
||||
)
|
||||
{
|
||||
return (_wcsicmp(pszName1, pszName2) == 0);
|
||||
}
|
||||
|
||||
VOID
|
||||
ReferenceRecord(
|
||||
ENVIRONMENT_VAR_ENTRY * pEntry
|
||||
)
|
||||
{
|
||||
pEntry->Reference();
|
||||
}
|
||||
|
||||
VOID
|
||||
DereferenceRecord(
|
||||
ENVIRONMENT_VAR_ENTRY * pEntry
|
||||
)
|
||||
{
|
||||
pEntry->Dereference();
|
||||
}
|
||||
|
||||
static
|
||||
VOID
|
||||
CopyToMultiSz(
|
||||
ENVIRONMENT_VAR_ENTRY * pEntry,
|
||||
PVOID pvData
|
||||
)
|
||||
{
|
||||
STRU strTemp;
|
||||
MULTISZ *pMultiSz = static_cast<MULTISZ *>(pvData);
|
||||
DBG_ASSERT(pMultiSz);
|
||||
DBG_ASSERT(pEntry);
|
||||
strTemp.Copy(pEntry->QueryName());
|
||||
strTemp.Append(pEntry->QueryValue());
|
||||
pMultiSz->Append(strTemp.QueryStr());
|
||||
}
|
||||
|
||||
static
|
||||
VOID
|
||||
CopyToTable(
|
||||
ENVIRONMENT_VAR_ENTRY * pEntry,
|
||||
PVOID pvData
|
||||
)
|
||||
{
|
||||
// best effort copy, ignore the failure
|
||||
ENVIRONMENT_VAR_ENTRY * pNewEntry = new ENVIRONMENT_VAR_ENTRY();
|
||||
if (pNewEntry != NULL)
|
||||
{
|
||||
pNewEntry->Initialize(pEntry->QueryName(), pEntry->QueryValue());
|
||||
ENVIRONMENT_VAR_HASH *pHash = static_cast<ENVIRONMENT_VAR_HASH *>(pvData);
|
||||
DBG_ASSERT(pHash);
|
||||
pHash->InsertRecord(pNewEntry);
|
||||
// Need to dereference as InsertRecord references it now
|
||||
pNewEntry->Dereference();
|
||||
}
|
||||
}
|
||||
|
||||
private:
|
||||
ENVIRONMENT_VAR_HASH(const ENVIRONMENT_VAR_HASH &);
|
||||
void operator=(const ENVIRONMENT_VAR_HASH &);
|
||||
};
|
||||
|
|
@ -0,0 +1,192 @@
|
|||
// Copyright (c) .NET Foundation and contributors. All rights reserved.
|
||||
// Licensed under the MIT license. See LICENSE file in the project root for full license information.
|
||||
|
||||
#include "stdafx.h"
|
||||
|
||||
fx_ver_t::fx_ver_t(int major, int minor, int patch, const std::wstring& pre, const std::wstring& build)
|
||||
: m_major(major)
|
||||
, m_minor(minor)
|
||||
, m_patch(patch)
|
||||
, m_pre(pre)
|
||||
, m_build(build)
|
||||
{
|
||||
}
|
||||
|
||||
fx_ver_t::fx_ver_t(int major, int minor, int patch, const std::wstring& pre)
|
||||
: fx_ver_t(major, minor, patch, pre, TEXT(""))
|
||||
{
|
||||
}
|
||||
|
||||
fx_ver_t::fx_ver_t(int major, int minor, int patch)
|
||||
: fx_ver_t(major, minor, patch, TEXT(""), TEXT(""))
|
||||
{
|
||||
}
|
||||
|
||||
bool fx_ver_t::operator ==(const fx_ver_t& b) const
|
||||
{
|
||||
return compare(*this, b) == 0;
|
||||
}
|
||||
|
||||
bool fx_ver_t::operator !=(const fx_ver_t& b) const
|
||||
{
|
||||
return !operator ==(b);
|
||||
}
|
||||
|
||||
bool fx_ver_t::operator <(const fx_ver_t& b) const
|
||||
{
|
||||
return compare(*this, b) < 0;
|
||||
}
|
||||
|
||||
bool fx_ver_t::operator >(const fx_ver_t& b) const
|
||||
{
|
||||
return compare(*this, b) > 0;
|
||||
}
|
||||
|
||||
bool fx_ver_t::operator <=(const fx_ver_t& b) const
|
||||
{
|
||||
return compare(*this, b) <= 0;
|
||||
}
|
||||
|
||||
bool fx_ver_t::operator >=(const fx_ver_t& b) const
|
||||
{
|
||||
return compare(*this, b) >= 0;
|
||||
}
|
||||
|
||||
std::wstring fx_ver_t::as_str() const
|
||||
{
|
||||
std::wstringstream stream;
|
||||
stream << m_major << TEXT(".") << m_minor << TEXT(".") << m_patch;
|
||||
if (!m_pre.empty())
|
||||
{
|
||||
stream << m_pre;
|
||||
}
|
||||
if (!m_build.empty())
|
||||
{
|
||||
stream << TEXT("+") << m_build;
|
||||
}
|
||||
return stream.str();
|
||||
}
|
||||
|
||||
/* static */
|
||||
int fx_ver_t::compare(const fx_ver_t&a, const fx_ver_t& b)
|
||||
{
|
||||
// compare(u.v.w-p+b, x.y.z-q+c)
|
||||
if (a.m_major != b.m_major)
|
||||
{
|
||||
return (a.m_major > b.m_major) ? 1 : -1;
|
||||
}
|
||||
|
||||
if (a.m_minor != b.m_minor)
|
||||
{
|
||||
return (a.m_minor > b.m_minor) ? 1 : -1;
|
||||
}
|
||||
|
||||
if (a.m_patch != b.m_patch)
|
||||
{
|
||||
return (a.m_patch > b.m_patch) ? 1 : -1;
|
||||
}
|
||||
|
||||
if (a.m_pre.empty() != b.m_pre.empty())
|
||||
{
|
||||
// Either a is empty or b is empty
|
||||
return a.m_pre.empty() ? 1 : -1;
|
||||
}
|
||||
|
||||
// Either both are empty or both are non-empty (may be equal)
|
||||
int pre_cmp = a.m_pre.compare(b.m_pre);
|
||||
if (pre_cmp != 0)
|
||||
{
|
||||
return pre_cmp;
|
||||
}
|
||||
|
||||
return a.m_build.compare(b.m_build);
|
||||
}
|
||||
|
||||
bool try_stou(const std::wstring& str, unsigned* num)
|
||||
{
|
||||
if (str.empty())
|
||||
{
|
||||
return false;
|
||||
}
|
||||
if (str.find_first_not_of(TEXT("0123456789")) != std::wstring::npos)
|
||||
{
|
||||
return false;
|
||||
}
|
||||
*num = (unsigned)std::stoul(str);
|
||||
return true;
|
||||
}
|
||||
|
||||
bool parse_internal(const std::wstring& ver, fx_ver_t* fx_ver, bool parse_only_production)
|
||||
{
|
||||
size_t maj_start = 0;
|
||||
size_t maj_sep = ver.find(TEXT('.'));
|
||||
if (maj_sep == std::wstring::npos)
|
||||
{
|
||||
return false;
|
||||
}
|
||||
unsigned major = 0;
|
||||
if (!try_stou(ver.substr(maj_start, maj_sep), &major))
|
||||
{
|
||||
return false;
|
||||
}
|
||||
|
||||
size_t min_start = maj_sep + 1;
|
||||
size_t min_sep = ver.find(TEXT('.'), min_start);
|
||||
if (min_sep == std::wstring::npos)
|
||||
{
|
||||
return false;
|
||||
}
|
||||
|
||||
unsigned minor = 0;
|
||||
if (!try_stou(ver.substr(min_start, min_sep - min_start), &minor))
|
||||
{
|
||||
return false;
|
||||
}
|
||||
|
||||
unsigned patch = 0;
|
||||
size_t pat_start = min_sep + 1;
|
||||
size_t pat_sep = ver.find_first_not_of(TEXT("0123456789"), pat_start);
|
||||
if (pat_sep == std::wstring::npos)
|
||||
{
|
||||
if (!try_stou(ver.substr(pat_start), &patch))
|
||||
{
|
||||
return false;
|
||||
}
|
||||
|
||||
*fx_ver = fx_ver_t(major, minor, patch);
|
||||
return true;
|
||||
}
|
||||
|
||||
if (parse_only_production)
|
||||
{
|
||||
// This is a prerelease or has build suffix.
|
||||
return false;
|
||||
}
|
||||
|
||||
if (!try_stou(ver.substr(pat_start, pat_sep - pat_start), &patch))
|
||||
{
|
||||
return false;
|
||||
}
|
||||
|
||||
size_t pre_start = pat_sep;
|
||||
size_t pre_sep = ver.find(TEXT('+'), pre_start);
|
||||
if (pre_sep == std::wstring::npos)
|
||||
{
|
||||
*fx_ver = fx_ver_t(major, minor, patch, ver.substr(pre_start));
|
||||
return true;
|
||||
}
|
||||
else
|
||||
{
|
||||
size_t build_start = pre_sep + 1;
|
||||
*fx_ver = fx_ver_t(major, minor, patch, ver.substr(pre_start, pre_sep - pre_start), ver.substr(build_start));
|
||||
return true;
|
||||
}
|
||||
}
|
||||
|
||||
/* static */
|
||||
bool fx_ver_t::parse(const std::wstring& ver, fx_ver_t* fx_ver, bool parse_only_production)
|
||||
{
|
||||
bool valid = parse_internal(ver, fx_ver, parse_only_production);
|
||||
assert(!valid || fx_ver->as_str() == ver);
|
||||
return valid;
|
||||
}
|
||||
|
|
@ -0,0 +1,46 @@
|
|||
// Copyright (c) .NET Foundation and contributors. All rights reserved.
|
||||
// Licensed under the MIT license. See LICENSE file in the project root for full license information.
|
||||
|
||||
#pragma once
|
||||
|
||||
// Note: This is not SemVer (esp., in comparing pre-release part, fx_ver_t does not
|
||||
// compare multiple dot separated identifiers individually.) ex: 1.0.0-beta.2 vs. 1.0.0-beta.11
|
||||
struct fx_ver_t
|
||||
{
|
||||
fx_ver_t(int major, int minor, int patch);
|
||||
fx_ver_t(int major, int minor, int patch, const std::wstring& pre);
|
||||
fx_ver_t(int major, int minor, int patch, const std::wstring& pre, const std::wstring& build);
|
||||
|
||||
int get_major() const { return m_major; }
|
||||
int get_minor() const { return m_minor; }
|
||||
int get_patch() const { return m_patch; }
|
||||
|
||||
void set_major(int m) { m_major = m; }
|
||||
void set_minor(int m) { m_minor = m; }
|
||||
void set_patch(int p) { m_patch = p; }
|
||||
|
||||
bool is_prerelease() const { return !m_pre.empty(); }
|
||||
|
||||
std::wstring as_str() const;
|
||||
std::wstring prerelease_glob() const;
|
||||
std::wstring patch_glob() const;
|
||||
|
||||
bool operator ==(const fx_ver_t& b) const;
|
||||
bool operator !=(const fx_ver_t& b) const;
|
||||
bool operator <(const fx_ver_t& b) const;
|
||||
bool operator >(const fx_ver_t& b) const;
|
||||
bool operator <=(const fx_ver_t& b) const;
|
||||
bool operator >=(const fx_ver_t& b) const;
|
||||
|
||||
static bool parse(const std::wstring& ver, fx_ver_t* fx_ver, bool parse_only_production = false);
|
||||
|
||||
private:
|
||||
int m_major;
|
||||
int m_minor;
|
||||
int m_patch;
|
||||
std::wstring m_pre;
|
||||
std::wstring m_build;
|
||||
|
||||
static int compare(const fx_ver_t&a, const fx_ver_t& b);
|
||||
};
|
||||
|
||||
|
|
@ -0,0 +1,259 @@
|
|||
// Copyright (c) .NET Foundation. All rights reserved.
|
||||
// Licensed under the MIT License. See License.txt in the project root for license information.
|
||||
|
||||
#include "stdafx.h"
|
||||
|
||||
HOSTFXR_UTILITY::HOSTFXR_UTILITY()
|
||||
{
|
||||
}
|
||||
|
||||
|
||||
HOSTFXR_UTILITY::~HOSTFXR_UTILITY()
|
||||
{
|
||||
}
|
||||
|
||||
HRESULT
|
||||
HOSTFXR_UTILITY::FindHostFxrDll(
|
||||
ASPNETCORE_CONFIG *pConfig,
|
||||
STRU* struHostFxrDllLocation,
|
||||
BOOL* fStandAlone
|
||||
)
|
||||
{
|
||||
HRESULT hr = S_OK;
|
||||
|
||||
// If the process path isn't dotnet, assume we are a standalone appliction.
|
||||
// TODO: this should be a path equivalent check
|
||||
if (!(pConfig->QueryProcessPath()->Equals(L".\\dotnet")
|
||||
|| pConfig->QueryProcessPath()->Equals(L"dotnet")
|
||||
|| pConfig->QueryProcessPath()->Equals(L".\\dotnet.exe")
|
||||
|| pConfig->QueryProcessPath()->Equals(L"dotnet.exe")))
|
||||
{
|
||||
// hostfxr is in the same folder, parse and use it.
|
||||
hr = GetStandaloneHostfxrLocation(struHostFxrDllLocation, pConfig);
|
||||
*fStandAlone = TRUE;
|
||||
}
|
||||
else
|
||||
{
|
||||
hr = GetPortableHostfxrLocation(struHostFxrDllLocation);
|
||||
fStandAlone = FALSE;
|
||||
}
|
||||
|
||||
return hr;
|
||||
}
|
||||
|
||||
//
|
||||
// Runs a standalone appliction.
|
||||
// The folder structure looks like this:
|
||||
// Application/
|
||||
// hostfxr.dll
|
||||
// Application.exe
|
||||
// Application.dll
|
||||
// etc.
|
||||
// We get the full path to hostfxr.dll and Application.dll and run hostfxr_main,
|
||||
// passing in Application.dll.
|
||||
// Assuming we don't need Application.exe as the dll is the actual application.
|
||||
//
|
||||
HRESULT
|
||||
HOSTFXR_UTILITY::GetStandaloneHostfxrLocation(
|
||||
STRU* struHostfxrPath,
|
||||
ASPNETCORE_CONFIG *pConfig
|
||||
)
|
||||
{
|
||||
HRESULT hr = S_OK;
|
||||
HANDLE hFileHandle = INVALID_HANDLE_VALUE;
|
||||
SECURITY_ATTRIBUTES saAttr;
|
||||
|
||||
// Get the full path to the exe and check if it exists
|
||||
if (FAILED(hr = UTILITY::ConvertPathToFullPath(L"\\hostfxr.dll",
|
||||
pConfig->QueryApplicationPhysicalPath()->QueryStr(),
|
||||
struHostfxrPath)))
|
||||
{
|
||||
goto Finished;
|
||||
}
|
||||
|
||||
|
||||
saAttr.nLength = sizeof(SECURITY_ATTRIBUTES);
|
||||
saAttr.bInheritHandle = TRUE;
|
||||
saAttr.lpSecurityDescriptor = NULL;
|
||||
|
||||
hFileHandle = CreateFile(struHostfxrPath->QueryStr(),
|
||||
GENERIC_READ,
|
||||
FILE_SHARE_READ,
|
||||
&saAttr,
|
||||
OPEN_EXISTING,
|
||||
FILE_ATTRIBUTE_NORMAL,
|
||||
NULL);
|
||||
if (hFileHandle == INVALID_HANDLE_VALUE)
|
||||
{
|
||||
// Treat access isseu as File not found
|
||||
hr = ERROR_FILE_NOT_FOUND;
|
||||
goto Finished;
|
||||
}
|
||||
else
|
||||
{
|
||||
CloseHandle(hFileHandle);
|
||||
}
|
||||
|
||||
Finished:
|
||||
return hr;
|
||||
}
|
||||
|
||||
HRESULT
|
||||
HOSTFXR_UTILITY::GetPortableHostfxrLocation(
|
||||
STRU* struHostfxrPath
|
||||
)
|
||||
{
|
||||
HRESULT hr = S_OK;
|
||||
|
||||
STRU struSystemPathVariable;
|
||||
STRU strDotnetExeLocation;
|
||||
STRU strHostFxrSearchExpression;
|
||||
STRU strHighestDotnetVersion;
|
||||
PWSTR pwzDelimeterContext = NULL;
|
||||
PCWSTR pszDotnetLocation = NULL;
|
||||
PCWSTR pszDotnetExeString(L"dotnet.exe");
|
||||
DWORD dwCopyLength;
|
||||
BOOL fFound = FALSE;
|
||||
HANDLE hFileHandle = INVALID_HANDLE_VALUE;
|
||||
SECURITY_ATTRIBUTES saAttr;
|
||||
std::vector<std::wstring> vVersionFolders;
|
||||
|
||||
if (FAILED(hr))
|
||||
{
|
||||
goto Finished;
|
||||
}
|
||||
|
||||
// Get the System PATH value.
|
||||
if (!UTILITY::GetSystemPathVariable(L"PATH", &struSystemPathVariable))
|
||||
{
|
||||
hr = ERROR_BAD_ENVIRONMENT;
|
||||
goto Finished;
|
||||
}
|
||||
|
||||
// Split on ';', checking to see if dotnet.exe exists in any folders.
|
||||
pszDotnetLocation = wcstok_s(struSystemPathVariable.QueryStr(), L";", &pwzDelimeterContext);
|
||||
while (pszDotnetLocation != NULL)
|
||||
{
|
||||
dwCopyLength = (DWORD) wcsnlen_s(pszDotnetLocation, 260);
|
||||
|
||||
// We store both the exe and folder locations as we eventually need to check inside of host\\fxr
|
||||
// which doesn't need the dotnet.exe portion of the string
|
||||
hr = strDotnetExeLocation.Copy(pszDotnetLocation, dwCopyLength);
|
||||
if (FAILED(hr))
|
||||
{
|
||||
goto Finished;
|
||||
}
|
||||
|
||||
if (dwCopyLength > 0 && pszDotnetLocation[dwCopyLength - 1] != L'\\')
|
||||
{
|
||||
hr = strDotnetExeLocation.Append(L"\\");
|
||||
if (FAILED(hr))
|
||||
{
|
||||
goto Finished;
|
||||
}
|
||||
}
|
||||
|
||||
hr = struHostfxrPath->Copy(strDotnetExeLocation);
|
||||
if (FAILED(hr))
|
||||
{
|
||||
goto Finished;
|
||||
}
|
||||
|
||||
hr = strDotnetExeLocation.Append(pszDotnetExeString);
|
||||
if (FAILED(hr))
|
||||
{
|
||||
goto Finished;
|
||||
}
|
||||
|
||||
saAttr.nLength = sizeof(SECURITY_ATTRIBUTES);
|
||||
saAttr.bInheritHandle = TRUE;
|
||||
saAttr.lpSecurityDescriptor = NULL;
|
||||
|
||||
hFileHandle = CreateFile(strDotnetExeLocation.QueryStr(),
|
||||
GENERIC_READ,
|
||||
FILE_SHARE_READ,
|
||||
&saAttr,
|
||||
OPEN_EXISTING,
|
||||
FILE_ATTRIBUTE_NORMAL,
|
||||
NULL);
|
||||
if (hFileHandle != INVALID_HANDLE_VALUE)
|
||||
{
|
||||
// means we found the folder with a dotnet.exe inside of it.
|
||||
fFound = TRUE;
|
||||
CloseHandle(hFileHandle);
|
||||
break;
|
||||
}
|
||||
pszDotnetLocation = wcstok_s(NULL, L";", &pwzDelimeterContext);
|
||||
}
|
||||
|
||||
if (!fFound)
|
||||
{
|
||||
// could not find dotnet.exe, error out
|
||||
hr = ERROR_BAD_ENVIRONMENT;
|
||||
goto Finished;
|
||||
}
|
||||
|
||||
hr = struHostfxrPath->Append(L"host\\fxr");
|
||||
if (FAILED(hr))
|
||||
{
|
||||
goto Finished;
|
||||
}
|
||||
|
||||
if (!UTILITY::DirectoryExists(struHostfxrPath))
|
||||
{
|
||||
// error, not found the folder
|
||||
hr = ERROR_BAD_ENVIRONMENT;
|
||||
goto Finished;
|
||||
}
|
||||
|
||||
// Find all folders under host\\fxr\\ for version numbers.
|
||||
hr = strHostFxrSearchExpression.Copy(struHostfxrPath);
|
||||
if (FAILED(hr))
|
||||
{
|
||||
goto Finished;
|
||||
}
|
||||
|
||||
hr = strHostFxrSearchExpression.Append(L"\\*");
|
||||
if (FAILED(hr))
|
||||
{
|
||||
goto Finished;
|
||||
}
|
||||
|
||||
// As we use the logic from core-setup, we are opting to use std here.
|
||||
// TODO remove all uses of std?
|
||||
UTILITY::FindDotNetFolders(strHostFxrSearchExpression.QueryStr(), &vVersionFolders);
|
||||
|
||||
if (vVersionFolders.size() == 0)
|
||||
{
|
||||
// no core framework was found
|
||||
hr = ERROR_BAD_ENVIRONMENT;
|
||||
goto Finished;
|
||||
}
|
||||
|
||||
hr = UTILITY::FindHighestDotNetVersion(vVersionFolders, &strHighestDotnetVersion);
|
||||
if (FAILED(hr))
|
||||
{
|
||||
goto Finished;
|
||||
}
|
||||
hr = struHostfxrPath->Append(L"\\");
|
||||
if (FAILED(hr))
|
||||
{
|
||||
goto Finished;
|
||||
}
|
||||
|
||||
hr = struHostfxrPath->Append(strHighestDotnetVersion.QueryStr());
|
||||
if (FAILED(hr))
|
||||
{
|
||||
goto Finished;
|
||||
|
||||
}
|
||||
|
||||
hr = struHostfxrPath->Append(L"\\hostfxr.dll");
|
||||
if (FAILED(hr))
|
||||
{
|
||||
goto Finished;
|
||||
}
|
||||
|
||||
Finished:
|
||||
return hr;
|
||||
}
|
||||
|
|
@ -0,0 +1,33 @@
|
|||
// Copyright (c) .NET Foundation. All rights reserved.
|
||||
// Licensed under the MIT License. See License.txt in the project root for license information.
|
||||
|
||||
#pragma once
|
||||
|
||||
class HOSTFXR_UTILITY
|
||||
{
|
||||
public:
|
||||
HOSTFXR_UTILITY();
|
||||
~HOSTFXR_UTILITY();
|
||||
|
||||
static
|
||||
HRESULT
|
||||
FindHostFxrDll(
|
||||
ASPNETCORE_CONFIG *pConfig,
|
||||
STRU* struHostFxrDllLocation,
|
||||
BOOL* fStandAlone
|
||||
);
|
||||
|
||||
static
|
||||
HRESULT
|
||||
GetStandaloneHostfxrLocation(
|
||||
STRU* struHostfxrPath,
|
||||
ASPNETCORE_CONFIG *pConfig
|
||||
);
|
||||
|
||||
static
|
||||
HRESULT
|
||||
GetPortableHostfxrLocation(
|
||||
STRU* struHostfxrPath
|
||||
);
|
||||
};
|
||||
|
||||
|
|
@ -0,0 +1,44 @@
|
|||
// Copyright (c) .NET Foundation. All rights reserved.
|
||||
// Licensed under the MIT License. See License.txt in the project root for license information.
|
||||
|
||||
#include "stdafx.h"
|
||||
|
||||
REQUEST_HANDLER::REQUEST_HANDLER(
|
||||
_In_ IHttpContext *pW3Context,
|
||||
_In_ HTTP_MODULE_ID *pModuleId,
|
||||
_In_ APPLICATION *pApplication)
|
||||
: m_cRefs(1)
|
||||
{
|
||||
m_pW3Context = pW3Context;
|
||||
m_pApplication = pApplication;
|
||||
m_pModuleId = *pModuleId;
|
||||
}
|
||||
|
||||
|
||||
REQUEST_HANDLER::~REQUEST_HANDLER()
|
||||
{
|
||||
}
|
||||
|
||||
VOID
|
||||
REQUEST_HANDLER::ReferenceRequestHandler(
|
||||
VOID
|
||||
) const
|
||||
{
|
||||
InterlockedIncrement(&m_cRefs);
|
||||
}
|
||||
|
||||
|
||||
VOID
|
||||
REQUEST_HANDLER::DereferenceRequestHandler(
|
||||
VOID
|
||||
) const
|
||||
{
|
||||
DBG_ASSERT(m_cRefs != 0);
|
||||
|
||||
LONG cRefs = 0;
|
||||
if ((cRefs = InterlockedDecrement(&m_cRefs)) == 0)
|
||||
{
|
||||
delete this;
|
||||
}
|
||||
|
||||
}
|
||||
|
|
@ -0,0 +1,59 @@
|
|||
// Copyright (c) .NET Foundation. All rights reserved.
|
||||
// Licensed under the MIT License. See License.txt in the project root for license information.
|
||||
|
||||
#pragma once
|
||||
|
||||
#include "stdafx.h"
|
||||
#include "application.h"
|
||||
|
||||
//
|
||||
// Abstract class
|
||||
//
|
||||
class REQUEST_HANDLER
|
||||
{
|
||||
public:
|
||||
REQUEST_HANDLER(
|
||||
_In_ IHttpContext *pW3Context,
|
||||
_In_ HTTP_MODULE_ID *pModuleId,
|
||||
_In_ APPLICATION *pApplication
|
||||
);
|
||||
|
||||
virtual
|
||||
REQUEST_NOTIFICATION_STATUS
|
||||
OnExecuteRequestHandler() = 0;
|
||||
|
||||
virtual
|
||||
REQUEST_NOTIFICATION_STATUS
|
||||
OnAsyncCompletion(
|
||||
DWORD cbCompletion,
|
||||
HRESULT hrCompletionStatus
|
||||
) = 0;
|
||||
|
||||
virtual
|
||||
VOID
|
||||
TerminateRequest(
|
||||
bool fClientInitiated
|
||||
) = 0;
|
||||
|
||||
virtual
|
||||
~REQUEST_HANDLER(
|
||||
VOID
|
||||
);
|
||||
|
||||
VOID
|
||||
ReferenceRequestHandler(
|
||||
VOID
|
||||
) const;
|
||||
|
||||
virtual
|
||||
VOID
|
||||
DereferenceRequestHandler(
|
||||
VOID
|
||||
) const;
|
||||
|
||||
protected:
|
||||
mutable LONG m_cRefs;
|
||||
IHttpContext* m_pW3Context;
|
||||
APPLICATION* m_pApplication;
|
||||
HTTP_MODULE_ID m_pModuleId;
|
||||
};
|
||||
|
|
@ -0,0 +1,4 @@
|
|||
// Copyright (c) .NET Foundation. All rights reserved.
|
||||
// Licensed under the MIT License. See License.txt in the project root for license information.
|
||||
|
||||
#include "stdafx.h"
|
||||
|
|
@ -0,0 +1,28 @@
|
|||
// Copyright (c) .NET Foundation. All rights reserved.
|
||||
// Licensed under the MIT License. See License.txt in the project root for license information.
|
||||
|
||||
#pragma once
|
||||
|
||||
#include "targetver.h"
|
||||
|
||||
#define WIN32_LEAN_AND_MEAN // Exclude rarely-used stuff from Windows headers
|
||||
|
||||
#include <Windows.h>
|
||||
#include <httpserv.h>
|
||||
#include <wchar.h>
|
||||
#include <vector>
|
||||
#include <sstream>
|
||||
#include "..\IISLib\hashtable.h"
|
||||
#include "..\IISLib\stringu.h"
|
||||
#include "..\IISLib\stringa.h"
|
||||
#include "..\IISLib\multisz.h"
|
||||
#include "..\IISLib\dbgutil.h"
|
||||
#include "..\IISLib\ahutil.h"
|
||||
#include "..\IISLib\hashfn.h"
|
||||
#include "environmentvariablehash.h"
|
||||
#include "utility.h"
|
||||
#include "aspnetcoreconfig.h"
|
||||
#include "application.h"
|
||||
#include "requesthandler.h"
|
||||
#include "fx_ver.h"
|
||||
#include "hostfxr_utility.h"
|
||||
|
|
@ -0,0 +1,8 @@
|
|||
#pragma once
|
||||
|
||||
// Including SDKDDKVer.h defines the highest available Windows platform.
|
||||
|
||||
// If you wish to build your application for a previous Windows platform, include WinSDKVer.h and
|
||||
// set the _WIN32_WINNT macro to the platform you wish to support before including SDKDDKVer.h.
|
||||
|
||||
#include <SDKDDKVer.h>
|
||||
|
|
@ -0,0 +1,609 @@
|
|||
// Copyright (c) .NET Foundation. All rights reserved.
|
||||
// Licensed under the MIT License. See License.txt in the project root for license information.
|
||||
|
||||
#include"stdafx.h"
|
||||
|
||||
// static
|
||||
HRESULT
|
||||
UTILITY::SplitUrl(
|
||||
PCWSTR pszDestinationUrl,
|
||||
BOOL *pfSecure,
|
||||
STRU *pstrDestination,
|
||||
STRU *pstrUrl
|
||||
)
|
||||
/*++
|
||||
|
||||
Routine Description:
|
||||
|
||||
Split the URL specified for forwarding into its specific components
|
||||
The format of the URL looks like
|
||||
http[s]://destination[:port]/path
|
||||
when port is omitted, the default port for that specific protocol is used
|
||||
when host is omitted, it gets the same value as the destination
|
||||
|
||||
Arguments:
|
||||
|
||||
pszDestinationUrl - the url to be split up
|
||||
pfSecure - SSL to be used in forwarding?
|
||||
pstrDestination - destination
|
||||
pDestinationPort - port
|
||||
pstrUrl - URL
|
||||
|
||||
Return Value:
|
||||
|
||||
HRESULT
|
||||
|
||||
--*/
|
||||
{
|
||||
HRESULT hr;
|
||||
|
||||
//
|
||||
// First determine if the target is secure
|
||||
//
|
||||
if (_wcsnicmp(pszDestinationUrl, L"http://", 7) == 0)
|
||||
{
|
||||
*pfSecure = FALSE;
|
||||
pszDestinationUrl += 7;
|
||||
}
|
||||
else if (_wcsnicmp(pszDestinationUrl, L"https://", 8) == 0)
|
||||
{
|
||||
*pfSecure = TRUE;
|
||||
pszDestinationUrl += 8;
|
||||
}
|
||||
else
|
||||
{
|
||||
return HRESULT_FROM_WIN32(ERROR_INVALID_DATA);
|
||||
}
|
||||
|
||||
if (*pszDestinationUrl == L'\0')
|
||||
{
|
||||
return HRESULT_FROM_WIN32(ERROR_INVALID_DATA);
|
||||
}
|
||||
|
||||
//
|
||||
// Find the 3rd slash corresponding to the url
|
||||
//
|
||||
LPCWSTR pszSlash = wcschr(pszDestinationUrl, L'/');
|
||||
if (pszSlash == NULL)
|
||||
{
|
||||
if (FAILED(hr = pstrUrl->Copy(L"/", 1)) ||
|
||||
FAILED(hr = pstrDestination->Copy(pszDestinationUrl)))
|
||||
{
|
||||
return hr;
|
||||
}
|
||||
}
|
||||
else
|
||||
{
|
||||
if (FAILED(hr = pstrUrl->Copy(pszSlash)) ||
|
||||
FAILED(hr = pstrDestination->Copy(pszDestinationUrl,
|
||||
(DWORD)(pszSlash - pszDestinationUrl))))
|
||||
{
|
||||
return hr;
|
||||
}
|
||||
}
|
||||
|
||||
return S_OK;
|
||||
}
|
||||
|
||||
// Change a hexadecimal digit to its numerical equivalent
|
||||
#define TOHEX( ch ) \
|
||||
((ch) > L'9' ? \
|
||||
(ch) >= L'a' ? \
|
||||
(ch) - L'a' + 10 : \
|
||||
(ch) - L'A' + 10 \
|
||||
: (ch) - L'0')
|
||||
|
||||
// static
|
||||
HRESULT
|
||||
UTILITY::UnEscapeUrl(
|
||||
PCWSTR pszUrl,
|
||||
DWORD cchUrl,
|
||||
bool fCopyQuery,
|
||||
STRA * pstrResult
|
||||
)
|
||||
{
|
||||
HRESULT hr;
|
||||
CHAR pch[2];
|
||||
pch[1] = '\0';
|
||||
DWORD cchStart = 0;
|
||||
DWORD index = 0;
|
||||
|
||||
while (index < cchUrl &&
|
||||
(fCopyQuery || pszUrl[index] != L'?'))
|
||||
{
|
||||
switch (pszUrl[index])
|
||||
{
|
||||
case L'%':
|
||||
if (iswxdigit(pszUrl[index+1]) && iswxdigit(pszUrl[index+2]))
|
||||
{
|
||||
if (index > cchStart &&
|
||||
FAILED(hr = pstrResult->AppendW(pszUrl + cchStart,
|
||||
index - cchStart)))
|
||||
{
|
||||
return hr;
|
||||
}
|
||||
cchStart = index+3;
|
||||
|
||||
pch[0] = static_cast<CHAR>(TOHEX(pszUrl[index+1]) * 16 +
|
||||
TOHEX(pszUrl[index+2]));
|
||||
if (FAILED(hr = pstrResult->Append(pch, 1)))
|
||||
{
|
||||
return hr;
|
||||
}
|
||||
index += 3;
|
||||
break;
|
||||
}
|
||||
|
||||
__fallthrough;
|
||||
default:
|
||||
index++;
|
||||
}
|
||||
}
|
||||
|
||||
if (index > cchStart)
|
||||
{
|
||||
return pstrResult->AppendW(pszUrl + cchStart,
|
||||
index - cchStart);
|
||||
}
|
||||
|
||||
return S_OK;
|
||||
}
|
||||
|
||||
// static
|
||||
HRESULT
|
||||
UTILITY::UnEscapeUrl(
|
||||
PCWSTR pszUrl,
|
||||
DWORD cchUrl,
|
||||
STRU * pstrResult
|
||||
)
|
||||
{
|
||||
HRESULT hr;
|
||||
WCHAR pch[2];
|
||||
pch[1] = L'\0';
|
||||
DWORD cchStart = 0;
|
||||
DWORD index = 0;
|
||||
bool fInQuery = FALSE;
|
||||
|
||||
while (index < cchUrl)
|
||||
{
|
||||
switch (pszUrl[index])
|
||||
{
|
||||
case L'%':
|
||||
if (iswxdigit(pszUrl[index+1]) && iswxdigit(pszUrl[index+2]))
|
||||
{
|
||||
if (index > cchStart &&
|
||||
FAILED(hr = pstrResult->Append(pszUrl + cchStart,
|
||||
index - cchStart)))
|
||||
{
|
||||
return hr;
|
||||
}
|
||||
cchStart = index+3;
|
||||
|
||||
pch[0] = static_cast<WCHAR>(TOHEX(pszUrl[index+1]) * 16 +
|
||||
TOHEX(pszUrl[index+2]));
|
||||
if (FAILED(hr = pstrResult->Append(pch, 1)))
|
||||
{
|
||||
return hr;
|
||||
}
|
||||
index += 3;
|
||||
if (pch[0] == L'?')
|
||||
{
|
||||
fInQuery = TRUE;
|
||||
}
|
||||
break;
|
||||
}
|
||||
|
||||
index++;
|
||||
break;
|
||||
|
||||
case L'/':
|
||||
if (fInQuery)
|
||||
{
|
||||
if (index > cchStart &&
|
||||
FAILED(hr = pstrResult->Append(pszUrl + cchStart,
|
||||
index - cchStart)))
|
||||
{
|
||||
return hr;
|
||||
}
|
||||
cchStart = index+1;
|
||||
|
||||
if (FAILED(hr = pstrResult->Append(L"\\", 1)))
|
||||
{
|
||||
return hr;
|
||||
}
|
||||
index += 1;
|
||||
break;
|
||||
}
|
||||
|
||||
__fallthrough;
|
||||
default:
|
||||
index++;
|
||||
}
|
||||
}
|
||||
|
||||
if (index > cchStart)
|
||||
{
|
||||
return pstrResult->Append(pszUrl + cchStart,
|
||||
index - cchStart);
|
||||
}
|
||||
|
||||
return S_OK;
|
||||
}
|
||||
|
||||
HRESULT
|
||||
UTILITY::EscapeAbsPath(
|
||||
IHttpRequest * pRequest,
|
||||
STRU * strEscapedUrl
|
||||
)
|
||||
{
|
||||
HRESULT hr = S_OK;
|
||||
STRU strAbsPath;
|
||||
LPCWSTR pszAbsPath = NULL;
|
||||
LPCWSTR pszFindStr = NULL;
|
||||
|
||||
hr = strAbsPath.Copy( pRequest->GetRawHttpRequest()->CookedUrl.pAbsPath,
|
||||
pRequest->GetRawHttpRequest()->CookedUrl.AbsPathLength / sizeof(WCHAR) );
|
||||
if(FAILED(hr))
|
||||
{
|
||||
goto Finished;
|
||||
}
|
||||
|
||||
pszAbsPath = strAbsPath.QueryStr();
|
||||
pszFindStr = wcschr(pszAbsPath, L'?');
|
||||
|
||||
while(pszFindStr != NULL)
|
||||
{
|
||||
strEscapedUrl->Append( pszAbsPath, pszFindStr - pszAbsPath);
|
||||
strEscapedUrl->Append(L"%3F");
|
||||
pszAbsPath = pszFindStr + 1;
|
||||
pszFindStr = wcschr(pszAbsPath, L'?');
|
||||
}
|
||||
|
||||
strEscapedUrl->Append(pszAbsPath);
|
||||
strEscapedUrl->Append(pRequest->GetRawHttpRequest()->CookedUrl.pQueryString,
|
||||
pRequest->GetRawHttpRequest()->CookedUrl.QueryStringLength / sizeof(WCHAR));
|
||||
|
||||
Finished:
|
||||
return hr;
|
||||
}
|
||||
|
||||
// static
|
||||
bool
|
||||
UTILITY::IsValidAttributeNameChar(
|
||||
WCHAR ch
|
||||
)
|
||||
{
|
||||
//
|
||||
// Values based on ASP.NET rendering for cookie names. RFC 2965 is not clear
|
||||
// what the non-special characters are.
|
||||
//
|
||||
return ch == L'\t' || (ch > 31 && ch < 127);
|
||||
}
|
||||
|
||||
// static
|
||||
bool
|
||||
UTILITY::FindInMultiString(
|
||||
PCWSTR pszMultiString,
|
||||
PCWSTR pszStringToFind
|
||||
)
|
||||
{
|
||||
while (*pszMultiString != L'\0')
|
||||
{
|
||||
if (wcscmp(pszMultiString, pszStringToFind) == 0)
|
||||
{
|
||||
return TRUE;
|
||||
}
|
||||
pszMultiString += wcslen(pszMultiString) + 1;
|
||||
}
|
||||
|
||||
return FALSE;
|
||||
}
|
||||
|
||||
// static
|
||||
bool
|
||||
UTILITY::IsValidQueryStringName(
|
||||
PCWSTR pszName
|
||||
)
|
||||
{
|
||||
while (*pszName != L'\0')
|
||||
{
|
||||
WCHAR c = *pszName;
|
||||
if (c != L'-' && c != L'_' && c != L'+' &&
|
||||
c != L'.' && c != L'*' && c != L'$' && c != L'%' && c != L',' &&
|
||||
!iswalnum(c))
|
||||
{
|
||||
return FALSE;
|
||||
}
|
||||
pszName++;
|
||||
}
|
||||
|
||||
return TRUE;
|
||||
}
|
||||
|
||||
// static
|
||||
bool
|
||||
UTILITY::IsValidHeaderName(
|
||||
PCWSTR pszName
|
||||
)
|
||||
{
|
||||
while (*pszName != L'\0')
|
||||
{
|
||||
WCHAR c = *pszName;
|
||||
if (c != L'-' && c != L'_' && c != L'+' &&
|
||||
c != L'.' && c != L'*' && c != L'$' && c != L'%'
|
||||
&& !iswalnum(c))
|
||||
{
|
||||
return FALSE;
|
||||
}
|
||||
pszName++;
|
||||
}
|
||||
|
||||
return TRUE;
|
||||
}
|
||||
|
||||
HRESULT
|
||||
UTILITY::IsPathUnc(
|
||||
__in LPCWSTR pszPath,
|
||||
__out BOOL * pfIsUnc
|
||||
)
|
||||
{
|
||||
HRESULT hr = S_OK;
|
||||
STRU strTempPath;
|
||||
|
||||
if ( pszPath == NULL || pfIsUnc == NULL )
|
||||
{
|
||||
hr = E_INVALIDARG;
|
||||
goto Finished;
|
||||
}
|
||||
|
||||
hr = MakePathCanonicalizationProof( (LPWSTR) pszPath, &strTempPath );
|
||||
if ( FAILED(hr) )
|
||||
{
|
||||
goto Finished;
|
||||
}
|
||||
|
||||
//
|
||||
// MakePathCanonicalizationProof will map \\?\UNC, \\.\UNC and \\ to \\?\UNC
|
||||
//
|
||||
(*pfIsUnc) = ( _wcsnicmp( strTempPath.QueryStr(), L"\\\\?\\UNC\\", 8 /* sizeof \\?\UNC\ */) == 0 );
|
||||
|
||||
Finished:
|
||||
|
||||
return hr;
|
||||
}
|
||||
|
||||
HRESULT
|
||||
UTILITY::ConvertPathToFullPath(
|
||||
_In_ LPCWSTR pszPath,
|
||||
_In_ LPCWSTR pszRootPath,
|
||||
_Out_ STRU* pStruFullPath
|
||||
)
|
||||
{
|
||||
HRESULT hr = S_OK;
|
||||
STRU strFileFullPath;
|
||||
LPWSTR pszFullPath = NULL;
|
||||
|
||||
// if relative path, prefix with root path and then convert to absolute path.
|
||||
if ( pszPath[0] == L'.' )
|
||||
{
|
||||
hr = strFileFullPath.Copy(pszRootPath);
|
||||
if(FAILED(hr))
|
||||
{
|
||||
goto Finished;
|
||||
}
|
||||
|
||||
if(!strFileFullPath.EndsWith(L"\\"))
|
||||
{
|
||||
hr = strFileFullPath.Append(L"\\");
|
||||
if(FAILED(hr))
|
||||
{
|
||||
goto Finished;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
hr = strFileFullPath.Append( pszPath );
|
||||
if (FAILED(hr))
|
||||
{
|
||||
goto Finished;
|
||||
}
|
||||
|
||||
pszFullPath = new WCHAR[ strFileFullPath.QueryCCH() + 1];
|
||||
if ( pszFullPath == NULL )
|
||||
{
|
||||
hr = E_OUTOFMEMORY;
|
||||
goto Finished;
|
||||
}
|
||||
|
||||
if(_wfullpath( pszFullPath,
|
||||
strFileFullPath.QueryStr(),
|
||||
strFileFullPath.QueryCCH() + 1 ) == NULL )
|
||||
{
|
||||
hr = HRESULT_FROM_WIN32( ERROR_INVALID_PARAMETER );
|
||||
goto Finished;
|
||||
}
|
||||
|
||||
// convert to canonical path
|
||||
hr = MakePathCanonicalizationProof( pszFullPath, pStruFullPath );
|
||||
if (FAILED(hr))
|
||||
{
|
||||
goto Finished;
|
||||
}
|
||||
|
||||
Finished:
|
||||
|
||||
if ( pszFullPath != NULL )
|
||||
{
|
||||
delete[] pszFullPath;
|
||||
pszFullPath = NULL;
|
||||
}
|
||||
|
||||
return hr;
|
||||
}
|
||||
|
||||
HRESULT
|
||||
UTILITY::EnsureDirectoryPathExist(
|
||||
_In_ LPCWSTR pszPath
|
||||
)
|
||||
{
|
||||
HRESULT hr = S_OK;
|
||||
STRU struPath;
|
||||
DWORD dwPosition = 0;
|
||||
BOOL fDone = FALSE;
|
||||
BOOL fUnc = FALSE;
|
||||
|
||||
struPath.Copy(pszPath);
|
||||
hr = IsPathUnc(pszPath, &fUnc);
|
||||
if (FAILED(hr))
|
||||
{
|
||||
goto Finished;
|
||||
}
|
||||
if (fUnc)
|
||||
{
|
||||
// "\\?\UNC\"
|
||||
dwPosition = 8;
|
||||
}
|
||||
else if (struPath.IndexOf(L'?', 0) != -1)
|
||||
{
|
||||
// sceanrio "\\?\"
|
||||
dwPosition = 4;
|
||||
}
|
||||
while (!fDone)
|
||||
{
|
||||
dwPosition = struPath.IndexOf(L'\\', dwPosition + 1);
|
||||
if (dwPosition == -1)
|
||||
{
|
||||
// not found '/'
|
||||
fDone = TRUE;
|
||||
goto Finished;
|
||||
}
|
||||
else if (dwPosition ==0)
|
||||
{
|
||||
hr = ERROR_INTERNAL_ERROR;
|
||||
goto Finished;
|
||||
}
|
||||
else if (struPath.QueryStr()[dwPosition-1] == L':')
|
||||
{
|
||||
// skip volume case
|
||||
continue;
|
||||
}
|
||||
else
|
||||
{
|
||||
struPath.QueryStr()[dwPosition] = L'\0';
|
||||
}
|
||||
|
||||
if (!CreateDirectory(struPath.QueryStr(), NULL) &&
|
||||
ERROR_ALREADY_EXISTS != GetLastError())
|
||||
{
|
||||
hr = HRESULT_FROM_WIN32(GetLastError());
|
||||
fDone = TRUE;
|
||||
goto Finished;
|
||||
}
|
||||
struPath.QueryStr()[dwPosition] = L'\\';
|
||||
}
|
||||
|
||||
Finished:
|
||||
return hr;
|
||||
}
|
||||
|
||||
HRESULT
|
||||
UTILITY::FindHighestDotNetVersion(
|
||||
_In_ std::vector<std::wstring> vFolders,
|
||||
_Out_ STRU *pstrResult
|
||||
)
|
||||
{
|
||||
HRESULT hr = S_OK;
|
||||
fx_ver_t max_ver(-1, -1, -1);
|
||||
for (const auto& dir : vFolders)
|
||||
{
|
||||
fx_ver_t fx_ver(-1, -1, -1);
|
||||
if (fx_ver_t::parse(dir, &fx_ver, false))
|
||||
{
|
||||
// TODO using max instead of std::max works
|
||||
max_ver = max(max_ver, fx_ver);
|
||||
}
|
||||
}
|
||||
|
||||
hr = pstrResult->Copy(max_ver.as_str().c_str());
|
||||
|
||||
// we check FAILED(hr) outside of function
|
||||
return hr;
|
||||
}
|
||||
|
||||
BOOL
|
||||
UTILITY::DirectoryExists(
|
||||
_In_ STRU *pstrPath
|
||||
)
|
||||
{
|
||||
WIN32_FILE_ATTRIBUTE_DATA data;
|
||||
|
||||
if (pstrPath->IsEmpty())
|
||||
{
|
||||
return false;
|
||||
}
|
||||
|
||||
return GetFileAttributesExW(pstrPath->QueryStr(), GetFileExInfoStandard, &data);
|
||||
}
|
||||
|
||||
BOOL
|
||||
UTILITY::GetSystemPathVariable(
|
||||
_In_ PCWSTR pszEnvironmentVariable,
|
||||
_Out_ STRU *pstrResult
|
||||
)
|
||||
{
|
||||
DWORD dwLength;
|
||||
PWSTR pszBuffer = NULL;
|
||||
BOOL fSucceeded = FALSE;
|
||||
|
||||
if (pszEnvironmentVariable == NULL)
|
||||
{
|
||||
goto Finished;
|
||||
}
|
||||
pstrResult->Reset();
|
||||
dwLength = GetEnvironmentVariableW(pszEnvironmentVariable, NULL, 0);
|
||||
|
||||
if (dwLength == 0)
|
||||
{
|
||||
goto Finished;
|
||||
}
|
||||
|
||||
pszBuffer = new WCHAR[dwLength];
|
||||
if (GetEnvironmentVariableW(pszEnvironmentVariable, pszBuffer, dwLength) == 0)
|
||||
{
|
||||
goto Finished;
|
||||
}
|
||||
|
||||
pstrResult->Copy(pszBuffer);
|
||||
|
||||
fSucceeded = TRUE;
|
||||
|
||||
Finished:
|
||||
if (pszBuffer != NULL) {
|
||||
delete[] pszBuffer;
|
||||
}
|
||||
return fSucceeded;
|
||||
}
|
||||
|
||||
VOID
|
||||
UTILITY::FindDotNetFolders(
|
||||
_In_ PCWSTR pszPath,
|
||||
_Out_ std::vector<std::wstring> *pvFolders
|
||||
)
|
||||
{
|
||||
HANDLE handle = NULL;
|
||||
WIN32_FIND_DATAW data = { 0 };
|
||||
|
||||
handle = FindFirstFileExW(pszPath, FindExInfoStandard, &data, FindExSearchNameMatch, NULL, 0);
|
||||
if (handle == INVALID_HANDLE_VALUE)
|
||||
{
|
||||
return;
|
||||
}
|
||||
|
||||
do
|
||||
{
|
||||
std::wstring folder(data.cFileName);
|
||||
pvFolders->push_back(folder);
|
||||
} while (FindNextFileW(handle, &data));
|
||||
|
||||
FindClose(handle);
|
||||
}
|
||||
|
|
@ -0,0 +1,119 @@
|
|||
// Copyright (c) .NET Foundation. All rights reserved.
|
||||
// Licensed under the MIT License. See License.txt in the project root for license information.
|
||||
|
||||
#pragma once
|
||||
|
||||
class UTILITY
|
||||
{
|
||||
public:
|
||||
|
||||
static
|
||||
HRESULT
|
||||
SplitUrl(
|
||||
PCWSTR pszDestinationUrl,
|
||||
BOOL *pfSecure,
|
||||
STRU *pstrDestination,
|
||||
STRU *pstrUrl
|
||||
);
|
||||
|
||||
static
|
||||
HRESULT
|
||||
UnEscapeUrl(
|
||||
PCWSTR pszUrl,
|
||||
DWORD cchUrl,
|
||||
bool fCopyQuery,
|
||||
STRA * pstrResult
|
||||
);
|
||||
|
||||
static
|
||||
HRESULT
|
||||
UnEscapeUrl(
|
||||
PCWSTR pszUrl,
|
||||
DWORD cchUrl,
|
||||
STRU * pstrResult
|
||||
);
|
||||
|
||||
static HRESULT
|
||||
EscapeAbsPath(
|
||||
IHttpRequest * pRequest,
|
||||
STRU * strEscapedUrl
|
||||
);
|
||||
|
||||
static
|
||||
bool
|
||||
IsValidAttributeNameChar(
|
||||
WCHAR ch
|
||||
);
|
||||
|
||||
static
|
||||
bool
|
||||
IsValidQueryStringName(
|
||||
PCWSTR pszName
|
||||
);
|
||||
|
||||
static
|
||||
bool
|
||||
IsValidHeaderName(
|
||||
PCWSTR pszName
|
||||
);
|
||||
|
||||
static
|
||||
bool
|
||||
FindInMultiString(
|
||||
PCWSTR pszMultiString,
|
||||
PCWSTR pszStringToFind
|
||||
);
|
||||
|
||||
static
|
||||
HRESULT
|
||||
IsPathUnc(
|
||||
__in LPCWSTR pszPath,
|
||||
__out BOOL * pfIsUnc
|
||||
);
|
||||
|
||||
static
|
||||
HRESULT
|
||||
ConvertPathToFullPath(
|
||||
_In_ LPCWSTR pszPath,
|
||||
_In_ LPCWSTR pszRootPath,
|
||||
_Out_ STRU* pStrFullPath
|
||||
);
|
||||
|
||||
static
|
||||
HRESULT
|
||||
EnsureDirectoryPathExist(
|
||||
_In_ LPCWSTR pszPath
|
||||
);
|
||||
|
||||
static
|
||||
BOOL
|
||||
DirectoryExists(
|
||||
_In_ STRU *pstrPath
|
||||
);
|
||||
|
||||
static
|
||||
BOOL
|
||||
GetSystemPathVariable(
|
||||
_In_ PCWSTR pszEnvironmentVariable,
|
||||
_Out_ STRU *pstrResult
|
||||
);
|
||||
|
||||
static
|
||||
VOID
|
||||
FindDotNetFolders(
|
||||
_In_ PCWSTR pszPath,
|
||||
_Out_ std::vector<std::wstring> *pvFolders
|
||||
);
|
||||
|
||||
static
|
||||
HRESULT
|
||||
FindHighestDotNetVersion(
|
||||
_In_ std::vector<std::wstring> vFolders,
|
||||
_Out_ STRU *pstrResult
|
||||
);
|
||||
|
||||
private:
|
||||
|
||||
UTILITY() {}
|
||||
~UTILITY() {}
|
||||
};
|
||||
|
|
@ -0,0 +1,186 @@
|
|||
<?xml version="1.0" encoding="utf-8"?>
|
||||
<Project DefaultTargets="Build" ToolsVersion="15.0" xmlns="http://schemas.microsoft.com/developer/msbuild/2003">
|
||||
<ItemGroup Label="ProjectConfigurations">
|
||||
<ProjectConfiguration Include="Debug|Win32">
|
||||
<Configuration>Debug</Configuration>
|
||||
<Platform>Win32</Platform>
|
||||
</ProjectConfiguration>
|
||||
<ProjectConfiguration Include="Debug|x64">
|
||||
<Configuration>Debug</Configuration>
|
||||
<Platform>x64</Platform>
|
||||
</ProjectConfiguration>
|
||||
<ProjectConfiguration Include="Release|Win32">
|
||||
<Configuration>Release</Configuration>
|
||||
<Platform>Win32</Platform>
|
||||
</ProjectConfiguration>
|
||||
<ProjectConfiguration Include="Release|x64">
|
||||
<Configuration>Release</Configuration>
|
||||
<Platform>x64</Platform>
|
||||
</ProjectConfiguration>
|
||||
</ItemGroup>
|
||||
<PropertyGroup Label="Globals">
|
||||
<ProjectGuid>{4787A64F-9A3E-4867-A55A-70CB4B2B2FFE}</ProjectGuid>
|
||||
<Keyword>Win32Proj</Keyword>
|
||||
<RootNamespace>IISLib</RootNamespace>
|
||||
<ProjectName>IISLib</ProjectName>
|
||||
<WindowsTargetPlatformVersion>8.1</WindowsTargetPlatformVersion>
|
||||
</PropertyGroup>
|
||||
<Import Project="$(VCTargetsPath)\Microsoft.Cpp.Default.props" />
|
||||
<PropertyGroup Condition="'$(Configuration)|$(Platform)'=='Debug|Win32'" Label="Configuration">
|
||||
<ConfigurationType>StaticLibrary</ConfigurationType>
|
||||
<UseDebugLibraries>true</UseDebugLibraries>
|
||||
<PlatformToolset>v141</PlatformToolset>
|
||||
<CharacterSet>Unicode</CharacterSet>
|
||||
</PropertyGroup>
|
||||
<PropertyGroup Condition="'$(Configuration)|$(Platform)'=='Debug|x64'" Label="Configuration">
|
||||
<ConfigurationType>StaticLibrary</ConfigurationType>
|
||||
<UseDebugLibraries>true</UseDebugLibraries>
|
||||
<PlatformToolset>v141</PlatformToolset>
|
||||
<CharacterSet>Unicode</CharacterSet>
|
||||
</PropertyGroup>
|
||||
<PropertyGroup Condition="'$(Configuration)|$(Platform)'=='Release|Win32'" Label="Configuration">
|
||||
<ConfigurationType>StaticLibrary</ConfigurationType>
|
||||
<UseDebugLibraries>false</UseDebugLibraries>
|
||||
<PlatformToolset>v141</PlatformToolset>
|
||||
<WholeProgramOptimization>true</WholeProgramOptimization>
|
||||
<CharacterSet>Unicode</CharacterSet>
|
||||
</PropertyGroup>
|
||||
<PropertyGroup Condition="'$(Configuration)|$(Platform)'=='Release|x64'" Label="Configuration">
|
||||
<ConfigurationType>StaticLibrary</ConfigurationType>
|
||||
<UseDebugLibraries>false</UseDebugLibraries>
|
||||
<PlatformToolset>v141</PlatformToolset>
|
||||
<WholeProgramOptimization>true</WholeProgramOptimization>
|
||||
<CharacterSet>Unicode</CharacterSet>
|
||||
</PropertyGroup>
|
||||
<Import Project="$(VCTargetsPath)\Microsoft.Cpp.props" />
|
||||
<ImportGroup Label="ExtensionSettings">
|
||||
</ImportGroup>
|
||||
<ImportGroup Label="PropertySheets" Condition="'$(Configuration)|$(Platform)'=='Debug|Win32'">
|
||||
<Import Project="$(UserRootDir)\Microsoft.Cpp.$(Platform).user.props" Condition="exists('$(UserRootDir)\Microsoft.Cpp.$(Platform).user.props')" Label="LocalAppDataPlatform" />
|
||||
</ImportGroup>
|
||||
<ImportGroup Condition="'$(Configuration)|$(Platform)'=='Debug|x64'" Label="PropertySheets">
|
||||
<Import Project="$(UserRootDir)\Microsoft.Cpp.$(Platform).user.props" Condition="exists('$(UserRootDir)\Microsoft.Cpp.$(Platform).user.props')" Label="LocalAppDataPlatform" />
|
||||
</ImportGroup>
|
||||
<ImportGroup Label="PropertySheets" Condition="'$(Configuration)|$(Platform)'=='Release|Win32'">
|
||||
<Import Project="$(UserRootDir)\Microsoft.Cpp.$(Platform).user.props" Condition="exists('$(UserRootDir)\Microsoft.Cpp.$(Platform).user.props')" Label="LocalAppDataPlatform" />
|
||||
</ImportGroup>
|
||||
<ImportGroup Condition="'$(Configuration)|$(Platform)'=='Release|x64'" Label="PropertySheets">
|
||||
<Import Project="$(UserRootDir)\Microsoft.Cpp.$(Platform).user.props" Condition="exists('$(UserRootDir)\Microsoft.Cpp.$(Platform).user.props')" Label="LocalAppDataPlatform" />
|
||||
</ImportGroup>
|
||||
<PropertyGroup Label="UserMacros" />
|
||||
<PropertyGroup />
|
||||
<ItemDefinitionGroup Condition="'$(Configuration)|$(Platform)'=='Debug|Win32'">
|
||||
<ClCompile>
|
||||
<PrecompiledHeader>
|
||||
</PrecompiledHeader>
|
||||
<WarningLevel>Level3</WarningLevel>
|
||||
<Optimization>Disabled</Optimization>
|
||||
<PreprocessorDefinitions>WIN32;_DEBUG;_LIB;%(PreprocessorDefinitions)</PreprocessorDefinitions>
|
||||
<SDLCheck>true</SDLCheck>
|
||||
<DebugInformationFormat>ProgramDatabase</DebugInformationFormat>
|
||||
<RuntimeLibrary>MultiThreadedDebug</RuntimeLibrary>
|
||||
<ShowIncludes>false</ShowIncludes>
|
||||
</ClCompile>
|
||||
<Link>
|
||||
<SubSystem>Windows</SubSystem>
|
||||
<GenerateDebugInformation>true</GenerateDebugInformation>
|
||||
</Link>
|
||||
</ItemDefinitionGroup>
|
||||
<ItemDefinitionGroup Condition="'$(Configuration)|$(Platform)'=='Debug|x64'">
|
||||
<ClCompile>
|
||||
<PrecompiledHeader>
|
||||
</PrecompiledHeader>
|
||||
<WarningLevel>Level4</WarningLevel>
|
||||
<Optimization>Disabled</Optimization>
|
||||
<PreprocessorDefinitions>WIN32;_DEBUG;_LIB;%(PreprocessorDefinitions)</PreprocessorDefinitions>
|
||||
<SDLCheck>true</SDLCheck>
|
||||
<DebugInformationFormat>ProgramDatabase</DebugInformationFormat>
|
||||
<RuntimeLibrary>MultiThreadedDebug</RuntimeLibrary>
|
||||
<ShowIncludes>false</ShowIncludes>
|
||||
</ClCompile>
|
||||
<Link>
|
||||
<SubSystem>Windows</SubSystem>
|
||||
<GenerateDebugInformation>true</GenerateDebugInformation>
|
||||
</Link>
|
||||
</ItemDefinitionGroup>
|
||||
<ItemDefinitionGroup Condition="'$(Configuration)|$(Platform)'=='Release|Win32'">
|
||||
<ClCompile>
|
||||
<WarningLevel>Level4</WarningLevel>
|
||||
<PrecompiledHeader>
|
||||
</PrecompiledHeader>
|
||||
<Optimization>MaxSpeed</Optimization>
|
||||
<FunctionLevelLinking>true</FunctionLevelLinking>
|
||||
<IntrinsicFunctions>true</IntrinsicFunctions>
|
||||
<PreprocessorDefinitions>WIN32;NDEBUG;_LIB;%(PreprocessorDefinitions)</PreprocessorDefinitions>
|
||||
<SDLCheck>true</SDLCheck>
|
||||
<RuntimeLibrary>MultiThreaded</RuntimeLibrary>
|
||||
<ShowIncludes>false</ShowIncludes>
|
||||
</ClCompile>
|
||||
<Link>
|
||||
<SubSystem>Windows</SubSystem>
|
||||
<GenerateDebugInformation>true</GenerateDebugInformation>
|
||||
<EnableCOMDATFolding>true</EnableCOMDATFolding>
|
||||
<OptimizeReferences>true</OptimizeReferences>
|
||||
</Link>
|
||||
</ItemDefinitionGroup>
|
||||
<ItemDefinitionGroup Condition="'$(Configuration)|$(Platform)'=='Release|x64'">
|
||||
<ClCompile>
|
||||
<WarningLevel>Level4</WarningLevel>
|
||||
<PrecompiledHeader>
|
||||
</PrecompiledHeader>
|
||||
<Optimization>MaxSpeed</Optimization>
|
||||
<FunctionLevelLinking>true</FunctionLevelLinking>
|
||||
<IntrinsicFunctions>true</IntrinsicFunctions>
|
||||
<PreprocessorDefinitions>WIN32;NDEBUG;_LIB;%(PreprocessorDefinitions)</PreprocessorDefinitions>
|
||||
<SDLCheck>true</SDLCheck>
|
||||
<RuntimeLibrary>MultiThreaded</RuntimeLibrary>
|
||||
<ShowIncludes>false</ShowIncludes>
|
||||
</ClCompile>
|
||||
<Link>
|
||||
<SubSystem>Windows</SubSystem>
|
||||
<GenerateDebugInformation>true</GenerateDebugInformation>
|
||||
<EnableCOMDATFolding>true</EnableCOMDATFolding>
|
||||
<OptimizeReferences>true</OptimizeReferences>
|
||||
</Link>
|
||||
</ItemDefinitionGroup>
|
||||
<ItemGroup>
|
||||
<ClInclude Include="acache.h" />
|
||||
<ClInclude Include="ahutil.h" />
|
||||
<ClInclude Include="base64.h" />
|
||||
<ClInclude Include="buffer.h" />
|
||||
<ClInclude Include="datetime.h" />
|
||||
<ClInclude Include="dbgutil.h" />
|
||||
<ClInclude Include="hashfn.h" />
|
||||
<ClInclude Include="hashtable.h" />
|
||||
<ClInclude Include="listentry.h" />
|
||||
<ClInclude Include="macros.h" />
|
||||
<ClInclude Include="multisz.h" />
|
||||
<ClInclude Include="multisza.h" />
|
||||
<ClInclude Include="ntassert.h" />
|
||||
<ClInclude Include="percpu.h" />
|
||||
<ClInclude Include="precomp.h" />
|
||||
<ClInclude Include="prime.h" />
|
||||
<ClInclude Include="pudebug.h" />
|
||||
<ClInclude Include="reftrace.h" />
|
||||
<ClInclude Include="rwlock.h" />
|
||||
<ClInclude Include="stringa.h" />
|
||||
<ClInclude Include="stringu.h" />
|
||||
<ClInclude Include="tracelog.h" />
|
||||
<ClInclude Include="treehash.h" />
|
||||
</ItemGroup>
|
||||
<ItemGroup>
|
||||
<ClCompile Include="acache.cxx" />
|
||||
<ClCompile Include="ahutil.cpp" />
|
||||
<ClCompile Include="base64.cpp" />
|
||||
<ClCompile Include="multisz.cpp" />
|
||||
<ClCompile Include="multisza.cpp" />
|
||||
<ClCompile Include="reftrace.c" />
|
||||
<ClCompile Include="stringa.cpp" />
|
||||
<ClCompile Include="stringu.cpp" />
|
||||
<ClCompile Include="tracelog.c" />
|
||||
<ClCompile Include="util.cxx" />
|
||||
</ItemGroup>
|
||||
<Import Project="$(VCTargetsPath)\Microsoft.Cpp.targets" />
|
||||
<ImportGroup Label="ExtensionTargets">
|
||||
</ImportGroup>
|
||||
</Project>
|
||||
|
|
@ -0,0 +1,443 @@
|
|||
// Copyright (c) .NET Foundation. All rights reserved.
|
||||
// Licensed under the MIT License. See License.txt in the project root for license information.
|
||||
|
||||
#include "precomp.h"
|
||||
|
||||
LONG ALLOC_CACHE_HANDLER::sm_nFillPattern = 0xACA50000;
|
||||
HANDLE ALLOC_CACHE_HANDLER::sm_hHeap;
|
||||
|
||||
//
|
||||
// This class is used to implement the free list. We cast the free'd
|
||||
// memory block to a FREE_LIST_HEADER*. The signature is used to guard against
|
||||
// double deletion. We also fill memory with a pattern.
|
||||
//
|
||||
class FREE_LIST_HEADER
|
||||
{
|
||||
public:
|
||||
SLIST_ENTRY ListEntry;
|
||||
DWORD dwSignature;
|
||||
|
||||
enum
|
||||
{
|
||||
FREE_SIGNATURE = (('A') | ('C' << 8) | ('a' << 16) | (('$' << 24) | 0x80)),
|
||||
};
|
||||
};
|
||||
|
||||
ALLOC_CACHE_HANDLER::ALLOC_CACHE_HANDLER(
|
||||
VOID
|
||||
) : m_nThreshold(0),
|
||||
m_cbSize(0),
|
||||
m_pFreeLists(NULL),
|
||||
m_nTotal(0)
|
||||
{
|
||||
}
|
||||
|
||||
ALLOC_CACHE_HANDLER::~ALLOC_CACHE_HANDLER(
|
||||
VOID
|
||||
)
|
||||
{
|
||||
if (m_pFreeLists != NULL)
|
||||
{
|
||||
CleanupLookaside();
|
||||
m_pFreeLists->Dispose();
|
||||
m_pFreeLists = NULL;
|
||||
}
|
||||
}
|
||||
|
||||
HRESULT
|
||||
ALLOC_CACHE_HANDLER::Initialize(
|
||||
DWORD cbSize,
|
||||
LONG nThreshold
|
||||
)
|
||||
{
|
||||
HRESULT hr = S_OK;
|
||||
|
||||
m_nThreshold = nThreshold;
|
||||
if ( m_nThreshold > 0xffff)
|
||||
{
|
||||
//
|
||||
// This will be compared against QueryDepthSList return value (USHORT).
|
||||
//
|
||||
m_nThreshold = 0xffff;
|
||||
}
|
||||
|
||||
if ( IsPageheapEnabled() )
|
||||
{
|
||||
//
|
||||
// Disable acache.
|
||||
//
|
||||
m_nThreshold = 0;
|
||||
}
|
||||
|
||||
//
|
||||
// Make sure the block is big enough to hold a FREE_LIST_HEADER.
|
||||
//
|
||||
m_cbSize = cbSize;
|
||||
m_cbSize = max(m_cbSize, sizeof(FREE_LIST_HEADER));
|
||||
|
||||
//
|
||||
// Round up the block size to a multiple of the size of a LONG (for
|
||||
// the fill pattern in Free()).
|
||||
//
|
||||
m_cbSize = (m_cbSize + sizeof(LONG) - 1) & ~(sizeof(LONG) - 1);
|
||||
|
||||
#if defined(_MSC_VER) && _MSC_VER >= 1600 // VC10
|
||||
auto Init = [] (SLIST_HEADER* pHead)
|
||||
{
|
||||
InitializeSListHead(pHead);
|
||||
};
|
||||
#else
|
||||
class Functor
|
||||
{
|
||||
public:
|
||||
void operator()(SLIST_HEADER* pHead)
|
||||
{
|
||||
InitializeSListHead(pHead);
|
||||
}
|
||||
} Init;
|
||||
#endif
|
||||
|
||||
hr = PER_CPU<SLIST_HEADER>::Create(Init,
|
||||
&m_pFreeLists );
|
||||
if (FAILED(hr))
|
||||
{
|
||||
goto Finished;
|
||||
}
|
||||
|
||||
m_nFillPattern = InterlockedIncrement(&sm_nFillPattern);
|
||||
|
||||
Finished:
|
||||
|
||||
return hr;
|
||||
}
|
||||
|
||||
// static
|
||||
HRESULT
|
||||
ALLOC_CACHE_HANDLER::StaticInitialize(
|
||||
VOID
|
||||
)
|
||||
{
|
||||
//
|
||||
// Since the memory allocated is fixed size,
|
||||
// a heap is not really needed, allocations can be done
|
||||
// using VirtualAllocEx[Numa]. For now use Windows Heap.
|
||||
//
|
||||
// Be aware that creating one private heap consumes more
|
||||
// virtual address space for the worker process.
|
||||
//
|
||||
sm_hHeap = GetProcessHeap();
|
||||
return S_OK;
|
||||
}
|
||||
|
||||
|
||||
// static
|
||||
VOID
|
||||
ALLOC_CACHE_HANDLER::StaticTerminate(
|
||||
VOID
|
||||
)
|
||||
{
|
||||
sm_hHeap = NULL;
|
||||
}
|
||||
|
||||
VOID
|
||||
ALLOC_CACHE_HANDLER::CleanupLookaside(
|
||||
VOID
|
||||
)
|
||||
/*++
|
||||
Description:
|
||||
This function cleans up the lookaside list by removing storage space.
|
||||
|
||||
Arguments:
|
||||
None.
|
||||
|
||||
Returns:
|
||||
None
|
||||
--*/
|
||||
{
|
||||
//
|
||||
// Free up all the entries in the list.
|
||||
// Don't use InterlockedFlushSList, in order to work
|
||||
// memory must be 16 bytes aligned and currently it is 64.
|
||||
//
|
||||
|
||||
#if defined(_MSC_VER) && _MSC_VER >= 1600 // VC10
|
||||
auto Predicate = [=] (SLIST_HEADER * pListHeader)
|
||||
{
|
||||
PSLIST_ENTRY pl;
|
||||
LONG NodesToDelete = QueryDepthSList( pListHeader );
|
||||
|
||||
pl = InterlockedPopEntrySList( pListHeader );
|
||||
while ( pl != NULL && --NodesToDelete >= 0 )
|
||||
{
|
||||
InterlockedDecrement( &m_nTotal);
|
||||
|
||||
::HeapFree( sm_hHeap, 0, pl );
|
||||
|
||||
pl = InterlockedPopEntrySList(pListHeader);
|
||||
}
|
||||
};
|
||||
#else
|
||||
class Functor
|
||||
{
|
||||
public:
|
||||
explicit Functor(ALLOC_CACHE_HANDLER * pThis) : _pThis(pThis)
|
||||
{
|
||||
}
|
||||
void operator()(SLIST_HEADER * pListHeader)
|
||||
{
|
||||
PSLIST_ENTRY pl;
|
||||
LONG NodesToDelete = QueryDepthSList( pListHeader );
|
||||
|
||||
pl = InterlockedPopEntrySList( pListHeader );
|
||||
while ( pl != NULL && --NodesToDelete >= 0 )
|
||||
{
|
||||
InterlockedDecrement( &_pThis->m_nTotal);
|
||||
|
||||
::HeapFree( sm_hHeap, 0, pl );
|
||||
|
||||
pl = InterlockedPopEntrySList(pListHeader);
|
||||
}
|
||||
}
|
||||
private:
|
||||
ALLOC_CACHE_HANDLER * _pThis;
|
||||
} Predicate(this);
|
||||
#endif
|
||||
|
||||
m_pFreeLists ->ForEach(Predicate);
|
||||
}
|
||||
|
||||
LPVOID
|
||||
ALLOC_CACHE_HANDLER::Alloc(
|
||||
VOID
|
||||
)
|
||||
{
|
||||
LPVOID pMemory = NULL;
|
||||
|
||||
if ( m_nThreshold > 0 )
|
||||
{
|
||||
SLIST_HEADER * pListHeader = m_pFreeLists ->GetLocal();
|
||||
pMemory = (LPVOID) InterlockedPopEntrySList(pListHeader); // get the real object
|
||||
|
||||
if (pMemory != NULL)
|
||||
{
|
||||
FREE_LIST_HEADER* pfl = (FREE_LIST_HEADER*) pMemory;
|
||||
//
|
||||
// If the signature is wrong then somebody's been scribbling
|
||||
// on memory that they've freed.
|
||||
//
|
||||
DBG_ASSERT(pfl->dwSignature == FREE_LIST_HEADER::FREE_SIGNATURE);
|
||||
}
|
||||
}
|
||||
|
||||
if ( pMemory == NULL )
|
||||
{
|
||||
//
|
||||
// No free entry. Need to alloc a new object.
|
||||
//
|
||||
pMemory = (LPVOID) ::HeapAlloc( sm_hHeap,
|
||||
0,
|
||||
m_cbSize );
|
||||
|
||||
if ( pMemory != NULL )
|
||||
{
|
||||
//
|
||||
// Update counters.
|
||||
//
|
||||
m_nTotal++;
|
||||
}
|
||||
}
|
||||
|
||||
if ( pMemory == NULL )
|
||||
{
|
||||
SetLastError( ERROR_NOT_ENOUGH_MEMORY );
|
||||
}
|
||||
else
|
||||
{
|
||||
FREE_LIST_HEADER* pfl = (FREE_LIST_HEADER*) pMemory;
|
||||
pfl->dwSignature = 0; // clear; just in case caller never overwrites
|
||||
}
|
||||
|
||||
return pMemory;
|
||||
}
|
||||
|
||||
VOID
|
||||
ALLOC_CACHE_HANDLER::Free(
|
||||
__in LPVOID pMemory
|
||||
)
|
||||
{
|
||||
//
|
||||
// Assume that this is allocated using the Alloc() function.
|
||||
//
|
||||
DBG_ASSERT(NULL != pMemory);
|
||||
|
||||
//
|
||||
// Use a signature to check against double deletions.
|
||||
//
|
||||
FREE_LIST_HEADER* pfl = (FREE_LIST_HEADER*) pMemory;
|
||||
DBG_ASSERT(pfl->dwSignature != FREE_LIST_HEADER::FREE_SIGNATURE);
|
||||
|
||||
//
|
||||
// Start filling the space beyond the portion overlaid by the initial
|
||||
// FREE_LIST_HEADER. Fill at most 6 DWORDS.
|
||||
//
|
||||
LONG* pl = (LONG*) (pfl+1);
|
||||
|
||||
for (LONG cb = (LONG)min(6 * sizeof(LONG),m_cbSize) - sizeof(FREE_LIST_HEADER);
|
||||
cb > 0;
|
||||
cb -= sizeof(LONG))
|
||||
{
|
||||
*pl++ = m_nFillPattern;
|
||||
}
|
||||
|
||||
//
|
||||
// Now, set the signature.
|
||||
//
|
||||
pfl->dwSignature = FREE_LIST_HEADER::FREE_SIGNATURE;
|
||||
|
||||
//
|
||||
// Store the items in the alloc cache.
|
||||
//
|
||||
SLIST_HEADER * pListHeader = m_pFreeLists ->GetLocal();
|
||||
|
||||
if ( QueryDepthSList(pListHeader) >= m_nThreshold )
|
||||
{
|
||||
//
|
||||
// Threshold for free entries is exceeded. Free the object to
|
||||
// process pool.
|
||||
//
|
||||
::HeapFree( sm_hHeap, 0, pMemory );
|
||||
}
|
||||
else
|
||||
{
|
||||
//
|
||||
// Store the given pointer in the single linear list
|
||||
//
|
||||
InterlockedPushEntrySList(pListHeader, &pfl->ListEntry);
|
||||
}
|
||||
}
|
||||
|
||||
DWORD
|
||||
ALLOC_CACHE_HANDLER::QueryDepthForAllSLists(
|
||||
VOID
|
||||
)
|
||||
/*++
|
||||
|
||||
Description:
|
||||
|
||||
Aggregates the total count of elements in all lists.
|
||||
|
||||
Arguments:
|
||||
|
||||
None.
|
||||
|
||||
Return Value:
|
||||
|
||||
Total count (snapshot).
|
||||
|
||||
--*/
|
||||
{
|
||||
DWORD Count = 0;
|
||||
|
||||
if (m_pFreeLists != NULL)
|
||||
{
|
||||
#if defined(_MSC_VER) && _MSC_VER >= 1600 // VC10
|
||||
auto Predicate = [&Count] (SLIST_HEADER * pListHeader)
|
||||
{
|
||||
Count += QueryDepthSList(pListHeader);
|
||||
};
|
||||
#else
|
||||
class Functor
|
||||
{
|
||||
public:
|
||||
explicit Functor(DWORD& Count) : _Count(Count)
|
||||
{
|
||||
}
|
||||
void operator()(SLIST_HEADER * pListHeader)
|
||||
{
|
||||
_Count += QueryDepthSList(pListHeader);
|
||||
}
|
||||
private:
|
||||
DWORD& _Count;
|
||||
} Predicate(Count);
|
||||
#endif
|
||||
//
|
||||
// [&Count] means that the method can modify local variable Count.
|
||||
//
|
||||
m_pFreeLists ->ForEach(Predicate);
|
||||
}
|
||||
|
||||
return Count;
|
||||
}
|
||||
|
||||
// static
|
||||
BOOL
|
||||
ALLOC_CACHE_HANDLER::IsPageheapEnabled(
|
||||
VOID
|
||||
)
|
||||
{
|
||||
BOOL fRet = FALSE;
|
||||
BOOL fLockedHeap = FALSE;
|
||||
HMODULE hModule = NULL;
|
||||
HANDLE hHeap = NULL;
|
||||
PROCESS_HEAP_ENTRY heapEntry = {0};
|
||||
|
||||
//
|
||||
// If verifier.dll is loaded - we are running under app verifier == pageheap is enabled
|
||||
//
|
||||
hModule = GetModuleHandle( L"verifier.dll" );
|
||||
if ( hModule != NULL )
|
||||
{
|
||||
hModule = NULL;
|
||||
fRet = TRUE;
|
||||
goto Finished;
|
||||
}
|
||||
|
||||
//
|
||||
// Create a heap for calling heapwalk
|
||||
// otherwise HeapWalk turns off lookasides for a useful heap
|
||||
//
|
||||
hHeap = ::HeapCreate( 0, 0, 0 );
|
||||
if ( hHeap == NULL )
|
||||
{
|
||||
fRet = FALSE;
|
||||
goto Finished;
|
||||
}
|
||||
|
||||
fRet = ::HeapLock( hHeap );
|
||||
if ( !fRet )
|
||||
{
|
||||
goto Finished;
|
||||
}
|
||||
fLockedHeap = TRUE;
|
||||
|
||||
//
|
||||
// If HeapWalk is unsupported -> then running page heap
|
||||
//
|
||||
fRet = ::HeapWalk( hHeap, &heapEntry );
|
||||
if ( !fRet )
|
||||
{
|
||||
if ( GetLastError() == ERROR_INVALID_FUNCTION )
|
||||
{
|
||||
fRet = TRUE;
|
||||
goto Finished;
|
||||
}
|
||||
}
|
||||
|
||||
fRet = FALSE;
|
||||
|
||||
Finished:
|
||||
|
||||
if ( fLockedHeap )
|
||||
{
|
||||
fLockedHeap = FALSE;
|
||||
DBG_REQUIRE( ::HeapUnlock( hHeap ) );
|
||||
}
|
||||
|
||||
if ( hHeap )
|
||||
{
|
||||
DBG_REQUIRE( ::HeapDestroy( hHeap ) );
|
||||
hHeap = NULL;
|
||||
}
|
||||
|
||||
return fRet;
|
||||
}
|
||||
|
|
@ -0,0 +1,115 @@
|
|||
// Copyright (c) .NET Foundation. All rights reserved.
|
||||
// Licensed under the MIT License. See License.txt in the project root for license information.
|
||||
|
||||
#pragma once
|
||||
|
||||
#include "percpu.h"
|
||||
|
||||
class ALLOC_CACHE_HANDLER
|
||||
{
|
||||
public:
|
||||
|
||||
ALLOC_CACHE_HANDLER(
|
||||
VOID
|
||||
);
|
||||
|
||||
~ALLOC_CACHE_HANDLER(
|
||||
VOID
|
||||
);
|
||||
|
||||
HRESULT
|
||||
Initialize(
|
||||
DWORD cbSize,
|
||||
LONG nThreshold
|
||||
);
|
||||
|
||||
LPVOID
|
||||
Alloc(
|
||||
VOID
|
||||
);
|
||||
|
||||
VOID
|
||||
Free(
|
||||
__in LPVOID pMemory
|
||||
);
|
||||
|
||||
|
||||
private:
|
||||
|
||||
VOID
|
||||
CleanupLookaside(
|
||||
VOID
|
||||
);
|
||||
|
||||
DWORD
|
||||
QueryDepthForAllSLists(
|
||||
VOID
|
||||
);
|
||||
|
||||
LONG m_nThreshold;
|
||||
DWORD m_cbSize;
|
||||
|
||||
PER_CPU<SLIST_HEADER> * m_pFreeLists;
|
||||
|
||||
//
|
||||
// Total heap allocations done over the lifetime.
|
||||
// Note that this is not interlocked, it is just a hint for debugging.
|
||||
//
|
||||
volatile LONG m_nTotal;
|
||||
|
||||
LONG m_nFillPattern;
|
||||
|
||||
public:
|
||||
|
||||
static
|
||||
HRESULT
|
||||
StaticInitialize(
|
||||
VOID
|
||||
);
|
||||
|
||||
static
|
||||
VOID
|
||||
StaticTerminate(
|
||||
VOID
|
||||
);
|
||||
|
||||
static
|
||||
BOOL
|
||||
IsPageheapEnabled();
|
||||
|
||||
private:
|
||||
|
||||
static LONG sm_nFillPattern;
|
||||
static HANDLE sm_hHeap;
|
||||
};
|
||||
|
||||
|
||||
// You can use ALLOC_CACHE_HANDLER as a per-class allocator
|
||||
// in your C++ classes. Add the following to your class definition:
|
||||
//
|
||||
// protected:
|
||||
// static ALLOC_CACHE_HANDLER* sm_palloc;
|
||||
// public:
|
||||
// static void* operator new(size_t s)
|
||||
// {
|
||||
// IRTLASSERT(s == sizeof(C));
|
||||
// IRTLASSERT(sm_palloc != NULL);
|
||||
// return sm_palloc->Alloc();
|
||||
// }
|
||||
// static void operator delete(void* pv)
|
||||
// {
|
||||
// IRTLASSERT(pv != NULL);
|
||||
// if (sm_palloc != NULL)
|
||||
// sm_palloc->Free(pv);
|
||||
// }
|
||||
//
|
||||
// Obviously, you must initialize sm_palloc before you can allocate
|
||||
// any objects of this class.
|
||||
//
|
||||
// Note that if you derive a class from this base class, the derived class
|
||||
// must also provide its own operator new and operator delete. If not, the
|
||||
// base class's allocator will be called, but the size of the derived
|
||||
// object will almost certainly be larger than that of the base object.
|
||||
// Furthermore, the allocator will not be used for arrays of objects
|
||||
// (override operator new[] and operator delete[]), but this is a
|
||||
// harder problem since the allocator works with one fixed size.
|
||||
File diff suppressed because it is too large
Load Diff
|
|
@ -0,0 +1,258 @@
|
|||
// Copyright (c) .NET Foundation. All rights reserved.
|
||||
// Licensed under the MIT License. See License.txt in the project root for license information.
|
||||
|
||||
#pragma once
|
||||
#include<Windows.h>
|
||||
|
||||
HRESULT
|
||||
SetElementProperty(
|
||||
IN IAppHostElement * pElement,
|
||||
IN CONST WCHAR * szPropName,
|
||||
IN CONST VARIANT * varPropValue
|
||||
);
|
||||
|
||||
HRESULT
|
||||
SetElementStringProperty(
|
||||
IN IAppHostElement * pElement,
|
||||
IN CONST WCHAR * szPropName,
|
||||
IN CONST WCHAR * szPropValue
|
||||
);
|
||||
|
||||
HRESULT
|
||||
GetElementStringProperty(
|
||||
IN IAppHostElement * pElement,
|
||||
IN CONST WCHAR * szPropName,
|
||||
OUT BSTR * pbstrPropValue
|
||||
);
|
||||
|
||||
HRESULT
|
||||
GetElementStringProperty(
|
||||
IN IAppHostElement * pElement,
|
||||
IN CONST WCHAR * szPropName,
|
||||
OUT STRU * pstrPropValue
|
||||
);
|
||||
|
||||
HRESULT
|
||||
GetElementBoolProperty(
|
||||
IN IAppHostElement * pElement,
|
||||
IN LPCWSTR pszPropertyName,
|
||||
OUT BOOL * pBool
|
||||
);
|
||||
|
||||
HRESULT
|
||||
GetElementBoolProperty(
|
||||
IN IAppHostElement * pElement,
|
||||
IN LPCWSTR pszPropertyName,
|
||||
OUT bool * pBool
|
||||
);
|
||||
|
||||
HRESULT
|
||||
GetElementChildByName(
|
||||
IN IAppHostElement * pElement,
|
||||
IN LPCWSTR pszElementName,
|
||||
OUT IAppHostElement ** ppChildElement
|
||||
);
|
||||
|
||||
HRESULT
|
||||
GetElementDWORDProperty(
|
||||
IN IAppHostElement * pElement,
|
||||
IN LPCWSTR pszPropertyName,
|
||||
OUT DWORD * pdwValue
|
||||
);
|
||||
|
||||
HRESULT
|
||||
GetElementLONGLONGProperty(
|
||||
IN IAppHostElement * pElement,
|
||||
IN LPCWSTR pszPropertyName,
|
||||
OUT LONGLONG * pllValue
|
||||
);
|
||||
|
||||
|
||||
HRESULT
|
||||
GetElementRawTimeSpanProperty(
|
||||
IN IAppHostElement * pElement,
|
||||
IN LPCWSTR pszPropertyName,
|
||||
OUT ULONGLONG * pulonglong
|
||||
);
|
||||
|
||||
#define FIND_ELEMENT_CASE_SENSITIVE 0x00000000
|
||||
#define FIND_ELEMENT_CASE_INSENSITIVE 0x00000001
|
||||
|
||||
HRESULT
|
||||
DeleteElementFromCollection(
|
||||
IAppHostElementCollection *pCollection,
|
||||
CONST WCHAR * szKeyName,
|
||||
CONST WCHAR * szKeyValue,
|
||||
ULONG BehaviorFlags,
|
||||
BOOL * pfDeleted
|
||||
);
|
||||
|
||||
HRESULT
|
||||
DeleteAllElementsFromCollection(
|
||||
IAppHostElementCollection *pCollection,
|
||||
CONST WCHAR * szKeyName,
|
||||
CONST WCHAR * szKeyValue,
|
||||
ULONG BehaviorFlags,
|
||||
UINT * pNumDeleted
|
||||
);
|
||||
|
||||
HRESULT
|
||||
FindElementInCollection(
|
||||
IAppHostElementCollection *pCollection,
|
||||
CONST WCHAR * szKeyName,
|
||||
CONST WCHAR * szKeyValue,
|
||||
ULONG BehaviorFlags,
|
||||
OUT ULONG * pIndex
|
||||
);
|
||||
|
||||
HRESULT
|
||||
VariantAssign(
|
||||
IN OUT VARIANT * pv,
|
||||
IN CONST WCHAR * sz
|
||||
);
|
||||
|
||||
HRESULT
|
||||
GetLocationFromFile(
|
||||
IN IAppHostAdminManager * pAdminMgr,
|
||||
IN CONST WCHAR * szConfigPath,
|
||||
IN CONST WCHAR * szLocationPath,
|
||||
OUT IAppHostConfigLocation ** ppLocation,
|
||||
OUT BOOL * pFound
|
||||
);
|
||||
|
||||
HRESULT
|
||||
GetSectionFromLocation(
|
||||
IN IAppHostConfigLocation * pLocation,
|
||||
IN CONST WCHAR * szSectionName,
|
||||
OUT IAppHostElement ** ppSectionElement,
|
||||
OUT BOOL * pFound
|
||||
);
|
||||
|
||||
HRESULT
|
||||
GetAdminElement(
|
||||
IN IAppHostAdminManager * pAdminMgr,
|
||||
IN CONST WCHAR * szConfigPath,
|
||||
IN CONST WCHAR * szElementName,
|
||||
OUT IAppHostElement ** pElement
|
||||
);
|
||||
|
||||
HRESULT
|
||||
ClearAdminElement(
|
||||
IN IAppHostAdminManager * pAdminMgr,
|
||||
IN CONST WCHAR * szConfigPath,
|
||||
IN CONST WCHAR * szElementName
|
||||
);
|
||||
|
||||
HRESULT
|
||||
ClearElementFromAllSites(
|
||||
IN IAppHostAdminManager * pAdminMgr,
|
||||
IN CONST WCHAR * szConfigPath,
|
||||
IN CONST WCHAR * szElementName
|
||||
);
|
||||
|
||||
HRESULT
|
||||
ClearElementFromAllLocations(
|
||||
IN IAppHostAdminManager * pAdminMgr,
|
||||
IN CONST WCHAR * szConfigPath,
|
||||
IN CONST WCHAR * szElementName
|
||||
);
|
||||
|
||||
HRESULT
|
||||
ClearLocationElements(
|
||||
IN IAppHostConfigLocation * pLocation,
|
||||
IN CONST WCHAR * szElementName
|
||||
);
|
||||
|
||||
HRESULT
|
||||
CompareElementName(
|
||||
IN IAppHostElement * pElement,
|
||||
IN CONST WCHAR * szNameToMatch,
|
||||
OUT BOOL * pMatched
|
||||
);
|
||||
|
||||
HRESULT
|
||||
ClearChildElementsByName(
|
||||
IN IAppHostChildElementCollection * pCollection,
|
||||
IN CONST WCHAR * szElementName,
|
||||
OUT BOOL * pFound
|
||||
);
|
||||
|
||||
HRESULT
|
||||
GetSitesCollection(
|
||||
IN IAppHostAdminManager * pAdminMgr,
|
||||
IN CONST WCHAR * szConfigPath,
|
||||
OUT IAppHostElementCollection ** pSitesCollection
|
||||
);
|
||||
|
||||
HRESULT
|
||||
GetLocationCollection(
|
||||
IN IAppHostAdminManager * pAdminMgr,
|
||||
IN CONST WCHAR * szConfigPath,
|
||||
OUT IAppHostConfigLocationCollection ** pLocationCollection
|
||||
);
|
||||
|
||||
struct ENUM_INDEX
|
||||
{
|
||||
VARIANT Index;
|
||||
ULONG Count;
|
||||
};
|
||||
|
||||
HRESULT
|
||||
FindFirstElement(
|
||||
IN IAppHostElementCollection * pCollection,
|
||||
OUT ENUM_INDEX * pIndex,
|
||||
OUT IAppHostElement ** pElement
|
||||
);
|
||||
|
||||
HRESULT
|
||||
FindNextElement(
|
||||
IN IAppHostElementCollection * pCollection,
|
||||
IN OUT ENUM_INDEX * pIndex,
|
||||
OUT IAppHostElement ** pElement
|
||||
);
|
||||
|
||||
HRESULT
|
||||
FindFirstChildElement(
|
||||
IN IAppHostChildElementCollection * pCollection,
|
||||
OUT ENUM_INDEX * pIndex,
|
||||
OUT IAppHostElement ** pElement
|
||||
);
|
||||
|
||||
HRESULT
|
||||
FindNextChildElement(
|
||||
IN IAppHostChildElementCollection * pCollection,
|
||||
IN OUT ENUM_INDEX * pIndex,
|
||||
OUT IAppHostElement ** pElement
|
||||
);
|
||||
|
||||
HRESULT
|
||||
FindFirstLocation(
|
||||
IN IAppHostConfigLocationCollection * pCollection,
|
||||
OUT ENUM_INDEX * pIndex,
|
||||
OUT IAppHostConfigLocation ** pLocation
|
||||
);
|
||||
|
||||
HRESULT
|
||||
FindNextLocation(
|
||||
IN IAppHostConfigLocationCollection * pCollection,
|
||||
IN OUT ENUM_INDEX * pIndex,
|
||||
OUT IAppHostConfigLocation ** pLocation
|
||||
);
|
||||
|
||||
HRESULT
|
||||
FindFirstLocationElement(
|
||||
IN IAppHostConfigLocation * pLocation,
|
||||
OUT ENUM_INDEX * pIndex,
|
||||
OUT IAppHostElement ** pElement
|
||||
);
|
||||
|
||||
HRESULT
|
||||
FindNextLocationElement(
|
||||
IN IAppHostConfigLocation * pLocation,
|
||||
IN OUT ENUM_INDEX * pIndex,
|
||||
OUT IAppHostElement ** pElement
|
||||
);
|
||||
|
||||
HRESULT GetSharedConfigEnabled(
|
||||
BOOL * pfIsSharedConfig
|
||||
);
|
||||
|
|
@ -0,0 +1,482 @@
|
|||
// Copyright (c) .NET Foundation. All rights reserved.
|
||||
// Licensed under the MIT License. See License.txt in the project root for license information.
|
||||
|
||||
#include "precomp.h"
|
||||
|
||||
DWORD
|
||||
Base64Encode(
|
||||
__in_bcount(cbDecodedBufferSize) VOID * pDecodedBuffer,
|
||||
IN DWORD cbDecodedBufferSize,
|
||||
__out_ecount_opt(cchEncodedStringSize) PWSTR pszEncodedString,
|
||||
IN DWORD cchEncodedStringSize,
|
||||
__out_opt DWORD * pcchEncoded
|
||||
)
|
||||
/*++
|
||||
|
||||
Routine Description:
|
||||
|
||||
Decode a base64-encoded string.
|
||||
|
||||
Arguments:
|
||||
|
||||
pDecodedBuffer (IN) - buffer to encode.
|
||||
cbDecodedBufferSize (IN) - size of buffer to encode.
|
||||
cchEncodedStringSize (IN) - size of the buffer for the encoded string.
|
||||
pszEncodedString (OUT) = the encoded string.
|
||||
pcchEncoded (OUT) - size in characters of the encoded string.
|
||||
|
||||
Return Values:
|
||||
|
||||
0 - success.
|
||||
E_OUTOFMEMORY
|
||||
|
||||
--*/
|
||||
{
|
||||
static WCHAR rgchEncodeTable[64] = {
|
||||
'A', 'B', 'C', 'D', 'E', 'F', 'G', 'H', 'I', 'J', 'K', 'L', 'M',
|
||||
'N', 'O', 'P', 'Q', 'R', 'S', 'T', 'U', 'V', 'W', 'X', 'Y', 'Z',
|
||||
'a', 'b', 'c', 'd', 'e', 'f', 'g', 'h', 'i', 'j', 'k', 'l', 'm',
|
||||
'n', 'o', 'p', 'q', 'r', 's', 't', 'u', 'v', 'w', 'x', 'y', 'z',
|
||||
'0', '1', '2', '3', '4', '5', '6', '7', '8', '9', '+', '/'
|
||||
};
|
||||
|
||||
DWORD ib;
|
||||
DWORD ich;
|
||||
DWORD cchEncoded;
|
||||
BYTE b0, b1, b2;
|
||||
BYTE * pbDecodedBuffer = (BYTE *) pDecodedBuffer;
|
||||
|
||||
// Calculate encoded string size.
|
||||
cchEncoded = 1 + (cbDecodedBufferSize + 2) / 3 * 4;
|
||||
|
||||
if (NULL != pcchEncoded) {
|
||||
*pcchEncoded = cchEncoded;
|
||||
}
|
||||
|
||||
if (cchEncodedStringSize == 0 && pszEncodedString == NULL) {
|
||||
return ERROR_SUCCESS;
|
||||
}
|
||||
|
||||
if (cchEncodedStringSize < cchEncoded) {
|
||||
// Given buffer is too small to hold encoded string.
|
||||
return ERROR_INSUFFICIENT_BUFFER;
|
||||
}
|
||||
|
||||
// Encode data byte triplets into four-byte clusters.
|
||||
ib = ich = 0;
|
||||
while (ib < cbDecodedBufferSize) {
|
||||
b0 = pbDecodedBuffer[ib++];
|
||||
b1 = (ib < cbDecodedBufferSize) ? pbDecodedBuffer[ib++] : 0;
|
||||
b2 = (ib < cbDecodedBufferSize) ? pbDecodedBuffer[ib++] : 0;
|
||||
|
||||
//
|
||||
// The checks below for buffer overflow seems redundant to me.
|
||||
// But it's the only way I can find to keep OACR quiet so it
|
||||
// will have to do.
|
||||
//
|
||||
|
||||
pszEncodedString[ich++] = rgchEncodeTable[b0 >> 2];
|
||||
if ( ich >= cchEncodedStringSize )
|
||||
{
|
||||
DBG_ASSERT( FALSE );
|
||||
return ERROR_BUFFER_OVERFLOW;
|
||||
}
|
||||
|
||||
pszEncodedString[ich++] = rgchEncodeTable[((b0 << 4) & 0x30) | ((b1 >> 4) & 0x0f)];
|
||||
if ( ich >= cchEncodedStringSize )
|
||||
{
|
||||
DBG_ASSERT( FALSE );
|
||||
return ERROR_BUFFER_OVERFLOW;
|
||||
}
|
||||
|
||||
pszEncodedString[ich++] = rgchEncodeTable[((b1 << 2) & 0x3c) | ((b2 >> 6) & 0x03)];
|
||||
if ( ich >= cchEncodedStringSize )
|
||||
{
|
||||
DBG_ASSERT( FALSE );
|
||||
return ERROR_BUFFER_OVERFLOW;
|
||||
}
|
||||
|
||||
pszEncodedString[ich++] = rgchEncodeTable[b2 & 0x3f];
|
||||
if ( ich >= cchEncodedStringSize )
|
||||
{
|
||||
DBG_ASSERT( FALSE );
|
||||
return ERROR_BUFFER_OVERFLOW;
|
||||
}
|
||||
}
|
||||
|
||||
// Pad the last cluster as necessary to indicate the number of data bytes
|
||||
// it represents.
|
||||
switch (cbDecodedBufferSize % 3) {
|
||||
case 0:
|
||||
break;
|
||||
case 1:
|
||||
pszEncodedString[ich - 2] = '=';
|
||||
__fallthrough;
|
||||
case 2:
|
||||
pszEncodedString[ich - 1] = '=';
|
||||
break;
|
||||
}
|
||||
|
||||
// Null-terminate the encoded string.
|
||||
pszEncodedString[ich++] = '\0';
|
||||
|
||||
DBG_ASSERT(ich == cchEncoded);
|
||||
|
||||
return ERROR_SUCCESS;
|
||||
}
|
||||
|
||||
|
||||
DWORD
|
||||
Base64Decode(
|
||||
__in PCWSTR pszEncodedString,
|
||||
__out_opt VOID * pDecodeBuffer,
|
||||
__in DWORD cbDecodeBufferSize,
|
||||
__out_opt DWORD * pcbDecoded
|
||||
)
|
||||
/*++
|
||||
|
||||
Routine Description:
|
||||
|
||||
Decode a base64-encoded string.
|
||||
|
||||
Arguments:
|
||||
|
||||
pszEncodedString (IN) - base64-encoded string to decode.
|
||||
cbDecodeBufferSize (IN) - size in bytes of the decode buffer.
|
||||
pbDecodeBuffer (OUT) - holds the decoded data.
|
||||
pcbDecoded (OUT) - number of data bytes in the decoded data (if success or
|
||||
STATUS_BUFFER_TOO_SMALL).
|
||||
|
||||
Return Values:
|
||||
|
||||
0 - success.
|
||||
E_OUTOFMEMORY
|
||||
E_INVALIDARG
|
||||
|
||||
--*/
|
||||
{
|
||||
#define NA (255)
|
||||
#define DECODE(x) (((ULONG)(x) < sizeof(rgbDecodeTable)) ? rgbDecodeTable[x] : NA)
|
||||
|
||||
static BYTE rgbDecodeTable[128] = {
|
||||
NA, NA, NA, NA, NA, NA, NA, NA, NA, NA, NA, NA, NA, NA, NA, NA, // 0-15
|
||||
NA, NA, NA, NA, NA, NA, NA, NA, NA, NA, NA, NA, NA, NA, NA, NA, // 16-31
|
||||
NA, NA, NA, NA, NA, NA, NA, NA, NA, NA, NA, 62, NA, NA, NA, 63, // 32-47
|
||||
52, 53, 54, 55, 56, 57, 58, 59, 60, 61, NA, NA, NA, 0, NA, NA, // 48-63
|
||||
NA, 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, // 64-79
|
||||
15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, NA, NA, NA, NA, NA, // 80-95
|
||||
NA, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 36, 37, 38, 39, 40, // 96-111
|
||||
41, 42, 43, 44, 45, 46, 47, 48, 49, 50, 51, NA, NA, NA, NA, NA, // 112-127
|
||||
};
|
||||
|
||||
DWORD cbDecoded;
|
||||
DWORD cchEncodedSize;
|
||||
DWORD ich;
|
||||
DWORD ib;
|
||||
BYTE b0, b1, b2, b3;
|
||||
BYTE * pbDecodeBuffer = (BYTE *) pDecodeBuffer;
|
||||
|
||||
cchEncodedSize = (DWORD)wcslen(pszEncodedString);
|
||||
if (NULL != pcbDecoded) {
|
||||
*pcbDecoded = 0;
|
||||
}
|
||||
|
||||
if ((0 == cchEncodedSize) || (0 != (cchEncodedSize % 4))) {
|
||||
// Input string is not sized correctly to be base64.
|
||||
return ERROR_INVALID_PARAMETER;
|
||||
}
|
||||
|
||||
// Calculate decoded buffer size.
|
||||
cbDecoded = (cchEncodedSize + 3) / 4 * 3;
|
||||
if (pszEncodedString[cchEncodedSize-1] == '=') {
|
||||
if (pszEncodedString[cchEncodedSize-2] == '=') {
|
||||
// Only one data byte is encoded in the last cluster.
|
||||
cbDecoded -= 2;
|
||||
}
|
||||
else {
|
||||
// Only two data bytes are encoded in the last cluster.
|
||||
cbDecoded -= 1;
|
||||
}
|
||||
}
|
||||
|
||||
if (NULL != pcbDecoded) {
|
||||
*pcbDecoded = cbDecoded;
|
||||
}
|
||||
|
||||
if (cbDecodeBufferSize == 0 && pDecodeBuffer == NULL) {
|
||||
return ERROR_SUCCESS;
|
||||
}
|
||||
|
||||
if (cbDecoded > cbDecodeBufferSize) {
|
||||
// Supplied buffer is too small.
|
||||
return ERROR_INSUFFICIENT_BUFFER;
|
||||
}
|
||||
|
||||
// Decode each four-byte cluster into the corresponding three data bytes.
|
||||
ich = ib = 0;
|
||||
while (ich < cchEncodedSize) {
|
||||
b0 = DECODE(pszEncodedString[ich]); ich++;
|
||||
b1 = DECODE(pszEncodedString[ich]); ich++;
|
||||
b2 = DECODE(pszEncodedString[ich]); ich++;
|
||||
b3 = DECODE(pszEncodedString[ich]); ich++;
|
||||
|
||||
if ((NA == b0) || (NA == b1) || (NA == b2) || (NA == b3)) {
|
||||
// Contents of input string are not base64.
|
||||
return ERROR_INVALID_PARAMETER;
|
||||
}
|
||||
|
||||
pbDecodeBuffer[ib++] = (b0 << 2) | (b1 >> 4);
|
||||
|
||||
if (ib < cbDecoded) {
|
||||
pbDecodeBuffer[ib++] = (b1 << 4) | (b2 >> 2);
|
||||
|
||||
if (ib < cbDecoded) {
|
||||
pbDecodeBuffer[ib++] = (b2 << 6) | b3;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
DBG_ASSERT(ib == cbDecoded);
|
||||
|
||||
return ERROR_SUCCESS;
|
||||
}
|
||||
|
||||
|
||||
DWORD
|
||||
Base64Encode(
|
||||
__in_bcount(cbDecodedBufferSize) VOID * pDecodedBuffer,
|
||||
IN DWORD cbDecodedBufferSize,
|
||||
__out_ecount_opt(cchEncodedStringSize) PSTR pszEncodedString,
|
||||
IN DWORD cchEncodedStringSize,
|
||||
__out_opt DWORD * pcchEncoded
|
||||
)
|
||||
/*++
|
||||
|
||||
Routine Description:
|
||||
|
||||
Decode a base64-encoded string.
|
||||
|
||||
Arguments:
|
||||
|
||||
pDecodedBuffer (IN) - buffer to encode.
|
||||
cbDecodedBufferSize (IN) - size of buffer to encode.
|
||||
cchEncodedStringSize (IN) - size of the buffer for the encoded string.
|
||||
pszEncodedString (OUT) = the encoded string.
|
||||
pcchEncoded (OUT) - size in characters of the encoded string.
|
||||
|
||||
Return Values:
|
||||
|
||||
0 - success.
|
||||
E_OUTOFMEMORY
|
||||
|
||||
--*/
|
||||
{
|
||||
static CHAR rgchEncodeTable[64] = {
|
||||
'A', 'B', 'C', 'D', 'E', 'F', 'G', 'H', 'I', 'J', 'K', 'L', 'M',
|
||||
'N', 'O', 'P', 'Q', 'R', 'S', 'T', 'U', 'V', 'W', 'X', 'Y', 'Z',
|
||||
'a', 'b', 'c', 'd', 'e', 'f', 'g', 'h', 'i', 'j', 'k', 'l', 'm',
|
||||
'n', 'o', 'p', 'q', 'r', 's', 't', 'u', 'v', 'w', 'x', 'y', 'z',
|
||||
'0', '1', '2', '3', '4', '5', '6', '7', '8', '9', '+', '/'
|
||||
};
|
||||
|
||||
DWORD ib;
|
||||
DWORD ich;
|
||||
DWORD cchEncoded;
|
||||
BYTE b0, b1, b2;
|
||||
BYTE * pbDecodedBuffer = (BYTE *) pDecodedBuffer;
|
||||
|
||||
// Calculate encoded string size.
|
||||
cchEncoded = 1 + (cbDecodedBufferSize + 2) / 3 * 4;
|
||||
|
||||
if (NULL != pcchEncoded) {
|
||||
*pcchEncoded = cchEncoded;
|
||||
}
|
||||
|
||||
if (cchEncodedStringSize == 0 && pszEncodedString == NULL) {
|
||||
return ERROR_SUCCESS;
|
||||
}
|
||||
|
||||
if (cchEncodedStringSize < cchEncoded) {
|
||||
// Given buffer is too small to hold encoded string.
|
||||
return ERROR_INSUFFICIENT_BUFFER;
|
||||
}
|
||||
|
||||
// Encode data byte triplets into four-byte clusters.
|
||||
ib = ich = 0;
|
||||
while (ib < cbDecodedBufferSize) {
|
||||
b0 = pbDecodedBuffer[ib++];
|
||||
b1 = (ib < cbDecodedBufferSize) ? pbDecodedBuffer[ib++] : 0;
|
||||
b2 = (ib < cbDecodedBufferSize) ? pbDecodedBuffer[ib++] : 0;
|
||||
|
||||
//
|
||||
// The checks below for buffer overflow seems redundant to me.
|
||||
// But it's the only way I can find to keep OACR quiet so it
|
||||
// will have to do.
|
||||
//
|
||||
|
||||
pszEncodedString[ich++] = rgchEncodeTable[b0 >> 2];
|
||||
if ( ich >= cchEncodedStringSize )
|
||||
{
|
||||
DBG_ASSERT( FALSE );
|
||||
return ERROR_BUFFER_OVERFLOW;
|
||||
}
|
||||
|
||||
pszEncodedString[ich++] = rgchEncodeTable[((b0 << 4) & 0x30) | ((b1 >> 4) & 0x0f)];
|
||||
if ( ich >= cchEncodedStringSize )
|
||||
{
|
||||
DBG_ASSERT( FALSE );
|
||||
return ERROR_BUFFER_OVERFLOW;
|
||||
}
|
||||
|
||||
pszEncodedString[ich++] = rgchEncodeTable[((b1 << 2) & 0x3c) | ((b2 >> 6) & 0x03)];
|
||||
if ( ich >= cchEncodedStringSize )
|
||||
{
|
||||
DBG_ASSERT( FALSE );
|
||||
return ERROR_BUFFER_OVERFLOW;
|
||||
}
|
||||
|
||||
pszEncodedString[ich++] = rgchEncodeTable[b2 & 0x3f];
|
||||
if ( ich >= cchEncodedStringSize )
|
||||
{
|
||||
DBG_ASSERT( FALSE );
|
||||
return ERROR_BUFFER_OVERFLOW;
|
||||
}
|
||||
}
|
||||
|
||||
// Pad the last cluster as necessary to indicate the number of data bytes
|
||||
// it represents.
|
||||
switch (cbDecodedBufferSize % 3) {
|
||||
case 0:
|
||||
break;
|
||||
case 1:
|
||||
pszEncodedString[ich - 2] = '=';
|
||||
__fallthrough;
|
||||
case 2:
|
||||
pszEncodedString[ich - 1] = '=';
|
||||
break;
|
||||
}
|
||||
|
||||
// Null-terminate the encoded string.
|
||||
pszEncodedString[ich++] = '\0';
|
||||
|
||||
DBG_ASSERT(ich == cchEncoded);
|
||||
|
||||
return ERROR_SUCCESS;
|
||||
}
|
||||
|
||||
|
||||
DWORD
|
||||
Base64Decode(
|
||||
__in PCSTR pszEncodedString,
|
||||
__out_opt VOID * pDecodeBuffer,
|
||||
__in DWORD cbDecodeBufferSize,
|
||||
__out_opt DWORD * pcbDecoded
|
||||
)
|
||||
/*++
|
||||
|
||||
Routine Description:
|
||||
|
||||
Decode a base64-encoded string.
|
||||
|
||||
Arguments:
|
||||
|
||||
pszEncodedString (IN) - base64-encoded string to decode.
|
||||
cbDecodeBufferSize (IN) - size in bytes of the decode buffer.
|
||||
pbDecodeBuffer (OUT) - holds the decoded data.
|
||||
pcbDecoded (OUT) - number of data bytes in the decoded data (if success or
|
||||
STATUS_BUFFER_TOO_SMALL).
|
||||
|
||||
Return Values:
|
||||
|
||||
0 - success.
|
||||
E_OUTOFMEMORY
|
||||
E_INVALIDARG
|
||||
|
||||
--*/
|
||||
{
|
||||
#define NA (255)
|
||||
#define DECODE(x) (((ULONG)(x) < sizeof(rgbDecodeTable)) ? rgbDecodeTable[x] : NA)
|
||||
|
||||
static BYTE rgbDecodeTable[128] = {
|
||||
NA, NA, NA, NA, NA, NA, NA, NA, NA, NA, NA, NA, NA, NA, NA, NA, // 0-15
|
||||
NA, NA, NA, NA, NA, NA, NA, NA, NA, NA, NA, NA, NA, NA, NA, NA, // 16-31
|
||||
NA, NA, NA, NA, NA, NA, NA, NA, NA, NA, NA, 62, NA, NA, NA, 63, // 32-47
|
||||
52, 53, 54, 55, 56, 57, 58, 59, 60, 61, NA, NA, NA, 0, NA, NA, // 48-63
|
||||
NA, 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, // 64-79
|
||||
15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, NA, NA, NA, NA, NA, // 80-95
|
||||
NA, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 36, 37, 38, 39, 40, // 96-111
|
||||
41, 42, 43, 44, 45, 46, 47, 48, 49, 50, 51, NA, NA, NA, NA, NA, // 112-127
|
||||
};
|
||||
|
||||
DWORD cbDecoded;
|
||||
DWORD cchEncodedSize;
|
||||
DWORD ich;
|
||||
DWORD ib;
|
||||
BYTE b0, b1, b2, b3;
|
||||
BYTE * pbDecodeBuffer = (BYTE *) pDecodeBuffer;
|
||||
|
||||
cchEncodedSize = (DWORD)strlen(pszEncodedString);
|
||||
if (NULL != pcbDecoded) {
|
||||
*pcbDecoded = 0;
|
||||
}
|
||||
|
||||
if ((0 == cchEncodedSize) || (0 != (cchEncodedSize % 4))) {
|
||||
// Input string is not sized correctly to be base64.
|
||||
return ERROR_INVALID_PARAMETER;
|
||||
}
|
||||
|
||||
// Calculate decoded buffer size.
|
||||
cbDecoded = (cchEncodedSize + 3) / 4 * 3;
|
||||
if (pszEncodedString[cchEncodedSize-1] == '=') {
|
||||
if (pszEncodedString[cchEncodedSize-2] == '=') {
|
||||
// Only one data byte is encoded in the last cluster.
|
||||
cbDecoded -= 2;
|
||||
}
|
||||
else {
|
||||
// Only two data bytes are encoded in the last cluster.
|
||||
cbDecoded -= 1;
|
||||
}
|
||||
}
|
||||
|
||||
if (NULL != pcbDecoded) {
|
||||
*pcbDecoded = cbDecoded;
|
||||
}
|
||||
|
||||
if (cbDecodeBufferSize == 0 && pDecodeBuffer == NULL) {
|
||||
return ERROR_SUCCESS;
|
||||
}
|
||||
|
||||
if (cbDecoded > cbDecodeBufferSize) {
|
||||
// Supplied buffer is too small.
|
||||
return ERROR_INSUFFICIENT_BUFFER;
|
||||
}
|
||||
|
||||
// Decode each four-byte cluster into the corresponding three data bytes.
|
||||
ich = ib = 0;
|
||||
while (ich < cchEncodedSize) {
|
||||
b0 = DECODE(pszEncodedString[ich]); ich++;
|
||||
b1 = DECODE(pszEncodedString[ich]); ich++;
|
||||
b2 = DECODE(pszEncodedString[ich]); ich++;
|
||||
b3 = DECODE(pszEncodedString[ich]); ich++;
|
||||
|
||||
if ((NA == b0) || (NA == b1) || (NA == b2) || (NA == b3)) {
|
||||
// Contents of input string are not base64.
|
||||
return ERROR_INVALID_PARAMETER;
|
||||
}
|
||||
|
||||
pbDecodeBuffer[ib++] = (b0 << 2) | (b1 >> 4);
|
||||
|
||||
if (ib < cbDecoded) {
|
||||
pbDecodeBuffer[ib++] = (b1 << 4) | (b2 >> 2);
|
||||
|
||||
if (ib < cbDecoded) {
|
||||
pbDecodeBuffer[ib++] = (b2 << 6) | b3;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
DBG_ASSERT(ib == cbDecoded);
|
||||
|
||||
return ERROR_SUCCESS;
|
||||
}
|
||||
|
||||
|
|
@ -0,0 +1,42 @@
|
|||
// Copyright (c) .NET Foundation. All rights reserved.
|
||||
// Licensed under the MIT License. See License.txt in the project root for license information.
|
||||
|
||||
#ifndef _BASE64_H_
|
||||
#define _BASE64_H_
|
||||
|
||||
DWORD
|
||||
Base64Encode(
|
||||
__in_bcount( cbDecodedBufferSize ) VOID * pDecodedBuffer,
|
||||
IN DWORD cbDecodedBufferSize,
|
||||
__out_ecount_opt( cchEncodedStringSize ) PWSTR pszEncodedString,
|
||||
IN DWORD cchEncodedStringSize,
|
||||
__out_opt DWORD * pcchEncoded
|
||||
);
|
||||
|
||||
DWORD
|
||||
Base64Decode(
|
||||
__in PCWSTR pszEncodedString,
|
||||
__out_opt VOID * pDecodeBuffer,
|
||||
__in DWORD cbDecodeBufferSize,
|
||||
__out_opt DWORD * pcbDecoded
|
||||
);
|
||||
|
||||
DWORD
|
||||
Base64Encode(
|
||||
__in_bcount( cbDecodedBufferSize ) VOID * pDecodedBuffer,
|
||||
IN DWORD cbDecodedBufferSize,
|
||||
__out_ecount_opt( cchEncodedStringSize ) PSTR pszEncodedString,
|
||||
IN DWORD cchEncodedStringSize,
|
||||
__out_opt DWORD * pcchEncoded
|
||||
);
|
||||
|
||||
DWORD
|
||||
Base64Decode(
|
||||
__in PCSTR pszEncodedString,
|
||||
__out_opt VOID * pDecodeBuffer,
|
||||
__in DWORD cbDecodeBufferSize,
|
||||
__out_opt DWORD * pcbDecoded
|
||||
);
|
||||
|
||||
#endif // _BASE64_HXX_
|
||||
|
||||
|
|
@ -0,0 +1,271 @@
|
|||
// Copyright (c) .NET Foundation. All rights reserved.
|
||||
// Licensed under the MIT License. See License.txt in the project root for license information.
|
||||
|
||||
#pragma once
|
||||
|
||||
#include <crtdbg.h>
|
||||
|
||||
|
||||
//
|
||||
// BUFFER_T class shouldn't be used directly. Use BUFFER specialization class instead.
|
||||
// The only BUFFER_T partners are STRU and STRA classes.
|
||||
// BUFFER_T cannot hold other but primitive types since it doesn't call
|
||||
// constructor and destructor.
|
||||
//
|
||||
// Note: Size is in bytes.
|
||||
//
|
||||
template<typename T, DWORD LENGTH>
|
||||
class BUFFER_T
|
||||
{
|
||||
public:
|
||||
|
||||
BUFFER_T()
|
||||
: m_cbBuffer( sizeof(m_rgBuffer) ),
|
||||
m_fHeapAllocated( false ),
|
||||
m_pBuffer(m_rgBuffer)
|
||||
/*++
|
||||
Description:
|
||||
|
||||
Default constructor where the inline buffer is used.
|
||||
|
||||
Arguments:
|
||||
|
||||
None.
|
||||
|
||||
Returns:
|
||||
|
||||
None.
|
||||
|
||||
--*/
|
||||
{
|
||||
}
|
||||
|
||||
BUFFER_T(
|
||||
__inout_bcount(cbInit) T* pbInit,
|
||||
__in DWORD cbInit
|
||||
) : m_pBuffer( pbInit ),
|
||||
m_cbBuffer( cbInit ),
|
||||
m_fHeapAllocated( false )
|
||||
/*++
|
||||
Description:
|
||||
|
||||
Instantiate BUFFER, initially using pbInit as buffer
|
||||
This is useful for stack-buffers and inline-buffer class members
|
||||
(see STACK_BUFFER and INLINE_BUFFER_INIT below)
|
||||
|
||||
BUFFER does not free pbInit.
|
||||
|
||||
Arguments:
|
||||
|
||||
pbInit - Initial buffer to use.
|
||||
cbInit - Size of pbInit in bytes (not in elements).
|
||||
|
||||
Returns:
|
||||
|
||||
None.
|
||||
|
||||
--*/
|
||||
{
|
||||
_ASSERTE( NULL != pbInit );
|
||||
_ASSERTE( cbInit > 0 );
|
||||
}
|
||||
|
||||
~BUFFER_T()
|
||||
{
|
||||
if( IsHeapAllocated() )
|
||||
{
|
||||
_ASSERTE( NULL != m_pBuffer );
|
||||
HeapFree( GetProcessHeap(), 0, m_pBuffer );
|
||||
m_pBuffer = NULL;
|
||||
m_cbBuffer = 0;
|
||||
m_fHeapAllocated = false;
|
||||
}
|
||||
}
|
||||
|
||||
T*
|
||||
QueryPtr(
|
||||
VOID
|
||||
) const
|
||||
{
|
||||
//
|
||||
// Return pointer to data buffer.
|
||||
//
|
||||
return m_pBuffer;
|
||||
}
|
||||
|
||||
DWORD
|
||||
QuerySize(
|
||||
VOID
|
||||
) const
|
||||
{
|
||||
//
|
||||
// Return number of bytes.
|
||||
//
|
||||
return m_cbBuffer;
|
||||
}
|
||||
|
||||
__success(return == true)
|
||||
bool
|
||||
Resize(
|
||||
const SIZE_T cbNewSize,
|
||||
const bool fZeroMemoryBeyondOldSize = false
|
||||
)
|
||||
/*++
|
||||
Description:
|
||||
|
||||
Resizes the buffer.
|
||||
|
||||
Arguments:
|
||||
|
||||
cbNewSize - Size in bytes to grow to.
|
||||
fZeroMemoryBeyondOldSize
|
||||
- Whether to zero the region of memory of the
|
||||
new buffer beyond the original size.
|
||||
|
||||
Returns:
|
||||
|
||||
TRUE on success, FALSE on failure.
|
||||
|
||||
--*/
|
||||
{
|
||||
PVOID pNewMem;
|
||||
|
||||
if ( cbNewSize <= m_cbBuffer )
|
||||
{
|
||||
return true;
|
||||
}
|
||||
|
||||
if ( cbNewSize > MAXDWORD )
|
||||
{
|
||||
SetLastError( ERROR_INVALID_PARAMETER );
|
||||
return false;
|
||||
}
|
||||
|
||||
DWORD dwHeapAllocFlags = fZeroMemoryBeyondOldSize ? HEAP_ZERO_MEMORY : 0;
|
||||
|
||||
if( IsHeapAllocated() )
|
||||
{
|
||||
pNewMem = HeapReAlloc( GetProcessHeap(), dwHeapAllocFlags, m_pBuffer, cbNewSize );
|
||||
}
|
||||
else
|
||||
{
|
||||
pNewMem = HeapAlloc( GetProcessHeap(), dwHeapAllocFlags, cbNewSize );
|
||||
}
|
||||
|
||||
if( pNewMem == NULL )
|
||||
{
|
||||
SetLastError( ERROR_NOT_ENOUGH_MEMORY );
|
||||
return false;
|
||||
}
|
||||
|
||||
if( !IsHeapAllocated() )
|
||||
{
|
||||
//
|
||||
// First time this block is allocated. Copy over old contents.
|
||||
//
|
||||
memcpy_s( pNewMem, static_cast<DWORD>(cbNewSize), m_pBuffer, m_cbBuffer );
|
||||
m_fHeapAllocated = true;
|
||||
}
|
||||
|
||||
m_pBuffer = reinterpret_cast<T*>(pNewMem);
|
||||
m_cbBuffer = static_cast<DWORD>(cbNewSize);
|
||||
|
||||
_ASSERTE( m_pBuffer != NULL );
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
private:
|
||||
|
||||
bool
|
||||
IsHeapAllocated(
|
||||
VOID
|
||||
) const
|
||||
{
|
||||
return m_fHeapAllocated;
|
||||
}
|
||||
|
||||
//
|
||||
// The default inline buffer.
|
||||
// This member should be at the beginning for alignment purposes.
|
||||
//
|
||||
T m_rgBuffer[LENGTH];
|
||||
|
||||
//
|
||||
// Is m_pBuffer dynamically allocated?
|
||||
//
|
||||
bool m_fHeapAllocated;
|
||||
|
||||
//
|
||||
// Size of the buffer as requested by client in bytes.
|
||||
//
|
||||
DWORD m_cbBuffer;
|
||||
|
||||
//
|
||||
// Pointer to buffer.
|
||||
//
|
||||
__field_bcount_full(m_cbBuffer)
|
||||
T* m_pBuffer;
|
||||
};
|
||||
|
||||
//
|
||||
// Resizes the buffer by 2 if the ideal size is bigger
|
||||
// than the buffer length. That give us lg(n) allocations.
|
||||
//
|
||||
// Use template inferring like:
|
||||
//
|
||||
// BUFFER buff;
|
||||
// hr = ResizeBufferByTwo(buff, 100);
|
||||
//
|
||||
template<typename T, DWORD LENGTH>
|
||||
HRESULT
|
||||
ResizeBufferByTwo(
|
||||
BUFFER_T<T,LENGTH>& Buffer,
|
||||
SIZE_T cbIdealSize,
|
||||
bool fZeroMemoryBeyondOldSize = false
|
||||
)
|
||||
{
|
||||
if (cbIdealSize > Buffer.QuerySize())
|
||||
{
|
||||
if (!Buffer.Resize(max(cbIdealSize, static_cast<SIZE_T>(Buffer.QuerySize() * 2)),
|
||||
fZeroMemoryBeyondOldSize))
|
||||
{
|
||||
return E_OUTOFMEMORY;
|
||||
}
|
||||
}
|
||||
return S_OK;
|
||||
}
|
||||
|
||||
|
||||
//
|
||||
//
|
||||
// Lots of code uses BUFFER class to store a bunch of different
|
||||
// structures, so m_rgBuffer needs to be 8 byte aligned when it is used
|
||||
// as an opaque buffer.
|
||||
//
|
||||
#define INLINED_BUFFER_LEN 32
|
||||
typedef BUFFER_T<BYTE, INLINED_BUFFER_LEN> BUFFER;
|
||||
|
||||
//
|
||||
// Assumption of macros below for pointer alignment purposes
|
||||
//
|
||||
C_ASSERT( sizeof(VOID*) <= sizeof(ULONGLONG) );
|
||||
|
||||
//
|
||||
// Declare a BUFFER that will use stack memory of <size>
|
||||
// bytes. If the buffer overflows then a heap buffer will be allocated.
|
||||
//
|
||||
#define STACK_BUFFER( _name, _size ) \
|
||||
ULONGLONG __aqw##_name[ ( ( (_size) + sizeof(ULONGLONG) - 1 ) / sizeof(ULONGLONG) ) ]; \
|
||||
BUFFER _name( (BYTE*)__aqw##_name, sizeof(__aqw##_name) )
|
||||
|
||||
//
|
||||
// Macros for declaring and initializing a BUFFER that will use inline memory
|
||||
// of <size> bytes as a member of an object.
|
||||
//
|
||||
#define INLINE_BUFFER( _name, _size ) \
|
||||
ULONGLONG __aqw##_name[ ( ( (_size) + sizeof(ULONGLONG) - 1 ) / sizeof(ULONGLONG) ) ]; \
|
||||
BUFFER _name;
|
||||
|
||||
#define INLINE_BUFFER_INIT( _name ) \
|
||||
_name( (BYTE*)__aqw##_name, sizeof( __aqw##_name ) )
|
||||
|
|
@ -0,0 +1,14 @@
|
|||
// Copyright (c) .NET Foundation. All rights reserved.
|
||||
// Licensed under the MIT License. See License.txt in the project root for license information.
|
||||
|
||||
#ifndef _DATETIME_H_
|
||||
#define _DATETIME_H_
|
||||
|
||||
BOOL
|
||||
StringTimeToFileTime(
|
||||
PCSTR pszTime,
|
||||
ULONGLONG * pulTime
|
||||
);
|
||||
|
||||
#endif
|
||||
|
||||
|
|
@ -0,0 +1,102 @@
|
|||
// Copyright (c) .NET Foundation. All rights reserved.
|
||||
// Licensed under the MIT License. See License.txt in the project root for license information.
|
||||
|
||||
#ifndef _DBGUTIL_H_
|
||||
#define _DBGUTIL_H_
|
||||
|
||||
#include <crtdbg.h>
|
||||
|
||||
//
|
||||
// TODO
|
||||
// Using _CrtDbg implementation. If hooking is desired
|
||||
// wrappers should be provided here so that we can reimplement
|
||||
// if neecessary.
|
||||
//
|
||||
// IF_DEBUG/DEBUG FLAGS
|
||||
//
|
||||
// registry configuration
|
||||
//
|
||||
|
||||
//
|
||||
// Debug error levels for DEBUG_FLAGS_VAR.
|
||||
//
|
||||
|
||||
#define DEBUG_FLAG_INFO 0x00000001
|
||||
#define DEBUG_FLAG_WARN 0x00000002
|
||||
#define DEBUG_FLAG_ERROR 0x00000004
|
||||
|
||||
//
|
||||
// Predefined error level values. These are backwards from the
|
||||
// windows definitions.
|
||||
//
|
||||
|
||||
#define DEBUG_FLAGS_INFO (DEBUG_FLAG_ERROR | DEBUG_FLAG_WARN | DEBUG_FLAG_INFO)
|
||||
#define DEBUG_FLAGS_WARN (DEBUG_FLAG_ERROR | DEBUG_FLAG_WARN)
|
||||
#define DEBUG_FLAGS_ERROR (DEBUG_FLAG_ERROR)
|
||||
#define DEBUG_FLAGS_ANY (DEBUG_FLAG_INFO | DEBUG_FLAG_WARN | DEBUG_FLAG_ERROR)
|
||||
|
||||
//
|
||||
// Global variables to control tracing. Generally per module
|
||||
//
|
||||
|
||||
#ifndef DEBUG_FLAGS_VAR
|
||||
#define DEBUG_FLAGS_VAR g_dwDebugFlags
|
||||
#endif
|
||||
|
||||
#ifndef DEBUG_LABEL_VAR
|
||||
#define DEBUG_LABEL_VAR g_szDebugLabel
|
||||
#endif
|
||||
|
||||
extern PCSTR DEBUG_LABEL_VAR;
|
||||
extern DWORD DEBUG_FLAGS_VAR;
|
||||
|
||||
//
|
||||
// Module should make this declaration globally.
|
||||
//
|
||||
|
||||
#define DECLARE_DEBUG_PRINT_OBJECT( _pszLabel_ ) \
|
||||
PCSTR DEBUG_LABEL_VAR = _pszLabel_; \
|
||||
DWORD DEBUG_FLAGS_VAR = DEBUG_FLAGS_ANY; \
|
||||
|
||||
#define DECLARE_DEBUG_PRINT_OBJECT2( _pszLabel_, _dwLevel_ ) \
|
||||
PCSTR DEBUG_LABEL_VAR = _pszLabel_; \
|
||||
DWORD DEBUG_FLAGS_VAR = _dwLevel_; \
|
||||
|
||||
//
|
||||
// This doesn't do anything now. Should be safe to call in dll main.
|
||||
//
|
||||
|
||||
#define CREATE_DEBUG_PRINT_OBJECT
|
||||
|
||||
//
|
||||
// Trace macros
|
||||
//
|
||||
|
||||
#define DBG_CONTEXT _CRT_WARN, __FILE__, __LINE__, DEBUG_LABEL_VAR
|
||||
|
||||
#ifdef DEBUG
|
||||
#define DBGINFO(args) \
|
||||
{if (DEBUG_FLAGS_VAR & DEBUG_FLAG_INFO) { _CrtDbgReport args; }}
|
||||
#define DBGWARN(args) \
|
||||
{if (DEBUG_FLAGS_VAR & DEBUG_FLAG_WARN) { _CrtDbgReport args; }}
|
||||
#define DBGERROR(args) \
|
||||
{if (DEBUG_FLAGS_VAR & DEBUG_FLAG_ERROR) { _CrtDbgReport args; }}
|
||||
#else
|
||||
#define DBGINFO
|
||||
#define DBGWARN
|
||||
#define DBGERROR
|
||||
#endif
|
||||
|
||||
#define DBGPRINTF DBGINFO
|
||||
|
||||
//
|
||||
// Simple error traces
|
||||
//
|
||||
|
||||
#define DBGERROR_HR( _hr_ ) \
|
||||
DBGERROR((DBG_CONTEXT, "hr=0x%x\n", _hr_))
|
||||
|
||||
#define DBGERROR_STATUS( _status_ ) \
|
||||
DBGERROR((DBG_CONTEXT, "status=%d\n", _status_))
|
||||
|
||||
#endif
|
||||
|
|
@ -0,0 +1,325 @@
|
|||
// Copyright (c) .NET Foundation. All rights reserved.
|
||||
// Licensed under the MIT License. See License.txt in the project root for license information.
|
||||
|
||||
#ifndef __HASHFN_H__
|
||||
#define __HASHFN_H__
|
||||
|
||||
|
||||
// Produce a scrambled, randomish number in the range 0 to RANDOM_PRIME-1.
|
||||
// Applying this to the results of the other hash functions is likely to
|
||||
// produce a much better distribution, especially for the identity hash
|
||||
// functions such as Hash(char c), where records will tend to cluster at
|
||||
// the low end of the hashtable otherwise. LKRhash applies this internally
|
||||
// to all hash signatures for exactly this reason.
|
||||
|
||||
inline DWORD
|
||||
HashScramble(DWORD dwHash)
|
||||
{
|
||||
// Here are 10 primes slightly greater than 10^9
|
||||
// 1000000007, 1000000009, 1000000021, 1000000033, 1000000087,
|
||||
// 1000000093, 1000000097, 1000000103, 1000000123, 1000000181.
|
||||
|
||||
// default value for "scrambling constant"
|
||||
const DWORD RANDOM_CONSTANT = 314159269UL;
|
||||
// large prime number, also used for scrambling
|
||||
const DWORD RANDOM_PRIME = 1000000007UL;
|
||||
|
||||
return (RANDOM_CONSTANT * dwHash) % RANDOM_PRIME ;
|
||||
}
|
||||
|
||||
|
||||
// Faster scrambling function suggested by Eric Jacobsen
|
||||
|
||||
inline DWORD
|
||||
HashRandomizeBits(DWORD dw)
|
||||
{
|
||||
return (((dw * 1103515245 + 12345) >> 16)
|
||||
| ((dw * 69069 + 1) & 0xffff0000));
|
||||
}
|
||||
|
||||
|
||||
// Small prime number used as a multiplier in the supplied hash functions
|
||||
const DWORD HASH_MULTIPLIER = 101;
|
||||
|
||||
#undef HASH_SHIFT_MULTIPLY
|
||||
|
||||
#ifdef HASH_SHIFT_MULTIPLY
|
||||
# define HASH_MULTIPLY(dw) (((dw) << 7) - (dw))
|
||||
#else
|
||||
# define HASH_MULTIPLY(dw) ((dw) * HASH_MULTIPLIER)
|
||||
#endif
|
||||
|
||||
// Fast, simple hash function that tends to give a good distribution.
|
||||
// Apply HashScramble to the result if you're using this for something
|
||||
// other than LKRhash.
|
||||
|
||||
inline DWORD
|
||||
HashString(
|
||||
const char* psz,
|
||||
DWORD dwHash = 0)
|
||||
{
|
||||
// force compiler to use unsigned arithmetic
|
||||
const unsigned char* upsz = (const unsigned char*) psz;
|
||||
|
||||
for ( ; *upsz; ++upsz)
|
||||
dwHash = HASH_MULTIPLY(dwHash) + *upsz;
|
||||
|
||||
return dwHash;
|
||||
}
|
||||
|
||||
inline DWORD
|
||||
HashString(
|
||||
__in_ecount(cch) const char* psz,
|
||||
__in DWORD cch,
|
||||
__in DWORD dwHash
|
||||
)
|
||||
{
|
||||
// force compiler to use unsigned arithmetic
|
||||
const unsigned char* upsz = (const unsigned char*) psz;
|
||||
|
||||
for (DWORD Index = 0;
|
||||
Index < cch;
|
||||
++Index, ++upsz)
|
||||
{
|
||||
dwHash = HASH_MULTIPLY(dwHash) + *upsz;
|
||||
}
|
||||
|
||||
return dwHash;
|
||||
}
|
||||
|
||||
|
||||
// Unicode version of above
|
||||
|
||||
inline DWORD
|
||||
HashString(
|
||||
const wchar_t* pwsz,
|
||||
DWORD dwHash = 0)
|
||||
{
|
||||
for ( ; *pwsz; ++pwsz)
|
||||
dwHash = HASH_MULTIPLY(dwHash) + *pwsz;
|
||||
|
||||
return dwHash;
|
||||
}
|
||||
|
||||
// Based on length of the string instead of null-terminating character
|
||||
|
||||
inline DWORD
|
||||
HashString(
|
||||
__in_ecount(cch) const wchar_t* pwsz,
|
||||
__in DWORD cch,
|
||||
__in DWORD dwHash
|
||||
)
|
||||
{
|
||||
for (DWORD Index = 0;
|
||||
Index < cch;
|
||||
++Index, ++pwsz)
|
||||
{
|
||||
dwHash = HASH_MULTIPLY(dwHash) + *pwsz;
|
||||
}
|
||||
|
||||
return dwHash;
|
||||
}
|
||||
|
||||
|
||||
// Quick-'n'-dirty case-insensitive string hash function.
|
||||
// Make sure that you follow up with _stricmp or _mbsicmp. You should
|
||||
// also cache the length of strings and check those first. Caching
|
||||
// an uppercase version of a string can help too.
|
||||
// Again, apply HashScramble to the result if using with something other
|
||||
// than LKRhash.
|
||||
// Note: this is not really adequate for MBCS strings.
|
||||
|
||||
inline DWORD
|
||||
HashStringNoCase(
|
||||
const char* psz,
|
||||
DWORD dwHash = 0)
|
||||
{
|
||||
const unsigned char* upsz = (const unsigned char*) psz;
|
||||
|
||||
for ( ; *upsz; ++upsz)
|
||||
dwHash = HASH_MULTIPLY(dwHash)
|
||||
+ (*upsz & 0xDF); // strip off lowercase bit
|
||||
|
||||
return dwHash;
|
||||
}
|
||||
|
||||
inline DWORD
|
||||
HashStringNoCase(
|
||||
__in_ecount(cch)
|
||||
const char* psz,
|
||||
SIZE_T cch,
|
||||
DWORD dwHash)
|
||||
{
|
||||
const unsigned char* upsz = (const unsigned char*) psz;
|
||||
|
||||
for (SIZE_T Index = 0;
|
||||
Index < cch;
|
||||
++Index, ++upsz)
|
||||
{
|
||||
dwHash = HASH_MULTIPLY(dwHash)
|
||||
+ (*upsz & 0xDF); // strip off lowercase bit
|
||||
}
|
||||
return dwHash;
|
||||
}
|
||||
|
||||
|
||||
// Unicode version of above
|
||||
|
||||
inline DWORD
|
||||
HashStringNoCase(
|
||||
const wchar_t* pwsz,
|
||||
DWORD dwHash = 0)
|
||||
{
|
||||
for ( ; *pwsz; ++pwsz)
|
||||
dwHash = HASH_MULTIPLY(dwHash) + (*pwsz & 0xFFDF);
|
||||
|
||||
return dwHash;
|
||||
}
|
||||
|
||||
// Unicode version of above with length
|
||||
|
||||
inline DWORD
|
||||
HashStringNoCase(
|
||||
__in_ecount(cch)
|
||||
const wchar_t* pwsz,
|
||||
SIZE_T cch,
|
||||
DWORD dwHash)
|
||||
{
|
||||
for (SIZE_T Index = 0;
|
||||
Index < cch;
|
||||
++Index, ++pwsz)
|
||||
{
|
||||
dwHash = HASH_MULTIPLY(dwHash) + (*pwsz & 0xFFDF);
|
||||
}
|
||||
return dwHash;
|
||||
}
|
||||
|
||||
|
||||
// HashBlob returns the hash of a blob of arbitrary binary data.
|
||||
//
|
||||
// Warning: HashBlob is generally not the right way to hash a class object.
|
||||
// Consider:
|
||||
// class CFoo {
|
||||
// public:
|
||||
// char m_ch;
|
||||
// double m_d;
|
||||
// char* m_psz;
|
||||
// };
|
||||
//
|
||||
// inline DWORD Hash(const CFoo& rFoo)
|
||||
// { return HashBlob(&rFoo, sizeof(CFoo)); }
|
||||
//
|
||||
// This is the wrong way to hash a CFoo for two reasons: (a) there will be
|
||||
// a 7-byte gap between m_ch and m_d imposed by the alignment restrictions
|
||||
// of doubles, which will be filled with random data (usually non-zero for
|
||||
// stack variables), and (b) it hashes the address (rather than the
|
||||
// contents) of the string m_psz. Similarly,
|
||||
//
|
||||
// bool operator==(const CFoo& rFoo1, const CFoo& rFoo2)
|
||||
// { return memcmp(&rFoo1, &rFoo2, sizeof(CFoo)) == 0; }
|
||||
//
|
||||
// does the wrong thing. Much better to do this:
|
||||
//
|
||||
// DWORD Hash(const CFoo& rFoo)
|
||||
// {
|
||||
// return HashString(rFoo.m_psz,
|
||||
// HASH_MULTIPLIER * Hash(rFoo.m_ch)
|
||||
// + Hash(rFoo.m_d));
|
||||
// }
|
||||
//
|
||||
// Again, apply HashScramble if using with something other than LKRhash.
|
||||
|
||||
inline DWORD
|
||||
HashBlob(
|
||||
const void* pv,
|
||||
size_t cb,
|
||||
DWORD dwHash = 0)
|
||||
{
|
||||
const BYTE * pb = static_cast<const BYTE *>(pv);
|
||||
|
||||
while (cb-- > 0)
|
||||
dwHash = HASH_MULTIPLY(dwHash) + *pb++;
|
||||
|
||||
return dwHash;
|
||||
}
|
||||
|
||||
|
||||
|
||||
//
|
||||
// Overloaded hash functions for all the major builtin types.
|
||||
// Again, apply HashScramble to result if using with something other than
|
||||
// LKRhash.
|
||||
//
|
||||
|
||||
inline DWORD Hash(const char* psz)
|
||||
{ return HashString(psz); }
|
||||
|
||||
inline DWORD Hash(const unsigned char* pusz)
|
||||
{ return HashString(reinterpret_cast<const char*>(pusz)); }
|
||||
|
||||
inline DWORD Hash(const signed char* pssz)
|
||||
{ return HashString(reinterpret_cast<const char*>(pssz)); }
|
||||
|
||||
inline DWORD Hash(const wchar_t* pwsz)
|
||||
{ return HashString(pwsz); }
|
||||
|
||||
inline DWORD
|
||||
Hash(
|
||||
const GUID* pguid,
|
||||
DWORD dwHash = 0)
|
||||
{
|
||||
|
||||
return * reinterpret_cast<const DWORD *>(const_cast<GUID*>(pguid)) + dwHash;
|
||||
}
|
||||
|
||||
// Identity hash functions: scalar values map to themselves
|
||||
inline DWORD Hash(char c)
|
||||
{ return c; }
|
||||
|
||||
inline DWORD Hash(unsigned char uc)
|
||||
{ return uc; }
|
||||
|
||||
inline DWORD Hash(signed char sc)
|
||||
{ return sc; }
|
||||
|
||||
inline DWORD Hash(short sh)
|
||||
{ return sh; }
|
||||
|
||||
inline DWORD Hash(unsigned short ush)
|
||||
{ return ush; }
|
||||
|
||||
inline DWORD Hash(int i)
|
||||
{ return i; }
|
||||
|
||||
inline DWORD Hash(unsigned int u)
|
||||
{ return u; }
|
||||
|
||||
inline DWORD Hash(long l)
|
||||
{ return l; }
|
||||
|
||||
inline DWORD Hash(unsigned long ul)
|
||||
{ return ul; }
|
||||
|
||||
inline DWORD Hash(float f)
|
||||
{
|
||||
// be careful of rounding errors when computing keys
|
||||
union {
|
||||
float f;
|
||||
DWORD dw;
|
||||
} u;
|
||||
u.f = f;
|
||||
return u.dw;
|
||||
}
|
||||
|
||||
inline DWORD Hash(double dbl)
|
||||
{
|
||||
// be careful of rounding errors when computing keys
|
||||
union {
|
||||
double dbl;
|
||||
DWORD dw[2];
|
||||
} u;
|
||||
u.dbl = dbl;
|
||||
return u.dw[0] * HASH_MULTIPLIER + u.dw[1];
|
||||
}
|
||||
|
||||
#endif // __HASHFN_H__
|
||||
|
|
@ -0,0 +1,666 @@
|
|||
// Copyright (c) .NET Foundation. All rights reserved.
|
||||
// Licensed under the MIT License. See License.txt in the project root for license information.
|
||||
|
||||
#pragma once
|
||||
|
||||
#include <crtdbg.h>
|
||||
#include "rwlock.h"
|
||||
#include "prime.h"
|
||||
|
||||
template <class _Record>
|
||||
class HASH_NODE
|
||||
{
|
||||
template <class _Record, class _Key>
|
||||
friend class HASH_TABLE;
|
||||
|
||||
HASH_NODE(
|
||||
_Record * pRecord,
|
||||
DWORD dwHash
|
||||
) : _pNext (NULL),
|
||||
_pRecord (pRecord),
|
||||
_dwHash (dwHash)
|
||||
{}
|
||||
|
||||
~HASH_NODE()
|
||||
{
|
||||
_ASSERTE(_pRecord == NULL);
|
||||
}
|
||||
|
||||
private:
|
||||
// Next node in the hash table look-aside
|
||||
HASH_NODE<_Record> *_pNext;
|
||||
|
||||
// actual record
|
||||
_Record * _pRecord;
|
||||
|
||||
// hash value
|
||||
DWORD _dwHash;
|
||||
};
|
||||
|
||||
template <class _Record, class _Key>
|
||||
class HASH_TABLE
|
||||
{
|
||||
protected:
|
||||
typedef BOOL
|
||||
(PFN_DELETE_IF)(
|
||||
_Record * pRecord,
|
||||
PVOID pvContext
|
||||
);
|
||||
|
||||
typedef VOID
|
||||
(PFN_APPLY)(
|
||||
_Record * pRecord,
|
||||
PVOID pvContext
|
||||
);
|
||||
|
||||
public:
|
||||
HASH_TABLE(
|
||||
VOID
|
||||
)
|
||||
: _ppBuckets( NULL ),
|
||||
_nBuckets( 0 ),
|
||||
_nItems( 0 )
|
||||
{
|
||||
}
|
||||
|
||||
virtual
|
||||
~HASH_TABLE();
|
||||
|
||||
virtual
|
||||
VOID
|
||||
ReferenceRecord(
|
||||
_Record * pRecord
|
||||
) = 0;
|
||||
|
||||
virtual
|
||||
VOID
|
||||
DereferenceRecord(
|
||||
_Record * pRecord
|
||||
) = 0;
|
||||
|
||||
virtual
|
||||
_Key
|
||||
ExtractKey(
|
||||
_Record * pRecord
|
||||
) = 0;
|
||||
|
||||
virtual
|
||||
DWORD
|
||||
CalcKeyHash(
|
||||
_Key key
|
||||
) = 0;
|
||||
|
||||
virtual
|
||||
BOOL
|
||||
EqualKeys(
|
||||
_Key key1,
|
||||
_Key key2
|
||||
) = 0;
|
||||
|
||||
DWORD
|
||||
Count(
|
||||
VOID
|
||||
) const;
|
||||
|
||||
bool
|
||||
IsInitialized(
|
||||
VOID
|
||||
) const;
|
||||
|
||||
virtual
|
||||
VOID
|
||||
Clear();
|
||||
|
||||
HRESULT
|
||||
Initialize(
|
||||
DWORD nBucketSize
|
||||
);
|
||||
|
||||
virtual
|
||||
VOID
|
||||
FindKey(
|
||||
_Key key,
|
||||
_Record ** ppRecord
|
||||
);
|
||||
|
||||
virtual
|
||||
HRESULT
|
||||
InsertRecord(
|
||||
_Record * pRecord
|
||||
);
|
||||
|
||||
virtual
|
||||
VOID
|
||||
DeleteKey(
|
||||
_Key key
|
||||
);
|
||||
|
||||
virtual
|
||||
VOID
|
||||
DeleteIf(
|
||||
PFN_DELETE_IF pfnDeleteIf,
|
||||
PVOID pvContext
|
||||
);
|
||||
|
||||
VOID
|
||||
Apply(
|
||||
PFN_APPLY pfnApply,
|
||||
PVOID pvContext
|
||||
);
|
||||
|
||||
private:
|
||||
|
||||
__success(*ppNode != NULL && return != FALSE)
|
||||
BOOL
|
||||
FindNodeInternal(
|
||||
_Key key,
|
||||
DWORD dwHash,
|
||||
__deref_out
|
||||
HASH_NODE<_Record> ** ppNode,
|
||||
__deref_opt_out
|
||||
HASH_NODE<_Record> *** pppPreviousNodeNextPointer = NULL
|
||||
);
|
||||
|
||||
VOID
|
||||
DeleteNode(
|
||||
HASH_NODE<_Record> * pNode
|
||||
)
|
||||
{
|
||||
if (pNode->_pRecord != NULL)
|
||||
{
|
||||
DereferenceRecord(pNode->_pRecord);
|
||||
pNode->_pRecord = NULL;
|
||||
}
|
||||
|
||||
delete pNode;
|
||||
}
|
||||
|
||||
VOID
|
||||
RehashTableIfNeeded(
|
||||
VOID
|
||||
);
|
||||
|
||||
HASH_NODE<_Record> ** _ppBuckets;
|
||||
DWORD _nBuckets;
|
||||
DWORD _nItems;
|
||||
//
|
||||
// Allow to use lock object in const methods.
|
||||
//
|
||||
mutable
|
||||
CWSDRWLock _tableLock;
|
||||
};
|
||||
|
||||
template <class _Record, class _Key>
|
||||
HRESULT
|
||||
HASH_TABLE<_Record,_Key>::Initialize(
|
||||
DWORD nBuckets
|
||||
)
|
||||
{
|
||||
HRESULT hr = S_OK;
|
||||
|
||||
if ( nBuckets == 0 )
|
||||
{
|
||||
hr = E_INVALIDARG;
|
||||
goto Failed;
|
||||
}
|
||||
|
||||
if (nBuckets >= MAXDWORD/sizeof(HASH_NODE<_Record> *))
|
||||
{
|
||||
hr = E_INVALIDARG;
|
||||
goto Failed;
|
||||
}
|
||||
|
||||
_ASSERTE(_ppBuckets == NULL );
|
||||
if ( _ppBuckets != NULL )
|
||||
{
|
||||
hr = E_INVALIDARG;
|
||||
goto Failed;
|
||||
}
|
||||
|
||||
hr = _tableLock.Init();
|
||||
if ( FAILED( hr ) )
|
||||
{
|
||||
goto Failed;
|
||||
}
|
||||
|
||||
_ppBuckets = (HASH_NODE<_Record> **)HeapAlloc(
|
||||
GetProcessHeap(),
|
||||
HEAP_ZERO_MEMORY,
|
||||
nBuckets*sizeof(HASH_NODE<_Record> *));
|
||||
if (_ppBuckets == NULL)
|
||||
{
|
||||
hr = HRESULT_FROM_WIN32(ERROR_NOT_ENOUGH_MEMORY);
|
||||
goto Failed;
|
||||
}
|
||||
_nBuckets = nBuckets;
|
||||
|
||||
return S_OK;
|
||||
|
||||
Failed:
|
||||
|
||||
if (_ppBuckets)
|
||||
{
|
||||
HeapFree(GetProcessHeap(),
|
||||
0,
|
||||
_ppBuckets);
|
||||
_ppBuckets = NULL;
|
||||
}
|
||||
|
||||
return hr;
|
||||
}
|
||||
|
||||
|
||||
template <class _Record, class _Key>
|
||||
HASH_TABLE<_Record,_Key>::~HASH_TABLE()
|
||||
{
|
||||
if (_ppBuckets == NULL)
|
||||
{
|
||||
return;
|
||||
}
|
||||
|
||||
_ASSERTE(_nItems == 0);
|
||||
|
||||
HeapFree(GetProcessHeap(),
|
||||
0,
|
||||
_ppBuckets);
|
||||
_ppBuckets = NULL;
|
||||
_nBuckets = 0;
|
||||
}
|
||||
|
||||
template< class _Record, class _Key>
|
||||
DWORD
|
||||
HASH_TABLE<_Record,_Key>::Count() const
|
||||
{
|
||||
return _nItems;
|
||||
}
|
||||
|
||||
template< class _Record, class _Key>
|
||||
bool
|
||||
HASH_TABLE<_Record,_Key>::IsInitialized(
|
||||
VOID
|
||||
) const
|
||||
{
|
||||
return _ppBuckets != NULL;
|
||||
}
|
||||
|
||||
|
||||
template <class _Record, class _Key>
|
||||
VOID
|
||||
HASH_TABLE<_Record,_Key>::Clear()
|
||||
{
|
||||
HASH_NODE<_Record> *pCurrent;
|
||||
HASH_NODE<_Record> *pNext;
|
||||
|
||||
// This is here in the off cases where someone instantiates a hashtable
|
||||
// and then does an automatic "clear" before its destruction WITHOUT
|
||||
// ever initializing it.
|
||||
if ( ! _tableLock.QueryInited() )
|
||||
{
|
||||
return;
|
||||
}
|
||||
|
||||
_tableLock.ExclusiveAcquire();
|
||||
|
||||
for (DWORD i=0; i<_nBuckets; i++)
|
||||
{
|
||||
pCurrent = _ppBuckets[i];
|
||||
_ppBuckets[i] = NULL;
|
||||
while (pCurrent != NULL)
|
||||
{
|
||||
pNext = pCurrent->_pNext;
|
||||
DeleteNode(pCurrent);
|
||||
pCurrent = pNext;
|
||||
}
|
||||
}
|
||||
|
||||
_nItems = 0;
|
||||
_tableLock.ExclusiveRelease();
|
||||
}
|
||||
|
||||
template <class _Record, class _Key>
|
||||
__success(*ppNode != NULL && return != FALSE)
|
||||
BOOL
|
||||
HASH_TABLE<_Record,_Key>::FindNodeInternal(
|
||||
_Key key,
|
||||
DWORD dwHash,
|
||||
__deref_out
|
||||
HASH_NODE<_Record> ** ppNode,
|
||||
__deref_opt_out
|
||||
HASH_NODE<_Record> *** pppPreviousNodeNextPointer
|
||||
)
|
||||
/*++
|
||||
Return value indicates whether the item is found
|
||||
key, dwHash - key and hash for the node to find
|
||||
ppNode - on successful return, the node found, on failed return, the first
|
||||
node with hash value greater than the node to be found
|
||||
pppPreviousNodeNextPointer - the pointer to previous node's _pNext
|
||||
|
||||
This routine may be called under either read or write lock
|
||||
--*/
|
||||
{
|
||||
HASH_NODE<_Record> **ppPreviousNodeNextPointer;
|
||||
HASH_NODE<_Record> *pNode;
|
||||
BOOL fFound = FALSE;
|
||||
|
||||
ppPreviousNodeNextPointer = _ppBuckets + (dwHash % _nBuckets);
|
||||
pNode = *ppPreviousNodeNextPointer;
|
||||
while (pNode != NULL)
|
||||
{
|
||||
if (pNode->_dwHash == dwHash)
|
||||
{
|
||||
if (EqualKeys(key,
|
||||
ExtractKey(pNode->_pRecord)))
|
||||
{
|
||||
fFound = TRUE;
|
||||
break;
|
||||
}
|
||||
}
|
||||
else if (pNode->_dwHash > dwHash)
|
||||
{
|
||||
break;
|
||||
}
|
||||
|
||||
ppPreviousNodeNextPointer = &(pNode->_pNext);
|
||||
pNode = *ppPreviousNodeNextPointer;
|
||||
}
|
||||
|
||||
__analysis_assume( (pNode == NULL && fFound == FALSE) ||
|
||||
(pNode != NULL && fFound == TRUE ) );
|
||||
*ppNode = pNode;
|
||||
if (pppPreviousNodeNextPointer != NULL)
|
||||
{
|
||||
*pppPreviousNodeNextPointer = ppPreviousNodeNextPointer;
|
||||
}
|
||||
return fFound;
|
||||
}
|
||||
|
||||
template <class _Record, class _Key>
|
||||
VOID
|
||||
HASH_TABLE<_Record,_Key>::FindKey(
|
||||
_Key key,
|
||||
_Record ** ppRecord
|
||||
)
|
||||
{
|
||||
HASH_NODE<_Record> *pNode;
|
||||
|
||||
*ppRecord = NULL;
|
||||
|
||||
DWORD dwHash = CalcKeyHash(key);
|
||||
|
||||
_tableLock.SharedAcquire();
|
||||
|
||||
if (FindNodeInternal(key, dwHash, &pNode) &&
|
||||
pNode->_pRecord != NULL)
|
||||
{
|
||||
ReferenceRecord(pNode->_pRecord);
|
||||
*ppRecord = pNode->_pRecord;
|
||||
}
|
||||
|
||||
_tableLock.SharedRelease();
|
||||
}
|
||||
|
||||
template <class _Record, class _Key>
|
||||
HRESULT
|
||||
HASH_TABLE<_Record,_Key>::InsertRecord(
|
||||
_Record * pRecord
|
||||
)
|
||||
/*++
|
||||
This method inserts a node for this record and also empty nodes for paths
|
||||
in the heirarchy leading upto this path
|
||||
|
||||
The insert is done under only a read-lock - this is possible by keeping
|
||||
the hashes in a bucket in increasing order and using interlocked operations
|
||||
to actually insert the item in the hash-bucket lookaside list and the parent
|
||||
children list
|
||||
|
||||
Returns HRESULT_FROM_WIN32(ERROR_ALREADY_EXISTS) if the record already exists.
|
||||
Never leak this error to the end user because "*file* already exists" may be confusing.
|
||||
--*/
|
||||
{
|
||||
BOOL fLocked = FALSE;
|
||||
_Key key = ExtractKey(pRecord);
|
||||
DWORD dwHash = CalcKeyHash(key);
|
||||
HRESULT hr = S_OK;
|
||||
HASH_NODE<_Record> * pNewNode;
|
||||
HASH_NODE<_Record> * pNextNode;
|
||||
HASH_NODE<_Record> ** ppPreviousNodeNextPointer;
|
||||
|
||||
//
|
||||
// Ownership of pRecord is not transferred to pNewNode yet, so remember
|
||||
// to either set it to null before deleting pNewNode or add an extra
|
||||
// reference later - this is to make sure we do not do an extra ref/deref
|
||||
// which users may view as getting flushed out of the hash-table
|
||||
//
|
||||
pNewNode = new HASH_NODE<_Record>(pRecord, dwHash);
|
||||
if (pNewNode == NULL)
|
||||
{
|
||||
hr = HRESULT_FROM_WIN32(ERROR_NOT_ENOUGH_MEMORY);
|
||||
goto Finished;
|
||||
}
|
||||
|
||||
_tableLock.SharedAcquire();
|
||||
fLocked = TRUE;
|
||||
|
||||
do
|
||||
{
|
||||
//
|
||||
// Find the right place to add this node
|
||||
//
|
||||
if (FindNodeInternal(key, dwHash, &pNextNode, &ppPreviousNodeNextPointer))
|
||||
{
|
||||
//
|
||||
// If node already there, return error
|
||||
//
|
||||
pNewNode->_pRecord = NULL;
|
||||
DeleteNode(pNewNode);
|
||||
|
||||
//
|
||||
// We should never leak this error to the end user
|
||||
// because "file already exists" may be confusing.
|
||||
//
|
||||
hr = HRESULT_FROM_WIN32(ERROR_ALREADY_EXISTS);
|
||||
goto Finished;
|
||||
}
|
||||
|
||||
//
|
||||
// If another node got inserted in between, we will have to retry
|
||||
//
|
||||
pNewNode->_pNext = pNextNode;
|
||||
} while (InterlockedCompareExchangePointer((PVOID *)ppPreviousNodeNextPointer,
|
||||
pNewNode,
|
||||
pNextNode) != pNextNode);
|
||||
// pass ownership of pRecord now
|
||||
if (pRecord != NULL)
|
||||
{
|
||||
ReferenceRecord(pRecord);
|
||||
pRecord = NULL;
|
||||
}
|
||||
InterlockedIncrement((LONG *)&_nItems);
|
||||
|
||||
Finished:
|
||||
|
||||
if (fLocked)
|
||||
{
|
||||
_tableLock.SharedRelease();
|
||||
}
|
||||
|
||||
if (SUCCEEDED(hr))
|
||||
{
|
||||
RehashTableIfNeeded();
|
||||
}
|
||||
|
||||
return hr;
|
||||
}
|
||||
|
||||
template <class _Record, class _Key>
|
||||
VOID
|
||||
HASH_TABLE<_Record,_Key>::DeleteKey(
|
||||
_Key key
|
||||
)
|
||||
{
|
||||
HASH_NODE<_Record> *pNode;
|
||||
HASH_NODE<_Record> **ppPreviousNodeNextPointer;
|
||||
|
||||
DWORD dwHash = CalcKeyHash(key);
|
||||
|
||||
_tableLock.ExclusiveAcquire();
|
||||
|
||||
if (FindNodeInternal(key, dwHash, &pNode, &ppPreviousNodeNextPointer))
|
||||
{
|
||||
*ppPreviousNodeNextPointer = pNode->_pNext;
|
||||
DeleteNode(pNode);
|
||||
_nItems--;
|
||||
}
|
||||
|
||||
_tableLock.ExclusiveRelease();
|
||||
}
|
||||
|
||||
template <class _Record, class _Key>
|
||||
VOID
|
||||
HASH_TABLE<_Record,_Key>::DeleteIf(
|
||||
PFN_DELETE_IF pfnDeleteIf,
|
||||
PVOID pvContext
|
||||
)
|
||||
{
|
||||
HASH_NODE<_Record> *pNode;
|
||||
HASH_NODE<_Record> **ppPreviousNodeNextPointer;
|
||||
|
||||
_tableLock.ExclusiveAcquire();
|
||||
|
||||
for (DWORD i=0; i<_nBuckets; i++)
|
||||
{
|
||||
ppPreviousNodeNextPointer = _ppBuckets + i;
|
||||
pNode = *ppPreviousNodeNextPointer;
|
||||
while (pNode != NULL)
|
||||
{
|
||||
//
|
||||
// Non empty nodes deleted based on DeleteIf, empty nodes deleted
|
||||
// if they have no children
|
||||
//
|
||||
if (pfnDeleteIf(pNode->_pRecord, pvContext))
|
||||
{
|
||||
*ppPreviousNodeNextPointer = pNode->_pNext;
|
||||
DeleteNode(pNode);
|
||||
_nItems--;
|
||||
}
|
||||
else
|
||||
{
|
||||
ppPreviousNodeNextPointer = &pNode->_pNext;
|
||||
}
|
||||
|
||||
pNode = *ppPreviousNodeNextPointer;
|
||||
}
|
||||
}
|
||||
|
||||
_tableLock.ExclusiveRelease();
|
||||
}
|
||||
|
||||
template <class _Record, class _Key>
|
||||
VOID
|
||||
HASH_TABLE<_Record,_Key>::Apply(
|
||||
PFN_APPLY pfnApply,
|
||||
PVOID pvContext
|
||||
)
|
||||
{
|
||||
HASH_NODE<_Record> *pNode;
|
||||
|
||||
_tableLock.SharedAcquire();
|
||||
|
||||
for (DWORD i=0; i<_nBuckets; i++)
|
||||
{
|
||||
pNode = _ppBuckets[i];
|
||||
while (pNode != NULL)
|
||||
{
|
||||
if (pNode->_pRecord != NULL)
|
||||
{
|
||||
pfnApply(pNode->_pRecord, pvContext);
|
||||
}
|
||||
|
||||
pNode = pNode->_pNext;
|
||||
}
|
||||
}
|
||||
|
||||
_tableLock.SharedRelease();
|
||||
}
|
||||
|
||||
template <class _Record, class _Key>
|
||||
VOID
|
||||
HASH_TABLE<_Record,_Key>::RehashTableIfNeeded(
|
||||
VOID
|
||||
)
|
||||
{
|
||||
HASH_NODE<_Record> **ppBuckets;
|
||||
DWORD nBuckets;
|
||||
HASH_NODE<_Record> *pNode;
|
||||
HASH_NODE<_Record> *pNextNode;
|
||||
HASH_NODE<_Record> **ppNextPointer;
|
||||
HASH_NODE<_Record> *pNewNextNode;
|
||||
DWORD nNewBuckets;
|
||||
|
||||
//
|
||||
// If number of items has become too many, we will double the hash table
|
||||
// size (we never reduce it however)
|
||||
//
|
||||
if (_nItems <= PRIME::GetPrime(2*_nBuckets))
|
||||
{
|
||||
return;
|
||||
}
|
||||
|
||||
_tableLock.ExclusiveAcquire();
|
||||
|
||||
nNewBuckets = PRIME::GetPrime(2*_nBuckets);
|
||||
|
||||
if (_nItems <= nNewBuckets)
|
||||
{
|
||||
goto Finished;
|
||||
}
|
||||
|
||||
nBuckets = nNewBuckets;
|
||||
if (nBuckets >= 0xffffffff/sizeof(HASH_NODE<_Record> *))
|
||||
{
|
||||
goto Finished;
|
||||
}
|
||||
ppBuckets = (HASH_NODE<_Record> **)HeapAlloc(
|
||||
GetProcessHeap(),
|
||||
HEAP_ZERO_MEMORY,
|
||||
nBuckets*sizeof(HASH_NODE<_Record> *));
|
||||
if (ppBuckets == NULL)
|
||||
{
|
||||
goto Finished;
|
||||
}
|
||||
|
||||
//
|
||||
// Take out nodes from the old hash table and insert in the new one, make
|
||||
// sure to keep the hashes in increasing order
|
||||
//
|
||||
for (DWORD i=0; i<_nBuckets; i++)
|
||||
{
|
||||
pNode = _ppBuckets[i];
|
||||
while (pNode != NULL)
|
||||
{
|
||||
pNextNode = pNode->_pNext;
|
||||
|
||||
ppNextPointer = ppBuckets + (pNode->_dwHash % nBuckets);
|
||||
pNewNextNode = *ppNextPointer;
|
||||
while (pNewNextNode != NULL &&
|
||||
pNewNextNode->_dwHash <= pNode->_dwHash)
|
||||
{
|
||||
ppNextPointer = &pNewNextNode->_pNext;
|
||||
pNewNextNode = pNewNextNode->_pNext;
|
||||
}
|
||||
pNode->_pNext = pNewNextNode;
|
||||
*ppNextPointer = pNode;
|
||||
|
||||
pNode = pNextNode;
|
||||
}
|
||||
}
|
||||
|
||||
HeapFree(GetProcessHeap(), 0, _ppBuckets);
|
||||
_ppBuckets = ppBuckets;
|
||||
_nBuckets = nBuckets;
|
||||
ppBuckets = NULL;
|
||||
|
||||
Finished:
|
||||
|
||||
_tableLock.ExclusiveRelease();
|
||||
}
|
||||
|
|
@ -0,0 +1,163 @@
|
|||
// Copyright (c) .NET Foundation. All rights reserved.
|
||||
// Licensed under the MIT License. See License.txt in the project root for license information.
|
||||
|
||||
#pragma once
|
||||
|
||||
#ifndef _LIST_ENTRY_H
|
||||
#define _LIST_ENTRY_H
|
||||
|
||||
//
|
||||
// Doubly-linked list manipulation routines.
|
||||
//
|
||||
|
||||
|
||||
#define InitializeListHead32(ListHead) (\
|
||||
(ListHead)->Flink = (ListHead)->Blink = PtrToUlong((ListHead)))
|
||||
|
||||
|
||||
FORCEINLINE
|
||||
VOID
|
||||
InitializeListHead(
|
||||
IN PLIST_ENTRY ListHead
|
||||
)
|
||||
{
|
||||
ListHead->Flink = ListHead->Blink = ListHead;
|
||||
}
|
||||
|
||||
FORCEINLINE
|
||||
BOOLEAN
|
||||
IsListEmpty(
|
||||
IN const LIST_ENTRY * ListHead
|
||||
)
|
||||
{
|
||||
return (BOOLEAN)(ListHead->Flink == ListHead);
|
||||
}
|
||||
|
||||
FORCEINLINE
|
||||
BOOLEAN
|
||||
RemoveEntryList(
|
||||
IN PLIST_ENTRY Entry
|
||||
)
|
||||
{
|
||||
PLIST_ENTRY Blink;
|
||||
PLIST_ENTRY Flink;
|
||||
|
||||
Flink = Entry->Flink;
|
||||
Blink = Entry->Blink;
|
||||
Blink->Flink = Flink;
|
||||
Flink->Blink = Blink;
|
||||
return (BOOLEAN)(Flink == Blink);
|
||||
}
|
||||
|
||||
FORCEINLINE
|
||||
PLIST_ENTRY
|
||||
RemoveHeadList(
|
||||
IN PLIST_ENTRY ListHead
|
||||
)
|
||||
{
|
||||
PLIST_ENTRY Flink;
|
||||
PLIST_ENTRY Entry;
|
||||
|
||||
Entry = ListHead->Flink;
|
||||
Flink = Entry->Flink;
|
||||
ListHead->Flink = Flink;
|
||||
Flink->Blink = ListHead;
|
||||
return Entry;
|
||||
}
|
||||
|
||||
|
||||
|
||||
FORCEINLINE
|
||||
PLIST_ENTRY
|
||||
RemoveTailList(
|
||||
IN PLIST_ENTRY ListHead
|
||||
)
|
||||
{
|
||||
PLIST_ENTRY Blink;
|
||||
PLIST_ENTRY Entry;
|
||||
|
||||
Entry = ListHead->Blink;
|
||||
Blink = Entry->Blink;
|
||||
ListHead->Blink = Blink;
|
||||
Blink->Flink = ListHead;
|
||||
return Entry;
|
||||
}
|
||||
|
||||
|
||||
FORCEINLINE
|
||||
VOID
|
||||
InsertTailList(
|
||||
IN PLIST_ENTRY ListHead,
|
||||
IN PLIST_ENTRY Entry
|
||||
)
|
||||
{
|
||||
PLIST_ENTRY Blink;
|
||||
|
||||
Blink = ListHead->Blink;
|
||||
Entry->Flink = ListHead;
|
||||
Entry->Blink = Blink;
|
||||
Blink->Flink = Entry;
|
||||
ListHead->Blink = Entry;
|
||||
}
|
||||
|
||||
|
||||
FORCEINLINE
|
||||
VOID
|
||||
InsertHeadList(
|
||||
IN PLIST_ENTRY ListHead,
|
||||
IN PLIST_ENTRY Entry
|
||||
)
|
||||
{
|
||||
PLIST_ENTRY Flink;
|
||||
|
||||
Flink = ListHead->Flink;
|
||||
Entry->Flink = Flink;
|
||||
Entry->Blink = ListHead;
|
||||
Flink->Blink = Entry;
|
||||
ListHead->Flink = Entry;
|
||||
}
|
||||
|
||||
FORCEINLINE
|
||||
VOID
|
||||
AppendTailList(
|
||||
IN PLIST_ENTRY ListHead,
|
||||
IN PLIST_ENTRY ListToAppend
|
||||
)
|
||||
{
|
||||
PLIST_ENTRY ListEnd = ListHead->Blink;
|
||||
|
||||
ListHead->Blink->Flink = ListToAppend;
|
||||
ListHead->Blink = ListToAppend->Blink;
|
||||
ListToAppend->Blink->Flink = ListHead;
|
||||
ListToAppend->Blink = ListEnd;
|
||||
}
|
||||
|
||||
FORCEINLINE
|
||||
PSINGLE_LIST_ENTRY
|
||||
PopEntryList(
|
||||
PSINGLE_LIST_ENTRY ListHead
|
||||
)
|
||||
{
|
||||
PSINGLE_LIST_ENTRY FirstEntry;
|
||||
FirstEntry = ListHead->Next;
|
||||
if (FirstEntry != NULL) {
|
||||
ListHead->Next = FirstEntry->Next;
|
||||
}
|
||||
|
||||
return FirstEntry;
|
||||
}
|
||||
|
||||
|
||||
FORCEINLINE
|
||||
VOID
|
||||
PushEntryList(
|
||||
PSINGLE_LIST_ENTRY ListHead,
|
||||
PSINGLE_LIST_ENTRY Entry
|
||||
)
|
||||
{
|
||||
Entry->Next = ListHead->Next;
|
||||
ListHead->Next = Entry;
|
||||
}
|
||||
|
||||
|
||||
#endif
|
||||
|
|
@ -0,0 +1,63 @@
|
|||
// Copyright (c) .NET Foundation. All rights reserved.
|
||||
// Licensed under the MIT License. See License.txt in the project root for license information.
|
||||
|
||||
#ifndef _MACROS_H
|
||||
#define _MACROS_H
|
||||
|
||||
//
|
||||
// The DIFF macro should be used around an expression involving pointer
|
||||
// subtraction. The expression passed to DIFF is cast to a size_t type,
|
||||
// allowing the result to be easily assigned to any 32-bit variable or
|
||||
// passed to a function expecting a 32-bit argument.
|
||||
//
|
||||
|
||||
#define DIFF(x) ((size_t)(x))
|
||||
|
||||
// Change a hexadecimal digit to its numerical equivalent
|
||||
#define TOHEX( ch ) \
|
||||
((ch) > L'9' ? \
|
||||
(ch) >= L'a' ? \
|
||||
(ch) - L'a' + 10 : \
|
||||
(ch) - L'A' + 10 \
|
||||
: (ch) - L'0')
|
||||
|
||||
|
||||
// Change a number to its Hexadecimal equivalent
|
||||
|
||||
#define TODIGIT( nDigit ) \
|
||||
(CHAR)((nDigit) > 9 ? \
|
||||
(nDigit) - 10 + 'A' \
|
||||
: (nDigit) + '0')
|
||||
|
||||
|
||||
inline int
|
||||
SAFEIsSpace(UCHAR c)
|
||||
{
|
||||
return isspace( c );
|
||||
}
|
||||
|
||||
inline int
|
||||
SAFEIsAlNum(UCHAR c)
|
||||
{
|
||||
return isalnum( c );
|
||||
}
|
||||
|
||||
inline int
|
||||
SAFEIsAlpha(UCHAR c)
|
||||
{
|
||||
return isalpha( c );
|
||||
}
|
||||
|
||||
inline int
|
||||
SAFEIsXDigit(UCHAR c)
|
||||
{
|
||||
return isxdigit( c );
|
||||
}
|
||||
|
||||
inline int
|
||||
SAFEIsDigit(UCHAR c)
|
||||
{
|
||||
return isdigit( c );
|
||||
}
|
||||
|
||||
#endif // _MACROS_H
|
||||
|
|
@ -0,0 +1,474 @@
|
|||
// Copyright (c) .NET Foundation. All rights reserved.
|
||||
// Licensed under the MIT License. See License.txt in the project root for license information.
|
||||
|
||||
|
||||
#pragma warning (disable : 4267)
|
||||
|
||||
#include "precomp.h"
|
||||
#include "multisz.h"
|
||||
#include <tchar.h>
|
||||
|
||||
//
|
||||
// Private Definitions
|
||||
//
|
||||
|
||||
#define MAXULONG 4294967295
|
||||
#define ISWHITE( ch ) ((ch) == L' ' || (ch) == L'\t' || (ch) == L'\r')
|
||||
|
||||
//
|
||||
// When appending data, this is the extra amount we request to avoid
|
||||
// reallocations
|
||||
//
|
||||
#define STR_SLOP 128
|
||||
|
||||
|
||||
DWORD
|
||||
MULTISZ::CalcLength( const WCHAR * str,
|
||||
LPDWORD pcStrings )
|
||||
{
|
||||
DWORD count = 0;
|
||||
DWORD total = 1;
|
||||
DWORD len;
|
||||
|
||||
while( *str ) {
|
||||
len = ::wcslen( str ) + 1;
|
||||
total += len;
|
||||
str += len;
|
||||
count++;
|
||||
}
|
||||
|
||||
if( pcStrings != NULL ) {
|
||||
*pcStrings = count;
|
||||
}
|
||||
|
||||
return total;
|
||||
|
||||
} // MULTISZ::CalcLength
|
||||
|
||||
|
||||
BOOL
|
||||
MULTISZ::FindString( const WCHAR * str )
|
||||
{
|
||||
|
||||
WCHAR * multisz;
|
||||
|
||||
//
|
||||
// Sanity check.
|
||||
//
|
||||
|
||||
DBG_ASSERT( QueryStr() != NULL );
|
||||
DBG_ASSERT( str != NULL );
|
||||
DBG_ASSERT( *str != '\0' );
|
||||
|
||||
//
|
||||
// Scan it.
|
||||
//
|
||||
|
||||
multisz = QueryStr();
|
||||
|
||||
while( *multisz != '\0' ) {
|
||||
|
||||
if( !::wcscmp( multisz, str ) ) {
|
||||
|
||||
return TRUE;
|
||||
|
||||
}
|
||||
|
||||
multisz += ::wcslen( multisz ) + 1;
|
||||
|
||||
}
|
||||
|
||||
return FALSE;
|
||||
|
||||
} // MULTISZ::FindString
|
||||
|
||||
|
||||
BOOL
|
||||
MULTISZ::FindStringNoCase( const WCHAR * str )
|
||||
{
|
||||
|
||||
WCHAR * multisz;
|
||||
|
||||
//
|
||||
// Sanity check.
|
||||
//
|
||||
|
||||
DBG_ASSERT( QueryStr() != NULL );
|
||||
DBG_ASSERT( str != NULL );
|
||||
DBG_ASSERT( *str != '\0' );
|
||||
|
||||
//
|
||||
// Scan it.
|
||||
//
|
||||
|
||||
multisz = QueryStr();
|
||||
|
||||
while( *multisz != '\0' ) {
|
||||
|
||||
if( !_wcsicmp( multisz, str ) ) {
|
||||
|
||||
return TRUE;
|
||||
|
||||
}
|
||||
|
||||
multisz += wcslen( multisz ) + 1;
|
||||
|
||||
}
|
||||
|
||||
return FALSE;
|
||||
|
||||
} // MULTISZ::FindStringNoCase
|
||||
|
||||
|
||||
VOID
|
||||
MULTISZ::AuxInit( const WCHAR * pInit )
|
||||
{
|
||||
BOOL fRet;
|
||||
|
||||
if ( pInit )
|
||||
{
|
||||
DWORD cStrings;
|
||||
int cbCopy = CalcLength( pInit, &cStrings ) * sizeof(WCHAR);
|
||||
fRet = Resize( cbCopy );
|
||||
|
||||
if ( fRet ) {
|
||||
CopyMemory( QueryPtr(), pInit, cbCopy );
|
||||
m_cchLen = (cbCopy)/sizeof(WCHAR);
|
||||
m_cStrings = cStrings;
|
||||
} else {
|
||||
// BUFFER::SetValid( FALSE);
|
||||
}
|
||||
|
||||
} else {
|
||||
|
||||
Reset();
|
||||
|
||||
}
|
||||
|
||||
} // MULTISZ::AuxInit()
|
||||
|
||||
|
||||
/*******************************************************************
|
||||
|
||||
NAME: MULTISZ::AuxAppend
|
||||
|
||||
SYNOPSIS: Appends the string onto the multisz.
|
||||
|
||||
ENTRY: Object to append
|
||||
********************************************************************/
|
||||
|
||||
BOOL MULTISZ::AuxAppend( const WCHAR * pStr, UINT cbStr, BOOL fAddSlop )
|
||||
{
|
||||
DBG_ASSERT( pStr != NULL );
|
||||
|
||||
UINT cbThis = QueryCB();
|
||||
|
||||
DBG_ASSERT( cbThis >= 2 );
|
||||
|
||||
if( cbThis == 4 ) {
|
||||
|
||||
//
|
||||
// It's empty, so start at the beginning.
|
||||
//
|
||||
|
||||
cbThis = 0;
|
||||
|
||||
} else {
|
||||
|
||||
//
|
||||
// It's not empty, so back up over the final terminating NULL.
|
||||
//
|
||||
|
||||
cbThis -= sizeof(WCHAR);
|
||||
|
||||
}
|
||||
|
||||
//
|
||||
// Only resize when we have to. When we do resize, we tack on
|
||||
// some extra space to avoid extra reallocations.
|
||||
//
|
||||
// Note: QuerySize returns the requested size of the string buffer,
|
||||
// *not* the strlen of the buffer
|
||||
//
|
||||
|
||||
//AcIncrement( CacMultiszAppend);
|
||||
|
||||
//
|
||||
// Check for the arithmetic overflow
|
||||
//
|
||||
// ( 2 * sizeof( WCHAR ) ) is for the double terminator
|
||||
//
|
||||
ULONGLONG cb64Required = (ULONGLONG)cbThis + cbStr + 2 * sizeof(WCHAR);
|
||||
if ( cb64Required > MAXULONG )
|
||||
{
|
||||
SetLastError( ERROR_ARITHMETIC_OVERFLOW );
|
||||
return FALSE;
|
||||
}
|
||||
if ( QuerySize() < (DWORD) cb64Required )
|
||||
{
|
||||
ULONGLONG cb64AllocSize = cb64Required + (fAddSlop ? STR_SLOP : 0 );
|
||||
//
|
||||
// Check for the arithmetic overflow
|
||||
//
|
||||
if ( cb64AllocSize > MAXULONG )
|
||||
{
|
||||
SetLastError( ERROR_ARITHMETIC_OVERFLOW );
|
||||
return FALSE;
|
||||
}
|
||||
if ( !Resize( (DWORD) cb64AllocSize ) )
|
||||
return FALSE;
|
||||
}
|
||||
|
||||
// copy the exact string and tack on the double terminator
|
||||
memcpy( (BYTE *) QueryPtr() + cbThis,
|
||||
pStr,
|
||||
cbStr);
|
||||
*(WCHAR *)((BYTE *)QueryPtr() + cbThis + cbStr) = L'\0';
|
||||
*(WCHAR *)((BYTE *)QueryPtr() + cbThis + cbStr + sizeof(WCHAR) ) = L'\0';
|
||||
|
||||
m_cchLen = CalcLength( (const WCHAR *)QueryPtr(), &m_cStrings );
|
||||
return TRUE;
|
||||
|
||||
} // MULTISZ::AuxAppend()
|
||||
|
||||
|
||||
#if 0
|
||||
|
||||
BOOL
|
||||
MULTISZ::CopyToBuffer( WCHAR * lpszBuffer, LPDWORD lpcch) const
|
||||
/*++
|
||||
Description:
|
||||
Copies the string into the WCHAR buffer passed in if the buffer
|
||||
is sufficient to hold the translated string.
|
||||
If the buffer is small, the function returns small and sets *lpcch
|
||||
to contain the required number of characters.
|
||||
|
||||
Arguments:
|
||||
lpszBuffer pointer to WCHAR buffer which on return contains
|
||||
the UNICODE version of string on success.
|
||||
lpcch pointer to DWORD containing the length of the buffer.
|
||||
If *lpcch == 0 then the function returns TRUE with
|
||||
the count of characters required stored in *lpcch.
|
||||
Also in this case lpszBuffer is not affected.
|
||||
Returns:
|
||||
TRUE on success.
|
||||
FALSE on failure. Use GetLastError() for further details.
|
||||
--*/
|
||||
{
|
||||
BOOL fReturn = TRUE;
|
||||
|
||||
if ( lpcch == NULL) {
|
||||
SetLastError( ERROR_INVALID_PARAMETER);
|
||||
return ( FALSE);
|
||||
}
|
||||
|
||||
if ( *lpcch == 0) {
|
||||
|
||||
//
|
||||
// Inquiring the size of buffer alone
|
||||
//
|
||||
*lpcch = QueryCCH() + 1; // add one character for terminating null
|
||||
} else {
|
||||
|
||||
//
|
||||
// Copy after conversion from ANSI to Unicode
|
||||
//
|
||||
int iRet;
|
||||
iRet = MultiByteToWideChar( CP_ACP,
|
||||
MB_PRECOMPOSED | MB_ERR_INVALID_CHARS,
|
||||
QueryStrA(), QueryCCH() + 1,
|
||||
lpszBuffer, (int )*lpcch);
|
||||
|
||||
if ( iRet == 0 || iRet != (int ) *lpcch) {
|
||||
|
||||
//
|
||||
// Error in conversion.
|
||||
//
|
||||
fReturn = FALSE;
|
||||
}
|
||||
}
|
||||
|
||||
return ( fReturn);
|
||||
} // MULTISZ::CopyToBuffer()
|
||||
#endif
|
||||
|
||||
BOOL
|
||||
MULTISZ::CopyToBuffer( __out_ecount_opt(*lpcch) WCHAR * lpszBuffer, LPDWORD lpcch) const
|
||||
/*++
|
||||
Description:
|
||||
Copies the string into the WCHAR buffer passed in if the buffer
|
||||
is sufficient to hold the translated string.
|
||||
If the buffer is small, the function returns small and sets *lpcch
|
||||
to contain the required number of characters.
|
||||
|
||||
Arguments:
|
||||
lpszBuffer pointer to WCHAR buffer which on return contains
|
||||
the string on success.
|
||||
lpcch pointer to DWORD containing the length of the buffer.
|
||||
If *lpcch == 0 then the function returns TRUE with
|
||||
the count of characters required stored in lpcch.
|
||||
Also in this case lpszBuffer is not affected.
|
||||
Returns:
|
||||
TRUE on success.
|
||||
FALSE on failure. Use GetLastError() for further details.
|
||||
--*/
|
||||
{
|
||||
BOOL fReturn = TRUE;
|
||||
|
||||
if ( lpcch == NULL) {
|
||||
SetLastError( ERROR_INVALID_PARAMETER);
|
||||
return ( FALSE);
|
||||
}
|
||||
|
||||
register DWORD cch = QueryCCH();
|
||||
|
||||
if ( *lpcch >= cch) {
|
||||
|
||||
DBG_ASSERT( lpszBuffer);
|
||||
memcpy( lpszBuffer, QueryStr(), cch * sizeof(WCHAR));
|
||||
} else {
|
||||
DBG_ASSERT( *lpcch < cch);
|
||||
SetLastError( ERROR_INSUFFICIENT_BUFFER);
|
||||
fReturn = FALSE;
|
||||
}
|
||||
|
||||
*lpcch = cch;
|
||||
|
||||
return ( fReturn);
|
||||
} // MULTISZ::CopyToBuffer()
|
||||
|
||||
BOOL
|
||||
MULTISZ::Equals(
|
||||
MULTISZ* pmszRhs
|
||||
)
|
||||
//
|
||||
// Compares this to pmszRhs, returns TRUE if equal
|
||||
//
|
||||
{
|
||||
DBG_ASSERT( NULL != pmszRhs );
|
||||
|
||||
PCWSTR pszLhs = First( );
|
||||
PCWSTR pszRhs = pmszRhs->First( );
|
||||
|
||||
if( m_cStrings != pmszRhs->m_cStrings )
|
||||
{
|
||||
return FALSE;
|
||||
}
|
||||
|
||||
while( NULL != pszLhs )
|
||||
{
|
||||
DBG_ASSERT( NULL != pszRhs );
|
||||
|
||||
if( 0 != wcscmp( pszLhs, pszRhs ) )
|
||||
{
|
||||
return FALSE;
|
||||
}
|
||||
|
||||
pszLhs = Next( pszLhs );
|
||||
pszRhs = pmszRhs->Next( pszRhs );
|
||||
}
|
||||
|
||||
return TRUE;
|
||||
}
|
||||
|
||||
HRESULT
|
||||
SplitCommaDelimitedString(
|
||||
PCWSTR pszList,
|
||||
BOOL fTrimEntries,
|
||||
BOOL fRemoveEmptyEntries,
|
||||
MULTISZ * pmszList
|
||||
)
|
||||
/*++
|
||||
|
||||
Routine Description:
|
||||
|
||||
Split comma delimited string into a multisz. Additional leading empty
|
||||
entries after the first are discarded.
|
||||
|
||||
Arguments:
|
||||
|
||||
pszList - List to split up
|
||||
fTrimEntries - Whether each entry should be trimmed before added to multisz
|
||||
fRemoveEmptyEntries - Whether empty entires should be discarded
|
||||
pmszList - Filled with MULTISZ list
|
||||
|
||||
Return Value:
|
||||
|
||||
HRESULT
|
||||
|
||||
--*/
|
||||
{
|
||||
HRESULT hr = S_OK;
|
||||
|
||||
if ( pszList == NULL ||
|
||||
pmszList == NULL )
|
||||
{
|
||||
DBG_ASSERT( FALSE );
|
||||
hr = HRESULT_FROM_WIN32( ERROR_INVALID_PARAMETER );
|
||||
goto Finished;
|
||||
}
|
||||
|
||||
pmszList->Reset();
|
||||
|
||||
/*
|
||||
pszCurrent: start of the current entry which may be the comma that
|
||||
precedes the next entry if the entry is empty
|
||||
|
||||
pszNext: the comma that precedes the next entry. If
|
||||
pszCurrent == pszNext, then the entry is empty
|
||||
|
||||
pszEnd: just past the end of the current entry
|
||||
*/
|
||||
|
||||
for ( PCWSTR pszCurrent = pszList,
|
||||
pszNext = wcschr( pszCurrent, L',' )
|
||||
;
|
||||
;
|
||||
pszCurrent = pszNext + 1,
|
||||
pszNext = wcschr( pszCurrent, L',' ) )
|
||||
{
|
||||
PCWSTR pszEnd = NULL;
|
||||
|
||||
if ( pszNext != NULL )
|
||||
{
|
||||
pszEnd = pszNext;
|
||||
}
|
||||
else
|
||||
{
|
||||
pszEnd = pszCurrent + wcslen( pszCurrent );
|
||||
}
|
||||
|
||||
if ( fTrimEntries )
|
||||
{
|
||||
while ( pszCurrent < pszEnd && ISWHITE( pszCurrent[ 0 ] ) )
|
||||
{
|
||||
pszCurrent++;
|
||||
}
|
||||
|
||||
while ( pszEnd > pszCurrent && ISWHITE( pszEnd[ -1 ] ) )
|
||||
{
|
||||
pszEnd--;
|
||||
}
|
||||
}
|
||||
|
||||
if ( pszCurrent != pszEnd || !fRemoveEmptyEntries )
|
||||
{
|
||||
if ( !pmszList->Append( pszCurrent, (DWORD) ( pszEnd - pszCurrent ) ) )
|
||||
{
|
||||
hr = HRESULT_FROM_WIN32( GetLastError() );
|
||||
goto Finished;
|
||||
}
|
||||
}
|
||||
|
||||
if ( pszNext == NULL )
|
||||
{
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
Finished:
|
||||
|
||||
return hr;
|
||||
}
|
||||
|
||||
#pragma warning(default:4267)
|
||||
|
|
@ -0,0 +1,225 @@
|
|||
// Copyright (c) .NET Foundation. All rights reserved.
|
||||
// Licensed under the MIT License. See License.txt in the project root for license information.
|
||||
|
||||
#ifndef _MULTISZ_H_
|
||||
#define _MULTISZ_H_
|
||||
|
||||
#include "stringu.h"
|
||||
#include "ntassert.h"
|
||||
|
||||
/*++
|
||||
class MULTISZ:
|
||||
|
||||
Intention:
|
||||
A light-weight multi-string class supporting encapsulated string class.
|
||||
|
||||
This object is derived from BUFFER class.
|
||||
It maintains following state:
|
||||
|
||||
m_fValid - whether this object is valid -
|
||||
used only by MULTISZ() init functions
|
||||
* NYI: I need to kill this someday *
|
||||
m_cchLen - string length cached when we update the string.
|
||||
m_cStrings - number of strings.
|
||||
|
||||
Member Functions:
|
||||
There are two categories of functions:
|
||||
1) Safe Functions - which do integrity checking of state
|
||||
2) UnSafe Functions - which do not do integrity checking, but
|
||||
enable writing to the data stream freely.
|
||||
(someday this will be enabled as Safe versions without
|
||||
problem for users)
|
||||
|
||||
--*/
|
||||
class MULTISZ : public BUFFER
|
||||
{
|
||||
public:
|
||||
|
||||
MULTISZ()
|
||||
: BUFFER (),
|
||||
m_cchLen ( 0),
|
||||
m_cStrings(0)
|
||||
{ Reset(); }
|
||||
|
||||
// creates a stack version of the MULTISZ object - uses passed in stack buffer
|
||||
// MULTISZ does not free this pbInit on its own.
|
||||
MULTISZ( __in_bcount(cbInit) WCHAR * pbInit, DWORD cbInit)
|
||||
: BUFFER( (BYTE *) pbInit, cbInit),
|
||||
m_cchLen (0),
|
||||
m_cStrings(0)
|
||||
{}
|
||||
|
||||
MULTISZ( const WCHAR * pchInit )
|
||||
: BUFFER (),
|
||||
m_cchLen ( 0),
|
||||
m_cStrings(0)
|
||||
{ AuxInit(pchInit); }
|
||||
|
||||
MULTISZ( const MULTISZ & str )
|
||||
: BUFFER (),
|
||||
m_cchLen ( 0),
|
||||
m_cStrings(0)
|
||||
{ AuxInit( str.QueryStr()); }
|
||||
|
||||
// BOOL IsValid(VOID) const { return ( BUFFER::IsValid()) ; }
|
||||
//
|
||||
// Checks and returns TRUE if this string has no valid data else FALSE
|
||||
//
|
||||
BOOL IsEmpty( VOID) const { return ( *QueryStr() == L'\0'); }
|
||||
|
||||
BOOL Append( const WCHAR * pchInit ) {
|
||||
return ((pchInit != NULL) ? (AuxAppend( pchInit,
|
||||
(DWORD) (::wcslen(pchInit)) * sizeof(WCHAR)
|
||||
)) :
|
||||
TRUE);
|
||||
}
|
||||
|
||||
|
||||
BOOL Append( const WCHAR * pchInit, DWORD cchLen ) {
|
||||
return ((pchInit != NULL) ? (AuxAppend( pchInit,
|
||||
cchLen * sizeof(WCHAR))) :
|
||||
TRUE);
|
||||
}
|
||||
|
||||
BOOL Append( STRU & str )
|
||||
{ return AuxAppend( str.QueryStr(),
|
||||
(str.QueryCCH()) * sizeof(WCHAR)); }
|
||||
|
||||
// Resets the internal string to be NULL string. Buffer remains cached.
|
||||
VOID Reset( VOID)
|
||||
{ DBG_ASSERT( QueryPtr() != NULL);
|
||||
QueryStr()[0] = L'\0';
|
||||
QueryStr()[1] = L'\0';
|
||||
m_cchLen = 2;
|
||||
m_cStrings = 0;
|
||||
}
|
||||
|
||||
BOOL Copy( const WCHAR * pchInit, IN DWORD cbLen ) {
|
||||
if ( QueryPtr() ) { Reset(); }
|
||||
return ( (pchInit != NULL) ?
|
||||
AuxAppend( pchInit, cbLen, FALSE ):
|
||||
TRUE);
|
||||
}
|
||||
|
||||
BOOL Copy( const MULTISZ & str )
|
||||
{ return ( Copy(str.QueryStr(), str.QueryCB())); }
|
||||
|
||||
//
|
||||
// Returns the number of bytes in the string including the terminating
|
||||
// NULLs
|
||||
//
|
||||
UINT QueryCB( VOID ) const
|
||||
{ return ( m_cchLen * sizeof(WCHAR)); }
|
||||
|
||||
//
|
||||
// Returns # of characters in the string including the terminating NULLs
|
||||
//
|
||||
UINT QueryCCH( VOID ) const { return (m_cchLen); }
|
||||
|
||||
//
|
||||
// Returns # of strings in the multisz.
|
||||
//
|
||||
|
||||
DWORD QueryStringCount( VOID ) const { return m_cStrings; }
|
||||
|
||||
//
|
||||
// Makes a copy of the stored string in given buffer
|
||||
//
|
||||
BOOL CopyToBuffer( __out_ecount_opt(*lpcch) WCHAR * lpszBuffer, LPDWORD lpcch) const;
|
||||
|
||||
//
|
||||
// Return the string buffer
|
||||
//
|
||||
WCHAR * QueryStrA( VOID ) const { return ( QueryStr()); }
|
||||
WCHAR * QueryStr( VOID ) const { return ((WCHAR *) QueryPtr()); }
|
||||
|
||||
//
|
||||
// Makes a clone of the current string in the string pointer passed in.
|
||||
//
|
||||
BOOL
|
||||
Clone( OUT MULTISZ * pstrClone) const
|
||||
{
|
||||
return ((pstrClone == NULL) ?
|
||||
(SetLastError(ERROR_INVALID_PARAMETER), FALSE) :
|
||||
(pstrClone->Copy( *this))
|
||||
);
|
||||
} // MULTISZ::Clone()
|
||||
|
||||
//
|
||||
// Recalculates the length of *this because we've modified the buffers
|
||||
// directly
|
||||
//
|
||||
|
||||
VOID RecalcLen( VOID )
|
||||
{ m_cchLen = MULTISZ::CalcLength( QueryStr(), &m_cStrings ); }
|
||||
|
||||
//
|
||||
// Calculate total character length of a MULTI_SZ, including the
|
||||
// terminating NULLs.
|
||||
//
|
||||
|
||||
static DWORD CalcLength( const WCHAR * str,
|
||||
LPDWORD pcStrings = NULL );
|
||||
|
||||
//
|
||||
// Determine if the MULTISZ contains a specific string.
|
||||
//
|
||||
|
||||
BOOL FindString( const WCHAR * str );
|
||||
|
||||
BOOL FindString( STRU & str )
|
||||
{ return FindString( str.QueryStr() ); }
|
||||
|
||||
//
|
||||
// Determine if the MULTISZ contains a specific string - case-insensitive
|
||||
//
|
||||
|
||||
BOOL FindStringNoCase( const WCHAR * str );
|
||||
|
||||
BOOL FindStringNoCase( STRU & str )
|
||||
{ return FindStringNoCase( str.QueryStr() ); }
|
||||
|
||||
//
|
||||
// Used for scanning a multisz.
|
||||
//
|
||||
|
||||
const WCHAR * First( VOID ) const
|
||||
{ return *QueryStr() == L'\0' ? NULL : QueryStr(); }
|
||||
|
||||
const WCHAR * Next( const WCHAR * Current ) const
|
||||
{ Current += ::wcslen( Current ) + 1;
|
||||
return *Current == L'\0' ? NULL : Current; }
|
||||
|
||||
BOOL
|
||||
Equals(
|
||||
MULTISZ* pmszRhs
|
||||
);
|
||||
|
||||
private:
|
||||
|
||||
DWORD m_cchLen;
|
||||
DWORD m_cStrings;
|
||||
VOID AuxInit( const WCHAR * pInit );
|
||||
BOOL AuxAppend( const WCHAR * pInit,
|
||||
UINT cbStr, BOOL fAddSlop = TRUE );
|
||||
|
||||
};
|
||||
|
||||
//
|
||||
// Quick macro for declaring a MULTISZ that will use stack memory of <size>
|
||||
// bytes. If the buffer overflows then a heap buffer will be allocated
|
||||
//
|
||||
|
||||
#define STACK_MULTISZ( name, size ) WCHAR __ach##name[size]; \
|
||||
MULTISZ name( __ach##name, sizeof( __ach##name ))
|
||||
|
||||
HRESULT
|
||||
SplitCommaDelimitedString(
|
||||
PCWSTR pszList,
|
||||
BOOL fTrimEntries,
|
||||
BOOL fRemoveEmptyEntries,
|
||||
MULTISZ * pmszList
|
||||
);
|
||||
|
||||
#endif // !_MULTISZ_HXX_
|
||||
|
||||
|
|
@ -0,0 +1,408 @@
|
|||
// Copyright (c) .NET Foundation. All rights reserved.
|
||||
// Licensed under the MIT License. See License.txt in the project root for license information.
|
||||
|
||||
#pragma warning (disable : 4267)
|
||||
#include "precomp.h"
|
||||
#include "multisza.h"
|
||||
#include <tchar.h>
|
||||
|
||||
//
|
||||
// Private Definitions
|
||||
//
|
||||
|
||||
#define MAXULONG 4294967295
|
||||
#define ISWHITE( ch ) ((ch) == L' ' || (ch) == L'\t' || (ch) == L'\r')
|
||||
|
||||
//
|
||||
// When appending data, this is the extra amount we request to avoid
|
||||
// reallocations
|
||||
//
|
||||
#define STR_SLOP 128
|
||||
|
||||
|
||||
DWORD
|
||||
MULTISZA::CalcLength( const CHAR * str,
|
||||
LPDWORD pcStrings )
|
||||
{
|
||||
DWORD count = 0;
|
||||
DWORD total = 1;
|
||||
DWORD len;
|
||||
|
||||
while( *str ) {
|
||||
len = ::strlen( str ) + 1;
|
||||
total += len;
|
||||
str += len;
|
||||
count++;
|
||||
}
|
||||
|
||||
if( pcStrings != NULL ) {
|
||||
*pcStrings = count;
|
||||
}
|
||||
|
||||
return total;
|
||||
|
||||
} // MULTISZA::CalcLength
|
||||
|
||||
|
||||
BOOL
|
||||
MULTISZA::FindString( const CHAR * str )
|
||||
{
|
||||
|
||||
CHAR * multisz;
|
||||
|
||||
//
|
||||
// Sanity check.
|
||||
//
|
||||
|
||||
DBG_ASSERT( QueryStr() != NULL );
|
||||
DBG_ASSERT( str != NULL );
|
||||
DBG_ASSERT( *str != '\0' );
|
||||
|
||||
//
|
||||
// Scan it.
|
||||
//
|
||||
|
||||
multisz = QueryStr();
|
||||
|
||||
while( *multisz != '\0' ) {
|
||||
|
||||
if( !::strcmp( multisz, str ) ) {
|
||||
|
||||
return TRUE;
|
||||
|
||||
}
|
||||
|
||||
multisz += ::strlen( multisz ) + 1;
|
||||
|
||||
}
|
||||
|
||||
return FALSE;
|
||||
|
||||
} // MULTISZA::FindString
|
||||
|
||||
|
||||
BOOL
|
||||
MULTISZA::FindStringNoCase( const CHAR * str )
|
||||
{
|
||||
|
||||
CHAR * multisz;
|
||||
|
||||
//
|
||||
// Sanity check.
|
||||
//
|
||||
|
||||
DBG_ASSERT( QueryStr() != NULL );
|
||||
DBG_ASSERT( str != NULL );
|
||||
DBG_ASSERT( *str != '\0' );
|
||||
|
||||
//
|
||||
// Scan it.
|
||||
//
|
||||
|
||||
multisz = QueryStr();
|
||||
|
||||
while( *multisz != '\0' ) {
|
||||
|
||||
if( !_stricmp( multisz, str ) ) {
|
||||
|
||||
return TRUE;
|
||||
|
||||
}
|
||||
|
||||
multisz += strlen( multisz ) + 1;
|
||||
|
||||
}
|
||||
|
||||
return FALSE;
|
||||
|
||||
} // MULTISZA::FindStringNoCase
|
||||
|
||||
|
||||
VOID
|
||||
MULTISZA::AuxInit( const CHAR * pInit )
|
||||
{
|
||||
BOOL fRet;
|
||||
|
||||
if ( pInit )
|
||||
{
|
||||
DWORD cStrings;
|
||||
int cbCopy = CalcLength( pInit, &cStrings ) * sizeof(CHAR);
|
||||
fRet = Resize( cbCopy );
|
||||
|
||||
if ( fRet ) {
|
||||
CopyMemory( QueryPtr(), pInit, cbCopy );
|
||||
m_cchLen = (cbCopy)/sizeof(CHAR);
|
||||
m_cStrings = cStrings;
|
||||
} else {
|
||||
// BUFFER::SetValid( FALSE);
|
||||
}
|
||||
|
||||
} else {
|
||||
|
||||
Reset();
|
||||
|
||||
}
|
||||
|
||||
} // MULTISZA::AuxInit()
|
||||
|
||||
|
||||
/*******************************************************************
|
||||
|
||||
NAME: MULTISZA::AuxAppend
|
||||
|
||||
SYNOPSIS: Appends the string onto the MULTISZA.
|
||||
|
||||
ENTRY: Object to append
|
||||
********************************************************************/
|
||||
|
||||
BOOL MULTISZA::AuxAppend( const CHAR * pStr, UINT cbStr, BOOL fAddSlop )
|
||||
{
|
||||
DBG_ASSERT( pStr != NULL );
|
||||
|
||||
UINT cbThis = QueryCB();
|
||||
|
||||
if( cbThis == 2 ) {
|
||||
|
||||
//
|
||||
// It's empty, so start at the beginning.
|
||||
//
|
||||
|
||||
cbThis = 0;
|
||||
|
||||
} else {
|
||||
|
||||
//
|
||||
// It's not empty, so back up over the final terminating NULL.
|
||||
//
|
||||
|
||||
cbThis -= sizeof(CHAR);
|
||||
|
||||
}
|
||||
|
||||
//
|
||||
// Only resize when we have to. When we do resize, we tack on
|
||||
// some extra space to avoid extra reallocations.
|
||||
//
|
||||
// Note: QuerySize returns the requested size of the string buffer,
|
||||
// *not* the strlen of the buffer
|
||||
//
|
||||
|
||||
//AcIncrement( CacMultiszAppend);
|
||||
|
||||
//
|
||||
// Check for the arithmetic overflow
|
||||
//
|
||||
// ( 2 * sizeof( CHAR ) ) is for the double terminator
|
||||
//
|
||||
ULONGLONG cb64Required = (ULONGLONG)cbThis + cbStr + 2 * sizeof(CHAR);
|
||||
if ( cb64Required > MAXULONG )
|
||||
{
|
||||
SetLastError( ERROR_ARITHMETIC_OVERFLOW );
|
||||
return FALSE;
|
||||
}
|
||||
if ( QuerySize() < (DWORD) cb64Required )
|
||||
{
|
||||
ULONGLONG cb64AllocSize = cb64Required + (fAddSlop ? STR_SLOP : 0 );
|
||||
//
|
||||
// Check for the arithmetic overflow
|
||||
//
|
||||
if ( cb64AllocSize > MAXULONG )
|
||||
{
|
||||
SetLastError( ERROR_ARITHMETIC_OVERFLOW );
|
||||
return FALSE;
|
||||
}
|
||||
if ( !Resize( (DWORD) cb64AllocSize ) )
|
||||
return FALSE;
|
||||
}
|
||||
|
||||
// copy the exact string and tack on the double terminator
|
||||
memcpy( (BYTE *) QueryPtr() + cbThis,
|
||||
pStr,
|
||||
cbStr);
|
||||
*(CHAR *)((BYTE *)QueryPtr() + cbThis + cbStr) = L'\0';
|
||||
*(CHAR *)((BYTE *)QueryPtr() + cbThis + cbStr + sizeof(CHAR) ) = L'\0';
|
||||
|
||||
m_cchLen = CalcLength( (const CHAR *)QueryPtr(), &m_cStrings );
|
||||
return TRUE;
|
||||
|
||||
} // MULTISZA::AuxAppend()
|
||||
|
||||
BOOL
|
||||
MULTISZA::CopyToBuffer( __out_ecount_opt(*lpcch) CHAR * lpszBuffer, LPDWORD lpcch) const
|
||||
/*++
|
||||
Description:
|
||||
Copies the string into the CHAR buffer passed in if the buffer
|
||||
is sufficient to hold the translated string.
|
||||
If the buffer is small, the function returns small and sets *lpcch
|
||||
to contain the required number of characters.
|
||||
|
||||
Arguments:
|
||||
lpszBuffer pointer to CHAR buffer which on return contains
|
||||
the string on success.
|
||||
lpcch pointer to DWORD containing the length of the buffer.
|
||||
If *lpcch == 0 then the function returns TRUE with
|
||||
the count of characters required stored in lpcch.
|
||||
Also in this case lpszBuffer is not affected.
|
||||
Returns:
|
||||
TRUE on success.
|
||||
FALSE on failure. Use GetLastError() for further details.
|
||||
--*/
|
||||
{
|
||||
BOOL fReturn = TRUE;
|
||||
|
||||
if ( lpcch == NULL) {
|
||||
SetLastError( ERROR_INVALID_PARAMETER);
|
||||
return ( FALSE);
|
||||
}
|
||||
|
||||
register DWORD cch = QueryCCH();
|
||||
|
||||
if ( *lpcch >= cch) {
|
||||
|
||||
DBG_ASSERT( lpszBuffer);
|
||||
memcpy( lpszBuffer, QueryStr(), cch * sizeof(CHAR));
|
||||
} else {
|
||||
DBG_ASSERT( *lpcch < cch);
|
||||
SetLastError( ERROR_INSUFFICIENT_BUFFER);
|
||||
fReturn = FALSE;
|
||||
}
|
||||
|
||||
*lpcch = cch;
|
||||
|
||||
return ( fReturn);
|
||||
} // MULTISZA::CopyToBuffer()
|
||||
|
||||
BOOL
|
||||
MULTISZA::Equals(
|
||||
MULTISZA* pmszRhs
|
||||
)
|
||||
//
|
||||
// Compares this to pmszRhs, returns TRUE if equal
|
||||
//
|
||||
{
|
||||
DBG_ASSERT( NULL != pmszRhs );
|
||||
|
||||
PCSTR pszLhs = First( );
|
||||
PCSTR pszRhs = pmszRhs->First( );
|
||||
|
||||
if( m_cStrings != pmszRhs->m_cStrings )
|
||||
{
|
||||
return FALSE;
|
||||
}
|
||||
|
||||
while( NULL != pszLhs )
|
||||
{
|
||||
DBG_ASSERT( NULL != pszRhs );
|
||||
|
||||
if( 0 != strcmp( pszLhs, pszRhs ) )
|
||||
{
|
||||
return FALSE;
|
||||
}
|
||||
|
||||
pszLhs = Next( pszLhs );
|
||||
pszRhs = pmszRhs->Next( pszRhs );
|
||||
}
|
||||
|
||||
return TRUE;
|
||||
}
|
||||
|
||||
HRESULT
|
||||
SplitCommaDelimitedString(
|
||||
PCSTR pszList,
|
||||
BOOL fTrimEntries,
|
||||
BOOL fRemoveEmptyEntries,
|
||||
MULTISZA * pmszList
|
||||
)
|
||||
/*++
|
||||
|
||||
Routine Description:
|
||||
|
||||
Split comma delimited string into a MULTISZA. Additional leading empty
|
||||
entries after the first are discarded.
|
||||
|
||||
Arguments:
|
||||
|
||||
pszList - List to split up
|
||||
fTrimEntries - Whether each entry should be trimmed before added to MULTISZA
|
||||
fRemoveEmptyEntries - Whether empty entires should be discarded
|
||||
pmszList - Filled with MULTISZA list
|
||||
|
||||
Return Value:
|
||||
|
||||
HRESULT
|
||||
|
||||
--*/
|
||||
{
|
||||
HRESULT hr = S_OK;
|
||||
|
||||
if ( pszList == NULL ||
|
||||
pmszList == NULL )
|
||||
{
|
||||
DBG_ASSERT( FALSE );
|
||||
hr = HRESULT_FROM_WIN32( ERROR_INVALID_PARAMETER );
|
||||
goto Finished;
|
||||
}
|
||||
|
||||
pmszList->Reset();
|
||||
|
||||
/*
|
||||
pszCurrent: start of the current entry which may be the comma that
|
||||
precedes the next entry if the entry is empty
|
||||
|
||||
pszNext: the comma that precedes the next entry. If
|
||||
pszCurrent == pszNext, then the entry is empty
|
||||
|
||||
pszEnd: just past the end of the current entry
|
||||
*/
|
||||
|
||||
for ( PCSTR pszCurrent = pszList,
|
||||
pszNext = strchr( pszCurrent, L',' )
|
||||
;
|
||||
;
|
||||
pszCurrent = pszNext + 1,
|
||||
pszNext = strchr( pszCurrent, L',' ) )
|
||||
{
|
||||
PCSTR pszEnd = NULL;
|
||||
|
||||
if ( pszNext != NULL )
|
||||
{
|
||||
pszEnd = pszNext;
|
||||
}
|
||||
else
|
||||
{
|
||||
pszEnd = pszCurrent + strlen( pszCurrent );
|
||||
}
|
||||
|
||||
if ( fTrimEntries )
|
||||
{
|
||||
while ( pszCurrent < pszEnd && ISWHITE( pszCurrent[ 0 ] ) )
|
||||
{
|
||||
pszCurrent++;
|
||||
}
|
||||
|
||||
while ( pszEnd > pszCurrent && ISWHITE( pszEnd[ -1 ] ) )
|
||||
{
|
||||
pszEnd--;
|
||||
}
|
||||
}
|
||||
|
||||
if ( pszCurrent != pszEnd || !fRemoveEmptyEntries )
|
||||
{
|
||||
if ( !pmszList->Append( pszCurrent, (DWORD) ( pszEnd - pszCurrent ) ) )
|
||||
{
|
||||
hr = HRESULT_FROM_WIN32( GetLastError() );
|
||||
goto Finished;
|
||||
}
|
||||
}
|
||||
|
||||
if ( pszNext == NULL )
|
||||
{
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
Finished:
|
||||
|
||||
return hr;
|
||||
}
|
||||
#pragma warning(default:4267)
|
||||
|
|
@ -0,0 +1,226 @@
|
|||
// Copyright (c) .NET Foundation. All rights reserved.
|
||||
// Licensed under the MIT License. See License.txt in the project root for license information.
|
||||
|
||||
#ifndef _MULTISZA_H_
|
||||
#define _MULTISZA_H_
|
||||
|
||||
#include <Windows.h>
|
||||
#include "stringa.h"
|
||||
|
||||
|
||||
/*++
|
||||
class MULTISZ:
|
||||
|
||||
Intention:
|
||||
A light-weight multi-string class supporting encapsulated string class.
|
||||
|
||||
This object is derived from BUFFER class.
|
||||
It maintains following state:
|
||||
|
||||
m_fValid - whether this object is valid -
|
||||
used only by MULTISZ() init functions
|
||||
* NYI: I need to kill this someday *
|
||||
m_cchLen - string length cached when we update the string.
|
||||
m_cStrings - number of strings.
|
||||
|
||||
Member Functions:
|
||||
There are two categories of functions:
|
||||
1) Safe Functions - which do integrity checking of state
|
||||
2) UnSafe Functions - which do not do integrity checking, but
|
||||
enable writing to the data stream freely.
|
||||
(someday this will be enabled as Safe versions without
|
||||
problem for users)
|
||||
|
||||
--*/
|
||||
class MULTISZA : public BUFFER
|
||||
{
|
||||
public:
|
||||
|
||||
MULTISZA()
|
||||
: BUFFER (),
|
||||
m_cchLen ( 0),
|
||||
m_cStrings(0)
|
||||
{ Reset(); }
|
||||
|
||||
// creates a stack version of the MULTISZA object - uses passed in stack buffer
|
||||
// MULTISZA does not free this pbInit on its own.
|
||||
MULTISZA( __in_bcount(cbInit) CHAR * pbInit, DWORD cbInit)
|
||||
: BUFFER( (BYTE *) pbInit, cbInit),
|
||||
m_cchLen (0),
|
||||
m_cStrings(0)
|
||||
{}
|
||||
|
||||
MULTISZA( const CHAR * pchInit )
|
||||
: BUFFER (),
|
||||
m_cchLen ( 0),
|
||||
m_cStrings(0)
|
||||
{ AuxInit(pchInit); }
|
||||
|
||||
MULTISZA( const MULTISZA & str )
|
||||
: BUFFER (),
|
||||
m_cchLen ( 0),
|
||||
m_cStrings(0)
|
||||
{ AuxInit( str.QueryStr()); }
|
||||
|
||||
// BOOL IsValid(VOID) const { return ( BUFFER::IsValid()) ; }
|
||||
//
|
||||
// Checks and returns TRUE if this string has no valid data else FALSE
|
||||
//
|
||||
BOOL IsEmpty( VOID) const { return ( *QueryStr() == L'\0'); }
|
||||
|
||||
BOOL Append( const CHAR * pchInit ) {
|
||||
return ((pchInit != NULL) ? (AuxAppend( pchInit,
|
||||
(DWORD) (::strlen(pchInit)) * sizeof(CHAR)
|
||||
)) :
|
||||
TRUE);
|
||||
}
|
||||
|
||||
|
||||
BOOL Append( const CHAR * pchInit, DWORD cchLen ) {
|
||||
return ((pchInit != NULL) ? (AuxAppend( pchInit,
|
||||
cchLen * sizeof(CHAR))) :
|
||||
TRUE);
|
||||
}
|
||||
|
||||
BOOL Append( STRA & str )
|
||||
{ return AuxAppend( str.QueryStr(),
|
||||
(str.QueryCCH()) * sizeof(CHAR)); }
|
||||
|
||||
// Resets the internal string to be NULL string. Buffer remains cached.
|
||||
VOID Reset( VOID)
|
||||
{ DBG_ASSERT( QueryPtr() != NULL);
|
||||
QueryStr()[0] = L'\0';
|
||||
QueryStr()[1] = L'\0';
|
||||
m_cchLen = 2;
|
||||
m_cStrings = 0;
|
||||
}
|
||||
|
||||
BOOL Copy( const CHAR * pchInit, IN DWORD cbLen ) {
|
||||
if ( QueryPtr() ) { Reset(); }
|
||||
return ( (pchInit != NULL) ?
|
||||
AuxAppend( pchInit, cbLen, FALSE ):
|
||||
TRUE);
|
||||
}
|
||||
|
||||
BOOL Copy( const MULTISZA & str )
|
||||
{ return ( Copy(str.QueryStr(), str.QueryCB())); }
|
||||
|
||||
//
|
||||
// Returns the number of bytes in the string including the terminating
|
||||
// NULLs
|
||||
//
|
||||
UINT QueryCB( VOID ) const
|
||||
{ return ( m_cchLen * sizeof(CHAR)); }
|
||||
|
||||
//
|
||||
// Returns # of characters in the string including the terminating NULLs
|
||||
//
|
||||
UINT QueryCCH( VOID ) const { return (m_cchLen); }
|
||||
|
||||
//
|
||||
// Returns # of strings in the MULTISZA.
|
||||
//
|
||||
|
||||
DWORD QueryStringCount( VOID ) const { return m_cStrings; }
|
||||
|
||||
//
|
||||
// Makes a copy of the stored string in given buffer
|
||||
//
|
||||
BOOL CopyToBuffer( __out_ecount_opt(*lpcch) CHAR * lpszBuffer, LPDWORD lpcch) const;
|
||||
|
||||
//
|
||||
// Return the string buffer
|
||||
//
|
||||
CHAR * QueryStrA( VOID ) const { return ( QueryStr()); }
|
||||
CHAR * QueryStr( VOID ) const { return ((CHAR *) QueryPtr()); }
|
||||
|
||||
//
|
||||
// Makes a clone of the current string in the string pointer passed in.
|
||||
//
|
||||
BOOL
|
||||
Clone( OUT MULTISZA * pstrClone) const
|
||||
{
|
||||
return ((pstrClone == NULL) ?
|
||||
(SetLastError(ERROR_INVALID_PARAMETER), FALSE) :
|
||||
(pstrClone->Copy( *this))
|
||||
);
|
||||
} // MULTISZA::Clone()
|
||||
|
||||
//
|
||||
// Recalculates the length of *this because we've modified the buffers
|
||||
// directly
|
||||
//
|
||||
|
||||
VOID RecalcLen( VOID )
|
||||
{ m_cchLen = MULTISZA::CalcLength( QueryStr(), &m_cStrings ); }
|
||||
|
||||
//
|
||||
// Calculate total character length of a MULTI_SZ, including the
|
||||
// terminating NULLs.
|
||||
//
|
||||
|
||||
static DWORD CalcLength( const CHAR * str,
|
||||
LPDWORD pcStrings = NULL );
|
||||
|
||||
//
|
||||
// Determine if the MULTISZA contains a specific string.
|
||||
//
|
||||
|
||||
BOOL FindString( const CHAR * str );
|
||||
|
||||
BOOL FindString( STRA & str )
|
||||
{ return FindString( str.QueryStr() ); }
|
||||
|
||||
//
|
||||
// Determine if the MULTISZA contains a specific string - case-insensitive
|
||||
//
|
||||
|
||||
BOOL FindStringNoCase( const CHAR * str );
|
||||
|
||||
BOOL FindStringNoCase( STRA & str )
|
||||
{ return FindStringNoCase( str.QueryStr() ); }
|
||||
|
||||
//
|
||||
// Used for scanning a MULTISZA.
|
||||
//
|
||||
|
||||
const CHAR * First( VOID ) const
|
||||
{ return *QueryStr() == L'\0' ? NULL : QueryStr(); }
|
||||
|
||||
const CHAR * Next( const CHAR * Current ) const
|
||||
{ Current += ::strlen( Current ) + 1;
|
||||
return *Current == L'\0' ? NULL : Current; }
|
||||
|
||||
BOOL
|
||||
Equals(
|
||||
MULTISZA* pmszRhs
|
||||
);
|
||||
|
||||
private:
|
||||
|
||||
DWORD m_cchLen;
|
||||
DWORD m_cStrings;
|
||||
VOID AuxInit( const CHAR * pInit );
|
||||
BOOL AuxAppend( const CHAR * pInit,
|
||||
UINT cbStr, BOOL fAddSlop = TRUE );
|
||||
|
||||
};
|
||||
|
||||
//
|
||||
// Quick macro for declaring a MULTISZA that will use stack memory of <size>
|
||||
// bytes. If the buffer overflows then a heap buffer will be allocated
|
||||
//
|
||||
|
||||
#define STACK_MULTISZA( name, size ) CHAR __ach##name[size]; \
|
||||
MULTISZA name( __ach##name, sizeof( __ach##name ))
|
||||
|
||||
HRESULT
|
||||
SplitCommaDelimitedString(
|
||||
PCSTR pszList,
|
||||
BOOL fTrimEntries,
|
||||
BOOL fRemoveEmptyEntries,
|
||||
MULTISZA * pmszList
|
||||
);
|
||||
|
||||
#endif // !_MULTISZA_HXX_
|
||||
|
||||
|
|
@ -0,0 +1,32 @@
|
|||
// Copyright (c) .NET Foundation. All rights reserved.
|
||||
// Licensed under the MIT License. See License.txt in the project root for license information.
|
||||
|
||||
#pragma once
|
||||
|
||||
#ifdef _ASSERTE
|
||||
#undef _ASSERTE
|
||||
#endif
|
||||
|
||||
#ifdef ASSERT
|
||||
#undef ASSERT
|
||||
#endif
|
||||
|
||||
#if defined( DBG ) && DBG
|
||||
#define SX_ASSERT( _x ) ( (VOID)( ( ( _x ) ) ? TRUE : ( __annotation( L"Debug", L"AssertFail", L#_x ), DbgRaiseAssertionFailure(), FALSE ) ) )
|
||||
#define SX_ASSERTMSG( _m, _x ) ( (VOID)( ( ( _x ) ) ? TRUE : ( __annotation( L"Debug", L"AssertFail", L##_m ), DbgRaiseAssertionFailure(), FALSE ) ) )
|
||||
#define SX_VERIFY( _x ) SX_ASSERT( _x )
|
||||
#define _ASSERTE( _x ) SX_ASSERT( _x )
|
||||
#define ASSERT( _x ) SX_ASSERT( _x )
|
||||
#define assert( _x ) SX_ASSERT( _x )
|
||||
#define DBG_ASSERT( _x ) SX_ASSERT( _x )
|
||||
#define DBG_REQUIRE( _x ) SX_ASSERT( _x )
|
||||
#else
|
||||
#define SX_ASSERT( _x ) ( (VOID)0 )
|
||||
#define SX_ASSERTMSG( _m, _x ) ( (VOID)0 )
|
||||
#define SX_VERIFY( _x ) ( (VOID)( ( _x ) ? TRUE : FALSE ) )
|
||||
#define _ASSERTE( _x ) ( (VOID)0 )
|
||||
#define assert( _x ) ( (VOID)0 )
|
||||
#define DBG_ASSERT( _x ) ( (VOID)0 )
|
||||
#define DBG_REQUIRE( _x ) ((VOID)(_x))
|
||||
#endif
|
||||
|
||||
|
|
@ -0,0 +1,305 @@
|
|||
// Copyright (c) .NET Foundation. All rights reserved.
|
||||
// Licensed under the MIT License. See License.txt in the project root for license information.
|
||||
|
||||
#pragma once
|
||||
|
||||
template<typename T>
|
||||
class PER_CPU
|
||||
{
|
||||
public:
|
||||
|
||||
template<typename FunctionInitializer>
|
||||
inline
|
||||
static
|
||||
HRESULT
|
||||
Create(
|
||||
FunctionInitializer Initializer,
|
||||
__deref_out PER_CPU<T> ** ppInstance
|
||||
);
|
||||
|
||||
inline
|
||||
T *
|
||||
GetLocal(
|
||||
VOID
|
||||
);
|
||||
|
||||
template<typename FunctionForEach>
|
||||
inline
|
||||
VOID
|
||||
ForEach(
|
||||
FunctionForEach Function
|
||||
);
|
||||
|
||||
inline
|
||||
VOID
|
||||
Dispose(
|
||||
VOID
|
||||
);
|
||||
|
||||
private:
|
||||
|
||||
PER_CPU(
|
||||
VOID
|
||||
)
|
||||
{
|
||||
//
|
||||
// Don't perform any operation during constructor.
|
||||
// Constructor will never be called.
|
||||
//
|
||||
}
|
||||
|
||||
~PER_CPU(
|
||||
VOID
|
||||
)
|
||||
{
|
||||
//
|
||||
// Don't perform any operation during destructor.
|
||||
// Constructor will never be called.
|
||||
//
|
||||
}
|
||||
|
||||
template<typename FunctionInitializer>
|
||||
HRESULT
|
||||
Initialize(
|
||||
FunctionInitializer Initializer,
|
||||
DWORD NumberOfVariables,
|
||||
DWORD Alignment
|
||||
);
|
||||
|
||||
T *
|
||||
GetObject(
|
||||
DWORD Index
|
||||
);
|
||||
|
||||
static
|
||||
HRESULT
|
||||
GetProcessorInformation(
|
||||
__out DWORD * pCacheLineSize,
|
||||
__out DWORD * pNumberOfProcessors
|
||||
);
|
||||
|
||||
//
|
||||
// Pointer to the begining of the inlined array.
|
||||
//
|
||||
PVOID m_pVariables;
|
||||
SIZE_T m_Alignment;
|
||||
SIZE_T m_VariablesCount;
|
||||
};
|
||||
|
||||
template<typename T>
|
||||
template<typename FunctionInitializer>
|
||||
inline
|
||||
// static
|
||||
HRESULT
|
||||
PER_CPU<T>::Create(
|
||||
FunctionInitializer Initializer,
|
||||
__deref_out PER_CPU<T> ** ppInstance
|
||||
)
|
||||
{
|
||||
HRESULT hr = S_OK;
|
||||
DWORD CacheLineSize = 0;
|
||||
DWORD ObjectCacheLineSize = 0;
|
||||
DWORD NumberOfProcessors = 0;
|
||||
PER_CPU<T> * pInstance = NULL;
|
||||
|
||||
hr = GetProcessorInformation(&CacheLineSize,
|
||||
&NumberOfProcessors);
|
||||
if (FAILED(hr))
|
||||
{
|
||||
goto Finished;
|
||||
}
|
||||
|
||||
if (sizeof(T) > CacheLineSize)
|
||||
{
|
||||
//
|
||||
// Round to the next multiple of the cache line size.
|
||||
//
|
||||
ObjectCacheLineSize = (sizeof(T) + CacheLineSize-1) & (CacheLineSize-1);
|
||||
}
|
||||
else
|
||||
{
|
||||
ObjectCacheLineSize = CacheLineSize;
|
||||
}
|
||||
|
||||
//
|
||||
// Calculate the size of the PER_CPU<T> object, including the array.
|
||||
// The first cache line is for the member variables and the array
|
||||
// starts in the next cache line.
|
||||
//
|
||||
SIZE_T Size = CacheLineSize + NumberOfProcessors * ObjectCacheLineSize;
|
||||
|
||||
pInstance = (PER_CPU<T>*) _aligned_malloc(Size, CacheLineSize);
|
||||
if (pInstance == NULL)
|
||||
{
|
||||
hr = E_OUTOFMEMORY;
|
||||
goto Finished;
|
||||
}
|
||||
ZeroMemory(pInstance, Size);
|
||||
|
||||
//
|
||||
// The array start in the 2nd cache line.
|
||||
//
|
||||
pInstance->m_pVariables = reinterpret_cast<PBYTE>(pInstance) + CacheLineSize;
|
||||
|
||||
//
|
||||
// Pass a disposer for disposing initialized items in case of failure.
|
||||
//
|
||||
hr = pInstance->Initialize(Initializer,
|
||||
NumberOfProcessors,
|
||||
ObjectCacheLineSize);
|
||||
if (FAILED(hr))
|
||||
{
|
||||
goto Finished;
|
||||
}
|
||||
|
||||
*ppInstance = pInstance;
|
||||
pInstance = NULL;
|
||||
|
||||
Finished:
|
||||
|
||||
if (pInstance != NULL)
|
||||
{
|
||||
//
|
||||
// Free the instance without disposing it.
|
||||
//
|
||||
pInstance->Dispose();
|
||||
pInstance = NULL;
|
||||
}
|
||||
|
||||
return hr;
|
||||
}
|
||||
|
||||
template<typename T>
|
||||
inline
|
||||
T *
|
||||
PER_CPU<T>::GetLocal(
|
||||
VOID
|
||||
)
|
||||
{
|
||||
// Use GetCurrentProcessorNumber (up to 64 logical processors) instead of
|
||||
// GetCurrentProcessorNumberEx (more than 64 logical processors) because
|
||||
// the number of processors are not densely packed per group.
|
||||
// The idea of distributing variables per CPU is to have
|
||||
// a scalability multiplier (could be NUMA node instead).
|
||||
//
|
||||
// Make sure the index don't go beyond the array size, if that happens,
|
||||
// there won't be even distribution, but still better
|
||||
// than one single variable.
|
||||
//
|
||||
return GetObject(GetCurrentProcessorNumber());
|
||||
}
|
||||
|
||||
template<typename T>
|
||||
inline
|
||||
T *
|
||||
PER_CPU<T>::GetObject(
|
||||
DWORD Index
|
||||
)
|
||||
{
|
||||
return reinterpret_cast<T*>(static_cast<PBYTE>(m_pVariables) + Index * m_Alignment);
|
||||
}
|
||||
|
||||
template<typename T>
|
||||
template<typename FunctionForEach>
|
||||
inline
|
||||
VOID
|
||||
PER_CPU<T>::ForEach(
|
||||
FunctionForEach Function
|
||||
)
|
||||
{
|
||||
for(DWORD Index = 0; Index < m_VariablesCount; ++Index)
|
||||
{
|
||||
T * pObject = GetObject(Index);
|
||||
Function(pObject);
|
||||
}
|
||||
}
|
||||
|
||||
template<typename T>
|
||||
VOID
|
||||
PER_CPU<T>::Dispose(
|
||||
VOID
|
||||
)
|
||||
{
|
||||
_aligned_free(this);
|
||||
}
|
||||
|
||||
template<typename T>
|
||||
template<typename FunctionInitializer>
|
||||
inline
|
||||
HRESULT
|
||||
PER_CPU<T>::Initialize(
|
||||
FunctionInitializer Initializer,
|
||||
DWORD NumberOfVariables,
|
||||
DWORD Alignment
|
||||
)
|
||||
/*++
|
||||
|
||||
Routine Description:
|
||||
|
||||
Initialize each object using the initializer function.
|
||||
If initialization for any object fails, it dispose the
|
||||
objects that were successfully initialized.
|
||||
|
||||
Arguments:
|
||||
|
||||
Initializer - Function for initialize one object.
|
||||
Signature: HRESULT Func(T*)
|
||||
Dispose - Function for disposing initialized objects in case of failure.
|
||||
Signature: void Func(T*)
|
||||
NumberOfVariables - The length of the array of variables.
|
||||
Alignment - Alignment to use for avoiding false sharing.
|
||||
|
||||
Return:
|
||||
|
||||
HRESULT - E_OUTOFMEMORY
|
||||
|
||||
--*/
|
||||
{
|
||||
HRESULT hr = S_OK;
|
||||
DWORD Index = 0;
|
||||
|
||||
m_VariablesCount = NumberOfVariables;
|
||||
m_Alignment = Alignment;
|
||||
|
||||
for (; Index < m_VariablesCount; ++Index)
|
||||
{
|
||||
T * pObject = GetObject(Index);
|
||||
Initializer(pObject);
|
||||
}
|
||||
|
||||
return hr;
|
||||
}
|
||||
|
||||
template<typename T>
|
||||
// static
|
||||
HRESULT
|
||||
PER_CPU<T>::GetProcessorInformation(
|
||||
__out DWORD * pCacheLineSize,
|
||||
__out DWORD * pNumberOfProcessors
|
||||
)
|
||||
/*++
|
||||
|
||||
Routine Description:
|
||||
|
||||
Gets the CPU cache-line size for the current system.
|
||||
This information is used for avoiding CPU false sharing.
|
||||
|
||||
Arguments:
|
||||
|
||||
pCacheLineSize - The processor cache-line size.
|
||||
pNumberOfProcessors - Maximum number of processors per group.
|
||||
|
||||
Return:
|
||||
|
||||
HRESULT - E_OUTOFMEMORY
|
||||
|
||||
--*/
|
||||
{
|
||||
SYSTEM_INFO SystemInfo = { };
|
||||
|
||||
GetSystemInfo(&SystemInfo);
|
||||
*pNumberOfProcessors = SystemInfo.dwNumberOfProcessors;
|
||||
*pCacheLineSize = SYSTEM_CACHE_ALIGNMENT_SIZE;
|
||||
|
||||
return S_OK;
|
||||
}
|
||||
|
|
@ -0,0 +1,22 @@
|
|||
// Copyright (c) .NET Foundation. All rights reserved.
|
||||
// Licensed under the MIT License. See License.txt in the project root for license information.
|
||||
|
||||
#include <windows.h>
|
||||
#include <ahadmin.h>
|
||||
#pragma warning( disable:4127 )
|
||||
#include <string.h>
|
||||
#include <stdio.h>
|
||||
#include <stdlib.h>
|
||||
#include <atlcomcli.h>
|
||||
#include <strsafe.h>
|
||||
#include <intsafe.h>
|
||||
|
||||
#include "macros.h"
|
||||
#include "stringu.h"
|
||||
#include "stringa.h"
|
||||
#include "dbgutil.h"
|
||||
#include "ntassert.h"
|
||||
#include "ahutil.h"
|
||||
#include "acache.h"
|
||||
//#include "base64.hxx"
|
||||
|
||||
|
|
@ -0,0 +1,85 @@
|
|||
// Copyright (c) .NET Foundation. All rights reserved.
|
||||
// Licensed under the MIT License. See License.txt in the project root for license information.
|
||||
|
||||
#pragma once
|
||||
|
||||
#include <math.h>
|
||||
#include <stdlib.h>
|
||||
|
||||
//
|
||||
// Pre-calculated prime numbers (up to 10,049,369).
|
||||
//
|
||||
extern __declspec(selectany) const DWORD g_Primes [] = {
|
||||
3, 7, 11, 17, 23, 29, 37, 47, 59, 71, 89, 107, 131, 163, 197, 239, 293, 353, 431, 521, 631,
|
||||
761, 919, 1103, 1327, 1597, 1931, 2333, 2801, 3371, 4049, 4861, 5839, 7013, 8419, 10103,
|
||||
12143, 14591, 17519, 21023, 25229, 30293, 36353, 43627, 52361, 62851, 75431, 90523, 108631,
|
||||
130363, 156437, 187751, 225307, 270371, 324449, 389357, 467237, 560689, 672827, 807403,
|
||||
968897, 1162687, 1395263, 1674319, 2009191, 2411033, 2893249, 3471899, 4166287, 4999559,
|
||||
5999471, 7199369, 7849369, 8649369, 9249369, 10049369
|
||||
};
|
||||
|
||||
class PRIME
|
||||
{
|
||||
public:
|
||||
|
||||
static
|
||||
DWORD
|
||||
GetPrime(
|
||||
DWORD dwMinimum
|
||||
)
|
||||
{
|
||||
//
|
||||
// Try to use the precalculated numbers.
|
||||
//
|
||||
for ( DWORD Index = 0; Index < _countof( g_Primes ); Index++ )
|
||||
{
|
||||
DWORD dwCandidate = g_Primes[Index];
|
||||
if ( dwCandidate >= dwMinimum )
|
||||
{
|
||||
return dwCandidate;
|
||||
}
|
||||
}
|
||||
|
||||
//
|
||||
// Do calculation.
|
||||
//
|
||||
for ( DWORD dwCandidate = dwMinimum | 1;
|
||||
dwCandidate < MAXDWORD;
|
||||
dwCandidate += 2 )
|
||||
{
|
||||
if ( IsPrime( dwCandidate ) )
|
||||
{
|
||||
return dwCandidate;
|
||||
}
|
||||
}
|
||||
return dwMinimum;
|
||||
}
|
||||
|
||||
private:
|
||||
|
||||
static
|
||||
BOOL
|
||||
IsPrime(
|
||||
DWORD dwCandidate
|
||||
)
|
||||
{
|
||||
if ((dwCandidate & 1) == 0)
|
||||
{
|
||||
return ( dwCandidate == 2 );
|
||||
}
|
||||
|
||||
DWORD dwMax = static_cast<DWORD>(sqrt(static_cast<double>(dwCandidate)));
|
||||
|
||||
for ( DWORD Index = 3; Index <= dwMax; Index += 2 )
|
||||
{
|
||||
if ( (dwCandidate % Index) == 0 )
|
||||
{
|
||||
return FALSE;
|
||||
}
|
||||
}
|
||||
return TRUE;
|
||||
}
|
||||
|
||||
PRIME() {}
|
||||
~PRIME() {}
|
||||
};
|
||||
|
|
@ -0,0 +1,736 @@
|
|||
// Copyright (c) .NET Foundation. All rights reserved.
|
||||
// Licensed under the MIT License. See License.txt in the project root for license information.
|
||||
|
||||
# ifndef _PUDEBUG_H_
|
||||
# define _PUDEBUG_H_
|
||||
|
||||
#ifndef _NO_TRACING_
|
||||
# define _NO_TRACING_
|
||||
#endif // _NO_TRACING_
|
||||
|
||||
/************************************************************
|
||||
* Include Headers
|
||||
************************************************************/
|
||||
|
||||
# ifdef __cplusplus
|
||||
extern "C" {
|
||||
# endif // __cplusplus
|
||||
|
||||
# include <windows.h>
|
||||
|
||||
# ifndef dllexp
|
||||
# define dllexp __declspec( dllexport)
|
||||
# endif // dllexp
|
||||
|
||||
#include <specstrings.h>
|
||||
|
||||
#ifndef IN_OUT
|
||||
#define IN_OUT __inout
|
||||
#endif
|
||||
|
||||
/***********************************************************
|
||||
* Macros
|
||||
************************************************************/
|
||||
|
||||
enum PRINT_REASONS {
|
||||
PrintNone = 0x0, // Nothing to be printed
|
||||
PrintError = 0x1, // An error message
|
||||
PrintWarning = 0x2, // A warning message
|
||||
PrintLog = 0x3, // Just logging. Indicates a trace of where ...
|
||||
PrintMsg = 0x4, // Echo input message
|
||||
PrintCritical = 0x5, // Print and Exit
|
||||
PrintAssertion= 0x6 // Printing for an assertion failure
|
||||
};
|
||||
|
||||
|
||||
enum DEBUG_OUTPUT_FLAGS {
|
||||
DbgOutputNone = 0x0, // None
|
||||
DbgOutputKdb = 0x1, // Output to Kernel Debugger
|
||||
DbgOutputLogFile = 0x2, // Output to LogFile
|
||||
DbgOutputTruncate = 0x4, // Truncate Log File if necessary
|
||||
DbgOutputStderr = 0x8, // Send output to std error
|
||||
DbgOutputBackup = 0x10, // Make backup of debug file ?
|
||||
DbgOutputMemory = 0x20, // Dump to memory buffer
|
||||
DbgOutputAll = 0xFFFFFFFF // All the bits set.
|
||||
};
|
||||
|
||||
|
||||
# define MAX_LABEL_LENGTH ( 100)
|
||||
|
||||
|
||||
// The following flags are used internally to track what level of tracing we
|
||||
// are currently using. Bitmapped for extensibility.
|
||||
#define DEBUG_FLAG_ODS 0x00000001
|
||||
//#define DEBUG_FLAG_INFO 0x00000002
|
||||
//#define DEBUG_FLAG_WARN 0x00000004
|
||||
//#define DEBUG_FLAG_ERROR 0x00000008
|
||||
// The following are used internally to determine whether to log or not based
|
||||
// on what the current state is
|
||||
//#define DEBUG_FLAGS_INFO (DEBUG_FLAG_ODS | DEBUG_FLAG_INFO)
|
||||
//#define DEBUG_FLAGS_WARN (DEBUG_FLAG_ODS | DEBUG_FLAG_INFO | DEBUG_FLAG_WARN)
|
||||
//#define DEBUG_FLAGS_ERROR (DEBUG_FLAG_ODS | DEBUG_FLAG_INFO | DEBUG_FLAG_WARN | DEBUG_FLAG_ERROR)
|
||||
|
||||
#define DEBUG_FLAGS_ANY (DEBUG_FLAG_INFO | DEBUG_FLAG_WARN | DEBUG_FLAG_ERROR)
|
||||
|
||||
//
|
||||
// user of DEBUG infrastructure may choose unique variable name for DEBUG_FLAGS
|
||||
// that's specially useful for cases where DEBUG infrastructure is used within
|
||||
// static library (static library may prefer to maintain it's own DebugFlags independent
|
||||
// on the main program it links to
|
||||
//
|
||||
#ifndef DEBUG_FLAGS_VAR
|
||||
#define DEBUG_FLAGS_VAR g_dwDebugFlags
|
||||
#endif
|
||||
|
||||
extern
|
||||
#ifdef __cplusplus
|
||||
"C"
|
||||
# endif // _cplusplus
|
||||
DWORD DEBUG_FLAGS_VAR ; // Debugging Flags
|
||||
|
||||
# define DECLARE_DEBUG_VARIABLE()
|
||||
|
||||
# define SET_DEBUG_FLAGS( dwFlags) DEBUG_FLAGS_VAR = dwFlags
|
||||
# define GET_DEBUG_FLAGS() ( DEBUG_FLAGS_VAR )
|
||||
|
||||
# define LOAD_DEBUG_FLAGS_FROM_REG(hkey, dwDefault) \
|
||||
DEBUG_FLAGS_VAR = PuLoadDebugFlagsFromReg((hkey), (dwDefault))
|
||||
|
||||
# define LOAD_DEBUG_FLAGS_FROM_REG_STR(pszRegKey, dwDefault) \
|
||||
DEBUG_FLAGS_VAR = PuLoadDebugFlagsFromRegStr((pszRegKey), (dwDefault))
|
||||
|
||||
# define SAVE_DEBUG_FLAGS_IN_REG(hkey, dwDbg) \
|
||||
PuSaveDebugFlagsInReg((hkey), (dwDbg))
|
||||
|
||||
# define DEBUG_IF( arg, s) if ( DEBUG_ ## arg & GET_DEBUG_FLAGS()) { \
|
||||
s \
|
||||
} else {}
|
||||
|
||||
# define IF_DEBUG( arg) if ( DEBUG_## arg & GET_DEBUG_FLAGS())
|
||||
|
||||
|
||||
/*++
|
||||
class DEBUG_PRINTS
|
||||
|
||||
This class is responsible for printing messages to log file / kernel debugger
|
||||
|
||||
Currently the class supports only member functions for <ANSI> char.
|
||||
( not unicode-strings).
|
||||
|
||||
--*/
|
||||
|
||||
|
||||
typedef struct _DEBUG_PRINTS {
|
||||
|
||||
CHAR m_rgchLabel[MAX_LABEL_LENGTH];
|
||||
CHAR m_rgchLogFilePath[MAX_PATH];
|
||||
CHAR m_rgchLogFileName[MAX_PATH];
|
||||
HANDLE m_LogFileHandle;
|
||||
HANDLE m_StdErrHandle;
|
||||
BOOL m_fInitialized;
|
||||
BOOL m_fBreakOnAssert;
|
||||
DWORD m_dwOutputFlags;
|
||||
VOID *m_pMemoryLog;
|
||||
} DEBUG_PRINTS, FAR * LPDEBUG_PRINTS;
|
||||
|
||||
|
||||
LPDEBUG_PRINTS
|
||||
PuCreateDebugPrintsObject(
|
||||
IN const char * pszPrintLabel,
|
||||
IN DWORD dwOutputFlags);
|
||||
|
||||
//
|
||||
// frees the debug prints object and closes any file if necessary.
|
||||
// Returns NULL on success or returns pDebugPrints on failure.
|
||||
//
|
||||
LPDEBUG_PRINTS
|
||||
PuDeleteDebugPrintsObject(
|
||||
IN_OUT LPDEBUG_PRINTS pDebugPrints);
|
||||
|
||||
|
||||
VOID
|
||||
PuDbgPrint(
|
||||
IN_OUT LPDEBUG_PRINTS pDebugPrints,
|
||||
IN const char * pszFilePath,
|
||||
IN int nLineNum,
|
||||
IN const char * pszFunctionName,
|
||||
IN const char * pszFormat,
|
||||
...);
|
||||
// arglist
|
||||
VOID
|
||||
PuDbgPrintW(
|
||||
IN_OUT LPDEBUG_PRINTS pDebugPrints,
|
||||
IN const char * pszFilePath,
|
||||
IN int nLineNum,
|
||||
IN const char * pszFunctionName,
|
||||
IN const WCHAR * pszFormat,
|
||||
...); // arglist
|
||||
|
||||
// PuDbgPrintError is similar to PuDbgPrint() but allows
|
||||
// one to print error code in friendly manner
|
||||
VOID
|
||||
PuDbgPrintError(
|
||||
IN_OUT LPDEBUG_PRINTS pDebugPrints,
|
||||
IN const char * pszFilePath,
|
||||
IN int nLineNum,
|
||||
IN const char * pszFunctionName,
|
||||
IN DWORD dwError,
|
||||
IN const char * pszFormat,
|
||||
...); // arglist
|
||||
|
||||
/*++
|
||||
PuDbgDump() does not do any formatting of output.
|
||||
It just dumps the given message onto the debug destinations.
|
||||
--*/
|
||||
VOID
|
||||
PuDbgDump(
|
||||
IN_OUT LPDEBUG_PRINTS pDebugPrints,
|
||||
IN const char * pszFilePath,
|
||||
IN int nLineNum,
|
||||
IN const char * pszFunctionName,
|
||||
IN const char * pszDump
|
||||
);
|
||||
|
||||
//
|
||||
// PuDbgAssertFailed() *must* be __cdecl to properly capture the
|
||||
// thread context at the time of the failure.
|
||||
//
|
||||
|
||||
INT
|
||||
__cdecl
|
||||
PuDbgAssertFailed(
|
||||
IN_OUT LPDEBUG_PRINTS pDebugPrints,
|
||||
IN const char * pszFilePath,
|
||||
IN int nLineNum,
|
||||
IN const char * pszFunctionName,
|
||||
IN const char * pszExpression,
|
||||
IN const char * pszMessage);
|
||||
|
||||
INT
|
||||
WINAPI
|
||||
PuDbgPrintAssertFailed(
|
||||
IN_OUT LPDEBUG_PRINTS pDebugPrints,
|
||||
IN const char * pszFilePath,
|
||||
IN int nLineNum,
|
||||
IN const char * pszFunctionName,
|
||||
IN const char * pszExpression,
|
||||
IN const char * pszMessage);
|
||||
|
||||
VOID
|
||||
PuDbgCaptureContext (
|
||||
OUT PCONTEXT ContextRecord
|
||||
);
|
||||
|
||||
VOID
|
||||
PuDbgPrintCurrentTime(
|
||||
IN_OUT LPDEBUG_PRINTS pDebugPrints,
|
||||
IN const char * pszFilePath,
|
||||
IN int nLineNum,
|
||||
IN const char * pszFunctionName
|
||||
);
|
||||
|
||||
VOID
|
||||
PuSetDbgOutputFlags(
|
||||
IN_OUT LPDEBUG_PRINTS pDebugPrints,
|
||||
IN DWORD dwFlags);
|
||||
|
||||
DWORD
|
||||
PuGetDbgOutputFlags(
|
||||
IN const LPDEBUG_PRINTS pDebugPrints);
|
||||
|
||||
|
||||
//
|
||||
// Following functions return Win32 error codes.
|
||||
// NO_ERROR if success
|
||||
//
|
||||
|
||||
DWORD
|
||||
PuOpenDbgPrintFile(
|
||||
IN_OUT LPDEBUG_PRINTS pDebugPrints,
|
||||
IN const char * pszFileName,
|
||||
IN const char * pszPathForFile);
|
||||
|
||||
DWORD
|
||||
PuReOpenDbgPrintFile(
|
||||
IN_OUT LPDEBUG_PRINTS pDebugPrints);
|
||||
|
||||
DWORD
|
||||
PuCloseDbgPrintFile(
|
||||
IN_OUT LPDEBUG_PRINTS pDebugPrints);
|
||||
|
||||
DWORD
|
||||
PuOpenDbgMemoryLog(
|
||||
IN_OUT LPDEBUG_PRINTS pDebugPrints);
|
||||
|
||||
DWORD
|
||||
PuCloseDbgMemoryLog(
|
||||
IN_OUT LPDEBUG_PRINTS pDebugPrints);
|
||||
|
||||
DWORD
|
||||
PuLoadDebugFlagsFromReg(IN HKEY hkey, IN DWORD dwDefault);
|
||||
|
||||
DWORD
|
||||
PuLoadDebugFlagsFromRegStr(IN LPCSTR pszRegKey, IN DWORD dwDefault);
|
||||
|
||||
DWORD
|
||||
PuSaveDebugFlagsInReg(IN HKEY hkey, IN DWORD dwDbg);
|
||||
|
||||
|
||||
# define PuPrintToKdb( pszOutput) \
|
||||
if ( pszOutput != NULL) { \
|
||||
OutputDebugString( pszOutput); \
|
||||
} else {}
|
||||
|
||||
|
||||
|
||||
# ifdef __cplusplus
|
||||
};
|
||||
# endif // __cplusplus
|
||||
|
||||
// begin_user_unmodifiable
|
||||
|
||||
|
||||
|
||||
/***********************************************************
|
||||
* Macros
|
||||
************************************************************/
|
||||
|
||||
|
||||
extern
|
||||
#ifdef __cplusplus
|
||||
"C"
|
||||
# endif // _cplusplus
|
||||
DEBUG_PRINTS * g_pDebug; // define a global debug variable
|
||||
|
||||
# if DBG
|
||||
|
||||
// For the CHK build we want ODS enabled. For an explanation of these flags see
|
||||
// the comment just after the definition of DBG_CONTEXT
|
||||
# define DECLARE_DEBUG_PRINTS_OBJECT() \
|
||||
DEBUG_PRINTS * g_pDebug = NULL; \
|
||||
DWORD DEBUG_FLAGS_VAR = DEBUG_FLAG_ERROR;
|
||||
|
||||
#else // !DBG
|
||||
|
||||
# define DECLARE_DEBUG_PRINTS_OBJECT() \
|
||||
DEBUG_PRINTS * g_pDebug = NULL; \
|
||||
DWORD DEBUG_FLAGS_VAR = 0;
|
||||
|
||||
#endif // !DBG
|
||||
|
||||
|
||||
//
|
||||
// Call the following macro as part of your initialization for program
|
||||
// planning to use the debugging class.
|
||||
//
|
||||
/** DEBUGDEBUG
|
||||
# define CREATE_DEBUG_PRINT_OBJECT( pszLabel) \
|
||||
g_pDebug = PuCreateDebugPrintsObject( pszLabel, DEFAULT_OUTPUT_FLAGS);\
|
||||
if ( g_pDebug == NULL) { \
|
||||
OutputDebugStringA( "Unable to Create Debug Print Object \n"); \
|
||||
}
|
||||
*/
|
||||
|
||||
//
|
||||
// Call the following macro once as part of the termination of program
|
||||
// which uses the debugging class.
|
||||
//
|
||||
# define DELETE_DEBUG_PRINT_OBJECT( ) \
|
||||
g_pDebug = PuDeleteDebugPrintsObject( g_pDebug);
|
||||
|
||||
|
||||
# define VALID_DEBUG_PRINT_OBJECT() \
|
||||
( ( g_pDebug != NULL) && g_pDebug->m_fInitialized)
|
||||
|
||||
|
||||
//
|
||||
// Use the DBG_CONTEXT without any surrounding braces.
|
||||
// This is used to pass the values for global DebugPrintObject
|
||||
// and File/Line information
|
||||
//
|
||||
//# define DBG_CONTEXT g_pDebug, __FILE__, __LINE__, __FUNCTION__
|
||||
|
||||
// The 3 main tracing macros, each one corresponds to a different level of
|
||||
// tracing
|
||||
|
||||
// The 3 main tracing macros, each one corresponds to a different level of
|
||||
// tracing
|
||||
//# define DBGINFO(args) {if (DEBUG_FLAGS_VAR & DEBUG_FLAGS_INFO) { PuDbgPrint args; }}
|
||||
//# define DBGWARN(args) {if (DEBUG_FLAGS_VAR & DEBUG_FLAGS_WARN) { PuDbgPrint args; }}
|
||||
//# define DBGERROR(args) {if (DEBUG_FLAGS_VAR & DEBUG_FLAGS_ERROR) { PuDbgPrint args; }}
|
||||
|
||||
# define DBGINFOW(args) {if (DEBUG_FLAGS_VAR & DEBUG_FLAGS_INFO) { PuDbgPrintW args; }}
|
||||
# define DBGWARNW(args) {if (DEBUG_FLAGS_VAR & DEBUG_FLAGS_WARN) { PuDbgPrintW args; }}
|
||||
# define DBGERRORW(args) {if (DEBUG_FLAGS_VAR & DEBUG_FLAGS_ERROR) { PuDbgPrintW args; }}
|
||||
|
||||
|
||||
//
|
||||
// DBGPRINTF() is printing function ( much like printf) but always called
|
||||
// with the DBG_CONTEXT as follows
|
||||
// DBGPRINTF( ( DBG_CONTEXT, format-string, arguments for format list));
|
||||
//
|
||||
# define DBGPRINTF DBGINFO
|
||||
|
||||
//
|
||||
// DPERROR() is printing function ( much like printf) but always called
|
||||
// with the DBG_CONTEXT as follows
|
||||
// DPERROR( ( DBG_CONTEXT, error, format-string,
|
||||
// arguments for format list));
|
||||
//
|
||||
# define DPERROR( args) {if (DEBUG_FLAGS_VAR & DEBUG_FLAGS_ERROR) { PuDbgPrintError args; }}
|
||||
|
||||
# if DBG
|
||||
|
||||
# define DBG_CODE(s) s /* echoes code in debugging mode */
|
||||
|
||||
// The same 3 main tracing macros however in this case the macros are only compiled
|
||||
// into the CHK build. This is necessary because some tracing info used functions or
|
||||
// variables which are not compiled into the FRE build.
|
||||
# define CHKINFO(args) { PuDbgPrint args; }
|
||||
# define CHKWARN(args) { PuDbgPrint args; }
|
||||
# define CHKERROR(args) { PuDbgPrint args; }
|
||||
|
||||
# define CHKINFOW(args) { PuDbgPrintW args; }
|
||||
# define CHKWARNW(args) { PuDbgPrintW args; }
|
||||
# define CHKERRORW(args) { PuDbgPrintW args; }
|
||||
|
||||
|
||||
#ifndef DBG_ASSERT
|
||||
# ifdef _PREFAST_
|
||||
# define DBG_ASSERT(exp) ((void)0) /* Do Nothing */
|
||||
# define DBG_ASSERT_MSG(exp, pszMsg) ((void)0) /* Do Nothing */
|
||||
# define DBG_REQUIRE( exp) ((void) (exp))
|
||||
# else // !_PREFAST_
|
||||
# define DBG_ASSERT( exp ) \
|
||||
( (VOID)( ( exp ) || ( DebugBreak(), \
|
||||
PuDbgPrintAssertFailed( DBG_CONTEXT, #exp, "" ) ) ) )
|
||||
|
||||
# define DBG_ASSERT_MSG( exp, pszMsg) \
|
||||
( (VOID)( ( exp ) || ( DebugBreak(), \
|
||||
PuDbgPrintAssertFailed( DBG_CONTEXT, #exp, pszMsg ) ) ) )
|
||||
|
||||
# define DBG_REQUIRE( exp ) \
|
||||
DBG_ASSERT( exp )
|
||||
# endif // !_PREFAST_
|
||||
#endif
|
||||
|
||||
|
||||
# define DBG_LOG() PuDbgPrint( DBG_CONTEXT, "\n" )
|
||||
|
||||
# define DBG_OPEN_LOG_FILE( pszFile, pszPath ) \
|
||||
PuOpenDbgPrintFile( g_pDebug, (pszFile), (pszPath) )
|
||||
|
||||
# define DBG_CLOSE_LOG_FILE( ) \
|
||||
PuCloseDbgPrintFile( g_pDebug )
|
||||
|
||||
# define DBG_OPEN_MEMORY_LOG( ) \
|
||||
PuOpenDbgMemoryLog( g_pDebug )
|
||||
|
||||
|
||||
# define DBGDUMP( args ) PuDbgDump args
|
||||
|
||||
# define DBGPRINT_CURRENT_TIME() PuDbgPrintCurrentTime( DBG_CONTEXT )
|
||||
|
||||
# else // !DBG
|
||||
|
||||
# define DBG_CODE(s) ((void)0) /* Do Nothing */
|
||||
|
||||
# define CHKINFO(args) ((void)0) /* Do Nothing */
|
||||
# define CHKWARN(args) ((void)0) /* Do Nothing */
|
||||
# define CHKERROR(args) ((void)0) /* Do Nothing */
|
||||
|
||||
# define CHKINFOW(args) ((void)0) /* Do Nothing */
|
||||
# define CHKWARNW(args) ((void)0) /* Do Nothing */
|
||||
# define CHKERRORW(args) ((void)0) /* Do Nothing */
|
||||
|
||||
#ifndef DBG_ASSERT
|
||||
# define DBG_ASSERT(exp) ((void)0) /* Do Nothing */
|
||||
|
||||
# define DBG_ASSERT_MSG(exp, pszMsg) ((void)0) /* Do Nothing */
|
||||
|
||||
# define DBG_REQUIRE( exp) ((void) (exp))
|
||||
#endif // !DBG_ASSERT
|
||||
|
||||
# define DBGDUMP( args) ((void)0) /* Do nothing */
|
||||
|
||||
# define DBG_LOG() ((void)0) /* Do Nothing */
|
||||
|
||||
# define DBG_OPEN_LOG_FILE( pszFile, pszPath) ((void)0) /* Do Nothing */
|
||||
|
||||
# define DBG_OPEN_MEMORY_LOG() ((void)0) /* Do Nothing */
|
||||
|
||||
# define DBG_CLOSE_LOG_FILE() ((void)0) /* Do Nothing */
|
||||
|
||||
# define DBGPRINT_CURRENT_TIME() ((void)0) /* Do Nothing */
|
||||
|
||||
# endif // !DBG
|
||||
|
||||
|
||||
// end_user_unmodifiable
|
||||
|
||||
// begin_user_unmodifiable
|
||||
|
||||
|
||||
#ifdef ASSERT
|
||||
# undef ASSERT
|
||||
#endif
|
||||
|
||||
|
||||
# define ASSERT( exp) DBG_ASSERT( exp)
|
||||
|
||||
|
||||
// end_user_unmodifiable
|
||||
|
||||
// begin_user_modifiable
|
||||
|
||||
//
|
||||
// Debugging constants consist of two pieces.
|
||||
// All constants in the range 0x0 to 0x8000 are reserved
|
||||
// User extensions may include additional constants (bit flags)
|
||||
//
|
||||
|
||||
# define DEBUG_API_ENTRY 0x00000001L
|
||||
# define DEBUG_API_EXIT 0x00000002L
|
||||
# define DEBUG_INIT_CLEAN 0x00000004L
|
||||
# define DEBUG_ERROR 0x00000008L
|
||||
|
||||
// End of Reserved Range
|
||||
# define DEBUG_RESERVED 0x00000FFFL
|
||||
|
||||
// end_user_modifiable
|
||||
|
||||
|
||||
|
||||
/***********************************************************
|
||||
* Platform Type related variables and macros
|
||||
************************************************************/
|
||||
|
||||
//
|
||||
// Enum for product types
|
||||
//
|
||||
|
||||
typedef enum _PLATFORM_TYPE {
|
||||
|
||||
PtInvalid = 0, // Invalid
|
||||
PtNtWorkstation = 1, // NT Workstation
|
||||
PtNtServer = 2, // NT Server
|
||||
|
||||
} PLATFORM_TYPE;
|
||||
|
||||
//
|
||||
// IISGetPlatformType is the function used to the platform type
|
||||
//
|
||||
|
||||
extern
|
||||
#ifdef __cplusplus
|
||||
"C"
|
||||
# endif // _cplusplus
|
||||
PLATFORM_TYPE
|
||||
IISGetPlatformType(
|
||||
VOID
|
||||
);
|
||||
|
||||
//
|
||||
// External Macros
|
||||
//
|
||||
|
||||
#define InetIsNtServer( _pt ) ((_pt) == PtNtServer)
|
||||
#define InetIsNtWksta( _pt ) ((_pt) == PtNtWorkstation)
|
||||
#define InetIsValidPT(_pt) ((_pt) != PtInvalid)
|
||||
|
||||
extern
|
||||
#ifdef __cplusplus
|
||||
"C"
|
||||
# endif // _cplusplus
|
||||
PLATFORM_TYPE g_PlatformType;
|
||||
|
||||
|
||||
// Use the DECLARE_PLATFORM_TYPE macro to declare the platform type
|
||||
#define DECLARE_PLATFORM_TYPE() \
|
||||
PLATFORM_TYPE g_PlatformType = PtInvalid;
|
||||
|
||||
// Use the INITIALIZE_PLATFORM_TYPE to init the platform type
|
||||
// This should typically go inside the DLLInit or equivalent place.
|
||||
#define INITIALIZE_PLATFORM_TYPE() \
|
||||
g_PlatformType = IISGetPlatformType();
|
||||
|
||||
//
|
||||
// Additional Macros to use the Platform Type
|
||||
//
|
||||
|
||||
#define TsIsNtServer( ) InetIsNtServer(g_PlatformType)
|
||||
#define TsIsNtWksta( ) InetIsNtWksta(g_PlatformType)
|
||||
#define IISIsValidPlatform() InetIsValidPT(g_PlatformType)
|
||||
#define IISPlatformType() (g_PlatformType)
|
||||
|
||||
|
||||
/***********************************************************
|
||||
* Some utility functions for Critical Sections
|
||||
************************************************************/
|
||||
|
||||
//
|
||||
// IISSetCriticalSectionSpinCount() provides a thunk for the
|
||||
// original NT4.0sp3 API SetCriticalSectionSpinCount() for CS with Spin counts
|
||||
// Users of this function should definitely dynlink with kernel32.dll,
|
||||
// Otherwise errors will surface to a large extent
|
||||
//
|
||||
extern
|
||||
# ifdef __cplusplus
|
||||
"C"
|
||||
# endif // _cplusplus
|
||||
DWORD
|
||||
IISSetCriticalSectionSpinCount(
|
||||
LPCRITICAL_SECTION lpCriticalSection,
|
||||
DWORD dwSpinCount
|
||||
);
|
||||
|
||||
|
||||
//
|
||||
// Macro for the calls to SetCriticalSectionSpinCount()
|
||||
//
|
||||
# define SET_CRITICAL_SECTION_SPIN_COUNT( lpCS, dwSpins) \
|
||||
IISSetCriticalSectionSpinCount( (lpCS), (dwSpins))
|
||||
|
||||
//
|
||||
// IIS_DEFAULT_CS_SPIN_COUNT is the default value of spins used by
|
||||
// Critical sections defined within IIS.
|
||||
// NYI: We should have to switch the individual values based on experiments!
|
||||
// Current value is an arbitrary choice
|
||||
//
|
||||
# define IIS_DEFAULT_CS_SPIN_COUNT (1000)
|
||||
|
||||
//
|
||||
// Initializes a critical section and sets its spin count
|
||||
// to IIS_DEFAULT_CS_SPIN_COUNT. Equivalent to
|
||||
// InitializeCriticalSectionAndSpinCount(lpCS, IIS_DEFAULT_CS_SPIN_COUNT),
|
||||
// but provides a safe thunking layer for older systems that don't provide
|
||||
// this API.
|
||||
//
|
||||
extern
|
||||
# ifdef __cplusplus
|
||||
"C"
|
||||
# endif // _cplusplus
|
||||
BOOL
|
||||
IISInitializeCriticalSection(
|
||||
LPCRITICAL_SECTION lpCriticalSection
|
||||
);
|
||||
|
||||
//
|
||||
// Macro for the calls to InitializeCriticalSection()
|
||||
//
|
||||
# define INITIALIZE_CRITICAL_SECTION(lpCS) IISInitializeCriticalSection(lpCS)
|
||||
|
||||
# endif /* _DEBUG_HXX_ */
|
||||
|
||||
//
|
||||
// The following macros allow the automatic naming of certain Win32 objects.
|
||||
// See IIS\SVCS\IISRTL\WIN32OBJ.C for details on the naming convention.
|
||||
//
|
||||
// Set IIS_NAMED_WIN32_OBJECTS to a non-zero value to enable named events,
|
||||
// semaphores, and mutexes.
|
||||
//
|
||||
|
||||
#if DBG
|
||||
#define IIS_NAMED_WIN32_OBJECTS 1
|
||||
#else
|
||||
#define IIS_NAMED_WIN32_OBJECTS 0
|
||||
#endif
|
||||
|
||||
#ifdef __cplusplus
|
||||
extern "C" {
|
||||
#endif
|
||||
|
||||
HANDLE
|
||||
PuDbgCreateEvent(
|
||||
__in LPSTR FileName,
|
||||
IN ULONG LineNumber,
|
||||
__in LPSTR MemberName,
|
||||
IN PVOID Address,
|
||||
IN BOOL ManualReset,
|
||||
IN BOOL InitialState
|
||||
);
|
||||
|
||||
HANDLE
|
||||
PuDbgCreateSemaphore(
|
||||
__in LPSTR FileName,
|
||||
IN ULONG LineNumber,
|
||||
__in LPSTR MemberName,
|
||||
IN PVOID Address,
|
||||
IN LONG InitialCount,
|
||||
IN LONG MaximumCount
|
||||
);
|
||||
|
||||
HANDLE
|
||||
PuDbgCreateMutex(
|
||||
__in LPSTR FileName,
|
||||
IN ULONG LineNumber,
|
||||
__in LPSTR MemberName,
|
||||
IN PVOID Address,
|
||||
IN BOOL InitialOwner
|
||||
);
|
||||
|
||||
#ifdef __cplusplus
|
||||
} // extern "C"
|
||||
#endif
|
||||
|
||||
#if IIS_NAMED_WIN32_OBJECTS
|
||||
|
||||
#define IIS_CREATE_EVENT( membername, address, manual, state ) \
|
||||
PuDbgCreateEvent( \
|
||||
(LPSTR)__FILE__, \
|
||||
(ULONG)__LINE__, \
|
||||
(membername), \
|
||||
(PVOID)(address), \
|
||||
(manual), \
|
||||
(state) \
|
||||
)
|
||||
|
||||
#define IIS_CREATE_SEMAPHORE( membername, address, initial, maximum ) \
|
||||
PuDbgCreateSemaphore( \
|
||||
(LPSTR)__FILE__, \
|
||||
(ULONG)__LINE__, \
|
||||
(membername), \
|
||||
(PVOID)(address), \
|
||||
(initial), \
|
||||
(maximum) \
|
||||
)
|
||||
|
||||
#define IIS_CREATE_MUTEX( membername, address, initial ) \
|
||||
PuDbgCreateMutex( \
|
||||
(LPSTR)__FILE__, \
|
||||
(ULONG)__LINE__, \
|
||||
(membername), \
|
||||
(PVOID)(address), \
|
||||
(initial) \
|
||||
)
|
||||
|
||||
#else // !IIS_NAMED_WIN32_OBJECTS
|
||||
|
||||
#define IIS_CREATE_EVENT( membername, address, manual, state ) \
|
||||
CreateEventA( \
|
||||
NULL, \
|
||||
(manual), \
|
||||
(state), \
|
||||
NULL \
|
||||
)
|
||||
|
||||
#define IIS_CREATE_SEMAPHORE( membername, address, initial, maximum ) \
|
||||
CreateSemaphoreA( \
|
||||
NULL, \
|
||||
(initial), \
|
||||
(maximum), \
|
||||
NULL \
|
||||
)
|
||||
|
||||
#define IIS_CREATE_MUTEX( membername, address, initial ) \
|
||||
CreateMutexA( \
|
||||
NULL, \
|
||||
(initial), \
|
||||
NULL \
|
||||
)
|
||||
|
||||
#endif // IIS_NAMED_WIN32_OBJECTS
|
||||
|
||||
|
||||
/************************ End of File ***********************/
|
||||
|
||||
|
|
@ -0,0 +1,229 @@
|
|||
// Copyright (c) .NET Foundation. All rights reserved.
|
||||
// Licensed under the MIT License. See License.txt in the project root for license information.
|
||||
|
||||
#include <windows.h>
|
||||
#include "dbgutil.h"
|
||||
#include "pudebug.h"
|
||||
#include "reftrace.h"
|
||||
|
||||
|
||||
PTRACE_LOG
|
||||
CreateRefTraceLog(
|
||||
IN LONG LogSize,
|
||||
IN LONG ExtraBytesInHeader
|
||||
)
|
||||
/*++
|
||||
|
||||
Routine Description:
|
||||
|
||||
Creates a new (empty) ref count trace log buffer.
|
||||
|
||||
Arguments:
|
||||
|
||||
LogSize - The number of entries in the log.
|
||||
|
||||
ExtraBytesInHeader - The number of extra bytes to include in the
|
||||
log header. This is useful for adding application-specific
|
||||
data to the log.
|
||||
|
||||
Return Value:
|
||||
|
||||
PTRACE_LOG - Pointer to the newly created log if successful,
|
||||
NULL otherwise.
|
||||
|
||||
--*/
|
||||
{
|
||||
|
||||
return CreateTraceLog(
|
||||
LogSize,
|
||||
ExtraBytesInHeader,
|
||||
sizeof(REF_TRACE_LOG_ENTRY)
|
||||
);
|
||||
|
||||
} // CreateRefTraceLog
|
||||
|
||||
|
||||
VOID
|
||||
DestroyRefTraceLog(
|
||||
IN PTRACE_LOG Log
|
||||
)
|
||||
/*++
|
||||
|
||||
Routine Description:
|
||||
|
||||
Destroys a ref count trace log buffer created with CreateRefTraceLog().
|
||||
|
||||
Arguments:
|
||||
|
||||
Log - The ref count trace log buffer to destroy.
|
||||
|
||||
Return Value:
|
||||
|
||||
None.
|
||||
|
||||
--*/
|
||||
{
|
||||
|
||||
DestroyTraceLog( Log );
|
||||
|
||||
} // DestroyRefTraceLog
|
||||
|
||||
|
||||
//
|
||||
// N.B. For RtlCaptureBacktrace() to work properly, the calling function
|
||||
// *must* be __cdecl, and must have a "normal" stack frame. So, we decorate
|
||||
// WriteRefTraceLog[Ex]() with the __cdecl modifier and disable the frame
|
||||
// pointer omission (FPO) optimization.
|
||||
//
|
||||
|
||||
//#pragma optimize( "y", off ) // disable frame pointer omission (FPO)
|
||||
#pragma optimize( "", off ) // disable frame pointer omission (FPO)
|
||||
|
||||
LONG
|
||||
__cdecl
|
||||
WriteRefTraceLog(
|
||||
IN PTRACE_LOG Log,
|
||||
IN LONG NewRefCount,
|
||||
IN CONST VOID * Context
|
||||
)
|
||||
/*++
|
||||
|
||||
Routine Description:
|
||||
|
||||
Writes a new entry to the specified ref count trace log. The entry
|
||||
written contains the updated reference count and a stack backtrace
|
||||
leading up to the current caller.
|
||||
|
||||
Arguments:
|
||||
|
||||
Log - The log to write to.
|
||||
|
||||
NewRefCount - The updated reference count.
|
||||
|
||||
Context - An uninterpreted context to associate with the log entry.
|
||||
|
||||
Return Value:
|
||||
|
||||
Index of entry in log.
|
||||
|
||||
--*/
|
||||
{
|
||||
|
||||
return WriteRefTraceLogEx(
|
||||
Log,
|
||||
NewRefCount,
|
||||
Context,
|
||||
REF_TRACE_EMPTY_CONTEXT, // suppress use of optional extra contexts
|
||||
REF_TRACE_EMPTY_CONTEXT,
|
||||
REF_TRACE_EMPTY_CONTEXT
|
||||
);
|
||||
|
||||
} // WriteRefTraceLog
|
||||
|
||||
|
||||
|
||||
|
||||
LONG
|
||||
__cdecl
|
||||
WriteRefTraceLogEx(
|
||||
IN PTRACE_LOG Log,
|
||||
IN LONG NewRefCount,
|
||||
IN CONST VOID * Context,
|
||||
IN CONST VOID * Context1, // optional extra context
|
||||
IN CONST VOID * Context2, // optional extra context
|
||||
IN CONST VOID * Context3 // optional extra context
|
||||
)
|
||||
/*++
|
||||
|
||||
Routine Description:
|
||||
|
||||
Writes a new "extended" entry to the specified ref count trace log.
|
||||
The entry written contains the updated reference count, stack backtrace
|
||||
leading up to the current caller and extra context information.
|
||||
|
||||
Arguments:
|
||||
|
||||
Log - The log to write to.
|
||||
|
||||
NewRefCount - The updated reference count.
|
||||
|
||||
Context - An uninterpreted context to associate with the log entry.
|
||||
Context1 - An uninterpreted context to associate with the log entry.
|
||||
Context2 - An uninterpreted context to associate with the log entry.
|
||||
Context3 - An uninterpreted context to associate with the log entry.
|
||||
|
||||
NOTE Context1/2/3 are "optional" in that the caller may suppress
|
||||
debug display of these values by passing REF_TRACE_EMPTY_CONTEXT
|
||||
for each of them.
|
||||
|
||||
Return Value:
|
||||
|
||||
Index of entry in log.
|
||||
|
||||
--*/
|
||||
{
|
||||
|
||||
REF_TRACE_LOG_ENTRY entry;
|
||||
ULONG hash;
|
||||
DWORD cStackFramesSkipped;
|
||||
|
||||
//
|
||||
// Initialize the entry.
|
||||
//
|
||||
|
||||
RtlZeroMemory(
|
||||
&entry,
|
||||
sizeof(entry)
|
||||
);
|
||||
|
||||
//
|
||||
// Set log entry members.
|
||||
//
|
||||
|
||||
entry.NewRefCount = NewRefCount;
|
||||
entry.Context = Context;
|
||||
entry.Thread = GetCurrentThreadId();
|
||||
entry.Context1 = Context1;
|
||||
entry.Context2 = Context2;
|
||||
entry.Context3 = Context3;
|
||||
|
||||
//
|
||||
// Capture the stack backtrace. Normally, we skip two stack frames:
|
||||
// one for this routine, and one for RtlCaptureBacktrace() itself.
|
||||
// For non-Ex callers who come in via WriteRefTraceLog,
|
||||
// we skip three stack frames.
|
||||
//
|
||||
|
||||
if ( entry.Context1 == REF_TRACE_EMPTY_CONTEXT
|
||||
&& entry.Context2 == REF_TRACE_EMPTY_CONTEXT
|
||||
&& entry.Context3 == REF_TRACE_EMPTY_CONTEXT
|
||||
) {
|
||||
|
||||
cStackFramesSkipped = 2;
|
||||
|
||||
} else {
|
||||
|
||||
cStackFramesSkipped = 1;
|
||||
|
||||
}
|
||||
|
||||
RtlCaptureStackBackTrace(
|
||||
cStackFramesSkipped,
|
||||
REF_TRACE_LOG_STACK_DEPTH,
|
||||
entry.Stack,
|
||||
&hash
|
||||
);
|
||||
|
||||
//
|
||||
// Write it to the log.
|
||||
//
|
||||
|
||||
return WriteTraceLog(
|
||||
Log,
|
||||
&entry
|
||||
);
|
||||
|
||||
} // WriteRefTraceLogEx
|
||||
|
||||
#pragma optimize( "", on ) // restore frame pointer omission (FPO)
|
||||
|
||||
|
|
@ -0,0 +1,87 @@
|
|||
// Copyright (c) .NET Foundation. All rights reserved.
|
||||
// Licensed under the MIT License. See License.txt in the project root for license information.
|
||||
|
||||
#ifndef _REFTRACE_H_
|
||||
#define _REFTRACE_H_
|
||||
|
||||
|
||||
#if defined(__cplusplus)
|
||||
extern "C" {
|
||||
#endif // __cplusplus
|
||||
|
||||
#include <Windows.h>
|
||||
#include "tracelog.h"
|
||||
|
||||
//
|
||||
// This is the number of stack backtrace values captured in each
|
||||
// trace log entry. This value is chosen to make the log entry
|
||||
// exactly twelve dwords long, making it a bit easier to interpret
|
||||
// from within the debugger without the debugger extension.
|
||||
//
|
||||
|
||||
#define REF_TRACE_LOG_STACK_DEPTH 9
|
||||
|
||||
// No-op value for the Context1,2,3 parameters of WriteRefTraceLogEx
|
||||
//#define REF_TRACE_EMPTY_CONTEXT ((PVOID) -1)
|
||||
#define REF_TRACE_EMPTY_CONTEXT NULL
|
||||
|
||||
|
||||
//
|
||||
// This defines the entry written to the trace log.
|
||||
//
|
||||
|
||||
typedef struct _REF_TRACE_LOG_ENTRY {
|
||||
|
||||
LONG NewRefCount;
|
||||
CONST VOID * Context;
|
||||
CONST VOID * Context1;
|
||||
CONST VOID * Context2;
|
||||
CONST VOID * Context3;
|
||||
DWORD Thread;
|
||||
PVOID Stack[REF_TRACE_LOG_STACK_DEPTH];
|
||||
|
||||
} REF_TRACE_LOG_ENTRY, *PREF_TRACE_LOG_ENTRY;
|
||||
|
||||
|
||||
//
|
||||
// Manipulators.
|
||||
//
|
||||
|
||||
PTRACE_LOG
|
||||
CreateRefTraceLog(
|
||||
IN LONG LogSize,
|
||||
IN LONG ExtraBytesInHeader
|
||||
);
|
||||
|
||||
VOID
|
||||
DestroyRefTraceLog(
|
||||
IN PTRACE_LOG Log
|
||||
);
|
||||
|
||||
LONG
|
||||
__cdecl
|
||||
WriteRefTraceLog(
|
||||
IN PTRACE_LOG Log,
|
||||
IN LONG NewRefCount,
|
||||
IN CONST VOID * Context
|
||||
);
|
||||
|
||||
LONG
|
||||
__cdecl
|
||||
WriteRefTraceLogEx(
|
||||
IN PTRACE_LOG Log,
|
||||
IN LONG NewRefCount,
|
||||
IN CONST VOID * Context,
|
||||
IN CONST VOID * Context1,
|
||||
IN CONST VOID * Context2,
|
||||
IN CONST VOID * Context3
|
||||
);
|
||||
|
||||
|
||||
#if defined(__cplusplus)
|
||||
} // extern "C"
|
||||
#endif // __cplusplus
|
||||
|
||||
|
||||
#endif // _REFTRACE_H_
|
||||
|
||||
|
|
@ -0,0 +1,193 @@
|
|||
// Copyright (c) .NET Foundation. All rights reserved.
|
||||
// Licensed under the MIT License. See License.txt in the project root for license information.
|
||||
|
||||
#pragma once
|
||||
|
||||
#if (_WIN32_WINNT < 0x600)
|
||||
|
||||
//
|
||||
// XP implementation.
|
||||
//
|
||||
class CWSDRWLock
|
||||
{
|
||||
public:
|
||||
|
||||
CWSDRWLock()
|
||||
: m_bInited(FALSE)
|
||||
{
|
||||
}
|
||||
|
||||
~CWSDRWLock()
|
||||
{
|
||||
if (m_bInited)
|
||||
{
|
||||
DeleteCriticalSection(&m_rwLock.critsec);
|
||||
CloseHandle(m_rwLock.ReadersDoneEvent);
|
||||
}
|
||||
}
|
||||
|
||||
BOOL QueryInited() const
|
||||
{
|
||||
return m_bInited;
|
||||
}
|
||||
|
||||
HRESULT Init()
|
||||
{
|
||||
HRESULT hr = S_OK;
|
||||
|
||||
if (FALSE == m_bInited)
|
||||
{
|
||||
m_rwLock.fWriterWaiting = FALSE;
|
||||
m_rwLock.LockCount = 0;
|
||||
if ( !InitializeCriticalSectionAndSpinCount( &m_rwLock.critsec, 0 ))
|
||||
{
|
||||
DWORD dwError = GetLastError();
|
||||
hr = HRESULT_FROM_WIN32(dwError);
|
||||
return hr;
|
||||
}
|
||||
|
||||
m_rwLock.ReadersDoneEvent = CreateEvent(NULL, FALSE, FALSE, NULL);
|
||||
if( NULL == m_rwLock.ReadersDoneEvent )
|
||||
{
|
||||
DWORD dwError = GetLastError();
|
||||
hr = HRESULT_FROM_WIN32(dwError);
|
||||
DeleteCriticalSection(&m_rwLock.critsec);
|
||||
return hr;
|
||||
}
|
||||
m_bInited = TRUE;
|
||||
}
|
||||
|
||||
return hr;
|
||||
}
|
||||
|
||||
void SharedAcquire()
|
||||
{
|
||||
EnterCriticalSection(&m_rwLock.critsec);
|
||||
InterlockedIncrement(&m_rwLock.LockCount);
|
||||
LeaveCriticalSection(&m_rwLock.critsec);
|
||||
}
|
||||
|
||||
void SharedRelease()
|
||||
{
|
||||
ReleaseRWLock();
|
||||
}
|
||||
|
||||
void ExclusiveAcquire()
|
||||
{
|
||||
EnterCriticalSection( &m_rwLock.critsec );
|
||||
|
||||
m_rwLock.fWriterWaiting = TRUE;
|
||||
|
||||
// check if there are any readers active
|
||||
if ( InterlockedExchangeAdd( &m_rwLock.LockCount, 0 ) > 0 )
|
||||
{
|
||||
//
|
||||
// Wait for all the readers to get done..
|
||||
//
|
||||
WaitForSingleObject( m_rwLock.ReadersDoneEvent, INFINITE );
|
||||
}
|
||||
m_rwLock.LockCount = -1;
|
||||
}
|
||||
|
||||
void ExclusiveRelease()
|
||||
{
|
||||
ReleaseRWLock();
|
||||
}
|
||||
|
||||
private:
|
||||
|
||||
BOOL m_bInited;
|
||||
|
||||
typedef struct _RW_LOCK
|
||||
{
|
||||
BOOL fWriterWaiting; // Is a writer waiting on the lock?
|
||||
LONG LockCount;
|
||||
CRITICAL_SECTION critsec;
|
||||
HANDLE ReadersDoneEvent;
|
||||
} RW_LOCK, *PRW_LOCK;
|
||||
|
||||
RW_LOCK m_rwLock;
|
||||
|
||||
private:
|
||||
|
||||
void ReleaseRWLock()
|
||||
{
|
||||
LONG Count = InterlockedDecrement( &m_rwLock.LockCount );
|
||||
|
||||
if ( 0 <= Count )
|
||||
{
|
||||
// releasing a read lock
|
||||
if (( m_rwLock.fWriterWaiting ) && ( 0 == Count ))
|
||||
{
|
||||
SetEvent( m_rwLock.ReadersDoneEvent );
|
||||
}
|
||||
}
|
||||
else
|
||||
{
|
||||
// Releasing a write lock
|
||||
m_rwLock.LockCount = 0;
|
||||
m_rwLock.fWriterWaiting = FALSE;
|
||||
LeaveCriticalSection(&m_rwLock.critsec);
|
||||
}
|
||||
}
|
||||
};
|
||||
|
||||
#else
|
||||
|
||||
//
|
||||
// Implementation for Windows Vista or greater.
|
||||
//
|
||||
class CWSDRWLock
|
||||
{
|
||||
public:
|
||||
|
||||
CWSDRWLock()
|
||||
{
|
||||
InitializeSRWLock(&m_rwLock);
|
||||
}
|
||||
|
||||
BOOL QueryInited()
|
||||
{
|
||||
return TRUE;
|
||||
}
|
||||
|
||||
|
||||
HRESULT Init()
|
||||
{
|
||||
//
|
||||
// Method defined to keep compatibility with CWSDRWLock class for XP.
|
||||
//
|
||||
return S_OK;
|
||||
}
|
||||
|
||||
void SharedAcquire()
|
||||
{
|
||||
AcquireSRWLockShared(&m_rwLock);
|
||||
}
|
||||
|
||||
void SharedRelease()
|
||||
{
|
||||
ReleaseSRWLockShared(&m_rwLock);
|
||||
}
|
||||
|
||||
void ExclusiveAcquire()
|
||||
{
|
||||
AcquireSRWLockExclusive(&m_rwLock);
|
||||
}
|
||||
|
||||
void ExclusiveRelease()
|
||||
{
|
||||
ReleaseSRWLockExclusive(&m_rwLock);
|
||||
}
|
||||
|
||||
private:
|
||||
|
||||
SRWLOCK m_rwLock;
|
||||
};
|
||||
|
||||
#endif
|
||||
|
||||
//
|
||||
// Rename the lock class to a more clear name.
|
||||
//
|
||||
typedef CWSDRWLock READ_WRITE_LOCK;
|
||||
File diff suppressed because it is too large
Load Diff
|
|
@ -0,0 +1,515 @@
|
|||
// Copyright (c) .NET Foundation. All rights reserved.
|
||||
// Licensed under the MIT License. See License.txt in the project root for license information.
|
||||
|
||||
#pragma once
|
||||
|
||||
#include "buffer.h"
|
||||
#include "macros.h"
|
||||
#include <strsafe.h>
|
||||
|
||||
class STRA
|
||||
{
|
||||
|
||||
public:
|
||||
|
||||
STRA(
|
||||
VOID
|
||||
);
|
||||
|
||||
STRA(
|
||||
__inout_ecount(cchInit) CHAR* pbInit,
|
||||
__in DWORD cchInit
|
||||
);
|
||||
|
||||
BOOL
|
||||
IsEmpty(
|
||||
VOID
|
||||
) const;
|
||||
|
||||
BOOL
|
||||
Equals(
|
||||
__in PCSTR pszRhs,
|
||||
__in BOOL fIgnoreCase = FALSE
|
||||
) const;
|
||||
|
||||
BOOL
|
||||
Equals(
|
||||
__in const STRA * pstrRhs,
|
||||
__in BOOL fIgnoreCase = FALSE
|
||||
) const;
|
||||
|
||||
BOOL
|
||||
Equals(
|
||||
__in const STRA & strRhs,
|
||||
__in BOOL fIgnoreCase = FALSE
|
||||
) const;
|
||||
|
||||
static
|
||||
BOOL
|
||||
Equals(
|
||||
__in PCSTR pszLhs,
|
||||
__in PCSTR pszRhs,
|
||||
__in bool fIgnoreCase = false
|
||||
)
|
||||
{
|
||||
// Return FALSE if either or both strings are NULL.
|
||||
if (!pszLhs || !pszRhs) return FALSE;
|
||||
|
||||
if( fIgnoreCase )
|
||||
{
|
||||
return ( 0 == _stricmp( pszLhs, pszRhs ) );
|
||||
}
|
||||
|
||||
return ( 0 == strcmp( pszLhs, pszRhs ) );
|
||||
}
|
||||
|
||||
VOID
|
||||
Trim();
|
||||
|
||||
BOOL
|
||||
StartsWith(
|
||||
__in const STRA * pStraPrefix,
|
||||
__in bool fIgnoreCase = FALSE
|
||||
) const;
|
||||
|
||||
BOOL
|
||||
StartsWith(
|
||||
__in const STRA & straPrefix,
|
||||
__in bool fIgnoreCase = FALSE
|
||||
) const;
|
||||
|
||||
BOOL
|
||||
StartsWith(
|
||||
__in PCSTR pszPrefix,
|
||||
__in bool fIgnoreCase = FALSE
|
||||
) const;
|
||||
|
||||
BOOL
|
||||
EndsWith(
|
||||
__in const STRA * pStraSuffix,
|
||||
__in bool fIgnoreCase = FALSE
|
||||
) const;
|
||||
|
||||
BOOL
|
||||
EndsWith(
|
||||
__in const STRA & straSuffix,
|
||||
__in bool fIgnoreCase = FALSE
|
||||
) const;
|
||||
|
||||
BOOL
|
||||
EndsWith(
|
||||
__in PCSTR pszSuffix,
|
||||
__in bool fIgnoreCase = FALSE
|
||||
) const;
|
||||
|
||||
INT
|
||||
IndexOf(
|
||||
__in CHAR charValue,
|
||||
__in DWORD dwStartIndex = 0
|
||||
) const;
|
||||
|
||||
INT
|
||||
IndexOf(
|
||||
__in PCSTR pszValue,
|
||||
__in DWORD dwStartIndex = 0
|
||||
) const;
|
||||
|
||||
INT
|
||||
LastIndexOf(
|
||||
__in CHAR charValue,
|
||||
__in DWORD dwStartIndex = 0
|
||||
) const;
|
||||
|
||||
DWORD
|
||||
QueryCB(
|
||||
VOID
|
||||
) const;
|
||||
|
||||
DWORD
|
||||
QueryCCH(
|
||||
VOID
|
||||
) const;
|
||||
|
||||
DWORD
|
||||
QuerySizeCCH(
|
||||
VOID
|
||||
) const;
|
||||
|
||||
DWORD
|
||||
QuerySize(
|
||||
VOID
|
||||
) const;
|
||||
|
||||
__nullterminated
|
||||
__bcount(this->m_cchLen)
|
||||
CHAR *
|
||||
QueryStr(
|
||||
VOID
|
||||
) const;
|
||||
|
||||
VOID
|
||||
Reset(
|
||||
VOID
|
||||
);
|
||||
|
||||
HRESULT
|
||||
Resize(
|
||||
__in DWORD cchSize
|
||||
);
|
||||
|
||||
HRESULT
|
||||
SyncWithBuffer(
|
||||
VOID
|
||||
);
|
||||
|
||||
HRESULT
|
||||
Copy(
|
||||
__in PCSTR pszCopy
|
||||
);
|
||||
|
||||
HRESULT
|
||||
Copy(
|
||||
__in_ecount(cbLen)
|
||||
PCSTR pszCopy,
|
||||
__in SIZE_T cbLen
|
||||
);
|
||||
|
||||
HRESULT
|
||||
Copy(
|
||||
__in const STRA * pstrRhs
|
||||
);
|
||||
|
||||
HRESULT
|
||||
Copy(
|
||||
__in const STRA & strRhs
|
||||
);
|
||||
|
||||
HRESULT
|
||||
CopyW(
|
||||
__in PCWSTR pszCopyW
|
||||
);
|
||||
|
||||
HRESULT
|
||||
CopyW(
|
||||
__in_ecount(cchLen)
|
||||
PCWSTR pszCopyW,
|
||||
__in SIZE_T cchLen,
|
||||
__in UINT CodePage = CP_UTF8,
|
||||
__in BOOL fFailIfNoTranslation = FALSE
|
||||
)
|
||||
{
|
||||
_ASSERTE( cchLen <= MAXDWORD );
|
||||
|
||||
return AuxAppendW(
|
||||
pszCopyW,
|
||||
static_cast<DWORD>(cchLen),
|
||||
0,
|
||||
CodePage,
|
||||
fFailIfNoTranslation
|
||||
);
|
||||
}
|
||||
|
||||
HRESULT
|
||||
CopyWTruncate(
|
||||
__in PCWSTR pszCopyWTruncate
|
||||
);
|
||||
|
||||
HRESULT
|
||||
CopyWTruncate(
|
||||
__in_ecount(cchLen)
|
||||
PCWSTR pszCopyWTruncate,
|
||||
__in SIZE_T cchLen
|
||||
);
|
||||
|
||||
HRESULT
|
||||
Append(
|
||||
__in PCSTR pszAppend
|
||||
);
|
||||
|
||||
HRESULT
|
||||
Append(
|
||||
__in_ecount(cbLen)
|
||||
PCSTR pszAppend,
|
||||
__in SIZE_T cbLen
|
||||
);
|
||||
|
||||
HRESULT
|
||||
Append(
|
||||
__in const STRA * pstrRhs
|
||||
);
|
||||
|
||||
HRESULT
|
||||
Append(
|
||||
__in const STRA & strRhs
|
||||
);
|
||||
|
||||
HRESULT
|
||||
AppendW(
|
||||
__in PCWSTR pszAppendW
|
||||
)
|
||||
{
|
||||
HRESULT hr;
|
||||
size_t cchLen;
|
||||
hr = StringCchLengthW( pszAppendW,
|
||||
STRSAFE_MAX_CCH,
|
||||
&cchLen );
|
||||
if ( FAILED( hr ) )
|
||||
{
|
||||
return hr;
|
||||
}
|
||||
return AppendW( pszAppendW, cchLen );
|
||||
}
|
||||
|
||||
HRESULT
|
||||
AppendW(
|
||||
__in_ecount(cchLen)
|
||||
PCWSTR pszAppendW,
|
||||
__in SIZE_T cchLen,
|
||||
__in UINT CodePage = CP_UTF8,
|
||||
__in BOOL fFailIfNoTranslation = FALSE
|
||||
)
|
||||
{
|
||||
_ASSERTE( cchLen <= MAXDWORD );
|
||||
if ( cchLen == 0 )
|
||||
{
|
||||
return S_OK;
|
||||
}
|
||||
return AuxAppendW(
|
||||
pszAppendW,
|
||||
static_cast<DWORD>(cchLen),
|
||||
QueryCB(),
|
||||
CodePage,
|
||||
fFailIfNoTranslation
|
||||
);
|
||||
}
|
||||
|
||||
HRESULT
|
||||
AppendWTruncate(
|
||||
__in PCWSTR pszAppendWTruncate
|
||||
);
|
||||
|
||||
HRESULT
|
||||
AppendWTruncate(
|
||||
__in_ecount(cchLen)
|
||||
PCWSTR pszAppendWTruncate,
|
||||
__in SIZE_T cchLen
|
||||
);
|
||||
|
||||
HRESULT
|
||||
CopyToBuffer(
|
||||
__out_bcount(*pcb) CHAR* pszBuffer,
|
||||
__inout DWORD * pcb
|
||||
) const;
|
||||
|
||||
HRESULT
|
||||
SetLen(
|
||||
__in DWORD cchLen
|
||||
);
|
||||
|
||||
HRESULT
|
||||
SafeSnprintf(
|
||||
__in __format_string
|
||||
PCSTR pszFormatString,
|
||||
...
|
||||
);
|
||||
|
||||
HRESULT
|
||||
SafeVsnprintf(
|
||||
__in __format_string
|
||||
PCSTR pszFormatString,
|
||||
va_list argsList
|
||||
);
|
||||
|
||||
HRESULT
|
||||
Escape(
|
||||
VOID
|
||||
);
|
||||
|
||||
HRESULT
|
||||
EscapeUtf8(
|
||||
VOID
|
||||
);
|
||||
|
||||
VOID
|
||||
Unescape(
|
||||
VOID
|
||||
);
|
||||
|
||||
HRESULT
|
||||
CopyWToUTF8Unescaped(
|
||||
__in LPCWSTR cpchStr
|
||||
);
|
||||
|
||||
HRESULT
|
||||
CopyWToUTF8Unescaped(
|
||||
__in_ecount(cch)
|
||||
LPCWSTR cpchStr,
|
||||
__in DWORD cch
|
||||
);
|
||||
|
||||
HRESULT
|
||||
CopyWToUTF8Escaped(
|
||||
__in LPCWSTR cpchStr
|
||||
);
|
||||
|
||||
HRESULT
|
||||
CopyWToUTF8Escaped(
|
||||
__in_ecount(cch)
|
||||
LPCWSTR cpchStr,
|
||||
__in DWORD cch
|
||||
);
|
||||
|
||||
private:
|
||||
|
||||
//
|
||||
// Avoid C++ errors. This object should never go through a copy
|
||||
// constructor, unintended cast or assignment.
|
||||
//
|
||||
STRA( const STRA &);
|
||||
STRA & operator = (const STRA &);
|
||||
|
||||
HRESULT
|
||||
AuxAppend(
|
||||
__in_ecount(cbLen)
|
||||
LPCSTR pStr,
|
||||
__in DWORD cbLen,
|
||||
__in DWORD cbOffset
|
||||
);
|
||||
|
||||
HRESULT
|
||||
AuxAppendW(
|
||||
__in_ecount(cchAppendW)
|
||||
PCWSTR pszAppendW,
|
||||
__in DWORD cchAppendW,
|
||||
__in DWORD cbOffset,
|
||||
__in UINT CodePage,
|
||||
__in BOOL fFailIfNoTranslation
|
||||
)
|
||||
{
|
||||
DWORD dwFlags = 0;
|
||||
|
||||
if( CP_ACP == CodePage )
|
||||
{
|
||||
dwFlags = WC_NO_BEST_FIT_CHARS;
|
||||
}
|
||||
else if( fFailIfNoTranslation && CodePage == CP_UTF8 )
|
||||
{
|
||||
//
|
||||
// WC_ERR_INVALID_CHARS is only supported in Longhorn or greater.
|
||||
//
|
||||
#if defined( NTDDI_VERSION ) && NTDDI_VERSION >= NTDDI_LONGHORN
|
||||
dwFlags |= WC_ERR_INVALID_CHARS;
|
||||
#else
|
||||
UNREFERENCED_PARAMETER(fFailIfNoTranslation);
|
||||
#endif
|
||||
}
|
||||
|
||||
return AuxAppendW( pszAppendW,
|
||||
cchAppendW,
|
||||
cbOffset,
|
||||
CodePage,
|
||||
fFailIfNoTranslation,
|
||||
dwFlags );
|
||||
}
|
||||
|
||||
HRESULT
|
||||
AuxAppendW(
|
||||
__in_ecount(cchAppendW)
|
||||
PCWSTR pszAppendW,
|
||||
__in DWORD cchAppendW,
|
||||
__in DWORD cbOffset,
|
||||
__in UINT CodePage,
|
||||
__in BOOL fFailIfNoTranslation,
|
||||
__in DWORD dwFlags
|
||||
);
|
||||
|
||||
HRESULT
|
||||
AuxAppendWTruncate(
|
||||
__in_ecount(cchAppendW)
|
||||
__in PCWSTR pszAppendW,
|
||||
__in DWORD cchAppendW,
|
||||
__in DWORD cbOffset
|
||||
);
|
||||
|
||||
static
|
||||
int
|
||||
ConvertUnicodeToCodePage(
|
||||
__in_ecount(dwStringLen)
|
||||
LPCWSTR pszSrcUnicodeString,
|
||||
__inout BUFFER_T<CHAR,1> * pbufDstAnsiString,
|
||||
__in DWORD dwStringLen,
|
||||
__in UINT uCodePage
|
||||
);
|
||||
|
||||
static
|
||||
HRESULT
|
||||
ConvertUnicodeToMultiByte(
|
||||
__in_ecount(dwStringLen)
|
||||
LPCWSTR pszSrcUnicodeString,
|
||||
__in BUFFER_T<CHAR,1> * pbufDstAnsiString,
|
||||
__in DWORD dwStringLen
|
||||
);
|
||||
|
||||
static
|
||||
HRESULT
|
||||
ConvertUnicodeToUTF8(
|
||||
__in_ecount(dwStringLen)
|
||||
LPCWSTR pszSrcUnicodeString,
|
||||
__in BUFFER_T<CHAR,1> * pbufDstAnsiString,
|
||||
__in DWORD dwStringLen
|
||||
);
|
||||
|
||||
typedef bool (* PFN_F_SHOULD_ESCAPE)(BYTE ch);
|
||||
|
||||
HRESULT
|
||||
EscapeInternal(
|
||||
PFN_F_SHOULD_ESCAPE pfnFShouldEscape
|
||||
);
|
||||
|
||||
//
|
||||
// Buffer with an inline buffer of 1,
|
||||
// enough to hold null-terminating character.
|
||||
//
|
||||
BUFFER_T<CHAR,1> m_Buff;
|
||||
DWORD m_cchLen;
|
||||
};
|
||||
|
||||
inline
|
||||
HRESULT
|
||||
AppendToString(
|
||||
ULONGLONG Number,
|
||||
STRA & String
|
||||
)
|
||||
{
|
||||
// prefast complains Append requires input
|
||||
// to be null terminated, so zero initialize
|
||||
// and pass the size of the buffer minus one
|
||||
// to _ui64toa_s
|
||||
CHAR chNumber[32] = {0};
|
||||
if (_ui64toa_s(Number,
|
||||
chNumber,
|
||||
sizeof(chNumber) - sizeof(CHAR),
|
||||
10) != 0)
|
||||
{
|
||||
return E_INVALIDARG;
|
||||
}
|
||||
return String.Append(chNumber);
|
||||
}
|
||||
|
||||
template<DWORD size>
|
||||
CHAR* InitHelper(__out CHAR (&psz)[size])
|
||||
{
|
||||
psz[0] = '\0';
|
||||
return psz;
|
||||
}
|
||||
|
||||
//
|
||||
// Heap operation reduction macros
|
||||
//
|
||||
#define STACK_STRA(name, size) CHAR __ach##name[size];\
|
||||
STRA name(InitHelper(__ach##name), sizeof(__ach##name))
|
||||
|
||||
#define INLINE_STRA(name, size) CHAR __ach##name[size];\
|
||||
STRA name;
|
||||
|
||||
#define INLINE_STRA_INIT(name) name(InitHelper(__ach##name), sizeof(__ach##name))
|
||||
File diff suppressed because it is too large
Load Diff
|
|
@ -0,0 +1,427 @@
|
|||
// Copyright (c) .NET Foundation. All rights reserved.
|
||||
// Licensed under the MIT License. See License.txt in the project root for license information.
|
||||
|
||||
#pragma once
|
||||
|
||||
#include "buffer.h"
|
||||
#include <strsafe.h>
|
||||
|
||||
class STRU
|
||||
{
|
||||
|
||||
public:
|
||||
|
||||
STRU(
|
||||
VOID
|
||||
);
|
||||
|
||||
STRU(
|
||||
__inout_ecount(cchInit) WCHAR* pbInit,
|
||||
__in DWORD cchInit
|
||||
);
|
||||
|
||||
BOOL
|
||||
IsEmpty(
|
||||
VOID
|
||||
) const;
|
||||
|
||||
BOOL
|
||||
Equals(
|
||||
__in const STRU * pstrRhs,
|
||||
__in BOOL fIgnoreCase = FALSE
|
||||
) const
|
||||
{
|
||||
_ASSERTE( pstrRhs != NULL );
|
||||
return Equals( pstrRhs->QueryStr(), fIgnoreCase );
|
||||
}
|
||||
|
||||
BOOL
|
||||
Equals(
|
||||
__in const STRU & strRhs,
|
||||
__in BOOL fIgnoreCase = FALSE
|
||||
) const
|
||||
{
|
||||
return Equals( strRhs.QueryStr(), fIgnoreCase );
|
||||
}
|
||||
|
||||
BOOL
|
||||
Equals(
|
||||
__in PCWSTR pszRhs,
|
||||
__in BOOL fIgnoreCase = FALSE
|
||||
) const
|
||||
{
|
||||
_ASSERTE( NULL != pszRhs );
|
||||
if ( NULL == pszRhs )
|
||||
{
|
||||
return FALSE;
|
||||
}
|
||||
|
||||
#if defined( NTDDI_VERSION ) && NTDDI_VERSION >= NTDDI_LONGHORN
|
||||
|
||||
return ( CSTR_EQUAL == CompareStringOrdinal( QueryStr(),
|
||||
QueryCCH(),
|
||||
pszRhs,
|
||||
-1,
|
||||
fIgnoreCase ) );
|
||||
#else
|
||||
|
||||
if( fIgnoreCase )
|
||||
{
|
||||
return ( 0 == _wcsicmp( QueryStr(), pszRhs ) );
|
||||
}
|
||||
return ( 0 == wcscmp( QueryStr(), pszRhs ) );
|
||||
|
||||
#endif
|
||||
}
|
||||
|
||||
|
||||
static
|
||||
BOOL
|
||||
Equals(
|
||||
__in PCWSTR pwszLhs,
|
||||
__in PCWSTR pwszRhs,
|
||||
__in bool fIgnoreCase = false
|
||||
)
|
||||
{
|
||||
// Return FALSE if either or both strings are NULL.
|
||||
if (!pwszLhs || !pwszRhs) return FALSE;
|
||||
|
||||
//
|
||||
// This method performs a ordinal string comparison when OS is Vista or
|
||||
// greater and a culture sensitive comparison if not (XP). This is
|
||||
// consistent with the existing Equals implementation (see above).
|
||||
//
|
||||
#if defined( NTDDI_VERSION ) && NTDDI_VERSION >= NTDDI_LONGHORN
|
||||
|
||||
return ( CSTR_EQUAL == CompareStringOrdinal( pwszLhs,
|
||||
-1,
|
||||
pwszRhs,
|
||||
-1,
|
||||
fIgnoreCase ) );
|
||||
#else
|
||||
|
||||
if( fIgnoreCase )
|
||||
{
|
||||
return ( 0 == _wcsicmp( pwszLhs, pwszRhs ) );
|
||||
}
|
||||
else
|
||||
{
|
||||
return ( 0 == wcscmp( pwszLhs, pwszRhs ) );
|
||||
}
|
||||
|
||||
#endif
|
||||
}
|
||||
|
||||
VOID
|
||||
Trim();
|
||||
|
||||
BOOL
|
||||
StartsWith(
|
||||
__in const STRU * pStruPrefix,
|
||||
__in bool fIgnoreCase = FALSE
|
||||
) const
|
||||
{
|
||||
_ASSERTE( pStruPrefix != NULL );
|
||||
return StartsWith( pStruPrefix->QueryStr(), fIgnoreCase );
|
||||
}
|
||||
|
||||
BOOL
|
||||
StartsWith(
|
||||
__in const STRU & struPrefix,
|
||||
__in bool fIgnoreCase = FALSE
|
||||
) const
|
||||
{
|
||||
return StartsWith( struPrefix.QueryStr(), fIgnoreCase );
|
||||
}
|
||||
|
||||
BOOL
|
||||
StartsWith(
|
||||
__in PCWSTR pwszPrefix,
|
||||
__in bool fIgnoreCase = FALSE
|
||||
) const;
|
||||
|
||||
BOOL
|
||||
EndsWith(
|
||||
__in const STRU * pStruSuffix,
|
||||
__in bool fIgnoreCase = FALSE
|
||||
) const
|
||||
{
|
||||
_ASSERTE( pStruSuffix != NULL );
|
||||
return EndsWith( pStruSuffix->QueryStr(), fIgnoreCase );
|
||||
}
|
||||
|
||||
BOOL
|
||||
EndsWith(
|
||||
__in const STRU & struSuffix,
|
||||
__in bool fIgnoreCase = FALSE
|
||||
) const
|
||||
{
|
||||
return EndsWith( struSuffix.QueryStr(), fIgnoreCase );
|
||||
}
|
||||
|
||||
BOOL
|
||||
EndsWith(
|
||||
__in PCWSTR pwszSuffix,
|
||||
__in bool fIgnoreCase = FALSE
|
||||
) const;
|
||||
|
||||
INT
|
||||
IndexOf(
|
||||
__in WCHAR charValue,
|
||||
__in DWORD dwStartIndex = 0
|
||||
) const;
|
||||
|
||||
INT
|
||||
IndexOf(
|
||||
__in PCWSTR pwszValue,
|
||||
__in DWORD dwStartIndex = 0
|
||||
) const;
|
||||
|
||||
INT
|
||||
LastIndexOf(
|
||||
__in WCHAR charValue,
|
||||
__in DWORD dwStartIndex = 0
|
||||
) const;
|
||||
|
||||
DWORD
|
||||
QueryCB(
|
||||
VOID
|
||||
) const;
|
||||
|
||||
DWORD
|
||||
QueryCCH(
|
||||
VOID
|
||||
) const;
|
||||
|
||||
DWORD
|
||||
QuerySizeCCH(
|
||||
VOID
|
||||
) const;
|
||||
|
||||
__nullterminated
|
||||
__ecount(this->m_cchLen)
|
||||
WCHAR*
|
||||
QueryStr(
|
||||
VOID
|
||||
) const;
|
||||
|
||||
VOID
|
||||
Reset(
|
||||
VOID
|
||||
);
|
||||
|
||||
HRESULT
|
||||
Resize(
|
||||
DWORD cchSize
|
||||
);
|
||||
|
||||
HRESULT
|
||||
SyncWithBuffer(
|
||||
VOID
|
||||
);
|
||||
|
||||
template<size_t size>
|
||||
HRESULT
|
||||
Copy(
|
||||
__in PCWSTR const (&rgpszStrings)[size]
|
||||
)
|
||||
//
|
||||
// Copies an array of strings declared as stack array. For example:
|
||||
//
|
||||
// LPCWSTR rgExample[] { L"one", L"two" };
|
||||
// hr = str.Copy( rgExample );
|
||||
//
|
||||
{
|
||||
Reset();
|
||||
|
||||
return AuxAppend( rgpszStrings, _countof( rgpszStrings ) );
|
||||
}
|
||||
|
||||
HRESULT
|
||||
Copy(
|
||||
__in PCWSTR pszCopy
|
||||
);
|
||||
|
||||
HRESULT
|
||||
Copy(
|
||||
__in_ecount(cchLen)
|
||||
PCWSTR pszCopy,
|
||||
SIZE_T cchLen
|
||||
);
|
||||
|
||||
HRESULT
|
||||
Copy(
|
||||
__in const STRU * pstrRhs
|
||||
);
|
||||
|
||||
HRESULT
|
||||
Copy(
|
||||
__in const STRU & str
|
||||
);
|
||||
|
||||
HRESULT
|
||||
CopyAndExpandEnvironmentStrings(
|
||||
__in PCWSTR pszSource
|
||||
);
|
||||
|
||||
HRESULT
|
||||
CopyA(
|
||||
__in PCSTR pszCopyA
|
||||
);
|
||||
|
||||
HRESULT
|
||||
CopyA(
|
||||
__in_bcount(cchLen)
|
||||
PCSTR pszCopyA,
|
||||
SIZE_T cchLen,
|
||||
UINT CodePage = CP_UTF8
|
||||
);
|
||||
|
||||
template<size_t size>
|
||||
HRESULT
|
||||
Append(
|
||||
__in PCWSTR const (&rgpszStrings)[size]
|
||||
)
|
||||
//
|
||||
// Appends an array of strings declared as stack array. For example:
|
||||
//
|
||||
// LPCWSTR rgExample[] { L"one", L"two" };
|
||||
// hr = str.Append( rgExample );
|
||||
//
|
||||
{
|
||||
return AuxAppend( rgpszStrings, _countof( rgpszStrings ) );
|
||||
}
|
||||
|
||||
HRESULT
|
||||
Append(
|
||||
__in PCWSTR pszAppend
|
||||
);
|
||||
|
||||
HRESULT
|
||||
Append(
|
||||
__in_ecount(cchLen)
|
||||
PCWSTR pszAppend,
|
||||
SIZE_T cchLen
|
||||
);
|
||||
|
||||
HRESULT
|
||||
Append(
|
||||
__in const STRU * pstrRhs
|
||||
);
|
||||
|
||||
HRESULT
|
||||
Append(
|
||||
__in const STRU & strRhs
|
||||
);
|
||||
|
||||
HRESULT
|
||||
AppendA(
|
||||
__in PCSTR pszAppendA
|
||||
);
|
||||
|
||||
HRESULT
|
||||
AppendA(
|
||||
__in_bcount(cchLen)
|
||||
PCSTR pszAppendA,
|
||||
SIZE_T cchLen,
|
||||
UINT CodePage = CP_UTF8
|
||||
);
|
||||
|
||||
HRESULT
|
||||
CopyToBuffer(
|
||||
__out_bcount(*pcb) WCHAR* pszBuffer,
|
||||
PDWORD pcb
|
||||
) const;
|
||||
|
||||
HRESULT
|
||||
SetLen(
|
||||
__in DWORD cchLen
|
||||
);
|
||||
|
||||
HRESULT
|
||||
SafeSnwprintf(
|
||||
__in PCWSTR pwszFormatString,
|
||||
...
|
||||
);
|
||||
|
||||
HRESULT
|
||||
SafeVsnwprintf(
|
||||
__in PCWSTR pwszFormatString,
|
||||
va_list argsList
|
||||
);
|
||||
|
||||
static
|
||||
HRESULT ExpandEnvironmentVariables(
|
||||
__in PCWSTR pszString,
|
||||
__out STRU * pstrExpandedString
|
||||
);
|
||||
|
||||
private:
|
||||
|
||||
//
|
||||
// Avoid C++ errors. This object should never go through a copy
|
||||
// constructor, unintended cast or assignment.
|
||||
//
|
||||
STRU( const STRU & );
|
||||
STRU & operator = ( const STRU & );
|
||||
|
||||
HRESULT
|
||||
AuxAppend(
|
||||
__in_ecount(cNumStrings)
|
||||
PCWSTR const rgpszStrings[],
|
||||
SIZE_T cNumStrings
|
||||
);
|
||||
|
||||
HRESULT
|
||||
AuxAppend(
|
||||
__in_bcount(cbStr)
|
||||
const WCHAR* pStr,
|
||||
SIZE_T cbStr,
|
||||
DWORD cbOffset
|
||||
);
|
||||
|
||||
HRESULT
|
||||
AuxAppendA(
|
||||
__in_bcount(cbStr)
|
||||
const CHAR* pStr,
|
||||
SIZE_T cbStr,
|
||||
DWORD cbOffset,
|
||||
UINT CodePage
|
||||
);
|
||||
|
||||
//
|
||||
// Buffer with an inline buffer of 1,
|
||||
// enough to hold null-terminating character.
|
||||
//
|
||||
BUFFER_T<WCHAR,1> m_Buff;
|
||||
DWORD m_cchLen;
|
||||
};
|
||||
|
||||
//
|
||||
// Helps to initialize an external buffer before
|
||||
// constructing the STRU object.
|
||||
//
|
||||
template<DWORD size>
|
||||
WCHAR* InitHelper(__out WCHAR (&psz)[size])
|
||||
{
|
||||
psz[0] = L'\0';
|
||||
return psz;
|
||||
}
|
||||
|
||||
//
|
||||
// Heap operation reduction macros
|
||||
//
|
||||
#define STACK_STRU(name, size) WCHAR __ach##name[size];\
|
||||
STRU name(InitHelper(__ach##name), sizeof(__ach##name)/sizeof(*__ach##name))
|
||||
|
||||
#define INLINE_STRU(name, size) WCHAR __ach##name[size];\
|
||||
STRU name;
|
||||
|
||||
#define INLINE_STRU_INIT(name) name(InitHelper(__ach##name), sizeof(__ach##name)/sizeof(*__ach##name))
|
||||
|
||||
|
||||
HRESULT
|
||||
MakePathCanonicalizationProof(
|
||||
IN PCWSTR pszName,
|
||||
OUT STRU * pstrPath
|
||||
);
|
||||
|
|
@ -0,0 +1,235 @@
|
|||
// Copyright (c) .NET Foundation. All rights reserved.
|
||||
// Licensed under the MIT License. See License.txt in the project root for license information.
|
||||
|
||||
#include <windows.h>
|
||||
#include "pudebug.h"
|
||||
#include "tracelog.h"
|
||||
#include <intsafe.h>
|
||||
|
||||
|
||||
#define ALLOC_MEM(cb) (PVOID)LocalAlloc( LPTR, (cb) )
|
||||
#define FREE_MEM(ptr) (VOID)LocalFree( (HLOCAL)(ptr) )
|
||||
|
||||
|
||||
|
||||
PTRACE_LOG
|
||||
CreateTraceLog(
|
||||
IN LONG LogSize,
|
||||
IN LONG ExtraBytesInHeader,
|
||||
IN LONG EntrySize
|
||||
)
|
||||
/*++
|
||||
|
||||
Routine Description:
|
||||
|
||||
Creates a new (empty) trace log buffer.
|
||||
|
||||
Arguments:
|
||||
|
||||
LogSize - The number of entries in the log.
|
||||
|
||||
ExtraBytesInHeader - The number of extra bytes to include in the
|
||||
log header. This is useful for adding application-specific
|
||||
data to the log.
|
||||
|
||||
EntrySize - The size (in bytes) of each entry.
|
||||
|
||||
Return Value:
|
||||
|
||||
PTRACE_LOG - Pointer to the newly created log if successful,
|
||||
NULL otherwise.
|
||||
|
||||
--*/
|
||||
{
|
||||
|
||||
ULONG ulTotalSize = 0;
|
||||
ULONG ulLogSize = 0;
|
||||
ULONG ulEntrySize = 0;
|
||||
ULONG ulTmpResult = 0;
|
||||
ULONG ulExtraBytesInHeader = 0;
|
||||
PTRACE_LOG log = NULL;
|
||||
HRESULT hr = S_OK;
|
||||
|
||||
//
|
||||
// Sanity check the parameters.
|
||||
//
|
||||
|
||||
//DBG_ASSERT( LogSize > 0 );
|
||||
//DBG_ASSERT( EntrySize > 0 );
|
||||
//DBG_ASSERT( ExtraBytesInHeader >= 0 );
|
||||
//DBG_ASSERT( ( EntrySize & 3 ) == 0 );
|
||||
|
||||
//
|
||||
// converting to unsigned long. Since all these values are positive
|
||||
// so its safe to cast them to their unsigned equivalent directly.
|
||||
//
|
||||
ulLogSize = (ULONG) LogSize;
|
||||
ulEntrySize = (ULONG) EntrySize;
|
||||
ulExtraBytesInHeader = (ULONG) ExtraBytesInHeader;
|
||||
|
||||
//
|
||||
// Check if the multiplication operation will overflow a LONG
|
||||
// ulTotalSize = LogSize * EntrySize;
|
||||
//
|
||||
hr = ULongMult( ulLogSize, ulEntrySize, &ulTotalSize );
|
||||
if ( FAILED(hr) )
|
||||
{
|
||||
SetLastError( ERROR_ARITHMETIC_OVERFLOW );
|
||||
return NULL;
|
||||
}
|
||||
|
||||
//
|
||||
// check for overflow in addition operation.
|
||||
// ulTmpResult = sizeof(TRACE_LOG) + ulExtraBytesInHeader
|
||||
//
|
||||
hr = ULongAdd( (ULONG) sizeof(TRACE_LOG), ulExtraBytesInHeader, &ulTmpResult );
|
||||
if ( FAILED(hr) )
|
||||
{
|
||||
SetLastError( ERROR_ARITHMETIC_OVERFLOW );
|
||||
return NULL;
|
||||
}
|
||||
|
||||
//
|
||||
// check for overflow in addition operation.
|
||||
// ulTotalSize = ulTotalSize + ulTmpResult;
|
||||
//
|
||||
hr = ULongAdd( ulTmpResult, ulTotalSize, &ulTotalSize );
|
||||
if ( FAILED(hr) )
|
||||
{
|
||||
SetLastError( ERROR_ARITHMETIC_OVERFLOW );
|
||||
return NULL;
|
||||
}
|
||||
|
||||
if ( ulTotalSize > (ULONG) 0x7FFFFFFF )
|
||||
{
|
||||
SetLastError( ERROR_ARITHMETIC_OVERFLOW );
|
||||
return NULL;
|
||||
}
|
||||
|
||||
//
|
||||
// Allocate & initialize the log structure.
|
||||
//
|
||||
|
||||
log = (PTRACE_LOG)ALLOC_MEM( ulTotalSize );
|
||||
|
||||
//
|
||||
// Initialize it.
|
||||
//
|
||||
|
||||
if( log != NULL ) {
|
||||
|
||||
RtlZeroMemory( log, ulTotalSize );
|
||||
|
||||
log->Signature = TRACE_LOG_SIGNATURE;
|
||||
log->LogSize = LogSize;
|
||||
log->NextEntry = -1;
|
||||
log->EntrySize = EntrySize;
|
||||
log->LogBuffer = (PUCHAR)( log + 1 ) + ExtraBytesInHeader;
|
||||
}
|
||||
|
||||
return log;
|
||||
|
||||
} // CreateTraceLog
|
||||
|
||||
|
||||
VOID
|
||||
DestroyTraceLog(
|
||||
IN PTRACE_LOG Log
|
||||
)
|
||||
/*++
|
||||
|
||||
Routine Description:
|
||||
|
||||
Destroys a trace log buffer created with CreateTraceLog().
|
||||
|
||||
Arguments:
|
||||
|
||||
Log - The trace log buffer to destroy.
|
||||
|
||||
Return Value:
|
||||
|
||||
None.
|
||||
|
||||
--*/
|
||||
{
|
||||
if ( Log != NULL ) {
|
||||
//DBG_ASSERT( Log->Signature == TRACE_LOG_SIGNATURE );
|
||||
|
||||
Log->Signature = TRACE_LOG_SIGNATURE_X;
|
||||
FREE_MEM( Log );
|
||||
}
|
||||
|
||||
} // DestroyTraceLog
|
||||
|
||||
|
||||
LONG
|
||||
WriteTraceLog(
|
||||
IN PTRACE_LOG Log,
|
||||
IN PVOID Entry
|
||||
)
|
||||
/*++
|
||||
|
||||
Routine Description:
|
||||
|
||||
Writes a new entry to the specified trace log.
|
||||
|
||||
Arguments:
|
||||
|
||||
Log - The log to write to.
|
||||
|
||||
Entry - Pointer to the data to write. This buffer is assumed to be
|
||||
Log->EntrySize bytes long.
|
||||
|
||||
Return Value:
|
||||
|
||||
Index of entry in log. This is useful for correlating the output
|
||||
of !inetdbg.ref to a particular point in the output debug stream
|
||||
|
||||
--*/
|
||||
{
|
||||
|
||||
PUCHAR target;
|
||||
ULONG index;
|
||||
|
||||
//DBG_ASSERT( Log != NULL );
|
||||
//DBG_ASSERT( Log->Signature == TRACE_LOG_SIGNATURE );
|
||||
//DBG_ASSERT( Entry != NULL );
|
||||
|
||||
//
|
||||
// Find the next slot, copy the entry to the slot.
|
||||
//
|
||||
|
||||
index = ( (ULONG) InterlockedIncrement( &Log->NextEntry ) ) % (ULONG) Log->LogSize;
|
||||
|
||||
//DBG_ASSERT( index < (ULONG) Log->LogSize );
|
||||
|
||||
target = Log->LogBuffer + ( index * Log->EntrySize );
|
||||
|
||||
RtlCopyMemory(
|
||||
target,
|
||||
Entry,
|
||||
Log->EntrySize
|
||||
);
|
||||
|
||||
return index;
|
||||
} // WriteTraceLog
|
||||
|
||||
|
||||
VOID
|
||||
ResetTraceLog(
|
||||
IN PTRACE_LOG Log
|
||||
)
|
||||
{
|
||||
|
||||
//DBG_ASSERT( Log != NULL );
|
||||
//DBG_ASSERT( Log->Signature == TRACE_LOG_SIGNATURE );
|
||||
|
||||
RtlZeroMemory(
|
||||
( Log + 1 ),
|
||||
Log->LogSize * Log->EntrySize
|
||||
);
|
||||
|
||||
Log->NextEntry = -1;
|
||||
|
||||
} // ResetTraceLog
|
||||
|
||||
|
|
@ -0,0 +1,105 @@
|
|||
// Copyright (c) .NET Foundation. All rights reserved.
|
||||
// Licensed under the MIT License. See License.txt in the project root for license information.
|
||||
|
||||
#ifndef _TRACELOG_H_
|
||||
#define _TRACELOG_H_
|
||||
|
||||
|
||||
#if defined(__cplusplus)
|
||||
extern "C" {
|
||||
#endif // __cplusplus
|
||||
|
||||
|
||||
typedef struct _TRACE_LOG {
|
||||
|
||||
//
|
||||
// Signature.
|
||||
//
|
||||
|
||||
LONG Signature;
|
||||
|
||||
//
|
||||
// The total number of entries available in the log.
|
||||
//
|
||||
|
||||
LONG LogSize;
|
||||
|
||||
//
|
||||
// The index of the next entry to use.
|
||||
//
|
||||
|
||||
LONG NextEntry;
|
||||
|
||||
//
|
||||
// The byte size of each entry.
|
||||
//
|
||||
|
||||
LONG EntrySize;
|
||||
|
||||
//
|
||||
// Pointer to the start of the circular buffer.
|
||||
//
|
||||
|
||||
PUCHAR LogBuffer;
|
||||
|
||||
//
|
||||
// The extra header bytes and actual log entries go here.
|
||||
//
|
||||
// BYTE ExtraHeaderBytes[ExtraBytesInHeader];
|
||||
// BYTE Entries[LogSize][EntrySize];
|
||||
//
|
||||
|
||||
} TRACE_LOG, *PTRACE_LOG;
|
||||
|
||||
|
||||
//
|
||||
// Log header signature.
|
||||
//
|
||||
|
||||
#define TRACE_LOG_SIGNATURE ((DWORD)'gOlT')
|
||||
#define TRACE_LOG_SIGNATURE_X ((DWORD)'golX')
|
||||
|
||||
|
||||
//
|
||||
// This macro maps a TRACE_LOG pointer to a pointer to the 'extra'
|
||||
// data associated with the log.
|
||||
//
|
||||
|
||||
#define TRACE_LOG_TO_EXTRA_DATA(log) (PVOID)( (log) + 1 )
|
||||
|
||||
|
||||
//
|
||||
// Manipulators.
|
||||
//
|
||||
|
||||
PTRACE_LOG
|
||||
CreateTraceLog(
|
||||
IN LONG LogSize,
|
||||
IN LONG ExtraBytesInHeader,
|
||||
IN LONG EntrySize
|
||||
);
|
||||
|
||||
VOID
|
||||
DestroyTraceLog(
|
||||
IN PTRACE_LOG Log
|
||||
);
|
||||
|
||||
LONG
|
||||
WriteTraceLog(
|
||||
IN PTRACE_LOG Log,
|
||||
IN PVOID Entry
|
||||
);
|
||||
|
||||
VOID
|
||||
ResetTraceLog(
|
||||
IN PTRACE_LOG Log
|
||||
);
|
||||
|
||||
|
||||
#if defined(__cplusplus)
|
||||
} // extern "C"
|
||||
#endif // __cplusplus
|
||||
|
||||
|
||||
#endif // _TRACELOG_H_
|
||||
|
||||
|
|
@ -0,0 +1,850 @@
|
|||
// Copyright (c) .NET Foundation. All rights reserved.
|
||||
// Licensed under the MIT License. See License.txt in the project root for license information.
|
||||
|
||||
#pragma once
|
||||
|
||||
#include <crtdbg.h>
|
||||
#include "rwlock.h"
|
||||
#include "prime.h"
|
||||
|
||||
template <class _Record>
|
||||
class TREE_HASH_NODE
|
||||
{
|
||||
template <class _Record>
|
||||
friend class TREE_HASH_TABLE;
|
||||
|
||||
private:
|
||||
// Next node in the hash table look-aside
|
||||
TREE_HASH_NODE<_Record> *_pNext;
|
||||
|
||||
// links in the tree structure
|
||||
TREE_HASH_NODE * _pParentNode;
|
||||
TREE_HASH_NODE * _pFirstChild;
|
||||
TREE_HASH_NODE * _pNextSibling;
|
||||
|
||||
// actual record
|
||||
_Record * _pRecord;
|
||||
|
||||
// hash value
|
||||
PCWSTR _pszPath;
|
||||
DWORD _dwHash;
|
||||
};
|
||||
|
||||
template <class _Record>
|
||||
class TREE_HASH_TABLE
|
||||
{
|
||||
protected:
|
||||
typedef BOOL
|
||||
(PFN_DELETE_IF)(
|
||||
_Record * pRecord,
|
||||
PVOID pvContext
|
||||
);
|
||||
|
||||
typedef VOID
|
||||
(PFN_APPLY)(
|
||||
_Record * pRecord,
|
||||
PVOID pvContext
|
||||
);
|
||||
|
||||
public:
|
||||
TREE_HASH_TABLE(
|
||||
BOOL fCaseSensitive
|
||||
) : _ppBuckets( NULL ),
|
||||
_nBuckets( 0 ),
|
||||
_nItems( 0 ),
|
||||
_fCaseSensitive( fCaseSensitive )
|
||||
{
|
||||
}
|
||||
|
||||
virtual
|
||||
~TREE_HASH_TABLE();
|
||||
|
||||
virtual
|
||||
VOID
|
||||
ReferenceRecord(
|
||||
_Record * pRecord
|
||||
) = 0;
|
||||
|
||||
virtual
|
||||
VOID
|
||||
DereferenceRecord(
|
||||
_Record * pRecord
|
||||
) = 0;
|
||||
|
||||
virtual
|
||||
PCWSTR
|
||||
GetKey(
|
||||
_Record * pRecord
|
||||
) = 0;
|
||||
|
||||
DWORD
|
||||
Count()
|
||||
{
|
||||
return _nItems;
|
||||
}
|
||||
|
||||
virtual
|
||||
VOID
|
||||
Clear();
|
||||
|
||||
HRESULT
|
||||
Initialize(
|
||||
DWORD nBucketSize
|
||||
);
|
||||
|
||||
DWORD
|
||||
CalcHash(
|
||||
PCWSTR pszKey
|
||||
)
|
||||
{
|
||||
return _fCaseSensitive ? HashString(pszKey) : HashStringNoCase(pszKey);
|
||||
}
|
||||
|
||||
virtual
|
||||
VOID
|
||||
FindKey(
|
||||
PCWSTR pszKey,
|
||||
_Record ** ppRecord
|
||||
);
|
||||
|
||||
virtual
|
||||
HRESULT
|
||||
InsertRecord(
|
||||
_Record * pRecord
|
||||
);
|
||||
|
||||
virtual
|
||||
VOID
|
||||
DeleteKey(
|
||||
PCWSTR pszKey
|
||||
);
|
||||
|
||||
virtual
|
||||
VOID
|
||||
DeleteIf(
|
||||
PFN_DELETE_IF pfnDeleteIf,
|
||||
PVOID pvContext
|
||||
);
|
||||
|
||||
VOID
|
||||
Apply(
|
||||
PFN_APPLY pfnApply,
|
||||
PVOID pvContext
|
||||
);
|
||||
|
||||
private:
|
||||
|
||||
BOOL
|
||||
FindNodeInternal(
|
||||
PCWSTR pszKey,
|
||||
DWORD dwHash,
|
||||
TREE_HASH_NODE<_Record> ** ppNode,
|
||||
TREE_HASH_NODE<_Record> *** pppPreviousNodeNextPointer = NULL
|
||||
);
|
||||
|
||||
HRESULT
|
||||
AddNodeInternal(
|
||||
PCWSTR pszPath,
|
||||
DWORD dwHash,
|
||||
_Record * pRecord,
|
||||
TREE_HASH_NODE<_Record> * pParentNode,
|
||||
TREE_HASH_NODE<_Record> ** ppNewNode
|
||||
);
|
||||
|
||||
HRESULT
|
||||
AllocateNode(
|
||||
PCWSTR pszPath,
|
||||
DWORD dwHash,
|
||||
_Record * pRecord,
|
||||
TREE_HASH_NODE<_Record> * pParentNode,
|
||||
TREE_HASH_NODE<_Record> ** ppNewNode
|
||||
);
|
||||
|
||||
VOID
|
||||
DeleteNode(
|
||||
TREE_HASH_NODE<_Record> * pNode
|
||||
)
|
||||
{
|
||||
if (pNode->_pRecord != NULL)
|
||||
{
|
||||
DereferenceRecord(pNode->_pRecord);
|
||||
pNode->_pRecord = NULL;
|
||||
}
|
||||
|
||||
HeapFree(GetProcessHeap(),
|
||||
0,
|
||||
pNode);
|
||||
}
|
||||
|
||||
VOID
|
||||
DeleteNodeInternal(
|
||||
TREE_HASH_NODE<_Record> ** ppPreviousNodeNextPointer,
|
||||
TREE_HASH_NODE<_Record> * pNode
|
||||
);
|
||||
|
||||
VOID
|
||||
RehashTableIfNeeded(
|
||||
VOID
|
||||
);
|
||||
|
||||
TREE_HASH_NODE<_Record> ** _ppBuckets;
|
||||
DWORD _nBuckets;
|
||||
DWORD _nItems;
|
||||
BOOL _fCaseSensitive;
|
||||
CWSDRWLock _tableLock;
|
||||
};
|
||||
|
||||
template <class _Record>
|
||||
HRESULT
|
||||
TREE_HASH_TABLE<_Record>::AllocateNode(
|
||||
PCWSTR pszPath,
|
||||
DWORD dwHash,
|
||||
_Record * pRecord,
|
||||
TREE_HASH_NODE<_Record> * pParentNode,
|
||||
TREE_HASH_NODE<_Record> ** ppNewNode
|
||||
)
|
||||
{
|
||||
//
|
||||
// Allocate enough extra space for pszPath
|
||||
//
|
||||
DWORD cchPath = (DWORD) wcslen(pszPath);
|
||||
if (cchPath >= ((0xffffffff - sizeof(TREE_HASH_NODE<_Record>))/sizeof(WCHAR) - 1))
|
||||
{
|
||||
return HRESULT_FROM_WIN32(ERROR_NOT_ENOUGH_MEMORY);
|
||||
}
|
||||
TREE_HASH_NODE<_Record> *pNode = (TREE_HASH_NODE<_Record> *)HeapAlloc(
|
||||
GetProcessHeap(),
|
||||
HEAP_ZERO_MEMORY,
|
||||
sizeof(TREE_HASH_NODE<_Record>) + (cchPath+1)*sizeof(WCHAR));
|
||||
if (pNode == NULL)
|
||||
{
|
||||
return HRESULT_FROM_WIN32(ERROR_NOT_ENOUGH_MEMORY);
|
||||
}
|
||||
|
||||
memcpy(pNode+1, pszPath, (cchPath+1)*sizeof(WCHAR));
|
||||
pNode->_pszPath = (PCWSTR)(pNode+1);
|
||||
pNode->_dwHash = dwHash;
|
||||
pNode->_pNext = pNode->_pNextSibling = pNode->_pFirstChild = NULL;
|
||||
pNode->_pParentNode = pParentNode;
|
||||
pNode->_pRecord = pRecord;
|
||||
|
||||
*ppNewNode = pNode;
|
||||
return S_OK;
|
||||
}
|
||||
|
||||
template <class _Record>
|
||||
HRESULT
|
||||
TREE_HASH_TABLE<_Record>::Initialize(
|
||||
DWORD nBuckets
|
||||
)
|
||||
{
|
||||
HRESULT hr = S_OK;
|
||||
|
||||
if ( nBuckets == 0 )
|
||||
{
|
||||
hr = E_INVALIDARG;
|
||||
goto Failed;
|
||||
}
|
||||
|
||||
hr = _tableLock.Init();
|
||||
if ( FAILED( hr ) )
|
||||
{
|
||||
goto Failed;
|
||||
}
|
||||
|
||||
if (nBuckets >= 0xffffffff/sizeof(TREE_HASH_NODE<_Record> *))
|
||||
{
|
||||
hr = E_INVALIDARG;
|
||||
goto Failed;
|
||||
}
|
||||
|
||||
_ppBuckets = (TREE_HASH_NODE<_Record> **)HeapAlloc(
|
||||
GetProcessHeap(),
|
||||
HEAP_ZERO_MEMORY,
|
||||
nBuckets*sizeof(TREE_HASH_NODE<_Record> *));
|
||||
if (_ppBuckets == NULL)
|
||||
{
|
||||
hr = HRESULT_FROM_WIN32(ERROR_NOT_ENOUGH_MEMORY);
|
||||
goto Failed;
|
||||
}
|
||||
_nBuckets = nBuckets;
|
||||
|
||||
return S_OK;
|
||||
|
||||
Failed:
|
||||
|
||||
if (_ppBuckets)
|
||||
{
|
||||
HeapFree(GetProcessHeap(),
|
||||
0,
|
||||
_ppBuckets);
|
||||
_ppBuckets = NULL;
|
||||
}
|
||||
|
||||
return hr;
|
||||
}
|
||||
|
||||
|
||||
template <class _Record>
|
||||
TREE_HASH_TABLE<_Record>::~TREE_HASH_TABLE()
|
||||
{
|
||||
if (_ppBuckets == NULL)
|
||||
{
|
||||
return;
|
||||
}
|
||||
|
||||
_ASSERTE(_nItems == 0);
|
||||
|
||||
HeapFree(GetProcessHeap(),
|
||||
0,
|
||||
_ppBuckets);
|
||||
_ppBuckets = NULL;
|
||||
_nBuckets = 0;
|
||||
}
|
||||
|
||||
template <class _Record>
|
||||
VOID
|
||||
TREE_HASH_TABLE<_Record>::Clear()
|
||||
{
|
||||
TREE_HASH_NODE<_Record> *pCurrent;
|
||||
TREE_HASH_NODE<_Record> *pNext;
|
||||
|
||||
_tableLock.ExclusiveAcquire();
|
||||
|
||||
for (DWORD i=0; i<_nBuckets; i++)
|
||||
{
|
||||
pCurrent = _ppBuckets[i];
|
||||
_ppBuckets[i] = NULL;
|
||||
while (pCurrent != NULL)
|
||||
{
|
||||
pNext = pCurrent->_pNext;
|
||||
DeleteNode(pCurrent);
|
||||
pCurrent = pNext;
|
||||
}
|
||||
}
|
||||
|
||||
_nItems = 0;
|
||||
_tableLock.ExclusiveRelease();
|
||||
}
|
||||
|
||||
template <class _Record>
|
||||
BOOL
|
||||
TREE_HASH_TABLE<_Record>::FindNodeInternal(
|
||||
PCWSTR pszKey,
|
||||
DWORD dwHash,
|
||||
TREE_HASH_NODE<_Record> ** ppNode,
|
||||
TREE_HASH_NODE<_Record> *** pppPreviousNodeNextPointer
|
||||
)
|
||||
/*++
|
||||
Return value indicates whether the item is found
|
||||
key, dwHash - key and hash for the node to find
|
||||
ppNode - on successful return, the node found, on failed return, the first
|
||||
node with hash value greater than the node to be found
|
||||
pppPreviousNodeNextPointer - the pointer to previous node's _pNext
|
||||
|
||||
This routine may be called under either read or write lock
|
||||
--*/
|
||||
{
|
||||
TREE_HASH_NODE<_Record> **ppPreviousNodeNextPointer;
|
||||
TREE_HASH_NODE<_Record> *pNode;
|
||||
BOOL fFound = FALSE;
|
||||
|
||||
ppPreviousNodeNextPointer = _ppBuckets + (dwHash % _nBuckets);
|
||||
pNode = *ppPreviousNodeNextPointer;
|
||||
while (pNode != NULL)
|
||||
{
|
||||
if (pNode->_dwHash == dwHash)
|
||||
{
|
||||
if (CompareStringOrdinal(pszKey,
|
||||
-1,
|
||||
pNode->_pszPath,
|
||||
-1,
|
||||
!_fCaseSensitive) == CSTR_EQUAL)
|
||||
{
|
||||
fFound = TRUE;
|
||||
break;
|
||||
}
|
||||
}
|
||||
else if (pNode->_dwHash > dwHash)
|
||||
{
|
||||
break;
|
||||
}
|
||||
|
||||
ppPreviousNodeNextPointer = &(pNode->_pNext);
|
||||
pNode = *ppPreviousNodeNextPointer;
|
||||
}
|
||||
|
||||
*ppNode = pNode;
|
||||
if (pppPreviousNodeNextPointer != NULL)
|
||||
{
|
||||
*pppPreviousNodeNextPointer = ppPreviousNodeNextPointer;
|
||||
}
|
||||
return fFound;
|
||||
}
|
||||
|
||||
template <class _Record>
|
||||
VOID
|
||||
TREE_HASH_TABLE<_Record>::FindKey(
|
||||
PCWSTR pszKey,
|
||||
_Record ** ppRecord
|
||||
)
|
||||
{
|
||||
TREE_HASH_NODE<_Record> *pNode;
|
||||
|
||||
*ppRecord = NULL;
|
||||
|
||||
DWORD dwHash = CalcHash(pszKey);
|
||||
|
||||
_tableLock.SharedAcquire();
|
||||
|
||||
if (FindNodeInternal(pszKey, dwHash, &pNode) &&
|
||||
pNode->_pRecord != NULL)
|
||||
{
|
||||
ReferenceRecord(pNode->_pRecord);
|
||||
*ppRecord = pNode->_pRecord;
|
||||
}
|
||||
|
||||
_tableLock.SharedRelease();
|
||||
}
|
||||
|
||||
template <class _Record>
|
||||
HRESULT
|
||||
TREE_HASH_TABLE<_Record>::AddNodeInternal(
|
||||
PCWSTR pszPath,
|
||||
DWORD dwHash,
|
||||
_Record * pRecord,
|
||||
TREE_HASH_NODE<_Record> * pParentNode,
|
||||
TREE_HASH_NODE<_Record> ** ppNewNode
|
||||
)
|
||||
/*++
|
||||
Return value is HRESULT indicating sucess or failure
|
||||
pszPath, dwHash, pRecord - path, hash value and record to be inserted
|
||||
pParentNode - this will be the parent of the node being inserted
|
||||
ppNewNode - on successful return, the new node created and inserted
|
||||
|
||||
This function may be called under a read or write lock
|
||||
--*/
|
||||
{
|
||||
TREE_HASH_NODE<_Record> *pNewNode;
|
||||
TREE_HASH_NODE<_Record> *pNextNode;
|
||||
TREE_HASH_NODE<_Record> **ppNextPointer;
|
||||
HRESULT hr;
|
||||
|
||||
//
|
||||
// Ownership of pRecord is not transferred to pNewNode yet, so remember
|
||||
// to either set it to null before deleting pNewNode or add an extra
|
||||
// reference later - this is to make sure we do not do an extra ref/deref
|
||||
// which users may view as getting flushed out of the hash-table
|
||||
//
|
||||
hr = AllocateNode(pszPath,
|
||||
dwHash,
|
||||
pRecord,
|
||||
pParentNode,
|
||||
&pNewNode);
|
||||
if (FAILED(hr))
|
||||
{
|
||||
return hr;
|
||||
}
|
||||
|
||||
do
|
||||
{
|
||||
//
|
||||
// Find the right place to add this node
|
||||
//
|
||||
|
||||
if (FindNodeInternal(pszPath, dwHash, &pNextNode, &ppNextPointer))
|
||||
{
|
||||
//
|
||||
// If node already there, record may still need updating
|
||||
//
|
||||
if (pRecord != NULL &&
|
||||
InterlockedCompareExchangePointer((PVOID *)&pNextNode->_pRecord,
|
||||
pRecord,
|
||||
NULL) == NULL)
|
||||
{
|
||||
ReferenceRecord(pRecord);
|
||||
hr = S_OK;
|
||||
}
|
||||
else
|
||||
{
|
||||
hr = HRESULT_FROM_WIN32(ERROR_ALREADY_EXISTS);
|
||||
}
|
||||
|
||||
// ownership of pRecord has either passed to existing record or
|
||||
// not to anyone at all
|
||||
pNewNode->_pRecord = NULL;
|
||||
DeleteNode(pNewNode);
|
||||
*ppNewNode = pNextNode;
|
||||
return hr;
|
||||
}
|
||||
|
||||
//
|
||||
// If another node got inserted in betwen, we will have to retry
|
||||
//
|
||||
pNewNode->_pNext = pNextNode;
|
||||
} while (InterlockedCompareExchangePointer((PVOID *)ppNextPointer,
|
||||
pNewNode,
|
||||
pNextNode) != pNextNode);
|
||||
// pass ownership of pRecord now
|
||||
if (pRecord != NULL)
|
||||
{
|
||||
ReferenceRecord(pRecord);
|
||||
pRecord = NULL;
|
||||
}
|
||||
InterlockedIncrement((LONG *)&_nItems);
|
||||
|
||||
//
|
||||
// update the parent
|
||||
//
|
||||
if (pParentNode != NULL)
|
||||
{
|
||||
ppNextPointer = &pParentNode->_pFirstChild;
|
||||
do
|
||||
{
|
||||
pNextNode = *ppNextPointer;
|
||||
pNewNode->_pNextSibling = pNextNode;
|
||||
} while (InterlockedCompareExchangePointer((PVOID *)ppNextPointer,
|
||||
pNewNode,
|
||||
pNextNode) != pNextNode);
|
||||
}
|
||||
|
||||
*ppNewNode = pNewNode;
|
||||
return S_OK;
|
||||
}
|
||||
|
||||
template <class _Record>
|
||||
HRESULT
|
||||
TREE_HASH_TABLE<_Record>::InsertRecord(
|
||||
_Record * pRecord
|
||||
)
|
||||
/*++
|
||||
This method inserts a node for this record and also empty nodes for paths
|
||||
in the heirarchy leading upto this path
|
||||
|
||||
The insert is done under only a read-lock - this is possible by keeping
|
||||
the hashes in a bucket in increasing order and using interlocked operations
|
||||
to actually insert the item in the hash-bucket lookaside list and the parent
|
||||
children list
|
||||
|
||||
Returns HRESULT_FROM_WIN32(ERROR_ALREADY_EXISTS) if the record already exists.
|
||||
Never leak this error to the end user because "*file* already exists" may be confusing.
|
||||
--*/
|
||||
{
|
||||
PCWSTR pszKey = GetKey(pRecord);
|
||||
STACK_STRU( strPartialPath, 256);
|
||||
PWSTR pszPartialPath;
|
||||
DWORD dwHash;
|
||||
DWORD cchEnd;
|
||||
HRESULT hr;
|
||||
TREE_HASH_NODE<_Record> *pParentNode = NULL;
|
||||
|
||||
hr = strPartialPath.Copy(pszKey);
|
||||
if (FAILED(hr))
|
||||
{
|
||||
goto Finished;
|
||||
}
|
||||
pszPartialPath = strPartialPath.QueryStr();
|
||||
|
||||
_tableLock.SharedAcquire();
|
||||
|
||||
//
|
||||
// First find the lowest parent node present
|
||||
//
|
||||
for (cchEnd = strPartialPath.QueryCCH() - 1; cchEnd > 0; cchEnd--)
|
||||
{
|
||||
if (pszPartialPath[cchEnd] == L'/' || pszPartialPath[cchEnd] == L'\\')
|
||||
{
|
||||
pszPartialPath[cchEnd] = L'\0';
|
||||
|
||||
dwHash = CalcHash(pszPartialPath);
|
||||
if (FindNodeInternal(pszPartialPath, dwHash, &pParentNode))
|
||||
{
|
||||
pszPartialPath[cchEnd] = pszKey[cchEnd];
|
||||
break;
|
||||
}
|
||||
pParentNode = NULL;
|
||||
}
|
||||
}
|
||||
|
||||
//
|
||||
// Now go ahead and add the rest of the tree (including our record)
|
||||
//
|
||||
for (; cchEnd <= strPartialPath.QueryCCH(); cchEnd++)
|
||||
{
|
||||
if (pszPartialPath[cchEnd] == L'\0')
|
||||
{
|
||||
dwHash = CalcHash(pszPartialPath);
|
||||
hr = AddNodeInternal(
|
||||
pszPartialPath,
|
||||
dwHash,
|
||||
(cchEnd == strPartialPath.QueryCCH()) ? pRecord : NULL,
|
||||
pParentNode,
|
||||
&pParentNode);
|
||||
if (FAILED(hr) &&
|
||||
hr != HRESULT_FROM_WIN32(ERROR_ALREADY_EXISTS))
|
||||
{
|
||||
goto Finished;
|
||||
}
|
||||
|
||||
pszPartialPath[cchEnd] = pszKey[cchEnd];
|
||||
}
|
||||
}
|
||||
|
||||
Finished:
|
||||
_tableLock.SharedRelease();
|
||||
|
||||
if (SUCCEEDED(hr))
|
||||
{
|
||||
RehashTableIfNeeded();
|
||||
}
|
||||
|
||||
return hr;
|
||||
}
|
||||
|
||||
template <class _Record>
|
||||
VOID
|
||||
TREE_HASH_TABLE<_Record>::DeleteNodeInternal(
|
||||
TREE_HASH_NODE<_Record> ** ppNextPointer,
|
||||
TREE_HASH_NODE<_Record> * pNode
|
||||
)
|
||||
/*++
|
||||
pNode is the node to be deleted
|
||||
ppNextPointer is the pointer to the previous node's next pointer pointing
|
||||
to this node
|
||||
|
||||
This function should be called under write-lock
|
||||
--*/
|
||||
{
|
||||
//
|
||||
// First remove this node from hash table
|
||||
//
|
||||
*ppNextPointer = pNode->_pNext;
|
||||
|
||||
//
|
||||
// Now fixup parent
|
||||
//
|
||||
if (pNode->_pParentNode != NULL)
|
||||
{
|
||||
ppNextPointer = &pNode->_pParentNode->_pFirstChild;
|
||||
while (*ppNextPointer != pNode)
|
||||
{
|
||||
ppNextPointer = &(*ppNextPointer)->_pNextSibling;
|
||||
}
|
||||
*ppNextPointer = pNode->_pNextSibling;
|
||||
}
|
||||
|
||||
//
|
||||
// Now remove all children recursively
|
||||
//
|
||||
TREE_HASH_NODE<_Record> *pChild = pNode->_pFirstChild;
|
||||
TREE_HASH_NODE<_Record> *pNextChild;
|
||||
while (pChild != NULL)
|
||||
{
|
||||
pNextChild = pChild->_pNextSibling;
|
||||
|
||||
ppNextPointer = _ppBuckets + (pChild->_dwHash % _nBuckets);
|
||||
while (*ppNextPointer != pChild)
|
||||
{
|
||||
ppNextPointer = &(*ppNextPointer)->_pNext;
|
||||
}
|
||||
pChild->_pParentNode = NULL;
|
||||
DeleteNodeInternal(ppNextPointer, pChild);
|
||||
|
||||
pChild = pNextChild;
|
||||
}
|
||||
|
||||
DeleteNode(pNode);
|
||||
_nItems--;
|
||||
}
|
||||
|
||||
template <class _Record>
|
||||
VOID
|
||||
TREE_HASH_TABLE<_Record>::DeleteKey(
|
||||
PCWSTR pszKey
|
||||
)
|
||||
{
|
||||
TREE_HASH_NODE<_Record> *pNode;
|
||||
TREE_HASH_NODE<_Record> **ppPreviousNodeNextPointer;
|
||||
|
||||
DWORD dwHash = CalcHash(pszKey);
|
||||
|
||||
_tableLock.ExclusiveAcquire();
|
||||
|
||||
if (FindNodeInternal(pszKey, dwHash, &pNode, &ppPreviousNodeNextPointer))
|
||||
{
|
||||
DeleteNodeInternal(ppPreviousNodeNextPointer, pNode);
|
||||
}
|
||||
|
||||
_tableLock.ExclusiveRelease();
|
||||
}
|
||||
|
||||
template <class _Record>
|
||||
VOID
|
||||
TREE_HASH_TABLE<_Record>::DeleteIf(
|
||||
PFN_DELETE_IF pfnDeleteIf,
|
||||
PVOID pvContext
|
||||
)
|
||||
{
|
||||
TREE_HASH_NODE<_Record> *pNode;
|
||||
TREE_HASH_NODE<_Record> **ppPreviousNodeNextPointer;
|
||||
BOOL fDelete;
|
||||
|
||||
_tableLock.ExclusiveAcquire();
|
||||
|
||||
for (DWORD i=0; i<_nBuckets; i++)
|
||||
{
|
||||
ppPreviousNodeNextPointer = _ppBuckets + i;
|
||||
pNode = *ppPreviousNodeNextPointer;
|
||||
while (pNode != NULL)
|
||||
{
|
||||
//
|
||||
// Non empty nodes deleted based on DeleteIf, empty nodes deleted
|
||||
// if they have no children
|
||||
//
|
||||
fDelete = FALSE;
|
||||
if (pNode->_pRecord != NULL)
|
||||
{
|
||||
if (pfnDeleteIf(pNode->_pRecord, pvContext))
|
||||
{
|
||||
fDelete = TRUE;
|
||||
}
|
||||
}
|
||||
else if (pNode->_pFirstChild == NULL)
|
||||
{
|
||||
fDelete = TRUE;
|
||||
}
|
||||
|
||||
if (fDelete)
|
||||
{
|
||||
if (pNode->_pFirstChild == NULL)
|
||||
{
|
||||
DeleteNodeInternal(ppPreviousNodeNextPointer, pNode);
|
||||
}
|
||||
else
|
||||
{
|
||||
DereferenceRecord(pNode->_pRecord);
|
||||
pNode->_pRecord = NULL;
|
||||
}
|
||||
}
|
||||
else
|
||||
{
|
||||
ppPreviousNodeNextPointer = &pNode->_pNext;
|
||||
}
|
||||
|
||||
pNode = *ppPreviousNodeNextPointer;
|
||||
}
|
||||
}
|
||||
|
||||
_tableLock.ExclusiveRelease();
|
||||
}
|
||||
|
||||
template <class _Record>
|
||||
VOID
|
||||
TREE_HASH_TABLE<_Record>::Apply(
|
||||
PFN_APPLY pfnApply,
|
||||
PVOID pvContext
|
||||
)
|
||||
{
|
||||
TREE_HASH_NODE<_Record> *pNode;
|
||||
|
||||
_tableLock.SharedAcquire();
|
||||
|
||||
for (DWORD i=0; i<_nBuckets; i++)
|
||||
{
|
||||
pNode = _ppBuckets[i];
|
||||
while (pNode != NULL)
|
||||
{
|
||||
if (pNode->_pRecord != NULL)
|
||||
{
|
||||
pfnApply(pNode->_pRecord, pvContext);
|
||||
}
|
||||
|
||||
pNode = pNode->_pNext;
|
||||
}
|
||||
}
|
||||
|
||||
_tableLock.SharedRelease();
|
||||
}
|
||||
|
||||
template <class _Record>
|
||||
VOID
|
||||
TREE_HASH_TABLE<_Record>::RehashTableIfNeeded(
|
||||
VOID
|
||||
)
|
||||
{
|
||||
TREE_HASH_NODE<_Record> **ppBuckets;
|
||||
DWORD nBuckets;
|
||||
TREE_HASH_NODE<_Record> *pNode;
|
||||
TREE_HASH_NODE<_Record> *pNextNode;
|
||||
TREE_HASH_NODE<_Record> **ppNextPointer;
|
||||
TREE_HASH_NODE<_Record> *pNewNextNode;
|
||||
DWORD nNewBuckets;
|
||||
|
||||
//
|
||||
// If number of items has become too many, we will double the hash table
|
||||
// size (we never reduce it however)
|
||||
//
|
||||
if (_nItems <= PRIME::GetPrime(2*_nBuckets))
|
||||
{
|
||||
return;
|
||||
}
|
||||
|
||||
_tableLock.ExclusiveAcquire();
|
||||
|
||||
nNewBuckets = PRIME::GetPrime(2*_nBuckets);
|
||||
|
||||
if (_nItems <= nNewBuckets)
|
||||
{
|
||||
goto Finished;
|
||||
}
|
||||
|
||||
nBuckets = nNewBuckets;
|
||||
if (nBuckets >= 0xffffffff/sizeof(TREE_HASH_NODE<_Record> *))
|
||||
{
|
||||
goto Finished;
|
||||
}
|
||||
ppBuckets = (TREE_HASH_NODE<_Record> **)HeapAlloc(
|
||||
GetProcessHeap(),
|
||||
HEAP_ZERO_MEMORY,
|
||||
nBuckets*sizeof(TREE_HASH_NODE<_Record> *));
|
||||
if (ppBuckets == NULL)
|
||||
{
|
||||
goto Finished;
|
||||
}
|
||||
|
||||
//
|
||||
// Take out nodes from the old hash table and insert in the new one, make
|
||||
// sure to keep the hashes in increasing order
|
||||
//
|
||||
for (DWORD i=0; i<_nBuckets; i++)
|
||||
{
|
||||
pNode = _ppBuckets[i];
|
||||
while (pNode != NULL)
|
||||
{
|
||||
pNextNode = pNode->_pNext;
|
||||
|
||||
ppNextPointer = ppBuckets + (pNode->_dwHash % nBuckets);
|
||||
pNewNextNode = *ppNextPointer;
|
||||
while (pNewNextNode != NULL &&
|
||||
pNewNextNode->_dwHash <= pNode->_dwHash)
|
||||
{
|
||||
ppNextPointer = &pNewNextNode->_pNext;
|
||||
pNewNextNode = pNewNextNode->_pNext;
|
||||
}
|
||||
pNode->_pNext = pNewNextNode;
|
||||
*ppNextPointer = pNode;
|
||||
|
||||
pNode = pNextNode;
|
||||
}
|
||||
}
|
||||
|
||||
HeapFree(GetProcessHeap(), 0, _ppBuckets);
|
||||
_ppBuckets = ppBuckets;
|
||||
_nBuckets = nBuckets;
|
||||
ppBuckets = NULL;
|
||||
|
||||
Finished:
|
||||
|
||||
_tableLock.ExclusiveRelease();
|
||||
}
|
||||
|
||||
|
|
@ -0,0 +1,74 @@
|
|||
// Copyright (c) .NET Foundation. All rights reserved.
|
||||
// Licensed under the MIT License. See License.txt in the project root for license information.
|
||||
|
||||
#include "precomp.h"
|
||||
|
||||
HRESULT
|
||||
MakePathCanonicalizationProof(
|
||||
IN PCWSTR pszName,
|
||||
OUT STRU * pstrPath
|
||||
)
|
||||
/*++
|
||||
|
||||
Routine Description:
|
||||
|
||||
This functions adds a prefix
|
||||
to the string, which is "\\?\UNC\" for a UNC path, and "\\?\" for
|
||||
other paths. This prefix tells Windows not to parse the path.
|
||||
|
||||
Arguments:
|
||||
|
||||
IN pszName - The path to be converted
|
||||
OUT pstrPath - Output path created
|
||||
|
||||
Return Values:
|
||||
|
||||
HRESULT
|
||||
|
||||
--*/
|
||||
{
|
||||
HRESULT hr;
|
||||
|
||||
if (pszName[0] == L'\\' && pszName[1] == L'\\')
|
||||
{
|
||||
//
|
||||
// If the path is already canonicalized, just return
|
||||
//
|
||||
|
||||
if ((pszName[2] == '?' || pszName[2] == '.') &&
|
||||
pszName[3] == '\\')
|
||||
{
|
||||
hr = pstrPath->Copy(pszName);
|
||||
|
||||
if (SUCCEEDED(hr))
|
||||
{
|
||||
//
|
||||
// If the path was in DOS form ("\\.\"),
|
||||
// we need to change it to Win32 from ("\\?\")
|
||||
//
|
||||
|
||||
pstrPath->QueryStr()[2] = L'?';
|
||||
}
|
||||
|
||||
return hr;
|
||||
}
|
||||
|
||||
pszName += 2;
|
||||
|
||||
|
||||
if (FAILED(hr = pstrPath->Copy(L"\\\\?\\UNC\\")))
|
||||
{
|
||||
return hr;
|
||||
}
|
||||
}
|
||||
else
|
||||
{
|
||||
if (FAILED(hr = pstrPath->Copy(L"\\\\?\\")))
|
||||
{
|
||||
return hr;
|
||||
}
|
||||
}
|
||||
|
||||
return pstrPath->Append(pszName);
|
||||
}
|
||||
|
||||
Binary file not shown.
|
|
@ -0,0 +1,287 @@
|
|||
<?xml version="1.0" encoding="utf-8"?>
|
||||
<Project DefaultTargets="Build" ToolsVersion="15.0" xmlns="http://schemas.microsoft.com/developer/msbuild/2003">
|
||||
<Import Project="..\..\Build\Build.Settings" />
|
||||
<ItemGroup Label="ProjectConfigurations">
|
||||
<ProjectConfiguration Include="Debug|Win32">
|
||||
<Configuration>Debug</Configuration>
|
||||
<Platform>Win32</Platform>
|
||||
</ProjectConfiguration>
|
||||
<ProjectConfiguration Include="Release|Win32">
|
||||
<Configuration>Release</Configuration>
|
||||
<Platform>Win32</Platform>
|
||||
</ProjectConfiguration>
|
||||
<ProjectConfiguration Include="Debug|x64">
|
||||
<Configuration>Debug</Configuration>
|
||||
<Platform>x64</Platform>
|
||||
</ProjectConfiguration>
|
||||
<ProjectConfiguration Include="Release|x64">
|
||||
<Configuration>Release</Configuration>
|
||||
<Platform>x64</Platform>
|
||||
</ProjectConfiguration>
|
||||
</ItemGroup>
|
||||
<PropertyGroup Label="Globals">
|
||||
<VCProjectVersion>15.0</VCProjectVersion>
|
||||
<ProjectGuid>{D57EA297-6DC2-4BC0-8C91-334863327863}</ProjectGuid>
|
||||
<Keyword>Win32Proj</Keyword>
|
||||
<RootNamespace>RequestHandler</RootNamespace>
|
||||
<WindowsTargetPlatformVersion>10.0.15063.0</WindowsTargetPlatformVersion>
|
||||
<ProjectName>RequestHandler</ProjectName>
|
||||
</PropertyGroup>
|
||||
<Import Project="$(VCTargetsPath)\Microsoft.Cpp.Default.props" />
|
||||
<PropertyGroup Condition="'$(Configuration)|$(Platform)'=='Debug|Win32'" Label="Configuration">
|
||||
<ConfigurationType>DynamicLibrary</ConfigurationType>
|
||||
<UseDebugLibraries>true</UseDebugLibraries>
|
||||
<PlatformToolset>v141</PlatformToolset>
|
||||
<CharacterSet>Unicode</CharacterSet>
|
||||
</PropertyGroup>
|
||||
<PropertyGroup Condition="'$(Configuration)|$(Platform)'=='Release|Win32'" Label="Configuration">
|
||||
<ConfigurationType>DynamicLibrary</ConfigurationType>
|
||||
<UseDebugLibraries>false</UseDebugLibraries>
|
||||
<PlatformToolset>v141</PlatformToolset>
|
||||
<WholeProgramOptimization>true</WholeProgramOptimization>
|
||||
<CharacterSet>Unicode</CharacterSet>
|
||||
</PropertyGroup>
|
||||
<PropertyGroup Condition="'$(Configuration)|$(Platform)'=='Debug|x64'" Label="Configuration">
|
||||
<ConfigurationType>DynamicLibrary</ConfigurationType>
|
||||
<UseDebugLibraries>true</UseDebugLibraries>
|
||||
<PlatformToolset>v141</PlatformToolset>
|
||||
<CharacterSet>Unicode</CharacterSet>
|
||||
</PropertyGroup>
|
||||
<PropertyGroup Condition="'$(Configuration)|$(Platform)'=='Release|x64'" Label="Configuration">
|
||||
<ConfigurationType>DynamicLibrary</ConfigurationType>
|
||||
<UseDebugLibraries>false</UseDebugLibraries>
|
||||
<PlatformToolset>v141</PlatformToolset>
|
||||
<WholeProgramOptimization>true</WholeProgramOptimization>
|
||||
<CharacterSet>Unicode</CharacterSet>
|
||||
</PropertyGroup>
|
||||
<Import Project="$(VCTargetsPath)\Microsoft.Cpp.props" />
|
||||
<ImportGroup Label="ExtensionSettings">
|
||||
</ImportGroup>
|
||||
<ImportGroup Label="Shared">
|
||||
</ImportGroup>
|
||||
<ImportGroup Label="PropertySheets" Condition="'$(Configuration)|$(Platform)'=='Debug|Win32'">
|
||||
<Import Project="$(UserRootDir)\Microsoft.Cpp.$(Platform).user.props" Condition="exists('$(UserRootDir)\Microsoft.Cpp.$(Platform).user.props')" Label="LocalAppDataPlatform" />
|
||||
</ImportGroup>
|
||||
<ImportGroup Label="PropertySheets" Condition="'$(Configuration)|$(Platform)'=='Release|Win32'">
|
||||
<Import Project="$(UserRootDir)\Microsoft.Cpp.$(Platform).user.props" Condition="exists('$(UserRootDir)\Microsoft.Cpp.$(Platform).user.props')" Label="LocalAppDataPlatform" />
|
||||
</ImportGroup>
|
||||
<ImportGroup Label="PropertySheets" Condition="'$(Configuration)|$(Platform)'=='Debug|x64'">
|
||||
<Import Project="$(UserRootDir)\Microsoft.Cpp.$(Platform).user.props" Condition="exists('$(UserRootDir)\Microsoft.Cpp.$(Platform).user.props')" Label="LocalAppDataPlatform" />
|
||||
</ImportGroup>
|
||||
<ImportGroup Label="PropertySheets" Condition="'$(Configuration)|$(Platform)'=='Release|x64'">
|
||||
<Import Project="$(UserRootDir)\Microsoft.Cpp.$(Platform).user.props" Condition="exists('$(UserRootDir)\Microsoft.Cpp.$(Platform).user.props')" Label="LocalAppDataPlatform" />
|
||||
</ImportGroup>
|
||||
<PropertyGroup Label="UserMacros" />
|
||||
<PropertyGroup Condition="'$(Configuration)|$(Platform)'=='Debug|Win32'">
|
||||
<LinkIncremental>true</LinkIncremental>
|
||||
<TargetName>aspnetcorerh</TargetName>
|
||||
<OutDir>$(SolutionDir)artifacts\build\AspNetCore\bin\$(Configuration)\$(Platform)\</OutDir>
|
||||
</PropertyGroup>
|
||||
<PropertyGroup Condition="'$(Configuration)|$(Platform)'=='Debug|x64'">
|
||||
<LinkIncremental>true</LinkIncremental>
|
||||
<TargetName>aspnetcorerh</TargetName>
|
||||
<OutDir>$(SolutionDir)artifacts\build\AspNetCore\bin\$(Configuration)\$(Platform)\</OutDir>
|
||||
</PropertyGroup>
|
||||
<PropertyGroup Condition="'$(Configuration)|$(Platform)'=='Release|Win32'">
|
||||
<LinkIncremental>false</LinkIncremental>
|
||||
<TargetName>aspnetcorerh</TargetName>
|
||||
<OutDir>$(SolutionDir)artifacts\build\AspNetCore\bin\$(Configuration)\$(Platform)\</OutDir>
|
||||
</PropertyGroup>
|
||||
<PropertyGroup Condition="'$(Configuration)|$(Platform)'=='Release|x64'">
|
||||
<LinkIncremental>false</LinkIncremental>
|
||||
<TargetName>aspnetcorerh</TargetName>
|
||||
<OutDir>$(SolutionDir)artifacts\build\AspNetCore\bin\$(Configuration)\$(Platform)\</OutDir>
|
||||
</PropertyGroup>
|
||||
<ItemDefinitionGroup Condition="'$(Configuration)|$(Platform)'=='Debug|Win32'">
|
||||
<ClCompile>
|
||||
<PrecompiledHeader>NotUsing</PrecompiledHeader>
|
||||
<WarningLevel>Level4</WarningLevel>
|
||||
<Optimization>Disabled</Optimization>
|
||||
<PreprocessorDefinitions>WIN32;_DEBUG;REQUESTHANDLER_EXPORTS;_WINDOWS;_USRDLL;%(PreprocessorDefinitions)</PreprocessorDefinitions>
|
||||
<PrecompiledHeaderFile>precomp.hxx</PrecompiledHeaderFile>
|
||||
<PrecompiledHeaderOutputFile>$(IntDir)$(TargetName).pch</PrecompiledHeaderOutputFile>
|
||||
<DebugInformationFormat>ProgramDatabase</DebugInformationFormat>
|
||||
<RuntimeLibrary>MultiThreadedDebug</RuntimeLibrary>
|
||||
<AdditionalIncludeDirectories>..\IISLib;..\CommonLib;.\Inc</AdditionalIncludeDirectories>
|
||||
<TreatWarningAsError>true</TreatWarningAsError>
|
||||
<SDLCheck>true</SDLCheck>
|
||||
<WholeProgramOptimization>true</WholeProgramOptimization>
|
||||
<PreprocessKeepComments>false</PreprocessKeepComments>
|
||||
<ExceptionHandling>SyncCThrow</ExceptionHandling>
|
||||
<StructMemberAlignment>8Bytes</StructMemberAlignment>
|
||||
<FunctionLevelLinking>true</FunctionLevelLinking>
|
||||
<RuntimeTypeInfo>false</RuntimeTypeInfo>
|
||||
<OmitDefaultLibName>true</OmitDefaultLibName>
|
||||
<CompileAs>CompileAsCpp</CompileAs>
|
||||
<IntrinsicFunctions>true</IntrinsicFunctions>
|
||||
</ClCompile>
|
||||
<Link>
|
||||
<SubSystem>Windows</SubSystem>
|
||||
<GenerateDebugInformation>true</GenerateDebugInformation>
|
||||
<AdditionalDependencies>kernel32.lib;user32.lib;advapi32.lib;ole32.lib;oleaut32.lib;uuid.lib;odbc32.lib;odbccp32.lib;ahadmin.lib;rpcrt4.lib;winhttp.lib;pdh.lib;ws2_32.lib;wbemuuid.lib;iphlpapi.lib;%(AdditionalDependencies)</AdditionalDependencies>
|
||||
<ModuleDefinitionFile>Source.def</ModuleDefinitionFile>
|
||||
</Link>
|
||||
</ItemDefinitionGroup>
|
||||
<ItemDefinitionGroup Condition="'$(Configuration)|$(Platform)'=='Debug|x64'">
|
||||
<ClCompile>
|
||||
<PrecompiledHeader>NotUsing</PrecompiledHeader>
|
||||
<WarningLevel>Level4</WarningLevel>
|
||||
<Optimization>Disabled</Optimization>
|
||||
<PreprocessorDefinitions>WIN32;_DEBUG;REQUESTHANDLER_EXPORTS;_WINDOWS;_USRDLL;%(PreprocessorDefinitions)</PreprocessorDefinitions>
|
||||
<PrecompiledHeaderFile>precomp.hxx</PrecompiledHeaderFile>
|
||||
<PrecompiledHeaderOutputFile>$(IntDir)$(TargetName).pch</PrecompiledHeaderOutputFile>
|
||||
<DebugInformationFormat>ProgramDatabase</DebugInformationFormat>
|
||||
<RuntimeLibrary>MultiThreadedDebug</RuntimeLibrary>
|
||||
<AdditionalIncludeDirectories>..\IISLib;..\CommonLib;.\Inc</AdditionalIncludeDirectories>
|
||||
<TreatWarningAsError>true</TreatWarningAsError>
|
||||
<SDLCheck>true</SDLCheck>
|
||||
<WholeProgramOptimization>true</WholeProgramOptimization>
|
||||
<PreprocessKeepComments>false</PreprocessKeepComments>
|
||||
<ExceptionHandling>SyncCThrow</ExceptionHandling>
|
||||
<StructMemberAlignment>8Bytes</StructMemberAlignment>
|
||||
<FunctionLevelLinking>true</FunctionLevelLinking>
|
||||
<RuntimeTypeInfo>false</RuntimeTypeInfo>
|
||||
<OmitDefaultLibName>true</OmitDefaultLibName>
|
||||
<CompileAs>CompileAsCpp</CompileAs>
|
||||
<IntrinsicFunctions>true</IntrinsicFunctions>
|
||||
</ClCompile>
|
||||
<Link>
|
||||
<SubSystem>Windows</SubSystem>
|
||||
<GenerateDebugInformation>true</GenerateDebugInformation>
|
||||
<AdditionalDependencies>kernel32.lib;user32.lib;advapi32.lib;ole32.lib;oleaut32.lib;uuid.lib;odbc32.lib;odbccp32.lib;ahadmin.lib;rpcrt4.lib;winhttp.lib;pdh.lib;ws2_32.lib;wbemuuid.lib;iphlpapi.lib;%(AdditionalDependencies)</AdditionalDependencies>
|
||||
<ModuleDefinitionFile>Source.def</ModuleDefinitionFile>
|
||||
</Link>
|
||||
</ItemDefinitionGroup>
|
||||
<ItemDefinitionGroup Condition="'$(Configuration)|$(Platform)'=='Release|Win32'">
|
||||
<ClCompile>
|
||||
<WarningLevel>Level4</WarningLevel>
|
||||
<PrecompiledHeader>NotUsing</PrecompiledHeader>
|
||||
<Optimization>MaxSpeed</Optimization>
|
||||
<FunctionLevelLinking>true</FunctionLevelLinking>
|
||||
<IntrinsicFunctions>true</IntrinsicFunctions>
|
||||
<PreprocessorDefinitions>WIN32;NDEBUG;_WINDOWS;_USRDLL;REQUESTHANDLER_EXPORTS;%(PreprocessorDefinitions)</PreprocessorDefinitions>
|
||||
<PrecompiledHeaderFile>precomp.hxx</PrecompiledHeaderFile>
|
||||
<RuntimeLibrary>MultiThreaded</RuntimeLibrary>
|
||||
<AdditionalIncludeDirectories>..\IISLib;..\CommonLib;.\Inc</AdditionalIncludeDirectories>
|
||||
<TreatWarningAsError>true</TreatWarningAsError>
|
||||
<SDLCheck>true</SDLCheck>
|
||||
<WholeProgramOptimization>true</WholeProgramOptimization>
|
||||
<PreprocessKeepComments>false</PreprocessKeepComments>
|
||||
<ExceptionHandling>SyncCThrow</ExceptionHandling>
|
||||
<StructMemberAlignment>8Bytes</StructMemberAlignment>
|
||||
<FunctionLevelLinking>true</FunctionLevelLinking>
|
||||
<RuntimeTypeInfo>false</RuntimeTypeInfo>
|
||||
<OmitDefaultLibName>true</OmitDefaultLibName>
|
||||
<CompileAs>CompileAsCpp</CompileAs>
|
||||
<IntrinsicFunctions>true</IntrinsicFunctions>
|
||||
</ClCompile>
|
||||
<Link>
|
||||
<SubSystem>Windows</SubSystem>
|
||||
<GenerateDebugInformation>false</GenerateDebugInformation>
|
||||
<EnableCOMDATFolding>true</EnableCOMDATFolding>
|
||||
<OptimizeReferences>true</OptimizeReferences>
|
||||
<ModuleDefinitionFile>Source.def</ModuleDefinitionFile>
|
||||
<AdditionalDependencies>kernel32.lib;user32.lib;gdi32.lib;winspool.lib;comdlg32.lib;advapi32.lib;shell32.lib;ole32.lib;oleaut32.lib;uuid.lib;ahadmin.lib;winhttp.lib;odbc32.lib;ws2_32.lib;odbccp32.lib;wbemuuid.lib;iphlpapi.lib;pdh.lib;rpcrt4.lib;%(AdditionalDependencies)</AdditionalDependencies>
|
||||
</Link>
|
||||
</ItemDefinitionGroup>
|
||||
<ItemDefinitionGroup Condition="'$(Configuration)|$(Platform)'=='Release|x64'">
|
||||
<ClCompile>
|
||||
<WarningLevel>Level4</WarningLevel>
|
||||
<PrecompiledHeader>NotUsing</PrecompiledHeader>
|
||||
<Optimization>MaxSpeed</Optimization>
|
||||
<FunctionLevelLinking>true</FunctionLevelLinking>
|
||||
<IntrinsicFunctions>true</IntrinsicFunctions>
|
||||
<PreprocessorDefinitions>NDEBUG;REQUESTHANDLER_EXPORTS;_WINDOWS;_USRDLL;%(PreprocessorDefinitions)</PreprocessorDefinitions>
|
||||
<PrecompiledHeaderFile>precomp.hxx</PrecompiledHeaderFile>
|
||||
<RuntimeLibrary>MultiThreaded</RuntimeLibrary>
|
||||
<AdditionalIncludeDirectories>..\IISLib;..\CommonLib;.\Inc</AdditionalIncludeDirectories>
|
||||
<TreatWarningAsError>true</TreatWarningAsError>
|
||||
<SDLCheck>true</SDLCheck>
|
||||
<WholeProgramOptimization>true</WholeProgramOptimization>
|
||||
<PreprocessKeepComments>false</PreprocessKeepComments>
|
||||
<ExceptionHandling>SyncCThrow</ExceptionHandling>
|
||||
<StructMemberAlignment>8Bytes</StructMemberAlignment>
|
||||
<FunctionLevelLinking>true</FunctionLevelLinking>
|
||||
<RuntimeTypeInfo>false</RuntimeTypeInfo>
|
||||
<OmitDefaultLibName>true</OmitDefaultLibName>
|
||||
<CompileAs>CompileAsCpp</CompileAs>
|
||||
<IntrinsicFunctions>true</IntrinsicFunctions>
|
||||
</ClCompile>
|
||||
<Link>
|
||||
<SubSystem>Windows</SubSystem>
|
||||
<GenerateDebugInformation>false</GenerateDebugInformation>
|
||||
<EnableCOMDATFolding>true</EnableCOMDATFolding>
|
||||
<OptimizeReferences>true</OptimizeReferences>
|
||||
<ModuleDefinitionFile>Source.def</ModuleDefinitionFile>
|
||||
<AdditionalDependencies>kernel32.lib;user32.lib;gdi32.lib;winspool.lib;comdlg32.lib;advapi32.lib;shell32.lib;ole32.lib;oleaut32.lib;uuid.lib;ahadmin.lib;rpcrt4.lib;winhttp.lib;pdh.lib;ws2_32.lib;wbemuuid.lib;iphlpapi.lib;%(AdditionalDependencies)</AdditionalDependencies>
|
||||
</Link>
|
||||
</ItemDefinitionGroup>
|
||||
<ItemGroup>
|
||||
<ClInclude Include="aspnetcore_event.h" />
|
||||
<ClInclude Include="aspnetcore_msg.h" />
|
||||
<ClInclude Include="disconnectcontext.h" />
|
||||
<ClInclude Include="sttimer.h" />
|
||||
<ClInclude Include="outofprocess\forwarderconnection.h" />
|
||||
<ClInclude Include="outofprocess\processmanager.h" />
|
||||
<ClInclude Include="outofprocess\protocolconfig.h" />
|
||||
<ClInclude Include="outofprocess\responseheaderhash.h" />
|
||||
<ClInclude Include="outofprocess\serverprocess.h" />
|
||||
<ClInclude Include="outofprocess\websockethandler.h" />
|
||||
<ClInclude Include="outofprocess\winhttphelper.h" />
|
||||
<ClInclude Include="precomp.hxx" />
|
||||
<ClInclude Include="resource.h" />
|
||||
<ClInclude Include=".\inprocess\inprocessapplication.h" />
|
||||
<ClInclude Include=".\inprocess\inprocesshandler.h" />
|
||||
<ClInclude Include=".\outofprocess\forwardinghandler.h" />
|
||||
<ClInclude Include=".\outofprocess\outprocessapplication.h" />
|
||||
</ItemGroup>
|
||||
<ItemGroup>
|
||||
<ClCompile Include="dllmain.cxx" />
|
||||
<ClCompile Include=".\inprocess\inprocessapplication.cpp" />
|
||||
<ClCompile Include=".\inprocess\inprocesshandler.cpp" />
|
||||
<ClCompile Include=".\outofprocess\forwardinghandler.cpp" />
|
||||
<ClCompile Include=".\outofprocess\outprocessapplication.cpp" />
|
||||
<ClCompile Include="managedexports.cxx" />
|
||||
<ClCompile Include="outofprocess\forwarderconnection.cxx" />
|
||||
<ClCompile Include="outofprocess\processmanager.cxx" />
|
||||
<ClCompile Include="outofprocess\protocolconfig.cxx" />
|
||||
<ClCompile Include="outofprocess\responseheaderhash.cxx" />
|
||||
<ClCompile Include="outofprocess\serverprocess.cxx" />
|
||||
<ClCompile Include="outofprocess\websockethandler.cxx" />
|
||||
<ClCompile Include="outofprocess\winhttphelper.cxx" />
|
||||
</ItemGroup>
|
||||
<ItemGroup>
|
||||
<CustomBuild Include="aspnetcore_msg.mc">
|
||||
<FileType>Document</FileType>
|
||||
<Command Condition="'$(Configuration)|$(Platform)'=='Debug|Win32'">mc %(FullPath)</Command>
|
||||
<Message Condition="'$(Configuration)|$(Platform)'=='Debug|Win32'">Compiling Event Messages ...</Message>
|
||||
<Outputs Condition="'$(Configuration)|$(Platform)'=='Debug|Win32'">%(Filename).rc;%(Filename).h;MSG0409.bin</Outputs>
|
||||
<Command Condition="'$(Configuration)|$(Platform)'=='Debug|x64'">mc %(FullPath)</Command>
|
||||
<Message Condition="'$(Configuration)|$(Platform)'=='Debug|x64'">Compiling Event Messages ...</Message>
|
||||
<Outputs Condition="'$(Configuration)|$(Platform)'=='Debug|x64'">%(Filename).rc;%(Filename).h;MSG0409.bin</Outputs>
|
||||
<Command Condition="'$(Configuration)|$(Platform)'=='Release|Win32'">mc %(FullPath)</Command>
|
||||
<Message Condition="'$(Configuration)|$(Platform)'=='Release|Win32'">Compiling Event Messages ...</Message>
|
||||
<Outputs Condition="'$(Configuration)|$(Platform)'=='Release|Win32'">%(Filename).rc;%(Filename).h;MSG0409.bin</Outputs>
|
||||
<Command Condition="'$(Configuration)|$(Platform)'=='Release|x64'">mc %(FullPath)</Command>
|
||||
<Message Condition="'$(Configuration)|$(Platform)'=='Release|x64'">Compiling Event Messages ...</Message>
|
||||
<Outputs Condition="'$(Configuration)|$(Platform)'=='Release|x64'">%(Filename).rc;%(Filename).h;MSG0409.bin</Outputs>
|
||||
</CustomBuild>
|
||||
</ItemGroup>
|
||||
<ItemGroup>
|
||||
<ProjectReference Include="..\CommonLib\CommonLib.vcxproj">
|
||||
<Project>{55494e58-e061-4c4c-a0a8-837008e72f85}</Project>
|
||||
</ProjectReference>
|
||||
<ProjectReference Include="..\IISLib\IISLib.vcxproj">
|
||||
<Project>{4787a64f-9a3e-4867-a55a-70cb4b2b2ffe}</Project>
|
||||
</ProjectReference>
|
||||
</ItemGroup>
|
||||
<ItemGroup>
|
||||
<ResourceCompile Include="RequestHandler.rc" />
|
||||
</ItemGroup>
|
||||
<ItemGroup>
|
||||
<None Include="Source.def" />
|
||||
</ItemGroup>
|
||||
<Import Project="$(VCTargetsPath)\Microsoft.Cpp.targets" />
|
||||
<ImportGroup Label="ExtensionTargets">
|
||||
</ImportGroup>
|
||||
</Project>
|
||||
|
|
@ -0,0 +1,109 @@
|
|||
<?xml version="1.0" encoding="utf-8"?>
|
||||
<Project ToolsVersion="4.0" xmlns="http://schemas.microsoft.com/developer/msbuild/2003">
|
||||
<ItemGroup>
|
||||
<ClCompile Include="dllmain.cxx" />
|
||||
<ClCompile Include=".\inprocess\inprocessapplication.cpp">
|
||||
<Filter>InProcess</Filter>
|
||||
</ClCompile>
|
||||
<ClCompile Include=".\inprocess\inprocesshandler.cpp">
|
||||
<Filter>InProcess</Filter>
|
||||
</ClCompile>
|
||||
<ClCompile Include="managedexports.cxx">
|
||||
<Filter>InProcess</Filter>
|
||||
</ClCompile>
|
||||
<ClCompile Include=".\outofprocess\forwardinghandler.cpp">
|
||||
<Filter>OutOfProcess</Filter>
|
||||
</ClCompile>
|
||||
<ClCompile Include=".\outofprocess\outprocessapplication.cpp">
|
||||
<Filter>OutOfProcess</Filter>
|
||||
</ClCompile>
|
||||
<ClCompile Include="outofprocess\processmanager.cxx">
|
||||
<Filter>OutOfProcess</Filter>
|
||||
</ClCompile>
|
||||
<ClCompile Include="outofprocess\protocolconfig.cxx">
|
||||
<Filter>OutOfProcess</Filter>
|
||||
</ClCompile>
|
||||
<ClCompile Include="outofprocess\serverprocess.cxx">
|
||||
<Filter>OutOfProcess</Filter>
|
||||
</ClCompile>
|
||||
<ClCompile Include="outofprocess\forwarderconnection.cxx">
|
||||
<Filter>OutOfProcess</Filter>
|
||||
</ClCompile>
|
||||
<ClCompile Include="outofprocess\responseheaderhash.cxx">
|
||||
<Filter>OutOfProcess</Filter>
|
||||
</ClCompile>
|
||||
<ClCompile Include="outofprocess\websockethandler.cxx">
|
||||
<Filter>OutOfProcess</Filter>
|
||||
</ClCompile>
|
||||
<ClCompile Include="outofprocess\winhttphelper.cxx">
|
||||
<Filter>OutOfProcess</Filter>
|
||||
</ClCompile>
|
||||
</ItemGroup>
|
||||
<ItemGroup>
|
||||
<ClInclude Include="precomp.hxx" />
|
||||
<ClInclude Include="resource.h" />
|
||||
<ClInclude Include=".\inprocess\inprocessapplication.h">
|
||||
<Filter>InProcess</Filter>
|
||||
</ClInclude>
|
||||
<ClInclude Include=".\inprocess\inprocesshandler.h">
|
||||
<Filter>InProcess</Filter>
|
||||
</ClInclude>
|
||||
<ClInclude Include=".\outofprocess\forwardinghandler.h">
|
||||
<Filter>OutOfProcess</Filter>
|
||||
</ClInclude>
|
||||
<ClInclude Include="outofprocess\forwarderconnection.h">
|
||||
<Filter>OutOfProcess</Filter>
|
||||
</ClInclude>
|
||||
<ClInclude Include="outofprocess\responseheaderhash.h">
|
||||
<Filter>OutOfProcess</Filter>
|
||||
</ClInclude>
|
||||
<ClInclude Include=".\outofprocess\outprocessapplication.h">
|
||||
<Filter>OutOfProcess</Filter>
|
||||
</ClInclude>
|
||||
<ClInclude Include="outofprocess\processmanager.h">
|
||||
<Filter>OutOfProcess</Filter>
|
||||
</ClInclude>
|
||||
<ClInclude Include="outofprocess\protocolconfig.h">
|
||||
<Filter>OutOfProcess</Filter>
|
||||
</ClInclude>
|
||||
<ClInclude Include="outofprocess\serverprocess.h">
|
||||
<Filter>OutOfProcess</Filter>
|
||||
</ClInclude>
|
||||
<ClInclude Include="outofprocess\responseheaderhash.h">
|
||||
<Filter>OutOfProcess</Filter>
|
||||
</ClInclude>
|
||||
<ClInclude Include="outofprocess\websockethandler.h">
|
||||
<Filter>OutOfProcess</Filter>
|
||||
</ClInclude>
|
||||
<ClInclude Include="outofprocess\winhttphelper.h">
|
||||
<Filter>OutOfProcess</Filter>
|
||||
</ClInclude>
|
||||
<ClInclude Include="disconnectcontext.h" />
|
||||
<ClInclude Include="aspnetcore_event.h" />
|
||||
<ClInclude Include="sttimer.h" />
|
||||
<ClInclude Include="aspnetcore_msg.h" />
|
||||
</ItemGroup>
|
||||
<ItemGroup>
|
||||
<ResourceCompile Include="RequestHandler.rc" />
|
||||
</ItemGroup>
|
||||
<ItemGroup>
|
||||
<None Include="Source.def" />
|
||||
</ItemGroup>
|
||||
<ItemGroup>
|
||||
<ResourceCompile Include="RequestHandler.rc" />
|
||||
</ItemGroup>
|
||||
<ItemGroup>
|
||||
<None Include="Source.def" />
|
||||
</ItemGroup>
|
||||
<ItemGroup>
|
||||
<Filter Include="InProcess">
|
||||
<UniqueIdentifier>{e567abb5-bac5-4f05-a320-5e25dcfc0000}</UniqueIdentifier>
|
||||
</Filter>
|
||||
<Filter Include="OutOfProcess">
|
||||
<UniqueIdentifier>{5568209f-269e-4d0a-bbb7-ba14f874ccb7}</UniqueIdentifier>
|
||||
</Filter>
|
||||
</ItemGroup>
|
||||
<ItemGroup>
|
||||
<CustomBuild Include="aspnetcore_msg.mc" />
|
||||
</ItemGroup>
|
||||
</Project>
|
||||
|
|
@ -0,0 +1,98 @@
|
|||
// Microsoft Visual C++ generated resource script.
|
||||
//
|
||||
#include "resource.h"
|
||||
#include "version.h"
|
||||
|
||||
#define APSTUDIO_READONLY_SYMBOLS
|
||||
/////////////////////////////////////////////////////////////////////////////
|
||||
//
|
||||
// Generated from the TEXTINCLUDE 2 resource.
|
||||
//
|
||||
#include "winres.h"
|
||||
|
||||
/////////////////////////////////////////////////////////////////////////////
|
||||
#undef APSTUDIO_READONLY_SYMBOLS
|
||||
|
||||
/////////////////////////////////////////////////////////////////////////////
|
||||
// English (United States) resources
|
||||
|
||||
#if !defined(AFX_RESOURCE_DLL) || defined(AFX_TARG_ENU)
|
||||
LANGUAGE 9, 1
|
||||
|
||||
#ifdef APSTUDIO_INVOKED
|
||||
/////////////////////////////////////////////////////////////////////////////
|
||||
//
|
||||
// TEXTINCLUDE
|
||||
//
|
||||
|
||||
1 TEXTINCLUDE
|
||||
BEGIN
|
||||
"resource.h\0"
|
||||
END
|
||||
|
||||
2 TEXTINCLUDE
|
||||
BEGIN
|
||||
"#include ""winres.h""\r\n"
|
||||
"\0"
|
||||
END
|
||||
|
||||
3 TEXTINCLUDE
|
||||
BEGIN
|
||||
"\r\n"
|
||||
"\0"
|
||||
END
|
||||
|
||||
#endif // APSTUDIO_INVOKED
|
||||
|
||||
#endif // English (United States) resources
|
||||
/////////////////////////////////////////////////////////////////////////////
|
||||
|
||||
/////////////////////////////////////////////////////////////////////////////
|
||||
//
|
||||
// Version
|
||||
//
|
||||
|
||||
VS_VERSION_INFO VERSIONINFO
|
||||
FILEVERSION FileVersion
|
||||
PRODUCTVERSION ProductVersion
|
||||
FILEFLAGSMASK 0x3fL
|
||||
#ifdef _DEBUG
|
||||
FILEFLAGS 0x1L
|
||||
#else
|
||||
FILEFLAGS 0x0L
|
||||
#endif
|
||||
FILEOS 0x40004L
|
||||
FILETYPE 0x2L
|
||||
FILESUBTYPE 0x0L
|
||||
BEGIN
|
||||
BLOCK "StringFileInfo"
|
||||
BEGIN
|
||||
BLOCK "040904b0"
|
||||
BEGIN
|
||||
VALUE "CompanyName", "Microsoft"
|
||||
VALUE "FileDescription", "IIS ASP.NET Core Module Request Handler"
|
||||
VALUE "FileVersion", FileVersionStr
|
||||
VALUE "InternalName", "aspnetcorerh.dll"
|
||||
VALUE "LegalCopyright", "Copyright (C) 2016"
|
||||
VALUE "OriginalFilename", "aspnetcorerh.dll"
|
||||
VALUE "ProductName", "ASP.NET Core Module"
|
||||
VALUE "ProductVersion", ProductVersionStr
|
||||
END
|
||||
END
|
||||
BLOCK "VarFileInfo"
|
||||
BEGIN
|
||||
VALUE "Translation", 0x409, 1200
|
||||
END
|
||||
END
|
||||
|
||||
|
||||
#ifndef APSTUDIO_INVOKED
|
||||
/////////////////////////////////////////////////////////////////////////////
|
||||
//
|
||||
// Generated from the TEXTINCLUDE 3 resource.
|
||||
//
|
||||
|
||||
|
||||
/////////////////////////////////////////////////////////////////////////////
|
||||
#endif // not APSTUDIO_INVOKED
|
||||
|
||||
|
|
@ -0,0 +1,6 @@
|
|||
LIBRARY aspnetcorerh
|
||||
|
||||
EXPORTS
|
||||
CreateApplication
|
||||
CreateRequestHandler
|
||||
|
||||
|
|
@ -0,0 +1,550 @@
|
|||
// Copyright (c) .NET Foundation. All rights reserved.
|
||||
// Licensed under the MIT License. See License.txt in the project root for license information.
|
||||
|
||||
#ifndef __ASPNETCOREEVENT_H__
|
||||
#define __ASPNETCOREEVENT_H__
|
||||
/*++
|
||||
|
||||
Module Name:
|
||||
|
||||
aspnetcore_event.h
|
||||
|
||||
Abstract:
|
||||
|
||||
Header file has been generated from mof file containing
|
||||
IIS trace event descriptions
|
||||
|
||||
--*/
|
||||
|
||||
//
|
||||
// Start of the new provider class WWWServerTraceProvider,
|
||||
// GUID: {3a2a4e84-4c21-4981-ae10-3fda0d9b0f83}
|
||||
// Description: IIS: WWW Server
|
||||
//
|
||||
|
||||
class WWWServerTraceProvider
|
||||
{
|
||||
public:
|
||||
static
|
||||
LPCGUID
|
||||
GetProviderGuid( VOID )
|
||||
// return GUID for the current event class
|
||||
{
|
||||
static const GUID ProviderGuid =
|
||||
{0x3a2a4e84,0x4c21,0x4981,{0xae,0x10,0x3f,0xda,0x0d,0x9b,0x0f,0x83}};
|
||||
return &ProviderGuid;
|
||||
};
|
||||
enum enumAreaFlags
|
||||
{
|
||||
// AspNetCore module events
|
||||
ANCM = 0x10000
|
||||
};
|
||||
static
|
||||
LPCWSTR
|
||||
TranslateEnumAreaFlagsToString( enum enumAreaFlags EnumValue)
|
||||
{
|
||||
switch( (DWORD) EnumValue )
|
||||
{
|
||||
case 0x10000: return L"ANCM";
|
||||
}
|
||||
return NULL;
|
||||
};
|
||||
|
||||
static
|
||||
BOOL
|
||||
CheckTracingEnabled(
|
||||
IHttpTraceContext * pHttpTraceContext,
|
||||
enumAreaFlags AreaFlags,
|
||||
DWORD dwVerbosity )
|
||||
{
|
||||
HRESULT hr;
|
||||
HTTP_TRACE_CONFIGURATION TraceConfig;
|
||||
TraceConfig.pProviderGuid = GetProviderGuid();
|
||||
hr = pHttpTraceContext->GetTraceConfiguration( &TraceConfig );
|
||||
if ( FAILED( hr ) || !TraceConfig.fProviderEnabled )
|
||||
{
|
||||
return FALSE;
|
||||
}
|
||||
if ( TraceConfig.dwVerbosity >= dwVerbosity &&
|
||||
( TraceConfig.dwAreas == (DWORD) AreaFlags ||
|
||||
( TraceConfig.dwAreas & (DWORD)AreaFlags ) == (DWORD)AreaFlags ) )
|
||||
{
|
||||
return TRUE;
|
||||
}
|
||||
return FALSE;
|
||||
};
|
||||
};
|
||||
|
||||
//
|
||||
// Start of the new event class ANCMEvents,
|
||||
// GUID: {82ADEAD7-12B2-4781-BDCA-5A4B6C757191}
|
||||
// Description: ANCM runtime events
|
||||
//
|
||||
|
||||
class ANCMEvents
|
||||
{
|
||||
public:
|
||||
static
|
||||
LPCGUID
|
||||
GetAreaGuid( VOID )
|
||||
// return GUID for the current event class
|
||||
{
|
||||
static const GUID AreaGuid =
|
||||
{0x82adead7,0x12b2,0x4781,{0xbd,0xca,0x5a,0x4b,0x6c,0x75,0x71,0x91}};
|
||||
return &AreaGuid;
|
||||
};
|
||||
|
||||
//
|
||||
// Event: mof class name ANCMAppStart,
|
||||
// Description: Start application success
|
||||
// EventTypeName: ANCM_START_APPLICATION_SUCCESS
|
||||
// EventType: 1
|
||||
// EventLevel: 4
|
||||
//
|
||||
|
||||
class ANCM_START_APPLICATION_SUCCESS
|
||||
{
|
||||
public:
|
||||
static
|
||||
HRESULT
|
||||
RaiseEvent(
|
||||
IHttpTraceContext * pHttpTraceContext,
|
||||
LPCGUID pContextId,
|
||||
LPCWSTR pAppDescription
|
||||
)
|
||||
//
|
||||
// Raise ANCM_START_APPLICATION_SUCCESS Event
|
||||
//
|
||||
{
|
||||
HTTP_TRACE_EVENT Event;
|
||||
Event.pProviderGuid = WWWServerTraceProvider::GetProviderGuid();
|
||||
Event.dwArea = WWWServerTraceProvider::ANCM;
|
||||
Event.pAreaGuid = ANCMEvents::GetAreaGuid();
|
||||
Event.dwEvent = 1;
|
||||
Event.pszEventName = L"ANCM_START_APPLICATION_SUCCESS";
|
||||
Event.dwEventVersion = 1;
|
||||
Event.dwVerbosity = 4;
|
||||
Event.cEventItems = 2;
|
||||
Event.pActivityGuid = NULL;
|
||||
Event.pRelatedActivityGuid = NULL;
|
||||
Event.dwTimeStamp = 0;
|
||||
Event.dwFlags = HTTP_TRACE_EVENT_FLAG_STATIC_DESCRIPTIVE_FIELDS;
|
||||
|
||||
// pActivityGuid, pRelatedActivityGuid, Timestamp to be filled in by IIS
|
||||
|
||||
HTTP_TRACE_EVENT_ITEM Items[ 2 ];
|
||||
Items[ 0 ].pszName = L"ContextId";
|
||||
Items[ 0 ].dwDataType = HTTP_TRACE_TYPE_LPCGUID; // mof type (object)
|
||||
Items[ 0 ].pbData = (PBYTE) pContextId;
|
||||
Items[ 0 ].cbData = 16;
|
||||
Items[ 0 ].pszDataDescription = NULL;
|
||||
Items[ 1 ].pszName = L"AppDescription";
|
||||
Items[ 1 ].dwDataType = HTTP_TRACE_TYPE_LPCWSTR; // mof type (string)
|
||||
Items[ 1 ].pbData = (PBYTE) pAppDescription;
|
||||
Items[ 1 ].cbData =
|
||||
( Items[ 1 ].pbData == NULL )? 0 : ( sizeof(WCHAR) * (1 + (DWORD) wcslen( (PWSTR) Items[ 1 ].pbData ) ) );
|
||||
Items[ 1 ].pszDataDescription = NULL;
|
||||
Event.pEventItems = Items;
|
||||
pHttpTraceContext->RaiseTraceEvent( &Event );
|
||||
return S_OK;
|
||||
};
|
||||
|
||||
static
|
||||
BOOL
|
||||
IsEnabled(
|
||||
IHttpTraceContext * pHttpTraceContext )
|
||||
// Check if tracing for this event is enabled
|
||||
{
|
||||
return WWWServerTraceProvider::CheckTracingEnabled(
|
||||
pHttpTraceContext,
|
||||
WWWServerTraceProvider::ANCM,
|
||||
4 ); //Verbosity
|
||||
};
|
||||
};
|
||||
//
|
||||
// Event: mof class name ANCMAppStartFail,
|
||||
// Description: Start application failed
|
||||
// EventTypeName: ANCM_START_APPLICATION_FAIL
|
||||
// EventType: 2
|
||||
// EventLevel: 2
|
||||
//
|
||||
|
||||
class ANCM_START_APPLICATION_FAIL
|
||||
{
|
||||
public:
|
||||
static
|
||||
HRESULT
|
||||
RaiseEvent(
|
||||
IHttpTraceContext * pHttpTraceContext,
|
||||
LPCGUID pContextId,
|
||||
LPCWSTR pFailureDescription
|
||||
)
|
||||
//
|
||||
// Raise ANCM_START_APPLICATION_FAIL Event
|
||||
//
|
||||
{
|
||||
HTTP_TRACE_EVENT Event;
|
||||
Event.pProviderGuid = WWWServerTraceProvider::GetProviderGuid();
|
||||
Event.dwArea = WWWServerTraceProvider::ANCM;
|
||||
Event.pAreaGuid = ANCMEvents::GetAreaGuid();
|
||||
Event.dwEvent = 2;
|
||||
Event.pszEventName = L"ANCM_START_APPLICATION_FAIL";
|
||||
Event.dwEventVersion = 1;
|
||||
Event.dwVerbosity = 2;
|
||||
Event.cEventItems = 2;
|
||||
Event.pActivityGuid = NULL;
|
||||
Event.pRelatedActivityGuid = NULL;
|
||||
Event.dwTimeStamp = 0;
|
||||
Event.dwFlags = HTTP_TRACE_EVENT_FLAG_STATIC_DESCRIPTIVE_FIELDS;
|
||||
|
||||
// pActivityGuid, pRelatedActivityGuid, Timestamp to be filled in by IIS
|
||||
|
||||
HTTP_TRACE_EVENT_ITEM Items[ 2 ];
|
||||
Items[ 0 ].pszName = L"ContextId";
|
||||
Items[ 0 ].dwDataType = HTTP_TRACE_TYPE_LPCGUID; // mof type (object)
|
||||
Items[ 0 ].pbData = (PBYTE) pContextId;
|
||||
Items[ 0 ].cbData = 16;
|
||||
Items[ 0 ].pszDataDescription = NULL;
|
||||
Items[ 1 ].pszName = L"FailureDescription";
|
||||
Items[ 1 ].dwDataType = HTTP_TRACE_TYPE_LPCWSTR; // mof type (string)
|
||||
Items[ 1 ].pbData = (PBYTE) pFailureDescription;
|
||||
Items[ 1 ].cbData =
|
||||
( Items[ 1 ].pbData == NULL )? 0 : ( sizeof(WCHAR) * (1 + (DWORD) wcslen( (PWSTR) Items[ 1 ].pbData ) ) );
|
||||
Items[ 1 ].pszDataDescription = NULL;
|
||||
Event.pEventItems = Items;
|
||||
pHttpTraceContext->RaiseTraceEvent( &Event );
|
||||
return S_OK;
|
||||
};
|
||||
|
||||
static
|
||||
BOOL
|
||||
IsEnabled(
|
||||
IHttpTraceContext * pHttpTraceContext )
|
||||
// Check if tracing for this event is enabled
|
||||
{
|
||||
return WWWServerTraceProvider::CheckTracingEnabled(
|
||||
pHttpTraceContext,
|
||||
WWWServerTraceProvider::ANCM,
|
||||
2 ); //Verbosity
|
||||
};
|
||||
};
|
||||
//
|
||||
// Event: mof class name ANCMForwardStart,
|
||||
// Description: Start fardwarding request
|
||||
// EventTypeName: ANCM_REQUEST_FORWARD_START
|
||||
// EventType: 3
|
||||
// EventLevel: 4
|
||||
//
|
||||
|
||||
class ANCM_REQUEST_FORWARD_START
|
||||
{
|
||||
public:
|
||||
static
|
||||
HRESULT
|
||||
RaiseEvent(
|
||||
IHttpTraceContext * pHttpTraceContext,
|
||||
LPCGUID pContextId
|
||||
)
|
||||
//
|
||||
// Raise ANCM_REQUEST_FORWARD_START Event
|
||||
//
|
||||
{
|
||||
HTTP_TRACE_EVENT Event;
|
||||
Event.pProviderGuid = WWWServerTraceProvider::GetProviderGuid();
|
||||
Event.dwArea = WWWServerTraceProvider::ANCM;
|
||||
Event.pAreaGuid = ANCMEvents::GetAreaGuid();
|
||||
Event.dwEvent = 3;
|
||||
Event.pszEventName = L"ANCM_REQUEST_FORWARD_START";
|
||||
Event.dwEventVersion = 1;
|
||||
Event.dwVerbosity = 4;
|
||||
Event.cEventItems = 1;
|
||||
Event.pActivityGuid = NULL;
|
||||
Event.pRelatedActivityGuid = NULL;
|
||||
Event.dwTimeStamp = 0;
|
||||
Event.dwFlags = HTTP_TRACE_EVENT_FLAG_STATIC_DESCRIPTIVE_FIELDS;
|
||||
|
||||
// pActivityGuid, pRelatedActivityGuid, Timestamp to be filled in by IIS
|
||||
|
||||
HTTP_TRACE_EVENT_ITEM Items[ 1 ];
|
||||
Items[ 0 ].pszName = L"ContextId";
|
||||
Items[ 0 ].dwDataType = HTTP_TRACE_TYPE_LPCGUID; // mof type (object)
|
||||
Items[ 0 ].pbData = (PBYTE) pContextId;
|
||||
Items[ 0 ].cbData = 16;
|
||||
Items[ 0 ].pszDataDescription = NULL;
|
||||
Event.pEventItems = Items;
|
||||
pHttpTraceContext->RaiseTraceEvent( &Event );
|
||||
return S_OK;
|
||||
};
|
||||
|
||||
static
|
||||
BOOL
|
||||
IsEnabled(
|
||||
IHttpTraceContext * pHttpTraceContext )
|
||||
// Check if tracing for this event is enabled
|
||||
{
|
||||
return WWWServerTraceProvider::CheckTracingEnabled(
|
||||
pHttpTraceContext,
|
||||
WWWServerTraceProvider::ANCM,
|
||||
4 ); //Verbosity
|
||||
};
|
||||
};
|
||||
//
|
||||
// Event: mof class name ANCMForwardEnd,
|
||||
// Description: Finish forwarding request
|
||||
// EventTypeName: ANCM_REQUEST_FORWARD_END
|
||||
// EventType: 4
|
||||
// EventLevel: 4
|
||||
//
|
||||
|
||||
class ANCM_REQUEST_FORWARD_END
|
||||
{
|
||||
public:
|
||||
static
|
||||
HRESULT
|
||||
RaiseEvent(
|
||||
IHttpTraceContext * pHttpTraceContext,
|
||||
LPCGUID pContextId
|
||||
)
|
||||
//
|
||||
// Raise ANCM_REQUEST_FORWARD_END Event
|
||||
//
|
||||
{
|
||||
HTTP_TRACE_EVENT Event;
|
||||
Event.pProviderGuid = WWWServerTraceProvider::GetProviderGuid();
|
||||
Event.dwArea = WWWServerTraceProvider::ANCM;
|
||||
Event.pAreaGuid = ANCMEvents::GetAreaGuid();
|
||||
Event.dwEvent = 4;
|
||||
Event.pszEventName = L"ANCM_REQUEST_FORWARD_END";
|
||||
Event.dwEventVersion = 1;
|
||||
Event.dwVerbosity = 4;
|
||||
Event.cEventItems = 1;
|
||||
Event.pActivityGuid = NULL;
|
||||
Event.pRelatedActivityGuid = NULL;
|
||||
Event.dwTimeStamp = 0;
|
||||
Event.dwFlags = HTTP_TRACE_EVENT_FLAG_STATIC_DESCRIPTIVE_FIELDS;
|
||||
|
||||
// pActivityGuid, pRelatedActivityGuid, Timestamp to be filled in by IIS
|
||||
|
||||
HTTP_TRACE_EVENT_ITEM Items[ 1 ];
|
||||
Items[ 0 ].pszName = L"ContextId";
|
||||
Items[ 0 ].dwDataType = HTTP_TRACE_TYPE_LPCGUID; // mof type (object)
|
||||
Items[ 0 ].pbData = (PBYTE) pContextId;
|
||||
Items[ 0 ].cbData = 16;
|
||||
Items[ 0 ].pszDataDescription = NULL;
|
||||
Event.pEventItems = Items;
|
||||
pHttpTraceContext->RaiseTraceEvent( &Event );
|
||||
return S_OK;
|
||||
};
|
||||
|
||||
static
|
||||
BOOL
|
||||
IsEnabled(
|
||||
IHttpTraceContext * pHttpTraceContext )
|
||||
// Check if tracing for this event is enabled
|
||||
{
|
||||
return WWWServerTraceProvider::CheckTracingEnabled(
|
||||
pHttpTraceContext,
|
||||
WWWServerTraceProvider::ANCM,
|
||||
4 ); //Verbosity
|
||||
};
|
||||
};
|
||||
//
|
||||
// Event: mof class name ANCMForwardFail,
|
||||
// Description: Forwarding request failure
|
||||
// EventTypeName: ANCM_REQUEST_FORWARD_FAIL
|
||||
// EventType: 5
|
||||
// EventLevel: 2
|
||||
//
|
||||
|
||||
class ANCM_REQUEST_FORWARD_FAIL
|
||||
{
|
||||
public:
|
||||
static
|
||||
HRESULT
|
||||
RaiseEvent(
|
||||
IHttpTraceContext * pHttpTraceContext,
|
||||
LPCGUID pContextId,
|
||||
ULONG ErrorCode
|
||||
)
|
||||
//
|
||||
// Raise ANCM_REQUEST_FORWARD_FAIL Event
|
||||
//
|
||||
{
|
||||
HTTP_TRACE_EVENT Event;
|
||||
Event.pProviderGuid = WWWServerTraceProvider::GetProviderGuid();
|
||||
Event.dwArea = WWWServerTraceProvider::ANCM;
|
||||
Event.pAreaGuid = ANCMEvents::GetAreaGuid();
|
||||
Event.dwEvent = 5;
|
||||
Event.pszEventName = L"ANCM_REQUEST_FORWARD_FAIL";
|
||||
Event.dwEventVersion = 1;
|
||||
Event.dwVerbosity = 2;
|
||||
Event.cEventItems = 2;
|
||||
Event.pActivityGuid = NULL;
|
||||
Event.pRelatedActivityGuid = NULL;
|
||||
Event.dwTimeStamp = 0;
|
||||
Event.dwFlags = HTTP_TRACE_EVENT_FLAG_STATIC_DESCRIPTIVE_FIELDS;
|
||||
|
||||
// pActivityGuid, pRelatedActivityGuid, Timestamp to be filled in by IIS
|
||||
|
||||
HTTP_TRACE_EVENT_ITEM Items[ 2 ];
|
||||
Items[ 0 ].pszName = L"ContextId";
|
||||
Items[ 0 ].dwDataType = HTTP_TRACE_TYPE_LPCGUID; // mof type (object)
|
||||
Items[ 0 ].pbData = (PBYTE) pContextId;
|
||||
Items[ 0 ].cbData = 16;
|
||||
Items[ 0 ].pszDataDescription = NULL;
|
||||
Items[ 1 ].pszName = L"ErrorCode";
|
||||
Items[ 1 ].dwDataType = HTTP_TRACE_TYPE_ULONG; // mof type (uint32)
|
||||
Items[ 1 ].pbData = (PBYTE) &ErrorCode;
|
||||
Items[ 1 ].cbData = 4;
|
||||
Items[ 1 ].pszDataDescription = NULL;
|
||||
Event.pEventItems = Items;
|
||||
pHttpTraceContext->RaiseTraceEvent( &Event );
|
||||
return S_OK;
|
||||
};
|
||||
|
||||
static
|
||||
BOOL
|
||||
IsEnabled(
|
||||
IHttpTraceContext * pHttpTraceContext )
|
||||
// Check if tracing for this event is enabled
|
||||
{
|
||||
return WWWServerTraceProvider::CheckTracingEnabled(
|
||||
pHttpTraceContext,
|
||||
WWWServerTraceProvider::ANCM,
|
||||
2 ); //Verbosity
|
||||
};
|
||||
};
|
||||
//
|
||||
// Event: mof class name ANCMWinHttpCallBack,
|
||||
// Description: Receiving callback from WinHttp
|
||||
// EventTypeName: ANCM_WINHTTP_CALLBACK
|
||||
// EventType: 6
|
||||
// EventLevel: 4
|
||||
//
|
||||
|
||||
class ANCM_WINHTTP_CALLBACK
|
||||
{
|
||||
public:
|
||||
static
|
||||
HRESULT
|
||||
RaiseEvent(
|
||||
IHttpTraceContext * pHttpTraceContext,
|
||||
LPCGUID pContextId,
|
||||
ULONG InternetStatus
|
||||
)
|
||||
//
|
||||
// Raise ANCM_WINHTTP_CALLBACK Event
|
||||
//
|
||||
{
|
||||
HTTP_TRACE_EVENT Event;
|
||||
Event.pProviderGuid = WWWServerTraceProvider::GetProviderGuid();
|
||||
Event.dwArea = WWWServerTraceProvider::ANCM;
|
||||
Event.pAreaGuid = ANCMEvents::GetAreaGuid();
|
||||
Event.dwEvent = 6;
|
||||
Event.pszEventName = L"ANCM_WINHTTP_CALLBACK";
|
||||
Event.dwEventVersion = 1;
|
||||
Event.dwVerbosity = 4;
|
||||
Event.cEventItems = 2;
|
||||
Event.pActivityGuid = NULL;
|
||||
Event.pRelatedActivityGuid = NULL;
|
||||
Event.dwTimeStamp = 0;
|
||||
Event.dwFlags = HTTP_TRACE_EVENT_FLAG_STATIC_DESCRIPTIVE_FIELDS;
|
||||
|
||||
// pActivityGuid, pRelatedActivityGuid, Timestamp to be filled in by IIS
|
||||
|
||||
HTTP_TRACE_EVENT_ITEM Items[ 2 ];
|
||||
Items[ 0 ].pszName = L"ContextId";
|
||||
Items[ 0 ].dwDataType = HTTP_TRACE_TYPE_LPCGUID; // mof type (object)
|
||||
Items[ 0 ].pbData = (PBYTE) pContextId;
|
||||
Items[ 0 ].cbData = 16;
|
||||
Items[ 0 ].pszDataDescription = NULL;
|
||||
Items[ 1 ].pszName = L"InternetStatus";
|
||||
Items[ 1 ].dwDataType = HTTP_TRACE_TYPE_ULONG; // mof type (uint32)
|
||||
Items[ 1 ].pbData = (PBYTE) &InternetStatus;
|
||||
Items[ 1 ].cbData = 4;
|
||||
Items[ 1 ].pszDataDescription = NULL;
|
||||
Event.pEventItems = Items;
|
||||
pHttpTraceContext->RaiseTraceEvent( &Event );
|
||||
return S_OK;
|
||||
};
|
||||
|
||||
static
|
||||
BOOL
|
||||
IsEnabled(
|
||||
IHttpTraceContext * pHttpTraceContext )
|
||||
// Check if tracing for this event is enabled
|
||||
{
|
||||
return WWWServerTraceProvider::CheckTracingEnabled(
|
||||
pHttpTraceContext,
|
||||
WWWServerTraceProvider::ANCM,
|
||||
4 ); //Verbosity
|
||||
};
|
||||
};
|
||||
//
|
||||
// Event: mof class name ANCMForwardEnd,
|
||||
// Description: Inprocess executing request failure
|
||||
// EventTypeName: ANCM_EXECUTE_REQUEST_FAIL
|
||||
// EventType: 7
|
||||
// EventLevel: 2
|
||||
//
|
||||
|
||||
class ANCM_EXECUTE_REQUEST_FAIL
|
||||
{
|
||||
public:
|
||||
static
|
||||
HRESULT
|
||||
RaiseEvent(
|
||||
IHttpTraceContext * pHttpTraceContext,
|
||||
LPCGUID pContextId,
|
||||
ULONG ErrorCode
|
||||
)
|
||||
//
|
||||
// Raise ANCM_EXECUTE_REQUEST_FAIL Event
|
||||
//
|
||||
{
|
||||
HTTP_TRACE_EVENT Event;
|
||||
Event.pProviderGuid = WWWServerTraceProvider::GetProviderGuid();
|
||||
Event.dwArea = WWWServerTraceProvider::ANCM;
|
||||
Event.pAreaGuid = ANCMEvents::GetAreaGuid();
|
||||
Event.dwEvent = 7;
|
||||
Event.pszEventName = L"ANCM_EXECUTE_REQUEST_FAIL";
|
||||
Event.dwEventVersion = 1;
|
||||
Event.dwVerbosity = 2;
|
||||
Event.cEventItems = 2;
|
||||
Event.pActivityGuid = NULL;
|
||||
Event.pRelatedActivityGuid = NULL;
|
||||
Event.dwTimeStamp = 0;
|
||||
Event.dwFlags = HTTP_TRACE_EVENT_FLAG_STATIC_DESCRIPTIVE_FIELDS;
|
||||
|
||||
// pActivityGuid, pRelatedActivityGuid, Timestamp to be filled in by IIS
|
||||
|
||||
HTTP_TRACE_EVENT_ITEM Items[ 2 ];
|
||||
Items[ 0 ].pszName = L"ContextId";
|
||||
Items[ 0 ].dwDataType = HTTP_TRACE_TYPE_LPCGUID; // mof type (object)
|
||||
Items[ 0 ].pbData = (PBYTE) pContextId;
|
||||
Items[ 0 ].cbData = 16;
|
||||
Items[ 0 ].pszDataDescription = NULL;
|
||||
Items[ 1 ].pszName = L"ErrorCode";
|
||||
Items[ 1 ].dwDataType = HTTP_TRACE_TYPE_ULONG; // mof type (uint32)
|
||||
Items[ 1 ].pbData = (PBYTE) &ErrorCode;
|
||||
Items[ 1 ].cbData = 4;
|
||||
Items[ 1 ].pszDataDescription = NULL;
|
||||
Event.pEventItems = Items;
|
||||
pHttpTraceContext->RaiseTraceEvent( &Event );
|
||||
return S_OK;
|
||||
};
|
||||
|
||||
static
|
||||
BOOL
|
||||
IsEnabled(
|
||||
IHttpTraceContext * pHttpTraceContext )
|
||||
// Check if tracing for this event is enabled
|
||||
{
|
||||
return WWWServerTraceProvider::CheckTracingEnabled(
|
||||
pHttpTraceContext,
|
||||
WWWServerTraceProvider::ANCM,
|
||||
2 ); //Verbosity
|
||||
};
|
||||
};
|
||||
};
|
||||
#endif
|
||||
|
|
@ -0,0 +1,165 @@
|
|||
/*++
|
||||
|
||||
Copyright (c) .NET Foundation. All rights reserved.
|
||||
Licensed under the MIT License. See License.txt in the project root for license information.
|
||||
|
||||
Module Name:
|
||||
|
||||
aspnetcore_msg.mc
|
||||
|
||||
Abstract:
|
||||
|
||||
Asp.Net Core Module localizable messages.
|
||||
|
||||
--*/
|
||||
|
||||
|
||||
#ifndef _ASPNETCORE_MSG_H_
|
||||
#define _ASPNETCORE_MSG_H_
|
||||
|
||||
//
|
||||
// Values are 32 bit values laid out as follows:
|
||||
//
|
||||
// 3 3 2 2 2 2 2 2 2 2 2 2 1 1 1 1 1 1 1 1 1 1
|
||||
// 1 0 9 8 7 6 5 4 3 2 1 0 9 8 7 6 5 4 3 2 1 0 9 8 7 6 5 4 3 2 1 0
|
||||
// +---+-+-+-----------------------+-------------------------------+
|
||||
// |Sev|C|R| Facility | Code |
|
||||
// +---+-+-+-----------------------+-------------------------------+
|
||||
//
|
||||
// where
|
||||
//
|
||||
// Sev - is the severity code
|
||||
//
|
||||
// 00 - Success
|
||||
// 01 - Informational
|
||||
// 10 - Warning
|
||||
// 11 - Error
|
||||
//
|
||||
// C - is the Customer code flag
|
||||
//
|
||||
// R - is a reserved bit
|
||||
//
|
||||
// Facility - is the facility code
|
||||
//
|
||||
// Code - is the facility's status code
|
||||
//
|
||||
//
|
||||
// Define the facility codes
|
||||
//
|
||||
|
||||
|
||||
//
|
||||
// Define the severity codes
|
||||
//
|
||||
|
||||
|
||||
//
|
||||
// MessageId: ASPNETCORE_EVENT_PROCESS_START_ERROR
|
||||
//
|
||||
// MessageText:
|
||||
//
|
||||
// %1
|
||||
//
|
||||
#define ASPNETCORE_EVENT_PROCESS_START_ERROR ((DWORD)0x000003E8L)
|
||||
|
||||
//
|
||||
// MessageId: ASPNETCORE_EVENT_PROCESS_START_SUCCESS
|
||||
//
|
||||
// MessageText:
|
||||
//
|
||||
// %1
|
||||
//
|
||||
#define ASPNETCORE_EVENT_PROCESS_START_SUCCESS ((DWORD)0x000003E9L)
|
||||
|
||||
//
|
||||
// MessageId: ASPNETCORE_EVENT_PROCESS_CRASH
|
||||
//
|
||||
// MessageText:
|
||||
//
|
||||
// %1
|
||||
//
|
||||
#define ASPNETCORE_EVENT_PROCESS_CRASH ((DWORD)0x000003EAL)
|
||||
|
||||
//
|
||||
// MessageId: ASPNETCORE_EVENT_RAPID_FAIL_COUNT_EXCEEDED
|
||||
//
|
||||
// MessageText:
|
||||
//
|
||||
// %1
|
||||
//
|
||||
#define ASPNETCORE_EVENT_RAPID_FAIL_COUNT_EXCEEDED ((DWORD)0x000003EBL)
|
||||
|
||||
//
|
||||
// MessageId: ASPNETCORE_EVENT_CONFIG_ERROR
|
||||
//
|
||||
// MessageText:
|
||||
//
|
||||
// %1
|
||||
//
|
||||
#define ASPNETCORE_EVENT_CONFIG_ERROR ((DWORD)0x000003ECL)
|
||||
|
||||
//
|
||||
// MessageId: ASPNETCORE_EVENT_GRACEFUL_SHUTDOWN_FAILURE
|
||||
//
|
||||
// MessageText:
|
||||
//
|
||||
// %1
|
||||
//
|
||||
#define ASPNETCORE_EVENT_GRACEFUL_SHUTDOWN_FAILURE ((DWORD)0x000003EDL)
|
||||
|
||||
//
|
||||
// MessageId: ASPNETCORE_EVENT_SENT_SHUTDOWN_HTTP_REQUEST
|
||||
//
|
||||
// MessageText:
|
||||
//
|
||||
// %1
|
||||
//
|
||||
#define ASPNETCORE_EVENT_SENT_SHUTDOWN_HTTP_REQUEST ((DWORD)0x000003EEL)
|
||||
|
||||
//
|
||||
// MessageId: ASPNETCORE_EVENT_LOAD_CLR_FALIURE
|
||||
//
|
||||
// MessageText:
|
||||
//
|
||||
// %1
|
||||
//
|
||||
#define ASPNETCORE_EVENT_LOAD_CLR_FALIURE ((DWORD)0x000003EFL)
|
||||
|
||||
//
|
||||
// MessageId: ASPNETCORE_EVENT_DUPLICATED_INPROCESS_APP
|
||||
//
|
||||
// MessageText:
|
||||
//
|
||||
// %1
|
||||
//
|
||||
#define ASPNETCORE_EVENT_DUPLICATED_INPROCESS_APP ((DWORD)0x000003F0L)
|
||||
|
||||
//
|
||||
// MessageId: ASPNETCORE_EVENT_MIXED_HOSTING_MODEL_ERROR
|
||||
//
|
||||
// MessageText:
|
||||
//
|
||||
// %1
|
||||
//
|
||||
#define ASPNETCORE_EVENT_MIXED_HOSTING_MODEL_ERROR ((DWORD)0x000003F1L)
|
||||
|
||||
//
|
||||
// MessageId: ASPNETCORE_EVENT_ADD_APPLICATION_ERROR
|
||||
//
|
||||
// MessageText:
|
||||
//
|
||||
// %1
|
||||
//
|
||||
#define ASPNETCORE_EVENT_ADD_APPLICATION_ERROR ((DWORD)0x000003F2L)
|
||||
|
||||
//
|
||||
// MessageId: ASPNETCORE_EVENT_INPROCESS_THREAD_EXIT
|
||||
//
|
||||
// MessageText:
|
||||
//
|
||||
// %1
|
||||
//
|
||||
#define ASPNETCORE_EVENT_INPROCESS_THREAD_EXIT ((DWORD)0x000003F3L)
|
||||
|
||||
|
||||
#endif // _ASPNETCORE_MODULE_MSG_H_
|
||||
Some files were not shown because too many files have changed in this diff Show More
Loading…
Reference in New Issue