SimpleYggGen-CPP/dev/basefiles.cpp

struct option 
{
	int proc = 0;
	int mode = 0;
	int log  = 0;
	int high = 0;
	std::string str_search;
	std::string rgx_search;
	std::string outputfile;
};

option conf;

int config()
{
	std::ifstream conffile ("sygcpp.conf");

	if(!conffile) // проверка наличия конфига
	{
		std::cout << " Configuration file not found... ";
		conffile.close();
		std::ofstream newconf ("sygcpp.conf"); // создание конфига
		if(!newconf)
		{
			std::cerr << "CREATION FAILED" << std::endl;
			return -1;
		}
		newconf << "####################################################################\n"
				<< "#               SimpleYggGen C++ configuration file.               #\n"
				<< "# If you have some errors, try delete this file and run SYG again. #\n"
				<< "####################################################################\n\n"
				<< "* Count of thread: 1\n\n"
				<< "  0 - by pattern, 1 - high address, 2 - search by pattern & high,\n"
				<< "  3 - regexp, 4 - search by regexp & high.\n"
				<< "* Mining option: 1\n\n"
				<< "  0 - console output only, 1 - log to file.\n"
				<< "* Logging mode: 1\n\n"
				<< "  High address search. Parameter is set in hexadecimal (0-9, a-f).\n"
				<< "* Start position (2xx): 14\n\n"
				<< "  Used when \"Mining mode\" set as 0 or 2.\n"
				<< "* Pattern: ::\n\n"
				<< "  Used when \"Mining mode\" set as 3 or 4. Extended grep type.\n"
				<< "* Regexp: ^2.*.(1:ace|ace:1)$";
		newconf.close();
		
		std::ifstream conffile ("sygcpp.conf");
		if(conffile)
			std::cout << "CREATED" << std::endl;
		config();
		return 0;
	} else { // чтение конфигурации
		std::string str_temp_read;
		std::string str_read;
		while(getline(conffile, str_temp_read))
			str_read += str_temp_read;
		conffile.close();
		
		struct check 
		{
			bool proc = false;
			bool mode = false;
			bool log  = false;
			bool high = false;
			bool str_search = false;
			bool rgx_search = false;
			
			bool ok()
			{
				if(proc & mode & log & high & str_search & rgx_search)
					return true;
				return false;
			}
		};
		check complete;
		
		std::istringstream ss_input(str_read); // чтение конфига
		while(!ss_input.eof())
		{
			ss_input >> str_temp_read;
			if(str_temp_read == "thread:") // поиск параметра по предыдущему слову
			{
				ss_input >> conf.proc; // запись в соответствующую переменную
				if(ss_input.fail())
				{
					std::cerr << " Count of thread value incorrect." << std::endl;
					return -2;
				}
				complete.proc = true;
			}
			if(str_temp_read == "option:")
			{
				ss_input >> conf.mode;
				if(ss_input.fail() || (conf.mode > 4 || conf.mode < 0))
				{
					std::cerr << " Mining option value incorrect." << std::endl;
					return -3;
				}
				complete.mode = true;
			}
			if(str_temp_read == "mode:")
			{
				ss_input >> conf.log;
				if(ss_input.fail() || (conf.log != 0 && conf.log != 1))
				{
					std::cerr << " Logging mode value incorrect." << std::endl;
					return -4;
				}
				complete.log = true;
			}
			if(str_temp_read == "(2xx):")
			{
				ss_input >> std::hex >> conf.high;
				if(ss_input.fail())
				{
					std::cerr << " Start position value incorrect." << std::endl;
					return -5;
				}
				complete.high = true;
			}
			if(str_temp_read == "Pattern:")
			{
				ss_input >> conf.str_search;
				if(ss_input.fail())
				{
					std::cerr << " Pattern value incorrect." << std::endl;
					return -6;
				}
				complete.str_search = true;
			}
			if(str_temp_read == "Regexp:")
			{
				ss_input >> conf.rgx_search;
				if(ss_input.fail())
				{
					std::cerr << " Regexp value incorrect." << std::endl;
					return -7;
				}
				complete.rgx_search = true;
			}
		}
		if(!complete.ok())
		{
			std::cerr << " Corrupted configuration file. Some parameters not found." << std::endl;
			return -8;
		}

		unsigned int processor_count = std::thread::hardware_concurrency(); // кол-во процессоров
		if (conf.proc > (int)processor_count)
			conf.proc = (int)processor_count;
		countsize = 800 << __bsrq(conf.proc);
	}
	return 0;
}

void DisplayConfig()
{
	// вывод конфигурации на экран
	std::cout << " Threads: " << conf.proc << ", ";

	if(conf.mode == 0)
		std::cout << "search by pattern (" << conf.str_search << "), ";
	else if(conf.mode == 1)
		std::cout << "search high addresses (2" << std::setw(2) << std::setfill('0') << 
			std::hex << conf.high << std::dec << "+), ";
	else if(conf.mode == 2)
		std::cout << "search by pattern & high (" << conf.str_search << " & 2" << 
			std::setw(2) << std::setfill('0') << std::hex << conf.high << std::dec <<"+), ";
	else if(conf.mode == 3)
		std::cout << "search by regexp (" << conf.rgx_search << "), ";
	else if(conf.mode == 4)
		std::cout << "search by regexp & high (" << conf.rgx_search << " & 2" << 
			std::setw(2) << std::setfill('0') << std::hex << conf.high << std::dec << "+), ";

	if(conf.log)
		std::cout << "logging to text file.";
	else
		std::cout << "console log only.";

	std::cout << std::endl << std::endl;
}

void testoutput()
{
	if(conf.log) // проверка включено ли логирование
	{
		if(conf.mode == 0)
			conf.outputfile = "syg-pattern.txt";
		else if(conf.mode == 1)
			conf.outputfile = "syg-high.txt";
		else if(conf.mode == 2)
			conf.outputfile = "syg-pattern-high.txt";
		else if(conf.mode == 3)
			conf.outputfile = "syg-regexp.txt";
		else if(conf.mode == 4)
			conf.outputfile = "syg-regexp-high.txt";

		std::ifstream test(conf.outputfile);
		if(!test) // проверка наличия выходного файла
		{
			test.close();
			std::ofstream output(conf.outputfile);
			output << "**************************************************************************\n"
			       << "Change EncryptionPublicKey and EncryptionPrivateKey to your yggdrasil.conf\n"
			       << "Windows: C:\\ProgramData\\Yggdrasil\\yggdrasil.conf\n"
			       << "Debian: /etc/yggdrasil.conf\n"
			       << "**************************************************************************\n";
			output.close();
		} else test.close();
	}
}