Shamusworld >> Repos - rln/blob - rln.c

   1 //
   2 // RLN - Reboot's Linker for the Atari Jaguar console system
   3 // Copyright (C) 199x, Allan K. Pratt, 2014-2015 Reboot & Friends
   4 //
   5
   6 #include "rln.h"
   7 //#include <assert.h>
   8
   9 unsigned errflag = 0;                           // Error flag, goes TRUE on error
  10 unsigned waitflag = 0;                          // Wait for any keypress flag
  11 unsigned versflag = 0;                          // Version banner has been shown flag
  12 unsigned aflag = 0;                                     // Absolute linking flag
  13 unsigned bflag = 0;                                     // Don't remove mulitply def locals flag
  14 unsigned cflag = 0;                                     // COF executable
  15 unsigned dflag = 0;                                     // Wait for key after link flag
  16 unsigned gflag = 0;                                     // Source level debug include flag
  17 unsigned lflag = 0;                                     // Add local symbols to output flag
  18 unsigned mflag = 0;                                     // Produce symbol load map flag
  19 unsigned oflag = 0;                                     // Output filename specified
  20 unsigned rflag = 0;                                     // Segment alignment size flag
  21 unsigned sflag = 0;                                     // Output only global symbols
  22 unsigned vflag = 0;                                     // Verbose flag
  23 unsigned wflag = 0;                                     // Show warnings flag
  24 unsigned zflag = 0;                                     // Suppress banner flag
  25 unsigned pflag = 0, uflag = 0;          // Unimplemented flags
  26 unsigned hd = 0;                                        // Index of next file handle to fill
  27 unsigned secalign = 7;                          // Section Alignment (8=phrase)
  28 unsigned tbase = 0;                                     // TEXT base address
  29 unsigned dbase = 0;                                     // DATA base address
  30 unsigned bbase = 0;                                     // BSS base address
  31 unsigned textoffset = 0;                        // COF TEXT segment offset
  32 unsigned dataoffset = 0;                        // COF DATA segment offset
  33 unsigned bssoffset = 0;                         // COF BSS segment offset
  34 unsigned displaybanner = 1;                     // Display version banner
  35 unsigned symoffset = 0;                         // Symbol table offset in output file
  36 unsigned dbgsymbase = 0;                        // Debug symbol base address
  37 int noheaderflag = 0;                           // No header flag for ABS files
  38 int hflags;                                                     // Value of the arg to -h option
  39 int ttype, dtype, btype;                        // Type flag: 0, -1, -2, -3, -4
  40 int tval, dval, bval;                           // Values of these abs bases
  41 int hflag[NHANDLES];                            // True for include files
  42 int handle[NHANDLES];                           // Open file handles
  43 int textsize, datasize, bsssize;        // Cumulative segment sizes
  44 char libdir[FARGSIZE * 3];                      // Library directory to search
  45 char ofile[FARGSIZE];                           // Output file name (.o)
  46 char * name[NHANDLES];                          // Associated file names
  47 char * cmdlnexec = NULL;                        // Executable name - pointer to ARGV[0]
  48 char * hsym1[SYMLEN];                           // First symbol for include files
  49 char * hsym2[SYMLEN];                           // Second symbol for include files
  50 struct OFILE * plist = NULL;            // Object image list pointer
  51 struct OFILE * plast;                           // Last object image list pointer
  52 struct OFILE * olist = NULL;            // Pointer to first object file in list
  53 struct OFILE * olast;                           // Pointer to last object file in list
  54 char * arPtr[512];
  55 uint32_t arIndex = 0;
  56 struct HREC * htable[NBUCKETS];         // Hash table
  57 struct HREC * unresolved = NULL;        // Pointer to unresolved hash list
  58 char * ost;                                                     // Output symbol table
  59 char * ost_ptr;                                         // Output symbol table; current pointer
  60 char * ost_end;                                         // Output symbol table; end pointer
  61 char * oststr;                                          // Output string table
  62 char * oststr_ptr;                                      // Output string table; current pointer
  63 char * oststr_end;                                      // Output string table; end pointer
  64 int ost_index = 0;                                      // Index of next ost addition
  65 uint8_t nullStr[1] = "\x00";            // Empty string
  66 struct HREC * arSymbol = NULL;          // Pointer to AR symbol table
  67
  68
  69 // Function prototypes
  70 struct HREC * LookupHREC(char *);
  71 char * PathTail(char *);
  72 void ShowHelp(void);
  73 void ShowVersion(void);
  74
  75
  76 //
  77 // Get a long word from memory
  78 //
  79 static inline uint32_t GetLong(uint8_t * src)
  80 {
  81         return (src[0] << 24) | (src[1] << 16) | (src[2] << 8) | src[3];
  82 }
  83
  84
  85 //
  86 // Put a long word into memory
  87 //
  88 static inline void PutLong(uint8_t * dest, uint32_t val)
  89 {
  90         *dest++ = (uint8_t)(val >> 24);
  91         *dest++ = (uint8_t)(val >> 16);
  92         *dest++ = (uint8_t)(val >> 8);
  93         *dest = (uint8_t)val;
  94 }
  95
  96
  97 //
  98 // Get a word from memory
  99 //
 100 static inline uint16_t GetWord(uint8_t * src)
 101 {
 102         return (src[0] << 8) | src[1];
 103 }
 104
 105
 106 //
 107 // Put a word into memory
 108 //
 109 static inline void PutWord(uint8_t * dest, uint16_t val)
 110 {
 111         *dest++ = (uint8_t)(val >> 8);
 112         *dest = (uint8_t)val;
 113 }
 114
 115
 116 //
 117 // Find passed in file's length in bytes
 118 // N.B.: This also resets the file's pointer to the start of the file
 119 //
 120 long FileSize(int fd)
 121 {
 122         long size = lseek(fd, 0, SEEK_END);
 123         lseek(fd, 0, SEEK_SET);
 124
 125         return size;
 126 }
 127
 128
 129 //
 130 // For this object file, add symbols to the output symbol table after
 131 // relocating them. Returns TRUE if OSTLookup returns an error (-1).
 132 //
 133 int DoSymbols(struct OFILE * ofile)
 134 {
 135         int type;
 136         long value;
 137         int index;
 138         int j;
 139         struct HREC * hptr;
 140         uint32_t tsoSave, dsoSave, bsoSave;
 141
 142         // Point to first symbol record in the object file
 143         char * symptr = (ofile->o_image + 32
 144                 + ofile->o_header.tsize
 145                 + ofile->o_header.dsize
 146                 + ofile->o_header.absrel.reloc.tsize
 147                 + ofile->o_header.absrel.reloc.dsize);
 148
 149         // Point to end of symbol record in the object file
 150         char * symend = symptr + ofile->o_header.ssize;
 151
 152         uint32_t tsegoffset = ofile->segBase[TEXT];
 153         uint32_t dsegoffset = ofile->segBase[DATA];
 154         uint32_t bsegoffset = ofile->segBase[BSS];
 155
 156         // Save segment vars, so we can restore them if needed
 157         tsoSave = tsegoffset, dsoSave = dsegoffset, bsoSave = bsegoffset;
 158
 159         // Process each record in the object's symbol table
 160         for(; symptr!=symend; symptr+=12)
 161         {
 162                 index = GetLong(symptr + 0);    // Obtain symbol string index
 163                 type  = GetLong(symptr + 4);    // Obtain symbol type
 164                 value = GetLong(symptr + 8);    // Obtain symbol value
 165
 166                 // Global/External symbols have a pre-processing stage
 167                 // N.B.: This destroys the t/d/bsegoffset discovered above. So if a
 168                 //       local symbol follows a global/exported one, it gets wrong
 169                 //       info! [Should be FIXED now]
 170                 if (type & 0x01000000)
 171                 {
 172                         // Obtain the string table index for the relocation symbol, look
 173                         // for it in the globals hash table to obtain information on that
 174                         // symbol.
 175                         hptr = LookupHREC(symend + index);
 176
 177                         if (hptr == NULL)
 178                         {
 179                                 // Try to find it in the OST
 180                                 int ostIndex = OSTLookup(symend + index);
 181
 182                                 if (ostIndex == -1)
 183                                 {
 184                                         printf("DoSymbols(): Symbol not found in hash table: '%s' (%s)\n", symend + index, ofile->o_name);
 185                                         return 1;
 186                                 }
 187
 188                                 if (vflag > 1)
 189                                         printf("DoSymbols(): Skipping symbol '%s' (%s) found in OST...\n", symend + index, ofile->o_name);
 190
 191                                 // If the symbol is not in any .a or .o units, it must be one
 192                                 // of the injected ones (_TEXT_E, _DATA_E, or _BSS_E), so skip
 193                                 // it [or maybe not? In verbose mode, we see nothing...!]
 194                                 continue;
 195                         }
 196
 197                         tsegoffset = hptr->h_ofile->segBase[TEXT];
 198                         dsegoffset = hptr->h_ofile->segBase[DATA];
 199                         bsegoffset = hptr->h_ofile->segBase[BSS];
 200
 201                         // Update type with global type
 202                         type = hptr->h_type;
 203
 204                         // Remove global flag if absolute
 205                         if (type == (T_GLBL | T_ABS))
 206                                 type = T_ABS;
 207
 208                         // If the global/external has a value then update that value in
 209                         // accordance with the segment sizes of the object file it
 210                         // originates from
 211                         if (hptr->h_value)
 212                         {
 213                                 switch (hptr->h_type & 0x0E000000)
 214                                 {
 215                                 case T_ABS:
 216                                 case T_TEXT:
 217                                         value = hptr->h_value;
 218                                         break;
 219                                 case T_DATA:
 220                                         value = hptr->h_value - hptr->h_ofile->o_header.tsize;
 221                                         break;
 222                                 case T_BSS:
 223                                         value = hptr->h_value
 224                                                 - (hptr->h_ofile->o_header.tsize
 225                                                 + hptr->h_ofile->o_header.dsize);
 226                                         break;
 227                                 default:
 228                                         if (vflag > 1)
 229                                                 printf("DoSymbols: No adjustment made for symbol: %s (%s) = %X\n", symend + index, ofile->o_name, hptr->h_value);
 230                                 }
 231                         }
 232                 }
 233                 // If *not* a global/external, use the info from passed in object
 234                 else
 235                         tsegoffset = tsoSave, dsegoffset = dsoSave, bsegoffset = bsoSave;
 236
 237                 // Process and update the value dependent on whether the symbol is a
 238                 // debug symbol or not
 239                 // N.B.: Debug symbols are currently not supported
 240                 if (type & 0xF0000000)
 241                 {
 242                         // DEBUG SYMBOL
 243                         // Set the correct debug symbol base address (TEXT segment)
 244 #if 0
 245                         dbgsymbase = 0;
 246
 247                         for(j=0; (unsigned)j<dosymi; j++)
 248                                 dbgsymbase += obj_segsize[j][0];
 249 #else
 250                         dbgsymbase = ofile->segBase[TEXT];
 251 #endif
 252
 253                         switch (type & 0xFF000000)
 254                         {
 255                         case 0x64000000:
 256                                 value = tval + dbgsymbase;
 257                                 break;
 258                         case 0x44000000:
 259                         case 0x46000000:
 260                         case 0x48000000:
 261                                 value = tval + dbgsymbase + value;
 262                         default:
 263                                 break;
 264                         }
 265
 266                         PutLong(symptr + 8, value);
 267                 }
 268                 else
 269                 {
 270                         // NON-DEBUG SYMBOL
 271                         // Now make modifications to the symbol value, local or global,
 272                         // based on the segment sizes of the object file currently being
 273                         // processed.
 274                         switch (type & T_SEG)
 275                         {
 276                         case T_ABS:
 277                                 break;
 278                         case T_TEXT:
 279                                 value = tbase + tsegoffset + value;
 280                                 PutLong(symptr + 8, value);
 281                                 break;
 282                         case T_DATA:
 283                                 if (type & T_GLBL)
 284                                         value = dbase + dsegoffset + value;
 285                                 else
 286                                         value = dbase + dsegoffset + (value
 287                                                 - ofile->o_header.tsize);
 288
 289                                 PutLong(symptr + 8, value);
 290                                 break;
 291                         case T_BSS:
 292                                 if (type & T_GLBL)
 293                                         value = bbase + bsegoffset + value;
 294                                 else
 295                                         value = bbase + bsegoffset
 296                                                 + (value - (ofile->o_header.tsize
 297                                                 + ofile->o_header.dsize));
 298
 299                                 PutLong(symptr + 8, value);
 300                                 break;
 301                         default:
 302                                 break;
 303                         }
 304                 }
 305
 306                 // Add to output symbol table if global/extern, or local flag is set
 307                 if (isglobal(type) || lflag)
 308                 {
 309                         if (vflag > 1)
 310                                 printf("DoSymbols: Adding symbol: %s (%s) to OST...\n", symend + index, ofile->o_name);
 311
 312                         index = OSTAdd(symend + index, type, value);
 313
 314                         if (index == -1)
 315                         {
 316                                 printf("DoSymbols(): Failed to add symbol '%s' to OST!\n", symend + index);
 317                                 return 1;
 318                         }
 319                 }
 320         }
 321
 322         return 0;
 323 }
 324
 325
 326 //
 327 // Free up hash memory
 328 //
 329 void FreeHashes(void)
 330 {
 331         int i;
 332
 333         for(i=0; i<NBUCKETS; i++)
 334         {
 335                 struct HREC * hptr = htable[i];
 336
 337                 while (hptr)
 338                 {
 339                         struct HREC * htemp = hptr->h_next;
 340                         free(hptr);
 341                         hptr = htemp;
 342                 }
 343         }
 344 }
 345
 346
 347 //
 348 // Add all global and external symbols to the output symbol table
 349 // [This is confusing--is it adding globals or locals? common == local!
 350 //  but then again, we see this in the header:
 351 //  #define T_COMMON  (T_GLOBAL | T_EXTERN) but that could be just bullshit.]
 352 //
 353 // Common symbols have a different number in the "value" field of the symbol
 354 // table (!0) than purely external symbols do (0). So you have to look at the
 355 // type (T_GLBL) *and* the value to determine if it's a common symbol.
 356 //
 357 long DoCommon(void)
 358 {
 359         struct HREC * hptr;
 360         int i;
 361
 362         for(i=0; i<NBUCKETS; i++)
 363         {
 364                 for(hptr=htable[i]; hptr!=NULL; hptr=hptr->h_next)
 365                 {
 366 //NO!                   if (iscommon(hptr->h_type))
 367                         if (isglobal(hptr->h_type))// || isextern(hptr->h_type))
 368                         {
 369 // Skip if in *.a file... (does nothing)
 370 //if (hptr->h_ofile->isArchiveFile)
 371 //      continue;
 372
 373 //Is this true? Couldn't an absolute be exported???
 374                                 if (hptr->h_type == (T_GLBL | T_ABS))
 375                                         hptr->h_type = T_ABS;   // Absolutes *can't* be externals
 376
 377                                 if (OSTAdd(hptr->h_sym, hptr->h_type, hptr->h_value) == -1)
 378                                         return -1;
 379                         }
 380                 }
 381         }
 382
 383         return 0;
 384 }
 385
 386
 387 //
 388 // Add a symbol's name, type, and value to the OST.
 389 // Returns the index of the symbol in OST, or -1 for error.
 390 //
 391 int OSTAdd(char * name, int type, long value)
 392 {
 393         int ost_offset_p, ost_offset_e = 0;     // OST table offsets for position calcs
 394         int ostresult;                                          // OST index result
 395         int slen = strlen(name);
 396
 397         // If the OST or OST string table has not been initialised then do so
 398         if (ost_index == 0)
 399         {
 400                 ost = malloc(OST_BLOCK);
 401                 oststr = malloc(OST_BLOCK);
 402
 403                 if (ost == NULL)
 404                 {
 405                         printf("OST memory allocation error.\n");
 406                         return -1;
 407                 }
 408
 409                 if (oststr == NULL)
 410                 {
 411                         printf("OSTSTR memory allocation error.\n");
 412                         return -1;
 413                 }
 414
 415                 ost_ptr = ost;                                          // Set OST start pointer
 416                 ost_end = ost + OST_BLOCK;                      // Set OST end pointer
 417
 418                 PutLong(oststr, 0x00000004);            // Just null long for now
 419                 oststr_ptr = oststr + 4;                        // Skip size of str table long (incl null long)
 420                 PutLong(oststr_ptr, 0x00000000);        // Null terminating long
 421                 oststr_end = oststr + OST_BLOCK;
 422         }
 423         else
 424         {
 425                 // If next symbol record exceeds current allocation then expand symbol
 426                 // table and/or symbol string table.
 427                 ost_offset_p = (ost_ptr - ost);
 428                 ost_offset_e = (ost_end - ost);
 429
 430                 // 3 x uint32_t (12 bytes)
 431                 if ((ost_ptr + 12) > ost_end)
 432                 {
 433                         // We want to allocate the current size of the OST + another block.
 434                         ost = realloc(ost, ost_offset_e + OST_BLOCK);
 435
 436                         if (ost == NULL)
 437                         {
 438                                 printf("OST memory reallocation error.\n");
 439                                 return -1;
 440                         }
 441
 442                         ost_ptr = ost + ost_offset_p;
 443                         ost_end = (ost + ost_offset_e) + OST_BLOCK;
 444                 }
 445
 446                 ost_offset_p = (oststr_ptr - oststr);
 447                 ost_offset_e = (oststr_end - oststr);
 448
 449                 // string length + terminating NULL + uint32_t (terminal long)
 450                 if ((oststr_ptr + (slen + 1 + 4)) > oststr_end)
 451                 {
 452                         oststr = realloc(oststr, ost_offset_e + OST_BLOCK);
 453
 454                         if (oststr == NULL)
 455                         {
 456                                 printf("OSTSTR memory reallocation error.\n");
 457                                 return -1;
 458                         }
 459
 460                         oststr_ptr = oststr + ost_offset_p;
 461                         oststr_end = (oststr + ost_offset_e) + OST_BLOCK;
 462                 }
 463         }
 464
 465         // If this is a debug symbol and the include debug symbol flag (-g) is not
 466         // set then do nothing
 467         if ((type & 0xF0000000) && !gflag)
 468         {
 469                 // Do nothing
 470                 return 0;
 471         }
 472
 473         // Get symbol index in OST, if any (-1 if not found)
 474         ostresult = OSTLookup(name);
 475
 476         // If the symbol is in the output symbol table and the bflag is set
 477         // (don't remove multiply defined locals) and this is not an
 478         // external/global symbol *** OR *** the symbol is not in the output
 479         // symbol table then add it.
 480         if (((ostresult != -1) && bflag && !(type & 0x01000000))
 481                 || ((ostresult != -1) && gflag && (type & 0xF0000000))
 482                 || (ostresult == -1))
 483         {
 484                 if ((type & 0xF0000000) == 0x40000000)
 485                         PutLong(ost_ptr, 0x00000000);   // Zero string table offset for dbg line
 486                 else
 487                         PutLong(ost_ptr, (oststr_ptr - oststr));        // String table offset of symbol string
 488
 489                 PutLong(ost_ptr + 4, type);
 490                 PutLong(ost_ptr + 8, value);
 491                 ost_ptr += 12;
 492
 493                 // If the symbol type is anything but a debug line information
 494                 // symbol then write the symbol string to the string table
 495                 if ((type & 0xF0000000) != 0x40000000)
 496                 {
 497                         strcpy(oststr_ptr, name);               // Put symbol name in string table
 498                         *(oststr_ptr + slen) = '\0';    // Add null terminating character
 499                         oststr_ptr += (slen + 1);
 500                         PutLong(oststr_ptr, 0x00000000);        // Null terminating long
 501                         PutLong(oststr, (oststr_ptr - oststr)); // Update size of string table
 502                 }
 503
 504                 if (vflag > 1)
 505                         printf("OSTAdd: (%s), type=$%08X, val=$%08lX\n", name, type, value);
 506
 507 // is ost_index pointing one past?
 508 // does this return the same regardless of if its ++n or n++?
 509 // no. it returns the value of ost_index *before* it's incremented.
 510                 return ++ost_index;
 511         }
 512
 513         return ostresult;
 514 }
 515
 516
 517 //
 518 // Return the index of a symbol in the output symbol table
 519 // N.B.: This is a 1-based index! (though there's no real reason for it to be)
 520 //
 521 int OSTLookup(char * sym)
 522 {
 523         int i;
 524         int stro = 4;           // Offset in string table
 525
 526         for(i=0; i<ost_index; i++)
 527         {
 528                 if (strcmp(oststr + stro, sym) == 0)
 529                         return i + 1;
 530
 531                 stro += strlen(oststr + stro) + 1;
 532         }
 533
 534         return -1;
 535 }
 536
 537
 538 //
 539 // Add unresolved externs to the output symbol table
 540 // N.B.: Only adds unresolved symbols *if* they're not already in the OST
 541 //
 542 int DoUnresolved(void)
 543 {
 544         struct HREC * hptr = unresolved;
 545
 546         // Add to OST while unresolved list is valid
 547         while (hptr != NULL)
 548         {
 549                 if (OSTAdd(hptr->h_sym, T_GLBL, 0L) == -1)
 550                         return 1;
 551
 552                 if (vflag > 1)
 553                         printf("DoUnresolved(): '%s' (%s:$%08X) in OST\n", hptr->h_sym, hptr->h_ofile->o_name, hptr->h_type);
 554
 555                 struct HREC * htemp = hptr->h_next;
 556                 free(hptr);
 557                 hptr = htemp;
 558         }
 559
 560         unresolved = NULL;
 561         return 0;
 562 }
 563
 564
 565 //
 566 // Update object file TEXT and DATA segments based on relocation records. Take
 567 // in an OFILE header and flag (T_TEXT, T_DATA) to process. Return (0) is
 568 // successful or non-zero (1) if failed.
 569 //
 570 int RelocateSegment(struct OFILE * ofile, int flag)
 571 {
 572         char * symtab;                  // Start of symbol table
 573         char * symbols;                 // Start of symbols
 574         char * sptr;                    // Start of segment data
 575         char * rptr;                    // Start of segment relocation records
 576         unsigned symidx;                // Offset to symbol
 577         unsigned addr;                  // Relocation address
 578         unsigned rflg;                  // Relocation flags
 579         unsigned olddata;               // Old segment data at reloc address
 580         unsigned newdata = 0;   // New segment data at reloc address
 581         unsigned pad;                   // Temporary to calculate phrase padding
 582         int i;                                  // Iterator
 583         char sym[SYMLEN];               // String for symbol name/hash search
 584         int ssidx;                              // Segment size table index
 585         unsigned glblreloc;             // Global relocation flag
 586         unsigned absreloc;              // Absolute relocation flag
 587         unsigned relreloc;              // Relative relocation flag
 588         unsigned swcond;                // Switch statement condition
 589         unsigned relocsize;             // Relocation record size
 590
 591         // If there is no TEXT relocation data for the selected object file segment
 592         // then update the COF TEXT segment offset allowing for the phrase padding
 593         if ((flag == T_TEXT) && !ofile->o_header.absrel.reloc.tsize)
 594         {
 595                 // SCPCD : we should not increment the textoffset before the end of processing the object file, else data section will point to wrong textoffset
 596                 return 0;
 597         }
 598
 599         // If there is no DATA relocation data for the selected object file segment
 600         // then update the COF DATA and BSS segment offsets allowing for the phrase
 601         // padding
 602         if ((flag == T_DATA) && !ofile->o_header.absrel.reloc.dsize)
 603         {
 604                 // SCPCD : the T_DATA is the last section of the file, we can now increment the textoffset, dataoffset and bssoffset
 605
 606                 // TEXT segment size plus padding
 607                 pad = ((ofile->o_header.tsize + secalign) & ~secalign);
 608                 textoffset += (ofile->o_header.tsize + (pad - ofile->o_header.tsize));
 609
 610                 if (vflag > 1)
 611                         printf("RelocateSegment(%s, TEXT) : No relocation data\n", ofile->o_name);
 612
 613                 // DATA segment size plus padding
 614                 pad = ((ofile->o_header.dsize + secalign) & ~secalign);
 615                 dataoffset += (ofile->o_header.dsize + (pad - ofile->o_header.dsize));
 616
 617                 // BSS segment size plus padding
 618                 pad = ((ofile->o_header.bsize + secalign) & ~secalign);
 619                 bssoffset += (ofile->o_header.bsize + (pad - ofile->o_header.bsize));
 620
 621                 if (vflag > 1)
 622                         printf("RelocateSegment(%s, DATA) : No relocation data\n", ofile->o_name);
 623
 624                 return 0;
 625         }
 626
 627         if (vflag > 1)
 628                 printf("RelocateSegment(%s, %s) : Processing Relocation Data\n",
 629                         ofile->o_name, flag == T_DATA ? "DATA" : "TEXT");
 630
 631         // Obtain pointer to start of symbol table
 632         symtab = (ofile->o_image + 32 + ofile->o_header.tsize
 633                 + ofile->o_header.dsize
 634                 + ofile->o_header.absrel.reloc.tsize
 635                 + ofile->o_header.absrel.reloc.dsize);
 636
 637         // Obtain pointer to start of symbols
 638         symbols = symtab + ofile->o_header.ssize;
 639
 640         // Obtain pointer to start of TEXT segment
 641         sptr = ofile->o_image + 32;
 642
 643         // Obtain pointer to start of TEXT relocation records
 644         rptr = sptr + (ofile->o_header.tsize + ofile->o_header.dsize);
 645
 646         relocsize = ofile->o_header.absrel.reloc.tsize;
 647
 648     if (vflag)
 649         printf("RELOCSIZE :: %d  Records = %d\n", relocsize, relocsize / 8);
 650
 651         // Update pointers if DATA relocation records are being processed
 652         if (flag == T_DATA)
 653         {
 654                 sptr += ofile->o_header.tsize;              // Start of DATA segment
 655                 rptr += ofile->o_header.absrel.reloc.tsize; // Start of DATA relocation records
 656                 relocsize = ofile->o_header.absrel.reloc.dsize;
 657         }
 658
 659         // Process each relocation record for the TEXT segment
 660         for(i=0; i<(int)relocsize; i+=8)
 661         {
 662                 // Obtain both the relocation address and the relocation flags from the
 663                 // object file image
 664                 addr = GetLong(rptr);
 665                 rflg = GetLong(rptr + 4);
 666                 glblreloc = (rflg & 0x00000010 ? 1 : 0);// Set global relocation flag
 667                 absreloc = (rflg & 0x00000040 ? 1 : 0); // Set absolute relocation flag
 668                 relreloc = (rflg & 0x000000A0 ? 1 : 0); // Set relative relocation flag
 669
 670                 // Additional processing required for global relocations
 671                 if (glblreloc)
 672                 {
 673                         // Obtain the string table index for the relocation symbol, look
 674                         // for it in the globals hash table to obtain information on that
 675                         // symbol. For the hash calculation to work correctly it must be
 676                         // placed in a 'clean' string before looking it up.
 677                         symidx = GetLong(symtab + ((rflg >> 8) * 12));
 678                         memset(sym, 0, SYMLEN);
 679                         strcpy(sym, symbols + symidx);
 680                         olddata = newdata = 0;   // Initialise old and new segment data
 681                         ssidx = OSTLookup(sym);
 682                         newdata = GetLong(ost + ((ssidx - 1) * 12) + 8);
 683                 }
 684
 685                 // Obtain the existing long word segment data and flip words if the
 686                 // relocation flags indicate it relates to a RISC MOVEI instruction
 687                 olddata = GetLong(sptr + addr);
 688
 689                 if (rflg & 0x01)
 690                         olddata = _SWAPWORD(olddata);
 691
 692                 // Process record dependant on segment it relates to; TEXT, DATA or
 693                 // BSS. Construct a new relocated segment long word based on the
 694                 // required segment base address, the segment data offset in the
 695                 // resulting COF file and the offsets from the incoming object file.
 696                 swcond = (rflg & 0xFFFFFF00);
 697
 698                 if (!glblreloc)
 699                 {
 700                         switch (swcond)
 701                         {
 702                         case 0x00000200:          // Absolute Value
 703                                 break;
 704                         case 0x00000400:          // TEXT segment relocation record
 705                                 // SCPCD : the symbol point to a text segment, we should use the textoffset
 706                                         newdata = tbase + textoffset + olddata;
 707
 708                                 break;
 709                         case 0x00000600:          // DATA segment relocation record
 710                                 newdata = dbase + dataoffset
 711                                         + (olddata - ofile->o_header.tsize);
 712
 713                                 break;
 714                         case 0x00000800:          // BSS segment relocation record
 715                                 newdata = bbase + bssoffset
 716                                         + (olddata - (ofile->o_header.tsize
 717                                         + ofile->o_header.dsize));
 718
 719                                 break;
 720                         }
 721                 }
 722                 else
 723                 {
 724                         if (!relreloc)
 725                                 newdata += olddata;
 726                 }
 727
 728                 // Set absolute (long) or relative (word) address of symbol
 729                 if (absreloc)
 730                 {
 731                         // Flip the new long word segment data if the relocation record
 732                         // indicated a RISC MOVEI instruction and place the resulting data
 733                         // back in the COF segment
 734                         if (rflg & 0x01)
 735                                 newdata = _SWAPWORD(newdata);
 736
 737                         PutLong(sptr + addr, newdata);
 738                 }
 739                 else if (relreloc)
 740                 {
 741                         PutWord(sptr + addr, newdata - tbase - addr - ofile->o_tbase);
 742                 }
 743
 744                 // Shamus: Let's output some info to aid in debugging this crap
 745                 if (vflag > 1)
 746                 {
 747                         char ssiString[128];
 748                         ssiString[0] = 0;
 749
 750                         if (glblreloc)
 751                                 sprintf(ssiString, " [ssi:%i]", ssidx);
 752
 753                         printf("RelocateSegment($%08X): %s, $%08X: $%08X => $%08X%s\n", rflg, (glblreloc ? sym : "(LOCAL)"), addr, olddata, GetLong(sptr + addr), ssiString);
 754                 }
 755
 756                 rptr += 8;     // Point to the next relocation record
 757         }
 758
 759         // Update the COF segment offset allowing for the phrase padding.
 760         // SCPCD : we should not increment the textoffset before the end of processing the object file, else data section will point to wrong textoffset
 761         if (flag == T_DATA)
 762         {
 763                 // TEXT segment plus padding
 764                 pad = ((ofile->o_header.tsize + secalign) & ~secalign);
 765                 textoffset += (ofile->o_header.tsize + (pad - ofile->o_header.tsize));
 766
 767                 // DATA segment plus padding
 768                 pad = ((ofile->o_header.dsize + secalign) & ~secalign);
 769                 dataoffset += (ofile->o_header.dsize + (pad - ofile->o_header.dsize));
 770
 771                 // BSS segment plus padding
 772                 pad = ((ofile->o_header.bsize + secalign) & ~secalign);
 773                 bssoffset += (ofile->o_header.bsize + (pad - ofile->o_header.bsize));
 774         }
 775
 776         // Return value, should always be zero
 777         return 0;
 778 }
 779
 780
 781 //
 782 // Add a path character to the end of string 's' if it doesn't already end with
 783 // one. The last occurrance of '/' or '\' in the string is assumed to be the
 784 // path character.
 785 //
 786 // This is fucking shit. We know what the path delimiter is, its FUCKING
 787 // DEFINED IN THE FUCKING HEADER. FOR FUCKS SAKE. AND YES, HOPE TO GOD THERE'S
 788 // ENOUGH SPACE IN THE PASSED IN BUFFER TO HOLD THE EXTRA CHARACTER!
 789 //
 790 void AppendPathDelimiter(char * s)
 791 {
 792 #if 0
 793         // And hope to God that there's enough space in the buffer...
 794         char pathchar = 0;
 795
 796         while (*s)
 797         {
 798                 if (*s == '/' || *s == '\\')
 799                         pathchar = *s;
 800
 801                 s++;
 802         }
 803
 804         s--;
 805
 806         if (*s == pathchar)
 807                 return;
 808
 809         *++s = pathchar;
 810         *++s = 0;
 811 #else
 812         int length = strlen(s);
 813
 814         if (s[length - 1] != PATH_DELIMITER)
 815         {
 816                 s[length] = PATH_DELIMITER;
 817                 s[length + 1] = 0;      // BUFFER OVERFLOW!!!! FFFFFFFFUUUUUUUUUUUU
 818         }
 819 #endif
 820 }
 821
 822
 823 //
 824 // Try to open "name", "name.o", "${libdir}name", "${libdir}name.o". Return the
 825 // handle of the file successfully opened. p_name is updated to point to a
 826 // malloc()'ed string which is the name which actually got opened. p_name will
 827 // return unchanged if the file can't be found.
 828 //
 829 int TryOpenFile(char ** p_name)
 830 {
 831         char * name = *p_name;
 832
 833         // Note that libdir will be an empty string if there is none specified
 834         char * tmpbuf = malloc(strlen(name) + strlen(libdir) + 3);
 835
 836         if (tmpbuf == NULL)
 837         {
 838                 printf("TryOpenFile() : out of memory\n");
 839                 return -1;
 840         }
 841
 842         strcpy(tmpbuf, name);
 843         int hasdot = (strrchr(tmpbuf, '.') > strrchr(tmpbuf, PATH_DELIMITER));
 844         // Try to open file as passed first
 845         int fd = open(tmpbuf, _OPEN_FLAGS);
 846
 847         if (fd >= 0)
 848                 goto ok;
 849
 850         if (!hasdot)
 851         {
 852                 // Try to open file with '.o' added
 853                 strcat(tmpbuf, ".o");
 854                 fd = open(tmpbuf, _OPEN_FLAGS);
 855
 856                 if (fd >= 0)
 857                         goto ok;
 858         }
 859
 860         // Try the libdir only if the name isn't already anchored
 861         // Shamus: WTH, this makes no sense... Why the ':'? Is this a Macintosh??
 862 //      if (*name != '/' && *name != '\\' && !strchr(name, ':'))
 863         if ((*name != PATH_DELIMITER) && (strchr(name, ':') == NULL))
 864         {
 865                 strcpy(tmpbuf, libdir);
 866                 // Add a trailing path char if there isn't one already
 867                 AppendPathDelimiter(tmpbuf);
 868                 strcat(tmpbuf, name);
 869
 870                 if ((fd = open(tmpbuf, _OPEN_FLAGS)) >= 0)
 871                         goto ok;
 872
 873                 if (!hasdot)
 874                 {
 875                         strcat(tmpbuf, ".o");
 876
 877                         if ((fd = open(tmpbuf, _OPEN_FLAGS)) >= 0)
 878                                 goto ok;
 879                 }
 880         }
 881
 882         // Couldn't open file at all
 883         return -1;
 884
 885 // There are worse things... :-P
 886 ok:
 887         tmpbuf = realloc(tmpbuf, strlen(tmpbuf) + 1);
 888
 889         if (tmpbuf == NULL)
 890         {
 891                 printf("TryOpenFile() : out of memory\n");
 892                 return -1;
 893         }
 894
 895         *p_name = tmpbuf;
 896         return fd;
 897 }
 898
 899
 900 //
 901 // What it says on the tin
 902 //
 903 void WriteARName(struct OFILE * p)
 904 {
 905         int flag = *(p->o_arname);
 906         printf("%s%s%s", (flag ? (char *)(p->o_arname) : ""), (flag ? ":" : ""), p->o_name);
 907 }
 908
 909
 910 //
 911 // Collect file names and handles in a buffer so there is less disk activity.
 912 // Call DoFile with flag FALSE for normal object files and archives.
 913 // Call it with flag TRUE and a symbol name for include files (-i).
 914 //
 915 int DoFile(char * fname, int incFlag, char * sym)
 916 {
 917         // Verbose information
 918         if (vflag)
 919         {
 920                 printf("DoFile() : `%s' %s", fname, incFlag ? "INCLUDE" : "NORMAL");
 921
 922                 if (incFlag)
 923                         printf(" symbol %s", sym);
 924
 925                 printf("\n");
 926         }
 927
 928         // Reached maximum file handles
 929         if (hd == NHANDLES)
 930         {
 931                 if (ProcessFiles())
 932                         return 1;
 933         }
 934
 935         // Attempt to open input file
 936         int fd = TryOpenFile(&fname);
 937
 938         if (fd < 0)
 939         {
 940                 printf("Cannot find input module %s\n", fname);
 941                 return 1;
 942         }
 943
 944         // The file is open; save its info in the handle and name arrays
 945         handle[hd] = fd;
 946         name[hd] = fname;               // This is the name from TryOpenFile()
 947         hflag[hd] = incFlag;
 948
 949         // Include files
 950         if (incFlag)
 951         {
 952                 int temp = strlen(sym);         // Get symbol length
 953
 954                 // 100 chars is max length of a symbol
 955                 if (temp > 99)
 956                 {
 957                         sym[99] = '\0';
 958                         temp = 99;
 959                 }
 960
 961                 // Malloc enough space for two symbols, then build the second one.
 962                 // Second one may be one character longer than first
 963                 if ((hsym1[hd] = malloc((long)temp + 1)) == NULL
 964                         || (hsym2[hd] = malloc((long)temp + 2)) == NULL)
 965                 {
 966                         printf("DoFile() : out of memory for include-file symbols\n");
 967                         return 1;
 968                 }
 969
 970                 strcpy(hsym1[hd], sym);
 971                 strcpy(hsym2[hd], sym);
 972
 973                 if (temp == 99)
 974                 {
 975                         if (sym[99] == 'x')
 976                         {
 977                                 printf("Last char of %s is already 'x': choose another name\n", sym);
 978                                 return 1;
 979                         }
 980
 981                         hsym2[hd][99] = 'x';
 982                 }
 983                 else
 984                 {
 985                         hsym2[hd][temp] = 'x';
 986                         hsym2[hd][temp+1] = '\0';
 987                 }
 988         }
 989
 990         // Increment next handle index
 991         hd++;
 992         // No problems
 993         return 0;
 994 }
 995
 996
 997 //
 998 // Pad TEXT or DATA segment to the requested boundary
 999 //
1000 int PadSegment(FILE * fd, long segsize, int value)
1001 {
1002         int i;
1003         char padarray[32];
1004         char * padptr;
1005
1006         // Determine the number of padding bytes that are needed
1007         long padsize = (segsize + secalign) & ~secalign;
1008         padsize = padsize - segsize;
1009
1010         // Fill pad array if padding is required
1011         if (padsize)
1012         {
1013                 padptr = padarray;
1014
1015                 for(i=0; i<16; i++)
1016                 {
1017                         PutWord(padptr, value);
1018                         padptr += 2;
1019                 }
1020
1021                 symoffset += padsize;
1022
1023                 // Write padding bytes
1024                 if (fwrite(padarray, padsize, 1, fd) != 1)
1025                         return 1;
1026         }
1027
1028         return 0;
1029 }
1030
1031
1032 //
1033 // Write the output file
1034 //
1035 int WriteOutputFile(struct OHEADER * header)
1036 {
1037         unsigned osize;                                         // Object segment size
1038         struct OFILE * otemp;                           // Object file pointer
1039         int i, j;                                                       // Iterators
1040         char himage[0x168];                                     // Header image (COF = 0xA8)
1041         uint32_t tsoff, dsoff, bsoff;           // Segment offset values
1042         unsigned index, type, value;            // Symbol table index, type and value
1043         short abstype;                                          // ABS symbol type
1044         char symbol[14];                                        // Symbol record for ABS files
1045         int slen;                                                       // Symbol string length
1046
1047         symoffset = 0;                                          // Initialise symbol offset
1048
1049         // Add correct output extension if none
1050         if (strchr(ofile, '.') == NULL)
1051         {
1052                 if (aflag && cflag)
1053                         strcat(ofile, ".cof");          // COF files
1054                 else if (aflag && !cflag)
1055                         strcat(ofile, ".abs");          // ABS files
1056                 else
1057                         strcat(ofile, ".o");            // Object files (partial linking etc)
1058         }
1059
1060         FILE * fd = fopen(ofile, "wb");         // Attempt to open output file
1061
1062         if (!fd)
1063         {
1064                 printf("Can't open output file %s\n", ofile);
1065                 return 1;
1066         }
1067
1068         // Build the output file header
1069         // Absolute (COF) header
1070         if (cflag)
1071         {
1072                 tsoff = dsoff = bsoff = 0xA8;   // Initialises segment offsets
1073
1074                 // Process each object file segment size to obtain a cumulative segment
1075                 // size for both the TEXT and DATA segments
1076                 for(otemp=olist; otemp!=NULL; otemp=otemp->o_next)
1077                 {
1078                         dsoff += otemp->segSize[TEXT];
1079                         bsoff += otemp->segSize[TEXT] + otemp->segSize[DATA];
1080                 }
1081
1082                 // Currently this only builds a COF absolute file. Conditionals and
1083                 // additional code will need to be added for ABS and partial linking.
1084
1085                 // Build the COF_HDR
1086                 PutWord(himage + 0,   0x0150               ); // Magic Number (0x0150)
1087                 PutWord(himage + 2,   0x0003               ); // Sections Number (3)
1088                 PutLong(himage + 4,   0x00000000           ); // Date (0L)
1089                 PutLong(himage + 8,   dsoff + header->dsize); // Offset to Symbols Section
1090                 PutLong(himage + 12,  ost_index);             // Number of Symbols
1091                 PutWord(himage + 16,  0x001C               ); // Size of RUN_HDR (0x1C)
1092                 PutWord(himage + 18,  0x0003               ); // Executable Flags (3)
1093
1094                 // Build the RUN_HDR
1095                 PutLong(himage + 20,  0x00000107           ); // Magic/vstamp
1096                 PutLong(himage + 24,  header->tsize        ); // TEXT size in bytes
1097                 PutLong(himage + 28,  header->dsize        ); // DATA size in bytes
1098                 PutLong(himage + 32,  header->bsize        ); // BSS size in bytes
1099                 PutLong(himage + 36,  tbase                ); // Start of executable, normally @TEXT
1100                 PutLong(himage + 40,  tbase                ); // @TEXT
1101                 PutLong(himage + 44,  dbase                ); // @DATA
1102
1103                 // Build the TEXT SEC_HDR
1104                 PutLong(himage + 48,  0x2E746578           );
1105                 PutLong(himage + 52,  0x74000000           ); // ".text"
1106                 PutLong(himage + 56,  tbase                ); // TEXT START
1107                 PutLong(himage + 60,  tbase                ); // TEXT START
1108                 PutLong(himage + 64,  header->tsize        ); // TEXT size in bytes
1109                 PutLong(himage + 68,  tsoff                ); // Offset to section data in file
1110                 PutLong(himage + 72,  0x00000000           ); // Offset to section reloc in file (0L)
1111                 PutLong(himage + 76,  0x00000000           ); // Offset to debug lines structures (0L)
1112                 PutLong(himage + 80,  0x00000000           ); // Nreloc/nlnno (0L)
1113                 PutLong(himage + 84,  0x00000020           ); // SEC_FLAGS: STYP_TEXT
1114
1115                 // Build the DATA SEC_HDR
1116                 PutLong(himage + 88,  0x2E646174           );
1117                 PutLong(himage + 92,  0x61000000           ); // ".data"
1118                 PutLong(himage + 96,  dbase                ); // DATA START
1119                 PutLong(himage + 100, dbase                ); // DATA START
1120                 PutLong(himage + 104, header->dsize        ); // DATA size in bytes
1121                 PutLong(himage + 108, dsoff                ); // Offset to section data in file
1122                 PutLong(himage + 112, 0x00000000           ); // Offset to section reloc in file (0L)
1123                 PutLong(himage + 116, 0x00000000           ); // Offset to debugging lines structures (0L)
1124                 PutLong(himage + 120, 0x00000000           ); // Nreloc/nlnno (0L)
1125                 PutLong(himage + 124, 0x00000040           ); // SEC_FLAGS: STYP_DATA
1126
1127                 // Build the BSS SEC_HDR
1128                 PutLong(himage + 128, 0x2E627373           );
1129                 PutLong(himage + 132, 0x00000000           ); // ".bss"
1130                 PutLong(himage + 136, bbase                ); // BSS START
1131                 PutLong(himage + 140, bbase                ); // BSS START
1132                 PutLong(himage + 144, header->bsize        ); // BSS size in bytes
1133                 PutLong(himage + 148, bsoff                ); // Offset to section data in file
1134                 PutLong(himage + 152, 0x00000000           ); // Offset to section reloc in file (0L)
1135                 PutLong(himage + 156, 0x00000000           ); // Offset to debugging lines structures (0L)
1136                 PutLong(himage + 160, 0x00000000           ); // Nreloc/nlnno (0L)
1137                 PutLong(himage + 164, 0x00000080           ); // SEC_FLAGS: STYP_BSS
1138
1139                 symoffset = 168;                              // Update symbol offset
1140         }
1141         // Absolute (ABS) header
1142         else
1143         {
1144                 // Build the ABS header
1145                 PutWord(himage + 0,   0x601B               ); // Magic Number (0x601B)
1146                 PutLong(himage + 2,   header->tsize        ); // TEXT segment size
1147                 PutLong(himage + 6,   header->dsize        ); // DATA segment size
1148                 PutLong(himage + 10,  header->bsize        ); // BSS segment size
1149                 PutLong(himage + 14,  ost_index * 14       ); // Symbol table size (?)
1150                 PutLong(himage + 18,  0x00000000           ); //
1151                 PutLong(himage + 22,  tbase                ); // TEXT base address
1152                 PutWord(himage + 26,  0xFFFF               ); // Flags (?)
1153                 PutLong(himage + 28,  dbase                ); // DATA base address
1154                 PutLong(himage + 32,  bbase                ); // BSS base address
1155
1156                 symoffset = 36;                               // Update symbol offset
1157         }
1158
1159         // Write the header, but not if noheaderflag
1160         // Absolute (ABS) header
1161         if (!cflag)
1162         {
1163                 if (!noheaderflag)
1164                         if (fwrite(himage, 36, 1, fd) != 1)
1165                                 goto werror;
1166         }
1167         // Absolute (COF) header
1168         else
1169         {
1170                 if (fwrite(himage, 168, 1, fd) != 1)
1171                         goto werror;
1172         }
1173
1174         // Write the TEXT segment of each object file
1175         for(otemp=olist; otemp!=NULL; otemp=otemp->o_next)
1176         {
1177                 osize = otemp->o_header.tsize;
1178
1179                 // Write only if segment has size
1180                 if (osize)
1181                 {
1182                         if (vflag > 1)
1183                                 printf("Writing TEXT Segment of %s\n", otemp->o_name);
1184
1185                         if (fwrite(otemp->o_image + 32, osize, 1, fd) != 1)
1186                                 goto werror;
1187
1188                         // Pad to required alignment boundary
1189                         if (PadSegment(fd, osize, 0x0000))
1190                                 goto werror;
1191
1192                         symoffset += osize;
1193                 }
1194         }
1195
1196         // Write the DATA segment of each object file
1197         for(otemp=olist; otemp!=NULL; otemp=otemp->o_next)
1198         {
1199                 osize = otemp->o_header.dsize;
1200
1201                 // Write only if the segment has size
1202                 if (osize)
1203                 {
1204                         if (vflag > 1)
1205                                 printf("Writing DATA Segment of %s\n", otemp->o_name);
1206
1207                         if (fwrite((otemp->o_image + 32 + otemp->o_header.tsize), osize, 1, fd) != 1)
1208                                 goto werror;
1209
1210                         // Pad to required alignment boundary
1211                         if (PadSegment(fd, osize, 0))
1212                                 goto werror;
1213
1214                         symoffset += osize;
1215                 }
1216         }
1217
1218         if (!noheaderflag)
1219         {
1220                 // Write the symbols table and string table
1221                 // Absolute (COF) symbol/string table
1222                 if (cflag)
1223                 {
1224                         if (header->ssize)
1225                         {
1226                                 if (fwrite(ost, (ost_ptr - ost), 1, fd) != 1)
1227                                         goto werror;
1228
1229                                 if (fwrite(oststr, (oststr_ptr - oststr), 1, fd) != 1)
1230                                         goto werror;
1231                         }
1232                 }
1233                 // Absolute (ABS) symbol/string table
1234                 else
1235                 {
1236                         // The symbol and string table have been created as part of the
1237                         // DoSymbols() function and the output symbol and string tables are
1238                         // in COF format. For an ABS file we need to process through this
1239                         // to create the 14 character long combined symbol and string
1240                         // table. Format of symbol table in ABS: AAAAAAAATTVVVV, where
1241                         // (A)=STRING, (T)=TYPE & (V)=VALUE
1242
1243                         for(i=0; i<ost_index; i++)
1244                         {
1245                                 memset(symbol, 0, 14);          // Initialise symbol record
1246                                 abstype = 0;                            // Initialise ABS symbol type
1247                                 slen = 0;                                       // Initialise symbol string length
1248                                 index = GetLong(ost + (i * 12));        // Get symbol index
1249                                 type  = GetLong((ost + (i * 12)) + 4);  // Get symbol type
1250
1251                                 // Skip debug symbols
1252                                 if (type & 0xF0000000)
1253                                         continue;
1254
1255                                 // Get symbol value
1256                                 value = GetLong((ost + (i * 12)) + 8);
1257                                 slen = strlen(oststr + index);
1258
1259                                 // Get symbol string (maximum 8 chars)
1260                                 if (slen > 8)
1261                                 {
1262                                         for(j=0; j<8; j++)
1263                                                 *(symbol + j) = *(oststr + index + j);
1264                                 }
1265                                 else
1266                                 {
1267                                         for(j=0; j<slen; j++)
1268                                                 *(symbol + j) = *(oststr + index + j);
1269                                 }
1270
1271                                 // Modify to ABS symbol type
1272                                 switch (type)
1273                                 {
1274                                 case 0x02000000: abstype = (short)ABST_DEFINED;                           break;
1275                                 case 0x04000000: abstype = (short)ABST_DEFINED | ABST_TEXT;               break;
1276                                 case 0x05000000: abstype = (short)ABST_DEFINED | ABST_GLOBAL | ABST_TEXT; break;
1277                                 case 0x06000000: abstype = (short)ABST_DEFINED | ABST_DATA;               break;
1278                                 case 0x07000000: abstype = (short)ABST_DEFINED | ABST_GLOBAL | ABST_DATA; break;
1279                                 case 0x08000000: abstype = (short)ABST_DEFINED | ABST_BSS;                break;
1280                                 case 0x09000000: abstype = (short)ABST_DEFINED | ABST_GLOBAL | ABST_BSS;  break;
1281                                 default:
1282                                         printf("warning (WriteOutputFile): ABS, cannot determine symbol type ($%08X) [%s]\n", type, symbol);
1283 //                                      type = 0;
1284                                         break;
1285                                 }
1286
1287                                 PutWord(symbol + 8, abstype);   // Write back new ABS type
1288                                 PutLong(symbol + 10, value);    // Write back value
1289
1290                                 // Write symbol record
1291                                 if (fwrite(symbol, 14, 1, fd) != 1)
1292                                         goto werror;
1293                         }
1294                 }
1295         }
1296
1297         if (fclose(fd))
1298         {
1299                 printf("Close error on output file %s\n", ofile);
1300                 return 1;
1301         }
1302
1303         return 0;
1304
1305 werror:
1306         printf("Write error on output file %s\n", ofile);
1307         fclose(fd);                     // Try to close output file anyway
1308         return 1;
1309 }
1310
1311
1312 //
1313 // Display the symbol load map
1314 //
1315 int ShowSymbolLoadMap(struct OHEADER * header)
1316 {
1317         unsigned i, o;                  // Inner and outer loop iterators
1318         unsigned c;                             // Column number
1319         unsigned index;                 // Symbol string index
1320         unsigned type;                  // Symbol type
1321         unsigned value;                 // Symbol value
1322         char * symbol;                  // Symbol string value
1323
1324         if (ost_index == 0)
1325                 return 0;                       // Return if no symbols to map
1326
1327         printf("LOAD MAP\n\n");
1328
1329         // Outer loop for each of the symbol areas to map out;
1330         // 0 = NON-RELOCATABLE SYMBOLS
1331         // 1 = TEXT-SEGMENT RELOCATABLE SYMBOLS
1332         // 2 = DATA-SEGMENT RELOCATABLE SYMBOLS
1333         // 3 = BSS-SEGMENT RELOCATABLE SYMBOLS
1334         for(o=0; o<4; o++)
1335         {
1336                 // Display the correct map header for the symbols being processed
1337                 switch (o)
1338                 {
1339                 case 0: printf("NON-RELOCATABLE SYMBOLS\n\n");          break;
1340                 case 1: printf("TEXT-SEGMENT RELOCATABLE SYMBOLS\n\n"); break;
1341                 case 2: printf("DATA-SEGMENT RELOCATABLE SYMBOLS\n\n"); break;
1342                 case 3: printf("BSS-SEGMENT RELOCATABLE SYMBOLS\n\n");  break;
1343                 }
1344
1345                 c = 0;                          // Initialise column number
1346
1347                 // Inner loop to process each record in the symbol table
1348                 for(i=0; i<(unsigned)ost_index; i++)
1349                 {
1350                         index  = GetLong(ost + (i * 12));               // Get symbol string index
1351                         type   = GetLong(ost + (i * 12) + 4);   // Get symbol type
1352                         value  = GetLong(ost + (i * 12) + 8);   // Get symbol value
1353                         symbol = oststr + index;                                // Get symbol string
1354
1355                         // Display only three columns
1356                         if (c == 3)
1357                         {
1358                                 printf("\n");
1359                                 c = 0;
1360                         }
1361
1362                         // If local symbols not included and the type is local then go to
1363                         // next symbol record
1364                         if (!lflag & !(type & 0x01000000))
1365                                 continue;
1366
1367                         // Output each symbol to the display, dependant on type
1368                         switch (o)
1369                         {
1370                         case 0:
1371                                 // Non-relocatable symbols
1372                                 if (type == 0x02000000 || type == 0x03000000)
1373                                 {
1374                                         printf("%-8s %c  %08X   ", symbol, (type & 0x01000000) ? 'G' : 'L', value);
1375                                         c++;
1376                                 }
1377
1378                                 break;
1379                         case 1:
1380                                 // TEXT segment relocatable symbols
1381                                 if (type == 0x04000000 || type == 0x05000000)
1382                                 {
1383                                         printf("%-8s %c  %08X   ", symbol, (type & 0x01000000) ? 'G' : 'L', value);
1384                                         c++;
1385                                 }
1386
1387                                 break;
1388                         case 2:
1389                                 // DATA segment relocatble symbols
1390                                 if (type == 0x06000000 || type == 0x07000000)
1391                                 {
1392                                         printf("%-8s %c  %08X   ", symbol, (type & 0x01000000) ? 'G' : 'L', value);
1393                                         c++;
1394                                 }
1395
1396                                 break;
1397                         case 3:
1398                                 // BSS segment relocatable symbols
1399                                 if (type == 0x08000000 || type == 0x09000000)
1400                                 {
1401                                         printf("%-8s %c  %08X   ", symbol, (type & 0x01000000) ? 'G' : 'L', value);
1402                                         c++;
1403                                 }
1404
1405                                 break;
1406                         }
1407                 }
1408
1409                 printf("\n\n");
1410         }
1411
1412         return 0;
1413 }
1414
1415
1416 //
1417 // Stuff the (long) value of a string into the value argument. RETURNS TRUE if
1418 // the string doesn't parse.  Parses only as a hexadecimal string.
1419 //
1420 int GetHexValue(char * string, int * value)
1421 {
1422         *value = 0;
1423
1424         while (isxdigit(*string))
1425         {
1426                 if (isdigit(*string))
1427                 {
1428                         *value = (*value << 4) + (*string++ - '0');
1429                 }
1430                 else
1431                 {
1432                         if (isupper(*string))
1433                                 *string = tolower(*string);
1434
1435                         *value = (*value << 4) + ((*string++ - 'a') + 10);
1436                 }
1437         }
1438
1439         if (*string != '\0')
1440         {
1441                 printf("Invalid hexadecimal value");
1442                 return 1;
1443         }
1444
1445         return 0;
1446 }
1447
1448
1449 //
1450 // Create one big .o file from the images already in memory, returning a
1451 // pointer to an OHEADER. Note that the oheader is just the header for the
1452 // output (plus some other information). The text, data, and fixups are all
1453 // still in the ofile images hanging off the global 'olist'.
1454 //
1455 struct OHEADER * MakeOutputObject()
1456 {
1457         unsigned tptr, dptr, bptr;      // Bases in runtime model
1458         int ret = 0;                            // Return value
1459         struct OHEADER * header;        // Output header pointer
1460
1461         // Initialize cumulative segment sizes
1462         textsize = datasize = bsssize = 0;
1463
1464         // For each object file, accumulate the sizes of the segments but remove
1465         // those object files which are unused
1466         struct OFILE * oprev = NULL;    // Init previous obj file list ptr
1467         struct OFILE * otemp = olist;   // Set temp pointer to object file list
1468
1469         while (otemp != NULL)
1470         {
1471                 // If the object is unused, discard it...
1472                 if ((otemp->o_flags & O_USED) == 0)
1473                 {
1474                         if (wflag)
1475                         {
1476                                 printf("Unused object file ");
1477                                 WriteARName(otemp);
1478                                 printf(" discarded.\n");
1479                         }
1480
1481                         // Drop the entry from the linked list
1482                         if (oprev == NULL)
1483                                 olist = otemp->o_next;
1484                         else
1485                                 oprev->o_next = otemp->o_next;
1486
1487                         // Free the object entry if it's not an archive file
1488                         if (!otemp->isArchiveFile)
1489                                 free(otemp->o_image);
1490                 }
1491                 else
1492                 {
1493                         // Save accumulated addresses in the object
1494                         otemp->segBase[TEXT] = textsize;
1495                         otemp->segBase[DATA] = datasize;
1496                         otemp->segBase[BSS]  = bsssize;
1497
1498                         // Increment total of segment sizes ensuring requested alignment
1499                         textsize += (otemp->o_header.tsize + secalign) & ~secalign;
1500                         datasize += (otemp->o_header.dsize + secalign) & ~secalign;
1501                         bsssize  += (otemp->o_header.bsize + secalign) & ~secalign;
1502                         oprev = otemp;
1503                 }
1504
1505                 // Go to next object file list pointer
1506                 otemp = otemp->o_next;
1507         }
1508
1509         // Update base addresses and inject the symbols _TEXT_E, _DATA_E and _BSS_E
1510         // into the OST
1511         tbase = tval;
1512
1513         if (!dval)
1514         {
1515                 // DATA follows TEXT
1516                 dbase = tval + textsize;
1517
1518                 if (!bval)
1519                         // BSS follows DATA
1520                         bbase = tval + textsize + datasize;
1521                 else
1522                         // BSS is independent of DATA
1523                         bbase = bval;
1524         }
1525         else
1526         {
1527                 // DATA is independent of TEXT
1528                 dbase = dval;
1529
1530                 if (!bval)
1531                         // BSS follows DATA
1532                         bbase = dval + datasize;
1533                 else
1534                         // BSS is independent of DATA
1535                         bbase = bval;
1536         }
1537
1538         // Inject segment end labels, for C compilers that expect this shite
1539         OSTAdd("_TEXT_E", 0x05000000, tbase + textsize);
1540         OSTAdd("_DATA_E", 0x07000000, dbase + datasize);
1541         OSTAdd("_BSS_E",  0x09000000, bbase + bsssize);
1542
1543         // Place each unresolved symbol in the output symbol table
1544         // N.B.: It only gets here to do this if user passes in -u flag
1545         //       [Only used here, once]
1546         if (DoUnresolved())
1547                 return NULL;
1548
1549         // Initialise base addresses
1550         tptr = dptr = bptr = 0;
1551
1552         // For each file, relocate its symbols and add them to the output symbol
1553         // table
1554         otemp = olist;
1555         oprev = NULL;
1556
1557         while (otemp != NULL)
1558         {
1559                 otemp->o_tbase = tptr;
1560                 otemp->o_dbase = dptr;
1561                 otemp->o_bbase = bptr;
1562                 tptr += (otemp->o_header.tsize + secalign) & ~secalign;
1563                 dptr += (otemp->o_header.dsize + secalign) & ~secalign;
1564                 bptr += (otemp->o_header.bsize + secalign) & ~secalign;
1565
1566                 // For each symbol, (conditionally) add it to the ost
1567                 // For ARCHIVE markers, this adds the symbol for the file & returns
1568                 // (Shamus: N.B. it does no such thing ATM)
1569                 // [Only used here, once]
1570                 if (DoSymbols(otemp))
1571                         return NULL;
1572
1573                 oprev = otemp;
1574                 otemp = otemp->o_next;
1575         }
1576
1577         // Places all the externs, globals etc into the output symbol table
1578         if (DoCommon() == -1)
1579                 return NULL;
1580
1581         // Create a new output file header
1582         header = new_oheader();
1583
1584         if (header == NULL)
1585         {
1586                 printf("MakeOutputObject: out of memory!\n");
1587                 return NULL;
1588         }
1589
1590         // Fill in the output header. Does not match the actual output but values
1591         // used as reference
1592         header->magic = 0x0150;                         // COF magic number
1593         header->tsize = textsize;                       // TEXT segment size
1594         header->dsize = datasize;                       // DATA segment size
1595         header->bsize = bsssize;                        // BSS segment size
1596         header->ssize = (ost_ptr - ost);        // Symbol table size
1597         header->ostbase = ost;                          // Output symbol table base address
1598
1599         // For each object file, relocate its TEXT and DATA segments. OR the result
1600         // into ret so all files get moved (and errors reported) before returning
1601         // with the error condition
1602         for(otemp=olist; otemp!=NULL; otemp=otemp->o_next)
1603         {
1604                 ret |= RelocateSegment(otemp, T_TEXT); // TEXT segment relocations
1605                 ret |= RelocateSegment(otemp, T_DATA); // DATA segment relocations
1606         }
1607
1608         // Done with global symbol hash tables
1609         FreeHashes();
1610
1611         return (ret ? (struct OHEADER *)NULL : header);
1612 }
1613
1614
1615 //
1616 // Add symbol to hash list
1617 //
1618 int AddSymbolToHashList(struct HREC ** hptr, char * sym, struct OFILE * ofile,
1619         long value, int type)
1620 {
1621         struct HREC * htemp = new_hrec();
1622
1623         if (htemp == NULL)
1624         {
1625                 printf("Out of memory\n");
1626                 return 1;
1627         }
1628
1629         // Shamus: Moar testing...
1630         if (vflag > 1)
1631         {
1632                 printf("AddSymbolToHashList(): hptr=$%08X, sym=\"%s\", ofile=$%08X, value=$%X, type=$%X\n", hptr, sym, ofile, value, type);
1633         }
1634
1635         // Populate hash record
1636         memset(htemp->h_sym, 0, SYMLEN);
1637         strcpy(htemp->h_sym, sym);
1638         htemp->h_ofile = ofile;
1639         htemp->h_value = value;
1640         htemp->h_type = type;
1641
1642         // Add new hash to the front of the list (hence the ** for hptr)
1643         htemp->h_next = *hptr;
1644         *hptr = htemp;
1645
1646         return 0;
1647 }
1648
1649
1650 //
1651 // Add symbol to the unresolved symbols hash table (really, it's a linked list)
1652 //
1653 int AddUnresolvedSymbol(char * sym, struct OFILE * ofile)
1654 {
1655         if (vflag > 1)
1656                 printf("AddUnresolvedSymbol(%s, %s)\n", sym, ofile->o_name);
1657
1658         return AddSymbolToHashList(&unresolved, sym, ofile, 0L, 0);
1659 }
1660
1661
1662 //
1663 // Remove the HREC from the unresolved symbol list, and pass back a pointer
1664 // to the spot where the HREC was.
1665 //
1666 struct HREC * RemoveUnresolvedSymbol(struct HREC * hrec)
1667 {
1668         struct HREC * ptr = unresolved;
1669         struct HREC * previous = NULL;
1670
1671         while ((ptr != hrec) && (ptr != NULL))
1672         {
1673                 previous = ptr;
1674                 ptr = ptr->h_next;
1675         }
1676
1677         // Not found...!
1678         if (ptr == NULL)
1679                 return NULL;
1680
1681         struct HREC * next = ptr->h_next;
1682
1683         // Remove the head if nothing previous, otherwise, remove what we found
1684         if (previous == NULL)
1685                 unresolved = next;
1686         else
1687                 previous->h_next = next;
1688
1689         free(ptr);
1690         return next;
1691 }
1692
1693
1694 //
1695 // Add symbol to the unresolved symbols hash table
1696 //
1697 int AddARSymbol(char * sym, struct OFILE * ofile)
1698 {
1699         if (vflag > 1)
1700                 printf("AddARSymbol(%s, %s)\n", sym, ofile->o_name);
1701
1702         return AddSymbolToHashList(&arSymbol, sym, ofile, 0L, 0);
1703 }
1704
1705
1706 //
1707 // Generate hash value from the 1st 15 characters of the symbol modulo the
1708 // number of buckets in the hash.
1709 //
1710 int GetHash(char * s)
1711 {
1712         // For this to be consistent, the symbol MUST be zeroed out beforehand!
1713         // N.B.: strncpy() pads zeroes for us, if the symbol is less than 15 chars.
1714         char c[15];
1715         strncpy(c, s, 15);
1716
1717         int i = (c[0] + c[1] + c[2] + c[3] + c[4] + c[5] + c[6] + c[7] + c[8]
1718                 + c[9] + c[10] + c[11] + c[12] + c[13] + c[14]) % NBUCKETS;
1719         return i;
1720 }
1721
1722
1723 //
1724 // Lookup a symbol in the hash table.
1725 // Returns either a pointer to the HREC or NULL if not found.
1726 //
1727 struct HREC * LookupHREC(char * symbol)
1728 {
1729         struct HREC * hptr = htable[GetHash(symbol)];
1730
1731         while (hptr != NULL)
1732         {
1733 //This is utter failure...
1734 //              if (symcmp(symbol, hptr->h_sym))  <-- left here for giggles :D  - LinkoVitch
1735                 // Return hash record pointer if found
1736                 if (strcmp(symbol, hptr->h_sym) == 0)
1737                         return hptr;
1738
1739                 hptr = hptr->h_next;
1740         }
1741
1742         return NULL;
1743 }
1744
1745
1746 //
1747 // Lookup a symbol in the AR symbol table.
1748 // Returns either a pointer to the HREC or NULL if not found.
1749 //
1750 struct HREC * LookupARHREC(char * symbol)
1751 {
1752         struct HREC * hptr = arSymbol;
1753
1754         while (hptr != NULL)
1755         {
1756                 // Return hash record pointer if found
1757                 if (strcmp(symbol, hptr->h_sym) == 0)
1758                         return hptr;
1759
1760                 hptr = hptr->h_next;
1761         }
1762
1763         return NULL;
1764 }
1765
1766
1767 //
1768 // Add the imported symbols from this file to unresolved, and the global and
1769 // common (???) symbols to the exported hash table.
1770 //
1771 // Change old-style commons (type == T_EXTERN, value != 0) to new-style ones
1772 // (type == (T_GLOBAL | T_EXTERN)). [??? O_o]
1773 // [N.B.: Whoever wrote the above didn't know what the fuck they were talking
1774 //        about. Commons (globals) are exactly what they are calling 'old
1775 //        style'. Also note, that there is no "T_GLOBAL" or "T_EXTERN" symbols
1776 //        defined anywhere in the code.]
1777 //
1778 int AddSymbols(struct OFILE * Ofile)
1779 {
1780         struct HREC * hptr;                     // Hash record pointer
1781
1782         if (vflag > 1)
1783         {
1784                 printf("AddSymbols: for file %s\n", Ofile->o_name);
1785                 printf("            t_bbase = $%X\n", Ofile->o_tbase);
1786                 printf("            d_bbase = $%X\n", Ofile->o_dbase);
1787                 printf("            o_bbase = $%X\n", Ofile->o_bbase);
1788                 printf("            tsize = $%X\n", Ofile->o_header.tsize);
1789                 printf("            dsize = $%X\n", Ofile->o_header.dsize);
1790                 printf("            bsize = $%X\n", Ofile->o_header.bsize);
1791                 printf("            reloc.tsize = $%X\n", Ofile->o_header.absrel.reloc.tsize);
1792                 printf("            reloc.dsize = $%X\n", Ofile->o_header.absrel.reloc.dsize);
1793         }
1794
1795         // Get base pointer, start of sym fixups
1796         char * ptr = Ofile->o_image + 32
1797                 + Ofile->o_header.tsize
1798                 + Ofile->o_header.dsize
1799                 + Ofile->o_header.absrel.reloc.tsize
1800                 + Ofile->o_header.absrel.reloc.dsize;
1801         char * sfix = ptr;                                                      // Set symbol fixup pointer
1802         char * sstr = sfix + Ofile->o_header.ssize;     // Set symbol table pointer
1803         long nsymbols = Ofile->o_header.ssize / 12;     // Obtain number of symbols
1804
1805         while (nsymbols)
1806         {
1807                 long index = GetLong(sfix);                             // Get symbol string index
1808                 long type  = GetLong(sfix + 4);                 // Get symbol type
1809                 long value = GetLong(sfix + 8);                 // Get symbol value
1810
1811                 if ((Ofile->isArchiveFile) && !(Ofile->o_flags & O_USED))
1812                 {
1813                         if ((type & T_GLBL) && (type & (T_SEG | T_ABS)))
1814                                 if (AddARSymbol(sstr + index, Ofile))
1815                                         return 1;
1816                 }
1817                 else if (type == T_GLBL)
1818                 {
1819                         // Global symbol that may or may not be in the current unit
1820                         hptr = LookupHREC(sstr + index);
1821
1822                         if (hptr != NULL)
1823                                 hptr->h_ofile->o_flags |= O_USED;       // Mark .o file as used
1824                         // Otherwise, *maybe* add to unresolved list
1825                         else
1826                         {
1827                                 // Check to see if this is a common symbol; if so, add it to
1828                                 // the hash list...
1829                                 if (value != 0)
1830                                 {
1831                                         // Actually, we need to convert this to a BSS symbol,
1832                                         // increase the size of the BSS segment for this object, &
1833                                         // add it to the hash list
1834                                         uint32_t bssLocation = Ofile->o_header.tsize + Ofile->o_header.dsize + Ofile->o_header.bsize;
1835                                         Ofile->o_header.bsize += value;
1836                                         Ofile->segSize[BSS] += value;
1837                                         type |= T_BSS;
1838                                         value = bssLocation;
1839                                         PutLong(sfix + 4, type);
1840                                         PutLong(sfix + 8, value);
1841
1842                                         if (vflag > 1)
1843                                                 printf("AddSymbols: Resetting common label to BSS label\n");
1844
1845                                         if (AddSymbolToHashList(&htable[GetHash(sstr + index)],
1846                                                 sstr + index, Ofile, value, type))
1847                                                 return 1;                               // Error if addition failed
1848                                 }
1849                                 // Make sure it's not a built-in external...
1850                                 else if ((strcmp(sstr + index, "_TEXT_E") != 0)
1851                                         && (strcmp(sstr + index, "_DATA_E") != 0)
1852                                         && (strcmp(sstr + index, "_BSS_E") != 0))
1853                                 {
1854                                         if (AddUnresolvedSymbol(sstr + index, Ofile))
1855                                                 return 1;                               // Error if addition failed
1856                                 }
1857                         }
1858                 }
1859                 else if ((type & T_GLBL) && (type & (T_SEG | T_ABS)))
1860                 {
1861                         hptr = LookupHREC(sstr + index);
1862
1863                         // Symbol isn't in the table, so try to add it:
1864                         if (hptr == NULL)
1865                         {
1866                                 if (AddSymbolToHashList(&htable[GetHash(sstr + index)],
1867                                         sstr + index, Ofile, value, type))
1868                                         return 1;
1869                         }
1870                         else
1871                         {
1872                                 // Symbol already exists, decide what to do about it
1873                                 // [N.B.: This isn't a check for a common symbol...
1874                                 //        BEWARE OF BAD INTERPRETATIONS!!]
1875                                 if (iscommon(hptr->h_type))
1876                                 {
1877                                         // Mismatch: common came first; warn and keep the global
1878                                         if (wflag)
1879                                         {
1880                                                 printf("Warning: %s: global from ", sstr + index);
1881                                                 WriteARName(Ofile);
1882                                                 printf(" used, common from ");
1883                                                 WriteARName(hptr->h_ofile);
1884                                                 printf(" discarded.\n");
1885                                         }
1886
1887                                         hptr->h_ofile = Ofile;
1888                                         hptr->h_type = type;
1889                                         hptr->h_value = value;
1890                                 }
1891                                 else
1892                                 {
1893                                         // Global exported by another ofile; warn and make this one
1894                                         // extern
1895                                         if (wflag)
1896                                         {
1897                                                 printf("Duplicate symbol %s: ", sstr + index);
1898                                                 WriteARName(hptr->h_ofile);
1899                                                 printf(" used, ");
1900                                                 WriteARName(Ofile);
1901                                                 printf(" discarded\n");
1902                                         }
1903
1904                                         // Set the global in this unit to pure external
1905                                         // (is this a good idea? what if the other one is a ref to
1906                                         // this one???)
1907                                         PutLong(sfix + 4, T_GLBL);
1908                                 }
1909                         }
1910                 }
1911
1912                 sfix += 12;                     // Increment symbol fixup pointer
1913                 nsymbols--;                     // Decrement num of symbols to process
1914         }
1915
1916         // Success loading symbols
1917         return 0;
1918 }
1919
1920
1921 //
1922 // Process object file for symbols
1923 //
1924 int DoItem(struct OFILE * obj)
1925 {
1926         // Allocate memory for object record ptr
1927         struct OFILE * Ofile = new_ofile();
1928
1929         if (Ofile == NULL)
1930         {
1931                 printf("Out of memory while processing %s\n", obj->o_name);
1932                 return 1;
1933         }
1934
1935         // Starting after all pathnames, etc., copy .o file name to Ofile
1936         char * temp = PathTail(obj->o_name);
1937
1938         // Check filename length
1939         if (strlen(temp) > FNLEN - 1)
1940         {
1941                 printf("File name too long: %s\n", temp);
1942                 return 1;
1943         }
1944
1945         // Check archive name length
1946         if (strlen(obj->o_arname) > (FNLEN - 1))
1947         {
1948                 printf("Archive name too long: %s\n", obj->o_arname);
1949                 return 1;
1950         }
1951
1952         strcpy(Ofile->o_name, temp);            // Store filename
1953         strcpy(Ofile->o_arname, obj->o_arname); // Store archive name
1954
1955         // Initialise object record information
1956         Ofile->o_next  = NULL;
1957         Ofile->o_tbase = 0;
1958         Ofile->o_dbase = 0;
1959         Ofile->o_bbase = 0;
1960         Ofile->o_flags = obj->o_flags;
1961         Ofile->o_image = obj->o_image;
1962         Ofile->isArchiveFile = obj->isArchiveFile;
1963         Ofile->segSize[TEXT] = obj->segSize[TEXT];
1964         Ofile->segSize[DATA] = obj->segSize[DATA];
1965         Ofile->segSize[BSS]  = obj->segSize[BSS];
1966         char * ptr = obj->o_image;
1967
1968         Ofile->o_header.magic = GetLong(ptr);
1969         Ofile->o_header.tsize = GetLong(ptr + 4);
1970         Ofile->o_header.dsize = GetLong(ptr + 8);
1971         Ofile->o_header.bsize = GetLong(ptr + 12);
1972         Ofile->o_header.ssize = GetLong(ptr + 16);
1973         Ofile->o_header.absrel.reloc.tsize = GetLong(ptr + 24);
1974         Ofile->o_header.absrel.reloc.dsize = GetLong(ptr + 28);
1975
1976         // Round BSS off to alignment boundary (??? isn't this already done ???)
1977         Ofile->o_header.bsize = (Ofile->o_header.bsize + secalign) & ~secalign;
1978
1979         if ((Ofile->o_header.dsize & 7) && wflag)
1980         {
1981                 printf("Warning: data segment size of ");
1982                 WriteARName(Ofile);
1983                 printf(" is not a phrase multiple\n");
1984         }
1985
1986         // Check for odd segment sizes
1987         if ((Ofile->o_header.tsize & 1) || (Ofile->o_header.dsize & 1)
1988                 || (Ofile->o_header.bsize & 1))
1989         {
1990                 printf("Error: odd-sized segment in ");
1991                 WriteARName(Ofile);
1992                 printf("; link aborted.\n");
1993                 return 1;
1994         }
1995
1996         if (AddSymbols(Ofile))
1997                 return 1;
1998
1999         // Add this file to the olist
2000         if (olist == NULL)
2001                 olist = Ofile;
2002         else
2003                 olast->o_next = Ofile;
2004
2005         olast = Ofile;
2006         return 0;
2007 }
2008
2009
2010 //
2011 // Handle items in processing list.
2012 //
2013 // After loading all objects, archives & include files, we now go and process
2014 // each item on the processing list (plist). Once this is done, we go through
2015 // any unresolved symbols left and see if they have shown up.
2016 //
2017 int ProcessLists(void)
2018 {
2019         // Process object file list first (adds symbols from each unit & creates
2020         // the olist)
2021         while (plist != NULL)
2022         {
2023                 if (DoItem(plist))
2024                         return 1;
2025
2026                 struct OFILE * ptemp = plist;
2027                 plist = plist->o_next;
2028                 free(ptemp);
2029         }
2030
2031         struct HREC * uptr;
2032
2033         // Process the unresolved symbols list. This may involve pulling in symbols
2034         // from any included .a units. Such units are lazy linked by default; we
2035         // generally don't want everything they provide, just what's referenced.
2036         for(uptr=unresolved; uptr!=NULL; )
2037         {
2038                 if (vflag > 1)
2039                         printf("LookupHREC(%s) => ", uptr->h_sym);
2040
2041                 struct HREC * htemp = LookupHREC(uptr->h_sym);
2042
2043                 if (htemp != NULL)
2044                 {
2045                         // Found it in the symbol table!
2046                         if (vflag > 1)
2047                                 printf("%s in %s (=$%06X)\n", (isglobal(htemp->h_type) ? "global" : "common"), htemp->h_ofile->o_name, htemp->h_value);
2048
2049                         // Mark the .o unit that the symbol is in as seen & remove from the
2050                         // unresolved list
2051                         htemp->h_ofile->o_flags |= O_USED;
2052                         uptr = RemoveUnresolvedSymbol(uptr);
2053                 }
2054                 else
2055                 {
2056                         if (vflag > 1)
2057                                 printf("NULL\n");
2058
2059                         // Check to see if the unresolved symbol is on the AR symbol list.
2060                         htemp = LookupARHREC(uptr->h_sym);
2061
2062                         // If the unresolved symbol is in a .o unit that is unused, we can
2063                         // drop it; same if the unresolved symbol is in the exported AR
2064                         // symbol list. Otherwise, go to the next unresolved symbol.
2065                         if (!(uptr->h_ofile->o_flags & O_USED) || (htemp != NULL))
2066                                 uptr = RemoveUnresolvedSymbol(uptr);
2067                         else
2068                                 uptr = uptr->h_next;
2069
2070                         // Now that we've possibly deleted the symbol from unresolved list
2071                         // that was also in the AR list, we add the symbols from this .o
2072                         // unit to the symbol table, mark the .o unit as used, and restart
2073                         // scanning the unresolved list as there is a good possibility that
2074                         // the symbols in the unit we're adding has unresolved symbols as
2075                         // well.
2076                         if (htemp != NULL)
2077                         {
2078                                 htemp->h_ofile->o_flags |= O_USED;
2079                                 AddSymbols(htemp->h_ofile);
2080                                 uptr = unresolved;
2081                         }
2082                 }
2083         }
2084
2085         // Show files used if the user requests it.
2086         if (vflag > 1)
2087         {
2088                 printf("Files used:\n");
2089                 struct OFILE * filePtr = olist;
2090
2091                 while (filePtr != NULL)
2092                 {
2093                         if (filePtr->o_flags & O_USED)
2094                         {
2095                                 printf("   %s%s%s\n", filePtr->o_name, (filePtr->isArchiveFile ? ":" : ""), (filePtr->isArchiveFile ? filePtr->o_arname : nullStr));
2096                         }
2097
2098                         filePtr = filePtr->o_next;
2099                 }
2100         }
2101
2102         return 0;
2103 }
2104
2105
2106 //
2107 // Extract filename from path
2108 //
2109 char * PathTail(char * name)
2110 {
2111         // Find last occurance of PATH_DELIMETER
2112         char * temp = strrchr(name, PATH_DELIMITER);
2113
2114         // Return what was passed in if path delimiter was not found
2115         if (temp == NULL)
2116                 return name;
2117
2118         return temp + 1;
2119 }
2120
2121
2122 //
2123 // Add input file to processing list
2124 //
2125 int AddToProcessingList(char * ptr, char * fname, char * arname, uint8_t arFile, uint32_t tSize, uint32_t dSize, uint32_t bSize)
2126 {
2127         if (plist == NULL)
2128         {
2129                 // First time object record allocation
2130                 plist = new_ofile();
2131                 plast = plist;
2132         }
2133         else
2134         {
2135                 // Next object record allocation
2136                 plast->o_next = new_ofile();
2137                 plast = plast->o_next;
2138         }
2139
2140         if (plast == NULL)
2141         {
2142                 printf("Out of memory.\n");             // Error if memory allocation fails
2143                 return 1;
2144         }
2145
2146         // Discard paths from filenames...
2147         fname = PathTail(fname);
2148         arname = PathTail(arname);
2149
2150         // Check for filename length errors...
2151         if (strlen(fname) > (FNLEN - 1))
2152         {
2153                 printf("File name too long: %s (sorry!)\n", fname);
2154                 return 1;
2155         }
2156
2157         if (strlen(arname) > (FNLEN - 1))
2158         {
2159                 printf("AR file name too long: %s (sorry!)\n", arname);
2160                 return 1;
2161         }
2162
2163         strcpy(plast->o_name, fname);           // Store filename sans path
2164         strcpy(plast->o_arname, arname);        // Store archive name sans path
2165         plast->o_image = ptr;                           // Store data pointer
2166         plast->o_flags = (arFile ? 0 : O_USED); // File is used if NOT in archive
2167         plast->o_next = NULL;                           // Initialise next record pointer
2168         plast->isArchiveFile = arFile;          // Shamus: Temp until can sort it out
2169         plast->segSize[TEXT] = tSize;
2170         plast->segSize[DATA] = dSize;
2171         plast->segSize[BSS]  = bSize;
2172
2173         return 0;                                                       // Return without errors
2174 }
2175
2176
2177 //
2178 // Process in binary include files and add them to the processing list. This
2179 // routine takes in the binary file and creates an 'object' file in memory.
2180 // Sym1/Sym2 point to the start and end of data.
2181 //
2182 // Image size for include files is:
2183 // Header ....... 32 bytes
2184 // Data ......... dsize
2185 // Sym fixups ... 2 * 12 bytes
2186 // Symbol size .. 4 bytes (Value to include symbols and terminating null)
2187 // Symbols ...... (strlen(sym1) + 1) + (strlen(sym2) + 1)
2188 // Terminate .... 4 bytes (0x00000000)
2189 //
2190 int LoadInclude(char * fname, int handle, char * sym1, char * sym2, int segment)
2191 {
2192         char * ptr, * sptr;
2193         int i;
2194         unsigned symtype = 0;
2195         uint32_t tSize = 0, dSize = 0, bSize = 0;
2196
2197         long fsize = FileSize(handle);          // Get size of include file
2198         long dsize = (fsize + secalign) & ~secalign;    // Align size to boundary
2199         int sym1len = strlen(sym1) + 1;         // Get sym1 length + null termination
2200         int sym2len = strlen(sym2) + 1;         // Get sym2 length + null termination
2201         long size = 32 + dsize + 24 + 4 + sym1len + sym2len + 4;
2202
2203         // Use calloc so the header & fixups initialize to zero
2204         // Allocate object image memory
2205         if ((ptr = calloc(size, 1)) == NULL)
2206         {
2207                 printf("Out of memory while including %s\n", fname);
2208                 close(handle);
2209                 return 1;
2210         }
2211
2212         // Read in binary data
2213         if (read(handle, ptr + 32, fsize) != fsize)
2214         {
2215                 printf("File read error on %s\n", fname);
2216                 close(handle);
2217                 free(ptr);
2218                 return 1;
2219         }
2220
2221         close(handle);
2222
2223         // Build this image's dummy header
2224         PutLong(ptr, 0x00000107);              // Magic number
2225
2226         if (segment)
2227         {
2228                 PutLong(ptr+4, dsize);             // Text size
2229                 PutLong(ptr+8, 0L);                // Data size
2230                 symtype = 0x05000000;
2231                 tSize = dsize;
2232         }
2233         else
2234         {
2235                 PutLong(ptr+4, 0L);                // Text size
2236                 PutLong(ptr+8, dsize);             // Data size
2237                 symtype = 0x07000000;
2238                 dSize = dsize;
2239         }
2240
2241         PutLong(ptr+12, 0L);                   // BSS size
2242         PutLong(ptr+16, 24);                   // Symbol table size
2243         PutLong(ptr+20, 0L);                   // Entry point
2244         PutLong(ptr+24, 0L);                   // TEXT relocation size
2245         PutLong(ptr+28, 0L);                   // DATA relocation size
2246
2247         sptr = ptr + 32 + dsize;               // Set sptr to symbol table location
2248
2249         PutLong(sptr,    4L);                  // String offset of symbol1
2250         PutLong(sptr+4,  symtype);             // Symbol type
2251         PutLong(sptr+8,  0x00000000);          // Symbol has no value (START)
2252         PutLong(sptr+12, 4L + (sym2len - 1));  // String offset of symbol2
2253         PutLong(sptr+16, symtype);             // Symbol type
2254         PutLong(sptr+20, dsize);               // Symbol is data size (END)
2255
2256         sptr = ptr + 32 + dsize + 24;          // Set sptr to symbol table size loc
2257
2258         PutLong(sptr, sym1len + 4L);           // Size of symbol table
2259
2260         sptr = ptr + 32 + dsize + 24 + 4;      // Set sptr to symbol table location
2261
2262         for(i=0; i<(sym1len-1); i++)           // Write symbol1 to string table
2263                 sptr[i] = *sym1++;
2264
2265         sptr += (sym1len - 1);                 // Step past symbol string
2266         *sptr = '\0';                          // Terminate symbol string
2267         sptr += 1;                             // Step past termination
2268
2269         for(i=0; i<(sym2len-1); i++)           // Write symbol2 to string table
2270                 sptr[i] = *sym2++;
2271
2272         sptr += (sym2len - 1);                 // Step past symbol string
2273         *sptr = '\0';                          // Terminate symbol string
2274         sptr += 1;                             // Step past termination
2275
2276         PutLong(sptr, 0L);                     // Terminating long for object file
2277
2278         return AddToProcessingList(ptr, fname, nullStr, 0, tSize, dSize, bSize);
2279 }
2280
2281
2282 //
2283 // Takes a file name, gets in its image, puts it on plist. The image may
2284 // already be in memory: If so, the ptr arg is non-null.  If so, the file is
2285 // already closed. Note that the file is already open (from DoFile()). RETURNS
2286 // a pointer to the OFILE structure for this file, so you can diddle its flags
2287 // (DoFile sets O_USED for files on the command line).
2288 //
2289 int LoadObject(char * fname, int fd, char * ptr)
2290 {
2291         uint32_t tSize = 0, dSize = 0, bSize = 0;
2292
2293         if (ptr == NULL)
2294         {
2295                 long size = FileSize(fd);
2296
2297                 // Allocate memory for file data
2298                 ptr = malloc(size);
2299
2300                 if (ptr == NULL)
2301                 {
2302                         printf("Out of memory while processing %s\n", fname);
2303                         close(fd);
2304                         return 1;
2305                 }
2306
2307                 // Read in file data
2308                 if (read(fd, ptr, size) != size)
2309                 {
2310                         printf("File read error on %s\n", fname);
2311                         close(fd);
2312                         free(ptr);
2313                         return 1;
2314                 }
2315
2316                 tSize = (GetLong(ptr + 4)  + secalign) & ~secalign;
2317                 dSize = (GetLong(ptr + 8)  + secalign) & ~secalign;
2318                 bSize = (GetLong(ptr + 12) + secalign) & ~secalign;
2319                 close(fd);
2320         }
2321
2322         // Now add this image to the list of pending ofiles (plist)
2323         return AddToProcessingList(ptr, fname, nullStr, 0, tSize, dSize, bSize);
2324 }
2325
2326
2327 //
2328 // What it says on the tin: check for a .o suffix on the passed in string
2329 //
2330 uint8_t HasDotOSuffix(char * s)
2331 {
2332         char * temp = strrchr(s, '.');
2333
2334         if ((temp == NULL) || (strncmp(temp, ".o", 2) != 0))
2335                 return 0;
2336
2337         return 1;
2338 }
2339
2340
2341 //
2342 // Process an ar archive file (*.a)
2343 //
2344 int LoadArchive(char * fname, int fd)
2345 {
2346         // Read in the archive file to memory and process
2347         long size = FileSize(fd);
2348         char * ptr = malloc(size);
2349         char * endPtr = ptr + size;
2350         char * longFilenames = NULL;
2351
2352         if (ptr == NULL)
2353         {
2354                 printf("Out of memory while processing %s\n", fname);
2355                 close(fd);
2356                 return 1;
2357         }
2358
2359         if (read(fd, ptr, size) != size)
2360         {
2361                 printf("File read error on %s\n", fname);
2362                 close(fd);
2363                 free(ptr);
2364                 return 1;
2365         }
2366
2367         close(fd);
2368
2369         // Save the pointer for later...
2370         arPtr[arIndex++] = ptr;
2371         char objName[FNLEN];
2372         char objSize[11];
2373         int i;
2374 //printf("\nProcessing AR file \"%s\"...\n", fname);
2375         ptr += 8;
2376
2377         // Loop through all objects in the archive and process them
2378         do
2379         {
2380                 memset(objName, 0, 17);
2381                 objSize[10] = 0;
2382
2383                 for(i=0; i<16; i++)
2384                 {
2385 //                      if ((ptr[i] == '/') || (ptr[i] == ' '))
2386                         if ((ptr[i] == ' ') && (i != 0))
2387                         {
2388                                 objName[i] = 0;
2389                                 break;
2390                         }
2391
2392                         objName[i] = ptr[i];
2393                 }
2394
2395                 for(i=0; i<10; i++)
2396                 {
2397                         if (ptr[48 + i] == ' ')
2398                         {
2399                                 objSize[i] = 0;
2400                                 break;
2401                         }
2402
2403                         objSize[i] = ptr[48 + i];
2404                 }
2405
2406                 // Check to see if a long filename was requested
2407                 if (objName[0] == 0x20)
2408                 {
2409                         uint32_t fnSize = atoi(objName + 1);
2410
2411                         if (longFilenames != NULL)
2412                         {
2413                                 i = 0;
2414                                 char * currentFilename = longFilenames + fnSize;
2415
2416                                 while (*currentFilename != 0x0A)
2417                                         objName[i++] = *currentFilename++;
2418
2419                                 objName[i] = 0;
2420                         }
2421                 }
2422
2423                 if ((strncmp(objName, "ARFILENAMES/", 12) == 0) || (strncmp(objName, "//", 2) == 0))
2424                 {
2425                         longFilenames = ptr + 60;
2426                 }
2427                 else if (HasDotOSuffix(objName))
2428                 {
2429
2430                         // Strip off any trailing forward slash at end of object name
2431                         int lastChar = strlen(objName) - 1;
2432
2433                         if (objName[lastChar] == '/')
2434                                 objName[lastChar] = 0;
2435
2436 //printf("Processing object \"%s\" (size == %i, obj_index == %i)...\n", objName, atoi(objSize), obj_index);
2437                         uint32_t tSize = (GetLong(ptr + 60 + 4)  + secalign) & ~secalign;
2438                         uint32_t dSize = (GetLong(ptr + 60 + 8)  + secalign) & ~secalign;
2439                         uint32_t bSize = (GetLong(ptr + 60 + 12) + secalign) & ~secalign;
2440
2441                         if (AddToProcessingList(ptr + 60, objName, fname, 1, tSize, dSize, bSize))
2442                                 return 1;
2443                 }
2444
2445                 uint32_t size = atoi(objSize);
2446                 size += (size & 0x01 ? 1 : 0);
2447                 ptr += 60 + size;
2448         }
2449         while (ptr < endPtr);
2450
2451         return 0;
2452 }
2453
2454
2455 //
2456 // Process files (*.o, *.a) passed in on the command line
2457 //
2458 int ProcessFiles(void)
2459 {
2460         int i;
2461         char magic[8];          // Magic header number (4 bytes for *.o, 8 for *.a)
2462
2463         // Process all file handles
2464         for(i=0; i<(int)hd; i++)
2465         {
2466                 // Verbose mode information
2467                 if (vflag == 1)
2468                         printf("Read file %s%s\n", name[i], (hflag[i] ? " (include)" : ""));
2469
2470                 if (!hflag[i])
2471                 {
2472                         // Attempt to read file magic number (OBJECT/ARCHIVE FILES)
2473                         if (read(handle[i], magic, 8) != 8)
2474                         {
2475                                 printf("Error reading file %s\n", name[i]);
2476                                 close(handle[i]);
2477                                 return 1;
2478                         }
2479
2480                         lseek(handle[i], 0L, 0);        // Reset to start of input file
2481
2482                         // Look for RMAC/MAC/GCC (a.out) object files
2483                         if ((GetLong(magic) & 0xFFFF) == 0x0107)
2484                         {
2485                                 // Process input object file
2486                                 if (LoadObject(name[i], handle[i], 0L))
2487                                         return 1;
2488                         }
2489                         // Otherwise, look for an object archive file
2490                         else if (strncmp(magic, "!<arch>\x0A", 8) == 0)
2491                         {
2492                                 if (LoadArchive(name[i], handle[i]))
2493                                         return 1;
2494                         }
2495                         else
2496                         {
2497                                 // Close file and error
2498                                 printf("%s is not a supported object or archive file\n", name[i]);
2499                                 printf("Magic == [%02X][%02X][%02X][%02X]\n", magic[0], magic[1], magic[2], magic[3]);
2500                                 close(handle[i]);
2501                                 return 1;
2502                         }
2503                 }
2504                 else
2505                 {
2506                         // INCLUDE FILES
2507                         // If hflag[i] is 1, include this in the data segment; if 2, put it
2508                         // in text segment
2509                         if (LoadInclude(name[i], handle[i], hsym1[i], hsym2[i], hflag[i] - 1))
2510                                 return 1;
2511                 }
2512         }
2513
2514         // Free include, symbol & object handles
2515         for(i=0; i<(int)hd; i++)
2516         {
2517                 free(name[i]);
2518
2519                 if (hflag[i])
2520                 {
2521                         free(hsym1[i]);
2522                         free(hsym2[i]);
2523                 }
2524         }
2525
2526         // Reset next handle indicator
2527         hd = 0;
2528         return 0;
2529 }
2530
2531
2532 //
2533 // Load newargv with pointers to arguments found in the buffer
2534 //
2535 int parse(char * buf, char * newargv[])
2536 {
2537         int i = 1;
2538
2539         if (vflag)
2540                 printf("begin parsing\n");
2541
2542         while (1)
2543         {
2544                 while (*buf && strchr(",\t\n\r\14 ", *buf))
2545                         buf++;
2546
2547                 /* test for eof */
2548                 if (*buf == '\0' || *buf == 26)
2549                 {
2550                         if (i == 0)
2551                         {
2552                                 printf("No commands in command file\n");
2553                                 return -1;
2554                         }
2555                         else
2556                         {
2557                                 return i;
2558                         }
2559                 }
2560
2561                 /* test for comment */
2562                 if (*buf == '#')
2563                 {
2564                         /* found a comment; skip to next \n and start over */
2565                         while (*buf && *buf != '\n')
2566                                 buf++;
2567
2568                         continue;
2569                 }
2570
2571                 if (i == MAXARGS)
2572                 {
2573                         printf("Too many arguments in command file\n");
2574                         return -1;
2575                 }
2576
2577                 newargv[i] = buf;
2578
2579                 while (!strchr(",\t\n\r\14 ", *buf))
2580                 {
2581                         if (*buf == '\0' || *buf == 26)
2582                         {
2583                                 printf("Finished parsing %d args\n", i);
2584                                 return i;
2585                         }
2586
2587                         buf++;
2588                 }
2589
2590                 *buf++ = '\0';
2591
2592                 if (vflag)
2593                         printf("argv[%d] = \"%s\"\n", i, newargv[i]);
2594
2595                 i++;
2596         }
2597 }
2598
2599
2600 //
2601 // Process in a link command file
2602 //
2603 int docmdfile(char * fname)
2604 {
2605         int fd;                                     // File descriptor
2606         unsigned size;                              // Command file size
2607         char * ptr;                                 // Pointer
2608         int newargc;                                // New argument count
2609         char * (*newargv)[];                        // New argument value array
2610
2611         // Verbose information
2612         if (vflag > 1)
2613                 printf("docmdfile(%s)\n", fname);
2614
2615         // Allocate memory for new argument values
2616         newargv = malloc((long)(sizeof(char *) * MAXARGS));
2617
2618         if (!newargv)
2619         {
2620                 printf("Out of memory.\n");
2621                 return 1;
2622         }
2623
2624         // Attempt to open and read in the command file
2625         if (fname)
2626         {
2627                 if ((fd = open(fname, _OPEN_FLAGS)) < 0)
2628                 {
2629                         printf("Cannot open command file %s.\n", fname);
2630                         return 1;
2631                 }
2632
2633                 size = FileSize(fd);
2634
2635                 if ((ptr = malloc(size + 1)) == NULL)
2636                 {
2637                         printf("Out of memory.\n");
2638                         close(fd);
2639                         return 1;
2640                 }
2641
2642                 if (read(fd, ptr, size) != (int)size)
2643                 {
2644                         printf("Read error on command file %s.\n", fname);
2645                         close(fd);
2646                         return 1;
2647                 }
2648
2649                 *(ptr + size) = 0;                      // Null terminate the buffer
2650                 close(fd);
2651         }
2652         else
2653         {
2654                 printf("No command filename specified\n");
2655                 return 1;
2656         }
2657
2658         // Parse the command file
2659         if ((newargc = parse(ptr, *newargv)) == -1)
2660         {
2661                 return 1;
2662         }
2663
2664         // Process the inputted flags
2665         if (doargs(newargc, *newargv))
2666         {
2667                 printf("docmdfile: doargs returns TRUE\n");
2668                 return 1;
2669         }
2670
2671         free(ptr);
2672         free(newargv);
2673
2674         return 0;
2675 }
2676
2677
2678 //
2679 // Take an argument list and parse the command line
2680 //
2681 int doargs(int argc, char * argv[])
2682 {
2683         int i = 1;                                      // Iterator
2684         int c;                                          // Command line character
2685         char * ifile, * isym;           // File name and symbol name for -i
2686
2687         // Parse through option switches & files
2688         while (i < argc)
2689         {
2690                 // Process command line switches
2691                 if (argv[i][0] == '-')
2692                 {
2693                         if (!argv[i][1])
2694                         {
2695                                 printf("Illegal option argument: %s\n\n", argv[i]);
2696                                 ShowHelp();
2697                                 return 1;
2698                         }
2699
2700                         c = argv[i++][1];                       // Get next character in command line
2701
2702                         // Process command line switch
2703                         switch (c)
2704                         {
2705                         case '?':                                       // Display usage information
2706                         case 'h':
2707                         case 'H':
2708                                 ShowVersion();
2709                                 ShowHelp();
2710                                 return 1;
2711                         case 'a':
2712                         case 'A':                                       // Set absolute linking on
2713                                 if (aflag)
2714                                         warn('a', 1);
2715
2716                                 if (i + 2 >= argc)
2717                                 {
2718                                         printf("Not enough arguments to -a\n");
2719                                         return 1;
2720                                 }
2721
2722                                 aflag = 1;                              // Set abs link flag
2723
2724                                 // Segment order is TEXT, DATA, BSS
2725                                 // Text segment can be 'r' or a value
2726                                 if ((*argv[i] == 'r' || *argv[i] == 'R') && !argv[i][1])
2727                                 {
2728                                         ttype = -1;                     // TEXT segment is relocatable
2729                                 }
2730                                 else if ((*argv[i] == 'x' || *argv[i] == 'X'))
2731                                 {
2732                                         printf("Error in text-segment address: cannot be contiguous\n");
2733                                         return 1;
2734                                 }
2735                                 else if (GetHexValue(argv[i], &tval))
2736                                 {
2737                                         printf("Error in text-segment address: %s is not 'r' or an address.", argv[i]);
2738                                         return 1;
2739                                 }
2740
2741                                 i++;
2742
2743                                 // Data segment can be 'r', 'x' or a value
2744                                 if ((*argv[i] == 'r' || *argv[i] == 'R') && !argv[i][1])
2745                                 {
2746                                         dtype = -1;                     // DATA segment is relocatable
2747                                 }
2748                                 else if ((*argv[i] == 'x' || *argv[i] == 'X'))
2749                                 {
2750                                         dtype = -2;                     // DATA follows TEXT
2751                                 }
2752                                 else if (GetHexValue(argv[i], &dval))
2753                                 {
2754                                         printf("Error in data-segment address: %s is not 'r', 'x' or an address.", argv[i]);
2755                                         return 1;
2756                                 }
2757
2758                                 i++;
2759
2760                                 // BSS segment can be 'r', 'x' or a value
2761                                 if ((*argv[i] == 'r' || *argv[i] == 'R') && !argv[i][1])
2762                                 {
2763                                         btype = -1;                     // BSS segment is relocatable
2764                                 }
2765                                 else if ((*argv[i] == 'x' || *argv[i] == 'X'))
2766                                 {
2767                                         btype = -3;                     // BSS follows DATA
2768                                 }
2769                                 else if (GetHexValue(argv[i], &bval))
2770                                 {
2771                                         printf("Error in bss-segment address: %s is not 'r', 'x[td]', or an address.", argv[i]);
2772                                         return 1;
2773                                 }
2774
2775                                 i++;
2776                                 break;
2777                         case 'b':
2778                         case 'B':                                       // Don't remove muliply defined locals
2779                                 if (bflag)
2780                                         warn('b', 1);
2781
2782                                 bflag = 1;
2783                                 break;
2784                         case 'c':
2785                         case 'C':                                       // Process a command file
2786                                 if (i == argc)
2787                                 {
2788                                         printf("Not enough arguments to -c\n");
2789                                         return 1;
2790                                 }
2791
2792                                 if (docmdfile(argv[i++]))
2793                                 {
2794                                         return 1;
2795                                 }
2796
2797                                 break;
2798                         case 'd':
2799                         case 'D':                                       // Wait for "return" before exiting
2800                                 if (dflag)
2801                                         warn('d', 0);
2802
2803                                 dflag = 1;
2804                                 waitflag = 1;
2805                                 break;
2806                         case 'e':
2807                         case 'E':                                       // Output COFF (absolute only)
2808                                 cflag = 1;
2809                                 break;
2810                         case 'g':
2811                         case 'G':                                       // Output source level debugging
2812                                 printf("\'g\' flag not currently implemented\n");
2813                                 gflag = 0;
2814                                 /*
2815                                 if (gflag) warn('g', 1);
2816                                 gflag = 1;
2817                                 */
2818                                 break;
2819                         case 'i':
2820                         case 'I':                                       // Include binary file
2821                                 if (i + 2 > argc)
2822                                 {
2823                                         printf("Not enough arguments to -i\n");
2824                                         return 1;
2825                                 }
2826
2827                                 ifile = argv[i++];
2828                                 isym = argv[i++];
2829
2830                                 // handle -ii (No truncation)
2831                                 if ((argv[i-3][2] == 'i') || (argv[i-3][2] == 'I'))
2832                                 {
2833                                         if (!cflag)
2834                                                 printf("warning: (-ii) COFF format output not specified\n");
2835                                 }
2836                                 // handle -i (Truncation)
2837                                 else
2838                                 {
2839                                         if (strlen(isym) > 8)
2840                                                 isym[8] = '\0';
2841                                 }
2842
2843                                 // Place include files in the DATA segment only
2844                                 if (DoFile(ifile, DSTSEG_D, isym))
2845                                         return 1;
2846
2847                                 break;
2848                         case 'l':
2849                         case 'L':                                       // Add local symbols
2850                                 if (lflag)
2851                                         warn('l', 1);
2852
2853                                 lflag = 1;
2854                                 break;
2855                         case 'm':
2856                         case 'M':                                       // Produce load symbol map
2857                                 if (mflag)
2858                                         warn('m', 1);
2859
2860                                 mflag = 1;
2861                                 break;
2862                         case 'n':
2863                         case 'N':                                       // Output no header to .abs file
2864                                 if (noheaderflag)
2865                                         warn('n', 1);
2866
2867                                 noheaderflag = 1;
2868                                 break;
2869                         case 'o':
2870                         case 'O':                                       // Specify an output file
2871                                 if (oflag)
2872                                         warn('o', 1);
2873
2874                                 oflag = 1;
2875
2876                                 if (i >= argc)
2877                                 {
2878                                         printf("No output filename following -o switch\n");
2879                                         return 1;
2880                                 }
2881
2882                                 if (strlen(argv[i]) > FARGSIZE - 5)
2883                                 {
2884                                         printf("Output file name too long (sorry!)\n");
2885                                         return 1;
2886                                 }
2887
2888                                 strcpy(ofile, argv[i++]);
2889                                 break;
2890                         case 'r':
2891                         case 'R':                                       // Section alignment size
2892                                 if (rflag)
2893                                         warn('r', 1);
2894
2895                                 rflag = 1;
2896
2897                                 switch (argv[i-1][2])
2898                                 {
2899                                         case 'w': case 'W': secalign = 1;  break; // Word alignment
2900                                         case 'l': case 'L': secalign = 3;  break; // Long alignment
2901                                         case 'p': case 'P': secalign = 7;  break; // Phrase alignment
2902                                         case 'd': case 'D': secalign = 15; break; // Double phrase alignment
2903                                         case 'q': case 'Q': secalign = 31; break; // Quad phrase alignment
2904                                         default:            secalign = 7;  break; // Default phrase alignment
2905                                 }
2906
2907                                 break;
2908                         case 's':
2909                         case 'S':                                       // Output only global symbols
2910                                 if (sflag)
2911                                         warn('s', 1);
2912
2913                                 sflag = 1;
2914                                 break;
2915                         case 'u':
2916                         case 'U':                                       // Undefined symbols
2917                                 uflag++;
2918                                 break;
2919                         case 'v':
2920                         case 'V':                                       // Verbose information
2921                                 if (!vflag && !versflag)
2922                                 {
2923                                         ShowVersion();
2924                                 }
2925
2926                                 vflag++;
2927                                 break;
2928                         case 'w':
2929                         case 'W':                                       // Show warnings flag
2930                                 if (wflag)
2931                                         warn('w', 1);
2932
2933                                 wflag = 1;
2934                                 break;
2935                         case 'z':
2936                         case 'Z':                                       // Suppress banner flag
2937                                 if (zflag)
2938                                         warn('z', 1);
2939
2940                                 zflag = 1;
2941                                 break;
2942                         default:
2943                                 printf("unknown option argument `%c'\n", c);
2944                                 return 1;
2945                         }
2946                 }
2947                 else
2948                 {
2949                         // Not a switch, then process as a file
2950                         if (DoFile(argv[i++], 0, NULL))
2951                                 return 1;
2952                 }
2953         }
2954
2955         if (!oflag && vflag)
2956         {
2957                 strcpy(ofile, "output");
2958                 printf("Output file is %s[.ext]\n", ofile);
2959         }
2960
2961         if (oflag && vflag)
2962                 printf("Output file is %s\n", ofile);
2963
2964         if (sflag)
2965                 lflag = 0;
2966
2967         // No problems encountered
2968         return 0;
2969 }
2970
2971
2972 //
2973 // Display version information
2974 //
2975 void ShowVersion(void)
2976 {
2977         if (displaybanner)// && vflag)
2978         {
2979                 printf(
2980                 "      _\n"
2981                 " _ __| |_ ___\n"
2982                 "| '__| | '_  \\\n"
2983                 "| |  | | | | |\n"
2984                 "|_|  |_|_| |_|\n"
2985                 "\nReboot's Linker for Atari Jaguar\n"
2986                 "Copyright (c) 199x Allan K. Pratt, 2014-2015 Reboot\n"
2987                 "V%i.%i.%i %s (%s)\n\n", MAJOR, MINOR, PATCH, __DATE__, PLATFORM);
2988         }
2989 }
2990
2991
2992 //
2993 // Display command line help
2994 //
2995 void ShowHelp(void)
2996 {
2997         printf("Usage:\n");
2998         printf("    %s [-options] file(s)\n", cmdlnexec);
2999         printf("\n");
3000         printf("Options:\n");
3001         printf("   -? or -h                display usage information\n");
3002         printf("   -a <text> <data> <bss>  output absolute file\n");
3003         printf("                           hex value: segment address\n");
3004         printf("                           r: relocatable segment\n");
3005         printf("                           x: contiguous segment\n");
3006         printf("   -b                      don't remove multiply defined local labels\n");
3007         printf("   -c <fname>              add contents of <fname> to command line\n");
3008         printf("   -d                      wait for key after link\n");
3009         printf("   -e                      output COF absolute file\n");
3010         printf("   -g                      output source-level debugging\n");
3011         printf("   -i <fname> <label>      incbin <fname> and set <label>\n");
3012         printf("   -l                      add local symbols\n");
3013         printf("   -m                      produce load symbols map\n");
3014         printf("   -n                      output no file header to .abs file\n");
3015         printf("   -o <fname>              set output name\n");
3016         printf("   -r<size>                section alignment size\n");
3017         printf("                           w: word (2 bytes)\n");
3018         printf("                           l: long (4 bytes)\n");
3019         printf("                           p: phrase (8 bytes, default alignment)\n");
3020         printf("                           d: double phrase (16 bytes)\n");
3021         printf("                           q: quad phrase (32 bytes)\n");
3022         printf("   -s                      output only global symbols\n");
3023         printf("   -u                      allow unresolved symbols (experimental)\n");
3024         printf("   -v                      set verbose mode\n");
3025         printf("   -w                      show linker warnings\n");
3026         printf("   -z                      suppress banner\n");
3027         printf("\n");
3028 }
3029
3030
3031 //
3032 // Application exit
3033 //
3034 void ExitLinker(void)
3035 {
3036         char tempbuf[128];
3037
3038         // Display link status if verbose mode
3039         if (vflag)
3040                 printf("Link %s.\n", errflag ? "aborted" : "complete");
3041
3042         // Wait for return key if requested
3043         if (waitflag)
3044         {
3045                 printf("\nPress the [RETURN] key to continue. ");
3046                 char * c = fgets(tempbuf, 128, stdin);
3047         }
3048
3049         exit(errflag);
3050 }
3051
3052
3053 int main(int argc, char * argv[])
3054 {
3055         cmdlnexec = argv[0];                    // Obtain executable name
3056         char * s = getenv("RLNPATH");   // Attempt to obtain env variable
3057
3058         if (s)
3059                 strcpy(libdir, s);                      // Store it if found
3060
3061         // Initialize some vars
3062         tval = dval = bval = 0;
3063         ttype = dtype = btype = 0;
3064
3065         // Parse the command line
3066         if (doargs(argc, argv))
3067         {
3068                 errflag = 1;
3069                 ExitLinker();
3070         }
3071
3072         if (!zflag && !vflag)
3073         {
3074                 ShowVersion();                          // Display version information
3075                 versflag = 1;                           // We've dumped the version banner
3076         }
3077
3078         // Load in specified files/objects and add to processing list
3079         if (ProcessFiles())
3080         {
3081                 errflag = 1;
3082                 ExitLinker();
3083         }
3084
3085         // Work in items in processing list & deal with unresolved list
3086         if (ProcessLists())
3087         {
3088                 errflag = 1;
3089                 ExitLinker();
3090         }
3091
3092         // Check that there is something to link
3093         if (olist == NULL)
3094         {
3095                 ShowHelp();
3096                 ExitLinker();
3097         }
3098
3099         // Report unresolved externals
3100         if (unresolved != NULL)
3101         {
3102                 printf("UNRESOLVED SYMBOLS\n");
3103
3104                 // Don't list them if two -u's or more
3105                 if (uflag < 2)
3106                 {
3107                         struct HREC * utemp = unresolved;
3108
3109                         while (utemp != NULL)
3110                         {
3111                                 printf("\t%s (", utemp->h_sym);
3112                                 WriteARName(utemp->h_ofile);
3113                                 printf(")\n");
3114                                 utemp = utemp->h_next;
3115                         }
3116                 }
3117
3118                 if (!uflag)
3119                 {
3120                         errflag = 1;
3121                         ExitLinker();
3122                 }
3123         }
3124
3125         // Make one output object from input objects
3126         struct OHEADER * header = MakeOutputObject();
3127
3128         if (header == NULL)
3129         {
3130                 errflag = 1;
3131                 ExitLinker();
3132         }
3133
3134         // Partial linking
3135         if (pflag)
3136         {
3137                 printf("TO DO: Partial linking\n");
3138                 errflag = 1;
3139         }
3140         // Relocatable linking
3141         else if (!aflag)
3142         {
3143                 printf("TO DO: Relocatable linking\n");
3144                 errflag = 1;
3145         }
3146         // Absolute linking
3147         else
3148         {
3149                 if (vflag)
3150                         printf("Absolute linking (%s)\n", (cflag ? "COF" : "ABS"));
3151
3152                 if (vflag > 1)
3153                         printf("Header magic is 0x%04X\n", (unsigned int)header->magic);
3154
3155                 if (WriteOutputFile(header))
3156                         errflag = 1;
3157         }
3158
3159         // Display the loaded symbols map
3160         if (mflag)
3161                 if (ShowSymbolLoadMap(header))
3162                         errflag = 1;
3163
3164         // Display segment size summary
3165         if (vflag)
3166         {
3167                 printf("\n");
3168                 printf("+---------+----------+----------+----------+\n");
3169                 printf("| Segment |     TEXT |     DATA |      BSS |\n");
3170                 printf("| Sizes   |----------+----------+----------|\n");
3171                 printf("| (Hex)   | %8X | %8X | %8X |\n", (unsigned int)header->tsize, (unsigned int)header->dsize, (unsigned int)header->bsize);
3172                 printf("+---------+----------+----------+----------+\n\n");
3173         }
3174
3175         free(header);
3176         ExitLinker();
3177 }
3178