Add initial QL support classes for coroutines

Matthew Gretton-Dann · igfoo · commit 5df5e6dfcecd · 2020-09-07T20:37:01.000+01:00
Add classes for expressions co_yield and co_await.
Adds classes for statements co_return and `for co_await`.
diff --git a/cpp/ql/src/semmle/code/cpp/exprs/Expr.qll b/cpp/ql/src/semmle/code/cpp/exprs/Expr.qll
@@ -1268,3 +1268,32 @@ class SpaceshipExpr extends BinaryOperation, @spaceshipexpr {
 
   override string getOperator() { result = "<=>" }
 }
+
+/**
+ * A C/C++ co_await expression
+ * ```
+ * co_await foo();
+ * ```
+ */
+class CoAwaitExpr extends UnaryOperation, @co_await {
+  override string getAPrimaryQlClass() { result = "CoAwaitExpr" }
+
+  override string getOperator() { result = "co_await" }
+
+  override int getPrecedence() { result = 16 }
+}
+
+
+/**
+ * A C/C++ co_yield expression
+ * ```
+ * co_yield 1;
+ * ```
+ */
+class CoYieldExpr extends UnaryOperation, @co_yield {
+  override string getAPrimaryQlClass() { result = "CoYieldExpr" }
+
+  override string getOperator() { result = "co_yield" }
+
+  override int getPrecedence() { result = 2 }
+}
diff --git a/cpp/ql/src/semmlecode.cpp.dbscheme b/cpp/ql/src/semmlecode.cpp.dbscheme
@@ -1857,6 +1857,8 @@ case @stmt.kind of
 |  38 = @stmt_range_based_for_co_await
 ;
 
+@range_based_for_stmt = @stmt_range_based_for | @stmt_range_based_for_co_await;
+
 type_vla(
     int type_id: @type ref,
     int decl: @stmt_vla_decl ref

Original file line number	Diff line number	Diff line change
`@@ -1857,6 +1857,8 @@ case @stmt.kind of`
`1857`	`1857`	`\| 38 = @stmt_range_based_for_co_await`
`1858`	`1858`	`;`
`1859`	`1859`
	`1860`	`+@range_based_for_stmt = @stmt_range_based_for \| @stmt_range_based_for_co_await;`
	`1861`	`+`
`1860`	`1862`	`type_vla(`
`1861`	`1863`	`int type_id: @type ref,`
`1862`	`1864`	`int decl: @stmt_vla_decl ref`