source: trunk/libtransmission/utils.c @ 10239

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

(trunk) #2938 "crash when adding a torrent by URL from an ftp source over RPC" -- fixed in trunk for 1.91

  • Property svn:keywords set to Date Rev Author Id
File size: 31.4 KB
Line 
1/*
2 * This file Copyright (C) 2009-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: utils.c 10239 2010-02-20 15:57:05Z charles $
11 */
12
13#ifdef HAVE_MEMMEM
14 #define _GNU_SOURCE /* glibc's string.h needs this to pick up memmem */
15#endif
16
17#include <assert.h>
18#include <ctype.h> /* isalpha, tolower */
19#include <errno.h>
20#include <math.h> /* pow */
21#include <stdarg.h>
22#include <stdio.h>
23#include <stdlib.h>
24#include <string.h> /* strerror, memset, memmem */
25
26#include <libgen.h> /* basename */
27#include <sys/time.h>
28#include <sys/types.h>
29#include <sys/stat.h>
30#include <unistd.h> /* usleep, stat, getcwd */
31
32#include "event.h"
33
34#ifdef WIN32
35 #include <direct.h> /* _getcwd */
36 #include <windows.h> /* Sleep */
37#endif
38
39#include "transmission.h"
40#include "fdlimit.h"
41#include "ConvertUTF.h"
42#include "list.h"
43#include "net.h"
44#include "utils.h"
45#include "platform.h"
46#include "version.h"
47
48
49int                   messageLevel = TR_MSG_INF;
50static tr_lock *      messageLock = NULL;
51static tr_bool        messageQueuing = FALSE;
52static tr_msg_list *  messageQueue = NULL;
53static tr_msg_list ** messageQueueTail = &messageQueue;
54static int            messageQueueCount = 0;
55
56#ifndef WIN32
57    /* make null versions of these win32 functions */
58    static inline int IsDebuggerPresent( void ) { return FALSE; }
59    static inline void OutputDebugString( const void * unused UNUSED ) { }
60#endif
61
62/***
63****
64***/
65
66time_t transmission_now = 0;
67
68void
69tr_timeUpdate( time_t now )
70{
71    transmission_now = now;
72}
73
74/***
75****
76***/
77
78void
79tr_msgInit( void )
80{
81    const char * env = getenv( "TR_DEBUG" );
82    messageLevel = ( env ? atoi( env ) : 0 ) + 1;
83    messageLevel = MAX( 1, messageLevel );
84
85    if( messageLock == NULL )
86        messageLock = tr_lockNew( );
87}
88
89FILE*
90tr_getLog( void )
91{
92    static tr_bool initialized = FALSE;
93    static FILE * file = NULL;
94
95    if( !initialized )
96    {
97        const char * str = getenv( "TR_DEBUG_FD" );
98        int          fd = 0;
99        if( str && *str )
100            fd = atoi( str );
101        switch( fd )
102        {
103            case 1:
104                file = stdout; break;
105
106            case 2:
107                file = stderr; break;
108
109            default:
110                file = NULL; break;
111        }
112        initialized = TRUE;
113    }
114
115    return file;
116}
117
118void
119tr_setMessageLevel( int level )
120{
121    tr_lockLock( messageLock );
122
123    messageLevel = MAX( 0, level );
124
125    tr_lockUnlock( messageLock );
126}
127
128int
129tr_getMessageLevel( void )
130{
131    int ret;
132    tr_lockLock( messageLock );
133
134    ret = messageLevel;
135
136    tr_lockUnlock( messageLock );
137    return ret;
138}
139
140void
141tr_setMessageQueuing( tr_bool enabled )
142{
143    tr_lockLock( messageLock );
144
145    messageQueuing = enabled;
146
147    tr_lockUnlock( messageLock );
148}
149
150tr_bool
151tr_getMessageQueuing( void )
152{
153    int ret;
154    tr_lockLock( messageLock );
155
156    ret = messageQueuing;
157
158    tr_lockUnlock( messageLock );
159    return ret;
160}
161
162tr_msg_list *
163tr_getQueuedMessages( void )
164{
165    tr_msg_list * ret;
166    tr_lockLock( messageLock );
167
168    ret = messageQueue;
169    messageQueue = NULL;
170    messageQueueTail = &messageQueue;
171
172    messageQueueCount = 0;
173
174    tr_lockUnlock( messageLock );
175    return ret;
176}
177
178void
179tr_freeMessageList( tr_msg_list * list )
180{
181    tr_msg_list * next;
182
183    while( NULL != list )
184    {
185        next = list->next;
186        free( list->message );
187        free( list->name );
188        free( list );
189        list = next;
190    }
191}
192
193/**
194***
195**/
196
197struct tm *
198tr_localtime_r( const time_t *_clock, struct tm *_result )
199{
200#ifdef HAVE_LOCALTIME_R
201    return localtime_r( _clock, _result );
202#else
203    struct tm *p = localtime( _clock );
204    if( p )
205        *(_result) = *p;
206    return p;
207#endif
208}
209
210char*
211tr_getLogTimeStr( char * buf, int buflen )
212{
213    char           tmp[64];
214    struct tm      now_tm;
215    struct timeval tv;
216    time_t         seconds;
217    int            milliseconds;
218
219    gettimeofday( &tv, NULL );
220
221    seconds = tv.tv_sec;
222    tr_localtime_r( &seconds, &now_tm );
223    strftime( tmp, sizeof( tmp ), "%H:%M:%S", &now_tm );
224    milliseconds = (int)( tv.tv_usec / 1000 );
225    tr_snprintf( buf, buflen, "%s.%03d", tmp, milliseconds );
226
227    return buf;
228}
229
230tr_bool
231tr_deepLoggingIsActive( void )
232{
233    static int8_t deepLoggingIsActive = -1;
234
235    if( deepLoggingIsActive < 0 )
236        deepLoggingIsActive = IsDebuggerPresent() || (tr_getLog()!=NULL);
237
238    return deepLoggingIsActive != 0;
239}
240
241void
242tr_deepLog( const char  * file,
243            int           line,
244            const char  * name,
245            const char  * fmt,
246            ... )
247{
248    FILE * fp = tr_getLog( );
249    if( fp || IsDebuggerPresent( ) )
250    {
251        va_list           args;
252        char              timestr[64];
253        struct evbuffer * buf = evbuffer_new( );
254        char *            base = tr_basename( file );
255
256        evbuffer_add_printf( buf, "[%s] ",
257                            tr_getLogTimeStr( timestr, sizeof( timestr ) ) );
258        if( name )
259            evbuffer_add_printf( buf, "%s ", name );
260        va_start( args, fmt );
261        evbuffer_add_vprintf( buf, fmt, args );
262        va_end( args );
263        evbuffer_add_printf( buf, " (%s:%d)\n", base, line );
264        /* FIXME(libevent2) ifdef this out for nonwindows platforms */
265        OutputDebugString( EVBUFFER_DATA( buf ) );
266        if(fp) /* FIXME(libevent2) tr_getLog() should return an fd, then use evbuffer_write() here ) */
267            (void) fwrite( EVBUFFER_DATA( buf ), 1, EVBUFFER_LENGTH( buf ), fp );
268
269        tr_free( base );
270        evbuffer_free( buf );
271    }
272}
273
274/***
275****
276***/
277
278void
279tr_msg( const char * file, int line,
280        int level, const char * name,
281        const char * fmt, ... )
282{
283    const int err = errno; /* message logging shouldn't affect errno */
284    char buf[1024];
285    va_list ap;
286
287    if( messageLock != NULL )
288        tr_lockLock( messageLock );
289
290    /* build the text message */
291    *buf = '\0';
292    va_start( ap, fmt );
293    evutil_vsnprintf( buf, sizeof( buf ), fmt, ap );
294    va_end( ap );
295
296    OutputDebugString( buf );
297
298    if( *buf )
299    {
300        if( messageQueuing )
301        {
302            tr_msg_list * newmsg;
303            newmsg = tr_new0( tr_msg_list, 1 );
304            newmsg->level = level;
305            newmsg->when = tr_time( );
306            newmsg->message = tr_strdup( buf );
307            newmsg->file = file;
308            newmsg->line = line;
309            newmsg->name = tr_strdup( name );
310
311            *messageQueueTail = newmsg;
312            messageQueueTail = &newmsg->next;
313            ++messageQueueCount;
314
315            if( messageQueueCount > TR_MAX_MSG_LOG )
316            {
317                tr_msg_list * old = messageQueue;
318                messageQueue = old->next;
319                old->next = NULL;
320                tr_freeMessageList(old);
321
322                --messageQueueCount;
323
324                assert( messageQueueCount == TR_MAX_MSG_LOG );
325            }
326        }
327        else
328        {
329            char timestr[64];
330            FILE * fp;
331
332            fp = tr_getLog( );
333            if( fp == NULL )
334                fp = stderr;
335
336            tr_getLogTimeStr( timestr, sizeof( timestr ) );
337
338            if( name )
339                fprintf( fp, "[%s] %s: %s\n", timestr, name, buf );
340            else
341                fprintf( fp, "[%s] %s\n", timestr, buf );
342            fflush( fp );
343        }
344    }
345
346    if( messageLock != NULL )
347        tr_lockUnlock( messageLock );
348
349    errno = err;
350}
351
352/***
353****
354***/
355
356void
357tr_set_compare( const void * va,
358                size_t aCount,
359                const void * vb,
360                size_t bCount,
361                int compare( const void * a, const void * b ),
362                size_t elementSize,
363                tr_set_func in_a_cb,
364                tr_set_func in_b_cb,
365                tr_set_func in_both_cb,
366                void * userData )
367{
368    const uint8_t * a = (const uint8_t *) va;
369    const uint8_t * b = (const uint8_t *) vb;
370    const uint8_t * aend = a + elementSize * aCount;
371    const uint8_t * bend = b + elementSize * bCount;
372
373    while( a != aend || b != bend )
374    {
375        if( a == aend )
376        {
377            ( *in_b_cb )( (void*)b, userData );
378            b += elementSize;
379        }
380        else if( b == bend )
381        {
382            ( *in_a_cb )( (void*)a, userData );
383            a += elementSize;
384        }
385        else
386        {
387            const int val = ( *compare )( a, b );
388
389            if( !val )
390            {
391                ( *in_both_cb )( (void*)a, userData );
392                a += elementSize;
393                b += elementSize;
394            }
395            else if( val < 0 )
396            {
397                ( *in_a_cb )( (void*)a, userData );
398                a += elementSize;
399            }
400            else if( val > 0 )
401            {
402                ( *in_b_cb )( (void*)b, userData );
403                b += elementSize;
404            }
405        }
406    }
407}
408
409/***
410****
411***/
412
413#ifdef DISABLE_GETTEXT
414
415const char*
416tr_strip_positional_args( const char* str )
417{
418    const char * in = str;
419    static size_t bufsize = 0;
420    static char * buf = NULL;
421    const size_t  len = strlen( str );
422    char *        out;
423
424    if( bufsize < len )
425    {
426        bufsize = len * 2;
427        buf = tr_renew( char, buf, bufsize );
428    }
429
430    for( out = buf; *str; ++str )
431    {
432        *out++ = *str;
433        if( ( *str == '%' ) && isdigit( str[1] ) )
434        {
435            const char * tmp = str + 1;
436            while( isdigit( *tmp ) )
437                ++tmp;
438
439            if( *tmp == '$' )
440                str = tmp;
441        }
442    }
443    *out = '\0';
444
445    return strcmp( buf, in ) ? buf : in;
446}
447
448#endif
449
450/**
451***
452**/
453
454void
455tr_timerAdd( struct event * timer, int seconds, int microseconds )
456{
457    struct timeval tv;
458    tv.tv_sec = seconds;
459    tv.tv_usec = microseconds;
460
461    assert( tv.tv_sec >= 0 );
462    assert( tv.tv_usec >= 0 );
463    assert( tv.tv_usec < 1000000 );
464
465    evtimer_add( timer, &tv );
466}
467
468void
469tr_timerAddMsec( struct event * timer, int msec )
470{
471    const int seconds =  msec / 1000;
472    const int usec = (msec%1000) * 1000;
473    tr_timerAdd( timer, seconds, usec );
474}
475
476/**
477***
478**/
479
480uint8_t *
481tr_loadFile( const char * path,
482             size_t *     size )
483{
484    uint8_t * buf;
485    struct stat  sb;
486    int fd;
487    ssize_t n;
488    const char * err_fmt = _( "Couldn't read \"%1$s\": %2$s" );
489
490    /* try to stat the file */
491    errno = 0;
492    if( stat( path, &sb ) )
493    {
494        const int err = errno;
495        tr_dbg( err_fmt, path, tr_strerror( errno ) );
496        errno = err;
497        return NULL;
498    }
499
500    if( ( sb.st_mode & S_IFMT ) != S_IFREG )
501    {
502        tr_err( err_fmt, path, _( "Not a regular file" ) );
503        errno = EISDIR;
504        return NULL;
505    }
506
507    /* Load the torrent file into our buffer */
508    fd = tr_open_file_for_scanning( path );
509    if( fd < 0 )
510    {
511        const int err = errno;
512        tr_err( err_fmt, path, tr_strerror( errno ) );
513        errno = err;
514        return NULL;
515    }
516    buf = malloc( sb.st_size + 1 );
517    if( !buf )
518    {
519        const int err = errno;
520        tr_err( err_fmt, path, _( "Memory allocation failed" ) );
521        tr_close_file( fd );
522        errno = err;
523        return NULL;
524    }
525    n = read( fd, buf, sb.st_size );
526    if( n == -1 )
527    {
528        const int err = errno;
529        tr_err( err_fmt, path, tr_strerror( errno ) );
530        tr_close_file( fd );
531        free( buf );
532        errno = err;
533        return NULL;
534    }
535
536    tr_close_file( fd );
537    buf[ sb.st_size ] = '\0';
538    *size = sb.st_size;
539    return buf;
540}
541
542char*
543tr_basename( const char * path )
544{
545    char * tmp = tr_strdup( path );
546    char * ret = tr_strdup( basename( tmp ) );
547    tr_free( tmp );
548    return ret;
549}
550
551char*
552tr_dirname( const char * path )
553{
554    char * tmp = tr_strdup( path );
555    char * ret = tr_strdup( dirname( tmp ) );
556    tr_free( tmp );
557    return ret;
558}
559
560int
561tr_mkdir( const char * path,
562          int permissions
563#ifdef WIN32
564                       UNUSED
565#endif
566        )
567{
568#ifdef WIN32
569    if( path && isalpha( path[0] ) && path[1] == ':' && !path[2] )
570        return 0;
571    return mkdir( path );
572#else
573    return mkdir( path, permissions );
574#endif
575}
576
577int
578tr_mkdirp( const char * path_in,
579           int          permissions )
580{
581    char *      path = tr_strdup( path_in );
582    char *      p, * pp;
583    struct stat sb;
584    int         done;
585
586    /* walk past the root */
587    p = path;
588    while( *p == TR_PATH_DELIMITER )
589        ++p;
590
591    pp = p;
592    done = 0;
593    while( ( p =
594                strchr( pp, TR_PATH_DELIMITER ) ) || ( p = strchr( pp, '\0' ) ) )
595    {
596        if( !*p )
597            done = 1;
598        else
599            *p = '\0';
600
601        if( stat( path, &sb ) )
602        {
603            /* Folder doesn't exist yet */
604            if( tr_mkdir( path, permissions ) )
605            {
606                const int err = errno;
607                tr_err( _(
608                           "Couldn't create \"%1$s\": %2$s" ), path,
609                       tr_strerror( err ) );
610                tr_free( path );
611                errno = err;
612                return -1;
613            }
614        }
615        else if( ( sb.st_mode & S_IFMT ) != S_IFDIR )
616        {
617            /* Node exists but isn't a folder */
618            char * buf = tr_strdup_printf( _( "File \"%s\" is in the way" ), path );
619            tr_err( _( "Couldn't create \"%1$s\": %2$s" ), path_in, buf );
620            tr_free( buf );
621            tr_free( path );
622            errno = ENOTDIR;
623            return -1;
624        }
625
626        if( done )
627            break;
628
629        *p = TR_PATH_DELIMITER;
630        p++;
631        pp = p;
632    }
633
634    tr_free( path );
635    return 0;
636}
637
638char*
639tr_buildPath( const char *first_element, ... )
640{
641    size_t bufLen = 0;
642    const char * element;
643    char * buf;
644    char * pch;
645    va_list vl;
646
647    /* pass 1: allocate enough space for the string */
648    va_start( vl, first_element );
649    element = first_element;
650    while( element ) {
651        bufLen += strlen( element ) + 1;
652        element = (const char*) va_arg( vl, const char* );
653    }
654    pch = buf = tr_new( char, bufLen );
655    va_end( vl );
656
657    /* pass 2: build the string piece by piece */
658    va_start( vl, first_element );
659    element = first_element;
660    while( element ) {
661        const size_t elementLen = strlen( element );
662        memcpy( pch, element, elementLen );
663        pch += elementLen;
664        *pch++ = TR_PATH_DELIMITER;
665        element = (const char*) va_arg( vl, const char* );
666    }
667    va_end( vl );
668
669    /* terminate the string.  if nonempty, eat the unwanted trailing slash */
670    if( pch != buf )
671        --pch;
672    *pch++ = '\0';
673
674    /* sanity checks & return */
675    assert( pch - buf == (off_t)bufLen );
676    return buf;
677}
678
679/****
680*****
681****/
682
683char*
684tr_strndup( const void * in, int len )
685{
686    char * out = NULL;
687
688    if( len < 0 )
689    {
690        out = tr_strdup( in );
691    }
692    else if( in )
693    {
694        out = tr_malloc( len + 1 );
695        memcpy( out, in, len );
696        out[len] = '\0';
697    }
698
699    return out;
700}
701
702const char*
703tr_memmem( const char * haystack, size_t haystacklen,
704           const char * needle, size_t needlelen )
705{
706#ifdef HAVE_MEMMEM
707    return memmem( haystack, haystacklen, needle, needlelen );
708#else
709    size_t i;
710    if( !needlelen )
711        return haystack;
712    if( needlelen > haystacklen || !haystack || !needle )
713        return NULL;
714    for( i=0; i<=haystacklen-needlelen; ++i )
715        if( !memcmp( haystack+i, needle, needlelen ) )
716            return haystack+i;
717    return NULL;
718#endif
719}
720
721char*
722tr_strdup_printf( const char * fmt, ... )
723{
724    va_list ap;
725    char * ret;
726    size_t len;
727    char statbuf[2048];
728
729    va_start( ap, fmt );
730    len = evutil_vsnprintf( statbuf, sizeof( statbuf ), fmt, ap );
731    va_end( ap );
732    if( len < sizeof( statbuf ) )
733        ret = tr_strndup( statbuf, len );
734    else {
735        ret = tr_new( char, len + 1 );
736        va_start( ap, fmt );
737        evutil_vsnprintf( ret, len + 1, fmt, ap );
738        va_end( ap );
739    }
740
741    return ret;
742}
743
744const char*
745tr_strerror( int i )
746{
747    const char * ret = strerror( i );
748
749    if( ret == NULL )
750        ret = "Unknown Error";
751    return ret;
752}
753
754/****
755*****
756****/
757
758char*
759tr_strstrip( char * str )
760{
761    if( str != NULL )
762    {
763        size_t pos;
764        size_t len = strlen( str );
765
766        while( len && isspace( str[len - 1] ) )
767            --len;
768
769        for( pos = 0; pos < len && isspace( str[pos] ); )
770            ++pos;
771
772        len -= pos;
773        memmove( str, str + pos, len );
774        str[len] = '\0';
775    }
776
777    return str;
778}
779
780/****
781*****
782****/
783
784uint64_t
785tr_date( void )
786{
787    struct timeval tv;
788
789    gettimeofday( &tv, NULL );
790    return (uint64_t) tv.tv_sec * 1000 + ( tv.tv_usec / 1000 );
791}
792
793void
794tr_wait_msec( uint64_t delay_milliseconds )
795{
796#ifdef WIN32
797    Sleep( (DWORD)delay_milliseconds );
798#else
799    usleep( 1000 * delay_milliseconds );
800#endif
801}
802
803/***
804****
805***/
806
807int
808tr_snprintf( char * buf, size_t buflen, const char * fmt, ... )
809{
810    int     len;
811    va_list args;
812
813    va_start( args, fmt );
814    len = evutil_vsnprintf( buf, buflen, fmt, args );
815    va_end( args );
816    return len;
817}
818
819/*
820 * Copy src to string dst of size siz.  At most siz-1 characters
821 * will be copied.  Always NUL terminates (unless siz == 0).
822 * Returns strlen(src); if retval >= siz, truncation occurred.
823 */
824size_t
825tr_strlcpy( char *       dst,
826            const void * src,
827            size_t       siz )
828{
829#ifdef HAVE_STRLCPY
830    return strlcpy( dst, src, siz );
831#else
832    char *      d = dst;
833    const char *s = src;
834    size_t      n = siz;
835
836    assert( s );
837    assert( d );
838
839    /* Copy as many bytes as will fit */
840    if( n != 0 )
841    {
842        while( --n != 0 )
843        {
844            if( ( *d++ = *s++ ) == '\0' )
845                break;
846        }
847    }
848
849    /* Not enough room in dst, add NUL and traverse rest of src */
850    if( n == 0 )
851    {
852        if( siz != 0 )
853            *d = '\0'; /* NUL-terminate dst */
854        while( *s++ )
855            ;
856    }
857
858    return s - (char*)src - 1;  /* count does not include NUL */
859#endif
860}
861
862/***
863****
864***/
865
866double
867tr_getRatio( double numerator,
868             double denominator )
869{
870    double ratio;
871
872    if( denominator )
873        ratio = numerator / denominator;
874    else if( numerator )
875        ratio = TR_RATIO_INF;
876    else
877        ratio = TR_RATIO_NA;
878
879    return ratio;
880}
881
882void
883tr_sha1_to_hex( char * out, const uint8_t * sha1 )
884{
885    int i;
886    static const char hex[] = "0123456789abcdef";
887
888    for( i=0; i<20; ++i )
889    {
890        const unsigned int val = *sha1++;
891        *out++ = hex[val >> 4];
892        *out++ = hex[val & 0xf];
893    }
894
895    *out = '\0';
896}
897
898void
899tr_hex_to_sha1( uint8_t * out, const char * in )
900{
901    int i;
902    static const char hex[] = "0123456789abcdef";
903
904    for( i=0; i<20; ++i )
905    {
906        const int hi = strchr( hex, tolower( *in++ ) ) - hex;
907        const int lo = strchr( hex, tolower( *in++ ) ) - hex;
908        *out++ = (uint8_t)( (hi<<4) | lo );
909    }
910}
911
912/***
913****
914***/
915
916static tr_bool
917isValidURLChars( const char * url )
918{
919    const char * c;
920    static const char * rfc2396_valid_chars =
921        "abcdefghijklmnopqrstuvwxyz" /* lowalpha */
922        "ABCDEFGHIJKLMNOPQRSTUVWXYZ" /* upalpha */
923        "0123456789"                 /* digit */
924        "-_.!~*'()"                  /* mark */
925        ";/?:@&=+$,"                 /* reserved */
926        "<>#%<\""                    /* delims */
927        "{}|\\^[]`";                 /* unwise */
928
929    if( url == NULL )
930        return FALSE;
931
932    for( c = url; c && *c; ++c )
933        if( !strchr( rfc2396_valid_chars, *c ) )
934            return FALSE;
935
936    return TRUE;
937}
938
939/** @brief return TRUE if the url is a http or https url that Transmission understands */
940tr_bool
941tr_urlIsValidTracker( const char * url )
942{
943    tr_bool valid;
944    char * scheme = NULL;
945
946    valid = isValidURLChars( url ) 
947         && !tr_urlParse( url, -1, &scheme, NULL, NULL, NULL )
948         && ( scheme != NULL )
949         && ( !strcmp(scheme,"http") || !strcmp(scheme,"https") );
950
951    tr_free( scheme );
952    return valid;
953}
954
955/** @brief return TRUE if the url is a http or https or ftp or sftp url that Transmission understands */
956tr_bool
957tr_urlIsValid( const char * url )
958{
959    tr_bool valid;
960    char * scheme = NULL;
961
962    valid = isValidURLChars( url ) 
963         && !tr_urlParse( url, -1, &scheme, NULL, NULL, NULL )
964         && ( scheme != NULL )
965         && ( !strcmp(scheme,"http") || !strcmp(scheme,"https") || !strcmp(scheme,"ftp") || !strcmp(scheme,"sftp") );
966
967    tr_free( scheme );
968    return valid;
969}
970
971tr_bool
972tr_addressIsIP( const char * address )
973{
974    tr_address tempAddr;
975    return tr_pton(address, &tempAddr) != NULL;
976}
977
978int
979tr_urlParse( const char * url_in,
980             int          len,
981             char **      setme_protocol,
982             char **      setme_host,
983             int *        setme_port,
984             char **      setme_path )
985{
986    int          err;
987    int          port = 0;
988    int          n;
989    char *       tmp;
990    char *       pch;
991    const char * protocol = NULL;
992    const char * host = NULL;
993    const char * path = NULL;
994
995    tmp = tr_strndup( url_in, len );
996    if( ( pch = strstr( tmp, "://" ) ) )
997    {
998        *pch = '\0';
999        protocol = tmp;
1000        pch += 3;
1001/*fprintf( stderr, "protocol is [%s]... what's left is [%s]\n", protocol, pch);*/
1002        if( ( n = strcspn( pch, ":/" ) ) )
1003        {
1004            const int havePort = pch[n] == ':';
1005            host = pch;
1006            pch += n;
1007            *pch++ = '\0';
1008/*fprintf( stderr, "host is [%s]... what's left is [%s]\n", host, pch );*/
1009            if( havePort )
1010            {
1011                char * end;
1012                port = strtol( pch, &end, 10 );
1013                pch = end;
1014/*fprintf( stderr, "port is [%d]... what's left is [%s]\n", port, pch );*/
1015            }
1016            path = pch;
1017/*fprintf( stderr, "path is [%s]\n", path );*/
1018        }
1019    }
1020
1021    err = !host || !path || !protocol;
1022
1023    if( !err && !port )
1024    {
1025        if( !strcmp( protocol, "ftp" ) ) port = 21;
1026        if( !strcmp( protocol, "sftp" ) ) port = 22;
1027        if( !strcmp( protocol, "http" ) ) port = 80;
1028        if( !strcmp( protocol, "https" ) ) port = 443;
1029    }
1030
1031    if( !err )
1032    {
1033        if( setme_protocol ) *setme_protocol = tr_strdup( protocol );
1034
1035        if( setme_host ){ ( (char*)host )[-3] = ':'; *setme_host =
1036                              tr_strdup( host ); }
1037        if( setme_path ){ if( path[0] == '/' ) *setme_path = tr_strdup( path );
1038                          else { ( (char*)path )[-1] = '/'; *setme_path = tr_strdup( path - 1 ); } }
1039        if( setme_port ) *setme_port = port;
1040    }
1041
1042
1043    tr_free( tmp );
1044    return err;
1045}
1046
1047#include <string.h>
1048#include <openssl/sha.h>
1049#include <openssl/hmac.h>
1050#include <openssl/evp.h>
1051#include <openssl/bio.h>
1052#include <openssl/buffer.h>
1053
1054char *
1055tr_base64_encode( const void * input, int length, int * setme_len )
1056{
1057    int retlen = 0;
1058    char * ret = NULL;
1059
1060    if( input != NULL )
1061    {
1062        BIO * b64;
1063        BIO * bmem;
1064        BUF_MEM * bptr;
1065
1066        if( length < 1 )
1067            length = strlen( input );
1068
1069        bmem = BIO_new( BIO_s_mem( ) );
1070        b64 = BIO_new( BIO_f_base64( ) );
1071        b64 = BIO_push( b64, bmem );
1072        BIO_write( b64, input, length );
1073        (void) BIO_flush( b64 );
1074        BIO_get_mem_ptr( b64, &bptr );
1075        ret = tr_strndup( bptr->data, bptr->length );
1076        retlen = bptr->length;
1077        BIO_free_all( b64 );
1078    }
1079
1080    if( setme_len )
1081        *setme_len = retlen;
1082
1083    return ret;
1084}
1085
1086char *
1087tr_base64_decode( const void * input,
1088                  int          length,
1089                  int *        setme_len )
1090{
1091    char * ret;
1092    BIO *  b64;
1093    BIO *  bmem;
1094    int    retlen;
1095
1096    if( length < 1 )
1097        length = strlen( input );
1098
1099    ret = tr_new0( char, length );
1100    b64 = BIO_new( BIO_f_base64( ) );
1101    bmem = BIO_new_mem_buf( (unsigned char*)input, length );
1102    bmem = BIO_push( b64, bmem );
1103    retlen = BIO_read( bmem, ret, length );
1104    if( !retlen )
1105    {
1106        /* try again, but with the BIO_FLAGS_BASE64_NO_NL flag */
1107        BIO_free_all( bmem );
1108        b64 = BIO_new( BIO_f_base64( ) );
1109        BIO_set_flags( b64, BIO_FLAGS_BASE64_NO_NL );
1110        bmem = BIO_new_mem_buf( (unsigned char*)input, length );
1111        bmem = BIO_push( b64, bmem );
1112        retlen = BIO_read( bmem, ret, length );
1113    }
1114
1115    if( setme_len )
1116        *setme_len = retlen;
1117
1118    BIO_free_all( bmem );
1119    return ret;
1120}
1121
1122/***
1123****
1124***/
1125
1126int
1127tr_lowerBound( const void * key,
1128               const void * base,
1129               size_t       nmemb,
1130               size_t       size,
1131               int       (* compar)(const void* key, const void* arrayMember),
1132               tr_bool    * exact_match )
1133{
1134    size_t first = 0;
1135    const char * cbase = base;
1136    tr_bool exact = FALSE;
1137
1138    while( nmemb != 0 )
1139    {
1140        const size_t half = nmemb / 2;
1141        const size_t middle = first + half;
1142        const int c = compar( key, cbase + size*middle );
1143
1144        if( c <= 0 ) {
1145            if( c == 0 )
1146                exact = TRUE;
1147            nmemb = half;
1148        } else {
1149            first = middle + 1;
1150            nmemb = nmemb - half - 1;
1151        }
1152    }
1153
1154    *exact_match = exact;
1155
1156    return first;
1157}
1158
1159/***
1160****
1161***/
1162
1163char*
1164tr_utf8clean( const char * str, int max_len, tr_bool * err )
1165{
1166    char * ret;
1167    const char * end;
1168
1169    if( max_len < 0 )
1170        max_len = (int) strlen( str );
1171
1172    if( err != NULL )
1173        *err = FALSE;
1174
1175    if( tr_utf8_validate( str, max_len, &end  ) )
1176    {
1177        ret = tr_strndup( str, max_len );
1178    }
1179    else
1180    {
1181        const char zero = '\0';
1182        struct evbuffer * buf = evbuffer_new( );
1183
1184        while( !tr_utf8_validate ( str, max_len, &end ) )
1185        {
1186            const int good_len = end - str;
1187
1188            evbuffer_add( buf, str, good_len );
1189            max_len -= ( good_len + 1 );
1190            str += ( good_len + 1 );
1191            evbuffer_add( buf, "?", 1 );
1192
1193            if( err != NULL )
1194                *err = TRUE;
1195        }
1196
1197        evbuffer_add( buf, str, max_len );
1198        evbuffer_add( buf, &zero, 1 );
1199        ret = tr_memdup( EVBUFFER_DATA( buf ), EVBUFFER_LENGTH( buf ) );
1200        evbuffer_free( buf );
1201    }
1202
1203    assert( tr_utf8_validate( ret, -1, NULL ) );
1204    return ret;
1205}
1206
1207/***
1208****
1209***/
1210
1211struct number_range
1212{
1213    int low;
1214    int high;
1215};
1216
1217/**
1218 * This should be a single number (ex. "6") or a range (ex. "6-9").
1219 * Anything else is an error and will return failure.
1220 */
1221static tr_bool
1222parseNumberSection( const char * str, int len, struct number_range * setme )
1223{
1224    long a, b;
1225    tr_bool success;
1226    char * end;
1227    const int error = errno;
1228    char * tmp = tr_strndup( str, len );
1229
1230    errno = 0;
1231    a = b = strtol( tmp, &end, 10 );
1232    if( errno || ( end == tmp ) ) {
1233        success = FALSE;
1234    } else if( *end != '-' ) {
1235        b = a;
1236        success = TRUE;
1237    } else {
1238        const char * pch = end + 1;
1239        b = strtol( pch, &end, 10 );
1240        if( errno || ( pch == end ) )
1241            success = FALSE;
1242        else if( *end ) /* trailing data */
1243            success = FALSE;
1244        else
1245            success = TRUE;
1246    }
1247    tr_free( tmp );
1248
1249    setme->low = MIN( a, b );
1250    setme->high = MAX( a, b );
1251
1252    errno = error;
1253    return success;
1254}
1255
1256static int
1257compareInt( const void * va, const void * vb )
1258{
1259    const int a = *(const int *)va;
1260    const int b = *(const int *)vb;
1261    return a - b;
1262}
1263
1264/**
1265 * Given a string like "1-4" or "1-4,6,9,14-51", this allocates and returns an
1266 * array of setmeCount ints of all the values in the array.
1267 * For example, "5-8" will return [ 5, 6, 7, 8 ] and setmeCount will be 4.
1268 * It's the caller's responsibility to call tr_free() on the returned array.
1269 * If a fragment of the string can't be parsed, NULL is returned.
1270 */
1271int*
1272tr_parseNumberRange( const char * str_in, int len, int * setmeCount )
1273{
1274    int n = 0;
1275    int * uniq = NULL;
1276    char * str = tr_strndup( str_in, len );
1277    const char * walk;
1278    tr_list * ranges = NULL;
1279    tr_bool success = TRUE;
1280
1281    walk = str;
1282    while( walk && *walk && success ) {
1283        struct number_range range;
1284        const char * pch = strchr( walk, ',' );
1285        if( pch ) {
1286            success = parseNumberSection( walk, pch-walk, &range );
1287            walk = pch + 1;
1288        } else {
1289            success = parseNumberSection( walk, strlen( walk ), &range );
1290            walk += strlen( walk );
1291        }
1292        if( success )
1293            tr_list_append( &ranges, tr_memdup( &range, sizeof( struct number_range ) ) );
1294    }
1295
1296    if( !success )
1297    {
1298        *setmeCount = 0;
1299        uniq = NULL;
1300    }
1301    else
1302    {
1303        int i;
1304        int n2;
1305        tr_list * l;
1306        int * sorted = NULL;
1307
1308        /* build a sorted number array */
1309        n = n2 = 0;
1310        for( l=ranges; l!=NULL; l=l->next ) {
1311            const struct number_range * r = l->data;
1312            n += r->high + 1 - r->low;
1313        }
1314        sorted = tr_new( int, n );
1315        for( l=ranges; l!=NULL; l=l->next ) {
1316            const struct number_range * r = l->data;
1317            int i;
1318            for( i=r->low; i<=r->high; ++i )
1319                sorted[n2++] = i;
1320        }
1321        qsort( sorted, n, sizeof( int ), compareInt );
1322        assert( n == n2 );
1323
1324        /* remove duplicates */
1325        uniq = tr_new( int, n );
1326        for( i=n=0; i<n2; ++i )
1327            if( !n || uniq[n-1] != sorted[i] )
1328                uniq[n++] = sorted[i];
1329
1330        tr_free( sorted );
1331    }
1332
1333    /* cleanup */
1334    tr_list_free( &ranges, tr_free );
1335    tr_free( str );
1336
1337    /* return the result */
1338    *setmeCount = n;
1339    return uniq;
1340}
1341
1342/***
1343****
1344***/
1345
1346double
1347tr_truncd( double x, int decimal_places )
1348{
1349    static const int multiplier[] = { 1, 10, 100, 1000, 10000, 100000, 1000000, 10000000 };
1350    const int64_t i = multiplier[decimal_places];
1351    double x2 = (int64_t)(x*i);
1352    return x2 / i;
1353}
1354
1355char*
1356tr_strratio( char * buf, size_t buflen, double ratio, const char * infinity )
1357{
1358    if( (int)ratio == TR_RATIO_NA )
1359        tr_strlcpy( buf, _( "None" ), buflen );
1360    else if( (int)ratio == TR_RATIO_INF )
1361        tr_strlcpy( buf, infinity, buflen );
1362    else if( ratio < 10.0 )
1363        tr_snprintf( buf, buflen, "%.2f", tr_truncd( ratio, 2 ) );
1364    else if( ratio < 100.0 )
1365        tr_snprintf( buf, buflen, "%.1f", tr_truncd( ratio, 1 ) );
1366    else
1367        tr_snprintf( buf, buflen, "%'.0f", ratio );
1368    return buf;
1369}
1370
1371/***
1372****
1373***/
1374
1375int
1376tr_moveFile( const char * oldpath, const char * newpath, tr_bool * renamed )
1377{
1378    int in;
1379    int out;
1380    char * buf;
1381    struct stat st;
1382    off_t bytesLeft;
1383    off_t buflen;
1384
1385    /* make sure the old file exists */
1386    if( stat( oldpath, &st ) ) {
1387        const int err = errno;
1388        errno = err;
1389        return -1;
1390    }
1391    if( !S_ISREG( st.st_mode ) ) {
1392        errno = ENOENT;
1393        return -1;
1394    }
1395    bytesLeft = st.st_size;
1396
1397    /* make sure the target directory exists */
1398    {
1399        char * newdir = tr_dirname( newpath );
1400        int i = tr_mkdirp( newdir, 0777 );
1401        tr_free( newdir );
1402        if( i )
1403            return i;
1404    }
1405
1406    /* they might be on the same filesystem... */
1407    {
1408        const int i = rename( oldpath, newpath );
1409        if( renamed != NULL )
1410            *renamed = i == 0;
1411        if( !i )
1412            return 0;
1413    }
1414
1415    /* copy the file */
1416    in = tr_open_file_for_scanning( oldpath );
1417    out = tr_open_file_for_writing( newpath );
1418    buflen = stat( newpath, &st ) ? 4096 : st.st_blksize;
1419    buf = tr_new( char, buflen );
1420    while( bytesLeft > 0 )
1421    {
1422        ssize_t bytesWritten;
1423        const off_t bytesThisPass = MIN( bytesLeft, buflen );
1424        const int numRead = read( in, buf, bytesThisPass );
1425        if( numRead < 0 )
1426            break;
1427        bytesWritten = write( out, buf, numRead );
1428        if( bytesWritten < 0 )
1429            break;
1430        bytesLeft -= bytesWritten;
1431    }
1432
1433    /* cleanup */
1434    tr_free( buf );
1435    tr_close_file( out );
1436    tr_close_file( in );
1437    if( bytesLeft != 0 )
1438        return -1;
1439
1440    unlink( oldpath );
1441    return 0;
1442}
Note: See TracBrowser for help on using the repository browser.