don't crash when removing out of bounds
This commit is contained in:
parent
db42c21754
commit
16470f2592
2 changed files with 10 additions and 1 deletions
10
ds.c
10
ds.c
|
|
@ -67,7 +67,7 @@ void
|
||||||
return NULL;
|
return NULL;
|
||||||
}
|
}
|
||||||
|
|
||||||
for (i = -1, cur = *ll;; i++, cur = cur->next) {
|
for (i = -1, cur = *ll; cur; i++, cur = cur->next) {
|
||||||
if (i + 1 == idx) {
|
if (i + 1 == idx) {
|
||||||
if (idx == 0) {
|
if (idx == 0) {
|
||||||
rm = cur;
|
rm = cur;
|
||||||
|
|
@ -86,6 +86,10 @@ void
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
if (!rm) {
|
||||||
|
return NULL;
|
||||||
|
}
|
||||||
|
|
||||||
data = rm->data;
|
data = rm->data;
|
||||||
free(rm);
|
free(rm);
|
||||||
return data;
|
return data;
|
||||||
|
|
@ -127,6 +131,10 @@ void
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
if (!rm) {
|
||||||
|
return NULL;
|
||||||
|
}
|
||||||
|
|
||||||
data = rm->data;
|
data = rm->data;
|
||||||
free(rm);
|
free(rm);
|
||||||
return data;
|
return data;
|
||||||
|
|
|
||||||
1
test.c
1
test.c
|
|
@ -17,6 +17,7 @@ main(int argc, char *argv[])
|
||||||
ds_ll_foreach(ds_sll_t, ll) {
|
ds_ll_foreach(ds_sll_t, ll) {
|
||||||
puts(cur->data);
|
puts(cur->data);
|
||||||
}
|
}
|
||||||
|
ds_sll_remove(&ll, 1234);
|
||||||
char *pa = ds_sll_remove(&ll, 0); // [ "two" ]
|
char *pa = ds_sll_remove(&ll, 0); // [ "two" ]
|
||||||
char *pb = ds_sll_remove(&ll, 0); // [ ]
|
char *pb = ds_sll_remove(&ll, 0); // [ ]
|
||||||
|
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue