summaryrefslogtreecommitdiffstats
blob: 38f70dbf24c1488c56feb5bce2c23c17662b5798 (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
92
93
94
95
96
97
98
/*
 * Copyright (c) 2010 Remko Tronçon
 * Licensed under the GNU General Public License v3.
 * See Documentation/Licenses/GPLv3.txt for more information.
 */

#include <boost/bind.hpp>
#include <boost/thread.hpp>
#include <iostream>

#include <Swiften/Client/Client.h>
#include <Swiften/Network/TimerFactory.h>
#include <Swiften/Network/BoostNetworkFactories.h>
#include <Swiften/EventLoop/EventLoop.h>
#include <Swiften/EventLoop/SimpleEventLoop.h>
#include <Swiften/Roster/GetRosterRequest.h>
#include <Swiften/Client/ClientXMLTracer.h>

using namespace Swift;

static SimpleEventLoop eventLoop;
static BoostNetworkFactories networkFactories(&eventLoop);

static Client* client = 0;
static bool rosterReceived = false;
enum TestStage {
	FirstConnect,
	Reconnect
};
static TestStage stage;
static ClientOptions options;

static void handleDisconnected(boost::optional<ClientError> e) {
	std::cout << "Disconnected: " << (e ? e.get().getType() : ClientError::UnknownError) << std::endl;
	if (stage == FirstConnect) {
		stage = Reconnect;
		client->connect(options);
	}
	else {
		eventLoop.stop();
	}
}

static void handleRosterReceived(boost::shared_ptr<Payload>) {
	rosterReceived = true;
	std::cout << "Disconnecting" << std::endl;
	client->disconnect();
}

static void handleConnected() {
	std::cout << "Connected" << std::endl;
	rosterReceived = false;
	GetRosterRequest::ref rosterRequest = GetRosterRequest::create(client->getIQRouter());
	rosterRequest->onResponse.connect(boost::bind(&handleRosterReceived, _1));
	rosterRequest->send();
}

int main(int, char**) {
	char* jid = getenv("SWIFT_CLIENTTEST_JID");
	if (!jid) {
		std::cerr << "Please set the SWIFT_CLIENTTEST_JID environment variable" << std::endl;
		return -1;
	}
	char* pass = getenv("SWIFT_CLIENTTEST_PASS");
	if (!pass) {
		std::cerr << "Please set the SWIFT_CLIENTTEST_PASS environment variable" << std::endl;
		return -1;
	}

	char* boshHost = getenv("SWIFT_CLIENTTEST_BOSH_HOST");
	char* boshPort = getenv("SWIFT_CLIENTTEST_BOSH_PORT");
	char* boshPath = getenv("SWIFT_CLIENTTEST_BOSH_PATH");

	if (boshHost && boshPort && boshPath) {
		std::cout << "Using BOSH with URL: http://" << boshHost << ":" << boshPort << "/" << boshPath << std::endl;
		options.boshURL = URL("http", boshHost, atoi(boshPort), boshPath);
	}

	client = new Swift::Client(JID(jid), std::string(pass), &networkFactories);
	ClientXMLTracer* tracer = new ClientXMLTracer(client, !options.boshURL.isEmpty());
	client->onConnected.connect(&handleConnected);
	client->onDisconnected.connect(boost::bind(&handleDisconnected, _1));
	client->setAlwaysTrustCertificates();
	stage = FirstConnect;
	client->connect(options);

	{
		Timer::ref timer = networkFactories.getTimerFactory()->createTimer(60000);
		timer->onTick.connect(boost::bind(&SimpleEventLoop::stop, &eventLoop));
		timer->start();

		eventLoop.run();
	}

	delete tracer;
	delete client;
	return !rosterReceived;
}