Please note that diffs are not public domain; they are subject to the copyright notices on the relevant files. =================================================================== RCS file: /ftp/cvs/cvsroot/src/lib/libterminfo/term.c,v rcsdiff: /ftp/cvs/cvsroot/src/lib/libterminfo/term.c,v: warning: Unknown phrases like `commitid ...;' are present. retrieving revision 1.13.2.1 retrieving revision 1.19 diff -u -p -r1.13.2.1 -r1.19 --- src/lib/libterminfo/term.c 2012/10/30 18:59:18 1.13.2.1 +++ src/lib/libterminfo/term.c 2015/11/26 01:03:22 1.19 @@ -1,4 +1,4 @@ -/* $NetBSD: term.c,v 1.13.2.1 2012/10/30 18:59:18 yamt Exp $ */ +/* $NetBSD: term.c,v 1.19 2015/11/26 01:03:22 christos Exp $ */ /* * Copyright (c) 2009, 2010, 2011 The NetBSD Foundation, Inc. @@ -28,7 +28,7 @@ */ #include -__RCSID("$NetBSD: term.c,v 1.13.2.1 2012/10/30 18:59:18 yamt Exp $"); +__RCSID("$NetBSD: term.c,v 1.19 2015/11/26 01:03:22 christos Exp $"); #include @@ -54,6 +54,23 @@ const char *_ti_database; #include "compiled_terms.c" static int +allocset(void *pp, int init, size_t nelem, size_t elemsize) +{ + void **p = pp; + if (*p) { + memset(*p, init, nelem * elemsize); + return 0; + } + + if ((*p = calloc(nelem, elemsize)) == NULL) + return -1; + + if (init != 0) + memset(*p, init, nelem * elemsize); + return 0; +} + +static int _ti_readterm(TERMINAL *term, const char *cap, size_t caplen, int flags) { uint8_t ver; @@ -68,20 +85,22 @@ _ti_readterm(TERMINAL *term, const char return -1; } - term->flags = calloc(TIFLAGMAX + 1, sizeof(char)); - if (term->flags == NULL) - return -1; - term->nums = malloc((TINUMMAX + 1) * sizeof(short)); - if (term->nums == NULL) + + if (allocset(&term->flags, 0, TIFLAGMAX + 1, sizeof(*term->flags)) == -1) return -1; - memset(term->nums, (short)-1, (TINUMMAX + 1) * sizeof(short)); - term->strs = calloc(TISTRMAX + 1, sizeof(char *)); - if (term->strs == NULL) + + if (allocset(&term->nums, -1, TINUMMAX + 1, sizeof(*term->nums)) == -1) return -1; - term->_arealen = caplen; - term->_area = malloc(term->_arealen); - if (term->_area == NULL) + + if (allocset(&term->strs, 0, TISTRMAX + 1, sizeof(*term->strs)) == -1) return -1; + + if (term->_arealen != caplen) { + term->_arealen = caplen; + term->_area = realloc(term->_area, term->_arealen); + if (term->_area == NULL) + return -1; + } memcpy(term->_area, cap, term->_arealen); cap = term->_area; @@ -119,7 +138,7 @@ _ti_readterm(TERMINAL *term, const char term->flags[ind] = 0; } } - + num = le16dec(cap); cap += sizeof(uint16_t); if (num != 0) { @@ -134,7 +153,7 @@ _ti_readterm(TERMINAL *term, const char cap += sizeof(uint16_t); } } - + num = le16dec(cap); cap += sizeof(uint16_t); if (num != 0) { @@ -154,13 +173,20 @@ _ti_readterm(TERMINAL *term, const char cap += len; } } - + num = le16dec(cap); cap += sizeof(uint16_t); if (num != 0) { - term->_nuserdefs = le16dec(cap); - term->_userdefs = malloc(sizeof(*term->_userdefs) * num); + num = le16dec(cap); cap += sizeof(uint16_t); + if (num != term->_nuserdefs) { + free(term->_userdefs); + term->_userdefs = NULL; + term->_nuserdefs = num; + } + if (allocset(&term->_userdefs, 0, term->_nuserdefs, + sizeof(*term->_userdefs)) == -1) + return -1; for (num = 0; num < term->_nuserdefs; num++) { ud = &term->_userdefs[num]; len = le16dec(cap); @@ -204,7 +230,14 @@ _ti_readterm(TERMINAL *term, const char return -1; } } + } else { + term->_nuserdefs = 0; + if (term->_userdefs) { + free(term->_userdefs); + term->_userdefs = NULL; + } } + return 1; } @@ -243,7 +276,7 @@ _ti_dbgetterm(TERMINAL *term, const char data8 = data; if (data8[0] != 1) goto fail; - } else if (data8[0] != 1) + } else if (data8[0] != 1) goto fail; else if (klen + 3 >= len || le16dec(data8 + 1) != klen) goto fail; @@ -350,13 +383,13 @@ _ti_findterm(TERMINAL *term, const char if (e != NULL) { if (c == NULL) e = strdup(e); /* So we don't destroy env */ - if (e == NULL) + if (e == NULL) tic = NULL; - else + else { tic = _ti_compile(e, TIC_WARNING | TIC_ALIAS | TIC_DESCRIPTION | TIC_EXTRA); - if (c == NULL && e != NULL) free(e); + } if (tic != NULL && ticcmp(tic, name) == 0) { len = _ti_flatten(&f, tic); if (len != -1) {