1 /* $NetBSD: emit.c,v 1.22 2023/07/13 08:40:38 rillig Exp $ */ 2 3 /* 4 * Copyright (c) 1994, 1995 Jochen Pohl 5 * All Rights Reserved. 6 * 7 * Redistribution and use in source and binary forms, with or without 8 * modification, are permitted provided that the following conditions 9 * are met: 10 * 1. Redistributions of source code must retain the above copyright 11 * notice, this list of conditions and the following disclaimer. 12 * 2. Redistributions in binary form must reproduce the above copyright 13 * notice, this list of conditions and the following disclaimer in the 14 * documentation and/or other materials provided with the distribution. 15 * 3. All advertising materials mentioning features or use of this software 16 * must display the following acknowledgement: 17 * This product includes software developed by Jochen Pohl for 18 * The NetBSD Project. 19 * 4. The name of the author may not be used to endorse or promote products 20 * derived from this software without specific prior written permission. 21 * 22 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR 23 * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES 24 * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. 25 * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT, 26 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT 27 * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, 28 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY 29 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 30 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF 31 * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 32 */ 33 34 #if HAVE_NBTOOL_CONFIG_H 35 #include "nbtool_config.h" 36 #endif 37 38 #include <sys/cdefs.h> 39 #if defined(__RCSID) 40 __RCSID("$NetBSD: emit.c,v 1.22 2023/07/13 08:40:38 rillig Exp $"); 41 #endif 42 43 #include <stdio.h> 44 #include <string.h> 45 46 #include "lint.h" 47 48 /* name and handle of output file */ 49 static const char *loname; 50 static FILE *lout; 51 52 /* output buffer data */ 53 static ob_t ob; 54 55 static void outxbuf(void); 56 57 58 /* 59 * initialize output 60 */ 61 void 62 outopen(const char *name) 63 { 64 65 loname = name; 66 67 /* Open output file */ 68 if ((lout = fopen(name, "w")) == NULL) 69 err(1, "cannot open '%s'", name); 70 71 /* Create output buffer */ 72 ob.o_len = 1024; 73 ob.o_end = (ob.o_buf = ob.o_next = xmalloc(ob.o_len)) + ob.o_len; 74 } 75 76 /* 77 * flush output buffer and close file 78 */ 79 void 80 outclose(void) 81 { 82 83 outclr(); 84 if (fclose(lout) == EOF) 85 err(1, "cannot close '%s'", loname); 86 } 87 88 /* 89 * resize output buffer 90 */ 91 static void 92 outxbuf(void) 93 { 94 95 size_t next = (size_t)(ob.o_next - ob.o_buf); 96 ob.o_len *= 2; 97 ob.o_buf = xrealloc(ob.o_buf, ob.o_len); 98 ob.o_end = ob.o_buf + ob.o_len; 99 ob.o_next = ob.o_buf + next; 100 } 101 102 /* 103 * reset output buffer 104 * if it is not empty, it is flushed 105 */ 106 void 107 outclr(void) 108 { 109 110 if (ob.o_buf != ob.o_next) { 111 outchar('\n'); 112 size_t sz = (size_t)(ob.o_next - ob.o_buf); 113 if (sz > ob.o_len) 114 errx(1, "internal error: outclr"); 115 if (fwrite(ob.o_buf, sz, 1, lout) != 1) 116 err(1, "cannot write to %s", loname); 117 ob.o_next = ob.o_buf; 118 } 119 } 120 121 /* 122 * write a character to the output buffer 123 */ 124 void 125 outchar(char c) 126 { 127 128 if (ob.o_next == ob.o_end) 129 outxbuf(); 130 *ob.o_next++ = c; 131 } 132 133 /* 134 * write a string to the output buffer 135 * the string must not contain any characters which 136 * should be quoted 137 */ 138 void 139 outstrg(const char *s) 140 { 141 142 while (*s != '\0') { 143 if (ob.o_next == ob.o_end) 144 outxbuf(); 145 *ob.o_next++ = *s++; 146 } 147 } 148 149 /* write an integer value to the output buffer */ 150 void 151 outint(int i) 152 { 153 154 if ((size_t)(ob.o_end - ob.o_next) < 3 * sizeof(int)) 155 outxbuf(); 156 ob.o_next += snprintf(ob.o_next, ob.o_end - ob.o_next, "%d", i); 157 } 158 159 /* write a name to the output buffer, preceded by its length */ 160 void 161 outname(const char *name) 162 { 163 outint((int)strlen(name)); 164 outstrg(name); 165 } 166 167 /* write the name of the .c source */ 168 void 169 outsrc(const char *name) 170 { 171 172 outclr(); 173 outchar('S'); 174 outstrg(name); 175 } 176