| // Copyright 2014 Google Inc. All Rights Reserved. |
| // |
| // Licensed under the Apache License, Version 2.0 (the "License"); |
| // you may not use this file except in compliance with the License. |
| // You may obtain a copy of the License at |
| // |
| // http://www.apache.org/licenses/LICENSE-2.0 |
| // |
| // Unless required by applicable law or agreed to in writing, software |
| // distributed under the License is distributed on an "AS IS" BASIS, |
| // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| // See the License for the specific language governing permissions and |
| // limitations under the License. |
| // |
| // Utilities for COM objects, error codes etc. |
| |
| #ifndef SYZYGY_COMMON_COM_UTILS_H_ |
| #define SYZYGY_COMMON_COM_UTILS_H_ |
| |
| #include <windows.h> |
| #include <wtypes.h> |
| #include <ostream> |
| |
| namespace common { |
| |
| // 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); |
| |
| } // namespace common |
| |
| #endif // SYZYGY_COMMON_COM_UTILS_H_ |