[frontend/backend] add obfuscation in payload info

Signed-off-by: Marine LM <[email protected]>
OpenBAS-Platform · Dec 20, 2024 · 651090b · 651090b
1 parent 0d04053
commit 651090b
Show file tree

Hide file tree

Showing 4 changed files with 29 additions and 0 deletions.
diff --git a/openbas-api/src/main/java/io/openbas/rest/atomic_testing/form/StatusPayloadOutput.java b/openbas-api/src/main/java/io/openbas/rest/atomic_testing/form/StatusPayloadOutput.java
@@ -46,6 +46,9 @@ public class StatusPayloadOutput {
   @JsonProperty("payload_arguments")
   private List<PayloadArgument> arguments = new ArrayList<>();
 
+  @JsonProperty("payload_obfuscator")
+  private String obfuscator;
+
   @JsonProperty("payload_prerequisites")
   private List<PayloadPrerequisite> prerequisites = new ArrayList<>();
 

diff --git a/openbas-api/src/main/java/io/openbas/utils/PayloadMapper.java b/openbas-api/src/main/java/io/openbas/utils/PayloadMapper.java
@@ -5,6 +5,7 @@
 import static io.openbas.database.model.Executable.EXECUTABLE_TYPE;
 import static io.openbas.database.model.FileDrop.FILE_DROP_TYPE;
 import static io.openbas.database.model.NetworkTraffic.NETWORK_TRAFFIC_TYPE;
+import static java.util.Optional.ofNullable;
 
 import io.openbas.database.model.*;
 import io.openbas.rest.atomic_testing.form.AttackPatternSimple;
@@ -27,6 +28,14 @@ public StatusPayloadOutput getStatusPayloadOutputFromInject(Optional<Inject> inj
       Optional<InjectorContract> injectorContractOpt = inject.get().getInjectorContract();
       if (injectorContractOpt.isPresent()) {
         InjectorContract injectorContract = injectorContractOpt.get();
+
+        if (ofNullable(inject.get().getContent())
+            .map(c -> c.has("obfuscator"))
+            .orElse(Boolean.FALSE)) {
+          String obfuscator = inject.get().getContent().findValue("obfuscator").asText();
+          statusPayloadOutputBuilder.obfuscator(obfuscator);
+        }
+
         if (inject.get().getStatus().isEmpty()) {
           Payload payload = injectorContract.getPayload();
           statusPayloadOutputBuilder

diff --git a/...as-front/src/admin/components/atomic_testings/atomic_testing/AtomicTestingPayloadInfo.tsx b/...as-front/src/admin/components/atomic_testings/atomic_testing/AtomicTestingPayloadInfo.tsx
@@ -152,6 +152,22 @@ const AtomicTestingPayloadInfo: FunctionComponent<Props> = () => {
                         );
                       })
                     )}
+                { payloadOutput.payload_obfuscator
+                && (
+                  <>
+                    <Typography
+                      variant="h3"
+                      gutterBottom
+                      style={{ marginTop: 20 }}
+                    >
+                      {t('Obfuscator')}
+                    </Typography>
+                    <Typography key="obfuscator" variant="body2">
+                      {payloadOutput.payload_obfuscator}
+                    </Typography>
+                  </>
+
+                )}
                 <Typography
                   variant="h3"
                   gutterBottom

diff --git a/openbas-front/src/utils/api-types.d.ts b/openbas-front/src/utils/api-types.d.ts
@@ -3243,6 +3243,7 @@ export interface StatusPayloadOutput {
   payload_description?: string;
   payload_external_id?: string;
   payload_name?: string;
+  payload_obfuscator?: string;
   payload_platforms?: ("Linux" | "Windows" | "MacOS" | "Container" | "Service" | "Generic" | "Internal" | "Unknown")[];
   payload_prerequisites?: PayloadPrerequisite[];
   /** @uniqueItems true */