From 690910665029e3e14544d59904dc999abb1222d9 Mon Sep 17 00:00:00 2001
From: xsalon00 <salonmarek@gmail.com>
Date: Nov 13 2021 23:37:39 +0000
Subject: Improved test cleaning + custom test example


---
diff --git a/tests/fpd_tests/common/iframe.html b/tests/fpd_tests/common/iframe.html
index a5a96c0..6f76a26 100644
--- a/tests/fpd_tests/common/iframe.html
+++ b/tests/fpd_tests/common/iframe.html
@@ -9,7 +9,6 @@ SPDX-License-Identifier: GPL-3.0-or-later
 		<title>Fingerprinting Detection</title>
 		<script language="javascript" src="helpers.js"></script>
 		<!--SCRIPTS_S-->
-
 		<!--SCRIPTS_E-->
 	</head>
 	<body>
@@ -17,7 +16,6 @@ SPDX-License-Identifier: GPL-3.0-or-later
 	<script>
 		var resultsAcc = {};	
 		//CALL_S
-
 		//CALL_E
 		window.top.postMessage({res: resultsAcc}, '*');
 	</script>
diff --git a/tests/fpd_tests/common/worker.js b/tests/fpd_tests/common/worker.js
index 0c8a3fd..c71de4f 100644
--- a/tests/fpd_tests/common/worker.js
+++ b/tests/fpd_tests/common/worker.js
@@ -22,11 +22,9 @@
 
 importScripts('helpers.js');
 //SCRIPTS_S
-
 //SCRIPTS_E
 
 var resultsAcc = {};
 //CALL_S
-
 //CALL_E
 postMessage(resultsAcc);
\ No newline at end of file
diff --git a/tests/fpd_tests/index.html b/tests/fpd_tests/index.html
index 4582423..b5f5bfd 100644
--- a/tests/fpd_tests/index.html
+++ b/tests/fpd_tests/index.html
@@ -9,7 +9,6 @@ SPDX-License-Identifier: GPL-3.0-or-later
 		<title>Fingerprinting Detection Tests</title>
 		<script language="javascript" src="./common/helpers.js"></script>
 		<!--SCRIPTS_S-->
-
 		<!--SCRIPTS_E-->
 		<script>
 			var resDivs = "<div id=\"passed\" style=\"color: green\">• PASSED: </div>"
@@ -23,7 +22,6 @@ SPDX-License-Identifier: GPL-3.0-or-later
 				
 				var resultsAcc = {};
 				//CALL_S
-
 				//CALL_E
 				updateDoc(resultsAcc);
 				messageExtension("direct");
diff --git a/tests/fpd_tests/resources.js b/tests/fpd_tests/resources.js
new file mode 100644
index 0000000..e69de29
--- /dev/null
+++ b/tests/fpd_tests/resources.js
diff --git a/tests/fpd_tests/start_fpd_test.sh b/tests/fpd_tests/start_fpd_test.sh
index 56cdf6e..08cde7b 100644
--- a/tests/fpd_tests/start_fpd_test.sh
+++ b/tests/fpd_tests/start_fpd_test.sh
@@ -42,10 +42,12 @@ done
 # update html and worker files to include and call test functions
 echo -ne "UPDATING ./common/* FILES FOR THE LATEST CONFIGURATION"
 
+# clear script placeholders in test files
 sed -i "/.*<!--SCRIPTS_S-->.*/,/.*<!--SCRIPTS_E-->.*/{/.*<!--SCRIPTS_S-->.*/!{/.*<!--SCRIPTS_E-->.*/!d}}" ./index.html
 sed -i "/.*<!--SCRIPTS_S-->.*/,/.*<!--SCRIPTS_E-->.*/{/.*<!--SCRIPTS_S-->.*/!{/.*<!--SCRIPTS_E-->.*/!d}}" ./common/iframe.html
 sed -i "/.*\/\/SCRIPTS_S.*/,/.*\/\/SCRIPTS_E.*/{/.*\/\/SCRIPTS_S.*/!{/.*\/\/SCRIPTS_E.*/!d}}" ./common/worker.js
 
+# add test scripts to source code of testing website
 test_functions=()
 test_files=`find ${tests_path} -type f -name "*.js"`
 for file in $test_files
@@ -59,11 +61,13 @@ done
 
 echo -ne "....."
 
+# clear function placeholders in test files
 for file in ./index.html ./common/iframe.html ./common/worker.js
 do
 	sed -i "/.*\/\/CALL_S.*/,/.*\/\/CALL_E.*/{/.*\/\/CALL_S.*/!{/.*\/\/CALL_E.*/!d}}" $file
 done
 
+# add calls of testing functions to source code of testing website
 for fce in "${test_functions[@]}"
 do
 	for file in ./index.html ./common/iframe.html ./common/worker.js
@@ -243,7 +247,20 @@ cd $fpd_test_path
 trap_ctrlc()
 {
 	echo ""
-    echo -ne "PERFORMING CLEAN-UP AND REVERTING FILES....."
+    echo -ne "CLEANING FPD TESTING FILES TO ORIGINAL STATE....."
+
+	sed -i "/.*<!--SCRIPTS_S-->.*/,/.*<!--SCRIPTS_E-->.*/{/.*<!--SCRIPTS_S-->.*/!{/.*<!--SCRIPTS_E-->.*/!d}}" ./index.html
+	sed -i "/.*<!--SCRIPTS_S-->.*/,/.*<!--SCRIPTS_E-->.*/{/.*<!--SCRIPTS_S-->.*/!{/.*<!--SCRIPTS_E-->.*/!d}}" ./common/iframe.html
+	sed -i "/.*\/\/SCRIPTS_S.*/,/.*\/\/SCRIPTS_E.*/{/.*\/\/SCRIPTS_S.*/!{/.*\/\/SCRIPTS_E.*/!d}}" ./common/worker.js
+
+	for file in ./index.html ./common/iframe.html ./common/worker.js
+	do
+		sed -i "/.*\/\/CALL_S.*/,/.*\/\/CALL_E.*/{/.*\/\/CALL_S.*/!{/.*\/\/CALL_E.*/!d}}" $file
+	done
+
+	echo ".....DONE!"
+	
+    echo -ne "PERFORMING CLEAN-UP AND REVERTING EXTENSION FILES....."
 	
 	cd $default_extension_path
 	rm -f ./fpd_test.js
diff --git a/tests/fpd_tests/tests/custom.js b/tests/fpd_tests/tests/custom.js
new file mode 100644
index 0000000..f25dcd7
--- /dev/null
+++ b/tests/fpd_tests/tests/custom.js
@@ -0,0 +1,48 @@
+/** \file
+ * \brief Example of custom FPD test file
+ *
+ *  \author Copyright (C) 2021  Marek Salon
+ *
+ *  \license SPDX-License-Identifier: GPL-3.0-or-later
+ */
+//
+//  This program is free software: you can redistribute it and/or modify
+//  it under the terms of the GNU General Public License as published by
+//  the Free Software Foundation, either version 3 of the License, or
+//  (at your option) any later version.
+//
+//  This program is distributed in the hope that it will be useful,
+//  but WITHOUT ANY WARRANTY; without even the implied warranty of
+//  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+//  GNU General Public License for more details.
+//
+//  You should have received a copy of the GNU General Public License
+//  along with this program.  If not, see <https://www.gnu.org/licenses/>.
+//
+
+// definition of test function must have "test_" prefix and "wrappers" parameter
+function test_canvasFp(wrappers) {
+	// wrap function body to try-catch construction - count only successful accesses
+	try {
+		var canvas = document.createElement('canvas');
+		canvas.height = 60;
+		canvas.width = 400;
+		var canvasContext = canvas.getContext('2d');
+		canvas.style.display = 'inline';
+		canvasContext.textBaseline = 'alphabetic';
+		canvasContext.fillStyle = '#f60';
+		canvasContext.fillRect(125, 1, 62, 20);
+		canvasContext.fillStyle = '#069';
+		canvasContext.font = '11pt no-real-font-123';
+		canvasContext.fillText("co ja viem, \uD83D\uDE03", 2, 15);
+		console.log(canvasContext.font)
+		canvasContext.font = 'bold 48px serif';
+
+		// end of function - have to specify all function accesses to resources that supposed to be wrapped and tested 
+		// use 'addWrapper' method like here:
+		addWrapper(wrappers, "CanvasRenderingContext2D.prototype.fillStyle", "set", 2);
+		addWrapper(wrappers, "CanvasRenderingContext2D.prototype.fillText", "call", 1);
+		addWrapper(wrappers, "CanvasRenderingContext2D.prototype.font", "set", 2);
+	}
+	catch {}
+}
\ No newline at end of file

