support basic generators

javascript/ql/src/semmle/javascript/Generators.qll

@@ -0,0 +1,21 @@
+import javascript
+private import semmle.javascript.dataflow.internal.PreCallGraphStep
+
+/**
+ * Classes and predicates for modelling data-flow for generator functions.
+ */
+private module GeneratorDataFlow {
+  private import DataFlow::PseudoProperties
+
+  private class ArrayIteration extends PreCallGraphStep {
+    override predicate storeStep(DataFlow::Node pred, DataFlow::SourceNode succ, string prop) {
+      exists(DataFlow::FunctionNode f | f.getFunction().isGenerator() |
+        prop = iteratorElement() and
+        exists(YieldExpr yield | yield.getContainer() = f.getFunction() |
+          pred.asExpr() = yield.getOperand()
+        ) and
+        succ = f.getReturnNode()
+      )
+    }
+  }
+}