Skip to content

Commit

Permalink
[wasm] spread WS based timers over next 6 minutes to prevent heavy th…
Browse files Browse the repository at this point in the history
…rottling (#57745)

spread WS based timers over next 6 minutes to prevent heavy throttling
  • Loading branch information
pavelsavara committed Aug 26, 2021
1 parent 1fa304a commit 5eaa911
Show file tree
Hide file tree
Showing 8 changed files with 334 additions and 10 deletions.
Expand Up @@ -19,6 +19,8 @@ Project("{2150E333-8FDC-42A3-9474-1A3956D46DE8}") = "ref", "ref", "{A0314AC5-E49
EndProject
Project("{2150E333-8FDC-42A3-9474-1A3956D46DE8}") = "src", "src", "{6F9A42A0-A04B-4CD0-B8C9-9A728274C851}"
EndProject
Project("{9A19103F-16F7-4668-BE54-9A1E7A4F7556}") = "System.Net.WebSockets.Client.Wasm.Tests", "tests\wasm\System.Net.WebSockets.Client.Wasm.Tests.csproj", "{CA20532A-33B3-4DC0-92D2-EA6D7987D59F}"
EndProject
Global
GlobalSection(SolutionConfigurationPlatforms) = preSolution
Debug|Any CPU = Debug|Any CPU
Expand Down Expand Up @@ -53,6 +55,10 @@ Global
{6B9721B2-00D0-41EC-96B9-6428CF1830CF}.Debug|Any CPU.Build.0 = Debug|Any CPU
{6B9721B2-00D0-41EC-96B9-6428CF1830CF}.Release|Any CPU.ActiveCfg = Release|Any CPU
{6B9721B2-00D0-41EC-96B9-6428CF1830CF}.Release|Any CPU.Build.0 = Release|Any CPU
{CA20532A-33B3-4DC0-92D2-EA6D7987D59F}.Debug|Any CPU.ActiveCfg = Debug|Any CPU
{CA20532A-33B3-4DC0-92D2-EA6D7987D59F}.Debug|Any CPU.Build.0 = Debug|Any CPU
{CA20532A-33B3-4DC0-92D2-EA6D7987D59F}.Release|Any CPU.ActiveCfg = Release|Any CPU
{CA20532A-33B3-4DC0-92D2-EA6D7987D59F}.Release|Any CPU.Build.0 = Release|Any CPU
EndGlobalSection
GlobalSection(SolutionProperties) = preSolution
HideSolutionNode = FALSE
Expand All @@ -65,6 +71,7 @@ Global
{0CD4C24D-7746-46F0-8D47-A396882B5468} = {6F9A42A0-A04B-4CD0-B8C9-9A728274C851}
{8CD4D190-F656-4970-9AE9-4A9F8B30A2F8} = {6F9A42A0-A04B-4CD0-B8C9-9A728274C851}
{59A23CAB-D098-495F-A467-74C7553FF5BB} = {6F9A42A0-A04B-4CD0-B8C9-9A728274C851}
{CA20532A-33B3-4DC0-92D2-EA6D7987D59F} = {BEE2F256-0489-4809-AB20-27ADB2D0E10C}
EndGlobalSection
GlobalSection(ExtensibilityGlobals) = postSolution
SolutionGuid = {D91D7DC5-24CC-4716-A357-8170C4EB1C32}
Expand Down
Expand Up @@ -255,12 +255,12 @@ public async Task ReceiveAsync_MultipleOutstandingReceiveOperations_Throws(Uri s
"ReceiveAsync"),
ex.Message);

Assert.Equal(WebSocketState.Aborted, cws.State);
Assert.True(WebSocketState.Aborted == cws.State, cws.State+" state when InvalidOperationException");
}
else if (ex is WebSocketException)
{
// Multiple cases.
Assert.Equal(WebSocketState.Aborted, cws.State);
Assert.True(WebSocketState.Aborted == cws.State, cws.State + " state when WebSocketException");

WebSocketError errCode = (ex as WebSocketException).WebSocketErrorCode;
Assert.True(
Expand All @@ -269,7 +269,7 @@ public async Task ReceiveAsync_MultipleOutstandingReceiveOperations_Throws(Uri s
}
else if (ex is OperationCanceledException)
{
Assert.Equal(WebSocketState.Aborted, cws.State);
Assert.True(WebSocketState.Aborted == cws.State, cws.State + " state when OperationCanceledException");
}
else
{
Expand Down
@@ -0,0 +1,181 @@
// Licensed to the .NET Foundation under one or more agreements.
// The .NET Foundation licenses this file to you under the MIT license.

using System.Collections.Generic;
using System.Linq;
using System.Net.WebSockets.Client.Tests;
using System.Runtime.InteropServices;
using System.Text;
using System.Threading;
using System.Threading.Tasks;

using Xunit;
using Xunit.Abstractions;
using Xunit.Sdk;

namespace System.Net.WebSockets.Client.Wasm.Tests
{
// https://developer.chrome.com/blog/timer-throttling-in-chrome-88/
// https://docs.google.com/document/d/11FhKHRcABGS4SWPFGwoL6g0ALMqrFKapCk5ZTKKupEk/view
// requires chromium based browser
// requires minimized browser or browser tab out of focus, browser can't be headless
// requires --enable-features=IntensiveWakeUpThrottling:grace_period_seconds/1 chromeDriver flags
// doesn't work with --disable-background-timer-throttling
[TestCaseOrderer("System.Net.WebSockets.Client.Wasm.Tests.AlphabeticalOrderer", "System.Net.WebSockets.Client.Wasm.Tests")]
public class BrowserTimerThrottlingTest : ClientWebSocketTestBase
{
public static bool IsBrowser => RuntimeInformation.IsOSPlatform(OSPlatform.Create("BROWSER"));
const double moreThanLightThrottlingThreshold = 1900;
const double detectLightThrottlingThreshold = 900;
const double webSocketMessageFrequency = 45000;
const double fastTimeoutFrequency = 100;

public BrowserTimerThrottlingTest(ITestOutputHelper output) : base(output) { }

[ConditionalFact(nameof(PlatformDetection.IsBrowser))]
[OuterLoop] // involves long delay
// this test is influenced by usage of WS on the same browser tab in previous unit tests. we may need to wait long time for it to fizzle down
public async Task DotnetTimersAreHeavilyThrottledWithoutWebSocket()
{
double maxDelayMs = 0;
double maxLightDelayMs = 0;
DateTime start = DateTime.Now;
CancellationTokenSource cts = new CancellationTokenSource();

using (var timer = new Timers.Timer(fastTimeoutFrequency))
{
DateTime last = DateTime.Now;
timer.AutoReset = true;
timer.Enabled = true;
timer.Elapsed += (object? source, Timers.ElapsedEventArgs? e) =>
{
var ms = (e.SignalTime - last).TotalMilliseconds;
if (maxDelayMs < ms)
{
maxDelayMs = ms;
}
if (ms > moreThanLightThrottlingThreshold)
{
#if DEBUG
Console.WriteLine("Too slow tick " + ms);
#endif
// stop, we are throttled heavily, this is what we are looking for
cts.Cancel();
}
else if (ms > detectLightThrottlingThreshold)
{
maxLightDelayMs = ms;
// we are lightly throttled
#if DEBUG
Console.WriteLine("Slow tick NO-WS " + ms);
#endif
}
last = e.SignalTime;
};

// test it for 10 minutes
try { await Task.Delay(10 * 60 * 1000, cts.Token); } catch (Exception) { }
timer.Close();
}
Assert.True(maxDelayMs > detectLightThrottlingThreshold, "Expect that it throttled lightly " + maxDelayMs);
Assert.True(maxDelayMs > moreThanLightThrottlingThreshold, "Expect that it was heavily throttled " + maxDelayMs);
}

[ConditionalFact(nameof(WebSocketsSupported), nameof(PlatformDetection.IsBrowser))]
[OuterLoop] // involves long delay
public async Task WebSocketKeepsDotnetTimersOnlyLightlyThrottled()
{
double maxDelayMs = 0;
double maxLightDelayMs = 0;
DateTime start = DateTime.Now;
CancellationTokenSource cts = new CancellationTokenSource();

using (ClientWebSocket cws = await WebSocketHelper.GetConnectedWebSocket(Test.Common.Configuration.WebSockets.RemoteEchoServer, TimeOutMilliseconds, _output))
{
await SendAndReceive(cws, "test");
using (var timer = new Timers.Timer(fastTimeoutFrequency))
{
DateTime last = DateTime.Now;
DateTime lastSent = DateTime.MinValue;
timer.AutoReset = true;
timer.Enabled = true;
timer.Elapsed += async (object? source, Timers.ElapsedEventArgs? e) =>
{
var ms = (e.SignalTime - last).TotalMilliseconds;
var msSent = (e.SignalTime - lastSent).TotalMilliseconds;
if (maxDelayMs < ms)
{
maxDelayMs = ms;
}
if (ms > moreThanLightThrottlingThreshold)
{
// fail fast, we are throttled heavily
#if DEBUG
Console.WriteLine("Too slow tick " + ms);
#endif
cts.Cancel();
}
else if (ms > detectLightThrottlingThreshold)
{
maxLightDelayMs = ms;
// we are lightly throttled
#if DEBUG
Console.WriteLine("Slow tick WS " + ms);
#endif
}
if (msSent > webSocketMessageFrequency)
{
await SendAndReceive(cws, "test");
lastSent = DateTime.Now;
}
last = e.SignalTime;
};

// test it for 10 minutes
try { await Task.Delay(10 * 60 * 1000, cts.Token); } catch (Exception) { }
timer.Close();
}
await cws.CloseAsync(WebSocketCloseStatus.NormalClosure, "WebSocketKeepsDotnetTimersOnlyLightlyThrottled", CancellationToken.None);
}
Assert.True(maxDelayMs > detectLightThrottlingThreshold, "Expect that it throttled lightly " + maxDelayMs);
Assert.True(maxDelayMs < moreThanLightThrottlingThreshold, "Expect that it wasn't heavily throttled " + maxDelayMs);
}

private async static Task SendAndReceive(ClientWebSocket cws, string message)
{
try
{
byte[] buffer = Encoding.UTF8.GetBytes(message);
await cws.SendAsync(new ArraySegment<byte>(buffer), WebSocketMessageType.Text, true, CancellationToken.None);

var receiveBuffer = new byte[100];
var receiveSegment = new ArraySegment<byte>(receiveBuffer);
WebSocketReceiveResult recvRet = await cws.ReceiveAsync(receiveSegment, CancellationToken.None);
#if DEBUG
Console.WriteLine("SendAndReceive");
#endif
}
catch (OperationCanceledException)
{
}
catch (Exception ex)
{
#if DEBUG
Console.WriteLine("SendAndReceive fail:" + ex);
#endif
}
}
}

// this is just for convinience, as the second test has side-effect to running page, the first test would take longer if they are in another order
public class AlphabeticalOrderer : ITestCaseOrderer
{
public IEnumerable<TTestCase> OrderTestCases<TTestCase>(IEnumerable<TTestCase> testCases)
where TTestCase : ITestCase
{
List<TTestCase> result = testCases.ToList();
result.Sort((x, y) => StringComparer.Ordinal.Compare(x.TestMethod.Method.Name, y.TestMethod.Method.Name));
return result;
}
}
}
@@ -0,0 +1,66 @@
<Project Sdk="Microsoft.NET.Sdk">
<PropertyGroup>
<StringResourcesPath>../../src/Resources/Strings.resx</StringResourcesPath>
<TargetFrameworks>$(NetCoreAppCurrent)-Browser</TargetFrameworks>
<DefineConstants>$(DefineConstants);NETSTANDARD</DefineConstants>
</PropertyGroup>

<PropertyGroup Condition="'$(TargetOS)' == 'Browser'">
<WasmXHarnessArgs>--background-throttling</WasmXHarnessArgs>
<Scenario>WasmTestOnBrowser</Scenario>
<TestArchiveTestsRoot>$(TestArchiveRoot)browseronly/</TestArchiveTestsRoot>
<TestArchiveTestsDir>$(TestArchiveTestsRoot)$(OSPlatformConfig)/</TestArchiveTestsDir>
<DefineConstants>$(DefineConstants);TARGET_BROWSER</DefineConstants>
</PropertyGroup>

<Import Condition="'$(TargetOS)' == 'Browser'" Project="$(CommonTestPath)System/Net/Prerequisites/LocalEchoServer.props" />

<!-- Browser specific files -->
<ItemGroup Condition="'$(TargetOS)' == 'Browser'">
<ProjectReference Include="$(CommonTestPath)System/Net/Prerequisites/NetCoreServer/NetCoreServer.csproj" ReferenceOutputAssembly="false"/>
<ProjectReference Include="$(CommonTestPath)System/Net/Prerequisites/RemoteLoopServer/RemoteLoopServer.csproj" ReferenceOutputAssembly="false"/>
<Compile Include="$(CommonTestPath)System\Net\WebSockets\WebSocketStream.cs"
Link="Common\System\Net\WebSockets\WebSocketStream.cs" />
</ItemGroup>

<!-- Do not reference these assemblies from the TargetingPack since we are building part of the source code for tests. -->
<ItemGroup>
<DefaultReferenceExclusion Include="System.Configuration" />
</ItemGroup>
<ItemGroup>
<Compile Include="$(CommonTestPath)System\Net\Capability.Security.cs"
Link="Common\System\Net\Capability.Security.cs" />
<Compile Include="$(CommonTestPath)System\Net\Configuration.cs"
Link="Common\System\Net\Configuration.cs" />
<Compile Include="$(CommonTestPath)System\Net\Configuration.Certificates.cs"
Link="Common\System\Net\Configuration.Certificates.cs" />
<Compile Include="$(CommonTestPath)System\Net\Configuration.Http.cs"
Link="Common\System\Net\Configuration.Http.cs" />
<Compile Include="$(CommonTestPath)System\Net\Configuration.Security.cs"
Link="Common\System\Net\Configuration.Security.cs" />
<Compile Include="$(CommonTestPath)System\Net\Configuration.WebSockets.cs"
Link="Common\System\Net\Configuration.WebSockets.cs" />
<Compile Include="$(CommonTestPath)System\Net\EventSourceTestLogging.cs"
Link="Common\System\Net\EventSourceTestLogging.cs" />
<Compile Include="$(CommonTestPath)System\Net\Http\LoopbackProxyServer.cs"
Link="Common\System\Net\Http\LoopbackProxyServer.cs" />
<Compile Include="$(CommonTestPath)System\Net\Http\LoopbackServer.cs"
Link="Common\System\Net\Http\LoopbackServer.cs" />
<Compile Include="$(CommonTestPath)System\Net\Http\GenericLoopbackServer.cs"
Link="Common\System\Net\Http\GenericLoopbackServer.cs" />
<Compile Include="$(CommonTestPath)System\Security\Cryptography\PlatformSupport.cs"
Link="CommonTest\System\Security\Cryptography\PlatformSupport.cs" />
<Compile Include="$(CommonTestPath)System\Threading\Tasks\TaskTimeoutExtensions.cs"
Link="Common\System\Threading\Tasks\TaskTimeoutExtensions.cs" />
<Compile Include="..\ClientWebSocketTestBase.cs" />
<Compile Include="..\WebSocketHelper.cs" />
<Compile Include="..\LoopbackHelper.cs" />
<Compile Include="..\ResourceHelper.cs" />
<Compile Include="..\WebSocketData.cs" />
<Compile Include="BrowserTimerThrottlingTest.cs" />
</ItemGroup>
<ItemGroup>
<PackageReference Include="System.Net.TestData" Version="$(SystemNetTestDataVersion)" />
</ItemGroup>

</Project>
@@ -0,0 +1,25 @@

Microsoft Visual Studio Solution File, Format Version 12.00
# Visual Studio Version 17
VisualStudioVersion = 17.0.31521.260
MinimumVisualStudioVersion = 10.0.40219.1
Project("{9A19103F-16F7-4668-BE54-9A1E7A4F7556}") = "System.Net.WebSockets.Client.Wasm.Tests", "System.Net.WebSockets.Client.Wasm.Tests.csproj", "{DA6DF153-169E-485D-90A4-EDBDA54B32A6}"
EndProject
Global
GlobalSection(SolutionConfigurationPlatforms) = preSolution
Debug|Any CPU = Debug|Any CPU
Release|Any CPU = Release|Any CPU
EndGlobalSection
GlobalSection(ProjectConfigurationPlatforms) = postSolution
{DA6DF153-169E-485D-90A4-EDBDA54B32A6}.Debug|Any CPU.ActiveCfg = Debug|Any CPU
{DA6DF153-169E-485D-90A4-EDBDA54B32A6}.Debug|Any CPU.Build.0 = Debug|Any CPU
{DA6DF153-169E-485D-90A4-EDBDA54B32A6}.Release|Any CPU.ActiveCfg = Release|Any CPU
{DA6DF153-169E-485D-90A4-EDBDA54B32A6}.Release|Any CPU.Build.0 = Release|Any CPU
EndGlobalSection
GlobalSection(SolutionProperties) = preSolution
HideSolutionNode = FALSE
EndGlobalSection
GlobalSection(ExtensibilityGlobals) = postSolution
SolutionGuid = {627DB5DF-DE31-44B4-B5FD-E5666467145E}
EndGlobalSection
EndGlobal
3 changes: 3 additions & 0 deletions src/libraries/tests.proj
Expand Up @@ -227,6 +227,9 @@
<ProjectExclusions Include="$(MSBuildThisFileDirectory)System.Security.Cryptography.Primitives\tests\System.Security.Cryptography.Primitives.Tests.csproj" />
<ProjectExclusions Include="$(MSBuildThisFileDirectory)System.Security.Cryptography.Xml\tests\System.Security.Cryptography.Xml.Tests.csproj" />
<ProjectExclusions Include="$(MSBuildThisFileDirectory)System.Security.Cryptography.X509Certificates\tests\System.Security.Cryptography.X509Certificates.Tests.csproj" />

<!-- This OuterLoop test requires browser UI, but the Helix agents are headless -->
<ProjectExclusions Include="$(MSBuildThisFileDirectory)System.Net.WebSockets.Client\tests\wasm\System.Net.WebSockets.Client.Wasm.Tests.csproj" />
</ItemGroup>

<ItemGroup Condition="'$(TargetOS)' == 'Browser' and '$(BuildAOTTestsOnHelix)' == 'true' and '$(RunDisabledWasmTests)' != 'true'">
Expand Down
25 changes: 22 additions & 3 deletions src/mono/wasm/runtime/binding_support.js
Expand Up @@ -160,6 +160,16 @@ var BindingSupportLib = {
return Promise.resolve(js_obj) === js_obj ||
((typeof js_obj === "object" || typeof js_obj === "function") && typeof js_obj.then === "function")
};
this.isChromium = false;
if (globalThis.navigator) {
var nav = globalThis.navigator;
if (nav.userAgentData && nav.userAgentData.brands) {
this.isChromium = nav.userAgentData.brands.some((i) => i.brand == 'Chromium');
}
else if (globalThis.navigator.userAgent) {
this.isChromium = nav.userAgent.includes("Chrome");
}
}

this._empty_string = "";
this._empty_string_ptr = 0;
Expand Down Expand Up @@ -372,7 +382,7 @@ var BindingSupportLib = {
return this._wrap_delegate_gc_handle_as_function(gc_handle);
},

_wrap_delegate_gc_handle_as_function: function (gc_handle) {
_wrap_delegate_gc_handle_as_function: function (gc_handle, after_listener_callback) {
this.bindings_lazy_init ();

// see if we have js owned instance for this gc_handle already
Expand All @@ -384,7 +394,11 @@ var BindingSupportLib = {
result = function() {
const delegateRoot = MONO.mono_wasm_new_root (BINDING.get_js_owned_object_by_gc_handle(gc_handle));
try {
return BINDING.call_method (result[BINDING.delegate_invoke_symbol], delegateRoot.value, result[BINDING.delegate_invoke_signature_symbol], arguments);
const res = BINDING.call_method(result[BINDING.delegate_invoke_symbol], delegateRoot.value, result[BINDING.delegate_invoke_signature_symbol], arguments);
if (after_listener_callback) {
after_listener_callback();
}
return res;
} finally {
delegateRoot.release();
}
Expand Down Expand Up @@ -2042,7 +2056,12 @@ var BindingSupportLib = {
var obj = BINDING.mono_wasm_get_jsobj_from_js_handle(objHandle);
if (!obj)
throw new Error("ERR09: Invalid JS object handle for '"+sName+"'");
var listener = BINDING._wrap_delegate_gc_handle_as_function(listener_gc_handle);

const prevent_timer_throttling = !BINDING.isChromium || obj.constructor.name !== 'WebSocket'
? null
: () => MONO.prevent_timer_throttling(0);

var listener = BINDING._wrap_delegate_gc_handle_as_function(listener_gc_handle, prevent_timer_throttling);
if (!listener)
throw new Error("ERR10: Invalid listener gc_handle");

Expand Down

0 comments on commit 5eaa911

Please sign in to comment.