sb_sigutil.cc revision 2a99a7e74a7f215066514fe81d2bfa6639d9eddd
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" 145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/file_util.h" 152a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#include "base/files/file_path.h" 165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/logging.h" 175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/memory/ref_counted.h" 185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "chrome/browser/safe_browsing/signature_util.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) 405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) scoped_refptr<safe_browsing::SignatureUtil> sig_util( 415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) new safe_browsing::SignatureUtil()); 425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) safe_browsing::ClientDownloadRequest_SignatureInfo signature_info; 435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) sig_util->CheckSignature(cmd_line->GetSwitchValuePath(kExecutable), 445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) &signature_info); 455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) std::string serialized_info = signature_info.SerializeAsString(); 475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int bytes_written = file_util::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