Swift: Extract whether a function type is throwing or async.

swift/codegen/schema.yml

@@ -63,6 +63,7 @@ AnyFunctionType:
   param_types: Type*
   param_labels: string*
   is_throwing: predicate
+  is_async: predicate
 
 AnyGenericType:
   _extends: Type

swift/extractor/visitors/TypeVisitor.h

@@ -211,6 +211,14 @@ class TypeVisitor : public TypeVisitorBase<TypeVisitor> {
       }
       ++i;
     }
+
+    if (type->isThrowing()) {
+      dispatcher_.emit(AnyFunctionTypeIsThrowingTrap{label});
+    }
+
+    if (type->isAsync()) {
+      dispatcher_.emit(AnyFunctionTypeIsAsyncTrap{label});
+    }
   }
 
   void emitBoundGenericType(swift::BoundGenericType* type, TrapLabel<BoundGenericTypeTag> label) {

swift/ql/lib/codeql/swift/generated/type/AnyFunctionType.qll

@@ -27,4 +27,6 @@ class AnyFunctionTypeBase extends @any_function_type, Type {
   int getNumberOfParamLabels() { result = count(getAParamLabel()) }
 
   predicate isThrowing() { any_function_type_is_throwing(this) }
+
+  predicate isAsync() { any_function_type_is_async(this) }
 }

swift/ql/lib/swift.dbscheme

@@ -176,6 +176,11 @@ any_function_type_is_throwing(
   int id: @any_function_type ref
 );
 
+#keyset[id]
+any_function_type_is_async(
+  int id: @any_function_type ref
+);
+
 @any_generic_type =
   @nominal_or_bound_generic_nominal_type
 | @unbound_generic_type

swift/ql/test/extractor-tests/types/Types.expected

@@ -38,3 +38,4 @@
 | types.swift:21:10:21:10 | f | (Int) -> Int |
 | types.swift:21:10:21:13 | call to ... | Int |
 | types.swift:21:12:21:12 | x | Int |
+| types.swift:25:10:25:10 | 42 | Int |

swift/ql/test/extractor-tests/types/types.swift

@@ -20,3 +20,11 @@ func f(x: Int, y: Int) -> Int {
 func g(_ f: (Int) -> Int, _ x: Int) -> Int {
   return f(x)
 }
+
+func throwingFunc() throws -> Int {
+  return 42
+}
+
+func asyncFunction(x: Int) async { }
+
+func throwingAndAsyncFunction(x: Int) async throws { }