source: trunk/libtransmission/utils.c @ 9647

Last change on this file since 9647 was 9647, checked in by livings124, 12 years ago

cap libT's message log

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