[backend] Add payload elevation required

openbas-api/src/main/java/io/openbas/migration/V3_37__Add_column_elevation_required_payload.java

@@ -0,0 +1,17 @@
+package io.openbas.migration;
+
+import org.flywaydb.core.api.migration.BaseJavaMigration;
+import org.flywaydb.core.api.migration.Context;
+import org.springframework.stereotype.Component;
+
+import java.sql.Statement;
+
+@Component
+public class V3_37__Add_column_elevation_required_payload extends BaseJavaMigration {
+
+    @Override
+    public void migrate(final Context context) throws Exception {
+        final Statement select = context.getConnection().createStatement();
+        select.execute("ALTER TABLE payloads ADD payload_elevation_required bool default false;");
+    }
+}

openbas-api/src/main/java/io/openbas/rest/payload/form/PayloadUpsertInput.java

@@ -76,6 +76,9 @@ public class PayloadUpsertInput {
 
     @JsonProperty("payload_attack_patterns")
     private List<String> attackPatternsExternalIds = new ArrayList<>();
+
+    @JsonProperty("payload_elevation_required")
+    private boolean elevationRequired;
 }
 
 

openbas-model/src/main/java/io/openbas/database/model/Payload.java

@@ -14,6 +14,7 @@
 import jakarta.validation.constraints.NotBlank;
 import jakarta.validation.constraints.NotNull;
 import lombok.Data;
+import lombok.Getter;
 import lombok.Setter;
 import org.hibernate.annotations.Type;
 import org.hibernate.annotations.UuidGenerator;
@@ -94,6 +95,11 @@ public enum PAYLOAD_STATUS {
   @JsonProperty("payload_cleanup_command")
   private String cleanupCommand;
 
+  @Getter
+  @Column(name = "payload_elevation_required")
+  @JsonProperty("payload_elevation_required")
+  private boolean elevationRequired;
+
   @Setter
   @Type(JsonType.class)
   @Column(name = "payload_arguments")
@@ -159,7 +165,7 @@ public enum PAYLOAD_STATUS {
 
   @JsonProperty("payload_collector_type")
   private String getCollectorType() {
-    return this.getCollector() != null ? this.getCollector().getType() : null;
+    return this.collector != null ? this.collector.getType() : null;
   }
 
   @Override