Create metric: appsec.waf.error

Author: sezen-datadog

dd-java-agent/appsec/src/main/java/com/datadog/appsec/gateway/AppSecRequestContext.java

@@ -130,6 +130,9 @@ public class AppSecRequestContext implements DataBundle, Closeable {
   private volatile int raspInternalErrors;
   private volatile int raspInvalidObjectErrors;
   private volatile int raspInvalidArgumentErrors;
+  private volatile int wafInternalErrors;
+  private volatile int wafInvalidObjectErrors;
+  private volatile int wafInvalidArgumentErrors;
 
   // keep a reference to the last published usr.id
   private volatile String userId;
@@ -157,6 +160,17 @@ public class AppSecRequestContext implements DataBundle, Closeable {
           AtomicIntegerFieldUpdater.newUpdater(
               AppSecRequestContext.class, "raspInvalidArgumentErrors");
 
+  private static final AtomicIntegerFieldUpdater<AppSecRequestContext> WAF_INTERNAL_ERRORS_UPDATER =
+      AtomicIntegerFieldUpdater.newUpdater(AppSecRequestContext.class, "wafInternalErrors");
+  private static final AtomicIntegerFieldUpdater<AppSecRequestContext>
+      WAF_INVALID_OBJECT_ERRORS_UPDATER =
+          AtomicIntegerFieldUpdater.newUpdater(
+              AppSecRequestContext.class, "wafInvalidObjectErrors");
+  private static final AtomicIntegerFieldUpdater<AppSecRequestContext>
+      WAF_INVALID_ARGUMENT_ERRORS_UPDATER =
+          AtomicIntegerFieldUpdater.newUpdater(
+              AppSecRequestContext.class, "wafInvalidArgumentErrors");
+
   // to be called by the Event Dispatcher
   public void addAll(DataBundle newData) {
     for (Map.Entry<Address<?>, Object> entry : newData) {
@@ -222,6 +236,22 @@ public void increaseRaspErrorCode(int code) {
     }
   }
 
+  public void increaseWafErrorCode(int code) {
+    switch (code) {
+      case DD_WAF_RUN_INTERNAL_ERROR:
+        WAF_INTERNAL_ERRORS_UPDATER.incrementAndGet(this);
+        break;
+      case DD_WAF_RUN_INVALID_OBJECT_ERROR:
+        WAF_INVALID_OBJECT_ERRORS_UPDATER.incrementAndGet(this);
+        break;
+      case DD_WAF_RUN_INVALID_ARGUMENT_ERROR:
+        WAF_INVALID_ARGUMENT_ERRORS_UPDATER.incrementAndGet(this);
+        break;
+      default:
+        break;
+    }
+  }
+
   public int getWafTimeouts() {
     return wafTimeouts;
   }
@@ -232,11 +262,11 @@ public int getRaspTimeouts() {
 
   public int getRaspError(int code) {
     switch (code) {
-      case -3:
+      case DD_WAF_RUN_INTERNAL_ERROR:
         return raspInternalErrors;
-      case -2:
+      case DD_WAF_RUN_INVALID_OBJECT_ERROR:
         return raspInvalidObjectErrors;
-      case -1:
+      case DD_WAF_RUN_INVALID_ARGUMENT_ERROR:
         return raspInvalidArgumentErrors;
       default:
         return 0;