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

[backend/federation] Enforce note visibility checks in AP controller

Browse source
This commit is contained in:
Laura Hausmann 2024-02-12 05:09:06 +01:00
parent c5d773e46a
commit a1fbe6f9b5
No known key found for this signature in database
GPG key ID: D044E84C5BE01605
4 changed files with 24 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
Iceshrimp.Backend/Controllers/ActivityPubController.cs
View file

@ -28,7 +28,11 @@ public class ActivityPubController : Controller {
[ProducesResponseType(StatusCodes.Status404NotFound, Type = typeof(ErrorResponse))]
public async Task<IActionResult> GetNote(string id, [FromServices] DatabaseContext db,
[FromServices] NoteRenderer noteRenderer) {
var note = await db.Notes.IncludeCommonProperties().FirstOrDefaultAsync(p => p.Id == id);
var actor = HttpContext.GetActor();
var note = await db.Notes
.IncludeCommonProperties()
.EnsureVisibleFor(actor)
.FirstOrDefaultAsync(p => p.Id == id);
if (note == null) return NotFound();
var rendered = await noteRenderer.RenderAsync(note);
var compacted = LdHelpers.Compact(rendered);

4
Iceshrimp.Backend/Core/Federation/ActivityPub/NoteRenderer.cs
View file

@ -13,8 +13,8 @@ public class NoteRenderer(IOptions<Config.InstanceSection> config, MfmConverter
public async Task<ASNote> RenderAsync(Note note, List<Note.MentionedUser>? mentions = null) {
var id = $"https://{config.Value.WebDomain}/notes/{note.Id}";
var userId = $"https://{config.Value.WebDomain}/users/{note.User.Id}";
var replyId = note.ReplyId != null
? new ASObjectBase($"https://{config.Value.WebDomain}/notes/{note.ReplyId}")
var replyId = note.Reply != null
? new ASObjectBase(note.Reply.Uri ?? $"https://{config.Value.WebDomain}/notes/{note.ReplyId}")
: null;
mentions ??= await db.Users

2
Iceshrimp.Backend/Core/Middleware/AuthenticationMiddleware.cs
View file

@ -65,7 +65,7 @@ public class AuthenticateAttribute(params string[] scopes) : Attribute {
public readonly string[] Scopes = scopes;
}
public static class HttpContextExtensions {
public static partial class HttpContextExtensions {
private const string Key = "session";
private const string MastodonKey = "masto-session";

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

@ -2,6 +2,7 @@ using System.Diagnostics.CodeAnalysis;
using System.Net;
using Iceshrimp.Backend.Core.Configuration;
using Iceshrimp.Backend.Core.Database;
using Iceshrimp.Backend.Core.Database.Tables;
using Iceshrimp.Backend.Core.Federation.Cryptography;
using Iceshrimp.Backend.Core.Services;
using Microsoft.EntityFrameworkCore;
@ -65,6 +66,8 @@ public class AuthorizedFetchMiddleware(
//TODO: re-fetch key once if signature validation fails, to properly support key rotation
//TODO: Check for LD signature as well
ctx.SetActor(key.User);
}
await next(ctx);
@ -73,4 +76,17 @@ public class AuthorizedFetchMiddleware(
public class AuthorizedFetchAttribute(bool forceBody = false) : Attribute {
public bool ForceBody { get; } = forceBody;
}
public static partial class HttpContextExtensions {
private const string ActorKey = "auth-fetch-user";
internal static void SetActor(this HttpContext ctx, User actor) {
ctx.Items.Add(ActorKey, actor);
}
public static User? GetActor(this HttpContext ctx) {
ctx.Items.TryGetValue(ActorKey, out var actor);
return actor as User;
}
}