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 <CoreFoundation/CFStreamAbstract.h> 40#include <WebKitSystemInterface/WebKitSystemInterface.h> 41#include <sys/types.h> 42#include <wtf/Assertions.h> 43#include <wtf/HashMap.h> 44#include <wtf/RetainPtr.h> 45#include <wtf/text/CString.h> 46 47#define USE_V1_CFSTREAM_CALLBACKS 48#ifdef USE_V1_CFSTREAM_CALLBACKS 49typedef CFReadStreamCallBacksV1 WCReadStreamCallBacks; 50#else 51typedef CFReadStreamCallBacks WCReadStreamCallBacks; 52#endif 53 54namespace WebCore { 55 56void setHTTPBody(CFMutableURLRequestRef request, PassRefPtr<FormData> formData) 57{ 58 if (!formData) { 59 wkCFURLRequestSetHTTPRequestBodyParts(request, 0); 60 return; 61 } 62 63 size_t count = formData->elements().size(); 64 65 if (count == 0) 66 return; 67 68 // Handle the common special case of one piece of form data, with no files. 69 if (count == 1) { 70 const FormDataElement& element = formData->elements()[0]; 71 if (element.m_type == FormDataElement::data) { 72 CFDataRef data = CFDataCreate(0, reinterpret_cast<const UInt8 *>(element.m_data.data()), element.m_data.size()); 73 CFURLRequestSetHTTPRequestBody(request, data); 74 CFRelease(data); 75 return; 76 } 77 } 78 79 RetainPtr<CFMutableArrayRef> array(AdoptCF, CFArrayCreateMutable(0, 0, &kCFTypeArrayCallBacks)); 80 81 for (size_t i = 0; i < count; ++i) { 82 const FormDataElement& element = formData->elements()[i]; 83 if (element.m_type == FormDataElement::data) { 84 RetainPtr<CFDataRef> data(AdoptCF, CFDataCreate(0, reinterpret_cast<const UInt8*>(element.m_data.data()), element.m_data.size())); 85 CFArrayAppendValue(array.get(), data.get()); 86 } else { 87 RetainPtr<CFStringRef> filename(AdoptCF, element.m_filename.createCFString()); 88 CFArrayAppendValue(array.get(), filename.get()); 89 } 90 } 91 92 wkCFURLRequestSetHTTPRequestBodyParts(request, array.get()); 93} 94 95PassRefPtr<FormData> httpBodyFromRequest(CFURLRequestRef request) 96{ 97 if (RetainPtr<CFDataRef> bodyData = CFURLRequestCopyHTTPRequestBody(request)) 98 return FormData::create(CFDataGetBytePtr(bodyData.get()), CFDataGetLength(bodyData.get())); 99 100 if (RetainPtr<CFArrayRef> bodyParts = wkCFURLRequestCopyHTTPRequestBodyParts(request)) { 101 RefPtr<FormData> formData = FormData::create(); 102 103 CFIndex count = CFArrayGetCount(bodyParts.get()); 104 for (CFIndex i = 0; i < count; i++) { 105 CFTypeRef bodyPart = CFArrayGetValueAtIndex(bodyParts.get(), i); 106 CFTypeID typeID = CFGetTypeID(bodyPart); 107 if (typeID == CFStringGetTypeID()) { 108 String filename = (CFStringRef)bodyPart; 109 formData->appendFile(filename); 110 } else if (typeID == CFDataGetTypeID()) { 111 CFDataRef data = (CFDataRef)bodyPart; 112 formData->appendData(CFDataGetBytePtr(data), CFDataGetLength(data)); 113 } else 114 ASSERT_NOT_REACHED(); 115 } 116 return formData.release(); 117 } 118 119 // FIXME: what to do about arbitrary body streams? 120 return 0; 121} 122 123} // namespace WebCore 124 125#endif // USE(CFNETWORK) 126