transmission/gtk/tr-io.c

344 lines
8.1 KiB
C
Raw Normal View History

/******************************************************************************
* $Id$
*
2008-01-01 17:20:20 +00:00
* Copyright (c) 2006-2008 Transmission authors and contributors
*
* Permission is hereby granted, free of charge, to any person obtaining a
* copy of this software and associated documentation files (the "Software"),
* to deal in the Software without restriction, including without limitation
* the rights to use, copy, modify, merge, publish, distribute, sublicense,
* and/or sell copies of the Software, and to permit persons to whom the
* Software is furnished to do so, subject to the following conditions:
*
* The above copyright notice and this permission notice shall be included in
* all copies or substantial portions of the Software.
*
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
* AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
* LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
* FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
* DEALINGS IN THE SOFTWARE.
*****************************************************************************/
2006-07-16 19:39:23 +00:00
#include <sys/types.h>
#ifdef WIN32
#include <winsock2.h>
#include <ws2tcpip.h>
#include <io.h>
#define EWOULDBLOCK WSAEWOULDBLOCK
#define ECONNREFUSED WSAECONNREFUSED
#define ECONNABORTED WSAECONNABORTED
#else
2006-07-16 19:39:23 +00:00
#include <sys/socket.h>
#endif
2006-07-16 19:39:23 +00:00
#include <errno.h>
#include <string.h> /* memset, memmove */
#include <unistd.h> /* read, write */
2006-07-16 19:39:23 +00:00
#include <glib.h>
#include <libevent/evutil.h> /* evutil_make_socket_nonblocking */
2008-02-26 01:19:26 +00:00
#include "tr-io.h"
2006-07-16 19:39:23 +00:00
#include "util.h"
#define IO_BLOCKSIZE (1024)
2006-07-16 19:39:23 +00:00
struct iosource {
GSource source;
GPollFD infd;
GPollFD outfd;
ioidfunc_t sent;
iodatafunc_t received;
ionewfunc_t accepted;
iofunc_t closed;
void *cbdata;
char *inbuf;
size_t inused;
size_t inmax;
GSList *outbufs;
2006-07-16 19:39:23 +00:00
unsigned int lastid;
};
struct iooutbuf {
char *data;
size_t len;
size_t off;
2006-07-16 19:39:23 +00:00
unsigned int id;
};
static gboolean
2008-01-19 05:04:38 +00:00
io_prepare(GSource *source UNUSED, gint *timeout_) {
2006-07-16 19:39:23 +00:00
*timeout_ = -1;
return FALSE;
}
static gboolean
io_check(GSource *source) {
struct iosource *io = (struct iosource*)source;
if(io->infd.revents)
return TRUE;
if(NULL != io->outbufs && io->outfd.revents)
return TRUE;
else
return FALSE;
}
static void
io_disconnect(struct iosource *io, int err) {
if(NULL != io->closed) {
errno = err;
io->closed((GSource*)io, io->cbdata);
2006-07-16 19:39:23 +00:00
}
if(NULL != io->outbufs)
g_source_remove_poll((GSource*)io, &io->outfd);
g_source_remove_poll((GSource*)io, &io->infd);
g_source_remove(g_source_get_id((GSource*)io));
g_source_unref((GSource*)io);
2006-07-16 19:39:23 +00:00
}
static void
io_biggify(char **buf, size_t used, size_t *max) {
2006-07-16 19:39:23 +00:00
if(used + IO_BLOCKSIZE > *max) {
*max += IO_BLOCKSIZE;
*buf = g_renew(char, *buf, *max);
}
}
static void
io_read(struct iosource *io) {
ssize_t res = 0;
gboolean newdata = FALSE;
size_t used;
2006-07-16 19:39:23 +00:00
int err = 0;
g_source_ref((GSource*)io);
do {
if(!newdata && 0 < res)
newdata = TRUE;
io->inused += res;
io_biggify(&io->inbuf, io->inused, &io->inmax);
errno = 0;
res = read(io->infd.fd, io->inbuf + io->inused, io->inmax - io->inused);
if(0 > res)
err = errno;
} while(0 < res);
if(NULL == io->received)
io->inused = 0;
else if(newdata) {
used = io->received((GSource*)io, io->inbuf, io->inused, io->cbdata);
if(used > io->inused)
used = io->inused;
if(0 < used) {
if(used < io->inused)
memmove(io->inbuf, io->inbuf + used, io->inused - used);
io->inused -= used;
}
}
if(0 != err && EAGAIN != err)
io_disconnect(io, err);
else if(0 == res)
io_disconnect(io, 0);
g_source_unref((GSource*)io);
}
static void
io_accept(struct iosource *io) {
int fd;
socklen_t len;
len = io->inmax;
if(0 > (fd = accept(io->infd.fd, (struct sockaddr*)io->inbuf, &len))) {
if(EAGAIN == errno || ECONNABORTED == errno || EWOULDBLOCK == errno)
return;
io_disconnect(io, errno);
}
io->accepted((GSource*)io, fd, (struct sockaddr*)io->inbuf, len, io->cbdata);
}
static void
freeoutbuf(struct iooutbuf *buf) {
if(NULL != buf->data)
g_free(buf->data);
g_free(buf);
}
2006-07-16 19:39:23 +00:00
static void
io_write(struct iosource *io) {
struct iooutbuf *buf;
ssize_t res = 1;
int err = 0;
g_source_ref((GSource*)io);
while(NULL != io->outbufs && 0 == err) {
buf = io->outbufs->data;
while(buf->off < buf->len && 0 < res) {
errno = 0;
res = write(io->outfd.fd, buf->data + buf->off, buf->len - buf->off);
if(0 > res)
err = errno;
else
buf->off += res;
}
if(buf->off >= buf->len) {
io->outbufs = g_slist_remove(io->outbufs, buf);
2006-07-16 19:39:23 +00:00
if(NULL == io->outbufs)
g_source_remove_poll((GSource*)io, &io->outfd);
if(NULL != io->sent)
io->sent((GSource*)io, buf->id, io->cbdata);
freeoutbuf(buf);
}
}
if(0 != err && EAGAIN != err)
io_disconnect(io, err);
g_source_unref((GSource*)io);
}
static gboolean
io_dispatch(GSource *source, GSourceFunc callback UNUSED,
gpointer gdata UNUSED) {
struct iosource *io = (struct iosource*)source;
if(io->infd.revents & (G_IO_ERR | G_IO_HUP) ||
io->outfd.revents & G_IO_ERR)
io_disconnect(io, 0 /* XXX how do I get errors here? */ );
else if(io->infd.revents & G_IO_IN)
(NULL == io->accepted ? io_read : io_accept)(io);
else if(io->outfd.revents & G_IO_OUT)
io_write(io);
else
return FALSE;
return TRUE;
}
2006-07-16 19:39:23 +00:00
static void
io_finalize(GSource *source UNUSED) {
struct iosource *io = (struct iosource*)source;
g_slist_foreach(io->outbufs, (GFunc)freeoutbuf, NULL);
g_slist_free(io->outbufs);
g_free(io->inbuf);
2006-07-16 19:39:23 +00:00
io->outbufs = NULL;
io->inbuf = NULL;
}
static GSourceFuncs sourcefuncs = {
io_prepare,
io_check,
io_dispatch,
io_finalize,
NULL,
NULL
};
static struct iosource *
newsource(void) {
GSource *source = g_source_new(&sourcefuncs, sizeof(struct iosource));
struct iosource *io = (struct iosource*)source;
io->sent = NULL;
io->received = NULL;
io->accepted = NULL;
io->closed = NULL;
io->cbdata = NULL;
memset(&io->infd, 0, sizeof(io->infd));
io->infd.fd = -1;
memset(&io->outfd, 0, sizeof(io->outfd));
io->outfd.fd = -1;
io->inbuf = NULL;
io->inused = 0;
io->inmax = 0;
io->outbufs = NULL;
io->lastid = 0;
return io;
}
GSource *
io_new(int fd, ioidfunc_t sent, iodatafunc_t received,
iofunc_t closed, void *cbdata) {
struct iosource *io;
if( evutil_make_socket_nonblocking( fd ) )
return NULL;
io = newsource();
io->sent = sent;
io->received = received;
io->closed = closed;
io->cbdata = cbdata;
io->infd.fd = fd;
io->infd.events = G_IO_IN | G_IO_HUP | G_IO_ERR;
io->infd.revents = 0;
io->outfd.fd = fd;
io->outfd.events = G_IO_OUT | G_IO_ERR;
io->outfd.revents = 0;
g_source_add_poll((GSource*)io, &io->infd);
g_source_attach((GSource*)io, NULL);
return (GSource*)io;
}
GSource *
io_new_listening(int fd, socklen_t len, ionewfunc_t accepted,
iofunc_t closed, void *cbdata) {
struct iosource *io;
g_assert(NULL != accepted);
if( evutil_make_socket_nonblocking( fd ) )
return NULL;
io = newsource();
io->accepted = accepted;
io->closed = closed;
io->cbdata = cbdata;
io->infd.fd = fd;
io->infd.events = G_IO_IN | G_IO_ERR;
io->infd.revents = 0;
io->inbuf = g_new(char, len);
io->inmax = len;
g_source_add_poll((GSource*)io, &io->infd);
g_source_attach((GSource*)io, NULL);
return (GSource*)io;
}
unsigned int
io_send_keepdata(GSource *source, void *data, size_t len) {
struct iosource *io = (struct iosource*)source;
struct iooutbuf *buf = g_new(struct iooutbuf, 1);
buf->data = data;
buf->len = len;
buf->off = 0;
io->lastid++;
buf->id = io->lastid;
2006-07-16 19:39:23 +00:00
if(NULL != io->outbufs)
io->outbufs = g_slist_append(io->outbufs, buf);
else {
io->outbufs = g_slist_append(io->outbufs, buf);
g_source_add_poll(source, &io->outfd);
}
2006-07-16 19:39:23 +00:00
return io->lastid;
2006-07-16 19:39:23 +00:00
}