mirror of https://github.com/tfussell/xlnt
145 lines
4.1 KiB
C++
145 lines
4.1 KiB
C++
/*
|
|
PARTIO SOFTWARE
|
|
Copyright 2010 Disney Enterprises, Inc. All rights reserved
|
|
|
|
Redistribution and use in source and binary forms, with or without
|
|
modification, are permitted provided that the following conditions are
|
|
met:
|
|
|
|
* Redistributions of source code must retain the above copyright
|
|
notice, this list of conditions and the following disclaimer.
|
|
|
|
* Redistributions in binary form must reproduce the above copyright
|
|
notice, this list of conditions and the following disclaimer in
|
|
the documentation and/or other materials provided with the
|
|
distribution.
|
|
|
|
* The names "Disney", "Walt Disney Pictures", "Walt Disney Animation
|
|
Studios" or the names of its contributors may NOT be used to
|
|
endorse or promote products derived from this software without
|
|
specific prior written permission from Walt Disney Pictures.
|
|
|
|
Disclaimer: THIS SOFTWARE IS PROVIDED BY WALT DISNEY PICTURES AND
|
|
CONTRIBUTORS "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING,
|
|
BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY, FITNESS
|
|
FOR A PARTICULAR PURPOSE, NONINFRINGEMENT AND TITLE ARE DISCLAIMED.
|
|
IN NO EVENT SHALL WALT DISNEY PICTURES, THE COPYRIGHT HOLDER OR
|
|
CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
|
|
EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
|
|
PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
|
|
PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND BASED ON ANY
|
|
THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
|
|
(INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
|
|
OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGES.
|
|
*/
|
|
|
|
#pragma once
|
|
|
|
#include <iostream>
|
|
#include <memory>
|
|
#include <unordered_map>
|
|
#include <vector>
|
|
|
|
#include <xlnt/utils/path.hpp>
|
|
|
|
namespace xlnt {
|
|
namespace detail {
|
|
|
|
/// <summary>
|
|
/// A structure representing the header that occurs before each compressed file in a ZIP
|
|
/// archive and again at the end of the file with more information.
|
|
/// </summary>
|
|
struct zip_file_header
|
|
{
|
|
std::uint16_t version = 20;
|
|
std::uint16_t flags = 0;
|
|
std::uint16_t compression_type = 8;
|
|
std::uint16_t stamp_date, stamp_time = 0;
|
|
std::uint32_t crc = 0;
|
|
std::uint32_t compressed_size = 0;
|
|
std::uint32_t uncompressed_size = 0;
|
|
std::string filename;
|
|
std::string comment;
|
|
std::vector<std::uint8_t> extra;
|
|
std::uint32_t header_offset = 0;
|
|
};
|
|
|
|
/// <summary>
|
|
/// Writes a series of uncompressed binary file data as ostreams into another ostream
|
|
/// according to the ZIP format.
|
|
/// </summary>
|
|
class zip_file_writer
|
|
{
|
|
public:
|
|
/// <summary>
|
|
/// Construct a new zip_file_writer which writes a ZIP archive to the given stream.
|
|
/// </summary>
|
|
zip_file_writer(std::ostream &stream);
|
|
|
|
/// <summary>
|
|
/// Destructor.
|
|
/// </summary>
|
|
virtual ~zip_file_writer();
|
|
|
|
/// <summary>
|
|
/// Returns a pointer to a streambuf which compresses the data it receives.
|
|
/// </summary>
|
|
std::unique_ptr<std::streambuf> open(const path &file);
|
|
|
|
private:
|
|
std::vector<zip_file_header> file_headers_;
|
|
std::ostream &destination_stream_;
|
|
};
|
|
|
|
/// <summary>
|
|
/// Reads an archive containing a number of files from an istream and allows them
|
|
/// to be decompressed into an istream.
|
|
/// </summary>
|
|
class zip_file_reader
|
|
{
|
|
public:
|
|
/// <summary>
|
|
/// Construct a new zip_file_reader which reads a ZIP archive from the given stream.
|
|
/// </summary>
|
|
zip_file_reader(std::istream &stream);
|
|
|
|
/// <summary>
|
|
/// Destructor.
|
|
/// </summary>
|
|
virtual ~zip_file_reader();
|
|
|
|
/// <summary>
|
|
///
|
|
/// </summary>
|
|
std::unique_ptr<std::streambuf> open(const path &file);
|
|
|
|
/// <summary>
|
|
///
|
|
/// </summary>
|
|
std::vector<path> files() const;
|
|
|
|
/// <summary>
|
|
///
|
|
/// </summary>
|
|
bool has_file(const path &filename) const;
|
|
|
|
private:
|
|
/// <summary>
|
|
///
|
|
/// </summary>
|
|
bool read_central_header();
|
|
|
|
/// <summary>
|
|
///
|
|
/// </summary>
|
|
std::unordered_map<std::string, zip_file_header> file_headers_;
|
|
|
|
/// <summary>
|
|
///
|
|
/// </summary>
|
|
std::istream &source_stream_;
|
|
};
|
|
|
|
} // namespace detail
|
|
} // namespace xlnt
|