mirror
/
transmission
mirror of https://github.com/transmission/transmission
1
0
Fork 0
Code Issues Releases Wiki Activity

(gtk) group the ipc functions by call order

This commit is contained in:
Charles Kerr 2008-02-26 21:05:13 +00:00
parent 6728ff934f
commit e960eaf7b2
2 changed files with 548 additions and 636 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

815
gtk/ipc.c
View File

@ -90,125 +90,210 @@ struct constate
} u; } u;
}; };
static void
serv_bind(struct constate *con);
static void
rmsock(void);
static gboolean
client_connect(char *path, struct constate *con);
static void
srv_io_accept(GSource *source, int fd, struct sockaddr *sa, socklen_t len,
void *vdata);
static size_t
srv_io_received( GSource * source, void * data, size_t len, void * vdata );
static size_t
cli_io_received( GSource * source, void * data, size_t len, void * vdata );
static void
client_sendmsg( struct constate * con );
static void
destroycon(struct constate *con);
static void
all_io_closed(GSource *source, void *vdata);
static void
cli_io_sent( GSource * source, size_t id, void * vdata );
static void
smsg_add( enum ipc_msg id, benc_val_t * val, int64_t tag, void * arg );
static void
smsg_addone( enum ipc_msg id, benc_val_t * val, int64_t tag, void * arg );
static void
smsg_quit( enum ipc_msg id, benc_val_t * val, int64_t tag, void * arg );
static void
smsg_noop( enum ipc_msg id, benc_val_t * val, int64_t tag, void * arg );
static void
smsg_info( enum ipc_msg id, benc_val_t * val, int64_t tag, void * arg );
static void
smsg_infoall( enum ipc_msg id, benc_val_t * val, int64_t tag, void * arg );
static int
addinfo( TrTorrent * tor, enum ipc_msg msgid, int torid, int types,
benc_val_t * val );
static void
smsg_look( enum ipc_msg id, benc_val_t * val, int64_t tag, void * arg );
static void
smsg_tor( enum ipc_msg id, benc_val_t * val, int64_t tag, void * arg );
static void
smsg_torall( enum ipc_msg id, benc_val_t * val, int64_t tag, void * arg );
static void
smsg_pref( enum ipc_msg id, benc_val_t * val, int64_t tag, void * arg );
static void
smsg_int( enum ipc_msg id, benc_val_t * val, int64_t tag, void * arg );
static void
smsg_str( enum ipc_msg id, benc_val_t * val, int64_t tag, void * arg );
static void
smsg_sup( enum ipc_msg id, benc_val_t * val, int64_t tag, void * arg );
static void
all_default( enum ipc_msg id, benc_val_t * val, int64_t tag, void * arg );
static gboolean
simpleresp( struct constate * con, int64_t tag, enum ipc_msg id );
static TrTorrent *
findtorid( TrCore * core, int id, GtkTreeIter * iter );
static TrTorrent *
findtorhash( TrCore * core, const char * hash, int * id );
/* this is only used on the server */ /* this is only used on the server */
static char *gl_sockpath = NULL; static char *gl_sockpath = NULL;
void static gboolean
ipc_socket_setup( GtkWindow * parent, TrCore * core ) simpleresp( struct constate * con, int64_t tag, enum ipc_msg id )
{ {
struct constate *con; uint8_t * buf;
size_t size;
con = g_new0(struct constate, 1); buf = ipc_mkempty( con->ipc, &size, id, tag );
con->source = NULL; if( NULL == buf )
con->fd = -1;
con->type = CON_SERV;
con->msgs = ipc_initmsgs();
if( NULL == con->msgs ||
0 > ipc_addmsg( con->msgs, IPC_MSG_ADDMANYFILES, smsg_add ) ||
0 > ipc_addmsg( con->msgs, IPC_MSG_ADDONEFILE, smsg_addone ) ||
0 > ipc_addmsg( con->msgs, IPC_MSG_AUTOMAP, smsg_int ) ||
0 > ipc_addmsg( con->msgs, IPC_MSG_AUTOSTART, smsg_int ) ||
0 > ipc_addmsg( con->msgs, IPC_MSG_DIR, smsg_str ) ||
0 > ipc_addmsg( con->msgs, IPC_MSG_DOWNLIMIT, smsg_int ) ||
0 > ipc_addmsg( con->msgs, IPC_MSG_GETAUTOMAP, smsg_pref ) ||
0 > ipc_addmsg( con->msgs, IPC_MSG_GETAUTOSTART, smsg_pref ) ||
0 > ipc_addmsg( con->msgs, IPC_MSG_GETDIR, smsg_pref ) ||
0 > ipc_addmsg( con->msgs, IPC_MSG_GETDOWNLIMIT, smsg_pref ) ||
0 > ipc_addmsg( con->msgs, IPC_MSG_GETINFO, smsg_info ) ||
0 > ipc_addmsg( con->msgs, IPC_MSG_GETINFOALL, smsg_infoall ) ||
0 > ipc_addmsg( con->msgs, IPC_MSG_GETPEX, smsg_pref ) ||
0 > ipc_addmsg( con->msgs, IPC_MSG_GETPORT, smsg_pref ) ||
0 > ipc_addmsg( con->msgs, IPC_MSG_GETSTAT, smsg_info ) ||
0 > ipc_addmsg( con->msgs, IPC_MSG_GETSTATALL, smsg_infoall ) ||
0 > ipc_addmsg( con->msgs, IPC_MSG_GETUPLIMIT, smsg_pref ) ||
0 > ipc_addmsg( con->msgs, IPC_MSG_LOOKUP, smsg_look ) ||
0 > ipc_addmsg( con->msgs, IPC_MSG_NOOP, smsg_noop ) ||
0 > ipc_addmsg( con->msgs, IPC_MSG_PEX, smsg_int ) ||
0 > ipc_addmsg( con->msgs, IPC_MSG_PORT, smsg_int ) ||
0 > ipc_addmsg( con->msgs, IPC_MSG_QUIT, smsg_quit ) ||
0 > ipc_addmsg( con->msgs, IPC_MSG_REMOVE, smsg_tor ) ||
0 > ipc_addmsg( con->msgs, IPC_MSG_REMOVEALL, smsg_torall ) ||
0 > ipc_addmsg( con->msgs, IPC_MSG_START, smsg_tor ) ||
0 > ipc_addmsg( con->msgs, IPC_MSG_STARTALL, smsg_torall ) ||
0 > ipc_addmsg( con->msgs, IPC_MSG_STOP, smsg_tor ) ||
0 > ipc_addmsg( con->msgs, IPC_MSG_STOPALL, smsg_torall ) ||
0 > ipc_addmsg( con->msgs, IPC_MSG_SUP, smsg_sup ) ||
0 > ipc_addmsg( con->msgs, IPC_MSG_UPLIMIT, smsg_int ) )
{ {
errmsg( con->u.serv.wind, _("Failed to set up IPC: %s"), return FALSE;
g_strerror( errno ) ); }
g_free( con );
io_send_keepdata( con->source, buf, size );
return TRUE;
}
static void
all_default( enum ipc_msg id, benc_val_t * val UNUSED, int64_t tag, void * arg )
{
switch( id )
{
case IPC_MSG_FAIL:
case IPC_MSG_NOTSUP:
case IPC_MSG_BAD:
case IPC_MSG_OK:
break;
default:
simpleresp( arg, tag, IPC_MSG_NOTSUP );
break;
}
}
static void
destroycon(struct constate *con) {
con->source = NULL;
if(0 <= con->fd)
EVUTIL_CLOSESOCKET(con->fd);
con->fd = -1;
ipc_freecon( con->ipc );
switch(con->type) {
case CON_SERV:
break;
case CON_CLIENT:
ipc_freemsgs( con->msgs );
freestrlist(con->u.client.files);
g_main_loop_quit(con->u.client.loop);
break;
}
}
static void
cli_io_sent( GSource * source UNUSED, size_t id, void *vdata )
{
struct constate_client *cli = &((struct constate*)vdata)->u.client;
if(0 < id && cli->msgid == id) {
*(cli->succeeded) = TRUE;
destroycon(vdata);
}
}
static void
client_sendmsg( struct constate * con )
{
struct constate_client * cli = &con->u.client;
GList * ii;
uint8_t * buf;
size_t size;
benc_val_t packet, * val;
int saved;
switch( cli->msg )
{
case IPC_MSG_ADDMANYFILES:
val = ipc_initval( con->ipc, cli->msg, -1, &packet, TYPE_LIST );
if( NULL == val ||
tr_bencListReserve( val, g_list_length( cli->files ) ) )
{
perror( "malloc" );
destroycon( con );
return;
}
for( ii = cli->files; NULL != ii; ii = ii->next )
{
tr_bencInitStr( tr_bencListAdd( val ), ii->data, -1, 0 );
}
buf = ipc_mkval( &packet, &size );
saved = errno;
tr_bencFree( &packet );
g_list_free( cli->files );
cli->files = NULL;
break;
case IPC_MSG_QUIT:
buf = ipc_mkempty( con->ipc, &size, cli->msg, -1 );
saved = errno;
break;
default:
g_assert_not_reached();
return; return;
} }
ipc_setdefmsg( con->msgs, all_default ); if( NULL == buf )
{
errno = saved;
perror( "malloc" );
destroycon( con );
return;
}
con->u.serv.wind = parent; cli->msgid = io_send_keepdata( con->source, buf, size );
con->u.serv.core = core; }
g_object_add_weak_pointer( G_OBJECT( core ), &con->u.serv.core ); static size_t
cli_io_received( GSource * source UNUSED, void * data, size_t len,
void * vdata )
{
struct constate * con = vdata;
struct constate_client * cli = &con->u.client;
ssize_t res;
serv_bind(con); if( IPC_MIN_MSG_LEN > len )
{
return 0;
}
res = ipc_parse( con->ipc, data, len, con );
if( 0 > res )
{
switch( errno )
{
case EPERM:
g_message( _("Bad IPC protocol version") );
break;
case EINVAL:
g_message( _("IPC protocol parse error") );
break;
default:
g_message( _("IPC parsing failed: %s"), g_strerror( errno ) );
break;
}
destroycon( con );
return 0;
}
if( HASVERS( con->ipc ) && 0 == cli->msgid )
{
client_sendmsg( con );
}
return res;
}
static void
all_io_closed(GSource *source UNUSED, void *vdata) {
struct constate *con = vdata;
destroycon(con);
}
static gboolean
client_connect(char *path, struct constate *con) {
struct sockaddr_un addr;
uint8_t * buf;
size_t size;
if(0 > (con->fd = socket(AF_UNIX, SOCK_STREAM, 0))) {
g_message( _("Failed to create socket: %s"), g_strerror(errno));
return FALSE;
}
memset(&addr, 0, sizeof(addr));
addr.sun_family = AF_UNIX;
strncpy(addr.sun_path, path, sizeof(addr.sun_path) - 1);
if(0 > connect(con->fd, (struct sockaddr*)&addr, SUN_LEN(&addr))) {
g_message( _("Failed to connect to %s: %s"), path, g_strerror(errno));
return FALSE;
}
con->source = io_new(con->fd, cli_io_sent, cli_io_received,
all_io_closed, con);
if( NULL == con->source )
{
EVUTIL_CLOSESOCKET( con->fd );
return FALSE;
}
buf = ipc_mkvers( &size, "Transmission GTK+ " LONG_VERSION_STRING );
if( NULL == buf )
{
EVUTIL_CLOSESOCKET( con->fd );
return FALSE;
}
io_send_keepdata( con->source, buf, size );
return TRUE;
} }
static gboolean static gboolean
@ -272,49 +357,6 @@ ipc_sendquit_blocking( void )
return blocking_client( IPC_MSG_QUIT, NULL ); return blocking_client( IPC_MSG_QUIT, NULL );
} }
/* open a local socket for clients connections */
static void
serv_bind(struct constate *con) {
struct sockaddr_un sa;
rmsock();
gl_sockpath = cf_sockname();
if(0 > (con->fd = socket(AF_LOCAL, SOCK_STREAM, 0)))
goto fail;
memset(&sa, 0, sizeof(sa));
sa.sun_family = AF_LOCAL;
strncpy(sa.sun_path, gl_sockpath, sizeof(sa.sun_path) - 1);
/* unlink any existing socket file before trying to create ours */
unlink(gl_sockpath);
if(0 > bind(con->fd, (struct sockaddr *)&sa, SUN_LEN(&sa))) {
/* bind may fail if there was already a socket, so try twice */
unlink(gl_sockpath);
if(0 > bind(con->fd, (struct sockaddr *)&sa, SUN_LEN(&sa)))
goto fail;
}
if(0 > listen(con->fd, 5))
goto fail;
con->source = io_new_listening(con->fd, sizeof(struct sockaddr_un),
srv_io_accept, all_io_closed, con);
g_atexit(rmsock);
return;
fail:
errmsg(con->u.serv.wind, _("Failed to set up socket: %s"),
g_strerror(errno));
if(0 <= con->fd)
EVUTIL_CLOSESOCKET(con->fd);
con->fd = -1;
rmsock();
}
static void static void
rmsock(void) { rmsock(void) {
if(NULL != gl_sockpath) { if(NULL != gl_sockpath) {
@ -323,87 +365,6 @@ rmsock(void) {
} }
} }
static gboolean
client_connect(char *path, struct constate *con) {
struct sockaddr_un addr;
uint8_t * buf;
size_t size;
if(0 > (con->fd = socket(AF_UNIX, SOCK_STREAM, 0))) {
g_message( _("Failed to create socket: %s"), g_strerror(errno));
return FALSE;
}
memset(&addr, 0, sizeof(addr));
addr.sun_family = AF_UNIX;
strncpy(addr.sun_path, path, sizeof(addr.sun_path) - 1);
if(0 > connect(con->fd, (struct sockaddr*)&addr, SUN_LEN(&addr))) {
g_message( _("Failed to connect to %s: %s"), path, g_strerror(errno));
return FALSE;
}
con->source = io_new(con->fd, cli_io_sent, cli_io_received,
all_io_closed, con);
if( NULL == con->source )
{
EVUTIL_CLOSESOCKET( con->fd );
return FALSE;
}
buf = ipc_mkvers( &size, "Transmission GTK+ " LONG_VERSION_STRING );
if( NULL == buf )
{
EVUTIL_CLOSESOCKET( con->fd );
return FALSE;
}
io_send_keepdata( con->source, buf, size );
return TRUE;
}
static void
srv_io_accept(GSource *source UNUSED, int fd, struct sockaddr *sa UNUSED,
socklen_t len UNUSED, void *vdata) {
struct constate *con = vdata;
struct constate *newcon;
uint8_t * buf;
size_t size;
newcon = g_new(struct constate, 1);
memcpy(newcon, con, sizeof(*newcon));
newcon->fd = fd;
newcon->ipc = ipc_newcon( con->msgs );
if( NULL == newcon->ipc )
{
g_free( newcon );
EVUTIL_CLOSESOCKET( fd );
return;
}
newcon->source = io_new(fd, NULL, srv_io_received, all_io_closed, newcon);
if( NULL == newcon->source )
{
ipc_freecon( newcon->ipc );
g_free( newcon );
EVUTIL_CLOSESOCKET( fd );
return;
}
buf = ipc_mkvers( &size, "Transmission GTK+ " LONG_VERSION_STRING );
if( NULL == buf )
{
ipc_freecon( newcon->ipc );
g_free( newcon );
EVUTIL_CLOSESOCKET( fd );
return;
}
io_send_keepdata( newcon->source, buf, size );
}
static size_t static size_t
srv_io_received( GSource * source UNUSED, void * data, size_t len, srv_io_received( GSource * source UNUSED, void * data, size_t len,
void * vdata ) void * vdata )
@ -445,134 +406,88 @@ srv_io_received( GSource * source UNUSED, void * data, size_t len,
return res; return res;
} }
static size_t
cli_io_received( GSource * source UNUSED, void * data, size_t len,
void * vdata )
{
struct constate * con = vdata;
struct constate_client * cli = &con->u.client;
ssize_t res;
if( IPC_MIN_MSG_LEN > len )
{
return 0;
}
res = ipc_parse( con->ipc, data, len, con );
if( 0 > res )
{
switch( errno )
{
case EPERM:
g_message( _("Bad IPC protocol version") );
break;
case EINVAL:
g_message( _("IPC protocol parse error") );
break;
default:
g_message( _("IPC parsing failed: %s"), g_strerror( errno ) );
break;
}
destroycon( con );
return 0;
}
if( HASVERS( con->ipc ) && 0 == cli->msgid )
{
client_sendmsg( con );
}
return res;
}
static void static void
client_sendmsg( struct constate * con ) srv_io_accept(GSource *source UNUSED, int fd, struct sockaddr *sa UNUSED,
{ socklen_t len UNUSED, void *vdata) {
struct constate_client * cli = &con->u.client; struct constate *con = vdata;
GList * ii; struct constate *newcon;
uint8_t * buf; uint8_t * buf;
size_t size; size_t size;
benc_val_t packet, * val;
int saved;
switch( cli->msg ) newcon = g_new(struct constate, 1);
memcpy(newcon, con, sizeof(*newcon));
newcon->fd = fd;
newcon->ipc = ipc_newcon( con->msgs );
if( NULL == newcon->ipc )
{ {
case IPC_MSG_ADDMANYFILES: g_free( newcon );
val = ipc_initval( con->ipc, cli->msg, -1, &packet, TYPE_LIST ); EVUTIL_CLOSESOCKET( fd );
if( NULL == val ||
tr_bencListReserve( val, g_list_length( cli->files ) ) )
{
perror( "malloc" );
destroycon( con );
return;
}
for( ii = cli->files; NULL != ii; ii = ii->next )
{
tr_bencInitStr( tr_bencListAdd( val ), ii->data, -1, 0 );
}
buf = ipc_mkval( &packet, &size );
saved = errno;
tr_bencFree( &packet );
g_list_free( cli->files );
cli->files = NULL;
break;
case IPC_MSG_QUIT:
buf = ipc_mkempty( con->ipc, &size, cli->msg, -1 );
saved = errno;
break;
default:
g_assert_not_reached();
return; return;
} }
newcon->source = io_new(fd, NULL, srv_io_received, all_io_closed, newcon);
if( NULL == newcon->source )
{
ipc_freecon( newcon->ipc );
g_free( newcon );
EVUTIL_CLOSESOCKET( fd );
return;
}
buf = ipc_mkvers( &size, "Transmission GTK+ " LONG_VERSION_STRING );
if( NULL == buf ) if( NULL == buf )
{ {
errno = saved; ipc_freecon( newcon->ipc );
perror( "malloc" ); g_free( newcon );
destroycon( con ); EVUTIL_CLOSESOCKET( fd );
return; return;
} }
cli->msgid = io_send_keepdata( con->source, buf, size ); io_send_keepdata( newcon->source, buf, size );
} }
/* open a local socket for clients connections */
static void static void
destroycon(struct constate *con) { serv_bind(struct constate *con) {
con->source = NULL; struct sockaddr_un sa;
rmsock();
gl_sockpath = cf_sockname();
if(0 > (con->fd = socket(AF_LOCAL, SOCK_STREAM, 0)))
goto fail;
memset(&sa, 0, sizeof(sa));
sa.sun_family = AF_LOCAL;
strncpy(sa.sun_path, gl_sockpath, sizeof(sa.sun_path) - 1);
/* unlink any existing socket file before trying to create ours */
unlink(gl_sockpath);
if(0 > bind(con->fd, (struct sockaddr *)&sa, SUN_LEN(&sa))) {
/* bind may fail if there was already a socket, so try twice */
unlink(gl_sockpath);
if(0 > bind(con->fd, (struct sockaddr *)&sa, SUN_LEN(&sa)))
goto fail;
}
if(0 > listen(con->fd, 5))
goto fail;
con->source = io_new_listening(con->fd, sizeof(struct sockaddr_un),
srv_io_accept, all_io_closed, con);
g_atexit(rmsock);
return;
fail:
errmsg(con->u.serv.wind, _("Failed to set up socket: %s"),
g_strerror(errno));
if(0 <= con->fd) if(0 <= con->fd)
EVUTIL_CLOSESOCKET(con->fd); EVUTIL_CLOSESOCKET(con->fd);
con->fd = -1; con->fd = -1;
ipc_freecon( con->ipc ); rmsock();
switch(con->type) {
case CON_SERV:
break;
case CON_CLIENT:
ipc_freemsgs( con->msgs );
freestrlist(con->u.client.files);
g_main_loop_quit(con->u.client.loop);
break;
}
}
static void
all_io_closed(GSource *source UNUSED, void *vdata) {
struct constate *con = vdata;
destroycon(con);
}
static void
cli_io_sent( GSource * source UNUSED, size_t id, void *vdata )
{
struct constate_client *cli = &((struct constate*)vdata)->u.client;
if(0 < id && cli->msgid == id) {
*(cli->succeeded) = TRUE;
destroycon(vdata);
}
} }
static void static void
@ -677,6 +592,54 @@ smsg_noop( enum ipc_msg id UNUSED, benc_val_t * val UNUSED, int64_t tag,
simpleresp( arg, tag, IPC_MSG_OK ); simpleresp( arg, tag, IPC_MSG_OK );
} }
static TrTorrent *
findtorid( TrCore * core, int id, GtkTreeIter * iter )
{
GtkTreeModel * model;
GtkTreeIter myiter;
int rowid;
TrTorrent * tor;
if( NULL == iter )
{
iter = &myiter;
}
model = tr_core_model( core );
if( gtk_tree_model_get_iter_first( model, iter ) )
{
do
{
gtk_tree_model_get( model, iter, MC_ID, &rowid, -1 );
if( rowid == id )
{
gtk_tree_model_get( model, iter, MC_TORRENT, &tor, -1 );
g_object_unref( tor );
return tor;
}
}
while( gtk_tree_model_iter_next( model, iter ) );
}
return NULL;
}
static int
addinfo( TrTorrent * tor, enum ipc_msg msgid, int torid, int types,
benc_val_t * val )
{
if( IPC_MSG_INFO == msgid )
{
const tr_info * inf = tr_torrent_info( tor );
return ipc_addinfo( val, torid, inf, types );
}
else
{
const tr_stat * st = tr_torrent_stat( tor );
return ipc_addstat( val, torid, st, types );
}
}
static void static void
smsg_info( enum ipc_msg id, benc_val_t * val, int64_t tag, void * arg ) smsg_info( enum ipc_msg id, benc_val_t * val, int64_t tag, void * arg )
{ {
@ -801,20 +764,32 @@ smsg_infoall( enum ipc_msg id, benc_val_t * val, int64_t tag, void * arg )
} }
} }
static int static TrTorrent *
addinfo( TrTorrent * tor, enum ipc_msg msgid, int torid, int types, findtorhash( TrCore * core, const char * hash, int * torid )
benc_val_t * val )
{ {
if( IPC_MSG_INFO == msgid ) GtkTreeModel * model;
GtkTreeIter iter;
char * rowhash;
TrTorrent * tor;
model = tr_core_model( core );
if( gtk_tree_model_get_iter_first( model, &iter ) )
{ {
const tr_info * inf = tr_torrent_info( tor ); do
return ipc_addinfo( val, torid, inf, types );
}
else
{ {
const tr_stat * st = tr_torrent_stat( tor ); gtk_tree_model_get( model, &iter, MC_HASH, &rowhash, -1 );
return ipc_addstat( val, torid, st, types ); if( 0 == strcmp( hash, rowhash ) )
{
gtk_tree_model_get( model, &iter, MC_ID, torid,
MC_TORRENT, &tor, -1 );
g_object_unref( tor );
return tor;
} }
}
while( gtk_tree_model_iter_next( model, &iter ) );
}
return NULL;
} }
static void static void
@ -1157,95 +1132,61 @@ smsg_sup( enum ipc_msg id UNUSED, benc_val_t * val, int64_t tag, void * arg )
} }
} }
static void void
all_default( enum ipc_msg id, benc_val_t * val UNUSED, int64_t tag, void * arg ) ipc_socket_setup( GtkWindow * parent, TrCore * core )
{ {
switch( id ) struct constate *con;
con = g_new0(struct constate, 1);
con->source = NULL;
con->fd = -1;
con->type = CON_SERV;
con->msgs = ipc_initmsgs();
if( NULL == con->msgs ||
0 > ipc_addmsg( con->msgs, IPC_MSG_ADDMANYFILES, smsg_add ) ||
0 > ipc_addmsg( con->msgs, IPC_MSG_ADDONEFILE, smsg_addone ) ||
0 > ipc_addmsg( con->msgs, IPC_MSG_AUTOMAP, smsg_int ) ||
0 > ipc_addmsg( con->msgs, IPC_MSG_AUTOSTART, smsg_int ) ||
0 > ipc_addmsg( con->msgs, IPC_MSG_DIR, smsg_str ) ||
0 > ipc_addmsg( con->msgs, IPC_MSG_DOWNLIMIT, smsg_int ) ||
0 > ipc_addmsg( con->msgs, IPC_MSG_GETAUTOMAP, smsg_pref ) ||
0 > ipc_addmsg( con->msgs, IPC_MSG_GETAUTOSTART, smsg_pref ) ||
0 > ipc_addmsg( con->msgs, IPC_MSG_GETDIR, smsg_pref ) ||
0 > ipc_addmsg( con->msgs, IPC_MSG_GETDOWNLIMIT, smsg_pref ) ||
0 > ipc_addmsg( con->msgs, IPC_MSG_GETINFO, smsg_info ) ||
0 > ipc_addmsg( con->msgs, IPC_MSG_GETINFOALL, smsg_infoall ) ||
0 > ipc_addmsg( con->msgs, IPC_MSG_GETPEX, smsg_pref ) ||
0 > ipc_addmsg( con->msgs, IPC_MSG_GETPORT, smsg_pref ) ||
0 > ipc_addmsg( con->msgs, IPC_MSG_GETSTAT, smsg_info ) ||
0 > ipc_addmsg( con->msgs, IPC_MSG_GETSTATALL, smsg_infoall ) ||
0 > ipc_addmsg( con->msgs, IPC_MSG_GETUPLIMIT, smsg_pref ) ||
0 > ipc_addmsg( con->msgs, IPC_MSG_LOOKUP, smsg_look ) ||
0 > ipc_addmsg( con->msgs, IPC_MSG_NOOP, smsg_noop ) ||
0 > ipc_addmsg( con->msgs, IPC_MSG_PEX, smsg_int ) ||
0 > ipc_addmsg( con->msgs, IPC_MSG_PORT, smsg_int ) ||
0 > ipc_addmsg( con->msgs, IPC_MSG_QUIT, smsg_quit ) ||
0 > ipc_addmsg( con->msgs, IPC_MSG_REMOVE, smsg_tor ) ||
0 > ipc_addmsg( con->msgs, IPC_MSG_REMOVEALL, smsg_torall ) ||
0 > ipc_addmsg( con->msgs, IPC_MSG_START, smsg_tor ) ||
0 > ipc_addmsg( con->msgs, IPC_MSG_STARTALL, smsg_torall ) ||
0 > ipc_addmsg( con->msgs, IPC_MSG_STOP, smsg_tor ) ||
0 > ipc_addmsg( con->msgs, IPC_MSG_STOPALL, smsg_torall ) ||
0 > ipc_addmsg( con->msgs, IPC_MSG_SUP, smsg_sup ) ||
0 > ipc_addmsg( con->msgs, IPC_MSG_UPLIMIT, smsg_int ) )
{ {
case IPC_MSG_FAIL: errmsg( con->u.serv.wind, _("Failed to set up IPC: %s"),
case IPC_MSG_NOTSUP: g_strerror( errno ) );
case IPC_MSG_BAD: g_free( con );
case IPC_MSG_OK: return;
break;
default:
simpleresp( arg, tag, IPC_MSG_NOTSUP );
break;
} }
}
ipc_setdefmsg( con->msgs, all_default );
static gboolean
simpleresp( struct constate * con, int64_t tag, enum ipc_msg id ) con->u.serv.wind = parent;
{ con->u.serv.core = core;
uint8_t * buf;
size_t size; g_object_add_weak_pointer( G_OBJECT( core ), &con->u.serv.core );
buf = ipc_mkempty( con->ipc, &size, id, tag ); serv_bind(con);
if( NULL == buf )
{
return FALSE;
}
io_send_keepdata( con->source, buf, size );
return TRUE;
}
static TrTorrent *
findtorid( TrCore * core, int id, GtkTreeIter * iter )
{
GtkTreeModel * model;
GtkTreeIter myiter;
int rowid;
TrTorrent * tor;
if( NULL == iter )
{
iter = &myiter;
}
model = tr_core_model( core );
if( gtk_tree_model_get_iter_first( model, iter ) )
{
do
{
gtk_tree_model_get( model, iter, MC_ID, &rowid, -1 );
if( rowid == id )
{
gtk_tree_model_get( model, iter, MC_TORRENT, &tor, -1 );
g_object_unref( tor );
return tor;
}
}
while( gtk_tree_model_iter_next( model, iter ) );
}
return NULL;
}
static TrTorrent *
findtorhash( TrCore * core, const char * hash, int * torid )
{
GtkTreeModel * model;
GtkTreeIter iter;
char * rowhash;
TrTorrent * tor;
model = tr_core_model( core );
if( gtk_tree_model_get_iter_first( model, &iter ) )
{
do
{
gtk_tree_model_get( model, &iter, MC_HASH, &rowhash, -1 );
if( 0 == strcmp( hash, rowhash ) )
{
gtk_tree_model_get( model, &iter, MC_ID, torid,
MC_TORRENT, &tor, -1 );
g_object_unref( tor );
return tor;
}
}
while( gtk_tree_model_iter_next( model, &iter ) );
}
return NULL;
} }

355
gtk/tr-io.c
View File

@ -60,146 +60,6 @@ struct iooutbuf {
unsigned int id; unsigned int id;
}; };
static struct iosource *
newsource(void);
static void
freeoutbuf(struct iooutbuf *buf);
static gboolean
io_prepare(GSource *source, gint *timeout_);
static gboolean
io_check(GSource *source);
static gboolean
io_dispatch(GSource *source, GSourceFunc callback, gpointer gdata);
static void
io_finalize(GSource *source);
static void
io_accept(struct iosource *io);
static void
io_read(struct iosource *io);
static void
io_write(struct iosource *io);
static void
io_disconnect(struct iosource *io, int err);
static GSourceFuncs sourcefuncs = {
io_prepare,
io_check,
io_dispatch,
io_finalize,
NULL,
NULL
};
static int
nonblock(int fd)
{
const int err = evutil_make_socket_nonblocking( fd );
return err;
}
GSource *
io_new(int fd, ioidfunc_t sent, iodatafunc_t received,
iofunc_t closed, void *cbdata) {
struct iosource *io;
if( nonblock( 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( nonblock( 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;
}
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;
}
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;
if(NULL != io->outbufs)
io->outbufs = g_list_append(io->outbufs, buf);
else {
io->outbufs = g_list_append(io->outbufs, buf);
g_source_add_poll(source, &io->outfd);
}
return io->lastid;
}
static void
freeoutbuf(struct iooutbuf *buf) {
if(NULL != buf->data)
g_free(buf->data);
g_free(buf);
}
static gboolean static gboolean
io_prepare(GSource *source UNUSED, gint *timeout_) { io_prepare(GSource *source UNUSED, gint *timeout_) {
*timeout_ = -1; *timeout_ = -1;
@ -218,36 +78,19 @@ io_check(GSource *source) {
return FALSE; return FALSE;
} }
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;
}
static void static void
io_finalize(GSource *source UNUSED) { io_disconnect(struct iosource *io, int err) {
struct iosource *io = (struct iosource*)source; if(NULL != io->closed) {
errno = err;
if(NULL != io->outbufs) { io->closed((GSource*)io, io->cbdata);
g_list_foreach(io->outbufs, (GFunc)freeoutbuf, NULL);
g_list_free(io->outbufs);
} }
if(NULL != io->inbuf) if(NULL != io->outbufs)
g_free(io->inbuf); 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);
} }
static void static void
@ -258,21 +101,6 @@ io_biggify(char **buf, size_t used, size_t *max) {
} }
} }
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 static void
io_read(struct iosource *io) { io_read(struct iosource *io) {
ssize_t res = 0; ssize_t res = 0;
@ -313,6 +141,28 @@ io_read(struct iosource *io) {
g_source_unref((GSource*)io); 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);
}
static void static void
io_write(struct iosource *io) { io_write(struct iosource *io) {
struct iooutbuf *buf; struct iooutbuf *buf;
@ -348,17 +198,138 @@ io_write(struct iosource *io) {
g_source_unref((GSource*)io); 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;
}
static void static void
io_disconnect(struct iosource *io, int err) { io_finalize(GSource *source UNUSED) {
if(NULL != io->closed) { struct iosource *io = (struct iosource*)source;
errno = err;
io->closed((GSource*)io, io->cbdata); if(NULL != io->outbufs) {
g_list_foreach(io->outbufs, (GFunc)freeoutbuf, NULL);
g_list_free(io->outbufs);
} }
if(NULL != io->outbufs) if(NULL != io->inbuf)
g_source_remove_poll((GSource*)io, &io->outfd); g_free(io->inbuf);
}
g_source_remove_poll((GSource*)io, &io->infd);
g_source_remove(g_source_get_id((GSource*)io)); static GSourceFuncs sourcefuncs = {
g_source_unref((GSource*)io); 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;
if(NULL != io->outbufs)
io->outbufs = g_list_append(io->outbufs, buf);
else {
io->outbufs = g_list_append(io->outbufs, buf);
g_source_add_poll(source, &io->outfd);
}
return io->lastid;
} }