| | | 1 | | using System.Security.Claims; |
| | | 2 | | using Elsa.Extensions; |
| | | 3 | | using Elsa.Options; |
| | | 4 | | using JetBrains.Annotations; |
| | | 5 | | using Microsoft.AspNetCore.Authentication.JwtBearer; |
| | | 6 | | using Microsoft.AspNetCore.Authorization; |
| | | 7 | | using Microsoft.AspNetCore.Http; |
| | | 8 | | using Microsoft.Extensions.Options; |
| | | 9 | | |
| | | 10 | | namespace Elsa.Requirements; |
| | | 11 | | |
| | | 12 | | /// <summary> |
| | | 13 | | /// Adds security-root bootstrap permissions to the current user when explicit localhost permission grants are enabled a |
| | | 14 | | /// </summary> |
| | | 15 | | public class LocalHostPermissionRequirement : IAuthorizationRequirement |
| | | 16 | | { |
| | | 17 | | } |
| | | 18 | | |
| | | 19 | | /// <inheritdoc /> |
| | | 20 | | [PublicAPI] |
| | | 21 | | public class LocalHostPermissionRequirementHandler : AuthorizationHandler<LocalHostPermissionRequirement> |
| | | 22 | | { |
| | 0 | 23 | | private static readonly string[] BootstrapPermissions = |
| | 0 | 24 | | [ |
| | 0 | 25 | | "create:application", |
| | 0 | 26 | | "create:user", |
| | 0 | 27 | | "create:role" |
| | 0 | 28 | | ]; |
| | | 29 | | |
| | | 30 | | private readonly IHttpContextAccessor _httpContextAccessor; |
| | | 31 | | private readonly IOptions<LocalHostPermissionRequirementOptions> _options; |
| | | 32 | | |
| | | 33 | | /// <inheritdoc /> |
| | 0 | 34 | | public LocalHostPermissionRequirementHandler(IHttpContextAccessor httpContextAccessor) : this( |
| | 0 | 35 | | httpContextAccessor, |
| | 0 | 36 | | Microsoft.Extensions.Options.Options.Create(new LocalHostPermissionRequirementOptions())) |
| | | 37 | | { |
| | 0 | 38 | | } |
| | | 39 | | |
| | | 40 | | /// <inheritdoc /> |
| | 293 | 41 | | public LocalHostPermissionRequirementHandler(IHttpContextAccessor httpContextAccessor, IOptions<LocalHostPermissionR |
| | | 42 | | { |
| | 293 | 43 | | _httpContextAccessor = httpContextAccessor; |
| | 293 | 44 | | _options = options; |
| | 293 | 45 | | } |
| | | 46 | | |
| | | 47 | | /// <inheritdoc /> |
| | | 48 | | protected override Task HandleRequirementAsync(AuthorizationHandlerContext context, LocalHostPermissionRequirement r |
| | | 49 | | { |
| | 0 | 50 | | if (!_options.Value.EnableLocalHostPermissionGrant) |
| | 0 | 51 | | return Task.CompletedTask; |
| | | 52 | | |
| | 0 | 53 | | if (_httpContextAccessor.HttpContext?.Request.IsLocal() != true) |
| | 0 | 54 | | return Task.CompletedTask; |
| | | 55 | | |
| | 0 | 56 | | if (context.User.Identities.Any(x => x.IsAuthenticated)) |
| | | 57 | | { |
| | 0 | 58 | | if (HasBootstrapPermissions(context.User)) |
| | 0 | 59 | | context.Succeed(requirement); |
| | | 60 | | |
| | 0 | 61 | | return Task.CompletedTask; |
| | | 62 | | } |
| | | 63 | | |
| | 0 | 64 | | var identity = new ClaimsIdentity(JwtBearerDefaults.AuthenticationScheme); |
| | 0 | 65 | | identity.AddClaims(BootstrapPermissions.Select(permission => new Claim(PermissionNames.ClaimType, permission))); |
| | 0 | 66 | | context.User.AddIdentity(identity); |
| | | 67 | | |
| | 0 | 68 | | context.Succeed(requirement); |
| | 0 | 69 | | return Task.CompletedTask; |
| | | 70 | | } |
| | | 71 | | |
| | | 72 | | private static bool HasBootstrapPermissions(ClaimsPrincipal user) |
| | | 73 | | { |
| | 0 | 74 | | var permissions = user.FindAll(PermissionNames.ClaimType).Select(x => x.Value).ToHashSet(StringComparer.Ordinal) |
| | 0 | 75 | | return permissions.Contains(PermissionNames.All) || BootstrapPermissions.All(permissions.Contains); |
| | | 76 | | } |
| | | 77 | | } |