/* load-ld.c
 *
 * $Id$
 *
 * Copyright 1990, 1991, 1992, 1993, 1994, 1995, Oliver Laumann, Berlin
 * Copyright 2002, 2003 Sam Hocevar <sam@zoy.org>, Paris
 *
 * This software was derived from Elk 1.2, which was Copyright 1987, 1988,
 * 1989, Nixdorf Computer AG and TELES GmbH, Berlin (Elk 1.2 has been written
 * by Oliver Laumann for TELES Telematic Services, Berlin, in a joint project
 * between TELES and Nixdorf Microprocessor Engineering, Berlin).
 *
 * Oliver Laumann, TELES GmbH, Nixdorf Computer AG and Sam Hocevar, as co-
 * owners or individual owners of copyright in this software, grant to any
 * person or company a worldwide, royalty free, license to
 *
 *    i) copy this software,
 *   ii) prepare derivative works based on this software,
 *  iii) distribute copies of this software or derivative works,
 *   iv) perform this software, or
 *    v) display this software,
 *
 * provided that this notice is not removed and that neither Oliver Laumann
 * nor Teles nor Nixdorf are deemed to have made any representations as to
 * the suitability of this software for any purpose nor are held responsible
 * for any defects of this software.
 *
 * THERE IS ABSOLUTELY NO WARRANTY FOR THIS SOFTWARE.
 */

#include AOUT_H
#include <errno.h>
#include <fcntl.h>
#include <sys/types.h>

#ifndef O_BINARY
#  define O_BINARY 0
#endif

#ifdef ECOFF
#  ifdef CACHECTL_H
#    include CACHECTL_H
#  endif

struct headers {
    struct filehdr fhdr;
    struct aouthdr aout;
    struct scnhdr section[3];
};
#endif

extern void Free_Symbols (SYMTAB *);
extern void Call_Initializers (SYMTAB *, char *, int);

extern void *sbrk();
extern char *getenv();

static char *Loader_Output;
static char *tmpdir;

void Load_Object (Object names) {
#ifdef ECOFF
    struct headers hdr;
#else
    struct exec hdr;
#endif
    register char *brk, *obrk, *lp, *li;
    char *buf;
    register int n, f, len, liblen;
    Object port, tail, fullnames, libs;
    FILE *fp;
    GC_Node3;
    Alloca_Begin;

    li = Loader_Input;
    if (!li)
        li = A_Out_Name;
    if (!Loader_Output) {
        if (!(tmpdir = getenv ("TMPDIR")))
            tmpdir = "/tmp";
        Loader_Output = Safe_Malloc (strlen (tmpdir) + 20);
    }
    sprintf (Loader_Output, "%s/ldXXXXXX", tmpdir);
    (void)mktemp (Loader_Output);

    port = tail = fullnames = Null;
    GC_Link3 (port, tail, fullnames);
    for (len = 0, tail = names; !Nullp (tail); tail = Cdr (tail)) {
        port = General_Open_File (Car (tail), P_INPUT, Var_Get (V_Load_Path));
        fullnames = Cons (PORT(port)->name, fullnames);
        len += STRING(Car (fullnames))->size + 1;
        (void)P_Close_Input_Port (port);
    }
    GC_Unlink;

    libs = Var_Get (V_Load_Libraries);
    if (TYPE(libs) == T_String) {
        liblen = STRING(libs)->size;
        lp = STRING(libs)->data;
    } else {
        liblen = 3; lp = "-lc";
    }

    Alloca (buf, char*, strlen (A_Out_Name) + len + liblen + 100);

    obrk = brk = (char *)sbrk (0);
    brk = (char *)((int)brk + 7 & ~7);

#if defined(hp9000s300) || defined(__hp9000s300) || defined(__hp9000s300__)
    sprintf (buf, "%s -N %s -A %s -R %x -o %s ",
#else
    sprintf (buf, "%s -N %s -A %s -T %x -o %s ",
#endif
        LD_NAME, INC_LDFLAGS, li, (unsigned int)brk, Loader_Output);

    for (tail = fullnames; !Nullp (tail); tail = Cdr (tail)) {
        register struct S_String *str = STRING(Car (tail));
        strncat (buf, str->data, str->size);
        strcat (buf, " ");
    }
    strncat (buf, lp, liblen);

    if (Verb_Load)
        printf ("[%s]\n", buf);
    if (system (buf) != 0) {
        (void)unlink (Loader_Output);
        Primitive_Error ("system linker failed");
    }
    Disable_Interrupts;               /* To ensure that f gets closed */
    if ((f = open (Loader_Output, O_RDONLY|O_BINARY)) == -1) {
        (void)unlink (Loader_Output);
        Primitive_Error ("cannot open tempfile");
    }
    if (Loader_Input)
        (void)unlink (Loader_Input);
    else
        Loader_Input = Safe_Malloc (strlen (tmpdir) + 20);
    strcpy (Loader_Input, Loader_Output);
    if (read (f, (char *)&hdr, sizeof (hdr)) != sizeof (hdr)) {
err:
        close (f);
        Primitive_Error ("corrupt tempfile (`ld' is broken)");
    }
#ifdef ECOFF
    n = hdr.aout.tsize + hdr.aout.dsize + hdr.aout.bsize;
#else
    n = hdr.a_text + hdr.a_data + hdr.a_bss;
#endif
    if ((char *)sbrk (n + brk-obrk) == (char *)-1) {
        close (f);
        Primitive_Error ("not enough memory to load object file");
    }
    memset (brk, 0, n);
#ifdef ECOFF
    n -= hdr.aout.bsize;
    (void)lseek (f, (off_t)hdr.section[0].s_scnptr, 0);
#else
    n -= hdr.a_bss;
#endif
    if (read (f, brk, n) != n)
        goto err;
    if ((fp = fdopen (f, O_BINARY ? "rb" : "r")) == NULL) {
        close (f);
        Primitive_Error ("cannot fdopen object file");
    }
    if (The_Symbols)
        Free_Symbols (The_Symbols);
    The_Symbols = Snarf_Symbols (fp, &hdr);
    (void)fclose (fp);
#if defined(ECOFF) && defined(CACHECTL_H)
    if (cacheflush (brk, n, BCACHE) == -1) {
        extern int errno;
        Saved_Errno = errno;
        Primitive_Error ("cacheflush failed: ~E");
    }
#endif
    Call_Initializers (The_Symbols, brk, PR_CONSTRUCTOR);
    Call_Initializers (The_Symbols, brk, PR_EXTENSION);
    Enable_Interrupts;
    Alloca_End;
}

void Finit_Load () {
    if (Loader_Input)
        (void)unlink (Loader_Input);
}

void Fork_Load () {
    char *newlink;

    if (Loader_Input) {
        Disable_Interrupts;
        newlink = Safe_Malloc (strlen (tmpdir) + 20);
        sprintf (newlink, "%s/ldXXXXXX", tmpdir);
        (void)mktemp (newlink);
        (void)link (Loader_Input, newlink);
        free (Loader_Input);
        Loader_Input = newlink;
        Enable_Interrupts;
    }
}