diff options
Diffstat (limited to 'src/Cedar/Win32Com.cpp')
| -rw-r--r-- | src/Cedar/Win32Com.cpp | 979 |
1 files changed, 979 insertions, 0 deletions
diff --git a/src/Cedar/Win32Com.cpp b/src/Cedar/Win32Com.cpp new file mode 100644 index 00000000..0500fd00 --- /dev/null +++ b/src/Cedar/Win32Com.cpp @@ -0,0 +1,979 @@ +// SoftEther VPN Source Code
+// Cedar Communication Module
+//
+// SoftEther VPN Server, Client and Bridge are free software under GPLv2.
+//
+// Copyright (c) 2012-2014 Daiyuu Nobori.
+// Copyright (c) 2012-2014 SoftEther VPN Project, University of Tsukuba, Japan.
+// Copyright (c) 2012-2014 SoftEther Corporation.
+//
+// All Rights Reserved.
+//
+// http://www.softether.org/
+//
+// Author: Daiyuu Nobori
+// Comments: Tetsuo Sugiyama, Ph.D.
+//
+//
+// This program is free software; you can redistribute it and/or
+// modify it under the terms of the GNU General Public License
+// version 2 as published by the Free Software Foundation.
+//
+// This program is distributed in the hope that it will be useful,
+// but WITHOUT ANY WARRANTY; without even the implied warranty of
+// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+// GNU General Public License for more details.
+//
+// You should have received a copy of the GNU General Public License version 2
+// along with this program; if not, write to the Free Software
+// Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
+//
+// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+// EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+// MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.
+// IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY
+// CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT,
+// TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE
+// SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
+//
+// THE LICENSE AGREEMENT IS ATTACHED ON THE SOURCE-CODE PACKAGE
+// AS "LICENSE.TXT" FILE. READ THE TEXT FILE IN ADVANCE TO USE THE SOFTWARE.
+//
+//
+// THIS SOFTWARE IS DEVELOPED IN JAPAN, AND DISTRIBUTED FROM JAPAN,
+// UNDER JAPANESE LAWS. YOU MUST AGREE IN ADVANCE TO USE, COPY, MODIFY,
+// MERGE, PUBLISH, DISTRIBUTE, SUBLICENSE, AND/OR SELL COPIES OF THIS
+// SOFTWARE, THAT ANY JURIDICAL DISPUTES WHICH ARE CONCERNED TO THIS
+// SOFTWARE OR ITS CONTENTS, AGAINST US (SOFTETHER PROJECT, SOFTETHER
+// CORPORATION, DAIYUU NOBORI OR OTHER SUPPLIERS), OR ANY JURIDICAL
+// DISPUTES AGAINST US WHICH ARE CAUSED BY ANY KIND OF USING, COPYING,
+// MODIFYING, MERGING, PUBLISHING, DISTRIBUTING, SUBLICENSING, AND/OR
+// SELLING COPIES OF THIS SOFTWARE SHALL BE REGARDED AS BE CONSTRUED AND
+// CONTROLLED BY JAPANESE LAWS, AND YOU MUST FURTHER CONSENT TO
+// EXCLUSIVE JURISDICTION AND VENUE IN THE COURTS SITTING IN TOKYO,
+// JAPAN. YOU MUST WAIVE ALL DEFENSES OF LACK OF PERSONAL JURISDICTION
+// AND FORUM NON CONVENIENS. PROCESS MAY BE SERVED ON EITHER PARTY IN
+// THE MANNER AUTHORIZED BY APPLICABLE LAW OR COURT RULE.
+//
+// USE ONLY IN JAPAN. DO NOT USE IT IN OTHER COUNTRIES. IMPORTING THIS
+// SOFTWARE INTO OTHER COUNTRIES IS AT YOUR OWN RISK. SOME COUNTRIES
+// PROHIBIT ENCRYPTED COMMUNICATIONS. USING THIS SOFTWARE IN OTHER
+// COUNTRIES MIGHT BE RESTRICTED.
+//
+//
+// DEAR SECURITY EXPERTS
+// ---------------------
+//
+// If you find a bug or a security vulnerability please kindly inform us
+// about the problem immediately so that we can fix the security problem
+// to protect a lot of users around the world as soon as possible.
+//
+// Our e-mail address for security reports is:
+// softether-vpn-security [at] softether.org
+//
+// Please note that the above e-mail address is not a technical support
+// inquiry address. If you need technical assistance, please visit
+// http://www.softether.org/ and ask your question on the users forum.
+//
+// Thank you for your cooperation.
+
+
+// Win32Com.c
+// Win32 COM module call
+
+#include <GlobalConst.h>
+
+#ifdef WIN32
+
+#define WIN32COM_CPP
+
+#define _WIN32_DCOM
+
+//#define _WIN32_WINNT 0x0502
+//#define WINVER 0x0502
+#include <winsock2.h>
+#include <windows.h>
+#include <wincrypt.h>
+#include <wininet.h>
+#include <Wbemidl.h>
+#include <comdef.h>
+#include <Mshtmhst.h>
+#include <shlobj.h>
+#include <commctrl.h>
+#include <Dbghelp.h>
+#include <iphlpapi.h>
+#include <Natupnp.h>
+#include <devguid.h>
+#include <regstr.h>
+#include <cfgmgr32.h>
+#include <tchar.h>
+#include <objbase.h>
+#include <Setupapi.h>
+#include "netcfgn.h"
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+#include <wchar.h>
+#include <stdarg.h>
+#include <time.h>
+#include <errno.h>
+extern "C"
+{
+#include <Mayaqua/Mayaqua.h>
+#include <Cedar/Cedar.h>
+}
+#include "../PenCore/resource.h"
+
+// Add a UPnP port
+bool Win32UPnPAddPort(UINT outside_port, UINT inside_port, bool udp, char *local_ip, wchar_t *description, bool remove_before_add)
+{
+ bool ret = false;
+ HRESULT hr;
+ IUPnPNAT *nat = NULL;
+ wchar_t ip_str[MAX_SIZE];
+ BSTR bstr_ip, bstr_description, bstr_protocol;
+ wchar_t *protocol_str = (udp ? L"UDP" : L"TCP");
+ // Validate arguments
+ if (outside_port == 0 || outside_port >= 65536 || inside_port == 0 || inside_port >= 65536 ||
+ IsEmptyStr(local_ip) || UniIsEmptyStr(description))
+ {
+ return false;
+ }
+
+ StrToUni(ip_str, sizeof(ip_str), local_ip);
+ bstr_ip = SysAllocString(ip_str);
+ bstr_description = SysAllocString(description);
+ bstr_protocol = SysAllocString(protocol_str);
+
+ hr = CoCreateInstance(CLSID_UPnPNAT, NULL, CLSCTX_INPROC_SERVER, IID_IUPnPNAT, (void **)&nat);
+
+ if (SUCCEEDED(hr))
+ {
+ if (nat != NULL)
+ {
+ IStaticPortMappingCollection *collection = NULL;
+ hr = nat->get_StaticPortMappingCollection(&collection);
+
+ if (SUCCEEDED(hr))
+ {
+ if (collection != NULL)
+ {
+ IStaticPortMapping *mapping = NULL;
+
+ if (remove_before_add)
+ {
+ hr = collection->Remove((long)outside_port, bstr_protocol);
+ }
+
+ hr = collection->Add((long)outside_port, bstr_protocol, (long)inside_port,
+ bstr_ip, VARIANT_TRUE, bstr_description, &mapping);
+
+ if (SUCCEEDED(hr))
+ {
+ ret = true;
+
+ if (mapping != NULL)
+ {
+ mapping->Release();
+ }
+ }
+
+ collection->Release();
+ }
+ else
+ {
+ WHERE;
+ }
+ }
+ else
+ {
+ WHERE;
+ }
+
+ nat->Release();
+ }
+ else
+ {
+ WHERE;
+ }
+ }
+ else
+ {
+ WHERE;
+ }
+
+ SysFreeString(bstr_ip);
+ SysFreeString(bstr_description);
+ SysFreeString(bstr_protocol);
+
+ return ret;
+}
+
+// Install the NDIS protocol driver
+bool InstallNdisProtocolDriver(wchar_t *inf_path, wchar_t *id, UINT lock_timeout)
+{
+ bool ret = false;
+ HRESULT hr;
+ INetCfg *pNetCfg;
+ HINSTANCE hSetupApiDll = NULL;
+ BOOL (WINAPI *_SetupCopyOEMInfW)(PCWSTR, PCWSTR, DWORD, DWORD, PWSTR, DWORD, PDWORD, PWSTR *) = NULL;
+ BOOL (WINAPI *_SetupUninstallOEMInfW)(PCWSTR, DWORD, PVOID) = NULL;
+ // Validate arguments
+ if (inf_path == NULL || id == NULL)
+ {
+ return false;
+ }
+
+ hSetupApiDll = LoadLibraryA("setupapi.dll");
+ if (hSetupApiDll == NULL)
+ {
+ WHERE;
+ goto LABEL_CLEANUP;
+ }
+
+ _SetupCopyOEMInfW =
+ (UINT (__stdcall *)(PCWSTR,PCWSTR,DWORD,DWORD,PWSTR,DWORD,PDWORD,PWSTR *))
+ GetProcAddress(hSetupApiDll, "SetupCopyOEMInfW");
+
+ _SetupUninstallOEMInfW =
+ (UINT (__stdcall *)(PCWSTR,DWORD,PVOID))
+ GetProcAddress(hSetupApiDll, "SetupUninstallOEMInfW");
+
+ if (_SetupCopyOEMInfW == NULL || _SetupUninstallOEMInfW == NULL)
+ {
+ WHERE;
+ goto LABEL_CLEANUP;
+ }
+
+ hr = CoCreateInstance(CLSID_CNetCfg, NULL, CLSCTX_INPROC_SERVER, IID_INetCfg, (void **)&pNetCfg);
+
+ if (SUCCEEDED(hr))
+ {
+ INetCfgLock *pLock;
+
+ hr = pNetCfg->QueryInterface(IID_INetCfgLock, (PVOID*)&pLock);
+
+ if (SUCCEEDED(hr))
+ {
+ LPWSTR locked_by;
+
+ hr = pLock->AcquireWriteLock(lock_timeout, L"SoftEther VPN", &locked_by);
+
+ if (SUCCEEDED(hr))
+ {
+ hr = pNetCfg->Initialize(NULL);
+
+ if (SUCCEEDED(hr))
+ {
+ wchar_t inf_dir[MAX_PATH];
+
+ GetDirNameFromFilePathW(inf_dir, sizeof(inf_dir), inf_path);
+
+ if (_SetupCopyOEMInfW(inf_path, inf_dir, SPOST_PATH, 0, NULL, 0, NULL, 0))
+ {
+ INetCfgClassSetup *pSetup;
+
+ hr = pNetCfg->QueryNetCfgClass(&GUID_DEVCLASS_NETTRANS, IID_INetCfgClassSetup, (void **)&pSetup);
+
+ if (SUCCEEDED(hr))
+ {
+ OBO_TOKEN token;
+ INetCfgComponent *pComponent;
+
+ Zero(&token, sizeof(token));
+
+ token.Type = OBO_USER;
+
+ hr = pSetup->Install(id, &token, 0, 0, NULL, NULL, &pComponent);
+
+ if (SUCCEEDED(hr))
+ {
+ pNetCfg->Apply();
+
+ ret = true;
+ }
+ else
+ {
+ WHERE;
+ Debug("0x%x\n", hr);
+ }
+
+ pSetup->Release();
+ }
+ else
+ {
+ WHERE;
+ }
+
+ if (ret == false)
+ {
+ wchar_t dst_inf_name[MAX_PATH];
+ DWORD dst_inf_name_size = MAX_PATH;
+
+ if (_SetupCopyOEMInfW(inf_path, inf_dir, SPOST_PATH, SP_COPY_REPLACEONLY,
+ dst_inf_name, dst_inf_name_size, &dst_inf_name_size, NULL) == false &&
+ GetLastError() == ERROR_FILE_EXISTS)
+ {
+ _SetupUninstallOEMInfW(dst_inf_name, 0, NULL);
+ }
+ }
+ }
+ else
+ {
+ WHERE;
+ }
+ }
+ else
+ {
+ WHERE;
+ }
+
+ pLock->ReleaseWriteLock();
+ }
+ else
+ {
+ WHERE;
+ }
+
+ pLock->Release();
+ }
+
+ pNetCfg->Release();
+ }
+ else
+ {
+ WHERE;
+ }
+
+LABEL_CLEANUP:
+
+ if (hSetupApiDll != NULL)
+ {
+ FreeLibrary(hSetupApiDll);
+ }
+
+ return ret;
+}
+
+typedef struct FOLDER_DLG_INNER_DATA
+{
+ wchar_t *default_dir;
+} FOLDER_DLG_INNER_DATA;
+
+int CALLBACK FolderDlgInnerCallbackA(HWND hWnd, UINT msg, LPARAM lParam, LPARAM lData)
+{
+ FOLDER_DLG_INNER_DATA *data = (FOLDER_DLG_INNER_DATA *)lData;
+ LPITEMIDLIST pidl;
+
+ switch (msg)
+ {
+ case BFFM_INITIALIZED:
+ if (data->default_dir != NULL)
+ {
+ char *default_dir_a = CopyUniToStr(data->default_dir);
+
+ SendMessage(hWnd, BFFM_SETSELECTIONA, true, (LPARAM)default_dir_a);
+
+ Free(default_dir_a);
+ }
+ break;
+
+ case BFFM_SELCHANGED:
+ pidl = (LPITEMIDLIST)lParam;
+
+ if (pidl)
+ {
+ char tmp[MAX_PATH];
+
+ Zero(tmp, sizeof(tmp));
+ if (SHGetPathFromIDListA(pidl, tmp))
+ {
+ SendMessage(hWnd, BFFM_ENABLEOK, 0, 1);
+ }
+ else
+ {
+ SendMessage(hWnd, BFFM_ENABLEOK, 0, 0);
+ }
+ }
+ break;
+ }
+
+ return 0;
+}
+
+char *FolderDlgInnerA(HWND hWnd, wchar_t *title, char *default_dir)
+{
+ BROWSEINFOA info;
+ char display_name[MAX_PATH];
+ FOLDER_DLG_INNER_DATA data;
+ LPMALLOC pMalloc;
+ char *ret = NULL;
+ char *title_a;
+ if (UniIsEmptyStr(title))
+ {
+ title = NULL;
+ }
+ if (IsEmptyStr(default_dir))
+ {
+ default_dir = NULL;
+ }
+
+ Zero(&data, sizeof(data));
+ data.default_dir = CopyStrToUni(default_dir);
+
+ Zero(display_name, sizeof(display_name));
+ Zero(&info, sizeof(info));
+ info.hwndOwner = hWnd;
+ info.pidlRoot = NULL;
+ info.pszDisplayName = display_name;
+ title_a = CopyUniToStr(title);
+ info.lpszTitle = title_a;
+ info.ulFlags = BIF_NEWDIALOGSTYLE | BIF_RETURNONLYFSDIRS | BIF_VALIDATE | BIF_SHAREABLE;
+ info.lpfn = FolderDlgInnerCallbackA;
+ info.lParam = (LPARAM)&data;
+
+ if (SUCCEEDED(SHGetMalloc(&pMalloc)))
+ {
+ LPITEMIDLIST pidl;
+
+ pidl = SHBrowseForFolderA(&info);
+
+ if (pidl)
+ {
+ char tmp[MAX_PATH];
+
+ if (SHGetPathFromIDListA(pidl, tmp))
+ {
+ ret = CopyStr(tmp);
+ }
+
+ pMalloc->Free(pidl);
+ }
+
+ pMalloc->Release();
+ }
+
+ Free(data.default_dir);
+ Free(title_a);
+
+ return ret;
+}
+
+int CALLBACK FolderDlgInnerCallbackW(HWND hWnd, UINT msg, LPARAM lParam, LPARAM lData)
+{
+ FOLDER_DLG_INNER_DATA *data = (FOLDER_DLG_INNER_DATA *)lData;
+ LPITEMIDLIST pidl;
+
+ switch (msg)
+ {
+ case BFFM_INITIALIZED:
+ if (data->default_dir != NULL)
+ {
+ SendMessage(hWnd, BFFM_SETSELECTIONW, true, (LPARAM)data->default_dir);
+ }
+ break;
+
+ case BFFM_SELCHANGED:
+ pidl = (LPITEMIDLIST)lParam;
+
+ if (pidl)
+ {
+ wchar_t tmp[MAX_PATH];
+
+ Zero(tmp, sizeof(tmp));
+ if (SHGetPathFromIDListW(pidl, tmp))
+ {
+ SendMessage(hWnd, BFFM_ENABLEOK, 0, 1);
+ }
+ else
+ {
+ SendMessage(hWnd, BFFM_ENABLEOK, 0, 0);
+ }
+ }
+ break;
+ }
+
+ return 0;
+}
+
+wchar_t *FolderDlgInnerW(HWND hWnd, wchar_t *title, wchar_t *default_dir)
+{
+ BROWSEINFOW info;
+ wchar_t display_name[MAX_PATH];
+ FOLDER_DLG_INNER_DATA data;
+ LPMALLOC pMalloc;
+ wchar_t *ret = NULL;
+ if (UniIsEmptyStr(title))
+ {
+ title = NULL;
+ }
+ if (UniIsEmptyStr(default_dir))
+ {
+ default_dir = NULL;
+ }
+
+ Zero(&data, sizeof(data));
+ data.default_dir = default_dir;
+
+ Zero(display_name, sizeof(display_name));
+ Zero(&info, sizeof(info));
+ info.hwndOwner = hWnd;
+ info.pidlRoot = NULL;
+ info.pszDisplayName = display_name;
+ info.lpszTitle = title;
+ info.ulFlags = BIF_NEWDIALOGSTYLE | BIF_RETURNONLYFSDIRS | BIF_VALIDATE | BIF_SHAREABLE;
+ info.lpfn = FolderDlgInnerCallbackW;
+ info.lParam = (LPARAM)&data;
+
+ if (SUCCEEDED(SHGetMalloc(&pMalloc)))
+ {
+ LPITEMIDLIST pidl;
+
+ pidl = SHBrowseForFolderW(&info);
+
+ if (pidl)
+ {
+ wchar_t tmp[MAX_PATH];
+
+ if (SHGetPathFromIDListW(pidl, tmp))
+ {
+ ret = CopyUniStr(tmp);
+ }
+
+ pMalloc->Free(pidl);
+ }
+
+ pMalloc->Release();
+ }
+
+ return ret;
+}
+
+
+class CModule
+{
+public:
+ CModule()
+ {
+ m_hInstLib = NULL;
+ }
+ CModule( HINSTANCE hInstLib )
+ {
+ m_hInstLib = NULL;
+ this->Attach( hInstLib );
+ }
+ CModule( LPCTSTR pszModuleName )
+ {
+ m_hInstLib = NULL;
+ this->LoadLibrary( pszModuleName );
+ }
+ virtual ~CModule()
+ {
+ this->FreeLibrary();
+ }
+
+public:
+ BOOL Attach( HINSTANCE hInstLib )
+ {
+ this->FreeLibrary();
+ m_hInstLib = hInstLib;
+
+ return TRUE;
+ }
+ BOOL Detach()
+ {
+ m_hInstLib = NULL;
+
+ return TRUE;
+ }
+
+public:
+ HMODULE GetHandle()
+ {
+ return m_hInstLib;
+ }
+ // Load the DLL
+ HINSTANCE LoadLibrary( LPCTSTR pszModuleName )
+ {
+ this->FreeLibrary();
+ m_hInstLib = ::LoadLibrary( pszModuleName );
+
+ return m_hInstLib;
+ }
+ // Release the DLL
+ BOOL FreeLibrary()
+ {
+ if (m_hInstLib == NULL)
+ {
+ return FALSE;
+ }
+
+ BOOL bResult = ::FreeLibrary( m_hInstLib );
+ m_hInstLib = NULL;
+
+ return bResult;
+ }
+ // Get the address of the function
+ FARPROC GetProcAddress( LPCTSTR pszProcName )
+ {
+ if (m_hInstLib == NULL)
+ {
+ return NULL;
+ }
+
+ return ::GetProcAddress(m_hInstLib, pszProcName);
+ }
+ // Get a handle to the information block of resource with the specified name and the type
+ HRSRC FindResource(LPCTSTR lpName, LPCTSTR lpType)
+ {
+ if (m_hInstLib == NULL)
+ {
+ return NULL;
+ }
+
+ return ::FindResource(m_hInstLib, lpName, lpType);
+ }
+ // Load the specified resource
+ HGLOBAL LoadResource(HRSRC hResInfo)
+ {
+ if (m_hInstLib == NULL)
+ {
+ return NULL;
+ }
+
+ return ::LoadResource(m_hInstLib, hResInfo);
+ }
+
+protected:
+ HINSTANCE m_hInstLib;
+};
+
+
+
+static HRESULT _ShowHTMLDialog(
+ HWND hwndParent,
+ IMoniker* pMk,
+ VARIANT* pvarArgIn = NULL,
+ WCHAR* pchOptions = NULL,
+ VARIANT* pvarArgOut = NULL)
+{
+ HRESULT hr = S_OK;
+
+ try
+ {
+ CModule Module("MSHTML.DLL");
+ if (Module.GetHandle() == NULL)
+ {
+ return E_FAIL;
+ }
+
+ SHOWHTMLDIALOGFN* fnShowHTMLDialog =
+ (SHOWHTMLDIALOGFN*)Module.GetProcAddress("ShowHTMLDialog");
+ if (fnShowHTMLDialog == NULL)
+ {
+ return E_FAIL;
+ }
+
+ hr = (*fnShowHTMLDialog)(hwndParent, pMk, pvarArgIn, pchOptions, pvarArgOut);
+ if (FAILED(hr))
+ {
+ return hr;
+ }
+ }
+ catch (...)
+ {
+ return E_FAIL;
+ }
+
+ return hr;
+}
+
+HRESULT ShowHTMLDialogFromURL(HWND hwndParent,wchar_t *szURL,VARIANT* pvarArgIn,WCHAR* pchOptions,VARIANT* pvarArgOut)
+{
+ HRESULT hr = S_OK;
+
+ try
+ {
+ IMonikerPtr spMoniker;
+ hr = ::CreateURLMoniker(NULL, szURL, &spMoniker);
+ if (FAILED(hr))
+ {
+ return hr;
+ }
+
+ hr = ::_ShowHTMLDialog(hwndParent, spMoniker, pvarArgIn, pchOptions, pvarArgOut);
+ if (FAILED(hr))
+ {
+ return hr;
+ }
+ }
+ catch (...)
+ {
+ return E_FAIL;
+ }
+
+ return hr;
+}
+
+// Create a shortcut
+bool CreateLinkInnerA(char *filename, char *target, char *workdir, char *args,
+ char *comment, char *icon, UINT icon_index)
+{
+ HRESULT r;
+ wchar_t tmp[MAX_SIZE];
+ IShellLinkA* pShellLink;
+ IPersistFile* pPersistFile;
+
+ r = CoCreateInstance(CLSID_ShellLink, NULL, CLSCTX_INPROC_SERVER, IID_IShellLinkA, (void **)&pShellLink);
+ if (FAILED(r))
+ {
+ return false;
+ }
+
+ r = pShellLink->QueryInterface(IID_IPersistFile,(void **)&pPersistFile);
+ if (FAILED(r))
+ {
+ pShellLink->Release();
+ return false;
+ }
+
+ r = pShellLink->SetPath(target);
+ if (FAILED(r))
+ {
+ pShellLink->Release();
+ pPersistFile->Release();
+ return false;
+ }
+
+ if (workdir != NULL)
+ {
+ r = pShellLink->SetWorkingDirectory(workdir);
+ if (FAILED(r))
+ {
+ pShellLink->Release();
+ pPersistFile->Release();
+ return false;
+ }
+ }
+
+ if (args != NULL)
+ {
+ r = pShellLink->SetArguments(args);
+ if (FAILED(r))
+ {
+ pShellLink->Release();
+ pPersistFile->Release();
+ return false;
+ }
+ }
+
+ if (comment != NULL)
+ {
+ r = pShellLink->SetDescription(comment);
+ if (FAILED(r))
+ {
+ pShellLink->Release();
+ pPersistFile->Release();
+ return false;
+ }
+ }
+
+ if (icon != NULL)
+ {
+ r = pShellLink->SetIconLocation(icon, icon_index);
+ if (FAILED(r))
+ {
+ pShellLink->Release();
+ pPersistFile->Release();
+ return false;
+ }
+ }
+
+ StrToUni(tmp, sizeof(tmp), filename);
+ r = pPersistFile->Save(tmp, true);
+ if (FAILED(r))
+ {
+ pShellLink->Release();
+ pPersistFile->Release();
+ return false;
+ }
+
+ pShellLink->Release();
+ pPersistFile->Release();
+ return true;
+}
+bool CreateLinkInner(wchar_t *filename, wchar_t *target, wchar_t *workdir, wchar_t *args,
+ wchar_t *comment, wchar_t *icon, UINT icon_index)
+{
+ HRESULT r;
+ bool ret;
+ IShellLinkW* pShellLink;
+ IPersistFile* pPersistFile;
+
+ if (OS_IS_WINDOWS_9X(GetOsInfo()->OsType))
+ {
+ char *a1, *a2, *a3, *a4, *a5, *a6;
+ a1 = CopyUniToStr(filename);
+ a2 = CopyUniToStr(target);
+ a3 = CopyUniToStr(workdir);
+ a4 = CopyUniToStr(args);
+ a5 = CopyUniToStr(icon);
+ a6 = CopyUniToStr(comment);
+
+ ret = CreateLinkInnerA(a1, a2, a3, a4, a6, a5, icon_index);
+
+ Free(a1);
+ Free(a2);
+ Free(a3);
+ Free(a4);
+ Free(a5);
+ Free(a6);
+
+ return ret;
+ }
+
+ r = CoCreateInstance(CLSID_ShellLink, NULL, CLSCTX_INPROC_SERVER, IID_IShellLinkW, (void **)&pShellLink);
+ if (FAILED(r))
+ {
+ return false;
+ }
+
+ r = pShellLink->QueryInterface(IID_IPersistFile,(void **)&pPersistFile);
+ if (FAILED(r))
+ {
+ pShellLink->Release();
+ return false;
+ }
+
+ r = pShellLink->SetPath(target);
+ if (FAILED(r))
+ {
+ pShellLink->Release();
+ pPersistFile->Release();
+ return false;
+ }
+
+ if (workdir != NULL)
+ {
+ r = pShellLink->SetWorkingDirectory(workdir);
+ if (FAILED(r))
+ {
+ pShellLink->Release();
+ pPersistFile->Release();
+ return false;
+ }
+ }
+
+ if (comment != NULL)
+ {
+ r = pShellLink->SetDescription(comment);
+ if (FAILED(r))
+ {
+ pShellLink->Release();
+ pPersistFile->Release();
+ return false;
+ }
+ }
+
+ if (args != NULL)
+ {
+ r = pShellLink->SetArguments(args);
+ if (FAILED(r))
+ {
+ pShellLink->Release();
+ pPersistFile->Release();
+ return false;
+ }
+ }
+
+ if (icon != NULL)
+ {
+ r = pShellLink->SetIconLocation(icon, icon_index);
+ if (FAILED(r))
+ {
+ pShellLink->Release();
+ pPersistFile->Release();
+ return false;
+ }
+ }
+
+ r = pPersistFile->Save(filename, true);
+ if (FAILED(r))
+ {
+ pShellLink->Release();
+ pPersistFile->Release();
+ return false;
+ }
+
+ pShellLink->Release();
+ pPersistFile->Release();
+ return true;
+}
+
+extern "C"
+{
+
+// Show the folder selection dialog
+wchar_t *FolderDlgW(HWND hWnd, wchar_t *title, wchar_t *default_dir)
+{
+ wchar_t *ret;
+
+ if (MsIsNt() == false)
+ {
+ char *default_dir_a = CopyUniToStr(default_dir);
+ char *ret_a = FolderDlgA(hWnd, title, default_dir_a);
+
+ ret = CopyStrToUni(ret_a);
+ Free(ret_a);
+ Free(default_dir_a);
+
+ return ret;
+ }
+
+ ret = FolderDlgInnerW(hWnd, title, default_dir);
+
+ return ret;
+}
+char *FolderDlgA(HWND hWnd, wchar_t *title, char *default_dir)
+{
+ char *ret;
+
+ ret = FolderDlgInnerA(hWnd, title, default_dir);
+
+ return ret;
+}
+
+// Create a shortcut
+bool CreateLink(wchar_t *filename, wchar_t *target, wchar_t *workdir, wchar_t *args,
+ wchar_t *comment, wchar_t *icon, UINT icon_index)
+{
+ if (filename == NULL || target == NULL)
+ {
+ return false;
+ }
+
+ return CreateLinkInner(filename, target, workdir, args, comment, icon, icon_index);
+}
+
+// Show the HTML
+void ShowHtml(HWND hWnd, char *url, wchar_t *option)
+{
+ wchar_t tmp[MAX_SIZE];
+ // Validate arguments
+ if (url == NULL || option == NULL)
+ {
+ return;
+ }
+
+ StrToUni(tmp, sizeof(tmp), url);
+
+ ShowHTMLDialogFromURL(hWnd, tmp, NULL, option, NULL);
+}
+
+}
+
+#endif
+
+// Developed by SoftEther VPN Project at University of Tsukuba in Japan.
+// Department of Computer Science has dozens of overly-enthusiastic geeks.
+// Join us: http://www.tsukuba.ac.jp/english/admission/
|