Implement KEY ID type of IKE Identification

This commit:
- Creates a new class that represents KEY ID
- Supports decoding ID payload using KEY ID
- Adds tests to verify decoding and encoding

Bug: 142139930
Test: atest FrameworksIkeTests(all tests passed)
Change-Id: I35c9c6509fd82df75ced7f5382d0a7ff6a90266a

1 files changed, 31 insertions, 0 deletions

diff --git a/tests/iketests/src/java/com/android/ike/ikev2/message/IkeIdPayloadTest.java b/tests/iketests/src/java/com/android/ike/ikev2/message/IkeIdPayloadTest.java
index cccbba12..14d05264 100644
--- a/tests/iketests/src/java/com/android/ike/ikev2/message/IkeIdPayloadTest.java
+++ b/tests/iketests/src/java/com/android/ike/ikev2/message/IkeIdPayloadTest.java
@@ -25,6 +25,7 @@ import com.android.ike.ikev2.IkeFqdnIdentification;
 import com.android.ike.ikev2.IkeIdentification;
 import com.android.ike.ikev2.IkeIpv4AddrIdentification;
 import com.android.ike.ikev2.IkeIpv6AddrIdentification;
+import com.android.ike.ikev2.IkeKeyIdIdentification;
 import com.android.ike.ikev2.IkeRfc822AddrIdentification;
 import com.android.ike.ikev2.exceptions.AuthenticationFailedException;
 
@@ -59,6 +60,12 @@ public final class IkeIdPayloadTest {
             "03000000616e64726f6964696b65406578616d706c652e636f6d";
     private static final String RFC822_NAME = "androidike@example.com";
 
+    private static final String KEY_ID_PAYLOAD_HEX_STRING =
+            "250000170b000000616E64726F6964496B654B65794964";
+    private static final String KEY_ID_PAYLOAD_BODY_HEX_STRING =
+            "0b000000616E64726F6964496B654B65794964";
+    private static final byte[] KEY_ID = "androidIkeKeyId".getBytes();
+
     private static final int ID_TYPE_OFFSET = 0;
 
     @Test
@@ -113,6 +120,18 @@ public final class IkeIdPayloadTest {
     }
 
     @Test
+    public void testDecodeKeyIdPayload() throws Exception {
+        byte[] inputPacket = TestUtils.hexStringToByteArray(KEY_ID_PAYLOAD_BODY_HEX_STRING);
+        IkeIdPayload payload =
+                new IkeIdPayload(false /*critical*/, inputPacket, true /*isInitiator*/);
+
+        assertEquals(IkePayload.PAYLOAD_TYPE_ID_INITIATOR, payload.payloadType);
+        assertEquals(IkeIdentification.ID_TYPE_KEY_ID, payload.ikeId.idType);
+        IkeKeyIdIdentification ikeId = (IkeKeyIdIdentification) payload.ikeId;
+        assertArrayEquals(KEY_ID, ikeId.keyId);
+    }
+
+    @Test
     public void testDecodeUnsupportedIdType() throws Exception {
         byte[] inputPacket =
                 TestUtils.hexStringToByteArray(IPV4_ADDR_ID_PAYLOAD_RESPONDER_BODY_HEX_STRING);
@@ -175,4 +194,16 @@ public final class IkeIdPayloadTest {
         byte[] expectedBytes = TestUtils.hexStringToByteArray(RFC822_ADDR_ID_PAYLOAD_HEX_STRING);
         assertArrayEquals(expectedBytes, inputBuffer.array());
     }
+
+    @Test
+    public void testConstructAndEncodeKeyIdPayload() throws Exception {
+        IkeIdPayload payload =
+                new IkeIdPayload(true /*isInitiator*/, new IkeKeyIdIdentification(KEY_ID));
+
+        ByteBuffer inputBuffer = ByteBuffer.allocate(payload.getPayloadLength());
+        payload.encodeToByteBuffer(IkePayload.PAYLOAD_TYPE_CERT, inputBuffer);
+
+        byte[] expectedBytes = TestUtils.hexStringToByteArray(KEY_ID_PAYLOAD_HEX_STRING);
+        assertArrayEquals(expectedBytes, inputBuffer.array());
+    }
 }