// // Copyright (C) 2015 The Android Open Source Project // // Licensed under the Apache License, Version 2.0 (the "License"); // you may not use this file except in compliance with the License. // You may obtain a copy of the License at // // http://www.apache.org/licenses/LICENSE-2.0 // // Unless required by applicable law or agreed to in writing, software // distributed under the License is distributed on an "AS IS" BASIS, // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. // See the License for the specific language governing permissions and // limitations under the License. // #ifndef UPDATE_ENGINE_PAYLOAD_GENERATOR_BLOB_FILE_WRITER_H_ #define UPDATE_ENGINE_PAYLOAD_GENERATOR_BLOB_FILE_WRITER_H_ #include #include #include namespace chromeos_update_engine { class BlobFileWriter { public: // Create the BlobFileWriter object that will manage the blobs stored to // |blob_fd| in a thread safe way. BlobFileWriter(int blob_fd, off_t* blob_file_size) : blob_fd_(blob_fd), blob_file_size_(blob_file_size) {} // Store the passed |blob| in the blob file. Returns the offset at which it // was stored, or -1 in case of failure. off_t StoreBlob(const brillo::Blob& blob); // The number of |total_blobs| is the number of blobs that will be stored but // is only used for logging purposes. If not set or set to 0, logging will be // skipped. This function will also reset the number of stored blobs to 0. void SetTotalBlobs(size_t total_blobs); private: size_t total_blobs_{0}; size_t stored_blobs_{0}; // The file and its size are protected with the |blob_mutex_|. int blob_fd_; off_t* blob_file_size_; base::Lock blob_mutex_; DISALLOW_COPY_AND_ASSIGN(BlobFileWriter); }; } // namespace chromeos_update_engine #endif // UPDATE_ENGINE_PAYLOAD_GENERATOR_BLOB_FILE_WRITER_H_