Skip to content

Instantly share code, notes, and snippets.

@MatteoPiovanelli-Laser
Last active January 16, 2018 11:24
Show Gist options
  • Save MatteoPiovanelli-Laser/6eca32667403bead62f8d9a989fa954f to your computer and use it in GitHub Desktop.
Save MatteoPiovanelli-Laser/6eca32667403bead62f8d9a989fa954f to your computer and use it in GitHub Desktop.
draft of idea to enable better auth cookie management in orchard cms
public class DefaultAuthCookieManager : IAuthCookieManager {
private const int _cookieVersion = 3;
private readonly ShellSettings _settings;
private readonly IClock _clock;
private readonly IMembershipService _membershipService;
private readonly IHttpContextAccessor _httpContextAccessor;
private readonly ISslSettingsProvider _sslSettingsProvider;
private readonly IMembershipValidationService _membershipValidationService;
public DefaultAuthCookieManager()
ShellSettings settings,
IClock clock,
IMembershipService membershipService,
IHttpContextAccessor httpContextAccessor,
ISslSettingsProvider sslSettingsProvider,
IMembershipValidationService membershipValidationService) {
_settings = settings;
_clock = clock;
_membershipService = membershipService;
_httpContextAccessor = httpContextAccessor;
_sslSettingsProvider = sslSettingsProvider;
_membershipValidationService = membershipValidationService;
ExpirationTimeSpan = TimeSpan.FromDays(30);
}
public TimeSpan ExpirationTimeSpan { get; set; }
public void CreateAndAddCookie(IUser user, bool createPersistentCookie) {
var now = _clock.UtcNow.ToLocalTime();
// The cookie user data is "{userName.Base64};{tenant}".
// The username is encoded to Base64 to prevent collisions with the ';' seprarator.
var userData = String.Concat(user.UserName.ToBase64(), ";", _settings.Name);
var ticket = new FormsAuthenticationTicket(
_cookieVersion,
user.UserName,
now,
now.Add(ExpirationTimeSpan),
createPersistentCookie,
userData,
FormsAuthentication.FormsCookiePath);
var encryptedTicket = FormsAuthentication.Encrypt(ticket);
var cookie = new HttpCookie(FormsAuthentication.FormsCookieName, encryptedTicket) {
HttpOnly = true,
Secure = _sslSettingsProvider.GetRequiresSSL(),
Path = FormsAuthentication.FormsCookiePath
};
var httpContext = _httpContextAccessor.Current();
if (!String.IsNullOrEmpty(_settings.RequestUrlPrefix)) {
cookie.Path = GetCookiePath(httpContext);
}
if (FormsAuthentication.CookieDomain != null) {
cookie.Domain = FormsAuthentication.CookieDomain;
}
if (createPersistentCookie) {
cookie.Expires = ticket.Expiration;
}
httpContext.Response.Cookies.Add(cookie);
}
public void InvalidateCookie() {
// overwritting the authentication cookie for the given tenant
var httpContext = _httpContextAccessor.Current();
var rFormsCookie = new HttpCookie(FormsAuthentication.FormsCookieName, "") {
Expires = DateTime.Now.AddYears(-1),
};
if (!String.IsNullOrEmpty(_settings.RequestUrlPrefix)) {
rFormsCookie.Path = GetCookiePath(httpContext);
}
httpContext.Response.Cookies.Add(rFormsCookie);
}
public IUser GetAuthenticatedUser() {
var httpContext = _httpContextAccessor.Current();
if (httpContext.IsBackgroundContext() || !httpContext.Request.IsAuthenticated || !(httpContext.User.Identity is FormsIdentity)) {
return null;
}
var formsIdentity = (FormsIdentity)httpContext.User.Identity;
var userData = formsIdentity.Ticket.UserData ?? "";
// The cookie user data is {userName.Base64};{tenant}.
var userDataSegments = userData.Split(';');
if (userDataSegments.Length < 2) {
return null;
}
var userDataName = userDataSegments[0];
var userDataTenant = userDataSegments[1];
try {
userDataName = userDataName.FromBase64();
}
catch {
return null;
}
if (!String.Equals(userDataTenant, _settings.Name, StringComparison.Ordinal)) {
return null;
}
return _membershipService.GetUser(userDataName);
}
private string GetCookiePath(HttpContextBase httpContext) {
var cookiePath = httpContext.Request.ApplicationPath;
if (cookiePath != null && cookiePath.Length > 1) {
cookiePath += '/';
}
cookiePath += _settings.RequestUrlPrefix;
return cookiePath;
}
}
using System;
using System.Web;
using System.Web.Security;
using Orchard.Environment.Configuration;
using Orchard.Logging;
using Orchard.Mvc;
using Orchard.Mvc.Extensions;
using Orchard.Services;
using Orchard.Utility.Extensions;
namespace Orchard.Security.Providers {
public class FormsAuthenticationService : IAuthenticationService {
private const int _cookieVersion = 3;
private readonly ShellSettings _settings;
private readonly IClock _clock;
private readonly IMembershipService _membershipService;
private readonly IHttpContextAccessor _httpContextAccessor;
private readonly ISslSettingsProvider _sslSettingsProvider;
private readonly IMembershipValidationService _membershipValidationService;
private readonly IAuthCookieManager _authCookieManager;
private IUser _signedInUser;
private bool _isAuthenticated;
// This fixes a performance issue when the forms authentication cookie is set to a
// user name not mapped to an actual Orchard user content item. If the request is
// authenticated but a null user is returned, multiple calls to GetAuthenticatedUser
// will cause multiple DB invocations, slowing down the request. We therefore
// remember if the current user is a non-Orchard user between invocations.
private bool _isNonOrchardUser;
public FormsAuthenticationService(
ShellSettings settings,
IClock clock,
IMembershipService membershipService,
IHttpContextAccessor httpContextAccessor,
ISslSettingsProvider sslSettingsProvider,
IMembershipValidationService membershipValidationService,
IAuthCookieManager authCookieManager) {
_settings = settings;
_clock = clock;
_membershipService = membershipService;
_httpContextAccessor = httpContextAccessor;
_sslSettingsProvider = sslSettingsProvider;
_membershipValidationService = membershipValidationService;
_authCookieManager = authCookieManager;
Logger = NullLogger.Instance;
ExpirationTimeSpan = TimeSpan.FromDays(30);
}
public ILogger Logger { get; set; }
public TimeSpan ExpirationTimeSpan { get; set; }
public void SignIn(IUser user, bool createPersistentCookie) {
_authCookieManager.CreateAndAddCookie(user, createPersistentCookie);
_isAuthenticated = true;
_isNonOrchardUser = false;
_signedInUser = user;
}
public void SignOut() {
_signedInUser = null;
_isAuthenticated = false;
FormsAuthentication.SignOut();
_authCookieManager.InvalidateCookie();
}
public void SetAuthenticatedUserForRequest(IUser user) {
_signedInUser = user;
_isAuthenticated = true;
_isNonOrchardUser = false;
}
public IUser GetAuthenticatedUser() {
if (_isNonOrchardUser)
return null;
if (_signedInUser != null || _isAuthenticated)
return _signedInUser;
_signedInUser = _authCookieManager.GetAuthenticatedUser();
if (_signedInUser == null || !_membershipValidationService.CanAuthenticateWithCookie(_signedInUser)) {
_isNonOrchardUser = true;
return null;
}
_isAuthenticated = true;
return _signedInUser;
}
}
}
public interface IAuthCookieManager : IDependency {
void CreateAndAddCookie(IUser user, bool createPersistentCookie);
void InvalidateCookie();
IUser GetAuthenticatedUser();
}
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment