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
36ELFTC_VCSID("$Id: elftc_demangle.c 3296 2016-01-09 14:17:28Z jkoshy $");
37
38static unsigned int
39is_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
60static char *
61demangle(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
75int
76elftc_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