1// Copyright 2014 The Chromium Authors. All rights reserved.
2// Use of this source code is governed by a BSD-style license that can be
3// found in the LICENSE file.
4
5#include "remoting/signaling/jingle_info_request.h"
6
7#include "base/bind.h"
8#include "base/message_loop/message_loop.h"
9#include "base/stl_util.h"
10#include "base/strings/string_number_conversions.h"
11#include "base/time/time.h"
12#include "net/base/net_util.h"
13#include "remoting/signaling/iq_sender.h"
14#include "third_party/libjingle/source/talk/xmpp/constants.h"
15#include "third_party/webrtc/base/socketaddress.h"
16#include "third_party/webrtc/libjingle/xmllite/xmlelement.h"
17
18namespace remoting {
19
20const int kRequestTimeoutSeconds = 5;
21
22JingleInfoRequest::JingleInfoRequest(SignalStrategy* signal_strategy)
23    : iq_sender_(signal_strategy) {
24}
25
26JingleInfoRequest::~JingleInfoRequest() {}
27
28void JingleInfoRequest::Send(const OnJingleInfoCallback& callback) {
29  on_jingle_info_cb_ = callback;
30  scoped_ptr<buzz::XmlElement> iq_body(
31      new buzz::XmlElement(buzz::QN_JINGLE_INFO_QUERY, true));
32  request_ = iq_sender_.SendIq(
33      buzz::STR_GET, buzz::STR_EMPTY, iq_body.Pass(),
34      base::Bind(&JingleInfoRequest::OnResponse, base::Unretained(this)));
35  if (!request_) {
36    // If we failed to send IqRequest it means that SignalStrategy is
37    // disconnected. Notify the caller.
38    std::vector<rtc::SocketAddress> stun_hosts;
39    std::vector<std::string> relay_hosts;
40    std::string relay_token;
41    on_jingle_info_cb_.Run(relay_token, relay_hosts, stun_hosts);
42    return;
43  }
44  request_->SetTimeout(base::TimeDelta::FromSeconds(kRequestTimeoutSeconds));
45}
46
47void JingleInfoRequest::OnResponse(IqRequest* request,
48                                   const buzz::XmlElement* stanza) {
49  std::vector<rtc::SocketAddress> stun_hosts;
50  std::vector<std::string> relay_hosts;
51  std::string relay_token;
52
53  if (!stanza) {
54    LOG(WARNING) << "Jingle info request has timed out.";
55    on_jingle_info_cb_.Run(relay_token, relay_hosts, stun_hosts);
56    return;
57  }
58
59  const buzz::XmlElement* query =
60      stanza->FirstNamed(buzz::QN_JINGLE_INFO_QUERY);
61  if (query == NULL) {
62    LOG(WARNING) << "No Jingle info found in Jingle Info query response."
63                 << stanza->Str();
64    on_jingle_info_cb_.Run(relay_token, relay_hosts, stun_hosts);
65    return;
66  }
67
68  const buzz::XmlElement* stun = query->FirstNamed(buzz::QN_JINGLE_INFO_STUN);
69  if (stun) {
70    for (const buzz::XmlElement* server =
71         stun->FirstNamed(buzz::QN_JINGLE_INFO_SERVER);
72         server != NULL;
73         server = server->NextNamed(buzz::QN_JINGLE_INFO_SERVER)) {
74      std::string host = server->Attr(buzz::QN_JINGLE_INFO_HOST);
75      std::string port_str = server->Attr(buzz::QN_JINGLE_INFO_UDP);
76      if (host != buzz::STR_EMPTY && port_str != buzz::STR_EMPTY) {
77        int port;
78        if (!base::StringToInt(port_str, &port)) {
79          LOG(WARNING) << "Unable to parse port in stanza" << stanza->Str();
80          continue;
81        }
82
83        stun_hosts.push_back(rtc::SocketAddress(host, port));
84      }
85    }
86  }
87
88  const buzz::XmlElement* relay = query->FirstNamed(buzz::QN_JINGLE_INFO_RELAY);
89  if (relay) {
90    relay_token = relay->TextNamed(buzz::QN_JINGLE_INFO_TOKEN);
91    for (const buzz::XmlElement* server =
92         relay->FirstNamed(buzz::QN_JINGLE_INFO_SERVER);
93         server != NULL;
94         server = server->NextNamed(buzz::QN_JINGLE_INFO_SERVER)) {
95      std::string host = server->Attr(buzz::QN_JINGLE_INFO_HOST);
96      if (host != buzz::STR_EMPTY)
97        relay_hosts.push_back(host);
98    }
99  }
100
101  on_jingle_info_cb_.Run(relay_token, relay_hosts, stun_hosts);
102}
103
104}  // namespace remoting
105