tccpe: support leading underscore for symbols

To make this the default, enable this line in libtcc.c:tcc_new: #if defined(TCC_TARGET_PE) && 0 s->leading_underscore = 1; and then recompile tcc and also libtcc1.a

grischka committed 15 years ago

Commit a3ebdd01 parent 11b2d333 files · +42 −24Browse files

Unified · Split

3 files

libtcc.c +1 −1
tccpe.c +39 −23
win32/lib/chkstk.S +2 −0

libtcc.c+1−1

 #ifdef CHAR_IS_UNSIGNED
     s->char_is_unsigned = 1;
 #endif
+    /* enable this if you want symbols with leading underscore on windows: */
 #if defined(TCC_TARGET_PE) && 0
-    /* XXX: currently the PE linker is not ready to support that */
     s->leading_underscore = 1;
 #endif
     if (s->section_align == 0)

tccpe.c+39−23

 /* --------------------------------------------*/
-static const char* get_alt_symbol(char *buffer, const char *symbol)
+static const char *pe_export_name(TCCState *s1, ElfW(Sym) *sym)
 {
-    const char *p;
+    const char *name = symtab_section->link->data + sym->st_name;
-    p = strrchr(symbol, '@');
+    if (s1->leading_underscore && name[0] == '_' && !(sym->st_other & 2))
-    if (p && isnum(p[1]) && symbol[0] == '_') { /* stdcall decor */
+        return name + 1;
-        strcpy(buffer, symbol+1)[p-symbol-1] = 0;
+    return name;
-    } else if (symbol[0] != '_') { /* try non-ansi function */
-        buffer[0] = '_', strcpy(buffer + 1, symbol);
-    } else if (0 == memcmp(symbol, "__imp__", 7)) { /* mingw 2.0 */
-        strcpy(buffer, symbol + 6);
-    } else if (0 == memcmp(symbol, "_imp___", 7)) { /* mingw 3.7 */
-        strcpy(buffer, symbol + 6);
-    } else {
-        return symbol;
-    }
-    return buffer;
 }
-static int pe_find_import(TCCState * s1, const char *symbol)
+static int pe_find_import(TCCState * s1, ElfW(Sym) *sym)
 {
     char buffer[200];
-    const char *s;
+    const char *s, *p;
     int sym_index, n = 0;
     do {
-        s = n ? get_alt_symbol(buffer, symbol) : symbol;
+        s = pe_export_name(s1, sym);
+        if (n) {
+            /* second try: */
+	    if (sym->st_other & 2) {
+                /* try w/0 stdcall deco (windows API convention) */
+	        p = strrchr(s, '@');
+	        if (!p || s[0] != '_')
+	            break;
+	        strcpy(buffer, s+1)[p-s-1] = 0;
+	    } else if (s[0] != '_') { /* try non-ansi function */
+	        buffer[0] = '_', strcpy(buffer + 1, s);
+	    } else if (0 == memcmp(s, "__imp__", 7)) { /* mingw 2.0 */
+	        strcpy(buffer, s + 6);
+	    } else if (0 == memcmp(s, "_imp___", 7)) { /* mingw 3.7 */
+	        strcpy(buffer, s + 6);
+	    } else {
+	        break;
+	    }
+	    s = buffer;
+        }
         sym_index = find_elf_sym(s1->dynsymtab_section, s);
         // printf("find (%d) %d %s\n", n, sym_index, s);
     } while (0 == sym_index && ++n < 2);
     sym_end = symtab_section->data_offset / sizeof(ElfW(Sym));
     for (sym_index = 1; sym_index < sym_end; ++sym_index) {
         sym = (ElfW(Sym)*)symtab_section->data + sym_index;
-        name = symtab_section->link->data + sym->st_name;
+        name = pe_export_name(pe->s1, sym);
         if ((sym->st_other & 1)
             /* export only symbols from actually written sections */
             && pe->s1->sections[sym->st_shndx]->sh_addr) {
             const char *name = symtab_section->link->data + sym->st_name;
             unsigned type = ELFW_ST_TYPE(sym->st_info);
-            int imp_sym = pe_find_import(pe->s1, name);
+            int imp_sym = pe_find_import(pe->s1, sym);
             struct import_symbol *is;
             if (0 == imp_sym)
     start_symbol =
         TCC_OUTPUT_MEMORY == s1->output_type
-        ? PE_GUI == pe_type ? "_runwinmain" : NULL
+        ? PE_GUI == pe_type ? "__runwinmain" : "_main"
-        : PE_DLL == pe_type ? PE_STDSYM("_dllstart","@12")
+        : PE_DLL == pe_type ? PE_STDSYM("__dllstart","@12")
-        : PE_GUI == pe_type ? "_winstart" : "_start"
+        : PE_GUI == pe_type ? "__winstart" : "__start"
         ;
+    if (!s1->leading_underscore || strchr(start_symbol, '@')) {
+        ++start_symbol;
+        if (start_symbol[0] != '_')
+            start_symbol = NULL;
+    }
     /* grab the startup code from libtcc1 */
     if (start_symbol)
         add_elf_sym(symtab_section,

win32/lib/chkstk.S+2−0

@@ -150,6 +150,8 @@ seh_scopetable: · 149 unmodified lines
150	seh_handler:	150	seh_handler:
151	jmp _except_handler3	151	jmp _except_handler3
152		152
		153	.globl ___try__
		154	___try__:
153	.globl __try__	155	.globl __try__
154	__try__:	156	__try__:
155	push %ebp	157	push %ebp

tccpe: support leading underscore for symbols

Keyboard shortcuts

General

Issues and pull requests

Issues, pull requests and Inbox

Repository code