1*b31f5e70Schristos /* $NetBSD: expandm.c,v 1.9 2019/01/23 02:00:00 christos Exp $ */ 2de284403Schristos 3de284403Schristos /*- 472194717Skre * Copyright (c) 2019 The NetBSD Foundation, Inc. 5de284403Schristos * All rights reserved. 6de284403Schristos * 7de284403Schristos * This code is derived from software contributed to The NetBSD Foundation 8de284403Schristos * by Christos Zoulas. 9de284403Schristos * 10de284403Schristos * Redistribution and use in source and binary forms, with or without 11de284403Schristos * modification, are permitted provided that the following conditions 12de284403Schristos * are met: 13de284403Schristos * 1. Redistributions of source code must retain the above copyright 14de284403Schristos * notice, this list of conditions and the following disclaimer. 15de284403Schristos * 2. Redistributions in binary form must reproduce the above copyright 16de284403Schristos * notice, this list of conditions and the following disclaimer in the 17de284403Schristos * documentation and/or other materials provided with the distribution. 18de284403Schristos * 19de284403Schristos * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS 20de284403Schristos * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED 21de284403Schristos * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR 22de284403Schristos * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS 23de284403Schristos * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR 24de284403Schristos * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF 25de284403Schristos * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS 26de284403Schristos * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN 27de284403Schristos * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) 28de284403Schristos * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE 29de284403Schristos * POSSIBILITY OF SUCH DAMAGE. 30de284403Schristos */ 31de284403Schristos #include <sys/cdefs.h> 32*b31f5e70Schristos __RCSID("$NetBSD: expandm.c,v 1.9 2019/01/23 02:00:00 christos Exp $"); 33de284403Schristos 3451621f70Schristos #include <limits.h> 35de284403Schristos #include <stdio.h> 36de284403Schristos #include <string.h> 37de284403Schristos #include <stdlib.h> 38de284403Schristos #include <errno.h> 39de284403Schristos 40de284403Schristos #include "expandm.h" 41de284403Schristos 4251621f70Schristos #ifdef TEST 4351621f70Schristos #undef INT_MAX 4451621f70Schristos #define INT_MAX 31 4551621f70Schristos #endif 4651621f70Schristos 4751621f70Schristos 480f2ce139Skre const char * __attribute__((__format_arg__(1))) 49c5cf555eSchristos expandm(const char *fmt, const char *sf, char **rbuf) 50de284403Schristos { 51f1ee8649Skre const int err = errno; 52f1ee8649Skre const char *e = NULL; 53de284403Schristos char *buf, *m, *nbuf; 54de284403Schristos const char *ptr; 55de284403Schristos 5672194717Skre buf = NULL; 5772194717Skre for (ptr = fmt; (m = strstr(ptr, "%m")) != NULL; ptr = m + 2) { 58de284403Schristos size_t cnt = 0; 5951621f70Schristos 60de284403Schristos for (char *p = m; p >= ptr && *p == '%'; p--) 61de284403Schristos cnt++; 6251621f70Schristos 6351621f70Schristos size_t nlen = (size_t)(m - ptr); 64*b31f5e70Schristos /* 65*b31f5e70Schristos * we can't exceed INT_MAX because int is used as 66*b31f5e70Schristos * a format width 67*b31f5e70Schristos */ 68*b31f5e70Schristos if (__predict_false(nlen >= INT_MAX)) { 69*b31f5e70Schristos size_t blen = buf ? strlen(buf) : 0; 70*b31f5e70Schristos size_t tlen = nlen + blen; 7151621f70Schristos 72*b31f5e70Schristos /* 73*b31f5e70Schristos * We can't exceed PTRDIFF_MAX because we would 74*b31f5e70Schristos * not be able to address the pointers 75*b31f5e70Schristos */ 76*b31f5e70Schristos if (tlen >= PTRDIFF_MAX) { 77*b31f5e70Schristos errno = EINVAL; 78*b31f5e70Schristos goto out; 79*b31f5e70Schristos } 80*b31f5e70Schristos 81*b31f5e70Schristos nbuf = realloc(buf, tlen + 1); 8251621f70Schristos if (nbuf == NULL) 8351621f70Schristos goto out; 8451621f70Schristos 8551621f70Schristos memcpy(nbuf + blen, ptr, nlen); 86*b31f5e70Schristos nbuf[tlen] = '\0'; 8751621f70Schristos ptr += nlen; 8851621f70Schristos buf = nbuf; 8951621f70Schristos } 9051621f70Schristos 91f1ee8649Skre if (__predict_true(e == NULL && (cnt & 1) != 0)) 92f1ee8649Skre e = strerror(err); 93de284403Schristos if (asprintf(&nbuf, "%s%.*s%s", buf ? buf : "", 94d501129cSchristos (int)(m - ptr), ptr, (cnt & 1) ? e : "%m") == -1) 95de284403Schristos goto out; 96de284403Schristos free(buf); 97de284403Schristos buf = nbuf; 98de284403Schristos } 99de284403Schristos 100de284403Schristos if (asprintf(&nbuf, "%s%s%s", buf ? buf : "", ptr, sf ? sf : "") == -1) 101de284403Schristos goto out; 102de284403Schristos 103de284403Schristos free(buf); 104c5cf555eSchristos if (rbuf) 105bd7d4285Skre *rbuf = nbuf; 106f1ee8649Skre errno = err; 107de284403Schristos return nbuf; 108de284403Schristos out: 109de284403Schristos free(buf); 110c5cf555eSchristos if (rbuf) 111c5cf555eSchristos *rbuf = NULL; 112f1ee8649Skre errno = err; 1130f2ce139Skre return fmt; 114de284403Schristos } 115de284403Schristos 116de284403Schristos #ifdef TEST 117de284403Schristos int 118de284403Schristos main(int argc, char *argv[]) 119de284403Schristos { 120de284403Schristos errno = ERANGE; 1210f2ce139Skre printf("%s\n", expandm(argc > 1 ? argv[1] : "Message %%m=%m: %%%m%%", 1220f2ce139Skre "...", NULL)); 123de284403Schristos return 0; 124de284403Schristos } 125de284403Schristos #endif 126