1*4882a593Smuzhiyun // SPDX-License-Identifier: GPL-2.0
2*4882a593Smuzhiyun #include <sys/types.h>
3*4882a593Smuzhiyun #include <stdio.h>
4*4882a593Smuzhiyun #include <stdlib.h>
5*4882a593Smuzhiyun #include <string.h>
6*4882a593Smuzhiyun #include "symbol.h"
7*4882a593Smuzhiyun
8*4882a593Smuzhiyun #include "demangle-java.h"
9*4882a593Smuzhiyun
10*4882a593Smuzhiyun #include <linux/ctype.h>
11*4882a593Smuzhiyun #include <linux/kernel.h>
12*4882a593Smuzhiyun
13*4882a593Smuzhiyun enum {
14*4882a593Smuzhiyun MODE_PREFIX = 0,
15*4882a593Smuzhiyun MODE_CLASS = 1,
16*4882a593Smuzhiyun MODE_FUNC = 2,
17*4882a593Smuzhiyun MODE_TYPE = 3,
18*4882a593Smuzhiyun MODE_CTYPE = 4, /* class arg */
19*4882a593Smuzhiyun };
20*4882a593Smuzhiyun
21*4882a593Smuzhiyun #define BASE_ENT(c, n) [c - 'A']=n
22*4882a593Smuzhiyun static const char *base_types['Z' - 'A' + 1] = {
23*4882a593Smuzhiyun BASE_ENT('B', "byte" ),
24*4882a593Smuzhiyun BASE_ENT('C', "char" ),
25*4882a593Smuzhiyun BASE_ENT('D', "double" ),
26*4882a593Smuzhiyun BASE_ENT('F', "float" ),
27*4882a593Smuzhiyun BASE_ENT('I', "int" ),
28*4882a593Smuzhiyun BASE_ENT('J', "long" ),
29*4882a593Smuzhiyun BASE_ENT('S', "short" ),
30*4882a593Smuzhiyun BASE_ENT('Z', "boolean" ),
31*4882a593Smuzhiyun };
32*4882a593Smuzhiyun
33*4882a593Smuzhiyun /*
34*4882a593Smuzhiyun * demangle Java symbol between str and end positions and stores
35*4882a593Smuzhiyun * up to maxlen characters into buf. The parser starts in mode.
36*4882a593Smuzhiyun *
37*4882a593Smuzhiyun * Use MODE_PREFIX to process entire prototype till end position
38*4882a593Smuzhiyun * Use MODE_TYPE to process return type if str starts on return type char
39*4882a593Smuzhiyun *
40*4882a593Smuzhiyun * Return:
41*4882a593Smuzhiyun * success: buf
42*4882a593Smuzhiyun * error : NULL
43*4882a593Smuzhiyun */
44*4882a593Smuzhiyun static char *
__demangle_java_sym(const char * str,const char * end,char * buf,int maxlen,int mode)45*4882a593Smuzhiyun __demangle_java_sym(const char *str, const char *end, char *buf, int maxlen, int mode)
46*4882a593Smuzhiyun {
47*4882a593Smuzhiyun int rlen = 0;
48*4882a593Smuzhiyun int array = 0;
49*4882a593Smuzhiyun int narg = 0;
50*4882a593Smuzhiyun const char *q;
51*4882a593Smuzhiyun
52*4882a593Smuzhiyun if (!end)
53*4882a593Smuzhiyun end = str + strlen(str);
54*4882a593Smuzhiyun
55*4882a593Smuzhiyun for (q = str; q != end; q++) {
56*4882a593Smuzhiyun
57*4882a593Smuzhiyun if (rlen == (maxlen - 1))
58*4882a593Smuzhiyun break;
59*4882a593Smuzhiyun
60*4882a593Smuzhiyun switch (*q) {
61*4882a593Smuzhiyun case 'L':
62*4882a593Smuzhiyun if (mode == MODE_PREFIX || mode == MODE_TYPE) {
63*4882a593Smuzhiyun if (mode == MODE_TYPE) {
64*4882a593Smuzhiyun if (narg)
65*4882a593Smuzhiyun rlen += scnprintf(buf + rlen, maxlen - rlen, ", ");
66*4882a593Smuzhiyun narg++;
67*4882a593Smuzhiyun }
68*4882a593Smuzhiyun if (mode == MODE_PREFIX)
69*4882a593Smuzhiyun mode = MODE_CLASS;
70*4882a593Smuzhiyun else
71*4882a593Smuzhiyun mode = MODE_CTYPE;
72*4882a593Smuzhiyun } else
73*4882a593Smuzhiyun buf[rlen++] = *q;
74*4882a593Smuzhiyun break;
75*4882a593Smuzhiyun case 'B':
76*4882a593Smuzhiyun case 'C':
77*4882a593Smuzhiyun case 'D':
78*4882a593Smuzhiyun case 'F':
79*4882a593Smuzhiyun case 'I':
80*4882a593Smuzhiyun case 'J':
81*4882a593Smuzhiyun case 'S':
82*4882a593Smuzhiyun case 'Z':
83*4882a593Smuzhiyun if (mode == MODE_TYPE) {
84*4882a593Smuzhiyun if (narg)
85*4882a593Smuzhiyun rlen += scnprintf(buf + rlen, maxlen - rlen, ", ");
86*4882a593Smuzhiyun rlen += scnprintf(buf + rlen, maxlen - rlen, "%s", base_types[*q - 'A']);
87*4882a593Smuzhiyun while (array--)
88*4882a593Smuzhiyun rlen += scnprintf(buf + rlen, maxlen - rlen, "[]");
89*4882a593Smuzhiyun array = 0;
90*4882a593Smuzhiyun narg++;
91*4882a593Smuzhiyun } else
92*4882a593Smuzhiyun buf[rlen++] = *q;
93*4882a593Smuzhiyun break;
94*4882a593Smuzhiyun case 'V':
95*4882a593Smuzhiyun if (mode == MODE_TYPE) {
96*4882a593Smuzhiyun rlen += scnprintf(buf + rlen, maxlen - rlen, "void");
97*4882a593Smuzhiyun while (array--)
98*4882a593Smuzhiyun rlen += scnprintf(buf + rlen, maxlen - rlen, "[]");
99*4882a593Smuzhiyun array = 0;
100*4882a593Smuzhiyun } else
101*4882a593Smuzhiyun buf[rlen++] = *q;
102*4882a593Smuzhiyun break;
103*4882a593Smuzhiyun case '[':
104*4882a593Smuzhiyun if (mode != MODE_TYPE)
105*4882a593Smuzhiyun goto error;
106*4882a593Smuzhiyun array++;
107*4882a593Smuzhiyun break;
108*4882a593Smuzhiyun case '(':
109*4882a593Smuzhiyun if (mode != MODE_FUNC)
110*4882a593Smuzhiyun goto error;
111*4882a593Smuzhiyun buf[rlen++] = *q;
112*4882a593Smuzhiyun mode = MODE_TYPE;
113*4882a593Smuzhiyun break;
114*4882a593Smuzhiyun case ')':
115*4882a593Smuzhiyun if (mode != MODE_TYPE)
116*4882a593Smuzhiyun goto error;
117*4882a593Smuzhiyun buf[rlen++] = *q;
118*4882a593Smuzhiyun narg = 0;
119*4882a593Smuzhiyun break;
120*4882a593Smuzhiyun case ';':
121*4882a593Smuzhiyun if (mode != MODE_CLASS && mode != MODE_CTYPE)
122*4882a593Smuzhiyun goto error;
123*4882a593Smuzhiyun /* safe because at least one other char to process */
124*4882a593Smuzhiyun if (isalpha(*(q + 1)) && mode == MODE_CLASS)
125*4882a593Smuzhiyun rlen += scnprintf(buf + rlen, maxlen - rlen, ".");
126*4882a593Smuzhiyun if (mode == MODE_CLASS)
127*4882a593Smuzhiyun mode = MODE_FUNC;
128*4882a593Smuzhiyun else if (mode == MODE_CTYPE)
129*4882a593Smuzhiyun mode = MODE_TYPE;
130*4882a593Smuzhiyun break;
131*4882a593Smuzhiyun case '/':
132*4882a593Smuzhiyun if (mode != MODE_CLASS && mode != MODE_CTYPE)
133*4882a593Smuzhiyun goto error;
134*4882a593Smuzhiyun rlen += scnprintf(buf + rlen, maxlen - rlen, ".");
135*4882a593Smuzhiyun break;
136*4882a593Smuzhiyun default :
137*4882a593Smuzhiyun buf[rlen++] = *q;
138*4882a593Smuzhiyun }
139*4882a593Smuzhiyun }
140*4882a593Smuzhiyun buf[rlen] = '\0';
141*4882a593Smuzhiyun return buf;
142*4882a593Smuzhiyun error:
143*4882a593Smuzhiyun return NULL;
144*4882a593Smuzhiyun }
145*4882a593Smuzhiyun
146*4882a593Smuzhiyun /*
147*4882a593Smuzhiyun * Demangle Java function signature (openJDK, not GCJ)
148*4882a593Smuzhiyun * input:
149*4882a593Smuzhiyun * str: string to parse. String is not modified
150*4882a593Smuzhiyun * flags: comobination of JAVA_DEMANGLE_* flags to modify demangling
151*4882a593Smuzhiyun * return:
152*4882a593Smuzhiyun * if input can be demangled, then a newly allocated string is returned.
153*4882a593Smuzhiyun * if input cannot be demangled, then NULL is returned
154*4882a593Smuzhiyun *
155*4882a593Smuzhiyun * Note: caller is responsible for freeing demangled string
156*4882a593Smuzhiyun */
157*4882a593Smuzhiyun char *
java_demangle_sym(const char * str,int flags)158*4882a593Smuzhiyun java_demangle_sym(const char *str, int flags)
159*4882a593Smuzhiyun {
160*4882a593Smuzhiyun char *buf, *ptr;
161*4882a593Smuzhiyun char *p;
162*4882a593Smuzhiyun size_t len, l1 = 0;
163*4882a593Smuzhiyun
164*4882a593Smuzhiyun if (!str)
165*4882a593Smuzhiyun return NULL;
166*4882a593Smuzhiyun
167*4882a593Smuzhiyun /* find start of retunr type */
168*4882a593Smuzhiyun p = strrchr(str, ')');
169*4882a593Smuzhiyun if (!p)
170*4882a593Smuzhiyun return NULL;
171*4882a593Smuzhiyun
172*4882a593Smuzhiyun /*
173*4882a593Smuzhiyun * expansion factor estimated to 3x
174*4882a593Smuzhiyun */
175*4882a593Smuzhiyun len = strlen(str) * 3 + 1;
176*4882a593Smuzhiyun buf = malloc(len);
177*4882a593Smuzhiyun if (!buf)
178*4882a593Smuzhiyun return NULL;
179*4882a593Smuzhiyun
180*4882a593Smuzhiyun buf[0] = '\0';
181*4882a593Smuzhiyun if (!(flags & JAVA_DEMANGLE_NORET)) {
182*4882a593Smuzhiyun /*
183*4882a593Smuzhiyun * get return type first
184*4882a593Smuzhiyun */
185*4882a593Smuzhiyun ptr = __demangle_java_sym(p + 1, NULL, buf, len, MODE_TYPE);
186*4882a593Smuzhiyun if (!ptr)
187*4882a593Smuzhiyun goto error;
188*4882a593Smuzhiyun
189*4882a593Smuzhiyun /* add space between return type and function prototype */
190*4882a593Smuzhiyun l1 = strlen(buf);
191*4882a593Smuzhiyun buf[l1++] = ' ';
192*4882a593Smuzhiyun }
193*4882a593Smuzhiyun
194*4882a593Smuzhiyun /* process function up to return type */
195*4882a593Smuzhiyun ptr = __demangle_java_sym(str, p + 1, buf + l1, len - l1, MODE_PREFIX);
196*4882a593Smuzhiyun if (!ptr)
197*4882a593Smuzhiyun goto error;
198*4882a593Smuzhiyun
199*4882a593Smuzhiyun return buf;
200*4882a593Smuzhiyun error:
201*4882a593Smuzhiyun free(buf);
202*4882a593Smuzhiyun return NULL;
203*4882a593Smuzhiyun }
204