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

using namespace EPRCmdAPI;

int _tmain(int argc, _TCHAR* argv[])
{
	if (argc == 1)
	{
		wprintf( L"usage: flowstats capturename [adaptername] [hostname] [username] [password]rn" );
		return 0;
	}

	_bstr_t bstrCaptureName = L"Capture";
	if (argc > 1) bstrCaptureName = argv[1];

	CComBSTR bstrAdapterName = L"Local Area Connection";
	if (argc > 2) bstrAdapterName = argv[2];

	CComBSTR bstrHostName;
	if (argc > 3) bstrHostName = argv[3];

	CComBSTR bstrUserName;
	if (argc > 4) bstrUserName = argv[4];

	CComBSTR bstrPassword;
	if (argc > 5) bstrPassword = argv[5];

	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;
		}

		if (bstrHostName.Length() == 0)
		{
			hr = spCmdMgr->Connect( L"localhost", 6367, 30000 );
		}
		else
		{
			CComBSTR bstrAuth( L"" );
			CComBSTR bstrDomain( L"" );
			hr = spCmdMgr->ConnectWithCred( BSTR( bstrHostName ), 6367, 
				BSTR( bstrAuth ), 
				BSTR( bstrDomain ), 
				BSTR( bstrUserName ), BSTR( bstrPassword ),
				30000 );
		}

		if (FAILED(hr))
		{
			wprintf( L"Failed to connect to OmniEnginern" );
			return -1;
		}
		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, BSTR( bstrAdapterName ) );

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

		IPeekConversationListPtr spConversationListPtr;
		spConversationListPtr = spCmdMgr->GetConversationListAll( bstrCapture );
		long nCount = spConversationListPtr->GetCount();

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

		for (int i=0; i < nCount; i++)
		{
			IPeekConversationStatPtr spFlow = spConversationListPtr->GetItem( i );
			_bstr_t bstrSrcIP = spFlow->GetSrcIPAddress();
			_bstr_t bstrDstIP = spFlow->GetDestIPAddress();
			double nPacketsReceived = spFlow->GetPacketsReceived();
			double nPacketsSent = spFlow->GetPacketsSent();
			double nBytesReceived = spFlow->GetBytesReceived();
			double nBytesSent = spFlow->GetBytesSent();

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

	}

	CoUninitialize();
	return 0;
}