Swift: Add data flow tests for RawRepresentable, OptionSet.

Author: geoffw0

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

@@ -0,0 +1,66 @@
+
+// --- stubs ---
+
+// --- tests ---
+
+func sourceInt() -> Int { return 0 }
+func sourceUInt() -> UInt { return 0 }
+func sink(arg: Any) {}
+
+// ---
+
+enum MyRawRepresentable : RawRepresentable {
+  case valueOne
+  case valueTwo
+
+	init?(rawValue: Int) {
+    switch rawValue {
+      case 1: self = .valueOne
+      case 2: self = .valueTwo
+      default: return nil
+    }
+  }
+
+  var rawValue: Int {
+    switch self {
+      case .valueOne: return 1
+      case .valueTwo: return 2
+    }
+  }
+}
+
+func testRawRepresentable() {
+  let rr1 = MyRawRepresentable.valueOne
+  let rr2 = MyRawRepresentable(rawValue: 1)!
+  let rr3 = MyRawRepresentable(rawValue: sourceInt())!
+
+  sink(arg: rr1)
+  sink(arg: rr2)
+  sink(arg: rr3) // $ MISSING: tainted=
+
+  sink(arg: rr1.rawValue)
+  sink(arg: rr2.rawValue)
+  sink(arg: rr3.rawValue) // $ MISSING: tainted=
+}
+
+// ---
+
+struct MyOptionSet : OptionSet {
+	let rawValue: UInt
+
+	static let red = MyOptionSet(rawValue: 1 << 0)
+	static let green = MyOptionSet(rawValue: 1 << 1)
+	static let blue = MyOptionSet(rawValue: 1 << 2)
+}
+
+func testOptionSet() {
+	sink(arg: MyOptionSet.red)
+	sink(arg: MyOptionSet([.red, .green]))
+	sink(arg: MyOptionSet(rawValue: 0))
+	sink(arg: MyOptionSet(rawValue: sourceUInt())) // $ MISSING: tainted=
+
+	sink(arg: MyOptionSet.red.rawValue)
+	sink(arg: MyOptionSet([.red, .green]).rawValue)
+	sink(arg: MyOptionSet(rawValue: 0).rawValue)
+	sink(arg: MyOptionSet(rawValue: sourceUInt()).rawValue) // $ MISSING: tainted=
+}