CryptoStream có thể được trả lại và vẫn có mọi thứ được xử lý đúng không?

Question

Nếu tôi có một CryptoStream mà tôi muốn vượt qua trở lại cho người sử dụng, cách tiếp cận ngây thơ sẽ là

public Stream GetDecryptedFileStream(string inputFile, byte[] key, byte[] iv) 
{ 
    var fsCrypt = new FileStream(inputFile, FileMode.Open, FileAccess.Read, FileShare.Read); 
    var rmCrypto = new RijndaelManaged(); 
    var transform = rmCrypto.CreateDecryptor(key, iv); 
    var cs = new CryptoStream(fsCrypt, transform, CryptoStreamMode.Read); 

    return cs; 
} 

Tôi biết rằng khi tôi vứt bỏ các CryptoStream nằm dưới FileStreamwill also be disposed. Vấn đề tôi đang gặp phải là tôi phải làm gì với rmCrypto và transform? RijndaelManaged và ICryptoTransform là các lớp dùng một lần, nhưng việc xử lý luồng không loại bỏ hai đối tượng đó.

Cách chính xác để xử lý tình huống này là gì?

Answer 1

Tôi muốn xem xét việc tạo lớp của riêng bạn kết thúc tốt đẹp luồng và sau đó bạn có thể quản lý việc loại bỏ các luồng này. Đôi khi dọc theo những dòng này (xin lỗi - không biết loại đối tượng biến đổi nằm ngoài đầu của tôi).

public CryptoStreamWrapper : Stream, IDisposable 
{ 
    public CryptoStreamWrapper(CryptoStream stream, RijndaelManaged rmCrypto, IDisposable transform) 
    { 
     this.transform = transform; 
     this.rmCrypto = rmCrypto; 
     this.stream = stream; 
    } 

    public void Dispose() 
    { 
     this.transform.Dispose(); 
     this.rmCrypto.Dispose(); 
     this.stream.Dispose(); 
    } 
} 

Answer 2

Ian đánh tôi đến khái niệm cơ bản (go upvote him), nhưng CryptoStream bản thân là không được niêm phong để nó là tầm thường để tạo ra một lớp được thừa kế mà kết thúc tốt đẹp những điều mà cần phải được xử lý.

/// <summary> 
/// Creates a class that creates a <see cref="CryptoStream"/> and wraps the disposing action of all the associated objects 
/// </summary> 
class ReturnableCryptoStream : CryptoStream 
{ 
    private readonly ICryptoTransform _transform; 
    private readonly IDisposable _algorithm; 

    public ReturnableCryptoStream(Stream stream, ICryptoTransform transform, CryptoStreamMode mode) 
     : this(stream, transform, mode, null) 
    { 
    } 

    public ReturnableCryptoStream(Stream stream, ICryptoTransform transform, CryptoStreamMode mode, IDisposable algorithm) 
     : base(stream, transform, mode) 
    { 
     _transform = transform; 
     _algorithm = algorithm; 
    } 

    protected override void Dispose(bool disposing) 
    { 
     base.Dispose(disposing); 
     if (disposing) 
     { 
      if (_transform != null) 
       _transform.Dispose(); 

      if (_algorithm != null) 
       _algorithm.Dispose(); 
     } 
    } 
} 

Được sử dụng như

public Stream GetDecryptedFileStream(string inputFile, byte[] key, byte[] iv) 
{ 
    var fsCrypt = new FileStream(inputFile, FileMode.Open, FileAccess.Read, FileShare.Read); 
    var rmCrypto = new RijndaelManaged(); 
    var transform = rmCrypto.CreateDecryptor(key, iv); 
    var cs = new ReturnableCryptoStream(fsCrypt, transform, CryptoStreamMode.Read, rmCrypto); 

    return cs; 
}