| /* |
| * Copyright (C) 2016 The Android Open Source Project |
| * |
| * Licensed under the Apache License, Version 2.0 (the "License"); |
| * you may not use this file except in compliance with the License. |
| * You may obtain a copy of the License at |
| * |
| * http://www.apache.org/licenses/LICENSE-2.0 |
| * |
| * Unless required by applicable law or agreed to in writing, software |
| * distributed under the License is distributed on an "AS IS" BASIS, |
| * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| * See the License for the specific language governing permissions and |
| * limitations under the License. |
| */ |
| |
| package art; |
| |
| import java.util.ArrayList; |
| import java.util.Collections; |
| import java.util.Random; |
| |
| public class Test906 { |
| public static void run() throws Exception { |
| doTest(); |
| } |
| |
| // Number of times we will try to count the heap in various ways. If we are unlucky and end up in |
| // the middle of a GC we could incorrectly fail. This is expected to be incredibly rare so 10 |
| // retries should be more than sufficient. |
| private static final int ITERATE_RETRIES = 10; |
| private static final class Foobar {} |
| |
| private static void testHeapCount() throws Exception { |
| IllegalStateException lastThrow = new IllegalStateException( |
| "Failed to get consistent counts after " + ITERATE_RETRIES + " retries"); |
| Foobar[] foobars = new Foobar[123]; |
| for (int i = 0; i < foobars.length; i++) { |
| foobars[i] = new Foobar(); |
| } |
| for (int i = 0; i < ITERATE_RETRIES; i++) { |
| try { |
| int all = iterateThroughHeapCount(0, null, Integer.MAX_VALUE); |
| int tagged = iterateThroughHeapCount(HEAP_FILTER_OUT_UNTAGGED, null, Integer.MAX_VALUE); |
| int untagged = iterateThroughHeapCount(HEAP_FILTER_OUT_TAGGED, null, Integer.MAX_VALUE); |
| int taggedClass = iterateThroughHeapCount(HEAP_FILTER_OUT_CLASS_UNTAGGED, null, |
| Integer.MAX_VALUE); |
| int untaggedClass = iterateThroughHeapCount(HEAP_FILTER_OUT_CLASS_TAGGED, null, |
| Integer.MAX_VALUE); |
| int filteredClass = iterateThroughHeapCount(0, Foobar.class, Integer.MAX_VALUE); |
| |
| if (all != tagged + untagged) { |
| throw new IllegalStateException("Instances: " + all + " != " + tagged + " + " + untagged); |
| } |
| if (all != taggedClass + untaggedClass) { |
| throw new IllegalStateException("By class: " + all + " != " + taggedClass + " + " + |
| untaggedClass); |
| } |
| if (filteredClass != foobars.length) { |
| throw new IllegalStateException( |
| "Missed objects of foobar type. " + filteredClass + " != " + foobars.length); |
| } |
| if (tagged != 6) { |
| throw new IllegalStateException(tagged + " tagged objects"); |
| } |
| if (taggedClass != 2) { |
| throw new IllegalStateException(tagged + " objects with tagged class"); |
| } |
| if (all == tagged) { |
| throw new IllegalStateException("All objects tagged"); |
| } |
| if (all == taggedClass) { |
| throw new IllegalStateException("All objects have tagged class"); |
| } |
| // Everything worked! |
| return; |
| } catch (IllegalStateException e) { |
| lastThrow.addSuppressed(e); |
| } |
| } |
| throw lastThrow; |
| } |
| |
| private static Object[] GenTs(Class<?> k) throws Exception { |
| Object[] ret = new Object[new Random().nextInt(100) + 10]; |
| for (int i = 0; i < ret.length; i++) { |
| ret[i] = k.newInstance(); |
| } |
| return ret; |
| } |
| |
| private static void checkEq(int a, int b) { |
| if (a != b) { |
| Error e = new Error("Failed: Expected equal " + a + " and " + b); |
| System.out.println(e); |
| e.printStackTrace(System.out); |
| } |
| } |
| |
| public static class Foo {} |
| public static class Bar extends Foo {} |
| public static class Baz extends Bar {} |
| public static class Alpha extends Bar {} |
| public static class MISSING extends Baz {} |
| public interface Iface {} |
| public static class Beta implements Iface {} |
| public static class Gamma implements Iface {} |
| public static class Delta extends Beta {} |
| private static void testIterateOverInstances() throws Exception { |
| Object[] foos = GenTs(Foo.class); |
| Object[] bars = GenTs(Bar.class); |
| Object[] bazs = GenTs(Baz.class); |
| Object[] alphas = GenTs(Alpha.class); |
| Object[] betas = GenTs(Beta.class); |
| Object[] gammas = GenTs(Gamma.class); |
| Object[] deltas = GenTs(Delta.class); |
| checkEq(0, iterateOverInstancesCount(MISSING.class)); |
| checkEq(alphas.length, iterateOverInstancesCount(Alpha.class)); |
| checkEq(bazs.length, iterateOverInstancesCount(Baz.class)); |
| checkEq(bazs.length + alphas.length + bars.length, iterateOverInstancesCount(Bar.class)); |
| checkEq(bazs.length + alphas.length + bars.length + foos.length, |
| iterateOverInstancesCount(Foo.class)); |
| checkEq(betas.length + gammas.length + deltas.length, |
| iterateOverInstancesCount(Iface.class)); |
| } |
| |
| public static void doTest() throws Exception { |
| A a = new A(); |
| B b = new B(); |
| B b2 = new B(); |
| C c = new C(); |
| A[] aArray = new A[5]; |
| String s = "Hello World"; |
| |
| setTag(a, 1); |
| setTag(b, 2); |
| setTag(b2, 3); |
| setTag(aArray, 4); |
| setTag(s, 5); |
| setTag(B.class, 100); |
| |
| testHeapCount(); |
| |
| testIterateOverInstances(); |
| |
| long classTags[] = new long[100]; |
| long sizes[] = new long[100]; |
| long tags[] = new long[100]; |
| int lengths[] = new int[100]; |
| |
| int n = iterateThroughHeapData(HEAP_FILTER_OUT_UNTAGGED, null, classTags, sizes, tags, lengths); |
| System.out.println(sort(n, classTags, sizes, tags, lengths)); |
| |
| iterateThroughHeapAdd(HEAP_FILTER_OUT_UNTAGGED, null); |
| n = iterateThroughHeapData(HEAP_FILTER_OUT_UNTAGGED, null, classTags, sizes, tags, lengths); |
| System.out.println(sort(n, classTags, sizes, tags, lengths)); |
| |
| System.out.println(iterateThroughHeapString(getTag(s))); |
| System.out.println(getTag(s)); |
| |
| boolean[] zArray = new boolean[] { false, true }; |
| setTag(zArray, 1); |
| System.out.println(iterateThroughHeapPrimitiveArray(getTag(zArray))); |
| System.out.println(getTag(zArray)); |
| |
| byte[] bArray = new byte[] { 1, 2, 3 }; |
| setTag(bArray, 1); |
| System.out.println(iterateThroughHeapPrimitiveArray(getTag(bArray))); |
| System.out.println(getTag(bArray)); |
| |
| char[] cArray = new char[] { 'A', 'Z' }; |
| setTag(cArray, 1); |
| System.out.println(iterateThroughHeapPrimitiveArray(getTag(cArray))); |
| System.out.println(getTag(cArray)); |
| |
| short[] sArray = new short[] { 1, 2, 3 }; |
| setTag(sArray, 1); |
| System.out.println(iterateThroughHeapPrimitiveArray(getTag(sArray))); |
| System.out.println(getTag(sArray)); |
| |
| int[] iArray = new int[] { 1, 2, 3 }; |
| setTag(iArray, 1); |
| System.out.println(iterateThroughHeapPrimitiveArray(getTag(iArray))); |
| System.out.println(getTag(iArray)); |
| |
| float[] fArray = new float[] { 0.0f, 1.0f }; |
| setTag(fArray, 1); |
| System.out.println(iterateThroughHeapPrimitiveArray(getTag(fArray))); |
| System.out.println(getTag(fArray)); |
| |
| long[] lArray = new long[] { 1, 2, 3 }; |
| setTag(lArray, 1); |
| System.out.println(iterateThroughHeapPrimitiveArray(getTag(lArray))); |
| System.out.println(getTag(lArray)); |
| |
| double[] dArray = new double[] { 0.0, 1.0 }; |
| setTag(dArray, 1); |
| System.out.println(iterateThroughHeapPrimitiveArray(getTag(dArray))); |
| System.out.println(getTag(dArray)); |
| |
| // Force GCs to clean up dirt. |
| Runtime.getRuntime().gc(); |
| Runtime.getRuntime().gc(); |
| |
| doTestPrimitiveFieldsClasses(); |
| |
| doTestPrimitiveFieldsIntegral(); |
| |
| // Force GCs to clean up dirt. |
| Runtime.getRuntime().gc(); |
| Runtime.getRuntime().gc(); |
| |
| doTestPrimitiveFieldsFloat(); |
| |
| // Force GCs to clean up dirt. |
| Runtime.getRuntime().gc(); |
| Runtime.getRuntime().gc(); |
| } |
| |
| private static void doTestPrimitiveFieldsClasses() { |
| System.out.println("doTestPrimitiveFieldsClasses"); |
| setTag(IntObject.class, 10000); |
| System.out.println(iterateThroughHeapPrimitiveFields(10000)); |
| System.out.println(getTag(IntObject.class)); |
| setTag(IntObject.class, 0); |
| |
| setTag(FloatObject.class, 10000); |
| System.out.println(iterateThroughHeapPrimitiveFields(10000)); |
| System.out.println(getTag(FloatObject.class)); |
| setTag(FloatObject.class, 0); |
| |
| boolean correctHeapValue = false; |
| setTag(Inf1.class, 10000); |
| String heapTrace = iterateThroughHeapPrimitiveFields(10000); |
| |
| if (!checkInitialized(Inf1.class)) { |
| correctHeapValue = heapTrace.equals("10000@0 (static, int, index=0) 0000000000000000"); |
| } else { |
| correctHeapValue = heapTrace.equals("10000@0 (static, int, index=0) 0000000000000001"); |
| } |
| |
| if (!correctHeapValue) |
| System.out.println("Heap Trace for Inf1 is not as expected:\n" + heapTrace); |
| |
| System.out.println(getTag(Inf1.class)); |
| setTag(Inf1.class, 0); |
| |
| setTag(Inf2.class, 10000); |
| heapTrace = iterateThroughHeapPrimitiveFields(10000); |
| |
| if (!checkInitialized(Inf2.class)) { |
| correctHeapValue = heapTrace.equals("10000@0 (static, int, index=1) 0000000000000000"); |
| } else { |
| correctHeapValue = heapTrace.equals("10000@0 (static, int, index=1) 0000000000000001"); |
| } |
| |
| if (!correctHeapValue) |
| System.out.println("Heap Trace for Inf2 is not as expected:\n" + heapTrace); |
| System.out.println(getTag(Inf2.class)); |
| |
| setTag(Inf2.class, 0); |
| } |
| |
| private static void doTestPrimitiveFieldsIntegral() { |
| System.out.println("doTestPrimitiveFieldsIntegral"); |
| IntObject intObject = new IntObject(); |
| setTag(intObject, 10000); |
| System.out.println(iterateThroughHeapPrimitiveFields(10000)); |
| System.out.println(getTag(intObject)); |
| } |
| |
| private static void doTestPrimitiveFieldsFloat() { |
| System.out.println("doTestPrimitiveFieldsFloat"); |
| FloatObject floatObject = new FloatObject(); |
| setTag(floatObject, 10000); |
| System.out.println(iterateThroughHeapPrimitiveFields(10000)); |
| System.out.println(getTag(floatObject)); |
| } |
| |
| static class A { |
| } |
| |
| static class B { |
| } |
| |
| static class C { |
| } |
| |
| static class HeapElem implements Comparable<HeapElem> { |
| long classTag; |
| long size; |
| long tag; |
| int length; |
| |
| public int compareTo(HeapElem other) { |
| if (tag != other.tag) { |
| return Long.compare(tag, other.tag); |
| } |
| if (classTag != other.classTag) { |
| return Long.compare(classTag, other.classTag); |
| } |
| if (size != other.size) { |
| return Long.compare(size, other.size); |
| } |
| return Integer.compare(length, other.length); |
| } |
| |
| public String toString() { |
| return "{tag=" + tag + ", class-tag=" + classTag + ", size=" + |
| (tag >= 100 ? "<class>" : size) // Class size is dependent on 32-bit vs 64-bit, |
| // so strip it. |
| + ", length=" + length + "}"; |
| } |
| } |
| |
| private static ArrayList<HeapElem> sort(int n, long classTags[], long sizes[], long tags[], |
| int lengths[]) { |
| ArrayList<HeapElem> ret = new ArrayList<HeapElem>(n); |
| for (int i = 0; i < n; i++) { |
| HeapElem elem = new HeapElem(); |
| elem.classTag = classTags[i]; |
| elem.size = sizes[i]; |
| elem.tag = tags[i]; |
| elem.length = lengths[i]; |
| ret.add(elem); |
| } |
| Collections.sort(ret); |
| return ret; |
| } |
| |
| private static interface Inf1 { |
| public final static int A = 1; |
| } |
| |
| private static interface Inf2 extends Inf1 { |
| public final static int B = 1; |
| } |
| |
| private static class IntObject implements Inf1 { |
| byte b = (byte)1; |
| char c= 'a'; |
| short s = (short)2; |
| int i = 3; |
| long l = 4; |
| Object o = new Object(); |
| static int sI = 5; |
| } |
| |
| private static class FloatObject extends IntObject implements Inf2 { |
| float f = 1.23f; |
| double d = 1.23; |
| Object p = new Object(); |
| static int sI = 6; |
| } |
| |
| private final static int HEAP_FILTER_OUT_TAGGED = 0x4; |
| private final static int HEAP_FILTER_OUT_UNTAGGED = 0x8; |
| private final static int HEAP_FILTER_OUT_CLASS_TAGGED = 0x10; |
| private final static int HEAP_FILTER_OUT_CLASS_UNTAGGED = 0x20; |
| |
| private static void setTag(Object o, long tag) { |
| Main.setTag(o, tag); |
| } |
| private static long getTag(Object o) { |
| return Main.getTag(o); |
| } |
| |
| private static native int iterateOverInstancesCount(Class<?> klass); |
| |
| private static native boolean checkInitialized(Class<?> klass); |
| private static native int iterateThroughHeapCount(int heapFilter, |
| Class<?> klassFilter, int stopAfter); |
| private static native int iterateThroughHeapData(int heapFilter, |
| Class<?> klassFilter, long classTags[], long sizes[], long tags[], int lengths[]); |
| private static native int iterateThroughHeapAdd(int heapFilter, |
| Class<?> klassFilter); |
| private static native String iterateThroughHeapString(long tag); |
| private static native String iterateThroughHeapPrimitiveArray(long tag); |
| private static native String iterateThroughHeapPrimitiveFields(long tag); |
| } |