Storage API: add safety check for cases when there is no storage used.

pull/466/head
Victor Julien 13 years ago
parent caf730d988
commit f06694d0c1

@ -52,7 +52,8 @@ void FlowFreeStorageById(Flow *f, int id) {
} }
void FlowFreeStorage(Flow *f) { void FlowFreeStorage(Flow *f) {
StorageFreeAll((Storage *)((void *)f + sizeof(Flow)), STORAGE_FLOW); if (FlowStorageSize() > 0)
StorageFreeAll((Storage *)((void *)f + sizeof(Flow)), STORAGE_FLOW);
} }
int FlowStorageRegister(const char *name, const unsigned int size, void *(*Init)(unsigned int), void (*Free)(void *)) { int FlowStorageRegister(const char *name, const unsigned int size, void *(*Init)(unsigned int), void (*Free)(void *)) {

@ -48,7 +48,8 @@ void HostFreeStorageById(Host *h, int id) {
} }
void HostFreeStorage(Host *h) { void HostFreeStorage(Host *h) {
StorageFreeAll((Storage *)((void *)h + sizeof(Host)), STORAGE_HOST); if (HostStorageSize() > 0)
StorageFreeAll((Storage *)((void *)h + sizeof(Host)), STORAGE_HOST);
} }
int HostStorageRegister(const char *name, const unsigned int size, void *(*Init)(unsigned int), void (*Free)(void *)) { int HostStorageRegister(const char *name, const unsigned int size, void *(*Init)(unsigned int), void (*Free)(void *)) {

Loading…
Cancel
Save