| // 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 REMOTING_SIGNALING_IQ_SENDER_H_ |
| #define REMOTING_SIGNALING_IQ_SENDER_H_ |
| |
| #include <map> |
| #include <memory> |
| #include <string> |
| |
| #include "base/callback.h" |
| #include "base/compiler_specific.h" |
| #include "base/macros.h" |
| #include "base/memory/weak_ptr.h" |
| #include "remoting/signaling/signal_strategy.h" |
| |
| namespace base { |
| class TimeDelta; |
| } // namespace base |
| |
| namespace jingle_xmpp { |
| class XmlElement; |
| } // namespace jingle_xmpp |
| |
| namespace remoting { |
| |
| class IqRequest; |
| class SignalStrategy; |
| |
| // IqSender handles sending iq requests and routing of responses to |
| // those requests. |
| class IqSender : public SignalStrategy::Listener { |
| public: |
| // Callback that is called when an Iq response is received. Called |
| // with the |response| set to nullptr in case of a timeout. |
| using ReplyCallback = |
| base::OnceCallback<void(IqRequest* request, |
| const jingle_xmpp::XmlElement* response)>; |
| |
| explicit IqSender(SignalStrategy* signal_strategy); |
| ~IqSender() override; |
| |
| // Send an iq stanza. Returns an IqRequest object that represends |
| // the request. |callback| is called when response to |stanza| is |
| // received. Destroy the returned IqRequest to cancel the callback. |
| // Caller must take ownership of the result. Result must be |
| // destroyed before sender is destroyed. |
| std::unique_ptr<IqRequest> SendIq( |
| std::unique_ptr<jingle_xmpp::XmlElement> stanza, |
| ReplyCallback callback); |
| |
| // Same as above, but also formats the message. |
| std::unique_ptr<IqRequest> SendIq( |
| const std::string& type, |
| const std::string& addressee, |
| std::unique_ptr<jingle_xmpp::XmlElement> iq_body, |
| ReplyCallback callback); |
| |
| // SignalStrategy::Listener implementation. |
| void OnSignalStrategyStateChange(SignalStrategy::State state) override; |
| bool OnSignalStrategyIncomingStanza(const jingle_xmpp::XmlElement* stanza) override; |
| |
| private: |
| typedef std::map<std::string, IqRequest*> IqRequestMap; |
| friend class IqRequest; |
| |
| // Helper function used to create iq stanzas. |
| static std::unique_ptr<jingle_xmpp::XmlElement> MakeIqStanza( |
| const std::string& type, |
| const std::string& addressee, |
| std::unique_ptr<jingle_xmpp::XmlElement> iq_body); |
| |
| // Removes |request| from the list of pending requests. Called by IqRequest. |
| void RemoveRequest(IqRequest* request); |
| |
| SignalStrategy* signal_strategy_; |
| IqRequestMap requests_; |
| |
| DISALLOW_COPY_AND_ASSIGN(IqSender); |
| }; |
| |
| // This call must only be used on the thread it was created on. |
| class IqRequest { |
| public: |
| IqRequest(IqSender* sender, |
| IqSender::ReplyCallback callback, |
| const std::string& addressee); |
| ~IqRequest(); |
| |
| // Sets timeout for the request. When the timeout expires the |
| // callback is called with the |response| set to nullptr. |
| void SetTimeout(base::TimeDelta timeout); |
| |
| private: |
| friend class IqSender; |
| |
| void CallCallback(const jingle_xmpp::XmlElement* stanza); |
| void OnTimeout(); |
| |
| // Called by IqSender when a response is received. |
| void OnResponse(const jingle_xmpp::XmlElement* stanza); |
| |
| void DeliverResponse(std::unique_ptr<jingle_xmpp::XmlElement> stanza); |
| |
| IqSender* sender_; |
| IqSender::ReplyCallback callback_; |
| std::string addressee_; |
| |
| base::WeakPtrFactory<IqRequest> weak_factory_{this}; |
| |
| DISALLOW_COPY_AND_ASSIGN(IqRequest); |
| }; |
| |
| } // namespace remoting |
| |
| #endif // REMOTING_SIGNALING_IQ_SENDER_H_ |