De-Clientize UDPSocket service

mojo/services/network/udp_socket_impl.h

 // 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 MOJO_SERVICES_NETWORK_UDP_SOCKET_IMPL_H_
 #define MOJO_SERVICES_NETWORK_UDP_SOCKET_IMPL_H_
 
 #include <deque>
 
 #include "base/macros.h"
 #include "base/memory/ref_counted.h"
 #include "mojo/services/network/public/interfaces/udp_socket.mojom.h"
 #include "net/base/ip_endpoint.h"
 #include "net/udp/udp_socket.h"
 #include "third_party/mojo/src/mojo/public/cpp/bindings/interface_impl.h"
+#include "third_party/mojo/src/mojo/public/cpp/bindings/strong_binding.h"
 
 namespace net {
 class IOBuffer;
 class IOBufferWithSize;
 }
 
 namespace mojo {
 
-class UDPSocketImpl : public InterfaceImpl<UDPSocket> {
+class UDPSocketImpl : public UDPSocket {
  public:
-  UDPSocketImpl();
+  // The lifetime of a new UDPSocketImpl is bound to the connection associated
+  // with |request|.
+  explicit UDPSocketImpl(InterfaceRequest<UDPSocket> request);
   ~UDPSocketImpl() override;
 
   // UDPSocket implementation.
   void AllowAddressReuse(
       const Callback<void(NetworkErrorPtr)>& callback) override;
 
   void Bind(
       NetAddressPtr addr,
       const Callback<void(NetworkErrorPtr, NetAddressPtr)>& callback) override;
 
   void Connect(
       NetAddressPtr remote_addr,
       const Callback<void(NetworkErrorPtr, NetAddressPtr)>& callback) override;
 
   void SetSendBufferSize(
       uint32_t size,
       const Callback<void(NetworkErrorPtr)>& callback) override;
 
   void SetReceiveBufferSize(
       uint32_t size,
       const Callback<void(NetworkErrorPtr)>& callback) override;
 
   void NegotiateMaxPendingSendRequests(
       uint32_t requested_size,
       const Callback<void(uint32_t)>& callback) override;
 
+  void SetReceiver(UDPSocketReceiverPtr receiver) override;
+
   void ReceiveMore(uint32_t datagram_number) override;
 
   void SendTo(NetAddressPtr dest_addr,
               Array<uint8_t> data,
               const Callback<void(NetworkErrorPtr)>& callback) override;
 
  private:
   enum State {
     NOT_BOUND_OR_CONNECTED,
     BOUND,
@@ skipping 15 matching lines @@
                 const Callback<void(NetworkErrorPtr)>& callback);
 
   void OnRecvFromCompleted(int net_result);
   void OnSendToCompleted(const Callback<void(NetworkErrorPtr)>& callback,
                          int net_result);
 
   bool IsBoundOrConnected() const {
     return state_ == BOUND || state_ == CONNECTED;
   }
 
+  StrongBinding<UDPSocket> binding_;
+
   net::UDPSocket socket_;
 
   State state_;
 
   bool allow_address_reuse_;
 
   // Non-null when there is a pending RecvFrom operation on |socket_|.
   scoped_refptr<net::IOBuffer> recvfrom_buffer_;
   // Non-null when there is a pending SendTo operation on |socket_|.
   scoped_refptr<net::IOBufferWithSize> sendto_buffer_;
 
+  // The address of the pending RecvFrom operation, if any.
   net::IPEndPoint recvfrom_address_;
 
+  // The result code the pending RecvFrom operation, if any.
+  int recvfrom_result_;
+
+  // The interface which gets data from fulfilled receive requests.
+  UDPSocketReceiverPtr receiver_;
+
   // How many more packets the client side expects to receive.
   size_t remaining_recv_slots_;
 
   // The queue owns the PendingSendRequest instances.
   std::deque<PendingSendRequest*> pending_send_requests_;
   // The maximum size of the |pending_send_requests_| queue.
   size_t max_pending_send_requests_;
 
   DISALLOW_COPY_AND_ASSIGN(UDPSocketImpl);
 };
 
 }  // namespace mojo
 
 #endif  // MOJO_SERVICES_NETWORK_UDP_SOCKET_IMPL_H_