protect from possible overflow

This commit is contained in:
V'itor Santos Costa 2016-02-11 05:55:53 -08:00
parent b13dc6812c
commit be835c73e6

View File

@ -343,10 +343,10 @@ Atom Yap_LookupMaybeWideAtomWithLength(
if (wide) {
wchar_t *ptr0;
ptr0 = (wchar_t *)Yap_AllocCodeSpace(sizeof(wchar_t) * (len0 + 1));
ptr0 = (wchar_t *)Yap_AllocCodeSpace(sizeof(wchar_t) * (len0 + 2));
if (!ptr0)
return NIL;
memcpy(ptr0, atom, len0 * sizeof(wchar_t));
memcpy(ptr0, atom, (len0+1) * sizeof(wchar_t));
ptr0[len0] = '\0';
at = LookupWideAtom(ptr0);
Yap_FreeCodeSpace((char *)ptr0);
@ -354,7 +354,7 @@ Atom Yap_LookupMaybeWideAtomWithLength(
} else {
unsigned char *ptr0;
ptr0 = Yap_AllocCodeSpace((len0 + 1));
ptr0 = Yap_AllocCodeSpace((len0 + 2));
if (!ptr0)
return NIL;
for (i = 0; i < len0; i++)