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

   1 /*
   2  * Copyright (C) 2013 University of Szeged
   3  * Copyright (C) 2018 Sony Interactive Entertainment Inc.
   4  * All rights reserved.
   5  *
   6  * Redistribution and use in source and binary forms, with or without
   7  * modification, are permitted provided that the following conditions
   8  * are met:
   9  * 1. Redistributions of source code must retain the above copyright
  10  *    notice, this list of conditions and the following disclaimer.
  11  * 2. Redistributions in binary form must reproduce the above copyright
  12  *    notice, this list of conditions and the following disclaimer in the
  13  *    documentation and/or other materials provided with the distribution.
  14  *
  15  * THIS SOFTWARE IS PROVIDED BY UNIVERSITY OF SZEGED ``AS IS'' AND ANY
  16  * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  17  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
  18  * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL UNIVERSITY OF SZEGED OR
  19  * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
  20  * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
  21  * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
  22  * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY
  23  * OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  24  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
  25  * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  26  */
  27
  28 #pragma once
  29
  30 #include <wtf/Vector.h>
  31 #include <wtf/text/WTFString.h>
  32
  33 namespace WebCore {
  34
  35 class CurlMultipartHandleClient;
  36 class CurlResponse;
  37 class SharedBuffer;
  38
  39 class CurlMultipartHandle {
  40 public:
  41     static std::unique_ptr<CurlMultipartHandle> createIfNeeded(CurlMultipartHandleClient&, const CurlResponse&);
  42
  43     CurlMultipartHandle(CurlMultipartHandleClient&, const String&);
  44     ~CurlMultipartHandle() { }
  45
  46     void didReceiveData(const SharedBuffer&);
  47     void didComplete();
  48
  49 private:
  50     enum class State {
  51         CheckBoundary,
  52         InBoundary,
  53         InHeader,
  54         InContent,
  55         EndBoundary,
  56         End
  57     };
  58
  59     static std::optional<String> extractBoundary(const CurlResponse&);
  60     static std::optional<String> extractBoundaryFromContentType(const String&);
  61
  62     bool processContent();
  63     bool checkForBoundary(size_t& boundaryStartPosition, size_t& lastPartialMatchPosition);
  64     size_t matchedLength(const char* data);
  65     bool parseHeadersIfPossible();
  66
  67     CurlMultipartHandleClient& m_client;
  68
  69     String m_boundary;
  70     Vector<char> m_buffer;
  71     Vector<String> m_headers;
  72
  73     State m_state { State::CheckBoundary };
  74 };
  75
  76 } // namespace WebCore