From 5315ded7861d218072991a5bfc33d7492fd0a052 Mon Sep 17 00:00:00 2001
From: "Jonathan \"Duke\" Leto" <jonathan@leto.net>
Date: Fri, 4 Sep 2020 19:31:05 -0700
Subject: [PATCH] Support wallet.dat outside of datadir

Ported from https://github.com/zcash/zcash/commit/6a7cfdea54a9469ac1234e32db9d2fed0bb146af
which did not apply cleanly.
---
 qa/rpc-tests/feature_walletfile.py | 51 ++++++++++++++++++++++++++++++
 src/wallet/wallet.cpp              | 17 ++++++++++
 2 files changed, 68 insertions(+)
 create mode 100755 qa/rpc-tests/feature_walletfile.py

diff --git a/qa/rpc-tests/feature_walletfile.py b/qa/rpc-tests/feature_walletfile.py
new file mode 100755
index 000000000..f38403574
--- /dev/null
+++ b/qa/rpc-tests/feature_walletfile.py
@@ -0,0 +1,51 @@
+#!/usr/bin/env python3
+# Copyright (c) 2017      The Bitcoin Core developers
+# Copyright (c) 2019-2020 The Hush developers
+# Distributed under the MIT software license, see the accompanying
+# file COPYING or https://www.opensource.org/licenses/mit-license.php
+"""Test wallet file location."""
+
+import os
+
+from test_framework.util import start_node, stop_node, assert_start_raises_init_error
+
+from test_framework.test_framework import BitcoinTestFramework
+
+class WalletFileTest(BitcoinTestFramework):
+    def set_test_params(self):
+        self.num_nodes = 1
+        self.setup_clean_chain = True
+
+    def run_test(self):
+        # test default wallet location
+        assert os.path.isfile(os.path.join(self.options.tmpdir, "node0", "regtest", "wallet.dat"))
+
+        # test alternative wallet file name in datadir
+        stop_node(self.nodes[0], 0)
+        self.nodes[0] = start_node(0, self.options.tmpdir, ["-wallet=altwallet.dat"])
+        assert os.path.isfile(os.path.join(self.options.tmpdir, "node0", "regtest", "altwallet.dat"))
+
+        # test wallet file outside datadir
+        tempname = os.path.join(self.options.tmpdir, "outsidewallet.dat")
+        stop_node(self.nodes[0], 0)
+        self.nodes[0] = start_node(0, self.options.tmpdir, ["-wallet=%s" % tempname])
+        assert os.path.isfile(tempname)
+
+        # test the case where absolute path does not exist
+        assert not os.path.isdir("/this_directory_must_not_exist")
+        invalidpath = os.path.join("/this_directory_must_not_exist/", "foo.dat")
+        stop_node(self.nodes[0], 0)
+        assert_start_raises_init_error(0, "-wallet=%s" % invalidpath,
+            "Error: Absolute path %s does not exist")
+
+        # relative path does not exist
+        invalidpath = os.path.join("wallet", "foo.dat")
+        assert_start_raises_init_error(0, "-wallet=%s" % invalidpath,
+            "Error: Relative path %s does not exist")
+
+        # create dir and retry
+        os.mkdir(os.path.join(self.options.tmpdir, "node0", "regtest", "wallet"))
+        self.nodes[0] = start_node(0, self.options.tmpdir, ["-wallet=%s" % invalidpath])
+
+if __name__ == '__main__':
+    WalletFileTest().main()
diff --git a/src/wallet/wallet.cpp b/src/wallet/wallet.cpp
index f519b5a46..1347ef974 100644
--- a/src/wallet/wallet.cpp
+++ b/src/wallet/wallet.cpp
@@ -668,6 +668,23 @@ void CWallet::Flush(bool shutdown)
 
 bool CWallet::Verify(const string& walletFile, string& warningString, string& errorString)
 {
+    LogPrintf("Using wallet %s\n", walletFile);
+    uiInterface.InitMessage(_("Verifying wallet..."));
+
+    if (walletFile != boost::filesystem::basename(walletFile) + boost::filesystem::extension(walletFile)) {
+        boost::filesystem::path path(walletFile);
+        if (path.is_absolute()) {
+            if (!boost::filesystem::exists(path.parent_path())) {
+                return UIError(strprintf(_("Absolute path %s does not exist!"), walletFile));
+            }
+        } else {
+            boost::filesystem::path full_path = GetDataDir() / path;
+            if (!boost::filesystem::exists(full_path.parent_path())) {
+                return UIError(strprintf(_("Relative path %s does not exist!"), walletFile));
+            }
+        }
+    }
+
     if (!bitdb.Open(GetDataDir()))
     {
         // try moving the database env out of the way