From 375124be51c796a53cce9716e83003ef37c7bcf5 Mon Sep 17 00:00:00 2001 From: "K.Takata" Date: Tue, 30 Jul 2019 12:08:33 +0900 Subject: st: Do error check only on non-Ruby --- st.c | 16 ++++++++++++++++ 1 file changed, 16 insertions(+) (limited to 'st.c') diff --git a/st.c b/st.c index 5d3588ff14..5d0c00cc0a 100644 --- a/st.c +++ b/st.c @@ -593,9 +593,15 @@ st_init_table_with_size(const struct st_hash_type *type, st_index_t size) #endif n = get_power2(size); +#ifndef RUBY + if (n < 0) + return NULL; +#endif tab = (st_table *) malloc(sizeof (st_table)); +#ifndef RUBY if (tab == NULL) return NULL; +#endif tab->type = type; tab->entry_power = n; tab->bin_power = features[n].bin_power; @@ -604,17 +610,21 @@ st_init_table_with_size(const struct st_hash_type *type, st_index_t size) tab->bins = NULL; else { tab->bins = (st_index_t *) malloc(bins_size(tab)); +#ifndef RUBY if (tab->bins == NULL) { free(tab); return NULL; } +#endif } tab->entries = (st_table_entry *) malloc(get_allocated_entries(tab) * sizeof(st_table_entry)); +#ifndef RUBY if (tab->entries == NULL) { st_free_table(tab); return NULL; } +#endif #ifdef ST_DEBUG memset(tab->entries, ST_INIT_VAL_BYTE, get_allocated_entries(tab) * sizeof(st_table_entry)); @@ -1312,24 +1322,30 @@ st_copy(st_table *old_tab) st_table *new_tab; new_tab = (st_table *) malloc(sizeof(st_table)); +#ifndef RUBY if (new_tab == NULL) return NULL; +#endif *new_tab = *old_tab; if (old_tab->bins == NULL) new_tab->bins = NULL; else { new_tab->bins = (st_index_t *) malloc(bins_size(old_tab)); +#ifndef RUBY if (new_tab->bins == NULL) { free(new_tab); return NULL; } +#endif } new_tab->entries = (st_table_entry *) malloc(get_allocated_entries(old_tab) * sizeof(st_table_entry)); +#ifndef RUBY if (new_tab->entries == NULL) { st_free_table(new_tab); return NULL; } +#endif MEMCPY(new_tab->entries, old_tab->entries, st_table_entry, get_allocated_entries(old_tab)); if (old_tab->bins != NULL) -- cgit v1.2.3