Skip to content

Commit e3867f0

Browse files
Rename classes to better reflect their counterparts in libddwaf (#8525)
1 parent d8de104 commit e3867f0

26 files changed

+454
-461
lines changed

Diff for: .circleci/config.continue.yml.j2

+1-1
Original file line numberDiff line numberDiff line change
@@ -36,7 +36,7 @@ instrumentation_modules: &instrumentation_modules "dd-java-agent/instrumentation
3636
debugger_modules: &debugger_modules "dd-java-agent/agent-debugger|dd-java-agent/agent-bootstrap|dd-java-agent/agent-builder|internal-api|communication|dd-trace-core"
3737
profiling_modules: &profiling_modules "dd-java-agent/agent-profiling"
3838

39-
default_system_tests_commit: &default_system_tests_commit 1de04c42cbd9783432258004e46eb5982bbc9fe5
39+
default_system_tests_commit: &default_system_tests_commit 55cc62fb333b8de1466f8a0ecfdfa55fe9f57031
4040

4141
parameters:
4242
nightly:

Diff for: dd-java-agent/appsec/build.gradle

+3-4
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,7 @@ dependencies {
1515
implementation project(':internal-api')
1616
implementation project(':communication')
1717
implementation project(':telemetry')
18-
implementation group: 'io.sqreen', name: 'libsqreen', version: '12.0.0'
18+
implementation group: 'io.sqreen', name: 'libsqreen', version: '13.0.1'
1919
implementation libs.moshi
2020

2121
testImplementation libs.bytebuddy
@@ -68,8 +68,8 @@ ext {
6868
minimumInstructionCoverage = 0.8
6969
excludedClassesCoverage = [
7070
'com.datadog.appsec.config.MergedAsmData.InvalidAsmDataException',
71-
'com.datadog.appsec.powerwaf.LibSqreenInitialization',
72-
'com.datadog.appsec.powerwaf.PowerWAFModule.PowerWAFDataCallback',
71+
'com.datadog.appsec.ddwaf.WafInitialization',
72+
'com.datadog.appsec.ddwaf.WAFModule.WAFDataCallback',
7373
'com.datadog.appsec.report.*',
7474
'com.datadog.appsec.config.AppSecConfigServiceImpl.SubscribeFleetServiceRunnable.1',
7575
'com.datadog.appsec.util.StandardizedLogging',
@@ -90,7 +90,6 @@ ext {
9090
'com.datadog.appsec.config.CurrentAppSecConfig',
9191
// equals() / hashCode() are not well covered
9292
'com.datadog.appsec.config.AppSecConfig.Helper',
93-
'com.datadog.appsec.powerwaf.PowerWAFModule.PowerWAFEventsCallback',
9493
// assert never fails
9594
'com.datadog.appsec.util.StandardizedLogging',
9695
'com.datadog.appsec.util.AbortStartupException',

Diff for: dd-java-agent/appsec/src/jmh/java/datadog/appsec/benchmark/BenchmarkUtil.java

+6-6
Original file line numberDiff line numberDiff line change
@@ -1,9 +1,9 @@
11
package datadog.appsec.benchmark;
22

33
import ch.qos.logback.classic.Logger;
4-
import io.sqreen.powerwaf.Powerwaf;
5-
import io.sqreen.powerwaf.exception.AbstractPowerwafException;
6-
import io.sqreen.powerwaf.exception.UnsupportedVMException;
4+
import com.datadog.ddwaf.Waf;
5+
import com.datadog.ddwaf.exception.AbstractWafException;
6+
import com.datadog.ddwaf.exception.UnsupportedVMException;
77
import java.lang.reflect.UndeclaredThrowableException;
88
import org.slf4j.LoggerFactory;
99

@@ -16,10 +16,10 @@ public static void disableLogging() {
1616
}
1717
}
1818

19-
public static void initializePowerwaf() {
19+
public static void initializeWaf() {
2020
try {
21-
Powerwaf.initialize(false);
22-
} catch (AbstractPowerwafException e) {
21+
Waf.initialize(false);
22+
} catch (AbstractWafException e) {
2323
throw new UndeclaredThrowableException(e);
2424
} catch (UnsupportedVMException e) {
2525
throw new UndeclaredThrowableException(e);

Diff for: dd-java-agent/appsec/src/jmh/java/datadog/appsec/benchmark/PowerwafBenchmark.java renamed to dd-java-agent/appsec/src/jmh/java/datadog/appsec/benchmark/WafBenchmark.java

+14-14
Original file line numberDiff line numberDiff line change
@@ -6,11 +6,11 @@
66
import com.datadog.appsec.config.AppSecConfig;
77
import com.datadog.appsec.config.AppSecConfigDeserializer;
88
import com.datadog.appsec.event.data.KnownAddresses;
9-
import io.sqreen.powerwaf.Additive;
10-
import io.sqreen.powerwaf.Powerwaf;
11-
import io.sqreen.powerwaf.PowerwafContext;
12-
import io.sqreen.powerwaf.PowerwafMetrics;
13-
import io.sqreen.powerwaf.exception.AbstractPowerwafException;
9+
import com.datadog.ddwaf.Waf;
10+
import com.datadog.ddwaf.WafContext;
11+
import com.datadog.ddwaf.WafHandle;
12+
import com.datadog.ddwaf.WafMetrics;
13+
import com.datadog.ddwaf.exception.AbstractWafException;
1414
import java.io.IOException;
1515
import java.io.InputStream;
1616
import java.util.ArrayList;
@@ -37,21 +37,21 @@
3737
@BenchmarkMode(Mode.AverageTime)
3838
@OutputTimeUnit(MICROSECONDS)
3939
@Fork(value = 3)
40-
public class PowerwafBenchmark {
40+
public class WafBenchmark {
4141

4242
static {
4343
BenchmarkUtil.disableLogging();
44-
BenchmarkUtil.initializePowerwaf();
44+
BenchmarkUtil.initializeWaf();
4545
}
4646

47-
PowerwafContext ctx;
47+
WafHandle ctx;
4848
Map<String, Object> wafData = new HashMap<>();
49-
Powerwaf.Limits limits = new Powerwaf.Limits(50, 500, 1000, 5000000, 5000000);
49+
Waf.Limits limits = new Waf.Limits(50, 500, 1000, 5000000, 5000000);
5050

5151
@Benchmark
5252
public void withMetrics() throws Exception {
53-
PowerwafMetrics metricsCollector = ctx.createMetrics();
54-
Additive add = ctx.openAdditive();
53+
WafMetrics metricsCollector = ctx.createMetrics();
54+
WafContext add = ctx.openContext();
5555
try {
5656
add.run(wafData, limits, metricsCollector);
5757
} finally {
@@ -61,7 +61,7 @@ public void withMetrics() throws Exception {
6161

6262
@Benchmark
6363
public void withoutMetrics() throws Exception {
64-
Additive add = ctx.openAdditive();
64+
WafContext add = ctx.openContext();
6565
try {
6666
add.run(wafData, limits, null);
6767
} finally {
@@ -70,12 +70,12 @@ public void withoutMetrics() throws Exception {
7070
}
7171

7272
@Setup(Level.Trial)
73-
public void setUp() throws AbstractPowerwafException, IOException {
73+
public void setUp() throws AbstractWafException, IOException {
7474
InputStream stream = getClass().getClassLoader().getResourceAsStream("test_multi_config.json");
7575
Map<String, AppSecConfig> cfg =
7676
Collections.singletonMap("waf", AppSecConfigDeserializer.INSTANCE.deserialize(stream));
7777
AppSecConfig waf = cfg.get("waf");
78-
ctx = Powerwaf.createContext("waf", waf.getRawConfig());
78+
ctx = Waf.createHandle("waf", waf.getRawConfig());
7979

8080
wafData.put(KnownAddresses.REQUEST_METHOD.getKey(), "POST");
8181
wafData.put(

Diff for: dd-java-agent/appsec/src/main/java/com/datadog/appsec/AppSecSystem.java

+2-2
Original file line numberDiff line numberDiff line change
@@ -6,10 +6,10 @@
66
import com.datadog.appsec.blocking.BlockingServiceImpl;
77
import com.datadog.appsec.config.AppSecConfigService;
88
import com.datadog.appsec.config.AppSecConfigServiceImpl;
9+
import com.datadog.appsec.ddwaf.WAFModule;
910
import com.datadog.appsec.event.EventDispatcher;
1011
import com.datadog.appsec.event.ReplaceableEventProducerService;
1112
import com.datadog.appsec.gateway.GatewayBridge;
12-
import com.datadog.appsec.powerwaf.PowerWAFModule;
1313
import com.datadog.appsec.util.AbortStartupException;
1414
import com.datadog.appsec.util.StandardizedLogging;
1515
import datadog.appsec.api.blocking.Blocking;
@@ -151,7 +151,7 @@ private static void loadModules(EventDispatcher eventDispatcher, Monitoring moni
151151
EventDispatcher.DataSubscriptionSet dataSubscriptionSet =
152152
new EventDispatcher.DataSubscriptionSet();
153153

154-
final List<AppSecModule> modules = Collections.singletonList(new PowerWAFModule(monitoring));
154+
final List<AppSecModule> modules = Collections.singletonList(new WAFModule(monitoring));
155155
for (AppSecModule module : modules) {
156156
log.debug("Starting appsec module {}", module.getName());
157157
try {

Diff for: dd-java-agent/appsec/src/main/java/com/datadog/appsec/api/security/AppSecSpanPostProcessor.java

+1-1
Original file line numberDiff line numberDiff line change
@@ -61,7 +61,7 @@ public void process(@Nonnull AgentSpan span, @Nonnull BooleanSupplier timeoutChe
6161
// XXX: Close the additive first. This is not strictly needed, but it'll prevent getting it
6262
// detected as a
6363
// missed request-ended event.
64-
ctx.closeAdditive();
64+
ctx.closeWafContext();
6565
ctx.close();
6666
} catch (Exception e) {
6767
log.debug("Error closing AppSecRequestContext", e);

Diff for: dd-java-agent/appsec/src/main/java/com/datadog/appsec/config/AppSecConfigServiceImpl.java

+1-1
Original file line numberDiff line numberDiff line change
@@ -405,7 +405,7 @@ private void setAppSecActivation(final AppSecFeatures.Asm asm) {
405405
AppSecSystem.setActive(newState);
406406
if (AppSecSystem.isActive()) {
407407
// On remote activation, we need to re-distribute the last known configuration.
408-
// This may trigger initializations, including PowerWAF if it was lazy loaded.
408+
// This may trigger initializations, including WAF if it was lazy loaded.
409409
this.currentAppSecConfig.dirtyStatus.markAllDirty();
410410
}
411411
}

Diff for: dd-java-agent/appsec/src/main/java/com/datadog/appsec/powerwaf/PowerWAFInitializationResultReporter.java renamed to dd-java-agent/appsec/src/main/java/com/datadog/appsec/ddwaf/WAFInitializationResultReporter.java

+5-5
Original file line numberDiff line numberDiff line change
@@ -1,21 +1,21 @@
1-
package com.datadog.appsec.powerwaf;
1+
package com.datadog.appsec.ddwaf;
22

33
import com.datadog.appsec.config.TraceSegmentPostProcessor;
44
import com.datadog.appsec.gateway.AppSecRequestContext;
55
import com.datadog.appsec.report.AppSecEvent;
6+
import com.datadog.ddwaf.RuleSetInfo;
7+
import com.datadog.ddwaf.Waf;
68
import com.squareup.moshi.JsonAdapter;
79
import com.squareup.moshi.Moshi;
810
import com.squareup.moshi.Types;
911
import datadog.trace.api.internal.TraceSegment;
1012
import datadog.trace.bootstrap.instrumentation.api.Tags;
11-
import io.sqreen.powerwaf.Powerwaf;
12-
import io.sqreen.powerwaf.RuleSetInfo;
1313
import java.util.Collection;
1414
import java.util.List;
1515
import java.util.Map;
1616
import java.util.concurrent.atomic.AtomicReference;
1717

18-
public class PowerWAFInitializationResultReporter implements TraceSegmentPostProcessor {
18+
public class WAFInitializationResultReporter implements TraceSegmentPostProcessor {
1919
private static final String WAF_VERSION = "_dd.appsec.waf.version";
2020
private static final String RULE_ERRORS = "_dd.appsec.event_rules.errors";
2121
private static final String RULES_LOADED = "_dd.appsec.event_rules.loaded";
@@ -48,7 +48,7 @@ public void processTraceSegment(
4848
segment.setTagTop(RULE_ERRORS, RULES_ERRORS_ADAPTER.toJson(report.getErrors()));
4949
segment.setTagTop(RULES_LOADED, report.getNumRulesOK());
5050
segment.setTagTop(RULE_ERROR_COUNT, report.getNumRulesError());
51-
segment.setTagTop(WAF_VERSION, Powerwaf.LIB_VERSION);
51+
segment.setTagTop(WAF_VERSION, Waf.LIB_VERSION);
5252

5353
segment.setTagTop(Tags.ASM_KEEP, true);
5454
}

0 commit comments

Comments
 (0)