2
0

Improved maintenance of marshal file & abort on ftl.

This commit is contained in:
Maarten Billemont 2019-07-28 21:03:37 -04:00
parent bf572a5e62
commit 360c0ade66
3 changed files with 59 additions and 31 deletions

View File

@ -106,17 +106,28 @@ MPMarshalledQuestion *mpw_marshal_question(
} }
MPMarshalledFile *mpw_marshal_file( MPMarshalledFile *mpw_marshal_file(
MPMarshalledUser *user, MPMarshalledData *data) { MPMarshalledFile *file, MPMarshalledUser *user, MPMarshalledData *data, MPMarshalInfo *info) {
MPMarshalledFile *file; if (!file) {
if (!user || !(file = malloc( sizeof( MPMarshalledFile ) ))) if (!(file = malloc( sizeof( MPMarshalledFile ) )))
return NULL; return NULL;
*file = (MPMarshalledFile){ *file = (MPMarshalledFile){};
.info = NULL, }
.user = user,
.data = data, if (user && user != file->user) {
}; mpw_marshal_user_free( &file->user );
file->user = user;
}
if (data && data != file->data) {
mpw_marshal_data_free( &file->data );
file->data = data;
}
if (info && info != file->info) {
mpw_marshal_info_free( &file->info );
file->info = info;
}
return file; return file;
} }
@ -153,6 +164,17 @@ void mpw_marshal_user_free(
mpw_free( user, sizeof( MPMarshalledUser ) ); mpw_free( user, sizeof( MPMarshalledUser ) );
} }
void mpw_marshal_data_free(
MPMarshalledData **data) {
if (!data || !*data)
return;
mpw_marshal_data_set_null( *data, NULL );
mpw_free_string( &(*data)->key );
mpw_free( data, sizeof( MPMarshalledData ) );
}
void mpw_marshal_file_free( void mpw_marshal_file_free(
MPMarshalledFile **file) { MPMarshalledFile **file) {
@ -160,12 +182,8 @@ void mpw_marshal_file_free(
return; return;
mpw_marshal_info_free( &(*file)->info ); mpw_marshal_info_free( &(*file)->info );
mpw_marshal_data_free( &(*file)->data );
mpw_marshal_user_free( &(*file)->user ); mpw_marshal_user_free( &(*file)->user );
if ((*file)->data) {
mpw_marshal_data_set_null( (*file)->data, NULL );
mpw_free_string( &(*file)->data->key );
mpw_free( &(*file)->data, sizeof( MPMarshalledData ) );
}
mpw_free( file, sizeof( MPMarshalledFile ) ); mpw_free( file, sizeof( MPMarshalledFile ) );
} }
@ -694,6 +712,16 @@ static const char *mpw_marshal_write_json(
const char *mpw_marshal_write( const char *mpw_marshal_write(
const MPMarshalFormat outFormat, MPMarshalledFile *file, MPMarshalError *error) { const MPMarshalFormat outFormat, MPMarshalledFile *file, MPMarshalError *error) {
if (!file) {
*error = (MPMarshalError){ .type = MPMarshalErrorMissing, "No file to marshal." };
return NULL;
}
if (file->data && file->data->key) {
*error = (MPMarshalError){ .type = MPMarshalErrorInternal, "Illegal file data." };
ftl( "Unexpected non-root file data." );
return NULL;
}
const char *out = NULL; const char *out = NULL;
switch (outFormat) { switch (outFormat) {
case MPMarshalFormatNone: case MPMarshalFormatNone:
@ -711,10 +739,7 @@ const char *mpw_marshal_write(
*error = (MPMarshalError){ MPMarshalErrorFormat, mpw_str( "Unsupported output format: %u", outFormat ) }; *error = (MPMarshalError){ MPMarshalErrorFormat, mpw_str( "Unsupported output format: %u", outFormat ) };
break; break;
} }
if (file) { mpw_marshal_file( file, NULL, NULL, mpw_marshal_read_info( out ) );
mpw_marshal_info_free( &file->info );
file->info = mpw_marshal_read_info( out );
}
return out; return out;
} }
@ -1057,7 +1082,7 @@ static MPMarshalledFile *mpw_marshal_read_flat(
mpw_free_strings( &fullName, &keyID, NULL ); mpw_free_strings( &fullName, &keyID, NULL );
mpw_free( &masterKey, MPMasterKeySize ); mpw_free( &masterKey, MPMasterKeySize );
MPMarshalledFile *file = mpw_marshal_file( user, NULL ); MPMarshalledFile *file = mpw_marshal_file( NULL, user, NULL, NULL );
if (!file) { if (!file) {
*error = (MPMarshalError){ MPMarshalErrorInternal, "Couldn't allocate a new marshal file." }; *error = (MPMarshalError){ MPMarshalErrorInternal, "Couldn't allocate a new marshal file." };
mpw_marshal_user_free( &user ); mpw_marshal_user_free( &user );
@ -1418,9 +1443,10 @@ static MPMarshalledFile *mpw_marshal_read_json(
mpw_set_json_data( data, json_file ); mpw_set_json_data( data, json_file );
json_object_put( json_file ); json_object_put( json_file );
MPMarshalledFile *file = mpw_marshal_file( user, data ); MPMarshalledFile *file = mpw_marshal_file( NULL, user, data, NULL );
if (!file) { if (!file) {
*error = (MPMarshalError){ MPMarshalErrorInternal, "Couldn't allocate a new marshal file." }; *error = (MPMarshalError){ MPMarshalErrorInternal, "Couldn't allocate a new marshal file." };
mpw_marshal_data_free( &data );
mpw_marshal_user_free( &user ); mpw_marshal_user_free( &user );
return NULL; return NULL;
} }
@ -1484,12 +1510,8 @@ MPMarshalledFile *mpw_marshal_read(
*error = (MPMarshalError){ MPMarshalErrorFormat, mpw_str( "Unsupported input format: %u", info->format ) }; *error = (MPMarshalError){ MPMarshalErrorFormat, mpw_str( "Unsupported input format: %u", info->format ) };
break; break;
} }
if (file) {
mpw_marshal_info_free( &(file->info) );
file->info = info;
}
return file; return mpw_marshal_file( file, NULL, NULL, info );
} }
const MPMarshalFormat mpw_format_named( const MPMarshalFormat mpw_format_named(

View File

@ -179,10 +179,10 @@ MPMarshalledSite *mpw_marshal_site(
* @return A question object (allocated), or NULL if the marshalled question couldn't be allocated. */ * @return A question object (allocated), or NULL if the marshalled question couldn't be allocated. */
MPMarshalledQuestion *mpw_marshal_question( MPMarshalledQuestion *mpw_marshal_question(
MPMarshalledSite *site, const char *keyword); MPMarshalledSite *site, const char *keyword);
/** Create a new file to marshal a user into. /** Create or update a marshal file descriptor.
* @return A file object (allocated), or NULL if the user is missing or the marshalled file couldn't be allocated. */ * @return The given file or new (allocated) if file is NULL; or NULL if the user is missing or the file couldn't be allocated. */
MPMarshalledFile *mpw_marshal_file( MPMarshalledFile *mpw_marshal_file(
MPMarshalledUser *user, MPMarshalledData *data); MPMarshalledFile *const file, MPMarshalledUser *user, MPMarshalledData *data, MPMarshalInfo *info);
//// Disposing. //// Disposing.
@ -191,6 +191,8 @@ void mpw_marshal_info_free(
MPMarshalInfo **info); MPMarshalInfo **info);
void mpw_marshal_user_free( void mpw_marshal_user_free(
MPMarshalledUser **user); MPMarshalledUser **user);
void mpw_marshal_data_free(
MPMarshalledData **data);
void mpw_marshal_file_free( void mpw_marshal_file_free(
MPMarshalledFile **file); MPMarshalledFile **file);

View File

@ -39,7 +39,11 @@ extern int mpw_verbosity;
#define mpw_log(level, format, ...) do { \ #define mpw_log(level, format, ...) do { \
if (mpw_verbosity >= level) { \ if (mpw_verbosity >= level) { \
mpw_log_do( level, format, ##__VA_ARGS__ ); \ mpw_log_do( level, format, ##__VA_ARGS__ ); \
}; } while (0) } \
if (level <= -2) { \
abort(); \
} \
} while (0)
#endif #endif
/** Logging internal state. */ /** Logging internal state. */