1/*
2 * Copyright (C) 2011 Google 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 are
6 * met:
7 *
8 * * Redistributions of source code must retain the above copyright
9 * notice, this list of conditions and the following disclaimer.
10 * * Redistributions in binary form must reproduce the above
11 * copyright notice, this list of conditions and the following disclaimer
12 * in the documentation and/or other materials provided with the
13 * distribution.
14 * * Neither the name of Google Inc. nor the names of its
15 * contributors may be used to endorse or promote products derived from
16 * this software without specific prior written permission.
17 *
18 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
19 * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
20 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
21 * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
22 * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
23 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
24 * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
25 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
26 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
27 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
28 * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
29 */
30
31#pragma once
32
33#include "ScriptExecutionContext.h"
34#include "ThreadableWebSocketChannel.h"
35#include "WebSocketChannelClient.h"
36#include "WorkerThreadableWebSocketChannel.h"
37#include <memory>
38#include <wtf/Forward.h>
39#include <wtf/Vector.h>
40#include <wtf/text/WTFString.h>
41
42namespace WebCore {
43
44class ScriptExecutionContext;
45class WebSocketChannelClient;
46
47class ThreadableWebSocketChannelClientWrapper : public ThreadSafeRefCounted<ThreadableWebSocketChannelClientWrapper> {
48public:
49 static Ref<ThreadableWebSocketChannelClientWrapper> create(ScriptExecutionContext&, WebSocketChannelClient&);
50
51 void clearSyncMethodDone();
52 void setSyncMethodDone();
53 bool syncMethodDone() const;
54
55 WorkerThreadableWebSocketChannel::Peer* peer() const;
56 void didCreateWebSocketChannel(WorkerThreadableWebSocketChannel::Peer*);
57 void clearPeer();
58
59 bool failedWebSocketChannelCreation() const;
60 void setFailedWebSocketChannelCreation();
61
62 // Subprotocol and extensions will be available when didConnect() callback is invoked.
63 String subprotocol() const;
64 void setSubprotocol(const String&);
65 String extensions() const;
66 void setExtensions(const String&);
67
68 ThreadableWebSocketChannel::SendResult sendRequestResult() const;
69 void setSendRequestResult(ThreadableWebSocketChannel::SendResult);
70
71 unsigned bufferedAmount() const;
72 void setBufferedAmount(unsigned);
73
74 void clearClient();
75
76 void didConnect();
77 void didReceiveMessage(const String& message);
78 void didReceiveBinaryData(Vector<uint8_t>&&);
79 void didUpdateBufferedAmount(unsigned bufferedAmount);
80 void didStartClosingHandshake();
81 void didClose(unsigned unhandledBufferedAmount, WebSocketChannelClient::ClosingHandshakeCompletionStatus, unsigned short code, const String& reason);
82 void didReceiveMessageError();
83 void didUpgradeURL();
84
85 void suspend();
86 void resume();
87
88private:
89 ThreadableWebSocketChannelClientWrapper(ScriptExecutionContext&, WebSocketChannelClient&);
90
91 void processPendingTasks();
92
93 ScriptExecutionContext& m_context;
94 WebSocketChannelClient* m_client;
95 WorkerThreadableWebSocketChannel::Peer* m_peer;
96 bool m_failedWebSocketChannelCreation;
97 bool m_syncMethodDone;
98 // ThreadSafeRefCounted must not have String member variables.
99 Vector<UChar> m_subprotocol;
100 Vector<UChar> m_extensions;
101 ThreadableWebSocketChannel::SendResult m_sendRequestResult;
102 unsigned m_bufferedAmount;
103 bool m_suspended;
104 Vector<std::unique_ptr<ScriptExecutionContext::Task>> m_pendingTasks;
105};
106
107} // namespace WebCore
108