Union/ProjectLighthouse
1
0
Fork 0
mirror of https://github.com/LBPUnion/ProjectLighthouse.git synced 2025-08-10 05:48:39 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

Rider really hates me now. More explicit types.

Browse source
This commit is contained in:
Michael VanOverbeek 2021-10-31 13:55:51 -04:00
commit e6e8022566
1 changed files with 9 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

18
ProjectLighthouse/Startup.cs
View file

@ -40,8 +40,8 @@ namespace LBPUnion.ProjectLighthouse
// This method gets called by the runtime. Use this method to configure the HTTP request pipeline. // This method gets called by the runtime. Use this method to configure the HTTP request pipeline.
public void Configure(IApplicationBuilder app, IWebHostEnvironment env) public void Configure(IApplicationBuilder app, IWebHostEnvironment env)
{ {
var computeDigests = true; bool computeDigests = true;
var serverDigestKey = Environment.GetEnvironmentVariable("SERVER_DIGEST_KEY"); string serverDigestKey = Environment.GetEnvironmentVariable("SERVER_DIGEST_KEY");
if (string.IsNullOrWhiteSpace(serverDigestKey)) if (string.IsNullOrWhiteSpace(serverDigestKey))
{ {
Logger.Log( Logger.Log(
@ -70,13 +70,13 @@ namespace LBPUnion.ProjectLighthouse
context.Request.EnableBuffering(); // Allows us to reset the position of Request.Body for later logging context.Request.EnableBuffering(); // Allows us to reset the position of Request.Body for later logging
// Client digest check. // Client digest check.
var authCookie = null as string; string authCookie;
if (!context.Request.Cookies.TryGetValue("MM_AUTH", out authCookie)) if (!context.Request.Cookies.TryGetValue("MM_AUTH", out authCookie))
authCookie = string.Empty; authCookie = string.Empty;
var digestPath = context.Request.Path; string digestPath = context.Request.Path;
var body = context.Request.Body; Stream body = context.Request.Body;
var clientRequestDigest = await HashHelper.ComputeDigest(digestPath, authCookie, body, serverDigestKey); string clientRequestDigest = await HashHelper.ComputeDigest(digestPath, authCookie, body, serverDigestKey);
// Check the digest we've just calculated against the X-Digest-A header if the game set the header. They should match. // Check the digest we've just calculated against the X-Digest-A header if the game set the header. They should match.
if (context.Request.Headers.TryGetValue("X-Digest-A", out var sentDigest)) if (context.Request.Headers.TryGetValue("X-Digest-A", out var sentDigest))
@ -93,8 +93,8 @@ namespace LBPUnion.ProjectLighthouse
context.Request.Body.Position = 0; context.Request.Body.Position = 0;
// This does the same as above, but for the response stream. // This does the same as above, but for the response stream.
using var responseBuffer = new MemoryStream(); using MemoryStream responseBuffer = new MemoryStream();
var oldResponseStream = context.Response.Body; Stream oldResponseStream = context.Response.Body;
context.Response.Body = responseBuffer; context.Response.Body = responseBuffer;
await next(); // Handle the request so we can get the status code from it await next(); // Handle the request so we can get the status code from it
@ -105,7 +105,7 @@ namespace LBPUnion.ProjectLighthouse
responseBuffer.Position = 0; responseBuffer.Position = 0;
// Compute the digest for the response. // Compute the digest for the response.
var serverDigest = await HashHelper.ComputeDigest(context.Request.Path, authCookie, string serverDigest = await HashHelper.ComputeDigest(context.Request.Path, authCookie,
responseBuffer, serverDigestKey); responseBuffer, serverDigestKey);
context.Response.Headers.Add("X-Digest-A", serverDigest); context.Response.Headers.Add("X-Digest-A", serverDigest);
} }