302 lines
8.5 KiB
C
302 lines
8.5 KiB
C
/* SPDX-License-Identifier: GPL-2.0-or-later
|
|
* Copyright 2001-2002 NaN Holding BV. All rights reserved. */
|
|
|
|
/** \file
|
|
* \ingroup bli
|
|
* WIN32-POSIX compatibility layer, MS-Windows-specific functions.
|
|
*/
|
|
|
|
#ifdef WIN32
|
|
|
|
# include <conio.h>
|
|
# include <shlwapi.h>
|
|
# include <stdio.h>
|
|
# include <stdlib.h>
|
|
|
|
# include "MEM_guardedalloc.h"
|
|
|
|
# define WIN32_SKIP_HKEY_PROTECTION /* Need to use HKEY. */
|
|
# include "BLI_fileops.h"
|
|
# include "BLI_path_util.h"
|
|
# include "BLI_string.h"
|
|
# include "BLI_utildefines.h"
|
|
# include "BLI_winstuff.h"
|
|
|
|
# include "utf_winfunc.h"
|
|
# include "utfconv.h"
|
|
|
|
/* FILE_MAXDIR + FILE_MAXFILE */
|
|
|
|
int BLI_windows_get_executable_dir(char *str)
|
|
{
|
|
char dir[FILE_MAXDIR];
|
|
int a;
|
|
/* Change to utf support. */
|
|
GetModuleFileName(NULL, str, FILE_MAX);
|
|
BLI_split_dir_part(str, dir, sizeof(dir)); /* shouldn't be relative */
|
|
a = strlen(dir);
|
|
if (dir[a - 1] == '\\') {
|
|
dir[a - 1] = 0;
|
|
}
|
|
|
|
strcpy(str, dir);
|
|
|
|
return 1;
|
|
}
|
|
|
|
static void register_blend_extension_failed(HKEY root, const bool background)
|
|
{
|
|
printf("failed\n");
|
|
if (root) {
|
|
RegCloseKey(root);
|
|
}
|
|
if (!background) {
|
|
MessageBox(0, "Could not register file extension.", "Blender error", MB_OK | MB_ICONERROR);
|
|
}
|
|
}
|
|
|
|
bool BLI_windows_register_blend_extension(const bool background)
|
|
{
|
|
LONG lresult;
|
|
HKEY hkey = 0;
|
|
HKEY root = 0;
|
|
BOOL usr_mode = false;
|
|
DWORD dwd = 0;
|
|
char buffer[256];
|
|
|
|
char BlPath[MAX_PATH];
|
|
char MBox[256];
|
|
|
|
printf("Registering file extension...");
|
|
GetModuleFileName(0, BlPath, MAX_PATH);
|
|
|
|
/* Replace the actual app name with the wrapper. */
|
|
{
|
|
char *blender_app = strstr(BlPath, "blender.exe");
|
|
if (blender_app != NULL) {
|
|
strcpy(blender_app, "blender-launcher.exe");
|
|
}
|
|
}
|
|
|
|
/* root is HKLM by default */
|
|
lresult = RegOpenKeyEx(HKEY_LOCAL_MACHINE, "Software\\Classes", 0, KEY_ALL_ACCESS, &root);
|
|
if (lresult != ERROR_SUCCESS) {
|
|
/* try HKCU on failure */
|
|
usr_mode = true;
|
|
lresult = RegOpenKeyEx(HKEY_CURRENT_USER, "Software\\Classes", 0, KEY_ALL_ACCESS, &root);
|
|
if (lresult != ERROR_SUCCESS) {
|
|
register_blend_extension_failed(0, background);
|
|
return false;
|
|
}
|
|
}
|
|
|
|
lresult = RegCreateKeyEx(
|
|
root, "blendfile", 0, NULL, REG_OPTION_NON_VOLATILE, KEY_ALL_ACCESS, NULL, &hkey, &dwd);
|
|
if (lresult == ERROR_SUCCESS) {
|
|
strcpy(buffer, "Blender File");
|
|
lresult = RegSetValueEx(hkey, NULL, 0, REG_SZ, (BYTE *)buffer, strlen(buffer) + 1);
|
|
RegCloseKey(hkey);
|
|
}
|
|
if (lresult != ERROR_SUCCESS) {
|
|
register_blend_extension_failed(root, background);
|
|
return false;
|
|
}
|
|
|
|
lresult = RegCreateKeyEx(root,
|
|
"blendfile\\shell\\open\\command",
|
|
0,
|
|
NULL,
|
|
REG_OPTION_NON_VOLATILE,
|
|
KEY_ALL_ACCESS,
|
|
NULL,
|
|
&hkey,
|
|
&dwd);
|
|
if (lresult == ERROR_SUCCESS) {
|
|
BLI_snprintf(buffer, sizeof(buffer), "\"%s\" \"%%1\"", BlPath);
|
|
lresult = RegSetValueEx(hkey, NULL, 0, REG_SZ, (BYTE *)buffer, strlen(buffer) + 1);
|
|
RegCloseKey(hkey);
|
|
}
|
|
if (lresult != ERROR_SUCCESS) {
|
|
register_blend_extension_failed(root, background);
|
|
return false;
|
|
}
|
|
|
|
lresult = RegCreateKeyEx(root,
|
|
"blendfile\\DefaultIcon",
|
|
0,
|
|
NULL,
|
|
REG_OPTION_NON_VOLATILE,
|
|
KEY_ALL_ACCESS,
|
|
NULL,
|
|
&hkey,
|
|
&dwd);
|
|
if (lresult == ERROR_SUCCESS) {
|
|
BLI_snprintf(buffer, sizeof(buffer), "\"%s\", 1", BlPath);
|
|
lresult = RegSetValueEx(hkey, NULL, 0, REG_SZ, (BYTE *)buffer, strlen(buffer) + 1);
|
|
RegCloseKey(hkey);
|
|
}
|
|
if (lresult != ERROR_SUCCESS) {
|
|
register_blend_extension_failed(root, background);
|
|
return false;
|
|
}
|
|
|
|
lresult = RegCreateKeyEx(
|
|
root, ".blend", 0, NULL, REG_OPTION_NON_VOLATILE, KEY_ALL_ACCESS, NULL, &hkey, &dwd);
|
|
if (lresult == ERROR_SUCCESS) {
|
|
strcpy(buffer, "blendfile");
|
|
lresult = RegSetValueEx(hkey, NULL, 0, REG_SZ, (BYTE *)buffer, strlen(buffer) + 1);
|
|
RegCloseKey(hkey);
|
|
}
|
|
if (lresult != ERROR_SUCCESS) {
|
|
register_blend_extension_failed(root, background);
|
|
return false;
|
|
}
|
|
|
|
# ifdef WITH_BLENDER_THUMBNAILER
|
|
{
|
|
char RegCmd[MAX_PATH * 2];
|
|
char InstallDir[FILE_MAXDIR];
|
|
char SysDir[FILE_MAXDIR];
|
|
BLI_windows_get_executable_dir(InstallDir);
|
|
GetSystemDirectory(SysDir, FILE_MAXDIR);
|
|
const char *ThumbHandlerDLL = "BlendThumb.dll";
|
|
snprintf(
|
|
RegCmd, MAX_PATH * 2, "%s\\regsvr32 /s \"%s\\%s\"", SysDir, InstallDir, ThumbHandlerDLL);
|
|
system(RegCmd);
|
|
}
|
|
# endif
|
|
|
|
RegCloseKey(root);
|
|
printf("success (%s)\n", usr_mode ? "user" : "system");
|
|
if (!background) {
|
|
BLI_snprintf(MBox,
|
|
sizeof(MBox),
|
|
"File extension registered for %s.",
|
|
usr_mode ?
|
|
"the current user. To register for all users, run as an administrator" :
|
|
"all users");
|
|
MessageBox(0, MBox, "Blender", MB_OK | MB_ICONINFORMATION);
|
|
}
|
|
return true;
|
|
}
|
|
|
|
/**
|
|
* Check the registry to see if there is an operation association to a file
|
|
* extension. Extension *should almost always contain a dot like `.txt`,
|
|
* but this does allow querying non - extensions *like "Directory", "Drive",
|
|
* "AllProtocols", etc - anything in Classes with a "shell" branch.
|
|
*/
|
|
static bool BLI_windows_file_operation_is_registered(const char *extension, const char *operation)
|
|
{
|
|
HKEY hKey;
|
|
HRESULT hr = AssocQueryKey(ASSOCF_INIT_IGNOREUNKNOWN,
|
|
ASSOCKEY_SHELLEXECCLASS,
|
|
(LPCTSTR)extension,
|
|
(LPCTSTR)operation,
|
|
&hKey);
|
|
if (SUCCEEDED(hr)) {
|
|
RegCloseKey(hKey);
|
|
return true;
|
|
}
|
|
return false;
|
|
}
|
|
|
|
bool BLI_windows_external_operation_supported(const char *filepath, const char *operation)
|
|
{
|
|
if (STREQ(operation, "open") || STREQ(operation, "properties")) {
|
|
return true;
|
|
}
|
|
|
|
if (BLI_is_dir(filepath)) {
|
|
return BLI_windows_file_operation_is_registered("Directory", operation);
|
|
}
|
|
|
|
const char *extension = BLI_path_extension(filepath);
|
|
return BLI_windows_file_operation_is_registered(extension, operation);
|
|
}
|
|
|
|
bool BLI_windows_external_operation_execute(const char *filepath, const char *operation)
|
|
{
|
|
WCHAR wpath[FILE_MAX];
|
|
if (conv_utf_8_to_16(filepath, wpath, ARRAY_SIZE(wpath)) != 0) {
|
|
return false;
|
|
}
|
|
|
|
WCHAR woperation[FILE_MAX];
|
|
if (conv_utf_8_to_16(operation, woperation, ARRAY_SIZE(woperation)) != 0) {
|
|
return false;
|
|
}
|
|
|
|
SHELLEXECUTEINFOW shellinfo = {0};
|
|
shellinfo.cbSize = sizeof(SHELLEXECUTEINFO);
|
|
shellinfo.fMask = SEE_MASK_INVOKEIDLIST;
|
|
shellinfo.lpVerb = woperation;
|
|
shellinfo.lpFile = wpath;
|
|
shellinfo.nShow = SW_SHOW;
|
|
|
|
return ShellExecuteExW(&shellinfo);
|
|
}
|
|
|
|
void BLI_windows_get_default_root_dir(char root[4])
|
|
{
|
|
char str[MAX_PATH + 1];
|
|
|
|
/* the default drive to resolve a directory without a specified drive
|
|
* should be the Windows installation drive, since this was what the OS
|
|
* assumes. */
|
|
if (GetWindowsDirectory(str, MAX_PATH + 1)) {
|
|
root[0] = str[0];
|
|
root[1] = ':';
|
|
root[2] = '\\';
|
|
root[3] = '\0';
|
|
}
|
|
else {
|
|
/* if GetWindowsDirectory fails, something has probably gone wrong,
|
|
* we are trying the blender install dir though */
|
|
if (GetModuleFileName(NULL, str, MAX_PATH + 1)) {
|
|
printf(
|
|
"Error! Could not get the Windows Directory - "
|
|
"Defaulting to Blender installation Dir!\n");
|
|
root[0] = str[0];
|
|
root[1] = ':';
|
|
root[2] = '\\';
|
|
root[3] = '\0';
|
|
}
|
|
else {
|
|
DWORD tmp;
|
|
int i;
|
|
int rc = 0;
|
|
/* now something has gone really wrong - still trying our best guess */
|
|
printf(
|
|
"Error! Could not get the Windows Directory - "
|
|
"Defaulting to first valid drive! Path might be invalid!\n");
|
|
tmp = GetLogicalDrives();
|
|
for (i = 2; i < 26; i++) {
|
|
if ((tmp >> i) & 1) {
|
|
root[0] = 'a' + i;
|
|
root[1] = ':';
|
|
root[2] = '\\';
|
|
root[3] = '\0';
|
|
if (GetFileAttributes(root) != 0xFFFFFFFF) {
|
|
rc = i;
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
if (0 == rc) {
|
|
printf("ERROR in 'BLI_windows_get_default_root_dir': can't find a valid drive!\n");
|
|
root[0] = 'C';
|
|
root[1] = ':';
|
|
root[2] = '\\';
|
|
root[3] = '\0';
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
#else
|
|
|
|
/* intentionally empty for UNIX */
|
|
|
|
#endif
|