Source/WebCore/platform/network/curl/CurlFormDataStream.h

   1 /*
   2  * Copyright (C) 2008 Apple Inc.  All rights reserved.
   3  * Copyright (C) 2017 Sony Interactive Entertainment Inc.
   4  *
   5  * Redistribution and use in source and binary forms, with or without
   6  * modification, are permitted provided that the following conditions
   7  * are met:
   8  * 1. Redistributions of source code must retain the above copyright
   9  *    notice, this list of conditions and the following disclaimer.
  10  * 2. Redistributions in binary form must reproduce the above copyright
  11  *    notice, this list of conditions and the following disclaimer in the
  12  *    documentation and/or other materials provided with the distribution.
  13  *
  14  * THIS SOFTWARE IS PROVIDED BY APPLE INC. ``AS IS'' AND ANY
  15  * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  16  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
  17  * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL APPLE INC. OR
  18  * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
  19  * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
  20  * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
  21  * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY
  22  * OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  23  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
  24  * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  25  */
  26
  27 #pragma once
  28
  29 #include "FileSystem.h"
  30 #include "FormData.h"
  31 #include <wtf/Vector.h>
  32
  33 namespace WebCore {
  34
  35 class CurlFormDataStream {
  36 public:
  37     CurlFormDataStream(const FormData*);
  38     WEBCORE_EXPORT ~CurlFormDataStream();
  39
  40     void clean();
  41
  42     size_t elementSize() { return m_formData ? m_formData->elements().size() : 0; }
  43
  44     const Vector<char>* getPostData();
  45     bool shouldUseChunkTransfer();
  46     unsigned long long totalSize();
  47
  48     std::optional<size_t> read(char*, size_t);
  49     unsigned long long totalReadSize() { return m_totalReadSize; }
  50
  51 private:
  52     void computeContentLength();
  53
  54     std::optional<size_t> readFromFile(const FormDataElement&, char*, size_t);
  55     std::optional<size_t> readFromData(const FormDataElement&, char*, size_t);
  56
  57     RefPtr<FormData> m_formData;
  58
  59     std::unique_ptr<Vector<char>> m_postData;
  60     bool m_isContentLengthUpdated { false };
  61     bool m_shouldUseChunkTransfer { false };
  62     unsigned long long m_totalSize { 0 };
  63     unsigned long long m_totalReadSize { 0 };
  64
  65     size_t m_elementPosition { 0 };
  66
  67     FileSystem::PlatformFileHandle m_fileHandle { FileSystem::invalidPlatformFileHandle };
  68     size_t m_dataOffset { 0 };
  69 };
  70
  71 } // namespace WebCore