ASP.NET Core中配置CORS

来源:互联网 发布:数据导入excel表格 编辑:程序博客网 时间:2024/05/16 07:39

域名支持通配符 http://www.cnblogs.com/dudu/p/5895424.html

ASP.NET Core 内置了对 CORS 的支持,使用很简单,只需先在 Startup 的 ConfigureServices() 中添加 CORS 策略:

复制代码
public void ConfigureServices(IServiceCollection services){    services.AddCors(options => options.AddPolicy(        "AllowSameDomain",        builder => builder.WithOrigins(            "http://www.cnblogs.com",            "https://q.cnblogs.com",            "https://zzk.cnblogs.com",            "https://i.cnblogs.com",            "https://news.cnblogs.com",            "https://job.cnblogs.com")));}
复制代码

然后在想启用 CORS 的控制器 Action 上应用这个策略:

[EnableCors("AllowSameDomain")]public IActionResult Markdown(){    return View();}

但是,当看到上面一堆网址时,当想到每增加一个二级域名都需要修改上面的代码时,一种不舒服的感觉油然而生,一种想偷懒的冲动涌上心头。

难道没有一劳永逸的方法吗?DNS解析中支持在域名中使用通配符(*.cnblogs.com),CA证书中也支持,如果这里的 CORS 策略也支持使用通配符,不就可以一劳永逸了吗?配置 CORS 策略的代码就可以简化为下面的样子:

复制代码
public void ConfigureServices(IServiceCollection services){    services.AddCors(options => options.AddPolicy(        "AllowSameDomain",        builder => builder.WithOrigins("*.cnblogs.com")));}
复制代码

不仅一劳永逸,而且代码更加简洁漂亮。

可是负责 ASP.NET CORS 的开发者没这么贴心,只能自己动手了。

从 github 签出 ASP.NET CORS 的源代码,找到其中根据域名进行判断的实现代码:

复制代码
public class CorsService : ICorsService{    public virtual void EvaluateRequest(HttpContext context, CorsPolicy policy, CorsResult result)    {        var origin = context.Request.Headers[CorsConstants.Origin];        if (StringValues.IsNullOrEmpty(origin) || !policy.AllowAnyOrigin && !policy.Origins.Contains(origin))        {            return;        }        AddOriginToResult(origin, policy, result);        result.SupportsCredentials = policy.SupportsCredentials;        AddHeaderValues(result.AllowedExposedHeaders, policy.ExposedHeaders);    }    public virtual void EvaluatePreflightRequest(HttpContext context, CorsPolicy policy, CorsResult result)    {        var origin = context.Request.Headers[CorsConstants.Origin];        if (StringValues.IsNullOrEmpty(origin) || !policy.AllowAnyOrigin && !policy.Origins.Contains(origin))        {            return;        }        //...    }}
复制代码

(这里竟然有重复代码,又增添了一份不舒服的感觉,暂且不管)

原来是通过 !policy.Origins.Contains(origin) 判断的,只要修改这个地方的判断代码,就能实现一劳永逸的偷懒目的。但是,这部分代码不是随意可以修改的,要走代码贡献流程,而且不一定被接受,目前还是先想办法扩展它吧。

英明的 ASP.NET CORS 开发者早就考虑了这个地方的扩展性,将 EvaluateRequest() 与 EvaluatePreflightRequest 定义为虚拟方法,我们只需定义一个子类继承自 CorsService ,覆盖这两个方法即可。

接下来就是解决如何覆盖的问题。把父类中的实现代码复制过来修改不可取,以后 ASP.NET CORS 升级了,这部分代码改了,就会带来问题。我们需要想办法在不改变现有处理逻辑的前提下,影响处理结果。

我们继续看 !policy.Origins.Contains(origin) ,policy.Origins 的类型是 IList<string> ,它存储的就是我们在定义 CORS 策略时添加的网址,所以,如果我们想要影响这里的判断结果,唯有改变 policy.Origins 的值。

根据当前的情况,我们可以把问题简化为下面的代码:

复制代码
public static void Main(string[] args){    IList<string> origins = new List<string>() { "*.cnblogs.com" };                var origin = "http://www.cnblogs.com";    //在origins中添加"http://www.cnblogs.com"    Assert.True(origins.Contains(origin));}
复制代码

接下来只需做一件事——集中精力把上面的注释变成代码。

。。。

我们将注释变成了下面的代码:

复制代码
private void EvaluateOriginForWildcard(IList<string> origins, string origin){    //只在没有匹配的origin的情况下进行操作    if (!origins.Contains(origin))    {        //查询所有以星号开头的origin        var wildcardDomains = origins.Where(o => o.StartsWith("*"));        if (wildcardDomains.Any())        {            //遍历以星号开头的origin            foreach (var wildcardDomain in wildcardDomains)            {                //如果以.cnblogs.com结尾                if (origin.EndsWith(wildcardDomain.Substring(1))                    //或者以//cnblogs.com结尾,针对http://cnblogs.com                    || origin.EndsWith("//" + wildcardDomain.Substring(2)))                {                    //将http://www.cnblogs.com添加至origins                    origins.Add(origin);                    break;                }            }        }    }}
复制代码

然后基于上面的代码实现 CorsService 的子类 WildcardCorsService :

复制代码
public class WildcardCorsService : CorsService{    public WildcardCorsService(IOptions<CorsOptions> options)        : base(options)    {    }    public override void EvaluateRequest(HttpContext context, CorsPolicy policy, CorsResult result)    {        var origin = context.Request.Headers[CorsConstants.Origin];        EvaluateOriginForWildcard(policy.Origins, origin);        base.EvaluateRequest(context, policy, result);    }    public override void EvaluatePreflightRequest(HttpContext context, CorsPolicy policy, CorsResult result)    {        var origin = context.Request.Headers[CorsConstants.Origin];        EvaluateOriginForWildcard(policy.Origins, origin);        base.EvaluatePreflightRequest(context, policy, result);    }    private void EvaluateOriginForWildcard(IList<string> origins, string origin)    {        //...    }}
复制代码

然后将其注入:

services.Add(ServiceDescriptor.Transient<ICorsService, WildcardCorsService>());

(注:这里要用 Add ,不要用 TryAdd ,因为在 service.AddMvc 中已经把 CorsService 注入了,用 Add 才能覆盖 CorsService 的注入。)

最终 ConfigureServices() 中的代码变成了这样:

复制代码
public void ConfigureServices(IServiceCollection services){    services.AddMvc();    services.Add(ServiceDescriptor.Transient<ICorsService, WildcardCorsService>());    services.Configure<CorsOptions>(options => options.AddPolicy(        "AllowSameDomain",        builder => builder.WithOrigins("*.cnblogs.com")));}
复制代码

一劳永逸的目标就此达成,不舒服的感觉烟消云散。


原创粉丝点击