void CheckForUpdatesWorker(void *param)
{
if (checking) return;
/*
// this check doesn't work on some systems - not sure which or why
if(!(GetSystemMetrics(SM_NETWORK) & 1)) {
ShowError(TranslateT("No network - aborting update check"));
NLog("worker thread aborting - no network");
return 1; // no network
}
*/
NLog("CheckForUpdatesWorker thread starting");
checking = true;
WriteLastCheckTime();
bool confirm = !(((DWORD)param & 1) == 1);
bool restart = !(((DWORD)param & 2) == 2); // if restart is false, then we're doing an 'update and shutdown'
if(hwndOptions) PostMessage(hwndOptions, WMU_CHECKING, 0, 0);
bool use_popup = options.popup_notify && ArePopupsEnabled();
FilenameMap fn_map(5, CompareFileNameStruct);
if (use_popup)
{
ShowPopup(0, TranslateT("Checking for Updates"), _T(""), POPFLAG_SAVEHWND, -1);
}
else
{
CreateProgressWindow();
SendMessage(hwndProgress, WM_SETTEXT, 0, (LPARAM)TranslateT("Progress - Checking for updates..."));
if (options.use_xml_backend) SendMessage(hwndProgress, WMU_SETMESSAGE, (WPARAM)TranslateT("Downloading XML data"), 0);
else SendMessage(hwndProgress, WMU_SETMESSAGE, (WPARAM)TranslateT("Checking for updates"), 0);
}
EnterCriticalSection(&list_cs);
if (options.use_xml_backend)
{
if (UpdateXMLData(MC_PLUGINS)) // prevent double error messages (in some cases)
{
// iterate through the registered plugins
if (!use_popup) SendMessage(hwndProgress, WMU_SETMESSAGE, (WPARAM)TranslateT("Scanning plugins folder"), 0);
ScanPlugins(&fn_map, &update_list);
if (UpdateXMLData(MC_LOCALIZATION))
{
if (!use_popup) SendMessage(hwndProgress, WMU_SETMESSAGE, (WPARAM)TranslateT("Scanning language packs"), 0);
ScanLangpacks(&fn_map, &update_list);
}
if(!use_popup) SendMessage(hwndProgress, WMU_SETMESSAGE, (WPARAM)TranslateT("Updating component file listing ids"), 0);
UpdateFLIDs(update_list);
}
if (!use_popup) SendMessage(hwndProgress, WMU_SETMESSAGE, (WPARAM)TranslateT("Checking for updates"), 0);
}
UpdateList update_list2(update_list);
LeaveCriticalSection(&list_cs);
int count = update_list2.getCount(), index = 0;
TCHAR msg[512];
TCHAR *temp_str;
UpdateList todo;
for(index = 0; index < count; index++)
{
if(update_list2[index].update_options.enabled)
{
mir_sntprintf(msg, SIZEOF(msg), TranslateT("Checking plugin: %s"), (temp_str = GetTString(update_list2[index].update.szComponentName)));
mir_free(temp_str);
}
else
{
mir_sntprintf(msg, SIZEOF(msg), TranslateT("Skipping plugin: %s"), (temp_str = GetTString(update_list2[index].update.szComponentName)));
mir_free(temp_str);
}
if(!use_popup) {
SendMessage(hwndProgress, WMU_SETMESSAGE, (WPARAM)msg, 0);
SendMessage(hwndProgress, WMU_SETPROGRESS, (WPARAM)(int)(index * 100.0 / count), 0);
} //else if(hwndPop) // disabled - just annoying
//ChangePopupText(hwndPop, msg);
if (update_list2[index].update_options.enabled)
{
char *nv;
bool beta;
if (nv = UpdateRequired(update_list2[index], &beta))
{
todo.insert(new UpdateInternal(update_list2[index]));
todo[todo.getCount()-1].newVersion = nv;
todo[todo.getCount()-1].update_options.use_beta = beta;
}
}
if (!use_popup && hwndProgress == NULL)
{
RemoveFolder(options.temp_folder);
break; // user closed progress window - cancel
}
}
ProgressWindowDone();
if (hwndPop) PostMessage(hwndPop, WMU_CLOSEPOP, 0 , 0);
if(options.use_xml_backend) {
FreeXMLData(MC_PLUGINS);
FreeXMLData(MC_LOCALIZATION);
}
bool restore_status = true;
if (todo.getCount())
{
int cd_ret = CD_OK;
if (confirm)
{
if (use_popup)
{
ShowPopup(0, TranslateT("Updates Available"), TranslateT("Updated Miranda components detected.\nClick here to install."), POPFLAG_SAVEHWND, -1);
DWORD ret;
while ((ret = WaitForSingleObject(hEventPop, 200)) == WAIT_TIMEOUT && !Miranda_Terminated());
if (!pop_cancelled && ret == WAIT_OBJECT_0 && !Miranda_Terminated())
{
cd_ret = DialogBoxParam(hInst, MAKEINTRESOURCE(IDD_CONFIRMUPDATES), GetDesktopWindow(), DlgProcConfirm, (LPARAM)&todo);
}
else
{
if(hwndOptions) PostMessage(hwndOptions, WMU_DONECHECKING, 0, 0);
RestoreStatus();
for(int i=0; i<todo.getCount(); ++i)
free(todo[i].newVersion);
checking = false;
Netlib_CloseHandle(hNetlibHttp);
hNetlibHttp = NULL;
return;
}
}
else
cd_ret = DialogBoxParam(hInst, MAKEINTRESOURCE(IDD_CONFIRMUPDATES), GetDesktopWindow(), DlgProcConfirm, (LPARAM)&todo);
}
if (!confirm || cd_ret == CD_CONFALL || cd_ret == CD_NOINSTALL || cd_ret == CD_OK)
{
bool conf_all = (cd_ret == CD_CONFALL), no_install = (cd_ret == CD_NOINSTALL);
// ensure the backup folder exists (either create it or return non-zero signifying error)
if (options.backup && !CreatePath(options.backup_folder))
{
//MessageBox(0, Translate("Could not create backup folder"), Translate("Error"), MB_OK | MB_ICONERROR);
ShowError(TranslateT("Could not create backup folder"));
if (hwndOptions) PostMessage(hwndOptions, WMU_DONECHECKING, 0, 0);
RestoreStatus();
for (int i=0; i<todo.getCount(); ++i)
free(todo[i].newVersion);
checking = false;
Netlib_CloseHandle(hNetlibHttp);
hNetlibHttp = NULL;
return;
}
bool dll_enabled_or_langpack = DownloadUpdates(todo, &fn_map, confirm ? false : options.auto_dll_only);
if (todo.getCount() && !no_install)
{
if (!conf_all || DialogBox(hInst, MAKEINTRESOURCE(IDD_CONFIRMCOMPONENTS), GetDesktopWindow(), DlgProcConfirmComponents) == IDOK)
{
if (!dll_enabled_or_langpack && restart)
{
// we're not doing an 'update and shutdown', and we're not updating any active dlls...so just install
// get folders
TCHAR root_folder[MAX_PATH], plugins_folder[MAX_PATH];
GetRootDir(root_folder);
_tcscpy(plugins_folder, root_folder);
_tcscat(plugins_folder, _T("\\Plugins"));
// move files
MoveFiles(0, options.temp_folder, plugins_folder, options.backup_folder, root_folder);
// rescan to get correct version numbers
ScanPlugins(0, 0);
}
else
{
if (ExternProcess(restart) == 0) // if restarting, don't restore status
restore_status = false;
}
}
}
}
for (int i=0; i<todo.getCount(); ++i)
free(todo[i].newVersion);
}
else if (!restart)
{
PostMessage((HWND)CallService(MS_CLUI_GETHWND, 0, 0), WM_COMMAND, ID_ICQ_EXIT, 0);
}
if (restore_status && restart) // restore status if we're not doing an 'update and shutdown', and the update was unsuccessful (or there was nothing to update, or no need to shutdown)
{
checking = false;
if (hwndOptions) PostMessage(hwndOptions, WMU_DONECHECKING, 0, 0);
RestoreStatus();
}
Netlib_CloseHandle(hNetlibHttp);
hNetlibHttp = NULL;
}
bool UpdateXMLData(const Category cat, const char *redirect_url /*= 0*/, int recurse_count /*=0*/) {
if(recurse_count > MAX_REDIRECT_RECURSE) {
PUShowMessageT(TranslateT("Updater: Error getting data - too many redirects"), SM_WARNING);
return false;
}
NETLIBHTTPREQUEST req = {0};
NETLIBHTTPHEADER etag_hdr = {0};
if(OldXMLDataExists(cat)) {
// ensure backend not checked more than once every MIN_XMLDATA_AGE hours
long age = OldXMLDataAge(cat);
if(age >= 0 && age < MIN_XMLDATA_AGE) { // get it only if our file is at least 8 hours old
#ifdef DEBUG_HTTP_POPUPS
char buff[512];
sprintf(buff, "XML Data is recent (%d hours old) - not downloading, using local copy", age);
PUShowMessage(buff, SM_NOTIFY);
#endif
return LoadOldXMLData(cat, false);
}
// add ETag header for conditional get
DBCONTACTGETSETTING cgs;
DBVARIANT dbv;
cgs.szModule = MODULE;
char buff[256];
strcpy(buff, "DataETag_");
strcat(buff, category_files[cat]);
cgs.szSetting = buff;
cgs.pValue = &dbv;
if(!CallService(MS_DB_CONTACT_GETSETTING, 0, (LPARAM)&cgs)) {
req.headersCount = 1;
req.headers = &etag_hdr;
etag_hdr.szName = "If-None-Match";
etag_hdr.szValue = _strdup(dbv.pszVal);
DBFreeVariant(&dbv);
}
}
req.cbSize = sizeof(req);
req.requestType = REQUEST_GET;
char URL[MAX_PATH];
if(!redirect_url) {
strcpy(URL, MIM_BACKEND_URL_PREFIX);
strcat(URL, category_files[cat]);
strcat(URL, ".bz2");
} else {
strcpy(URL, redirect_url);
}
req.szUrl = URL;
req.flags = NLHRF_HTTP11;
req.nlc = hNetlibHttp;
if (CallService(MS_SYSTEM_GETVERSION, 0, 0) >= PLUGIN_MAKE_VERSION(0,9,0,5))
req.flags |= NLHRF_PERSISTENT | NLHRF_REDIRECT;
NETLIBHTTPREQUEST *resp = (NETLIBHTTPREQUEST *)CallService(MS_NETLIB_HTTPTRANSACTION, (WPARAM)hNetlibUser, (LPARAM)&req);
free(etag_hdr.szValue);
if (!resp)
{
hNetlibHttp = NULL;
if (!Miranda_Terminated())
{
int err = GetLastError();
if (err)
{
TCHAR buff[512];
int len = mir_sntprintf(buff, SIZEOF(buff), TranslateT("Failed to download XML data: "));
FormatMessage(FORMAT_MESSAGE_FROM_SYSTEM, 0, err, 0, buff + len, 512 - len, 0);
ShowError(buff);
//MessageBox(0, buff + len, TranslateT("Updater: Error Downloading XML Data"), MB_OK | MB_ICONWARNING);
}
else
{
ShowError(TranslateT("Failed to download XML data - Response is NULL"));
//MessageBox(0, TranslateT("Error downloading XML data...\nResponse is NULL"), TranslateT("Updater Error"), MB_OK | MB_ICONWARNING);
NLog("Failed to download XML data - Response is NULL");
}
}
return LoadOldXMLData(cat, false);
}
else if (resp->resultCode == 304) { // 'Not Modified' response
hNetlibHttp = resp->nlc;
CallService(MS_NETLIB_FREEHTTPREQUESTSTRUCT, 0, (LPARAM)resp);
resp = 0;
#ifdef DEBUG_HTTP_POPUPS
PUShowMessage("XML Data unchanged - using local copy", SM_NOTIFY);
#endif
// mark data as current
return LoadOldXMLData(cat, true);
} else if(resp->resultCode >= 300 && resp->resultCode < 400) { // redirect response
hNetlibHttp = resp->nlc;
// get new location
bool ret = false;
for(int i = 0; i < resp->headersCount; i++) {
//MessageBox(0, resp->headers[i].szValue, resp->headers[i].szName, MB_OK);
if(strcmp(resp->headers[i].szName, "Location") == 0) {
ret = UpdateXMLData(cat, resp->headers[i].szValue, recurse_count + 1);
break;
}
}
CallService(MS_NETLIB_FREEHTTPREQUESTSTRUCT, 0, (LPARAM)resp);
resp = 0;
if(!ret) return LoadOldXMLData(cat, false);
return ret;
} else if(resp->resultCode != 200) {
hNetlibHttp = resp->nlc;
TCHAR buff[512];
mir_sntprintf(buff, SIZEOF(buff), TranslateT("Failed to download XML data - Invalid response, code %d"), resp->resultCode);
ShowError(buff);
NLog(buff);
CallService(MS_NETLIB_FREEHTTPREQUESTSTRUCT, 0, (LPARAM)resp);
resp = 0;
return LoadOldXMLData(cat, false);
}
// resp->resultCode == 200
hNetlibHttp = resp->nlc;
if(!bz2_decompress_xml(resp->pData, resp->dataLength, &pData[cat], &dataLength[cat])) {
ShowError(TranslateT("Failed to decompress XML data"));
CallService(MS_NETLIB_FREEHTTPREQUESTSTRUCT, 0, (LPARAM)resp);
resp = 0;
return LoadOldXMLData(cat, false);
}
// store date header and data for response 'Not Modified' (304) above
for(int i = 0; i < resp->headersCount; i++) {
//MessageBox(0, resp->headers[i].szValue, resp->headers[i].szName, MB_OK);
if(strcmp(resp->headers[i].szName, "ETag") == 0) {
//MessageBox(0, resp->headers[i].szValue, "Storing ETag", MB_OK);
char buff[256];
strcpy(buff, "DataETag_");
strcat(buff, category_files[cat]);
DBWriteContactSettingString(0, MODULE, buff, resp->headers[i].szValue);
}
}
CallService(MS_NETLIB_FREEHTTPREQUESTSTRUCT, 0, (LPARAM)resp);
resp = 0;
SaveXMLData(cat);
doc[cat] = ezxml_parse_str((char*)pData[cat], strlen((char*)pData[cat]));
if (!doc[cat]) return false;
return true;
}
