1/* 2 * Copyright (C) 2013 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 * 1. Redistributions of source code must retain the above copyright 8 * notice, this list of conditions and the following disclaimer. 9 * 2. Redistributions in binary form must reproduce the above copyright 10 * notice, this list of conditions and the following disclaimer in the 11 * documentation and/or other materials provided with the distribution. 12 * 13 * THIS SOFTWARE IS PROVIDED BY APPLE INC. AND ITS CONTRIBUTORS ``AS IS'' 14 * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, 15 * THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR 16 * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL APPLE INC. OR ITS CONTRIBUTORS 17 * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR 18 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF 19 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS 20 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN 21 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) 22 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF 23 * THE POSSIBILITY OF SUCH DAMAGE. 24 */ 25 26#include "config.h" 27#include "CookieStorageShimLibrary.h" 28 29#if ENABLE(NETWORK_PROCESS) && !PLATFORM(IOS) 30 31#include <WebCore/DynamicLinkerInterposing.h> 32 33extern "C" CFDictionaryRef _CFHTTPCookieStorageCopyRequestHeaderFieldsForURL(CFAllocatorRef inAllocator, CFHTTPCookieStorageRef inCookieStorage, CFURLRef inRequestURL); 34 35namespace WebKit { 36 37extern "C" void WebKitCookieStorageShimInitialize(const CookieStorageShimCallbacks&); 38 39static CookieStorageShimCallbacks cookieStorageShimCallbacks; 40 41class ShimProtector { 42public: 43 ShimProtector() { ++m_count; } 44 ~ShimProtector() { --m_count; } 45 static unsigned count() { return m_count; } 46private: 47 __thread static unsigned m_count; 48}; 49 50__thread unsigned ShimProtector::m_count = 0; 51 52static CFDictionaryRef shimCFHTTPCookieStorageCopyRequestHeaderFieldsForURL(CFAllocatorRef inAllocator, CFHTTPCookieStorageRef inCookieStorage, CFURLRef inRequestURL) 53{ 54 ShimProtector protector; 55 56 do { 57 // Protect against uninitialized callbacks: 58 if (!cookieStorageShimCallbacks.cookieStorageCopyRequestHeaderFieldsForURL) 59 break; 60 61 // Protect against accidental recursion: 62 if (ShimProtector::count() > 1) 63 break; 64 65 CFDictionaryRef results = cookieStorageShimCallbacks.cookieStorageCopyRequestHeaderFieldsForURL(inCookieStorage, inRequestURL); 66 if (!results) 67 break; 68 69 return results; 70 } while (0); 71 72 // If we've failed to retrieve the cookies manually, fall back to the original imposed function: 73 return _CFHTTPCookieStorageCopyRequestHeaderFieldsForURL(inAllocator, inCookieStorage, inRequestURL); 74} 75 76DYLD_INTERPOSE(shimCFHTTPCookieStorageCopyRequestHeaderFieldsForURL, _CFHTTPCookieStorageCopyRequestHeaderFieldsForURL); 77 78__attribute__((visibility("default"))) 79void WebKitCookieStorageShimInitialize(const CookieStorageShimCallbacks& callbacks) 80{ 81 // Because the value of cookieStorageShimCallbacks will be read from mulitple threads, 82 // only allow it to be initialized once. 83 static int initialized = 0; 84 if (!OSAtomicCompareAndSwapInt(0, 1, &initialized)) { 85 return; 86 } 87 88 cookieStorageShimCallbacks = callbacks; 89} 90 91} 92 93#endif // ENABLE(NETWORK_PROCESS) && !PLATFORM(IOS) 94