// Copyright (c) 2010 The Chromium OS Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#ifndef VBOOT_REFERENCE_FIRMWARE_UTILITY_H_
#define VBOOT_REFERENCE_FIRMWARE_UTILITY_H_

#include <string>

class FirmwareImage;
struct RSAPublicKey;

namespace vboot_reference {

// A class for handling verified boot firmware images.
class FirmwareUtility {
 public:
  FirmwareUtility();
  ~FirmwareUtility();

  // Print usage to stderr.
  void PrintUsage(void);

  // Parse command line options and populate data members.
  // Return true on success, false on failure.
  bool ParseCmdLineOptions(int argc, char* argv[]);

  // Generate a verified boot image by reading firmware data from in_file_.
  // Return true on success, false on failure.
  bool GenerateSignedImage();

  // Verify a previously generated signed firmware image using the root key read
  // from [root_key_pub_file_].
  bool VerifySignedImage();

  // Output the verified boot image to out_file_.
  void OutputSignedImage();


  bool is_generate() { return is_generate_; }
  bool is_verify() { return is_verify_; }

 private:

  // Check if all options were specified and sane.
  // Return true on success, false on failure.
  bool CheckOptions();

  FirmwareImage* image_;
  RSAPublicKey* root_key_pub_;
  std::string root_key_file_;
  std::string root_key_pub_file_;
  int firmware_version_;
  std::string firmware_key_file_;
  std::string firmware_key_pub_file_;
  int firmware_key_version_;
  int firmware_sign_algorithm_;
  std::string in_file_;
  std::string out_file_;
  bool is_generate_;  // Are we generating a new image?
  bool is_verify_;  // Are we just verifying an already signed image?
};

}  // namespace vboot_reference

#endif  // VBOOT_REFERENCE_FIRMWARE_UTILITY_H_