source: trunk/libtransmission/utils.c @ 9573

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

(trunk libT) #2595: "file too large error?" -- this can generate an EFBIG warning sparse preallocation is on and the peer makes an invalid request. At worst this kind of oddball case should generate a debug message notice, not a torrent error. Reported by Waldorf.

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