source: trunk/libtransmission/utils.c @ 9833

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

(trunk libT) remove dead code -- tr_assert()

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