source: trunk/libtransmission/utils.c @ 8808

Last change on this file since 8808 was 8808, checked in by charles, 13 years ago

(trunk) #2271: rounding problem in printf()

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