1 // 2 // Copyright (C) 2009 The Android Open Source Project 3 // 4 // Licensed under the Apache License, Version 2.0 (the "License"); 5 // you may not use this file except in compliance with the License. 6 // You may obtain a copy of the License at 7 // 8 // http://www.apache.org/licenses/LICENSE-2.0 9 // 10 // Unless required by applicable law or agreed to in writing, software 11 // distributed under the License is distributed on an "AS IS" BASIS, 12 // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 13 // See the License for the specific language governing permissions and 14 // limitations under the License. 15 // 16 17 #ifndef UPDATE_ENGINE_PAYLOAD_CONSUMER_BZIP_EXTENT_WRITER_H_ 18 #define UPDATE_ENGINE_PAYLOAD_CONSUMER_BZIP_EXTENT_WRITER_H_ 19 20 #include <bzlib.h> 21 #include <memory> 22 #include <vector> 23 24 #include <brillo/secure_blob.h> 25 26 #include "update_engine/common/utils.h" 27 #include "update_engine/payload_consumer/extent_writer.h" 28 29 // BzipExtentWriter is a concrete ExtentWriter subclass that bzip-decompresses 30 // what it's given in Write. It passes the decompressed data to an underlying 31 // ExtentWriter. 32 33 namespace chromeos_update_engine { 34 35 class BzipExtentWriter : public ExtentWriter { 36 public: BzipExtentWriter(std::unique_ptr<ExtentWriter> next)37 explicit BzipExtentWriter(std::unique_ptr<ExtentWriter> next) 38 : next_(std::move(next)) { 39 memset(&stream_, 0, sizeof(stream_)); 40 } 41 ~BzipExtentWriter() override = default; 42 43 bool Init(FileDescriptorPtr fd, 44 const std::vector<Extent>& extents, 45 uint32_t block_size) override; 46 bool Write(const void* bytes, size_t count) override; 47 bool EndImpl() override; 48 49 private: 50 std::unique_ptr<ExtentWriter> next_; // The underlying ExtentWriter. 51 bz_stream stream_; // the libbz2 stream 52 brillo::Blob input_buffer_; 53 }; 54 55 } // namespace chromeos_update_engine 56 57 #endif // UPDATE_ENGINE_PAYLOAD_CONSUMER_BZIP_EXTENT_WRITER_H_ 58