code.vuplus.com Git - vuplus_webkit/blob - Source/WebCore/platform/network/cf/FormDataStreamCFNet.cpp

   1 /*
   2  * Copyright (C) 2005, 2006, 2007 Apple Inc.  All rights reserved.
   3  *
   4  * Redistribution and use in source and binary forms, with or without
   5  * modification, are permitted provided that the following conditions
   6  * are met:
   7  *
   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  * 3.  Neither the name of Apple Computer, Inc. ("Apple") nor the names of
  14  *     its contributors may be used to endorse or promote products derived
  15  *     from this software without specific prior written permission.
  16  *
  17  * THIS SOFTWARE IS PROVIDED BY APPLE AND ITS CONTRIBUTORS "AS IS" AND ANY
  18  * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
  19  * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
  20  * DISCLAIMED. IN NO EVENT SHALL APPLE OR ITS CONTRIBUTORS BE LIABLE FOR ANY
  21  * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
  22  * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
  23  * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
  24  * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  25  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
  26  * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  27  */
  28
  29 /* originally written by Becky Willrich, additional code by Darin Adler */
  30
  31 #include "config.h"
  32 #include "FormDataStreamCFNet.h"
  33
  34 #if USE(CFNETWORK)
  35
  36 #include "FileSystem.h"
  37 #include "FormData.h"
  38 #include <CFNetwork/CFURLRequestPriv.h>
  39 #include <sys/types.h>
  40 #include <wtf/Assertions.h>
  41 #include <wtf/HashMap.h>
  42 #include <wtf/RetainPtr.h>
  43 #include <wtf/text/CString.h>
  44
  45 #if PLATFORM(MAC)
  46 #include "WebCoreSystemInterface.h"
  47 #elif PLATFORM(WIN)
  48 #include <WebKitSystemInterface/WebKitSystemInterface.h>
  49 #endif
  50
  51 namespace WebCore {
  52
  53 void setHTTPBody(CFMutableURLRequestRef request, PassRefPtr<FormData> formData)
  54 {
  55     if (!formData) {
  56         wkCFURLRequestSetHTTPRequestBodyParts(request, 0);
  57         return;
  58     }
  59
  60     size_t count = formData->elements().size();
  61
  62     if (count == 0)
  63         return;
  64
  65     // Handle the common special case of one piece of form data, with no files.
  66     if (count == 1) {
  67         const FormDataElement& element = formData->elements()[0];
  68         if (element.m_type == FormDataElement::data) {
  69             CFDataRef data = CFDataCreate(0, reinterpret_cast<const UInt8 *>(element.m_data.data()), element.m_data.size());
  70             CFURLRequestSetHTTPRequestBody(request, data);
  71             CFRelease(data);
  72             return;
  73         }
  74     }
  75
  76     RetainPtr<CFMutableArrayRef> array(AdoptCF, CFArrayCreateMutable(0, 0, &kCFTypeArrayCallBacks));
  77
  78     for (size_t i = 0; i < count; ++i) {
  79         const FormDataElement& element = formData->elements()[i];
  80         if (element.m_type == FormDataElement::data) {
  81             RetainPtr<CFDataRef> data(AdoptCF, CFDataCreate(0, reinterpret_cast<const UInt8*>(element.m_data.data()), element.m_data.size()));
  82             CFArrayAppendValue(array.get(), data.get());
  83         } else {
  84             RetainPtr<CFStringRef> filename(AdoptCF, element.m_filename.createCFString());
  85             CFArrayAppendValue(array.get(), filename.get());
  86         }
  87     }
  88
  89     wkCFURLRequestSetHTTPRequestBodyParts(request, array.get());
  90 }
  91
  92 PassRefPtr<FormData> httpBodyFromRequest(CFURLRequestRef request)
  93 {
  94     RetainPtr<CFDataRef> bodyData(AdoptCF, CFURLRequestCopyHTTPRequestBody(request));
  95     if (bodyData)
  96         return FormData::create(CFDataGetBytePtr(bodyData.get()), CFDataGetLength(bodyData.get()));
  97
  98     RetainPtr<CFArrayRef> bodyParts(AdoptCF, wkCFURLRequestCopyHTTPRequestBodyParts(request));
  99     if (bodyParts) {
 100         RefPtr<FormData> formData = FormData::create();
 101
 102         CFIndex count = CFArrayGetCount(bodyParts.get());
 103         for (CFIndex i = 0; i < count; i++) {
 104             CFTypeRef bodyPart = CFArrayGetValueAtIndex(bodyParts.get(), i);
 105             CFTypeID typeID = CFGetTypeID(bodyPart);
 106             if (typeID == CFStringGetTypeID()) {
 107                 String filename = (CFStringRef)bodyPart;
 108                 formData->appendFile(filename);
 109             } else if (typeID == CFDataGetTypeID()) {
 110                 CFDataRef data = (CFDataRef)bodyPart;
 111                 formData->appendData(CFDataGetBytePtr(data), CFDataGetLength(data));
 112             } else
 113                 ASSERT_NOT_REACHED();
 114         }
 115         return formData.release();
 116     }
 117
 118     // FIXME: what to do about arbitrary body streams?
 119     return 0;
 120 }
 121
 122 } // namespace WebCore
 123
 124 #endif // USE(CFNETWORK)