diff --git a/build.psm1 b/build.psm1 index d30ce8ee24..625996e4a2 100644 --- a/build.psm1 +++ b/build.psm1 @@ -331,6 +331,10 @@ cmd.exe /C cd /d "$location" "&" "$($vcVarsPath)\vcvarsall.bat" "$NativeHostArch 'pwrshplugin.pdb' ) Build-NativeWindowsBinaries "ON" $coreClrBinaries + + # Place the remoting configuration script in the same directory + # as the binary so it will get published. + Copy-Item .\Install-PowerShellRemoting.ps1 ($script:Options).Top } } finally { Pop-Location diff --git a/src/powershell-native/Install-PowerShellRemoting.ps1 b/src/powershell-native/Install-PowerShellRemoting.ps1 new file mode 100644 index 0000000000..e481797ede --- /dev/null +++ b/src/powershell-native/Install-PowerShellRemoting.ps1 @@ -0,0 +1,165 @@ +##################################################################################################### +# +# Registers the WinRM endpoint for this instance of PowerShell. +# +# Assumptions: +# 1. This script is run from within the PowerShell that it will register as a WinRM endpoint. +# 2. The CoreCLR and the the PowerShell assemblies are side-by-side in $PSHOME +# 3. Plugins are registered by version number. Only one plugin can be automatically registered +# per PowerShell version. However, multiple endpoints may be manually registered for a given +# plugin. +# +##################################################################################################### + +param +( + [Switch]$Verbose +) + +function Register-WinRmPlugin +{ + param + ( + # + # Expected Example: + # %windir%\\system32\\PowerShell\\6.0.0\\pwrshplugin.dll + # + [string] + [parameter(Mandatory=$true)] + [ValidateNotNullOrEmpty()] + $pluginAbsolutePath, + + # + # Expected Example: microsoft.ops.5.1 + # + [string] + [parameter(Mandatory=$true)] + [ValidateNotNullOrEmpty()] + $pluginEndpointName + ) + + $header = "Windows Registry Editor Version 5.00`n`n" + + $regKeyFormatString = "[HKEY_LOCAL_MACHINE\SOFTWARE\Microsoft\Windows\CurrentVersion\WSMAN\Plugin\{0}]`n" + + $regKeyName = '"ConfigXML"="{0}"' + + # + # Example Values: + # + # Filename = %windir%\\system32\\PowerShell\\6.0.0\\pwrshplugin.dll + # Name = PowerShell.6.0.0 + # + $regKeyValueFormatString = '' + $valueString = $regKeyValueFormatString -f $pluginEndpointName, $pluginAbsolutePath + $keyValuePair = $regKeyName -f $valueString + + $regKey = $regKeyFormatString -f $pluginEndpointName + + $fileName = "$pluginEndpointName.reg" + + Set-Content -path .\$fileName "$header$regKey$keyValuePair`n" + + Write-Verbose "Performing WinRM registration with: $fileName" + reg.exe import .\$fileName + + # Clean up +# Remove-Item .\$fileName +} + +function Generate-PluginConfigFile +{ + param + ( + [string] + [parameter(Mandatory=$true)] + [ValidateNotNullOrEmpty()] + $pluginFile + ) + + # This always overwrites the file with a new version of it if the + # script is invoked multiple times. + Set-Content -Path $pluginFile -Value "PSHOMEDIR=$PSHOME" + Add-Content -Path $pluginFile -Value "CORECLRDIR=$PSHOME" + + Write-Verbose "Created Plugin Config File: $pluginFile" +} + +###################### +# # +# Install the plugin # +# # +###################### + +if (! ([Security.Principal.WindowsPrincipal] [Security.Principal.WindowsIdentity]::GetCurrent()).IsInRole([Security.Principal.WindowsBuiltInRole] "Administrator")) +{ + Write-Error "WinRM registration requires Administrator rights. To run this cmdlet, start PowerShell with the `"Run as administrator`" option." + Break +} + +$originalVerbosePreference = $VerbosePreference +if ($Verbose) +{ + $VerbosePreference = "Continue" +} + +$powershellVersion = $PSVersionTable.PSVersion +$pluginBasePath = Join-Path "$env:WINDIR\System32\PowerShell" $powerShellVersion + +$resolvedPluginAbsolutePath = "" +if (! (Test-Path $pluginBasePath)) +{ + Write-Verbose "Creating $pluginBasePath" + $resolvedPluginAbsolutePath = New-Item -Type Directory -Path $pluginBasePath +} +else +{ + $resolvedPluginAbsolutePath = Resolve-Path $pluginBasePath +} + +# The registration reg file requires "\\" instead of "\" in its path so it is properly escaped in the XML +$pluginRawPath = Join-Path $resolvedPluginAbsolutePath "pwrshplugin.dll" +$fixedPluginPath = $pluginRawPath -replace '\\','\\' + +# This is forced to ensure the the file is placed correctly +copy-item $PSHOME\pwrshplugin.dll $resolvedPluginAbsolutePath -Force -Verbose + +$pluginFile = Join-Path $resolvedPluginAbsolutePath "RemotePowerShellConfig.txt" +Generate-PluginConfigFile $pluginFile + +$pluginEndpointName = "powershell.$powershellVersion" + +# Register the plugin +Register-WinRmPlugin $fixedPluginPath $pluginEndpointName + +# Reset verbosity +if ($Verbose) +{ + $VerbosePreference = $originalVerbosePreference +} + +#################################################################### +# # +# Validations to confirm that everything was registered correctly. # +# # +#################################################################### + +if (! (Test-Path $pluginFile)) +{ + throw "WinRM Plugin configuration file not created. Expected = $pluginFile" +} + +if (! (Test-Path $resolvedPluginAbsolutePath\pwrshplugin.dll)) +{ + throw "WinRM Plugin DLL missing. Expected = $resolvedPluginAbsolutePath\pwrshplugin.dll" +} + +try +{ + Get-PSSessionConfiguration $pluginEndpointName +} +catch [Microsoft.PowerShell.Commands.WriteErrorException] +{ + Write-Error "No remoting session configuration matches the name $pluginEndpointName." +} + diff --git a/src/powershell-win-core/project.json b/src/powershell-win-core/project.json index 98413566fe..b8c815db42 100644 --- a/src/powershell-win-core/project.json +++ b/src/powershell-win-core/project.json @@ -46,6 +46,9 @@ } }, "include": [ + "pwrshplugin.dll", + "pwrshplugin.pdb", + "Install-PowerShellRemoting.ps1", "../../powershell.version" ], },