2011-12-13 09:04:12 +00:00
# include "mphlr.h"
2011-12-18 21:40:02 +00:00
# include "rhizome.h"
2011-12-13 09:04:12 +00:00
2011-12-18 21:40:02 +00:00
long long rhizome_space = 0 ;
char * rhizome_datastore_path = NULL ;
2011-12-17 01:41:32 +00:00
2011-12-18 21:40:02 +00:00
sqlite3 * rhizome_db = NULL ;
2011-12-13 09:04:12 +00:00
int rhizome_opendb ( )
{
if ( rhizome_db ) return 0 ;
char dbname [ 1024 ] ;
if ( ! rhizome_datastore_path ) {
fprintf ( stderr , " Cannot open rhizome database -- no path specified \n " ) ;
exit ( 1 ) ;
}
if ( strlen ( rhizome_datastore_path ) > 1000 ) {
fprintf ( stderr , " Cannot open rhizome database -- data store path is too long \n " ) ;
exit ( 1 ) ;
}
snprintf ( dbname , 1024 , " %s/rhizome.db " , rhizome_datastore_path ) ;
int r = sqlite3_open ( dbname , & rhizome_db ) ;
if ( r ) {
fprintf ( stderr , " SQLite could not open database: %s \n " , sqlite3_errmsg ( rhizome_db ) ) ;
exit ( 1 ) ;
}
/* Read Rhizome configuration, and write it back out as we understand it. */
char conf [ 1024 ] ;
snprintf ( conf , 1024 , " %s/rhizome.conf " , rhizome_datastore_path ) ;
FILE * f = fopen ( conf , " r " ) ;
if ( f ) {
char line [ 1024 ] ;
line [ 0 ] = 0 ; fgets ( line , 1024 , f ) ;
while ( line [ 0 ] ) {
if ( sscanf ( line , " space=%lld " , & rhizome_space ) = = 1 ) {
rhizome_space * = 1024 ; /* Units are kilobytes */
}
line [ 0 ] = 0 ; fgets ( line , 1024 , f ) ;
}
fclose ( f ) ;
}
f = fopen ( conf , " w " ) ;
if ( f ) {
fprintf ( f , " space=%lld \n " , rhizome_space / 1024LL ) ;
fclose ( f ) ;
}
/* Create tables if required */
if ( sqlite3_exec ( rhizome_db , " PRAGMA auto_vacuum=2; " , NULL , NULL , NULL ) ) {
fprintf ( stderr , " SQLite could enable incremental vacuuming: %s \n " , sqlite3_errmsg ( rhizome_db ) ) ;
exit ( 1 ) ;
}
if ( sqlite3_exec ( rhizome_db , " CREATE TABLE IF NOT EXISTS GROUPS(id text not null primary key, priority integer, manifest blob, groupsecret blob); " , NULL , NULL , NULL ) )
{
fprintf ( stderr , " SQLite could not create GROUPS table: %s \n " , sqlite3_errmsg ( rhizome_db ) ) ;
exit ( 1 ) ;
}
if ( sqlite3_exec ( rhizome_db , " CREATE TABLE IF NOT EXISTS MANIFESTS(id text not null primary key, manifest blob, version integer, privatekey blob); " , NULL , NULL , NULL ) )
{
fprintf ( stderr , " SQLite could not create MANIFESTS table: %s \n " , sqlite3_errmsg ( rhizome_db ) ) ;
exit ( 1 ) ;
}
if ( sqlite3_exec ( rhizome_db , " CREATE TABLE IF NOT EXISTS FILES(id text not null primary key, data blob, length integer, highestpriority integer); " , NULL , NULL , NULL ) )
{
fprintf ( stderr , " SQLite could not create FILES table: %s \n " , sqlite3_errmsg ( rhizome_db ) ) ;
exit ( 1 ) ;
}
if ( sqlite3_exec ( rhizome_db , " CREATE TABLE IF NOT EXISTS FILEMANIFESTS(fileid text not null primary key, manifestid text not null); " , NULL , NULL , NULL ) )
{
fprintf ( stderr , " SQLite could not create FILEMANIFESTS table: %s \n " , sqlite3_errmsg ( rhizome_db ) ) ;
exit ( 1 ) ;
}
if ( sqlite3_exec ( rhizome_db , " CREATE TABLE IF NOT EXISTS MANIFESTGROUPS(manifestid text not null primary key, groupid text not null); " , NULL , NULL , NULL ) )
{
fprintf ( stderr , " SQLite could not create MANIFESTGROUPS table: %s \n " , sqlite3_errmsg ( rhizome_db ) ) ;
exit ( 1 ) ;
}
/* XXX Setup special groups, e.g., Serval Software and Serval Optional Data */
return 0 ;
}
/*
Convenience wrapper for executing an SQL command that returns a single int64 value
*/
long long sqlite_exec_int64 ( char * sqlformat , . . . )
{
if ( ! rhizome_db ) rhizome_opendb ( ) ;
va_list ap , ap2 ;
char sqlstatement [ 8192 ] ;
va_start ( ap , sqlformat ) ;
va_copy ( ap2 , ap ) ;
vsnprintf ( sqlstatement , 8192 , sqlformat , ap2 ) ; sqlstatement [ 8191 ] = 0 ;
va_end ( ap ) ;
sqlite3_stmt * statement ;
if ( sqlite3_prepare_v2 ( rhizome_db , sqlstatement , - 1 , & statement , NULL ) ! = SQLITE_OK )
{
sqlite3_close ( rhizome_db ) ;
rhizome_db = NULL ;
return WHY ( " Could not prepare sql statement. " ) ;
}
if ( sqlite3_step ( statement ) = = SQLITE_ROW )
{
if ( sqlite3_column_count ( statement ) ! = 1 ) {
sqlite3_finalize ( statement ) ;
return - 1 ;
}
long long result = sqlite3_column_int ( statement , 0 ) ;
sqlite3_finalize ( statement ) ;
return result ;
}
sqlite3_finalize ( statement ) ;
return - 1 ;
}
long long rhizome_database_used_bytes ( )
{
long long db_page_size = sqlite_exec_int64 ( " PRAGMA page_size; " ) ;
long long db_page_count = sqlite_exec_int64 ( " PRAGMA page_count; " ) ;
long long db_free_page_count = sqlite_exec_int64 ( " PRAGMA free_count; " ) ;
return db_page_size * ( db_page_count - db_free_page_count ) ;
}
int rhizome_make_space ( int group_priority , long long bytes )
{
sqlite3_stmt * statement ;
/* Asked for impossibly large amount */
if ( bytes > = ( rhizome_space - 65536 ) ) return - 1 ;
long long db_used = rhizome_database_used_bytes ( ) ;
/* If there is already enough space now, then do nothing more */
if ( db_used < = ( rhizome_space - bytes - 65536 ) ) return 0 ;
/* Okay, not enough space, so free up some. */
char sql [ 1024 ] ;
snprintf ( sql , 1024 , " select id,length from files where highestpriority<%d order by descending length " , group_priority ) ;
if ( sqlite3_prepare_v2 ( rhizome_db , sql , - 1 , & statement , NULL ) ! = SQLITE_OK )
{
fprintf ( stderr , " SQLite error running query '%s': %s \n " , sql , sqlite3_errmsg ( rhizome_db ) ) ;
sqlite3_close ( rhizome_db ) ;
rhizome_db = NULL ;
exit ( - 1 ) ;
}
while ( bytes > ( rhizome_space - 65536 - rhizome_database_used_bytes ( ) ) & & sqlite3_step ( statement ) = = SQLITE_ROW )
{
/* Make sure we can drop this blob, and if so drop it, and recalculate number of bytes required */
char * id ;
long long length ;
/* Get values */
if ( sqlite3_column_type ( statement , 0 ) = = SQLITE_TEXT ) id = sqlite3_column_text ( statement , 0 ) ;
else {
fprintf ( stderr , " Incorrect type in id column of files table. \n " ) ;
continue ; }
if ( sqlite3_column_type ( statement , 1 ) = = SQLITE_INTEGER ) length = sqlite3_column_int ( statement , 1 ) ;
else {
fprintf ( stderr , " Incorrect type in length column of files table. \n " ) ;
continue ; }
/* Try to drop this file from storage, discarding any references that do not trump the priority of this
request . The query done earlier should ensure this , but it doesn ' t hurt to be paranoid , and it also
protects against inconsistency in the database . */
rhizome_drop_stored_file ( id , group_priority + 1 ) ;
}
sqlite3_finalize ( statement ) ;
long long equal_priority_larger_file_space_used = sqlite_exec_int64 ( " SELECT COUNT(length) FROM FILES WHERE highestpriority=%d and length>%lld " , group_priority , bytes ) ;
/* XXX Get rid of any equal priority files that are larger than this one */
/* XXX Get rid of any higher priority files that are not relevant in this time or location */
/* Couldn't make space */
return WHY ( " Not implemented " ) ;
}
/* Drop the specified file from storage, and any manifests that reference it,
provided that none of those manifests are being retained at a higher priority
than the maximum specified here . */
int rhizome_drop_stored_file ( char * id , int maximum_priority )
{
char sql [ 1024 ] ;
sqlite3_stmt * statement ;
int cannot_drop = 0 ;
if ( strlen ( id ) > 70 ) return - 1 ;
snprintf ( sql , 1024 , " select manifests.id from manifests,filemanifests where manifests.id==filemanifests.manifestid and filemanifests.fileid='%s' " ,
id ) ;
if ( sqlite3_prepare_v2 ( rhizome_db , sql , - 1 , & statement , NULL ) ! = SQLITE_OK )
{
fprintf ( stderr , " SQLite error running query '%s': %s \n " , sql , sqlite3_errmsg ( rhizome_db ) ) ;
sqlite3_close ( rhizome_db ) ;
rhizome_db = NULL ;
exit ( - 1 ) ;
}
while ( sqlite3_step ( statement ) = = SQLITE_ROW )
{
/* Find manifests for this file */
char * id ;
if ( sqlite3_column_type ( statement , 0 ) = = SQLITE_TEXT ) id = sqlite3_column_text ( statement , 0 ) ;
else {
fprintf ( stderr , " Incorrect type in id column of manifests table. \n " ) ;
continue ; }
/* Check that manifest is not part of a higher priority group.
If so , we cannot drop the manifest or the file .
However , we will keep iterating , as we can still drop any other manifests pointing to this file
that are lower priority , and thus free up a little space . */
if ( rhizome_manifest_priority ( id ) > maximum_priority ) {
cannot_drop = 1 ;
} else {
sqlite_exec_int64 ( " delete from filemanifests where manifestid='%s'; " , id ) ;
sqlite_exec_int64 ( " delete from manifests where manifestid='%s'; " , id ) ;
sqlite_exec_int64 ( " delete from manifestgroups where manifestid='%s'; " , id ) ;
}
}
sqlite3_finalize ( statement ) ;
if ( ! cannot_drop ) {
sqlite_exec_int64 ( " delete from filemanifests where fileid='%s'; " , id ) ;
sqlite_exec_int64 ( " delete from files where id='%s'; " , id ) ;
}
return 0 ;
}
/* XXX Requires a messy join that might be slow. */
int rhizome_manifest_priority ( char * id )
{
long long result = sqlite_exec_int64 ( " select max(groups.priorty) from groups,manifests,manifestgroups where manifests.id='%s' and groups.id=manifestgroups.groupid and manifestgroups.manifestid=manifests.id; " , id ) ;
return result ;
}
/* Import a bundle from the inbox folder.
Check that the manifest prototype is valid , and if so , complete it , and sign it if required and possible .
The file should be included in the specified rhizome groups , if possible .
( some groups may be closed groups that we do not have the private key for . )
*/
int rhizome_bundle_import ( char * bundle , char * groups [ ] , int verifyP , int checkFileP , int signP )
{
char filename [ 1024 ] ;
char manifestname [ 1024 ] ;
char buffer [ 1024 ] ;
snprintf ( filename , 1024 , " %s/import/file.%s " , rhizome_datastore_path , bundle ) ; filename [ 1023 ] = 0 ;
2011-12-18 19:16:34 +00:00
snprintf ( manifestname , 1024 , " %s/import/manifest.%s " , rhizome_datastore_path , bundle ) ; manifestname [ 1023 ] = 0 ;
2011-12-13 09:04:12 +00:00
/* Open files */
rhizome_manifest * m = rhizome_read_manifest_file ( manifestname ) ;
if ( ! m ) return WHY ( " Could not read manifest file. " ) ;
char hexhash [ SHA512_DIGEST_STRING_LENGTH ] ;
if ( checkFileP | | signP ) {
if ( rhizome_hash_file ( filename , hexhash ) )
{ rhizome_manifest_free ( m ) ; return WHY ( " Could not hash file. " ) ; }
}
if ( verifyP )
{
/* Make sure hashes match.
Make sure that no signature verification errors were spotted on loading . */
int verifyErrors = 0 ;
char mhexhash [ 1024 ] ;
if ( checkFileP ) {
if ( rhizome_manifest_get ( m , " filehash " , mhexhash ) = = 0 )
if ( strcmp ( hexhash , mhexhash ) ) verifyErrors + + ; }
if ( m - > signature_errors ) verifyErrors + = m - > signature_errors ;
if ( verifyErrors ) {
rhizome_manifest_free ( m ) ;
unlink ( manifestname ) ;
unlink ( filename ) ;
return WHY ( " Errors encountered verifying bundle manifest " ) ;
}
}
if ( ! verifyP ) {
if ( rhizome_manifest_get ( m , " id " , buffer ) ! = 0 ) {
/* No bundle id (256 bit random string being a public key in the NaCl CryptoSign crypto system),
so create one , and keep the private key handy . */
rhizome_manifest_createid ( m ) ;
}
rhizome_manifest_set ( m , " filehash " , hexhash ) ;
if ( rhizome_manifest_get ( m , " version " , buffer ) ! = 0 )
/* Version not set, so set one */
rhizome_manifest_set_ll ( m , " version " , overlay_time_in_ms ( ) ) ;
rhizome_manifest_set_ll ( m , " first_byte " , 0 ) ;
rhizome_manifest_set_ll ( m , " last_byte " , rhizome_file_size ( filename ) ) ;
}
/* Convert to final form for signing and writing to disk */
rhizome_manifest_pack_variables ( m ) ;
/* Sign it */
if ( signP ) rhizome_manifest_sign ( m ) ;
/* Add group memberships */
int i ;
for ( i = 0 ; groups [ i ] ; i + + ) rhizome_manifest_add_group ( m , groups [ i ] ) ;
/* Write manifest back to disk */
if ( rhizome_write_manifest_file ( m , manifestname ) ) {
rhizome_manifest_free ( m ) ;
return WHY ( " Could not write manifest file. " ) ;
}
/* Okay, it is written, and can be put directly into the rhizome database now */
int r = rhizome_store_bundle ( m , filename ) ;
if ( ! r ) {
unlink ( manifestname ) ;
unlink ( filename ) ;
return 0 ;
}
return - 1 ;
}
/* Update an existing Rhizome bundle */
int rhizome_bundle_push_update ( char * id , long long version , unsigned char * data , int appendP )
{
return WHY ( " Not implemented " ) ;
}
rhizome_manifest * rhizome_read_manifest_file ( char * filename )
{
rhizome_manifest * m = calloc ( sizeof ( rhizome_manifest ) , 1 ) ;
if ( ! m ) return NULL ;
FILE * f = fopen ( filename , " r " ) ;
2011-12-18 21:34:31 +00:00
if ( ! f ) { WHY ( " Could not open manifest file for reading. " ) ;
rhizome_manifest_free ( m ) ; return NULL ; }
2011-12-13 09:04:12 +00:00
m - > manifest_bytes = fread ( m - > manifestdata , 1 , MAX_MANIFEST_BYTES , f ) ;
fclose ( f ) ;
/* Parse out variables, signature etc */
int ofs = 0 ;
while ( ofs < m - > manifest_bytes & & m - > manifestdata [ ofs ] )
{
int i ;
char line [ 1024 ] , var [ 1024 ] , value [ 1024 ] ;
while ( ofs < m - > manifest_bytes & &
( m - > manifestdata [ ofs ] = = 0x0a | |
m - > manifestdata [ ofs ] = = 0x09 | |
m - > manifestdata [ ofs ] = = 0x20 | |
m - > manifestdata [ ofs ] = = 0x0d ) ) ofs + + ;
for ( i = 0 ; i < ( ofs - m - > manifest_bytes )
& & ( i < 1023 )
& & m - > manifestdata [ ofs ] ! = 0x00
& & m - > manifestdata [ ofs ] ! = 0x0d
& & m - > manifestdata [ ofs ] ! = 0x0a ; i + + )
line [ i ] = m - > manifestdata [ ofs + i ] ;
line [ i ] = 0 ;
/* Ignore blank lines */
if ( line [ 0 ] = = 0 ) continue ;
if ( sscanf ( line , " %[^=]=%[^ \n \r ] " , var , value ) = = 2 )
{
if ( rhizome_manifest_get ( m , var , NULL ) = = 0 ) {
WHY ( " Error in manifest file (duplicate variable -- keeping first value). " ) ;
}
if ( m - > var_count < MAX_MANIFEST_VARS )
{
m - > vars [ m - > var_count ] = strdup ( var ) ;
m - > values [ m - > var_count ] = strdup ( value ) ;
m - > var_count + + ;
}
}
else
{
/* Error in manifest file.
Silently ignore for now . */
WHY ( " Error in manifest file (badly formatted line). " ) ;
}
}
/* The null byte gets included in the check sum */
if ( ofs < m - > manifest_bytes ) ofs + + ;
/* Remember where the text ends */
int end_of_text = ofs ;
/* Calculate hash of the text part of the file, as we need to couple this with
each signature block to */
2011-12-18 21:58:39 +00:00
unsigned char manifest_hash [ crypto_hash_sha512_BYTES ] ;
2011-12-13 09:04:12 +00:00
crypto_hash ( manifest_hash , m - > manifestdata , end_of_text ) ;
/* Read signature blocks from file.
XXX - What additional information / restrictions should the
signatures have ? start / expiry times ? geo bounding box ?
Those elements all need to be included in the hash */
WHY ( " Signature verification not implemented " ) ;
WHY ( " Group membership signature reading not implemented (are we still doing it this way?) " ) ;
m - > manifest_bytes = end_of_text ;
WHY ( " Incomplete " ) ;
rhizome_manifest_free ( m ) ;
return NULL ;
}
int rhizome_hash_file ( char * filename , char * hash_out )
{
/* Gnarf! NaCl's crypto_hash() function needs the whole file passed in in one
go . Trouble is , we need to run Serval DNA on filesystems that lack mmap ( ) ,
and may be very resource constrained . Thus we need a streamable SHA - 512
implementation .
*/
FILE * f = fopen ( filename , " r " ) ;
if ( ! f ) return WHY ( " Could not open file for reading to calculage SHA512 hash. " ) ;
unsigned char buffer [ 8192 ] ;
int r ;
SHA512_CTX context ;
SHA512_Init ( & context ) ;
while ( ! feof ( f ) ) {
r = fread ( buffer , 1 , 8192 , f ) ;
if ( r > 0 ) SHA512_Update ( & context , buffer , r ) ;
}
SHA512_End ( & context , ( char * ) hash_out ) ;
return 0 ;
}
int rhizome_manifest_get ( rhizome_manifest * m , char * var , char * out )
{
int i ;
if ( ! m ) return - 1 ;
for ( i = 0 ; i < m - > var_count ; i + + )
if ( ! strcmp ( m - > vars [ i ] , var ) ) {
if ( out ) strcpy ( m - > values [ i ] , out ) ;
return 0 ;
}
return - 1 ;
}
2011-12-18 19:16:34 +00:00
long long rhizome_manifest_get_ll ( rhizome_manifest * m , char * var )
{
int i ;
if ( ! m ) return - 1 ;
for ( i = 0 ; i < m - > var_count ; i + + )
if ( ! strcmp ( m - > vars [ i ] , var ) )
return strtoll ( m - > values [ i ] , NULL , 10 ) ;
return - 1 ;
}
2011-12-13 09:04:12 +00:00
int rhizome_manifest_set ( rhizome_manifest * m , char * var , char * value )
{
int i ;
if ( ! m ) return - 1 ;
for ( i = 0 ; i < m - > var_count ; i + + )
if ( ! strcmp ( m - > vars [ i ] , var ) ) {
free ( m - > values [ i ] ) ;
m - > values [ i ] = strdup ( value ) ;
return 0 ;
}
if ( m - > var_count > = MAX_MANIFEST_VARS ) return - 1 ;
m - > vars [ m - > var_count ] = strdup ( var ) ;
m - > values [ m - > var_count ] = strdup ( value ) ;
m - > var_count + + ;
return 0 ;
}
int rhizome_manifest_set_ll ( rhizome_manifest * m , char * var , long long value )
{
char svalue [ 100 ] ;
snprintf ( svalue , 1024 , " %lld " , value ) ;
return rhizome_manifest_set ( m , var , svalue ) ;
}
long long rhizome_file_size ( char * filename )
{
FILE * f ;
/* XXX really should just use stat instead of opening the file */
f = fopen ( filename , " r " ) ;
fseek ( f , 0 , SEEK_END ) ;
long long size = ftello ( f ) ;
fclose ( f ) ;
return size ;
}
void rhizome_manifest_free ( rhizome_manifest * m )
{
if ( ! m ) return ;
int i ;
for ( i = 0 ; i < m - > var_count ; i + + )
{ free ( m - > vars [ i ] ) ; free ( m - > values [ i ] ) ; }
WHY ( " Doesn't free signatures yet " ) ;
free ( m ) ;
return ;
}
/* Convert variable list to string, complaining if it ends up
too long .
Signatures etc will be added later . */
int rhizome_manifest_pack_variables ( rhizome_manifest * m )
{
int i , ofs = 0 ;
for ( i = 0 ; i < m - > var_count ; i + + )
{
if ( ( ofs + strlen ( m - > vars [ i ] ) + 1 + strlen ( m - > values [ i ] ) + 1 + 1 ) > MAX_MANIFEST_BYTES )
return WHY ( " Manifest variables too long in total to fit in MAX_MANIFEST_BYTES " ) ;
snprintf ( ( char * ) & m - > manifestdata [ ofs ] , MAX_MANIFEST_BYTES - ofs , " %s=%s \n " ,
m - > vars [ i ] , m - > values [ i ] ) ;
ofs + = strlen ( ( char * ) & m - > manifestdata [ ofs ] ) ;
}
m - > manifest_bytes = ofs ;
return 0 ;
}
/* Sign this manifest using our own private CryptoSign key */
int rhizome_manifest_sign ( rhizome_manifest * m )
{
return WHY ( " Not implemented. " ) ;
}
int rhizome_write_manifest_file ( rhizome_manifest * m , char * filename )
{
if ( ! m ) return WHY ( " Manifest is null. " ) ;
if ( ! m - > finalised ) return WHY ( " Manifest must be finalised before it can be written. " ) ;
FILE * f = fopen ( filename , " w " ) ;
int r = fwrite ( m - > manifestdata , m - > manifest_bytes , 1 , f ) ;
fclose ( f ) ;
if ( r ! = 1 ) return WHY ( " Failed to fwrite() manifest file. " ) ;
return 0 ;
}
int rhizome_manifest_createid ( rhizome_manifest * m )
{
2011-12-17 01:41:32 +00:00
return crypto_sign_edwards25519sha512batch_keypair ( m - > cryptoSignPublic , m - > cryptoSignSecret ) ;
2011-12-13 09:04:12 +00:00
}
/*
Store the specified manifest into the sqlite database .
We assume that sufficient space has been made for us .
The manifest should be finalised , and so we don ' t need to
look at the underlying manifest file , but can just write m - > manifest_data
as a blob .
associated_filename needs to be read in and stored as a blob . Hopefully that
can be done in pieces so that we don ' t have memory exhaustion issues on small
architectures . However , we do know it ' s hash apriori from m , and so we can
2011-12-17 01:41:32 +00:00
skip loading the file in if it is already stored . mmap ( ) apparently works on
Linux FAT file systems , and is probably the best choice since it doesn ' t need
all pages to be in RAM at the same time .
SQLite does allow modifying of blobs once stored in the database .
The trick is to insert the blob as all zeroes using a special function , and then
substitute bytes in the blog progressively .
2011-12-13 09:04:12 +00:00
We need to also need to create the appropriate row ( s ) in the MANIFESTS , FILES ,
FILEMANIFESTS and MANIFESTGROUPS tables .
*/
int rhizome_store_bundle ( rhizome_manifest * m , char * associated_filename )
{
2011-12-18 19:16:34 +00:00
char sqlcmd [ 1024 ] ;
const char * cmdtail ;
2011-12-17 01:41:32 +00:00
if ( ! m - > finalised ) return WHY ( " Manifest was not finalised " ) ;
if ( m - > fileLength > 0 )
if ( rhizome_store_file ( associated_filename , m - > fileHexHash , m - > fileHighestPriority ) )
return WHY ( " Could not store associated file " ) ;
/* XXX Store manifest itself, and create relationships to groups etc. */
2011-12-18 19:16:34 +00:00
/* Store manifest */
snprintf ( sqlcmd , 1024 , " INSERT INTO MANIFESTS(id,manifest,version,privatekey,inserttime) VALUES('%s',?,%lld,'%s',%lld); " ,
2011-12-18 21:58:39 +00:00
rhizome_safe_encode ( m - > cryptoSignPublic , crypto_sign_edwards25519sha512batch_PUBLICKEYBYTES ) ,
2011-12-18 19:16:34 +00:00
m - > version ,
2011-12-18 21:58:39 +00:00
rhizome_safe_encode ( m - > cryptoSignSecret , crypto_sign_edwards25519sha512batch_SECRETKEYBYTES ) ,
2011-12-18 19:16:34 +00:00
overlay_time_in_ms ( ) ) ;
sqlite3_stmt * statement ;
if ( sqlite3_prepare_v2 ( rhizome_db , sqlcmd , strlen ( sqlcmd ) + 1 , & statement , & cmdtail )
! = SQLITE_OK )
return WHY ( sqlite3_errmsg ( rhizome_db ) ) ;
/* Bind appropriate sized zero-filled blob to data field */
sqlite3_bind_blob ( statement , 1 , m - > manifestdata , m - > manifest_bytes , SQLITE_TRANSIENT ) ;
if ( rhizome_finish_sqlstatement ( statement ) )
return WHY ( " SQLite3 failed to insert row for manifest " ) ;
2011-12-13 09:04:12 +00:00
return WHY ( " Not implemented. " ) ;
}
2011-12-18 19:16:34 +00:00
int rhizome_finish_sqlstatement ( sqlite3_stmt * statement )
{
/* Do actual insert, and abort if it fails */
int dud = 0 ;
if ( sqlite3_step ( statement ) ! = SQLITE_OK ) dud + + ;
if ( sqlite3_finalize ( statement ) ) dud + + ;
if ( dud ) WHY ( sqlite3_errmsg ( rhizome_db ) ) ;
return dud ;
}
/* Like sqlite_encode_binary(), but with a fixed rotation to make comparison of
string prefixes easier . Also , returns string directly for convenience .
The rotoring through four return strings is so that this function can be used
safely inline in sprintf ( ) type functions , which makes composition of sql statements
easier . */
int rse_rotor = 0 ;
char rse_out [ 4 ] [ 129 ] ;
char * rhizome_safe_encode ( unsigned char * in , int len )
{
char * r = rse_out [ rse_rotor ] ;
rse_rotor + + ;
rse_rotor & = 3 ;
int i , o = 0 ;
for ( i = 0 ; i < len ; i + + )
{
if ( o < = 127 )
switch ( in [ i ] )
{
case 0x00 : case 0x01 : case ' \' ' :
r [ o + + ] = 0x01 ;
r [ o + + ] = in [ i ] ^ 0x80 ;
break ;
default :
r [ o + + ] = in [ i ] ;
}
}
return r ;
}
2011-12-17 01:41:32 +00:00
/* The following function just stores the file (or silently returns if it already
exists ) .
The relationships of manifests to this file are the responsibility of the
caller . */
int rhizome_store_file ( char * file , char * hash , int priority ) {
int fd = open ( file , O_RDONLY ) ;
if ( fd < 0 ) return WHY ( " Could not open associated file " ) ;
struct stat stat ;
if ( fstat ( fd , & stat ) ) {
close ( fd ) ;
return WHY ( " Could not stat() associated file " ) ;
}
unsigned char * addr =
mmap ( NULL , stat . st_size , PROT_READ , MAP_FILE | MAP_SHARED , fd , 0 ) ;
if ( addr = = MAP_FAILED ) {
close ( fd ) ;
return WHY ( " mmap() of associated file failed . " ) ;
}
/* Get hash of file if not supplied */
char hexhash [ SHA512_DIGEST_STRING_LENGTH ] ;
if ( ! hash )
{
/* Hash the file */
SHA512_CTX c ;
SHA512_Init ( & c ) ;
SHA512_Update ( & c , addr , stat . st_size ) ;
SHA512_End ( & c , hexhash ) ;
hash = hexhash ;
}
/* INSERT INTO FILES(id as text, data blob, length integer, highestpriority integer).
BUT , we have to do this incrementally so that we can handle blobs larger than available memory .
This is possible using :
int sqlite3_bind_zeroblob ( sqlite3_stmt * , int , int n ) ;
That binds an all zeroes blob to a field . We can then populate the data by
opening a handle to the blob using :
int sqlite3_blob_write ( sqlite3_blob * , const void * z , int n , int iOffset ) ;
*/
char sqlcmd [ 1024 ] ;
const char * cmdtail ;
/* See if the file is already stored, and if so, don't bother storing it again */
int count = sqlite_exec_int64 ( " SELECT COUNT(*) FROM FILES WHERE id='%s' AND datavalid!=0; " , hash ) ;
if ( count = = 1 ) {
/* File is already stored, so just update the highestPriority field if required. */
long long storedPriority = sqlite_exec_int64 ( " SELECT highestPriority FROM FILES WHERE id='%s' AND datavalid!=0 " , hash ) ;
if ( storedPriority < priority )
{
snprintf ( sqlcmd , 1024 , " UPDATE FILES SET highestPriority=%d WHERE id='%s'; " ,
priority , hash ) ;
if ( sqlite3_exec ( rhizome_db , sqlcmd , NULL , NULL , NULL ) ! = SQLITE_OK ) {
close ( fd ) ;
WHY ( sqlite3_errmsg ( rhizome_db ) ) ;
return WHY ( " SQLite failed to update highestPriority field for stored file. " ) ;
}
}
close ( fd ) ;
return 0 ;
} else if ( count > 1 ) {
/* This should never happen! */
return WHY ( " Duplicate records for a file in the rhizome database. Database probably corrupt. " ) ;
}
2011-12-18 19:16:34 +00:00
/* Okay, so there are no records that match, but we should delete any half-baked record (with datavalid=0) so that the insert below doesn't fail.
Don ' t worry about the return result , since it might not delete any records . */
2011-12-17 01:41:32 +00:00
sqlite3_exec ( rhizome_db , " DELETE FROM FILES WHERE datavalid=0; " , NULL , NULL , NULL ) ;
snprintf ( sqlcmd , 1024 , " INSERT INTO FILES(id,data,length,highestpriority,datavalid) VALUES('%s',?,%lld,%d,0); " ,
hash , ( long long ) stat . st_size , priority ) ;
sqlite3_stmt * statement ;
if ( sqlite3_prepare_v2 ( rhizome_db , sqlcmd , strlen ( sqlcmd ) + 1 , & statement , & cmdtail )
! = SQLITE_OK )
{
close ( fd ) ;
return WHY ( sqlite3_errmsg ( rhizome_db ) ) ;
}
/* Bind appropriate sized zero-filled blob to data field */
sqlite3_bind_zeroblob ( statement , 1 , stat . st_size ) ;
/* Do actual insert, and abort if it fails */
int dud = 0 ;
if ( sqlite3_step ( statement ) ! = SQLITE_OK ) dud + + ;
if ( sqlite3_finalize ( statement ) ) dud + + ;
if ( dud ) {
close ( fd ) ;
WHY ( sqlite3_errmsg ( rhizome_db ) ) ;
return WHY ( " SQLite3 failed to insert row for file " ) ;
}
/* Get rowid for inserted row, so that we can modify the blob */
int rowid = sqlite3_last_insert_rowid ( rhizome_db ) ;
if ( rowid < 1 ) {
close ( fd ) ;
WHY ( sqlite3_errmsg ( rhizome_db ) ) ;
return WHY ( " SQLite3 failed return rowid of inserted row " ) ;
}
sqlite3_blob * blob ;
if ( sqlite3_blob_open ( rhizome_db , " main " , " FILES " , " data " , rowid ,
1 /* read/write */ ,
& blob ) ! = SQLITE_OK )
{
WHY ( sqlite3_errmsg ( rhizome_db ) ) ;
close ( fd ) ;
sqlite3_blob_close ( blob ) ;
return WHY ( " SQLite3 failed to open file blob for writing " ) ;
}
{
long long i ;
for ( i = 0 ; i < stat . st_size ; i + = 65536 )
{
int n = 65536 ;
if ( i + n > stat . st_size ) n = stat . st_size - i ;
if ( sqlite3_blob_write ( blob , & addr [ i ] , n , i ) ! = SQLITE_OK ) dud + + ;
}
}
if ( dud ) {
WHY ( sqlite3_errmsg ( rhizome_db ) ) ;
close ( fd ) ;
sqlite3_blob_close ( blob ) ;
return WHY ( " SQLite3 failed write all blob data " ) ;
}
/* Get things consistent */
sqlite3_exec ( rhizome_db , " COMMIT; " , NULL , NULL , NULL ) ;
return WHY ( " Not implemented " ) ;
}
2011-12-13 09:04:12 +00:00
/*
Adds a group that this bundle should be present in . If we have the means to sign
the bundle as a member of that group , then we create the appropriate signature block .
The group signature blocks , like all signature blocks , will be appended to the
manifest data during the finalisation process .
*/
int rhizome_manifest_add_group ( rhizome_manifest * m , char * groupid )
{
return WHY ( " Not implemented. " ) ;
}
2011-12-18 19:16:34 +00:00
int rhizome_manifest_finalise ( rhizome_manifest * m )
{
/* set fileLength */
/* set fileHexHash */
/* set fileHighestPriority based on group associations */
/* set version of manifest, either from version variable, or using current time */
if ( rhizome_manifest_get ( m , " version " , NULL ) )
{
/* No version set */
m - > version = overlay_time_in_ms ( ) ;
rhizome_manifest_set_ll ( m , " version " , m - > version ) ;
}
else
m - > version = rhizome_manifest_get_ll ( m , " version " ) ;
/* mark manifest as finalised */
return WHY ( " Not implemented. " ) ;
}