Skip to content

Commit 8891985

Browse files
committed
renames
1 parent 2659315 commit 8891985

File tree

8 files changed

+15
-15
lines changed

8 files changed

+15
-15
lines changed

dd-java-agent/appsec/src/jmh/java/datadog/appsec/benchmark/WafBenchmark.java

+2-2
Original file line numberDiff line numberDiff line change
@@ -80,7 +80,7 @@ public void setUp() throws AbstractWafException, IOException {
8080
AppSecConfig waf = cfg.get("waf");
8181
wafBuilder = new WafBuilder();
8282
wafBuilder.addOrUpdateConfig("waf", waf.getRawConfig());
83-
wafHandle = wafBuilder.buildWafHandleInstance(wafHandle);
83+
wafHandle = wafBuilder.buildWafHandleInstance();
8484
wafData.put(KnownAddresses.REQUEST_METHOD.getKey(), "POST");
8585
wafData.put(
8686
KnownAddresses.REQUEST_URI_RAW.getKey(), "/foo/bar?foo=bar&foo=xpto&foo=%3cscript%3e");
@@ -117,7 +117,7 @@ public void setUp() throws AbstractWafException, IOException {
117117
@TearDown(Level.Trial)
118118
public void teardown() {
119119
if (wafBuilder != null && !wafBuilder.isOnline()) {
120-
wafBuilder.destroy();
120+
wafBuilder.close();
121121
}
122122
}
123123
}

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

+1-1
Original file line numberDiff line numberDiff line change
@@ -155,7 +155,7 @@ public static void stop() {
155155
}
156156
Blocking.setBlockingService(BlockingService.NOOP);
157157
APP_SEC_CONFIG_SERVICE.close();
158-
wafBuilder.destroy();
158+
wafBuilder.close();
159159
}
160160

161161
private static void loadModules(EventDispatcher eventDispatcher, Monitoring monitoring) {

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

+3-3
Original file line numberDiff line numberDiff line change
@@ -29,8 +29,8 @@
2929
import com.datadog.appsec.ddwaf.WAFStatsReporter;
3030
import com.datadog.appsec.util.AbortStartupException;
3131
import com.datadog.appsec.util.StandardizedLogging;
32-
import com.datadog.ddwaf.RuleSetInfo;
3332
import com.datadog.ddwaf.WafBuilder;
33+
import com.datadog.ddwaf.WafDiagnostics;
3434
import com.datadog.ddwaf.exception.InvalidRuleSetException;
3535
import datadog.remoteconfig.ConfigurationChangesTypedListener;
3636
import datadog.remoteconfig.ConfigurationEndListener;
@@ -219,11 +219,11 @@ private ConfigurationChangesTypedListener<AppSecConfig> asmDDTypedListener() {
219219

220220
private void handleWafUpdateResultReport(String configKey, Map<String, Object> rawConfig) {
221221
try {
222-
RuleSetInfo initReport = wafBuilder.addOrUpdateConfig(configKey, rawConfig);
222+
WafDiagnostics initReport = wafBuilder.addOrUpdateConfig(configKey, rawConfig);
223223
initReporter.setReportForPublication(initReport);
224224
statsReporter.setRulesVersion(initReport.rulesetVersion);
225225
} catch (InvalidRuleSetException e) {
226-
initReporter.setReportForPublication(e.ruleSetInfo);
226+
initReporter.setReportForPublication(e.wafDiagnostics);
227227
throw new RuntimeException(e);
228228
}
229229
}

dd-java-agent/appsec/src/main/java/com/datadog/appsec/ddwaf/WAFInitializationResultReporter.java

+4-4
Original file line numberDiff line numberDiff line change
@@ -3,8 +3,8 @@
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;
76
import com.datadog.ddwaf.Waf;
7+
import com.datadog.ddwaf.WafDiagnostics;
88
import com.squareup.moshi.JsonAdapter;
99
import com.squareup.moshi.Moshi;
1010
import com.squareup.moshi.Types;
@@ -27,16 +27,16 @@ public class WAFInitializationResultReporter implements TraceSegmentPostProcesso
2727
.adapter(
2828
Types.newParameterizedType(
2929
Map.class, String.class, Types.newParameterizedType(List.class, String.class)));
30-
private final AtomicReference<RuleSetInfo> pendingReportRef = new AtomicReference<>();
30+
private final AtomicReference<WafDiagnostics> pendingReportRef = new AtomicReference<>();
3131

32-
public void setReportForPublication(RuleSetInfo report) {
32+
public void setReportForPublication(WafDiagnostics report) {
3333
this.pendingReportRef.set(report);
3434
}
3535

3636
@Override
3737
public void processTraceSegment(
3838
TraceSegment segment, AppSecRequestContext ctx, Collection<AppSecEvent> collectedEvents) {
39-
RuleSetInfo report = pendingReportRef.get();
39+
WafDiagnostics report = pendingReportRef.get();
4040
if (report == null) {
4141
return;
4242
}

dd-java-agent/appsec/src/main/java/com/datadog/appsec/ddwaf/WAFModule.java

+1-1
Original file line numberDiff line numberDiff line change
@@ -251,7 +251,7 @@ private boolean initOrUpdateWafHandle(
251251
config.dirtyStatus.markAllDirty();
252252
}
253253
try {
254-
wafHandle = wafBuilder.buildWafHandleInstance(null);
254+
wafHandle = wafBuilder.buildWafHandleInstance();
255255
} catch (AbstractWafException e) {
256256
log.info("Could not initialize waf handle, no rules were added!", e);
257257
throw new AppSecModuleActivationException(

dd-java-agent/appsec/src/test/groovy/com/datadog/appsec/config/AppSecConfigServiceImplSpecification.groovy

+1-1
Original file line numberDiff line numberDiff line change
@@ -57,7 +57,7 @@ class AppSecConfigServiceImplSpecification extends DDSpecification {
5757
void setUpTest() {
5858
WafInitialization.ONLINE
5959
if(wafBuilder != null && !wafBuilder.isOnline()) {
60-
wafBuilder.destroy()
60+
close()
6161
}
6262
wafBuilder = new WafBuilder()
6363
appSecConfigService = new AppSecConfigServiceImpl(config, poller, reconf, wafBuilder)

dd-java-agent/appsec/src/test/groovy/com/datadog/appsec/ddwaf/WAFModuleSpecification.groovy

+1-1
Original file line numberDiff line numberDiff line change
@@ -113,7 +113,7 @@ class WAFModuleSpecification extends DDSpecification {
113113

114114
private static void release() {
115115
if (wafBuilder?.isOnline()) {
116-
wafBuilder.destroy()
116+
wafBuilder.close()
117117
}
118118
}
119119

dd-java-agent/appsec/src/test/groovy/com/datadog/appsec/gateway/AppSecRequestContextSpecification.groovy

+2-2
Original file line numberDiff line numberDiff line change
@@ -209,14 +209,14 @@ class AppSecRequestContextSpecification extends DDSpecification {
209209
Waf.initialize false
210210

211211
if (wafBuilder?.online){
212-
wafBuilder.destroy()
212+
close()
213213
}
214214
wafBuilder = new WafBuilder()
215215
def stream = getClass().classLoader.getResourceAsStream("test_multi_config.json")
216216
def casc = new CurrentAppSecConfig()
217217
casc.ddConfig = AppSecConfigDeserializer.INSTANCE.deserialize(stream)
218218
wafBuilder.addOrUpdateConfig("test", casc.ddConfig.rawConfig)
219-
new WafContext(wafBuilder.buildWafHandleInstance(null))
219+
new WafContext(wafBuilder.buildWafHandleInstance())
220220
}
221221

222222
void 'close closes the wafContext'() {

0 commit comments

Comments
 (0)