summaryrefslogtreecommitdiffstats
path: root/utils/src/main
diff options
context:
space:
mode:
Diffstat (limited to 'utils/src/main')
-rw-r--r--utils/src/main/java/org/onap/policy/common/utils/coder/PropertyCoder.java139
1 files changed, 139 insertions, 0 deletions
diff --git a/utils/src/main/java/org/onap/policy/common/utils/coder/PropertyCoder.java b/utils/src/main/java/org/onap/policy/common/utils/coder/PropertyCoder.java
new file mode 100644
index 00000000..7b3c16d3
--- /dev/null
+++ b/utils/src/main/java/org/onap/policy/common/utils/coder/PropertyCoder.java
@@ -0,0 +1,139 @@
+/*
+ * ============LICENSE_START=======================================================
+ * ONAP PAP
+ * ================================================================================
+ * Copyright (C) 2019 AT&T Intellectual Property. All rights reserved.
+ * ================================================================================
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ * ============LICENSE_END=========================================================
+ */
+
+package org.onap.policy.common.utils.coder;
+
+import com.google.gson.Gson;
+import com.google.gson.JsonArray;
+import com.google.gson.JsonElement;
+import com.google.gson.JsonObject;
+import com.google.gson.JsonPrimitive;
+
+import java.io.Reader;
+import java.util.Map;
+import java.util.Map.Entry;
+import java.util.Set;
+import lombok.AccessLevel;
+import lombok.Getter;
+
+import org.apache.commons.lang3.StringUtils;
+import org.onap.policy.common.utils.security.CryptoCoder;
+import org.onap.policy.common.utils.security.CryptoUtils;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+
+/**
+ * JSON encoder and decoder using the "property" mechanism, which is currently gson.
+ *
+ */
+public class PropertyCoder {
+ /**
+ * Gson object used to encode and decode messages.
+ */
+ @Getter(AccessLevel.PROTECTED)
+ private static final Gson GSON = new Gson();
+
+ /**
+ * Decode json for encrypted password.
+ *
+ * @param json string
+ * @param keyProperty contains property within jsonObject for secretKey
+ * @param clazz class T object
+ * @return a class T object
+ */
+ public <T> T decode(String json, String keyProperty, Class<T> clazz) throws CoderException {
+ JsonElement jsonElement = GSON.fromJson(json, JsonElement.class);
+ return new MyDecoder(jsonElement, keyProperty).decrypt(jsonElement, clazz);
+ }
+
+ public <T> T decode(Reader reader, String keyProperty, Class<T> clazz) throws CoderException {
+ JsonElement jsonElement = GSON.fromJson(reader, JsonElement.class);
+ return new MyDecoder(jsonElement, keyProperty).decrypt(jsonElement, clazz);
+ }
+
+ private static class MyDecoder extends StandardCoder {
+ private CryptoCoder crypto = null;
+
+ MyDecoder(JsonElement jsonElement, String keyProperty) throws CoderException {
+ if (!jsonElement.isJsonObject()) {
+ return;
+ }
+ JsonObject jsonObject = jsonElement.getAsJsonObject();
+ // Use keyProperty from input to retrieve secretKey
+ String secretKey = jsonObject.get(keyProperty).getAsString();
+ if (!StringUtils.isBlank(secretKey)) {
+ crypto = new CryptoUtils(secretKey);
+ }
+ }
+
+ private <T> T decrypt(JsonElement jsonElement, Class<T> clazz) {
+ if (crypto == null) {
+ return fromJson(jsonElement, clazz);
+ }
+ JsonElement newElement = decrypt(jsonElement);
+ return fromJson(newElement, clazz);
+ }
+
+ private JsonElement decrypt(JsonElement jsonElement) {
+ if (jsonElement.isJsonObject()) {
+ return decryptObject(jsonElement.getAsJsonObject());
+ }
+ if (jsonElement.isJsonArray()) {
+ return decryptArray(jsonElement.getAsJsonArray());
+ }
+ if (!jsonElement.getAsJsonPrimitive().isString()) {
+ return jsonElement;
+ }
+ String value = jsonElement.getAsString();
+ if (!value.startsWith("enc:")) {
+ return jsonElement;
+ }
+ if (crypto != null) {
+ value = crypto.decrypt(value);
+ }
+ return new JsonPrimitive(value);
+ }
+
+ private JsonArray decryptArray(JsonArray jsonArray) {
+ if (crypto == null) {
+ return jsonArray;
+ }
+ JsonArray newArray = new JsonArray();
+ for (JsonElement element: jsonArray) {
+ newArray.add(decrypt(element));
+ }
+ return newArray;
+ }
+
+ private JsonObject decryptObject(JsonObject jsonObject) {
+ if (crypto == null) {
+ return jsonObject;
+ }
+ JsonObject newObject = new JsonObject();
+ Set<Entry<String, JsonElement>> entrySet = jsonObject.entrySet();
+ for (Map.Entry<String, JsonElement> entry : entrySet) {
+ String key = entry.getKey();
+ JsonElement jsonElement = decrypt(entry.getValue());
+ newObject.add(key, jsonElement);
+ }
+ return newObject;
+ }
+ }
+} \ No newline at end of file