SystemSan: repair pytorch-lightning-1.5.10

Author: swapgs

infra/experimental/SystemSan/Makefile

@@ -30,10 +30,10 @@ test:  all vuln.dict
 	$(call test_fuzzer,target_file,Arbitrary file open)
 
 pytorch-lightning-1.5.10:
-	cp SystemSan.cpp PoEs/pytorch-lightning-1.5.10/; \
+	cp SystemSan.cpp inspect_*.* PoEs/pytorch-lightning-1.5.10/; \
 	cd PoEs/pytorch-lightning-1.5.10/; \
-	docker build . --tag syssan_pytorch-lightning; \
-	docker run -t systemsan_pytorch-lightning:latest;
+	docker build . --tag systemsan_pytorch-lightning; \
+	docker run -it systemsan_pytorch-lightning:latest;
 
 node-shell-quote-v1.7.3:
 	cp SystemSan.cpp PoEs/node-shell-quote-v1.7.3/; \

infra/experimental/SystemSan/PoEs/pytorch-lightning-1.5.10/Dockerfile

@@ -21,7 +21,7 @@ RUN apt update && \
   git clone \
     --depth 1 \
     --branch 1.5.10 \
-    https://github.com/PyTorchLightning/pytorch-lightning.git
+    https://github.com/Lightning-AI/lightning.git
 
 COPY ./build.sh $SRC
 RUN  ./build.sh

infra/experimental/SystemSan/PoEs/pytorch-lightning-1.5.10/Makefile

@@ -2,7 +2,7 @@
 CXX     = clang++
 CFLAGS = -std=c++17 -Wall -Wextra -O3 -g3
 
-SystemSan: SystemSan.cpp
+SystemSan: SystemSan.cpp inspect_dns.cpp inspect_utils.cpp
 	$(CXX) $(CFLAGS) -lpthread -o $@ $^
 
 run: clean SystemSan fuzz_pytorch_lightning.py

infra/experimental/SystemSan/PoEs/pytorch-lightning-1.5.10/build.sh

@@ -16,7 +16,45 @@
 ################################################################################
 
 # Build and install project (using current CFLAGS, CXXFLAGS).
-cd pytorch-lightning
+cd lightning
+git apply <<EOF
+diff --git a/requirements.txt b/requirements.txt
+index dd34e9273..36f55e683 100644
+--- a/requirements.txt
++++ b/requirements.txt
+@@ -1,7 +1,7 @@
+ # the default package dependencies
+
+ numpy>=1.17.2
+-torch>=1.7.*
++torch>=1.7.1
+ future>=0.17.1  # required for builtins in setup.py
+ tqdm>=4.41.0
+ PyYAML>=5.1
+diff --git a/requirements/examples.txt b/requirements/examples.txt
+index 8591f9bd5..8c0d96bd1 100644
+--- a/requirements/examples.txt
++++ b/requirements/examples.txt
+@@ -1,3 +1,3 @@
+-torchvision>=0.8.*
++torchvision>=0.8.2
+ gym>=0.17.0
+ ipython[all]
+diff --git a/requirements/extra.txt b/requirements/extra.txt
+index 74743185e..52580b475 100644
+--- a/requirements/extra.txt
++++ b/requirements/extra.txt
+@@ -2,7 +2,7 @@
+
+ matplotlib>3.1
+ horovod>=0.21.2  # no need to install with [pytorch] as pytorch is already installed
+-torchtext>=0.8.*
++torchtext>=0.8.1
+ omegaconf>=2.0.5
+ hydra-core>=1.0.5
+ jsonargparse[signatures]>=4.0.4,<5.0.0
+EOF
+pip3 install --upgrade pip
 pip3 install .
 
 # Build fuzzers in $OUT.