source: trunk/libtransmission/utils.c @ 9604

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

(trunk libT) make tr_msgLoggingIsActive() an inlined function. suggested by BentMyWookie?

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