transmission/libtransmission/bencode.h

/*
 * This file Copyright (C) 2008 Charles Kerr <charles@rebelbase.com>
 *
 * This file is licensed by the GPL version 2.  Works owned by the
 * Transmission project are granted a special exemption to clause 2(b)
 * so that the bulk of its code can remain under the MIT license. 
 * This exemption does not extend to derived works not owned by
 * the Transmission project.
 *
 * $Id$
 */

#ifndef TR_BENCODE_H
#define TR_BENCODE_H 1

#include <inttypes.h> /* for int64_t */

enum
{
    TYPE_INT  = 1,
    TYPE_STR  = 2,
    TYPE_LIST = 4,
    TYPE_DICT = 8
};
    
typedef struct tr_benc
{
    char   type;
    union
    {
        int64_t i;
        struct
        {
            int    i;
            char * s;
        } s;
        struct
        {
            int alloc;
            int count;
            struct tr_benc * vals;
        } l;
    } val;
} tr_benc;

/***
****
***/

int tr_bencParse( const void      * buf,
                  const void      * bufend,
                  tr_benc         * setme_benc,
                  const uint8_t  ** setme_end );

int tr_bencLoad( const void  * buf,
                 int           buflen,
                 tr_benc     * setme_benc,
                 char       ** setme_end );

int    tr_bencLoadFile     ( const char * filename, tr_benc * );
int    tr_bencLoadJSONFile ( const char * filename, tr_benc * );

#if 0
void  tr_bencPrint( const tr_benc * );
#endif

void  tr_bencFree( tr_benc * );

char*  tr_bencSave         ( const tr_benc * val, int * len );
char*  tr_bencSaveAsJSON   ( const tr_benc * top, int * len );
int    tr_bencSaveFile     ( const char * filename, const tr_benc * );
int    tr_bencSaveJSONFile ( const char * filename, const tr_benc * );

void  tr_bencInitStr( tr_benc *, const void * str, int str_len );
void  tr_bencInitRaw( tr_benc *, const void * raw, size_t raw_len );
void  tr_bencInitInt( tr_benc *, int64_t num );
int   tr_bencInitDict( tr_benc *, int reserveCount );
int   tr_bencInitList( tr_benc *, int reserveCount );

/***
****
***/

int tr_bencListReserve( tr_benc *, int count );
tr_benc * tr_bencListAdd( tr_benc * );
tr_benc * tr_bencListAddInt( tr_benc *, int64_t val );
tr_benc * tr_bencListAddStr( tr_benc *, const char * val );
tr_benc * tr_bencListAddList( tr_benc *, int reserveCount );
tr_benc * tr_bencListAddDict( tr_benc *, int reserveCount );

int tr_bencListSize( const tr_benc * list );
tr_benc * tr_bencListChild( tr_benc * list, int n );

/***
****
***/

int tr_bencDictReserve( tr_benc *, int count );
int tr_bencDictRemove( tr_benc *, const char * key );
tr_benc * tr_bencDictAdd( tr_benc *, const char * key );
tr_benc * tr_bencDictAddDouble( tr_benc *, const char * key, double d );
tr_benc * tr_bencDictAddInt( tr_benc *, const char * key, int64_t val );
tr_benc * tr_bencDictAddStr( tr_benc *, const char * key, const char * val );
tr_benc * tr_bencDictAddList( tr_benc *, const char * key, int reserveCount );
tr_benc * tr_bencDictAddDict( tr_benc *, const char * key, int reserveCount );
tr_benc * tr_bencDictAddRaw( tr_benc *, const char * key,
                             const void *, size_t len );

tr_benc* tr_bencDictFind( tr_benc *, const char * key );
int tr_bencDictFindList( tr_benc *, const char * key, tr_benc ** setme );
int tr_bencDictFindDict( tr_benc *, const char * key, tr_benc ** setme );
int tr_bencDictFindInt( tr_benc *, const char * key, int64_t * setme );
int tr_bencDictFindDouble( tr_benc *, const char * key, double * setme );
int tr_bencDictFindStr( tr_benc *, const char * key, const char ** setme );
int tr_bencDictFindRaw( tr_benc *, const char * key,
                          const uint8_t ** setme_raw, size_t * setme_len );

/***
****
***/

int tr_bencGetInt( const tr_benc * val, int64_t * setme );
int tr_bencGetStr( const tr_benc * val, const char ** setme );
int tr_bencIsType( const tr_benc *, int type );
#define tr_bencIsInt( b )    tr_bencIsType( ( b ), TYPE_INT )
#define tr_bencIsDict( b )   tr_bencIsType( ( b ), TYPE_DICT )
#define tr_bencIsList( b )   tr_bencIsType( ( b ), TYPE_LIST )
#define tr_bencIsString( b ) tr_bencIsType( ( b ), TYPE_STR )

/**
***  Treat these as private -- they're only made public here
***  so that the unit tests can find them
**/

int  tr_bencParseInt( const uint8_t  * buf,
                      const uint8_t  * bufend,
                      const uint8_t ** setme_end, 
                      int64_t        * setme_val );

int  tr_bencParseStr( const uint8_t  * buf,
                      const uint8_t  * bufend,
                      const uint8_t ** setme_end, 
                      const uint8_t ** setme_str,
                      size_t         * setme_strlen );

#endif