Record Code Signature of Downloaded DMG files

chrome/utility/safe_browsing/mac/dmg_iterator.cc

 // Copyright 2015 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "chrome/utility/safe_browsing/mac/dmg_iterator.h"
 
 #include "chrome/utility/safe_browsing/mac/hfs.h"
 #include "chrome/utility/safe_browsing/mac/read_stream.h"
 
 namespace safe_browsing {
@@ skipping 17 matching lines @@
   for (size_t i = 0; i < udif_.GetNumberOfPartitions(); ++i) {
     if (udif_.GetPartitionType(i) == "Apple_HFS" ||
         udif_.GetPartitionType(i) == "Apple_HFSX") {
       partitions_.push_back(udif_.GetPartitionReadStream(i));
     }
   }
 
   return partitions_.size() > 0;
 }
 
+const std::vector<uint8_t>& DMGIterator::GetCodeSignature() {
+  return udif_.GetDmgSignatureData();
+}
+
 bool DMGIterator::Next() {
   // Iterate through all the HFS partitions in the DMG file.
   for (; current_partition_ < partitions_.size(); ++current_partition_) {
     if (!hfs_) {
       hfs_.reset(new HFSIterator(partitions_[current_partition_].get()));
       if (!hfs_->Open())
         continue;
     }
 
     // Iterate through the HFS filesystem until a concrete file is found.
@@ skipping 26 matching lines @@
 base::string16 DMGIterator::GetPath() {
   return hfs_->GetPath();
 }
 
 std::unique_ptr<ReadStream> DMGIterator::GetReadStream() {
   return hfs_->GetReadStream();
 }
 
 }  // namespace dmg
 }  // namespace safe_browsing