From 795b226b90361d49a6a80dceebdc4e3956fdfe1c Mon Sep 17 00:00:00 2001
From: Chris Eager <chris@signal.org>
Date: Wed, 24 Nov 2021 16:26:44 -0800
Subject: [PATCH] Mark methods that update `SignedPreKey`s as
 `@ChangesDeviceEnabledState`

---
 .../textsecuregcm/controllers/KeysController.java              | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/service/src/main/java/org/whispersystems/textsecuregcm/controllers/KeysController.java b/service/src/main/java/org/whispersystems/textsecuregcm/controllers/KeysController.java
index 22e33cbfc..bcaaa9013 100644
--- a/service/src/main/java/org/whispersystems/textsecuregcm/controllers/KeysController.java
+++ b/service/src/main/java/org/whispersystems/textsecuregcm/controllers/KeysController.java
@@ -31,6 +31,7 @@ import javax.ws.rs.core.MediaType;
 import javax.ws.rs.core.Response;
 import org.whispersystems.textsecuregcm.auth.Anonymous;
 import org.whispersystems.textsecuregcm.auth.AuthenticatedAccount;
+import org.whispersystems.textsecuregcm.auth.ChangesDeviceEnabledState;
 import org.whispersystems.textsecuregcm.auth.DisabledPermittedAuthenticatedAccount;
 import org.whispersystems.textsecuregcm.auth.OptionalAccess;
 import org.whispersystems.textsecuregcm.entities.PreKey;
@@ -93,6 +94,7 @@ public class KeysController {
   @Timed
   @PUT
   @Consumes(MediaType.APPLICATION_JSON)
+  @ChangesDeviceEnabledState
   public void setKeys(@Auth final DisabledPermittedAuthenticatedAccount disabledPermittedAuth,
       @Valid final PreKeyState preKeys,
       @QueryParam("identity") final Optional<String> identityType) {
@@ -217,6 +219,7 @@ public class KeysController {
   @PUT
   @Path("/signed")
   @Consumes(MediaType.APPLICATION_JSON)
+  @ChangesDeviceEnabledState
   public void setSignedKey(@Auth final AuthenticatedAccount auth,
       @Valid final SignedPreKey signedPreKey,
       @QueryParam("identity") final Optional<String> identityType) {