Home | History | Annotate | Download | only in generator 
      1 /*
      2  * Copyright (C) 2017 The Android Open Source Project
      3  *
      4  * Licensed under the Apache License, Version 2.0 (the "License");
      5  * you may not use this file except in compliance with the License.
      6  * You may obtain a copy of the License at
      7  *
      8  *      http://www.apache.org/licenses/LICENSE-2.0
      9  *
     10  * Unless required by applicable law or agreed to in writing, software
     11  * distributed under the License is distributed on an "AS IS" BASIS,
     12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
     13  * See the License for the specific language governing permissions and
     14  * limitations under the License.
     15  */
     16 
     17 import com.android.jack.annotations.CalledByInvokeCustom;
     18 import com.android.jack.annotations.Constant;
     19 import com.android.jack.annotations.LinkerMethodHandle;
     20 import com.android.jack.annotations.MethodHandleKind;
     21 
     22 import java.lang.invoke.CallSite;
     23 import java.lang.invoke.ConstantCallSite;
     24 import java.lang.invoke.MethodHandle;
     25 import java.lang.invoke.MethodHandles;
     26 import java.lang.invoke.MethodType;
     27 
     28 public class TestLinkerMethodMinimalArguments {
     29 
     30   private static int forceFailureType = 0;
     31 
     32   private static int FAILURE_TYPE_NONE = 0;
     33   private static int FAILURE_TYPE_LINKER_METHOD_RETURNS_NULL = 1;
     34   private static int FAILURE_TYPE_LINKER_METHOD_THROWS = 2;
     35   private static int FAILURE_TYPE_TARGET_METHOD_THROWS = 3;
     36 
     37   @CalledByInvokeCustom(
     38       invokeMethodHandle = @LinkerMethodHandle(
     39           kind = MethodHandleKind.INVOKE_STATIC,
     40           enclosingType = TestLinkerMethodMinimalArguments.class,
     41           argumentTypes = {MethodHandles.Lookup.class, String.class, MethodType.class},
     42           name = "linkerMethod"),
     43       name = "add",
     44       returnType = int.class,
     45       argumentTypes = {int.class, int.class})
     46   private static int add(int a, int b) {
     47     if (forceFailureType == FAILURE_TYPE_TARGET_METHOD_THROWS) {
     48       System.out.println("Throwing ArithmeticException in add()");
     49       throw new ArithmeticException("add");
     50     }
     51     return a + b;
     52   }
     53 
     54   @SuppressWarnings("unused")
     55   private static CallSite linkerMethod(MethodHandles.Lookup caller, String name,
     56                                        MethodType methodType) throws Throwable {
     57     System.out.println("linkerMethod failure type " + forceFailureType);
     58     MethodHandle mh_add =
     59         caller.findStatic(TestLinkerMethodMinimalArguments.class, name, methodType);
     60     if (forceFailureType == FAILURE_TYPE_LINKER_METHOD_RETURNS_NULL) {
     61       System.out.println("Returning null instead of CallSite for " + name + " " + methodType);
     62       return null;
     63     } else if (forceFailureType == FAILURE_TYPE_LINKER_METHOD_THROWS) {
     64       System.out.println("Throwing InstantiationException in linkerMethod()");
     65       throw new InstantiationException("linkerMethod");
     66     } else {
     67       return new ConstantCallSite(mh_add);
     68     }
     69   }
     70 
     71   public static void test(int failureType, int x, int y) throws Throwable {
     72     assertTrue(failureType >= FAILURE_TYPE_NONE);
     73     assertTrue(failureType <= FAILURE_TYPE_TARGET_METHOD_THROWS);
     74     forceFailureType = failureType;
     75     assertEquals(x + y, add(x, y));
     76     System.out.println("Failure Type + " + failureType + " (" + x + y+ ")");
     77   }
     78 
     79   public static void assertTrue(boolean value) {
     80     if (!value) {
     81       throw new AssertionError("assertTrue value: " + value);
     82     }
     83   }
     84 
     85   public static void assertEquals(byte b1, byte b2) {
     86     if (b1 == b2) { return; }
     87     throw new AssertionError("assertEquals b1: " + b1 + ", b2: " + b2);
     88   }
     89 
     90   public static void assertEquals(char c1, char c2) {
     91     if (c1 == c2) { return; }
     92     throw new AssertionError("assertEquals c1: " + c1 + ", c2: " + c2);
     93   }
     94 
     95   public static void assertEquals(short s1, short s2) {
     96     if (s1 == s2) { return; }
     97     throw new AssertionError("assertEquals s1: " + s1 + ", s2: " + s2);
     98   }
     99 
    100   public static void assertEquals(int i1, int i2) {
    101     if (i1 == i2) { return; }
    102     throw new AssertionError("assertEquals i1: " + i1 + ", i2: " + i2);
    103   }
    104 
    105   public static void assertEquals(long l1, long l2) {
    106     if (l1 == l2) { return; }
    107     throw new AssertionError("assertEquals l1: " + l1 + ", l2: " + l2);
    108   }
    109 
    110   public static void assertEquals(float f1, float f2) {
    111     if (f1 == f2) { return; }
    112     throw new AssertionError("assertEquals f1: " + f1 + ", f2: " + f2);
    113   }
    114 
    115   public static void assertEquals(double d1, double d2) {
    116     if (d1 == d2) { return; }
    117     throw new AssertionError("assertEquals d1: " + d1 + ", d2: " + d2);
    118   }
    119 
    120   public static void assertEquals(Object o, Object p) {
    121     if (o == p) { return; }
    122     if (o != null && p != null && o.equals(p)) { return; }
    123     throw new AssertionError("assertEquals: o1: " + o + ", o2: " + p);
    124   }
    125 
    126   public static void assertEquals(String s1, String s2) {
    127     if (s1 == s2) {
    128       return;
    129     }
    130 
    131     if (s1 != null && s2 != null && s1.equals(s2)) {
    132       return;
    133     }
    134 
    135     throw new AssertionError("assertEquals s1: " + s1 + ", s2: " + s2);
    136   }
    137 }
    138