chore: add user agent string (#2172)

Author: rishav-karanjit

DynamoDbEncryption/runtimes/java/src/main/sdkv2/com/amazonaws/services/dynamodbv2/datamodeling/sdkv2/encryption/providers/DirectKmsMaterialProvider.java

@@ -17,6 +17,7 @@
 import static com.amazonaws.services.dynamodbv2.datamodeling.sdkv2.encryption.materials.WrappedRawMaterials.CONTENT_KEY_ALGORITHM;
 import static com.amazonaws.services.dynamodbv2.datamodeling.sdkv2.encryption.materials.WrappedRawMaterials.ENVELOPE_KEY;
 import static com.amazonaws.services.dynamodbv2.datamodeling.sdkv2.encryption.materials.WrappedRawMaterials.KEY_WRAPPING_ALGORITHM;
+import static com.amazonaws.services.dynamodbv2.datamodeling.sdkv2.internal.Utils.loadVersion;
 
 import com.amazonaws.services.dynamodbv2.datamodeling.sdkv2.encryption.EncryptionContext;
 import com.amazonaws.services.dynamodbv2.datamodeling.sdkv2.encryption.exceptions.DynamoDbEncryptionException;
@@ -33,6 +34,8 @@
 import java.util.Map;
 import javax.crypto.SecretKey;
 import javax.crypto.spec.SecretKeySpec;
+import software.amazon.awssdk.awscore.AwsRequestOverrideConfiguration;
+import software.amazon.awssdk.core.ApiName;
 import software.amazon.awssdk.core.SdkBytes;
 import software.amazon.awssdk.services.dynamodb.model.AttributeValue;
 import software.amazon.awssdk.services.kms.KmsClient;
@@ -51,6 +54,8 @@
  */
 public class DirectKmsMaterialProvider implements EncryptionMaterialsProvider {
 
+  private static final String USER_AGENT = "DynamodbEncryptionSdkJava/";
+
   private static final String COVERED_ATTR_CTX_KEY = "aws-kms-ec-attr";
   private static final String SIGNING_KEY_ALGORITHM = "amzn-ddb-sig-alg";
   private static final String TABLE_NAME_EC_KEY = "*aws-kms-table*";
@@ -288,7 +293,7 @@ protected DecryptResponse decrypt(
     final DecryptRequest request,
     final EncryptionContext context
   ) {
-    return kms.decrypt(request);
+    return kms.decrypt(appendUserAgent(request));
   }
 
   /**
@@ -305,7 +310,7 @@ protected GenerateDataKeyResponse generateDataKey(
     final GenerateDataKeyRequest request,
     final EncryptionContext context
   ) {
-    return kms.generateDataKey(request);
+    return kms.generateDataKey(appendUserAgent(request));
   }
 
   /**
@@ -384,4 +389,29 @@ private static void populateKmsEcFromEc(
   public void refresh() {
     // No action needed
   }
+
+  private static AwsRequestOverrideConfiguration userAgentOverrideConfig() {
+    return AwsRequestOverrideConfiguration
+      .builder()
+      .addApiName(
+        ApiName.builder().name(USER_AGENT).version(loadVersion()).build()
+      )
+      .build();
+  }
+
+  private static DecryptRequest appendUserAgent(final DecryptRequest request) {
+    return request
+      .toBuilder()
+      .overrideConfiguration(userAgentOverrideConfig())
+      .build();
+  }
+
+  private static GenerateDataKeyRequest appendUserAgent(
+    final GenerateDataKeyRequest request
+  ) {
+    return request
+      .toBuilder()
+      .overrideConfiguration(userAgentOverrideConfig())
+      .build();
+  }
 }

DynamoDbEncryption/runtimes/java/src/main/sdkv2/com/amazonaws/services/dynamodbv2/datamodeling/sdkv2/internal/Utils.java

@@ -38,4 +38,8 @@ public static byte[] getRandom(int len) {
     getRng().nextBytes(result);
     return result;
   }
+
+  public static String loadVersion() {
+    return "4.x-legacy";
+  }
 }