From efae85db2cafff94ebd175b25bcae51a5e7f23bc Mon Sep 17 00:00:00 2001
From: Aigio Liu <lysoft.pw@foxmail.com>
Date: Sun, 24 Jul 2022 13:41:29 +0800
Subject: [PATCH] Add UseWinHttpHandler

---
 Directory.Packages.props                      |   1 +
 .../PlatformHttpMessageHandlerBuilder.cs      |   4 +-
 .../Pages/Settings/Settings_General.axaml     |  12 ++
 .../ST.Client.Windows.csproj                  |   1 +
 .../ArchiSteamFarmServiceImpl.cs              |   2 +-
 .../Settings/GeneralSettings.Avalonia.cs      |  10 ++
 .../UI/Resx/AppResources.Designer.cs          |   9 ++
 src/ST.Client/UI/Resx/AppResources.resx       |   3 +
 .../ServiceCollectionExtensions.cs            |   1 +
 src/Startup.cs                                | 145 +++++++++++++++---
 10 files changed, 164 insertions(+), 24 deletions(-)

diff --git a/Directory.Packages.props b/Directory.Packages.props
index 10c4abbf7..541c00886 100644
--- a/Directory.Packages.props
+++ b/Directory.Packages.props
@@ -84,6 +84,7 @@
     <!--<PackageVersion Include="CommunityToolkit.Mvvm" Version="7.1.2" />-->
     <PackageVersion Include="System.Text.Encoding.CodePages" Version="6.0.0" />
     <PackageVersion Include="System.Diagnostics.PerformanceCounter" Version="6.0.1" />
+    <PackageVersion Include="System.Net.Http.WinHttpHandler" Version="6.0.1" />
     
     <!-- AppCenter.* -->
     <PackageVersion Include="Microsoft.AppCenter" Version="4.5.1" />
diff --git a/src/ST.Client.Android/Services/Implementation/PlatformHttpMessageHandlerBuilder.cs b/src/ST.Client.Android/Services/Implementation/PlatformHttpMessageHandlerBuilder.cs
index a4b7c08d3..4cdbc9467 100644
--- a/src/ST.Client.Android/Services/Implementation/PlatformHttpMessageHandlerBuilder.cs
+++ b/src/ST.Client.Android/Services/Implementation/PlatformHttpMessageHandlerBuilder.cs
@@ -14,10 +14,10 @@ namespace System.Application.Services.Implementation
 #endif
         }
 
-        public static AndroidClientHandler CreateAndroidClientHandler()
+        public static AndroidClientHandler CreateAndroidClientHandler(AndroidClientHandler? handler = null)
         {
             // https://docs.microsoft.com/zh-cn/xamarin/android/app-fundamentals/http-stack?context=xamarin%2Fcross-platform&tabs=macos
-            var handler = new AndroidClientHandler();
+            handler ??= new AndroidClientHandler();
             AddFiddlerRootCertificate(handler);
             GeneralHttpClientFactory.SetProxyToHandler(GeneralHttpClientFactory.DefaultProxy, handler);
             return handler;
diff --git a/src/ST.Client.Desktop.Avalonia/Application/UI/Views/Pages/Settings/Settings_General.axaml b/src/ST.Client.Desktop.Avalonia/Application/UI/Views/Pages/Settings/Settings_General.axaml
index 5e36e3fb5..5de51b32f 100644
--- a/src/ST.Client.Desktop.Avalonia/Application/UI/Views/Pages/Settings/Settings_General.axaml
+++ b/src/ST.Client.Desktop.Avalonia/Application/UI/Views/Pages/Settings/Settings_General.axaml
@@ -69,6 +69,18 @@
         </DockPanel>
       </Border>
 
+      <Border Classes="Item" IsVisible="{markup:OnPlatform Windows}">
+        <DockPanel>
+          <StackPanel>
+            <TextBlock Text="{ReflectionBinding Path=Res.Settings_General_UseWinHttpHandler,Mode=OneWay,Source={x:Static resx:R.Current}}"/>
+            <TextBlock Classes="prompt"
+                       Text="{ReflectionBinding Path=Res.Settings_RestartEffective,Mode=OneWay,Source={x:Static resx:R.Current}}"/>
+          </StackPanel>
+
+          <ToggleSwitch IsChecked="{ReflectionBinding Source={x:Static ms:GeneralSettings.UseWinHttpHandler},Path=Value, Mode=TwoWay}"/>
+        </DockPanel>
+      </Border>
+
       <!--<Border Classes="Item"
               IsVisible="{markup:OnPlatform Windows}">
         <DockPanel>
diff --git a/src/ST.Client.Desktop.Windows/ST.Client.Windows.csproj b/src/ST.Client.Desktop.Windows/ST.Client.Windows.csproj
index fb390a23c..992504589 100644
--- a/src/ST.Client.Desktop.Windows/ST.Client.Windows.csproj
+++ b/src/ST.Client.Desktop.Windows/ST.Client.Windows.csproj
@@ -37,6 +37,7 @@
     <PackageReference Include="7-Zip.NativeAssets.Win32" />
     <PackageReference Include="Microsoft.Toolkit.Uwp.Notifications" />
     <PackageReference Include="System.Management" />
+    <PackageReference Include="System.Net.Http.WinHttpHandler" />
   </ItemGroup>
 
   <!--<ItemGroup Condition=" !$(TargetFramework.Contains('-windows')) ">
diff --git a/src/ST.Client/Services/Implementation/ArchiSteamFarmServiceImpl.cs b/src/ST.Client/Services/Implementation/ArchiSteamFarmServiceImpl.cs
index 8c34b5302..cf52ff33a 100644
--- a/src/ST.Client/Services/Implementation/ArchiSteamFarmServiceImpl.cs
+++ b/src/ST.Client/Services/Implementation/ArchiSteamFarmServiceImpl.cs
@@ -18,8 +18,8 @@ using System.Net;
 using System.Net.Http;
 using System.Text;
 using System.Threading.Tasks;
-using MSEXLog = Microsoft.Extensions.Logging;
 using CC = System.Common.Constants;
+using MSEXLog = Microsoft.Extensions.Logging;
 
 namespace System.Application.Services.Implementation
 {
diff --git a/src/ST.Client/Settings/GeneralSettings.Avalonia.cs b/src/ST.Client/Settings/GeneralSettings.Avalonia.cs
index 7d2029c74..abcb0f5ab 100644
--- a/src/ST.Client/Settings/GeneralSettings.Avalonia.cs
+++ b/src/ST.Client/Settings/GeneralSettings.Avalonia.cs
@@ -113,5 +113,15 @@ namespace System.Application.Settings
         ///// </summary>
         //public static SerializableProperty<bool> IsEnableLogRecord { get; }
         //    = GetProperty(defaultValue: false, autoSave: true);
+
+        static readonly SerializableProperty<bool>? _UseWinHttpHandler = OperatingSystem2.IsWindows() ? GetProperty(defaultValue: false, autoSave: true) : null;
+
+        /// <summary>
+        /// (仅 Windows)使用基于 Windows 的 WinHTTP 接口处理消息
+        /// <para>https://docs.microsoft.com/zh-cn/dotnet/api/system.net.http.winhttphandler?view=dotnet-plat-ext-6.0</para>
+        /// <para>https://docs.microsoft.com/zh-cn/dotnet/api/system.net.http.socketshttphandler?view=net-6.0</para>
+        /// </summary>
+        [SupportedOSPlatform("Windows7.0")]
+        public static SerializableProperty<bool> UseWinHttpHandler => _UseWinHttpHandler ?? throw new PlatformNotSupportedException();
     }
 }
diff --git a/src/ST.Client/UI/Resx/AppResources.Designer.cs b/src/ST.Client/UI/Resx/AppResources.Designer.cs
index da6f7cec7..04c05c4ba 100644
--- a/src/ST.Client/UI/Resx/AppResources.Designer.cs
+++ b/src/ST.Client/UI/Resx/AppResources.Designer.cs
@@ -5140,6 +5140,15 @@ namespace System.Application.UI.Resx {
             }
         }
         
+        /// <summary>
+        ///   查找类似 使用 WinHttpHandler 的本地化字符串。
+        /// </summary>
+        public static string Settings_General_UseWinHttpHandler {
+            get {
+                return ResourceManager.GetString("Settings_General_UseWinHttpHandler", resourceCulture);
+            }
+        }
+        
         /// <summary>
         ///   查找类似 语言 的本地化字符串。
         /// </summary>
diff --git a/src/ST.Client/UI/Resx/AppResources.resx b/src/ST.Client/UI/Resx/AppResources.resx
index 6ea99bf99..93cfc96c3 100644
--- a/src/ST.Client/UI/Resx/AppResources.resx
+++ b/src/ST.Client/UI/Resx/AppResources.resx
@@ -2380,4 +2380,7 @@ AAAAA-BBBBB-CCCC3</value>
   <data name="Notice_Type" xml:space="preserve">
     <value>消息类型</value>
   </data>
+  <data name="Settings_General_UseWinHttpHandler" xml:space="preserve">
+    <value>使用 WinHttpHandler</value>
+  </data>
 </root>
\ No newline at end of file
diff --git a/src/ST.Services.CloudService/ServiceCollectionExtensions.cs b/src/ST.Services.CloudService/ServiceCollectionExtensions.cs
index 5197fe0ae..7e8208ee6 100644
--- a/src/ST.Services.CloudService/ServiceCollectionExtensions.cs
+++ b/src/ST.Services.CloudService/ServiceCollectionExtensions.cs
@@ -16,6 +16,7 @@ namespace Microsoft.Extensions.DependencyInjection
         /// <typeparam name="T"></typeparam>
         /// <param name="services"></param>
         /// <param name="config"></param>
+        /// <param name="configureHandler"></param>
         /// <param name="useMock"></param>
         /// <returns></returns>
         public static IServiceCollection TryAddCloudServiceClient<T>(
diff --git a/src/Startup.cs b/src/Startup.cs
index 4998126af..2cc28c84e 100644
--- a/src/Startup.cs
+++ b/src/Startup.cs
@@ -72,6 +72,12 @@ using LogLevel = Microsoft.Extensions.Logging.LogLevel;
 #pragma warning restore SA1216 // Using static directives should be placed at the correct location
 #pragma warning restore SA1211 // Using alias directives should be ordered alphabetically by alias name
 #pragma warning restore IDE0079 // 请删除不必要的忽略
+using CreateHttpHandlerArgs = System.ValueTuple<
+    bool,
+    System.Net.DecompressionMethods,
+    System.Net.CookieContainer,
+    System.Net.IWebProxy?,
+    int>;
 
 namespace System.Application.UI
 {
@@ -429,32 +435,12 @@ namespace System.Application.UI
                 services.TryAddModelValidator();
 
                 // 添加服务端API调用
-#pragma warning disable SA1111 // Closing parenthesis should be on line of last parameter
                 services.TryAddCloudServiceClient<CloudServiceClient>(c =>
                 {
 #if NETCOREAPP3_0_OR_GREATER
                     c.DefaultRequestVersion = HttpVersion.Version30;
 #endif
-                }, configureHandler:
-#if NETCOREAPP2_1_OR_GREATER
-                () => new SocketsHttpHandler
-                {
-                    UseCookies = false,
-                    AutomaticDecompression = DecompressionMethods.Brotli | DecompressionMethods.GZip,
-                }
-#elif __ANDROID__
-                () =>
-                {
-                    var handler = PlatformHttpMessageHandlerBuilder.CreateAndroidClientHandler();
-                    handler.UseCookies = false;
-                    handler.AutomaticDecompression = DecompressionMethods.GZip;
-                    return handler;
-                }
-#else
-                null
-#endif
-                );
-#pragma warning restore SA1111 // Closing parenthesis should be on line of last parameter
+                }, configureHandler: ConfigureHandler());
 
                 services.AddAutoMapper();
 
@@ -629,6 +615,8 @@ namespace System.Application.UI
             {
                 if (isTrace) StartWatchTrace.Record(dispose: true);
             }
+
+            ArchiSteamFarm.Web.WebBrowser.CreateHttpHandlerDelegate = CreateHttpHandler;
         }
 
         /// <summary>
@@ -897,5 +885,120 @@ namespace System.Application.UI
 #endif
 
         #endregion
+
+        #region HttpMessageHandler
+
+        static Func<HttpMessageHandler>? ConfigureHandler()
+        {
+#if NETCOREAPP2_1_OR_GREATER
+#if WINDOWS
+            if (GeneralSettings.UseWinHttpHandler.Value)
+            {
+                return () => new WinHttpHandler
+                {
+                    WindowsProxyUsePolicy = WindowsProxyUsePolicy.DoNotUseProxy,
+                    CookieUsePolicy = CookieUsePolicy.IgnoreCookies,
+                    AutomaticDecompression = DecompressionMethods.Brotli | DecompressionMethods.GZip,
+                };
+            }
+            else
+#endif
+            {
+                return () => new SocketsHttpHandler
+                {
+                    UseCookies = false,
+                    AutomaticDecompression = DecompressionMethods.Brotli | DecompressionMethods.GZip,
+                };
+            }
+#elif __ANDROID__
+            return () =>
+            {
+                var handler = PlatformHttpMessageHandlerBuilder.CreateAndroidClientHandler();
+                handler.UseCookies = false;
+                handler.AutomaticDecompression = DecompressionMethods.GZip;
+                return handler;
+            };
+#else
+            return null;
+#endif
+        }
+
+        static HttpMessageHandler? CreateHttpHandler(CreateHttpHandlerArgs args)
+        {
+#if NETCOREAPP2_1_OR_GREATER
+#if WINDOWS
+            if (GeneralSettings.UseWinHttpHandler.Value)
+            {
+                var handler = new WinHttpHandler
+                {
+                    AutomaticRedirection = args.Item1,
+                    AutomaticDecompression = args.Item2,
+                    CookieContainer = args.Item3,
+                };
+                var proxy = args.Item4;
+                var useProxy = GeneralHttpClientFactory.UseWebProxy(proxy);
+                if (useProxy)
+                {
+                    handler.Proxy = proxy;
+                    handler.WindowsProxyUsePolicy = WindowsProxyUsePolicy.UseCustomProxy;
+                }
+                else
+                {
+                    handler.WindowsProxyUsePolicy = WindowsProxyUsePolicy.DoNotUseProxy;
+                }
+                if (!(args.Item5 < 1)) // https://github.com/dotnet/runtime/blob/v6.0.0/src/libraries/System.Net.Http/src/System/Net/Http/SocketsHttpHandler/SocketsHttpHandler.cs#L157
+                {
+                    handler.MaxConnectionsPerServer = args.Item5;
+                }
+                return handler;
+            }
+            else
+#endif
+            {
+                var handler = new SocketsHttpHandler
+                {
+                    AllowAutoRedirect = args.Item1,
+                    AutomaticDecompression = args.Item2,
+                    CookieContainer = args.Item3,
+                };
+                var proxy = args.Item4;
+                var useProxy = GeneralHttpClientFactory.UseWebProxy(proxy);
+                if (useProxy)
+                {
+                    handler.Proxy = proxy;
+                    handler.UseProxy = true;
+                }
+                if (!(args.Item5 < 1)) // https://github.com/dotnet/runtime/blob/v6.0.0/src/libraries/System.Net.Http/src/System/Net/Http/SocketsHttpHandler/SocketsHttpHandler.cs#L157
+                {
+                    handler.MaxConnectionsPerServer = args.Item5;
+                }
+                return handler;
+            }
+#elif __ANDROID__
+            var handler = new AndroidClientHandler
+            {
+                AllowAutoRedirect = args.Item1,
+                AutomaticDecompression = args.Item2,
+                CookieContainer = args.Item3,
+            };
+            PlatformHttpMessageHandlerBuilder.CreateAndroidClientHandler(handler);
+            var proxy = args.Item4;
+            var useProxy = GeneralHttpClientFactory.UseWebProxy(proxy);
+            if (useProxy)
+            {
+                handler.Proxy = proxy;
+                handler.UseProxy = true;
+            }
+            if (!(args.Item5 < 1)) // https://github.com/dotnet/runtime/blob/v6.0.0/src/libraries/System.Net.Http/src/System/Net/Http/SocketsHttpHandler/SocketsHttpHandler.cs#L157
+            {
+                handler.MaxConnectionsPerServer = args.Item5;
+            }
+            return handler;
+#else
+            return null;
+#endif
+        }
+
+        #endregion
     }
 }
\ No newline at end of file