notfire/Iceshrimp.NET
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

[backend/middleware] Improve error handling in AuthorizedFetchMiddlewre

Browse source
This commit is contained in:
Laura Hausmann 2024-02-14 02:53:01 +01:00
parent 9591255220
commit 2756e19399
No known key found for this signature in database
GPG key ID: D044E84C5BE01605
1 changed files with 8 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

11
Iceshrimp.Backend/Core/Middleware/AuthorizedFetchMiddleware.cs
View file

@ -41,12 +41,17 @@ public class AuthorizedFetchMiddleware(
// If we don't, we need to try to fetch it
if (key == null) {
var user = await userResolver.ResolveAsync(sig.KeyId);
key = await db.UserPublickeys.Include(p => p.User).FirstOrDefaultAsync(p => p.User == user);
try {
var user = await userResolver.ResolveAsync(sig.KeyId);
key = await db.UserPublickeys.Include(p => p.User).FirstOrDefaultAsync(p => p.User == user);
}
catch (Exception e) {
throw new GracefulException($"Failed to fetch key of signature user ({sig.KeyId}) - {e.Message}");
}
}
// If we still don't have the key, something went wrong and we need to throw an exception
if (key == null) throw new GracefulException("Failed to fetch key of signature user");
if (key == null) throw new GracefulException($"Failed to fetch key of signature user ({sig.KeyId})");
if (key.User.Host == null)
throw new GracefulException("Remote user must have a host");