Index: src/ch/UpdateChecker.h =================================================================== diff -u -r50ad2dc9f0b42ba432bb54e4a042582277410773 -r50007f112b77cba170e6c427fee5428bda2d9dc5 --- src/ch/UpdateChecker.h (.../UpdateChecker.h) (revision 50ad2dc9f0b42ba432bb54e4a042582277410773) +++ src/ch/UpdateChecker.h (.../UpdateChecker.h) (revision 50007f112b77cba170e6c427fee5428bda2d9dc5) @@ -48,7 +48,7 @@ ~CUpdateChecker(); /// Starts the 'check for updates' thread - bool AsyncCheckForUpdates(const wchar_t* pszSite, const wchar_t* pszLanguage, UpdateVersionInfo::EVersionType bCheckBeta, bool bOnlyIfConnected); + bool AsyncCheckForUpdates(const wchar_t* pszSite, const wchar_t* pszLanguage, UpdateVersionInfo::EVersionType eUpdateChannel, bool bOnlyIfConnected, bool bSendHeaders); /// Stops checking and cleanups the object void Cleanup(); @@ -63,6 +63,7 @@ const wchar_t* GetDownloadAddress() const { return m_strDownloadAddress; } const wchar_t* GetReleaseDate() const { return m_strReleaseDate; } const wchar_t* GetReleaseNotes() const { return m_strReleaseNotes; } + bool GetSendHeaders() const { return m_bSendHeaders; } protected: /// Thread function (handles most of the internet connection operation) @@ -75,8 +76,10 @@ /// Sets the versions and download address void SetVersionsAndAddress(PCTSTR pszAddress, PCTSTR pszNumericVersion, PCTSTR pszReadableVersion, PCTSTR pszReleaseDate, PCTSTR pszReleaseNotes); /// Retrieves the site address - void GetSiteAddress(CString& rstrAddress) const; + CString GetSiteAddress() const; + void SetSendHeaders(bool bSendHeaders); + /// Returns information if we're interested in beta versions UpdateVersionInfo::EVersionType GetUpdateChannel(); static std::wstring FormatDate(const boost::gregorian::date& date); @@ -91,13 +94,13 @@ CString m_strDownloadAddress; CString m_strReleaseDate; CString m_strReleaseNotes; - + bool m_bSendHeaders = true; + ECheckResult m_eResult; CAsyncHttpFile m_httpFile; UpdateHeaders m_tUpdateHeaders; - HANDLE m_hThread; HANDLE m_hKillEvent; mutable CRITICAL_SECTION m_cs;