blob: c65934d3ff1569c9b81fd826484a28f5fc686313 [file] [log] [blame]
/*
* 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);
}