371 lines
		
	
	
	
		
			11 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
			
		
		
	
	
			371 lines
		
	
	
	
		
			11 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
/*
 | 
						|
This file is part of Telegram Desktop,
 | 
						|
the official desktop version of Telegram messaging app, see https://telegram.org
 | 
						|
 | 
						|
Telegram Desktop is free software: you can redistribute it and/or modify
 | 
						|
it under the terms of the GNU General Public License as published by
 | 
						|
the Free Software Foundation, either version 3 of the License, or
 | 
						|
(at your option) any later version.
 | 
						|
 | 
						|
It is distributed in the hope that it will be useful,
 | 
						|
but WITHOUT ANY WARRANTY; without even the implied warranty of
 | 
						|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
 | 
						|
GNU General Public License for more details.
 | 
						|
 | 
						|
Full license: https://github.com/telegramdesktop/tdesktop/blob/master/LICENSE
 | 
						|
Copyright (c) 2014 John Preston, https://desktop.telegram.org
 | 
						|
*/
 | 
						|
#include "stdafx.h"
 | 
						|
#include "lang.h"
 | 
						|
#include "style.h"
 | 
						|
 | 
						|
#include "gui/filedialog.h"
 | 
						|
#include "boxes/confirmbox.h"
 | 
						|
 | 
						|
#include "application.h"
 | 
						|
 | 
						|
#include "intro/intropwdcheck.h"
 | 
						|
#include "intro/intro.h"
 | 
						|
 | 
						|
IntroPwdCheck::IntroPwdCheck(IntroWidget *parent) : IntroStage(parent),
 | 
						|
errorAlpha(0),
 | 
						|
_next(this, lang(lng_intro_submit), st::btnIntroNext),
 | 
						|
_salt(parent->getPwdSalt()),
 | 
						|
_hasRecovery(parent->getHasRecovery()),
 | 
						|
_hint(parent->getPwdHint()),
 | 
						|
_pwdField(this, st::inpIntroPassword, lang(lng_signin_password)),
 | 
						|
_codeField(this, st::inpIntroPassword, lang(lng_signin_code)),
 | 
						|
_toRecover(this, lang(lng_signin_recover)),
 | 
						|
_toPassword(this, lang(lng_signin_try_password)),
 | 
						|
_reset(this, lang(lng_signin_reset_account), st::btnRedLink),
 | 
						|
sentRequest(0) {
 | 
						|
	setVisible(false);
 | 
						|
	setGeometry(parent->innerRect());
 | 
						|
 | 
						|
	connect(&_next, SIGNAL(clicked()), this, SLOT(onSubmitPwd()));
 | 
						|
	connect(&checkRequest, SIGNAL(timeout()), this, SLOT(onCheckRequest()));
 | 
						|
	connect(&_toRecover, SIGNAL(clicked()), this, SLOT(onToRecover()));
 | 
						|
	connect(&_toPassword, SIGNAL(clicked()), this, SLOT(onToPassword()));
 | 
						|
	connect(&_pwdField, SIGNAL(changed()), this, SLOT(onInputChange()));
 | 
						|
	connect(&_codeField, SIGNAL(changed()), this, SLOT(onInputChange()));
 | 
						|
	connect(&_reset, SIGNAL(clicked()), this, SLOT(onReset()));
 | 
						|
 | 
						|
	_pwdField.setEchoMode(QLineEdit::Password);
 | 
						|
 | 
						|
	if (!_hint.isEmpty()) {
 | 
						|
		_hintText.setText(st::introFont, lng_signin_hint(lt_password_hint, _hint));
 | 
						|
	}
 | 
						|
	_codeField.hide();
 | 
						|
	_toPassword.hide();
 | 
						|
	_toRecover.show();
 | 
						|
	_reset.hide();
 | 
						|
 | 
						|
	setMouseTracking(true);
 | 
						|
}
 | 
						|
 | 
						|
void IntroPwdCheck::paintEvent(QPaintEvent *e) {
 | 
						|
	bool trivial = (rect() == e->rect());
 | 
						|
 | 
						|
	QPainter p(this);
 | 
						|
	if (!trivial) {
 | 
						|
		p.setClipRect(e->rect());
 | 
						|
	}
 | 
						|
	if (trivial || e->rect().intersects(textRect)) {
 | 
						|
		p.setFont(st::introHeaderFont->f);
 | 
						|
		p.drawText(textRect, lang(lng_signin_title), style::al_top);
 | 
						|
		p.setFont(st::introFont->f);
 | 
						|
		p.drawText(textRect, lang(_pwdField.isHidden() ? lng_signin_recover_desc : lng_signin_desc), style::al_bottom);
 | 
						|
	}
 | 
						|
	if (_pwdField.isHidden()) {
 | 
						|
		if (!_emailPattern.isEmpty()) {
 | 
						|
			p.drawText(QRect(textRect.x(), _pwdField.y() + _pwdField.height() + st::introFinishSkip, textRect.width(), st::introFont->height), _emailPattern, style::al_top);
 | 
						|
		}
 | 
						|
	} else if (!_hint.isEmpty()) {
 | 
						|
		_hintText.drawElided(p, _pwdField.x(), _pwdField.y() + _pwdField.height() + st::introFinishSkip, _pwdField.width(), 1, style::al_top);
 | 
						|
	}
 | 
						|
	if (animating() || error.length()) {
 | 
						|
		p.setOpacity(errorAlpha.current());
 | 
						|
 | 
						|
		QRect errRect((width() - st::introErrWidth) / 2, (_pwdField.y() + _pwdField.height() + st::introFinishSkip + st::introFont->height + _next.y() - st::introErrHeight) / 2, st::introErrWidth, st::introErrHeight);
 | 
						|
		p.setFont(st::introErrFont->f);
 | 
						|
		p.setPen(st::introErrColor->p);
 | 
						|
		p.drawText(errRect, error, QTextOption(style::al_center));
 | 
						|
 | 
						|
		p.setOpacity(1);
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
void IntroPwdCheck::resizeEvent(QResizeEvent *e) {
 | 
						|
	if (e->oldSize().width() != width()) {
 | 
						|
		_next.move((width() - _next.width()) / 2, st::introBtnTop);
 | 
						|
		_pwdField.move((width() - _pwdField.width()) / 2, st::introTextTop + st::introTextSize.height() + st::introCountry.top);
 | 
						|
		_codeField.move((width() - _codeField.width()) / 2, st::introTextTop + st::introTextSize.height() + st::introCountry.top);
 | 
						|
		_toRecover.move(_next.x() + (_pwdField.width() - _toRecover.width()) / 2, _next.y() + _next.height() + st::introFinishSkip);
 | 
						|
		_toPassword.move(_next.x() + (_pwdField.width() - _toPassword.width()) / 2, _next.y() + _next.height() + st::introFinishSkip);
 | 
						|
		_reset.move((width() - _reset.width()) / 2, _toRecover.y() + _toRecover.height() + st::introFinishSkip);
 | 
						|
	}
 | 
						|
	textRect = QRect((width() - st::introTextSize.width()) / 2, st::introTextTop, st::introTextSize.width(), st::introTextSize.height());
 | 
						|
}
 | 
						|
 | 
						|
void IntroPwdCheck::showError(const QString &err) {
 | 
						|
	if (!animating() && err == error) return;
 | 
						|
 | 
						|
	if (err.length()) {
 | 
						|
		error = err;
 | 
						|
		errorAlpha.start(1);
 | 
						|
	} else {
 | 
						|
		errorAlpha.start(0);
 | 
						|
	}
 | 
						|
	anim::start(this);
 | 
						|
}
 | 
						|
 | 
						|
bool IntroPwdCheck::animStep(float64 ms) {
 | 
						|
	float64 dt = ms / st::introErrDuration;
 | 
						|
 | 
						|
	bool res = true;
 | 
						|
	if (dt >= 1) {
 | 
						|
		res = false;
 | 
						|
		errorAlpha.finish();
 | 
						|
		if (!errorAlpha.current()) {
 | 
						|
			error = "";
 | 
						|
		}
 | 
						|
	} else {
 | 
						|
		errorAlpha.update(dt, st::introErrFunc);
 | 
						|
	}
 | 
						|
	update();
 | 
						|
	return res;
 | 
						|
}
 | 
						|
 | 
						|
void IntroPwdCheck::activate() {
 | 
						|
	show();
 | 
						|
	if (_pwdField.isHidden()) {
 | 
						|
		_codeField.setFocus();
 | 
						|
	} else {
 | 
						|
		_pwdField.setFocus();
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
void IntroPwdCheck::deactivate() {
 | 
						|
	hide();
 | 
						|
}
 | 
						|
 | 
						|
void IntroPwdCheck::stopCheck() {
 | 
						|
	checkRequest.stop();
 | 
						|
}
 | 
						|
 | 
						|
void IntroPwdCheck::onCheckRequest() {
 | 
						|
	int32 status = MTP::state(sentRequest);
 | 
						|
	if (status < 0) {
 | 
						|
		int32 leftms = -status;
 | 
						|
		if (leftms >= 1000) {
 | 
						|
			MTP::cancel(sentRequest);
 | 
						|
			sentRequest = 0;
 | 
						|
			if (!_pwdField.isEnabled()) {
 | 
						|
				_pwdField.setDisabled(false);
 | 
						|
				_codeField.setDisabled(false);
 | 
						|
				activate();
 | 
						|
			}
 | 
						|
		}
 | 
						|
	}
 | 
						|
	if (!sentRequest && status == MTP::RequestSent) {
 | 
						|
		stopCheck();
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
void IntroPwdCheck::pwdSubmitDone(bool recover, const MTPauth_Authorization &result) {
 | 
						|
	sentRequest = 0;
 | 
						|
	stopCheck();
 | 
						|
	if (recover) {
 | 
						|
		cSetPasswordRecovered(true);
 | 
						|
	}
 | 
						|
	_pwdField.setDisabled(false);
 | 
						|
	_codeField.setDisabled(false);
 | 
						|
	const MTPDauth_authorization &d(result.c_auth_authorization());
 | 
						|
	if (d.vuser.type() != mtpc_userSelf) { // wtf?
 | 
						|
		showError(lang(lng_server_error));
 | 
						|
		return;
 | 
						|
	}
 | 
						|
	intro()->finish(d.vuser);
 | 
						|
}
 | 
						|
 | 
						|
bool IntroPwdCheck::pwdSubmitFail(const RPCError &error) {
 | 
						|
	sentRequest = 0;
 | 
						|
	stopCheck();
 | 
						|
	_pwdField.setDisabled(false);
 | 
						|
	_codeField.setDisabled(false);
 | 
						|
	const QString &err = error.type();
 | 
						|
	if (err == "PASSWORD_HASH_INVALID") {
 | 
						|
		showError(lang(lng_signin_bad_password));
 | 
						|
		_pwdField.notaBene();
 | 
						|
		return true;
 | 
						|
	} else if (err == "PASSWORD_EMPTY") {
 | 
						|
		intro()->onIntroBack();
 | 
						|
	} else if (err.startsWith(qsl("FLOOD_WAIT_"))) {
 | 
						|
		showError(lang(lng_flood_error));
 | 
						|
		_pwdField.notaBene();
 | 
						|
		return true;
 | 
						|
	}
 | 
						|
	if (cDebug()) { // internal server error
 | 
						|
		showError(err + ": " + error.description());
 | 
						|
	} else {
 | 
						|
		showError(lang(lng_server_error));
 | 
						|
	}
 | 
						|
	_pwdField.setFocus();
 | 
						|
	return false;
 | 
						|
}
 | 
						|
 | 
						|
bool IntroPwdCheck::codeSubmitFail(const RPCError &error) {
 | 
						|
	sentRequest = 0;
 | 
						|
	stopCheck();
 | 
						|
	_pwdField.setDisabled(false);
 | 
						|
	_codeField.setDisabled(false);
 | 
						|
	const QString &err = error.type();
 | 
						|
	if (err == "PASSWORD_EMPTY") {
 | 
						|
		intro()->onIntroBack();
 | 
						|
		return true;
 | 
						|
	} else if (err == "PASSWORD_RECOVERY_NA") {
 | 
						|
		recoverStartFail(error);
 | 
						|
		return true;
 | 
						|
	} else if (err == "PASSWORD_RECOVERY_EXPIRED") {
 | 
						|
		_emailPattern = QString();
 | 
						|
		onToPassword();
 | 
						|
		return true;
 | 
						|
	} else if (err == "CODE_INVALID") {
 | 
						|
		showError(lang(lng_signin_wrong_code));
 | 
						|
		_codeField.notaBene();
 | 
						|
		return true;
 | 
						|
	} else if (err.startsWith(qsl("FLOOD_WAIT_"))) {
 | 
						|
		showError(lang(lng_flood_error));
 | 
						|
		_codeField.notaBene();
 | 
						|
		return true;
 | 
						|
	}
 | 
						|
	if (cDebug()) { // internal server error
 | 
						|
		showError(err + ": " + error.description());
 | 
						|
	} else {
 | 
						|
		showError(lang(lng_server_error));
 | 
						|
	}
 | 
						|
	_codeField.setFocus();
 | 
						|
	return false;
 | 
						|
}
 | 
						|
 | 
						|
void IntroPwdCheck::recoverStarted(const MTPauth_PasswordRecovery &result) {
 | 
						|
	_emailPattern = st::introFont->m.elidedText(lng_signin_recover_hint(lt_recover_email, qs(result.c_auth_passwordRecovery().vemail_pattern)), Qt::ElideRight, textRect.width());
 | 
						|
	update();
 | 
						|
}
 | 
						|
 | 
						|
bool IntroPwdCheck::recoverStartFail(const RPCError &error) {
 | 
						|
	stopCheck();
 | 
						|
	_pwdField.setDisabled(false);
 | 
						|
	_codeField.setDisabled(false);
 | 
						|
	_pwdField.show();
 | 
						|
	_codeField.hide();
 | 
						|
	_pwdField.setFocus();
 | 
						|
	update();
 | 
						|
	showError("");
 | 
						|
	return true;
 | 
						|
}
 | 
						|
 | 
						|
void IntroPwdCheck::onToRecover() {
 | 
						|
	if (_hasRecovery) {
 | 
						|
		if (sentRequest) {
 | 
						|
			MTP::cancel(sentRequest);
 | 
						|
			sentRequest = 0;
 | 
						|
		}
 | 
						|
		showError("");
 | 
						|
		_toRecover.hide();
 | 
						|
		_toPassword.show();
 | 
						|
		_pwdField.hide();
 | 
						|
		_pwdField.setText(QString());
 | 
						|
		_codeField.show();
 | 
						|
		_codeField.setFocus();
 | 
						|
		if (_emailPattern.isEmpty()) {
 | 
						|
			MTP::send(MTPauth_RequestPasswordRecovery(), rpcDone(&IntroPwdCheck::recoverStarted), rpcFail(&IntroPwdCheck::recoverStartFail));
 | 
						|
		}
 | 
						|
		update();
 | 
						|
	} else {
 | 
						|
		ConfirmBox *box = new ConfirmBox(lang(lng_signin_no_email_forgot), true);
 | 
						|
		App::wnd()->showLayer(box);
 | 
						|
		connect(box, SIGNAL(destroyed(QObject*)), this, SLOT(onToReset()));
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
void IntroPwdCheck::onToPassword() {
 | 
						|
	ConfirmBox *box = new ConfirmBox(lang(lng_signin_no_email_forgot), true);
 | 
						|
	App::wnd()->showLayer(box);
 | 
						|
	connect(box, SIGNAL(destroyed(QObject*)), this, SLOT(onToReset()));
 | 
						|
}
 | 
						|
 | 
						|
void IntroPwdCheck::onToReset() {
 | 
						|
	if (sentRequest) {
 | 
						|
		MTP::cancel(sentRequest);
 | 
						|
		sentRequest = 0;
 | 
						|
	}
 | 
						|
	_toRecover.show();
 | 
						|
	_toPassword.hide();
 | 
						|
	_pwdField.show();
 | 
						|
	_codeField.hide();
 | 
						|
	_codeField.setText(QString());
 | 
						|
	_pwdField.setFocus();
 | 
						|
	_reset.show();
 | 
						|
	update();
 | 
						|
}
 | 
						|
 | 
						|
void IntroPwdCheck::onReset() {
 | 
						|
	if (sentRequest) return;
 | 
						|
	ConfirmBox *box = new ConfirmBox(lang(lng_sigin_sure_reset), lang(lng_sigin_reset), QString(), st::btnRedDone);
 | 
						|
	connect(box, SIGNAL(confirmed()), this, SLOT(onResetSure()));
 | 
						|
	App::wnd()->showLayer(box);
 | 
						|
}
 | 
						|
 | 
						|
void IntroPwdCheck::onResetSure() {
 | 
						|
	if (sentRequest) return;
 | 
						|
	sentRequest = MTP::send(MTPaccount_DeleteAccount(MTP_string("Forgot password")), rpcDone(&IntroPwdCheck::deleteDone), rpcFail(&IntroPwdCheck::deleteFail));
 | 
						|
}
 | 
						|
 | 
						|
bool IntroPwdCheck::deleteFail(const RPCError &error) {
 | 
						|
	if (mtpIsFlood(error)) return false;
 | 
						|
	sentRequest = 0;
 | 
						|
	showError(lang(lng_server_error));
 | 
						|
	return true;
 | 
						|
}
 | 
						|
 | 
						|
void IntroPwdCheck::deleteDone(const MTPBool &v) {
 | 
						|
	App::wnd()->hideLayer();
 | 
						|
	intro()->onIntroNext();
 | 
						|
}
 | 
						|
 | 
						|
void IntroPwdCheck::onInputChange() {
 | 
						|
	showError("");
 | 
						|
}
 | 
						|
 | 
						|
void IntroPwdCheck::onSubmitPwd(bool force) {
 | 
						|
	if (sentRequest) return;
 | 
						|
	if (_pwdField.isHidden()) {
 | 
						|
		if (!force && !_codeField.isEnabled()) return;
 | 
						|
		QString code = _codeField.text().trimmed();
 | 
						|
		if (code.isEmpty()) {
 | 
						|
			_codeField.notaBene();
 | 
						|
			return;
 | 
						|
		}
 | 
						|
 | 
						|
		sentRequest = MTP::send(MTPauth_RecoverPassword(MTP_string(code)), rpcDone(&IntroPwdCheck::pwdSubmitDone, true), rpcFail(&IntroPwdCheck::codeSubmitFail));
 | 
						|
	} else {
 | 
						|
		if (!force && !_pwdField.isEnabled()) return;
 | 
						|
 | 
						|
		_pwdField.setDisabled(true);
 | 
						|
		setFocus();
 | 
						|
 | 
						|
		showError("");
 | 
						|
 | 
						|
		QByteArray pwdData = _salt + _pwdField.text().toUtf8() + _salt, pwdHash(32, Qt::Uninitialized);
 | 
						|
		hashSha256(pwdData.constData(), pwdData.size(), pwdHash.data());
 | 
						|
		sentRequest = MTP::send(MTPauth_CheckPassword(MTP_string(pwdHash)), rpcDone(&IntroPwdCheck::pwdSubmitDone, false), rpcFail(&IntroPwdCheck::pwdSubmitFail));
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
void IntroPwdCheck::onNext() {
 | 
						|
	onSubmitPwd();
 | 
						|
}
 | 
						|
 | 
						|
void IntroPwdCheck::onBack() {
 | 
						|
}
 |