Swift Taint Tests

CMakeLists.txt

@@ -125,7 +125,9 @@ if (NOT PHASAR_IN_TREE)
   set(CMAKE_INSTALL_RPATH_USE_LINK_PATH TRUE)
 endif()
 
-if (LLVM_ENABLE_LIBCXX)
+if(CMAKE_SYSTEM_NAME STREQUAL "Darwin")
+  set(PHASAR_STD_FILESYSTEM stdc++)
+elseif (LLVM_ENABLE_LIBCXX)
   set(PHASAR_STD_FILESYSTEM c++fs)
 else()
   set(PHASAR_STD_FILESYSTEM stdc++fs)

README.md

@@ -94,7 +94,23 @@ Done!
 
 
 ### Installing PhASAR a MacOS system
-Due to unfortunate updates to MacOS and the handling of C++, especially on the newer M1 processors, we can't support native development on Mac.
+To install PhASAR on a Mac you need to follow these steps:
+1. Install both LLVM 14 and LLVM 17 on your system using e.g. homebrew.
+2. Setup LLVM 17 as your compiler to compile PhASAR with using the following environment variables
+
+```
+// this must point to LLVM 17 !
+CC=/opt/homebrew/opt/llvm/bin/clang
+CXX=/opt/homebrew/opt/llvm/bin/clang++
+```
+3. Update your `LDFLAGS`, `CPPFLAGS`, and `PATH` to point to your LLVM 14 installation
+```
+export LDFLAGS="-L/opt/homebrew/opt/llvm@14/lib"
+export CPPFLAGS="-I/opt/homebrew/opt/llvm@14/include"
+export PATH="/opt/homebrew/opt/llvm@14/bin:$PATH"
+```
+4. Run `cmake .. -DCMAKE_CXX_COMPILER=$CXX -DPHASAR_BUILD_UNITTESTS=0 -DPHASAR_BUILD_IR=0 -G Ninja` and `ninja` to build PhASAR (currently unit tests break the build, because cmake tries to use LLVM 17 to create the IR. We are working on a solution for this)
+### Use Docker for development
 The easiest solution to develop PhASAR on a Mac right now is to use [dockers development environments](https://docs.docker.com/desktop/dev-environments/). Clone this repository as described in their documentation. Afterwards, you have to login once manually, as a root user by running `docker exec -it -u root <container name> /bin/bash` to complete the rest of the install process as described in this readme (install submodules, run bootstrap.sh, ...).
 Now you can just attach your docker container to VS Code or any other IDE, which supports remote development.
 
@@ -152,6 +168,7 @@ When using CMake to compile PhASAR the following optional parameters can be used
 | <b>PHASAR_CONFIG_INSTALL_PREFIX</b> : PATH | Path where PhASAR's configuration files will be installed if <br> ninja install” is invoked or the “install” <br> target is built. Expression will be evaluated within CMAKE_INSTALL_PREFIX, so prefer absolute paths (default is $(HOME)/.config/) |
 | <b>PHASAR_BUILD_DOC</b> : BOOL | Build PhASAR documentation (default is OFF) |
 | <b>PHASAR_BUILD_UNITTESTS</b> : BOOL | Build PhASAR unit tests (default is ON) |
+| <b>BUILD_SWIFT_TESTS</b> : BOOL | Builds the Swift tests (Swift compiler has to be installed manually beforehand!) (default is OFF) |
 | <b>PHASAR_BUILD_IR</b> : BOOL | Build PhASAR IR (required for running the unit tests) (default is ON) |
 | <b>PHASAR_BUILD_OPENSSL_TS_UNITTESTS</b> : BOOL | Build PhASAR unit tests that require OpenSSL (default is OFF) |
 | <b>PHASAR_ENABLE_PAMM</b> : STRING | Enable the performance measurement mechanism <br> ('Off', 'Core' or 'Full', default is Off) |

include/phasar/Utils/MemoryResource.h

@@ -2,7 +2,7 @@
 /// a workaround here
 
 #ifndef HAS_MEMORY_RESOURCE
-#if !defined(__has_include) || __has_include(<memory_resource>)
+#if !defined(__has_include) || __has_include(<memory_resource>) && !defined(__APPLE__)
 #define HAS_MEMORY_RESOURCE 1
 #include <memory_resource>
 #else

lib/PhasarLLVM/DataFlow/IfdsIde/Problems/IFDSTaintAnalysis.cpp

@@ -124,23 +124,7 @@ IFDSTaintAnalysis::FlowFunctionPtrType IFDSTaintAnalysis::getNormalFlowFunction(
     IFDSTaintAnalysis::n_t Curr, [[maybe_unused]] IFDSTaintAnalysis::n_t Succ) {
   // If a tainted value is stored, the store location must be tainted too
   if (const auto *Store = llvm::dyn_cast<llvm::StoreInst>(Curr)) {
-    struct TAFF : FlowFunction<IFDSTaintAnalysis::d_t> {
-      const llvm::StoreInst *Store;
-      TAFF(const llvm::StoreInst *S) : Store(S){};
-      std::set<IFDSTaintAnalysis::d_t>
-      computeTargets(IFDSTaintAnalysis::d_t Source) override {
-        if (Store->getValueOperand() == Source) {
-          return std::set<IFDSTaintAnalysis::d_t>{Store->getPointerOperand(),
-                                                  Source};
-        }
-        if (Store->getValueOperand() != Source &&
-            Store->getPointerOperand() == Source) {
-          return {};
-        }
-        return {Source};
-      }
-    };
-    return std::make_shared<TAFF>(Store);
+    return strongUpdateStore(Store);
   }
   // If a tainted value is loaded, the loaded value is of course tainted
   if (const auto *Load = llvm::dyn_cast<llvm::LoadInst>(Curr)) {
@@ -154,10 +138,13 @@ IFDSTaintAnalysis::FlowFunctionPtrType IFDSTaintAnalysis::getNormalFlowFunction(
   // Check if a tainted value is extracted and taint the targets of
   // the extract operation accordingly
   if (const auto *Extract = llvm::dyn_cast<llvm::ExtractValueInst>(Curr)) {
-
     return generateFlow(Extract, Extract->getAggregateOperand());
   }
 
+  if (const auto *Insert = llvm::dyn_cast<llvm::InsertValueInst>(Curr)) {
+    return generateFlow(Insert, Insert->getInsertedValueOperand());
+  }
+
   // Otherwise we do not care and leave everything as it is
   return Identity<IFDSTaintAnalysis::d_t>::getInstance();
 }
@@ -290,7 +277,6 @@ IFDSTaintAnalysis::getSummaryFlowFunction(
   // result should be tainted
   if (DestFun->getName().equals("$sSS1poiyS2S_SStFZ")) {
     const auto *CS = llvm::cast<llvm::CallBase>(CallSite);
-
     return generateFlowIf<d_t>(CallSite, [CS](d_t Source) {
       return ((Source == CS->getArgOperand(1)) ||
               (Source == CS->getArgOperand(3)));

test/llvm_swift_test_code/CMakeLists.txt

@@ -1 +1,2 @@
 add_subdirectory(linear_constant)
+add_subdirectory(taint_analysis)

test/llvm_swift_test_code/linear_constant/CMakeLists.txt

@@ -1,9 +1,15 @@
 file(GLOB lca_files *.swift)
 
-set(SWIFT_COMPILE_IR_FLAGS -emit-ir -suppress-warnings -g -parse-as-library -Onone -Xfrontend -disable-llvm-optzns -Xfrontend -disable-swift-specific-llvm-optzns)
+set(SWIFT_COMPILE_IR_FLAGS -emit-ir -suppress-warnings -g -parse-as-library -Onone)
 
 foreach(TEST_SRC ${lca_files})
   get_filename_component(TEST_SRC_FILE ${TEST_SRC} NAME_WE)
-  add_executable(${TEST_SRC_FILE}.ll ${TEST_SRC})
-  target_compile_options(${TEST_SRC_FILE}.ll PRIVATE ${SWIFT_COMPILE_IR_FLAGS})
+  set(TEST_OUTPUT "${CMAKE_CURRENT_BINARY_DIR}/${TEST_SRC_FILE}.ll")
+  add_custom_command(
+    OUTPUT ${TEST_OUTPUT}
+    COMMAND swiftc ${SWIFT_COMPILE_IR_FLAGS} -o ${TEST_OUTPUT} ${TEST_SRC}
+    DEPENDS ${TEST_SRC}
+  )
+  add_custom_target(${TEST_SRC_FILE} DEPENDS ${TEST_OUTPUT})
+  add_dependencies(PhasarUnitTests ${TEST_SRC_FILE})
 endforeach(TEST_SRC)

test/llvm_swift_test_code/linear_constant/call_05.swift

@@ -1,6 +1,6 @@
 @main
 struct MyMain {
     static func main() {
-      var i = CommandLine.arguments.count
+        var i = CommandLine.arguments.count
     }
 }

test/llvm_swift_test_code/linear_constant/call_11.swift

@@ -1,10 +1,10 @@
 @main
 struct MyMain {
-    static func bar(_ b: Int)-> Int {
+    static func bar(_ b: Int) -> Int {
         return b
     }
 
-    static func foo(_ a: Int)-> Int {
+    static func foo(_ a: Int) -> Int {
         return bar(a)
     }
 

test/llvm_swift_test_code/linear_constant/for_01.swift

@@ -1,10 +1,9 @@
-
 @main
 struct MyMain {
     static func main() {
-      var a = 0
-      for i in 0...9 {
-        a += i
-      }
+        var a = 0
+        for i in 0...9 {
+            a += i
+        }
     }
 }

test/llvm_swift_test_code/linear_constant/global_00.swift

@@ -0,0 +1,7 @@
+var y = 42
+@main
+struct MyMain {
+    static func main() {
+        y += 1
+    }
+}

test/llvm_swift_test_code/linear_constant/global_01.swift

@@ -5,8 +5,8 @@ var g3 = 3.1415
 @main
 struct MyMain {
     static func main() {
-      var i = 666
-      g1 = 42
-      g2 = g1
+        var i = 666
+        g1 = 42
+        g2 = g1
     }
 }

test/llvm_swift_test_code/linear_constant/global_02.swift

@@ -3,8 +3,8 @@ var g = 10
 @main
 struct MyMain {
     static func main() {
-      var i = g
-      i -= 20
-      g = i
+        var i = g
+        i -= 20
+        g = i
     }
 }

test/llvm_swift_test_code/linear_constant/global_03.swift

@@ -3,12 +3,12 @@ var g = 0
 @main
 struct MyMain {
 
-  static func foo() {
-    g += 1
-  }
+    static func foo() {
+        g += 1
+    }
     static func main() {
-      var i = 42
-      g += 1
-      foo()
+        var i = 42
+        g += 1
+        foo()
     }
 }

test/llvm_swift_test_code/linear_constant/global_04.swift

@@ -3,12 +3,12 @@ var g = 0
 @main
 struct MyMain {
 
-  static func foo (_ a: Int)-> Int {
-    return a + 1
-  }
+    static func foo(_ a: Int) -> Int {
+        return a + 1
+    }
 
     static func main() {
-      g += 1
-      var i = foo(g)
+        g += 1
+        var i = foo(g)
     }
 }

test/llvm_swift_test_code/linear_constant/global_06.swift

@@ -3,12 +3,12 @@ var g = 0
 @main
 struct MyMain {
 
-  static func foo ()-> Int {
-    return g + 1
-  }
+    static func foo() -> Int {
+        return g + 1
+    }
 
     static func main() {
-      g += 1
-      var i = foo()
+        g += 1
+        var i = foo()
     }
 }

test/llvm_swift_test_code/linear_constant/global_07.swift

@@ -3,21 +3,21 @@ var g = 0
 @main
 struct MyMain {
 
-  static func foo (_ a: Int)-> Int {
-    var x = a
-    x += g
-    return x
-  }
+    static func foo(_ a: Int) -> Int {
+        var x = a
+        x += g
+        return x
+    }
 
-  static func bar (_ b: Int)-> Int {
-    g += 1
-    return b + 1
-  }
+    static func bar(_ b: Int) -> Int {
+        g += 1
+        return b + 1
+    }
 
     static func main() {
-      g += 1
-      var i = 0
-      i = foo(10)
-      i = bar(3)
+        g += 1
+        var i = 0
+        i = foo(10)
+        i = bar(3)
     }
 }

test/llvm_swift_test_code/linear_constant/global_08.swift

@@ -3,21 +3,21 @@ var g = 1
 @main
 struct MyMain {
 
-  static func baz (_ c: Int)-> Int {
-    return g + c
-  }
+    static func baz(_ c: Int) -> Int {
+        return g + c
+    }
 
-  static func bar (_ b: Int)-> Int {
-    return baz(b + 1)
-  }
+    static func bar(_ b: Int) -> Int {
+        return baz(b + 1)
+    }
 
-  static func foo (_ a: Int)-> Int {
-    return bar(a + 1)
-  }
+    static func foo(_ a: Int) -> Int {
+        return bar(a + 1)
+    }
 
     static func main() {
-      g += 1
-      var i = 0
-      i = foo(1)
+        g += 1
+        var i = 0
+        i = foo(1)
     }
 }

test/llvm_swift_test_code/linear_constant/global_09.swift

@@ -3,26 +3,26 @@ var g1 = 1
 @main
 struct MyMain {
 
-  static func baz (_ c: Int)-> Int {
-    return c + 3
-  }
+    static func baz(_ c: Int) -> Int {
+        return c + 3
+    }
 
-  static func bar (_ b: Int)-> Int {
-    g1 += 1
-    return b + 1
-  }
+    static func bar(_ b: Int) -> Int {
+        g1 += 1
+        return b + 1
+    }
 
-  static func foo (_ a: Int)-> Int {
-    var x = a
-    x+=g1
-    return x
-  }
+    static func foo(_ a: Int) -> Int {
+        var x = a
+        x += g1
+        return x
+    }
 
     static func main() {
-      g1 += 1
-      var i = 0
-      i = foo(10)
-      i = bar(3)
-      i = baz(39)
+        g1 += 1
+        var i = 0
+        i = foo(10)
+        i = bar(3)
+        i = baz(39)
     }
 }

test/llvm_swift_test_code/linear_constant/global_11.swift

@@ -4,15 +4,14 @@ var g2 = 9001
 @main
 struct MyMain {
 
-  static func foo (_ x: Int)-> Int {
-    var a = x
-    a = a + 1
-    return a
-  }
-
+    static func foo(_ x: Int) -> Int {
+        var a = x
+        a = a + 1
+        return a
+    }
 
     static func main() {
-      var a = 13
-      a = foo(a)
+        var a = 13
+        a = foo(a)
     }
 }