mirror of
https://github.com/mozilla/gecko-dev.git
synced 2024-10-10 11:55:49 +00:00
f9a82146b7
Since the ApplyUpdate and SwitchToUpdatedApp functions are almost entirely the same this moves everything into ApplyUpdate. Removes most of the gonk code from application update. Makes client code and xpcshell tests use the same code for directory providers in nsUpdateDriver.cpp.
103 lines
2.9 KiB
C++
103 lines
2.9 KiB
C++
/* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
|
|
/* vim:set ts=2 sw=2 sts=2 et cindent: */
|
|
/* This Source Code Form is subject to the terms of the Mozilla Public
|
|
* License, v. 2.0. If a copy of the MPL was not distributed with this
|
|
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */
|
|
|
|
#ifndef nsUpdateDriver_h__
|
|
#define nsUpdateDriver_h__
|
|
|
|
#include "nscore.h"
|
|
#include "nsIUpdateService.h"
|
|
#include "nsIThread.h"
|
|
#include "nsCOMPtr.h"
|
|
#include "nsString.h"
|
|
#include "mozilla/Attributes.h"
|
|
|
|
class nsIFile;
|
|
|
|
#if defined(XP_WIN)
|
|
#include <windows.h>
|
|
typedef HANDLE ProcessType;
|
|
#elif defined(XP_UNIX)
|
|
typedef pid_t ProcessType;
|
|
#else
|
|
#include "prproces.h"
|
|
typedef PRProcess* ProcessType;
|
|
#endif
|
|
|
|
/**
|
|
* This function processes any available updates. As part of that process, it
|
|
* may exit the current process and relaunch it at a later time.
|
|
*
|
|
* Two directories are passed to this function: greDir (where the actual
|
|
* binary resides) and appDir (which contains application.ini for XULRunner
|
|
* apps). If this is not a XULRunner app then appDir is identical to greDir.
|
|
*
|
|
* The argc and argv passed to this function should be what is needed to
|
|
* relaunch the current process.
|
|
*
|
|
* The appVersion param passed to this function is the current application's
|
|
* version and is used to determine if an update's version is older than the
|
|
* current application version.
|
|
*
|
|
* If you want the update to be processed without restarting, set the restart
|
|
* parameter to false.
|
|
*
|
|
* This function does not modify appDir.
|
|
*/
|
|
nsresult ProcessUpdates(nsIFile *greDir, nsIFile *appDir, nsIFile *updRootDir,
|
|
int argc, char **argv, const char *appVersion,
|
|
bool restart = true, ProcessType *pid = nullptr);
|
|
|
|
// The implementation of the update processor handles the task of loading the
|
|
// updater application for staging an update.
|
|
// XXX ehsan this is living in this file in order to make use of the existing
|
|
// stuff here, we might want to move it elsewhere in the future.
|
|
class nsUpdateProcessor final : public nsIUpdateProcessor
|
|
{
|
|
public:
|
|
nsUpdateProcessor();
|
|
|
|
NS_DECL_THREADSAFE_ISUPPORTS
|
|
NS_DECL_NSIUPDATEPROCESSOR
|
|
|
|
private:
|
|
~nsUpdateProcessor();
|
|
|
|
struct StagedUpdateInfo {
|
|
StagedUpdateInfo()
|
|
: mArgc(0),
|
|
mArgv(nullptr),
|
|
mIsOSUpdate(false)
|
|
{}
|
|
~StagedUpdateInfo() {
|
|
for (int i = 0; i < mArgc; ++i) {
|
|
delete[] mArgv[i];
|
|
}
|
|
delete[] mArgv;
|
|
}
|
|
|
|
nsCOMPtr<nsIFile> mGREDir;
|
|
nsCOMPtr<nsIFile> mAppDir;
|
|
nsCOMPtr<nsIFile> mUpdateRoot;
|
|
nsCOMPtr<nsIFile> mOSApplyToDir;
|
|
int mArgc;
|
|
char **mArgv;
|
|
nsCString mAppVersion;
|
|
bool mIsOSUpdate;
|
|
};
|
|
|
|
private:
|
|
void StartStagedUpdate();
|
|
void WaitForProcess();
|
|
void UpdateDone();
|
|
void ShutdownWatcherThread();
|
|
|
|
private:
|
|
ProcessType mUpdaterPID;
|
|
nsCOMPtr<nsIThread> mProcessWatcher;
|
|
StagedUpdateInfo mInfo;
|
|
};
|
|
#endif // nsUpdateDriver_h__
|