crypto/signature_creator_win.cc

   1 // Copyright (c) 2012 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 #include "crypto/signature_creator.h"
   6
   7 #include "base/logging.h"
   8 #include "base/memory/scoped_ptr.h"
   9 #include "crypto/rsa_private_key.h"
  10
  11 namespace crypto {
  12
  13 // static
  14 SignatureCreator* SignatureCreator::Create(RSAPrivateKey* key) {
  15   scoped_ptr<SignatureCreator> result(new SignatureCreator);
  16   result->key_ = key;
  17
  18   if (!CryptCreateHash(key->provider(), CALG_SHA1, 0, 0,
  19                        result->hash_object_.receive())) {
  20     NOTREACHED();
  21     return NULL;
  22   }
  23
  24   return result.release();
  25 }
  26
  27 SignatureCreator::SignatureCreator() : key_(NULL), hash_object_(0) {}
  28
  29 SignatureCreator::~SignatureCreator() {}
  30
  31 bool SignatureCreator::Update(const uint8* data_part, int data_part_len) {
  32   if (!CryptHashData(hash_object_, data_part, data_part_len, 0)) {
  33     NOTREACHED();
  34     return false;
  35   }
  36
  37   return true;
  38 }
  39
  40 bool SignatureCreator::Final(std::vector<uint8>* signature) {
  41   DWORD signature_length = 0;
  42   if (!CryptSignHash(hash_object_, AT_SIGNATURE, NULL, 0, NULL,
  43                      &signature_length)) {
  44     return false;
  45   }
  46
  47   std::vector<uint8> temp;
  48   temp.resize(signature_length);
  49   if (!CryptSignHash(hash_object_, AT_SIGNATURE, NULL, 0, &temp.front(),
  50                      &signature_length)) {
  51     return false;
  52   }
  53
  54   temp.resize(signature_length);
  55   for (size_t i = temp.size(); i > 0; --i)
  56     signature->push_back(temp[i - 1]);
  57
  58   return true;
  59 }
  60
  61 }  // namespace crypto