diff --git a/releaseNote.md b/releaseNote.md index b420db971d7..e54496fcc6f 100644 --- a/releaseNote.md +++ b/releaseNote.md @@ -1,6 +1,7 @@ ## Features ## Bugs - Fixed an issue where container environment variables names or values could escape the docker command (#2108) +- Sanitize Windows ENVs (#2280) ## Misc diff --git a/src/Runner.Sdk/ProcessInvoker.cs b/src/Runner.Sdk/ProcessInvoker.cs index 78a9f2dd27e..43117ef47c5 100644 --- a/src/Runner.Sdk/ProcessInvoker.cs +++ b/src/Runner.Sdk/ProcessInvoker.cs @@ -264,7 +264,17 @@ public async Task ExecuteAsync( { foreach (KeyValuePair kvp in environment) { +#if OS_WINDOWS + string tempKey = String.IsNullOrWhiteSpace(kvp.Key) ? kvp.Key : kvp.Key.Split('\0')[0]; + string tempValue = String.IsNullOrWhiteSpace(kvp.Value) ? kvp.Value : kvp.Value.Split('\0')[0]; + if(!String.IsNullOrWhiteSpace(tempKey)) + { + _proc.StartInfo.Environment[tempKey] = tempValue; + } +#else _proc.StartInfo.Environment[kvp.Key] = kvp.Value; + +#endif } } diff --git a/src/Test/L0/ProcessInvokerL0.cs b/src/Test/L0/ProcessInvokerL0.cs index 629abb0a010..3f0985175cc 100644 --- a/src/Test/L0/ProcessInvokerL0.cs +++ b/src/Test/L0/ProcessInvokerL0.cs @@ -129,7 +129,76 @@ public async Task SetCIEnv() } } } +#if OS_WINDOWS + [Fact] + [Trait("Level", "L0")] + [Trait("Category", "Common")] + public async Task SetTestEnvWithNullInKey() + { + using (TestHostContext hc = new(this)) + { + Tracing trace = hc.GetTrace(); + + Int32 exitCode = -1; + var processInvoker = new ProcessInvokerWrapper(); + processInvoker.Initialize(hc); + var stdout = new List(); + var stderr = new List(); + processInvoker.OutputDataReceived += (object sender, ProcessDataReceivedEventArgs e) => + { + trace.Info(e.Data); + stdout.Add(e.Data); + }; + processInvoker.ErrorDataReceived += (object sender, ProcessDataReceivedEventArgs e) => + { + trace.Info(e.Data); + stderr.Add(e.Data); + }; + + exitCode = await processInvoker.ExecuteAsync("", "cmd.exe", "/c \"echo %TEST%\"", new Dictionary() { { "TEST\0second", "first" } }, CancellationToken.None); + + + trace.Info("Exit Code: {0}", exitCode); + Assert.Equal(0, exitCode); + Assert.Equal("first", stdout.First(x => !string.IsNullOrWhiteSpace(x))); + + } + } + [Fact] + [Trait("Level", "L0")] + [Trait("Category", "Common")] + public async Task SetTestEnvWithNullInValue() + { + using (TestHostContext hc = new(this)) + { + Tracing trace = hc.GetTrace(); + + Int32 exitCode = -1; + var processInvoker = new ProcessInvokerWrapper(); + processInvoker.Initialize(hc); + var stdout = new List(); + var stderr = new List(); + processInvoker.OutputDataReceived += (object sender, ProcessDataReceivedEventArgs e) => + { + trace.Info(e.Data); + stdout.Add(e.Data); + }; + processInvoker.ErrorDataReceived += (object sender, ProcessDataReceivedEventArgs e) => + { + trace.Info(e.Data); + stderr.Add(e.Data); + }; + + exitCode = await processInvoker.ExecuteAsync("", "cmd.exe", "/c \"echo %TEST%\"", new Dictionary() { { "TEST", "first\0second" } }, CancellationToken.None); + + trace.Info("Exit Code: {0}", exitCode); + Assert.Equal(0, exitCode); + Assert.Equal("first", stdout.First(x => !string.IsNullOrWhiteSpace(x))); + + } + } +#endif [Fact] [Trait("Level", "L0")] [Trait("Category", "Common")] diff --git a/src/runnerversion b/src/runnerversion index 05eb5382601..e8555cba73e 100644 --- a/src/runnerversion +++ b/src/runnerversion @@ -1 +1 @@ -2.293.1 +2.293.2