Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Minor refactoring concering user.setpassword #3139

Merged
merged 3 commits into from
Jan 24, 2023
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
13 changes: 6 additions & 7 deletions src/api/core/accounts.rs
Original file line number Diff line number Diff line change
Expand Up @@ -161,8 +161,7 @@ pub async fn _register(data: JsonUpcase<RegisterData>, mut conn: DbConn) -> Json
user.client_kdf_type = client_kdf_type;
}

user.set_password(&data.MasterPasswordHash, None);
user.akey = data.Key;
user.set_password(&data.MasterPasswordHash, &data.Key, None);
user.password_hint = password_hint;

// Add extra fields if present
Expand Down Expand Up @@ -318,9 +317,10 @@ async fn post_password(

user.set_password(
&data.NewMasterPasswordHash,
&data.Key,
Some(vec![String::from("post_rotatekey"), String::from("get_contacts"), String::from("get_public_keys")]),
);
user.akey = data.Key;

let save_result = user.save(&mut conn).await;

nt.send_user_update(UpdateType::LogOut, &user).await;
Expand Down Expand Up @@ -350,8 +350,7 @@ async fn post_kdf(data: JsonUpcase<ChangeKdfData>, headers: Headers, mut conn: D

user.client_kdf_iter = data.KdfIterations;
user.client_kdf_type = data.Kdf;
user.set_password(&data.NewMasterPasswordHash, None);
user.akey = data.Key;
user.set_password(&data.NewMasterPasswordHash, &data.Key, None);
let save_result = user.save(&mut conn).await;

nt.send_user_update(UpdateType::LogOut, &user).await;
Expand Down Expand Up @@ -560,8 +559,8 @@ async fn post_email(
user.email_new = None;
user.email_new_token = None;

user.set_password(&data.NewMasterPasswordHash, None);
user.akey = data.Key;
user.set_password(&data.NewMasterPasswordHash, &data.Key, None);

let save_result = user.save(&mut conn).await;

nt.send_user_update(UpdateType::LogOut, &user).await;
Expand Down
5 changes: 2 additions & 3 deletions src/api/core/emergency_access.rs
Original file line number Diff line number Diff line change
Expand Up @@ -644,7 +644,7 @@ async fn password_emergency_access(

let data: EmergencyAccessPasswordData = data.into_inner().data;
let new_master_password_hash = &data.NewMasterPasswordHash;
let key = data.Key;
let key = &data.Key;

let requesting_user = headers.user;
let emergency_access = match EmergencyAccess::find_by_uuid(&emer_id, &mut conn).await {
Expand All @@ -662,8 +662,7 @@ async fn password_emergency_access(
};

// change grantor_user password
grantor_user.set_password(new_master_password_hash, None);
grantor_user.akey = key;
grantor_user.set_password(new_master_password_hash, key, None);
grantor_user.save(&mut conn).await?;

// Disable TwoFactor providers since they will otherwise block logins
Expand Down
5 changes: 3 additions & 2 deletions src/db/models/user.rs
Original file line number Diff line number Diff line change
Expand Up @@ -147,17 +147,18 @@ impl User {
/// # Arguments
///
/// * `password` - A str which contains a hashed version of the users master password.
/// * `new_key` - A String which contains the new aKey value of the users master password.
/// * `allow_next_route` - A Option<Vec<String>> with the function names of the next allowed (rocket) routes.
/// These routes are able to use the previous stamp id for the next 2 minutes.
/// After these 2 minutes this stamp will expire.
///
pub fn set_password(&mut self, password: &str, allow_next_route: Option<Vec<String>>) {
pub fn set_password(&mut self, password: &str, new_key: &str, allow_next_route: Option<Vec<String>>) {
self.password_hash = crypto::hash_password(password.as_bytes(), &self.salt, self.password_iterations as u32);

if let Some(route) = allow_next_route {
self.set_stamp_exception(route);
}

self.akey = String::from(new_key);
self.reset_security_stamp()
}

Expand Down