Skip to content

Commit

Permalink
Validate that folders in paths don't start or end with a space
Browse files Browse the repository at this point in the history
(cherry picked from commit 316b5cbf75b45ef9a25f96ce1f2fbed25ad94296)
  • Loading branch information
markus101 authored and mynameisbogdan committed Apr 28, 2024
1 parent ab7debb commit 5a6b131
Show file tree
Hide file tree
Showing 2 changed files with 58 additions and 3 deletions.
28 changes: 27 additions & 1 deletion src/NzbDrone.Common.Test/PathExtensionFixture.cs
Expand Up @@ -3,6 +3,7 @@
using FluentAssertions;
using Moq;
using NUnit.Framework;
using NzbDrone.Common.Disk;
using NzbDrone.Common.EnvironmentInfo;
using NzbDrone.Common.Extensions;
using NzbDrone.Test.Common;
Expand Down Expand Up @@ -34,7 +35,7 @@ private IAppFolderInfo GetIAppDirectoryInfo()
[TestCase(@"\\Testserver\\Test\", @"\\Testserver\Test")]
[TestCase(@"\\Testserver\Test\file.ext", @"\\Testserver\Test\file.ext")]
[TestCase(@"\\Testserver\Test\file.ext\\", @"\\Testserver\Test\file.ext")]
[TestCase(@"\\Testserver\Test\file.ext \\", @"\\Testserver\Test\file.ext")]
[TestCase(@"\\Testserver\Test\file.ext ", @"\\Testserver\Test\file.ext")]
[TestCase(@"//CAPITAL//lower// ", @"\\CAPITAL\lower")]
public void Clean_Path_Windows(string dirty, string clean)
{
Expand Down Expand Up @@ -315,5 +316,30 @@ public void GetAncestorFolders_should_return_all_ancestors_in_path_Linux()
result[2].Should().Be(@"TV");
result[3].Should().Be(@"Series Title");
}

[TestCase(@"C:\Test\")]
[TestCase(@"C:\Test")]
[TestCase(@"C:\Test\TV\")]
[TestCase(@"C:\Test\TV")]
public void IsPathValid_should_be_true(string path)
{
path.AsOsAgnostic().IsPathValid(PathValidationType.CurrentOs).Should().BeTrue();
}

[TestCase(@"C:\Test \")]
[TestCase(@"C:\Test ")]
[TestCase(@"C:\ Test\")]
[TestCase(@"C:\ Test")]
[TestCase(@"C:\Test \TV")]
[TestCase(@"C:\ Test\TV")]
[TestCase(@"C:\Test \TV\")]
[TestCase(@"C:\ Test\TV\")]
[TestCase(@" C:\Test\TV\")]
[TestCase(@" C:\Test\TV")]

public void IsPathValid_should_be_false(string path)
{
path.AsOsAgnostic().IsPathValid(PathValidationType.CurrentOs).Should().BeFalse();
}
}
}
33 changes: 31 additions & 2 deletions src/NzbDrone.Common/Extensions/PathExtensions.cs
@@ -1,6 +1,7 @@
using System;
using System.Collections.Generic;
using System.IO;
using System.Linq;
using System.Text.RegularExpressions;
using NzbDrone.Common.Disk;
using NzbDrone.Common.EnsureThat;
Expand Down Expand Up @@ -28,6 +29,12 @@ public static class PathExtensions

public static string CleanFilePath(this string path)
{
if (path.IsNotNullOrWhiteSpace())
{
// Trim trailing spaces before checking if the path is valid so validation doesn't fail for something we can fix.
path = path.TrimEnd(' ');
}

Ensure.That(path, () => path).IsNotNullOrWhiteSpace();
Ensure.That(path, () => path).IsValidPath(PathValidationType.AnyOs);

Expand All @@ -36,10 +43,10 @@ public static string CleanFilePath(this string path)
// UNC
if (!info.FullName.Contains('/') && info.FullName.StartsWith(@"\\"))
{
return info.FullName.TrimEnd('/', '\\', ' ');
return info.FullName.TrimEnd('/', '\\');
}

return info.FullName.TrimEnd('/').Trim('\\', ' ');
return info.FullName.TrimEnd('/').Trim('\\');
}

public static bool PathNotEquals(this string firstPath, string secondPath, StringComparison? comparison = null)
Expand Down Expand Up @@ -143,6 +150,23 @@ public static bool IsPathValid(this string path, PathValidationType validationTy
return false;
}

if (path.Trim() != path)
{
return false;
}

var directoryInfo = new DirectoryInfo(path);

while (directoryInfo != null)
{
if (directoryInfo.Name.Trim() != directoryInfo.Name)
{
return false;
}

directoryInfo = directoryInfo.Parent;
}

if (validationType == PathValidationType.AnyOs)
{
return IsPathValidForWindows(path) || IsPathValidForNonWindows(path);
Expand Down Expand Up @@ -253,6 +277,11 @@ public static string ProcessNameToExe(this string processName)
return processName;
}

public static string CleanPath(this string path)
{
return Path.Join(path.Split(Path.DirectorySeparatorChar).Select(s => s.Trim()).ToArray());
}

public static string GetAppDataPath(this IAppFolderInfo appFolderInfo)
{
return appFolderInfo.AppDataFolder;
Expand Down

0 comments on commit 5a6b131

Please sign in to comment.