summaryrefslogtreecommitdiffstats
path: root/browser/components/migration/nsEdgeMigrationUtils.cpp
blob: a8d76c140581170d98dec858f7640f1d29735283 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
/* 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/. */

#include "nsEdgeMigrationUtils.h"

#include "mozilla/dom/Promise.h"
#include "nsCOMPtr.h"
#include "nsIEventTarget.h"

#include <windows.h>

namespace mozilla {

NS_IMPL_ISUPPORTS(nsEdgeMigrationUtils, nsIEdgeMigrationUtils)

NS_IMETHODIMP
nsEdgeMigrationUtils::IsDbLocked(nsIFile* aFile, JSContext* aCx,
                                 dom::Promise** aPromise) {
  NS_ENSURE_ARG_POINTER(aFile);

  nsString path;
  nsresult rv = aFile->GetPath(path);
  NS_ENSURE_SUCCESS(rv, rv);

  ErrorResult err;
  RefPtr<dom::Promise> promise =
      dom::Promise::Create(xpc::CurrentNativeGlobal(aCx), err);

  if (MOZ_UNLIKELY(err.Failed())) {
    return err.StealNSResult();
  }

  nsMainThreadPtrHandle<dom::Promise> promiseHolder(
      new nsMainThreadPtrHolder<dom::Promise>("nsEdgeMigrationUtils Promise",
                                              promise));

  NS_DispatchBackgroundTask(NS_NewRunnableFunction(
      __func__,
      [promiseHolder = std::move(promiseHolder), path = std::move(path)]() {
        HANDLE file = ::CreateFileW(path.get(), GENERIC_READ, FILE_SHARE_READ,
                                    nullptr, OPEN_EXISTING, 0, nullptr);

        bool locked = true;
        if (file != INVALID_HANDLE_VALUE) {
          locked = false;
          ::CloseHandle(file);
        }

        NS_DispatchToMainThread(NS_NewRunnableFunction(
            __func__, [promiseHolder = std::move(promiseHolder), locked]() {
              promiseHolder.get()->MaybeResolve(locked);
            }));
      }));

  promise.forget(aPromise);

  return NS_OK;
}

}  // namespace mozilla