cmdline.cpp

来自「HDCP Key加密程序」· C++ 代码 · 共 82 行

// CmdLine.cpp: implementation of the CCmdLine class.
//
//////////////////////////////////////////////////////////////////////

#include "stdafx.h"
#include "PWKeyEncrypt.h"
#include "CmdLine.h"

#ifdef _DEBUG
#undef THIS_FILE
static char THIS_FILE[]=__FILE__;
#define new DEBUG_NEW
#endif

//////////////////////////////////////////////////////////////////////
// Construction/Destruction
//////////////////////////////////////////////////////////////////////

CCmdLine::CCmdLine()
{
	m_strIniFile		=_T("");
	m_strOutputDir		=_T("");
	m_strKeyFile		=_T("");
	m_strFileFormat		=_T("");

	m_nStartKey			=0;
	m_nNumKeys			=0;

	m_bEncrypt			=TRUE;
	m_bGenerate			=FALSE;
}

CCmdLine::~CCmdLine()
{

}

void CCmdLine::ParseParam( LPCTSTR lpszParam, BOOL bFlag, BOOL bLast )
{
	CCommandLineInfo::ParseParam(lpszParam, bFlag, bLast);

	if (bFlag && lpszParam[0]=='i')
	{
		m_strIniFile = &lpszParam[1];
	}

	if (bFlag && lpszParam[0]=='f')
	{
		m_strKeyFile = &lpszParam[1];
	}

	if (bFlag && lpszParam[0]=='d')
	{
		m_strOutputDir = &lpszParam[1];
	}

	if (bFlag && lpszParam[0]=='o')
	{
		m_strFileFormat = &lpszParam[1];
	}

	if (bFlag && lpszParam[0]=='s')
	{
		m_nStartKey = atoi(&lpszParam[1]);
	}

	if (bFlag && lpszParam[0]=='n')
	{
		m_nNumKeys = atoi(&lpszParam[1]);
	}

	if (bFlag && lpszParam[0]=='e')
	{
		m_bEncrypt = (lpszParam[1]=='t'||lpszParam[1]=='T');
	}

	if (bFlag && lpszParam[0]=='g')
	{
		m_bGenerate = TRUE;
	}
}