type: Make an assert()ion independent of the last entry of an enum.
[cparser] / mangle.c
index 83fb373..e79d08a 100644 (file)
--- a/mangle.c
+++ b/mangle.c
@@ -1,21 +1,6 @@
 /*
  * This file is part of cparser.
- * Copyright (C) 2007-2008 Matthias Braun <matze@braunis.de>
- *
- * This program is free software; you can redistribute it and/or
- * modify it under the terms of the GNU General Public License
- * as published by the Free Software Foundation; either version 2
- * of the License, or (at your option) any later version.
- *
- * This program is distributed in the hope that it will be useful,
- * but WITHOUT ANY WARRANTY; without even the implied warranty of
- * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
- * GNU General Public License for more details.
- *
- * You should have received a copy of the GNU General Public License
- * along with this program; if not, write to the Free Software
- * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA
- * 02111-1307, USA.
+ * Copyright (C) 2012 Matthias Braun <matze@braunis.de>
  */
 #include <config.h>
 
@@ -27,7 +12,6 @@
 #include "symbol_t.h"
 #include "mangle.h"
 #include "diagnostic.h"
-#include "ast2firm.h"
 #include "lang_features.h"
 #include "adt/error.h"
 
@@ -38,8 +22,8 @@ static void mangle_type(type_t *type);
 static char get_atomic_type_mangle(atomic_type_kind_t kind)
 {
        switch (kind) {
-       case ATOMIC_TYPE_INVALID: break;
        case ATOMIC_TYPE_VOID:        return 'v';
+       case ATOMIC_TYPE_WCHAR_T:     return 'w';
        case ATOMIC_TYPE_BOOL:        return 'b';
        case ATOMIC_TYPE_CHAR:        return 'c';
        case ATOMIC_TYPE_SCHAR:       return 'a';
@@ -56,7 +40,7 @@ static char get_atomic_type_mangle(atomic_type_kind_t kind)
        case ATOMIC_TYPE_FLOAT:       return 'f';
        case ATOMIC_TYPE_DOUBLE:      return 'd';
        }
-       panic("invalid atomic type in mangler");
+       panic("invalid atomic type");
 }
 
 static void mangle_atomic_type(const atomic_type_t *type)
@@ -70,6 +54,12 @@ static void mangle_pointer_type(const pointer_type_t *type)
        mangle_type(type->points_to);
 }
 
+static void mangle_reference_type(const reference_type_t *type)
+{
+       obstack_1grow(&obst, 'R');
+       mangle_type(type->refers_to);
+}
+
 static void mangle_parameters(const function_type_t *type)
 {
        if (type->unspecified_parameters)
@@ -143,13 +133,13 @@ static void mangle_array_type(const array_type_t *type)
        mangle_type(type->element_type);
 }
 
-static void mangle_complex_type(const complex_type_t *type)
+static void mangle_complex_type(const atomic_type_t *type)
 {
        obstack_1grow(&obst, 'C');
        obstack_1grow(&obst, get_atomic_type_mangle(type->akind));
 }
 
-static void mangle_imaginary_type(const imaginary_type_t *type)
+static void mangle_imaginary_type(const atomic_type_t *type)
 {
        obstack_1grow(&obst, 'G');
        obstack_1grow(&obst, get_atomic_type_mangle(type->akind));
@@ -182,6 +172,9 @@ static void mangle_type(type_t *orig_type)
        case TYPE_POINTER:
                mangle_pointer_type(&type->pointer);
                return;
+       case TYPE_REFERENCE:
+               mangle_reference_type(&type->reference);
+               return;
        case TYPE_FUNCTION:
                mangle_function_type(&type->function);
                return;
@@ -196,33 +189,42 @@ static void mangle_type(type_t *orig_type)
                mangle_array_type(&type->array);
                return;
        case TYPE_COMPLEX:
-               mangle_complex_type(&type->complex);
+               mangle_complex_type(&type->atomic);
                return;
        case TYPE_IMAGINARY:
-               mangle_imaginary_type(&type->imaginary);
+               mangle_imaginary_type(&type->atomic);
                return;
-       case TYPE_INVALID:
-               panic("invalid type encountered while mangling");
        case TYPE_ERROR:
                panic("error type encountered while mangling");
-       case TYPE_BUILTIN:
        case TYPE_TYPEDEF:
        case TYPE_TYPEOF:
                panic("typeref not resolved while manging?!?");
-
-       case TYPE_BITFIELD:
-               panic("no mangling for this type implemented yet");
-               break;
        }
        panic("invalid type encountered while mangling");
 }
 
+static void mangle_namespace(entity_t *entity)
+{
+       for (entity_t *e = entity->base.parent_entity; e != NULL;
+            e = e->base.parent_entity) {
+           /* TODO: we need something similar (or the same?) for classes */
+               if (e->kind == ENTITY_NAMESPACE) {
+                       mangle_namespace(e);
+                       print_name(e->base.symbol->string);
+                       return;
+               }
+       }
+}
+
 static void mangle_entity(entity_t *entity)
 {
        obstack_1grow(&obst, '_');
        obstack_1grow(&obst, 'Z');
 
-       /* TODO: mangle scope */
+       if (entity->base.parent_entity != NULL) {
+               obstack_1grow(&obst, 'N');
+               mangle_namespace(entity);
+       }
 
        print_name(entity->base.symbol->string);
 
@@ -252,55 +254,53 @@ ident *create_name_win32(entity_t *entity)
 
        assert(is_declaration(entity));
 
-       if (entity->declaration.modifiers & DM_DLLIMPORT) {
-               /* add prefix for imported symbols */
-               obstack_printf(o, "__imp_");
-       }
-
        if (entity->kind == ENTITY_FUNCTION) {
-               cc_kind_t cc = entity->declaration.type->function.calling_convention;
+               type_t *type = skip_typeref(entity->declaration.type);
+               assert(is_type_function(type));
+
+               if (entity->declaration.modifiers & DM_DLLIMPORT)
+                       /* add prefix for imported symbols */
+                       obstack_printf(o, "__imp_");
+
+               cc_kind_t cc = type->function.calling_convention;
 
                /* calling convention prefix */
                switch (cc) {
-                       case CC_DEFAULT:
-                       case CC_CDECL:
-                       case CC_STDCALL:  obstack_1grow(o, '_'); break;
-                       case CC_FASTCALL: obstack_1grow(o, '@'); break;
-                       default:          panic("unhandled calling convention");
+               case CC_DEFAULT:
+               case CC_CDECL:
+               case CC_STDCALL:  obstack_1grow(o, '_'); break;
+               case CC_FASTCALL: obstack_1grow(o, '@'); break;
+               default:          panic("unhandled calling convention");
                }
 
-               switch (entity->declaration.type->function.linkage) {
-                       case LINKAGE_INVALID:
-                               break;
-
-                       case LINKAGE_C:
-                               obstack_printf(o, "%s", entity->base.symbol->string);
-                               break;
+               switch (type->function.linkage) {
+               case LINKAGE_C:
+                       obstack_printf(o, "%s", entity->base.symbol->string);
+                       break;
 
-                       case LINKAGE_CXX:
-                               mangle_entity(entity);
-                               break;
+               case LINKAGE_CXX:
+                       mangle_entity(entity);
+                       break;
                }
 
                /* calling convention suffix */
                switch (cc) {
-                       case CC_DEFAULT:
-                       case CC_CDECL:
-                               break;
-
-                       case CC_STDCALL:
-                       case CC_FASTCALL: {
-                               ir_type  *irtype = get_ir_type(entity->declaration.type);
-                               unsigned size    = 0;
-                               for (int i = get_method_n_params(irtype) - 1; i >= 0; --i) {
-                                       size += get_type_size_bytes(get_method_param_type(irtype, i));
-                               }
-                               obstack_printf(o, "@%u", size);
-                               break;
+               case CC_DEFAULT:
+               case CC_CDECL:
+                       break;
+
+               case CC_STDCALL:
+               case CC_FASTCALL: {
+                       unsigned size = 0;
+                       for (function_parameter_t const* i = type->function.parameters; i; i = i->next) {
+                               size += get_type_size(i->type);
                        }
+                       obstack_printf(o, "@%u", size);
+                       break;
+               }
 
-                       default:
-                               panic("unhandled calling convention");
+               default:
+                       panic("unhandled calling convention");
                }
        } else {
                obstack_printf(o, "_%s", entity->base.symbol->string);
@@ -317,22 +317,24 @@ ident *create_name_win32(entity_t *entity)
  */
 ident *create_name_linux_elf(entity_t *entity)
 {
-       bool needs_mangling = false;
+       const char *name = entity->base.symbol->string;
 
        if (entity->kind == ENTITY_FUNCTION) {
-               switch (entity->declaration.type->function.linkage) {
-                       case LINKAGE_INVALID: break;
-                       case LINKAGE_C:       break;
-                       case LINKAGE_CXX:     needs_mangling = true; break;
+               type_t *type = skip_typeref(entity->declaration.type);
+               assert(is_type_function(type));
+               switch (type->function.linkage) {
+                       case LINKAGE_C:
+                               if (entity->function.actual_name != NULL)
+                                       name = entity->function.actual_name->string;
+                               break;
+                       case LINKAGE_CXX:
+                               // TODO What about __REDIRECT/actual_name with mangling?
+                               mangle_entity(entity);
+                               return make_id_from_obst();
                }
        }
 
-       if (needs_mangling) {
-               mangle_entity(entity);
-               return make_id_from_obst();
-       }
-
-       return new_id_from_str(entity->base.symbol->string);
+       return new_id_from_str(name);
 }
 
 /**
@@ -343,6 +345,18 @@ ident *create_name_linux_elf(entity_t *entity)
  */
 ident *create_name_macho(entity_t *entity)
 {
+       if (entity->kind == ENTITY_FUNCTION) {
+               type_t *type = skip_typeref(entity->declaration.type);
+               assert(is_type_function(type));
+
+               switch (type->function.linkage) {
+                       default:
+                               if (entity->function.actual_name != NULL)
+                                       return new_id_from_str(entity->function.actual_name->string);
+                               break;
+               }
+       }
+
        obstack_printf(&obst, "_%s", entity->base.symbol->string);
        return make_id_from_obst();
 }