source: trunk/libtransmission/utils.c @ 9709

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

(trunk libT) new utility function tr_timerAddMsec()

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