Swift: Add taint tests for 'Set'.

Author: geoffw0

swift/ql/test/library-tests/dataflow/taint/libraries/set.swift

@@ -0,0 +1,107 @@
+
+// --- stubs ---
+
+// --- tests ---
+
+func source(_ label: String) -> Int { return 0 }
+func sink(arg: Any) {}
+
+// ---
+
+func testSet(ix: Int) {
+  let goodSet = Set([1, 2, 3])
+  sink(arg: goodSet)
+  sink(arg: goodSet.randomElement()!)
+  sink(arg: goodSet.min()!)
+  sink(arg: goodSet.max()!)
+  sink(arg: goodSet.firstIndex(of: 1)!)
+  sink(arg: goodSet[goodSet.firstIndex(of: 1)!])
+  sink(arg: goodSet.first!)
+  for elem in goodSet {
+    sink(arg: elem)
+  }
+
+  let taintedSet = Set([1, 2, source("t1")])
+  sink(arg: taintedSet) // $ tainted=t1
+  sink(arg: taintedSet.randomElement()!) // $ tainted=t1
+  sink(arg: taintedSet.min()!) // $ MISSING: tainted=t1
+  sink(arg: taintedSet.max()!) // $ MISSING: tainted=t1
+  sink(arg: taintedSet.firstIndex(of: source("t2"))!)
+  sink(arg: taintedSet[taintedSet.firstIndex(of: source("t3"))!]) // $ tainted=t1
+  sink(arg: taintedSet.first!) // $ tainted=t1
+  for elem in taintedSet {
+    sink(arg: elem) // $ tainted=t1
+  }
+  for (ix, elem) in taintedSet.enumerated() {
+    sink(arg: ix)
+    sink(arg: elem) // $ MISSING: tainte=t1
+  }
+  taintedSet.forEach {
+    elem in
+    sink(arg: elem) // $ MISSING: tainted=
+  }
+
+  var set1 = Set<Int>()
+  set1.insert(1)
+  sink(arg: set1.randomElement()!)
+  set1.insert(source("t4"))
+  sink(arg: set1.randomElement()!) // $ tainted=t4
+  set1.insert(2)
+  sink(arg: set1.randomElement()!) // $ tainted=t4
+  set1.removeAll()
+  sink(arg: set1.randomElement()!) // $ SPURIOUS: tainted=t4
+
+  var set2 = Set<Int>()
+  set2.update(with: source("t5"))
+  sink(arg: set2.randomElement()!) // $ MISSING: tainted=t5
+
+  var set3 = Set([source("t6")])
+  sink(arg: set3.randomElement()!) // $ tainted=t6
+  let (inserted, previous) = set3.insert(source("t7"))
+  sink(arg: inserted)
+  sink(arg: previous) // $ tainted=t6 tainted=t7
+  let previous2 = set3.update(with: source("t8"))
+  sink(arg: previous2!) // $ MISSING: tainted=t8
+  let previous3 = set3.remove(source("t9"))
+  sink(arg: previous3!) // $ MISSING: tainted=t9
+  let previous4 = set3.removeFirst()
+  sink(arg: previous4) // $ MISSING: tainted=t6
+  let previous5 = set3.remove(at: set3.firstIndex(of: source("t10"))!)
+  sink(arg: previous5) // $ MISSING: tainted=t6
+
+  sink(arg: goodSet.union(goodSet).randomElement()!)
+  sink(arg: goodSet.union(taintedSet).randomElement()!) // $ MISSING: tainted=t1
+  sink(arg: taintedSet.union(goodSet).randomElement()!) // $ MISSING: tainted=t1
+  sink(arg: taintedSet.union(taintedSet).randomElement()!) // $ MISSING: tainted=t1
+
+  var set4 = Set<Int>()
+  set4.formUnion(goodSet)
+  sink(arg: set4.randomElement()!)
+  set4.formUnion(taintedSet)
+  sink(arg: set4.randomElement()!) // $ MISSING: tainted=t1
+  set4.formUnion(goodSet)
+  sink(arg: set4.randomElement()!) // $ MISSING: tainted=t1
+
+  sink(arg: goodSet.intersection(goodSet).randomElement()!)
+  sink(arg: goodSet.intersection(taintedSet).randomElement()!)
+  sink(arg: taintedSet.intersection(goodSet).randomElement()!)
+  sink(arg: taintedSet.intersection(taintedSet).randomElement()!) // $ MISSING: tainted=t1
+
+  sink(arg: goodSet.symmetricDifference(goodSet).randomElement()!)
+  sink(arg: goodSet.symmetricDifference(taintedSet).randomElement()!) // $ MISSING: tainted=t1
+  sink(arg: taintedSet.symmetricDifference(goodSet).randomElement()!) // $ MISSING: tainted=t1
+  sink(arg: taintedSet.symmetricDifference(taintedSet).randomElement()!) // $ MISSING: tainted=t1
+
+  sink(arg: goodSet.subtracting(goodSet).randomElement()!)
+  sink(arg: goodSet.subtracting(taintedSet).randomElement()!)
+  sink(arg: taintedSet.subtracting(goodSet).randomElement()!) // $ MISSING: tainted=t1
+  sink(arg: taintedSet.subtracting(taintedSet).randomElement()!) // $ MISSING: tainted=t1
+
+  sink(arg: taintedSet.sorted().randomElement()!) // $ MISSING: tainted=t1
+  sink(arg: taintedSet.shuffled().randomElement()!) // $ MISSING: tainted=t1
+
+  sink(arg: taintedSet.lazy[taintedSet.firstIndex(of: source("t11"))!]) // $ tainted=t1
+
+  var it = taintedSet.makeIterator()
+  sink(arg: it.next()!) // $ tainted=t1
+}