source: trunk/libtransmission/fdlimit.c @ 10321

Last change on this file since 10321 was 10321, checked in by charles, 12 years ago

(trunk libT) add an fsync() before closing files

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