Simple tool to populate a signature info protocol buffer from an executable.

chrome/tools/safe_browsing/sb_sigutil.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.
+//
+// Simple tool that uses the SignatureUtil class to extract signature
+// information from an executable.  The output is an encoded
+// ClientDownloadRequest_SignatureInfo protocol buffer.
+//
+// Example usage: sb_sigutil --executable=blah.exe --output=siginfo.pb
+
+#include <string>
+
+#include "base/command_line.h"
+#include "base/file_path.h"
+#include "base/file_util.h"
+#include "base/logging.h"
+#include "base/memory/ref_counted.h"
+#include "chrome/browser/safe_browsing/signature_util.h"
+#include "chrome/common/safe_browsing/csd.pb.h"
+
+// Command-line switch for the executable to extract a signature from.
+static const char kExecutable[] = "executable";
+
+// File to write the output protocol buffer to.
+static const char kOutputFile[] = "output";
+
+int main(int argc, char* argv[]) {
+  CommandLine::Init(argc, argv);
+
+  CommandLine* cmd_line = CommandLine::ForCurrentProcess();
+  if (!cmd_line->HasSwitch(kExecutable)) {
+    LOG(ERROR) << "Must specify executable to open with --executable";
+    return 1;
+  }
+  if (!cmd_line->HasSwitch(kOutputFile)) {
+    LOG(ERROR) << "Must specify output file with --output";
+    return 1;
+  }
+
+  scoped_refptr<safe_browsing::SignatureUtil> sig_util(
+      new safe_browsing::SignatureUtil());
+  safe_browsing::ClientDownloadRequest_SignatureInfo signature_info;
+  sig_util->CheckSignature(cmd_line->GetSwitchValuePath(kExecutable),
+                           &signature_info);
+
+  std::string serialized_info = signature_info.SerializeAsString();
+  int bytes_written = file_util::WriteFile(
+      cmd_line->GetSwitchValuePath(kOutputFile),
+      serialized_info.data(),
+      serialized_info.size());
+  if (bytes_written != static_cast<int>(serialized_info.size())) {
+    LOG(ERROR) << "Error writing output file";
+    return 1;
+  }
+
+  return 0;
+}