client: use GQueue instead of sllist.h for deferred_send

Another custom data structore converted to GLib.
This commit is contained in:
Max Kellermann 2008-10-31 09:18:11 +01:00
parent 90e9079142
commit 94cbdffb72
1 changed files with 42 additions and 34 deletions

View File

@ -22,7 +22,6 @@
#include "log.h" #include "log.h"
#include "listen.h" #include "listen.h"
#include "permission.h" #include "permission.h"
#include "sllist.h"
#include "utils.h" #include "utils.h"
#include "ioops.h" #include "ioops.h"
#include "main_notify.h" #include "main_notify.h"
@ -57,6 +56,11 @@ static size_t client_max_command_list_size =
static size_t client_max_output_buffer_size = static size_t client_max_output_buffer_size =
CLIENT_MAX_OUTPUT_BUFFER_SIZE_DEFAULT; CLIENT_MAX_OUTPUT_BUFFER_SIZE_DEFAULT;
struct deferred_buffer {
size_t size;
char data[sizeof(long)];
};
struct client { struct client {
struct list_head siblings; struct list_head siblings;
@ -74,7 +78,7 @@ struct client {
GSList *cmd_list; /* for when in list mode */ GSList *cmd_list; /* for when in list mode */
int cmd_list_OK; /* print OK after each command execution */ int cmd_list_OK; /* print OK after each command execution */
size_t cmd_list_size; /* mem cmd_list consumes */ size_t cmd_list_size; /* mem cmd_list consumes */
struct sllnode *deferred_send; /* for output if client is slow */ GQueue *deferred_send; /* for output if client is slow */
size_t deferred_bytes; /* mem deferred_send consumes */ size_t deferred_bytes; /* mem deferred_send consumes */
unsigned int num; /* client number */ unsigned int num; /* client number */
@ -138,7 +142,7 @@ static void client_init(struct client *client, int fd)
set_nonblocking(fd); set_nonblocking(fd);
client->lastTime = time(NULL); client->lastTime = time(NULL);
client->cmd_list = NULL; client->cmd_list = NULL;
client->deferred_send = NULL; client->deferred_send = g_queue_new();
client->deferred_bytes = 0; client->deferred_bytes = 0;
client->num = next_client_num++; client->num = next_client_num++;
client->send_buf_used = 0; client->send_buf_used = 0;
@ -161,10 +165,15 @@ static void new_cmd_list_ptr(struct client *client, char *s)
client->cmd_list = g_slist_prepend(client->cmd_list, g_strdup(s)); client->cmd_list = g_slist_prepend(client->cmd_list, g_strdup(s));
} }
static void
deferred_buffer_free(gpointer data, mpd_unused gpointer user_data)
{
struct deferred_buffer *buffer = data;
g_free(buffer);
}
static void client_close(struct client *client) static void client_close(struct client *client)
{ {
struct sllnode *buf;
assert(num_clients > 0); assert(num_clients > 0);
assert(!list_empty(&clients)); assert(!list_empty(&clients));
list_del(&client->siblings); list_del(&client->siblings);
@ -177,14 +186,8 @@ static void client_close(struct client *client)
client->cmd_list = NULL; client->cmd_list = NULL;
} }
if ((buf = client->deferred_send)) { g_queue_foreach(client->deferred_send, deferred_buffer_free, NULL);
do { g_queue_free(client->deferred_send);
struct sllnode *prev = buf;
buf = buf->next;
free(prev);
} while (buf);
client->deferred_send = NULL;
}
SECURE("client %i: closed\n", client->num); SECURE("client %i: closed\n", client->num);
free(client); free(client);
@ -430,7 +433,8 @@ static void client_manager_register_read_fd(fd_set * fds, int *fdmax)
addListenSocketsToFdSet(fds, fdmax); addListenSocketsToFdSet(fds, fdmax);
list_for_each_entry(client, &clients, siblings) { list_for_each_entry(client, &clients, siblings) {
if (!client_is_expired(client) && !client->deferred_send) { if (!client_is_expired(client) &&
g_queue_is_empty(client->deferred_send)) {
FD_SET(client->fd, fds); FD_SET(client->fd, fds);
if (*fdmax < client->fd) if (*fdmax < client->fd)
*fdmax = client->fd; *fdmax = client->fd;
@ -446,7 +450,7 @@ static void client_manager_register_write_fd(fd_set * fds, int *fdmax)
list_for_each_entry(client, &clients, siblings) { list_for_each_entry(client, &clients, siblings) {
if (client->fd >= 0 && !client_is_expired(client) if (client->fd >= 0 && !client_is_expired(client)
&& client->deferred_send) { && !g_queue_is_empty(client->deferred_send)) {
FD_SET(client->fd, fds); FD_SET(client->fd, fds);
if (*fdmax < client->fd) if (*fdmax < client->fd)
*fdmax = client->fd; *fdmax = client->fd;
@ -594,12 +598,14 @@ void client_manager_expire(void)
static void client_write_deferred(struct client *client) static void client_write_deferred(struct client *client)
{ {
struct sllnode *buf;
ssize_t ret = 0; ssize_t ret = 0;
buf = client->deferred_send; while (!g_queue_is_empty(client->deferred_send)) {
while (buf) { struct deferred_buffer *buf =
g_queue_peek_head(client->deferred_send);
assert(buf->size > 0); assert(buf->size > 0);
assert(buf->size <= client->deferred_bytes);
ret = write(client->fd, buf->data, buf->size); ret = write(client->fd, buf->data, buf->size);
if (ret < 0) if (ret < 0)
@ -607,22 +613,21 @@ static void client_write_deferred(struct client *client)
else if ((size_t)ret < buf->size) { else if ((size_t)ret < buf->size) {
assert(client->deferred_bytes >= (size_t)ret); assert(client->deferred_bytes >= (size_t)ret);
client->deferred_bytes -= ret; client->deferred_bytes -= ret;
buf->data = (char *)buf->data + ret;
buf->size -= ret; buf->size -= ret;
memmove(buf->data, buf->data + ret, buf->size);
} else { } else {
struct sllnode *tmp = buf; size_t decr = sizeof(*buf) -
size_t decr = (buf->size + sizeof(struct sllnode)); sizeof(buf->data) + buf->size;
assert(client->deferred_bytes >= decr); assert(client->deferred_bytes >= decr);
client->deferred_bytes -= decr; client->deferred_bytes -= decr;
buf = buf->next; free(buf);
free(tmp); g_queue_pop_head(client->deferred_send);
client->deferred_send = buf;
} }
client->lastTime = time(NULL); client->lastTime = time(NULL);
} }
if (!client->deferred_send) { if (g_queue_is_empty(client->deferred_send)) {
DEBUG("client %i: buffer empty %lu\n", client->num, DEBUG("client %i: buffer empty %lu\n", client->num,
(unsigned long)client->deferred_bytes); (unsigned long)client->deferred_bytes);
assert(client->deferred_bytes == 0); assert(client->deferred_bytes == 0);
@ -637,11 +642,13 @@ static void client_write_deferred(struct client *client)
static void client_defer_output(struct client *client, static void client_defer_output(struct client *client,
const void *data, size_t length) const void *data, size_t length)
{ {
struct sllnode **buf_r; size_t alloc;
struct deferred_buffer *buf;
assert(length > 0); assert(length > 0);
client->deferred_bytes += sizeof(struct sllnode) + length; alloc = sizeof(*buf) - sizeof(buf->data) + length;
client->deferred_bytes += alloc;
if (client->deferred_bytes > client_max_output_buffer_size) { if (client->deferred_bytes > client_max_output_buffer_size) {
ERROR("client %i: output buffer size (%lu) is " ERROR("client %i: output buffer size (%lu) is "
"larger than the max (%lu)\n", "larger than the max (%lu)\n",
@ -653,10 +660,11 @@ static void client_defer_output(struct client *client,
return; return;
} }
buf_r = &client->deferred_send; buf = g_malloc(alloc);
while (*buf_r != NULL) buf->size = length;
buf_r = &(*buf_r)->next; memcpy(buf->data, data, length);
*buf_r = new_sllnode(data, length);
g_queue_push_tail(client->deferred_send, buf);
} }
static void client_write_direct(struct client *client, static void client_write_direct(struct client *client,
@ -665,7 +673,7 @@ static void client_write_direct(struct client *client,
ssize_t ret; ssize_t ret;
assert(length > 0); assert(length > 0);
assert(client->deferred_send == NULL); assert(g_queue_is_empty(client->deferred_send));
if ((ret = write(client->fd, data, length)) < 0) { if ((ret = write(client->fd, data, length)) < 0) {
if (errno == EAGAIN || errno == EINTR) { if (errno == EAGAIN || errno == EINTR) {
@ -679,7 +687,7 @@ static void client_write_direct(struct client *client,
client_defer_output(client, data + ret, length - ret); client_defer_output(client, data + ret, length - ret);
} }
if (client->deferred_send) if (!g_queue_is_empty(client->deferred_send))
DEBUG("client %i: buffer created\n", client->num); DEBUG("client %i: buffer created\n", client->num);
} }
@ -688,7 +696,7 @@ static void client_write_output(struct client *client)
if (client_is_expired(client) || !client->send_buf_used) if (client_is_expired(client) || !client->send_buf_used)
return; return;
if (client->deferred_send != NULL) if (!g_queue_is_empty(client->deferred_send))
client_defer_output(client, client->send_buf, client_defer_output(client, client->send_buf,
client->send_buf_used); client->send_buf_used);
else else