#include "stdafx.h"
#import "EPRCmdAPI.tlb" rename("value", "val")

using namespace EPRCmdAPI;

int _tmain(int argc, _TCHAR* argv[])
{
	_bstr_t bstrCaptureName = L"Capture";
	if (argc > 1)
	{
		bstrCaptureName = argv[1];
	}

	HRESULT hr;

	hr = CoInitializeEx(NULL, 0);
	if ( FAILED( hr ) ) {
		wprintf( L"Failed to initialize COM.rn" );
		return 1;
	}

	// create a block inside the coinitialize
	{
		CComPtr spCmdMgr;
		hr = spCmdMgr.CoCreateInstance( _T( "EPRCmdAPI.WPRemoteCmdMgr" ) );
		if (FAILED(hr))
		{
			wprintf( L"Failed to initialize OmniScriptrn" );
			return -1;
		}

		hr = spCmdMgr->Connect( L"localhost", 6367, 30000 );
		if (FAILED(hr))
		{
			wprintf( L"Failed to connect to OmniEnginern" );
			return -1;
		}

		//IPeekRemoteAdapterListPtr spRemoteAdapterList;
		//spRemoteAdapterList = spCmdMgr->GetRemoteAdapterList();

		IPeekRemoteCaptureListPtr spCaptureListPtr;
		spCaptureListPtr = spCmdMgr->GetCaptureList( );

		BOOL bExists = FALSE;
		LONG lCount = spCaptureListPtr->GetCount();
		IPeekRemoteCaptureInfoPtr spCaptureInfo;
		_bstr_t bstrCapture;

		for (long i=0; i < lCount; i++ )
		{
			spCaptureInfo = spCaptureListPtr->GetItem( i );
			_bstr_t bstrCaptureName2 = spCaptureInfo->GetName();
			
			if (bstrCaptureName == bstrCaptureName2)
			{
				bstrCapture = spCaptureInfo->GetID();
				bExists = TRUE;
				break;
			}
		}

		if (bExists == FALSE)
		{
			bstrCapture = spCmdMgr->CreateNewCapture( 
				bstrCaptureName, TRUE, FALSE, 0, 0, 0, 0, 0, 
				FALSE, 0, 512, L"Local Area Connection" );

				hr = spCmdMgr->StartCapture( bstrCapture );
				if (FAILED(hr))
				{
					wprintf( L"Failed to start capturern" );
					return -1;
				}
		}

		IPeekRemoteNodeListPtr spRemoteNodeListPtr;
		spRemoteNodeListPtr = spCmdMgr->GetNodeListAll( bstrCapture );
		long nCount = spRemoteNodeListPtr->GetCount();

		_tprintf( _T("IP                   Packets In     Packets Out        Bytes In       Bytes Outrn") );

		for (int i=0; i < nCount; i++)
		{
			IPeekSimpleNodeStatPtr spNode = spRemoteNodeListPtr->GetItem( i );
			_bstr_t bstrIP = spNode->GetIPAddr();
			unsigned long nPacketsReceived = spNode->GetPacketsReceived();
			unsigned long nPacketsSent = spNode->GetPacketsSent();
			double nBytesReceived = spNode->GetBytesReceived();
			double nBytesSent = spNode->GetBytesSent();

			_tprintf( _T("%-15s %15u %15u %15.0f %15.0frn"), (LPCTSTR)bstrIP, 
				nPacketsReceived,
				nPacketsSent,
				nBytesReceived,
				nBytesSent);
		}

	}

	CoUninitialize();


	return 0;
}