forked from KolibriOS/kolibrios
555 lines
14 KiB
C
555 lines
14 KiB
C
/*
|
|
* TCCMEOS.C - KolibriOS/MenuetOS file output for the TinyC Compiler
|
|
*
|
|
* Copyright (c) 2006 Andrey Khalyavin
|
|
* Copyright (c) 2021 Coldy (KX extension)
|
|
*
|
|
* This library is free software; you can redistribute it and/or
|
|
* modify it under the terms of the GNU Lesser General Public
|
|
* License as published by the Free Software Foundation; either
|
|
* version 2 of the License, or (at your option) any later version.
|
|
*
|
|
* This library 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
|
|
* Lesser General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU Lesser General Public
|
|
* License along with this library; if not, write to the Free Software
|
|
* Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
|
|
*/
|
|
|
|
#include "tcc.h"
|
|
|
|
typedef struct {
|
|
char magic[8];
|
|
int version;
|
|
int entry_point;
|
|
int image_size;
|
|
int memory_size;
|
|
int stack;
|
|
int params;
|
|
int argv;
|
|
} IMAGE_MEOS_FILE_HEADER,*PIMAGE_MEOS_FILE_HEADER;
|
|
typedef struct _meos_section_info{
|
|
int sh_addr;
|
|
void* data;
|
|
int data_size;
|
|
int sec_num;
|
|
struct _meos_section_info* next;
|
|
} meos_section_info;
|
|
#ifdef TCC_TARGET_KX
|
|
typedef struct {
|
|
void* data;
|
|
int data_size;
|
|
} kx_import_table;
|
|
#endif
|
|
typedef struct {
|
|
TCCState* s1;
|
|
IMAGE_MEOS_FILE_HEADER header;
|
|
meos_section_info* code_sections;
|
|
meos_section_info* data_sections;
|
|
#ifdef TCC_TARGET_KX
|
|
kx_import_table* imp_table;
|
|
#endif
|
|
meos_section_info* bss_sections;
|
|
} me_info;
|
|
|
|
#ifdef TCC_TARGET_KX
|
|
void kx_init(me_info* me);
|
|
void kx_build_imports(me_info* me);
|
|
long kx_get_header_length(me_info* me);
|
|
void kx_write_header(me_info* me, FILE* f);
|
|
void kx_write_imports(me_info* me, FILE* f);
|
|
void kx_free(me_info* me);
|
|
#endif
|
|
|
|
int tcc_output_dbgme(const char *filename, me_info* me);
|
|
|
|
|
|
meos_section_info* findsection(me_info* me,int num)
|
|
{
|
|
meos_section_info* si;
|
|
for(si=me->code_sections;si;si=si->next)
|
|
{
|
|
if (si->sec_num==num)
|
|
return si;
|
|
}
|
|
for (si=me->data_sections;si;si=si->next)
|
|
{
|
|
if (si->sec_num==num)
|
|
return si;
|
|
}
|
|
for (si=me->bss_sections;si;si=si->next)
|
|
{
|
|
if (si->sec_num==num)
|
|
return si;
|
|
}
|
|
return (meos_section_info*)0;
|
|
}
|
|
|
|
void build_reloc(me_info* me)
|
|
{
|
|
int flag;
|
|
Elf32_Rel *rel = 0, *rel_ = 0, *rel_end = 0;
|
|
Section *sr;
|
|
meos_section_info* s;
|
|
meos_section_info* ss;
|
|
s=me->code_sections;
|
|
rel=0;
|
|
rel_end=0;
|
|
flag=0;
|
|
for(;;)
|
|
{
|
|
sr=me->s1->sections[s->sec_num]->reloc;
|
|
if (sr)
|
|
{
|
|
rel = (Elf32_Rel *) sr->data;
|
|
rel_end = (Elf32_Rel *) (sr->data + sr->data_offset);
|
|
}
|
|
rel_=rel;
|
|
while (rel_<rel_end){
|
|
rel=rel_;
|
|
int type = ELF32_R_TYPE(rel->r_info);
|
|
rel_=rel+1;
|
|
if (type != R_386_PC32 && type != R_386_32)
|
|
continue;
|
|
int sym = ELF32_R_SYM(rel->r_info);
|
|
if (sym>symtab_section->data_offset/sizeof(Elf32_Sym))
|
|
continue;
|
|
Elf32_Sym* esym = ((Elf32_Sym *)symtab_section->data)+sym;
|
|
int sect=esym->st_shndx;
|
|
int sh_addr;
|
|
ss=findsection(me,sect);
|
|
if (ss==0)
|
|
{
|
|
const char *sym_name = strtab_section->data + esym->st_name;
|
|
#ifdef TCC_TARGET_KX
|
|
int sym_index = find_elf_sym(me->s1->dynsymtab_section, sym_name);
|
|
Elf32_Sym* dyn_sym;
|
|
if (sym_index == 0) {
|
|
#endif
|
|
tcc_error_noabort("undefined symbol '%s'", sym_name);
|
|
continue;
|
|
#ifdef TCC_TARGET_KX
|
|
}
|
|
dyn_sym = &((ElfW(Sym) *)me->s1->dynsymtab_section->data)[sym_index];
|
|
sh_addr = dyn_sym->st_value;
|
|
if (sh_addr == 0) {
|
|
tcc_error_noabort("symbol '%s' has zero value", sym_name);
|
|
continue;
|
|
}
|
|
#endif
|
|
}
|
|
else
|
|
sh_addr = ss->sh_addr;
|
|
if (rel->r_offset>s->data_size)
|
|
continue;
|
|
if (type==R_386_PC32)
|
|
*(int*)(rel->r_offset+s->data)+=/*ss->*/sh_addr+esym->st_value-rel->r_offset-s->sh_addr;
|
|
else if (type==R_386_32)
|
|
*(int*)(rel->r_offset+s->data)+=/*ss->*/sh_addr+esym->st_value;
|
|
}
|
|
rel=rel_;
|
|
s=s->next;
|
|
if (s==0) // what about multiple BSS sections ?
|
|
{
|
|
if (flag) break;
|
|
s=me->data_sections;
|
|
if (s==0) break;
|
|
flag=1;
|
|
continue;
|
|
}
|
|
}
|
|
}
|
|
|
|
void assign_addresses(me_info* me)
|
|
{
|
|
int i;
|
|
meos_section_info* si;
|
|
for (i=1;i<me->s1->nb_sections;i++)
|
|
{
|
|
Section* s=me->s1->sections[i];
|
|
if (strcmp(".text",s->name)==0)
|
|
{
|
|
si=tcc_malloc(sizeof(meos_section_info));
|
|
si->data=s->data;
|
|
si->data_size=s->data_offset;
|
|
si->next=me->code_sections;
|
|
si->sec_num=i;
|
|
me->code_sections=si;
|
|
continue;
|
|
}
|
|
if (strcmp(".data",s->name)==0)
|
|
{
|
|
si=tcc_malloc(sizeof(meos_section_info));
|
|
si->data=s->data;
|
|
si->data_size=s->data_offset;
|
|
si->next=me->data_sections;
|
|
si->sec_num=i;
|
|
me->data_sections=si;
|
|
continue;
|
|
}
|
|
if (strcmp(".bss",s->name)==0)
|
|
{
|
|
si=tcc_malloc(sizeof(meos_section_info));
|
|
si->data=s->data;
|
|
si->data_size=s->data_offset;
|
|
si->next=me->bss_sections;
|
|
si->sec_num=i;
|
|
me->bss_sections=si;
|
|
continue;
|
|
}
|
|
}
|
|
int addr;
|
|
addr=sizeof(IMAGE_MEOS_FILE_HEADER);
|
|
#ifdef TCC_TARGET_KX
|
|
addr += kx_get_header_length(me);
|
|
#endif
|
|
for (si=me->code_sections;si;si=si->next)
|
|
{
|
|
si->sh_addr=addr;
|
|
addr+=si->data_size;
|
|
}
|
|
for (si=me->data_sections;si;si=si->next)
|
|
{
|
|
si->sh_addr=addr;
|
|
addr+=si->data_size;
|
|
}
|
|
me->header.image_size=addr;
|
|
#ifdef TCC_TARGET_KX
|
|
kx_build_imports(me);
|
|
addr = me->header.image_size;
|
|
#endif
|
|
for (si=me->bss_sections;si;si=si->next)
|
|
{
|
|
si->sh_addr=addr;
|
|
addr+=si->data_size;
|
|
}
|
|
if (me->s1->pe_stack_size < 4096)
|
|
addr+=4096;
|
|
else
|
|
addr += me->s1->pe_stack_size;
|
|
addr=(addr+4)&(~3);
|
|
me->header.stack=addr;
|
|
me->header.memory_size=addr;
|
|
build_reloc(me);
|
|
}
|
|
|
|
|
|
const char *tcc_get_symbol_name(int st_name)
|
|
// return string by index from stringtable section
|
|
{
|
|
const char *sym_name = strtab_section->data + st_name;
|
|
return sym_name;
|
|
}
|
|
|
|
int tcc_find_symbol_me(me_info* me, const char *sym_name, int* addr)
|
|
{
|
|
int i, symtab = 0, strtab = 0;
|
|
*addr = 0;
|
|
for (i=1;i<me->s1->nb_sections;i++)
|
|
{
|
|
Section* s;
|
|
s=me->s1->sections[i];
|
|
if (strcmp(s->name,".symtab")==0)
|
|
{
|
|
symtab=i;
|
|
}
|
|
if (strcmp(s->name,".strtab")==0)
|
|
{
|
|
strtab=i;
|
|
}
|
|
}
|
|
if (symtab==0 || strtab==0)
|
|
{
|
|
tcc_error_noabort("undefined sections .symtab or .strtab on linking '%s'", sym_name);
|
|
return 0;
|
|
}
|
|
Elf32_Sym* s,*se;
|
|
char* name;
|
|
s=(Elf32_Sym*)me->s1->sections[symtab]->data;
|
|
se=(Elf32_Sym*)(((void*)s)+me->s1->sections[symtab]->data_offset);
|
|
name=(char*)me->s1->sections[strtab]->data;
|
|
while (s<se)
|
|
{
|
|
if (strcmp(name+s->st_name,sym_name)==0)
|
|
{
|
|
*addr = s->st_value+findsection(me,s->st_shndx)->sh_addr;
|
|
return 1;
|
|
}
|
|
s++;
|
|
}
|
|
tcc_error_noabort("undefined symbol '%s'", sym_name);
|
|
return 0;
|
|
}
|
|
|
|
const char* me_magic="MENUET01";
|
|
int tcc_output_me(TCCState* s1,const char *filename)
|
|
{
|
|
me_info me;
|
|
int i;
|
|
FILE* f;
|
|
//printf("%d\n",s1->nb_sections);
|
|
memset(&me,0,sizeof(me));
|
|
me.s1=s1;
|
|
#ifdef TCC_TARGET_KX
|
|
kx_init(&me);
|
|
#endif
|
|
tcc_add_runtime(s1);
|
|
relocate_common_syms();
|
|
assign_addresses(&me);
|
|
|
|
if (s1->do_debug)
|
|
tcc_output_dbgme(filename, &me);
|
|
|
|
if (!tcc_find_symbol_me(&me, "start", &me.header.entry_point) |
|
|
!tcc_find_symbol_me(&me, "__argv", &me.header.params) |
|
|
!tcc_find_symbol_me(&me, "__path", &me.header.argv)) {
|
|
exit(1);
|
|
}
|
|
|
|
if((f=fopen(filename,"wb"))==NULL){
|
|
tcc_error("could not create '%s': %s", filename, strerror(errno));
|
|
}
|
|
|
|
for (i=0;i<8;i++)
|
|
me.header.magic[i]=me_magic[i];
|
|
fwrite(&me.header,1,sizeof(IMAGE_MEOS_FILE_HEADER),f);
|
|
#ifdef TCC_TARGET_KX
|
|
kx_write_header(&me, f);
|
|
#endif
|
|
meos_section_info* si;
|
|
for(si=me.code_sections;si;si=si->next)
|
|
fwrite(si->data,1,si->data_size,f);
|
|
for (si=me.data_sections;si;si=si->next)
|
|
fwrite(si->data,1,si->data_size,f);
|
|
#ifdef TCC_TARGET_KX
|
|
kx_write_imports(&me, f);
|
|
kx_free(&me);
|
|
#else
|
|
if (!s1->nobss)
|
|
{
|
|
for (si=me.bss_sections;si;si=si->next)
|
|
{
|
|
if (si->data == NULL)
|
|
{
|
|
// printf("\nError! BSS data is NULL! size:%i",(int)si->data_size);
|
|
si->data = calloc(si->data_size, 1);
|
|
}
|
|
fwrite(si->data, 1, si->data_size, f);
|
|
}
|
|
}
|
|
/*
|
|
if (me.bss_sections) // Siemargl testin, what we lose
|
|
{
|
|
tcc_error_noabort("We lose .BSS section when linking KOS32 executable");
|
|
}
|
|
*/
|
|
#endif
|
|
fclose(f);
|
|
return 0;
|
|
}
|
|
|
|
#if !defined(_WIN32) && !defined(TCC_TARGET_MEOS_LINUX)
|
|
|
|
static inline int get_current_folder(char* buf, int bufsize){
|
|
register int val;
|
|
asm volatile ("int $0x40":"=a"(val):"a"(30), "b"(2), "c"(buf), "d"(bufsize));
|
|
return val;
|
|
}
|
|
|
|
|
|
char *getcwd(char *buf, size_t size)
|
|
{
|
|
int rc = get_current_folder(buf, size);
|
|
if (rc > size)
|
|
{
|
|
errno = ERANGE;
|
|
return 0;
|
|
}
|
|
else
|
|
return buf;
|
|
}
|
|
|
|
#endif
|
|
|
|
|
|
static FILE *src_file;
|
|
static int next_src_line;
|
|
|
|
void close_source_file()
|
|
{
|
|
if (src_file)
|
|
fclose(src_file);
|
|
src_file = NULL;
|
|
}
|
|
|
|
void load_source_file(char *fname)
|
|
{
|
|
close_source_file();
|
|
src_file = fopen(fname,"rt");
|
|
if (!src_file) return;
|
|
next_src_line = 1;
|
|
}
|
|
|
|
int get_src_lines(char *buf, int sz, int start, int end)
|
|
// 1 if read
|
|
{
|
|
char line[255], *ch;
|
|
strcpy(buf, "");
|
|
if (!src_file) return 0;
|
|
while (next_src_line < start) // skip
|
|
{
|
|
ch = fgets(line, sizeof line, src_file);
|
|
if (!ch) return 0;
|
|
next_src_line++;
|
|
}
|
|
while (next_src_line <= end)
|
|
{
|
|
ch = fgets(line, sizeof line, src_file);
|
|
if (!ch) return 0;
|
|
next_src_line++;
|
|
strncat(buf, line, sz - strlen(buf) - 1);
|
|
}
|
|
// remove newlines
|
|
for (ch = buf; *ch; ch++)
|
|
if (strchr("\t\n\r", *ch)) *ch = ' ';
|
|
|
|
return 1;
|
|
}
|
|
|
|
int tcc_output_dbgme(const char *filename, me_info* me)
|
|
// by Siemargl. Writes filename.dbg file for source code level debuggin with MTDBG
|
|
// return 1 on error
|
|
{
|
|
FILE *fdbg;
|
|
char fname[400],
|
|
buf[80]; // no more fits in mtdbg string
|
|
|
|
strcpy(fname, filename);
|
|
strcat(fname, ".dbg");
|
|
fdbg = fopen(fname,"wt");
|
|
if (!fdbg) return 1;
|
|
|
|
meos_section_info *si, *ss;
|
|
fputs(".text\n", fdbg); // just for mtbg
|
|
|
|
// print symbol table with resolved addresses
|
|
Elf32_Sym* esym;
|
|
for (esym = (Elf32_Sym*)symtab_section->data; esym <= (Elf32_Sym*)(symtab_section->data + symtab_section->data_offset); esym++)
|
|
{
|
|
if (esym->st_info == 0 || esym->st_info == 4) continue;
|
|
int sect = esym->st_shndx;
|
|
ss = findsection(me, sect);
|
|
const char *sym_name = strtab_section->data + esym->st_name;
|
|
if (ss == 0)
|
|
{
|
|
fprintf(fdbg, "undefined symbol '%s' type(%d)\n", sym_name, esym->st_info);
|
|
continue;
|
|
}
|
|
fprintf(fdbg, "0x%X %s\n", ss->sh_addr + esym->st_value, sym_name); // removed type(%d) esym->st_info
|
|
}
|
|
|
|
fputs(".text source code links\n", fdbg); // just for mtbg
|
|
// print symbol table with resolved addresses
|
|
Stab_Sym *stab;
|
|
char *str = "", *cur_file = "???", cur_fun[255];
|
|
int cur_line = 0, cur_fun_addr = 0, fun_flag = 0;
|
|
strcpy(cur_fun, "fn???");
|
|
for (stab = (Stab_Sym*)stab_section->data; stab <= (Stab_Sym*)(stab_section->data + stab_section->data_offset); stab++)
|
|
{
|
|
str = "";
|
|
switch(stab->n_type)
|
|
{
|
|
case 100: // source file, or path
|
|
if (stab->n_strx)
|
|
{
|
|
cur_file = stabstr_section->data + stab->n_strx;
|
|
load_source_file(cur_file);
|
|
}
|
|
else
|
|
cur_file = "???";
|
|
strcpy(cur_fun, "fn???");
|
|
cur_line = 0;
|
|
cur_fun_addr = 0;
|
|
fun_flag = 0;
|
|
break;
|
|
case 36: // func
|
|
cur_fun_addr = 0;
|
|
if (stab->n_strx)
|
|
{
|
|
strcpy(cur_fun, stabstr_section->data + stab->n_strx);
|
|
str = strchr(cur_fun, ':');
|
|
if (str) *str = '\0';
|
|
tcc_find_symbol_me(me, cur_fun, &cur_fun_addr);
|
|
cur_line = stab->n_desc;
|
|
fun_flag = 1;
|
|
//fprintf(fdbg, "0x%X %s() line(%d)\n", cur_fun_addr, cur_fun, cur_line); // commented as conflicted with direct address
|
|
}
|
|
else
|
|
strcpy(cur_fun, "fn???");
|
|
break;
|
|
case 68: // N_SLINE
|
|
if (stab->n_value == 0 ) continue; // skip zero offset line
|
|
if (fun_flag) // skip string {, as duplicates address
|
|
{
|
|
fun_flag = 0;
|
|
continue;
|
|
}
|
|
|
|
int line;
|
|
if (stab->n_desc > cur_line)
|
|
line = cur_line + 1;
|
|
else
|
|
line = cur_line;
|
|
//fprintf(fdbg, "0x%X LINES %d-%d \n", cur_fun_addr + stab->n_value, line, stab->n_desc);
|
|
if (get_src_lines(buf, sizeof buf, line, stab->n_desc))
|
|
fprintf(fdbg, "0x%X %s\n", cur_fun_addr + stab->n_value, buf);
|
|
|
|
cur_line = stab->n_desc;
|
|
break;
|
|
default:
|
|
continue;
|
|
}
|
|
/*
|
|
if (stab->n_strx)
|
|
str = stabstr_section->data + stab->n_strx;
|
|
fprintf(fdbg, "0x%X type(%d) str(%s) desc(%d)\n", stab->n_value, stab->n_type, str, stab->n_desc);
|
|
*/
|
|
}
|
|
|
|
/* for(; si; si = si->next)
|
|
{
|
|
Section *sr;
|
|
Elf32_Rel *rel, *rel_end;
|
|
for(sr = me->s1->sections[si->sec_num]->reloc; sr; )
|
|
{
|
|
for (rel = (Elf32_Rel *) sr->data, rel_end = (Elf32_Rel *) (sr->data + sr->data_offset); rel < rel_end; rel++)
|
|
{
|
|
int type = ELF32_R_TYPE(rel->r_info);
|
|
if (type != R_386_PC32 && type != R_386_32)
|
|
continue;
|
|
int sym = ELF32_R_SYM(rel->r_info);
|
|
if (sym > symtab_section->data_offset / sizeof(Elf32_Sym))
|
|
continue;
|
|
Elf32_Sym* esym = ((Elf32_Sym*)symtab_section->data) + sym;
|
|
int sect = esym->st_shndx;
|
|
ss = findsection(me, sect);
|
|
const char *sym_name = strtab_section->data + esym->st_name;
|
|
if (ss == 0)
|
|
{
|
|
fprintf(fdbg, "undefined symbol '%s'\n", sym_name);
|
|
continue;
|
|
}
|
|
if (rel->r_offset > si->data_size) continue;
|
|
fprintf(fdbg, "\t0x%X %s\n", ss->sh_addr + esym->st_value, sym_name);
|
|
}
|
|
}
|
|
}
|
|
*/
|
|
close_source_file();
|
|
fclose(fdbg);
|
|
return 0;
|
|
}
|