test/2000-virtual-list-structural/src/Main.java

/*
 * Copyright (C) 2017 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 art.*;
import java.util.*;
import java.lang.invoke.*;
import java.io.*;
import java.util.zip.*;

public class Main {
  public static final String TEST_NAME = "2000-virtual-list-structural";
  public static final boolean PRINT_COUNT = false;
  public static MethodHandles.Lookup lookup = MethodHandles.publicLookup();
  public static MethodHandle getcnt;
  public static MethodHandle get_total_cnt;
  public static void GetHandles() throws Throwable {
    getcnt = lookup.findGetter(AbstractCollection.class, "cnt", Integer.TYPE);
    get_total_cnt = lookup.findStaticGetter(AbstractCollection.class, "TOTAL_COUNT", Integer.TYPE);
  }

  public static byte[] GetDexBytes() throws Throwable {
    String jar_loc = System.getenv("DEX_LOCATION") + "/" + TEST_NAME + "-ex.jar";
    try (ZipFile zip = new ZipFile(new File(jar_loc))) {
      ZipEntry entry = zip.getEntry("classes.dex");
      try (InputStream is = zip.getInputStream(entry)) {
        byte[] res = new byte[(int)entry.getSize()];
        is.read(res);
        return res;
      }
    }
  }
  public static void PrintListAndData(AbstractCollection<String> c) throws Throwable {
    if (PRINT_COUNT) {
      System.out.println("List is: " + c + " count = " + getcnt.invoke(c) + " TOTAL_COUNT = " + get_total_cnt.invoke());
    } else {
      System.out.println("List is: " + c);
    }
  }
  public static void main(String[] args) throws Throwable {
    AbstractCollection<String> l1 = (AbstractCollection<String>)Arrays.asList("a", "b", "c", "d");
    AbstractCollection<String> l2 = new ArrayList<>();
    l2.add("1");
    l2.add("2");
    l2.add("3");
    l2.add("4");
    Redefinition.doCommonStructuralClassRedefinition(AbstractCollection.class, GetDexBytes());
    GetHandles();
    AbstractCollection<String> l3 = new HashSet<>(l2);
    AbstractCollection<String> l4 = new LinkedList<>(l1);
    PrintListAndData(l1);
    PrintListAndData(l2);
    for (int i = 0; i < 1000; i++) {
      l2.add("xyz: " + i);
    }
    PrintListAndData(l2);
    PrintListAndData(l3);
    PrintListAndData(l4);
    CheckLE(getcnt.invoke(l1), get_total_cnt.invoke());
    CheckLE(getcnt.invoke(l2), get_total_cnt.invoke());
    CheckLE(getcnt.invoke(l3), get_total_cnt.invoke());
    CheckLE(getcnt.invoke(l4), get_total_cnt.invoke());
    CheckEQ(getcnt.invoke(l1), 0);
    CheckLE(getcnt.invoke(l2), 0);
    CheckLE(getcnt.invoke(l1), getcnt.invoke(l2));
    CheckLE(getcnt.invoke(l1), getcnt.invoke(l3));
    CheckLE(getcnt.invoke(l1), getcnt.invoke(l4));
    CheckLE(getcnt.invoke(l2), getcnt.invoke(l3));
    CheckLE(getcnt.invoke(l2), getcnt.invoke(l4));
    CheckLE(getcnt.invoke(l3), getcnt.invoke(l4));
  }
  public static void CheckEQ(Object a, int b) {
    CheckEQ(((Integer)a).intValue(), b);
  }
  public static void CheckLE(Object a, Object b) {
    CheckLE(((Integer)a).intValue(), ((Integer)b).intValue());
  }
  public static void CheckEQ(int a, int b) {
    if (a != b) {
      throw new Error(a + " is not equal to " + b);
    }
  }
  public static void CheckLE(int a, int b) {
    if (!(a <= b)) {
      throw new Error(a + " is not less than or equal to " + b);
    }
  }
}