Upgrade utilities (Update checker and snapshot downloader/application)

configure.ac

@@ -985,6 +985,7 @@ if test x$use_pkgconfig = xyes; then
     [
       PKG_CHECK_MODULES([SSL], [libssl],, [AC_MSG_ERROR(openssl  not found.)])
       PKG_CHECK_MODULES([CRYPTO], [libcrypto],,[AC_MSG_ERROR(libcrypto  not found.)])
+      PKG_CHECK_MODULES([LIBZIP], [libzip],,[AC_MSG_ERROR(libzip  not found.)])
       PKG_CHECK_MODULES([CURL], [libcurl],,[AC_MSG_ERROR(libcurl  not found.)])
       if test x$use_qr != xno; then
         BITCOIN_QT_CHECK([PKG_CHECK_MODULES([QR], [libqrencode], [have_qrencode=yes], [have_qrencode=no])])
@@ -998,6 +999,9 @@ else
   AC_CHECK_HEADER([openssl/ssl.h],, AC_MSG_ERROR(libssl headers missing),)
   AC_CHECK_LIB([ssl], [main],SSL_LIBS=-lssl, AC_MSG_ERROR(libssl missing))
 
+  AC_CHECK_HEADER([zipconf.h],,AC_MSG_ERROR(libzip headers missing))
+  AC_CHECK_LIB([zip], [main], LIBZIP_LIBS=-lzip, AC_MSG_ERROR(libzip missing))
+
   AC_CHECK_HEADER([curl/curl.h],, AC_MSG_ERROR(libcurl headers missing),)
   AC_CHECK_LIB([curl], [main],, AC_MSG_ERROR(libcurl missing))
 

depends/packages/libzip.mk

@@ -41,4 +41,3 @@ endef
 define $(package)_stage_cmds
   $(MAKE) DESTDIR=$($(package)_staging_dir) install
 endef
-

src/Makefile.am

@@ -6,7 +6,7 @@ DIST_SUBDIRS = univalue
 
 AM_LDFLAGS = ${libcurl_LIBS} $(PTHREAD_CFLAGS) $(LIBTOOL_LDFLAGS) $(HARDENED_LDFLAGS)
 AM_CXXFLAGS = $(HARDENED_CXXFLAGS) $(ERROR_CXXFLAGS)
-AM_CPPFLAGS = ${libcurl_CFLAGS} $(HARDENED_CPPFLAGS) -DSTATICLIB -DCURL_STATICLIB -DMINIUPNP_STATICLIB
+AM_CPPFLAGS = ${libcurl_CFLAGS} $(HARDENED_CPPFLAGS) -DSTATICLIB -DCURL_STATICLIB -DMINIUPNP_STATICLIB -DZIP_STATIC -DNN_STATIC_LIB
 EXTRA_LIBRARIES =
 
 if EMBEDDED_UNIVALUE
@@ -132,6 +132,7 @@ GRIDCOIN_CORE_H = \
 	txdb-leveldb.h \
 	ui_interface.h \
 	uint256.h \
+        upgrade.h \
 	util/memory.h \
 	util/reverse_iterator.h \
 	util/strencodings.h \
@@ -202,6 +203,7 @@ GRIDCOIN_CORE_CPP = addrdb.cpp \
 	sync.cpp \
 	txdb-leveldb.cpp \
 	uint256.cpp \
+        upgrade.cpp \
 	util/strencodings.cpp \
 	util.cpp \
 	version.cpp \

src/Makefile.qt.include

@@ -180,6 +180,7 @@ GRIDCOINRESEARCH_QT_H = \
   qt/transactionrecord.h \
   qt/transactiontablemodel.h \
   qt/transactionview.h \
+  qt/upgradeqt.h \
   qt/votingdialog.h \
   qt/walletmodel.h \
   qt/winshutdownmonitor.h
@@ -222,6 +223,7 @@ GRIDCOINRESEARCH_QT_CPP = \
   qt/transactionrecord.cpp \
   qt/transactiontablemodel.cpp \
   qt/transactionview.cpp \
+  qt/upgradeqt.cpp \
   qt/votingdialog.cpp \
   qt/walletmodel.cpp \
   qt/winshutdownmonitor.cpp

src/gridcoinresearchd.cpp

@@ -15,6 +15,7 @@
 #include "rpcserver.h"
 #include "rpcclient.h"
 #include "ui_interface.h"
+#include "upgrade.h"
 
 #include <boost/thread.hpp>
 #include <boost/algorithm/string/predicate.hpp>
@@ -85,16 +86,52 @@ bool AppInit(int argc, char* argv[])
             return false;
         }
 
-        LogPrintf("AppInit");
-
         if (!boost::filesystem::is_directory(GetDataDir(false)))
         {
             fprintf(stderr, "Error: Specified directory does not exist\n");
             Shutdown(NULL);
         }
 
+        /** Check here config file incase TestNet is set there and not in mapArgs **/
         ReadConfigFile(mapArgs, mapMultiArgs);
 
+        // Check to see if the user requested a snapshot and we are not running TestNet!
+        if (mapArgs.count("-snapshotdownload") && !mapArgs.count("-testnet"))
+        {
+            Upgrade Snapshot;
+
+            // Let's check make sure gridcoin is not already running in the data directory.
+            // Use new probe feature
+            if (!LockDirectory(GetDataDir(), ".lock", false))
+            {
+                fprintf(stderr, "Cannot obtain a lock on data directory %s.  Gridcoin is probably already running.", GetDataDir().string().c_str());
+
+                exit(1);
+            }
+
+            else
+            {
+                try
+                {
+                    Snapshot.SnapshotMain();
+                }
+
+                catch (std::runtime_error& e)
+                {
+                    LogPrintf("Snapshot Downloader: Runtime exception occured in SanpshotMain() (%s)", e.what());
+
+                    Snapshot.DeleteSnapshot();
+
+                    exit(1);
+                }
+            }
+
+            // Delete snapshot file
+            Snapshot.DeleteSnapshot();
+        }
+
+        LogPrintf("AppInit");
+
         // Command-line RPC  - Test this - ensure single commands execute and exit please.
         for (int i = 1; i < argc; i++)
             if (!IsSwitchChar(argv[i][0]) && !boost::algorithm::istarts_with(argv[i], "gridcoinresearchd"))

src/init.cpp

@@ -17,6 +17,7 @@
 #include "neuralnet/neuralnet.h"
 #include "neuralnet/researcher.h"
 #include "neuralnet/tally.h"
+#include "upgrade.h"
 
 #include <boost/filesystem.hpp>
 #include <boost/filesystem/fstream.hpp>
@@ -57,12 +58,13 @@ extern unsigned int nActiveBeforeSB;
 extern bool fExplorer;
 extern bool fUseFastIndex;
 extern boost::filesystem::path pathScraper;
-
+bool fSnapshotRequest = false;
 // Dump addresses to banlist.dat every 5 minutes (300 s)
 static constexpr int DUMP_BANS_INTERVAL = 300;
 
 std::unique_ptr<BanMan> g_banman;
-
+/** Update checker pointer for CScheduler; **/
+std::unique_ptr<Upgrade> g_UpdateChecker;
 
 //////////////////////////////////////////////////////////////////////////////
 //
@@ -285,7 +287,15 @@ std::string HelpMessage()
         "  -rpcssl                                  " + _("Use OpenSSL (https) for JSON-RPC connections") + "\n" +
         "  -rpcsslcertificatechainfile=<file.cert>  " + _("Server certificate file (default: server.cert)") + "\n" +
         "  -rpcsslprivatekeyfile=<file.pem>         " + _("Server private key (default: server.pem)") + "\n" +
-        "  -rpcsslciphers=<ciphers>                 " + _("Acceptable ciphers (default: TLSv1.2+HIGH:TLSv1+HIGH:!SSLv2:!aNULL:!eNULL:!3DES:@STRENGTH)") + "\n";
+        "  -rpcsslciphers=<ciphers>                 " + _("Acceptable ciphers (default: TLSv1.2+HIGH:TLSv1+HIGH:!SSLv2:!aNULL:!eNULL:!3DES:@STRENGTH)") + "\n"
+
+        "\n" + _("Update/Snapshot options:") + "\n"
+        "  -snapshotdownload            " + _("Download and apply latest snapshot") + "\n"
+        "  -snapshoturl=<url>           " + _("Optional: Specify url of snapshot.zip file (ex: https://sub.domain.com/location/snapshot.zip)") + "\n"
+        "  -snapshotsha256url=<url>     " + _("Optional: Specify url of snapshot.sha256 file (ex: https://sub.domain.com/location/snapshot.sha256)") + "\n"
+        "  -disableupdatecheck          " + _("Optional: Disable update checks by wallet") + "\n"
+        "  -updatecheckinterval=<hours> " + _("Optional: Specify custom update interval checks in hours (Default: 24 hours (minimum 1 hour))") + "\n"
+        "  -updatecheckurl=<url>        " + _("Optional: Specify url of update version checks (ex: https://sub.domain.com/location/latest") + "\n";
 
     return strUsage;
 }
@@ -1043,6 +1053,53 @@ bool AppInit2(ThreadHandlerPtr threads)
         g_banman->DumpBanlist();
     }, DUMP_BANS_INTERVAL * 1000);
 
+    /** If this is not TestNet we check for updates on startup and daily **/
+    /** We still add to the scheduler regardless of the users choice however the choice is respected when they opt out**/
+    if (!fTestNet)
+    {
+        int64_t UpdateCheckInterval = 24;
+
+        // Save some cycles and only so this area if the argument exists
+        if (mapArgs.count("-updatecheckinterval"))
+        {
+            try
+            {
+                UpdateCheckInterval = GetArg("-updatecheckinterval", 24);
+                // trivial: Don't allow checks less then 1 hour apart of update checks to prevent server DDoS (what is a good value)
+                if (UpdateCheckInterval < 1)
+                {
+                    LogPrintf("UpdateChecker: Update check interval too small of %" PRId64 "; Defaulting to 24 hour intervals", UpdateCheckInterval);
+
+                    UpdateCheckInterval = 24;
+                }
+            }
+
+            catch (const std::exception& ex)
+            {
+                // Tell them the exception and what they had put in place
+                LogPrintf("UpdateChecker: Exception occured while obtaining interval for update checks (ex: %s -updatecheckinterval=%s); Defaulting to 24 hour intervals", ex.what(), GetArgument("-updatecheckinterval", ""));
+
+                UpdateCheckInterval = 24;
+            }
+        }
+
+        scheduler.scheduleEvery([]{g_UpdateChecker->CheckForLatestUpdate();}, UpdateCheckInterval * 60 * 60 * 1000);
+
+        if (!GetBoolArg("-disableupdatecheck", false))
+        {
+            LogPrintf("UpdateChecker: Update checks scheduled every %" PRId64 " hours.", UpdateCheckInterval);
+
+            LogPrintf("Updatechecker: Performing startup update check.");
+
+            g_UpdateChecker->CheckForLatestUpdate();
+        }
+
+        else
+            LogPrintf("UpdateChecker: Update checks are disabled by user.");
+    }
+
+    else
+        LogPrintf("UpdateChecker: Update checks are disable for TestNet.");
 
     return true;
 }

src/init.h

@@ -22,4 +22,5 @@ std::string HelpMessage();
 std::string VersionMessage();
 std::string LogSomething();
 
+extern bool fSnapshotRequest;
 #endif

src/noui.cpp

@@ -21,9 +21,20 @@ static bool noui_ThreadSafeAskFee(int64_t nFeeRequired, const std::string& strCa
     return true;
 }
 
+static int noui_UpdateMessageBox(const std::string& version, const std::string& message)
+{
+    std::string caption = _("Gridcoin Update Available");
+
+    LogPrintf("%s:\r\n%s", caption, message);
+    fprintf(stderr, "\r\n%s:\r\n%s\r\n%s\r\n", caption.c_str(), version.c_str(), message.c_str());
+
+    return 0;
+}
+
 void noui_connect()
 {
     // Connect bitcoind signal handlers
     uiInterface.ThreadSafeMessageBox.connect(noui_ThreadSafeMessageBox);
     uiInterface.ThreadSafeAskFee.connect(noui_ThreadSafeAskFee);
+    uiInterface.UpdateMessageBox.connect(noui_UpdateMessageBox);
 }