/***********************************************************************

  $Id: killproc.c 551 2007-02-28 01:33:51Z gyoung $

  Kill a process

  Copyright (c) 1993-98 M. Kimes
  Copyright (c) 2005, 2006 Steven H. Levine

  24 May 05 SHL Rework Win_Error usage
  14 Jul 06 SHL Use Runtime_Error
  29 Jul 06 SHL Use xfgets
  03 Nov 06 SHL Renames
  03 Nov 06 SHL Count thread usage

***********************************************************************/

#define INCL_DOSERRORS
#define INCL_DOS
#define INCL_WIN
#include <os2.h>

#include <stdlib.h>
#include <stdio.h>
#include <string.h>
#include <time.h>
#include <ctype.h>
#include <process.h>
#include <limits.h>

#include "procstat.h"
#include "fm3dll.h"
#include "fm3dlg.h"
#include "fm3str.h"

#pragma data_seg(DATA2)

static PSZ pszSrcFile = __FILE__;

#pragma alloc_text(KILLPROC,FillKillListThread,FillKillListThread2,GetDosPgmName,KillDlgProc)

CHAR *GetDosPgmName(PID pid, CHAR * string)
{
  HSWITCH hs;
  SWCNTRL swctl;
  PCH pch;

  *string = 0;
  hs = WinQuerySwitchHandle(0, pid);
  if (hs) {
    WinQuerySwitchEntry(hs, &swctl);
    pch = swctl.szSwtitle;
    while (*pch) {
      if (*pch < 0x10)
	if (pch != swctl.szSwtitle && *(pch - 1) == 0x20)
	  memmove(pch, pch + 1, strlen(pch));
	else {
	  *pch = 0x20;
	  pch++;
	}
      else
	pch++;
    }
    strcpy(string, swctl.szSwtitle);
  }
  if (!*string)
    strcpy(string, GetPString(IDS_UNKNOWNDOSPROCTEXT));
  return string;
}

static VOID FillKillListThread2(VOID * arg)
{
  HWND hwnd = *(HWND *) arg;
  CHAR s[1036];
  HAB thab;
  HMQ thmq;
  INT rc;
  PROCESSINFO *ppi;
  BUFFHEADER *pbh;
  MODINFO *pmi;

  thab = WinInitialize(0);
  thmq = WinCreateMsgQueue(thab, 0);
  WinCancelShutdown(thmq, TRUE);
  IncrThreadUsage();

  WinSendDlgItemMsg(hwnd, KILL_LISTBOX, LM_DELETEALL, MPVOID, MPVOID);
  rc = DosAllocMem((PVOID) & pbh, USHRT_MAX + 4096,
		   PAG_COMMIT | OBJ_TILE | PAG_READ | PAG_WRITE);
  if (rc)
    Dos_Error(MB_CANCEL, rc, HWND_DESKTOP, pszSrcFile, __LINE__,
	      GetPString(IDS_OUTOFMEMORY));
  else {
    rc = DosQProcStatus(pbh, USHRT_MAX);
    if (!rc) {
      ppi = pbh->ppi;
      while (ppi->ulEndIndicator != PROCESS_END_INDICATOR) {
	if (ppi->pid != mypid) {
	  pmi = pbh->pmi;
	  while (pmi && ppi->hModRef != pmi->hMod)
	    pmi = pmi->pNext;
	  if (pmi) {
	    sprintf(s, "%04x ", ppi->pid);
	    if (!stricmp(pmi->szModName, "SYSINIT"))
	      GetDosPgmName(ppi->pid, s + strlen(s));
	    else {
	      if (*pmi->szModName)
		strcat(s, pmi->szModName);
	      else
		strcat(s, GetPString(IDS_UNKNOWNPROCTEXT));
	    }
	    if (WinIsWindow(thab, hwnd)) {
	      WinSendDlgItemMsg(hwnd, KILL_LISTBOX, LM_INSERTITEM,
				MPFROM2SHORT(LIT_SORTASCENDING, 0),
				MPFROMP(s));
	    }
	    else
	      break;
	  }
	}
	ppi = (PPROCESSINFO) (ppi->ptiFirst + ppi->usThreadCount);
      }					// while
    }
    DosFreeMem(pbh);
  }

  if (WinIsWindow(thab, hwnd))
    PostMsg(hwnd, UM_CONTAINER_FILLED, MPVOID, MPVOID);
  WinDestroyMsgQueue(thmq);
  DecrThreadUsage();
  WinTerminate(thab);
}

static VOID FillKillListThread(VOID * arg)
{
  HWND hwnd = *(HWND *) arg;
  CHAR s[1036], progname[1027], *p;
  HAB thab;
  HMQ thmq;
  FILE *fp;
  BOOL foundstart = FALSE;
  INT rc;
  CHAR *startstring = "Process and Thread Information";
  CHAR *endstring = "System Semaphore Information";
  PID pid;
  HFILE oldstdout, newstdout;

  DosError(FERR_DISABLEHARDERR);

  thab = WinInitialize(0);
  thmq = WinCreateMsgQueue(thab, 0);
  WinCancelShutdown(thmq, TRUE);
  IncrThreadUsage();

  WinSendDlgItemMsg(hwnd, KILL_LISTBOX, LM_DELETEALL, MPVOID, MPVOID);
  strcpy(s, "$PSTAT#$.#$#");
  unlinkf("%s", s);
  fp = fopen(s, "w");
  if (!fp) {
    Win_Error(NULLHANDLE, HWND_DESKTOP, __FILE__, __LINE__,
	      GetPString(IDS_REDIRECTERRORTEXT));
    goto Abort;
  }
  else {
    newstdout = -1;
    rc = DosDupHandle(fileno(stdout), &newstdout);
    if (rc)
      Dos_Error(MB_CANCEL, rc, hwnd, __FILE__, __LINE__, "DosDupHandle");
    oldstdout = fileno(stdout);
    DosDupHandle(fileno(fp), &oldstdout);
    rc = runemf2(SEPARATE | INVISIBLE | FULLSCREEN | BACKGROUND | WAIT,
		 hwnd, NULL, NULL, "%s", "PSTAT.EXE /C");
    oldstdout = fileno(stdout);
    DosDupHandle(newstdout, &oldstdout);
    DosClose(newstdout);
    fclose(fp);
    // fixme to be gone?
    if (rc == -1) {
      saymsg(MB_CANCEL,
	     hwnd,
	     GetPString(IDS_ARGHTEXT), GetPString(IDS_CANTRUNPSTATTEXT));
      goto Abort;
    }
  }
  fp = fopen(s, "r");
  if (fp) {
    while (!feof(fp)) {
      strset(s, 0);
      if (!xfgets(s, 1025, fp, pszSrcFile, __LINE__))
	break;
      if (!foundstart) {
	if (*s == ' ' && strstr(s, startstring))
	  foundstart = TRUE;
      }
      else {
	if (*s == ' ' && strstr(s, endstring))
	  break;
	if (*s == ' ' && s[5] == ' ' && isxdigit(s[1]) &&
	    isxdigit(s[2]) && isxdigit(s[3]) && isxdigit(s[4])) {
	  p = &s[1];
	  pid = strtol(&s[1], &p, 16);
	  if (pid && pid != mypid) {
	    strcpy(progname, &s[30]);
	    p = strchr(progname, ' ');
	    if (p)
	      *p = 0;
	    if (!stristr(progname, "\\PSTAT.EXE")) {
	      sprintf(s, "%04x %s", pid, progname);
	      WinSendDlgItemMsg(hwnd,
				KILL_LISTBOX,
				LM_INSERTITEM,
				MPFROM2SHORT(LIT_SORTASCENDING, 0),
				MPFROMP(s));
	    }
	  }
	}
      }
    }
    fclose(fp);
  }
Abort:
  DosForceDelete("$PSTAT#$.#$#");
  PostMsg(hwnd, UM_CONTAINER_FILLED, MPVOID, MPVOID);
  WinDestroyMsgQueue(thmq);
  DecrThreadUsage();
  WinTerminate(thab);
}

MRESULT EXPENTRY KillDlgProc(HWND hwnd, ULONG msg, MPARAM mp1, MPARAM mp2)
{
  SHORT sSelect;
  PID pid;
  static BOOL listdone;
  static HPOINTER hptrIcon = (HPOINTER) 0;

  switch (msg) {
  case WM_INITDLG:
    hptrIcon = WinLoadPointer(HWND_DESKTOP, FM3ModHandle, KILL_FRAME);
    WinDefDlgProc(hwnd, WM_SETICON, MPFROMLONG(hptrIcon), MPVOID);
    WinCheckButton(hwnd, KILL_CHECKBOX, fUseQProcStat);
    PostMsg(hwnd, WM_COMMAND, MPFROM2SHORT(KILL_RESCAN, 0), MPVOID);
    break;

  case UM_CONTAINER_FILLED:
    listdone = TRUE;
    if ((SHORT) WinSendDlgItemMsg(hwnd,
				  KILL_LISTBOX,
				  LM_QUERYITEMCOUNT, MPVOID, MPVOID) == 0) {
      if (!fUseQProcStat)
	saymsg(MB_CANCEL,
	       hwnd,
	       GetPString(IDS_ICHOKEDTEXT), GetPString(IDS_ISPSTATTHERETEXT));
      else
	saymsg(MB_CANCEL,
	       hwnd,
	       GetPString(IDS_ICHOKEDTEXT),
	       GetPString(IDS_DOSQPROCSTATFAILEDTEXT));
    }
    return 0;

  case WM_CONTROL:
    switch (SHORT1FROMMP(mp1)) {
    case KILL_CHECKBOX:
      fUseQProcStat = WinQueryButtonCheckstate(hwnd, KILL_CHECKBOX);
      PrfWriteProfileData(fmprof,
			  FM3Str,
			  "UseQProcStat", &fUseQProcStat, sizeof(BOOL));
      PostMsg(hwnd, WM_COMMAND, MPFROM2SHORT(KILL_RESCAN, 0), MPVOID);
      break;

    case KILL_LISTBOX:
      switch (SHORT2FROMMP(mp2)) {
      case LN_ENTER:
	WinSendDlgItemMsg(hwnd, DID_OK, BM_CLICK, MPFROMSHORT(TRUE), MPVOID);
	break;
      }
      break;

    default:
      break;
    }
    return 0;

  case WM_ADJUSTWINDOWPOS:
    PostMsg(hwnd, UM_STRETCH, MPVOID, MPVOID);
    break;

  case UM_STRETCH:
    {
      SWP swpC, swp, swpH;

      WinQueryWindowPos(hwnd, &swp);
      if (!(swp.fl & (SWP_HIDE | SWP_MINIMIZE))) {
	WinQueryWindowPos(WinWindowFromID(hwnd, KILL_LISTBOX), &swpC);
	WinQueryWindowPos(WinWindowFromID(hwnd, KILL_HDR), &swpH);
	WinSetWindowPos(WinWindowFromID(hwnd, KILL_LISTBOX), HWND_TOP,
			SysVal(SV_CXSIZEBORDER),
			swpC.y,
			swp.cx - (SysVal(SV_CXSIZEBORDER) * 2),
			((swp.cy - swpC.y) - (SysVal(SV_CYTITLEBAR) +
					      SysVal(SV_CYSIZEBORDER)) -
			 (swpH.cy + 8)), SWP_MOVE | SWP_SIZE);
	WinSetWindowPos(WinWindowFromID(hwnd, KILL_HDR), HWND_TOP,
			SysVal(SV_CXSIZEBORDER) + 4,
			swpC.y + ((swp.cy - swpC.y) -
				  (SysVal(SV_CYTITLEBAR) +
				   SysVal(SV_CYSIZEBORDER)) -
				  (swpH.cy + 4)), swpH.cx, swpH.cy, SWP_MOVE);
      }
    }
    return 0;

  case WM_COMMAND:
    switch (SHORT1FROMMP(mp1)) {
    case KILL_RESCAN:
      listdone = FALSE;
      if (fUseQProcStat) {
	if (_beginthread(FillKillListThread2,
			 NULL, 65536, (PVOID) & hwnd) != -1)
	  DosSleep(250L);
	else
	  WinDismissDlg(hwnd, 0);
      }
      else {
	if (_beginthread(FillKillListThread,
			 NULL, 65536, (PVOID) & hwnd) != -1)
	  DosSleep(250L);
	else
	  WinDismissDlg(hwnd, 0);
      }
      break;

    case KILL_SHOW:
    case DID_OK:
      sSelect = (USHORT) WinSendDlgItemMsg(hwnd,
					   KILL_LISTBOX,
					   LM_QUERYSELECTION,
					   MPFROMSHORT(LIT_FIRST), MPVOID);
      if (sSelect >= 0) {

	CHAR s[31], *p;
	APIRET error;

	*s = 0;
	WinSendDlgItemMsg(hwnd,
			  KILL_LISTBOX,
			  LM_QUERYITEMTEXT,
			  MPFROM2SHORT(sSelect, 30), MPFROMP(s));
	if (*s) {
	  p = s;
	  pid = strtol(s, &p, 16);
	  if (pid) {
	    if (SHORT1FROMMP(mp1) == DID_OK) {
	      error = DosKillProcess(DKP_PROCESS, pid);
	      if (error && error != ERROR_INVALID_PROCID) {
		Dos_Error(MB_CANCEL,
			  error,
			  hwnd,
			  __FILE__,
			  __LINE__, GetPString(IDS_DOSKILLFAILEDTEXT));
	      }
	      else
		WinSendDlgItemMsg(hwnd,
				  KILL_LISTBOX,
				  LM_DELETEITEM,
				  MPFROM2SHORT(sSelect, 0), MPVOID);
	    }
	    else if (!ShowSession(hwnd, pid))
	      Notify(GetPString(IDS_SORRYCANTSHOWTEXT));
	  }
	}
      }
      break;

    case DID_CANCEL:
      if (!listdone)
	Runtime_Error(pszSrcFile, __LINE__, "busy");
      else
	WinDismissDlg(hwnd, 0);
      break;

    case IDM_HELP:
      saymsg(MB_ENTER | MB_ICONASTERISK,
	     hwnd,
	     GetPString(IDS_KILLPROCHELPTITLETEXT),
	     GetPString(IDS_KILLPROCHELPTEXT));
      break;
    }
    return 0;

  case WM_CLOSE:
    if (!listdone) {
      Runtime_Error(pszSrcFile, __LINE__, "busy");
      return 0;
    }
    break;

  case WM_DESTROY:
    if (hptrIcon)
      WinDestroyPointer(hptrIcon);
    hptrIcon = (HPOINTER) 0;
    break;
  }
  return WinDefDlgProc(hwnd, msg, mp1, mp2);
}