changepasswordtask.cpp

#include "changepasswordtask.h"

#include "signetcliapplication.h"


extern "C" {
#include "signetdev/host/signetdev.h"
}

#include <string>
#include <iostream>

void changePasswordTask::help()
{
	std::cout << "Change the device's master password" << std::endl
		  << std::endl
		  << "Usage: signet-cli change-password"<< std::endl << std::endl;
}

changePasswordTask::changePasswordTask()
{
}

bool changePasswordTask::canStart()
{
	return true;
}

bool changePasswordTask::start()
{
	int token;
	::signetdev_startup(NULL, &token);
	return true;
}

void changePasswordTask::cmdResponse(void *cb_user_param, int cmd_token, int cmd, int end_device_state, int messages_remaining, int resp_code, void *resp_data)
{
	Q_UNUSED(cb_user_param);
	Q_UNUSED(cmd_token);
	Q_UNUSED(messages_remaining);
	Q_UNUSED(end_device_state);
	int token;
	switch(cmd) {
	case SIGNETDEV_CMD_STARTUP: {
		signetdev_startup_resp_data *resp = (signetdev_startup_resp_data *) resp_data;
		std::cout << "Enter old password:";
		std::cout.flush();
		std::string str;
		std::getline(std::cin, str);
		QString oldPassword(str.c_str());
		std::cout << "Enter new password:";
		std::cout.flush();
		std::getline(std::cin, str);
		QString newPassword(str.c_str());
		std::cout << "Repeat new password:";
		std::cout.flush();
		std::getline(std::cin, str);
		QString newPasswordRepeat(str.c_str());

		if (newPassword.compare(newPasswordRepeat)) {
			std::cout << "New passwords don't match" << std::endl;
			QCoreApplication::quit();
			return;
		}

		SignetCLIApplication *app = SignetCLIApplication::get();
		u8 oldKey[LOGIN_KEY_SZ];
		u8 newKey[LOGIN_KEY_SZ];
		std::cout << "Please wait while authentication keys are being generated..." << std::endl;
		app->generateKey(oldPassword, oldKey, resp->hashfn, resp->salt);
		app->generateKey(newPassword, newKey, resp->hashfn, resp->salt);
		::signetdev_change_master_password(NULL, &token,
						   oldKey, LOGIN_KEY_SZ,
						   newKey, LOGIN_KEY_SZ,
						   resp->hashfn, HASH_FN_SZ,
						   resp->salt, SALT_SZ_V2);
		std::cout << "Press and hold the device button to change your master password..." << std::endl;
	} break;
	case SIGNETDEV_CMD_CHANGE_MASTER_PASSWORD: {
		if (resp_code != OKAY) {
			switch (resp_code) {
			case BAD_PASSWORD:
				std::cout << "Change password failed. Old password incorrect" << std::endl;
				break;
			default:
				std::cout << "Change password failed. Code " << resp_code << std::endl;
				break;
			}
			QCoreApplication::quit();
			return;
		} else {
			std::cout << "Password changed successfully" << std::endl;
			QCoreApplication::quit();
		}
		} break;
	}
}