internal StreamingMessage ProcessWorkerInitRequest(StreamingMessage request)
{
var workerInitRequest = request.WorkerInitRequest;
Environment.SetEnvironmentVariable("AZUREPS_HOST_ENVIRONMENT", $"AzureFunctions/{workerInitRequest.HostVersion}");
StreamingMessage response = NewStreamingMessageTemplate(
request.RequestId,
StreamingMessage.ContentOneofCase.WorkerInitResponse,
out StatusResult status);
response.WorkerInitResponse.Capabilities.Add("RpcHttpBodyOnly", "true");
// If the environment variable is set, spin up the custom named pipe server.
// This is typically used for debugging. It will throw a friendly exception if the
// pipe name is not a valid pipename.
string pipeName = Environment.GetEnvironmentVariable("PSWorkerCustomPipeName");
if (!string.IsNullOrEmpty(pipeName))
{
RpcLogger.WriteSystemLog(LogLevel.Trace, string.Format(PowerShellWorkerStrings.SpecifiedCustomPipeName, pipeName));
RemoteSessionNamedPipeServer.CreateCustomNamedPipeServer(pipeName);
}
return(response);
}
public void TestCustomPipeNameCreationTooLongOnNonWindows()
{
var longPipeName = "DoggoipsumwaggywagssmolborkingdoggowithalongsnootforpatsdoingmeafrightenporgoYapperporgolongwatershoobcloudsbigolpupperlengthboy";
if (!Platform.IsWindows)
{
Assert.Throws <InvalidOperationException>(() =>
RemoteSessionNamedPipeServer.CreateCustomNamedPipeServer(longPipeName));
}
else
{
RemoteSessionNamedPipeServer.CreateCustomNamedPipeServer(longPipeName);
Assert.True(File.Exists(GetPipePath(longPipeName)));
}
}
public void TestCustomPipeNameCreation()
{
string pipeNameForFirstCall = Path.GetRandomFileName();
string pipeNameForSecondCall = Path.GetRandomFileName();
RemoteSessionNamedPipeServer.CreateCustomNamedPipeServer(pipeNameForFirstCall);
Assert.True(File.Exists(GetPipePath(pipeNameForFirstCall)));
// The second call to this method would override the first named pipe.
RemoteSessionNamedPipeServer.CreateCustomNamedPipeServer(pipeNameForSecondCall);
Assert.True(File.Exists(GetPipePath(pipeNameForSecondCall)));
// Previous pipe should have been cleaned up.
Assert.False(File.Exists(GetPipePath(pipeNameForFirstCall)));
}
internal StreamingMessage ProcessWorkerInitRequest(StreamingMessage request)
{
StreamingMessage response = NewStreamingMessageTemplate(
request.RequestId,
StreamingMessage.ContentOneofCase.WorkerInitResponse,
out StatusResult status);
// If the environment variable is set, spin up the custom named pipe server.
// This is typically used for debugging. It will throw a friendly exception if the
// pipe name is not a valid pipename.
string pipeName = Environment.GetEnvironmentVariable("PSWorkerCustomPipeName");
if (!string.IsNullOrEmpty(pipeName))
{
RpcLogger.WriteSystemLog(LogLevel.Trace, string.Format(PowerShellWorkerStrings.SpecifiedCustomPipeName, pipeName));
RemoteSessionNamedPipeServer.CreateCustomNamedPipeServer(pipeName);
}
return(response);
}