save

[protos/libecoli.git] / lib / ecoli_vec.c

diff --git a/lib/ecoli_vec.c b/lib/ecoli_vec.c
index 001891e..b3249b5 100644 (file)
--- a/lib/ecoli_vec.c
+++ b/lib/ecoli_vec.c
@@ -93,13 +93,11 @@ int ec_vec_add_by_ref(struct ec_vec *vec, void *ptr)
                new_vec = ec_realloc(vec->vec, vec->elt_size * (vec->len + 1));
                if (new_vec == NULL)
                        return -ENOMEM;
+               vec->size = vec->len + 1;
        }
 
        vec->vec = new_vec;
-       if (vec->copy)
-               vec->copy(get_obj(vec, vec->len), ptr);
-       else
-               memcpy(get_obj(vec, vec->len), ptr, vec->elt_size);
+       memcpy(get_obj(vec, vec->len), ptr, vec->elt_size);
        vec->len++;
 
        return 0;
@@ -180,6 +178,9 @@ fail:
 
 size_t ec_vec_len(const struct ec_vec *vec)
 {
+       if (vec == NULL)
+               return 0;
+
        return vec->len;
 }
 
@@ -206,13 +207,12 @@ void ec_vec_free(struct ec_vec *vec)
 
 int ec_vec_get(void *ptr, const struct ec_vec *vec, size_t idx)
 {
-       if (vec == NULL || idx >= vec->len)
-               return -EINVAL;
+       if (vec == NULL || idx >= vec->len) {
+               errno = EINVAL;
+               return -1;
+       }
 
-       if (vec->copy)
-               vec->copy(ptr, get_obj(vec, idx));
-       else
-               memcpy(ptr, get_obj(vec, idx), vec->elt_size);
+       memcpy(ptr, get_obj(vec, idx), vec->elt_size);
 
        return 0;
 }
@@ -459,15 +459,15 @@ static int ec_vec_testcase(void)
 
        if (ec_vec_get(&vals, vec, 0) < 0)
                GOTO_FAIL;
-       if (strcmp(vals, "0"))
+       if (vals == NULL || strcmp(vals, "0"))
                GOTO_FAIL;
        if (ec_vec_get(&vals, vec, 1) < 0)
                GOTO_FAIL;
-       if (strcmp(vals, "1"))
+       if (vals == NULL || strcmp(vals, "1"))
                GOTO_FAIL;
        if (ec_vec_get(&vals, vec, 2) < 0)
                GOTO_FAIL;
-       if (strcmp(vals, "2"))
+       if (vals == NULL || strcmp(vals, "2"))
                GOTO_FAIL;
 
        ec_vec_free(vec);