source: trunk/libtransmission/platform.c @ 1969

Last change on this file since 1969 was 1969, checked in by livings124, 14 years ago

patch from Charles Kerr: perform file checking one at a time (fixes #124)

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