Add a field to the network service proto of Tsunami to keep track of supported HTTP methods.

PiperOrigin-RevId: 591823049
Change-Id: I81dc68aa44023248538ecde5fa5f73fb96b9ff89

Author: tooryx

common/src/main/java/com/google/tsunami/common/data/NetworkServiceUtils.java

@@ -68,7 +68,10 @@ public static boolean isWebService(Optional<String> serviceName) {
 
   public static boolean isWebService(NetworkService networkService) {
     checkNotNull(networkService);
-    return isWebService(Optional.of(networkService.getServiceName()));
+    // A web-service is a service that is either flagged as http by nmap or one that supports at
+    // least one HTTP method.
+    return (networkService.getSupportedHttpMethodsCount() > 0)
+        || isWebService(Optional.of(networkService.getServiceName()));
   }
 
   public static boolean isPlainHttp(NetworkService networkService) {

common/src/test/java/com/google/tsunami/common/data/NetworkServiceUtilsTest.java

@@ -105,6 +105,17 @@ public void isWebService_whenCapitalizedHttpService_ignoresCaseAndReturnsTrue()
         .isTrue();
   }
 
+  @Test
+  public void isWebService_whenHasAtLeastOneHttpMethod_returnsTrue() {
+    assertThat(
+            NetworkServiceUtils.isWebService(
+                NetworkService.newBuilder()
+                    .setServiceName("irrelevantService")
+                    .addSupportedHttpMethods("IrrelevantMethodName")
+                    .build()))
+        .isTrue();
+  }
+
   @Test
   public void isWebService_whenNonWebService_returnsFalse() {
     assertThat(

proto/network_service.proto

@@ -58,6 +58,9 @@ message NetworkService {
 
   // List of supported SSL versions (e.g. TLSv1, SSLv3, ...) on the service.
   repeated string supported_ssl_versions = 9;
+
+  // List of supported HTTP methods (e.g. POST, GET, ...) on the service.
+  repeated string supported_http_methods = 10;
 }
 
 // Context information about a specific network service.