source: trunk/libtransmission/platform.c @ 2034

Last change on this file since 2034 was 2034, checked in by joshe, 14 years ago

Add some code for verbose UPnP logging which can be enabled at compile-time.

  • Property svn:keywords set to Date Rev Author Id
File size: 16.1 KB
Line 
1/******************************************************************************
2 * $Id: platform.c 2034 2007-06-10 22:26:59Z joshe $
3 *
4 * Copyright (c) 2005 Transmission authors and contributors
5 *
6 * Permission is hereby granted, free of charge, to any person obtaining a
7 * copy of this software and associated documentation files (the "Software"),
8 * to deal in the Software without restriction, including without limitation
9 * the rights to use, copy, modify, merge, publish, distribute, sublicense,
10 * and/or sell copies of the Software, and to permit persons to whom the
11 * Software is furnished to do so, subject to the following conditions:
12 *
13 * The above copyright notice and this permission notice shall be included in
14 * all copies or substantial portions of the Software.
15 *
16 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
17 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
18 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
19 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
20 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
21 * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
22 * DEALINGS IN THE SOFTWARE.
23 *****************************************************************************/
24
25#ifdef SYS_BEOS
26  #include <fs_info.h>
27  #include <FindDirectory.h>
28#endif
29#include <sys/types.h>
30#include <dirent.h>
31
32#include "transmission.h"
33
34#if !defined( SYS_BEOS ) && !defined( __AMIGAOS4__ )
35
36#include <pwd.h>
37
38const char *
39tr_getHomeDirectory( void )
40{
41    static char     homeDirectory[MAX_PATH_LENGTH];
42    static int      init = 0;
43    char          * envHome;
44    struct passwd * pw;
45
46    if( init )
47    {
48        return homeDirectory;
49    }
50
51    envHome = getenv( "HOME" );
52    if( NULL == envHome )
53    {
54        pw = getpwuid( getuid() );
55        endpwent();
56        if( NULL == pw )
57        {
58            /* XXX need to handle this case */
59            return NULL;
60        }
61        envHome = pw->pw_dir;
62    }
63
64    snprintf( homeDirectory, MAX_PATH_LENGTH, "%s", envHome );
65    init = 1;
66
67    return homeDirectory;
68}
69
70#else
71
72const char *
73tr_getHomeDirectory( void )
74{
75    /* XXX */
76    return "";
77}
78
79#endif /* !SYS_BEOS && !__AMIGAOS4__ */
80
81static void
82tr_migrateResume( const char *oldDirectory, const char *newDirectory )
83{
84    DIR * dirh;
85    struct dirent * dirp;
86    char oldFile[MAX_PATH_LENGTH];
87    char newFile[MAX_PATH_LENGTH];
88
89    if( ( dirh = opendir( oldDirectory ) ) )
90    {
91        while( ( dirp = readdir( dirh ) ) )
92        {
93            if( strncmp( "resume.", dirp->d_name, 7 ) )
94            {
95                continue;
96            }
97            snprintf( oldFile, MAX_PATH_LENGTH, "%s/%s",
98                      oldDirectory, dirp->d_name );
99            snprintf( newFile, MAX_PATH_LENGTH, "%s/%s",
100                      newDirectory, dirp->d_name );
101            rename( oldFile, newFile );
102        }
103
104        closedir( dirh );
105    }
106}
107
108char * tr_getPrefsDirectory()
109{
110    static char prefsDirectory[MAX_PATH_LENGTH];
111    static int  init = 0;
112
113    if( init )
114    {
115        return prefsDirectory;
116    }
117
118#ifdef SYS_BEOS
119        find_directory( B_USER_SETTINGS_DIRECTORY, dev_for_path("/boot"),
120                        true, prefsDirectory, MAX_PATH_LENGTH );
121        strcat( prefsDirectory, "/Transmission" );
122#elif defined( SYS_DARWIN )
123    snprintf( prefsDirectory, MAX_PATH_LENGTH,
124              "%s/Library/Application Support/Transmission",
125              tr_getHomeDirectory() );
126#elif defined(__AMIGAOS4__)
127    snprintf( prefsDirectory, MAX_PATH_LENGTH, "PROGDIR:.transmission" );
128#else
129    snprintf( prefsDirectory, MAX_PATH_LENGTH, "%s/.transmission",
130              tr_getHomeDirectory() );
131#endif
132
133    tr_mkdir( prefsDirectory );
134    init = 1;
135
136#ifdef SYS_DARWIN
137    char oldDirectory[MAX_PATH_LENGTH];
138    snprintf( oldDirectory, MAX_PATH_LENGTH, "%s/.transmission",
139              tr_getHomeDirectory() );
140    tr_migrateResume( oldDirectory, prefsDirectory );
141    rmdir( oldDirectory );
142#endif
143
144    return prefsDirectory;
145}
146
147char * tr_getCacheDirectory()
148{
149    static char cacheDirectory[MAX_PATH_LENGTH];
150    static int  init = 0;
151
152    if( init )
153    {
154        return cacheDirectory;
155    }
156
157#ifdef SYS_BEOS
158    /* XXX hey Bryan, is this fine with you? */
159    snprintf( cacheDirectory, MAX_PATH_LENGTH, "%s/Cache",
160              tr_getPrefsDirectory() );
161#elif defined( SYS_DARWIN )
162    snprintf( cacheDirectory, MAX_PATH_LENGTH, "%s/Library/Caches/Transmission",
163              tr_getHomeDirectory() );
164#else
165    snprintf( cacheDirectory, MAX_PATH_LENGTH, "%s/cache",
166              tr_getPrefsDirectory() );
167#endif
168
169    tr_mkdir( cacheDirectory );
170    init = 1;
171
172    if( strcmp( tr_getPrefsDirectory(), cacheDirectory ) )
173    {
174        tr_migrateResume( tr_getPrefsDirectory(), cacheDirectory );
175    }
176
177    return cacheDirectory;
178}
179
180char * tr_getTorrentsDirectory()
181{
182    static char torrentsDirectory[MAX_PATH_LENGTH];
183    static int  init = 0;
184
185    if( init )
186    {
187        return torrentsDirectory;
188    }
189
190#ifdef SYS_BEOS
191    /* XXX hey Bryan, is this fine with you? */
192    snprintf( torrentsDirectory, MAX_PATH_LENGTH, "%s/Torrents",
193              tr_getPrefsDirectory() );
194#elif defined( SYS_DARWIN )
195    snprintf( torrentsDirectory, MAX_PATH_LENGTH, "%s/Torrents",
196              tr_getPrefsDirectory() );
197#else
198    snprintf( torrentsDirectory, MAX_PATH_LENGTH, "%s/torrents",
199              tr_getPrefsDirectory() );
200#endif
201
202    tr_mkdir( torrentsDirectory );
203    init = 1;
204
205    return torrentsDirectory;
206}
207
208static void ThreadFunc( void * _t )
209{
210    tr_thread_t * t = _t;
211
212#ifdef SYS_BEOS
213    /* This is required because on BeOS, SIGINT is sent to each thread,
214       which kills them not nicely */
215    signal( SIGINT, SIG_IGN );
216#endif
217
218    tr_dbg( "Thread '%s' started", t->name );
219    t->func( t->arg );
220    tr_dbg( "Thread '%s' exited", t->name );
221}
222
223void tr_threadCreate( tr_thread_t * t, void (*func)(void *), void * arg,
224                      char * name )
225{
226    t->func = func;
227    t->arg  = arg;
228    t->name = strdup( name );
229#ifdef SYS_BEOS
230    t->thread = spawn_thread( (void *) ThreadFunc, name,
231                              B_NORMAL_PRIORITY, t );
232    resume_thread( t->thread );
233#else
234    pthread_create( &t->thread, NULL, (void *) ThreadFunc, t );
235#endif
236}
237
238const tr_thread_t THREAD_EMPTY = { NULL, NULL, NULL, 0 };
239
240void tr_threadJoin( tr_thread_t * t )
241{
242    if (t->func != NULL)
243    {
244#ifdef SYS_BEOS
245        long exit;
246        wait_for_thread( t->thread, &exit );
247#else
248        pthread_join( t->thread, NULL );
249#endif
250        tr_dbg( "Thread '%s' joined", t->name );
251        free( t->name );
252        t->name = NULL;
253        t->func = NULL;
254    }
255}
256
257void tr_lockInit( tr_lock_t * l )
258{
259#ifdef SYS_BEOS
260    *l = create_sem( 1, "" );
261#else
262    pthread_mutex_init( l, NULL );
263#endif
264}
265
266void tr_lockClose( tr_lock_t * l )
267{
268#ifdef SYS_BEOS
269    delete_sem( *l );
270#else
271    pthread_mutex_destroy( l );
272#endif
273}
274
275
276void tr_condInit( tr_cond_t * c )
277{
278#ifdef SYS_BEOS
279    *c = -1;
280#else
281    pthread_cond_init( c, NULL );
282#endif
283}
284
285void tr_condWait( tr_cond_t * c, tr_lock_t * l )
286{
287#ifdef SYS_BEOS
288    *c = find_thread( NULL );
289    release_sem( *l );
290    suspend_thread( *c );
291    acquire_sem( *l );
292    *c = -1;
293#else
294    pthread_cond_wait( c, l );
295#endif
296}
297
298void tr_condSignal( tr_cond_t * c )
299{
300#ifdef SYS_BEOS
301    while( *c != -1 )
302    {
303        thread_info info;
304        get_thread_info( *c, &info );
305        if( info.state == B_THREAD_SUSPENDED )
306        {
307            resume_thread( *c );
308            break;
309        }
310        snooze( 5000 );
311    }
312#else
313    pthread_cond_signal( c );
314#endif
315}
316
317void tr_condClose( tr_cond_t * c )
318{
319#ifdef SYS_BEOS
320    *c = -1; /* Shut up gcc */
321#else
322    pthread_cond_destroy( c );
323#endif
324}
325
326
327#if defined( BSD )
328
329#include <sys/sysctl.h>
330#include <net/route.h>
331
332static uint8_t *
333getroute( int * buflen );
334static int
335parseroutes( uint8_t * buf, int len, struct in_addr * addr );
336
337int
338tr_getDefaultRoute( struct in_addr * addr )
339{
340    uint8_t * buf;
341    int len;
342
343    buf = getroute( &len );
344    if( NULL == buf )
345    {
346        tr_err( "failed to get default route (BSD)" );
347        return 1;
348    }
349
350    len = parseroutes( buf, len, addr );
351    free( buf );
352
353    return len;
354}
355
356#ifndef SA_SIZE
357#define ROUNDUP( a, size ) \
358    ( ( (a) & ( (size) - 1 ) ) ? ( 1 + ( (a) | ( (size) - 1 ) ) ) : (a) )
359#define SA_SIZE( sap ) \
360    ( sap->sa_len ? ROUNDUP( (sap)->sa_len, sizeof( u_long ) ) : \
361                    sizeof( u_long ) )
362#endif /* !SA_SIZE */
363#define NEXT_SA( sap ) \
364    (struct sockaddr *) ( (caddr_t) (sap) + ( SA_SIZE( (sap) ) ) )
365
366static uint8_t *
367getroute( int * buflen )
368{
369    int     mib[6];
370    size_t  len;
371    uint8_t * buf;
372
373    mib[0] = CTL_NET;
374    mib[1] = PF_ROUTE;
375    mib[2] = 0;
376    mib[3] = AF_INET;
377    mib[4] = NET_RT_FLAGS;
378    mib[5] = RTF_GATEWAY;
379
380    if( sysctl( mib, 6, NULL, &len, NULL, 0 ) )
381    {
382        if( ENOENT != errno )
383        {
384            tr_err( "sysctl net.route.0.inet.flags.gateway failed (%s)",
385                    strerror( errno ) );
386        }
387        *buflen = 0;
388        return NULL;
389    }
390
391    buf = malloc( len );
392    if( NULL == buf )
393    {
394        *buflen = 0;
395        return NULL;
396    }
397
398    if( sysctl( mib, 6, buf, &len, NULL, 0 ) )
399    {
400        tr_err( "sysctl net.route.0.inet.flags.gateway failed (%s)",
401                strerror( errno ) );
402        free( buf );
403        *buflen = 0;
404        return NULL;
405    }
406
407    *buflen = len;
408
409    return buf;
410}
411
412static int
413parseroutes( uint8_t * buf, int len, struct in_addr * addr )
414{
415    uint8_t            * end;
416    struct rt_msghdr   * rtm;
417    struct sockaddr    * sa;
418    struct sockaddr_in * sin;
419    int                  ii;
420    struct in_addr       dest, gw;
421
422    end = buf + len;
423    while( end > buf + sizeof( *rtm ) )
424    {
425        rtm = (struct rt_msghdr *) buf;
426        buf += rtm->rtm_msglen;
427        if( end >= buf )
428        {
429            dest.s_addr = INADDR_NONE;
430            gw.s_addr   = INADDR_NONE;
431            sa = (struct sockaddr *) ( rtm + 1 );
432
433            for( ii = 0; ii < RTAX_MAX && (uint8_t *) sa < buf; ii++ )
434            {
435                if( buf < (uint8_t *) NEXT_SA( sa ) )
436                {
437                    break;
438                }
439
440                if( rtm->rtm_addrs & ( 1 << ii ) )
441                {
442                    if( AF_INET == sa->sa_family )
443                    {
444                        sin = (struct sockaddr_in *) sa;
445                        switch( ii )
446                        {
447                            case RTAX_DST:
448                                dest = sin->sin_addr;
449                                break;
450                            case RTAX_GATEWAY:
451                                gw = sin->sin_addr;
452                                break;
453                        }
454                    }
455                    sa = NEXT_SA( sa );
456                }
457            }
458
459            if( INADDR_ANY == dest.s_addr && INADDR_NONE != gw.s_addr )
460            {
461                *addr = gw;
462                return 0;
463            }
464        }
465    }
466
467    return 1;
468}
469
470#elif defined( linux ) || defined( __linux ) || defined( __linux__ )
471
472#include <linux/types.h>
473#include <linux/netlink.h>
474#include <linux/rtnetlink.h>
475
476#define SEQNUM 195909
477
478static int
479getsock( void );
480static uint8_t *
481getroute( int fd, unsigned int * buflen );
482static int
483parseroutes( uint8_t * buf, unsigned int len, struct in_addr * addr );
484
485int
486tr_getDefaultRoute( struct in_addr * addr )
487{
488    int fd, ret;
489    unsigned int len;
490    uint8_t * buf;
491
492    ret = 1;
493    fd = getsock();
494    if( 0 <= fd )
495    {
496        while( ret )
497        {
498            buf = getroute( fd, &len );
499            if( NULL == buf )
500            {
501                break;
502            }
503            ret = parseroutes( buf, len, addr );
504            free( buf );
505        }
506        close( fd );
507    }
508
509    if( ret )
510    {
511        tr_err( "failed to get default route (Linux)" );
512    }
513
514    return ret;
515}
516
517static int
518getsock( void )
519{
520    int fd, flags;
521    struct
522    {
523        struct nlmsghdr nlh;
524        struct rtgenmsg rtg;
525    } req;
526    struct sockaddr_nl snl;
527
528    fd = socket( PF_NETLINK, SOCK_DGRAM, NETLINK_ROUTE );
529    if( 0 > fd )
530    {
531        tr_err( "failed to create routing socket (%s)", strerror( errno ) );
532        return -1;
533    }
534
535    flags = fcntl( fd, F_GETFL );
536    if( 0 > flags || 0 > fcntl( fd, F_SETFL, O_NONBLOCK | flags ) )
537    {
538        tr_err( "failed to set socket nonblocking (%s)", strerror( errno ) );
539        close( fd );
540        return -1;
541    }
542
543    bzero( &snl, sizeof( snl ) );
544    snl.nl_family = AF_NETLINK;
545
546    bzero( &req, sizeof( req ) );
547    req.nlh.nlmsg_len = NLMSG_LENGTH( sizeof( req.rtg ) );
548    req.nlh.nlmsg_type = RTM_GETROUTE;
549    req.nlh.nlmsg_flags = NLM_F_REQUEST | NLM_F_DUMP;
550    req.nlh.nlmsg_seq = SEQNUM;
551    req.nlh.nlmsg_pid = 0;
552    req.rtg.rtgen_family = AF_INET;
553
554    if( 0 > sendto( fd, &req, sizeof( req ), 0,
555                    (struct sockaddr *) &snl, sizeof( snl ) ) )
556    {
557        tr_err( "failed to write to routing socket (%s)", strerror( errno ) );
558        close( fd );
559        return -1;
560    }
561
562    return fd;
563}
564
565static uint8_t *
566getroute( int fd, unsigned int * buflen )
567{
568    void             * buf;
569    unsigned int       len;
570    ssize_t            res;
571    struct sockaddr_nl snl;
572    socklen_t          slen;
573
574    len = 8192;
575    buf = calloc( 1, len );
576    if( NULL == buf )
577    {
578        *buflen = 0;
579        return NULL;
580    }
581
582    for( ;; )
583    {
584        bzero( &snl, sizeof( snl ) );
585        slen = sizeof( snl );
586        res = recvfrom( fd, buf, len, 0, (struct sockaddr *) &snl, &slen );
587        if( 0 > res )
588        {
589            if( EAGAIN != errno )
590            {
591                tr_err( "failed to read from routing socket (%s)",
592                        strerror( errno ) );
593            }
594            free( buf );
595            *buflen = 0;
596            return NULL;
597        }
598        if( slen < sizeof( snl ) || AF_NETLINK != snl.nl_family )
599        {
600            tr_err( "bad address" );
601            free( buf );
602            *buflen = 0;
603            return NULL;
604        }
605
606        if( 0 == snl.nl_pid )
607        {
608            break;
609        }
610    }
611
612    *buflen = res;
613
614    return buf;
615}
616
617static int
618parseroutes( uint8_t * buf, unsigned int len, struct in_addr * addr )
619{
620    struct nlmsghdr * nlm;
621    struct nlmsgerr * nle;
622    struct rtmsg    * rtm;
623    struct rtattr   * rta;
624    int               rtalen;
625    struct in_addr    gw, dst;
626
627    nlm = ( struct nlmsghdr * ) buf;
628    while( NLMSG_OK( nlm, len ) )
629    {
630        gw.s_addr = INADDR_ANY;
631        dst.s_addr = INADDR_ANY;
632        if( NLMSG_ERROR == nlm->nlmsg_type )
633        {
634            nle = (struct nlmsgerr *) NLMSG_DATA( nlm );
635            if( NLMSG_LENGTH( NLMSG_ALIGN( sizeof( struct nlmsgerr ) ) ) >
636                nlm->nlmsg_len )
637            {
638                tr_err( "truncated netlink error" );
639            }
640            else
641            {
642                tr_err( "netlink error (%s)", strerror( nle->error ) );
643            }
644            return 1;
645        }
646        else if( RTM_NEWROUTE == nlm->nlmsg_type && SEQNUM == nlm->nlmsg_seq &&
647                 getpid() == (pid_t) nlm->nlmsg_pid &&
648                 NLMSG_LENGTH( sizeof( struct rtmsg ) ) <= nlm->nlmsg_len )
649        {
650            rtm = NLMSG_DATA( nlm );
651            rta = RTM_RTA( rtm );
652            rtalen = RTM_PAYLOAD( nlm );
653
654            while( RTA_OK( rta, rtalen ) )
655            {
656                if( sizeof( struct in_addr ) <= RTA_PAYLOAD( rta ) )
657                {
658                    switch( rta->rta_type )
659                    {
660                        case RTA_GATEWAY:
661                            memcpy( &gw, RTA_DATA( rta ), sizeof( gw ) );
662                            break;
663                        case RTA_DST:
664                            memcpy( &dst, RTA_DATA( rta ), sizeof( dst ) );
665                            break;
666                    }
667                }
668                rta = RTA_NEXT( rta, rtalen );
669            }
670        }
671
672        if( INADDR_NONE != gw.s_addr && INADDR_ANY != gw.s_addr &&
673            INADDR_ANY == dst.s_addr )
674        {
675            *addr = gw;
676            return 0;
677        }
678
679        nlm = NLMSG_NEXT( nlm, len );
680    }
681
682    return 1;
683}
684
685#else /* not BSD or Linux */
686
687int
688tr_getDefaultRoute( struct in_addr * addr UNUSED )
689{
690    tr_inf( "don't know how to get default route on this platform" );
691    return 1;
692}
693
694#endif
Note: See TracBrowser for help on using the repository browser.