diff options
Diffstat (limited to 'src/VBox/Main/include/DataStreamImpl.h')
-rw-r--r-- | src/VBox/Main/include/DataStreamImpl.h | 78 |
1 files changed, 78 insertions, 0 deletions
diff --git a/src/VBox/Main/include/DataStreamImpl.h b/src/VBox/Main/include/DataStreamImpl.h new file mode 100644 index 00000000..8239dcd7 --- /dev/null +++ b/src/VBox/Main/include/DataStreamImpl.h @@ -0,0 +1,78 @@ +/* $Id: DataStreamImpl.h $ */ +/** @file + * VirtualBox COM class implementation + */ + +/* + * Copyright (C) 2018-2022 Oracle and/or its affiliates. + * + * This file is part of VirtualBox base platform packages, as + * available from https://www.virtualbox.org. + * + * This program is free software; you can redistribute it and/or + * modify it under the terms of the GNU General Public License + * as published by the Free Software Foundation, in version 3 of the + * License. + * + * This program is distributed in the hope that it will be useful, but + * WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU + * General Public License for more details. + * + * You should have received a copy of the GNU General Public License + * along with this program; if not, see <https://www.gnu.org/licenses>. + * + * SPDX-License-Identifier: GPL-3.0-only + */ + +#ifndef MAIN_INCLUDED_DataStreamImpl_h +#define MAIN_INCLUDED_DataStreamImpl_h +#ifndef RT_WITHOUT_PRAGMA_ONCE +# pragma once +#endif + +#include "DataStreamWrap.h" + +#include <iprt/circbuf.h> +#include <iprt/semaphore.h> + +class ATL_NO_VTABLE DataStream + : public DataStreamWrap +{ +public: + DECLARE_COMMON_CLASS_METHODS(DataStream) + + HRESULT FinalConstruct(); + void FinalRelease(); + + HRESULT init(unsigned long aBufferSize); + void uninit(); + + /// Feed data into the stream, used by the stream source. + /// Blocks if the internal buffer cannot take anything, otherwise + /// as much as the internal buffer can hold is taken (if smaller + /// than @a cbWrite). Modeled after RTStrmWriteEx. + int i_write(const void *pvBuf, size_t cbWrite, size_t *pcbWritten); + + /// Marks the end of the stream. + int i_close(); + +private: + // wrapped IDataStream attributes and methods + HRESULT getReadSize(ULONG *aReadSize); + HRESULT read(ULONG aSize, ULONG aTimeoutMS, std::vector<BYTE> &aData); + +private: + /** The temporary buffer the conversion process writes into and the user reads from. */ + PRTCIRCBUF m_pBuffer; + /** Event semaphore for waiting until data is available. */ + RTSEMEVENT m_hSemEvtDataAvail; + /** Event semaphore for waiting until there is room in the buffer for writing. */ + RTSEMEVENT m_hSemEvtBufSpcAvail; + /** Flag whether the end of stream flag is set. */ + bool m_fEos; +}; + +#endif /* !MAIN_INCLUDED_DataStreamImpl_h */ + +/* vi: set tabstop=4 shiftwidth=4 expandtab: */ |