Minki/linux
1
0
Fork 1
mirror of https://github.com/torvalds/linux.git synced 2024-11-21 19:41:42 +00:00
Code Issues Packages Projects Releases Wiki Activity

kallsyms: use xmalloc() and xrealloc()

Browse Source 
When malloc() or realloc() fails, there is not much userspace programs
can do. xmalloc() and xrealloc() are useful to bail out on a memory
allocation failure.

Signed-off-by: Masahiro Yamada <masahiroy@kernel.org>
This commit is contained in:
Masahiro Yamada 2024-08-12 21:48:52 +09:00
parent 4c2598e3b6
commit aeaa4283a3
1 changed files with 5 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
scripts/kallsyms.c
View File

@ -27,6 +27,8 @@
#include <ctype.h> #include <ctype.h>
#include <limits.h> #include <limits.h>
#include <xalloc.h>
#define ARRAY_SIZE(arr) (sizeof(arr) / sizeof(arr[0])) #define ARRAY_SIZE(arr) (sizeof(arr) / sizeof(arr[0]))
#define KSYM_NAME_LEN 512 #define KSYM_NAME_LEN 512
@ -168,12 +170,7 @@ static struct sym_entry *read_symbol(FILE *in, char **buf, size_t *buf_len)
* compressed together */ * compressed together */
len++; len++;
sym = malloc(sizeof(*sym) + len + 1); sym = xmalloc(sizeof(*sym) + len + 1);
if (!sym) {
fprintf(stderr, "kallsyms failure: "
"unable to allocate required amount of memory\n");
exit(EXIT_FAILURE);
}
sym->addr = addr; sym->addr = addr;
sym->len = len; sym->len = len;
sym->sym[0] = type; sym->sym[0] = type;
@ -278,12 +275,7 @@ static void read_map(const char *in)
if (table_cnt >= table_size) { if (table_cnt >= table_size) {
table_size += 10000; table_size += 10000;
table = realloc(table, sizeof(*table) * table_size); table = xrealloc(table, sizeof(*table) * table_size);
if (!table) {
fprintf(stderr, "out of memory\n");
fclose(fp);
exit (1);
}
} }
table[table_cnt++] = sym; table[table_cnt++] = sym;
@ -391,12 +383,7 @@ static void write_src(void)
/* table of offset markers, that give the offset in the compressed stream /* table of offset markers, that give the offset in the compressed stream
* every 256 symbols */ * every 256 symbols */
markers_cnt = (table_cnt + 255) / 256; markers_cnt = (table_cnt + 255) / 256;
markers = malloc(sizeof(*markers) * markers_cnt); markers = xmalloc(sizeof(*markers) * markers_cnt);
if (!markers) {
fprintf(stderr, "kallsyms failure: "
"unable to allocate required memory\n");
exit(EXIT_FAILURE);
}
output_label("kallsyms_names"); output_label("kallsyms_names");
off = 0; off = 0;