1*4887Schin /***********************************************************************
2*4887Schin *                                                                      *
3*4887Schin *               This software is part of the ast package               *
4*4887Schin *           Copyright (c) 1985-2007 AT&T Knowledge Ventures            *
5*4887Schin *                      and is licensed under the                       *
6*4887Schin *                  Common Public License, Version 1.0                  *
7*4887Schin *                      by AT&T Knowledge Ventures                      *
8*4887Schin *                                                                      *
9*4887Schin *                A copy of the License is available at                 *
10*4887Schin *            http://www.opensource.org/licenses/cpl1.0.txt             *
11*4887Schin *         (with md5 checksum 059e8cd6165cb4c31e351f2b69388fd9)         *
12*4887Schin *                                                                      *
13*4887Schin *              Information and Software Systems Research               *
14*4887Schin *                            AT&T Research                             *
15*4887Schin *                           Florham Park NJ                            *
16*4887Schin *                                                                      *
17*4887Schin *                 Glenn Fowler <gsf@research.att.com>                  *
18*4887Schin *                  David Korn <dgk@research.att.com>                   *
19*4887Schin *                   Phong Vo <kpv@research.att.com>                    *
20*4887Schin *                                                                      *
21*4887Schin ***********************************************************************/
22*4887Schin #pragma prototyped
23*4887Schin 
24*4887Schin /*
25*4887Schin  * Glenn Fowler
26*4887Schin  * AT&T Research
27*4887Schin  *
28*4887Schin  * return printf(3) format signature given format string
29*4887Schin  * the format signature contains one char per format optionally preceded
30*4887Schin  * by the number of `*' args
31*4887Schin  *	c	char
32*4887Schin  *	d	double
33*4887Schin  *	D	long double
34*4887Schin  *	f	float
35*4887Schin  *	h	short
36*4887Schin  *	i	int
37*4887Schin  *	j	long long
38*4887Schin  *	l	long
39*4887Schin  *	p	void*
40*4887Schin  *	s	string
41*4887Schin  *	t	ptrdiff_t
42*4887Schin  *	z	size_t
43*4887Schin  *	?	unknown
44*4887Schin  */
45*4887Schin 
46*4887Schin #include <ast.h>
47*4887Schin #include <ctype.h>
48*4887Schin 
49*4887Schin char*
50*4887Schin fmtfmt(const char* as)
51*4887Schin {
52*4887Schin 	register char*	s = (char*)as;
53*4887Schin 	char*		buf;
54*4887Schin 	int		i;
55*4887Schin 	int		c;
56*4887Schin 	int		a;
57*4887Schin 	int		q;
58*4887Schin 	int		x;
59*4887Schin 	int		t;
60*4887Schin 	int		m;
61*4887Schin 	int		n;
62*4887Schin 	int		z;
63*4887Schin 	char		formats[256];
64*4887Schin 	unsigned int	extra[elementsof(formats)];
65*4887Schin 
66*4887Schin 	z = 1;
67*4887Schin 	i = m = 0;
68*4887Schin 	for (;;)
69*4887Schin 	{
70*4887Schin 		switch (*s++)
71*4887Schin 		{
72*4887Schin 		case 0:
73*4887Schin 			break;
74*4887Schin 		case '%':
75*4887Schin 			if (*s == '%')
76*4887Schin 				continue;
77*4887Schin 			n = 0;
78*4887Schin 			a = 0;
79*4887Schin 			q = 0;
80*4887Schin 			t = '?';
81*4887Schin 			x = 0;
82*4887Schin 			for (;;)
83*4887Schin 			{
84*4887Schin 				switch (c = *s++)
85*4887Schin 				{
86*4887Schin 				case 0:
87*4887Schin 					s--;
88*4887Schin 					break;
89*4887Schin 				case '(':
90*4887Schin 					q++;
91*4887Schin 					continue;
92*4887Schin 				case ')':
93*4887Schin 					if (--q <= 0)
94*4887Schin 						n = 0;
95*4887Schin 					continue;
96*4887Schin 				case '0': case '1': case '2': case '3':
97*4887Schin 				case '4': case '5': case '6': case '7':
98*4887Schin 				case '8': case '9':
99*4887Schin 					n = n * 10 + (c - '0');
100*4887Schin 					continue;
101*4887Schin 				case '$':
102*4887Schin 					a = n;
103*4887Schin 					n = 0;
104*4887Schin 					continue;
105*4887Schin 				case '*':
106*4887Schin 					x++;
107*4887Schin 					n = 0;
108*4887Schin 					continue;
109*4887Schin 				case 'h':
110*4887Schin 					if (!q)
111*4887Schin 						t = t == 'h' ? 'c' : 'h';
112*4887Schin 					continue;
113*4887Schin 				case 'l':
114*4887Schin 					if (!q)
115*4887Schin 						t = t == 'l' ? 'j' : 'l';
116*4887Schin 					continue;
117*4887Schin 				case 'j':
118*4887Schin 				case 't':
119*4887Schin 				case 'z':
120*4887Schin 					if (!q)
121*4887Schin 						t = c;
122*4887Schin 					continue;
123*4887Schin 				case 'c':
124*4887Schin 				case 'p':
125*4887Schin 				case 's':
126*4887Schin 					if (!q)
127*4887Schin 					{
128*4887Schin 						t = c;
129*4887Schin 						break;
130*4887Schin 					}
131*4887Schin 					continue;
132*4887Schin 				case 'e':
133*4887Schin 				case 'g':
134*4887Schin 					if (!q)
135*4887Schin 					{
136*4887Schin 						switch (t)
137*4887Schin 						{
138*4887Schin 						case 'j':
139*4887Schin 							t = 'D';
140*4887Schin 							break;
141*4887Schin 						default:
142*4887Schin 							t = 'd';
143*4887Schin 							break;
144*4887Schin 						}
145*4887Schin 						break;
146*4887Schin 					}
147*4887Schin 					continue;
148*4887Schin 				case 'f':
149*4887Schin 					if (!q)
150*4887Schin 					{
151*4887Schin 						switch (t)
152*4887Schin 						{
153*4887Schin 						case 'j':
154*4887Schin 							t = 'D';
155*4887Schin 							break;
156*4887Schin 						case 'l':
157*4887Schin 							t = 'd';
158*4887Schin 							break;
159*4887Schin 						default:
160*4887Schin 							t = c;
161*4887Schin 							break;
162*4887Schin 						}
163*4887Schin 						break;
164*4887Schin 					}
165*4887Schin 					continue;
166*4887Schin 				default:
167*4887Schin 					if (!q && isalpha(c))
168*4887Schin 					{
169*4887Schin 						if (t == '?')
170*4887Schin 							t = 'i';
171*4887Schin 						break;
172*4887Schin 					}
173*4887Schin 					n = 0;
174*4887Schin 					continue;
175*4887Schin 				}
176*4887Schin 				break;
177*4887Schin 			}
178*4887Schin 			if (a)
179*4887Schin 				i = a;
180*4887Schin 			else
181*4887Schin 				i++;
182*4887Schin 			if (i < elementsof(formats))
183*4887Schin 			{
184*4887Schin 				formats[i] = t;
185*4887Schin 				if (extra[i] = x)
186*4887Schin 					do z++; while (x /= 10);
187*4887Schin 				if (m < i)
188*4887Schin 					m = i;
189*4887Schin 			}
190*4887Schin 			continue;
191*4887Schin 		default:
192*4887Schin 			continue;
193*4887Schin 		}
194*4887Schin 		break;
195*4887Schin 	}
196*4887Schin 	s = buf = fmtbuf(m + z);
197*4887Schin 	for (i = 1; i <= m; i++)
198*4887Schin 	{
199*4887Schin 		if (extra[i])
200*4887Schin 			s += sfsprintf(s, 10, "%d", extra[m]);
201*4887Schin 		*s++ = formats[i];
202*4887Schin 	}
203*4887Schin 	*s = 0;
204*4887Schin 	return buf;
205*4887Schin }
206