mirror of
https://github.com/irssi/irssi.git
synced 2024-09-08 04:26:01 -04:00
working sometimes because of missing "return 1". git-svn-id: http://svn.irssi.org/repos/irssi/trunk@2325 dbcabf3a-b0e7-0310-adc4-f8d773084564
193 lines
4.7 KiB
C
193 lines
4.7 KiB
C
/*
|
|
write-buffer.c : irssi
|
|
|
|
Copyright (C) 2001 Timo Sirainen
|
|
|
|
This program 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 2 of the License, or
|
|
(at your option) any later version.
|
|
|
|
This program 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 this program; if not, write to the Free Software
|
|
Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
|
|
*/
|
|
|
|
#include "module.h"
|
|
#include "signals.h"
|
|
#include "commands.h"
|
|
#include "settings.h"
|
|
#include "write-buffer.h"
|
|
|
|
#define BUFFER_BLOCK_SIZE 2048
|
|
|
|
typedef struct {
|
|
char *active_block;
|
|
int active_block_pos;
|
|
|
|
GSList *blocks;
|
|
} BUFFER_REC;
|
|
|
|
static GSList *empty_blocks;
|
|
static GHashTable *buffers;
|
|
static int block_count;
|
|
|
|
static int write_buffer_max_blocks;
|
|
static int timeout_tag;
|
|
|
|
static void write_buffer_new_block(BUFFER_REC *rec)
|
|
{
|
|
char *block;
|
|
|
|
if (empty_blocks == NULL)
|
|
block = g_malloc(BUFFER_BLOCK_SIZE);
|
|
else {
|
|
block = empty_blocks->data;
|
|
empty_blocks = g_slist_remove(empty_blocks, block);
|
|
}
|
|
|
|
block_count++;
|
|
rec->active_block = block;
|
|
rec->active_block_pos = 0;
|
|
rec->blocks = g_slist_append(rec->blocks, block);
|
|
}
|
|
|
|
int write_buffer(int handle, const void *data, int size)
|
|
{
|
|
BUFFER_REC *rec;
|
|
const char *cdata = data;
|
|
int next_size;
|
|
|
|
if (write_buffer_max_blocks <= 0) {
|
|
/* no write buffer */
|
|
return write(handle, data, size);
|
|
}
|
|
|
|
if (size <= 0)
|
|
return size;
|
|
|
|
rec = g_hash_table_lookup(buffers, GINT_TO_POINTER(handle));
|
|
if (rec == NULL) {
|
|
rec = g_new0(BUFFER_REC, 1);
|
|
write_buffer_new_block(rec);
|
|
g_hash_table_insert(buffers, GINT_TO_POINTER(handle), rec);
|
|
}
|
|
|
|
while (size > 0) {
|
|
if (rec->active_block_pos == BUFFER_BLOCK_SIZE)
|
|
write_buffer_new_block(rec);
|
|
|
|
next_size = size < BUFFER_BLOCK_SIZE-rec->active_block_pos ?
|
|
size : BUFFER_BLOCK_SIZE-rec->active_block_pos;
|
|
memcpy(rec->active_block+rec->active_block_pos,
|
|
cdata, next_size);
|
|
|
|
rec->active_block_pos += next_size;
|
|
cdata += next_size;
|
|
size -= next_size;
|
|
}
|
|
|
|
if (block_count > write_buffer_max_blocks)
|
|
write_buffer_flush();
|
|
|
|
return size;
|
|
}
|
|
|
|
static int write_buffer_flush_rec(void *handlep, BUFFER_REC *rec)
|
|
{
|
|
GSList *tmp;
|
|
int handle, size;
|
|
|
|
handle = GPOINTER_TO_INT(handlep);
|
|
for (tmp = rec->blocks; tmp != NULL; tmp = tmp->next) {
|
|
size = tmp->data != rec->active_block ? BUFFER_BLOCK_SIZE :
|
|
rec->active_block_pos;
|
|
write(handle, tmp->data, size);
|
|
}
|
|
|
|
empty_blocks = g_slist_concat(empty_blocks, rec->blocks);
|
|
g_free(rec);
|
|
return TRUE;
|
|
}
|
|
|
|
void write_buffer_flush(void)
|
|
{
|
|
g_slist_foreach(empty_blocks, (GFunc) g_free, NULL);
|
|
g_slist_free(empty_blocks);
|
|
empty_blocks = NULL;
|
|
|
|
g_hash_table_foreach_remove(buffers,
|
|
(GHRFunc) write_buffer_flush_rec, NULL);
|
|
block_count = 0;
|
|
}
|
|
|
|
static int flush_timeout(void)
|
|
{
|
|
write_buffer_flush();
|
|
return 1;
|
|
}
|
|
|
|
static void read_settings(void)
|
|
{
|
|
int msecs;
|
|
|
|
write_buffer_flush();
|
|
|
|
write_buffer_max_blocks = settings_get_int("write_buffer_kb") *
|
|
1024/BUFFER_BLOCK_SIZE;
|
|
|
|
if (settings_get_int("write_buffer_mins") > 0) {
|
|
msecs = settings_get_int("write_buffer_mins")*60*1000;
|
|
if (timeout_tag == -1) {
|
|
timeout_tag = g_timeout_add(msecs,
|
|
(GSourceFunc) flush_timeout,
|
|
NULL);
|
|
}
|
|
} else if (timeout_tag != -1) {
|
|
g_source_remove(timeout_tag);
|
|
timeout_tag = -1;
|
|
}
|
|
}
|
|
|
|
static void cmd_flushbuffer(void)
|
|
{
|
|
write_buffer_flush();
|
|
}
|
|
|
|
void write_buffer_init(void)
|
|
{
|
|
settings_add_int("misc", "write_buffer_mins", 0);
|
|
settings_add_int("misc", "write_buffer_kb", 0);
|
|
|
|
buffers = g_hash_table_new((GHashFunc) g_direct_hash,
|
|
(GCompareFunc) g_direct_equal);
|
|
|
|
empty_blocks = NULL;
|
|
block_count = 0;
|
|
|
|
timeout_tag = -1;
|
|
read_settings();
|
|
signal_add("setup changed", (SIGNAL_FUNC) read_settings);
|
|
command_bind("flushbuffer", NULL, (SIGNAL_FUNC) cmd_flushbuffer);
|
|
}
|
|
|
|
void write_buffer_deinit(void)
|
|
{
|
|
if (timeout_tag != -1)
|
|
g_source_remove(timeout_tag);
|
|
|
|
write_buffer_flush();
|
|
g_hash_table_destroy(buffers);
|
|
|
|
g_slist_foreach(empty_blocks, (GFunc) g_free, NULL);
|
|
g_slist_free(empty_blocks);
|
|
|
|
signal_remove("setup changed", (SIGNAL_FUNC) read_settings);
|
|
command_unbind("flushbuffer", (SIGNAL_FUNC) cmd_flushbuffer);
|
|
}
|