Update test.cpp

ihsinme · web-flow · commit 2dc85e183c56 · 2022-03-03T10:20:41.000+03:00
diff --git a/cpp/ql/test/experimental/query-tests/Security/CWE/CWE-754/semmle/tests/test.cpp b/cpp/ql/test/experimental/query-tests/Security/CWE/CWE-754/semmle/tests/test.cpp
@@ -35,14 +35,18 @@ int functionWork1_() {
   return i;
 }
 
-int functionWork1b() {
+int functionWork1b(int retIndex) {
   int i;
   char a[10];
   int b;
   int *p = &b;
   scanf("%i", &i); // BAD
   scanf("%s", a); // BAD
   scanf("%i", p); // BAD
+  if(retIndex == 0)
+    return (int)a;
+  if(retIndex == 1)
+    return *p;
   return i;
 }
 
