9de27f16bc
* Publishing R5 content * Updated ade revision * updated readme * add possibility to build CPU plugin with Intel MKL package
73 lines
2.0 KiB
C++
73 lines
2.0 KiB
C++
// Copyright (C) 2018 Intel Corporation
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
//
|
|
|
|
/**
|
|
* @brief WINAPI compatible loader for a shared object
|
|
* @file win_shared_object_loader.h
|
|
*/
|
|
#pragma once
|
|
|
|
#include "../../ie_api.h"
|
|
#include "../ie_exception.hpp"
|
|
|
|
// Avoidance of Windows.h to include winsock library.
|
|
#define _WINSOCKAPI_
|
|
// Avoidance of Windows.h to define min/max.
|
|
#ifndef NOMINMAX
|
|
#define NOMINMAX
|
|
#endif
|
|
#include <windows.h>
|
|
#include <direct.h>
|
|
|
|
namespace InferenceEngine {
|
|
namespace details {
|
|
|
|
/**
|
|
* @brief This class provides an OS shared module abstraction
|
|
*/
|
|
class SharedObjectLoader {
|
|
private:
|
|
HMODULE shared_object;
|
|
|
|
public:
|
|
/**
|
|
* @brief Loads a library with the name specified. The library is loaded according to the
|
|
* WinAPI LoadLibrary rules
|
|
* @param pluginName Full or relative path to the plugin library
|
|
*/
|
|
explicit SharedObjectLoader(LPCTSTR pluginName) {
|
|
char cwd[1024];
|
|
shared_object = LoadLibrary(pluginName);
|
|
if (!shared_object) {
|
|
THROW_IE_EXCEPTION << "Cannot load library '"
|
|
<< pluginName << "': "
|
|
<< GetLastError()
|
|
<< " from cwd: " << _getcwd(cwd, 1024);
|
|
}
|
|
}
|
|
~SharedObjectLoader() {
|
|
FreeLibrary(shared_object);
|
|
}
|
|
|
|
/**
|
|
* @brief Searches for a function symbol in the loaded module
|
|
* @param symbolName Name of function to find
|
|
* @return A pointer to the function if found
|
|
* @throws InferenceEngineException if the function is not found
|
|
*/
|
|
void *get_symbol(const char* symbolName) const {
|
|
if (!shared_object) {
|
|
THROW_IE_EXCEPTION << "Cannot get '" << symbolName << "' content from unknown library!";
|
|
}
|
|
auto procAddr = reinterpret_cast<void*>(GetProcAddress(shared_object, symbolName));
|
|
if (procAddr == nullptr)
|
|
THROW_IE_EXCEPTION << "GetProcAddress cannot locate method '" << symbolName << "': " << GetLastError();
|
|
|
|
return procAddr;
|
|
}
|
|
};
|
|
|
|
} // namespace details
|
|
} // namespace InferenceEngine
|