forked from mirrors/gecko-dev
718f8615db
This patch adds application info (Name and Publisher for now) in the about:third-party page if a module is a part of an installed application, which is registered in the registry and shown in Windows Control Panel. To achieve this, we parse the registry to collect installed applications in the background task. Differential Revision: https://phabricator.services.mozilla.com/D109306
94 lines
2.5 KiB
C++
94 lines
2.5 KiB
C++
/* -*- Mode: C++; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
|
|
/* vim: set ts=8 sts=2 et sw=2 tw=80: */
|
|
/* 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 __MsiDatabase_h__
|
|
#define __MsiDatabase_h__
|
|
|
|
#include "mozilla/Maybe.h"
|
|
#include "mozilla/UniquePtr.h"
|
|
#include "nsWindowsHelpers.h"
|
|
|
|
#include <windows.h>
|
|
#include <msi.h>
|
|
#include <msiquery.h>
|
|
|
|
namespace mozilla {
|
|
|
|
class MsiDatabase final {
|
|
static UniquePtr<wchar_t[]> GetRecordString(MSIHANDLE aRecord,
|
|
UINT aFieldIndex);
|
|
|
|
nsAutoMsiHandle mDatabase;
|
|
|
|
MsiDatabase() = default;
|
|
explicit MsiDatabase(const wchar_t* aDatabasePath);
|
|
|
|
public:
|
|
// A callback function passed to ExecuteSingleColumnQuery uses this type
|
|
// to control the enumeration loop.
|
|
enum class CallbackResult { Continue, Stop };
|
|
|
|
static Maybe<MsiDatabase> FromProductId(const wchar_t* aProductId);
|
|
|
|
MsiDatabase(const MsiDatabase&) = delete;
|
|
MsiDatabase& operator=(const MsiDatabase&) = delete;
|
|
MsiDatabase(MsiDatabase&& aOther) : mDatabase(aOther.mDatabase.disown()) {}
|
|
MsiDatabase& operator=(MsiDatabase&& aOther) {
|
|
if (this != &aOther) {
|
|
mDatabase.own(aOther.mDatabase.disown());
|
|
}
|
|
return *this;
|
|
}
|
|
|
|
explicit operator bool() const;
|
|
|
|
template <typename CallbackT>
|
|
bool ExecuteSingleColumnQuery(const wchar_t* aQuery,
|
|
const CallbackT& aCallback) const {
|
|
MSIHANDLE handle;
|
|
UINT ret = ::MsiDatabaseOpenViewW(mDatabase, aQuery, &handle);
|
|
if (ret != ERROR_SUCCESS) {
|
|
return false;
|
|
}
|
|
|
|
nsAutoMsiHandle view(handle);
|
|
|
|
ret = ::MsiViewExecute(view, 0);
|
|
if (ret != ERROR_SUCCESS) {
|
|
return false;
|
|
}
|
|
|
|
for (;;) {
|
|
ret = ::MsiViewFetch(view, &handle);
|
|
if (ret == ERROR_NO_MORE_ITEMS) {
|
|
break;
|
|
} else if (ret != ERROR_SUCCESS) {
|
|
return false;
|
|
}
|
|
|
|
nsAutoMsiHandle record(handle);
|
|
UniquePtr<wchar_t[]> guidStr = GetRecordString(record, 1);
|
|
if (!guidStr) {
|
|
continue;
|
|
}
|
|
|
|
CallbackResult result = aCallback(guidStr.get());
|
|
if (result == CallbackResult::Continue) {
|
|
continue;
|
|
} else if (result == CallbackResult::Stop) {
|
|
break;
|
|
} else {
|
|
MOZ_ASSERT_UNREACHABLE("Unexpected CallbackResult.");
|
|
}
|
|
}
|
|
|
|
return true;
|
|
}
|
|
};
|
|
|
|
} // namespace mozilla
|
|
|
|
#endif // __MsiDatabase_h__
|