From 419c655533126a379c5cc3f97ba59fbf4d71410b Mon Sep 17 00:00:00 2001
From: varjolintu <sami.vanttinen@protonmail.com>
Date: Fri, 11 May 2018 14:03:50 +0300
Subject: [PATCH] Credential retrieval fix

---
 keepassxc-browser/background/event.js   |  3 +++
 keepassxc-browser/background/keepass.js | 11 +++++++++++
 keepassxc-browser/keepassxc-browser.js  |  2 +-
 3 files changed, 15 insertions(+), 1 deletion(-)

diff --git a/keepassxc-browser/background/event.js b/keepassxc-browser/background/event.js
index 8d1b39c..f0f6fee 100755
--- a/keepassxc-browser/background/event.js
+++ b/keepassxc-browser/background/event.js
@@ -159,6 +159,9 @@ kpxcEvent.onReconnect = function(callback, tab) {
     // Add a small timeout after reconnecting. Just to make sure. It's not pretty, I know :(
     setTimeout(() => {
         keepass.reconnect(callback, tab).then((configured) => {
+            browser.tabs.sendMessage(tab.id, {
+                action: 'redetect_fields'
+            });
             kpxcEvent.showStatus(configured, tab, callback);
         });
     }, 500);
diff --git a/keepassxc-browser/background/keepass.js b/keepassxc-browser/background/keepass.js
index e8fe811..711a5e9 100755
--- a/keepassxc-browser/background/keepass.js
+++ b/keepassxc-browser/background/keepass.js
@@ -839,6 +839,17 @@ keepass.onNativeMessage = function(response) {
         keepass.testAssociation((associationResponse) => {
             keepass.isConfigured().then((configured) => {
                 keepass.updatePopup(configured ? 'normal' : 'cross');
+
+                // Send message to content script
+                browser.tabs.query({ active: true, currentWindow: true }).then((tabs) => {
+                    if (tabs.length) {
+                        browser.tabs.sendMessage(tabs[0].id, {
+                            action: 'check_database_hash',
+                            hash: {old: keepass.previousDatabaseHash, new: keepass.databaseHash}
+                        });
+                        keepass.previousDatabaseHash = keepass.databaseHash;
+                    }
+                });
             });
         }, null);
     }
diff --git a/keepassxc-browser/keepassxc-browser.js b/keepassxc-browser/keepassxc-browser.js
index 490f675..222dc8d 100755
--- a/keepassxc-browser/keepassxc-browser.js
+++ b/keepassxc-browser/keepassxc-browser.js
@@ -1202,6 +1202,7 @@ cip.detectDatabaseChange = function(response) {
                 args: [ true ]    // Set polling to true, this is an internal function call
             });
         } else if (response.new !== 'no-hash' && response.new !== response.old) {
+            _called.retrieveCredentials = false;
             browser.runtime.sendMessage({
                 action: 'load_settings',
             }).then((response) => {
@@ -1257,7 +1258,6 @@ cip.initCredentialFields = function(forceCall) {
         } 
 
         if (cip.settings.autoRetrieveCredentials && _called.retrieveCredentials === false && (cip.url && cip.submitUrl)) {
-            _called.retrieveCredentials = true;
             browser.runtime.sendMessage({
                 action: 'retrieve_credentials',
                 args: [ cip.url, cip.submitUrl ]