diff options
Diffstat (limited to 'runtime/POSIX/fd_init.c')
| -rw-r--r-- | runtime/POSIX/fd_init.c | 18 |
1 files changed, 15 insertions, 3 deletions
diff --git a/runtime/POSIX/fd_init.c b/runtime/POSIX/fd_init.c index 8845fc9c..a8d557e7 100644 --- a/runtime/POSIX/fd_init.c +++ b/runtime/POSIX/fd_init.c @@ -14,12 +14,9 @@ #include "klee/klee.h" #include <assert.h> -#include <stdio.h> #include <stdlib.h> #include <string.h> #include <sys/stat.h> -#include <sys/syscall.h> -#include <unistd.h> exe_file_system_t __exe_fs; @@ -46,6 +43,9 @@ exe_sym_env_t __exe_env = { static void __create_new_dfile(exe_disk_file_t *dfile, unsigned size, const char *name, struct stat64 *defaults) { struct stat64 *s = malloc(sizeof(*s)); + if (!s) + klee_report_error(__FILE__, __LINE__, "out of memory in klee_init_env", "user.err"); + const char *sp; char sname[64]; for (sp=name; *sp; ++sp) @@ -56,6 +56,8 @@ static void __create_new_dfile(exe_disk_file_t *dfile, unsigned size, dfile->size = size; dfile->contents = malloc(dfile->size); + if (!dfile->contents) + klee_report_error(__FILE__, __LINE__, "out of memory in klee_init_env", "user.err"); klee_make_symbolic(dfile->contents, dfile->size, name); klee_make_symbolic(s, sizeof(*s), sname); @@ -118,6 +120,9 @@ void klee_init_fds(unsigned n_files, unsigned file_length, __exe_fs.n_sym_files = n_files; __exe_fs.sym_files = malloc(sizeof(*__exe_fs.sym_files) * n_files); + if (n_files && !__exe_fs.sym_files) + klee_report_error(__FILE__, __LINE__, "out of memory in klee_init_env", "user.err"); + for (k=0; k < n_files; k++) { name[0] = 'A' + k; __create_new_dfile(&__exe_fs.sym_files[k], file_length, name, &s); @@ -126,6 +131,8 @@ void klee_init_fds(unsigned n_files, unsigned file_length, /* setting symbolic stdin */ if (stdin_length) { __exe_fs.sym_stdin = malloc(sizeof(*__exe_fs.sym_stdin)); + if (!__exe_fs.sym_stdin) + klee_report_error(__FILE__, __LINE__, "out of memory in klee_init_env", "user.err"); __create_new_dfile(__exe_fs.sym_stdin, stdin_length, "stdin", &s); __exe_env.fds[0].dfile = __exe_fs.sym_stdin; } @@ -138,6 +145,9 @@ void klee_init_fds(unsigned n_files, unsigned file_length, __exe_fs.close_fail = malloc(sizeof(*__exe_fs.close_fail)); __exe_fs.ftruncate_fail = malloc(sizeof(*__exe_fs.ftruncate_fail)); __exe_fs.getcwd_fail = malloc(sizeof(*__exe_fs.getcwd_fail)); + if (!(__exe_fs.read_fail && __exe_fs.write_fail && __exe_fs.close_fail + && __exe_fs.ftruncate_fail && __exe_fs.getcwd_fail)) + klee_report_error(__FILE__, __LINE__, "out of memory in klee_init_env", "user.err"); klee_make_symbolic(__exe_fs.read_fail, sizeof(*__exe_fs.read_fail), "read_fail"); klee_make_symbolic(__exe_fs.write_fail, sizeof(*__exe_fs.write_fail), "write_fail"); @@ -149,6 +159,8 @@ void klee_init_fds(unsigned n_files, unsigned file_length, /* setting symbolic stdout */ if (sym_stdout_flag) { __exe_fs.sym_stdout = malloc(sizeof(*__exe_fs.sym_stdout)); + if (!__exe_fs.sym_stdout) + klee_report_error(__FILE__, __LINE__, "out of memory in klee_init_env", "user.err"); __create_new_dfile(__exe_fs.sym_stdout, 1024, "stdout", &s); __exe_env.fds[1].dfile = __exe_fs.sym_stdout; __exe_fs.stdout_writes = 0; |