blob: 169d1e481a87111b1d08782904c25c6c8b12af09 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
|
// Copyright (c) 2012 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#ifndef NET_SPDY_SPDY_WEBSOCKET_STREAM_H_
#define NET_SPDY_SPDY_WEBSOCKET_STREAM_H_
#include "base/basictypes.h"
#include "base/gtest_prod_util.h"
#include "base/memory/ref_counted.h"
#include "base/memory/scoped_ptr.h"
#include "base/memory/weak_ptr.h"
#include "base/time/time.h"
#include "net/base/completion_callback.h"
#include "net/base/request_priority.h"
#include "net/spdy/spdy_framer.h"
#include "net/spdy/spdy_header_block.h"
#include "net/spdy/spdy_session.h"
#include "net/spdy/spdy_stream.h"
namespace net {
// The SpdyWebSocketStream is a WebSocket-specific type of stream known to a
// SpdySession. WebSocket's opening handshake is converted to SPDY's
// SYN_STREAM/SYN_REPLY. WebSocket frames are encapsulated as SPDY data frames.
class NET_EXPORT_PRIVATE SpdyWebSocketStream
: public SpdyStream::Delegate {
public:
// Delegate handles asynchronous events.
class NET_EXPORT_PRIVATE Delegate {
public:
// Called when InitializeStream() finishes asynchronously. This delegate is
// called if InitializeStream() returns ERR_IO_PENDING. |status| indicates
// network error.
virtual void OnCreatedSpdyStream(int status) = 0;
// Called on corresponding to OnSendHeadersComplete() or SPDY's SYN frame
// has been sent.
virtual void OnSentSpdyHeaders() = 0;
// Called on corresponding to OnResponseHeadersUpdated() or
// SPDY's SYN_STREAM, SYN_REPLY, or HEADERS frames are
// received. This callback may be called multiple times as SPDY's
// delegate does.
virtual void OnSpdyResponseHeadersUpdated(
const SpdyHeaderBlock& response_headers) = 0;
// Called when data is sent.
virtual void OnSentSpdyData(size_t bytes_sent) = 0;
// Called when data is received.
virtual void OnReceivedSpdyData(scoped_ptr<SpdyBuffer> buffer) = 0;
// Called when SpdyStream is closed.
virtual void OnCloseSpdyStream() = 0;
protected:
virtual ~Delegate() {}
};
SpdyWebSocketStream(const base::WeakPtr<SpdySession>& spdy_session,
Delegate* delegate);
virtual ~SpdyWebSocketStream();
// Initializes SPDY stream for the WebSocket.
// It might create SPDY stream asynchronously. In this case, this method
// returns ERR_IO_PENDING and call OnCreatedSpdyStream delegate with result
// after completion. In other cases, delegate does not be called.
int InitializeStream(const GURL& url,
RequestPriority request_priority,
const BoundNetLog& stream_net_log);
int SendRequest(scoped_ptr<SpdyHeaderBlock> headers);
int SendData(const char* data, int length);
void Close();
// SpdyStream::Delegate
virtual void OnRequestHeadersSent() OVERRIDE;
virtual SpdyResponseHeadersStatus OnResponseHeadersUpdated(
const SpdyHeaderBlock& response_headers) OVERRIDE;
virtual void OnDataReceived(scoped_ptr<SpdyBuffer> buffer) OVERRIDE;
virtual void OnDataSent() OVERRIDE;
virtual void OnClose(int status) OVERRIDE;
private:
friend class SpdyWebSocketStreamTest;
FRIEND_TEST_ALL_PREFIXES(SpdyWebSocketStreamTest, Basic);
void OnSpdyStreamCreated(int status);
base::WeakPtrFactory<SpdyWebSocketStream> weak_ptr_factory_;
SpdyStreamRequest stream_request_;
base::WeakPtr<SpdyStream> stream_;
const base::WeakPtr<SpdySession> spdy_session_;
size_t pending_send_data_length_;
Delegate* delegate_;
DISALLOW_COPY_AND_ASSIGN(SpdyWebSocketStream);
};
} // namespace net
#endif // NET_SPDY_SPDY_WEBSOCKET_STREAM_H_
|
