summaryrefslogtreecommitdiffstats
path: root/toolkit/mozapps/update/updater/archivereader.h
blob: 1f18327f1d55acb1a04956c4bd0ba18c0037ea27 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
/* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
/* vim:set ts=2 sw=2 sts=2 et cindent: */
/* This Source Code Form is subject to the terms of the Mozilla Public
 * License, v. 2.0. If a copy of the MPL was not distributed with this
 * file, You can obtain one at http://mozilla.org/MPL/2.0/. */

#ifndef ArchiveReader_h__
#define ArchiveReader_h__

#include <stdio.h>
#include "mar.h"

#ifdef XP_WIN
#  include <windows.h>

typedef WCHAR NS_tchar;
#else
typedef char NS_tchar;
#endif

// This class provides an API to extract files from an update archive.
class ArchiveReader {
 public:
  ArchiveReader() = default;
  ~ArchiveReader() { Close(); }

  int Open(const NS_tchar* path);
  int VerifySignature();
  int VerifyProductInformation(const char* MARChannelID,
                               const char* appVersion);
  void Close();

  int ExtractFile(const char* item, const NS_tchar* destination);
  int ExtractFileToStream(const char* item, FILE* fp);

 private:
  int ExtractItemToStream(const MarItem* item, FILE* fp);

  MarFile* mArchive = nullptr;
  uint8_t* mInBuf = nullptr;
  uint8_t* mOutBuf = nullptr;
  size_t mInBufSize = 262144;
  size_t mOutBufSize = 262144;
};

#endif  // ArchiveReader_h__