add check_valid_indexHEADmaster

1 files changed, 10 insertions, 6 deletions

diff --git a/libvector.c b/libvector.c
index 29abc9d..04461b8 100644
--- a/libvector.c
+++ b/libvector.c
@@ -10,6 +10,7 @@ vector_t vector_init(int size){
     return vec;
 }
 
+
 int size(vector_t vec){
     return vec.cur_size;
 }
@@ -25,11 +26,15 @@ int is_empty(vector_t vec){
     return 0;
 }
 
-int at(vector_t vec, int index){
-    if (index >= size(vec) || index < 0){
+void check_valid_index(vector_t vec, int index){
+    if (index > size(vec) || index < 0){
         printf("Index out of bound!!\n");
         exit(-1);
     }
+}
+
+int at(vector_t vec, int index){
+    check_valid_index(vec, index);
 
     return vec.arr[index];
 }
@@ -40,7 +45,9 @@ void push(vector_t *vec, int value){
     vec->cur_size++;
 }
 
+
 void insert(vector_t *vec, int index, int val){
+    check_valid_index(*vec, index);
     resize_vec_check(vec);
     
     int tmp;
@@ -53,10 +60,7 @@ void insert(vector_t *vec, int index, int val){
 }
 
 void delete_vec(vector_t *vec, int index){
-    if (!(index < vec->cur_size)){
-       printf("Invalid Index!\nFailed to Delete Element\n");
-       exit(-1);
-    }
+    check_valid_index(*vec, index);
     int len = vec->cur_size - 1;
 
     for (int i = index; i < len; i++){