Root commit for new Compass repository

This is the Alpha version of ComPASS, originally developed sometime in
2014.

1 files changed, 412 insertions, 0 deletions

diff --git a/Compass.cpp b/Compass.cpp
new file mode 100644
index 0000000..c5d2a3d
--- /dev/null
+++ b/Compass.cpp
@@ -0,0 +1,412 @@
+#include "Compass.h"

+

+Compass::Compass() {

+}

+

+Compass::~Compass() {

+}

+

+void Compass::perform(int argc, char* argv[]) {

+	// Figure out action

+	if (argc < 2) {

+		std::cerr << ERR_NO_ACTION << std::endl;

+		std::cout << "Run 'compass help' for more information." << std::endl;

+		return;

+	}

+

+	try {

+		std::string action(argv[1]);

+

+		// Need Help?

+		if (action == "help") {

+			initNoArgs(argc, argv);

+			help();

+		}

+

+		// Supported Actions

+		else if (action == "initialize") {

+			initNoArgs(argc, argv);

+			initialize();

+		}

+

+		else if (action == "create") {

+			std::string service = initOneArg(argc, argv);

+			create(service);

+		}

+

+		else if (action == "walk" && ALLOW_WALK) {

+			initNoArgs(argc, argv);

+			walk();

+		}

+

+		else if (action == "rekey") {

+			initNoArgs(argc, argv);

+			rekey();

+		}

+

+		else if (action == "get") {

+			std::string service = initOneArg(argc, argv);

+			get(service);

+		}

+

+		else if (action == "delete") {

+			std::string service = initOneArg(argc, argv);

+			_delete(service);

+		}

+

+		else if (action == "show") {

+			initNoArgs(argc, argv);

+			show("", false);

+		}

+

+		else if (action == "show1") {

+			std::string service = initOneArg(argc, argv);

+			show(service, false);

+		}

+

+		else if (action == "showlike") {

+			std::string service = initOneArg(argc, argv);

+			show(service, true);

+		}

+

+		else if (action == "random") {

+			initNoArgs(argc, argv);

+			random();

+		}

+

+		else if (action == "markreset") {

+			initNoArgs(argc, argv);

+			markReset("");

+		}

+

+		else if (action == "markreset1") {

+			std::string service = initOneArg(argc, argv);

+			markReset(service);

+		}

+

+		else if (action == "checkreset") {

+			initNoArgs(argc, argv);

+			checkReset("");

+		}

+

+		else if (action == "checkreset1") {

+			std::string service = initOneArg(argc, argv);

+			checkReset(service);

+		}

+

+		else if (action == "ruser") {

+			std::string service = initOneArg(argc, argv);

+			ruser(service);

+		}

+

+		else if (action == "rpass") {

+			std::string service = initOneArg(argc, argv);

+			rpass(service);

+		}

+

+		else {

+			std::cerr << ERR_INVALID_ACTION << std::endl;

+			std::cout << "Run 'compass help' for more information." << std::endl;

+		}

+	}

+	catch (const int e) {

+		std::cerr << ERR_GENERIC << std::endl;

+	}

+	catch (const CryptoPP::Exception& e) {

+		std::cerr << ERR_GENERIC << std::endl;

+	}

+	catch (const double e) {

+		std::cerr << ERR_INPUT << std::endl;

+	}

+	catch (const bool e) {

+		// do nothing (message was produced by called function)

+	}

+}

+

+void Compass::initNoArgs(int argc, char* argv[]) {

+	opt.parseArgv(2, argc, argv);

+}

+

+std::string Compass::initOneArg(int argc, char* argv[]) {

+	if (argc < 3) {

+		std::cerr << ERR_REQ_PARAMS_MISSING << std::endl;

+		throw false;

+	}

+

+	std::string arg(argv[2]);

+	opt.checkString(arg);

+	opt.parseArgv(3, argc, argv);

+	return arg;

+}

+

+void Compass::showResult(Credential c) {

+	std::cout << "Username is: " << c.username << std::endl;

+

+	if (opt.pp)

+		std::cout << "Password is: " << c.password << std::endl;

+	else

+		pasteClipboard(c.password, !opt.cc, false);

+}

+

+void Compass::pasteClipboard(std::string text, bool clean, bool recur) {

+	#ifdef WIN32

+

+		if (OpenClipboard(NULL)) {

+			HGLOBAL clipbuffer;

+			char* buffer;

+			EmptyClipboard();

+			const char* source = text.c_str();

+			clipbuffer = GlobalAlloc(GMEM_DDESHARE, strlen(source) + 1);

+			buffer = (char*)GlobalLock(clipbuffer);

+			strcpy(buffer, LPCSTR(source));

+			GlobalUnlock(clipbuffer);

+			SetClipboardData(CF_TEXT, clipbuffer);

+			CloseClipboard();

+			std::cout << MSG_CLIPBOARD << std::endl;

+		}

+		else {

+			std::cerr << ERR_CLIPBOARD << std::endl;

+		}

+

+	#else

+	#ifdef __APPLE__

+

+		std::string sink = "pbcopy";

+

+	#else

+

+		std::string sink = "xclip -selection c";

+

+	#endif // __APPLE__

+

+		// Mac/Linux procedure

+		std::string cmd = "printf %s '" + text + "' | " + sink;

+		system(cmd.c_str());

+		if (!recur) std::cout << MSG_CLIPBOARD << std::endl;

+		else exit(EXIT_SUCCESS);

+

+		// Cleanup

+		if (clean) {

+			pid_t pid = fork();

+			if (pid == 0) {

+				usleep(SECONDS_PSWD_CLEANUP * 1000000);

+				pasteClipboard(CLEAN_STRING, clean, true);

+			}

+			else if (pid < 0) {

+				std::cerr << WARN_CANT_CLEANUP << std::endl;

+			}

+		}

+

+	#endif // WIN32

+

+	if (!clean) std::cout << WARN_WONT_CLEANUP << std::endl;

+}

+

+void Compass::help() {

+	Help h;

+	h.dispGeneral();

+}

+

+void Compass::initialize() {

+	std::cout << "WARNING: This will destroy the keychain (if one exists) at '" << opt.dr << "'" << std::endl;

+	std::cout << "You may still abort this action by pressing CTRL-C now. If you wish to procede, create a new master password for your keychain below..." << std::endl;

+	std::cout << std::endl;

+

+	Keychain* kc = Keychain::newKeychain(opt.dr);

+	kc->saveKeychain();

+	delete kc;

+}

+

+void Compass::create(std::string service) {

+	Keychain* kc = Keychain::loadKeychain(opt.rh, opt.pt, opt.dr);

+	Credential c;

+

+	if (opt.user != "" && opt.pass != "")

+		c = kc->create(service, opt.user, opt.pass);

+

+	else if (opt.user != "")

+		c = kc->create(service, opt.user, opt.getPasswordSpec());

+

+	else if (opt.pass != "")

+		c = kc->create(service, opt.pass);

+

+	else

+		c = kc->create(service, opt.getPasswordSpec());

+

+	std::cout << "Created new credential for " << service << "(" << kc->getServiceCount(service) << ")" << std::endl;

+	showResult(c);

+

+	kc->saveKeychain();

+	delete kc;

+}

+

+void Compass::walk() {

+	Keychain* kc = Keychain::loadKeychain(opt.rh, opt.pt, opt.dr);

+	kc->walk();

+	delete kc;

+}

+

+void Compass::rekey() {

+	Keychain* kc = Keychain::loadKeychain(opt.rh, opt.pt, opt.dr);

+	Cryptor::rekey();

+

+	std::cout << "Enter a new master password for your keychain... (CTRL-C to abort)" << std::endl;

+	kc->saveKeychain();

+	delete kc;

+}

+

+void Compass::get(std::string service) {

+	Keychain* kc = Keychain::loadKeychain(opt.rh, opt.pt, opt.dr);

+	Credential c;

+

+	if (opt.cn != -1)

+		c = kc->get(service, opt.cn);

+

+	else if (opt.user != "")

+		c = kc->get(service, opt.user);

+

+	else

+		c = kc->get(service);

+

+	std::cout << "Found from " << service << "(" << kc->getServiceCount(service) << ")" << std::endl;

+	showResult(c);

+

+	if (c.reset) {

+		std::cout << "************************************************" << std::endl;

+		std::cout << "* Notice: this credential is marked for reset! *" << std::endl;

+		std::cout << "************************************************" << std::endl;

+	}

+

+	delete kc;

+}

+

+void Compass::_delete(std::string service) {

+	Keychain* kc = Keychain::loadKeychain(opt.rh, opt.pt, opt.dr);

+	bool b;

+

+	if (opt.cn != -1)

+		b = kc->_delete(service, opt.cn);

+

+	else if (opt.user != "")

+		b = kc->_delete(service, opt.user);

+

+	else

+		b = kc->_delete(service);

+

+	if (b)

+		std::cout << service << "(" << kc->getServiceCount(service) << ") has been fully removed from the keychain." << std::endl;

+	else

+		std::cout << service << "(" << kc->getServiceCount(service) << ") has been partially removed from the keychain." << std::endl;

+

+	kc->saveKeychain();

+	delete kc;

+}

+

+void Compass::show(std::string service, bool like) {

+	Keychain* kc = Keychain::loadKeychain(opt.rh, opt.pt, opt.dr);

+

+	if (service == "")

+		kc->show();

+

+	else if (like)

+		kc->showlike(service);

+

+	else

+		kc->show1(service);

+

+	delete kc;

+}

+

+void Compass::random() {

+	std::cout << Cryptor::createRandomPassword(opt.getPasswordSpec()) << std::endl;

+}

+

+void Compass::markReset(std::string service) {

+	Keychain* kc = Keychain::loadKeychain(opt.rh, opt.pt, opt.dr);

+	int i;

+

+	if (service == "")

+		i = kc->markResetAll();

+

+	else

+		i = kc->markResetOne(service);

+

+	std::cout << "Marked " << i << " credentials for reset." << std::endl;

+

+	kc->saveKeychain();

+	delete kc;

+}

+

+void Compass::checkReset(std::string service) {

+	Keychain* kc = Keychain::loadKeychain(opt.rh, opt.pt, opt.dr);

+	int i;

+

+	if (service == "")

+		i = kc->checkResetAll();

+

+	else

+		i = kc->checkResetOne(service);

+

+	std::cout << i << " credentials are marked for reset." << std::endl;

+	delete kc;

+}

+

+void Compass::ruser(std::string service) {

+	Keychain* kc = Keychain::loadKeychain(opt.rh, opt.pt, opt.dr);

+	Credential c;

+

+	if (opt.user == "") {

+		if (opt.cn == -1)

+			c = kc->ruser(service);

+

+		else

+			c = kc->ruser(service, opt.cn);

+	}

+

+	else {

+		if (opt.cn == -1)

+			c = kc->ruser(service, opt.user);

+

+		else

+			c = kc->ruser(service, opt.user, opt.cn);

+	}

+

+	std::cout << "Altered from " << service << "(" << kc->getServiceCount(service) << ")" << std::endl;

+	showResult(c);

+

+	kc->saveKeychain();

+	delete kc;

+}

+

+void Compass::rpass(std::string service) {

+	Keychain* kc = Keychain::loadKeychain(opt.rh, opt.pt, opt.dr);

+	Credential c;

+

+	if (opt.pass == "") {

+		if (opt.user == "" && opt.cn == -1)

+			c = kc->rpass(service, opt.getPasswordSpec());

+

+		else if (opt.cn != -1)

+			c = kc->rpass(service, opt.getPasswordSpec(), opt.cn);

+

+		else

+			c = kc->rpass(service, opt.getPasswordSpec(), opt.user);

+	}

+

+	else {

+		if (opt.user == "" && opt.cn == -1)

+			c = kc->rpass(service, opt.pass);

+

+		else if (opt.cn != -1)

+			c = kc->rpass(service, opt.pass, opt.cn);

+

+		else

+			c = kc->rpass(service, opt.pass, opt.user);

+	}

+

+	std::cout << "Altered from " << service << "(" << kc->getServiceCount(service) << ")" << std::endl;

+	showResult(c);

+

+	kc->saveKeychain();

+	delete kc;

+}