Changed the max length define

keepassxreboot · Apr 1, 2018 · d666625 · d666625
1 parent 8c6a7ce
commit d666625
Show file tree

Hide file tree

Showing 4 changed files with 12 additions and 13 deletions.
diff --git a/src/browser/BrowserAction.cpp b/src/browser/BrowserAction.cpp
@@ -20,6 +20,7 @@
 #include <QJsonParseError>
 #include "BrowserAction.h"
 #include "BrowserSettings.h"
+#include "NativeMessagingBase.h"
 #include "sodium.h"
 #include "sodium/crypto_box.h"
 #include "sodium/randombytes.h"
@@ -456,7 +457,7 @@ QString BrowserAction::encrypt(const QString plaintext, const QString nonce)
     std::vector<unsigned char> sk(sa.cbegin(), sa.cend());
 
     std::vector<unsigned char> e;
-    e.resize(max_length);
+    e.resize(NATIVE_MSG_MAX_LENGTH);
 
     if (m.empty() || n.empty() || ck.empty() || sk.empty()) {
         return QString();
@@ -484,7 +485,7 @@ QByteArray BrowserAction::decrypt(const QString encrypted, const QString nonce)
     std::vector<unsigned char> sk(sa.cbegin(), sa.cend());
 
     std::vector<unsigned char> d;
-    d.resize(max_length);
+    d.resize(NATIVE_MSG_MAX_LENGTH);
 
     if (m.empty() || n.empty() || ck.empty() || sk.empty()) {
         return QByteArray();

diff --git a/src/browser/BrowserService.h b/src/browser/BrowserService.h
@@ -25,8 +25,6 @@
 #include "gui/DatabaseTabWidget.h"
 #include "core/Entry.h"
 
-enum { max_length = 1024*1024 };
-
 class BrowserService : public QObject
 {
     Q_OBJECT

diff --git a/src/browser/NativeMessagingBase.h b/src/browser/NativeMessagingBase.h
@@ -37,6 +37,8 @@
 #include <sys/socket.h> 
 #endif
 
+static const int NATIVE_MSG_MAX_LENGTH = 1024*1024;
+
 class NativeMessagingBase : public QObject
 {
     Q_OBJECT

diff --git a/src/browser/NativeMessagingHost.cpp b/src/browser/NativeMessagingHost.cpp
@@ -142,19 +142,17 @@ void NativeMessagingHost::newLocalConnection()
 void NativeMessagingHost::newLocalMessage()
 {
     QLocalSocket* socket = qobject_cast<QLocalSocket*>(QObject::sender());
-    if (socket) {
-        socket->setReadBufferSize(max_length);
-        int socketDesc = socket->socketDescriptor();
-        if (socketDesc) {
-            int max = max_length;
-            res = setsockopt(socketDesc, SOL_SOCKET, SO_SNDBUF, &max, sizeof(max));
-        }
-    }
-
     if (!socket || socket->bytesAvailable() <= 0) {
         return;
     }
 
+    socket->setReadBufferSize(NATIVE_MSG_MAX_LENGTH);
+    int socketDesc = socket->socketDescriptor();
+    if (socketDesc) {
+        int max = NATIVE_MSG_MAX_LENGTH;
+        setsockopt(socketDesc, SOL_SOCKET, SO_SNDBUF, &max, sizeof(max));
+    }
+
     QByteArray arr = socket->readAll();
     if (arr.isEmpty()) {
         return;