Shamusworld >> Repos - rmac/blob - direct.c

   1 //
   2 // RMAC - Reboot's Macro Assembler for the Atari Jaguar Console System
   3 // DIRECT.C - Directive Handling
   4 // Copyright (C) 199x Landon Dyer, 2017 Reboot and Friends
   5 // RMAC derived from MADMAC v1.07 Written by Landon Dyer, 1986
   6 // Source utilised with the kind permission of Landon Dyer
   7 //
   8
   9 #include "direct.h"
  10 #include "sect.h"
  11 #include "riscasm.h"
  12 #include "error.h"
  13 #include "token.h"
  14 #include "procln.h"
  15 #include "expr.h"
  16 #include "mach.h"
  17 #include "listing.h"
  18 #include "mark.h"
  19 #include "symbol.h"
  20
  21 #define DEF_KW
  22 #include "kwtab.h"
  23
  24 TOKEN exprbuf[128];                     // Expression buffer
  25 SYM * symbolPtr[1000000];       // Symbol pointers table
  26 static long unused;                     // For supressing 'write' warnings
  27 char buffer[256];                       // Scratch buffer for messages
  28
  29
  30 // Directive handler table
  31 int (*dirtab[])() = {
  32         d_org,                          // 0 org
  33         d_even,                         // 1 even
  34         d_6502,                     // 2 .6502
  35         d_68000,                        // 3 .68000
  36         d_bss,                          // 4 bss
  37         d_data,                         // 5 data
  38         d_text,                         // 6 text
  39         d_abs,                          // 7 abs
  40         d_comm,                         // 8 comm
  41         (void *)d_init,         // 9 init
  42         d_cargs,                        // 10 cargs
  43         (void *)d_goto,         // 11 goto
  44         (void *)d_dc,           // 12 dc
  45         (void *)d_ds,           // 13 ds
  46         d_undmac,                       // 14 undefmac
  47         d_gpu,                          // 15 .gpu
  48         d_dsp,                          // 16 .dsp
  49         (void *)d_dcb,          // 17 dcb
  50         d_unimpl,                       // 18* set
  51         d_unimpl,                       // 19* reg
  52         d_unimpl,                       // 20 dump
  53         d_incbin,                       // 21 .incbin //load
  54         d_unimpl,                       // 22 disable
  55         d_unimpl,                       // 23 enable
  56         d_globl,                        // 24 globl
  57         d_regbank0,                     // 25 .regbank0
  58         d_regbank1,                     // 26 .regbank1
  59         d_unimpl,                       // 27 xdef
  60         d_assert,                       // 28 assert
  61         d_unimpl,                       // 29* if
  62         d_unimpl,                       // 30* endif
  63         d_unimpl,                       // 31* endc
  64         d_unimpl,                       // 32* iif
  65         d_include,                      // 33 include
  66         fpop,                           // 34 end
  67         d_unimpl,                       // 35* macro
  68         ExitMacro,                      // 36* exitm
  69         d_unimpl,                       // 37* endm
  70         d_list,                         // 38 list
  71         d_nlist,                        // 39 nlist
  72         d_long,                         // 40* rept
  73         d_phrase,                       // 41* endr
  74         d_dphrase,                      // 42 struct
  75         d_qphrase,                      // 43 ends
  76         d_title,                        // 44 title
  77         d_subttl,                       // 45 subttl
  78         eject,                          // 46 eject
  79         d_error,                        // 47 error
  80         d_warn,                         // 48 warn
  81         d_noclear,                      // 49 .noclear
  82         d_equrundef,            // 50 .equrundef/.regundef
  83         d_ccundef,                      // 51 .ccundef
  84         d_print,                        // 52 .print
  85         d_cstruct,                      // 53 .cstruct
  86         d_jpad,                         // 54 .jpad (deprecated)
  87         d_nojpad,                       // 55 .nojpad (deprecated)
  88         d_gpumain,                      // 56 .gpumain (deprecated)
  89         d_prgflags,                     // 57 .prgflags
  90         d_opt,                          // 58 .opt
  91 };
  92
  93
  94 //
  95 // .error - Abort compilation, printing an error message
  96 //
  97 int d_error(char *str)
  98 {
  99         if (*tok == EOL)
 100                 return error("error directive encountered - aborting assembling");
 101         else
 102         {
 103                 switch(*tok)
 104                 {
 105                 case STRING:
 106                         return error(string[tok[1]]);
 107                         break;
 108                 default:
 109                         return error("error directive encountered - aborting assembling");
 110                 }
 111         }
 112 }
 113
 114
 115 //
 116 // .warn - Just display a warning on screen
 117 //
 118 int d_warn(char *str)
 119 {
 120         if (*tok == EOL)
 121                 return warn("WARNING WARNING WARNING");
 122         else
 123         {
 124                 switch(*tok)
 125                 {
 126                 case STRING:
 127                         return warn(string[tok[1]]);
 128                         break;
 129                 default:
 130                         return warn("WARNING WARNING WARNING");
 131                 }
 132         }
 133 }
 134
 135
 136 //
 137 // .org - Set origin
 138 //
 139 int d_org(void)
 140 {
 141     VALUE address;
 142
 143     if (!rgpu && !rdsp && !m6502)
 144         return error(".org permitted only in gpu/dsp and 6502 sections");
 145
 146     if (rgpu | rdsp)
 147     {
 148         orgaddr = 0;
 149
 150         if (abs_expr(&address) == ERROR)
 151         {
 152             error("cannot determine org'd address");
 153             return ERROR;
 154         }
 155
 156         orgaddr = address;
 157         orgactive = 1;
 158         //printf("ORG: address = $%08X...\n", orgaddr);
 159
 160         return 0;
 161     }
 162     else
 163     {
 164         // 6502
 165         // We also kludge `lsloc' so the listing generator doesn't
 166         // try to spew out megabytes.
 167             VALUE eval;
 168
 169             //if (m6502 == 0) return error(".org permitted only in .6502 section");
 170             if (abs_expr(&eval) < 0) return 0;
 171             if (eval >= 0x10000L) return error(range_error);
 172
 173         if (sloc != currentorg[0])
 174         {
 175             currentorg[1] = sloc;
 176             currentorg += 2;
 177         }
 178
 179         currentorg[0] = eval;
 180             ch_size = 0;
 181             lsloc = sloc = eval;
 182             chptr = scode->chptr + eval;
 183         orgactive = 1;
 184         orgaddr = eval;
 185             at_eol();
 186         return 0;
 187     }
 188 }
 189
 190
 191 //
 192 // Print directive
 193 //
 194 int d_print(void)
 195 {
 196         char prntstr[LNSIZ];            // String for PRINT directive
 197         char format[LNSIZ];                     // Format for PRINT directive
 198         int formatting = 0;                     // Formatting on/off
 199         int wordlong = 0;                       // WORD = 0, LONG = 1
 200         int outtype = 0;                        // 0:hex, 1:decimal, 2:unsigned
 201
 202         VALUE eval;                                     // Expression value
 203         WORD eattr;                                     // Expression attributes
 204         SYM * esym;                                     // External symbol involved in expr.
 205         TOKEN r_expr[EXPRSIZE];
 206
 207         while (*tok != EOL)
 208         {
 209                 switch(*tok)
 210                 {
 211                 case STRING:
 212                         sprintf(prntstr, "%s", string[tok[1]]);
 213                         printf("%s", prntstr);
 214
 215                         if (list_fd)
 216                                 unused = write(list_fd, prntstr, (LONG)strlen(prntstr));
 217
 218                         tok += 2;
 219                         break;
 220                 case '/':
 221                         formatting = 1;
 222
 223                         if (tok[1] != SYMBOL)
 224                                 goto token_err;
 225
 226 //                      strcpy(prntstr, (char *)tok[2]);
 227                         strcpy(prntstr, string[tok[2]]);
 228
 229                         switch(prntstr[0])
 230                         {
 231                         case 'l': case 'L': wordlong = 1; break;
 232                         case 'w': case 'W': wordlong = 0; break;
 233                         case 'x': case 'X': outtype  = 0; break;
 234                         case 'd': case 'D': outtype  = 1; break;
 235                         case 'u': case 'U': outtype  = 2; break;
 236                         default:
 237                                 error("unknown print format flag");
 238                                 return ERROR;
 239                         }
 240
 241                         tok += 3;
 242                         break;
 243                 case ',':
 244                         tok++;
 245                         break;
 246                 default:
 247                         if (expr(r_expr, &eval, &eattr, &esym) != OK)
 248                                 goto token_err;
 249                         else
 250                         {
 251                                 switch(outtype)
 252                                 {
 253                                 case 0: strcpy(format, "%X"); break;
 254                                 case 1: strcpy(format, "%d" ); break;
 255                                 case 2: strcpy(format, "%u" ); break;
 256                                 }
 257
 258                                 if (wordlong)
 259                                         sprintf(prntstr, format, eval);
 260                                 else
 261                                         sprintf(prntstr, format, eval & 0xFFFF);
 262
 263                                 printf("%s", prntstr);
 264
 265                                 if (list_fd)
 266                                         unused = write(list_fd, prntstr, (LONG)strlen(prntstr));
 267
 268                                 formatting = 0;
 269                                 wordlong = 0;
 270                                 outtype = 0;
 271                         }
 272
 273                         break;
 274                 }
 275         }
 276
 277         printf("\n");
 278
 279         return 0;
 280
 281 token_err:
 282         error("illegal print token");
 283         return ERROR;
 284 }
 285
 286
 287 //
 288 // Undefine an equated condition code
 289 //
 290 int d_ccundef(void)
 291 {
 292         SYM * ccname;
 293
 294         // Check that we are in a RISC section
 295         if (!rgpu && !rdsp)
 296         {
 297                 error(".ccundef must be defined in .gpu/.dsp section");
 298                 return ERROR;
 299         }
 300
 301         if (*tok != SYMBOL)
 302         {
 303                 error("syntax error; expected symbol");
 304                 return ERROR;
 305         }
 306
 307         ccname = lookup(string[tok[1]], LABEL, 0);
 308
 309         // Make sure symbol is a valid ccdef
 310         if (!ccname || !(ccname->sattre & EQUATEDCC))
 311         {
 312                 error("invalid equated condition name specified");
 313                 return ERROR;
 314         }
 315
 316         ccname->sattre |= UNDEF_CC;
 317
 318         return 0;
 319 }
 320
 321
 322 //
 323 // Undefine an equated register
 324 //
 325 int d_equrundef(void)
 326 {
 327         SYM * regname;
 328
 329         // Check that we are in a RISC section
 330         if (!rgpu && !rdsp)
 331                 return error(".equrundef/.regundef must be defined in .gpu/.dsp section");
 332
 333         while (*tok != EOL)
 334         {
 335                 // Skip preceeding or seperating commas (if any)
 336                 if (*tok == ',')
 337                         tok++;
 338
 339                 // Check we are dealing with a symbol
 340                 if (*tok != SYMBOL)
 341                         return error("syntax error; expected symbol");
 342
 343                 // Lookup and undef if equated register
 344                 regname = lookup(string[tok[1]], LABEL, 0);
 345
 346                 if (regname && (regname->sattre & EQUATEDREG))
 347                 {
 348                         // Reset the attributes of this symbol...
 349                         regname->sattr = 0;
 350                         regname->sattre &= ~(EQUATEDREG | BANK_0 | BANK_1);
 351                         regname->sattre |= UNDEF_EQUR;
 352                 }
 353
 354                 // Skip over symbol token and address
 355                 tok += 2;
 356         }
 357
 358         return 0;
 359 }
 360
 361
 362 //
 363 // Do not allow use of the CLR.L opcode
 364 //
 365 int d_noclear(void)
 366 {
 367         warn("CLR.L opcode ignored...");
 368         return 0;
 369 }
 370
 371
 372 //
 373 // Include binary file
 374 //
 375 int d_incbin(void)
 376 {
 377         int fd;
 378         int bytes = 0;
 379         long pos, size, bytesRead;
 380         char msg[256];
 381         char buf1[256];
 382         int i;
 383
 384         // Check to see if we're in BSS, and, if so, throw an error
 385         if (scattr & SBSS)
 386         {
 387                 errors("cannot include binary file \"%s\" in BSS section", string[tok[1]]);
 388                 return ERROR;
 389         }
 390
 391         if (*tok != STRING)
 392         {
 393                 error("syntax error; string missing");
 394                 return ERROR;
 395         }
 396
 397         // Attempt to open the include file in the current directory, then (if that
 398         // failed) try list of include files passed in the enviroment string or by
 399         // the "-d" option.
 400         if ((fd = open(string[tok[1]], _OPEN_INC)) < 0)
 401         {
 402                 for(i=0; nthpath("RMACPATH", i, buf1)!=0; i++)
 403                 {
 404                         fd = strlen(buf1);
 405
 406                         // Append path char if necessary
 407                         if (fd > 0 && buf1[fd - 1] != SLASHCHAR)
 408                                 strcat(buf1, SLASHSTRING);
 409
 410                         strcat(buf1, string[tok[1]]);
 411
 412                         if ((fd = open(buf1, _OPEN_INC)) >= 0)
 413                                 goto allright;
 414                 }
 415
 416                 return errors("cannot open: \"%s\"", string[tok[1]]);
 417         }
 418
 419 allright:
 420
 421         size = lseek(fd, 0L, SEEK_END);
 422         pos = lseek(fd, 0L, SEEK_SET);
 423         chcheck(size);
 424
 425         DEBUG { printf("INCBIN: File '%s' is %li bytes.\n", string[tok[1]], size); }
 426
 427         char * fileBuffer = (char *)malloc(size);
 428         bytesRead = read(fd, fileBuffer, size);
 429
 430         if (bytesRead != size)
 431         {
 432                 sprintf(msg, "was only able to read %li bytes from binary file (%s, %li bytes)", bytesRead, string[tok[1]], size);
 433                 error(msg);
 434                 return ERROR;
 435         }
 436
 437         memcpy(chptr, fileBuffer, size);
 438         chptr += size;
 439         sloc += size;
 440         ch_size += size;
 441
 442         if (orgactive)
 443                 orgaddr += size;
 444
 445         free(fileBuffer);
 446         close(fd);
 447         return 0;
 448 }
 449
 450
 451 //
 452 // Set RISC register banks
 453 //
 454 int d_regbank0(void)
 455 {
 456         // Set active register bank zero
 457         regbank = BANK_0;
 458         return 0;
 459 }
 460
 461
 462 int d_regbank1(void)
 463 {
 464         // Set active register bank one
 465         regbank = BANK_1;
 466         return 0;
 467 }
 468
 469
 470 //
 471 // Helper function, to cut down on mistakes & typing
 472 //
 473 static inline void SkipBytes(unsigned bytesToSkip)
 474 {
 475         if (!bytesToSkip)
 476                 return;
 477
 478         if ((scattr & SBSS) == 0)
 479         {
 480                 chcheck(bytesToSkip);
 481                 D_ZEROFILL(bytesToSkip);
 482         }
 483         else
 484         {
 485                 sloc += bytesToSkip;
 486
 487                 if (orgactive)
 488                         orgaddr += bytesToSkip;
 489         }
 490 }
 491
 492
 493 //
 494 // Adjust location to an EVEN value
 495 //
 496 int d_even(void)
 497 {
 498     if (m6502)
 499         return error(in_6502mode);
 500
 501         unsigned skip = (rgpu || rdsp ? orgaddr : sloc) & 0x01;
 502
 503         if (skip)
 504         {
 505                 if ((scattr & SBSS) == 0)
 506                 {
 507                         chcheck(1);
 508                         D_byte(0);
 509                 }
 510                 else
 511                 {
 512                         sloc++;
 513
 514                         if (orgactive)
 515                                 orgaddr++;
 516                 }
 517         }
 518
 519         return 0;
 520 }
 521
 522
 523 //
 524 // Adjust location to a LONG value
 525 //
 526 int d_long(void)
 527 {
 528         unsigned lower2Bits = (rgpu || rdsp ? orgaddr : sloc) & 0x03;
 529         unsigned bytesToSkip = (0x04 - lower2Bits) & 0x03;
 530         SkipBytes(bytesToSkip);
 531
 532         return 0;
 533 }
 534
 535
 536 //
 537 // Adjust location to a PHRASE value
 538 //
 539 // N.B.: We have to handle the GPU/DSP cases separately because you can embed
 540 //       RISC code in the middle of a regular 68K section. Also note that all
 541 //       of the alignment pseudo-ops will have to be fixed this way.
 542 //
 543 // This *must* behave differently when in a RISC section, as following sloc
 544 // (instead of orgaddr) will fuck things up royally. Note that we do it this
 545 // way because you can embed RISC code in a 68K section, and have the origin
 546 // pointing to a different alignment in the RISC section than the 68K section.
 547 //
 548 int d_phrase(void)
 549 {
 550         unsigned lower3Bits = (rgpu || rdsp ? orgaddr : sloc) & 0x07;
 551         unsigned bytesToSkip = (0x08 - lower3Bits) & 0x07;
 552         SkipBytes(bytesToSkip);
 553
 554         return 0;
 555 }
 556
 557
 558 //
 559 // Adjust location to a DPHRASE value
 560 //
 561 int d_dphrase(void)
 562 {
 563         unsigned lower4Bits = (rgpu || rdsp ? orgaddr : sloc) & 0x0F;
 564         unsigned bytesToSkip = (0x10 - lower4Bits) & 0x0F;
 565         SkipBytes(bytesToSkip);
 566
 567         return 0;
 568 }
 569
 570
 571 //
 572 // Adjust location to a QPHRASE value
 573 //
 574 int d_qphrase(void)
 575 {
 576         unsigned lower5Bits = (rgpu || rdsp ? orgaddr : sloc) & 0x1F;
 577         unsigned bytesToSkip = (0x20 - lower5Bits) & 0x1F;
 578         SkipBytes(bytesToSkip);
 579
 580         return 0;
 581 }
 582
 583
 584 //
 585 // Do auto-even.  This must be called ONLY if 'sloc' is odd.
 586 //
 587 // This is made hairy because, if there was a label on the line, we also have
 588 // to adjust its value. This won't work with more than one label on the line,
 589 // which is OK since multiple labels are only allowed in AS68 kludge mode, and
 590 // the C compiler is VERY paranoid and uses ".even" whenever it can
 591 //
 592 // N.B.: This probably needs the same fixes as above...
 593 //
 594 void auto_even(void)
 595 {
 596   if (cursect != M6502)
 597   {
 598         if (scattr & SBSS)
 599                 sloc++;                         // Bump BSS section
 600         else
 601                 D_byte(0);                      // Deposit 0.b in non-BSS
 602
 603         if (lab_sym != NULL)    // Bump label if we have to
 604                 lab_sym->svalue++;
 605   }
 606 }
 607
 608
 609 //
 610 // Unimplemened directive error
 611 //
 612 int d_unimpl(void)
 613 {
 614         return error("unimplemented directive");
 615 }
 616
 617
 618 //
 619 // Return absolute (not TDB) and defined expression or return an error
 620 //
 621 int abs_expr(VALUE * a_eval)
 622 {
 623         WORD eattr;
 624
 625         if (expr(exprbuf, a_eval, &eattr, NULL) < 0)
 626                 return ERROR;
 627
 628         if (!(eattr & DEFINED))
 629                 return error(undef_error);
 630
 631         if (eattr & TDB)
 632                 return error(rel_error);
 633
 634         return OK;
 635 }
 636
 637
 638 //
 639 // Hand symbols in a symbol-list to a function (kind of like mapcar...)
 640 //
 641 int symlist(int(* func)())
 642 {
 643         const char * em = "symbol list syntax";
 644
 645         for(;;)
 646         {
 647                 if (*tok != SYMBOL)
 648                         return error(em);
 649
 650                 if ((*func)(string[tok[1]]) != OK)
 651                         break;
 652
 653                 tok += 2;
 654
 655                 if (*tok == EOL)
 656                         break;
 657
 658                 if (*tok != ',')
 659                         return error(em);
 660
 661                 tok++;
 662         }
 663
 664         return 0;
 665 }
 666
 667
 668 //
 669 // .include "filename"
 670 //
 671 int d_include(void)
 672 {
 673         int j;
 674         int i;
 675         char * fn;
 676         char buf[128];
 677         char buf1[128];
 678
 679         if (*tok == STRING)                     // Leave strings ALONE
 680                 fn = string[*++tok];
 681         else if (*tok == SYMBOL)        // Try to append ".s" to symbols
 682         {
 683                 strcpy(buf, string[*++tok]);
 684                 fext(buf, ".s", 0);
 685                 fn = &buf[0];
 686         }
 687         else                                            // Punt if no STRING or SYMBOL
 688                 return error("missing filename");
 689
 690         // Make sure the user didn't try anything like:
 691         // .include equates.s
 692         if (*++tok != EOL)
 693                 return error("extra stuff after filename -- enclose it in quotes");
 694
 695         // Attempt to open the include file in the current directory, then (if that
 696         // failed) try list of include files passed in the enviroment string or by
 697         // the "-i" option.
 698         if ((j = open(fn, 0)) < 0)
 699         {
 700                 for(i=0; nthpath("RMACPATH", i, buf1)!=0; i++)
 701                 {
 702                         j = strlen(buf1);
 703
 704                         // Append path char if necessary
 705                         if (j > 0 && buf1[j - 1] != SLASHCHAR)
 706                                 strcat(buf1, SLASHSTRING);
 707
 708                         strcat(buf1, fn);
 709
 710                         if ((j = open(buf1, 0)) >= 0)
 711                                 goto allright;
 712                 }
 713
 714                 return errors("cannot open: \"%s\"", fn);
 715         }
 716
 717 allright:
 718         include(j, fn);
 719         return 0;
 720 }
 721
 722
 723 //
 724 // .assert expression [, expression...]
 725 //
 726 int d_assert(void)
 727 {
 728         WORD eattr;
 729         VALUE eval;
 730
 731         for(; expr(exprbuf, &eval, &eattr, NULL)==OK; ++tok)
 732         {
 733                 if (!(eattr & DEFINED))
 734                         return error("forward or undefined .assert");
 735
 736                 if (!eval)
 737                         return error("assert failure");
 738
 739                 if (*tok != ',')
 740                         break;
 741         }
 742
 743         at_eol();
 744         return 0;
 745 }
 746
 747
 748 //
 749 // .globl symbol [, symbol] <<<cannot make local symbols global>>>
 750 //
 751 int globl1(char * p)
 752 {
 753         SYM * sy;
 754
 755         if (*p == '.')
 756                 return error("cannot .globl local symbol");
 757
 758         if ((sy = lookup(p, LABEL, 0)) == NULL)
 759         {
 760                 sy = NewSymbol(p, LABEL, 0);
 761                 sy->svalue = 0;
 762                 sy->sattr = GLOBAL;
 763 //printf("glob1: Making global symbol: attr=%04X, eattr=%08X, %s\n", sy->sattr, sy->sattre, sy->sname);
 764         }
 765         else
 766                 sy->sattr |= GLOBAL;
 767
 768         return OK;
 769 }
 770
 771
 772 int d_globl(void)
 773 {
 774   if (m6502)
 775     return error(in_6502mode);
 776
 777         symlist(globl1);
 778         return 0;
 779 }
 780
 781
 782 //
 783 // .prgflags expression
 784 //
 785 int d_prgflags(void)
 786 {
 787         VALUE eval;
 788
 789         if (*tok == EOL)
 790                 return error("PRGFLAGS requires value");
 791         else if (abs_expr(&eval) == OK)
 792         {
 793                 PRGFLAGS=eval;
 794                 return 0;
 795         }
 796         else
 797         {
 798                 return error("PRGFLAGS requires value");
 799         }
 800 }
 801
 802
 803 //
 804 // .abs [expression]
 805 //
 806 int d_abs(void)
 807 {
 808         VALUE eval;
 809
 810     if (m6502)
 811         return error(in_6502mode);
 812
 813         SaveSection();
 814
 815         if (*tok == EOL)
 816                 eval = 0;
 817         else if (abs_expr(&eval) != OK)
 818                 return 0;
 819
 820         SwitchSection(ABS);
 821         sloc = eval;
 822         return 0;
 823 }
 824
 825
 826 //
 827 // Switch segments
 828 //
 829 int d_text(void)
 830 {
 831         if (rgpu || rdsp)
 832                 return error("directive forbidden in gpu/dsp mode");
 833     if (m6502)
 834         return error(in_6502mode);
 835
 836         if (cursect != TEXT)
 837         {
 838                 SaveSection();
 839                 SwitchSection(TEXT);
 840         }
 841
 842         return 0;
 843 }
 844
 845
 846 int d_data(void)
 847 {
 848         if (rgpu || rdsp)
 849                 return error("directive forbidden in gpu/dsp mode");
 850     if (m6502)
 851         return error(in_6502mode);
 852
 853         if (cursect != DATA)
 854         {
 855                 SaveSection();
 856                 SwitchSection(DATA);
 857         }
 858
 859         return 0;
 860 }
 861
 862
 863 int d_bss(void)
 864 {
 865         if (rgpu || rdsp)
 866                 return error("directive forbidden in gpu/dsp mode");
 867     if (m6502)
 868         return error(in_6502mode);
 869
 870         if (cursect != BSS)
 871         {
 872                 SaveSection();
 873                 SwitchSection(BSS);
 874         }
 875
 876         return 0;
 877 }
 878
 879
 880 //
 881 // .ds[.size] expression
 882 //
 883 int d_ds(WORD siz)
 884 {
 885         DEBUG { printf("Directive: .ds.[size] = %u, sloc = $%X\n", siz, sloc); }
 886
 887         VALUE eval;
 888
 889     if (cursect != M6502)
 890     {
 891             if ((siz != SIZB) && (sloc & 1))    // Automatic .even
 892                     auto_even();
 893     }
 894
 895         if (abs_expr(&eval) != OK)
 896                 return 0;
 897
 898         // Check to see if the value being passed in is negative (who the hell does
 899         // that?--nobody does; it's the code gremlins, or rum, that does it)
 900         if (eval < 0)
 901                 return error("negative sizes not allowed");
 902
 903         // In non-TDB section (BSS, ABS and M6502) just advance the location
 904         // counter appropriately. In TDB sections, deposit (possibly large) chunks
 905         // of zeroed memory....
 906         if ((scattr & SBSS) || cursect == M6502)
 907         {
 908                 listvalue(eval);
 909                 eval *= siz;
 910                 sloc += eval;
 911         if (cursect == M6502) chptr += eval;
 912                 just_bss = 1;                                   // No data deposited (8-bit CPU mode)
 913         }
 914         else
 915         {
 916                 dep_block(eval, siz, (VALUE)0, (WORD)(DEFINED | ABS), NULL);
 917         }
 918
 919         at_eol();
 920         return 0;
 921 }
 922
 923
 924 //
 925 // dc.b, dc.w / dc, dc.l, dc.i
 926 //
 927 int d_dc(WORD siz)
 928 {
 929         WORD eattr;
 930         VALUE eval;
 931         uint8_t * p;
 932
 933         if ((scattr & SBSS) != 0)
 934                 return error("illegal initialization of section");
 935
 936         // Do an auto_even if it's not BYTE sized (hmm, should we be doing this???)
 937         if (cursect != M6502 && (siz != SIZB) && (sloc & 1))
 938                 auto_even();
 939
 940         // Check to see if we're trying to set LONGS on a non 32-bit aligned
 941         // address in a GPU or DSP section, in their local RAM
 942         if ((siz == SIZL) && (orgaddr & 0x03)
 943                 && ((rgpu && (orgaddr >= 0xF03000) && (orgaddr <= 0xF03FFFF))
 944                 || (rdsp && (orgaddr >= 0xF1B000) && (orgaddr <= 0xF1CFFFF))))
 945                 warn("depositing LONGs on a non-long address in local RAM");
 946
 947         for(;; tok++)
 948         {
 949                 // dc.b 'string' [,] ...
 950                 if (siz == SIZB && *tok == STRING && (tok[2] == ',' || tok[2] == EOL))
 951                 {
 952                         uint32_t i = strlen(string[tok[1]]);
 953
 954                         if ((challoc - ch_size) < i)
 955                                 chcheck(i);
 956
 957                         for(p=string[tok[1]]; *p!=EOS; p++)
 958                                 D_byte(*p);
 959
 960                         tok += 2;
 961                         goto comma;
 962                 }
 963
 964                 int movei = 0; // MOVEI flag for dc.i
 965
 966                 if (*tok == DOTI)
 967                 {
 968                         movei = 1;
 969                         tok++;
 970                         siz = SIZL;
 971                 }
 972
 973                 // dc.x <expression>
 974                 SYM * esym = 0;
 975
 976                 if (expr(exprbuf, &eval, &eattr, &esym) != OK)
 977                         return 0;
 978
 979                 uint16_t tdb = eattr & TDB;
 980                 uint16_t defined = eattr & DEFINED;
 981
 982                 if ((challoc - ch_size) < 4)
 983                         chcheck(4);
 984
 985                 switch (siz)
 986                 {
 987                 case SIZB:
 988                         if (!defined)
 989                         {
 990                                 AddFixup(FU_BYTE | FU_SEXT, sloc, exprbuf);
 991                                 D_byte(0);
 992                         }
 993                         else
 994                         {
 995                                 if (eval + 0x100 >= 0x200)
 996                                 {
 997                                         sprintf(buffer, "%s (value = $%X)", range_error, eval);
 998                                         return error(buffer);
 999                                 }
1000
1001                                 if (tdb)
1002                                         return error("non-absolute byte value");
1003
1004                                 D_byte(eval);
1005                         }
1006
1007                         break;
1008                 case SIZW:
1009                 case SIZN:
1010                         if (!defined)
1011                         {
1012                                 AddFixup(FU_WORD | FU_SEXT, sloc, exprbuf);
1013                                 D_word(0);
1014                         }
1015                         else
1016                         {
1017                                 if (eval + 0x10000 >= 0x20000)
1018                                         return error(range_error);
1019
1020                                 if (tdb)
1021                                         MarkRelocatable(cursect, sloc, tdb, MWORD, NULL);
1022
1023                                 // Deposit 68000 or 6502 (byte-reversed) word
1024                     if (cursect != M6502)
1025                         {
1026                             D_word(eval);
1027                         }
1028                     else
1029                         {
1030                             D_rword(eval);
1031                         }
1032
1033                         }
1034
1035                         break;
1036                 case SIZL:
1037                 if (m6502)
1038                     return error(in_6502mode);
1039
1040                         if (!defined)
1041                         {
1042                                 if (movei)
1043                                         AddFixup(FU_LONG | FU_MOVEI, sloc, exprbuf);
1044                                 else
1045                                         AddFixup(FU_LONG, sloc, exprbuf);
1046
1047                                 D_long(0);
1048                         }
1049                         else
1050                         {
1051                                 if (tdb)
1052                                         MarkRelocatable(cursect, sloc, tdb, MLONG, NULL);
1053
1054                                 if (movei)
1055                                         eval = WORDSWAP32(eval);
1056
1057                                 D_long(eval);
1058                         }
1059                         break;
1060                 }
1061
1062 comma:
1063                 if (*tok != ',')
1064                         break;
1065         }
1066
1067         at_eol();
1068         return 0;
1069 }
1070
1071
1072 //
1073 // dcb[.siz] expr1,expr2 - Make 'expr1' copies of 'expr2'
1074 //
1075 int d_dcb(WORD siz)
1076 {
1077         VALUE evalc, eval;
1078         WORD eattr;
1079
1080         DEBUG { printf("dcb: section is %s%s%s (scattr=$%X)\n", (cursect & TEXT ? "TEXT" : ""), (cursect & DATA ? " DATA" : ""), (cursect & BSS ? "BSS" : ""), scattr); }
1081
1082         if ((scattr & SBSS) != 0)
1083                 return error("illegal initialization of section");
1084
1085         if (abs_expr(&evalc) != OK)
1086                 return 0;
1087
1088         if (*tok++ != ',')
1089                 return error("missing comma");
1090
1091         if (expr(exprbuf, &eval, &eattr, NULL) < 0)
1092                 return 0;
1093
1094         if (cursect != M6502 && (siz != SIZB) && (sloc & 1))
1095                 auto_even();
1096
1097         dep_block(evalc, siz, eval, eattr, exprbuf);
1098         return 0;
1099 }
1100
1101
1102 //
1103 // Generalized initialization directive
1104 //
1105 // .init[.siz] [#count,] expression [.size] , ...
1106 //
1107 // The size suffix on the ".init" directive becomes the default size of the
1108 // objects to deposit. If an item is preceeded with a sharp (immediate) sign
1109 // and an expression, it specifies a repeat count. The value to be deposited
1110 // may be followed by a size suffix, which overrides the default size.
1111 //
1112 int d_init(WORD def_siz)
1113 {
1114         VALUE count;
1115         VALUE eval;
1116         WORD eattr;
1117         WORD siz;
1118
1119         if ((scattr & SBSS) != 0)
1120                 return error(".init not permitted in BSS or ABS");
1121
1122         if (rgpu || rdsp)
1123                 return error("directive forbidden in gpu/dsp mode");
1124
1125         for(;;)
1126         {
1127                 // Get repeat count (defaults to 1)
1128                 if (*tok == '#')
1129                 {
1130                         ++tok;
1131
1132                         if (abs_expr(&count) != OK)
1133                                 return 0;
1134
1135                         if (*tok++ != ',')
1136                                 return error(comma_error);
1137                 }
1138                 else
1139                         count = 1;
1140
1141                 // Evaluate expression to deposit
1142                 if (expr(exprbuf, &eval, &eattr, NULL) < 0)
1143                         return 0;
1144
1145                 switch ((int)*tok++)
1146                 {                                 // Determine size of object to deposit
1147                 case DOTB: siz = SIZB; break;
1148                 case DOTW: siz = SIZB; break;
1149                 case DOTL: siz = SIZL; break;
1150                 default:
1151                         siz = def_siz;
1152                         tok--;
1153                         break;
1154                 }
1155
1156                 dep_block(count, siz, eval, eattr, exprbuf);
1157
1158                 switch ((int)*tok)
1159                 {
1160                 case EOL:
1161                         return 0;
1162                 case ',':
1163                         tok++;
1164                         continue;
1165                 default:
1166                         return error(comma_error);
1167                 }
1168         }
1169 }
1170
1171
1172 //
1173 // Deposit 'count' values of size 'siz' in the current (non-BSS) segment
1174 //
1175 int dep_block(VALUE count, WORD siz, VALUE eval, WORD eattr, TOKEN * exprbuf)
1176 {
1177         WORD tdb;
1178         WORD defined;
1179
1180         tdb = (WORD)(eattr & TDB);
1181         defined = (WORD)(eattr & DEFINED);
1182
1183         while (count--)
1184         {
1185                 if ((challoc - ch_size) < 4)
1186                         chcheck(4L);
1187
1188                 switch(siz)
1189                 {
1190                 case SIZB:
1191                         if (!defined)
1192                         {
1193                                 AddFixup(FU_BYTE | FU_SEXT, sloc, exprbuf);
1194                                 D_byte(0);
1195                         }
1196                         else
1197                         {
1198                                 if (tdb)
1199                                         return error("non-absolute byte value");
1200
1201                                 if (eval + 0x100 >= 0x200)
1202                                         return error(range_error);
1203
1204                                 D_byte(eval);
1205                         }
1206
1207                         break;
1208                 case SIZW:
1209                 case SIZN:
1210                         if (!defined)
1211                         {
1212                                 AddFixup(FU_WORD | FU_SEXT, sloc, exprbuf);
1213                                 D_word(0);
1214                         }
1215                         else
1216                         {
1217                                 if (tdb)
1218                                         MarkRelocatable(cursect, sloc, tdb, MWORD, NULL);
1219
1220                                 if (eval + 0x10000 >= 0x20000)
1221                                         return error(range_error);
1222
1223                                 // Deposit 68000 or 6502 (byte-reversed) word
1224                                 if (cursect != M6502)
1225                                 {
1226                                         D_word(eval);
1227                                 }
1228                                 else
1229                                 {
1230                                         D_rword(eval);
1231                                 }
1232
1233                         }
1234
1235                         break;
1236                 case SIZL:
1237                         if (m6502)
1238                                 return error(in_6502mode);
1239
1240                         if (!defined)
1241                         {
1242                                 AddFixup(FU_LONG, sloc, exprbuf);
1243                                 D_long(0);
1244                         }
1245                         else
1246                         {
1247                                 if (tdb)
1248                                         MarkRelocatable(cursect, sloc, tdb, MLONG, NULL);
1249
1250                                 D_long(eval);
1251                         }
1252
1253                         break;
1254                 }
1255         }
1256
1257         return 0;
1258 }
1259
1260
1261 //
1262 // .comm symbol, size
1263 //
1264 int d_comm(void)
1265 {
1266         SYM * sym;
1267         char * p;
1268         VALUE eval;
1269
1270         if (m6502)
1271                 return error(in_6502mode);
1272
1273         if (*tok != SYMBOL)
1274                 return error("missing symbol");
1275
1276         p = string[tok[1]];
1277         tok += 2;
1278
1279         if (*p == '.')                                                  // Cannot .comm a local symbol
1280                 return error(locgl_error);
1281
1282         if ((sym = lookup(p, LABEL, 0)) == NULL)
1283                 sym = NewSymbol(p, LABEL, 0);
1284         else
1285         {
1286                 if (sym->sattr & DEFINED)
1287                         return error(".comm symbol already defined");
1288         }
1289
1290         sym->sattr = GLOBAL | COMMON | BSS;
1291
1292         if (*tok++ != ',')
1293                 return error(comma_error);
1294
1295         if (abs_expr(&eval) != OK)                              // Parse size of common region
1296                 return 0;
1297
1298         sym->svalue = eval;                                             // Install common symbol's size
1299         at_eol();
1300         return 0;
1301 }
1302
1303
1304 //
1305 // .list - Turn listing on
1306 //
1307 int d_list(void)
1308 {
1309         if (list_flag)
1310                 listing++;
1311
1312         return 0;
1313 }
1314
1315
1316 //
1317 // .nlist - Turn listing off
1318 //
1319 int d_nlist(void)
1320 {
1321         if (list_flag)
1322                 listing--;
1323
1324         return 0;
1325 }
1326
1327
1328 //
1329 // .68000 - Back to 68000 TEXT segment
1330 //
1331 int d_68000(void)
1332 {
1333         rgpu = rdsp = 0;
1334         // Switching from gpu/dsp sections should reset any ORG'd Address
1335         orgactive = 0;
1336         orgwarning = 0;
1337         SaveSection();
1338         SwitchSection(TEXT);
1339         return 0;
1340 }
1341
1342
1343 //
1344 // .gpu - Switch to GPU assembler
1345 //
1346 int d_gpu(void)
1347 {
1348         if ((cursect != TEXT) && (cursect != DATA))
1349         {
1350                 error(".gpu can only be used in the TEXT or DATA segments");
1351                 return ERROR;
1352         }
1353
1354         // If previous section was dsp or 68000 then we need to reset ORG'd Addresses
1355         if (!rgpu)
1356         {
1357 //printf("Resetting ORG...\n");
1358                 orgactive = 0;
1359                 orgwarning = 0;
1360         }
1361 //else printf("NOT resetting ORG!\n");
1362
1363         rgpu = 1;                       // Set GPU assembly
1364         rdsp = 0;                       // Unset DSP assembly
1365         regbank = BANK_N;       // Set no default register bank
1366         return 0;
1367 }
1368
1369
1370 //
1371 // .dsp - Switch to DSP assembler
1372 //
1373 int d_dsp(void)
1374 {
1375         if ((cursect != TEXT) && (cursect != DATA))
1376         {
1377                 error(".dsp can only be used in the TEXT or DATA segments");
1378                 return ERROR;
1379         }
1380
1381         // If previous section was gpu or 68000 then we need to reset ORG'd Addresses
1382         if (!rdsp)
1383         {
1384                 orgactive = 0;
1385                 orgwarning = 0;
1386         }
1387
1388         rdsp = 1;                       // Set DSP assembly
1389         rgpu = 0;                       // Unset GPU assembly
1390         regbank = BANK_N;       // Set no default register bank
1391         return 0;
1392 }
1393
1394
1395 //
1396 // .cargs [#offset], symbol[.size], ...
1397 //
1398 // Lists of registers may also be mentioned; they just take up space. Good for
1399 // "documentation" purposes:
1400 //
1401 // .cargs a6, .arg1, .arg2, .arg3...
1402 //
1403 // Symbols thus created are ABS and EQUATED.
1404 //
1405 int d_cargs(void)
1406 {
1407         VALUE eval = 4;         // Default to 4 if no offset specified (to account for
1408                                                 // return address)
1409         WORD rlist;
1410         SYM * symbol;
1411         char * p;
1412         int env;
1413         int i;
1414
1415         if (rgpu || rdsp)
1416                 return error("directive forbidden in gpu/dsp mode");
1417
1418         if (*tok == '#')
1419         {
1420                 tok++;
1421
1422                 if (abs_expr(&eval) != OK)
1423                         return 0;
1424
1425                 // Eat the comma, if it's there
1426                 if (*tok == ',')
1427                         tok++;
1428         }
1429
1430         for(;;)
1431         {
1432                 if (*tok == SYMBOL)
1433                 {
1434                         p = string[tok[1]];
1435
1436                         // Set env to either local (dot prefixed) or global scope
1437                         env = (*p == '.' ? curenv : 0);
1438                         symbol = lookup(p, LABEL, env);
1439
1440                         if (symbol == NULL)
1441                         {
1442                                 symbol = NewSymbol(p, LABEL, env);
1443                                 symbol->sattr = 0;
1444                         }
1445                         else if (symbol->sattr & DEFINED)
1446                                 return errors("multiply-defined label '%s'", p);
1447
1448                         // Put symbol in "order of definition" list
1449                         AddToSymbolDeclarationList(symbol);
1450
1451                         symbol->sattr |= (ABS | DEFINED | EQUATED);
1452                         symbol->svalue = eval;
1453                         tok += 2;
1454
1455                         // What this does is eat any dot suffixes attached to a symbol. If
1456                         // it's a .L, it adds 4 to eval; if it's .W or .B, it adds 2. If
1457                         // there is no dot suffix, it assumes a size of 2.
1458                         switch ((int)*tok)
1459                         {
1460                         case DOTL:
1461                                 eval += 2;
1462                         case DOTB:
1463                         case DOTW:
1464                                 tok++;
1465                         }
1466
1467                         eval += 2;
1468                 }
1469                 else if (*tok >= KW_D0 && *tok <= KW_A7)
1470                 {
1471                         if (reglist(&rlist) < 0)
1472                                 return 0;
1473
1474                         for(i=0; i<16; i++, rlist>>=1)
1475                         {
1476                                 if (rlist & 1)
1477                                         eval += 4;
1478                         }
1479                 }
1480                 else
1481                 {
1482                         switch ((int)*tok)
1483                         {
1484                         case KW_USP:
1485                         case KW_SSP:
1486                         case KW_PC:
1487                                 eval += 2;
1488                                 // FALLTHROUGH
1489                         case KW_SR:
1490                         case KW_CCR:
1491                                 eval += 2;
1492                                 tok++;
1493                                 break;
1494                         case EOL:
1495                                 return 0;
1496                         default:
1497                                 return error(".cargs syntax");
1498                         }
1499                 }
1500
1501                 // Eat commas in between each argument, if they exist
1502                 if (*tok == ',')
1503                         tok++;
1504         }
1505 }
1506
1507
1508 //
1509 // .cstruct [#offset], symbol[.size], ...
1510 //
1511 // Lists of registers may also be mentioned; they just take up space. Good for
1512 // "documentation" purposes:
1513 //
1514 // .cstruct a6, .arg1, .arg2, .arg3...
1515 //
1516 // Symbols thus created are ABS and EQUATED. Note that this is for
1517 // compatibility with VBCC and the Remover's library. Thanks to GroovyBee for
1518 // the suggestion.
1519 //
1520 int d_cstruct(void)
1521 {
1522         VALUE eval = 0;         // Default, if no offset specified, is zero
1523         WORD rlist;
1524         SYM * symbol;
1525         char * symbolName;
1526         int env;
1527         int i;
1528
1529         if (rgpu || rdsp)
1530                 return error("directive forbidden in gpu/dsp mode");
1531
1532         if (*tok == '#')
1533         {
1534                 tok++;
1535
1536                 if (abs_expr(&eval) != OK)
1537                         return 0;
1538
1539                 // Eat the comma, if it's there
1540                 if (*tok == ',')
1541                         tok++;
1542         }
1543
1544         for(;;)
1545         {
1546                 if (*tok == SYMBOL)
1547                 {
1548                         symbolName = string[tok[1]];
1549
1550                         // Set env to either local (dot prefixed) or global scope
1551                         env = (symbolName[0] == '.' ? curenv : 0);
1552                         symbol = lookup(symbolName, LABEL, env);
1553
1554                         // If the symbol wasn't found, then define it. Otherwise, throw an
1555                         // error.
1556                         if (symbol == NULL)
1557                         {
1558                                 symbol = NewSymbol(symbolName, LABEL, env);
1559                                 symbol->sattr = 0;
1560                         }
1561                         else if (symbol->sattr & DEFINED)
1562                                 return errors("multiply-defined label '%s'", symbolName);
1563
1564                         // Put symbol in "order of definition" list
1565                         AddToSymbolDeclarationList(symbol);
1566
1567                         tok += 2;
1568
1569                         // Adjust label start address if it's a word or a long, as a byte
1570                         // label might have left us on an odd address.
1571                         switch ((int)*tok)
1572                         {
1573                         case DOTW:
1574                         case DOTL:
1575                                 eval += eval & 0x01;
1576                         }
1577
1578                         symbol->sattr |= (ABS | DEFINED | EQUATED);
1579                         symbol->svalue = eval;
1580
1581                         // Check for dot suffixes and adjust space accordingly (longs and
1582                         // words on an odd boundary get bumped to the next word aligned
1583                         // address). If no suffix, then throw an error.
1584                         switch ((int)*tok)
1585                         {
1586                         case DOTL:
1587                                 eval += 4;
1588                                 break;
1589                         case DOTW:
1590                                 eval += 2;
1591                                 break;
1592                         case DOTB:
1593                                 eval += 1;
1594                                 break;
1595                         default:
1596                                 return error("Symbol missing dot suffix in .cstruct construct");
1597                         }
1598
1599                         tok++;
1600                 }
1601                 else if (*tok >= KW_D0 && *tok <= KW_A7)
1602                 {
1603                         if (reglist(&rlist) < 0)
1604                                 return 0;
1605
1606                         for(i=0; i<16; i++, rlist>>=1)
1607                         {
1608                                 if (rlist & 1)
1609                                         eval += 4;
1610                         }
1611                 }
1612                 else
1613                 {
1614                         switch ((int)*tok)
1615                         {
1616                         case KW_USP:
1617                         case KW_SSP:
1618                         case KW_PC:
1619                                 eval += 2;
1620                                 // FALLTHROUGH
1621                         case KW_SR:
1622                         case KW_CCR:
1623                                 eval += 2;
1624                                 tok++;
1625                                 break;
1626                         case EOL:
1627                                 return 0;
1628                         default:
1629                                 return error(".cstruct syntax");
1630                         }
1631                 }
1632
1633                 // Eat commas in between each argument, if they exist
1634                 if (*tok == ',')
1635                         tok++;
1636         }
1637 }
1638
1639
1640 //
1641 // Undefine a macro - .undefmac macname [, macname...]
1642 //
1643 int undmac1(char * p)
1644 {
1645         SYM * symbol = lookup(p, MACRO, 0);
1646
1647         // If the macro symbol exists, cause it to disappear
1648         if (symbol != NULL)
1649                 symbol->stype = (BYTE)SY_UNDEF;
1650
1651         return OK;
1652 }
1653
1654
1655 int d_undmac(void)
1656 {
1657         symlist(undmac1);
1658         return 0;
1659 }
1660
1661
1662 int d_jpad(void)
1663 {
1664         warn("JPAD directive is deprecated/non-functional");
1665         return OK;
1666 }
1667
1668
1669 int d_nojpad(void)
1670 {
1671         warn("NOJPAD directive is deprecated/non-functional");
1672         return OK;
1673 }
1674
1675
1676 int d_gpumain(void)
1677 {
1678         return error("What the hell? Do you think we adhere to the Goof standard?");
1679 }
1680
1681
1682 //
1683 // .opt - turn a specific (or all) optimisation on or off
1684 //
1685 int d_opt(void)
1686 {
1687         while (*tok != EOL)
1688         {
1689                 if (*tok == STRING)
1690                 {
1691                         tok++;
1692                         char * tmpstr = string[*tok++];
1693
1694                         if (ParseOptimization(tmpstr) != OK)
1695                         {
1696                                 char temperr[256];
1697                                 sprintf(temperr, "unknown optimization flag '%s'", tmpstr);
1698                                 return error(temperr);
1699                         }
1700                 }
1701                 else
1702                         return error(".opt directive needs every switch enclosed inside quotation marks");
1703         }
1704
1705         return OK;
1706 }
1707