xref: /aosp_15_r20/art/test/1339-dead-reference-safe/src/ReachabilityFenceTest.java (revision 795d594fd825385562da6b089ea9b2033f3abf5a)

/*
 * 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.
 */

// DeadReferenceSafeTest, but with a reachabilityFence.

import dalvik.annotation.optimization.DeadReferenceSafe;
import java.lang.ref.Reference;
import java.util.concurrent.atomic.AtomicInteger;

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

  private static void $noinline$loop() {
    ReachabilityFenceTest x;
    // Each loop allocates INNER_ITERS ReachabilitySenstiveTest objects.
    for (int i = 0; i < INNER_ITERS; ++i) {
      // We've allocated i objects so far.
      x = new ReachabilityFenceTest();
      count += x.n;
      // x is dead here.
      if (i == 5) {
        // The rechabilityFence should keep the last allocated object reachable.
        // Thus the last instance should not be finalized.
        Main.$noinline$gcAndCheck(nFinalized, 5, "ReachabilityFence",
            "reachabilityFence failed to keep object live.");
      }
      Reference.reachabilityFence(x);
    }
  }

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

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

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

    $noinline$loop();

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