OSDN Git Service

Updated the UpdateChecker class to use cURL instead of Wget.
[mutilities/MUtilities.git] / include / MUtils / UpdateChecker.h
1 ///////////////////////////////////////////////////////////////////////////////
2 // MuldeR's Utilities for Qt
3 // Copyright (C) 2004-2018 LoRd_MuldeR <MuldeR2@GMX.de>
4 //
5 // This library is free software; you can redistribute it and/or
6 // modify it under the terms of the GNU Lesser General Public
7 // License as published by the Free Software Foundation; either
8 // version 2.1 of the License, or (at your option) any later version.
9 //
10 // This library is distributed in the hope that it will be useful,
11 // but WITHOUT ANY WARRANTY; without even the implied warranty of
12 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
13 // Lesser General Public License for more details.
14 //
15 // You should have received a copy of the GNU Lesser General Public
16 // License along with this library; if not, write to the Free Software
17 // Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA
18 //
19 // http://www.gnu.org/licenses/lgpl-2.1.txt
20 //////////////////////////////////////////////////////////////////////////////////
21
22 #pragma once
23
24 //MUtils
25 #include <MUtils/Global.h>
26
27 //Qt
28 #include <QThread>
29 #include <QDate>
30
31 ///////////////////////////////////////////////////////////////////////////////
32
33 namespace MUtils
34 {
35         class MUTILS_API UpdateCheckerInfo
36         {
37                 friend class UpdateChecker;
38
39         public:
40                 UpdateCheckerInfo(void);
41                 void resetInfo(void);
42                 bool isComplete(void);
43
44                 const quint32 &getBuildNo(void)          const { return m_buildNo;          }
45                 const QDate   &getBuildDate(void)        const { return m_buildDate;        }
46                 const QString &getDownloadSite(void)     const { return m_downloadSite;     }
47                 const QString &getDownloadAddress(void)  const { return m_downloadAddress;  }
48                 const QString &getDownloadFilename(void) const { return m_downloadFilename; }
49                 const QString &getDownloadFilecode(void) const { return m_downloadFilecode; }
50                 const QString &getDownloadChecksum(void) const { return m_downloadChecksum; }
51
52         private:
53                 quint32 m_buildNo;
54                 QDate m_buildDate;
55                 QString m_downloadSite;
56                 QString m_downloadAddress;
57                 QString m_downloadFilename;
58                 QString m_downloadFilecode;
59                 QString m_downloadChecksum;
60         };
61
62         // ----------------------------------------------------------------
63
64         class MUTILS_API UpdateChecker : public QThread
65         {
66                 Q_OBJECT
67
68         public:
69                 enum
70                 {
71                         UpdateStatus_NotStartedYet             = 0,
72                         UpdateStatus_CheckingConnection        = 1,
73                         UpdateStatus_FetchingUpdates           = 2,
74                         UpdateStatus_CompletedUpdateAvailable  = 3,
75                         UpdateStatus_CompletedNoUpdates        = 4,
76                         UpdateStatus_CompletedNewVersionOlder  = 5,
77                         UpdateStatus_ErrorNoConnection         = 6,
78                         UpdateStatus_ErrorConnectionTestFailed = 7,
79                         UpdateStatus_ErrorFetchUpdateInfo      = 8,
80                         UpdateStatus_CancelledByUser           = 9
81                 }
82                 update_status_t;
83
84                 UpdateChecker(const QString &binCurl, const QString &binGnuPG, const QString &binKeys, const QString &applicationId, const quint32 &installedBuildNo, const bool betaUpdates, const bool testMode = false);
85                 ~UpdateChecker(void);
86
87                 const int  getUpdateStatus(void)             const { return m_status; }
88                 const bool getSuccess(void)                  const { return m_success; };
89                 const int  getMaximumProgress(void)          const { return m_maxProgress; };
90                 const int  getCurrentProgress(void)          const { return m_progress; };
91                 const UpdateCheckerInfo *getUpdateInfo(void) const { return m_updateInfo.data(); }
92
93                 bool cancel(void) { return m_cancelled.ref(); }
94
95         public slots:
96                 void start(Priority = InheritPriority);
97
98         protected:
99                 void run(void);
100                 void checkForUpdates(void);
101                 void testMirrorsList(void);
102
103         signals:
104                 void statusChanged(const int status);
105                 void progressChanged(const int progress);
106                 void messageLogged(const QString &text);
107
108         private:
109                 const int m_maxProgress;
110                 QScopedPointer<UpdateCheckerInfo> m_updateInfo;
111         
112                 const bool m_betaUpdates;
113                 const bool m_testMode;
114
115                 const QString m_applicationId;
116                 const quint32 m_installedBuildNo;
117
118                 const QString m_binaryCurl;
119                 const QString m_binaryGnuPG;
120                 const QString m_binaryKeys;
121
122                 const QScopedPointer<const QHash<QString, QString>> m_environment;
123
124                 QAtomicInt m_success;
125                 QAtomicInt m_cancelled;
126
127                 int m_status;
128                 int m_progress;
129
130                 inline void setStatus(const int status);
131                 inline void setProgress(const int progress);
132                 inline void log(const QString &str1, const QString &str2 = QString(), const QString &str3 = QString(), const QString &str4 = QString());
133
134                 bool getUpdateInfo(const QString &url, const QString &outFileVers, const QString &outFileSign);
135                 bool tryContactHost(const QString &hostname, const int &timeoutMsec);
136                 bool parseVersionInfo(const QString &file, UpdateCheckerInfo *updateInfo);
137
138                 bool getFile(const QString &url, const QString &outFile, const unsigned int maxRedir = 8U);
139                 bool checkSignature(const QString &file, const QString &signature);
140                 bool tryUpdateMirror(UpdateCheckerInfo *updateInfo, const QString &url, const bool &quick);
141         };
142 }