From 9bf7f2bfeb3057ad8b49fc0361fee5036639dfe1 Mon Sep 17 00:00:00 2001 From: Egor Tensin Date: Mon, 11 May 2015 13:32:30 +0300 Subject: libservice: add virtual device wrappers --- utils/libservice/include/libservice/device.hpp | 71 ++++++++++++ utils/libservice/include/libservice/handle.hpp | 80 ++++++++++++++ utils/libservice/include/libservice/interface.hpp | 2 + utils/libservice/src/device.cpp | 126 ++++++++++++++++++++++ utils/libservice/src/handle.cpp | 29 +++++ 5 files changed, 308 insertions(+) create mode 100644 utils/libservice/include/libservice/device.hpp create mode 100644 utils/libservice/include/libservice/handle.hpp create mode 100644 utils/libservice/src/device.cpp create mode 100644 utils/libservice/src/handle.cpp (limited to 'utils') diff --git a/utils/libservice/include/libservice/device.hpp b/utils/libservice/include/libservice/device.hpp new file mode 100644 index 0000000..3a9540e --- /dev/null +++ b/utils/libservice/include/libservice/device.hpp @@ -0,0 +1,71 @@ +/** + * \file + * \author Egor Tensin + * \date 2015 + * \copyright This file is licensed under the terms of the MIT License. + * See LICENSE.txt for details. + */ + +#pragma once + +#include "common.hpp" +#include "handle.hpp" + +#include + +#include +#include + +namespace libservice +{ + class Device + { + public: + static Device open(const std::string& path); + + Device(Device&& other) LIBSERVICE_NOEXCEPT + { + swap(other); + } + + Device& operator=(Device other) LIBSERVICE_NOEXCEPT + { + swap(other); + return *this; + } + + void swap(Device& other) LIBSERVICE_NOEXCEPT + { + using std::swap; + swap(m_handle, other.m_handle); + } + + DWORD get_required_output_size(DWORD code, + const void* in_buf, + DWORD in_buf_size) const; + + DWORD send_control_code(DWORD code, + const void* in_buf, + DWORD in_buf_size, + void* out_buf, + DWORD out_buf_size) const; + + private: + explicit Device(Handle h) + : m_handle(std::move(h)) + { } + + Handle m_handle; + + Device(const Device&) = delete; + }; + + void swap(libservice::Device&, libservice::Device&) LIBSERVICE_NOEXCEPT; +} + +namespace std +{ + template <> + void swap( + libservice::Device&, libservice::Device&) LIBSERVICE_NOEXCEPT; +} diff --git a/utils/libservice/include/libservice/handle.hpp b/utils/libservice/include/libservice/handle.hpp new file mode 100644 index 0000000..1292090 --- /dev/null +++ b/utils/libservice/include/libservice/handle.hpp @@ -0,0 +1,80 @@ +/** + * \file + * \author Egor Tensin + * \date 2015 + * \copyright This file is licensed under the terms of the MIT License. + * See LICENSE.txt for details. + */ + +#pragma once + +#include "common.hpp" + +#include + +#include +#include +#include + +namespace libservice +{ + class Handle + { + public: + Handle() = default; + + explicit Handle(HANDLE raw) + : m_impl(raw) + { } + + Handle(Handle&& other) LIBSERVICE_NOEXCEPT + { + swap(other); + } + + Handle& operator=(Handle other) LIBSERVICE_NOEXCEPT + { + swap(other); + return *this; + } + + explicit operator bool() const + { + return static_cast(m_impl); + } + + explicit operator HANDLE() const + { + return m_impl.get(); + } + + void swap(Handle& other) LIBSERVICE_NOEXCEPT + { + using std::swap; + swap(m_impl, other.m_impl); + } + + private: + struct Deleter + { + void operator()(HANDLE raw) + { + ::CloseHandle(raw); + } + }; + + std::unique_ptr::type, Deleter> m_impl; + + Handle(const Handle&) = delete; + }; + + void swap(Handle& a, Handle& b) LIBSERVICE_NOEXCEPT; +} + +namespace std +{ + template <> + void swap( + libservice::Handle& a, + libservice::Handle& b) LIBSERVICE_NOEXCEPT; +} diff --git a/utils/libservice/include/libservice/interface.hpp b/utils/libservice/include/libservice/interface.hpp index 53751b2..e68439b 100644 --- a/utils/libservice/include/libservice/interface.hpp +++ b/utils/libservice/include/libservice/interface.hpp @@ -9,6 +9,8 @@ #pragma once #include "common.hpp" +#include "device.hpp" +#include "handle.hpp" #include "service.hpp" #include "service_handle.hpp" #include "service_manager.hpp" diff --git a/utils/libservice/src/device.cpp b/utils/libservice/src/device.cpp new file mode 100644 index 0000000..3cfb760 --- /dev/null +++ b/utils/libservice/src/device.cpp @@ -0,0 +1,126 @@ +/** + * \file + * \author Egor Tensin + * \date 2015 + * \copyright This file is licensed under the terms of the MIT License. + * See LICENSE.txt for details. + */ + +#include "libservice/common.hpp" +#include "libservice/device.hpp" +#include "libservice/handle.hpp" +#include "libservice/windows_error.hpp" + +#include + +#include +#include +#include + +namespace libservice +{ + namespace + { + Handle open_device(const std::string& path) + { + const auto raw = CreateFileA(path.c_str(), + GENERIC_READ | GENERIC_WRITE, + 0, + NULL, + OPEN_EXISTING, + FILE_ATTRIBUTE_NORMAL, + NULL); + + if (INVALID_HANDLE_VALUE == raw) + { + const auto ec = GetLastError(); + throw std::system_error(ec, WinErrorCategory::get(), LIBSERVICE_ERROR_PREFIX); + } + + return Handle(raw); + } + } + + Device Device::open(const std::string& path) + { + return Device(open_device(path)); + } + + DWORD Device::get_required_output_size( + DWORD code, + const void* in_buf, + DWORD in_buf_size) const + { + DWORD nbreq; + + DWORD ret = DeviceIoControl( + static_cast(m_handle), + code, + const_cast(in_buf), + in_buf_size, + NULL, + 0, + &nbreq, + NULL); + + if (0 == ret) + { + const auto ec = GetLastError(); + + switch (ec) + { + case ERROR_MORE_DATA: + return nbreq; + + default: + throw std::system_error(ec, WinErrorCategory::get(), LIBSERVICE_ERROR_PREFIX); + } + } + + return ret; + } + + DWORD Device::send_control_code( + DWORD code, + const void* in_buf, + DWORD in_buf_size, + void* out_buf, + DWORD out_buf_size) const + { + DWORD nbreq; + + DWORD ret = DeviceIoControl( + static_cast(m_handle), + code, + const_cast(in_buf), + in_buf_size, + out_buf, + out_buf_size, + &nbreq, + NULL); + + if (0 == ret) + { + const auto ec = GetLastError(); + throw std::system_error(ec, WinErrorCategory::get(), LIBSERVICE_ERROR_PREFIX); + } + + return ret; + } + + void swap(Device& a, Device& b) LIBSERVICE_NOEXCEPT + { + a.swap(b); + } +} + +namespace std +{ + template <> + void swap( + libservice::Device& a, + libservice::Device& b) + { + a.swap(b); + } +} diff --git a/utils/libservice/src/handle.cpp b/utils/libservice/src/handle.cpp new file mode 100644 index 0000000..2f9a65f --- /dev/null +++ b/utils/libservice/src/handle.cpp @@ -0,0 +1,29 @@ +/** + * \file + * \author Egor Tensin + * \date 2015 + * \copyright This file is licensed under the terms of the MIT License. + * See LICENSE.txt for details. + */ + +#include "libservice/common.hpp" +#include "libservice/handle.hpp" + +namespace libservice +{ + void swap(Handle& a, Handle& b) LIBSERVICE_NOEXCEPT + { + a.swap(b); + } +} + +namespace std +{ + template <> + void swap( + libservice::Handle& a, + libservice::Handle& b) LIBSERVICE_NOEXCEPT + { + a.swap(b); + } +} -- cgit v1.2.3