Add -supported command line option to describe if a given file is supported

courgette/courgette_tool.cc

 // Copyright (c) 2011 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include <vector>
 #include <string>
 
 #include "base/at_exit.h"
 #include "base/basictypes.h"
 #include "base/command_line.h"
 #include "base/file_path.h"
 #include "base/file_util.h"
 #include "base/logging.h"
 #include "base/string_number_conversions.h"
 #include "base/string_util.h"
 #include "base/utf_string_conversions.h"
 #include "courgette/third_party/bsdiff.h"
 #include "courgette/courgette.h"
 #include "courgette/streams.h"
 
 
 void PrintHelp() {
   fprintf(stderr,
     "Usage:\n"
+    "  courgette -supported <executable_file>\n"
     "  courgette -dis <executable_file> <binary_assembly_file>\n"
     "  courgette -asm <binary_assembly_file> <executable_file>\n"
     "  courgette -disadj <executable_file> <reference> <binary_assembly_file>\n"
     "  courgette -gen <v1> <v2> <patch>\n"
     "  courgette -apply <v1> <patch> <v2>\n"
     "\n");
 }
 
 void UsageProblem(const char* message) {
   fprintf(stderr, "%s", message);
@@ skipping 63 matching lines @@
 
   courgette::DeleteEncodedProgram(encoded);
 
   courgette::SinkStream sink;
   if (!sinks.CopyTo(&sink))
     Problem("Can't combine serialized encoded program streams.");
 
   WriteSinkToFile(&sink, output_file);
 }
 
+bool Supported(const FilePath& input_file) {
+  bool result = false;
+
+  std::string buffer = ReadOrFail(input_file, "input");
+
+  courgette::ExecutableType type;
+  size_t detected_length;
+
+  DetectExecutableType(buffer.c_str(), buffer.length(),
+                       &type,
+                       &detected_length);
+
+  // If the detection fails, we just fall back on UNKNOWN
+  std::string format = "Unsupported";
+
+  switch (type)
+  {
+    case courgette::EXE_UNKNOWN:
+      break;
+
+    case courgette::EXE_WIN_32_X86:
+      format = "Windows 32 PE";
+      result = true;
+      break;
+
+    case courgette::EXE_ELF_32_X86:
+      format = "ELF 32 X86";
+      result = true;
+      break;
+  }
+
+  printf("%s Executable\n", format.c_str());
+  return result;
+}
+
 void DisassembleAndAdjust(const FilePath& program_file,
                           const FilePath& model_file,
                           const FilePath& output_file) {
   std::string program_buffer = ReadOrFail(program_file, "program");
   std::string model_buffer = ReadOrFail(model_file, "reference");
 
   courgette::AssemblyProgram* program = NULL;
   const courgette::Status parse_program_status =
       courgette::ParseDetectedExecutable(program_buffer.c_str(),
                                          program_buffer.length(),
@@ skipping 266 matching lines @@
   const CommandLine& command_line = *CommandLine::ForCurrentProcess();
 
   (void)logging::InitLogging(
       FILE_PATH_LITERAL("courgette.log"),
       logging::LOG_TO_BOTH_FILE_AND_SYSTEM_DEBUG_LOG,
       logging::LOCK_LOG_FILE,
       logging::APPEND_TO_OLD_LOG_FILE,
       logging::DISABLE_DCHECK_FOR_NON_OFFICIAL_RELEASE_BUILDS);
   logging::SetMinLogLevel(logging::LOG_VERBOSE);
 
+  bool cmd_sup = command_line.HasSwitch("supported");
   bool cmd_dis = command_line.HasSwitch("dis");
   bool cmd_asm = command_line.HasSwitch("asm");
   bool cmd_disadj = command_line.HasSwitch("disadj");
   bool cmd_make_patch = command_line.HasSwitch("gen");
   bool cmd_apply_patch = command_line.HasSwitch("apply");
   bool cmd_make_bsdiff_patch = command_line.HasSwitch("genbsdiff");
   bool cmd_apply_bsdiff_patch = command_line.HasSwitch("applybsdiff");
   bool cmd_spread_1_adjusted = command_line.HasSwitch("gen1a");
   bool cmd_spread_1_unadjusted = command_line.HasSwitch("gen1u");
 
   std::vector<FilePath> values;
   const CommandLine::StringVector& args = command_line.GetArgs();
   for (size_t i = 0; i < args.size(); ++i) {
     values.push_back(FilePath(args[i]));
   }
 
   // '-repeat=N' is for debugging.  Running many iterations can reveal leaks and
   // bugs in cleanup.
   int repeat_count = 1;
   std::string repeat_switch = command_line.GetSwitchValueASCII("repeat");
   if (!repeat_switch.empty())
     if (!base::StringToInt(repeat_switch, &repeat_count))
       repeat_count = 1;
 
-  if (cmd_dis + cmd_asm + cmd_disadj + cmd_make_patch + cmd_apply_patch +
-      cmd_make_bsdiff_patch + cmd_apply_bsdiff_patch +
+  if (cmd_sup + cmd_dis + cmd_asm + cmd_disadj + cmd_make_patch +
+      cmd_apply_patch + cmd_make_bsdiff_patch + cmd_apply_bsdiff_patch +
       cmd_spread_1_adjusted + cmd_spread_1_unadjusted
       != 1)
     UsageProblem(
         "Must have exactly one of:\n"
-        "  -asm, -dis, -disadj, -gen or -apply, -genbsdiff or -applybsdiff.");
+        "  -supported -asm, -dis, -disadj, -gen or -apply, -genbsdiff"
+        " or -applybsdiff.");
 
   while (repeat_count-- > 0) {
-    if (cmd_dis) {
-      if (values.size() != 2)
-        UsageProblem("-dis <executable_file> <courgette_file>");
-      Disassemble(values[0], values[1]);
+    if (cmd_sup) {
+      if (values.size() != 1)
+        UsageProblem("-supported <executable_file>");
+      return !Supported(values[0]);
+    } else if (cmd_dis) {
+        if (values.size() != 2)
+          UsageProblem("-dis <executable_file> <courgette_file>");
+        Disassemble(values[0], values[1]);
     } else if (cmd_asm) {
       if (values.size() != 2)
         UsageProblem("-asm <courgette_file_input> <executable_file_output>");
       Assemble(values[0], values[1]);
     } else if (cmd_disadj) {
       if (values.size() != 3)
         UsageProblem("-disadj <executable_file> <model> <courgette_file>");
       DisassembleAndAdjust(values[0], values[1], values[2]);
     } else if (cmd_make_patch) {
       if (values.size() != 3)
@@ skipping 13 matching lines @@
       ApplyBSDiffPatch(values[0], values[1], values[2]);
     } else if (cmd_spread_1_adjusted || cmd_spread_1_unadjusted) {
       if (values.size() != 3)
         UsageProblem("-gen1[au] <old_file> <new_file> <patch_files_root>");
       DisassembleAdjustDiff(values[0], values[1], values[2],
                             cmd_spread_1_adjusted);
     } else {
       UsageProblem("No operation specified");
     }
   }
+
+  return 0;
 }