1 files changed, 93 insertions, 0 deletions
diff --git a/ceee/common/com_utils.h b/ceee/common/com_utils.h
new file mode 100644
index 0000000..e531ba4
--- /dev/null
+++ b/ceee/common/com_utils.h
@@ -0,0 +1,93 @@
+// Copyright (c) 2010 The Chromium Authors. All rights reserved.
+// Use of this source code is governed by a BSD-style license that can be
+// found in the LICENSE file.
+//
+// Utilities for COM objects, error codes etc.
+
+#ifndef CEEE_COMMON_COM_UTILS_H_
+#define CEEE_COMMON_COM_UTILS_H_
+
+#include <windows.h>
+#include <wtypes.h>
+
+#include <ostream>
+
+namespace com {
+
+// Returns the provided hr if it is an error, otherwise default_error.
+inline HRESULT AlwaysError(HRESULT hr, HRESULT default_error) {
+  if (FAILED(hr)) {
+    return hr;
+  } else {
+    return default_error;
+  }
+}
+
+// Returns the provided hr if it is an error, otherwise E_FAIL.
+inline HRESULT AlwaysError(HRESULT hr) {
+  return AlwaysError(hr, E_FAIL);
+}
+
+// Converts a Win32 result code to a HRESULT. If the 'win32_code'
+// does not indicate an error, it returns 'default_error'.
+inline HRESULT AlwaysErrorFromWin32(DWORD win32_code,
+                                    HRESULT default_error) {
+  HRESULT hr = HRESULT_FROM_WIN32(win32_code);
+  return AlwaysError(hr, default_error);
+}
+
+// Converts a Win32 result code to a HRESULT. If the 'win32_code'
+// does not indicate an error, it returns E_FAIL
+inline HRESULT AlwaysErrorFromWin32(DWORD win32_code) {
+  return AlwaysErrorFromWin32(win32_code, E_FAIL);
+}
+
+// Returns the HRESULT equivalent of GetLastError(), unless it does not
+// represent an error in which case it returns 'default_error'.
+inline HRESULT AlwaysErrorFromLastError(HRESULT default_error) {
+  return AlwaysErrorFromWin32(::GetLastError(), default_error);
+}
+
+// Returns the HRESULT equivalent of GetLastError(), unless it does not
+// represent an error in which case it returns E_FAIL.
+inline HRESULT AlwaysErrorFromLastError() {
+  return AlwaysErrorFromLastError(E_FAIL);
+}
+
+// Return the specified string, or the empty string if it is NULL.
+inline const wchar_t* ToString(BSTR str) {
+  return str ? str : L"";
+}
+
+// Logs HRESULTs verbosely, with the error code and human-readable error
+// text if available.
+class LogHr {
+ public:
+  explicit LogHr(HRESULT hr) : hr_(hr) {}
+ private:
+  HRESULT hr_;
+  friend std::ostream& operator<<(std::ostream&, const LogHr&);
+};
+
+std::ostream& operator<<(std::ostream& os, const LogHr& hr);
+
+// Logs Windows errors verbosely, with the error code and human-readable error
+// text if available.
+class LogWe {
+ public:
+  LogWe() : we_(::GetLastError()) {}
+  explicit LogWe(DWORD we) : we_(we) {}
+ private:
+  DWORD we_;
+  friend std::ostream& operator<<(std::ostream&, const LogWe&);
+};
+
+std::ostream& operator<<(std::ostream& os, const LogWe& we);
+
+// Use to register a .reg file that needs %MODULE_PATH% and %MODULE_BASENAME%
+// replaceables, but no %APPID%.
+HRESULT ModuleRegistrationWithoutAppid(int reg_id, BOOL should_register);
+
+}  // namespace com
+
+#endif  // CEEE_COMMON_COM_UTILS_H_