Record Code Signature of Downloaded DMG files

chrome/utility/safe_browsing/mac/udif.h

 // 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.
 
 #ifndef CHROME_UTILITY_SAFE_BROWSING_MAC_UDIF_H_
 #define CHROME_UTILITY_SAFE_BROWSING_MAC_UDIF_H_
 
 #include <CoreFoundation/CoreFoundation.h>
 #include <stddef.h>
 #include <stdint.h>
@@ skipping 33 matching lines @@
 class UDIFParser {
  public:
   // Constructs an instance from a stream.
   explicit UDIFParser(ReadStream* stream);
   ~UDIFParser();
 
   // Parses the UDIF file. This method must be called before any other method.
   // If this returns false, it is not legal to call any other methods.
   bool Parse();
 
+  // Returns the blob of DMG signature data.
+  const std::vector<uint8_t>& GetCodeSignature();
+
   // Returns the number of partitions in this UDIF image.
   size_t GetNumberOfPartitions();
 
   // Returns the partition name for a given partition number, in the range of
   // [0,GetNumberOfPartitions()). This will include the number, name, and type
   // of partition. E.g., "disk image (Apple_HFS : 2)".
   std::string GetPartitionName(size_t part_number);
 
   // Returns the partition type as a string for the given partition number.
   // E.g., "Apple_HFS" and "Apple_Free".
   std::string GetPartitionType(size_t part_number);
 
   // Returns the size of the partition in bytes.
   size_t GetPartitionSize(size_t part_number);
 
   // Returns a stream of the raw partition data for the given partition
   // number.
   std::unique_ptr<ReadStream> GetPartitionReadStream(size_t part_number);
 
  private:
   // Parses the blkx plist trailer structure.
   bool ParseBlkx();
 
   ReadStream* const stream_;  // The stream backing the UDIF image. Weak.
   std::vector<std::string> partition_names_;  // The names of all partitions.
   // All blocks in the UDIF image.
   std::vector<std::unique_ptr<const UDIFBlock>> blocks_;
   uint16_t block_size_;  // The image's block size, in bytes.
+  std::vector<uint8_t> signature_blob_;  // DMG signature.
 
   DISALLOW_COPY_AND_ASSIGN(UDIFParser);
 };
 
 }  // namespace dmg
 }  // namespace safe_browsing
 
 #endif  // CHROME_UTILITY_SAFE_BROWSING_MAC_UDIF_H_