ViewVC Help
View File | Revision Log | Show Annotations | Revision Graph | Root Listing
root/i-scream/projects/libstatgrab/src/libstatgrab/tools.c
(Generate patch)

Comparing projects/libstatgrab/src/libstatgrab/tools.c (file contents):
Revision 1.44 by ats, Tue Apr 6 17:42:16 2004 UTC vs.
Revision 1.48 by ats, Wed Apr 7 10:33:50 2004 UTC

# Line 45 | Line 45
45   #endif
46  
47   #include "tools.h"
48 + #include "statgrab.h"
49  
50   #ifdef SOLARIS
51   #ifdef HAVE_LIBDEVINFO_H
# Line 90 | Line 91 | static void add_mapping(char *bsd, char *svr){
91          mapping_t *map_end_ptr;
92  
93          if (mapping == NULL){
94 <                mapping = malloc(sizeof(mapping_t));
94 >                mapping = sg_malloc(sizeof(mapping_t));
95                  if (mapping == NULL) return;
96                  map_ptr = mapping;
97          }else{
# Line 105 | Line 106 | static void add_mapping(char *bsd, char *svr){
106                  /* We've reached end of list and not found the entry.. So we need to malloc
107                   * new mapping_t
108                   */
109 <                map_end_ptr->next = malloc(sizeof(mapping_t));
109 >                map_end_ptr->next = sg_malloc(sizeof(mapping_t));
110                  if (map_end_ptr->next == NULL) return;
111                  map_ptr = map_end_ptr->next;
112          }
113  
114          map_ptr->next = NULL;
115 <        map_ptr->bsd = strdup(bsd);
116 <        map_ptr->svr = strdup(svr);
115 >        map_ptr->bsd = NULL;
116 >        map_ptr->svr = NULL;
117 >        if (sg_update_string(&map_ptr->bsd, bsd) < 0
118 >            || sg_update_string(&map_ptr->svr, svr) < 0) {
119 >                return;
120 >        }
121  
122          return;
123   }
# Line 122 | Line 127 | static char *read_dir(char *disk_path){
127          DIR *dirp;
128          struct dirent *dp;
129          struct stat stbuf;
130 <        char *svr_name;
130 >        char *svr_name = NULL;
131          char current_dir[MAXPATHLEN];
132          char file_name[MAXPATHLEN];
133          char temp_name[MAXPATHLEN];
# Line 149 | Line 154 | static char *read_dir(char *disk_path){
154                          x = readlink(dir_dname, file_name, sizeof(file_name));
155                          file_name[x] = '\0';
156                          if (strcmp(file_name, temp_name) == 0) {
157 <                                svr_name = strdup(dp->d_name);
157 >                                if (sg_update_string(&svr_name,
158 >                                                     dp->d_name) < 0) {
159 >                                        return NULL;
160 >                                }
161                                  closedir(dirp);
162                                  return svr_name;
163                          }
# Line 171 | Line 179 | static int get_alias(char *alias){
179          char *value;
180          int instance;
181          if ((root_node = di_init("/", DINFOCPYALL)) == DI_NODE_NIL) {
182 <                return 1;
182 >                return -1;
183          }
184          node = di_drv_first_node(alias, root_node);
185          while (node != DI_NODE_NIL) {
# Line 213 | Line 221 | static int build_mapping(){
221          int found;
222  
223          if ((kc = kstat_open()) == NULL) {
224 <                return 1;
224 >                return -1;
225          }
226  
227          for (ksp = kc->kc_chain; ksp; ksp = ksp->ks_next) {
# Line 235 | Line 243 | static int build_mapping(){
243                                  if (x>=BIG_ENOUGH){
244                                          /* We've got bigger than we thought was massive */
245                                          /* If we hit this.. Make big enough bigger */
246 <                                        return 1;
246 >                                        return -1;
247                                  }
248                                  if( !strncmp(driver_list[x], device_name, BIG_ENOUGH)){
249                                          found = 1;
# Line 245 | Line 253 | static int build_mapping(){
253  
254                          if(!found){
255                                  if((get_alias(device_name)) != 0){
256 <                                        return 1;
256 >                                        return -1;
257                                  }
258                                  strncpy(driver_list[x], device_name, BIG_ENOUGH);
259                                  list_entries++;
# Line 275 | Line 283 | char *sg_f_read_line(FILE *f, const char *string){
283  
284   char *sg_get_string_match(char *line, regmatch_t *match){
285          int len=match->rm_eo - match->rm_so;
286 <        char *match_string=malloc(len+1);
286 >        char *match_string=sg_malloc(len+1);
287  
288          match_string=strncpy(match_string, line+match->rm_so, len);
289          match_string[len]='\0';
# Line 413 | Line 421 | int sg_update_string(char **dest, const char *src) {
421                  return 0;
422          }
423  
424 <        new = realloc(*dest, strlen(src) + 1);
424 >        new = sg_realloc(*dest, strlen(src) + 1);
425          if (new == NULL) {
426                  return -1;
427          }
# Line 479 | Line 487 | struct uvmexp *sg_get_uvmexp() {
487   int sg_init(){
488   #if (defined(FREEBSD) && !defined(FREEBSD5)) || defined(DFBSD)
489          if (sg_get_kvm() == NULL) {
490 <                return 1;
490 >                return -1;
491          }
492          if (sg_get_kvm2() == NULL) {
493 <                return 1;
493 >                return -1;
494          }
495   #endif
496   #ifdef SOLARIS
# Line 501 | Line 509 | int sg_drop_privileges() {
509          if (setegid(getgid()) != 0) return -1;
510          if (seteuid(getuid()) != 0) return -1;
511          return 0;
512 + }
513 +
514 + void *sg_realloc(void *ptr, size_t size) {
515 +        void *tmp = NULL;
516 +        tmp = realloc(ptr, size);
517 +        if(tmp == NULL) {
518 +                sg_set_error(SG_ERROR_MALLOC_FAILED, NULL);
519 +        }
520 +        return tmp;
521   }
522  

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines