412 lines
12 KiB
C
412 lines
12 KiB
C
/*
|
|
* relay-raw.c - functions for Relay raw data messages
|
|
*
|
|
* Copyright (C) 2003-2020 Sébastien Helleu <flashcode@flashtux.org>
|
|
*
|
|
* This file is part of WeeChat, the extensible chat client.
|
|
*
|
|
* WeeChat 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.
|
|
*
|
|
* WeeChat 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 WeeChat. If not, see <https://www.gnu.org/licenses/>.
|
|
*/
|
|
|
|
#include <stdlib.h>
|
|
#include <stdio.h>
|
|
#include <stdarg.h>
|
|
#include <string.h>
|
|
#include <time.h>
|
|
|
|
#include "../weechat-plugin.h"
|
|
#include "relay.h"
|
|
#include "relay-raw.h"
|
|
#include "relay-buffer.h"
|
|
#include "relay-client.h"
|
|
#include "relay-config.h"
|
|
|
|
|
|
struct t_gui_buffer *relay_raw_buffer = NULL;
|
|
|
|
int relay_raw_messages_count = 0;
|
|
struct t_relay_raw_message *relay_raw_messages = NULL;
|
|
struct t_relay_raw_message *last_relay_raw_message = NULL;
|
|
|
|
|
|
/*
|
|
* Prints a relay raw message.
|
|
*/
|
|
|
|
void
|
|
relay_raw_message_print (struct t_relay_raw_message *raw_message)
|
|
{
|
|
if (relay_raw_buffer && raw_message)
|
|
{
|
|
weechat_printf_date_tags (relay_raw_buffer,
|
|
raw_message->date, NULL,
|
|
"%s\t%s",
|
|
raw_message->prefix,
|
|
raw_message->message);
|
|
}
|
|
}
|
|
|
|
/*
|
|
* Opens relay raw buffer.
|
|
*/
|
|
|
|
void
|
|
relay_raw_open (int switch_to_buffer)
|
|
{
|
|
struct t_relay_raw_message *ptr_raw_message;
|
|
|
|
if (!relay_raw_buffer)
|
|
{
|
|
relay_raw_buffer = weechat_buffer_search (RELAY_PLUGIN_NAME,
|
|
RELAY_RAW_BUFFER_NAME);
|
|
if (!relay_raw_buffer)
|
|
{
|
|
relay_raw_buffer = weechat_buffer_new (
|
|
RELAY_RAW_BUFFER_NAME,
|
|
&relay_buffer_input_cb, NULL, NULL,
|
|
&relay_buffer_close_cb, NULL, NULL);
|
|
|
|
/* failed to create buffer ? then return */
|
|
if (!relay_raw_buffer)
|
|
return;
|
|
|
|
weechat_buffer_set (relay_raw_buffer,
|
|
"title", _("Relay raw messages"));
|
|
|
|
if (!weechat_buffer_get_integer (relay_raw_buffer, "short_name_is_set"))
|
|
{
|
|
weechat_buffer_set (relay_raw_buffer, "short_name",
|
|
RELAY_RAW_BUFFER_NAME);
|
|
}
|
|
weechat_buffer_set (relay_raw_buffer, "localvar_set_type", "debug");
|
|
weechat_buffer_set (relay_raw_buffer, "localvar_set_server", RELAY_RAW_BUFFER_NAME);
|
|
weechat_buffer_set (relay_raw_buffer, "localvar_set_channel", RELAY_RAW_BUFFER_NAME);
|
|
weechat_buffer_set (relay_raw_buffer, "localvar_set_no_log", "1");
|
|
|
|
/* disable all highlights on this buffer */
|
|
weechat_buffer_set (relay_raw_buffer, "highlight_words", "-");
|
|
|
|
/* print messages in list */
|
|
for (ptr_raw_message = relay_raw_messages; ptr_raw_message;
|
|
ptr_raw_message = ptr_raw_message->next_message)
|
|
{
|
|
relay_raw_message_print (ptr_raw_message);
|
|
}
|
|
}
|
|
}
|
|
|
|
if (relay_raw_buffer && switch_to_buffer)
|
|
weechat_buffer_set (relay_raw_buffer, "display", "1");
|
|
}
|
|
|
|
/*
|
|
* Frees a raw message and remove it from list.
|
|
*/
|
|
|
|
void
|
|
relay_raw_message_free (struct t_relay_raw_message *raw_message)
|
|
{
|
|
struct t_relay_raw_message *new_raw_messages;
|
|
|
|
if (!raw_message)
|
|
return;
|
|
|
|
/* remove message from raw messages list */
|
|
if (last_relay_raw_message == raw_message)
|
|
last_relay_raw_message = raw_message->prev_message;
|
|
if (raw_message->prev_message)
|
|
{
|
|
(raw_message->prev_message)->next_message = raw_message->next_message;
|
|
new_raw_messages = relay_raw_messages;
|
|
}
|
|
else
|
|
new_raw_messages = raw_message->next_message;
|
|
|
|
if (raw_message->next_message)
|
|
(raw_message->next_message)->prev_message = raw_message->prev_message;
|
|
|
|
/* free data */
|
|
if (raw_message->prefix)
|
|
free (raw_message->prefix);
|
|
if (raw_message->message)
|
|
free (raw_message->message);
|
|
|
|
free (raw_message);
|
|
|
|
relay_raw_messages = new_raw_messages;
|
|
|
|
relay_raw_messages_count--;
|
|
}
|
|
|
|
/*
|
|
* Frees all raw messages.
|
|
*/
|
|
|
|
void
|
|
relay_raw_message_free_all ()
|
|
{
|
|
while (relay_raw_messages)
|
|
{
|
|
relay_raw_message_free (relay_raw_messages);
|
|
}
|
|
}
|
|
|
|
/*
|
|
* Removes old raw messages if limit has been reached.
|
|
*/
|
|
|
|
void
|
|
relay_raw_message_remove_old ()
|
|
{
|
|
int max_messages;
|
|
|
|
max_messages = weechat_config_integer (relay_config_look_raw_messages);
|
|
while (relay_raw_messages && (relay_raw_messages_count >= max_messages))
|
|
{
|
|
relay_raw_message_free (relay_raw_messages);
|
|
}
|
|
}
|
|
|
|
/*
|
|
* Adds a new raw message to list.
|
|
*
|
|
* Returns pointer to new raw message, NULL if error.
|
|
*/
|
|
|
|
struct t_relay_raw_message *
|
|
relay_raw_message_add_to_list (time_t date, const char *prefix,
|
|
const char *message)
|
|
{
|
|
struct t_relay_raw_message *new_raw_message;
|
|
|
|
if (!prefix || !message)
|
|
return NULL;
|
|
|
|
relay_raw_message_remove_old ();
|
|
|
|
new_raw_message = malloc (sizeof (*new_raw_message));
|
|
if (new_raw_message)
|
|
{
|
|
new_raw_message->date = date;
|
|
new_raw_message->prefix = strdup (prefix);
|
|
new_raw_message->message = strdup (message);
|
|
|
|
/* add message to list */
|
|
new_raw_message->prev_message = last_relay_raw_message;
|
|
new_raw_message->next_message = NULL;
|
|
if (last_relay_raw_message)
|
|
last_relay_raw_message->next_message = new_raw_message;
|
|
else
|
|
relay_raw_messages = new_raw_message;
|
|
last_relay_raw_message = new_raw_message;
|
|
|
|
relay_raw_messages_count++;
|
|
}
|
|
|
|
return new_raw_message;
|
|
}
|
|
|
|
/*
|
|
* Adds a new raw message to list.
|
|
*/
|
|
|
|
void
|
|
relay_raw_message_add (struct t_relay_client *client,
|
|
enum t_relay_client_msg_type msg_type,
|
|
int flags,
|
|
const char *data, int data_size)
|
|
{
|
|
char *buf, *buf2, *buf3, prefix[256], prefix_arrow[16];
|
|
const unsigned char *ptr_buf;
|
|
const char *hexa = "0123456789ABCDEF";
|
|
int pos_buf, pos_buf2, char_size, i, length;
|
|
struct t_relay_raw_message *new_raw_message;
|
|
|
|
buf = NULL;
|
|
buf2 = NULL;
|
|
buf3 = NULL;
|
|
|
|
if (flags & RELAY_RAW_FLAG_BINARY)
|
|
{
|
|
/* binary message */
|
|
buf = weechat_string_hex_dump (data, data_size, 16, " > ", NULL);
|
|
snprintf (prefix, sizeof (prefix), " ");
|
|
}
|
|
else
|
|
{
|
|
/* text message */
|
|
buf = weechat_iconv_to_internal (NULL, data);
|
|
buf2 = weechat_string_replace (buf, "\r", "");
|
|
if (buf2)
|
|
{
|
|
free (buf);
|
|
buf = buf2;
|
|
buf2 = NULL;
|
|
}
|
|
buf2 = malloc ((strlen (buf) * 4) + 1);
|
|
if (buf2)
|
|
{
|
|
ptr_buf = (buf) ? (unsigned char *)buf : (unsigned char *)data;
|
|
pos_buf = 0;
|
|
pos_buf2 = 0;
|
|
while (ptr_buf[pos_buf])
|
|
{
|
|
if ((ptr_buf[pos_buf] < 32) && (ptr_buf[pos_buf] != '\n'))
|
|
{
|
|
buf2[pos_buf2++] = '\\';
|
|
buf2[pos_buf2++] = 'x';
|
|
buf2[pos_buf2++] = hexa[ptr_buf[pos_buf] / 16];
|
|
buf2[pos_buf2++] = hexa[ptr_buf[pos_buf] % 16];
|
|
pos_buf++;
|
|
}
|
|
else
|
|
{
|
|
char_size = weechat_utf8_char_size ((const char *)(ptr_buf + pos_buf));
|
|
for (i = 0; i < char_size; i++)
|
|
{
|
|
buf2[pos_buf2++] = ptr_buf[pos_buf++];
|
|
}
|
|
}
|
|
}
|
|
buf2[pos_buf2] = '\0';
|
|
}
|
|
}
|
|
|
|
if (!(flags & RELAY_RAW_FLAG_BINARY)
|
|
|| (msg_type == RELAY_CLIENT_MSG_PING)
|
|
|| (msg_type == RELAY_CLIENT_MSG_PONG)
|
|
|| (msg_type == RELAY_CLIENT_MSG_CLOSE))
|
|
{
|
|
/* build prefix with arrow */
|
|
prefix_arrow[0] = '\0';
|
|
switch (flags & (RELAY_RAW_FLAG_RECV | RELAY_RAW_FLAG_SEND))
|
|
{
|
|
case RELAY_RAW_FLAG_RECV:
|
|
strcpy (prefix_arrow, RELAY_RAW_PREFIX_RECV);
|
|
break;
|
|
case RELAY_RAW_FLAG_SEND:
|
|
strcpy (prefix_arrow, RELAY_RAW_PREFIX_SEND);
|
|
break;
|
|
default:
|
|
if (flags & RELAY_RAW_FLAG_RECV)
|
|
strcpy (prefix_arrow, RELAY_RAW_PREFIX_RECV);
|
|
else
|
|
strcpy (prefix_arrow, RELAY_RAW_PREFIX_SEND);
|
|
break;
|
|
}
|
|
|
|
if (client)
|
|
{
|
|
snprintf (prefix, sizeof (prefix), "%s%s %s[%s%d%s] %s%s%s%s",
|
|
(flags & RELAY_RAW_FLAG_SEND) ?
|
|
weechat_color ("chat_prefix_quit") :
|
|
weechat_color ("chat_prefix_join"),
|
|
prefix_arrow,
|
|
weechat_color ("chat_delimiters"),
|
|
weechat_color ("chat"),
|
|
client->id,
|
|
weechat_color ("chat_delimiters"),
|
|
weechat_color ("chat_server"),
|
|
relay_protocol_string[client->protocol],
|
|
(client->protocol_args) ? "." : "",
|
|
(client->protocol_args) ? client->protocol_args : "");
|
|
}
|
|
else
|
|
{
|
|
snprintf (prefix, sizeof (prefix), "%s%s",
|
|
(flags & RELAY_RAW_FLAG_SEND) ?
|
|
weechat_color ("chat_prefix_quit") :
|
|
weechat_color ("chat_prefix_join"),
|
|
prefix_arrow);
|
|
}
|
|
}
|
|
|
|
length = strlen (relay_client_msg_type_string[msg_type]) +
|
|
strlen ((buf2) ? buf2 : ((buf) ? buf : data)) + 1;
|
|
buf3 = malloc (length);
|
|
if (buf3)
|
|
{
|
|
snprintf (buf3, length, "%s%s",
|
|
relay_client_msg_type_string[msg_type],
|
|
(buf2) ? buf2 : ((buf) ? buf : data));
|
|
}
|
|
|
|
new_raw_message = relay_raw_message_add_to_list (
|
|
time (NULL),
|
|
prefix,
|
|
(buf3) ? buf3 : ((buf2) ? buf2 : ((buf) ? buf : data)));
|
|
|
|
if (new_raw_message)
|
|
{
|
|
if (relay_raw_buffer)
|
|
relay_raw_message_print (new_raw_message);
|
|
if (weechat_config_integer (relay_config_look_raw_messages) == 0)
|
|
relay_raw_message_free (new_raw_message);
|
|
}
|
|
|
|
if (buf)
|
|
free (buf);
|
|
if (buf2)
|
|
free (buf2);
|
|
if (buf3)
|
|
free (buf3);
|
|
}
|
|
|
|
/*
|
|
* Prints a message on relay raw buffer.
|
|
*/
|
|
|
|
void
|
|
relay_raw_print (struct t_relay_client *client,
|
|
enum t_relay_client_msg_type msg_type, int flags,
|
|
const char *data, int data_size)
|
|
{
|
|
/* auto-open Relay raw buffer if debug for irc plugin is >= 1 */
|
|
if (!relay_raw_buffer && (weechat_relay_plugin->debug >= 1))
|
|
relay_raw_open (0);
|
|
|
|
relay_raw_message_add (client, msg_type, flags, data, data_size);
|
|
}
|
|
|
|
/*
|
|
* Adds a raw message in an infolist.
|
|
*
|
|
* Returns:
|
|
* 1: OK
|
|
* 0: error
|
|
*/
|
|
|
|
int
|
|
relay_raw_add_to_infolist (struct t_infolist *infolist,
|
|
struct t_relay_raw_message *raw_message)
|
|
{
|
|
struct t_infolist_item *ptr_item;
|
|
|
|
if (!infolist || !raw_message)
|
|
return 0;
|
|
|
|
ptr_item = weechat_infolist_new_item (infolist);
|
|
if (!ptr_item)
|
|
return 0;
|
|
|
|
if (!weechat_infolist_new_var_time (ptr_item, "date", raw_message->date))
|
|
return 0;
|
|
if (!weechat_infolist_new_var_string (ptr_item, "prefix", raw_message->prefix))
|
|
return 0;
|
|
if (!weechat_infolist_new_var_string (ptr_item, "message", raw_message->message))
|
|
return 0;
|
|
|
|
return 1;
|
|
}
|