快捷導(dǎo)航

C++基于Boost庫實現(xiàn)命令行解析

更新時間：2021年06月24日 11:22:52 作者：lyshark

Boost庫中默認(rèn)自帶了一個功能強(qiáng)大的命令行參數(shù)解析器，以往我都是自己實現(xiàn)參數(shù)解析的，今天偶爾發(fā)現(xiàn)這個好東西，就來總結(jié)一下參數(shù)解析的基本用法，該庫需要引入program_options.hpp頭文件，即可使用了

第一次嘗試

#include <iostream>
#include <boost/program_options.hpp>

// 定義命名空間
namespace opt = boost::program_options;

int main(int argc, char const *argv[])
{
	opt::options_description desc("Usage: 32位端口快速掃描器\n\n options: \n");

	desc.add_options()
		("Address,a", opt::value<std::string>()->default_value("127.0.0.1"), "輸入一個IP地址"),
		("StartPort,s", opt::value<int>()->default_value(1024), "傳入掃描起始端口"),
		("EndPort,e", opt::value<int>()->default_value(65535), "傳入掃描結(jié)束端口"),
		("Help,h", "彈出幫助菜單");

	// 解析參數(shù)將值傳遞給virtual_map
	opt::variables_map virtual_map;

	try
	{
		opt::store(opt::parse_command_line(argc, argv, desc), virtual_map);
	}
	catch (...)
	{
		std::cout << "error \n";
		return 0;
	}

	// 參數(shù)解析完畢，處理
	if (virtual_map.count("Help"))
	{
		printf("幫助");
	}
	if (virtual_map.count("Address"))
	{
		std::cout << "找到" << virtual_map["Address"].as<std::string>() << std::endl;
	}
	if (virtual_map.empty())
	{
		std::cout << "no options\n";
	}
	return 0;
}

第二次嘗試

#include <iostream>  
#include <vector>  
#include <string>
#include <boost/program_options.hpp>  

namespace opt = boost::program_options;

int main(int argc, char const *argv[])
{
	int start_port = 1024, end_port = 65535;
	std::vector<std::string> address;

	opt::options_description opt("\nUsage: 32位端口快速掃描器 Ver:1.0 \n\n options: \n");
	opt.add_options()
		("address,a", opt::value<std::vector<std::string> >()->multitoken(), "指定地址")
		("start_port,s", opt::value<int>(&start_port)->default_value(1024), "開始端口")
		("end_port,e", opt::value<int>(&end_port)->default_value(65535), "結(jié)束端口")
		("help", "幫助菜單");

	opt::variables_map vm;
	try
	{
		opt::store(parse_command_line(argc, argv, opt), vm);
	}
	catch (...){
		std::cout << "command error！\n";
		return 0;
	}

	opt::notify(vm);
	if (vm.count("help"))
	{
		std::cout << opt << std::endl;
		return 0;
	}
	if (vm.count("address") && vm.count("start_port") && vm.count("end_port"))
	{
		//遍歷選項值  
		for (auto& str : vm["address"].as<std::vector<std::string> >())
			std::cout << str << " ";
		int x = vm["start_port"].as<int>();
		std::cout << x << std::endl;
	}
	return 0;
}

最終版

#include <iostream>
#include <boost/program_options.hpp>

namespace opt = boost::program_options;

int main(int argc, char const *argv[])
{
	opt::options_description des_cmd("\n Usage: 32位端口快速掃描器 Ver:1.0 \n\n Options: \n");
	des_cmd.add_options()
		("address,a", opt::value<std::string>()->default_value("127.0.0.1"), "指定地址")
		("start_port,s", opt::value<int>()->default_value(1024), "開始端口")
		("end_port,e", opt::value<int>()->default_value(65535), "結(jié)束端口")
		("help,h", "幫助菜單");

	opt::variables_map virtual_map;
	try
	{
		opt::store(opt::parse_command_line(argc, argv, des_cmd), virtual_map);
	}
	catch (...){ return 0; }
	
	// 定義消息
	opt::notify(virtual_map);
	
	// 無參數(shù)直接返回
	if (virtual_map.empty())
	{
		return 0;
	}
	else if (virtual_map.count("help") || virtual_map.count("h"))
	{
		std::cout << des_cmd << std::endl;
		return 0;
	}
	else if (virtual_map.count("address") && virtual_map.count("start_port") && virtual_map.count("end_port"))
	{
		std::cout << "Addr = " << virtual_map["address"].as<std::string>() << std::endl;
		std::cout << "StartPort = " << virtual_map["start_port"].as<int>() << std::endl;
		std::cout << "EndPort = " << virtual_map["end_port"].as<int>() << std::endl;
	}
	else
	{
		std::cout << "option error" << std::endl;
	}
	return 0;
}

命令行下使用help輸出幫助菜單，當(dāng)傳入三個參數(shù)時，即可解析到第二個判斷上，執(zhí)行相應(yīng)的函數(shù)即可。