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.
291 lines
7.2 KiB
291 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 <stdlib.h> |
|
//#include <stdio.h> |
|
#include <process.h> |
|
#include "main.h" |
|
#include "openvpn-gui-res.h" |
|
#include "options.h" |
|
|
|
extern struct options 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; |
|
char command_line[256]; |
|
char batch_file[100]; |
|
DWORD ExitCode; |
|
int i, TimeOut; |
|
TCHAR buf[1000]; |
|
|
|
/* Cut of extention from config filename and add "_up.bat". */ |
|
strncpy(batch_file, o.cnn[config].config_file, sizeof(batch_file)); |
|
batch_file[strlen(batch_file) - (strlen(o.ext_string)+1)]=0; |
|
strncat(batch_file, "_up.bat", sizeof(batch_file) - strlen(batch_file) - 1); |
|
mysnprintf(command_line, "%s\\%s", o.cnn[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 */ |
|
myLoadString(INFO_STATE_CONN_SCRIPT); |
|
SetDlgItemText(o.cnn[config].hwndStatus, TEXT_STATUS, buf); |
|
} |
|
|
|
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.cnn[config].config_dir, //start-up dir |
|
&start_info, |
|
&proc_info)) |
|
{ |
|
/* Running Script failed */ |
|
ShowLocalizedMsg(GUI_NAME, 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(GUI_NAME, ERR_GET_EXIT_CODE, command_line); |
|
return; |
|
} |
|
|
|
if (ExitCode != STILL_ACTIVE) |
|
{ |
|
if (ExitCode != 0) |
|
{ |
|
/* ConnectScript failed */ |
|
ShowLocalizedMsg(GUI_NAME, ERR_CONN_SCRIPT_FAILED, ExitCode); |
|
return; |
|
} |
|
return; |
|
} |
|
|
|
Sleep(1000); |
|
} |
|
|
|
/* UserInfo: Timeout */ |
|
ShowLocalizedMsg(GUI_NAME, 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; |
|
char command_line[256]; |
|
char batch_file[100]; |
|
DWORD ExitCode; |
|
int i, TimeOut; |
|
TCHAR buf[1000]; |
|
|
|
/* Append "_down.bat" to config name. */ |
|
strncpy(batch_file, o.cnn[config].config_name, sizeof(batch_file)); |
|
strncat(batch_file, "_down.bat", sizeof(batch_file) - strlen(batch_file) - 1); |
|
mysnprintf(command_line, "%s\\%s", o.cnn[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 */ |
|
myLoadString(INFO_STATE_DISCONN_SCRIPT); |
|
SetDlgItemText(o.cnn[config].hwndStatus, TEXT_STATUS, buf); |
|
} |
|
|
|
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.cnn[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; |
|
char command_line[256]; |
|
char batch_file[100]; |
|
DWORD ExitCode; |
|
int i, TimeOut; |
|
TCHAR buf[1000]; |
|
|
|
/* Append "_pre.bat" to config name. */ |
|
strncpy(batch_file, o.cnn[config].config_name, sizeof(batch_file)); |
|
strncat(batch_file, "_pre.bat", sizeof(batch_file) - strlen(batch_file) - 1); |
|
mysnprintf(command_line, "%s\\%s", o.cnn[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.cnn[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); |
|
} |
|
} |
|
|
|
|