blob: d88f2f3734129851a2514c20fadb22f907c03030 (
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
|
// Copyright 2014 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 MEDIA_CAST_NET_UDP_TRANSPORT_H_
#define MEDIA_CAST_NET_UDP_TRANSPORT_H_
#include "base/memory/ref_counted.h"
#include "base/memory/scoped_ptr.h"
#include "base/memory/weak_ptr.h"
#include "media/cast/cast_environment.h"
#include "media/cast/net/cast_transport_config.h"
#include "media/cast/net/cast_transport_sender.h"
#include "net/base/ip_endpoint.h"
#include "net/base/net_util.h"
#include "net/udp/udp_socket.h"
namespace net {
class IOBuffer;
class IPEndPoint;
class NetLog;
} // namespace net
namespace media {
namespace cast {
// This class implements UDP transport mechanism for Cast.
class UdpTransport : public PacketSender {
public:
// Construct a UDP transport.
// All methods must be called on |io_thread_proxy|.
// |local_end_point| specifies the address and port to bind and listen
// to incoming packets. If the value is 0.0.0.0:0 then a bind is not
// performed.
// |remote_end_point| specifies the address and port to send packets
// to. If the value is 0.0.0.0:0 the the end point is set to the source
// address of the first packet received.
UdpTransport(
net::NetLog* net_log,
const scoped_refptr<base::SingleThreadTaskRunner>& io_thread_proxy,
const net::IPEndPoint& local_end_point,
const net::IPEndPoint& remote_end_point,
const CastTransportStatusCallback& status_callback);
virtual ~UdpTransport();
// Start receiving packets. Packets are submitted to |packet_receiver|.
void StartReceiving(const PacketReceiverCallback& packet_receiver);
// Set a new DSCP value to the socket. The value will be set right before
// the next send.
void SetDscp(net::DiffServCodePoint dscp);
// PacketSender implementations.
virtual bool SendPacket(PacketRef packet,
const base::Closure& cb) OVERRIDE;
virtual int64 GetBytesSent() OVERRIDE;
private:
// Requests and processes packets from |udp_socket_|. This method is called
// once with |length_or_status| set to net::ERR_IO_PENDING to start receiving
// packets. Thereafter, it is called with some other value as the callback
// response from UdpSocket::RecvFrom().
void ReceiveNextPacket(int length_or_status);
// Schedule packet receiving, if needed.
void ScheduleReceiveNextPacket();
void OnSent(const scoped_refptr<net::IOBuffer>& buf,
PacketRef packet,
const base::Closure& cb,
int result);
const scoped_refptr<base::SingleThreadTaskRunner> io_thread_proxy_;
const net::IPEndPoint local_addr_;
net::IPEndPoint remote_addr_;
const scoped_ptr<net::UDPSocket> udp_socket_;
bool send_pending_;
bool receive_pending_;
bool client_connected_;
net::DiffServCodePoint next_dscp_value_;
scoped_ptr<Packet> next_packet_;
scoped_refptr<net::WrappedIOBuffer> recv_buf_;
net::IPEndPoint recv_addr_;
PacketReceiverCallback packet_receiver_;
const CastTransportStatusCallback status_callback_;
int bytes_sent_;
// NOTE: Weak pointers must be invalidated before all other member variables.
base::WeakPtrFactory<UdpTransport> weak_factory_;
DISALLOW_COPY_AND_ASSIGN(UdpTransport);
};
} // namespace cast
} // namespace media
#endif // MEDIA_CAST_NET_UDP_TRANSPORT_H_
|
