<#
.SYNOPSIS
Проверяет хэш-суммы файлов по списку.
.DESCRIPTION
Поддерживает чтение из файла, конвейера или строки.
Исправлено зависание при сканировании директорий (убран Resolve-Path -Relative, добавлена защита от рекурсивных симлинков).
.PARAMETER HashSource
Путь к файлу с хэшами, содержимое хэшей или массив строк.
.PARAMETER Algorithm
Алгоритм хэширования (MD5, SHA1, SHA256, SHA384, SHA512, MACTripleDES).
.PARAMETER StrictPath
Не приводить пути к нижнему регистру (регистрозависимое сравнение).
.PARAMETER TargetPath
Директория, в которой выполняется проверка файлов. По умолчанию: текущая директория.
.PARAMETER SkipChecksumFiles
Пропускать файлы, имена которых начинаются с "checksum".
#>
param(
[Parameter(Position=0, ValueFromPipeline=$true)]
[string[]]$HashSource,
[string]$Algorithm = "MD5",
[switch]$StrictPath,
[string]$TargetPath = ".",
[switch]$SkipChecksumFiles
)
begin {
$lines = @()
}
process {
if ($HashSource) {
$lines += $HashSource
}
}
end {
function Normalize-Path {
param([string]$Path)
$norm = $Path -replace '\\', '/'
if (-not $StrictPath) {
$norm = $norm.ToLowerInvariant()
}
$norm = $norm -replace '^\.[/\\]?', ''
return $norm
}
# --- 2. Валидация и нормализация целевой директории ---
if (-not (Test-Path -LiteralPath $TargetPath -PathType Container)) {
Write-Error "Target directory '$TargetPath' does not exist or is not a directory."
exit 1
}
$TargetPath = (Resolve-Path -LiteralPath $TargetPath).Path
# Добавляем слэш для корректного вычисления относительных путей
if (-not $TargetPath.EndsWith('\')) { $TargetPath += '\' }
# --- Чтение источника хэшей ---
if ($lines.Count -eq 0) {
# Нет входных данных — ищем стандартный файл
$defaultFile = "hashes.txt"
if (-not (Test-Path -LiteralPath $defaultFile)) {
Write-Error "No hash source provided and '$defaultFile' not found"
exit 1
}
$hashLines = Get-Content -LiteralPath $defaultFile -ErrorAction Stop
Write-Host "Reading hashes from default file: $defaultFile" -ForegroundColor Cyan
} elseif ($lines.Count -eq 1) {
# Одна строка — проверяем, файл ли это
$candidate = $lines[0].Trim()
if ($candidate -and (Test-Path -LiteralPath $candidate -ErrorAction SilentlyContinue)) {
$hashLines = Get-Content -LiteralPath $candidate -ErrorAction Stop
Write-Host "Reading hashes from file: $candidate" -ForegroundColor Cyan
} else {
# Содержимое хэшей — разбиваем на строки
$hashLines = $candidate -split '\r?\n' | Where-Object { $_.Trim() -ne '' }
Write-Host "Using inline hash content ($($hashLines.Count) lines)" -ForegroundColor Cyan
}
} else {
# Несколько строк (из конвейера или массива)
$hashLines = $lines | Where-Object { $_.Trim() -ne '' }
Write-Host "Using inline hash content ($($hashLines.Count) lines)" -ForegroundColor Cyan
}
# --- Парсинг хэшей ---
$expected = @{}
$invalidLines = 0
foreach ($line in $hashLines) {
if ($line -match '^(?<hash>[a-f0-9]{32}) \*(?<path>.+)$') {
$normPath = Normalize-Path $matches.path
$expected[$normPath] = $matches.hash.ToLowerInvariant()
} else {
$invalidLines++
}
}
if ($invalidLines -gt 0) {
Write-Warning "Skipped $invalidLines lines with invalid format"
}
if ($expected.Count -eq 0) {
Write-Error "No valid hash entries found"
exit 1
}
Write-Host "Loaded $($expected.Count) reference hashes" -ForegroundColor Cyan
# --- 1. Сканирование директории (оптимизировано) ---
$current = @{}
$processed = 0
Write-Host "Scanning directory: $TargetPath" -ForegroundColor Cyan
# -Attributes !ReparsePoint предотвращает зацикливание на junction points / симлинках
$files = Get-ChildItem -LiteralPath $TargetPath -Recurse -File -Attributes !ReparsePoint -ErrorAction SilentlyContinue
foreach ($file in $files) {
# --- 3. Пропуск файлов, начинающихся с checksum ---
if ($SkipChecksumFiles -and $file.Name -like 'checksum*') { continue }
# Быстрое вычисление относительного пути (замена медленному Resolve-Path -Relative)
$relPath = $file.FullName.Substring($TargetPath.Length)
$normPath = Normalize-Path $relPath
$hashObj = Get-FileHash -LiteralPath $file.FullName -Algorithm $Algorithm -ErrorAction SilentlyContinue
if ($hashObj -and $hashObj.Hash) {
$current[$normPath] = $hashObj.Hash.ToLowerInvariant()
$processed++
}
}
Write-Host "Processed $processed files in $TargetPath" -ForegroundColor Cyan
# --- Анализ расхождений ---
$missing = @()
$modified = @()
$newFiles = @()
$verified = 0
foreach ($path in $expected.Keys) {
if (-not $current.ContainsKey($path)) {
$missing += $path
} elseif ($current[$path] -ne $expected[$path]) {
$modified += [PSCustomObject]@{
Path = $path
Expected = $expected[$path]
Actual = $current[$path]
}
} else {
$verified++
}
}
foreach ($path in $current.Keys) {
if (-not $expected.ContainsKey($path)) {
$newFiles += $path
}
}
# --- Вывод результатов ---
Write-Host "`n=== VERIFICATION REPORT ===" -ForegroundColor Magenta
Write-Host "OK: $verified" -ForegroundColor Green
if ($modified.Count -gt 0) {
Write-Host "MODIFIED: $($modified.Count)" -ForegroundColor Red
foreach ($item in $modified) {
Write-Host " [!] $($item.Path)" -ForegroundColor Yellow
Write-Host " Expected: $($item.Expected)" -ForegroundColor DarkGray
Write-Host " Actual: $($item.Actual)" -ForegroundColor Red
}
}
if ($missing.Count -gt 0) {
Write-Host "MISSING: $($missing.Count)" -ForegroundColor Red
foreach ($path in $missing) {
Write-Host " [-] $path" -ForegroundColor Red
}
}
if ($newFiles.Count -gt 0) {
Write-Host "NEW FILES: $($newFiles.Count)" -ForegroundColor Cyan
foreach ($path in $newFiles) {
Write-Host " [+] $path" -ForegroundColor Cyan
}
}
if ($missing.Count -eq 0 -and $modified.Count -eq 0 -and $newFiles.Count -eq 0) {
Write-Host "`n[SUCCESS] All files verified successfully!" -ForegroundColor Green
exit 0
} else {
Write-Host "`n[WARNING] Verification failed: discrepancies found" -ForegroundColor Yellow
exit 1
}
}
