Mercurial > hg > orthanc
view OrthancFramework/Sources/HttpServer/HttpStreamTranscoder.h @ 4151:8c559dd5034b
Fix possible crash in HttpClient if sending multipart body (can occur in STOW-RS)
author | Sebastien Jodogne <s.jodogne@gmail.com> |
---|---|
date | Wed, 19 Aug 2020 11:59:02 +0200 |
parents | bf7b9edf6b81 |
children | 2d5209153b32 |
line wrap: on
line source
/** * Orthanc - A Lightweight, RESTful DICOM Store * Copyright (C) 2012-2016 Sebastien Jodogne, Medical Physics * Department, University Hospital of Liege, Belgium * Copyright (C) 2017-2020 Osimis S.A., Belgium * * This program is free software: you can redistribute it and/or * modify it under the terms of the GNU Lesser General Public License * as published by the Free Software Foundation, either version 3 of * the License, or (at your option) any later version. * * 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 * Lesser General Public License for more details. * * You should have received a copy of the GNU Lesser General Public * License along with this program. If not, see * <http://www.gnu.org/licenses/>. **/ #pragma once #include "BufferHttpSender.h" #include "../Compatibility.h" #include <memory> // For std::unique_ptr namespace Orthanc { class ORTHANC_PUBLIC HttpStreamTranscoder : public IHttpStreamAnswer { private: IHttpStreamAnswer& source_; CompressionType sourceCompression_; uint64_t bytesToSkip_; uint64_t skipped_; uint64_t currentChunkOffset_; bool ready_; std::unique_ptr<BufferHttpSender> uncompressed_; void ReadSource(std::string& buffer); HttpCompression SetupZlibCompression(bool deflateAllowed); public: HttpStreamTranscoder(IHttpStreamAnswer& source, CompressionType compression) : source_(source), sourceCompression_(compression), bytesToSkip_(0), skipped_(0), currentChunkOffset_(0), ready_(false) { } // This is the first method to be called virtual HttpCompression SetupHttpCompression(bool gzipAllowed, bool deflateAllowed); virtual bool HasContentFilename(std::string& filename) { return source_.HasContentFilename(filename); } virtual std::string GetContentType() { return source_.GetContentType(); } virtual uint64_t GetContentLength(); virtual bool ReadNextChunk(); virtual const char* GetChunkContent(); virtual size_t GetChunkSize(); }; }