Source/WebKit2/Shared/API/c/cf/WKStringCF.mm

   1 /*
   2  * Copyright (C) 2010 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 #import "config.h"
  27 #import "WKStringCF.h"
  28
  29 #import "WKAPICast.h"
  30 #import "WKNSString.h"
  31 #import <objc/runtime.h>
  32 #import <wtf/text/WTFString.h>
  33
  34 using namespace WebCore;
  35 using namespace WebKit;
  36
  37 #if WK_API_ENABLED
  38 static inline Class wkNSStringClass()
  39 {
  40     static dispatch_once_t once;
  41     static Class wkNSStringClass;
  42     dispatch_once(&once, ^{
  43         wkNSStringClass = [WKNSString class];
  44     });
  45     return wkNSStringClass;
  46 }
  47 #endif // WK_API_ENABLED
  48
  49 WKStringRef WKStringCreateWithCFString(CFStringRef cfString)
  50 {
  51 #if WK_API_ENABLED
  52     // Since WKNSString is an internal class with no subclasses, we can do a simple equality check.
  53     if (object_getClass((NSString *)cfString) == wkNSStringClass()) {
  54         return toAPI(static_cast<WebString*>(&[(WKNSString *)cfString retain]._apiObject));
  55     }
  56 #endif
  57     String string(cfString);
  58     return toCopiedAPI(string);
  59 }
  60
  61 CFStringRef WKStringCopyCFString(CFAllocatorRef allocatorRef, WKStringRef stringRef)
  62 {
  63     ASSERT(!toImpl(stringRef)->string().isNull());
  64
  65     // NOTE: This does not use StringImpl::createCFString() since that function
  66     // expects to be called on the thread running WebCore.
  67     if (toImpl(stringRef)->string().is8Bit())
  68         return CFStringCreateWithBytes(allocatorRef, reinterpret_cast<const UInt8*>(toImpl(stringRef)->string().characters8()), toImpl(stringRef)->string().length(), kCFStringEncodingISOLatin1, true);
  69     return CFStringCreateWithCharacters(allocatorRef, reinterpret_cast<const UniChar*>(toImpl(stringRef)->string().characters16()), toImpl(stringRef)->string().length());
  70 }