1*72194717Skre /* $NetBSD: expandm.c,v 1.8 2019/01/14 03:30:25 kre Exp $ */ 2de284403Schristos 3de284403Schristos /*- 4*72194717Skre * 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*72194717Skre __RCSID("$NetBSD: expandm.c,v 1.8 2019/01/14 03:30:25 kre 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 56*72194717Skre buf = NULL; 57*72194717Skre 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 if (__predict_false((m - ptr) >= INT_MAX)) { 6451621f70Schristos size_t blen = buf ? strlen(buf) : 0; 6551621f70Schristos size_t nlen = (size_t)(m - ptr); 6651621f70Schristos 6751621f70Schristos nbuf = realloc(buf, blen + nlen + 1); 6851621f70Schristos if (nbuf == NULL) 6951621f70Schristos goto out; 7051621f70Schristos 7151621f70Schristos memcpy(nbuf + blen, ptr, nlen); 7251621f70Schristos nbuf[blen + nlen] = '\0'; 7351621f70Schristos ptr += nlen; 7451621f70Schristos buf = nbuf; 7551621f70Schristos } 7651621f70Schristos 77f1ee8649Skre if (__predict_true(e == NULL && (cnt & 1) != 0)) 78f1ee8649Skre e = strerror(err); 79de284403Schristos if (asprintf(&nbuf, "%s%.*s%s", buf ? buf : "", 80d501129cSchristos (int)(m - ptr), ptr, (cnt & 1) ? e : "%m") == -1) 81de284403Schristos goto out; 82de284403Schristos free(buf); 83de284403Schristos buf = nbuf; 84de284403Schristos } 85de284403Schristos 86de284403Schristos if (asprintf(&nbuf, "%s%s%s", buf ? buf : "", ptr, sf ? sf : "") == -1) 87de284403Schristos goto out; 88de284403Schristos 89de284403Schristos free(buf); 90c5cf555eSchristos if (rbuf) 91bd7d4285Skre *rbuf = nbuf; 92f1ee8649Skre errno = err; 93de284403Schristos return nbuf; 94de284403Schristos out: 95de284403Schristos free(buf); 96c5cf555eSchristos if (rbuf) 97c5cf555eSchristos *rbuf = NULL; 98f1ee8649Skre errno = err; 990f2ce139Skre return fmt; 100de284403Schristos } 101de284403Schristos 102de284403Schristos #ifdef TEST 103de284403Schristos int 104de284403Schristos main(int argc, char *argv[]) 105de284403Schristos { 106de284403Schristos errno = ERANGE; 1070f2ce139Skre printf("%s\n", expandm(argc > 1 ? argv[1] : "Message %%m=%m: %%%m%%", 1080f2ce139Skre "...", NULL)); 109de284403Schristos return 0; 110de284403Schristos } 111de284403Schristos #endif 112