Encryption manager

WalletWasabi.Gui/Tabs/EncryptionManager/DecryptMessageView.xaml

@@ -3,21 +3,31 @@
   <controls:GroupBox Title="{Binding Title}" BorderThickness="0" Classes="docTabContainer">
     <Grid Classes="content" Margin="0 40 0 0">
       <Grid.RowDefinitions>
+        <RowDefinition Height="Auto" />
         <RowDefinition Height="Auto" />
         <RowDefinition Height="*" />
         <RowDefinition Height="*" />
         <RowDefinition Height="Auto" />
       </Grid.RowDefinitions>
-      <StackPanel Grid.Row="0">
-        <DockPanel IsVisible="{Binding !IsPublicKeyPresent}" LastChildFill="True">
-          <Button DockPanel.Dock="Right" Content="Get my public key" Command="{Binding MyPublicKeyCommand}" Width="200" />
-          <controls:NoparaPasswordBox Password="{Binding Password}" Watermark="Password" UseFloatingWatermark="True" />
-        </DockPanel>
-        <controls:ExtendedTextBox IsVisible="{Binding IsPublicKeyPresent}" Text="{Binding MyPublicKey}" Watermark="My public key" TextWrapping="Wrap" IsReadOnly="True" AcceptsReturn="True" UseFloatingWatermark="True" />
-      </StackPanel>
-      <controls:ExtendedTextBox Grid.Row="1" Text="{Binding EncryptedMessage}" Watermark="Encrypted message" UseFloatingWatermark="True" TextWrapping="Wrap" AcceptsReturn="True" />
-      <controls:ExtendedTextBox Grid.Row="2" Text="{Binding DecryptedMessage}" Watermark="Decrypted message" TextWrapping="Wrap" IsReadOnly="True" AcceptsReturn="True" UseFloatingWatermark="True" />
-      <DockPanel Grid.Row="3" LastChildFill="True">
+      <ListBox Grid.Row="0" Items="{Binding Addresses}" SelectedItem="{Binding SelectedItem}" MaxHeight="300">
+        <ListBox.ItemTemplate>
+          <DataTemplate>
+            <Grid ColumnDefinitions="* *">
+              <Grid.Styles>
+                <Style Selector="TextBlock">
+                  <Setter Property="VerticalAlignment" Value="Center" />
+                </Style>
+              </Grid.Styles>
+              <TextBlock Text="{Binding PubKey}" Grid.Column="0" Margin="10 0" />
+              <TextBlock Text="{Binding Label}" Grid.Column="1" Margin="10 0" />
+            </Grid>
+          </DataTemplate>
+        </ListBox.ItemTemplate>
+      </ListBox>
+      <controls:ExtendedTextBox Grid.Row="1" Text="{Binding MyPublicKey}" Watermark="My public key" TextWrapping="Wrap" IsReadOnly="True" AcceptsReturn="True" UseFloatingWatermark="True" />
+      <controls:ExtendedTextBox Grid.Row="2" Text="{Binding EncryptedMessage}" Watermark="Encrypted message" UseFloatingWatermark="True" TextWrapping="Wrap" AcceptsReturn="True" />
+      <controls:ExtendedTextBox Grid.Row="3" Text="{Binding DecryptedMessage}" Watermark="Decrypted message" TextWrapping="Wrap" IsReadOnly="True" AcceptsReturn="True" UseFloatingWatermark="True" />
+      <DockPanel Grid.Row="4" LastChildFill="True">
         <Button DockPanel.Dock="Right" Content="Decrypt" Command="{Binding DecryptCommand}" Width="100" />
         <TextBlock Text="{Binding WarningMessage}" Classes="warningMessage" TextWrapping="Wrap" />
       </DockPanel>

WalletWasabi.Gui/Tabs/EncryptionManager/DecryptMessageViewModel.cs

@@ -3,6 +3,8 @@
 using ReactiveUI;
 using System;
 using System.Collections.Generic;
+using System.Collections.ObjectModel;
+using System.Linq;
 using System.Text;
 using WalletWasabi.Gui.Tabs.EncryptionManager;
 using WalletWasabi.Gui.Tabs.WalletManager;
@@ -20,6 +22,8 @@ internal class DecryptMessageViewModel : CategoryViewModel
 		private string _myPublicKey;
 		private string _warningMessage;
 		private bool _isPublicKeyPresent;
+		private ObservableCollection<AddressPubKeyViewModel> _addresses;
+		private AddressPubKeyViewModel _selectedItem;
 
 		public string EncryptedMessage
 		{
@@ -57,6 +61,18 @@ public bool IsPublicKeyPresent
 			set => this.RaiseAndSetIfChanged(ref _isPublicKeyPresent, value);
 		}
 
+		public AddressPubKeyViewModel SelectedItem
+		{
+			get => _selectedItem;
+			set => this.RaiseAndSetIfChanged(ref _selectedItem, value);
+		}
+
+		public ObservableCollection<AddressPubKeyViewModel> Addresses
+		{
+			get => _addresses;
+			set => this.RaiseAndSetIfChanged(ref _addresses, value);
+		}
+
 		public ReactiveCommand DecryptCommand { get; }
 		public ReactiveCommand MyPublicKeyCommand { get; }
 
@@ -74,7 +90,7 @@ public DecryptMessageViewModel(EncryptionManagerViewModel owner) : base("Decrypt
 				() =>
 				{
 					WarningMessage = "";
-					DecryptedMessage = DecryptMessage(EncryptedMessage, Password);
+					DecryptedMessage = DecryptMessage(EncryptedMessage, MyPublicKey, Password);
 				},
 				canDecrypt
 			);
@@ -88,22 +104,39 @@ public DecryptMessageViewModel(EncryptionManagerViewModel owner) : base("Decrypt
 				() =>
 				{
 					WarningMessage = "";
-					MyPublicKey = Global.WalletService.KeyManager.EncryptedSecret.GetSecret(Password).PubKey.ToHex();
-					IsPublicKeyPresent = true;
-				}
+					DecryptedMessage = DecryptMessage(EncryptedMessage, MyPublicKey, Password);
+				},
+				canDecrypt
 			);
 
 			MyPublicKeyCommand.ThrownExceptions.Subscribe(ex =>
 			{
 				WarningMessage = ex.Message;
 			});
+
+			this.WhenAnyValue(x => x.SelectedItem)
+				.Subscribe((x) => MyPublicKey = x is null ? "" : x.PubKey);
+
+			IEnumerable<HdPubKey> keys = Global.WalletService.KeyManager.GetKeys();
+
+			Addresses = new ObservableCollection<AddressPubKeyViewModel>(keys.Select(a => new AddressPubKeyViewModel(a)));
 		}
 
-		private static string DecryptMessage(string message, string password)
+		private string DecryptMessage(string message, string pubkey, string password)
 		{
 			password = Guard.Correct(password);
-			var bitcoinPrivateKey = Global.WalletService.KeyManager.EncryptedSecret.GetSecret(password);
-			return bitcoinPrivateKey.PrivateKey.Decrypt(message);
+			HdPubKey hdPubKey = Global.WalletService.KeyManager.GetKeys().FirstOrDefault(k => k.PubKey.ToHex() == pubkey);
+			if (hdPubKey == null)
+			{
+				throw new InvalidOperationException("Could not fint the corresponting address in your wallet for that public key.");
+			}
+
+			(ExtKey secret, HdPubKey pubKey) secret = Global.WalletService.KeyManager.GetSecretsAndPubKeyPairs(password, hdPubKey.PubKey.ScriptPubKey).FirstOrDefault();
+			if (secret.Equals(default))
+			{
+				throw new InvalidOperationException("Could not fint the corresponting secret in your wallet for that ScriptPubKey");
+			}
+			return secret.secret.PrivateKey.Decrypt(message);
 		}
 	}
 }

WalletWasabi.Gui/ViewModels/AddressPubKeyViewModel.cs

@@ -0,0 +1,55 @@
+using System;
+using System.Collections.Generic;
+using System.Reactive.Disposables;
+using System.Text;
+using NBitcoin;
+using WalletWasabi.KeyManagement;
+
+namespace WalletWasabi.Gui.ViewModels
+{
+	public class AddressPubKeyViewModel : ViewModelBase, IDisposable
+	{
+		private CompositeDisposable Disposables { get; }
+		private HdPubKey _hdPubKey;
+
+		public string PubKey
+		{
+			get => _hdPubKey.PubKey.ToHex();
+		}
+
+		public string Label
+		{
+			get => _hdPubKey.Label;
+		}
+
+		public AddressPubKeyViewModel(HdPubKey hdpubkey)
+		{
+			_hdPubKey = hdpubkey;
+		}
+
+		#region IDisposable Support
+
+		private volatile bool _disposedValue = false; // To detect redundant calls
+
+		protected virtual void Dispose(bool disposing)
+		{
+			if (!_disposedValue)
+			{
+				if (disposing)
+				{
+					Disposables?.Dispose();
+				}
+				_disposedValue = true;
+			}
+		}
+
+		// This code added to correctly implement the disposable pattern.
+		public void Dispose()
+		{
+			// Do not change this code. Put cleanup code in Dispose(bool disposing) above.
+			Dispose(true);
+		}
+
+		#endregion IDisposable Support
+	}
+}