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 <utility> 23 24 #include <brillo/secure_blob.h> 25 26 #include "update_engine/payload_consumer/extent_writer.h" 27 28 // BzipExtentWriter is a concrete ExtentWriter subclass that bzip-decompresses 29 // what it's given in Write. It passes the decompressed data to an underlying 30 // ExtentWriter. 31 32 namespace chromeos_update_engine { 33 34 class BzipExtentWriter : public ExtentWriter { 35 public: BzipExtentWriter(std::unique_ptr<ExtentWriter> next)36 explicit BzipExtentWriter(std::unique_ptr<ExtentWriter> next) 37 : next_(std::move(next)) { 38 memset(&stream_, 0, sizeof(stream_)); 39 } 40 ~BzipExtentWriter() override; 41 42 bool Init(const google::protobuf::RepeatedPtrField<Extent>& extents, 43 uint32_t block_size) override; 44 bool Write(const void* bytes, size_t count) override; 45 46 private: 47 std::unique_ptr<ExtentWriter> next_; // The underlying ExtentWriter. 48 bz_stream stream_{}; // the libbz2 stream 49 brillo::Blob input_buffer_; 50 }; 51 52 } // namespace chromeos_update_engine 53 54 #endif // UPDATE_ENGINE_PAYLOAD_CONSUMER_BZIP_EXTENT_WRITER_H_ 55