1 /*-
2  * Copyright (c) 2009 Kai Wang
3  * All rights reserved.
4  *
5  * Redistribution and use in source and binary forms, with or without
6  * modification, are permitted provided that the following conditions
7  * are met:
8  * 1. Redistributions of source code must retain the above copyright
9  *    notice, this list of conditions and the following disclaimer
10  *    in this position and unchanged.
11  * 2. Redistributions in binary form must reproduce the above copyright
12  *    notice, this list of conditions and the following disclaimer in the
13  *    documentation and/or other materials provided with the distribution.
14  *
15  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR(S) ``AS IS'' AND ANY EXPRESS OR
16  * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
17  * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
18  * IN NO EVENT SHALL THE AUTHOR(S) BE LIABLE FOR ANY DIRECT, INDIRECT,
19  * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
20  * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
21  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
22  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
23  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
24  * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
25  */
26 
27 #include <sys/param.h>
28 #include <assert.h>
29 #include <errno.h>
30 #include <libelftc.h>
31 #include <stdlib.h>
32 #include <string.h>
33 
34 #include "_libelftc.h"
35 
36 ELFTC_VCSID("$Id: elftc_demangle.c 3296 2016-01-09 14:17:28Z jkoshy $");
37 
38 static unsigned int
39 is_mangled(const char *s, unsigned int style)
40 {
41 
42 	switch (style) {
43 	case ELFTC_DEM_ARM: return (is_cpp_mangled_ARM(s) ? style : 0);
44 	case ELFTC_DEM_GNU2: return (is_cpp_mangled_gnu2(s) ? style : 0);
45 	case ELFTC_DEM_GNU3: return (is_cpp_mangled_gnu3(s) ? style : 0);
46 	}
47 
48 	/* No style or invalid style spcified, try to guess. */
49 	if (is_cpp_mangled_gnu3(s))
50 		return (ELFTC_DEM_GNU3);
51 	if (is_cpp_mangled_gnu2(s))
52 		return (ELFTC_DEM_GNU2);
53 	if (is_cpp_mangled_ARM(s))
54 		return (ELFTC_DEM_ARM);
55 
56 	/* Cannot be demangled. */
57 	return (0);
58 }
59 
60 static char *
61 demangle(const char *s, unsigned int style, unsigned int rc)
62 {
63 
64 	(void) rc;			/* XXX */
65 	switch (style) {
66 	case ELFTC_DEM_ARM: return (cpp_demangle_ARM(s));
67 	case ELFTC_DEM_GNU2: return (cpp_demangle_gnu2(s));
68 	case ELFTC_DEM_GNU3: return (cpp_demangle_gnu3(s));
69 	default:
70 		assert(0);
71 		return (NULL);
72 	}
73 }
74 
75 int
76 elftc_demangle(const char *mangledname, char *buffer, size_t bufsize,
77     unsigned int flags)
78 {
79 	unsigned int style, rc;
80 	char *rlt;
81 
82 	style = flags & 0xFFFF;
83 	rc = flags >> 16;
84 
85 	if (mangledname == NULL ||
86 	    ((style = is_mangled(mangledname, style)) == 0)) {
87 		errno = EINVAL;
88 		return (-1);
89 	}
90 
91 	if ((rlt = demangle(mangledname, style, rc)) == NULL) {
92 		errno = EINVAL;
93 		return (-1);
94 	}
95 
96 	if (buffer == NULL || bufsize < strlen(rlt) + 1) {
97 		free(rlt);
98 		errno = ENAMETOOLONG;
99 		return (-1);
100 	}
101 
102 	strncpy(buffer, rlt, bufsize);
103 	buffer[bufsize - 1] = '\0';
104 	free(rlt);
105 
106 	return (0);
107 }
108