1 /* 2 * Copyright (C) 2009 Apple Inc. All rights reserved. 3 * Copyright (C) 2009 Google Inc. All rights reserved. 4 * 5 * Redistribution and use in source and binary forms, with or without 6 * modification, are permitted provided that the following conditions are 7 * met: 8 * 9 * * Redistributions of source code must retain the above copyright 10 * notice, this list of conditions and the following disclaimer. 11 * * Redistributions in binary form must reproduce the above 12 * copyright notice, this list of conditions and the following disclaimer 13 * in the documentation and/or other materials provided with the 14 * distribution. 15 * * Neither the name of Google Inc. nor the names of its 16 * contributors may be used to endorse or promote products derived from 17 * this software without specific prior written permission. 18 * 19 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS 20 * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT 21 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR 22 * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT 23 * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, 24 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT 25 * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, 26 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY 27 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 28 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE 29 * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 30 */ 31 32 #ifndef SocketStreamHandle_h 33 #define SocketStreamHandle_h 34 35 #include "AuthenticationClient.h" 36 #include "SocketStreamHandleBase.h" 37 #include <wtf/RetainPtr.h> 38 39 namespace WebCore { 40 41 class AuthenticationChallenge; 42 class Credential; 43 class SocketStreamHandleClient; 44 45 class SocketStreamHandle : public RefCounted<SocketStreamHandle>, public SocketStreamHandleBase, public AuthenticationClient { 46 public: create(const KURL & url,SocketStreamHandleClient * client)47 static PassRefPtr<SocketStreamHandle> create(const KURL& url, SocketStreamHandleClient* client) { return adoptRef(new SocketStreamHandle(url, client)); } 48 49 virtual ~SocketStreamHandle(); 50 51 using RefCounted<SocketStreamHandle>::ref; 52 using RefCounted<SocketStreamHandle>::deref; 53 54 private: 55 virtual int platformSend(const char* data, int length); 56 virtual void platformClose(); 57 58 SocketStreamHandle(const KURL&, SocketStreamHandleClient*); 59 void createStreams(); 60 void scheduleStreams(); 61 void chooseProxy(); 62 #ifndef BUILDING_ON_TIGER 63 void chooseProxyFromArray(CFArrayRef); 64 void executePACFileURL(CFURLRef); 65 void removePACRunLoopSource(); 66 RetainPtr<CFRunLoopSourceRef> m_pacRunLoopSource; 67 static void pacExecutionCallback(void* client, CFArrayRef proxyList, CFErrorRef error); 68 static void pacExecutionCallbackMainThread(void*); 69 static CFStringRef copyPACExecutionDescription(void*); 70 #endif 71 shouldUseSSL()72 bool shouldUseSSL() const { return m_url.protocolIs("wss"); } 73 74 void addCONNECTCredentials(CFHTTPMessageRef response); 75 76 static CFStringRef copyCFStreamDescription(void* ); 77 static void readStreamCallback(CFReadStreamRef, CFStreamEventType, void*); 78 static void writeStreamCallback(CFWriteStreamRef, CFStreamEventType, void*); 79 #if PLATFORM(WIN) 80 static void readStreamCallbackMainThread(void*); 81 static void writeStreamCallbackMainThread(void*); 82 #endif 83 void readStreamCallback(CFStreamEventType); 84 void writeStreamCallback(CFStreamEventType); 85 86 // No authentication for streams per se, but proxy may ask for credentials. 87 virtual void receivedCredential(const AuthenticationChallenge&, const Credential&); 88 virtual void receivedRequestToContinueWithoutCredential(const AuthenticationChallenge&); 89 virtual void receivedCancellation(const AuthenticationChallenge&); 90 refAuthenticationClient()91 virtual void refAuthenticationClient() { ref(); } derefAuthenticationClient()92 virtual void derefAuthenticationClient() { deref(); } 93 94 enum ConnectingSubstate { New, ExecutingPACFile, WaitingForCredentials, WaitingForConnect, Connected }; 95 ConnectingSubstate m_connectingSubstate; 96 97 enum ConnectionType { Unknown, Direct, SOCKSProxy, CONNECTProxy }; 98 ConnectionType m_connectionType; 99 RetainPtr<CFStringRef> m_proxyHost; 100 RetainPtr<CFNumberRef> m_proxyPort; 101 102 RetainPtr<CFHTTPMessageRef> m_proxyResponseMessage; 103 bool m_sentStoredCredentials; 104 RetainPtr<CFReadStreamRef> m_readStream; 105 RetainPtr<CFWriteStreamRef> m_writeStream; 106 107 RetainPtr<CFURLRef> m_httpsURL; // ws(s): replaced with https: 108 }; 109 110 } // namespace WebCore 111 112 #endif // SocketStreamHandle_h 113