1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. 2 // Use of this source code is governed by a BSD-style license that can be 3 // found in the LICENSE file. 4 // 5 // Simple tool that uses the SignatureUtil class to extract signature 6 // information from an executable. The output is an encoded 7 // ClientDownloadRequest_SignatureInfo protocol buffer. 8 // 9 // Example usage: sb_sigutil --executable=blah.exe --output=siginfo.pb 10 11 #include <string> 12 13 #include "base/command_line.h" 14 #include "base/file_util.h" 15 #include "base/files/file_path.h" 16 #include "base/logging.h" 17 #include "base/memory/ref_counted.h" 18 #include "chrome/browser/safe_browsing/signature_util.h" 19 #include "chrome/common/safe_browsing/csd.pb.h" 20 21 // Command-line switch for the executable to extract a signature from. 22 static const char kExecutable[] = "executable"; 23 24 // File to write the output protocol buffer to. 25 static const char kOutputFile[] = "output"; 26 27 int main(int argc, char* argv[]) { 28 CommandLine::Init(argc, argv); 29 30 CommandLine* cmd_line = CommandLine::ForCurrentProcess(); 31 if (!cmd_line->HasSwitch(kExecutable)) { 32 LOG(ERROR) << "Must specify executable to open with --executable"; 33 return 1; 34 } 35 if (!cmd_line->HasSwitch(kOutputFile)) { 36 LOG(ERROR) << "Must specify output file with --output"; 37 return 1; 38 } 39 40 scoped_refptr<safe_browsing::SignatureUtil> sig_util( 41 new safe_browsing::SignatureUtil()); 42 safe_browsing::ClientDownloadRequest_SignatureInfo signature_info; 43 sig_util->CheckSignature(cmd_line->GetSwitchValuePath(kExecutable), 44 &signature_info); 45 46 std::string serialized_info = signature_info.SerializeAsString(); 47 int bytes_written = file_util::WriteFile( 48 cmd_line->GetSwitchValuePath(kOutputFile), 49 serialized_info.data(), 50 serialized_info.size()); 51 if (bytes_written != static_cast<int>(serialized_info.size())) { 52 LOG(ERROR) << "Error writing output file"; 53 return 1; 54 } 55 56 return 0; 57 } 58
