commit a4a6d5349866e4a0064b6d2264ae8efc3d041658
author Keun Soo Yim <yim@google.com> Wed Jun 08 09:11:40 2016 -0700
committer Keun Soo Yim <yim@google.com> Fri Jun 10 19:13:04 2016 -0700
tree 9d24f928b213b09df6ad19b5d5e83c8d3306a029
parent 7ad64920847dd16d37738d8a6d25c6358156a87c

callback support from driver to agent

Change-Id: I62c39e50e93ed619bf7977df23e32facf1f29d29

sysfuzzer/vtscompiler/code_gen/CodeGenBase.cpp

diff --git a/sysfuzzer/vtscompiler/code_gen/CodeGenBase.cpp b/sysfuzzer/vtscompiler/code_gen/CodeGenBase.cpp
index b4050f0..897c018 100644
--- a/sysfuzzer/vtscompiler/code_gen/CodeGenBase.cpp
+++ b/sysfuzzer/vtscompiler/code_gen/CodeGenBase.cpp
@@ -63,6 +63,9 @@
 
   GenerateAllHeader(fuzzer_extended_class_name, h_ss, message);
   cpp_ss << endl << endl;
+  if (message.component_class() == HAL) {
+    GenerateCppBodyCallbackFunction(cpp_ss, message, fuzzer_extended_class_name);
+  }
 
   cpp_ss << endl;
   GenerateCppBodyFuzzFunction(cpp_ss, message, fuzzer_extended_class_name);
@@ -87,9 +90,13 @@
   h_ss << "#ifndef __VTS_SPEC_" << vts_name_ << "__" << endl;
   h_ss << "#define __VTS_SPEC_" << vts_name_ << "__" << endl;
   h_ss << endl;
+  h_ss << "#include <stdio.h>" << endl;
+  h_ss << "#include <stdarg.h>" << endl;
+  h_ss << "#include <stdlib.h>" << endl;
   h_ss << "#define LOG_TAG \"" << fuzzer_extended_class_name << "\"" << endl;
   h_ss << "#include <utils/Log.h>" << endl;
   h_ss << "#include \"common/fuzz_tester/FuzzerBase.h\"" << endl;
+  h_ss << "#include \"common/fuzz_tester/FuzzerCallbackBase.h\"" << endl;
   for (auto const& header : message.header()) {
     h_ss << "#include " << header << endl;
   }
@@ -131,7 +138,7 @@
   h_ss << ") { }" << endl;
   h_ss << " protected:" << endl;
   h_ss << "  bool Fuzz(FunctionSpecificationMessage* func_msg," << endl;
-  h_ss << "            void** result);" << endl;
+  h_ss << "            void** result, int agent_port);" << endl;
 
   // produce Fuzz method(s) for sub_struct(s).
   for (auto const& sub_struct : message.sub_struct()) {
@@ -157,7 +164,7 @@
     const StructSpecificationMessage& message, const string& parent_path) {
   h_ss << "  bool Fuzz_" << parent_path << message.name()
       << "(FunctionSpecificationMessage* func_msg," << endl;
-  h_ss << "            void** result);" << endl;
+  h_ss << "            void** result, int agent_port);" << endl;
 
   for (auto const& sub_struct : message.sub_struct()) {
     GenerateFuzzFunctionForSubStruct(
@@ -204,5 +211,12 @@
   }
   return component_name;
 }
+
+
+void CodeGenBase::GenerateCppBodyCallbackFunction(
+    std::stringstream& /*cpp_ss*/,
+    const InterfaceSpecificationMessage& /*message*/,
+    const string& /*fuzzer_extended_class_name*/) {}
+
 }  // namespace vts
 }  // namespace android