1 files changed, 45 insertions, 0 deletions
diff --git a/mobile/android/services/src/main/java/org/mozilla/gecko/fxa/login/TokensAndKeysState.java b/mobile/android/services/src/main/java/org/mozilla/gecko/fxa/login/TokensAndKeysState.java
new file mode 100644
index 000000000..b5121a4d4
--- /dev/null
+++ b/mobile/android/services/src/main/java/org/mozilla/gecko/fxa/login/TokensAndKeysState.java
@@ -0,0 +1,45 @@
+/* This Source Code Form is subject to the terms of the Mozilla Public
+ * License, v. 2.0. If a copy of the MPL was not distributed with this
+ * file, You can obtain one at http://mozilla.org/MPL/2.0/. */
+
+package org.mozilla.gecko.fxa.login;
+
+import org.mozilla.gecko.browserid.BrowserIDKeyPair;
+import org.mozilla.gecko.sync.ExtendedJSONObject;
+import org.mozilla.gecko.sync.Utils;
+
+public abstract class TokensAndKeysState extends State {
+  protected final byte[] sessionToken;
+  protected final byte[] kA;
+  protected final byte[] kB;
+  protected final BrowserIDKeyPair keyPair;
+
+  public TokensAndKeysState(StateLabel stateLabel, String email, String uid, byte[] sessionToken, byte[] kA, byte[] kB, BrowserIDKeyPair keyPair) {
+    super(stateLabel, email, uid, true);
+    Utils.throwIfNull(sessionToken, kA, kB, keyPair);
+    this.sessionToken = sessionToken;
+    this.kA = kA;
+    this.kB = kB;
+    this.keyPair = keyPair;
+  }
+
+  @Override
+  public ExtendedJSONObject toJSONObject() {
+    ExtendedJSONObject o = super.toJSONObject();
+    // Fields are non-null by constructor.
+    o.put("sessionToken", Utils.byte2Hex(sessionToken));
+    o.put("kA", Utils.byte2Hex(kA));
+    o.put("kB", Utils.byte2Hex(kB));
+    o.put("keyPair", keyPair.toJSONObject());
+    return o;
+  }
+
+  public byte[] getSessionToken() {
+    return sessionToken;
+  }
+
+  @Override
+  public Action getNeededAction() {
+    return Action.None;
+  }
+}