source: trunk/libtransmission/fdlimit.c @ 11531

Last change on this file since 11531 was 11531, checked in by charles, 11 years ago

(trunk libT) #3826 "all data fails its checksum test in uClibc 0.9.27" -- refinement of r11530 based on research by iz0bbz to find which version of uClibc added a fix.

  • Property svn:keywords set to Date Rev Author Id
File size: 21.6 KB
Line 
1/*
2 * This file Copyright (C) 2010 Mnemosyne LLC
3 *
4 * This file is licensed by the GPL version 2.  Works owned by the
5 * Transmission project are granted a special exemption to clause 2(b)
6 * so that the bulk of its code can remain under the MIT license.
7 * This exemption does not extend to derived works not owned by
8 * the Transmission project.
9 *
10 * $Id: fdlimit.c 11531 2010-12-14 14:41:23Z charles $
11 */
12
13#ifndef WIN32
14 #define HAVE_GETRLIMIT
15#endif
16
17#ifdef HAVE_POSIX_FADVISE
18 #ifdef _XOPEN_SOURCE
19  #undef _XOPEN_SOURCE
20 #endif
21 #define _XOPEN_SOURCE 600
22#endif
23
24#include <assert.h>
25#include <errno.h>
26#include <inttypes.h>
27#include <stdio.h>
28#include <stdlib.h>
29#include <string.h>
30#ifdef SYS_DARWIN
31 #include <fcntl.h>
32#endif
33
34#ifdef HAVE_FALLOCATE64
35  /* FIXME can't find the right #include voodoo to pick up the declaration.. */
36  extern int fallocate64( int fd, int mode, uint64_t offset, uint64_t len );
37#endif
38
39#ifdef HAVE_XFS_XFS_H
40 #include <xfs/xfs.h>
41#endif
42
43#include <sys/types.h>
44#include <sys/stat.h>
45#ifdef HAVE_GETRLIMIT
46 #include <sys/time.h> /* getrlimit */
47 #include <sys/resource.h> /* getrlimit */
48#endif
49#include <fcntl.h> /* O_LARGEFILE posix_fadvise */
50#include <unistd.h>
51
52#include "transmission.h"
53#include "fdlimit.h"
54#include "net.h"
55#include "platform.h" /* TR_PATH_MAX, TR_PATH_DELIMITER */
56#include "session.h"
57#include "torrent.h" /* tr_isTorrent() */
58#include "utils.h"
59
60#define dbgmsg( ... ) \
61    do { \
62        if( tr_deepLoggingIsActive( ) ) \
63            tr_deepLog( __FILE__, __LINE__, NULL, __VA_ARGS__ ); \
64    } while( 0 )
65
66/**
67***
68**/
69
70struct tr_openfile
71{
72    tr_bool          isWritable;
73    int              torrentId;
74    tr_file_index_t  fileNum;
75    char             filename[TR_PATH_MAX];
76    int              fd;
77    time_t           date;
78};
79
80struct tr_fdInfo
81{
82    int                   socketCount;
83    int                   socketLimit;
84    int                   publicSocketLimit;
85    int                   openFileLimit;
86    struct tr_openfile  * openFiles;
87};
88
89/***
90****
91****  Local Files
92****
93***/
94
95#ifndef O_LARGEFILE
96 #define O_LARGEFILE 0
97#endif
98
99static tr_bool
100preallocateFileSparse( int fd, uint64_t length )
101{
102    const char zero = '\0';
103    tr_bool success = 0;
104
105    if( !length )
106        success = TRUE;
107
108#ifdef HAVE_FALLOCATE64
109    if( !success ) /* fallocate64 is always preferred, so try it first */
110        success = !fallocate64( fd, 0, 0, length );
111#endif
112
113    if( !success ) /* fallback: the old-style seek-and-write */
114        success = ( lseek( fd, length-1, SEEK_SET ) != -1 )
115               && ( write( fd, &zero, 1 ) != -1 )
116               && ( ftruncate( fd, length ) != -1 );
117
118    return success;
119}
120
121static tr_bool
122preallocateFileFull( const char * filename, uint64_t length )
123{
124    tr_bool success = 0;
125
126#ifdef WIN32
127
128    HANDLE hFile = CreateFile( filename, GENERIC_WRITE, 0, 0, CREATE_NEW, FILE_FLAG_RANDOM_ACCESS, 0 );
129    if( hFile != INVALID_HANDLE_VALUE )
130    {
131        LARGE_INTEGER li;
132        li.QuadPart = length;
133        success = SetFilePointerEx( hFile, li, NULL, FILE_BEGIN ) && SetEndOfFile( hFile );
134        CloseHandle( hFile );
135    }
136
137#else
138
139    int flags = O_RDWR | O_CREAT | O_LARGEFILE;
140    int fd = open( filename, flags, 0666 );
141    if( fd >= 0 )
142    {
143# ifdef HAVE_FALLOCATE64
144       if( !success )
145       {
146           success = !fallocate64( fd, 0, 0, length );
147       }
148# endif
149# ifdef HAVE_XFS_XFS_H
150        if( !success && platform_test_xfs_fd( fd ) )
151        {
152            xfs_flock64_t fl;
153            fl.l_whence = 0;
154            fl.l_start = 0;
155            fl.l_len = length;
156            success = !xfsctl( NULL, fd, XFS_IOC_RESVSP64, &fl );
157        }
158# endif
159# ifdef SYS_DARWIN
160        if( !success )
161        {
162            fstore_t fst;
163            fst.fst_flags = F_ALLOCATECONTIG;
164            fst.fst_posmode = F_PEOFPOSMODE;
165            fst.fst_offset = 0;
166            fst.fst_length = length;
167            fst.fst_bytesalloc = 0;
168            success = !fcntl( fd, F_PREALLOCATE, &fst );
169        }
170# endif
171# ifdef HAVE_POSIX_FALLOCATE
172        if( !success )
173        {
174            success = !posix_fallocate( fd, 0, length );
175        }
176# endif
177
178        if( !success ) /* if nothing else works, do it the old-fashioned way */
179        {
180            uint8_t buf[ 4096 ];
181            memset( buf, 0, sizeof( buf ) );
182            success = TRUE;
183            while ( success && ( length > 0 ) )
184            {
185                const int thisPass = MIN( length, sizeof( buf ) );
186                success = write( fd, buf, thisPass ) == thisPass;
187                length -= thisPass;
188            }
189        }
190
191        close( fd );
192    }
193
194#endif
195
196    return success;
197}
198
199/* Like pread and pwrite, except that the position is undefined afterwards.
200   And of course they are not thread-safe. */
201
202/* https://trac.transmissionbt.com/ticket/3826 */
203#ifdef __UCLIBC__
204#define TR_UCLIBC_CHECK_VERSION(major,minor,micro) \
205    (__UCLIBC_MAJOR__ > (major) || \
206     (__UCLIBC_MAJOR__ == (major) && __UCLIBC_MINOR__ > (minor)) || \
207     (__UCLIBC_MAJOR__ == (major) && __UCLIBC_MINOR__ == (minor) && \
208      __UCLIBC_SUBLEVEL__ >= (micro)))
209#if !TR_UCLIBC_CHECK_VERSION(0,9,28)
210 #undef HAVE_PREAD
211 #undef HAVE_PWRITE
212#endif
213#endif
214
215#ifdef SYS_DARWIN
216 #define HAVE_PREAD
217 #define HAVE_PWRITE
218#endif
219
220ssize_t
221tr_pread( int fd, void *buf, size_t count, off_t offset )
222{
223#ifdef HAVE_PREAD
224    return pread( fd, buf, count, offset );
225#else
226    const off_t lrc = lseek( fd, offset, SEEK_SET );
227    if( lrc < 0 )
228        return -1;
229    return read( fd, buf, count );
230#endif
231}
232
233ssize_t
234tr_pwrite( int fd, const void *buf, size_t count, off_t offset )
235{
236#ifdef HAVE_PWRITE
237    return pwrite( fd, buf, count, offset );
238#else
239    const off_t lrc = lseek( fd, offset, SEEK_SET );
240    if( lrc < 0 )
241        return -1;
242    return write( fd, buf, count );
243#endif
244}
245
246int
247tr_prefetch( int fd UNUSED, off_t offset UNUSED, size_t count UNUSED )
248{
249#ifdef HAVE_POSIX_FADVISE
250    return posix_fadvise( fd, offset, count, POSIX_FADV_WILLNEED );
251#elif defined(SYS_DARWIN)
252    struct radvisory radv;
253    radv.ra_offset = offset;
254    radv.ra_count = count;
255    return fcntl( fd, F_RDADVISE, &radv );
256#else
257    return 0;
258#endif
259}
260
261int
262tr_open_file_for_writing( const char * filename )
263{
264    int flags = O_WRONLY | O_CREAT;
265#ifdef O_BINARY
266    flags |= O_BINARY;
267#endif
268#ifdef O_LARGEFILE
269    flags |= O_LARGEFILE;
270#endif
271    return open( filename, flags, 0666 );
272}
273
274int
275tr_open_file_for_scanning( const char * filename )
276{
277    int fd;
278    int flags;
279
280    /* build the flags */
281    flags = O_RDONLY;
282#ifdef O_SEQUENTIAL
283    flags |= O_SEQUENTIAL;
284#endif
285#ifdef O_BINARY
286    flags |= O_BINARY;
287#endif
288#ifdef O_LARGEFILE
289    flags |= O_LARGEFILE;
290#endif
291
292    /* open the file */
293    fd = open( filename, flags, 0666 );
294    if( fd >= 0 )
295    {
296        /* Set hints about the lookahead buffer and caching. It's okay
297           for these to fail silently, so don't let them affect errno */
298        const int err = errno;
299#ifdef HAVE_POSIX_FADVISE
300        posix_fadvise( fd, 0, 0, POSIX_FADV_SEQUENTIAL );
301#endif
302#ifdef SYS_DARWIN
303        fcntl( fd, F_NOCACHE, 1 );
304        fcntl( fd, F_RDAHEAD, 1 );
305#endif
306        errno = err;
307    }
308
309    return fd;
310}
311
312void
313tr_close_file( int fd )
314{
315#if defined(HAVE_POSIX_FADVISE)
316    /* Set hint about not caching this file.
317       It's okay for this to fail silently, so don't let it affect errno */
318    const int err = errno;
319    posix_fadvise( fd, 0, 0, POSIX_FADV_DONTNEED );
320    errno = err;
321#endif
322#ifdef SYS_DARWIN
323    /* it's unclear to me from the man pages if this actually flushes out the cache,
324     * but it couldn't hurt... */
325    fcntl( fd, F_NOCACHE, 1 );
326#endif
327    close( fd );
328}
329
330/**
331 * returns 0 on success, or an errno value on failure.
332 * errno values include ENOENT if the parent folder doesn't exist,
333 * plus the errno values set by tr_mkdirp() and open().
334 */
335static int
336TrOpenFile( tr_session             * session,
337            int                      i,
338            const char             * filename,
339            tr_bool                  doWrite,
340            tr_preallocation_mode    preallocationMode,
341            uint64_t                 desiredFileSize )
342{
343    int flags;
344    struct stat sb;
345    tr_bool alreadyExisted;
346    struct tr_openfile * file;
347
348    assert( tr_isSession( session ) );
349    assert( session->fdInfo != NULL );
350
351    file = &session->fdInfo->openFiles[i];
352
353    /* create subfolders, if any */
354    if( doWrite )
355    {
356        char * dir = tr_dirname( filename );
357        const int err = tr_mkdirp( dir, 0777 ) ? errno : 0;
358        if( err ) {
359            tr_err( _( "Couldn't create \"%1$s\": %2$s" ), dir, tr_strerror( err ) );
360            tr_free( dir );
361            return err;
362        }
363        tr_free( dir );
364    }
365
366    alreadyExisted = !stat( filename, &sb ) && S_ISREG( sb.st_mode );
367
368    if( doWrite && !alreadyExisted && ( preallocationMode == TR_PREALLOCATE_FULL ) )
369        if( preallocateFileFull( filename, desiredFileSize ) )
370            tr_dbg( _( "Preallocated file \"%s\"" ), filename );
371
372    /* open the file */
373    flags = doWrite ? ( O_RDWR | O_CREAT ) : O_RDONLY;
374#ifdef O_SEQUENTIAL
375    flags |= O_SEQUENTIAL;
376#endif
377#ifdef O_LARGEFILE
378    flags |= O_LARGEFILE;
379#endif
380#ifdef WIN32
381    flags |= O_BINARY;
382#endif
383    file->fd = open( filename, flags, 0666 );
384    if( file->fd == -1 )
385    {
386        const int err = errno;
387        tr_err( _( "Couldn't open \"%1$s\": %2$s" ), filename, tr_strerror( err ) );
388        return err;
389    }
390
391    /* If the file already exists and it's too large, truncate it.
392     * This is a fringe case that happens if a torrent's been updated
393     * and one of the updated torrent's files is smaller.
394     * http://trac.transmissionbt.com/ticket/2228
395     * https://bugs.launchpad.net/ubuntu/+source/transmission/+bug/318249
396     */
397    if( alreadyExisted && ( desiredFileSize < (uint64_t)sb.st_size ) )
398        ftruncate( file->fd, desiredFileSize );
399
400    if( doWrite && !alreadyExisted && ( preallocationMode == TR_PREALLOCATE_SPARSE ) )
401        preallocateFileSparse( file->fd, desiredFileSize );
402
403#ifdef HAVE_POSIX_FADVISE
404    /* this doubles the OS level readahead buffer, which in practice
405     * turns out to be a good thing, because many (most?) clients request
406     * chunks of blocks in order.
407     * It's okay for this to fail silently, so don't let it affect errno */
408    {
409        const int err = errno;
410        posix_fadvise( file->fd, 0, 0, POSIX_FADV_SEQUENTIAL );
411        errno = err;
412    }
413#endif
414
415#if defined( SYS_DARWIN )
416    /**
417     * 1. Enable readahead for reasons described above w/POSIX_FADV_SEQUENTIAL.
418     *
419     * 2. Disable OS-level caching due to user reports of adverse effects of
420     *    excessive inactive memory.  However this is experimental because
421     *    previous attempts at this have *also* had adverse effects (see r8198)
422     *
423     * It's okay for this to fail silently, so don't let it affect errno
424     */
425    {
426        const int err = errno;
427        fcntl( file->fd, F_NOCACHE, 1 );
428        fcntl( file->fd, F_RDAHEAD, 1 );
429        errno = err;
430    }
431#endif
432
433    return 0;
434}
435
436static inline tr_bool
437fileIsOpen( const struct tr_openfile * o )
438{
439    return o->fd >= 0;
440}
441
442static void
443TrCloseFile( struct tr_openfile * o )
444{
445    assert( o != NULL );
446    assert( fileIsOpen( o ) );
447
448    tr_close_file( o->fd );
449    o->fd = -1;
450}
451
452int
453tr_fdFileGetCached( tr_session       * session,
454                    int                torrentId,
455                    tr_file_index_t    fileNum,
456                    tr_bool            doWrite )
457{
458    struct tr_openfile * match = NULL;
459    struct tr_fdInfo * gFd;
460
461    assert( tr_isSession( session ) );
462    assert( session->fdInfo != NULL );
463    assert( torrentId > 0 );
464    assert( tr_isBool( doWrite ) );
465
466    gFd = session->fdInfo;
467
468    /* is it already open? */
469    {
470        int i;
471        struct tr_openfile * o;
472        for( i=0; i<gFd->openFileLimit; ++i )
473        {
474            o = &gFd->openFiles[i];
475
476            if( torrentId != o->torrentId )
477                continue;
478            if( fileNum != o->fileNum )
479                continue;
480            if( !fileIsOpen( o ) )
481                continue;
482
483            match = o;
484            break;
485        }
486    }
487
488    if( ( match != NULL ) && ( !doWrite || match->isWritable ) )
489    {
490        match->date = tr_time( );
491        return match->fd;
492    }
493
494    return -1;
495}
496
497/* returns an fd on success, or a -1 on failure and sets errno */
498int
499tr_fdFileCheckout( tr_session             * session,
500                   int                      torrentId,
501                   tr_file_index_t          fileNum,
502                   const char             * filename,
503                   tr_bool                  doWrite,
504                   tr_preallocation_mode    preallocationMode,
505                   uint64_t                 desiredFileSize )
506{
507    int i, winner = -1;
508    struct tr_fdInfo * gFd;
509    struct tr_openfile * o;
510
511    assert( tr_isSession( session ) );
512    assert( session->fdInfo != NULL );
513    assert( torrentId > 0 );
514    assert( filename && *filename );
515    assert( tr_isBool( doWrite ) );
516
517    gFd = session->fdInfo;
518
519    dbgmsg( "looking for file '%s', writable %c", filename, doWrite ? 'y' : 'n' );
520
521    /* is it already open? */
522    for( i=0; i<gFd->openFileLimit; ++i )
523    {
524        o = &gFd->openFiles[i];
525
526        if( torrentId != o->torrentId )
527            continue;
528        if( fileNum != o->fileNum )
529            continue;
530        if( !fileIsOpen( o ) )
531            continue;
532
533        if( doWrite && !o->isWritable )
534        {
535            dbgmsg( "found it!  it's open and available, but isn't writable. closing..." );
536            TrCloseFile( o );
537            break;
538        }
539
540        dbgmsg( "found it!  it's ready for use!" );
541        winner = i;
542        break;
543    }
544
545    dbgmsg( "it's not already open.  looking for an open slot or an old file." );
546    while( winner < 0 )
547    {
548        time_t date = tr_time( ) + 1;
549
550        /* look for the file that's been open longest */
551        for( i=0; i<gFd->openFileLimit; ++i )
552        {
553            o = &gFd->openFiles[i];
554
555            if( !fileIsOpen( o ) )
556            {
557                winner = i;
558                dbgmsg( "found an empty slot in %d", winner );
559                break;
560            }
561
562            if( date > o->date )
563            {
564                date = o->date;
565                winner = i;
566            }
567        }
568
569        assert( winner >= 0 );
570
571        if( fileIsOpen( &gFd->openFiles[winner] ) )
572        {
573            dbgmsg( "closing file \"%s\"", gFd->openFiles[winner].filename );
574            TrCloseFile( &gFd->openFiles[winner] );
575        }
576    }
577
578    assert( winner >= 0 );
579    o = &gFd->openFiles[winner];
580    if( !fileIsOpen( o ) )
581    {
582        const int err = TrOpenFile( session, winner, filename, doWrite,
583                                    preallocationMode, desiredFileSize );
584        if( err ) {
585            errno = err;
586            return -1;
587        }
588
589        dbgmsg( "opened '%s' in slot %d, doWrite %c", filename, winner,
590                doWrite ? 'y' : 'n' );
591        tr_strlcpy( o->filename, filename, sizeof( o->filename ) );
592        o->isWritable = doWrite;
593    }
594
595    dbgmsg( "checking out '%s' in slot %d", filename, winner );
596    o->torrentId = torrentId;
597    o->fileNum = fileNum;
598    o->date = tr_time( );
599    return o->fd;
600}
601
602void
603tr_fdFileClose( tr_session        * session,
604                const tr_torrent  * tor,
605                tr_file_index_t     fileNum )
606{
607    struct tr_openfile * o;
608    struct tr_fdInfo * gFd;
609    const struct tr_openfile * end;
610    const int torrentId = tr_torrentId( tor );
611
612    assert( tr_isSession( session ) );
613    assert( session->fdInfo != NULL );
614    assert( tr_isTorrent( tor ) );
615    assert( fileNum < tor->info.fileCount );
616
617    gFd = session->fdInfo;
618
619    for( o=gFd->openFiles, end=o+gFd->openFileLimit; o!=end; ++o )
620    {
621        if( torrentId != o->torrentId )
622            continue;
623        if( fileNum != o->fileNum )
624            continue;
625        if( !fileIsOpen( o ) )
626            continue;
627
628        dbgmsg( "tr_fdFileClose closing \"%s\"", o->filename );
629        TrCloseFile( o );
630    }
631}
632
633void
634tr_fdTorrentClose( tr_session * session, int torrentId )
635{
636    assert( tr_isSession( session ) );
637
638    if( session->fdInfo != NULL )
639    {
640        struct tr_openfile * o;
641        const struct tr_openfile * end;
642        struct tr_fdInfo * gFd = session->fdInfo;
643
644        for( o=gFd->openFiles, end=o+gFd->openFileLimit; o!=end; ++o )
645            if( fileIsOpen( o ) && ( o->torrentId == torrentId ) )
646                TrCloseFile( o );
647    }
648}
649
650/***
651****
652****  Sockets
653****
654***/
655
656int
657tr_fdSocketCreate( tr_session * session, int domain, int type )
658{
659    int s = -1;
660    struct tr_fdInfo * gFd;
661
662    assert( tr_isSession( session ) );
663    assert( session->fdInfo != NULL );
664
665    gFd = session->fdInfo;
666
667    if( gFd->socketCount < gFd->socketLimit )
668        if( ( s = socket( domain, type, 0 ) ) < 0 )
669        {
670            if( sockerrno != EAFNOSUPPORT )
671                tr_err( _( "Couldn't create socket: %s" ),
672                        tr_strerror( sockerrno ) );
673        }
674
675    if( s > -1 )
676        ++gFd->socketCount;
677
678    assert( gFd->socketCount >= 0 );
679
680    if( s >= 0 )
681    {
682        static tr_bool buf_logged = FALSE;
683        if( !buf_logged )
684        {
685            int i;
686            socklen_t size = sizeof( int );
687            buf_logged = TRUE;
688            getsockopt( s, SOL_SOCKET, SO_SNDBUF, &i, &size );
689            tr_dbg( "SO_SNDBUF size is %d", i );
690            getsockopt( s, SOL_SOCKET, SO_RCVBUF, &i, &size );
691            tr_dbg( "SO_RCVBUF size is %d", i );
692        }
693    }
694
695    return s;
696}
697
698int
699tr_fdSocketAccept( tr_session  * session,
700                   int           b,
701                   tr_address  * addr,
702                   tr_port     * port )
703{
704    int s;
705    unsigned int len;
706    struct tr_fdInfo * gFd;
707    struct sockaddr_storage sock;
708
709    assert( tr_isSession( session ) );
710    assert( session->fdInfo != NULL );
711    assert( addr );
712    assert( port );
713
714    gFd = session->fdInfo;
715
716    len = sizeof( struct sockaddr_storage );
717    s = accept( b, (struct sockaddr *) &sock, &len );
718
719    if( ( s >= 0 ) && gFd->socketCount > gFd->socketLimit )
720    {
721        tr_netCloseSocket( s );
722        s = -1;
723    }
724
725    if( s >= 0 )
726    {
727        /* "The ss_family field of the sockaddr_storage structure will always
728         * align with the family field of any protocol-specific structure." */
729        if( sock.ss_family == AF_INET )
730        {
731            struct sockaddr_in *si;
732            union { struct sockaddr_storage dummy; struct sockaddr_in si; } s;
733            s.dummy = sock;
734            si = &s.si;
735            addr->type = TR_AF_INET;
736            addr->addr.addr4.s_addr = si->sin_addr.s_addr;
737            *port = si->sin_port;
738        }
739        else
740        {
741            struct sockaddr_in6 *si;
742            union { struct sockaddr_storage dummy; struct sockaddr_in6 si; } s;
743            s.dummy = sock;
744            si = &s.si;
745            addr->type = TR_AF_INET6;
746            addr->addr.addr6 = si->sin6_addr;
747            *port = si->sin6_port;
748        }
749        ++gFd->socketCount;
750    }
751
752    return s;
753}
754
755void
756tr_fdSocketClose( tr_session * session, int fd )
757{
758    assert( tr_isSession( session ) );
759
760    if( session->fdInfo != NULL )
761    {
762        struct tr_fdInfo * gFd = session->fdInfo;
763
764        if( fd >= 0 )
765        {
766            tr_netCloseSocket( fd );
767            --gFd->socketCount;
768        }
769
770        assert( gFd->socketCount >= 0 );
771    }
772}
773
774/***
775****
776****  Startup / Shutdown
777****
778***/
779
780static void
781ensureSessionFdInfoExists( tr_session * session )
782{
783    assert( tr_isSession( session ) );
784
785    if( session->fdInfo == NULL )
786        session->fdInfo = tr_new0( struct tr_fdInfo, 1 );
787}
788
789void
790tr_fdClose( tr_session * session )
791{
792    struct tr_fdInfo * gFd;
793    struct tr_openfile * o;
794    const struct tr_openfile * end;
795
796    assert( tr_isSession( session ) );
797    assert( session->fdInfo != NULL );
798
799    gFd = session->fdInfo;
800
801    for( o=gFd->openFiles, end=o+gFd->openFileLimit; o!=end; ++o )
802        if( fileIsOpen( o ) )
803            TrCloseFile( o );
804
805    tr_free( gFd->openFiles );
806    tr_free( gFd );
807    session->fdInfo = NULL;
808}
809
810/***
811****
812***/
813
814void
815tr_fdSetFileLimit( tr_session * session, int limit )
816{
817    struct tr_fdInfo * gFd;
818
819    ensureSessionFdInfoExists( session );
820
821    gFd = session->fdInfo;
822
823    if( gFd->openFileLimit != limit )
824    {
825        int i;
826        struct tr_openfile * o;
827        const struct tr_openfile * end;
828
829        /* close any files we've got open  */
830        for( o=gFd->openFiles, end=o+gFd->openFileLimit; o!=end; ++o )
831            if( fileIsOpen( o ) )
832                TrCloseFile( o );
833
834        /* rebuild the openFiles array */
835        tr_free( gFd->openFiles );
836        gFd->openFiles = tr_new0( struct tr_openfile, limit );
837        gFd->openFileLimit = limit;
838        for( i=0; i<gFd->openFileLimit; ++i )
839            gFd->openFiles[i].fd = -1;
840    }
841}
842
843int
844tr_fdGetFileLimit( const tr_session * session )
845{
846    return session && session->fdInfo ? session->fdInfo->openFileLimit : -1;
847}
848
849void
850tr_fdSetPeerLimit( tr_session * session, int socketLimit )
851{
852    struct tr_fdInfo * gFd;
853
854    ensureSessionFdInfoExists( session );
855
856    gFd = session->fdInfo;
857
858#ifdef HAVE_GETRLIMIT
859    {
860        struct rlimit rlim;
861        const int NOFILE_BUFFER = 512;
862        const int open_max = sysconf( _SC_OPEN_MAX );
863        getrlimit( RLIMIT_NOFILE, &rlim );
864        rlim.rlim_cur = MAX( 1024, open_max );
865        rlim.rlim_cur = MIN( rlim.rlim_cur, rlim.rlim_max );
866        setrlimit( RLIMIT_NOFILE, &rlim );
867        tr_dbg( "setrlimit( RLIMIT_NOFILE, %d )", (int)rlim.rlim_cur );
868        gFd->socketLimit = MIN( socketLimit, (int)rlim.rlim_cur - NOFILE_BUFFER );
869    }
870#else
871    gFd->socketLimit = socketLimit;
872#endif
873    gFd->publicSocketLimit = socketLimit;
874
875    tr_dbg( "socket limit is %d", (int)gFd->socketLimit );
876}
877
878int
879tr_fdGetPeerLimit( const tr_session * session )
880{
881    return session && session->fdInfo ? session->fdInfo->publicSocketLimit : -1;
882}
Note: See TracBrowser for help on using the repository browser.