Home | History | Annotate | Download | only in install
      1 /*
      2  * Copyright (C) 2019 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 #pragma once
     18 
     19 #include <stdint.h>
     20 
     21 #include <functional>
     22 #include <map>
     23 #include <memory>
     24 #include <string>
     25 #include <vector>
     26 
     27 #include <ziparchive/zip_archive.h>
     28 
     29 #include "verifier.h"
     30 
     31 // This class serves as a wrapper for an OTA update package. It aims to provide the common
     32 // interface for both packages loaded in memory and packages read from fd.
     33 class Package : public VerifierInterface {
     34  public:
     35   static std::unique_ptr<Package> CreateMemoryPackage(
     36       const std::string& path, const std::function<void(float)>& set_progress);
     37   static std::unique_ptr<Package> CreateMemoryPackage(
     38       std::vector<uint8_t> content, const std::function<void(float)>& set_progress);
     39   static std::unique_ptr<Package> CreateFilePackage(const std::string& path,
     40                                                     const std::function<void(float)>& set_progress);
     41 
     42   virtual ~Package() = default;
     43 
     44   // Opens the package as a zip file and returns the ZipArchiveHandle.
     45   virtual ZipArchiveHandle GetZipArchiveHandle() = 0;
     46 
     47   // Updates the progress in fraction during package verification.
     48   void SetProgress(float progress) override;
     49 
     50  protected:
     51   // An optional function to update the progress.
     52   std::function<void(float)> set_progress_;
     53 };
     54