Whamcloud - gitweb
ss_create_invocation: fix error handling when memory allocation fails
[tools/e2fsprogs.git] / lib / ss / invocation.c
1 /*
2  * Copyright 1987, 1988 by MIT Student Information Processing Board
3  *
4  * Permission to use, copy, modify, and distribute this software and
5  * its documentation for any purpose is hereby granted, provided that
6  * the names of M.I.T. and the M.I.T. S.I.P.B. not be used in
7  * advertising or publicity pertaining to distribution of the software
8  * without specific, written prior permission.  M.I.T. and the
9  * M.I.T. S.I.P.B. make no representations about the suitability of
10  * this software for any purpose.  It is provided "as is" without
11  * express or implied warranty.
12  */
13
14 #include "config.h"
15 #ifdef HAS_STDLIB_H
16 #include <stdlib.h>
17 #endif
18 #include "ss_internal.h"
19 #define size    sizeof(ss_data *)
20 #ifdef HAVE_DLOPEN
21 #include <dlfcn.h>
22 #endif
23 #include <errno.h>
24
25 int ss_create_invocation(const char *subsystem_name, const char *version_string,
26                          void *info_ptr, ss_request_table *request_table_ptr,
27                          int *code_ptr)
28 {
29         int sci_idx;
30         ss_data *new_table = NULL;
31         ss_data **table = NULL;
32         ss_data **realloc_table = NULL;
33
34         *code_ptr = 0;
35         table = _ss_table;
36         new_table = (ss_data *) malloc(sizeof(ss_data));
37         if (!new_table)
38                 goto out;
39
40         if (table == (ss_data **) NULL) {
41                 table = (ss_data **) malloc(2 * size);
42                 if (!table)
43                         goto out;
44                 table[0] = table[1] = (ss_data *)NULL;
45         }
46         initialize_ss_error_table ();
47
48         for (sci_idx = 1; table[sci_idx] != (ss_data *)NULL; sci_idx++)
49                 ;
50         realloc_table = (ss_data **) realloc((char *)table,
51                                      ((unsigned)sci_idx+2)*size);
52         if (realloc_table == NULL)
53                 goto out;
54
55         table = realloc_table;
56         table[sci_idx+1] = (ss_data *) NULL;
57         table[sci_idx] = new_table;
58
59         new_table->subsystem_name = subsystem_name;
60         new_table->subsystem_version = version_string;
61         new_table->argv = (char **)NULL;
62         new_table->current_request = (char *)NULL;
63         new_table->info_dirs = (char **)malloc(sizeof(char *));
64         if (!new_table->info_dirs)
65                 goto out;
66
67         *new_table->info_dirs = (char *)NULL;
68         new_table->info_ptr = info_ptr;
69         new_table->prompt = malloc((unsigned)strlen(subsystem_name)+4);
70         if (!new_table->prompt)
71                 goto out;
72
73         strcpy(new_table->prompt, subsystem_name);
74         strcat(new_table->prompt, ":  ");
75 #ifdef silly
76         new_table->abbrev_info = ss_abbrev_initialize("/etc/passwd", code_ptr);
77 #else
78         new_table->abbrev_info = NULL;
79 #endif
80         new_table->flags.escape_disabled = 0;
81         new_table->flags.abbrevs_disabled = 0;
82         new_table->rqt_tables =
83                 (ss_request_table **) calloc(2, sizeof(ss_request_table *));
84         if (!new_table->rqt_tables)
85                 goto out;
86
87         *(new_table->rqt_tables) = request_table_ptr;
88         *(new_table->rqt_tables+1) = (ss_request_table *) NULL;
89
90         new_table->readline_handle = 0;
91         new_table->readline_shutdown = 0;
92         new_table->readline = 0;
93         new_table->add_history = 0;
94         new_table->redisplay = 0;
95         new_table->rl_completion_matches = 0;
96         _ss_table = table;
97 #if defined(HAVE_DLOPEN) && defined(SHARED_ELF_LIB)
98         ss_get_readline(sci_idx);
99 #endif
100         return(sci_idx);
101
102 out:
103         if (new_table) {
104                 free(new_table->prompt);
105                 free(new_table->info_dirs);
106         }
107         free(new_table);
108         free(table);
109         *code_ptr = ENOMEM;
110         return 0;
111
112 }
113
114 void
115 ss_delete_invocation(int sci_idx)
116 {
117         register ss_data *t;
118         int ignored_code;
119
120         t = ss_info(sci_idx);
121         free(t->prompt);
122         free(t->rqt_tables);
123         while(t->info_dirs[0] != (char *)NULL)
124                 ss_delete_info_dir(sci_idx, t->info_dirs[0], &ignored_code);
125         free(t->info_dirs);
126 #if defined(HAVE_DLOPEN) && defined(SHARED_ELF_LIB)
127         if (t->readline_shutdown)
128                 (*t->readline_shutdown)(t);
129 #endif
130         free(t);
131 }