internal/classes/DBOpsHelper.class.ps1
|
using namespace System.IO using namespace System.IO.Compression class DBOpsHelper { # Only keeps N last items in the path - helps to build relative paths static [string] SplitRelativePath ([string]$Path, [int]$Depth) { $returnPath = Split-Path -Path $Path -Leaf $parent = Split-Path -Path $Path -Parent while ($Depth-- -gt 0) { $returnPath = Join-Path -Path (Split-Path -Path $parent -Leaf) -ChildPath $returnPath $parent = Split-Path -Path $parent -Parent } return $returnPath } # Returns file contents as a binary array static [byte[]] GetBinaryFile ([string]$fileName) { $stream = [System.IO.File]::Open($fileName, [System.IO.FileMode]::Open, [System.IO.FileAccess]::Read, [System.IO.FileShare]::ReadWrite) $b = [byte[]]::new($stream.Length) try { $null = $stream.Read($b, 0, $b.Length) } catch { Stop-PSFFunction -EnableException $true -Message "Failed to read a binary stream from file $fileName" -ErrorRecord $_ } finally { $stream.Close() } return $b } # Converts a deflate stream into a memory stream - aka reads zip contents and writes them into memory static [System.IO.MemoryStream] ReadDeflateStream ([DeflateStream]$stream) { $memStream = [System.IO.MemoryStream]::new() $stream.CopyTo($memStream) $stream.Close() return $memStream } # Adds a new file entry into an opened ZipArchive object and fills it from the byte array static [void] WriteZipFile ([ZipArchive]$zipFile, [string]$fileName, [byte[]]$data) { #Remove old file entry if exists if ($zipFile.Mode -eq [ZipArchiveMode]::Update) { if ($oldEntry = $zipFile.GetEntry($fileName)) { $oldEntry.Delete() } } #Create new file entry $entry = $zipFile.CreateEntry($fileName) $writer = $entry.Open() #Write file contents $writer.Write($data, 0, $data.Length ) #Close the stream $writer.Close() } # Adds a new file entry into an opened ZipArchive object and fills it from file stream object - not used for now # static [void] WriteZipFileStream ([ZipArchive]$zipFile, [string]$fileName, [FileStream]$stream) { # $entry = $zipFile.CreateEntry($fileName) # $writer = $entry.Open() # $data = [byte[]]::new(4098) # #Read from stream and write file contents # while ($read = $stream.Read($data, 0, $data.Length)) { # $writer.Write($data, 0, $data.Length ) # } # #Close the stream # $writer.Close() # } # Returns an entry list from the archive file static [psobject[]] GetArchiveItems ([string]$fileName) { $entries = $null $zip = [Zipfile]::OpenRead($FileName) try { $entries = $zip.Entries | Select-Object * } catch { Stop-PSFFunction -EnableException $true -Message "Failed to get a list of files from archive $fileName" -ErrorRecord $_ } finally { $zip.Dispose() } return $entries } # Returns a specific entry from the archive file static [psobject[]] GetArchiveItem ([string]$fileName, [string[]]$itemName) { $zip = [Zipfile]::OpenRead($FileName) [psobject[]]$output = @() try { $entries = $zip.Entries | Where-Object { $_.FullName -in $itemName } foreach ($entry in $entries) { $bin = $null #Read deflate stream $stream = [DBOpsHelper]::ReadDeflateStream($entry.Open()) try { $bin = $stream.ToArray() } catch { Stop-PSFFunction -EnableException $true -Message "Failed to read deflate stream for item $itemName in archive $fileName" -ErrorRecord $_ } finally { $stream.Dispose() } $output += $entry | Select-Object * | Add-Member -MemberType NoteProperty -Name ByteArray -Value $bin -PassThru } } catch { Stop-PSFFunction -EnableException $true -Message "Failed to complete the deflate operation against archive $fileName" -ErrorRecord $_ } finally { $zip.Dispose() } return $output } # Converts byte array to hash string static [string] ToHexString([byte[]]$InputObject) { $outString = "0x" $InputObject | ForEach-Object { $outString += ("{0:X}" -f $_).PadLeft(2, "0") } return $outString } static [string] DecodeBinaryText ([byte[]]$Array) { $skipBytes = 0 # EF BB BF (UTF8) if ( $Array[0] -eq 0xef -and $Array[1] -eq 0xbb -and $Array[2] -eq 0xbf ) { $encoding = [System.Text.Encoding]::UTF8 $skipBytes = 3 } # 00 00 FE FF (UTF32 Big-Endian) elseif ($Array[0] -eq 0 -and $Array[1] -eq 0 -and $Array[2] -eq 0xfe -and $Array[3] -eq 0xff) { $encoding = [System.Text.Encoding]::UTF32 $skipBytes = 4 } # FF FE 00 00 (UTF32 Little-Endian) elseif ($Array[0] -eq 0xff -and $Array[1] -eq 0xfe -and $Array[2] -eq 0 -and $Array[3] -eq 0) { $encoding = [System.Text.Encoding]::UTF32 $skipBytes = 4 } # FE FF (UTF-16 Big-Endian) elseif ($Array[0] -eq 0xfe -and $Array[1] -eq 0xff) { $encoding = [System.Text.Encoding]::BigEndianUnicode $skipBytes = 2 } # FF FE (UTF-16 Little-Endian) elseif ($Array[0] -eq 0xff -and $Array[1] -eq 0xfe) { $encoding = [System.Text.Encoding]::Unicode $skipBytes = 2 } elseif ($Array[0] -eq 0x2b -and $Array[1] -eq 0x2f -and $Array[2] -eq 0x76 -and ($Array[3] -eq 0x38 -or $Array[3] -eq 0x39 -or $Array[3] -eq 0x2b -or $Array[3] -eq 0x2f)) { $encoding = [System.Text.Encoding]::UTF7 } else { $encoding = [System.Text.Encoding]::ASCII } return $encoding.GetString($Array, $skipBytes, $Array.Length - $skipBytes) } } |