-#include <lualib.h>

- lib_env = env.Clone()

- lib_env.UseFlags(env["LUA_FLAGS"])

- lib_env.UseFlags(env["SWIFTEN_FLAGS"])

- lib_env.UseFlags(env["SWIFTEN_DEP_FLAGS"])

- sluift_lib = lib_env.StaticLibrary("SluiftCore", [

- "Lua/Value.cpp",

- "sluift.cpp"

- ]);

- myenv.Append(LIBS = sluift_lib)

- if env["PLATFORM"] == "win32" :

- myenv.Append(CFLAGS = ["/TP"])

- else :

- myenv.Append(CFLAGS = ["-x", "c++"])

- env["SLUIFT"] = myenv.Program("sluift", [

- myenv.SharedLibrary("sluift", ["dll.c"])

-/*

- * Copyright (c) 2011 Remko Tronçon

- * Licensed under the GNU General Public License v3.

- * See Documentation/Licenses/GPLv3.txt for more information.

- */

-

-#pragma once

-

-#include <string>

-

-#include <Swiften/Client/ClientError.h>

-

-namespace Swift {

- class SluiftException : public std::exception {

- public:

- virtual ~SluiftException() throw() {}

-

- SluiftException(const std::string& reason) : reason(reason) {

- }

-

- SluiftException(const ClientError& error) {

- reason = "Disconnected: ";

- switch(error.getType()) {

- case ClientError::UnknownError: reason += "Unknown Error"; break;

- case ClientError::DomainNameResolveError: reason += "Unable to find server"; break;

- case ClientError::ConnectionError: reason += "Error connecting to server"; break;

- case ClientError::ConnectionReadError: reason += "Error while receiving server data"; break;

- case ClientError::ConnectionWriteError: reason += "Error while sending data to the server"; break;

- case ClientError::XMLError: reason += "Error parsing server data"; break;

- case ClientError::AuthenticationFailedError: reason += "Login/password invalid"; break;

- case ClientError::CompressionFailedError: reason += "Error while compressing stream"; break;

- case ClientError::ServerVerificationFailedError: reason += "Server verification failed"; break;

- case ClientError::NoSupportedAuthMechanismsError: reason += "Authentication mechanisms not supported"; break;

- case ClientError::UnexpectedElementError: reason += "Unexpected response"; break;

- case ClientError::ResourceBindError: reason += "Error binding resource"; break;

- case ClientError::RevokedError: reason += "Certificate got revoked"; break;

- case ClientError::RevocationCheckFailedError: reason += "Failed to do revokation check"; break;

- case ClientError::SessionStartError: reason += "Error starting session"; break;

- case ClientError::StreamError: reason += "Stream error"; break;

- case ClientError::TLSError: reason += "Encryption error"; break;

- case ClientError::ClientCertificateLoadError: reason += "Error loading certificate (Invalid password?)"; break;

- case ClientError::ClientCertificateError: reason += "Certificate not authorized"; break;

- case ClientError::UnknownCertificateError: reason += "Unknown certificate"; break;

- case ClientError::CertificateCardRemoved: reason += "Certificate card removed"; break;

- case ClientError::CertificateExpiredError: reason += "Certificate has expired"; break;

- case ClientError::CertificateNotYetValidError: reason += "Certificate is not yet valid"; break;

- case ClientError::CertificateSelfSignedError: reason += "Certificate is self-signed"; break;

- case ClientError::CertificateRejectedError: reason += "Certificate has been rejected"; break;

- case ClientError::CertificateUntrustedError: reason += "Certificate is not trusted"; break;

- case ClientError::InvalidCertificatePurposeError: reason += "Certificate cannot be used for encrypting your connection"; break;

- case ClientError::CertificatePathLengthExceededError: reason += "Certificate path length constraint exceeded"; break;

- case ClientError::InvalidCertificateSignatureError: reason += "Invalid certificate signature"; break;

- case ClientError::InvalidCAError: reason += "Invalid Certificate Authority"; break;

- case ClientError::InvalidServerIdentityError: reason += "Certificate does not match the host identity"; break;

- }

- }

-

- const std::string& getReason() const {

- return reason;

- }

-

- virtual const char* what() const throw() {

- return getReason().c_str();

- }

-

-

- private:

- std::string reason;

- };

-}

- * Copyright (c) 2011 Remko Tronçon

-#include "SluiftException.h"

- SluiftClient(const JID& jid, const std::string& password) : tracer(NULL) {

- throw SluiftException("Timeout while connecting");

- throw SluiftException(*error);

- try {

- SluiftClient* client = getClient(L);

- std::string host;

- if (lua_type(L, 2) != LUA_TNONE) {

- host = luaL_checkstring(L, 2);

- }

- if (host.empty()) {

- client->connect();

- }

- else {

- client->connect(host);

- }

- client->waitConnected();

- return 1;

- catch (const SluiftException& e) {

- return luaL_error(L, e.getReason().c_str());

- try {

- getClient(L)->connect();

- return 1;

- }

- catch (const SluiftException& e) {

- return luaL_error(L, e.getReason().c_str());

- }

- try {

- getClient(L)->waitConnected();

- return 1;

- }

- catch (const SluiftException& e) {

- return luaL_error(L, e.getReason().c_str());

- }

- try {

- getClient(L)->disconnect();

- return 1;

- }

- catch (const SluiftException& e) {

- return luaL_error(L, e.getReason().c_str());

- }

- try {

- eventLoop.runOnce();

- SluiftClient* client = getClient(L);

- luaL_checktype(L, 2, LUA_TTABLE);

- lua_getfield(L, 2, "name");

- const char* rawName = lua_tostring(L, -1);

- lua_getfield(L, 2, "version");

- const char* rawVersion = lua_tostring(L, -1);

- lua_getfield(L, 2, "os");

- const char* rawOS = lua_tostring(L, -1);

- client->setSoftwareVersion(rawName ? rawName : "", rawVersion ? rawVersion : "", rawOS ? rawOS : "");

- lua_pop(L, 3);

- lua_pushvalue(L, 1);

- return 1;

- }

- catch (const SluiftException& e) {

- return luaL_error(L, e.getReason().c_str());

- }

- try {

- eventLoop.runOnce();

- SluiftClient* client = getClient(L);

- Lua::Table contactsTable;

- foreach(const XMPPRosterItem& item, client->getRoster()) {

- std::string subscription;

- switch(item.getSubscription()) {

- case RosterItemPayload::None: subscription = "none"; break;

- case RosterItemPayload::To: subscription = "to"; break;

- case RosterItemPayload::From: subscription = "from"; break;

- case RosterItemPayload::Both: subscription = "both"; break;

- case RosterItemPayload::Remove: subscription = "remove"; break;

- }

- Lua::Value groups(std::vector<Lua::Value>(item.getGroups().begin(), item.getGroups().end()));

- Lua::Table itemTable = boost::assign::map_list_of

- ("jid", boost::make_shared<Lua::Value>(item.getJID().toString()))

- ("name", boost::make_shared<Lua::Value>(item.getName()))

- ("subscription", boost::make_shared<Lua::Value>(subscription))

- ("groups", boost::make_shared<Lua::Value>(std::vector<Lua::Value>(item.getGroups().begin(), item.getGroups().end())));

- contactsTable[item.getJID().toString()] = boost::make_shared<Lua::Value>(itemTable);

- }

- pushValue(L, contactsTable);

- return 1;

- }

- catch (const SluiftException& e) {

- return luaL_error(L, e.getReason().c_str());

- }

- try {

- SluiftClient* client = getClient(L);

- int timeout = -1;

- if (lua_type(L, 3) != LUA_TNONE) {

- timeout = boost::numeric_cast<int>(luaL_checknumber(L, 3));

- }

- ResponseSink<SoftwareVersion> sink;

- GetSoftwareVersionRequest::ref request = GetSoftwareVersionRequest::create(std::string(luaL_checkstring(L, 2)), client->getClient()->getIQRouter());

- boost::signals::scoped_connection c = request->onResponse.connect(boost::ref(sink));

- request->send();

- Watchdog watchdog(timeout, networkFactories.getTimerFactory());

- while (!watchdog.getTimedOut() && !sink.hasResponse()) {

- eventLoop.runUntilEvents();

- }

- ErrorPayload::ref error = sink.getResponseError();

- if (error || watchdog.getTimedOut()) {

- lua_pushnil(L);

- if (watchdog.getTimedOut()) {

- lua_pushstring(L, "Timeout");

- }

- else if (error->getCondition() == ErrorPayload::RemoteServerNotFound) {

- lua_pushstring(L, "Remote server not found");

- }

- // TODO

- else {

- lua_pushstring(L, "Error");

- }

- return 2;

- else if (SoftwareVersion::ref version = sink.getResponsePayload()) {

- Lua::Table result = boost::assign::map_list_of

- ("name", boost::make_shared<Lua::Value>(version->getName()))

- ("version", boost::make_shared<Lua::Value>(version->getVersion()))

- ("os", boost::make_shared<Lua::Value>(version->getOS()));

- Lua::pushValue(L, result);

- lua_pushnil(L);

- return 1;

- catch (const SluiftException& e) {

- return luaL_error(L, e.getReason().c_str());

- try {

- eventLoop.runOnce();

- getClient(L)->sendMessage(std::string(luaL_checkstring(L, 2)), luaL_checkstring(L, 3));

- lua_pushvalue(L, 1);

- return 1;

- }

- catch (const SluiftException& e) {

- return luaL_error(L, e.getReason().c_str());

- }

- try {

- eventLoop.runOnce();

- getClient(L)->sendPresence(std::string(luaL_checkstring(L, 2)));

- lua_pushvalue(L, 1);

- return 0;

- }

- catch (const SluiftException& e) {

- return luaL_error(L, e.getReason().c_str());

- }

- try {

- SluiftClient* client = getClient(L);

- JID jid;

- std::string data;

- int timeout = -1;

- if (lua_type(L, 3) == LUA_TSTRING) {

- jid = JID(std::string(luaL_checkstring(L, 2)));

- data = std::string(luaL_checkstring(L, 3));

- if (lua_type(L, 4) != LUA_TNONE) {

- timeout = boost::numeric_cast<int>(luaL_checknumber(L, 4));

- }

- }

- else {

- data = std::string(luaL_checkstring(L, 2));

- if (lua_type(L, 3) != LUA_TNONE) {

- timeout = boost::numeric_cast<int>(luaL_checknumber(L, 3));

- }

- }

- boost::optional<std::string> result = client->sendQuery(jid, IQ::Get, data, timeout);

- if (result) {

- lua_pushstring(L, result->c_str());

- else {

- lua_pushnil(L);

- return 1;

- catch (const SluiftException& e) {

- return luaL_error(L, e.getReason().c_str());

- try {

- SluiftClient* client = getClient(L);

- JID jid;

- std::string data;

- int timeout = -1;

- if (lua_type(L, 3) == LUA_TSTRING) {

- jid = JID(std::string(luaL_checkstring(L, 2)));

- data = std::string(luaL_checkstring(L, 3));

- if (lua_type(L, 4) != LUA_TNONE) {

- timeout = boost::numeric_cast<int>(luaL_checknumber(L, 4));

- }

- }

- else {

- data = std::string(luaL_checkstring(L, 2));

- if (lua_type(L, 3) != LUA_TNONE) {

- timeout = boost::numeric_cast<int>(luaL_checknumber(L, 3));

- }

- }

- boost::optional<std::string> result = client->sendQuery(jid, IQ::Set, data, timeout);

- if (result) {

- lua_pushstring(L, result->c_str());

- else {

- lua_pushnil(L);

- return 1;

- catch (const SluiftException& e) {

- return luaL_error(L, e.getReason().c_str());

- try {

- eventLoop.runOnce();

- getClient(L)->getClient()->sendData(std::string(luaL_checkstring(L, 2)));

- lua_pushvalue(L, 1);

- return 0;

- }

- catch (const SluiftException& e) {

- return luaL_error(L, e.getReason().c_str());

- }

- try {

- eventLoop.runOnce();

- SluiftClient* client = getClient(L);

- luaL_checktype(L, 2, LUA_TFUNCTION);

- int timeout = -1;

- if (lua_type(L, 3) != LUA_TNONE) {

- timeout = boost::numeric_cast<int>(lua_tonumber(L, 3));

- }

- while (true) {

- Stanza::ref event = client->getNextEvent(timeout);

- if (!event) {

- // We got a timeout

- lua_pushnil(L);

- return 1;

- }

- else {

- // Push the function and event on the stack

- lua_pushvalue(L, 2);

- pushEvent(L, event);

- int oldTop = lua_gettop(L) - 2;

- lua_call(L, 1, LUA_MULTRET);

- int returnValues = lua_gettop(L) - oldTop;

- if (returnValues > 0) {

- lua_remove(L, -1 - returnValues);

- return returnValues;

- }

- catch (const SluiftException& e) {

- return luaL_error(L, e.getReason().c_str());

- }

- try {

- eventLoop.runOnce();

- SluiftClient* client = getClient(L);

- int timeout = -1;

- if (lua_type(L, 2) != LUA_TNONE) {

- timeout = boost::numeric_cast<int>(lua_tonumber(L, 2));

- }

- pushEvent(L, client->getNextEvent(timeout));

- return 1;

- }

- catch (const SluiftException& e) {

- return luaL_error(L, e.getReason().c_str());

- try {

- eventLoop.runOnce();

- SluiftClient* client = getClient(L);

- RosterItemPayload item;

- if (lua_type(L, 2) == LUA_TTABLE) {

- lua_getfield(L, 2, "jid");

- const char* rawJID = lua_tostring(L, -1);

- if (rawJID) {

- item.setJID(std::string(rawJID));

- }

- lua_getfield(L, 2, "name");

- const char* rawName = lua_tostring(L, -1);

- if (rawName) {

- item.setName(rawName);

- }

- lua_getfield(L, 2, "groups");

- if (!lua_isnil(L, -1)) {

- if (lua_type(L, -1) == LUA_TTABLE) {

- for (size_t i = 1; i <= lua_objlen(L, -1); ++i) {

- lua_rawgeti(L, -1, boost::numeric_cast<int>(i));

- const char* rawGroup = lua_tostring(L, -1);

- if (rawGroup) {

- item.addGroup(rawGroup);

- }

- lua_pop(L, 1);

- }

- else {

- return luaL_error(L, "Groups should be a table");

- }

- else {

- item.setJID(luaL_checkstring(L, 2));

- }

-

- client->getRoster();

- if (!client->getClient()->getRoster()->containsJID(item.getJID())) {

- RosterPayload::ref roster = boost::make_shared<RosterPayload>();

- roster->addItem(item);

-

- ResponseSink<RosterPayload> sink;

- SetRosterRequest::ref request = SetRosterRequest::create(roster, client->getClient()->getIQRouter());

- boost::signals::scoped_connection c = request->onResponse.connect(boost::ref(sink));

- request->send();

- while (!sink.hasResponse()) {

- eventLoop.runUntilEvents();

- }

- if (sink.getResponseError()) {

- lua_pushboolean(L, false);

- return 1;

- client->getClient()->getSubscriptionManager()->requestSubscription(item.getJID());

- lua_pushboolean(L, true);

- return 1;

- catch (const SluiftException& e) {

- return luaL_error(L, e.getReason().c_str());

-}

-

-static int sluift_client_remove_contact(lua_State* L) {

- try {

- eventLoop.runOnce();

- SluiftClient* client = getClient(L);

- JID jid(luaL_checkstring(L, 2));

- roster->addItem(RosterItemPayload(JID(luaL_checkstring(L, 2)), "", RosterItemPayload::Remove));

- lua_pushboolean(L, !sink.getResponseError());

- return 1;

- }

- catch (const SluiftException& e) {

- return luaL_error(L, e.getReason().c_str());

- try {

- eventLoop.runOnce();

- SluiftClient* client = getClient(L);

- JID jid(luaL_checkstring(L, 2));

- client->getClient()->getSubscriptionManager()->confirmSubscription(jid);

- return 0;

- }

- catch (const SluiftException& e) {

- return luaL_error(L, e.getReason().c_str());

- }

- try {

- eventLoop.runOnce();

- SluiftClient* client = getClient(L);

- JID jid(luaL_checkstring(L, 2));

- client->getClient()->getSubscriptionManager()->cancelSubscription(jid);

- return 0;

- }

- catch (const SluiftException& e) {

- return luaL_error(L, e.getReason().c_str());

- }

- try {

- JID jid(std::string(luaL_checkstring(L, 1)));

- std::string password(luaL_checkstring(L, 2));

- SluiftClient** client = reinterpret_cast<SluiftClient**>(lua_newuserdata(L, sizeof(SluiftClient*)));

- luaL_getmetatable(L, SLUIFT_CLIENT);

- lua_setmetatable(L, -2);

- *client = new SluiftClient(jid, password);

- return 1;

- }

- catch (const SluiftException& e) {

- return luaL_error(L, e.getReason().c_str());

- }

- try {

- eventLoop.runOnce();

- int timeout = boost::numeric_cast<int>(luaL_checknumber(L, 1));

- Watchdog watchdog(timeout, networkFactories.getTimerFactory());

- while (!watchdog.getTimedOut()) {

- Swift::sleep(boost::numeric_cast<unsigned int>(std::min(100, timeout)));

- eventLoop.runOnce();

- }

- return 0;

- }

- catch (const SluiftException& e) {

- return luaL_error(L, e.getReason().c_str());

- * Copyright (c) 2011 Remko Tronçon

-#define SLUIFT_API extern