From c5abfe4cea91f7ffcbaaa72b8f636534a12629d0 Mon Sep 17 00:00:00 2001
From: Ben Gruver <bgruv@google.com>
Date: Tue, 5 Mar 2013 23:11:13 -0800
Subject: [PATCH] Annotate AnnotationDirectoryItems

---
 .../raw/AnnotationDirectoryItem.java          | 119 ++++++++++++++++++
 .../dexbacked/raw/AnnotationSetItem.java      |   9 ++
 .../dexbacked/raw/AnnotationSetRefList.java   |   9 ++
 .../dexbacked/raw/util/DexAnnotator.java      |  10 +-
 4 files changed, 143 insertions(+), 4 deletions(-)
 create mode 100644 dexlib2/src/main/java/org/jf/dexlib2/dexbacked/raw/AnnotationDirectoryItem.java

diff --git a/dexlib2/src/main/java/org/jf/dexlib2/dexbacked/raw/AnnotationDirectoryItem.java b/dexlib2/src/main/java/org/jf/dexlib2/dexbacked/raw/AnnotationDirectoryItem.java
new file mode 100644
index 00000000..712233bd
--- /dev/null
+++ b/dexlib2/src/main/java/org/jf/dexlib2/dexbacked/raw/AnnotationDirectoryItem.java
@@ -0,0 +1,119 @@
+/*
+ * Copyright 2013, Google Inc.
+ * All rights reserved.
+ *
+ * Redistribution and use in source and binary forms, with or without
+ * modification, are permitted provided that the following conditions are
+ * met:
+ *
+ *     * Redistributions of source code must retain the above copyright
+ * notice, this list of conditions and the following disclaimer.
+ *     * Redistributions in binary form must reproduce the above
+ * copyright notice, this list of conditions and the following disclaimer
+ * in the documentation and/or other materials provided with the
+ * distribution.
+ *     * Neither the name of Google Inc. nor the names of its
+ * contributors may be used to endorse or promote products derived from
+ * this software without specific prior written permission.
+ *
+ * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
+ * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
+ * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
+ * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
+ * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
+ * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
+ * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
+ * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
+ * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
+ * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
+ * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
+ */
+
+package org.jf.dexlib2.dexbacked.raw;
+
+import org.jf.dexlib2.dexbacked.raw.util.DexAnnotator;
+import org.jf.dexlib2.util.AnnotatedBytes;
+
+import javax.annotation.Nonnull;
+import javax.annotation.Nullable;
+
+public class AnnotationDirectoryItem {
+    public static final int CLASS_ANNOTATIONS_OFFSET  = 0;
+    public static final int FIELD_SIZE_OFFSET = 4;
+    public static final int ANNOTATED_METHOD_SIZE_OFFSET = 8;
+    public static final int ANNOTATED_PARAMETERS_SIZE = 12;
+
+    @Nonnull
+    public static SectionAnnotator makeAnnotator(@Nonnull DexAnnotator annotator, @Nonnull MapItem mapItem) {
+        return new SectionAnnotator(annotator, mapItem) {
+            @Nonnull @Override public String getItemName() {
+                return "annotation_directory_item";
+            }
+
+            @Override public int getItemAlignment() {
+                return 4;
+            }
+
+            @Override
+            protected void annotateItem(@Nonnull AnnotatedBytes out, int itemIndex, @Nullable String itemIdentity) {
+                int classAnnotationsOffset = dexFile.readSmallUint(out.getCursor());
+                out.annotate(4, "class_annotations_off = %s",
+                        AnnotationSetItem.getReferenceAnnotation(dexFile, classAnnotationsOffset));
+
+                int fieldsSize = dexFile.readSmallUint(out.getCursor());
+                out.annotate(4, "fields_size = %d", fieldsSize);
+
+                int annotatedMethodsSize = dexFile.readSmallUint(out.getCursor());
+                out.annotate(4, "annotated_methods_size = %d", annotatedMethodsSize);
+
+                int annotatedParameterSize = dexFile.readSmallUint(out.getCursor());
+                out.annotate(4, "annotated_parameters_size = %d", annotatedParameterSize);
+
+                if (fieldsSize > 0) {
+                    out.annotate(0, "field_annotations:");
+                    out.indent();
+                    for (int i=0; i<fieldsSize; i++) {
+                        out.annotate(0, "field_annotation[%d]", i);
+                        out.indent();
+                        int fieldIndex = dexFile.readSmallUint(out.getCursor());
+                        out.annotate(4, "%s", FieldIdItem.getReferenceAnnotation(dexFile, fieldIndex));
+                        int annotationOffset = dexFile.readSmallUint(out.getCursor());
+                        out.annotate(4, "%s", AnnotationSetItem.getReferenceAnnotation(dexFile, annotationOffset));
+                        out.deindent();
+                    }
+                    out.deindent();
+                }
+
+                if (annotatedMethodsSize > 0) {
+                    out.annotate(0, "method_annotations:");
+                    out.indent();
+                    for (int i=0; i<annotatedMethodsSize; i++) {
+                        out.annotate(0, "method_annotation[%d]", i);
+                        out.indent();
+                        int methodIndex = dexFile.readSmallUint(out.getCursor());
+                        out.annotate(4, "%s", MethodIdItem.getReferenceAnnotation(dexFile, methodIndex));
+                        int annotationOffset = dexFile.readSmallUint(out.getCursor());
+                        out.annotate(4, "%s", AnnotationSetItem.getReferenceAnnotation(dexFile, annotationOffset));
+                        out.deindent();
+                    }
+                    out.deindent();
+                }
+
+                if (annotatedParameterSize > 0) {
+                    out.annotate(0, "parameter_annotations:");
+                    out.indent();
+                    for (int i=0; i<annotatedParameterSize; i++) {
+                        out.annotate(0, "parameter_annotation[%d]", i);
+                        out.indent();
+                        int methodIndex = dexFile.readSmallUint(out.getCursor());
+                        out.annotate(4, "%s", MethodIdItem.getReferenceAnnotation(dexFile, methodIndex));
+                        int annotationOffset = dexFile.readSmallUint(out.getCursor());
+                        out.annotate(4, "%s", AnnotationSetRefList.getReferenceAnnotation(dexFile, annotationOffset));
+                        out.deindent();
+                    }
+                    out.deindent();
+                }
+            }
+        };
+    }
+}
diff --git a/dexlib2/src/main/java/org/jf/dexlib2/dexbacked/raw/AnnotationSetItem.java b/dexlib2/src/main/java/org/jf/dexlib2/dexbacked/raw/AnnotationSetItem.java
index 7ed99bc7..ca5d8d22 100644
--- a/dexlib2/src/main/java/org/jf/dexlib2/dexbacked/raw/AnnotationSetItem.java
+++ b/dexlib2/src/main/java/org/jf/dexlib2/dexbacked/raw/AnnotationSetItem.java
@@ -31,6 +31,7 @@
 
 package org.jf.dexlib2.dexbacked.raw;
 
+import org.jf.dexlib2.dexbacked.DexBackedDexFile;
 import org.jf.dexlib2.dexbacked.raw.util.DexAnnotator;
 import org.jf.dexlib2.util.AnnotatedBytes;
 
@@ -64,4 +65,12 @@ public class AnnotationSetItem {
             }
         };
     }
+
+    @Nonnull
+    public static String getReferenceAnnotation(@Nonnull DexBackedDexFile dexFile, int annotationSetOffset) {
+        if (annotationSetOffset == 0) {
+            return "annotation_set_item[NO_OFFSET]";
+        }
+        return String.format("annotation_set_item[0x%x]", annotationSetOffset);
+    }
 }
diff --git a/dexlib2/src/main/java/org/jf/dexlib2/dexbacked/raw/AnnotationSetRefList.java b/dexlib2/src/main/java/org/jf/dexlib2/dexbacked/raw/AnnotationSetRefList.java
index 13ab06dc..2b7ef012 100644
--- a/dexlib2/src/main/java/org/jf/dexlib2/dexbacked/raw/AnnotationSetRefList.java
+++ b/dexlib2/src/main/java/org/jf/dexlib2/dexbacked/raw/AnnotationSetRefList.java
@@ -31,6 +31,7 @@
 
 package org.jf.dexlib2.dexbacked.raw;
 
+import org.jf.dexlib2.dexbacked.DexBackedDexFile;
 import org.jf.dexlib2.dexbacked.raw.util.DexAnnotator;
 import org.jf.dexlib2.util.AnnotatedBytes;
 
@@ -64,4 +65,12 @@ public class AnnotationSetRefList {
             }
         };
     }
+
+    @Nonnull
+    public static String getReferenceAnnotation(@Nonnull DexBackedDexFile dexFile, int annotationSetRefListOffset) {
+        if (annotationSetRefListOffset == 0) {
+            return "annotation_set_ref_list[NO_OFFSET]";
+        }
+        return String.format("annotation_set_ref_list[0x%x]", annotationSetRefListOffset);
+    }
 }
diff --git a/dexlib2/src/main/java/org/jf/dexlib2/dexbacked/raw/util/DexAnnotator.java b/dexlib2/src/main/java/org/jf/dexlib2/dexbacked/raw/util/DexAnnotator.java
index fc3cb6a1..2a740823 100644
--- a/dexlib2/src/main/java/org/jf/dexlib2/dexbacked/raw/util/DexAnnotator.java
+++ b/dexlib2/src/main/java/org/jf/dexlib2/dexbacked/raw/util/DexAnnotator.java
@@ -139,6 +139,11 @@ public class DexAnnotator extends AnnotatedBytes {
                 case ItemType.ENCODED_ARRAY_ITEM:
                     annotators.put(mapItem.getType(), EncodedArrayItem.makeAnnotator(this, mapItem));
                     break;
+                case ItemType.ANNOTATION_DIRECTORY_ITEM:
+                    annotators.put(mapItem.getType(), AnnotationDirectoryItem.makeAnnotator(this, mapItem));
+                    break;
+                default:
+                    throw new RuntimeException(String.format("Unrecognized item type: 0x%x", mapItem.getType()));
             }
         }
     }
@@ -156,10 +161,7 @@ public class DexAnnotator extends AnnotatedBytes {
 
         for (MapItem mapItem: mapItems) {
             SectionAnnotator annotator = annotators.get(mapItem.getType());
-            //TODO: temporary null check, until all item types are implemented
-            if (annotator != null) {
-                annotator.annotateSection(this);
-            }
+            annotator.annotateSection(this);
         }
 
         dexFile.writeAnnotations(out, this);