1
0
mirror of https://github.com/profanity-im/profanity.git synced 2024-12-04 14:46:46 -05:00
profanity/src/xmpp/connection.c

553 lines
16 KiB
C
Raw Normal View History

/*
2013-02-02 14:47:41 -05:00
* connection.c
2012-02-20 15:07:38 -05:00
*
2013-01-10 21:05:29 -05:00
* Copyright (C) 2012, 2013 James Booth <boothj5@gmail.com>
*
2012-02-20 15:07:38 -05:00
* This file is part of Profanity.
*
* Profanity is free software: you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* Profanity is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with Profanity. If not, see <http://www.gnu.org/licenses/>.
*
*/
2013-02-02 15:55:58 -05:00
#include <assert.h>
2012-02-05 18:29:09 -05:00
#include <string.h>
#include <stdlib.h>
2012-08-25 20:50:50 -04:00
2012-02-25 22:15:42 -05:00
#include <strophe.h>
2012-02-05 18:29:09 -05:00
#include "chat_session.h"
2012-08-22 20:08:06 -04:00
#include "common.h"
2013-02-02 16:59:29 -05:00
#include "config/preferences.h"
2012-10-28 16:52:30 -04:00
#include "contact_list.h"
2013-01-12 18:10:56 -05:00
#include "jid.h"
2012-08-25 20:50:50 -04:00
#include "log.h"
2013-02-02 15:55:58 -05:00
#include "muc.h"
#include "profanity.h"
2013-02-03 17:46:32 -05:00
#include "xmpp/capabilities.h"
#include "xmpp/connection.h"
2013-02-02 15:55:58 -05:00
#include "xmpp/iq.h"
#include "xmpp/message.h"
#include "xmpp/presence.h"
#include "xmpp/stanza.h"
#include "xmpp/xmpp.h"
2012-02-05 18:29:09 -05:00
static struct _jabber_conn_t {
xmpp_log_t *log;
xmpp_ctx_t *ctx;
xmpp_conn_t *conn;
jabber_conn_status_t conn_status;
char *presence_message;
int priority;
int tls_disabled;
} jabber_conn;
2012-02-16 19:42:41 -05:00
static GHashTable *available_resources;
// for auto reconnect
static struct {
char *name;
char *passwd;
} saved_account;
static struct {
char *name;
char *jid;
char *passwd;
char *altdomain;
} saved_details;
static GTimer *reconnect_timer;
static log_level_t _get_log_level(xmpp_log_level_t xmpp_level);
static xmpp_log_level_t _get_xmpp_log_level();
static void _xmpp_file_logger(void * const userdata,
const xmpp_log_level_t level, const char * const area,
const char * const msg);
static xmpp_log_t * _xmpp_get_file_logger();
2012-08-18 21:44:46 -04:00
static jabber_conn_status_t _jabber_connect(const char * const fulljid,
const char * const passwd, const char * const altdomain);
static void _jabber_reconnect(void);
static void _connection_handler(xmpp_conn_t * const conn,
const xmpp_conn_event_t status, const int error,
2012-02-10 18:58:57 -05:00
xmpp_stream_error_t * const stream_error, void * const userdata);
2012-02-27 16:16:54 -05:00
static int _ping_timed_handler(xmpp_conn_t * const conn, void * const userdata);
void _connection_free_resources(void);
2012-07-24 18:19:48 -04:00
void
jabber_init(const int disable_tls)
{
log_info("Initialising XMPP");
jabber_conn.conn_status = JABBER_STARTED;
jabber_conn.presence_message = NULL;
jabber_conn.tls_disabled = disable_tls;
2013-01-28 17:24:47 -05:00
presence_init();
2013-02-03 17:46:32 -05:00
caps_init();
available_resources = g_hash_table_new_full(g_str_hash, g_str_equal, free,
(GDestroyNotify)resource_destroy);
}
jabber_conn_status_t
2013-02-03 19:37:10 -05:00
jabber_connect_with_account(const ProfAccount * const account,
const char * const passwd)
{
2013-02-03 18:46:20 -05:00
assert(account != NULL);
assert(passwd != NULL);
log_info("Connecting using account: %s", account->name);
// save account name and password for reconnect
saved_account.name = strdup(account->name);
saved_account.passwd = strdup(passwd);
2013-02-03 18:59:04 -05:00
// connect with fulljid
Jid *jidp = jid_create_from_bare_and_resource(account->jid, account->resource);
jabber_conn_status_t result = _jabber_connect(jidp->fulljid, passwd, account->server);
free(jidp);
return result;
}
jabber_conn_status_t
jabber_connect_with_details(const char * const jid,
const char * const passwd, const char * const altdomain)
{
2013-02-03 18:59:04 -05:00
assert(jid != NULL);
assert(passwd != NULL);
// save details for reconnect, remember name for account creating on success
saved_details.name = strdup(jid);
saved_details.passwd = strdup(passwd);
if (altdomain != NULL) {
saved_details.altdomain = strdup(altdomain);
} else {
saved_details.altdomain = NULL;
}
2013-02-03 18:59:04 -05:00
// use 'profanity' when no resourcepart in provided jid
Jid *jidp = jid_create(jid);
if (jidp->resourcepart == NULL) {
jid_destroy(jidp);
jidp = jid_create_from_bare_and_resource(jid, "profanity");
saved_details.jid = strdup(jidp->fulljid);
} else {
saved_details.jid = strdup(jid);
}
jid_destroy(jidp);
2013-02-03 18:59:04 -05:00
// connect with fulljid
log_info("Connecting without account, JID: %s", saved_details.jid);
return _jabber_connect(saved_details.jid, passwd, saved_details.altdomain);
}
2012-10-27 13:26:57 -04:00
void
2012-07-24 18:19:48 -04:00
jabber_disconnect(void)
2012-02-06 19:08:59 -05:00
{
// if connected, send end stream and wait for response
if (jabber_conn.conn_status == JABBER_CONNECTED) {
log_info("Closing connection");
jabber_conn.conn_status = JABBER_DISCONNECTING;
xmpp_disconnect(jabber_conn.conn);
2012-10-27 13:26:57 -04:00
while (jabber_get_connection_status() == JABBER_DISCONNECTING) {
jabber_process_events();
}
_connection_free_resources();
2012-02-19 20:42:29 -05:00
}
2013-02-03 18:46:20 -05:00
2013-02-03 17:56:23 -05:00
jabber_conn.conn_status = JABBER_STARTED;
FREE_SET_NULL(jabber_conn.presence_message);
2012-02-06 19:08:59 -05:00
}
2012-07-24 18:19:48 -04:00
void
jabber_process_events(void)
2012-02-06 19:08:59 -05:00
{
// run xmpp event loop if connected, connecting or disconnecting
if (jabber_conn.conn_status == JABBER_CONNECTED
|| jabber_conn.conn_status == JABBER_CONNECTING
|| jabber_conn.conn_status == JABBER_DISCONNECTING) {
xmpp_run_once(jabber_conn.ctx, 10);
// check timer and reconnect if disconnected and timer set
} else if (prefs_get_reconnect() != 0) {
if ((jabber_conn.conn_status == JABBER_DISCONNECTED) &&
(reconnect_timer != NULL)) {
if (g_timer_elapsed(reconnect_timer, NULL) > prefs_get_reconnect()) {
_jabber_reconnect();
}
}
}
2012-02-06 19:08:59 -05:00
}
void
2013-02-03 19:37:10 -05:00
jabber_set_autoping(const int seconds)
{
2013-01-12 16:48:24 -05:00
if (jabber_conn.conn_status == JABBER_CONNECTED) {
xmpp_timed_handler_delete(jabber_conn.conn, _ping_timed_handler);
if (seconds != 0) {
int millis = seconds * 1000;
xmpp_timed_handler_add(jabber_conn.conn, _ping_timed_handler, millis,
jabber_conn.ctx);
}
}
}
GList *
jabber_get_available_resources(void)
{
return g_hash_table_get_values(available_resources);
}
jabber_conn_status_t
jabber_get_connection_status(void)
{
return (jabber_conn.conn_status);
}
2013-01-28 17:24:47 -05:00
xmpp_conn_t *
connection_get_conn(void)
2013-01-28 17:24:47 -05:00
{
return jabber_conn.conn;
}
xmpp_ctx_t *
connection_get_ctx(void)
2013-01-28 17:24:47 -05:00
{
return jabber_conn.ctx;
}
const char *
jabber_get_jid(void)
{
return xmpp_conn_get_jid(jabber_conn.conn);
}
char *
jabber_get_presence_message(void)
{
return jabber_conn.presence_message;
}
2013-01-28 17:55:26 -05:00
char *
jabber_get_account_name(void)
{
return saved_account.name;
}
void
connection_set_presence_message(const char * const message)
2013-01-28 17:55:26 -05:00
{
FREE_SET_NULL(jabber_conn.presence_message);
2013-01-28 17:55:26 -05:00
if (message != NULL) {
jabber_conn.presence_message = strdup(message);
2013-01-28 17:55:26 -05:00
}
}
void
2013-02-03 19:37:10 -05:00
connection_set_priority(const int priority)
2013-01-28 17:55:26 -05:00
{
jabber_conn.priority = priority;
}
void
connection_add_available_resource(Resource *resource)
{
g_hash_table_replace(available_resources, strdup(resource->name), resource);
}
void
connection_remove_available_resource(const char * const resource)
{
g_hash_table_remove(available_resources, resource);
}
void
_connection_free_resources(void)
{
FREE_SET_NULL(saved_details.name);
FREE_SET_NULL(saved_details.jid);
FREE_SET_NULL(saved_details.passwd);
FREE_SET_NULL(saved_details.altdomain);
FREE_SET_NULL(saved_account.name);
FREE_SET_NULL(saved_account.passwd);
g_hash_table_remove_all(available_resources);
chat_sessions_clear();
2013-01-28 17:24:47 -05:00
presence_free_sub_requests();
2012-10-16 18:56:44 -04:00
xmpp_conn_release(jabber_conn.conn);
xmpp_ctx_free(jabber_conn.ctx);
xmpp_shutdown();
}
2013-01-28 17:24:47 -05:00
int
connection_error_handler(xmpp_conn_t * const conn, xmpp_stanza_t * const stanza,
void * const userdata)
{
2012-11-09 21:28:38 -05:00
gchar *err_msg = NULL;
gchar *from = xmpp_stanza_get_attribute(stanza, STANZA_ATTR_FROM);
xmpp_stanza_t *error_stanza = xmpp_stanza_get_child_by_name(stanza, STANZA_NAME_ERROR);
xmpp_stanza_t *text_stanza =
xmpp_stanza_get_child_by_name(error_stanza, STANZA_NAME_TEXT);
2012-11-05 16:48:13 -05:00
2012-11-09 21:28:38 -05:00
if (error_stanza == NULL) {
log_debug("error message without <error/> received");
2012-11-05 15:29:57 -05:00
} else {
2012-11-09 21:28:38 -05:00
// check for text
if (text_stanza != NULL) {
err_msg = xmpp_stanza_get_text(text_stanza);
prof_handle_error_message(from, err_msg);
// TODO : process 'type' attribute from <error/> [RFC6120, 8.3.2]
// otherwise show defined-condition
} else {
2012-11-09 21:28:38 -05:00
xmpp_stanza_t *err_cond = xmpp_stanza_get_children(error_stanza);
2012-08-15 18:52:54 -04:00
2012-11-09 21:28:38 -05:00
if (err_cond == NULL) {
log_debug("error message without <defined-condition/> or <text/> received");
2012-11-09 21:28:38 -05:00
} else {
err_msg = xmpp_stanza_get_name(err_cond);
prof_handle_error_message(from, err_msg);
// TODO : process 'type' attribute from <error/> [RFC6120, 8.3.2]
}
}
}
return 1;
}
static jabber_conn_status_t
_jabber_connect(const char * const fulljid, const char * const passwd,
const char * const altdomain)
{
2013-02-03 18:46:20 -05:00
assert(fulljid != NULL);
assert(passwd != NULL);
Jid *jid = jid_create(fulljid);
if (jid == NULL) {
log_error("Malformed JID not able to connect: %s", fulljid);
jabber_conn.conn_status = JABBER_DISCONNECTED;
return jabber_conn.conn_status;
} else if (jid->fulljid == NULL) {
log_error("Full JID required to connect, received: %s", fulljid);
jabber_conn.conn_status = JABBER_DISCONNECTED;
return jabber_conn.conn_status;
}
jid_destroy(jid);
2013-01-12 22:14:36 -05:00
log_info("Connecting as %s", fulljid);
xmpp_initialize();
jabber_conn.log = _xmpp_get_file_logger();
jabber_conn.ctx = xmpp_ctx_new(NULL, jabber_conn.log);
jabber_conn.conn = xmpp_conn_new(jabber_conn.ctx);
xmpp_conn_set_jid(jabber_conn.conn, fulljid);
xmpp_conn_set_pass(jabber_conn.conn, passwd);
if (jabber_conn.tls_disabled)
xmpp_conn_disable_tls(jabber_conn.conn);
2012-02-05 18:29:09 -05:00
int connect_status = xmpp_connect_client(jabber_conn.conn, altdomain, 0,
_connection_handler, jabber_conn.ctx);
if (connect_status == 0)
jabber_conn.conn_status = JABBER_CONNECTING;
else
jabber_conn.conn_status = JABBER_DISCONNECTED;
return jabber_conn.conn_status;
2012-02-05 18:29:09 -05:00
}
2013-01-28 20:02:40 -05:00
static void
_jabber_reconnect(void)
{
// reconnect with account.
ProfAccount *account = accounts_get_account(saved_account.name);
if (account == NULL) {
log_error("Unable to reconnect, account no longer exists: %s", saved_account.name);
} else {
char *fulljid = create_fulljid(account->jid, account->resource);
log_debug("Attempting reconnect with account %s", account->name);
_jabber_connect(fulljid, saved_account.passwd, account->server);
free(fulljid);
g_timer_start(reconnect_timer);
}
}
2012-07-24 18:19:48 -04:00
static void
_connection_handler(xmpp_conn_t * const conn,
const xmpp_conn_event_t status, const int error,
2012-02-10 18:58:57 -05:00
xmpp_stream_error_t * const stream_error, void * const userdata)
2012-02-05 18:29:09 -05:00
{
xmpp_ctx_t *ctx = (xmpp_ctx_t *)userdata;
// login success
2012-02-05 18:29:09 -05:00
if (status == XMPP_CONN_CONNECT) {
2013-02-26 17:02:03 -05:00
log_debug("Connection handler: XMPP_CONN_CONNECT");
// logged in with account
if (saved_account.name != NULL) {
2013-02-26 17:02:03 -05:00
log_debug("Connection handler: logged in with account name: %s", saved_account.name);
prof_handle_login_account_success(saved_account.name);
// logged in without account, use details to create new account
2012-12-09 17:14:38 -05:00
} else {
2013-02-26 17:02:03 -05:00
log_debug("Connection handler: logged in with jid: %s", saved_details.name);
accounts_add(saved_details.name, saved_details.altdomain);
accounts_set_jid(saved_details.name, saved_details.jid);
prof_handle_login_account_success(saved_details.name);
saved_account.name = strdup(saved_details.name);
saved_account.passwd = strdup(saved_details.passwd);
FREE_SET_NULL(saved_details.name);
FREE_SET_NULL(saved_details.jid);
FREE_SET_NULL(saved_details.passwd);
FREE_SET_NULL(saved_details.altdomain);
2012-12-09 17:14:38 -05:00
}
chat_sessions_init();
2013-01-28 19:45:03 -05:00
message_add_handlers();
2013-01-28 19:04:49 -05:00
presence_add_handlers();
iq_add_handlers();
if (prefs_get_autoping() != 0) {
int millis = prefs_get_autoping() * 1000;
xmpp_timed_handler_add(conn, _ping_timed_handler, millis, ctx);
}
2012-02-05 18:29:09 -05:00
2013-01-28 20:02:40 -05:00
iq_roster_request();
jabber_conn.conn_status = JABBER_CONNECTED;
if (prefs_get_reconnect() != 0) {
if (reconnect_timer != NULL) {
g_timer_destroy(reconnect_timer);
reconnect_timer = NULL;
}
}
} else if (status == XMPP_CONN_DISCONNECT) {
2013-02-26 17:02:03 -05:00
log_debug("Connection handler: XMPP_CONN_DISCONNECT");
// lost connection for unkown reason
if (jabber_conn.conn_status == JABBER_CONNECTED) {
2013-02-26 17:02:03 -05:00
log_debug("Connection handler: Lost connection for unknown reason");
prof_handle_lost_connection();
if (prefs_get_reconnect() != 0) {
assert(reconnect_timer == NULL);
reconnect_timer = g_timer_new();
// TODO: free resources but leave saved_user untouched
} else {
_connection_free_resources();
}
// login attempt failed
} else if (jabber_conn.conn_status != JABBER_DISCONNECTING) {
2013-02-26 17:02:03 -05:00
log_debug("Connection handler: Login failed");
if (reconnect_timer == NULL) {
2013-02-26 17:02:03 -05:00
log_debug("Connection handler: No reconnect timer");
prof_handle_failed_login();
_connection_free_resources();
} else {
2013-02-26 17:02:03 -05:00
log_debug("Connection handler: Restarting reconnect timer");
if (prefs_get_reconnect() != 0) {
g_timer_start(reconnect_timer);
}
// TODO: free resources but leave saved_user untouched
}
}
// close stream response from server after disconnect is handled too
jabber_conn.conn_status = JABBER_DISCONNECTED;
2013-02-26 17:02:03 -05:00
} else if (status == XMPP_CONN_FAIL) {
log_debug("Connection handler: XMPP_CONN_FAIL");
} else {
log_error("Connection handler: Unknown status");
2012-02-05 18:29:09 -05:00
}
}
2012-02-08 21:47:25 -05:00
2012-07-24 18:19:48 -04:00
static int
_ping_timed_handler(xmpp_conn_t * const conn, void * const userdata)
2012-02-27 16:16:54 -05:00
{
2012-02-27 18:43:46 -05:00
if (jabber_conn.conn_status == JABBER_CONNECTED) {
xmpp_ctx_t *ctx = (xmpp_ctx_t *)userdata;
2012-11-09 19:43:09 -05:00
xmpp_stanza_t *iq = stanza_create_ping_iq(ctx);
2012-02-27 18:43:46 -05:00
xmpp_send(conn, iq);
xmpp_stanza_release(iq);
}
2012-02-27 16:16:54 -05:00
return 1;
}
static log_level_t
2013-02-03 19:37:10 -05:00
_get_log_level(const xmpp_log_level_t xmpp_level)
{
if (xmpp_level == XMPP_LEVEL_DEBUG) {
return PROF_LEVEL_DEBUG;
} else if (xmpp_level == XMPP_LEVEL_INFO) {
return PROF_LEVEL_INFO;
} else if (xmpp_level == XMPP_LEVEL_WARN) {
return PROF_LEVEL_WARN;
} else {
return PROF_LEVEL_ERROR;
}
}
static xmpp_log_level_t
_get_xmpp_log_level()
{
log_level_t prof_level = log_get_filter();
if (prof_level == PROF_LEVEL_DEBUG) {
return XMPP_LEVEL_DEBUG;
} else if (prof_level == PROF_LEVEL_INFO) {
return XMPP_LEVEL_INFO;
} else if (prof_level == PROF_LEVEL_WARN) {
return XMPP_LEVEL_WARN;
} else {
return XMPP_LEVEL_ERROR;
}
}
static void
_xmpp_file_logger(void * const userdata, const xmpp_log_level_t level,
const char * const area, const char * const msg)
{
log_level_t prof_level = _get_log_level(level);
log_msg(prof_level, area, msg);
}
static xmpp_log_t *
_xmpp_get_file_logger()
{
xmpp_log_level_t level = _get_xmpp_log_level();
xmpp_log_t *file_log = malloc(sizeof(xmpp_log_t));
file_log->handler = _xmpp_file_logger;
file_log->userdata = &level;
return file_log;
}