Backup Wallet on change password (#578)

* backup wallet on changepass * move backup to cli * case * update nuget Co-authored-by: Shargon <shargon@gmail.com> Co-authored-by: Owen Zhang <38493437+superboyiii@users.noreply.github.com>
neo-project · May 7, 2020 · abc9f2d · abc9f2d
1 parent b96017f
commit abc9f2d
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 1 deletion.
diff --git a/neo-cli/CLI/MainService.Wallet.cs b/neo-cli/CLI/MainService.Wallet.cs
@@ -483,6 +483,25 @@ private void OnChangePasswordCommand()
                 return;
             }
 
+            if (CurrentWallet is NEP6Wallet wallet)
+            {
+                string backupFile = wallet.Path + ".bak";
+                if (!File.Exists(wallet.Path) || File.Exists(backupFile))
+                {
+                    Console.WriteLine("Wallet backup fail");
+                    return;
+                }
+                try
+                {
+                    File.Copy(wallet.Path, backupFile);
+                }
+                catch (IOException)
+                {
+                    Console.WriteLine("Wallet backup fail");
+                    return;
+                }
+            }
+
             bool succeed = CurrentWallet.ChangePassword(oldPassword, newPassword);
             if (succeed)
             {

diff --git a/neo-cli/neo-cli.csproj b/neo-cli/neo-cli.csproj
@@ -28,7 +28,7 @@
   </ItemGroup>
 
   <ItemGroup>
-    <PackageReference Include="Neo" Version="3.0.0-CI00906" />
+    <PackageReference Include="Neo" Version="3.0.0-CI00907" />
   </ItemGroup>
 
   <ItemGroup>