test/1339-dead-reference-safe/src/DeadReferenceUnsafeTest.java

/*
 * Copyright (C) 2019 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.
 */

import java.util.concurrent.atomic.AtomicInteger;

public final class DeadReferenceUnsafeTest {
  static AtomicInteger nFinalized = new AtomicInteger(0);
  private static final int INNER_ITERS = 10;
  static int count;
  int n = 1;

  private static void $noinline$loop() {
    DeadReferenceUnsafeTest x;
    // The loop allocates INNER_ITERS DeadReferenceUnsafeTest objects.
    for (int i = 0; i < INNER_ITERS; ++i) {
      // We've allocated i objects so far.
      x = new DeadReferenceUnsafeTest();
      count += x.n;
      // x is dead here.
      if (i == 5) {
        // Without dead reference elimination, the last object should be kept around,
        // and only 5 objects should be relcaimed here.
        Main.$noinline$gcAndCheck(nFinalized, 5, "DeadReferenceUnsafe",
            "Failed to keep dead reference live in unannotated code!");
      }
    }
  }

  private static void reset(int expected_count) {
    Runtime.getRuntime().gc();
    System.runFinalization();
    if (nFinalized.get() != expected_count) {
      System.out.println("DeadReferenceUnsafeTest: Wrong number of finalized objects:"
                         + nFinalized.get());
    }
    nFinalized.set(0);
  }

  protected void finalize() {
    nFinalized.incrementAndGet();
  }

  public static void runTest() {
    try {
      Main.ensureCompiled(DeadReferenceUnsafeTest.class, "$noinline$loop");
    } catch (NoSuchMethodException e) {
      System.out.println("Unexpectedly threw " + e);
    }

    $noinline$loop();

    if (count != INNER_ITERS) {
      System.out.println("DeadReferenceUnsafeTest: Final count wrong: " + count);
    }
    reset(INNER_ITERS);
  }
}