JSpecify mode: initial support for generic methods (with explicit type arguments at calls)

nullaway/src/main/java/com/uber/nullaway/NullAway.java

@@ -1853,6 +1853,10 @@ private Description handleInvocation(
       if (config.isJSpecifyMode()) {
         GenericsChecks.compareGenericTypeParameterNullabilityForCall(
             formalParams, actualParams, varArgsMethod, this, state);
+        if (!methodSymbol.getTypeParameters().isEmpty()) {
+          GenericsChecks.checkInstantiationForGenericMethodCalls(
+              tree, state, this, config, handler);
+        }
       }
     }
 

nullaway/src/main/java/com/uber/nullaway/generics/GenericsChecks.java

@@ -26,16 +26,19 @@
 import com.sun.tools.javac.code.Symbol;
 import com.sun.tools.javac.code.TargetType;
 import com.sun.tools.javac.code.Type;
+import com.sun.tools.javac.tree.JCTree;
 import com.uber.nullaway.CodeAnnotationInfo;
 import com.uber.nullaway.Config;
 import com.uber.nullaway.ErrorBuilder;
 import com.uber.nullaway.ErrorMessage;
 import com.uber.nullaway.NullAway;
 import com.uber.nullaway.Nullness;
 import com.uber.nullaway.handlers.Handler;
+import java.util.ArrayList;
 import java.util.HashMap;
 import java.util.List;
 import java.util.Map;
+import java.util.Objects;
 import javax.lang.model.type.ExecutableType;
 import javax.lang.model.type.TypeKind;
 import javax.lang.model.type.TypeVariable;
@@ -145,6 +148,81 @@ private static boolean[] getTypeParamsWithNullableUpperBound(
     return result;
   }
 
+  /**
+   * Checks instantiated generic arguments of generic method calls. {@code @Nullable} types can only
+   * be used for type variables that have a {@code @Nullable} upper bound.
+   *
+   * @param tree the tree representing the instantiated type
+   * @param state visitor state
+   * @param analysis the analysis object
+   * @param config the analysis config
+   * @param handler the handler instance
+   */
+  public static void checkInstantiationForGenericMethodCalls(
+      Tree tree, VisitorState state, NullAway analysis, Config config, Handler handler) {
+    List<? extends Tree> typeArguments = ((MethodInvocationTree) tree).getTypeArguments();
+    if (typeArguments.isEmpty()) {
+      return;
+    }
+    // get Nullable annotated type arguments
+    MethodInvocationTree methodTree = (MethodInvocationTree) tree;
+    Map<Integer, Tree> nullableTypeArguments = new HashMap<>();
+    for (int i = 0; i < typeArguments.size(); i++) {
+      Tree curTypeArg = typeArguments.get(i);
+      if (curTypeArg instanceof AnnotatedTypeTree) {
+        AnnotatedTypeTree annotatedType = (AnnotatedTypeTree) curTypeArg;
+        for (AnnotationTree annotation : annotatedType.getAnnotations()) {
+          Type annotationType = ASTHelpers.getType(annotation);
+          if (annotationType != null
+              && Nullness.isNullableAnnotation(annotationType.toString(), config)) {
+            nullableTypeArguments.put(i, curTypeArg);
+            break;
+          }
+        }
+      }
+    }
+    Symbol.MethodSymbol methodSymbol = ASTHelpers.getSymbol(methodTree);
+
+    // check if type variables are allowed to be Nullable
+    Type baseType = methodSymbol.asType();
+    List<Type> baseTypeVariables = baseType.getTypeArguments();
+    for (int i = 0; i < baseTypeVariables.size(); i++) {
+      if (nullableTypeArguments.containsKey(i)) {
+        Type typeVariable = baseTypeVariables.get(i);
+        Type upperBound = typeVariable.getUpperBound();
+        com.sun.tools.javac.util.List<Attribute.TypeCompound> annotationMirrors =
+            upperBound.getAnnotationMirrors();
+        boolean hasNullableAnnotation =
+            Nullness.hasNullableAnnotation(annotationMirrors.stream(), config)
+                || handler.onOverrideTypeParameterUpperBound(baseType.tsym.toString(), i);
+        // if type variable's upper bound does not have @Nullable annotation then the instantiation
+        // is invalid
+        if (!hasNullableAnnotation) {
+          reportInvalidTypeArgumentError(
+              nullableTypeArguments.get(i), methodSymbol, typeVariable, state, analysis);
+        }
+      }
+    }
+  }
+
+  private static void reportInvalidTypeArgumentError(
+      Tree tree,
+      Symbol.MethodSymbol methodSymbol,
+      Type typeVariable,
+      VisitorState state,
+      NullAway analysis) {
+    ErrorBuilder errorBuilder = analysis.getErrorBuilder();
+    ErrorMessage errorMessage =
+        new ErrorMessage(
+            ErrorMessage.MessageTypes.TYPE_PARAMETER_CANNOT_BE_NULLABLE,
+            String.format(
+                "Type argument cannot be @Nullable, as method %s's type variable %s is not @Nullable",
+                methodSymbol.toString(), typeVariable.tsym.toString()));
+    state.reportMatch(
+        errorBuilder.createErrorDescription(
+            errorMessage, analysis.buildDescription(tree), state, null));
+  }
+
   private static void reportInvalidInstantiationError(
       Tree tree, Type baseType, Type baseTypeVariable, VisitorState state, NullAway analysis) {
     ErrorBuilder errorBuilder = analysis.getErrorBuilder();
@@ -682,7 +760,8 @@ public static Nullness getGenericMethodReturnTypeNullness(
   /**
    * Computes the nullness of the return of a generic method at an invocation, in the context of the
    * declared type of its receiver argument. If the return type is a type variable, its nullness
-   * depends on the nullability of the corresponding type parameter in the receiver's type.
+   * depends on the nullability of the corresponding type parameter in the receiver's type or the
+   * type argument of the method call.
    *
    * <p>Consider the following example:
    *
@@ -715,6 +794,33 @@ public static Nullness getGenericReturnNullnessAtInvocation(
       MethodInvocationTree tree,
       VisitorState state,
       Config config) {
+    // If generic method invocation
+    if (!invokedMethodSymbol.getTypeParameters().isEmpty()) {
+      List<? extends Tree> typeArgumentTrees = tree.getTypeArguments();
+      com.sun.tools.javac.util.List<Type> explicitTypeArgs =
+          convertTreesToTypes(typeArgumentTrees); // Convert to Type objects
+
+      Type methodType = invokedMethodSymbol.type;
+      Type substitutedReturnType = null;
+      if (methodType instanceof Type.ForAll) {
+        Type.ForAll forAllType = (Type.ForAll) methodType;
+
+        // Extract the underlying MethodType (the actual signature)
+        Type.MethodType methodTypeInsideForAll = (Type.MethodType) forAllType.qtype;
+
+        // Substitute the argument and return types within the MethodType
+        substitutedReturnType =
+            state
+                .getTypes()
+                .subst(methodTypeInsideForAll.restype, forAllType.tvars, explicitTypeArgs);
+      }
+
+      if (substitutedReturnType != null
+          && Objects.equals(getTypeNullness(substitutedReturnType, config), Nullness.NULLABLE)) {
+        return Nullness.NULLABLE;
+      }
+    }
+
     if (!(tree.getMethodSelect() instanceof MemberSelectTree) || invokedMethodSymbol.isStatic()) {
       return Nullness.NONNULL;
     }
@@ -728,6 +834,18 @@ public static Nullness getGenericReturnNullnessAtInvocation(
     }
   }
 
+  private static com.sun.tools.javac.util.List<Type> convertTreesToTypes(
+      List<? extends Tree> typeArgumentTrees) {
+    List<Type> types = new ArrayList<>();
+    for (Tree tree : typeArgumentTrees) {
+      if (tree instanceof JCTree.JCExpression) {
+        JCTree.JCExpression expression = (JCTree.JCExpression) tree;
+        types.add(expression.type); // Retrieve the Type
+      }
+    }
+    return com.sun.tools.javac.util.List.from(types);
+  }
+
   /**
    * Computes the nullness of a formal parameter of a generic method at an invocation, in the
    * context of the declared type of its receiver argument. If the formal parameter's type is a type
@@ -768,6 +886,38 @@ public static Nullness getGenericParameterNullnessAtInvocation(
       MethodInvocationTree tree,
       VisitorState state,
       Config config) {
+    // If generic method invocation
+    if (!invokedMethodSymbol.getTypeParameters().isEmpty()) {
+      List<? extends Tree> typeArgumentTrees = tree.getTypeArguments();
+      com.sun.tools.javac.util.List<Type> explicitTypeArgs =
+          convertTreesToTypes(typeArgumentTrees); // Convert to Type objects
+
+      Type methodType = invokedMethodSymbol.type;
+      List<Type> substitutedParamTypes = null;
+      if (methodType instanceof Type.ForAll) {
+        Type.ForAll forAllType = (Type.ForAll) methodType;
+
+        // Extract the underlying MethodType (the actual signature)
+        Type.MethodType methodTypeInsideForAll = (Type.MethodType) forAllType.qtype;
+
+        // Substitute the argument and return types within the MethodType
+        substitutedParamTypes =
+            state
+                .getTypes()
+                .subst(
+                    methodTypeInsideForAll.argtypes,
+                    forAllType.tvars, // The type variables from the ForAll
+                    explicitTypeArgs // The actual type arguments from the method invocation
+                    );
+      }
+
+      if (substitutedParamTypes != null
+          && Objects.equals(
+              getTypeNullness(substitutedParamTypes.get(paramIndex), config), Nullness.NULLABLE)) {
+        return Nullness.NULLABLE;
+      }
+    }
+
     if (!(tree.getMethodSelect() instanceof MemberSelectTree) || invokedMethodSymbol.isStatic()) {
       return Nullness.NONNULL;
     }

nullaway/src/test/java/com/uber/nullaway/jspecify/GenericMethodTests.java

@@ -8,6 +8,74 @@
 
 public class GenericMethodTests extends NullAwayTestsBase {
 
+  @Test
+  public void genericNonNullIdentityFunction() {
+    makeHelper()
+        .addSourceLines(
+            "Test.java",
+            "package com.uber;",
+            "import org.jspecify.annotations.Nullable;",
+            "class Test {",
+            "  static <T> T nonNullIdentity(T t) {",
+            "    return t;",
+            "  }",
+            "  static void test() {",
+            "    // legal",
+            "    nonNullIdentity(new Object()).toString();",
+            "    // BUG: Diagnostic contains: passing @Nullable parameter 'null'",
+            "    nonNullIdentity(null);",
+            "    // BUG: Diagnostic contains: Type argument cannot be @Nullable",
+            "    Test.<@Nullable Object>nonNullIdentity(new Object());",
+            "  }",
+            "}")
+        .doTest();
+  }
+
+  @Test
+  public void genericNullAllowingIdentityFunction() {
+    makeHelper()
+        .addSourceLines(
+            "Test.java",
+            "package com.uber;",
+            "import org.jspecify.annotations.Nullable;",
+            "class Test {",
+            "  static <T extends @Nullable Object> T identity(T t) {",
+            "    return t;",
+            "  }",
+            "  static void test() {",
+            "    // legal",
+            "    identity(new Object()).toString();",
+            "    // also legal",
+            "    Test.<@Nullable Object>identity(null);",
+            "    // BUG: Diagnostic contains: dereferenced expression",
+            "    Test.<@Nullable Object>identity(null).toString();",
+            "  }",
+            "}")
+        .doTest();
+  }
+
+  @Test
+  public void multipleTypeVariablesMethodCall() {
+    makeHelper()
+        .addSourceLines(
+            "Test.java",
+            "package com.uber;",
+            "import org.jspecify.annotations.Nullable;",
+            "import org.jspecify.annotations.NonNull;",
+            "class Test {",
+            "    public static <T extends @Nullable Object, U> void twoTypeVars(T first, U second) {}",
+            "    static void test() {",
+            "       // legal",
+            "       Test.<@NonNull Object, Object>twoTypeVars(new Object(), new Object());",
+            "       // legal",
+            "       Test.<@Nullable Object, Object>twoTypeVars(null, new Object());",
+            "       // BUG: Diagnostic contains: Type argument cannot be @Nullable",
+            "       Test.<@Nullable Object, @Nullable Object>twoTypeVars(null, null);",
+            "    }",
+            "}")
+        .doTest();
+  }
+
   @Test
   @Ignore("requires generic method support")
   public void genericMethodAndVoidType() {