summaryrefslogtreecommitdiffstats
path: root/package/inc/ZipOutputStream.hxx
blob: d92f140f70f200cda7181acc0cb9680e780501f5 (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
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
/* -*- Mode: C++; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 4 -*- */
/*
 * This file is part of the LibreOffice project.
 *
 * 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/.
 *
 * This file incorporates work covered by the following license notice:
 *
 *   Licensed to the Apache Software Foundation (ASF) under one or more
 *   contributor license agreements. See the NOTICE file distributed
 *   with this work for additional information regarding copyright
 *   ownership. The ASF licenses this file to you 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 .
 */
#ifndef INCLUDED_PACKAGE_INC_ZIPOUTPUTSTREAM_HXX
#define INCLUDED_PACKAGE_INC_ZIPOUTPUTSTREAM_HXX

#include <com/sun/star/uno/Reference.hxx>
#include <com/sun/star/io/XOutputStream.hpp>

#include "ByteChucker.hxx"
#include <comphelper/threadpool.hxx>

#include <cstddef>
#include <exception>
#include <vector>

struct ZipEntry;
class ZipOutputEntry;
class ZipOutputEntryInThread;
class ZipPackageStream;

class ZipOutputStream
{
    css::uno::Reference< css::io::XOutputStream > m_xStream;
    ::std::vector < ZipEntry * > m_aZipList;
    std::shared_ptr<comphelper::ThreadTaskTag> mpThreadTaskTag;

    ByteChucker         m_aChucker;
    ZipEntry            *m_pCurrentEntry;
    std::vector< ZipOutputEntryInThread* > m_aEntries;
    std::exception_ptr m_aDeflateException;

public:
    ZipOutputStream(
        const css::uno::Reference< css::io::XOutputStream > &xOStream );
    ~ZipOutputStream();

    void addDeflatingThreadTask( ZipOutputEntryInThread *pEntry, std::unique_ptr<comphelper::ThreadTask> pThreadTask );

    /// @throws css::io::IOException
    /// @throws css::uno::RuntimeException
    void writeLOC( ZipEntry *pEntry, bool bEncrypt = false );
    /// @throws css::io::IOException
    /// @throws css::uno::RuntimeException
    void rawWrite( const css::uno::Sequence< sal_Int8 >& rBuffer );
    /// @throws css::io::IOException
    /// @throws css::uno::RuntimeException
    void rawCloseEntry( bool bEncrypt = false );

    /// @throws css::io::IOException
    /// @throws css::uno::RuntimeException
    void finish();
    const css::uno::Reference< css::io::XOutputStream >& getStream() const;

    static sal_uInt32 getCurrentDosTime();
    static void setEntry( ZipEntry *pEntry );

private:
    /// @throws css::io::IOException
    /// @throws css::uno::RuntimeException
    void writeEND(sal_uInt32 nOffset, sal_uInt32 nLength);
    /// @throws css::io::IOException
    /// @throws css::uno::RuntimeException
    void writeCEN( const ZipEntry &rEntry );
    /// @throws css::io::IOException
    /// @throws css::uno::RuntimeException
    void writeDataDescriptor( const ZipEntry &rEntry );
    void writeExtraFields( const ZipEntry& rEntry );

    // ScheduledThread handling helpers
    void consumeScheduledThreadTaskEntry(std::unique_ptr<ZipOutputEntryInThread> pCandidate);
    void consumeFinishedScheduledThreadTaskEntries();

public:
    void reduceScheduledThreadTasksToGivenNumberOrLess(
        std::size_t nThreadTasks);

    const std::shared_ptr<comphelper::ThreadTaskTag>& getThreadTaskTag() const { return mpThreadTaskTag; }
};

#endif

/* vim:set shiftwidth=4 softtabstop=4 expandtab: */