blob: f48eb2bfd68323169c0b5e516c3069e3f5dc249a (
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
|
#include <boost/bind.hpp>
#include <boost/thread.hpp>
#include "Swiften/Client/Client.h"
#include "Swiften/Network/BoostTimer.h"
#include "Swiften/EventLoop/MainEventLoop.h"
#include "Swiften/Client/ClientXMLTracer.h"
#include "Swiften/EventLoop/SimpleEventLoop.h"
#include "Swiften/Network/BoostIOServiceThread.h"
#include "Swiften/Network/MainBoostIOServiceThread.h"
#include "Swiften/Queries/Requests/GetDiscoInfoRequest.h"
using namespace Swift;
enum ExitCodes {OK = 0, CANNOT_CONNECT, CANNOT_AUTH, NO_RESPONSE, DISCO_ERROR};
SimpleEventLoop eventLoop;
Client* client = 0;
JID recipient;
int exitCode = CANNOT_CONNECT;
boost::bsignals::connection errorConnection;
void handleServerDiscoInfoResponse(boost::shared_ptr<DiscoInfo> /*info*/, const boost::optional<ErrorPayload>& error) {
if (!error) {
errorConnection.disconnect();
client->disconnect();
eventLoop.stop();
exitCode = OK;
} else {
errorConnection.disconnect();
exitCode = DISCO_ERROR;
}
}
void handleConnected() {
exitCode = NO_RESPONSE;
boost::shared_ptr<GetDiscoInfoRequest> discoInfoRequest(new GetDiscoInfoRequest(JID(), client));
discoInfoRequest->onResponse.connect(handleServerDiscoInfoResponse);
discoInfoRequest->send();
}
void handleError(const ClientError&) {
exitCode = CANNOT_AUTH;
eventLoop.stop();
}
int main(int argc, char* argv[]) {
if (argc < 4 || argc > 5) {
std::cerr << "Usage: " << argv[0] << " <jid> [<connect_host>] <password> <timeout_seconds>" << std::endl;
return -1;
}
int argi = 1;
String jid = argv[argi++];
String connectHost = "";
if (argc == 5) {
connectHost = argv[argi++];
}
client = new Swift::Client(JID(jid), String(argv[argi++]));
char* timeoutChar = argv[argi++];
int timeout = atoi(timeoutChar);
timeout = (timeout ? timeout : 30) * 1000;
ClientXMLTracer* tracer = new ClientXMLTracer(client);
client->onConnected.connect(&handleConnected);
errorConnection = client->onError.connect(&handleError);
std::cout << "Connecting to JID " << jid << " with timeout " << timeout << "ms on host: "; ;
if (!connectHost.isEmpty()) {
std::cout << connectHost << std::endl;
client->connect(connectHost);
} else {
std::cout << " Default" << std::endl;
client->connect();
}
{
boost::shared_ptr<BoostTimer> timer(new BoostTimer(timeout, &MainBoostIOServiceThread::getInstance().getIOService()));
timer->onTick.connect(boost::bind(&SimpleEventLoop::stop, &eventLoop));
timer->start();
eventLoop.run();
}
delete tracer;
delete client;
return exitCode;
}
|
Swift