avoid possible memory leak

This commit is contained in:
Axel Kohlmeyer
2018-03-16 19:11:30 -04:00
parent 3befd4b603
commit 8e5d3802ad

View File

@ -594,19 +594,21 @@ void *lammps_extract_fix(void *ptr, char *id, int style, int type,
Fix *fix = lmp->modify->fix[ifix];
if (style == 0) {
double *dptr = (double *) malloc(sizeof(double));
if (type == 0) {
if (!fix->scalar_flag) return NULL;
double *dptr = (double *) malloc(sizeof(double));
*dptr = fix->compute_scalar();
return (void *) dptr;
}
if (type == 1) {
if (!fix->vector_flag) return NULL;
double *dptr = (double *) malloc(sizeof(double));
*dptr = fix->compute_vector(i);
return (void *) dptr;
}
if (type == 2) {
if (!fix->array_flag) return NULL;
double *dptr = (double *) malloc(sizeof(double));
*dptr = fix->compute_array(i,j);
return (void *) dptr;
}