Commit 51ffc799 authored by justcoding121's avatar justcoding121 Committed by justcoding121

optimize

parent 2b5bfef9
...@@ -5,7 +5,6 @@ using System.Threading.Tasks; ...@@ -5,7 +5,6 @@ using System.Threading.Tasks;
using Titanium.Web.Proxy.EventArguments; using Titanium.Web.Proxy.EventArguments;
using Titanium.Web.Proxy.Models; using Titanium.Web.Proxy.Models;
using Titanium.Web.Proxy.Network.WinAuth; using Titanium.Web.Proxy.Network.WinAuth;
using Titanium.Web.Proxy.Extensions;
namespace Titanium.Web.Proxy namespace Titanium.Web.Proxy
{ {
...@@ -58,7 +57,7 @@ namespace Titanium.Web.Proxy ...@@ -58,7 +57,7 @@ namespace Titanium.Web.Proxy
{ {
authHeader = args.WebSession.Response authHeader = args.WebSession.Response
.NonUniqueResponseHeaders[headerName] .NonUniqueResponseHeaders[headerName]
.Where(x => authSchemes.Any(y => x.Value.ContainsIgnoreCase(y))) .Where(x => authSchemes.Any(y => x.Value.StartsWith(y, StringComparison.OrdinalIgnoreCase)))
.FirstOrDefault(); .FirstOrDefault();
} }
...@@ -79,7 +78,7 @@ namespace Titanium.Web.Proxy ...@@ -79,7 +78,7 @@ namespace Titanium.Web.Proxy
if (headerName != null) if (headerName != null)
{ {
authHeader = authSchemes.Any(x => args.WebSession.Response authHeader = authSchemes.Any(x => args.WebSession.Response
.ResponseHeaders[headerName].Value.ContainsIgnoreCase(x)) ? .ResponseHeaders[headerName].Value.StartsWith(x, StringComparison.OrdinalIgnoreCase)) ?
args.WebSession.Response.ResponseHeaders[headerName] : null; args.WebSession.Response.ResponseHeaders[headerName] : null;
} }
} }
...@@ -102,6 +101,7 @@ namespace Titanium.Web.Proxy ...@@ -102,6 +101,7 @@ namespace Titanium.Web.Proxy
var serverToken = authHeader.Value.Substring(scheme.Length + 1); var serverToken = authHeader.Value.Substring(scheme.Length + 1);
var clientToken = WinAuthHandler.GetFinalAuthToken(args.WebSession.Request.Host, serverToken, args.Id); var clientToken = WinAuthHandler.GetFinalAuthToken(args.WebSession.Request.Host, serverToken, args.Id);
args.WebSession.Request.RequestHeaders["Authorization"] args.WebSession.Request.RequestHeaders["Authorization"]
= new HttpHeader("Authorization", string.Concat(scheme, clientToken)); = new HttpHeader("Authorization", string.Concat(scheme, clientToken));
} }
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment