当前位置: 代码迷 >> 综合 >> boost 处理命令行选项参数
  详细解决方案

boost 处理命令行选项参数

热度:21   发布时间:2023-12-15 20:04:58.0
// genmac.cpp : 定义控制台应用程序的入口点。
//#define _CRTDBG_MAP_ALLOC
#include <stdlib.h>
#include <crtdbg.h>
#include "stdafx.h"
#include  <stdio.h>
#include <StrUtils.h>
#include <Utils.h>
#include <iostream>
#include <fstream>
#include <exception>
#include <boost/program_options.hpp>using namespace std;
using namespace boost;program_options::options_description bOptions("Options");
char* mPrgPath=NULL;
void showhelpMessage(const TCHAR* pvMsg=NULL) {if (pvMsg) {wcout << pvMsg << endl << endl;}cout << "This is genmac 1.0.7 that built by rocklee,ALCO MIS." << endl;cout << bOptions << endl;cout << "example: " << mPrgPath << " --from=\"01-02-03-04-05-06\" --to=\"01-02-03-04-05-ff\" --output=\"c:\\temp\\genmac.txt\"" << endl;
}
bool macgen(string pvsMacFrom , string pvsMacTo,string pvsOutput) {//char* lvlpcMacFrom = "20-D1-60-DB-4C-3B";//char* lvlpcMacTo = "20-D1-60-DB-4D-3C";if (pvsMacFrom.size() != 17 || pvsMacTo.size() != 17 ||!(pvsMacFrom.substr(2, 1) == "-" &&pvsMacFrom.substr(5, 1) == "-" && pvsMacFrom.substr(8, 1) == "-" && pvsMacFrom.substr(11, 1) == "-" && pvsMacFrom.substr(14, 1) == "-")||!(pvsMacTo.substr(2, 1) == "-" &&pvsMacTo.substr(5, 1) == "-" && pvsMacTo.substr(8, 1) == "-" && pvsMacTo.substr(11, 1) == "-" && pvsMacTo.substr(14, 1) == "-")) {showhelpMessage(_TEXT("Invalid Mac format ."));return false;}fstream _file;if (pvsOutput.size()!=0 ){_file.open(pvsOutput.data(), ios::in);if (_file){showhelpMessage(_TEXT("target file is exists."));return false;}_file.close();try{			_file.open(pvsOutput.data(), ios::out);			}catch (std::exception & e) {wstring lvwsTmp = freestyle::StrUtils::MBytesToWString(e.what());showhelpMessage(lvwsTmp.data());_file.close();return false;}}bool lvbOutputToConsole = pvsOutput.size() == 0;long long lvulMac1 = freestyle::hex_to_decimal(freestyle::StrUtils::replace_all_distinct(pvsMacFrom, "-", "").data());// atoll(freestyle::StrUtils::replace_all_distinct(string(lvlpcMac1), "-", "").data());long long lvulMac2 = freestyle::hex_to_decimal(freestyle::StrUtils::replace_all_distinct(pvsMacTo, "-", "").data());//atoll(freestyle::StrUtils::replace_all_distinct(string(lvlpcMac2), "-", "").data());int lvCC = 0;for (long long lvulItem = lvulMac1; lvulItem <= lvulMac2; ++lvulItem) {char lvlpcCC[10];sprintf_s(lvlpcCC, "%9d", ++lvCC);/*if (lvbOutputToConsole){cout << lvlpcCC << "  " << lvulItem << " ";}*/char lvlpcTmp[13];sprintf_s(lvlpcTmp, "%12llX", lvulItem);string lvsTmp(lvlpcTmp);lvsTmp.insert(2, "-");lvsTmp.insert(5, "-");lvsTmp.insert(8, "-");lvsTmp.insert(11, "-");lvsTmp.insert(14, "-");if (lvbOutputToConsole){cout << lvlpcCC << " ";cout << lvsTmp.data() << endl;}else {_file << lvsTmp.data() << endl;}}	_file.close();cout << "Total " << lvCC << " records created." << endl;return true;
}int main(int argc,char** argv)
{	_CrtSetDbgFlag(_CRTDBG_ALLOC_MEM_DF | _CRTDBG_LEAK_CHECK_DF);mPrgPath = argv[0];string lvsMacFrom, lvsMacTo,lvsOutFile;bOptions.add_options()("help", "Show help message")("from", program_options::value<string>(&lvsMacFrom), "MAC from")("to", program_options::value<string>(&lvsMacTo), "MAC to")("output", program_options::value<string>(&lvsOutFile), "output file name path");// parse program optionsprogram_options::variables_map mVMap;program_options::store(parse_command_line(argc, argv, bOptions), mVMap);program_options::notify(mVMap);// output help message if requiredif (mVMap.count("help")||mVMap.size()==0){showhelpMessage();return 1;}// process other optionif (!mVMap.count("from")){showhelpMessage(_TEXT("Missing from option."));return 1;}if (!mVMap.count("to")){showhelpMessage(_TEXT("Missing to option."));return 1;}if (macgen(lvsMacFrom, lvsMacTo, lvsOutFile)) {return 0;}else {return 1;}
}


上面演示了如何接受一个mac范围, 生成mac列表然后根据参数要求输出到指定文件或控制台. 主要留意program_options的add_options方法, 还有program_options::variables_map的count.


运行之:

C:\temp>genmac.exe
This is genmac 1.0.7 that built by rocklee,ALCO MIS.
Command Options:--help                Show help message--from arg            MAC from--to arg              MAC to--output arg          output file name path
C:\temp>genmac.exe --from="01-02-03-04-05-06"
Missing to option.This is genmac 1.0.7 that built by rocklee,ALCO MIS.
Command Options:--help                Show help message--from arg            MAC from--to arg              MAC to--output arg          output file name pathexample: genmac.exe --from="01-02-03-04-05-06" --to="01-02-03-04-05-ff" --output
="c:\temp\genmac.txt"

C:\temp>genmac.exe --from="01-02-03-04-05-06" --to="01-02-03-04-05-10"1  1-02-03-04-05-062  1-02-03-04-05-073  1-02-03-04-05-084  1-02-03-04-05-095  1-02-03-04-05-0A6  1-02-03-04-05-0B7  1-02-03-04-05-0C8  1-02-03-04-05-0D9  1-02-03-04-05-0E10  1-02-03-04-05-0F11  1-02-03-04-05-10
Total 11 records created.