/*
 * Copyright 1995, Silicon Graphics, Inc.
 * ALL RIGHTS RESERVED
 *
 * UNPUBLISHED -- Rights reserved under the copyright laws of the United
 * States.   Use of a copyright notice is precautionary only and does not
 * imply publication or disclosure.
 *
 * U.S. GOVERNMENT RESTRICTED RIGHTS LEGEND:
 * Use, duplication or disclosure by the Government is subject to restrictions
 * as set forth in FAR 52.227.19(c)(2) or subparagraph (c)(1)(ii) of the Rights
 * in Technical Data and Computer Software clause at DFARS 252.227-7013 and/or
 * in similar or successor clauses in the FAR, or the DOD or NASA FAR
 * Supplement.  Contractor/manufacturer is Silicon Graphics, Inc.,
 * 2011 N. Shoreline Blvd. Mountain View, CA 94039-7311.
 *
 * THE CONTENT OF THIS WORK CONTAINS CONFIDENTIAL AND PROPRIETARY
 * INFORMATION OF SILICON GRAPHICS, INC. ANY DUPLICATION, MODIFICATION,
 * DISTRIBUTION, OR DISCLOSURE IN ANY FORM, IN WHOLE, OR IN PART, IS STRICTLY
 * PROHIBITED WITHOUT THE PRIOR EXPRESS WRITTEN PERMISSION OF SILICON
 * GRAPHICS, INC.
 *
 */

/*
 * File:	buildtask.c
 * Creator:	hsa@sgi.com
 * Create Date:	Fri Dec 23 13:32:56 PST 1994
 *
 * This program is a simple linker for RSP tasks. It concatenates
 * objects compiled by rspasm, as well as their data segments, and
 * builds an object table in DMEM (actual usage of this information
 * is completely up to the ucode).
 *
 * The table in DMEM looks like this:
 *
 *			offset size dest
 * number of bits:	  32    16   16
 *
 * So each object entry is 2 words.
 *
 * The offset is 32 bits because it is convenient for the ucode
 * to compute the proper DRAM address once and write it back.
 *
 * (the ucode is responsible for filling in the dest)
 *
 * Dest is 16 bits because IMEM is only 4K.
 *
 * By convention, the first object listed should be the 'main'
 * module. 
 */


#include <stdio.h>

#define PROG_NAME		"buildtask"
#define BUILDTASK_VERSION	"1.0"

#define usage()	fprintf(stderr,"usage : %s [-d off] [-o outfile] [-s skiplen] [-l len] object0 [-s skiplen] [-l len] object1 ...\n",PROG_NAME);

char	*obj_names[256];
int	obj_skip[256];
int	obj_len[256];
char	*obj_prefile[256];
char	output_name[256];
char	tmp_name[256];
char	line[256];
int	dmem_files = -1;

/* .dat array: */
unsigned long int	dmem[1024];
int	data_ptr = 0, obj_ptr = 0;

typedef struct
{
    int		offset;
    int		size;
} table_td;

table_td	obj_table[256];

int
main(int argc, char *argv[])
{
    FILE	*in, *out, *sym, *dbg;
    int		i, j, obj_count = 0, dmem_offset = 0, obj_size,argsdone=0;
    unsigned long int	val, *ptr;
    char	c;

    strcpy(output_name, "a.out");

    fprintf(stderr,"%s : RSP Linker, Version %s, created %s %s\n",
	    PROG_NAME, BUILDTASK_VERSION, __DATE__, __TIME__);

    while ((argc > 1) && (argv[1][0] == '-')) {
	switch(argv[1][1]) {
	    
	  case 's':
	  case 'l':
	  case 'p':
	    argsdone=1;
	    break;

	  case 'd':
	    dmem_offset = strtol(argv[2], NULL, 0);
	    argc--;	
	    argv++;
	    break;
	    
	  case 'o':
	    strcpy(output_name, argv[2]);
	    argc--;	
	    argv++;
	    break;
	    
	  case 'f':
	    dmem_files = strtol(argv[2], NULL, 0);
	    argc--;	
	    argv++;
	    break;
	    
	  default:
	    fprintf(stderr,"%s : option [%s] not recognized.\n", 
		    PROG_NAME, argv[1]);
	    usage();
	    break;

	}
	if (argsdone) break;
	argc--;
	argv++;
    }

    if (argc < 2) {
	fprintf(stderr,"%s : no objects specified.\n",PROG_NAME);
	usage();
	exit(-1);
    }

    obj_skip[obj_count]=0;
    obj_len[obj_count]=-1;
    obj_prefile[obj_count]=NULL;
    for (i= 1; i<argc; i++) {
	if (strcmp(argv[i], "-s")==0) {
	    if (i+2>=argc) {
		fprintf(stderr,"%s : -s option must precede object name",
			PROG_NAME);
		usage();
		exit(-1);
	    }
	    i++;
	    obj_skip[obj_count] = strtol(argv[i], NULL, 0);
	    continue;
	}

	if (strcmp(argv[i], "-l")==0) {
	    if (i+2>=argc) {
		fprintf(stderr,"%s : -l option must precede object name",
			PROG_NAME);
		usage();
		exit(-1);
	    }
	    i++;
	    obj_len[obj_count] = strtol(argv[i], NULL, 0);
	    continue;
	}

	if (strcmp(argv[i], "-p")==0) {
	    if (i+2>=argc) {
		fprintf(stderr,"%s : -l option must precede object name",
			PROG_NAME);
		usage();
		exit(-1);
	    }
	    i++;
	    obj_prefile[obj_count] = argv[i];
	    continue;
	}

	if (strcmp(output_name, argv[i])==0) {
	    fprintf(stderr, 
		    "%s : output file [%s] same as input file [%s].\n",
		    "buildtask", output_name, argv[i]);
	    usage();
	    exit(-1);
	}

	obj_names[obj_count] = (char *) malloc(strlen(argv[i]) + 2);
	if (obj_names[obj_count] == NULL) {
	    fprintf(stderr,"%s : can't alloc memory.\n",PROG_NAME);
	    exit(-1);
	}
	strcpy(obj_names[obj_count++], argv[i]);
	obj_skip[obj_count]=0;
	obj_len[obj_count]=-1;

    }

    if (obj_count<1) {
	fprintf(stderr,"%s : no objects specified(2).\n",PROG_NAME);
	usage();
	exit(-1);
    }
    if (obj_skip[obj_count]!=0 || obj_len[obj_count]!=-1 || 
	obj_prefile[obj_count]!=NULL) {
	fprintf(stderr,"%s : -s,-l,and -p must precede object name.\n",
		PROG_NAME);
	usage();
	exit(-1);
    }

    fprintf(stderr,"%s : building task [%s] from objects:\n",
	    PROG_NAME, output_name);
    for (i=0; i<obj_count; i++)
	fprintf(stderr,"\t...loading object [%s]\n",obj_names[i]);
    
    fprintf(stderr,"%s : building DMEM object table at offset %08x:\n",
	    PROG_NAME, dmem_offset);

    /*
     * open output files.
     */
    if ((out=fopen(output_name,"w")) == NULL) {
	fprintf(stderr,"%s : cannot open output file [%s]\n",
		PROG_NAME, output_name);
	usage();
	exit(-1);
    }

    sprintf(tmp_name, "%s.sym", output_name);
    if ((sym=fopen(tmp_name,"w")) == NULL) {
	fprintf(stderr,"%s : cannot open .sym file [%s]\n",
		PROG_NAME, tmp_name);
	usage();
	exit(-1);
    }

    sprintf(tmp_name, "%s.dbg", output_name);
    if ((dbg=fopen(tmp_name,"w")) == NULL) {
	fprintf(stderr,"%s : cannot open .dbg file [%s]\n",
		PROG_NAME, tmp_name);
	usage();
	exit(-1);
    }

    /*
     * for each object:
     */
    for (i=0; i<obj_count; i++) {

	/*
	 * update the offset to the object table:
	 */
	obj_table[i].offset = obj_ptr;

	/*
	 * copy each object to the output file, determining the length:
	 */
	obj_size = 0;

	if (obj_prefile[i]) {
            /*
             * open input object file.
             */
            if ((in=fopen(obj_prefile[i],"r")) == NULL) {
                fprintf(stderr,"%s : cannot open input file [%s]\n",
                        PROG_NAME, obj_prefile[i]);
                exit(-1);
            }
    
            fread(&val, sizeof(unsigned long int), 1, in);
            while (!feof(in)) {
                fwrite(&val, sizeof(unsigned long int), 1, out);
                obj_size += 4;
                obj_ptr += 4;
                fread(&val, sizeof(unsigned long int), 1, in);
            }
            fclose(in);
	}

	/*
	 * open input object file.
	 */
	if ((in=fopen(obj_names[i],"r")) == NULL) {
	    fprintf(stderr,"%s : cannot open input file [%s]\n",
		    PROG_NAME, obj_names[i]);
	    exit(-1);
	}

	fread(&val, sizeof(unsigned long int), 1, in);
	while (!feof(in)) {
	    if (obj_skip[i] <= 0 && obj_len[i]!=0) {
	    	fwrite(&val, sizeof(unsigned long int), 1, out);
	    	obj_size += 4;
	    	obj_ptr += 4;
		obj_len[i]--;
	    } else {
		obj_skip[i]--;
	    }
	    fread(&val, sizeof(unsigned long int), 1, in);
	}
	fclose(in);

	/* must align to 64-bits for DMA transfer */
	if (obj_ptr & 0x04) {
	    val = 0x0;
	    fwrite(&val, sizeof(unsigned long int), 1, out);
	    obj_ptr += 4;
	    obj_size += 4;
	}

	/*
	 * update the object size to the object table:
	 */
	obj_table[i].size = obj_size - 1;
	
	/*
	 * copy the data into DMEM file:
	 */
	sprintf(tmp_name, "%s.dat", obj_names[i]);
	if (dmem_files<0 || i<dmem_files) {
	  if ((in=fopen(tmp_name,"r")) == NULL) {
	    fprintf(stderr,"%s : cannot open input data file [%s]\n",
		    PROG_NAME, tmp_name);
	  } else {
	    fprintf(stderr,"\t...copying data [%s]\n",tmp_name);
	    /*
	     * allow for empty .dat files.
	     */
	    fread(&val, sizeof(unsigned long int), 1, in);
	    while (!feof(in)) {
		dmem[data_ptr] = val;
		data_ptr++;
		if (data_ptr > 1024) data_ptr=1024;
		fread(&val, sizeof(unsigned long int), 1, in);
	    }
	    fclose(in);
	  }
	}

	/*
	 * copy the symbols into .sym file:
	 */
	sprintf(tmp_name, "%s.sym", obj_names[i]);
	if ((in=fopen(tmp_name,"r")) == NULL) {
	    fprintf(stderr,"%s : cannot open input .sym file [%s]\n",
		    PROG_NAME, tmp_name);
	} else {
	    while (fgets(line, 80, in) != NULL) {
		fprintf(sym,"%s",line);
	    }
	    fclose(in);
	}

	/*
	 * copy the symbols into .dbg file:
	 */
	sprintf(tmp_name, "%s.dbg", obj_names[i]);
	if ((in=fopen(tmp_name,"r")) == NULL) {
	    fprintf(stderr,"%s : cannot open input .dbg file [%s]\n",
		    PROG_NAME, tmp_name);
	} else {
	    while (fgets(line, 80, in) != NULL) {
		fprintf(dbg,"%s",line);
	    }
	    fclose(in);
	}

    }

    fclose(out);
    fclose(sym);
    fclose(dbg);

    /*
     * update offset tables in DMEM.
     *
     * The table in DMEM looks like this:
     *
     *			offset size dest
     * number of bits:	  32    16   16
     *
     * So each object is 2 words.
     *
     * (the ucode is responsible for filling in the dest)
     *
     */
    for (i=0; i<obj_count; i++) {
	ptr = (unsigned long int *) &(dmem[(dmem_offset >> 2) + (i*2)]);
	*ptr = htonl(obj_table[i].offset);
	ptr++;
#if 0
	*ptr &= 0x0000ffff;
	*ptr |= (obj_table[i].size << 16);
#else /*XXXblythe*/
	*(unsigned short*)ptr = htons(obj_table[i].size);
#endif
    }

    /*
     * write out DMEM file:
     */

    /*
     * open output file.
     */
    sprintf(tmp_name, "%s.dat", output_name);
    if ((out=fopen(tmp_name,"w")) == NULL) {
	fprintf(stderr,"%s : cannot open data output file [%s]\n",
		PROG_NAME, tmp_name);
	exit(-1);
    }

    fwrite(&(dmem[0]), sizeof(unsigned long int), data_ptr, out);
    fclose(out);
}