1 85231fd8 2006-05-21 devnull /* Copyright (c) 2002-2006 Lucent Technologies; see LICENSE */
  2 91c13e54 2004-02-29 devnull #include <stdarg.h>
  3 91c13e54 2004-02-29 devnull #include <string.h>
  4 e5aa96ac 2004-12-26 devnull #include "plan9.h"
  5 91c13e54 2004-02-29 devnull #include "fmt.h"
  6 91c13e54 2004-02-29 devnull #include "fmtdef.h"
  7 91c13e54 2004-02-29 devnull 
  8 91c13e54 2004-02-29 devnull /*
  9 91c13e54 2004-02-29 devnull  * How many bytes of output UTF will be produced by quoting (if necessary) this string?
 10 91c13e54 2004-02-29 devnull  * How many runes? How much of the input will be consumed?
 11 91c13e54 2004-02-29 devnull  * The parameter q is filled in by __quotesetup.
 12 91c13e54 2004-02-29 devnull  * The string may be UTF or Runes (s or r).
 13 91c13e54 2004-02-29 devnull  * Return count does not include NUL.
 14 91c13e54 2004-02-29 devnull  * Terminate the scan at the first of:
 15 91c13e54 2004-02-29 devnull  *	NUL in input
 16 91c13e54 2004-02-29 devnull  *	count exceeded in input
 17 91c13e54 2004-02-29 devnull  *	count exceeded on output
 18 91c13e54 2004-02-29 devnull  * *ninp is set to number of input bytes accepted.
 19 91c13e54 2004-02-29 devnull  * nin may be <0 initially, to avoid checking input by count.
 20 91c13e54 2004-02-29 devnull  */
 21 91c13e54 2004-02-29 devnull void
 22 91c13e54 2004-02-29 devnull __quotesetup(char *s, Rune *r, int nin, int nout, Quoteinfo *q, int sharp, int runesout)
 23 91c13e54 2004-02-29 devnull {
 24 91c13e54 2004-02-29 devnull 	int w;
 25 91c13e54 2004-02-29 devnull 	Rune c;
 26 91c13e54 2004-02-29 devnull 
 27 91c13e54 2004-02-29 devnull 	q->quoted = 0;
 28 91c13e54 2004-02-29 devnull 	q->nbytesout = 0;
 29 91c13e54 2004-02-29 devnull 	q->nrunesout = 0;
 30 91c13e54 2004-02-29 devnull 	q->nbytesin = 0;
 31 91c13e54 2004-02-29 devnull 	q->nrunesin = 0;
 32 91c13e54 2004-02-29 devnull 	if(sharp || nin==0 || (s && *s=='\0') || (r && *r=='\0')){
 33 91c13e54 2004-02-29 devnull 		if(nout < 2)
 34 91c13e54 2004-02-29 devnull 			return;
 35 91c13e54 2004-02-29 devnull 		q->quoted = 1;
 36 91c13e54 2004-02-29 devnull 		q->nbytesout = 2;
 37 91c13e54 2004-02-29 devnull 		q->nrunesout = 2;
 38 91c13e54 2004-02-29 devnull 	}
 39 df121a00 2004-12-26 devnull 	for(; nin!=0; nin--){
 40 91c13e54 2004-02-29 devnull 		if(s)
 41 91c13e54 2004-02-29 devnull 			w = chartorune(&c, s);
 42 91c13e54 2004-02-29 devnull 		else{
 43 91c13e54 2004-02-29 devnull 			c = *r;
 44 91c13e54 2004-02-29 devnull 			w = runelen(c);
 45 91c13e54 2004-02-29 devnull 		}
 46 91c13e54 2004-02-29 devnull 
 47 91c13e54 2004-02-29 devnull 		if(c == '\0')
 48 91c13e54 2004-02-29 devnull 			break;
 49 91c13e54 2004-02-29 devnull 		if(runesout){
 50 91c13e54 2004-02-29 devnull 			if(q->nrunesout+1 > nout)
 51 91c13e54 2004-02-29 devnull 				break;
 52 91c13e54 2004-02-29 devnull 		}else{
 53 91c13e54 2004-02-29 devnull 			if(q->nbytesout+w > nout)
 54 91c13e54 2004-02-29 devnull 				break;
 55 91c13e54 2004-02-29 devnull 		}
 56 91c13e54 2004-02-29 devnull 
 57 91c13e54 2004-02-29 devnull 		if((c <= L' ') || (c == L'\'') || (fmtdoquote!=nil && fmtdoquote(c))){
 58 91c13e54 2004-02-29 devnull 			if(!q->quoted){
 59 91c13e54 2004-02-29 devnull 				if(runesout){
 60 91c13e54 2004-02-29 devnull 					if(1+q->nrunesout+1+1 > nout)	/* no room for quotes */
 61 91c13e54 2004-02-29 devnull 						break;
 62 91c13e54 2004-02-29 devnull 				}else{
 63 91c13e54 2004-02-29 devnull 					if(1+q->nbytesout+w+1 > nout)	/* no room for quotes */
 64 91c13e54 2004-02-29 devnull 						break;
 65 91c13e54 2004-02-29 devnull 				}
 66 91c13e54 2004-02-29 devnull 				q->nrunesout += 2;	/* include quotes */
 67 91c13e54 2004-02-29 devnull 				q->nbytesout += 2;	/* include quotes */
 68 91c13e54 2004-02-29 devnull 				q->quoted = 1;
 69 91c13e54 2004-02-29 devnull 			}
 70 91c13e54 2004-02-29 devnull 			if(c == '\'')	{
 71 91c13e54 2004-02-29 devnull 				if(runesout){
 72 91c13e54 2004-02-29 devnull 					if(1+q->nrunesout+1 > nout)	/* no room for quotes */
 73 91c13e54 2004-02-29 devnull 						break;
 74 91c13e54 2004-02-29 devnull 				}else{
 75 91c13e54 2004-02-29 devnull 					if(1+q->nbytesout+w > nout)	/* no room for quotes */
 76 91c13e54 2004-02-29 devnull 						break;
 77 91c13e54 2004-02-29 devnull 				}
 78 91c13e54 2004-02-29 devnull 				q->nbytesout++;
 79 91c13e54 2004-02-29 devnull 				q->nrunesout++;	/* quotes reproduce as two characters */
 80 91c13e54 2004-02-29 devnull 			}
 81 91c13e54 2004-02-29 devnull 		}
 82 91c13e54 2004-02-29 devnull 
 83 91c13e54 2004-02-29 devnull 		/* advance input */
 84 91c13e54 2004-02-29 devnull 		if(s)
 85 91c13e54 2004-02-29 devnull 			s += w;
 86 91c13e54 2004-02-29 devnull 		else
 87 91c13e54 2004-02-29 devnull 			r++;
 88 91c13e54 2004-02-29 devnull 		q->nbytesin += w;
 89 91c13e54 2004-02-29 devnull 		q->nrunesin++;
 90 91c13e54 2004-02-29 devnull 
 91 91c13e54 2004-02-29 devnull 		/* advance output */
 92 91c13e54 2004-02-29 devnull 		q->nbytesout += w;
 93 91c13e54 2004-02-29 devnull 		q->nrunesout++;
 94 ab239cb5 2006-05-26 devnull 
 95 ab239cb5 2006-05-26 devnull #ifndef PLAN9PORT
 96 ab239cb5 2006-05-26 devnull 		/* ANSI requires precision in bytes, not Runes. */
 97 ab239cb5 2006-05-26 devnull 		nin-= w-1;	/* and then n-- in the loop */
 98 ab239cb5 2006-05-26 devnull #endif
 99 91c13e54 2004-02-29 devnull 	}
100 91c13e54 2004-02-29 devnull }
101 91c13e54 2004-02-29 devnull 
102 91c13e54 2004-02-29 devnull static int
103 91c13e54 2004-02-29 devnull qstrfmt(char *sin, Rune *rin, Quoteinfo *q, Fmt *f)
104 91c13e54 2004-02-29 devnull {
105 91c13e54 2004-02-29 devnull 	Rune r, *rm, *rme;
106 91c13e54 2004-02-29 devnull 	char *t, *s, *m, *me;
107 91c13e54 2004-02-29 devnull 	Rune *rt, *rs;
108 91c13e54 2004-02-29 devnull 	ulong fl;
109 91c13e54 2004-02-29 devnull 	int nc, w;
110 91c13e54 2004-02-29 devnull 
111 91c13e54 2004-02-29 devnull 	m = sin;
112 91c13e54 2004-02-29 devnull 	me = m + q->nbytesin;
113 91c13e54 2004-02-29 devnull 	rm = rin;
114 91c13e54 2004-02-29 devnull 	rme = rm + q->nrunesin;
115 91c13e54 2004-02-29 devnull 
116 91c13e54 2004-02-29 devnull 	fl = f->flags;
117 85231fd8 2006-05-21 devnull 	w = 0;
118 85231fd8 2006-05-21 devnull 	if(fl & FmtWidth)
119 85231fd8 2006-05-21 devnull 		w = f->width;
120 91c13e54 2004-02-29 devnull 	if(f->runes){
121 91c13e54 2004-02-29 devnull 		if(!(fl & FmtLeft) && __rfmtpad(f, w - q->nrunesout) < 0)
122 91c13e54 2004-02-29 devnull 			return -1;
123 91c13e54 2004-02-29 devnull 	}else{
124 91c13e54 2004-02-29 devnull 		if(!(fl & FmtLeft) && __fmtpad(f, w - q->nbytesout) < 0)
125 91c13e54 2004-02-29 devnull 			return -1;
126 91c13e54 2004-02-29 devnull 	}
127 91c13e54 2004-02-29 devnull 	t = (char*)f->to;
128 91c13e54 2004-02-29 devnull 	s = (char*)f->stop;
129 91c13e54 2004-02-29 devnull 	rt = (Rune*)f->to;
130 91c13e54 2004-02-29 devnull 	rs = (Rune*)f->stop;
131 91c13e54 2004-02-29 devnull 	if(f->runes)
132 91c13e54 2004-02-29 devnull 		FMTRCHAR(f, rt, rs, '\'');
133 91c13e54 2004-02-29 devnull 	else
134 91c13e54 2004-02-29 devnull 		FMTRUNE(f, t, s, '\'');
135 91c13e54 2004-02-29 devnull 	for(nc = q->nrunesin; nc > 0; nc--){
136 91c13e54 2004-02-29 devnull 		if(sin){
137 91c13e54 2004-02-29 devnull 			r = *(uchar*)m;
138 91c13e54 2004-02-29 devnull 			if(r < Runeself)
139 91c13e54 2004-02-29 devnull 				m++;
140 91c13e54 2004-02-29 devnull 			else if((me - m) >= UTFmax || fullrune(m, me-m))
141 91c13e54 2004-02-29 devnull 				m += chartorune(&r, m);
142 91c13e54 2004-02-29 devnull 			else
143 91c13e54 2004-02-29 devnull 				break;
144 91c13e54 2004-02-29 devnull 		}else{
145 91c13e54 2004-02-29 devnull 			if(rm >= rme)
146 91c13e54 2004-02-29 devnull 				break;
147 91c13e54 2004-02-29 devnull 			r = *(uchar*)rm++;
148 91c13e54 2004-02-29 devnull 		}
149 91c13e54 2004-02-29 devnull 		if(f->runes){
150 91c13e54 2004-02-29 devnull 			FMTRCHAR(f, rt, rs, r);
151 91c13e54 2004-02-29 devnull 			if(r == '\'')
152 91c13e54 2004-02-29 devnull 				FMTRCHAR(f, rt, rs, r);
153 91c13e54 2004-02-29 devnull 		}else{
154 91c13e54 2004-02-29 devnull 			FMTRUNE(f, t, s, r);
155 91c13e54 2004-02-29 devnull 			if(r == '\'')
156 91c13e54 2004-02-29 devnull 				FMTRUNE(f, t, s, r);
157 91c13e54 2004-02-29 devnull 		}
158 91c13e54 2004-02-29 devnull 	}
159 91c13e54 2004-02-29 devnull 
160 91c13e54 2004-02-29 devnull 	if(f->runes){
161 91c13e54 2004-02-29 devnull 		FMTRCHAR(f, rt, rs, '\'');
162 91c13e54 2004-02-29 devnull 		USED(rs);
163 91c13e54 2004-02-29 devnull 		f->nfmt += rt - (Rune *)f->to;
164 91c13e54 2004-02-29 devnull 		f->to = rt;
165 91c13e54 2004-02-29 devnull 		if(fl & FmtLeft && __rfmtpad(f, w - q->nrunesout) < 0)
166 91c13e54 2004-02-29 devnull 			return -1;
167 91c13e54 2004-02-29 devnull 	}else{
168 91c13e54 2004-02-29 devnull 		FMTRUNE(f, t, s, '\'');
169 91c13e54 2004-02-29 devnull 		USED(s);
170 91c13e54 2004-02-29 devnull 		f->nfmt += t - (char *)f->to;
171 91c13e54 2004-02-29 devnull 		f->to = t;
172 91c13e54 2004-02-29 devnull 		if(fl & FmtLeft && __fmtpad(f, w - q->nbytesout) < 0)
173 91c13e54 2004-02-29 devnull 			return -1;
174 91c13e54 2004-02-29 devnull 	}
175 91c13e54 2004-02-29 devnull 	return 0;
176 91c13e54 2004-02-29 devnull }
177 91c13e54 2004-02-29 devnull 
178 91c13e54 2004-02-29 devnull int
179 91c13e54 2004-02-29 devnull __quotestrfmt(int runesin, Fmt *f)
180 91c13e54 2004-02-29 devnull {
181 df121a00 2004-12-26 devnull 	int nin, outlen;
182 91c13e54 2004-02-29 devnull 	Rune *r;
183 91c13e54 2004-02-29 devnull 	char *s;
184 91c13e54 2004-02-29 devnull 	Quoteinfo q;
185 91c13e54 2004-02-29 devnull 
186 df121a00 2004-12-26 devnull 	nin = -1;
187 df121a00 2004-12-26 devnull 	if(f->flags&FmtPrec)
188 df121a00 2004-12-26 devnull 		nin = f->prec;
189 91c13e54 2004-02-29 devnull 	if(runesin){
190 91c13e54 2004-02-29 devnull 		r = va_arg(f->args, Rune *);
191 91c13e54 2004-02-29 devnull 		s = nil;
192 91c13e54 2004-02-29 devnull 	}else{
193 91c13e54 2004-02-29 devnull 		s = va_arg(f->args, char *);
194 91c13e54 2004-02-29 devnull 		r = nil;
195 91c13e54 2004-02-29 devnull 	}
196 91c13e54 2004-02-29 devnull 	if(!s && !r)
197 91c13e54 2004-02-29 devnull 		return __fmtcpy(f, (void*)"<nil>", 5, 5);
198 91c13e54 2004-02-29 devnull 
199 91c13e54 2004-02-29 devnull 	if(f->flush)
200 91c13e54 2004-02-29 devnull 		outlen = 0x7FFFFFFF;	/* if we can flush, no output limit */
201 91c13e54 2004-02-29 devnull 	else if(f->runes)
202 91c13e54 2004-02-29 devnull 		outlen = (Rune*)f->stop - (Rune*)f->to;
203 91c13e54 2004-02-29 devnull 	else
204 91c13e54 2004-02-29 devnull 		outlen = (char*)f->stop - (char*)f->to;
205 91c13e54 2004-02-29 devnull 
206 df121a00 2004-12-26 devnull 	__quotesetup(s, r, nin, outlen, &q, f->flags&FmtSharp, f->runes);
207 cbeb0b26 2006-04-01 devnull /*print("bytes in %d bytes out %d runes in %d runesout %d\n", q.nbytesin, q.nbytesout, q.nrunesin, q.nrunesout); */
208 91c13e54 2004-02-29 devnull 
209 91c13e54 2004-02-29 devnull 	if(runesin){
210 91c13e54 2004-02-29 devnull 		if(!q.quoted)
211 91c13e54 2004-02-29 devnull 			return __fmtrcpy(f, r, q.nrunesin);
212 91c13e54 2004-02-29 devnull 		return qstrfmt(nil, r, &q, f);
213 91c13e54 2004-02-29 devnull 	}
214 91c13e54 2004-02-29 devnull 
215 91c13e54 2004-02-29 devnull 	if(!q.quoted)
216 91c13e54 2004-02-29 devnull 		return __fmtcpy(f, s, q.nrunesin, q.nbytesin);
217 91c13e54 2004-02-29 devnull 	return qstrfmt(s, nil, &q, f);
218 91c13e54 2004-02-29 devnull }
219 91c13e54 2004-02-29 devnull 
220 91c13e54 2004-02-29 devnull int
221 91c13e54 2004-02-29 devnull quotestrfmt(Fmt *f)
222 91c13e54 2004-02-29 devnull {
223 91c13e54 2004-02-29 devnull 	return __quotestrfmt(0, f);
224 91c13e54 2004-02-29 devnull }
225 91c13e54 2004-02-29 devnull 
226 91c13e54 2004-02-29 devnull int
227 91c13e54 2004-02-29 devnull quoterunestrfmt(Fmt *f)
228 91c13e54 2004-02-29 devnull {
229 91c13e54 2004-02-29 devnull 	return __quotestrfmt(1, f);
230 91c13e54 2004-02-29 devnull }
231 91c13e54 2004-02-29 devnull 
232 91c13e54 2004-02-29 devnull void
233 91c13e54 2004-02-29 devnull quotefmtinstall(void)
234 91c13e54 2004-02-29 devnull {
235 91c13e54 2004-02-29 devnull 	fmtinstall('q', quotestrfmt);
236 91c13e54 2004-02-29 devnull 	fmtinstall('Q', quoterunestrfmt);
237 91c13e54 2004-02-29 devnull }
238 91c13e54 2004-02-29 devnull 
239 91c13e54 2004-02-29 devnull int
240 91c13e54 2004-02-29 devnull __needsquotes(char *s, int *quotelenp)
241 91c13e54 2004-02-29 devnull {
242 91c13e54 2004-02-29 devnull 	Quoteinfo q;
243 91c13e54 2004-02-29 devnull 
244 91c13e54 2004-02-29 devnull 	__quotesetup(s, nil, -1, 0x7FFFFFFF, &q, 0, 0);
245 91c13e54 2004-02-29 devnull 	*quotelenp = q.nbytesout;
246 91c13e54 2004-02-29 devnull 
247 91c13e54 2004-02-29 devnull 	return q.quoted;
248 91c13e54 2004-02-29 devnull }
249 91c13e54 2004-02-29 devnull 
250 91c13e54 2004-02-29 devnull int
251 91c13e54 2004-02-29 devnull __runeneedsquotes(Rune *r, int *quotelenp)
252 91c13e54 2004-02-29 devnull {
253 91c13e54 2004-02-29 devnull 	Quoteinfo q;
254 91c13e54 2004-02-29 devnull 
255 91c13e54 2004-02-29 devnull 	__quotesetup(nil, r, -1, 0x7FFFFFFF, &q, 0, 0);
256 91c13e54 2004-02-29 devnull 	*quotelenp = q.nrunesout;
257 91c13e54 2004-02-29 devnull 
258 91c13e54 2004-02-29 devnull 	return q.quoted;
259 91c13e54 2004-02-29 devnull }