/*
 *  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);
    }    
}