huahaofeng
/
aspnetcore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Less work on non-relative strings in UrlResolutionTagHelper

This commit is contained in:
Ben Adams 2016-02-06 14:01:16 +00:00 committed by N. Taylor Mullen
parent dca15c0a60
commit fd11d70fcd
1 changed files with 96 additions and 38 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

84
src/Microsoft.AspNetCore.Mvc.Razor/TagHelpers/UrlResolutionTagHelper.cs
View File

@ -219,25 +219,20 @@ namespace Microsoft.AspNetCore.Mvc.Razor.TagHelpers
protected bool TryResolveUrl(string url, out string resolvedUrl)
{
resolvedUrl = null;
if (url == null)
var start = FindRelativeStart(url);
if (start == -1)
{
return false;
}
var trimmedUrl = url.Trim(ValidAttributeWhitespaceChars);
var trimmedUrl = CreateTrimmedString(url, start);
// Before doing more work, ensure that the URL we're looking at is app-relative.
if (trimmedUrl.Length >= 2 && trimmedUrl[0] == '~' && trimmedUrl[1] == '/')
{
var urlHelper = UrlHelperFactory.GetUrlHelper(ViewContext);
resolvedUrl = urlHelper.Content(trimmedUrl);
return true;
}
return false;
}
/// <summary>
/// Tries to resolve the given <paramref name="url"/> value relative to the application's 'webroot' setting.
/// </summary>
@ -250,16 +245,14 @@ namespace Microsoft.AspNetCore.Mvc.Razor.TagHelpers
protected bool TryResolveUrl(string url, out IHtmlContent resolvedUrl)
{
resolvedUrl = null;
if (url == null)
var start = FindRelativeStart(url);
if (start == -1)
{
return false;
}
var trimmedUrl = url.Trim(ValidAttributeWhitespaceChars);
var trimmedUrl = CreateTrimmedString(url, start);
// Before doing more work, ensure that the URL we're looking at is app-relative.
if (trimmedUrl.Length >= 2 && trimmedUrl[0] == '~' && trimmedUrl[1] == '/')
{
var urlHelper = UrlHelperFactory.GetUrlHelper(ViewContext);
var appRelativeUrl = urlHelper.Content(trimmedUrl);
var postTildeSlashUrlValue = trimmedUrl.Substring(2);
@ -284,8 +277,73 @@ namespace Microsoft.AspNetCore.Mvc.Razor.TagHelpers
return true;
}
private static int FindRelativeStart(string url)
{
if (url == null || url.Length < 2)
{
return -1;
}
var maxTestLength = url.Length - 2;
var start = 0;
for (; start < url.Length; start++)
{
if (start > maxTestLength)
{
return -1;
}
if (!IsCharWhitespace(url[start]))
{
break;
}
}
// Before doing more work, ensure that the URL we're looking at is app-relative.
if (url[start] != '~' || url[start + 1] != '/')
{
return -1;
}
return start;
}
private static string CreateTrimmedString(string input, int start)
{
var end = input.Length - 1;
for (; end >= start; end--)
{
if (!IsCharWhitespace(input[end]))
{
break;
}
}
var len = end - start + 1;
// Substring returns same string if start == 0 && len == Length
return input.Substring(start, len);
}
private static bool IsCharWhitespace(char ch)
{
var i = 0;
for (; i < ValidAttributeWhitespaceChars.Length; i++)
{
if (ValidAttributeWhitespaceChars[i] == ch)
{
break;
}
}
if (i == ValidAttributeWhitespaceChars.Length)
{
// the character is not white space
return false;
}
return true;
}
private class EncodeFirstSegmentContent : IHtmlContent
{