Private/ConvertTo-EncryptedBin.ps1
|
function ConvertTo-EncryptedBin { [CmdletBinding(SupportsShouldProcess, DefaultParameterSetName = "SomethingFromPipeline")] param( [Parameter(Mandatory)] [String]$Key, [Parameter(Mandatory, ValueFromPipeline, ParameterSetName="FromFile")] [String]$FileIn, [Parameter(Mandatory, ParameterSetName="FromFile")] [String]$FileOut, [Parameter(Mandatory, ValueFromPipeline, ParameterSetName="FromString")] [String]$String ) process { $Csp = [System.Security.Cryptography.AesCryptoServiceProvider]::New() $Csp.Padding = [System.Security.Cryptography.PaddingMode]::PKCS7 $Csp.Mode = [System.Security.Cryptography.CipherMode]::CBC $Csp.BlockSize = 128 $Csp.KeySize = 256 $Csp.GenerateIV() $IV = $Csp.IV $SHA256 = [System.Security.Cryptography.SHA256]::Create() $KeySha = $SHA256.ComputeHash([System.Text.ASCIIEncoding]::UTF8.GetBytes($Key)) $Encryptor = $Csp.CreateEncryptor($KeySha,$IV) if($PsCmdlet.ParameterSetName -eq "FromFile"){ if($PSCmdlet.ShouldProcess($FileOut, "WriteByte")){ $FileInStream = [System.IO.FileStream]::New($FileIn,[System.IO.FileMode]::Open) $FileOutStream = [System.IO.FileStream]::new($FileOut,[System.IO.FileMode]::Create) $CryptoStream = [System.Security.Cryptography.CryptoStream]::New($FileOutStream, $Encryptor, [System.Security.Cryptography.CryptoStreamMode]::Write) $FileOutStream.Write($IV,0,$IV.Count) $ApproxSize = (Get-Item $FileIn).Length Write-Progress -Activity $FileIn -Status "Encrypting" -PercentComplete 0 $n = 0 $Completed = $False try { do { try { $Byte = $FileInStream.ReadByte() } catch { Throw "Read failed" } if($Byte -ne -1){ $CryptoStream.WriteByte($Byte) $n++ } if(($n % 102400) -eq 0){ $CurrentSize = (Get-Item $FileOut).Length if($ApproxSize -gt 0){ $PercentComplete = [Math]::Round(((1 - ($ApproxSize - $CurrentSize)/$ApproxSize) * 100), 1) Write-Progress -Activity $FileIn -Status "Encrypting $PercentComplete%" -PercentComplete $PercentComplete } } } while ($Byte -ne -1) $CryptoStream.FlushFinalBlock() $Completed = $True } finally { if(-not $Completed){ Remove-Item $FileOut Throw "Encryption failed" } } $FileInStream.Dispose() $FileOutStream.Dispose() $CryptoStream.Dispose() } } if($PsCmdlet.ParameterSetName -eq "FromString"){ $MemoryStream = [System.IO.MemoryStream]::New() $CryptoStream = [System.Security.Cryptography.CryptoStream]::New($MemoryStream, $Encryptor, [System.Security.Cryptography.CryptoStreamMode]::Write) $MemoryStream.Write($IV,0,$IV.Count) [System.Text.Encoding]::UTF8.GetBytes($String) | ForEach-Object { $CryptoStream.WriteByte($_) } $CryptoStream.FlushFinalBlock() # out string ConvertFrom-MemoryStream -MemoryStream $MemoryStream -ToBase64 $CryptoStream.Dispose() } } } |