Skip to content

Commit

Permalink
sth
Browse files Browse the repository at this point in the history
  • Loading branch information
mili-tan committed Aug 1, 2020
1 parent 9424497 commit f60f42d
Showing 1 changed file with 3 additions and 8 deletions.
11 changes: 3 additions & 8 deletions Arashi.Aoi/Routes/DNSRoutes.cs
Original file line number Diff line number Diff line change
Expand Up @@ -25,14 +25,9 @@ public static void DnsQueryRoute(IEndpointRouteBuilder endpoints)
{
context.Response.Headers.Add("X-Powered-By", "ArashiDNSP/ONE.Aoi");
var queryDictionary = context.Request.Query;
if (context.Request.Method == "POST" && context.Request.ContentType.Contains("dns-message"))
if (context.Request.BodyReader.TryRead(out var readResult))
ReturnContext(context, true, DnsQuery(DnsMessage.Parse(readResult.Buffer.ToArray()), context));
else
{
context.Response.StatusCode = StatusCodes.Status500InternalServerError;
await context.Response.WriteAsync("Query POST RAW data read error");
}
if (context.Request.Method == "POST" && context.Request.ContentType.Contains("dns-message") &&
context.Request.BodyReader.TryRead(out var readResult))
ReturnContext(context, true, DnsQuery(DnsMessage.Parse(readResult.Buffer.ToArray()), context));
else if (queryDictionary.ContainsKey("dns"))
ReturnContext(context, true, DnsQuery(DNSGet.FromWebBase64(context), context));
else if (queryDictionary.ContainsKey("name"))
Expand Down

0 comments on commit f60f42d

Please sign in to comment.