src/MassTransit/MessageData/EncryptedMessageDataRepository.cs - develop - vshmidt/masstransit

src/MassTransit/MessageData/EncryptedMessageDataRepository.cs

74 строки3 KB

Chris Patterson

Integration of all external packages into MassTransit, split out into Abstractions, Middleware, and the core MassTransit assembly.

22 янв 2022, 18:24

22 янв 2022, 18:24b5b4f50

namespace MassTransit.MessageData
{
    using System;
    using System.Collections.Specialized;
    using System.IO;
    using System.Security.Cryptography;
    using System.Threading;
    using System.Threading.Tasks;
    using Internals;
    using Serialization;


    public class EncryptedMessageDataRepository :
        IMessageDataRepository
    {
        readonly IMessageDataRepository _repository;
        readonly ICryptoStreamProvider _streamProvider;

        /// <summary>
        /// Provides encrypted stream support to ensure that message data is encrypted at rest.
        /// </summary>
        /// <param name="repository">The original message data repository where message data is stored.</param>
        /// <param name="streamProvider">The encrypted stream provider</param>
        public EncryptedMessageDataRepository(IMessageDataRepository repository, ICryptoStreamProvider streamProvider)
        {
            _repository = repository;
            _streamProvider = streamProvider;
        }

        public async Task<Stream> Get(Uri address, CancellationToken cancellationToken = new CancellationToken())
        {
            var stream = await _repository.Get(address, cancellationToken).ConfigureAwait(false);

            address.TryGetValueFromQueryString("keyId", out var keyId);

            return _streamProvider.GetDecryptStream(stream, keyId, CryptoStreamMode.Read);
        }

        public async Task<Uri> Put(Stream stream, TimeSpan? timeToLive = null, CancellationToken cancellationToken = new CancellationToken())
        {
            string keyId = null;

            using var cryptoStream = _streamProvider.GetEncryptStream(stream, keyId, CryptoStreamMode.Read);

            var address = await _repository.Put(cryptoStream, timeToLive, cancellationToken).ConfigureAwait(false);

            var addressBuilder = new UriBuilder(address);

            var parameters = new NameValueCollection();
            if (!string.IsNullOrWhiteSpace(addressBuilder.Query))
            {
                var query = addressBuilder.Query;

                if (query.Contains("?"))
                    query = query.Substring(query.IndexOf('?') + 1);

                foreach (var parameter in query.Split('&'))
                {
                    if (string.IsNullOrWhiteSpace(parameter))
                        continue;

                    var pair = parameter.Split('=');

                    parameters.Add(pair[0], pair.Length == 2 ? pair[1] : "");
                }
            }

            parameters["keyId"] = "";
            addressBuilder.Query = parameters.ToString();

            return addressBuilder.Uri;
        }
    }
}