hcq/MaxScale
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

query_classifier.cc: cleaned up and simplified skygw_get_canonical

Browse Source 
skygw_util.cc:fixed memory allocation issue where terminating byte wasn't counted. Added some error checks.
This commit is contained in:
VilhoRaatikka
2014-08-21 23:08:21 +03:00
parent 8d1eae6fde
commit 3a5b8ef64c
2 changed files with 27 additions and 25 deletions
Download Patch File Download Diff File Expand all files Collapse all files

32
utils/skygw_utils.cc
View File

@ -1882,7 +1882,7 @@ char* replace_literal(
const char* needle,
const char* replacement)
{
const char* prefix = "[ =',\\(]"; /*< ' ','=','(',''',',' are allowed before needle */
const char* prefix = "[ ='\",\\(]"; /*< ' ','=','(',''',''"',',' are allowed before needle */
const char* suffix = "[$^[:alnum:]]?"; /*< alpha-num chars aren't allowed after the needle */
char* search_re;
char* newstr;
@ -1894,9 +1894,27 @@ char* replace_literal(
size_t hlen = strlen(haystack);
search_re = (char *)malloc(strlen(prefix)+nlen+strlen(suffix)+1);
if (search_re == NULL)
{
fprintf(stderr, "Regex memory allocation failed : %s\n",
strerror(errno));
newstr = haystack;
goto retblock;
}
sprintf(search_re, "%s%s%s", prefix, needle, suffix);
/** Allocate memory for new string */
newstr = (char *)malloc(hlen-nlen+rlen);
/** Allocate memory for new string +1 for terminating byte */
newstr = (char *)malloc(hlen-nlen+rlen+1);
if (newstr == NULL)
{
fprintf(stderr, "Regex memory allocation failed : %s\n",
strerror(errno));
free(search_re);
newstr = haystack;
goto retblock;
}
rc = regcomp(&re, search_re, REG_EXTENDED);
ss_dassert(rc == 0);
@ -1910,7 +1928,7 @@ char* replace_literal(
search_re,
error_message);
free(search_re);
newstr = NULL;
newstr = haystack;
goto retblock;
}
rc = regexec(&re, haystack, 1, &match, 0);
@ -1918,7 +1936,7 @@ char* replace_literal(
if (rc != 0)
{
free(search_re);
newstr = NULL;
newstr = haystack;
goto retblock;
}
memcpy(newstr, haystack, match.rm_so+1);
@ -1927,8 +1945,8 @@ char* replace_literal(
memcpy(newstr+match.rm_so+1+rlen, haystack+match.rm_so+1+nlen, hlen-(match.rm_so+1)-nlen+1);
regfree(&re);
retblock:
free(haystack);
retblock:
return newstr;
}