MVC 5 FileContentResult操作结果权限和重定向

Ces*_*sar 2 filecontentresult asp.net-mvc-5

我有一个MVC 5应用程序,允许用户下载数据库中存储的文件。我正在使用FileContentResult操作方法来执行此操作。

我可以在整个应用程序中限制对该方法的访问,但是精明的用户可以找出操作URL并将类似这样的内容(localhost:50000 / Home / FileDownload?id = 13)粘贴到他们的浏览器中,并可以通过以下方式下载任何文件只是更改参数。

我想限制用户这样做。仅允许管理员角色和具有特定权限的用户,这些权限只能由数据库调用来确定是否下载文件。

我正在寻找的是,如果用户使用URL下载文件并且没有适当的权限,我想用消息重定向用户。

我想做类似下面的代码或类似的操作,但是出现以下错误:无法将类型'System.Web.Mvc.RedirectToRouteResult'隐式转换为'System.Web.Mvc.FileContentResult'

我知道我不能在这里使用return RedirectToAction(“ Index”),只是在寻找有关如何解决此问题的想法。

    public FileContentResult FileDownload(int id)
    {
        //Check user has file download permission
        bool UserHasPermission = Convert.ToInt32(context.CheckUserHasFileDownloadPermission(id)) == 0 ? false : true;

        if (User.IsInRole("Administrator") || UserHasPermission)
        {
            //declare byte array to get file content from database and string to store file name
            byte[] fileData;
            string fileName;
            //create object of LINQ to SQL class

            //using LINQ expression to get record from database for given id value
            var record = from p in context.UploadedFiles
                         where p.Id == id
                         select p;
            //only one record will be returned from database as expression uses condtion on primary field
            //so get first record from returned values and retrive file content (binary) and filename
            fileData = (byte[])record.First().FileData.ToArray();
            fileName = record.First().FileName;
            //return file and provide byte file content and file name

            return File(fileData, "text", fileName);
        }
        else
        {
            TempData["Message"] = "Record not found";

            return RedirectToAction("Index");
        }           
    }
Run Code Online (Sandbox Code Playgroud)

Sam*_*ari 6

由于这两个FileContentResultRedirectToRouteResult来自被继承ActionResult,简单地使用ActionResult,而不是FileContentResult为你操作的返回类型:

public ActionResult FileDownload(int id)
{
    if(IsUserCanDownloadFile()) // your logic here
    {
        // fetch the file
        return File(fileData, "text", fileName);
    }
    return RedirectToAction("Index");

}
Run Code Online (Sandbox Code Playgroud)

或者,如果您更喜欢属性,则可以编写自己的authorize属性来检查权限:

public class FileAccessAttribute : AuthorizeAttribute
{
    private string _keyName;

    public FileAccessAttribute (string keyName)
    {
        _keyName = keyName;
    }

    protected override bool AuthorizeCore(HttpContextBase httpContext)
    {
        // imagine you have a service which could check the Permission
        return base.AuthorizeCore(httpContext) 
            || (this.ContainsKey
                && _permissionService.CanDownload(httpContext.User.Identity.GetUserId(),
                    int.Parse(this.KeyValue.ToString()));
    }

    private bool ContainsKey
    {
        get
        {
            // for simplicity I just check route data 
            // in real world you might need to check query string too 
            return ((MvcHandler)HttpContext.Current.Handler).RequestContext
                .RouteData.Values.ContainsKey(_keyName);
        }
    }
    private object KeyValue
    {
        get
        {
            return ((MvcHandler)HttpContext.Current.Handler)
                .RequestContext.RouteData.Values[_keyName];
        }
    }
}
Run Code Online (Sandbox Code Playgroud)

现在,您可以在操作上装饰自定义属性:

[FileAccess("id", Roles ="Administrator")]
public FileContentResult FileDownload(int id)
{
    // fetch the file
    return File(fileData, "text", fileName);
}
Run Code Online (Sandbox Code Playgroud)