Update test.cpp

Author: ihsinme

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;
 }