mirror of https://github.com/OpenVPN/openvpn-gui
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
289 lines
7.2 KiB
289 lines
7.2 KiB
/*
|
|
* OpenVPN-GUI -- A Windows GUI for OpenVPN.
|
|
*
|
|
* Copyright (C) 2004 Mathias Sundman <mathias@nilings.se>
|
|
*
|
|
* This program is free software; you can redistribute it and/or modify
|
|
* it under the terms of the GNU General Public License as published by
|
|
* the Free Software Foundation; either version 2 of the License, or
|
|
* (at your option) any later version.
|
|
*
|
|
* 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
|
|
* along with this program (see the file COPYING included with this
|
|
* distribution); if not, write to the Free Software Foundation, Inc.,
|
|
* 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
|
|
*/
|
|
|
|
#include <windows.h>
|
|
#include <process.h>
|
|
#include <tchar.h>
|
|
|
|
#include "config.h"
|
|
#include "main.h"
|
|
#include "openvpn-gui-res.h"
|
|
#include "options.h"
|
|
#include "localization.h"
|
|
|
|
extern options_t o;
|
|
|
|
void RunConnectScript(int config, int run_as_service)
|
|
{
|
|
WIN32_FIND_DATA FindFileData;
|
|
HANDLE hFind;
|
|
STARTUPINFO start_info;
|
|
PROCESS_INFORMATION proc_info;
|
|
SECURITY_ATTRIBUTES sa;
|
|
SECURITY_DESCRIPTOR sd;
|
|
TCHAR command_line[256];
|
|
TCHAR batch_file[100];
|
|
DWORD ExitCode;
|
|
int i, TimeOut;
|
|
|
|
/* Cut of extention from config filename and add "_up.bat". */
|
|
_tcsncpy(batch_file, o.conn[config].config_file, _tsizeof(batch_file));
|
|
batch_file[_tcslen(batch_file) - (_tcslen(o.ext_string)+1)]=0;
|
|
_tcsncat(batch_file, _T("_up.bat"), _tsizeof(batch_file) - _tcslen(batch_file) - 1);
|
|
_sntprintf_0(command_line, _T("%s\\%s"), o.conn[config].config_dir, batch_file);
|
|
|
|
|
|
/* Return if no script exists */
|
|
hFind = FindFirstFile(command_line, &FindFileData);
|
|
if (hFind == INVALID_HANDLE_VALUE)
|
|
{
|
|
return;
|
|
}
|
|
|
|
FindClose(hFind);
|
|
|
|
if (!run_as_service)
|
|
{
|
|
/* UserInfo: Connect Script running */
|
|
SetDlgItemText(o.conn[config].hwndStatus, ID_TXT_STATUS, LoadLocalizedString(IDS_NFO_STATE_CONN_SCRIPT));
|
|
}
|
|
|
|
CLEAR (start_info);
|
|
CLEAR (proc_info);
|
|
CLEAR (sa);
|
|
CLEAR (sd);
|
|
|
|
/* fill in STARTUPINFO struct */
|
|
GetStartupInfo(&start_info);
|
|
start_info.cb = sizeof(start_info);
|
|
start_info.dwFlags = 0;
|
|
start_info.wShowWindow = SW_SHOWDEFAULT;
|
|
start_info.hStdInput = NULL;
|
|
start_info.hStdOutput = NULL;
|
|
|
|
if (!CreateProcess(NULL,
|
|
command_line, //commandline
|
|
NULL,
|
|
NULL,
|
|
TRUE,
|
|
((o.show_script_window[0] == '1') ? (DWORD) CREATE_NEW_CONSOLE :
|
|
(DWORD) CREATE_NO_WINDOW),
|
|
NULL,
|
|
o.conn[config].config_dir, //start-up dir
|
|
&start_info,
|
|
&proc_info))
|
|
{
|
|
/* Running Script failed */
|
|
ShowLocalizedMsg(IDS_ERR_RUN_CONN_SCRIPT, command_line);
|
|
return;
|
|
}
|
|
|
|
TimeOut = o.connectscript_timeout;
|
|
|
|
if (TimeOut == 0)
|
|
{
|
|
/* Don't check exist status, just return */
|
|
return;
|
|
}
|
|
|
|
for (i=0; i <= TimeOut; i++)
|
|
{
|
|
if (!GetExitCodeProcess(proc_info.hProcess, &ExitCode))
|
|
{
|
|
/* failed to get ExitCode */
|
|
ShowLocalizedMsg(IDS_ERR_GET_EXIT_CODE, command_line);
|
|
return;
|
|
}
|
|
|
|
if (ExitCode != STILL_ACTIVE)
|
|
{
|
|
if (ExitCode != 0)
|
|
{
|
|
/* ConnectScript failed */
|
|
ShowLocalizedMsg(IDS_ERR_CONN_SCRIPT_FAILED, ExitCode);
|
|
return;
|
|
}
|
|
return;
|
|
}
|
|
|
|
Sleep(1000);
|
|
}
|
|
|
|
/* UserInfo: Timeout */
|
|
ShowLocalizedMsg(IDS_ERR_RUN_CONN_SCRIPT_TIMEOUT, TimeOut);
|
|
|
|
}
|
|
|
|
void RunDisconnectScript(int config, int run_as_service)
|
|
{
|
|
WIN32_FIND_DATA FindFileData;
|
|
HANDLE hFind;
|
|
STARTUPINFO start_info;
|
|
PROCESS_INFORMATION proc_info;
|
|
SECURITY_ATTRIBUTES sa;
|
|
SECURITY_DESCRIPTOR sd;
|
|
TCHAR command_line[256];
|
|
TCHAR batch_file[100];
|
|
DWORD ExitCode;
|
|
int i, TimeOut;
|
|
|
|
/* Append "_down.bat" to config name. */
|
|
_tcsncpy(batch_file, o.conn[config].config_name, _tsizeof(batch_file));
|
|
_tcsncat(batch_file, _T("_down.bat"), _tsizeof(batch_file) - _tcslen(batch_file) - 1);
|
|
_sntprintf_0(command_line, _T("%s\\%s"), o.conn[config].config_dir, batch_file);
|
|
|
|
|
|
/* Return if no script exists */
|
|
hFind = FindFirstFile(command_line, &FindFileData);
|
|
if (hFind == INVALID_HANDLE_VALUE)
|
|
{
|
|
return;
|
|
}
|
|
|
|
FindClose(hFind);
|
|
|
|
if (!run_as_service)
|
|
{
|
|
/* UserInfo: Disconnect Script running */
|
|
SetDlgItemText(o.conn[config].hwndStatus, ID_TXT_STATUS, LoadLocalizedString(IDS_NFO_STATE_DISCONN_SCRIPT));
|
|
}
|
|
|
|
CLEAR (start_info);
|
|
CLEAR (proc_info);
|
|
CLEAR (sa);
|
|
CLEAR (sd);
|
|
|
|
/* fill in STARTUPINFO struct */
|
|
GetStartupInfo(&start_info);
|
|
start_info.cb = sizeof(start_info);
|
|
start_info.dwFlags = 0;
|
|
start_info.wShowWindow = SW_SHOWDEFAULT;
|
|
start_info.hStdInput = NULL;
|
|
start_info.hStdOutput = NULL;
|
|
|
|
if (!CreateProcess(NULL,
|
|
command_line, //commandline
|
|
NULL,
|
|
NULL,
|
|
TRUE,
|
|
((o.show_script_window[0] == '1') ? (DWORD) CREATE_NEW_CONSOLE :
|
|
(DWORD) CREATE_NO_WINDOW),
|
|
NULL,
|
|
o.conn[config].config_dir, //start-up dir
|
|
&start_info,
|
|
&proc_info))
|
|
{
|
|
return;
|
|
}
|
|
|
|
TimeOut = o.disconnectscript_timeout;
|
|
|
|
for (i=0; i <= TimeOut; i++)
|
|
{
|
|
if (!GetExitCodeProcess(proc_info.hProcess, &ExitCode))
|
|
{
|
|
return;
|
|
}
|
|
|
|
if (ExitCode != STILL_ACTIVE)
|
|
{
|
|
return;
|
|
}
|
|
|
|
Sleep(1000);
|
|
}
|
|
}
|
|
|
|
void RunPreconnectScript(int config)
|
|
{
|
|
WIN32_FIND_DATA FindFileData;
|
|
HANDLE hFind;
|
|
STARTUPINFO start_info;
|
|
PROCESS_INFORMATION proc_info;
|
|
SECURITY_ATTRIBUTES sa;
|
|
SECURITY_DESCRIPTOR sd;
|
|
TCHAR command_line[256];
|
|
TCHAR batch_file[100];
|
|
DWORD ExitCode;
|
|
int i, TimeOut;
|
|
|
|
/* Append "_pre.bat" to config name. */
|
|
_tcsncpy(batch_file, o.conn[config].config_name, _tsizeof(batch_file));
|
|
_tcsncat(batch_file, _T("_pre.bat"), _tsizeof(batch_file) - _tcslen(batch_file) - 1);
|
|
_sntprintf_0(command_line, _T("%s\\%s"), o.conn[config].config_dir, batch_file);
|
|
|
|
|
|
/* Return if no script exists */
|
|
hFind = FindFirstFile(command_line, &FindFileData);
|
|
if (hFind == INVALID_HANDLE_VALUE)
|
|
{
|
|
return;
|
|
}
|
|
|
|
FindClose(hFind);
|
|
|
|
CLEAR (start_info);
|
|
CLEAR (proc_info);
|
|
CLEAR (sa);
|
|
CLEAR (sd);
|
|
|
|
/* fill in STARTUPINFO struct */
|
|
GetStartupInfo(&start_info);
|
|
start_info.cb = sizeof(start_info);
|
|
start_info.dwFlags = 0;
|
|
start_info.wShowWindow = SW_SHOWDEFAULT;
|
|
start_info.hStdInput = NULL;
|
|
start_info.hStdOutput = NULL;
|
|
|
|
if (!CreateProcess(NULL,
|
|
command_line, //commandline
|
|
NULL,
|
|
NULL,
|
|
TRUE,
|
|
((o.show_script_window[0] == '1') ? (DWORD) CREATE_NEW_CONSOLE :
|
|
(DWORD) CREATE_NO_WINDOW),
|
|
NULL,
|
|
o.conn[config].config_dir, //start-up dir
|
|
&start_info,
|
|
&proc_info))
|
|
{
|
|
return;
|
|
}
|
|
|
|
TimeOut = o.preconnectscript_timeout;
|
|
|
|
for (i=0; i <= TimeOut; i++)
|
|
{
|
|
if (!GetExitCodeProcess(proc_info.hProcess, &ExitCode))
|
|
{
|
|
return;
|
|
}
|
|
|
|
if (ExitCode != STILL_ACTIVE)
|
|
{
|
|
return;
|
|
}
|
|
|
|
Sleep(1000);
|
|
}
|
|
}
|
|
|