[java] API Security sampling endpoints (#4209)

Author: smola

docs/weblog/README.md

@@ -90,6 +90,8 @@ Hello world!\n
 
 This endpoint is used for API security sampling and must accept a parameter `i` as an integer.
 
+The response status code must be `i`.
+
 The response body may contain the following text:
 
 ```

utils/build/docker/java/akka-http/src/main/scala/com/datadoghq/akka_http/AppSecRoutes.scala

@@ -83,6 +83,26 @@ object AppSecRoutes {
             }
           }
       } ~
+      path("api_security/sampling" / """\d{3}""".r) { (i) =>
+        get {
+          complete(
+            HttpResponse(
+              status = StatusCodes.custom(i.toInt, "some reason"),
+              entity = HttpEntity(ContentTypes.`text/plain(UTF-8)`, "Hello!\n")
+            )
+          )
+        }
+      } ~
+      path("api_security_sampling" / """\d{2,3}""".r) { (i) =>
+        get {
+          complete(
+            HttpResponse(
+              status = StatusCodes.OK,
+              entity = HttpEntity(ContentTypes.`text/plain(UTF-8)`, "OK!\n")
+            )
+          )
+        }
+      } ~
       path("params" / Segments) { segments: Seq[String] =>
         get {
           complete(segments.toString())

utils/build/docker/java/jersey-grizzly2/src/main/java/com/datadoghq/jersey/MyResource.java

@@ -111,6 +111,22 @@ public Response tagValuePost(@PathParam("tag_value") String value, @PathParam("s
         return tagValue(value, code);
     }
 
+    @GET
+    @Path("/api_security/sampling/{i}")
+    public Response apiSecuritySamplingWithStatus(@PathParam("i") int i) {
+        return Response.status(i)
+                .header("content-type", "text/plain")
+                .entity("Hello!\n").build();
+    }
+
+    @GET
+    @Path("/api_security_sampling/{i}")
+    public Response apiSecuritySampling(@PathParam("i") int i) {
+        return Response.status(200)
+                .header("content-type", "text/plain")
+                .entity("Hello!\n").build();
+    }
+
     @GET
     @Path("/params/{params: .*}")
     public String params(@PathParam("params") List<PathSegment> params) {

utils/build/docker/java/play/app/controllers/AppSecController.scala

@@ -94,6 +94,16 @@ class AppSecController @Inject()(cc: MessagesControllerComponents, ws: WSClient,
       .as("text/plain; charset=utf-8")
   }
 
+  def apiSecuritySamplingWithStatus(code: Int) = Action { request =>
+    Results.Status(code)("Hello!\n")
+      .as("text/plain; charset=utf-8")
+  }
+
+  def apiSecuritySampling(code: Int) = Action { request =>
+    Results.Status(200)("OK!\n")
+      .as("text/plain; charset=utf-8")
+  }
+
   def params(segments: Seq[String]) = Action {
     Results.Ok(segments.toString())
   }

utils/build/docker/java/play/conf/routes

@@ -3,6 +3,8 @@ GET  /healthcheck              controllers.AppSecController.healthcheck
 GET  /headers                  controllers.AppSecController.headers
 GET  /tag_value/:tag_value/:status_code   controllers.AppSecController.tagValue(tag_value: String, status_code: Int)
 POST /tag_value/:tag_value/:status_code   controllers.AppSecController.tagValuePost(tag_value: String, status_code: Int)
+GET  /api_security/sampling/:i controllers.AppSecController.apiSecuritySamplingWithStatus(i: Int)
+GET  /api_security_sampling/:i controllers.AppSecController.apiSecuritySampling(i: Int)
 GET  /params/*segments         controllers.AppSecController.params(segments: Seq[String])
 GET  /waf                      controllers.AppSecController.waf
 GET  /waf/                      controllers.AppSecController.waf

utils/build/docker/java/ratpack/src/main/java/com/datadoghq/ratpack/Main.java

@@ -177,6 +177,14 @@ public static void main(String[] args) throws Exception {
                                     ctx.getResponse().status(code).send("Value tagged");
                                 });
                             })
+                            .get("api_security/sampling/:i", ctx -> {
+                                final int i = Integer.parseInt(ctx.getPathTokens().get("i"));
+                                ctx.getResponse().status(i).send("Hello!\n");
+                            })
+                            .get("api_security_sampling/:i", ctx -> {
+                                final int i = Integer.parseInt(ctx.getPathTokens().get("i"));
+                                ctx.getResponse().status(200).send("OK!\n");
+                            })
                             .path("waf/:params?", ctx -> {
                                 HttpMethod method = ctx.getRequest().getMethod();
                                 if (method.equals(HttpMethod.GET)) {

utils/build/docker/java/resteasy-netty3/src/main/java/com/datadoghq/resteasy/MyResource.java

@@ -111,6 +111,22 @@ public Response tagValuePost(@PathParam("tag_value") String value, @PathParam("s
         return tagValue(value, code);
     }
 
+    @GET
+    @Path("/api_security/sampling/{i}")
+    public Response apiSecuritySamplingWithStatus(@PathParam("i") int i) {
+        return Response.status(i)
+                .header("content-type", "text/plain")
+                .entity("Hello!\n").build();
+    }
+
+    @GET
+    @Path("/api_security_sampling/{i}")
+    public Response apiSecuritySampling(@PathParam("i") int i) {
+        return Response.status(200)
+                .header("content-type", "text/plain")
+                .entity("Hello!\n").build();
+    }
+
     @GET
     @Path("/params/{params: .*}")
     public String params(@PathParam("params") List<PathSegment> params) {

utils/build/docker/java/spring-boot/src/main/java/com/datadoghq/system_tests/springboot/App.java

@@ -192,6 +192,16 @@ ResponseEntity<String> tagValueWithUrlencodedBody(@PathVariable final String tag
         return tagValue(tag_value, status_code);
     }
 
+    @GetMapping("/api_security/sampling/{i}")
+    ResponseEntity<String> apiSecuritySamplingWithStatus(@PathVariable final int i) {
+        return ResponseEntity.status(i).body("Hello!\n");
+    }
+
+    @GetMapping("/api_security_sampling/{i}")
+    ResponseEntity<String> apiSecuritySampling(@PathVariable final int i) {
+        return ResponseEntity.status(200).body("OK!\n");
+    }
+
     @RequestMapping("/waf/**")
     String waf() {
         return "Hello World!";

utils/build/docker/java/vertx3/src/main/java/com/datadoghq/vertx3/Main.java

@@ -97,6 +97,21 @@ public static void main(String[] args) {
                             .setStatusCode(Integer.parseInt(ctx.pathParam("status_code")))
                             .end("Value tagged");
                 });
+        router.get("/api_security/sampling/:i")
+                .produces("text/plain")
+                .handler(ctx -> {
+                    ctx.response()
+                            .setStatusCode(Integer.parseInt(ctx.pathParam("i")))
+                            .end("Hello!\n");
+                });
+        router.get("/api_security_sampling/:i")
+                .produces("text/plain")
+                .handler(ctx -> {
+                    final int i = Integer.parseInt(ctx.pathParam("i"));
+                    ctx.response()
+                            .setStatusCode(200)
+                            .end("Hello!\n");
+                });
         router.getWithRegex("/params(?:/([^/]*))?(?:/([^/]*))?(?:/([^/]*))?(?:/([^/]*))?(?:/([^/]*))?")
                 .produces("text/plain")
                 .handler(ctx ->

utils/build/docker/java/vertx4/src/main/java/com/datadoghq/vertx4/Main.java

@@ -94,6 +94,21 @@ public static void main(String[] args) {
                             .setStatusCode(Integer.parseInt(ctx.pathParam("status_code")))
                             .end("Value tagged");
                 });
+        router.get("/api_security/sampling/:i")
+                .produces("text/plain")
+                .handler(ctx -> {
+                    ctx.response()
+                            .setStatusCode(Integer.parseInt(ctx.pathParam("i")))
+                            .end("Hello!\n");
+                });
+        router.get("/api_security_sampling/:i")
+                .produces("text/plain")
+                .handler(ctx -> {
+                    final int i = Integer.parseInt(ctx.pathParam("i"));
+                    ctx.response()
+                            .setStatusCode(200)
+                            .end("Hello!\n");
+                });
         router.getWithRegex("/params(?:/([^/]*))?(?:/([^/]*))?(?:/([^/]*))?(?:/([^/]*))?(?:/([^/]*))?")
                 .produces("text/plain")
                 .handler(ctx ->