DictionaryGenerator/main.cpp at master · jpaav/DictionaryGenerator · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
#include "Target.hpp"
#include "dataStructures.hpp"
#include <iostream>
#include <ncurses.h>
#include <fstream>
#include <boost/archive/tmpdir.hpp>
#include <boost/archive/text_oarchive.hpp>
#include <boost/archive/text_iarchive.hpp>
#include <boost/serialization/base_object.hpp>
#include <boost/serialization/optional.hpp>
#include <boost/serialization/map.hpp>
#include <boost/serialization/shared_ptr.hpp>
#include <boost/serialization/list.hpp>
#include <boost/serialization/string.hpp>
#include <boost/serialization/version.hpp>
#include <boost/serialization/split_member.hpp>

//BOOST_CLASS_VERSION(Target, 0)

void save_profile(Target &t, std::string filename){
    // make an archive
    std::ofstream ofs(filename.c_str());
    boost::archive::text_oarchive oa(ofs);
    oa << t;
    std::cout << "Profile saved at " << filename << std::endl;
}

void load_profile(Target &t, std::string filename)
{
	std::ifstream ifs(filename.c_str());
	boost::archive::text_iarchive ia(ifs);
	ia >> t;
	std::cout << "Profile loaded from " << filename << std::endl;
}

int main(int argc, char const *argv[])
{
	if (argc < 2)
	{
		std::cout << "You need to either specify an existing file path or name a new one to be created.\n";
		return 1;
	}

	std::string profilePath (argv[1]);
	std::ifstream f;
	Target profile;
	bool quit = false;

	f.open(profilePath.c_str(), std::ios::binary);
	f.seekg(0, std::ios::end);
	int length = f.tellg();
	f.close();
	if (length != 0)
	{
		load_profile(profile, profilePath);
		std::cout << "TestVal: " << profile.getTestVal() << std::endl;
	}
	while(!quit)
	{
		//std::cout << "\n______________\n\nTo edit or add to profile data, press \"e\"\nTo generate a wordlist from this profile, press \"g\"\nTo quit, press \"q\"\n";
		char input = getchar();
		if (input == 'e')
		{
			profile.enterData();
			save_profile(profile, profilePath);
		}
		else if(input == 'g')
		{
			profile.generateList();
		}
		else if(input == 'q')
		{
			quit = true;
		}
		/*else
		{
			std::cout << "That wasn't one of your choices... try again?\n";
		}*/
	}

}