.NET Core使用中间件做防盗链

最新推荐文章于 2023-05-10 16:39:11 发布

关键我是你林哥啊

最新推荐文章于 2023-05-10 16:39:11 发布

阅读量674

点赞数

分类专栏：防盗链 .NET Core 文章标签： .net

本文链接：https://blog.youkuaiyun.com/qq_27337291/article/details/121613767

版权

.NET Core 同时被 2 个专栏收录

15 篇文章

订阅专栏

防盗链

1 篇文章

订阅专栏

本文介绍了如何在.NET Core 3.1中编写一个名为OuterImgMiddleware的中间件，用于限制对静态图片的访问。当请求的URL不包含.jpg或者来源不在本地主机时，该中间件会返回404错误页面，从而保护图片资源不被外部直接访问。确保app.UseMiddleware<OuterImgMiddleware>()在app.UseStaticFiles();之前调用以生效。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

本文使用.NET Core 3.1.

准备两张图片：

编写一个OuterImgMiddleware中间件，编写如下代码：

private readonly RequestDelegate _next;
        private readonly IWebHostEnvironment  _webHostEnvironment;

        public OuterImgMiddleware(RequestDelegate next, IWebHostEnvironment webHostEnvironment)
        {
            _webHostEnvironment = webHostEnvironment;
            _next = next;
        }

        public async Task Invoke(HttpContext context)
        {
            string url = context.Request.Path.Value;
            if (!url.Contains(".jpg"))
            {
                await _next(context);//走正常流程
                return;
            }

            string urlReferrer = context.Request.Headers["Referer"];
            if (string.IsNullOrWhiteSpace(urlReferrer))//直接访问
            {
                await this.SetForbiddenImage(context, _webHostEnvironment);//返回404图片
            }
            else if (!urlReferrer.Contains("localhost"))//非当前域名
            {
                await this.SetForbiddenImage(context, _webHostEnvironment);//返回404图片
            }
            else
            {
                await _next(context);//走正常流程
            }
        }
        /// <summary>
        /// 设置拒绝图片
        /// </summary>
        /// <param name="context"></param>
        /// <returns></returns>
        private async Task SetForbiddenImage(HttpContext context, IWebHostEnvironment webHostEnvironment)
        {
            string defaultImagePath = webHostEnvironment.WebRootPath+ "\\Error\\404.jpg";// "wwwroot/Document/Img/Forbidden.jpg";
            string path = Path.Combine(Directory.GetCurrentDirectory(), defaultImagePath);

            FileStream fs = File.OpenRead(path);
            byte[] bytes = new byte[fs.Length];
            await fs.ReadAsync(bytes, 0, bytes.Length);
            await context.Response.Body.WriteAsync(bytes, 0, bytes.Length);
        }

启动中间件。需要注意的是， app.UseMiddleware<OuterImgMiddleware>()应至于 app.UseStaticFiles();之前