refactor(bootstrap): Store public key as ToxPk rather than QString

Talkless · Feb 21, 2022 · 7aa8d71 · 7aa8d71
1 parent e7e30ad
commit 7aa8d71
Show file tree

Hide file tree

Showing 4 changed files with 9 additions and 6 deletions.
diff --git a/src/core/core.cpp b/src/core/core.cpp
@@ -837,8 +837,7 @@ void Core::bootstrapDht()
             address = dhtServer.ipv4.toLatin1();
         }
 
-        ToxPk pk{dhtServer.userId};
-        const uint8_t* pkPtr = pk.getData();
+        const uint8_t* pkPtr = dhtServer.publicKey.getData();
 
         Tox_Err_Bootstrap error;
         if (dhtServer.statusUdp) {

diff --git a/src/core/dhtserver.cpp b/src/core/dhtserver.cpp
@@ -32,7 +32,7 @@ bool DhtServer::operator==(const DhtServer& other) const
         && ipv4 == other.ipv4
         && ipv6 == other.ipv6
         && maintainer == other.maintainer
-        && userId == other.userId
+        && publicKey == other.publicKey
         && udpPort == other.udpPort
         && tcpPorts == other.tcpPorts);
 }

diff --git a/src/core/dhtserver.h b/src/core/dhtserver.h
@@ -19,6 +19,8 @@
 
 #pragma once
 
+#include "toxpk.h"
+
 #include <QString>
 #include <vector>
 
@@ -29,7 +31,7 @@ struct DhtServer
     QString ipv4;
     QString ipv6;
     QString maintainer;
-    QString userId;
+    ToxPk publicKey;
     quint16 udpPort;
     std::vector<uint16_t> tcpPorts;
 

diff --git a/src/net/bootstrapnodeupdater.cpp b/src/net/bootstrapnodeupdater.cpp
@@ -20,6 +20,8 @@
 #include "bootstrapnodeupdater.h"
 
 #include "src/persistence/paths.h"
+#include "src/core/toxpk.h"
+#include "src/core/toxid.h"
 
 #include <QDirIterator>
 #include <QFile>
@@ -120,7 +122,7 @@ void jsonNodeToDhtServer(const QJsonObject& node, QList<DhtServer>& outList)
     server.statusUdp = true;
     server.statusTcp = status_tcp;
     server.tcpPorts = tcp_ports;
-    server.userId = public_key;
+    server.publicKey = ToxPk{public_key};
     server.udpPort = udp_port_u16;
     server.maintainer = maintainer;
     server.ipv4 = ipv4_address;
@@ -182,7 +184,7 @@ QByteArray serialize(QList<DhtServer> nodes)
         nodeJson.insert(NodeFields::status_tcp, node.statusTcp);
         nodeJson.insert(NodeFields::ipv4, node.ipv4);
         nodeJson.insert(NodeFields::ipv6, node.ipv6);
-        nodeJson.insert(NodeFields::public_key, node.userId);
+        nodeJson.insert(NodeFields::public_key, node.publicKey.toString());
         nodeJson.insert(NodeFields::udp_port, node.udpPort);
         nodeJson.insert(NodeFields::maintainer, node.maintainer);