// PSLThreads.cpp : Implementation of CPSLThreads

#include "stdafx.h"
#include "PSLThread.h"
#include "PSLThreads.h"

CPSLThreads::CPSLThreads()
{
}

HRESULT CPSLThreads::OnIndexOutOfRange()
{
   return MakeException(EX_INDEXOUTOFRANGE);
}

HRESULT CPSLThreads::FinalConstruct()
{
   InternalUpdate();
   return S_OK;
}

void CPSLThreads::FinalRelease()
{
}

void CPSLThreads::InternalUpdate()
{
   CCritSecLock cs(m_csCollection);
   m_coll.clear();
   HANDLE hSnapshot = ::CreateToolhelp32Snapshot(TH32CS_SNAPTHREAD, 0);
   if(hSnapshot != INVALID_HANDLE_VALUE)
   {
      THREADENTRY32 te;
      te.dwSize = sizeof(te);
      if(::Thread32First(hSnapshot, &te))
      {
         DWORD dwProcessID = ::GetCurrentProcessId();
         do
         {
            if(te.dwSize >= FIELD_OFFSET(THREADENTRY32, th32OwnerProcessID) + sizeof(te.th32OwnerProcessID) && te.th32OwnerProcessID == dwProcessID)
            {
               CComObject<CPSLThread> * pThread = NULL;
               if(CComObject<CPSLThread>::CreateInstance(&pThread) == S_OK)
               {
                  pThread->Initialize(&te);
                  m_coll.push_back(CComPtr<IPSLThread>(pThread));
               }
            }
            te.dwSize = sizeof(te);
         }
         while(::Thread32Next(hSnapshot, &te));
      }
      ::CloseHandle(hSnapshot);
   }
}

////////////////////////////////////////////////////////////////////////
// Interface Implementation;
////////////////////////////////////////////////////////////////////////

STDMETHODIMP CPSLThreads::Update()
{
   PSL_BEGIN

   InternalUpdate();

   PSL_END
}

STDMETHODIMP CPSLThreads::Find(long ThreadID, IPSLThread ** ppValue)
{
   PSL_BEGIN

   *ppValue = NULL;
   CCritSecLock cs(m_csCollection);
   for(ThreadsType::iterator i = m_coll.begin();i != m_coll.end();i ++)
   {
      long ID = 0;
      if(i->m_T->get_ThreadID(&ID) == S_OK && ID == ThreadID)
      {
         IPSLThread * pThread = i->m_T;
         pThread->AddRef();
         *ppValue = pThread;
         break;
      }
   }

   PSL_END
}

STDMETHODIMP CPSLThreads::FindCurrent(IPSLThread ** ppValue)
{
   PSL_BEGIN

   *ppValue = NULL;
   CCritSecLock cs(m_csCollection);
   for(ThreadsType::iterator i = m_coll.begin();i != m_coll.end();i ++)
   {
      VARIANT_BOOL IsCurrent = VARIANT_FALSE;
      if(i->m_T->get_IsCurrent(&IsCurrent) == S_OK && IsCurrent)
      {
         IPSLThread * pThread = i->m_T;
         pThread->AddRef();
         *ppValue = pThread;
         break;
      }
   }

   PSL_END
}