1 files changed, 96 insertions, 0 deletions
diff --git a/src/main/java/org/whispersystems/libaxolotl/state/PreKeyBundle.java b/src/main/java/org/whispersystems/libaxolotl/state/PreKeyBundle.java
new file mode 100644
index 00000000..772bcc14
--- /dev/null
+++ b/src/main/java/org/whispersystems/libaxolotl/state/PreKeyBundle.java
@@ -0,0 +1,96 @@
+package org.whispersystems.libaxolotl.state;
+
+import org.whispersystems.libaxolotl.IdentityKey;
+import org.whispersystems.libaxolotl.ecc.ECPublicKey;
+
+/**
+ * A class that contains a remote PreKey and collection
+ * of associated items.
+ *
+ * @author Moxie Marlinspike
+ */
+public class PreKeyBundle {
+
+  private int         registrationId;
+
+  private int         deviceId;
+
+  private int         preKeyId;
+  private ECPublicKey preKeyPublic;
+
+  private int         signedPreKeyId;
+  private ECPublicKey signedPreKeyPublic;
+  private byte[]      signedPreKeySignature;
+
+  private IdentityKey identityKey;
+
+  public PreKeyBundle(int registrationId, int deviceId, int preKeyId, ECPublicKey preKeyPublic,
+                      int signedPreKeyId, ECPublicKey signedPreKeyPublic, byte[] signedPreKeySignature,
+                      IdentityKey identityKey)
+  {
+    this.registrationId        = registrationId;
+    this.deviceId              = deviceId;
+    this.preKeyId              = preKeyId;
+    this.preKeyPublic          = preKeyPublic;
+    this.signedPreKeyId        = signedPreKeyId;
+    this.signedPreKeyPublic    = signedPreKeyPublic;
+    this.signedPreKeySignature = signedPreKeySignature;
+    this.identityKey           = identityKey;
+  }
+
+  /**
+   * @return the device ID this PreKey belongs to.
+   */
+  public int getDeviceId() {
+    return deviceId;
+  }
+
+  /**
+   * @return the unique key ID for this PreKey.
+   */
+  public int getPreKeyId() {
+    return preKeyId;
+  }
+
+  /**
+   * @return the public key for this PreKey.
+   */
+  public ECPublicKey getPreKey() {
+    return preKeyPublic;
+  }
+
+  /**
+   * @return the unique key ID for this signed prekey.
+   */
+  public int getSignedPreKeyId() {
+    return signedPreKeyId;
+  }
+
+  /**
+   * @return the signed prekey for this PreKeyBundle.
+   */
+  public ECPublicKey getSignedPreKey() {
+    return signedPreKeyPublic;
+  }
+
+  /**
+   * @return the signature over the signed  prekey.
+   */
+  public byte[] getSignedPreKeySignature() {
+    return signedPreKeySignature;
+  }
+
+  /**
+   * @return the {@link org.whispersystems.libaxolotl.IdentityKey} of this PreKeys owner.
+   */
+  public IdentityKey getIdentityKey() {
+    return identityKey;
+  }
+
+  /**
+   * @return the registration ID associated with this PreKey.
+   */
+  public int getRegistrationId() {
+    return registrationId;
+  }
+}