aboutsummaryrefslogtreecommitdiffstats
path: root/library/cpp/streams/bzip2/bzip2.cpp
diff options
context:
space:
mode:
authorAnton Samokhvalov <pg83@yandex.ru>2022-02-10 16:45:17 +0300
committerDaniil Cherednik <dcherednik@yandex-team.ru>2022-02-10 16:45:17 +0300
commitd3a398281c6fd1d3672036cb2d63f842d2cb28c5 (patch)
treedd4bd3ca0f36b817e96812825ffaf10d645803f2 /library/cpp/streams/bzip2/bzip2.cpp
parent72cb13b4aff9bc9cf22e49251bc8fd143f82538f (diff)
downloadydb-d3a398281c6fd1d3672036cb2d63f842d2cb28c5.tar.gz
Restoring authorship annotation for Anton Samokhvalov <pg83@yandex.ru>. Commit 2 of 2.
Diffstat (limited to 'library/cpp/streams/bzip2/bzip2.cpp')
-rw-r--r--library/cpp/streams/bzip2/bzip2.cpp342
1 files changed, 171 insertions, 171 deletions
diff --git a/library/cpp/streams/bzip2/bzip2.cpp b/library/cpp/streams/bzip2/bzip2.cpp
index 37c484fed7..bccc5c6807 100644
--- a/library/cpp/streams/bzip2/bzip2.cpp
+++ b/library/cpp/streams/bzip2/bzip2.cpp
@@ -1,204 +1,204 @@
-#include "bzip2.h"
-
-#include <util/memory/addstorage.h>
-#include <util/generic/scope.h>
-
-#include <contrib/libs/libbz2/bzlib.h>
-
-class TBZipDecompress::TImpl: public TAdditionalStorage<TImpl> {
-public:
+#include "bzip2.h"
+
+#include <util/memory/addstorage.h>
+#include <util/generic/scope.h>
+
+#include <contrib/libs/libbz2/bzlib.h>
+
+class TBZipDecompress::TImpl: public TAdditionalStorage<TImpl> {
+public:
inline TImpl(IInputStream* input)
- : Stream_(input)
- {
- Zero(BzStream_);
- Init();
- }
-
+ : Stream_(input)
+ {
+ Zero(BzStream_);
+ Init();
+ }
+
inline ~TImpl() {
- Clear();
- }
-
- inline void Init() {
+ Clear();
+ }
+
+ inline void Init() {
if (BZ2_bzDecompressInit(&BzStream_, 0, 0) != BZ_OK) {
- ythrow TBZipDecompressError() << "can not init bzip engine";
- }
- }
-
+ ythrow TBZipDecompressError() << "can not init bzip engine";
+ }
+ }
+
inline void Clear() noexcept {
BZ2_bzDecompressEnd(&BzStream_);
- }
-
- inline size_t Read(void* buf, size_t size) {
- BzStream_.next_out = (char*)buf;
- BzStream_.avail_out = size;
-
- while (true) {
- if (BzStream_.avail_in == 0) {
- if (FillInputBuffer() == 0) {
- return 0;
- }
- }
-
+ }
+
+ inline size_t Read(void* buf, size_t size) {
+ BzStream_.next_out = (char*)buf;
+ BzStream_.avail_out = size;
+
+ while (true) {
+ if (BzStream_.avail_in == 0) {
+ if (FillInputBuffer() == 0) {
+ return 0;
+ }
+ }
+
switch (BZ2_bzDecompress(&BzStream_)) {
- case BZ_STREAM_END: {
- Clear();
- Init();
+ case BZ_STREAM_END: {
+ Clear();
+ Init();
[[fallthrough]];
- }
-
- case BZ_OK: {
- const size_t processed = size - BzStream_.avail_out;
-
- if (processed) {
- return processed;
- }
-
- break;
- }
-
- default:
- ythrow TBZipDecompressError() << "bzip error";
- }
- }
- }
-
- inline size_t FillInputBuffer() {
- BzStream_.next_in = (char*)AdditionalData();
- BzStream_.avail_in = Stream_->Read(BzStream_.next_in, AdditionalDataLength());
-
- return BzStream_.avail_in;
- }
-
-private:
+ }
+
+ case BZ_OK: {
+ const size_t processed = size - BzStream_.avail_out;
+
+ if (processed) {
+ return processed;
+ }
+
+ break;
+ }
+
+ default:
+ ythrow TBZipDecompressError() << "bzip error";
+ }
+ }
+ }
+
+ inline size_t FillInputBuffer() {
+ BzStream_.next_in = (char*)AdditionalData();
+ BzStream_.avail_in = Stream_->Read(BzStream_.next_in, AdditionalDataLength());
+
+ return BzStream_.avail_in;
+ }
+
+private:
IInputStream* Stream_;
- bz_stream BzStream_;
-};
-
+ bz_stream BzStream_;
+};
+
TBZipDecompress::TBZipDecompress(IInputStream* input, size_t bufLen)
- : Impl_(new (bufLen) TImpl(input))
-{
-}
-
+ : Impl_(new (bufLen) TImpl(input))
+{
+}
+
TBZipDecompress::~TBZipDecompress() {
-}
-
-size_t TBZipDecompress::DoRead(void* buf, size_t size) {
- return Impl_->Read(buf, size);
-}
-
-class TBZipCompress::TImpl: public TAdditionalStorage<TImpl> {
-public:
+}
+
+size_t TBZipDecompress::DoRead(void* buf, size_t size) {
+ return Impl_->Read(buf, size);
+}
+
+class TBZipCompress::TImpl: public TAdditionalStorage<TImpl> {
+public:
inline TImpl(IOutputStream* stream, size_t level)
- : Stream_(stream)
- {
- Zero(BzStream_);
-
+ : Stream_(stream)
+ {
+ Zero(BzStream_);
+
if (BZ2_bzCompressInit(&BzStream_, level, 0, 0) != BZ_OK) {
- ythrow TBZipCompressError() << "can not init bzip engine";
- }
-
- BzStream_.next_out = TmpBuf();
- BzStream_.avail_out = TmpBufLen();
- }
-
+ ythrow TBZipCompressError() << "can not init bzip engine";
+ }
+
+ BzStream_.next_out = TmpBuf();
+ BzStream_.avail_out = TmpBufLen();
+ }
+
inline ~TImpl() {
BZ2_bzCompressEnd(&BzStream_);
- }
-
- inline void Write(const void* buf, size_t size) {
- BzStream_.next_in = (char*)buf;
- BzStream_.avail_in = size;
-
- Y_DEFER {
- BzStream_.next_in = 0;
- BzStream_.avail_in = 0;
- };
-
- while (BzStream_.avail_in) {
+ }
+
+ inline void Write(const void* buf, size_t size) {
+ BzStream_.next_in = (char*)buf;
+ BzStream_.avail_in = size;
+
+ Y_DEFER {
+ BzStream_.next_in = 0;
+ BzStream_.avail_in = 0;
+ };
+
+ while (BzStream_.avail_in) {
const int ret = BZ2_bzCompress(&BzStream_, BZ_RUN);
-
- switch (ret) {
- case BZ_RUN_OK:
- continue;
-
- case BZ_PARAM_ERROR:
- case BZ_OUTBUFF_FULL:
- Stream_->Write(TmpBuf(), TmpBufLen() - BzStream_.avail_out);
- BzStream_.next_out = TmpBuf();
- BzStream_.avail_out = TmpBufLen();
-
- break;
-
- default:
- ythrow TBZipCompressError() << "bzip error(" << ret << ", " << BzStream_.avail_out << ")";
- }
- }
- }
-
- inline void Flush() {
- /*
- * TODO ?
- */
- }
-
- inline void Finish() {
+
+ switch (ret) {
+ case BZ_RUN_OK:
+ continue;
+
+ case BZ_PARAM_ERROR:
+ case BZ_OUTBUFF_FULL:
+ Stream_->Write(TmpBuf(), TmpBufLen() - BzStream_.avail_out);
+ BzStream_.next_out = TmpBuf();
+ BzStream_.avail_out = TmpBufLen();
+
+ break;
+
+ default:
+ ythrow TBZipCompressError() << "bzip error(" << ret << ", " << BzStream_.avail_out << ")";
+ }
+ }
+ }
+
+ inline void Flush() {
+ /*
+ * TODO ?
+ */
+ }
+
+ inline void Finish() {
int ret = BZ2_bzCompress(&BzStream_, BZ_FINISH);
-
- while (ret != BZ_STREAM_END) {
- Stream_->Write(TmpBuf(), TmpBufLen() - BzStream_.avail_out);
- BzStream_.next_out = TmpBuf();
- BzStream_.avail_out = TmpBufLen();
-
+
+ while (ret != BZ_STREAM_END) {
+ Stream_->Write(TmpBuf(), TmpBufLen() - BzStream_.avail_out);
+ BzStream_.next_out = TmpBuf();
+ BzStream_.avail_out = TmpBufLen();
+
ret = BZ2_bzCompress(&BzStream_, BZ_FINISH);
- }
-
- Stream_->Write(TmpBuf(), TmpBufLen() - BzStream_.avail_out);
- }
-
-private:
+ }
+
+ Stream_->Write(TmpBuf(), TmpBufLen() - BzStream_.avail_out);
+ }
+
+private:
inline char* TmpBuf() noexcept {
- return (char*)AdditionalData();
- }
-
+ return (char*)AdditionalData();
+ }
+
inline size_t TmpBufLen() const noexcept {
- return AdditionalDataLength();
- }
-
-private:
+ return AdditionalDataLength();
+ }
+
+private:
IOutputStream* Stream_;
- bz_stream BzStream_;
-};
-
+ bz_stream BzStream_;
+};
+
TBZipCompress::TBZipCompress(IOutputStream* out, size_t compressionLevel, size_t bufLen)
- : Impl_(new (bufLen) TImpl(out, compressionLevel))
-{
-}
-
+ : Impl_(new (bufLen) TImpl(out, compressionLevel))
+{
+}
+
TBZipCompress::~TBZipCompress() {
- try {
- Finish();
- } catch (...) {
- }
-}
-
-void TBZipCompress::DoWrite(const void* buf, size_t size) {
+ try {
+ Finish();
+ } catch (...) {
+ }
+}
+
+void TBZipCompress::DoWrite(const void* buf, size_t size) {
if (!Impl_) {
- ythrow TBZipCompressError() << "can not write to finished bzip stream";
+ ythrow TBZipCompressError() << "can not write to finished bzip stream";
}
- Impl_->Write(buf, size);
-}
-
-void TBZipCompress::DoFlush() {
+ Impl_->Write(buf, size);
+}
+
+void TBZipCompress::DoFlush() {
if (Impl_) {
Impl_->Flush();
}
-}
-
-void TBZipCompress::DoFinish() {
+}
+
+void TBZipCompress::DoFinish() {
THolder<TImpl> impl(Impl_.Release());
if (impl) {
impl->Finish();
}
-}
+}