2008-05-18 16:44:30 +00:00
|
|
|
/*
|
2010-01-04 21:00:47 +00:00
|
|
|
* This file Copyright (C) 2008-2010 Mnemosyne LLC
|
2008-05-18 16:44:30 +00:00
|
|
|
*
|
|
|
|
* This file is licensed by the GPL version 2. Works owned by the
|
|
|
|
* Transmission project are granted a special exemption to clause 2(b)
|
2008-09-23 19:11:04 +00:00
|
|
|
* so that the bulk of its code can remain under the MIT license.
|
2008-05-18 16:44:30 +00:00
|
|
|
* This exemption does not extend to derived works not owned by
|
|
|
|
* the Transmission project.
|
|
|
|
*
|
2008-05-28 17:17:12 +00:00
|
|
|
* $Id$
|
2008-05-18 16:44:30 +00:00
|
|
|
*/
|
|
|
|
|
|
|
|
#include <assert.h>
|
2008-10-06 13:36:06 +00:00
|
|
|
#include <errno.h>
|
2008-09-25 18:48:09 +00:00
|
|
|
#include <string.h> /* memcpy */
|
2008-09-25 05:03:39 +00:00
|
|
|
#include <limits.h> /* INT_MAX */
|
|
|
|
|
2008-09-26 00:58:06 +00:00
|
|
|
#include <sys/types.h> /* open */
|
|
|
|
#include <sys/stat.h> /* open */
|
|
|
|
#include <fcntl.h> /* open */
|
|
|
|
#include <unistd.h> /* close */
|
2008-06-05 16:23:03 +00:00
|
|
|
|
2008-11-09 15:43:31 +00:00
|
|
|
#ifdef HAVE_ZLIB
|
2008-10-01 20:23:57 +00:00
|
|
|
#include <zlib.h>
|
2008-10-01 15:53:56 +00:00
|
|
|
#endif
|
|
|
|
|
2009-06-03 15:47:08 +00:00
|
|
|
#include <event.h>
|
|
|
|
#include <evhttp.h>
|
2008-05-18 16:44:30 +00:00
|
|
|
|
|
|
|
#include "transmission.h"
|
2008-06-30 21:11:53 +00:00
|
|
|
#include "bencode.h"
|
2009-03-17 21:50:20 +00:00
|
|
|
#include "crypto.h"
|
2010-12-20 02:07:51 +00:00
|
|
|
#include "fdlimit.h"
|
2008-07-16 17:47:20 +00:00
|
|
|
#include "list.h"
|
2009-07-01 14:58:57 +00:00
|
|
|
#include "net.h"
|
2008-07-11 04:07:14 +00:00
|
|
|
#include "platform.h"
|
2009-05-14 17:21:07 +00:00
|
|
|
#include "ptrarray.h"
|
2008-09-05 14:31:58 +00:00
|
|
|
#include "rpcimpl.h"
|
2008-05-18 16:44:30 +00:00
|
|
|
#include "rpc-server.h"
|
2008-09-29 04:26:52 +00:00
|
|
|
#include "trevent.h"
|
2008-05-18 16:44:30 +00:00
|
|
|
#include "utils.h"
|
2008-09-25 18:48:09 +00:00
|
|
|
#include "web.h"
|
2008-05-18 16:44:30 +00:00
|
|
|
|
2009-05-08 14:56:11 +00:00
|
|
|
/* session-id is used to make cross-site request forgery attacks difficult.
|
|
|
|
* Don't disable this feature unless you really know what you're doing!
|
|
|
|
* http://en.wikipedia.org/wiki/Cross-site_request_forgery
|
|
|
|
* http://shiflett.org/articles/cross-site-request-forgeries
|
|
|
|
* http://www.webappsec.org/lists/websecurity/archive/2008-04/msg00037.html */
|
|
|
|
#define REQUIRE_SESSION_ID
|
|
|
|
|
2008-06-02 19:57:16 +00:00
|
|
|
#define MY_NAME "RPC Server"
|
2008-08-22 23:04:16 +00:00
|
|
|
#define MY_REALM "Transmission"
|
2008-09-25 05:03:39 +00:00
|
|
|
#define TR_N_ELEMENTS( ary ) ( sizeof( ary ) / sizeof( *ary ) )
|
|
|
|
|
2008-10-14 01:00:15 +00:00
|
|
|
#ifdef WIN32
|
|
|
|
#define strncasecmp _strnicmp
|
|
|
|
#endif
|
|
|
|
|
2008-05-18 16:44:30 +00:00
|
|
|
struct tr_rpc_server
|
|
|
|
{
|
2008-11-28 22:11:41 +00:00
|
|
|
tr_bool isEnabled;
|
|
|
|
tr_bool isPasswordEnabled;
|
|
|
|
tr_bool isWhitelistEnabled;
|
2008-12-01 20:51:01 +00:00
|
|
|
tr_port port;
|
2010-12-12 18:22:11 +00:00
|
|
|
char * url;
|
2009-04-15 21:05:58 +00:00
|
|
|
struct in_addr bindAddress;
|
2008-10-01 20:23:57 +00:00
|
|
|
struct evhttp * httpd;
|
2008-12-14 11:21:11 +00:00
|
|
|
tr_session * session;
|
2008-10-01 20:23:57 +00:00
|
|
|
char * username;
|
|
|
|
char * password;
|
2008-12-21 19:23:41 +00:00
|
|
|
char * whitelistStr;
|
|
|
|
tr_list * whitelist;
|
2008-12-30 20:40:48 +00:00
|
|
|
|
2009-05-08 14:56:11 +00:00
|
|
|
char * sessionId;
|
|
|
|
time_t sessionIdExpiresAt;
|
|
|
|
|
2008-12-30 20:40:48 +00:00
|
|
|
#ifdef HAVE_ZLIB
|
2009-07-02 20:20:00 +00:00
|
|
|
tr_bool isStreamInitialized;
|
2008-12-30 20:40:48 +00:00
|
|
|
z_stream stream;
|
|
|
|
#endif
|
2008-05-18 16:44:30 +00:00
|
|
|
};
|
|
|
|
|
2008-10-26 15:39:04 +00:00
|
|
|
#define dbgmsg( ... ) \
|
|
|
|
do { \
|
|
|
|
if( tr_deepLoggingIsActive( ) ) \
|
|
|
|
tr_deepLog( __FILE__, __LINE__, MY_NAME, __VA_ARGS__ ); \
|
|
|
|
} while( 0 )
|
|
|
|
|
2008-06-05 04:02:46 +00:00
|
|
|
|
2009-05-14 17:21:07 +00:00
|
|
|
/***
|
|
|
|
****
|
|
|
|
***/
|
|
|
|
|
|
|
|
static char*
|
|
|
|
get_current_session_id( struct tr_rpc_server * server )
|
|
|
|
{
|
2009-11-26 18:47:08 +00:00
|
|
|
const time_t now = tr_time( );
|
2009-05-14 17:21:07 +00:00
|
|
|
|
|
|
|
if( !server->sessionId || ( now >= server->sessionIdExpiresAt ) )
|
|
|
|
{
|
|
|
|
int i;
|
|
|
|
const int n = 48;
|
|
|
|
const char * pool = "0123456789abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ";
|
|
|
|
const size_t pool_size = strlen( pool );
|
2010-10-17 18:32:55 +00:00
|
|
|
unsigned char * buf = tr_new( unsigned char, n+1 );
|
2009-05-14 17:21:07 +00:00
|
|
|
|
2010-10-17 18:32:55 +00:00
|
|
|
tr_cryptoRandBuf( buf, n );
|
2009-05-14 17:21:07 +00:00
|
|
|
for( i=0; i<n; ++i )
|
2010-10-17 18:32:55 +00:00
|
|
|
buf[i] = pool[ buf[i] % pool_size ];
|
2009-05-14 17:21:07 +00:00
|
|
|
buf[n] = '\0';
|
|
|
|
|
|
|
|
tr_free( server->sessionId );
|
2010-10-17 18:32:55 +00:00
|
|
|
server->sessionId = (char*) buf;
|
2009-05-14 17:21:07 +00:00
|
|
|
server->sessionIdExpiresAt = now + (60*60); /* expire in an hour */
|
|
|
|
}
|
|
|
|
|
|
|
|
return server->sessionId;
|
|
|
|
}
|
|
|
|
|
|
|
|
|
2008-09-26 15:40:24 +00:00
|
|
|
/**
|
|
|
|
***
|
|
|
|
**/
|
|
|
|
|
|
|
|
static void
|
|
|
|
send_simple_response( struct evhttp_request * req,
|
|
|
|
int code,
|
|
|
|
const char * text )
|
|
|
|
{
|
|
|
|
const char * code_text = tr_webGetResponseStr( code );
|
2009-06-14 01:00:36 +00:00
|
|
|
struct evbuffer * body = evbuffer_new( );
|
2008-09-26 15:40:24 +00:00
|
|
|
|
2008-10-06 13:36:06 +00:00
|
|
|
evbuffer_add_printf( body, "<h1>%d: %s</h1>", code, code_text );
|
2008-09-26 15:40:24 +00:00
|
|
|
if( text )
|
2008-10-07 01:25:29 +00:00
|
|
|
evbuffer_add_printf( body, "%s", text );
|
2008-09-26 15:40:24 +00:00
|
|
|
evhttp_send_reply( req, code, code_text, body );
|
2008-12-30 20:32:00 +00:00
|
|
|
|
2009-06-14 01:00:36 +00:00
|
|
|
evbuffer_free( body );
|
2008-09-26 15:40:24 +00:00
|
|
|
}
|
|
|
|
|
2009-05-14 17:21:07 +00:00
|
|
|
struct tr_mimepart
|
|
|
|
{
|
|
|
|
char * headers;
|
|
|
|
int headers_len;
|
|
|
|
char * body;
|
|
|
|
int body_len;
|
|
|
|
};
|
|
|
|
|
|
|
|
static void
|
|
|
|
tr_mimepart_free( struct tr_mimepart * p )
|
|
|
|
{
|
|
|
|
tr_free( p->body );
|
|
|
|
tr_free( p->headers );
|
|
|
|
tr_free( p );
|
|
|
|
}
|
|
|
|
|
|
|
|
static void
|
|
|
|
extract_parts_from_multipart( const struct evkeyvalq * headers,
|
2010-12-20 02:07:51 +00:00
|
|
|
struct evbuffer * body,
|
2009-05-14 17:21:07 +00:00
|
|
|
tr_ptrArray * setme_parts )
|
|
|
|
{
|
|
|
|
const char * content_type = evhttp_find_header( headers, "Content-Type" );
|
|
|
|
const char * in = (const char*) EVBUFFER_DATA( body );
|
2010-12-20 02:07:51 +00:00
|
|
|
size_t inlen = evbuffer_get_length( body );
|
2009-05-14 17:21:07 +00:00
|
|
|
|
|
|
|
const char * boundary_key = "boundary=";
|
|
|
|
const char * boundary_key_begin = strstr( content_type, boundary_key );
|
|
|
|
const char * boundary_val = boundary_key_begin ? boundary_key_begin + strlen( boundary_key ) : "arglebargle";
|
|
|
|
char * boundary = tr_strdup_printf( "--%s", boundary_val );
|
|
|
|
const size_t boundary_len = strlen( boundary );
|
|
|
|
|
|
|
|
const char * delim = tr_memmem( in, inlen, boundary, boundary_len );
|
|
|
|
while( delim )
|
|
|
|
{
|
|
|
|
size_t part_len;
|
|
|
|
const char * part = delim + boundary_len;
|
|
|
|
|
|
|
|
inlen -= ( part - in );
|
|
|
|
in = part;
|
|
|
|
|
|
|
|
delim = tr_memmem( in, inlen, boundary, boundary_len );
|
|
|
|
part_len = delim ? (size_t)( delim - part ) : inlen;
|
|
|
|
|
|
|
|
if( part_len )
|
|
|
|
{
|
|
|
|
const char * rnrn = tr_memmem( part, part_len, "\r\n\r\n", 4 );
|
|
|
|
if( rnrn )
|
|
|
|
{
|
|
|
|
struct tr_mimepart * p = tr_new( struct tr_mimepart, 1 );
|
|
|
|
p->headers_len = rnrn - part;
|
|
|
|
p->headers = tr_strndup( part, p->headers_len );
|
|
|
|
p->body_len = (part+part_len) - (rnrn + 4);
|
|
|
|
p->body = tr_strndup( rnrn+4, p->body_len );
|
|
|
|
tr_ptrArrayAppend( setme_parts, p );
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
tr_free( boundary );
|
|
|
|
}
|
|
|
|
|
2008-06-30 21:11:53 +00:00
|
|
|
static void
|
2008-09-26 15:40:24 +00:00
|
|
|
handle_upload( struct evhttp_request * req,
|
|
|
|
struct tr_rpc_server * server )
|
2008-06-30 21:11:53 +00:00
|
|
|
{
|
2008-09-25 05:03:39 +00:00
|
|
|
if( req->type != EVHTTP_REQ_POST )
|
|
|
|
{
|
2008-09-26 00:58:06 +00:00
|
|
|
send_simple_response( req, 405, NULL );
|
2008-09-25 05:03:39 +00:00
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
2009-05-14 17:21:07 +00:00
|
|
|
int i;
|
|
|
|
int n;
|
|
|
|
tr_bool hasSessionId = FALSE;
|
|
|
|
tr_ptrArray parts = TR_PTR_ARRAY_INIT;
|
2008-06-30 21:11:53 +00:00
|
|
|
|
2008-09-25 05:03:39 +00:00
|
|
|
const char * query = strchr( req->uri, '?' );
|
2009-05-14 17:21:07 +00:00
|
|
|
const tr_bool paused = query && strstr( query + 1, "paused=true" );
|
2008-09-25 05:03:39 +00:00
|
|
|
|
2009-05-14 17:21:07 +00:00
|
|
|
extract_parts_from_multipart( req->input_headers, req->input_buffer, &parts );
|
|
|
|
n = tr_ptrArraySize( &parts );
|
2008-09-26 04:41:13 +00:00
|
|
|
|
2009-05-14 17:21:07 +00:00
|
|
|
/* first look for the session id */
|
|
|
|
for( i=0; i<n; ++i ) {
|
|
|
|
struct tr_mimepart * p = tr_ptrArrayNth( &parts, i );
|
|
|
|
if( tr_memmem( p->headers, p->headers_len, TR_RPC_SESSION_ID_HEADER, strlen( TR_RPC_SESSION_ID_HEADER ) ) )
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
if( i<n ) {
|
|
|
|
const struct tr_mimepart * p = tr_ptrArrayNth( &parts, i );
|
|
|
|
const char * ours = get_current_session_id( server );
|
|
|
|
const int ourlen = strlen( ours );
|
|
|
|
hasSessionId = ourlen<=p->body_len && !memcmp( p->body, ours, ourlen );
|
|
|
|
}
|
2008-09-25 05:03:39 +00:00
|
|
|
|
2009-05-14 17:21:07 +00:00
|
|
|
if( !hasSessionId )
|
2008-09-26 04:41:13 +00:00
|
|
|
{
|
2010-08-21 12:50:13 +00:00
|
|
|
int code = 409;
|
|
|
|
const char * codetext = tr_webGetResponseStr( code );
|
|
|
|
struct evbuffer * body = evbuffer_new( );
|
|
|
|
evbuffer_add_printf( body, "%s", "{ \"success\": false, \"msg\": \"Bad Session-Id\" }" );;
|
|
|
|
evhttp_send_reply( req, code, codetext, body );
|
|
|
|
evbuffer_free( body );
|
2009-05-14 17:21:07 +00:00
|
|
|
}
|
|
|
|
else for( i=0; i<n; ++i )
|
|
|
|
{
|
|
|
|
struct tr_mimepart * p = tr_ptrArrayNth( &parts, i );
|
2010-08-21 12:50:13 +00:00
|
|
|
int body_len = p->body_len;
|
|
|
|
tr_benc top, *args;
|
|
|
|
tr_benc test;
|
|
|
|
tr_bool have_source = FALSE;
|
|
|
|
char * body = p->body;
|
2009-05-14 17:21:07 +00:00
|
|
|
|
2010-08-21 12:50:13 +00:00
|
|
|
if( body_len >= 2 && !memcmp( &body[body_len - 2], "\r\n", 2 ) )
|
|
|
|
body_len -= 2;
|
2009-05-14 17:21:07 +00:00
|
|
|
|
2010-08-21 12:50:13 +00:00
|
|
|
tr_bencInitDict( &top, 2 );
|
|
|
|
tr_bencDictAddStr( &top, "method", "torrent-add" );
|
|
|
|
args = tr_bencDictAddDict( &top, "arguments", 2 );
|
|
|
|
tr_bencDictAddBool( args, "paused", paused );
|
|
|
|
|
|
|
|
if( tr_urlIsValid( body, body_len ) )
|
|
|
|
{
|
|
|
|
tr_bencDictAddRaw( args, "filename", body, body_len );
|
|
|
|
have_source = TRUE;
|
|
|
|
}
|
|
|
|
else if( !tr_bencLoad( body, body_len, &test, NULL ) )
|
|
|
|
{
|
|
|
|
char * b64 = tr_base64_encode( body, body_len, NULL );
|
2009-05-14 17:21:07 +00:00
|
|
|
tr_bencDictAddStr( args, "metainfo", b64 );
|
2010-08-21 12:50:13 +00:00
|
|
|
tr_free( b64 );
|
|
|
|
have_source = TRUE;
|
|
|
|
}
|
|
|
|
|
|
|
|
if( have_source )
|
|
|
|
{
|
|
|
|
struct evbuffer * json = evbuffer_new( );
|
|
|
|
tr_bencToBuf( &top, TR_FMT_JSON, json );
|
2009-05-14 17:21:07 +00:00
|
|
|
tr_rpc_request_exec_json( server->session,
|
|
|
|
EVBUFFER_DATA( json ),
|
2010-12-20 02:07:51 +00:00
|
|
|
evbuffer_get_length( json ),
|
2009-05-14 17:21:07 +00:00
|
|
|
NULL, NULL );
|
|
|
|
evbuffer_free( json );
|
2008-06-30 21:11:53 +00:00
|
|
|
}
|
2010-08-21 12:50:13 +00:00
|
|
|
|
|
|
|
tr_bencFree( &top );
|
2008-09-26 04:41:13 +00:00
|
|
|
}
|
2008-06-30 21:11:53 +00:00
|
|
|
|
2009-05-14 17:21:07 +00:00
|
|
|
tr_ptrArrayDestruct( &parts, (PtrArrayForeachFunc)tr_mimepart_free );
|
2008-06-30 21:11:53 +00:00
|
|
|
|
2010-08-21 12:50:13 +00:00
|
|
|
/* send "success" response */
|
|
|
|
{
|
|
|
|
int code = HTTP_OK;
|
|
|
|
const char * codetext = tr_webGetResponseStr( code );
|
|
|
|
struct evbuffer * body = evbuffer_new( );
|
|
|
|
evbuffer_add_printf( body, "%s", "{ \"success\": true, \"msg\": \"Torrent Added\" }" );;
|
|
|
|
evhttp_send_reply( req, code, codetext, body );
|
|
|
|
evbuffer_free( body );
|
|
|
|
}
|
2008-07-16 17:47:20 +00:00
|
|
|
}
|
2008-06-30 21:11:53 +00:00
|
|
|
}
|
|
|
|
|
2008-09-25 05:03:39 +00:00
|
|
|
static const char*
|
|
|
|
mimetype_guess( const char * path )
|
2008-08-06 00:24:05 +00:00
|
|
|
{
|
2008-09-25 05:03:39 +00:00
|
|
|
unsigned int i;
|
2008-09-26 15:40:24 +00:00
|
|
|
|
|
|
|
const struct
|
|
|
|
{
|
2008-10-01 20:23:57 +00:00
|
|
|
const char * suffix;
|
|
|
|
const char * mime_type;
|
2008-09-25 05:03:39 +00:00
|
|
|
} types[] = {
|
2010-03-19 18:48:37 +00:00
|
|
|
/* these are the ones we need for serving the web client's files... */
|
2008-10-06 13:36:06 +00:00
|
|
|
{ "css", "text/css" },
|
|
|
|
{ "gif", "image/gif" },
|
|
|
|
{ "html", "text/html" },
|
|
|
|
{ "ico", "image/vnd.microsoft.icon" },
|
|
|
|
{ "js", "application/javascript" },
|
|
|
|
{ "png", "image/png" }
|
2008-09-25 05:03:39 +00:00
|
|
|
};
|
|
|
|
const char * dot = strrchr( path, '.' );
|
|
|
|
|
2008-09-26 15:40:24 +00:00
|
|
|
for( i = 0; dot && i < TR_N_ELEMENTS( types ); ++i )
|
|
|
|
if( !strcmp( dot + 1, types[i].suffix ) )
|
2008-09-25 05:03:39 +00:00
|
|
|
return types[i].mime_type;
|
|
|
|
|
|
|
|
return "application/octet-stream";
|
2008-08-06 00:24:05 +00:00
|
|
|
}
|
|
|
|
|
2008-10-08 13:33:19 +00:00
|
|
|
static void
|
2010-12-20 02:07:51 +00:00
|
|
|
add_response( struct evhttp_request * req, struct tr_rpc_server * server,
|
|
|
|
struct evbuffer * out, struct evbuffer * content )
|
2008-10-01 15:53:56 +00:00
|
|
|
{
|
2008-11-09 15:43:31 +00:00
|
|
|
#ifndef HAVE_ZLIB
|
2010-12-20 02:07:51 +00:00
|
|
|
evbuffer_add_buffer( out, content );
|
2008-10-08 13:33:19 +00:00
|
|
|
#else
|
|
|
|
const char * key = "Accept-Encoding";
|
|
|
|
const char * encoding = evhttp_find_header( req->input_headers, key );
|
2010-08-04 20:57:08 +00:00
|
|
|
const int do_compress = encoding && strstr( encoding, "gzip" );
|
2008-10-03 17:38:14 +00:00
|
|
|
|
2010-08-04 20:57:08 +00:00
|
|
|
if( !do_compress )
|
2008-10-08 13:33:19 +00:00
|
|
|
{
|
2010-12-20 02:07:51 +00:00
|
|
|
evbuffer_add_buffer( out, content );
|
2008-10-01 20:23:57 +00:00
|
|
|
}
|
2008-10-08 13:33:19 +00:00
|
|
|
else
|
|
|
|
{
|
|
|
|
int state;
|
2010-12-20 02:07:51 +00:00
|
|
|
struct evbuffer_iovec iovec[1];
|
|
|
|
void * content_ptr = evbuffer_pullup( content, -1 );
|
|
|
|
const size_t content_len = evbuffer_get_length( content );
|
2009-06-02 18:21:23 +00:00
|
|
|
|
2009-07-02 20:20:00 +00:00
|
|
|
if( !server->isStreamInitialized )
|
|
|
|
{
|
2010-08-15 23:45:58 +00:00
|
|
|
int compressionLevel;
|
|
|
|
|
2009-07-02 20:20:00 +00:00
|
|
|
server->isStreamInitialized = TRUE;
|
|
|
|
server->stream.zalloc = (alloc_func) Z_NULL;
|
|
|
|
server->stream.zfree = (free_func) Z_NULL;
|
|
|
|
server->stream.opaque = (voidpf) Z_NULL;
|
2010-08-04 20:57:08 +00:00
|
|
|
|
|
|
|
/* zlib's manual says: "Add 16 to windowBits to write a simple gzip header
|
|
|
|
* and trailer around the compressed data instead of a zlib wrapper." */
|
2010-08-15 23:45:58 +00:00
|
|
|
#ifdef TR_EMBEDDED
|
|
|
|
compressionLevel = Z_DEFAULT_COMPRESSION;
|
|
|
|
#else
|
|
|
|
compressionLevel = Z_BEST_COMPRESSION;
|
|
|
|
#endif
|
|
|
|
deflateInit2( &server->stream, compressionLevel, Z_DEFLATED, 15+16, 8, Z_DEFAULT_STRATEGY );
|
2009-07-02 20:20:00 +00:00
|
|
|
}
|
|
|
|
|
2010-12-20 02:07:51 +00:00
|
|
|
server->stream.next_in = content_ptr;
|
2008-12-30 20:40:48 +00:00
|
|
|
server->stream.avail_in = content_len;
|
2008-10-01 20:23:57 +00:00
|
|
|
|
2008-10-08 13:39:44 +00:00
|
|
|
/* allocate space for the raw data and call deflate() just once --
|
|
|
|
* we won't use the deflated data if it's longer than the raw data,
|
|
|
|
* so it's okay to let deflate() run out of output buffer space */
|
2010-12-20 02:07:51 +00:00
|
|
|
evbuffer_reserve_space( out, content_len, iovec, 1 );
|
|
|
|
server->stream.next_out = iovec[0].iov_base;
|
|
|
|
server->stream.avail_out = iovec[0].iov_len;
|
2008-12-30 20:40:48 +00:00
|
|
|
state = deflate( &server->stream, Z_FINISH );
|
2008-10-08 13:39:44 +00:00
|
|
|
|
|
|
|
if( state == Z_STREAM_END )
|
|
|
|
{
|
2010-12-20 02:07:51 +00:00
|
|
|
iovec[0].iov_len -= server->stream.avail_out;
|
2008-11-01 22:09:16 +00:00
|
|
|
|
2008-11-19 16:12:00 +00:00
|
|
|
#if 0
|
2010-11-11 15:31:11 +00:00
|
|
|
fprintf( stderr, "compressed response is %.2f of original (raw==%zu bytes; compressed==%zu)\n",
|
2010-12-20 02:07:51 +00:00
|
|
|
(double)evbuffer_get_length(out)/content_len,
|
|
|
|
content_len, evbuffer_get_length(out) );
|
2008-11-19 16:12:00 +00:00
|
|
|
#endif
|
2008-10-08 13:33:19 +00:00
|
|
|
evhttp_add_header( req->output_headers,
|
2010-08-04 20:57:08 +00:00
|
|
|
"Content-Encoding", "gzip" );
|
2008-10-08 13:33:19 +00:00
|
|
|
}
|
2008-10-08 13:39:44 +00:00
|
|
|
else
|
|
|
|
{
|
2010-12-20 02:07:51 +00:00
|
|
|
memcpy( iovec[0].iov_base, content_ptr, content_len );
|
|
|
|
iovec[0].iov_len = content_len;
|
2008-10-08 13:39:44 +00:00
|
|
|
}
|
2008-10-08 13:33:19 +00:00
|
|
|
|
2010-12-20 02:07:51 +00:00
|
|
|
evbuffer_commit_space( out, iovec, 1 );
|
2008-12-30 20:40:48 +00:00
|
|
|
deflateReset( &server->stream );
|
2008-10-08 13:33:19 +00:00
|
|
|
}
|
2008-10-01 15:53:56 +00:00
|
|
|
#endif
|
|
|
|
}
|
|
|
|
|
2009-05-08 16:41:32 +00:00
|
|
|
static void
|
|
|
|
add_time_header( struct evkeyvalq * headers, const char * key, time_t value )
|
|
|
|
{
|
|
|
|
/* According to RFC 2616 this must follow RFC 1123's date format,
|
|
|
|
so use gmtime instead of localtime... */
|
2009-05-08 17:30:14 +00:00
|
|
|
char buf[128];
|
2009-05-08 16:41:32 +00:00
|
|
|
struct tm tm = *gmtime( &value );
|
|
|
|
strftime( buf, sizeof( buf ), "%a, %d %b %Y %H:%M:%S GMT", &tm );
|
|
|
|
evhttp_add_header( headers, key, buf );
|
|
|
|
}
|
|
|
|
|
2008-05-18 16:44:30 +00:00
|
|
|
static void
|
2008-09-26 15:40:24 +00:00
|
|
|
serve_file( struct evhttp_request * req,
|
2008-12-30 20:40:48 +00:00
|
|
|
struct tr_rpc_server * server,
|
2008-10-06 13:36:06 +00:00
|
|
|
const char * filename )
|
2008-05-18 16:44:30 +00:00
|
|
|
{
|
2008-09-25 05:03:39 +00:00
|
|
|
if( req->type != EVHTTP_REQ_GET )
|
2008-05-18 16:44:30 +00:00
|
|
|
{
|
2008-09-26 15:40:24 +00:00
|
|
|
evhttp_add_header( req->output_headers, "Allow", "GET" );
|
2008-09-25 18:48:09 +00:00
|
|
|
send_simple_response( req, 405, NULL );
|
2008-05-18 16:44:30 +00:00
|
|
|
}
|
2008-09-25 05:03:39 +00:00
|
|
|
else
|
2008-05-18 16:44:30 +00:00
|
|
|
{
|
2010-12-20 02:07:51 +00:00
|
|
|
void * file;
|
|
|
|
size_t file_len;
|
|
|
|
struct evbuffer * content = evbuffer_new( );
|
2008-10-06 13:36:06 +00:00
|
|
|
const int error = errno;
|
|
|
|
|
|
|
|
errno = 0;
|
2010-12-20 02:07:51 +00:00
|
|
|
file_len = 0;
|
|
|
|
file = tr_loadFile( filename, &file_len );
|
|
|
|
content = evbuffer_new( );
|
|
|
|
evbuffer_add_reference( content, file, file_len, evbuffer_ref_cleanup_tr_free, file );
|
2008-10-06 13:36:06 +00:00
|
|
|
|
2009-05-08 20:52:12 +00:00
|
|
|
if( errno )
|
|
|
|
{
|
|
|
|
char * tmp = tr_strdup_printf( "%s (%s)", filename, tr_strerror( errno ) );
|
|
|
|
send_simple_response( req, HTTP_NOTFOUND, tmp );
|
|
|
|
tr_free( tmp );
|
|
|
|
}
|
|
|
|
else
|
2008-09-25 05:03:39 +00:00
|
|
|
{
|
2008-10-06 16:33:33 +00:00
|
|
|
struct evbuffer * out;
|
2009-11-26 18:47:08 +00:00
|
|
|
const time_t now = tr_time( );
|
2008-10-06 13:36:06 +00:00
|
|
|
|
2008-10-06 16:33:33 +00:00
|
|
|
errno = error;
|
2009-06-14 01:00:36 +00:00
|
|
|
out = evbuffer_new( );
|
2009-05-08 16:41:32 +00:00
|
|
|
evhttp_add_header( req->output_headers, "Content-Type", mimetype_guess( filename ) );
|
|
|
|
add_time_header( req->output_headers, "Date", now );
|
|
|
|
add_time_header( req->output_headers, "Expires", now+(24*60*60) );
|
2010-12-20 02:07:51 +00:00
|
|
|
add_response( req, server, out, content );
|
2008-10-06 13:36:06 +00:00
|
|
|
evhttp_send_reply( req, HTTP_OK, "OK", out );
|
|
|
|
|
2009-06-14 01:00:36 +00:00
|
|
|
evbuffer_free( out );
|
2008-09-25 05:03:39 +00:00
|
|
|
}
|
2010-12-20 02:07:51 +00:00
|
|
|
|
|
|
|
evbuffer_free( content );
|
2008-07-16 17:47:20 +00:00
|
|
|
}
|
2008-05-18 16:44:30 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
static void
|
2010-03-19 18:48:37 +00:00
|
|
|
handle_web_client( struct evhttp_request * req,
|
|
|
|
struct tr_rpc_server * server )
|
2008-05-18 16:44:30 +00:00
|
|
|
{
|
2010-03-19 18:48:37 +00:00
|
|
|
const char * webClientDir = tr_getWebClientDir( server->session );
|
2008-09-25 05:03:39 +00:00
|
|
|
|
2010-03-19 18:48:37 +00:00
|
|
|
if( !webClientDir || !*webClientDir )
|
2008-10-07 01:25:29 +00:00
|
|
|
{
|
|
|
|
send_simple_response( req, HTTP_NOTFOUND,
|
|
|
|
"<p>Couldn't find Transmission's web interface files!</p>"
|
|
|
|
"<p>Users: to tell Transmission where to look, "
|
2009-05-29 19:17:12 +00:00
|
|
|
"set the TRANSMISSION_WEB_HOME environment "
|
2008-10-07 01:25:29 +00:00
|
|
|
"variable to the folder where the web interface's "
|
|
|
|
"index.html is located.</p>"
|
|
|
|
"<p>Package Builders: to set a custom default at compile time, "
|
|
|
|
"#define PACKAGE_DATA_DIR in libtransmission/platform.c "
|
|
|
|
"or tweak tr_getClutchDir() by hand.</p>" );
|
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
|
|
|
char * pch;
|
|
|
|
char * subpath;
|
|
|
|
|
2010-12-12 18:22:11 +00:00
|
|
|
subpath = tr_strdup( req->uri + strlen( server->url ) + 4 );
|
2008-10-07 01:25:29 +00:00
|
|
|
if(( pch = strchr( subpath, '?' )))
|
|
|
|
*pch = '\0';
|
2008-09-26 15:40:24 +00:00
|
|
|
|
2009-04-10 15:09:31 +00:00
|
|
|
if( strstr( subpath, ".." ) )
|
|
|
|
{
|
|
|
|
send_simple_response( req, HTTP_NOTFOUND, "<p>Tsk, tsk.</p>" );
|
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
|
|
|
char * filename = tr_strdup_printf( "%s%s%s",
|
2010-03-19 18:48:37 +00:00
|
|
|
webClientDir,
|
2009-04-10 15:09:31 +00:00
|
|
|
TR_PATH_DELIMITER_STR,
|
|
|
|
subpath && *subpath ? subpath : "index.html" );
|
|
|
|
serve_file( req, server, filename );
|
|
|
|
tr_free( filename );
|
|
|
|
}
|
2008-05-18 16:44:30 +00:00
|
|
|
|
2008-10-07 01:25:29 +00:00
|
|
|
tr_free( subpath );
|
|
|
|
}
|
2008-06-05 16:23:03 +00:00
|
|
|
}
|
|
|
|
|
2009-01-18 15:24:26 +00:00
|
|
|
struct rpc_response_data
|
|
|
|
{
|
|
|
|
struct evhttp_request * req;
|
|
|
|
struct tr_rpc_server * server;
|
|
|
|
};
|
|
|
|
|
|
|
|
static void
|
|
|
|
rpc_response_func( tr_session * session UNUSED,
|
2010-12-20 02:07:51 +00:00
|
|
|
struct evbuffer * response,
|
2009-01-18 15:24:26 +00:00
|
|
|
void * user_data )
|
|
|
|
{
|
|
|
|
struct rpc_response_data * data = user_data;
|
2009-06-14 01:00:36 +00:00
|
|
|
struct evbuffer * buf = evbuffer_new( );
|
2009-01-18 15:24:26 +00:00
|
|
|
|
2010-12-20 02:07:51 +00:00
|
|
|
add_response( data->req, data->server, buf, response );
|
2009-01-18 15:24:26 +00:00
|
|
|
evhttp_add_header( data->req->output_headers,
|
|
|
|
"Content-Type", "application/json; charset=UTF-8" );
|
|
|
|
evhttp_send_reply( data->req, HTTP_OK, "OK", buf );
|
|
|
|
|
2009-06-14 01:00:36 +00:00
|
|
|
evbuffer_free( buf );
|
2009-01-18 15:24:26 +00:00
|
|
|
tr_free( data );
|
|
|
|
}
|
|
|
|
|
|
|
|
|
2008-05-18 16:44:30 +00:00
|
|
|
static void
|
2008-09-26 15:40:24 +00:00
|
|
|
handle_rpc( struct evhttp_request * req,
|
2008-12-30 22:07:39 +00:00
|
|
|
struct tr_rpc_server * server )
|
2008-05-18 16:44:30 +00:00
|
|
|
{
|
2009-01-18 15:24:26 +00:00
|
|
|
struct rpc_response_data * data = tr_new0( struct rpc_response_data, 1 );
|
2008-06-05 04:02:46 +00:00
|
|
|
|
2009-01-18 15:24:26 +00:00
|
|
|
data->req = req;
|
|
|
|
data->server = server;
|
2009-08-12 14:40:32 +00:00
|
|
|
|
2008-09-25 05:03:39 +00:00
|
|
|
if( req->type == EVHTTP_REQ_GET )
|
2008-05-18 16:44:30 +00:00
|
|
|
{
|
2008-09-25 05:03:39 +00:00
|
|
|
const char * q;
|
2008-09-26 15:40:24 +00:00
|
|
|
if( ( q = strchr( req->uri, '?' ) ) )
|
2009-01-18 15:24:26 +00:00
|
|
|
tr_rpc_request_exec_uri( server->session, q+1, -1, rpc_response_func, data );
|
2008-09-25 05:03:39 +00:00
|
|
|
}
|
|
|
|
else if( req->type == EVHTTP_REQ_POST )
|
|
|
|
{
|
2008-12-30 22:07:39 +00:00
|
|
|
tr_rpc_request_exec_json( server->session,
|
|
|
|
EVBUFFER_DATA( req->input_buffer ),
|
2010-12-20 02:07:51 +00:00
|
|
|
evbuffer_get_length( req->input_buffer ),
|
2009-01-18 15:24:26 +00:00
|
|
|
rpc_response_func, data );
|
2008-12-30 22:07:39 +00:00
|
|
|
}
|
2008-10-02 16:50:05 +00:00
|
|
|
|
2008-09-25 05:03:39 +00:00
|
|
|
}
|
2008-07-17 14:45:31 +00:00
|
|
|
|
2008-12-21 19:23:41 +00:00
|
|
|
static tr_bool
|
2008-09-26 15:40:24 +00:00
|
|
|
isAddressAllowed( const tr_rpc_server * server,
|
|
|
|
const char * address )
|
|
|
|
{
|
2008-12-21 19:23:41 +00:00
|
|
|
tr_list * l;
|
2008-09-26 15:40:24 +00:00
|
|
|
|
2008-10-01 22:59:29 +00:00
|
|
|
if( !server->isWhitelistEnabled )
|
2008-12-21 19:23:41 +00:00
|
|
|
return TRUE;
|
2008-10-01 22:59:29 +00:00
|
|
|
|
2008-12-21 19:23:41 +00:00
|
|
|
for( l=server->whitelist; l!=NULL; l=l->next )
|
|
|
|
if( tr_wildmat( address, l->data ) )
|
|
|
|
return TRUE;
|
2008-09-26 15:40:24 +00:00
|
|
|
|
2008-12-21 19:23:41 +00:00
|
|
|
return FALSE;
|
2008-09-26 15:40:24 +00:00
|
|
|
}
|
|
|
|
|
2009-05-08 14:56:11 +00:00
|
|
|
static tr_bool
|
|
|
|
test_session_id( struct tr_rpc_server * server, struct evhttp_request * req )
|
|
|
|
{
|
|
|
|
const char * ours = get_current_session_id( server );
|
|
|
|
const char * theirs = evhttp_find_header( req->input_headers, TR_RPC_SESSION_ID_HEADER );
|
|
|
|
const tr_bool success = theirs && !strcmp( theirs, ours );
|
|
|
|
return success;
|
|
|
|
}
|
|
|
|
|
2008-09-25 05:03:39 +00:00
|
|
|
static void
|
2009-05-08 14:56:11 +00:00
|
|
|
handle_request( struct evhttp_request * req, void * arg )
|
2008-09-25 05:03:39 +00:00
|
|
|
{
|
|
|
|
struct tr_rpc_server * server = arg;
|
2008-07-17 14:45:31 +00:00
|
|
|
|
2008-09-26 15:40:24 +00:00
|
|
|
if( req && req->evcon )
|
2008-09-25 05:03:39 +00:00
|
|
|
{
|
|
|
|
const char * auth;
|
2009-05-08 14:56:11 +00:00
|
|
|
char * user = NULL;
|
|
|
|
char * pass = NULL;
|
2008-07-17 14:45:31 +00:00
|
|
|
|
2008-09-25 18:48:09 +00:00
|
|
|
evhttp_add_header( req->output_headers, "Server", MY_REALM );
|
2008-09-26 15:40:24 +00:00
|
|
|
|
2008-09-25 05:03:39 +00:00
|
|
|
auth = evhttp_find_header( req->input_headers, "Authorization" );
|
|
|
|
if( auth && !strncasecmp( auth, "basic ", 6 ) )
|
|
|
|
{
|
2008-09-26 15:40:24 +00:00
|
|
|
int plen;
|
2008-09-25 05:03:39 +00:00
|
|
|
char * p = tr_base64_decode( auth + 6, 0, &plen );
|
2008-09-26 15:40:24 +00:00
|
|
|
if( p && plen && ( ( pass = strchr( p, ':' ) ) ) )
|
|
|
|
{
|
2008-09-25 05:03:39 +00:00
|
|
|
user = p;
|
|
|
|
*pass++ = '\0';
|
|
|
|
}
|
|
|
|
}
|
2008-08-15 19:45:46 +00:00
|
|
|
|
2008-10-01 22:59:29 +00:00
|
|
|
if( !isAddressAllowed( server, req->remote_host ) )
|
2008-07-16 23:55:49 +00:00
|
|
|
{
|
2009-05-08 14:56:11 +00:00
|
|
|
send_simple_response( req, 403,
|
2008-12-21 19:23:41 +00:00
|
|
|
"<p>Unauthorized IP Address.</p>"
|
|
|
|
"<p>Either disable the IP address whitelist or add your address to it.</p>"
|
2008-12-23 21:34:07 +00:00
|
|
|
"<p>If you're editing settings.json, see the 'rpc-whitelist' and 'rpc-whitelist-enabled' entries.</p>"
|
|
|
|
"<p>If you're still using ACLs, use a whitelist instead. See the transmission-daemon manpage for details.</p>" );
|
2008-07-11 17:09:53 +00:00
|
|
|
}
|
2008-10-06 13:36:06 +00:00
|
|
|
else if( server->isPasswordEnabled
|
|
|
|
&& ( !pass || !user || strcmp( server->username, user )
|
2009-03-24 01:39:06 +00:00
|
|
|
|| !tr_ssha1_matches( server->password,
|
|
|
|
pass ) ) )
|
2008-07-16 23:55:49 +00:00
|
|
|
{
|
2008-09-25 05:03:39 +00:00
|
|
|
evhttp_add_header( req->output_headers,
|
2008-09-26 15:40:24 +00:00
|
|
|
"WWW-Authenticate",
|
|
|
|
"Basic realm=\"" MY_REALM "\"" );
|
2008-09-25 18:48:09 +00:00
|
|
|
send_simple_response( req, 401, "Unauthorized User" );
|
2008-06-05 04:02:46 +00:00
|
|
|
}
|
2010-12-12 18:22:11 +00:00
|
|
|
else if( strncmp( req->uri, server->url, strlen( server->url ) ) )
|
2008-07-16 23:55:49 +00:00
|
|
|
{
|
2010-12-12 18:22:11 +00:00
|
|
|
const char * protocol = "http";
|
|
|
|
const char * host = evhttp_find_header( req->input_headers, "Host" );
|
|
|
|
char * location = tr_strdup_printf( "%s://%s%sweb/", protocol, host, server->url );
|
|
|
|
evhttp_add_header( req->output_headers, "Location", location );
|
2008-09-25 18:48:09 +00:00
|
|
|
send_simple_response( req, HTTP_MOVEPERM, NULL );
|
2010-12-12 18:22:11 +00:00
|
|
|
tr_free( location );
|
2008-06-05 16:23:03 +00:00
|
|
|
}
|
2010-12-12 18:22:11 +00:00
|
|
|
else if( !strncmp( req->uri + strlen( server->url ), "web/", 4 ) )
|
2008-09-25 05:03:39 +00:00
|
|
|
{
|
2010-03-19 18:48:37 +00:00
|
|
|
handle_web_client( req, server );
|
2008-09-25 05:03:39 +00:00
|
|
|
}
|
2009-05-08 14:56:11 +00:00
|
|
|
#ifdef REQUIRE_SESSION_ID
|
|
|
|
else if( !test_session_id( server, req ) )
|
|
|
|
{
|
|
|
|
const char * sessionId = get_current_session_id( server );
|
|
|
|
char * tmp = tr_strdup_printf(
|
2009-08-13 16:50:50 +00:00
|
|
|
"<p>Your request had an invalid session-id header.</p>"
|
|
|
|
"<p>To fix this, follow these steps:"
|
|
|
|
"<ol><li> When reading a response, get its X-Transmission-Session-Id header and remember it"
|
|
|
|
"<li> Add the updated header to your outgoing requests"
|
|
|
|
"<li> When you get this 409 error message, resend your request with the updated header"
|
|
|
|
"</ol></p>"
|
|
|
|
"<p>This requirement has been added to help prevent "
|
|
|
|
"<a href=\"http://en.wikipedia.org/wiki/Cross-site_request_forgery\">CSRF</a> "
|
|
|
|
"attacks.</p>"
|
|
|
|
"<p><code>%s: %s</code></p>",
|
2009-05-08 20:52:12 +00:00
|
|
|
TR_RPC_SESSION_ID_HEADER, sessionId );
|
2009-05-08 14:56:11 +00:00
|
|
|
evhttp_add_header( req->output_headers, TR_RPC_SESSION_ID_HEADER, sessionId );
|
|
|
|
send_simple_response( req, 409, tmp );
|
|
|
|
tr_free( tmp );
|
|
|
|
}
|
|
|
|
#endif
|
2010-12-12 18:22:11 +00:00
|
|
|
else if( !strncmp( req->uri + strlen( server->url ), "rpc", 3 ) )
|
2008-09-25 05:03:39 +00:00
|
|
|
{
|
|
|
|
handle_rpc( req, server );
|
|
|
|
}
|
2010-12-12 18:22:11 +00:00
|
|
|
else if( !strncmp( req->uri + strlen( server->url ), "upload", 6 ) )
|
|
|
|
{
|
|
|
|
handle_upload( req, server );
|
|
|
|
}
|
2008-09-25 05:03:39 +00:00
|
|
|
else
|
2008-09-03 19:59:09 +00:00
|
|
|
{
|
2009-03-17 21:50:20 +00:00
|
|
|
send_simple_response( req, HTTP_NOTFOUND, req->uri );
|
2008-09-03 19:59:09 +00:00
|
|
|
}
|
2008-07-16 23:55:49 +00:00
|
|
|
|
2008-09-25 05:03:39 +00:00
|
|
|
tr_free( user );
|
2008-05-18 16:44:30 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2008-09-25 05:03:39 +00:00
|
|
|
static void
|
2008-10-01 04:26:38 +00:00
|
|
|
startServer( void * vserver )
|
2008-09-25 05:03:39 +00:00
|
|
|
{
|
2008-10-01 04:26:38 +00:00
|
|
|
tr_rpc_server * server = vserver;
|
2009-04-15 21:05:58 +00:00
|
|
|
tr_address addr;
|
2008-10-01 04:26:38 +00:00
|
|
|
|
2008-09-25 05:03:39 +00:00
|
|
|
if( !server->httpd )
|
2008-09-29 03:02:27 +00:00
|
|
|
{
|
2009-04-15 21:05:58 +00:00
|
|
|
addr.type = TR_AF_INET;
|
|
|
|
addr.addr.addr4 = server->bindAddress;
|
2008-09-29 04:26:52 +00:00
|
|
|
server->httpd = evhttp_new( tr_eventGetBase( server->session ) );
|
2009-04-15 21:05:58 +00:00
|
|
|
evhttp_bind_socket( server->httpd, tr_ntop_non_ts( &addr ),
|
|
|
|
server->port );
|
2008-09-30 16:04:41 +00:00
|
|
|
evhttp_set_gencb( server->httpd, handle_request, server );
|
2009-04-05 15:49:03 +00:00
|
|
|
|
2008-09-29 03:02:27 +00:00
|
|
|
}
|
2008-09-25 05:03:39 +00:00
|
|
|
}
|
|
|
|
|
2008-05-18 16:44:30 +00:00
|
|
|
static void
|
|
|
|
stopServer( tr_rpc_server * server )
|
|
|
|
{
|
2008-09-25 05:03:39 +00:00
|
|
|
if( server->httpd )
|
2008-05-18 16:44:30 +00:00
|
|
|
{
|
2008-09-25 05:03:39 +00:00
|
|
|
evhttp_free( server->httpd );
|
|
|
|
server->httpd = NULL;
|
2008-05-18 16:44:30 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2008-10-01 04:26:38 +00:00
|
|
|
static void
|
|
|
|
onEnabledChanged( void * vserver )
|
|
|
|
{
|
|
|
|
tr_rpc_server * server = vserver;
|
|
|
|
|
|
|
|
if( !server->isEnabled )
|
|
|
|
stopServer( server );
|
|
|
|
else
|
|
|
|
startServer( server );
|
|
|
|
}
|
|
|
|
|
2008-05-18 16:44:30 +00:00
|
|
|
void
|
2008-09-23 19:11:04 +00:00
|
|
|
tr_rpcSetEnabled( tr_rpc_server * server,
|
2008-12-13 23:39:12 +00:00
|
|
|
tr_bool isEnabled )
|
2008-05-18 16:44:30 +00:00
|
|
|
{
|
2008-12-13 23:39:12 +00:00
|
|
|
server->isEnabled = isEnabled;
|
2008-05-18 16:44:30 +00:00
|
|
|
|
2008-10-01 04:26:38 +00:00
|
|
|
tr_runInEventThread( server->session, onEnabledChanged, server );
|
2008-05-18 16:44:30 +00:00
|
|
|
}
|
|
|
|
|
2008-12-13 23:39:12 +00:00
|
|
|
tr_bool
|
2008-05-18 16:44:30 +00:00
|
|
|
tr_rpcIsEnabled( const tr_rpc_server * server )
|
|
|
|
{
|
2008-09-25 18:48:09 +00:00
|
|
|
return server->isEnabled;
|
2008-05-18 16:44:30 +00:00
|
|
|
}
|
|
|
|
|
2008-10-01 04:26:38 +00:00
|
|
|
static void
|
|
|
|
restartServer( void * vserver )
|
|
|
|
{
|
|
|
|
tr_rpc_server * server = vserver;
|
|
|
|
|
|
|
|
if( server->isEnabled )
|
|
|
|
{
|
|
|
|
stopServer( server );
|
|
|
|
startServer( server );
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2008-05-18 16:44:30 +00:00
|
|
|
void
|
2008-09-23 19:11:04 +00:00
|
|
|
tr_rpcSetPort( tr_rpc_server * server,
|
2008-12-01 20:51:01 +00:00
|
|
|
tr_port port )
|
2008-05-18 16:44:30 +00:00
|
|
|
{
|
2010-01-16 22:46:38 +00:00
|
|
|
assert( server != NULL );
|
|
|
|
|
2008-05-18 16:44:30 +00:00
|
|
|
if( server->port != port )
|
|
|
|
{
|
|
|
|
server->port = port;
|
|
|
|
|
2008-09-03 19:59:09 +00:00
|
|
|
if( server->isEnabled )
|
2008-10-01 04:26:38 +00:00
|
|
|
tr_runInEventThread( server->session, restartServer, server );
|
2008-05-18 16:44:30 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2008-12-01 20:51:01 +00:00
|
|
|
tr_port
|
2008-05-18 16:44:30 +00:00
|
|
|
tr_rpcGetPort( const tr_rpc_server * server )
|
|
|
|
{
|
|
|
|
return server->port;
|
|
|
|
}
|
|
|
|
|
2008-09-25 18:48:09 +00:00
|
|
|
void
|
2010-12-12 18:22:11 +00:00
|
|
|
tr_rpcSetUrl( tr_rpc_server * server, const char * url )
|
|
|
|
{
|
|
|
|
char * tmp = server->url;
|
|
|
|
server->url = tr_strdup( url );
|
|
|
|
dbgmsg( "setting our URL to [%s]", server->url );
|
|
|
|
tr_free( tmp );
|
|
|
|
}
|
|
|
|
|
|
|
|
const char*
|
|
|
|
tr_rpcGetUrl( const tr_rpc_server * server )
|
|
|
|
{
|
|
|
|
return server->url ? server->url : "";
|
|
|
|
}
|
|
|
|
|
|
|
|
void
|
|
|
|
tr_rpcSetWhitelist( tr_rpc_server * server, const char * whitelistStr )
|
2008-12-21 19:23:41 +00:00
|
|
|
{
|
|
|
|
void * tmp;
|
|
|
|
const char * walk;
|
|
|
|
|
|
|
|
/* keep the string */
|
2010-12-12 18:22:11 +00:00
|
|
|
tmp = server->whitelistStr;
|
2008-12-21 19:23:41 +00:00
|
|
|
server->whitelistStr = tr_strdup( whitelistStr );
|
2010-12-12 18:22:11 +00:00
|
|
|
tr_free( tmp );
|
2008-12-21 19:23:41 +00:00
|
|
|
|
|
|
|
/* clear out the old whitelist entries */
|
|
|
|
while(( tmp = tr_list_pop_front( &server->whitelist )))
|
|
|
|
tr_free( tmp );
|
|
|
|
|
|
|
|
/* build the new whitelist entries */
|
|
|
|
for( walk=whitelistStr; walk && *walk; ) {
|
|
|
|
const char * delimiters = " ,;";
|
|
|
|
const size_t len = strcspn( walk, delimiters );
|
|
|
|
char * token = tr_strndup( walk, len );
|
|
|
|
tr_list_append( &server->whitelist, token );
|
2008-12-21 19:35:38 +00:00
|
|
|
if( strcspn( token, "+-" ) < len )
|
|
|
|
tr_ninf( MY_NAME, "Adding address to whitelist: %s (And it has a '+' or '-'! Are you using an old ACL by mistake?)", token );
|
|
|
|
else
|
|
|
|
tr_ninf( MY_NAME, "Adding address to whitelist: %s", token );
|
2009-08-12 14:40:32 +00:00
|
|
|
|
2008-12-21 19:23:41 +00:00
|
|
|
if( walk[len]=='\0' )
|
|
|
|
break;
|
|
|
|
walk += len + 1;
|
|
|
|
}
|
2008-05-18 16:44:30 +00:00
|
|
|
}
|
|
|
|
|
2009-04-16 19:14:37 +00:00
|
|
|
const char*
|
2008-10-01 20:23:57 +00:00
|
|
|
tr_rpcGetWhitelist( const tr_rpc_server * server )
|
2008-05-18 16:44:30 +00:00
|
|
|
{
|
2009-04-16 19:14:37 +00:00
|
|
|
return server->whitelistStr ? server->whitelistStr : "";
|
2008-06-05 16:23:03 +00:00
|
|
|
}
|
|
|
|
|
2008-10-01 22:59:29 +00:00
|
|
|
void
|
|
|
|
tr_rpcSetWhitelistEnabled( tr_rpc_server * server,
|
2008-12-13 23:39:12 +00:00
|
|
|
tr_bool isEnabled )
|
2008-10-01 22:59:29 +00:00
|
|
|
{
|
|
|
|
server->isWhitelistEnabled = isEnabled != 0;
|
|
|
|
}
|
|
|
|
|
2008-12-13 23:39:12 +00:00
|
|
|
tr_bool
|
2008-10-01 22:59:29 +00:00
|
|
|
tr_rpcGetWhitelistEnabled( const tr_rpc_server * server )
|
|
|
|
{
|
|
|
|
return server->isWhitelistEnabled;
|
|
|
|
}
|
|
|
|
|
2008-06-05 16:23:03 +00:00
|
|
|
/****
|
|
|
|
***** PASSWORD
|
|
|
|
****/
|
|
|
|
|
|
|
|
void
|
2010-12-12 18:22:11 +00:00
|
|
|
tr_rpcSetUsername( tr_rpc_server * server, const char * username )
|
2008-06-05 16:23:03 +00:00
|
|
|
{
|
2010-12-12 18:22:11 +00:00
|
|
|
char * tmp = server->username;
|
2008-06-05 16:23:03 +00:00
|
|
|
server->username = tr_strdup( username );
|
|
|
|
dbgmsg( "setting our Username to [%s]", server->username );
|
2010-12-12 18:22:11 +00:00
|
|
|
tr_free( tmp );
|
2008-06-05 16:23:03 +00:00
|
|
|
}
|
|
|
|
|
2009-04-16 19:14:37 +00:00
|
|
|
const char*
|
2008-09-23 19:11:04 +00:00
|
|
|
tr_rpcGetUsername( const tr_rpc_server * server )
|
2008-06-05 16:23:03 +00:00
|
|
|
{
|
2009-04-16 19:14:37 +00:00
|
|
|
return server->username ? server->username : "";
|
2008-06-05 16:23:03 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
void
|
2008-09-23 19:11:04 +00:00
|
|
|
tr_rpcSetPassword( tr_rpc_server * server,
|
|
|
|
const char * password )
|
2008-06-05 16:23:03 +00:00
|
|
|
{
|
|
|
|
tr_free( server->password );
|
2009-03-24 01:39:06 +00:00
|
|
|
if( *password != '{' )
|
|
|
|
server->password = tr_ssha1( password );
|
|
|
|
else
|
|
|
|
server->password = strdup( password );
|
2008-06-05 16:23:03 +00:00
|
|
|
dbgmsg( "setting our Password to [%s]", server->password );
|
|
|
|
}
|
|
|
|
|
2009-04-16 19:14:37 +00:00
|
|
|
const char*
|
2008-09-23 19:11:04 +00:00
|
|
|
tr_rpcGetPassword( const tr_rpc_server * server )
|
2008-06-05 16:23:03 +00:00
|
|
|
{
|
2009-04-16 19:14:37 +00:00
|
|
|
return server->password ? server->password : "" ;
|
2008-06-05 16:23:03 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
void
|
2008-09-23 19:11:04 +00:00
|
|
|
tr_rpcSetPasswordEnabled( tr_rpc_server * server,
|
2008-12-13 23:39:12 +00:00
|
|
|
tr_bool isEnabled )
|
2008-06-05 16:23:03 +00:00
|
|
|
{
|
2008-12-13 23:39:12 +00:00
|
|
|
server->isPasswordEnabled = isEnabled;
|
|
|
|
dbgmsg( "setting 'password enabled' to %d", (int)isEnabled );
|
2008-06-05 16:23:03 +00:00
|
|
|
}
|
|
|
|
|
2008-12-13 23:39:12 +00:00
|
|
|
tr_bool
|
2008-06-05 16:23:03 +00:00
|
|
|
tr_rpcIsPasswordEnabled( const tr_rpc_server * server )
|
|
|
|
{
|
|
|
|
return server->isPasswordEnabled;
|
2008-05-18 16:44:30 +00:00
|
|
|
}
|
|
|
|
|
2009-04-15 21:05:58 +00:00
|
|
|
const char *
|
|
|
|
tr_rpcGetBindAddress( const tr_rpc_server * server )
|
|
|
|
{
|
|
|
|
tr_address addr;
|
|
|
|
addr.type = TR_AF_INET;
|
|
|
|
addr.addr.addr4 = server->bindAddress;
|
|
|
|
return tr_ntop_non_ts( &addr );
|
|
|
|
}
|
|
|
|
|
2008-06-05 16:23:03 +00:00
|
|
|
/****
|
|
|
|
***** LIFE CYCLE
|
|
|
|
****/
|
|
|
|
|
2008-10-01 04:26:38 +00:00
|
|
|
static void
|
|
|
|
closeServer( void * vserver )
|
2008-05-18 16:44:30 +00:00
|
|
|
{
|
2008-12-21 19:23:41 +00:00
|
|
|
void * tmp;
|
2008-10-01 04:26:38 +00:00
|
|
|
tr_rpc_server * s = vserver;
|
2008-10-01 20:23:57 +00:00
|
|
|
|
2008-05-18 16:44:30 +00:00
|
|
|
stopServer( s );
|
2008-12-21 19:23:41 +00:00
|
|
|
while(( tmp = tr_list_pop_front( &s->whitelist )))
|
|
|
|
tr_free( tmp );
|
2008-12-30 20:40:48 +00:00
|
|
|
#ifdef HAVE_ZLIB
|
2009-07-02 20:20:00 +00:00
|
|
|
if( s->isStreamInitialized )
|
|
|
|
deflateEnd( &s->stream );
|
2008-12-30 20:40:48 +00:00
|
|
|
#endif
|
2010-12-12 18:22:11 +00:00
|
|
|
tr_free( s->url );
|
2009-05-14 17:18:17 +00:00
|
|
|
tr_free( s->sessionId );
|
2008-12-21 19:23:41 +00:00
|
|
|
tr_free( s->whitelistStr );
|
2008-07-16 19:36:46 +00:00
|
|
|
tr_free( s->username );
|
|
|
|
tr_free( s->password );
|
2008-05-18 16:44:30 +00:00
|
|
|
tr_free( s );
|
|
|
|
}
|
|
|
|
|
2008-10-01 04:26:38 +00:00
|
|
|
void
|
|
|
|
tr_rpcClose( tr_rpc_server ** ps )
|
|
|
|
{
|
2008-10-01 20:23:57 +00:00
|
|
|
tr_runInEventThread( ( *ps )->session, closeServer, *ps );
|
2008-10-01 04:26:38 +00:00
|
|
|
*ps = NULL;
|
|
|
|
}
|
|
|
|
|
2008-05-18 16:44:30 +00:00
|
|
|
tr_rpc_server *
|
2009-10-23 03:41:36 +00:00
|
|
|
tr_rpcInit( tr_session * session, tr_benc * settings )
|
2008-05-18 16:44:30 +00:00
|
|
|
{
|
2008-06-02 19:57:16 +00:00
|
|
|
tr_rpc_server * s;
|
2009-01-13 16:35:06 +00:00
|
|
|
tr_bool found;
|
2009-03-29 23:05:32 +00:00
|
|
|
tr_bool boolVal;
|
2009-01-13 16:35:06 +00:00
|
|
|
int64_t i;
|
|
|
|
const char *str;
|
2009-04-15 21:05:58 +00:00
|
|
|
tr_address address;
|
2008-06-02 19:57:16 +00:00
|
|
|
|
|
|
|
s = tr_new0( tr_rpc_server, 1 );
|
2008-05-18 16:44:30 +00:00
|
|
|
s->session = session;
|
2009-01-13 16:35:06 +00:00
|
|
|
|
2009-03-29 23:05:32 +00:00
|
|
|
found = tr_bencDictFindBool( settings, TR_PREFS_KEY_RPC_ENABLED, &boolVal );
|
2009-01-13 16:35:06 +00:00
|
|
|
assert( found );
|
2009-03-29 23:05:32 +00:00
|
|
|
s->isEnabled = boolVal;
|
2009-01-13 16:35:06 +00:00
|
|
|
|
|
|
|
found = tr_bencDictFindInt( settings, TR_PREFS_KEY_RPC_PORT, &i );
|
|
|
|
assert( found );
|
|
|
|
s->port = i;
|
|
|
|
|
2010-12-12 18:22:11 +00:00
|
|
|
found = tr_bencDictFindStr( settings, TR_PREFS_KEY_RPC_URL, &str );
|
|
|
|
assert( found );
|
|
|
|
s->url = tr_strdup( str );
|
|
|
|
|
2009-03-29 23:05:32 +00:00
|
|
|
found = tr_bencDictFindBool( settings, TR_PREFS_KEY_RPC_WHITELIST_ENABLED, &boolVal );
|
2009-01-13 16:35:06 +00:00
|
|
|
assert( found );
|
2009-10-23 03:41:36 +00:00
|
|
|
tr_rpcSetWhitelistEnabled( s, boolVal );
|
2009-01-13 16:35:06 +00:00
|
|
|
|
2009-03-29 23:05:32 +00:00
|
|
|
found = tr_bencDictFindBool( settings, TR_PREFS_KEY_RPC_AUTH_REQUIRED, &boolVal );
|
2009-01-13 16:35:06 +00:00
|
|
|
assert( found );
|
2009-10-23 03:41:36 +00:00
|
|
|
tr_rpcSetPasswordEnabled( s, boolVal );
|
2009-01-13 16:35:06 +00:00
|
|
|
|
|
|
|
found = tr_bencDictFindStr( settings, TR_PREFS_KEY_RPC_WHITELIST, &str );
|
|
|
|
assert( found );
|
|
|
|
tr_rpcSetWhitelist( s, str ? str : "127.0.0.1" );
|
|
|
|
|
|
|
|
found = tr_bencDictFindStr( settings, TR_PREFS_KEY_RPC_USERNAME, &str );
|
|
|
|
assert( found );
|
2009-10-23 03:41:36 +00:00
|
|
|
tr_rpcSetUsername( s, str );
|
2009-01-13 16:35:06 +00:00
|
|
|
|
|
|
|
found = tr_bencDictFindStr( settings, TR_PREFS_KEY_RPC_PASSWORD, &str );
|
|
|
|
assert( found );
|
2009-10-23 03:41:36 +00:00
|
|
|
tr_rpcSetPassword( s, str );
|
2008-12-30 20:40:48 +00:00
|
|
|
|
2009-04-15 21:05:58 +00:00
|
|
|
found = tr_bencDictFindStr( settings, TR_PREFS_KEY_RPC_BIND_ADDRESS, &str );
|
|
|
|
assert( found );
|
|
|
|
if( tr_pton( str, &address ) == NULL ) {
|
|
|
|
tr_err( _( "%s is not a valid address" ), str );
|
|
|
|
address = tr_inaddr_any;
|
|
|
|
} else if( address.type != TR_AF_INET ) {
|
|
|
|
tr_err( _( "%s is not an IPv4 address. RPC listeners must be IPv4" ),
|
|
|
|
str );
|
|
|
|
address = tr_inaddr_any;
|
|
|
|
}
|
|
|
|
s->bindAddress = address.addr.addr4;
|
|
|
|
|
2009-01-13 16:35:06 +00:00
|
|
|
if( s->isEnabled )
|
2008-12-16 16:13:21 +00:00
|
|
|
{
|
2010-12-12 18:22:11 +00:00
|
|
|
tr_ninf( MY_NAME, _( "Serving RPC and Web requests on port 127.0.0.1:%d%s" ), (int) s->port, s->url );
|
2009-01-13 16:35:06 +00:00
|
|
|
tr_runInEventThread( session, startServer, s );
|
2008-12-16 16:13:21 +00:00
|
|
|
|
2009-01-13 16:35:06 +00:00
|
|
|
if( s->isWhitelistEnabled )
|
2009-06-23 14:15:23 +00:00
|
|
|
tr_ninf( MY_NAME, "%s", _( "Whitelist enabled" ) );
|
2008-12-16 16:13:21 +00:00
|
|
|
|
2009-01-13 16:35:06 +00:00
|
|
|
if( s->isPasswordEnabled )
|
2009-06-23 14:15:23 +00:00
|
|
|
tr_ninf( MY_NAME, "%s", _( "Password required" ) );
|
2008-12-16 16:13:21 +00:00
|
|
|
}
|
|
|
|
|
2008-09-23 19:11:04 +00:00
|
|
|
return s;
|
2008-05-18 16:44:30 +00:00
|
|
|
}
|