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
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
|
// Copyright (c) 2011 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.
#include "remoting/jingle_glue/iq_sender.h"
#include "base/logging.h"
#include "base/string_number_conversions.h"
#include "remoting/jingle_glue/signal_strategy.h"
#include "third_party/libjingle/source/talk/xmllite/xmlelement.h"
#include "third_party/libjingle/source/talk/xmpp/constants.h"
namespace remoting {
// static
buzz::XmlElement* IqSender::MakeIqStanza(const std::string& type,
const std::string& addressee,
buzz::XmlElement* iq_body) {
buzz::XmlElement* stanza = new buzz::XmlElement(buzz::QN_IQ);
stanza->AddAttr(buzz::QN_TYPE, type);
if (!addressee.empty())
stanza->AddAttr(buzz::QN_TO, addressee);
stanza->AddElement(iq_body);
return stanza;
}
IqSender::IqSender(SignalStrategy* signal_strategy)
: signal_strategy_(signal_strategy) {
signal_strategy_->AddListener(this);
}
IqSender::~IqSender() {
signal_strategy_->RemoveListener(this);
}
IqRequest* IqSender::SendIq(buzz::XmlElement* stanza,
const ReplyCallback& callback) {
std::string id = signal_strategy_->GetNextId();
stanza->AddAttr(buzz::QN_ID, id);
if (!signal_strategy_->SendStanza(stanza)) {
return NULL;
}
DCHECK(requests_.find(id) == requests_.end());
IqRequest* request = new IqRequest(this, callback);
if (!callback.is_null())
requests_[id] = request;
return request;
}
IqRequest* IqSender::SendIq(const std::string& type,
const std::string& addressee,
buzz::XmlElement* iq_body,
const ReplyCallback& callback) {
return SendIq(MakeIqStanza(type, addressee, iq_body), callback);
}
void IqSender::RemoveRequest(IqRequest* request) {
IqRequestMap::iterator it = requests_.begin();
while (it != requests_.end()) {
IqRequestMap::iterator cur = it;
++it;
if (cur->second == request) {
requests_.erase(cur);
break;
}
}
}
bool IqSender::OnIncomingStanza(const buzz::XmlElement* stanza) {
if (stanza->Name() != buzz::QN_IQ) {
LOG(WARNING) << "Received unexpected non-IQ packet " << stanza->Str();
return false;
}
const std::string& type = stanza->Attr(buzz::QN_TYPE);
if (type.empty()) {
LOG(WARNING) << "IQ packet missing type " << stanza->Str();
return false;
}
if (type != "result" && type != "error") {
return false;
}
const std::string& id = stanza->Attr(buzz::QN_ID);
if (id.empty()) {
LOG(WARNING) << "IQ packet missing id " << stanza->Str();
return false;
}
IqRequestMap::iterator it = requests_.find(id);
if (it == requests_.end()) {
return false;
}
IqRequest* request = it->second;
requests_.erase(it);
request->OnResponse(stanza);
return true;
}
IqRequest::IqRequest(IqSender* sender, const IqSender::ReplyCallback& callback)
: sender_(sender),
callback_(callback) {
}
IqRequest::~IqRequest() {
sender_->RemoveRequest(this);
}
void IqRequest::OnResponse(const buzz::XmlElement* stanza) {
DCHECK(!callback_.is_null());
IqSender::ReplyCallback callback(callback_);
callback_.Reset();
callback.Run(stanza);
}
} // namespace remoting
|
