2012-11-18 07:03:04 +04:00
|
|
|
/**
|
|
|
|
* FreeRDP: A Remote Desktop Protocol Implementation
|
|
|
|
* Addin Loader
|
|
|
|
*
|
|
|
|
* Copyright 2012 Marc-Andre Moreau <marcandre.moreau@gmail.com>
|
|
|
|
*
|
|
|
|
* 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.
|
|
|
|
*/
|
|
|
|
|
|
|
|
#ifdef HAVE_CONFIG_H
|
|
|
|
#include "config.h"
|
|
|
|
#endif
|
|
|
|
|
|
|
|
#include <stdio.h>
|
|
|
|
#include <stdlib.h>
|
|
|
|
#include <string.h>
|
|
|
|
|
|
|
|
#include <winpr/crt.h>
|
|
|
|
#include <winpr/path.h>
|
|
|
|
#include <winpr/library.h>
|
|
|
|
|
|
|
|
#include <freerdp/addin.h>
|
2015-11-05 16:02:07 +03:00
|
|
|
#include <freerdp/build-config.h>
|
2012-11-18 07:03:04 +04:00
|
|
|
|
2016-10-27 15:41:05 +03:00
|
|
|
#include <freerdp/log.h>
|
|
|
|
#define TAG FREERDP_TAG("addin")
|
2012-11-18 07:03:04 +04:00
|
|
|
|
2016-11-10 12:18:55 +03:00
|
|
|
static INLINE BOOL is_path_required(LPCSTR path, size_t len)
|
|
|
|
{
|
|
|
|
if (!path || (len <= 1))
|
|
|
|
return FALSE;
|
2017-04-10 11:38:38 +03:00
|
|
|
|
2016-11-10 12:18:55 +03:00
|
|
|
if (strcmp(path, ".") == 0)
|
|
|
|
return FALSE;
|
2017-04-10 11:38:38 +03:00
|
|
|
|
2016-11-10 12:18:55 +03:00
|
|
|
return TRUE;
|
|
|
|
}
|
|
|
|
|
2016-10-07 15:06:46 +03:00
|
|
|
LPSTR freerdp_get_library_install_path(void)
|
2012-11-18 07:03:04 +04:00
|
|
|
{
|
|
|
|
LPSTR pszPath;
|
|
|
|
size_t cchPath;
|
|
|
|
size_t cchLibraryPath;
|
|
|
|
size_t cchInstallPrefix;
|
2016-11-10 12:18:55 +03:00
|
|
|
BOOL needLibPath, needInstallPath;
|
2012-12-20 16:09:33 +04:00
|
|
|
LPCSTR pszLibraryPath = FREERDP_LIBRARY_PATH;
|
|
|
|
LPCSTR pszInstallPrefix = FREERDP_INSTALL_PREFIX;
|
2016-11-10 12:18:55 +03:00
|
|
|
cchLibraryPath = strlen(pszLibraryPath) + 1;
|
|
|
|
cchInstallPrefix = strlen(pszInstallPrefix) + 1;
|
|
|
|
cchPath = cchInstallPrefix + cchLibraryPath;
|
|
|
|
needInstallPath = is_path_required(pszInstallPrefix, cchInstallPrefix);
|
|
|
|
needLibPath = is_path_required(pszLibraryPath, cchLibraryPath);
|
|
|
|
|
|
|
|
if (!needInstallPath && !needLibPath)
|
|
|
|
return NULL;
|
|
|
|
|
2012-11-18 07:03:04 +04:00
|
|
|
pszPath = (LPSTR) malloc(cchPath + 1);
|
2016-10-07 15:06:46 +03:00
|
|
|
|
2015-06-16 16:42:07 +03:00
|
|
|
if (!pszPath)
|
|
|
|
return NULL;
|
2012-11-18 07:03:04 +04:00
|
|
|
|
2016-11-10 12:18:55 +03:00
|
|
|
if (needInstallPath)
|
|
|
|
{
|
|
|
|
CopyMemory(pszPath, pszInstallPrefix, cchInstallPrefix);
|
|
|
|
pszPath[cchInstallPrefix] = '\0';
|
|
|
|
}
|
2012-11-18 07:03:04 +04:00
|
|
|
|
2016-11-10 12:18:55 +03:00
|
|
|
if (needLibPath)
|
2015-06-16 16:42:07 +03:00
|
|
|
{
|
2016-11-10 12:18:55 +03:00
|
|
|
if (FAILED(NativePathCchAppendA(pszPath, cchPath + 1, pszLibraryPath)))
|
|
|
|
{
|
|
|
|
free(pszPath);
|
|
|
|
return NULL;
|
|
|
|
}
|
2015-06-16 16:42:07 +03:00
|
|
|
}
|
2012-11-18 07:03:04 +04:00
|
|
|
|
|
|
|
return pszPath;
|
|
|
|
}
|
|
|
|
|
2016-10-07 15:06:46 +03:00
|
|
|
LPSTR freerdp_get_dynamic_addin_install_path(void)
|
2012-11-18 07:03:04 +04:00
|
|
|
{
|
|
|
|
LPSTR pszPath;
|
|
|
|
size_t cchPath;
|
|
|
|
size_t cchAddinPath;
|
|
|
|
size_t cchInstallPrefix;
|
2016-11-10 12:18:55 +03:00
|
|
|
BOOL needLibPath, needInstallPath;
|
2012-12-20 16:09:33 +04:00
|
|
|
LPCSTR pszAddinPath = FREERDP_ADDIN_PATH;
|
|
|
|
LPCSTR pszInstallPrefix = FREERDP_INSTALL_PREFIX;
|
2016-11-10 12:18:55 +03:00
|
|
|
cchAddinPath = strlen(pszAddinPath) + 1;
|
|
|
|
cchInstallPrefix = strlen(pszInstallPrefix) + 1;
|
|
|
|
cchPath = cchInstallPrefix + cchAddinPath;
|
|
|
|
needInstallPath = is_path_required(pszInstallPrefix, cchInstallPrefix);
|
|
|
|
needLibPath = is_path_required(pszAddinPath, cchAddinPath);
|
|
|
|
|
|
|
|
if (!needInstallPath && !needLibPath)
|
|
|
|
return NULL;
|
|
|
|
|
|
|
|
pszPath = (LPSTR) calloc(cchPath + 1, sizeof(CHAR));
|
2016-10-07 15:06:46 +03:00
|
|
|
|
2015-06-17 23:08:02 +03:00
|
|
|
if (!pszPath)
|
|
|
|
return NULL;
|
2012-11-18 07:03:04 +04:00
|
|
|
|
2016-11-10 12:18:55 +03:00
|
|
|
if (needInstallPath)
|
|
|
|
{
|
|
|
|
CopyMemory(pszPath, pszInstallPrefix, cchInstallPrefix);
|
|
|
|
pszPath[cchInstallPrefix] = '\0';
|
|
|
|
}
|
2012-11-18 07:03:04 +04:00
|
|
|
|
2016-11-10 12:18:55 +03:00
|
|
|
if (needLibPath)
|
2015-06-16 16:42:07 +03:00
|
|
|
{
|
2016-11-10 12:18:55 +03:00
|
|
|
if (FAILED(NativePathCchAppendA(pszPath, cchPath + 1, pszAddinPath)))
|
|
|
|
{
|
|
|
|
free(pszPath);
|
|
|
|
return NULL;
|
|
|
|
}
|
2015-06-16 16:42:07 +03:00
|
|
|
}
|
2012-11-18 07:03:04 +04:00
|
|
|
|
|
|
|
return pszPath;
|
|
|
|
}
|
|
|
|
|
2016-10-07 15:06:46 +03:00
|
|
|
PVIRTUALCHANNELENTRY freerdp_load_dynamic_addin(LPCSTR pszFileName,
|
|
|
|
LPCSTR pszPath, LPCSTR pszEntryName)
|
2012-11-18 07:03:04 +04:00
|
|
|
{
|
2016-11-10 12:18:55 +03:00
|
|
|
LPSTR pszAddinInstallPath = freerdp_get_dynamic_addin_install_path();
|
|
|
|
PVIRTUALCHANNELENTRY entry = NULL;
|
|
|
|
BOOL bHasExt = TRUE;
|
2012-11-18 07:03:04 +04:00
|
|
|
PCSTR pszExt;
|
2016-11-10 12:18:55 +03:00
|
|
|
size_t cchExt = 0;
|
|
|
|
HINSTANCE library = NULL;
|
2012-11-18 07:03:04 +04:00
|
|
|
size_t cchFileName;
|
|
|
|
size_t cchFilePath;
|
2016-12-01 13:50:01 +03:00
|
|
|
LPSTR pszAddinFile = NULL;
|
2016-11-10 12:18:55 +03:00
|
|
|
LPSTR pszFilePath = NULL;
|
|
|
|
LPSTR pszRelativeFilePath = NULL;
|
2012-11-18 07:03:04 +04:00
|
|
|
size_t cchAddinFile;
|
|
|
|
size_t cchAddinInstallPath;
|
2016-11-10 12:18:55 +03:00
|
|
|
|
|
|
|
if (!pszFileName || !pszEntryName)
|
|
|
|
goto fail;
|
|
|
|
|
2012-11-18 07:03:04 +04:00
|
|
|
cchFileName = strlen(pszFileName);
|
|
|
|
|
2016-11-10 12:18:55 +03:00
|
|
|
/* Get file name with prefix and extension */
|
2015-06-22 19:52:13 +03:00
|
|
|
if (FAILED(PathCchFindExtensionA(pszFileName, cchFileName + 1, &pszExt)))
|
2012-11-18 07:03:04 +04:00
|
|
|
{
|
|
|
|
pszExt = PathGetSharedLibraryExtensionA(PATH_SHARED_LIB_EXT_WITH_DOT);
|
|
|
|
cchExt = strlen(pszExt);
|
|
|
|
bHasExt = FALSE;
|
|
|
|
}
|
|
|
|
|
|
|
|
if (bHasExt)
|
|
|
|
{
|
|
|
|
pszAddinFile = _strdup(pszFileName);
|
2016-10-07 15:06:46 +03:00
|
|
|
|
2015-06-17 23:08:02 +03:00
|
|
|
if (!pszAddinFile)
|
2016-11-10 12:18:55 +03:00
|
|
|
goto fail;
|
2012-11-18 07:03:04 +04:00
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
2015-11-05 16:02:07 +03:00
|
|
|
cchAddinFile = cchFileName + cchExt + 2 + sizeof(FREERDP_SHARED_LIBRARY_PREFIX);
|
2012-11-18 07:03:04 +04:00
|
|
|
pszAddinFile = (LPSTR) malloc(cchAddinFile + 1);
|
2016-10-07 15:06:46 +03:00
|
|
|
|
2015-06-17 23:08:02 +03:00
|
|
|
if (!pszAddinFile)
|
2016-11-10 12:18:55 +03:00
|
|
|
goto fail;
|
2016-10-07 15:06:46 +03:00
|
|
|
|
|
|
|
sprintf_s(pszAddinFile, cchAddinFile, FREERDP_SHARED_LIBRARY_PREFIX"%s%s",
|
|
|
|
pszFileName, pszExt);
|
2012-11-18 07:03:04 +04:00
|
|
|
}
|
2017-04-10 11:38:38 +03:00
|
|
|
|
2016-11-10 12:18:55 +03:00
|
|
|
cchAddinFile = strlen(pszAddinFile);
|
|
|
|
|
|
|
|
/* If a path is provided prefix the library name with it. */
|
|
|
|
if (pszPath)
|
|
|
|
{
|
|
|
|
size_t relPathLen = strlen(pszPath) + cchAddinFile + 1;
|
|
|
|
pszRelativeFilePath = calloc(relPathLen, sizeof(CHAR));
|
2017-04-10 11:38:38 +03:00
|
|
|
|
2016-11-10 12:18:55 +03:00
|
|
|
if (!pszRelativeFilePath)
|
|
|
|
goto fail;
|
2017-04-10 11:38:38 +03:00
|
|
|
|
2016-11-10 12:18:55 +03:00
|
|
|
sprintf_s(pszRelativeFilePath, relPathLen, "%s", pszRelativeFilePath);
|
|
|
|
NativePathCchAppendA(pszRelativeFilePath, relPathLen, pszAddinFile);
|
|
|
|
}
|
|
|
|
else
|
|
|
|
pszRelativeFilePath = _strdup(pszAddinFile);
|
|
|
|
|
|
|
|
if (!pszRelativeFilePath)
|
|
|
|
goto fail;
|
|
|
|
|
|
|
|
/* If a system prefix path is provided try these locations too. */
|
|
|
|
if (pszAddinInstallPath)
|
|
|
|
{
|
|
|
|
cchAddinInstallPath = strlen(pszAddinInstallPath);
|
|
|
|
cchFilePath = cchAddinInstallPath + cchFileName + 32;
|
|
|
|
pszFilePath = (LPSTR) malloc(cchFilePath + 1);
|
|
|
|
|
|
|
|
if (!pszFilePath)
|
|
|
|
goto fail;
|
|
|
|
|
|
|
|
CopyMemory(pszFilePath, pszAddinInstallPath, cchAddinInstallPath);
|
|
|
|
pszFilePath[cchAddinInstallPath] = '\0';
|
|
|
|
NativePathCchAppendA((LPSTR) pszFilePath, cchFilePath + 1, pszRelativeFilePath);
|
|
|
|
}
|
|
|
|
else
|
|
|
|
pszFilePath = _strdup(pszRelativeFilePath);
|
2012-11-18 07:03:04 +04:00
|
|
|
|
|
|
|
library = LoadLibraryA(pszFilePath);
|
2012-11-19 02:32:18 +04:00
|
|
|
|
2012-11-18 07:03:04 +04:00
|
|
|
if (!library)
|
2016-11-10 12:18:55 +03:00
|
|
|
goto fail;
|
2012-11-18 07:03:04 +04:00
|
|
|
|
2016-10-07 15:06:46 +03:00
|
|
|
entry = (PVIRTUALCHANNELENTRY)GetProcAddress(library, pszEntryName);
|
2016-11-10 12:18:55 +03:00
|
|
|
fail:
|
|
|
|
free(pszRelativeFilePath);
|
|
|
|
free(pszAddinFile);
|
|
|
|
free(pszFilePath);
|
|
|
|
free(pszAddinInstallPath);
|
2017-04-10 11:38:38 +03:00
|
|
|
|
2016-11-10 12:18:55 +03:00
|
|
|
if (!entry && library)
|
|
|
|
FreeLibrary(library);
|
2017-04-10 11:38:38 +03:00
|
|
|
|
2012-11-18 07:03:04 +04:00
|
|
|
return entry;
|
|
|
|
}
|
|
|
|
|
2016-10-07 15:06:46 +03:00
|
|
|
PVIRTUALCHANNELENTRY freerdp_load_dynamic_channel_addin_entry(LPCSTR pszName,
|
|
|
|
LPSTR pszSubsystem, LPSTR pszType, DWORD dwFlags)
|
2012-11-18 07:03:04 +04:00
|
|
|
{
|
2016-10-07 15:06:46 +03:00
|
|
|
PVIRTUALCHANNELENTRY entry;
|
2012-11-18 07:03:04 +04:00
|
|
|
LPSTR pszFileName;
|
2015-11-05 16:02:07 +03:00
|
|
|
size_t cchFileName = sizeof(FREERDP_SHARED_LIBRARY_PREFIX) + 32;
|
2012-11-18 07:03:04 +04:00
|
|
|
LPCSTR pszExtension;
|
2015-11-05 16:02:07 +03:00
|
|
|
LPCSTR pszPrefix = FREERDP_SHARED_LIBRARY_PREFIX;
|
2012-11-18 07:03:04 +04:00
|
|
|
pszExtension = PathGetSharedLibraryExtensionA(0);
|
|
|
|
|
|
|
|
if (pszName && pszSubsystem && pszType)
|
|
|
|
{
|
2016-10-07 15:06:46 +03:00
|
|
|
cchFileName += strlen(pszName) + strlen(pszSubsystem) + strlen(
|
|
|
|
pszType) + strlen(pszExtension);
|
2012-11-18 07:03:04 +04:00
|
|
|
pszFileName = (LPSTR) malloc(cchFileName);
|
2016-10-07 15:06:46 +03:00
|
|
|
|
2015-06-16 16:42:07 +03:00
|
|
|
if (!pszFileName)
|
|
|
|
return NULL;
|
2016-10-07 15:06:46 +03:00
|
|
|
|
|
|
|
sprintf_s(pszFileName, cchFileName, "%s%s-client-%s-%s.%s", pszPrefix, pszName,
|
|
|
|
pszSubsystem, pszType, pszExtension);
|
2012-11-18 07:03:04 +04:00
|
|
|
cchFileName = strlen(pszFileName);
|
|
|
|
}
|
|
|
|
else if (pszName && pszSubsystem)
|
|
|
|
{
|
2014-09-17 13:44:15 +04:00
|
|
|
cchFileName += strlen(pszName) + strlen(pszSubsystem) + strlen(pszExtension);
|
2012-11-18 07:03:04 +04:00
|
|
|
pszFileName = (LPSTR) malloc(cchFileName);
|
2016-10-07 15:06:46 +03:00
|
|
|
|
2015-06-16 16:42:07 +03:00
|
|
|
if (!pszFileName)
|
|
|
|
return NULL;
|
|
|
|
|
2016-10-07 15:06:46 +03:00
|
|
|
sprintf_s(pszFileName, cchFileName, "%s%s-client-%s.%s", pszPrefix, pszName,
|
|
|
|
pszSubsystem, pszExtension);
|
2012-11-18 07:03:04 +04:00
|
|
|
cchFileName = strlen(pszFileName);
|
|
|
|
}
|
|
|
|
else if (pszName)
|
|
|
|
{
|
2014-09-17 13:44:15 +04:00
|
|
|
cchFileName += strlen(pszName) + strlen(pszExtension);
|
2012-11-18 07:03:04 +04:00
|
|
|
pszFileName = (LPSTR) malloc(cchFileName);
|
2016-10-07 15:06:46 +03:00
|
|
|
|
2015-06-16 16:42:07 +03:00
|
|
|
if (!pszFileName)
|
|
|
|
return NULL;
|
|
|
|
|
2016-10-07 15:06:46 +03:00
|
|
|
sprintf_s(pszFileName, cchFileName, "%s%s-client.%s", pszPrefix, pszName,
|
|
|
|
pszExtension);
|
2012-11-18 07:03:04 +04:00
|
|
|
cchFileName = strlen(pszFileName);
|
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
|
|
|
return NULL;
|
|
|
|
}
|
|
|
|
|
|
|
|
if (pszSubsystem)
|
|
|
|
{
|
|
|
|
LPSTR pszEntryName;
|
|
|
|
size_t cchEntryName;
|
|
|
|
/* subsystem add-in */
|
|
|
|
cchEntryName = 64 + strlen(pszName);
|
|
|
|
pszEntryName = (LPSTR) malloc(cchEntryName + 1);
|
2016-10-07 15:06:46 +03:00
|
|
|
|
2015-06-16 16:42:07 +03:00
|
|
|
if (!pszEntryName)
|
|
|
|
{
|
|
|
|
free(pszFileName);
|
|
|
|
return NULL;
|
|
|
|
}
|
2012-11-18 07:03:04 +04:00
|
|
|
|
2016-10-07 15:06:46 +03:00
|
|
|
sprintf_s(pszEntryName, cchEntryName + 1, "freerdp_%s_client_subsystem_entry",
|
|
|
|
pszName);
|
2012-11-18 07:03:04 +04:00
|
|
|
entry = freerdp_load_dynamic_addin(pszFileName, NULL, pszEntryName);
|
2012-11-19 22:26:56 +04:00
|
|
|
free(pszEntryName);
|
|
|
|
free(pszFileName);
|
2015-06-16 16:42:07 +03:00
|
|
|
return entry;
|
2012-11-18 07:03:04 +04:00
|
|
|
}
|
|
|
|
|
2015-06-16 16:42:07 +03:00
|
|
|
/* channel add-in */
|
2012-11-19 22:26:56 +04:00
|
|
|
|
2015-06-16 16:42:07 +03:00
|
|
|
if (dwFlags & FREERDP_ADDIN_CHANNEL_STATIC)
|
2016-12-28 13:21:14 +03:00
|
|
|
{
|
|
|
|
if (dwFlags & FREERDP_ADDIN_CHANNEL_ENTRYEX)
|
|
|
|
entry = freerdp_load_dynamic_addin(pszFileName, NULL, "VirtualChannelEntryEx");
|
|
|
|
else
|
|
|
|
entry = freerdp_load_dynamic_addin(pszFileName, NULL, "VirtualChannelEntry");
|
|
|
|
}
|
2015-06-16 16:42:07 +03:00
|
|
|
else if (dwFlags & FREERDP_ADDIN_CHANNEL_DYNAMIC)
|
|
|
|
entry = freerdp_load_dynamic_addin(pszFileName, NULL, "DVCPluginEntry");
|
|
|
|
else if (dwFlags & FREERDP_ADDIN_CHANNEL_DEVICE)
|
|
|
|
entry = freerdp_load_dynamic_addin(pszFileName, NULL, "DeviceServiceEntry");
|
|
|
|
else
|
|
|
|
entry = freerdp_load_dynamic_addin(pszFileName, NULL, pszType);
|
2012-11-18 07:03:04 +04:00
|
|
|
|
2015-06-16 16:42:07 +03:00
|
|
|
free(pszFileName);
|
|
|
|
return entry;
|
2012-11-18 07:03:04 +04:00
|
|
|
}
|
|
|
|
|
2016-10-07 15:06:46 +03:00
|
|
|
static FREERDP_LOAD_CHANNEL_ADDIN_ENTRY_FN
|
|
|
|
freerdp_load_static_channel_addin_entry = NULL;
|
2012-11-18 07:03:04 +04:00
|
|
|
|
2016-10-07 15:06:46 +03:00
|
|
|
int freerdp_register_addin_provider(FREERDP_LOAD_CHANNEL_ADDIN_ENTRY_FN
|
|
|
|
provider, DWORD dwFlags)
|
2012-11-18 07:03:04 +04:00
|
|
|
{
|
|
|
|
freerdp_load_static_channel_addin_entry = provider;
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
2016-10-07 15:06:46 +03:00
|
|
|
PVIRTUALCHANNELENTRY freerdp_load_channel_addin_entry(LPCSTR pszName,
|
|
|
|
LPSTR pszSubsystem, LPSTR pszType, DWORD dwFlags)
|
2012-11-18 07:03:04 +04:00
|
|
|
{
|
2016-10-07 15:06:46 +03:00
|
|
|
PVIRTUALCHANNELENTRY entry = NULL;
|
2012-11-18 07:03:04 +04:00
|
|
|
|
|
|
|
if (freerdp_load_static_channel_addin_entry)
|
2016-11-16 17:52:24 +03:00
|
|
|
entry = freerdp_load_static_channel_addin_entry(pszName, pszSubsystem, pszType, dwFlags);
|
|
|
|
|
2012-11-18 07:03:04 +04:00
|
|
|
if (!entry)
|
2016-11-16 17:52:24 +03:00
|
|
|
entry = freerdp_load_dynamic_channel_addin_entry(pszName, pszSubsystem, pszType, dwFlags);
|
2012-11-18 07:03:04 +04:00
|
|
|
|
2016-10-27 15:41:05 +03:00
|
|
|
if (!entry)
|
|
|
|
WLog_WARN(TAG, "Failed to load channel %s [%s]", pszName, pszSubsystem);
|
|
|
|
|
2012-11-18 07:03:04 +04:00
|
|
|
return entry;
|
|
|
|
}
|