Expanding traffic_annotation_extractor clang tool to extract network calls.

tools/traffic_annotation/auditor/traffic_annotation_auditor.py

 #!/usr/bin/env python
 # Copyright (c) 2017 The Chromium Authors. All rights reserved.
 # Use of this source code is governed by a BSD-style license that can be
 # found in the LICENSE file.
 
 """This script is used to extract network traffic annotations from Chrome.
 Please refer to README.md for running steps."""
 
 import argparse
 import os
@@ skipping 57 matching lines @@
     raw_annotations += stdout_text
     if stderr_text:
       print stderr_text
   return raw_annotations
 
 
 def _ParsRawAnnotations(raw_annotations):
   """Parses raw annotations texts which are received from the clang tool.
   Args:
     raw_annotations: str Serialization of annotations and metadata. Each
-        annotation should have the following lines:
+        annotation should have either of the following lines:
         1- "==== NEW ANNOTATION ===="
         2- File path.
         3- Name of the function including this position.
         4- Line number.
         5- Function Type.
         6- Unique id of annotation.
         7- Completing id or group id, when applicable, empty otherwise.
         8- Serialization of annotation text (several lines)
         n- "==== ANNOTATION ENDS ===="
+        or:
+        1: "==== NEW CALL ===="
+        2: File path.
+        3: Name of the function in which the call is made.
+        4: Name of the called function.
+        5: Does the call have an annotation?
+        6: "==== CALL ENDS ===="
 
   Returns:
     annotations: ExtractedNetworkTrafficAnnotation A protobuf including all
       extracted annotations.
     metadata: list of dict List of metadata for each annotation. Each item
       includes the following fields:
         function_type: str Type of the function that defines the annotation.
         extra_id: str Possible prefix for annotation completion.
     errors: list of str List of errors.
   """
   annotations = traffic_annotation_pb2.ExtractedNetworkTrafficAnnotation()
   errors = []
   metadata = []
 
   lines = [line.strip("\r\n") for line in raw_annotations.split("\n")]
   current = 0
 
   try:
     while current < len(lines) - 1:
-      if lines[current] != "==== NEW ANNOTATION ====":
+      if lines[current] == "==== NEW ANNOTATION ====":
+        if current + 6 >= len(lines):
+          raise Exception(
+              "Not enough header lines at line %i." % current)
+
+        # Extract header lines.
+        source = traffic_annotation_pb2.NetworkTrafficAnnotation.TrafficSource()
+        source.file = lines[current + 1]
+        source.function = lines[current + 2]
+        source.line = int(lines[current + 3])
+        unique_id = lines[current + 5]
+
+        new_metadata = {'function_type': lines[current + 4],
+                        'extra_id': lines[current + 6],
+                        'unique_id_hash': _ComputeStringHash(unique_id)}
+        # Extract serialized proto.
+        current += 7
+        annotation_text = ""
+
+        while current < len(lines):
+          if lines[current] == "==== ANNOTATION ENDS ====":
+            break
+          else:
+            annotation_text += lines[current]
+          current += 1
+        else:
+          raise Exception(
+            "Error at line %i, expected annotation end tag." % current)
+        current += 1
+
+        # Process unittests and undefined tags.
+        if unique_id in ("test", "test_partial"):
+          continue
+        if unique_id in ("undefined", "missing"):
+          errors.append("Annotation is not defined for file '%s', line %i." %
+              (source.file, source.line))
+          continue
+
+        # Decode serialized proto.
+        annotation_proto = traffic_annotation_pb2.NetworkTrafficAnnotation()
+        try:
+          text_format.Parse(annotation_text, annotation_proto)
+        except Exception as error:
+          errors.append("Annotation in file '%s', line %i, has an error: %s" %
+              (source.file, source.line, error))
+
+        # Add new proto.
+        annotation_proto.unique_id = unique_id
+        annotation_proto.source.CopyFrom(source)
+        annotations.network_traffic_annotation.add().CopyFrom(annotation_proto)
+        metadata.append(new_metadata)
+      elif lines[current] == "==== NEW CALL ====":
+        # Ignore calls for now.
+        while current < len(lines):
+          if lines[current] == "==== CALL ENDS ====":
+            break
+          current += 1
+        else:
+          raise Exception(
+              "Error at line %i, expected call end tag." % current)
+        current += 1
+      else: # The line is neither new annotation nor new call.
         raise Exception(
-            "Error at line %i, expected starting new annotaion." % current)
-      if current + 5 >= len(lines):
-        raise Exception(
-            "Not enough header lines at line %i." % current)
-
-      # Extract header lines.
-      source = traffic_annotation_pb2.NetworkTrafficAnnotation.TrafficSource()
-      source.file = lines[current + 1]
-      source.function = lines[current + 2]
-      source.line = int(lines[current + 3])
-      unique_id = lines[current + 5]
-
-      new_metadata = {'function_type': lines[current + 4],
-                      'extra_id': lines[current + 6],
-                      'unique_id_hash': _ComputeStringHash(unique_id)}
-      # Extract serialized proto.
-      current += 7
-      annotation_text = ""
-
-      while current < len(lines):
-        current += 1
-        if lines[current - 1] == "==== ANNOTATION ENDS ====":
-          break
-        else:
-          annotation_text += lines[current - 1]
-      else:
-        raise Exception(
-          "Error at line %i, expected annotation end tag." % current)
-
-      # Process unittests and undefined tags.
-      if unique_id in ("test", "test_partial"):
-        continue
-      if unique_id in ("undefined", "missing"):
-        errors.append("Annotation is not defined for file '%s', line %i." %
-            (source.file, source.line))
-        continue
-
-      # Decode serialized proto.
-      annotation_proto = traffic_annotation_pb2.NetworkTrafficAnnotation()
-      try:
-        text_format.Parse(annotation_text, annotation_proto)
-      except Exception as error:
-        errors.append("Annotation in file '%s', line %i, has error: %s" %
-            (source.file, source.line, error))
-
-      # Add new proto.
-      annotation_proto.unique_id = unique_id
-      annotation_proto.source.CopyFrom(source)
-      annotations.network_traffic_annotation.add().CopyFrom(annotation_proto)
-      metadata.append(new_metadata)
+            "Error at line %i, expected starting new annotation or call." %
+            current)
 
   except Exception as error:
     errors.append(str(error))
 
   print "Extracted %i annotations with %i errors." % \
     (len(annotations.network_traffic_annotation), len(errors))
   return annotations, metadata, errors
 
 
 def _WriteSummaryFile(annotations, metadata, errors, file_path):
@@ skipping 96 matching lines @@
     _WriteSummaryFile(annotations, metadata, errors, args.summary_file)
 
   if args.hash_codes_file:
     _WriteHashCodesFile(annotations, metadata, args.hash_codes_file)
 
   return 0
 
 
 if __name__ == '__main__':
   sys.exit(main())