15821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Copyright (c) 2011 The Chromium Authors. All rights reserved. 25821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Use of this source code is governed by a BSD-style license that can be 35821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// found in the LICENSE file. 45821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// 55821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Simple tool that uses the SignatureUtil class to extract signature 65821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// information from an executable. The output is an encoded 75821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// ClientDownloadRequest_SignatureInfo protocol buffer. 85821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// 95821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Example usage: sb_sigutil --executable=blah.exe --output=siginfo.pb 105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <string> 125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/command_line.h" 142a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#include "base/files/file_path.h" 151320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci#include "base/files/file_util.h" 165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/logging.h" 175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/memory/ref_counted.h" 18effb81e5f8246d0db0270817048dc992db66e9fbBen Murdoch#include "chrome/browser/safe_browsing/binary_feature_extractor.h" 195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "chrome/common/safe_browsing/csd.pb.h" 205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Command-line switch for the executable to extract a signature from. 225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)static const char kExecutable[] = "executable"; 235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// File to write the output protocol buffer to. 255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)static const char kOutputFile[] = "output"; 265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)int main(int argc, char* argv[]) { 285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) CommandLine::Init(argc, argv); 295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) CommandLine* cmd_line = CommandLine::ForCurrentProcess(); 315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (!cmd_line->HasSwitch(kExecutable)) { 325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) LOG(ERROR) << "Must specify executable to open with --executable"; 335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return 1; 345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (!cmd_line->HasSwitch(kOutputFile)) { 365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) LOG(ERROR) << "Must specify output file with --output"; 375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return 1; 385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 40effb81e5f8246d0db0270817048dc992db66e9fbBen Murdoch scoped_refptr<safe_browsing::BinaryFeatureExtractor> extractor( 41effb81e5f8246d0db0270817048dc992db66e9fbBen Murdoch new safe_browsing::BinaryFeatureExtractor()); 425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) safe_browsing::ClientDownloadRequest_SignatureInfo signature_info; 43effb81e5f8246d0db0270817048dc992db66e9fbBen Murdoch extractor->CheckSignature(cmd_line->GetSwitchValuePath(kExecutable), 44effb81e5f8246d0db0270817048dc992db66e9fbBen Murdoch &signature_info); 455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) std::string serialized_info = signature_info.SerializeAsString(); 47a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles) int bytes_written = base::WriteFile( 485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) cmd_line->GetSwitchValuePath(kOutputFile), 495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) serialized_info.data(), 505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) serialized_info.size()); 515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (bytes_written != static_cast<int>(serialized_info.size())) { 525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) LOG(ERROR) << "Error writing output file"; 535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return 1; 545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return 0; 575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 58