pike.git/ src/ cpp.cmod

2018-01-04

2018-01-04 12:04:50 by Henrik Grubbström (Grubba) <grubba@grubba.org>

• 282a8a93b79e0babd66565f093c3087df5ad9c39 (21 lines) (+11/-10) ^{[ Show | Annotate ]}
  Branch: master

Cpp: Fixed potential memory leak on OOM.

add_predefine() would leak the macro name if it failed to
allocate space for the macro value.

3688:    while((tmp=first_predef))    {    free(tmp->name); -  free(tmp->value); +     first_predef=tmp->next;    free(tmp);    }
3734:   void add_predefine(const char *s)   {    struct pike_predef_s *tmp=ALLOC_STRUCT(pike_predef_s); -  char * pos=strchr(s,'='); +  size_t len = strlen(s); +  char *pos=strchr(s,'='); +     if(pos)    { -  tmp->name=xalloc(pos-s+1); -  memcpy(tmp->name,s,pos-s); +  tmp->name = xalloc(len+1); +  memcpy(tmp->name, s, len+1);    tmp->name[pos-s]=0;    -  tmp->value=xalloc(s+strlen(s)-pos); -  memcpy(tmp->value,pos+1,s+strlen(s)-pos); +  tmp->value = tmp->name + (pos-s) + 1;    }else{ -  tmp->name=xalloc(strlen(s)+1); -  memcpy(tmp->name,s,strlen(s)+1); +  tmp->name = xalloc(len + 1 + 4); +  memcpy(tmp->name,s,len + 1);    -  tmp->value=xalloc(4); +  tmp->value = tmp->name + len + 1;    memcpy(tmp->value," 1 ",4);    }    tmp->next = NULL; -  +     if (first_predef) {    last_predef->next = tmp;    last_predef = tmp;
3768:    while((tmp=first_predef))    {    free(tmp->name); -  free(tmp->value); +     first_predef=tmp->next;    free(tmp);    }