/* * OpenVPN-GUI -- A Windows GUI for OpenVPN. * * Copyright (C) 2004 Mathias Sundman * * 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 //#include //#include #include #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); } }