Potential fix for kyber errors

Schmiddiii · Dec 22, 2023 · 6339114 · 6339114
1 parent 6d7003f
commit 6339114
Showing 1 changed file with 11 additions and 3 deletions.
diff --git a/presage/src/manager/linking.rs b/presage/src/manager/linking.rs
@@ -137,11 +137,19 @@ impl<S: Store> Manager<S, Linking> {
                     &registration_data.service_ids
                 );
 
-                Ok(Manager {
+                let mut manager = Manager {
                     rng,
-                    store,
+                    store: store.clone(),
                     state: Registered::with_data(registration_data),
-                })
+                };
+
+                // Register pre-keys with the server. If this fails, this can lead to issues receiving, in that case clear the registration and propagate the error.
+                if let Err(e) = manager.register_pre_keys().await {
+                    store.clear_registration()?;
+                    Err(e)
+                } else {
+                    Ok(manager)
+                }
             }
             Err(e) => {
                 store.clear_registration()?;