2 // RMAC - Reboot's Macro Assembler for all Atari computers
3 // PROCLN.C - Line Processing
4 // Copyright (C) 199x Landon Dyer, 2011-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
22 #define DEF_KW // Declare keyword values
23 #include "kwtab.h" // Incl generated keyword tables & defs
25 #define DEF_MN // Incl 68k keyword definitions
26 #define DECL_MN // Incl 68k keyword state machine tables
33 #define DEF_MP // Include 6502 keyword definitions
34 #define DECL_MP // Include 6502 keyword state machine tables
37 IFENT * ifent; // Current ifent
38 static IFENT ifent0; // Root ifent
39 static IFENT * f_ifent; // Freelist of ifents
40 static int disabled; // Assembly conditionally disabled
41 int just_bss; // 1, ds.b in microprocessor mode
42 VALUE pcloc; // Value of "PC" at beginning of line
43 SYM * lab_sym; // Label on line (or NULL)
45 const char extra_stuff[] = "extra (unexpected) text found after addressing mode";
46 const char comma_error[] = "missing comma";
47 const char syntax_error[] = "syntax error";
48 const char locgl_error[] = "cannot GLOBL local symbol";
49 const char lab_ignored[] = "label ignored";
51 // Table to convert an addressing-mode number to a bitmask.
52 LONG amsktab[0124] = {
53 M_DREG, M_DREG, M_DREG, M_DREG,
54 M_DREG, M_DREG, M_DREG, M_DREG,
56 M_AREG, M_AREG, M_AREG, M_AREG,
57 M_AREG, M_AREG, M_AREG, M_AREG,
59 M_AIND, M_AIND, M_AIND, M_AIND,
60 M_AIND, M_AIND, M_AIND, M_AIND,
62 M_APOSTINC, M_APOSTINC, M_APOSTINC, M_APOSTINC,
63 M_APOSTINC, M_APOSTINC, M_APOSTINC, M_APOSTINC,
65 M_APREDEC, M_APREDEC, M_APREDEC, M_APREDEC,
66 M_APREDEC, M_APREDEC, M_APREDEC, M_APREDEC,
68 M_ADISP, M_ADISP, M_ADISP, M_ADISP,
69 M_ADISP, M_ADISP, M_ADISP, M_ADISP,
71 M_AINDEXED, M_AINDEXED, M_AINDEXED, M_AINDEXED,
72 M_AINDEXED, M_AINDEXED, M_AINDEXED, M_AINDEXED,
98 M_CREG, // 0120 (caches - TODO: is this correct or does it need its own bitfield?)
105 // Function prototypes
106 int HandleLabel(char *, int);
110 // Initialize line processor
112 void InitLineProcessor(void)
116 f_ifent = ifent0.if_prev = NULL;
126 int state; // Keyword machine state (output)
127 int j; // Random int, must be fast
128 char * p; // Random char ptr, must be fast
129 TOKEN * tk; // First token in line
130 char * label; // Symbol (or NULL)
131 char * equate; // Symbol (or NULL)
132 int labtyp = 0; // Label type (':', DCOLON)
133 int equtyp = 0; // Equ type ('=', DEQUALS)
134 VALUE eval; // Expression value
135 WORD eattr; // Expression attributes
136 SYM * esym; // External symbol involved in expr.
137 WORD siz = 0; // Size suffix to mnem/diretve/macro
138 LONG amsk0, amsk1; // Address-type masks for ea0, ea1
139 MNTAB * m; // Code generation table pointer
140 SYM * sy, * sy2; // Symbol (temp usage)
141 char * opname = NULL; // Name of dirctve/mnemonic/macro
142 int listflag; // 0: Don't call listeol()
143 WORD rmask; // Register list, for REG
144 int registerbank; // RISC register bank
145 int riscreg; // RISC register
146 listflag = 0; // Initialise listing flag
148 loop: // Line processing loop label
150 // Get another line of tokens
151 if (TokenizeLine() == TKEOF)
153 DEBUG { printf("Assemble: Found TKEOF flag...\n"); }
154 if (list_flag && listflag) // Flush last line of source
157 if (ifent->if_prev != NULL) // Check conditional token
158 error("hit EOF without finding matching .endif");
163 DEBUG { DumpTokenBuffer(); }
167 if (listflag && listing > 0)
168 listeol(); // Tell listing generator about EOL
170 lstout((char)(disabled ? '-' : lntag)); // Prepare new line for listing
171 listflag = 1; // OK to call `listeol' now
172 just_bss = 0; // Reset just_bss mode
175 state = -3; // No keyword (just EOL)
176 label = NULL; // No label
177 lab_sym = NULL; // No (exported) label
178 equate = NULL; // No equate
179 tk = tok; // Save first token in line
180 pcloc = (VALUE)sloc; // Set beginning-of-line PC
182 loop1: // Internal line processing loop
184 if (*tok == EOL) // Restart loop if end-of-line
187 // First token MUST be a symbol (Shamus: not sure why :-/)
190 if ((*tok >= KW_D0) && (*tok <= KW_R31))
191 error("cannot use reserved keyword as label name or .equ");
193 error("syntax error; expected symbol");
198 j = (int)tok[2]; // Skip equates (normal statements)
200 if (j == '=' || j == DEQUALS || j == SET || j == REG || j == EQUREG || j == CCDEF)
202 equate = string[tok[1]];
208 // Skip past label (but record it)
209 if (j == ':' || j == DCOLON)
212 label = string[tok[1]]; // Get label name
213 labtyp = tok[2]; // Get label type
214 tok += 3; // Go to next line token
217 // Looks like another label follows the previous one, so handle
218 // the previous one until there aren't any more
219 if (as68_flag && (*tok == SYMBOL && tok[2] == ':'))
221 if (HandleLabel(label, labtyp) != 0)
228 // EOL is legal here...
232 // Next token MUST be a symbol
233 if (*tok++ != SYMBOL)
235 error("syntax error; expected symbol");
239 opname = p = string[*tok++];
241 // Check to see if the SYMBOL is a keyword (a mnemonic or directive).
242 // On output, `state' will have one of the values:
243 // -3 there was no symbol (EOL)
244 // -2..-1 the symbol didn't match any keyword
245 // 0..499 vanilla directives (dc, ds, etc.)
246 // 500..999 electric directives (macro, rept, etc.)
247 // 1000..+ mnemonics (move, lsr, etc.)
248 for(state=0; state>=0;)
250 j = mnbase[state] + (int)tolowertab[*p];
252 // Reject, character doesn't match
253 if (mncheck[j] != state)
255 state = -1; // No match
259 // Must accept or reject at EOS
262 state = mnaccept[j]; // (-1 on no terminal match)
269 // Check for ".b" ".w" ".l" after directive, macro or mnemonic.
274 else if (*tok == DOTL)
276 else if (*tok == DOTB)
278 else if(*tok == DOTD)
280 else if(*tok == DOTP)
282 else if(*tok == DOTQ)
284 else if(*tok == DOTS)
286 else if(*tok == DOTX)
290 // Do special directives (500..999) (These must be handled in "real time")
291 if (state >= 500 && state < 1000)
304 case MN_IIF: // .iif --- immediate if
305 if (disabled || expr(exprbuf, &eval, &eattr, &esym) != OK)
308 if (!(eattr & DEFINED))
324 case MN_MACRO: // .macro --- macro definition
327 // Label on a macro definition is bad mojo... Warn the user
335 case MN_EXITM: // .exitm --- exit macro
336 case MN_ENDM: // .endm --- same as .exitm
349 // Handle labels on REPT directive lines...
352 if (HandleLabel(label, labtyp) != 0)
362 error("mis-nested .endr");
369 if (disabled) // Conditionally disabled code
375 // Pick global or local symbol enviroment
376 j = (*equate == '.' ? curenv : 0);
377 sy = lookup(equate, LABEL, j);
381 sy = NewSymbol(equate, LABEL, j);
384 if (equtyp == DEQUALS)
386 // Can't GLOBAL a local symbol
396 else if ((sy->sattr & DEFINED) && equtyp != SET)
398 if ((equtyp == EQUREG) && (sy->sattre & UNDEF_EQUR))
400 //REALLY? sy->sattre |= ~UNDEF_EQUR;
401 sy->sattre &= ~UNDEF_EQUR;
404 else if ((equtyp == CCDEF) && (sy->sattre & UNDEF_CC))
406 //REALLY? sy->sattre |= ~UNDEF_CC;
407 sy->sattre &= ~UNDEF_CC;
412 error("multiple equate to '%s'", sy->sname);
417 // Put symbol in "order of definition" list if it's not already there
418 AddToSymbolDeclarationList(sy);
420 // Parse value to equate symbol to;
424 if (equtyp == EQUREG)
426 //Linko's request to issue a warning on labels that equated to the same
427 //register would go here. Not sure how to implement it though. :-/
430 have an array of bools with 64 entries. Whenever a register is equated, set the
431 corresponding register bool to true. Whenever it's undef'ed, set it to false.
432 When checking to see if it's already been equated, issue a warning.
434 // Check that we are in a RISC section
437 error(".equr/.regequ must be defined in .gpu/.dsp section");
441 // Check for register to equate to
442 if ((*tok >= KW_R0) && (*tok <= KW_R31))
444 // sy->sattre = EQUATEDREG | RISCSYM; // Mark as equated register
445 sy->sattre = EQUATEDREG; // Mark as equated register
446 riscreg = (*tok - KW_R0);
447 //is there any reason to do this, since we're putting this in svalue?
448 //i'm thinking, no. Let's test that out! :-D
449 // sy->sattre |= (riscreg << 8); // Store register number
450 //everything seems to build fine without it... We'll leave it here Just In Case(tm)
452 #define DEBODGE_REGBANK
453 #ifdef DEBODGE_REGBANK
454 // Default is current state of "regbank"
455 registerbank = regbank;
457 // Default is no register bank specified
458 registerbank = BANK_N;
461 // Check for ",<bank #>" override notation
462 if ((tok[1] == ',') && (tok[2] == CONST))
464 // Advance token pointer to the constant
467 // Anything other than a 0 or a 1 will result in "No Bank"
469 registerbank = BANK_0;
471 registerbank = BANK_1;
474 #ifdef DEBODGE_REGBANK
475 sy->sattre |= registerbank; // Store register bank
477 // What needs to happen here is to prime registerbank with regbank, then use
478 // registerbank down below for the bank marking.
479 #warning "!!! regbank <-> registerbank confusion here !!!"
480 // The question here is why, if we're allowed to override the ".regbankN" rules
481 // above, then why is it using the one set by the directive in the extended
482 // attributes and not in what ends up in symbol->svalue?
483 // ".regbankN" is not an original Madmac directive, so it's suspect
484 sy->sattre |= regbank; // Store register bank
486 eattr = ABS | DEFINED | GLOBAL;
487 // & what does this $80000080 constant mean???
488 // eval = 0x80000080 + (riscreg) + (registerbank << 8);
492 // Checking for a register symbol
493 else if (tok[0] == SYMBOL)
495 sy2 = lookup(string[tok[1]], LABEL, j);
497 // Make sure symbol is a valid equreg
498 if (!sy2 || !(sy2->sattre & EQUATEDREG))
500 error("invalid GPU/DSP .equr/.regequ definition");
505 eattr = ABS | DEFINED | GLOBAL; // Copy symbols attributes
506 sy->sattre = sy2->sattre;
507 eval = (sy2->svalue & 0xFFFFF0FF);
513 error("invalid GPU/DSP .equr/.regequ definition");
517 else if (equtyp == REG)
519 if (reglist(&rmask) < 0)
523 eattr = ABS | DEFINED;
525 else if (equtyp == CCDEF)
527 sy->sattre |= EQUATEDCC;
528 eattr = ABS | DEFINED | GLOBAL;
530 if (tok[0] == SYMBOL)
532 sy2 = lookup(string[tok[1]], LABEL, j);
534 if (!sy2 || !(sy2->sattre & EQUATEDCC))
536 error("invalid gpu/dsp .ccdef definition");
541 eattr = ABS | DEFINED | GLOBAL;
542 sy->sattre = sy2->sattre;
547 else if (expr(exprbuf, &eval, &eattr, &esym) != OK)
551 else if (*tok == SYMBOL)
553 sy2 = lookup(string[tok[1]], LABEL, j);
555 if (sy2 && (sy2->sattre & EQUATEDREG))
557 sy->stype = sy2->stype;
558 sy->sattr = sy2->sattr;
559 sy->sattre = sy2->sattre;
560 //ICK sy->svalue = (sy2->svalue & 0xFFFFF0FF);
561 sy->svalue = sy2->svalue;
564 else if (expr(exprbuf, &eval, &eattr, &esym) != OK)
567 else if (expr(exprbuf, &eval, &eattr, &esym) != OK)
570 if (!(eattr & DEFINED))
576 sy->sattr |= eattr | EQUATED; // Symbol inherits value and attributes
579 if (list_flag) // Put value in listing
582 at_eol(); // Must be at EOL now
589 // Non-zero == error occurred
590 if (HandleLabel(label, labtyp) != 0)
598 // If we're in 6502 mode and are still in need of a mnemonic, then search
599 // for valid 6502 mnemonic.
600 if (m6502 && (state < 0 || state >= 1000))
603 state = kmatch(opname, mpbase, mpcheck, mptab, mpaccept);
605 for(state=0, p=opname; state>= 0; )
607 j = mpbase[state] + tolowertab[*p];
609 if (mpcheck[j] != state) // Reject, character doesn't match
611 state = -1; // No match
616 { // Must accept or reject at EOS
617 state = mpaccept[j]; // (-1 on no terminal match)
625 // Call 6502 code generator if we found a mnemonic
628 m6502cg(state - 2000);
633 // If we are in GPU or DSP mode and still in need of a mnemonic then search
635 if ((rgpu || rdsp) && (state < 0 || state >= 1000))
637 for(state=0, p=opname; state>=0;)
639 j = mrbase[state] + (int)tolowertab[*p];
641 // Reject, character doesn't match
642 if (mrcheck[j] != state)
644 state = -1; // No match
648 // Must accept or reject at EOS
651 state = mraccept[j]; // (-1 on no terminal match)
658 // Call RISC code generator if we found a mnemonic
661 GenerateRISCCode(state);
666 // Invoke macro or complain about bad mnemonic
669 if ((sy = lookup(opname, MACRO, 0)) != NULL)
670 InvokeMacro(sy, siz);
672 error("unknown op '%s'", opname);
677 // Call directive handlers
680 (*dirtab[state])(siz);
685 // o can't deposit instrs in BSS or ABS
686 // o do automatic .EVEN for instrs
687 // o allocate space for largest possible instr
688 // o can't do ".b" operations with an address register
691 error("cannot initialize non-storage (BSS) section");
695 if (sloc & 1) // Automatic .even
698 if (challoc - ch_size < 18) // Make sure have space in current chunk
701 m = &machtab[state - 1000];
703 // Call special-mode handler
704 if (m->mnattr & CGSPECIAL)
706 (*m->mnfunc)(m->mninst, siz);
710 if (amode(1) < 0) // Parse 0, 1 or 2 addr modes
716 amsk0 = amsktab[am0];
717 amsk1 = amsktab[am1];
719 // Catch attempts to use ".B" with an address register (yes, this check
720 // does work at this level)
721 if (siz == SIZB && (am0 == AREG || am1 == AREG))
723 error("cannot use '.b' with an address register");
727 // Keep a backup of chptr (used for optimisations during codegen)
728 chptr_opcode = chptr;
732 if ((m->mnattr & siz) && (amsk0 & m->mn0) != 0 && (amsk1 & m->mn1) != 0)
734 (*m->mnfunc)(m->mninst, siz);
738 m = &machtab[m->mncont];
744 // Handle the creation of labels
746 int HandleLabel(char * label, int labelType)
748 // Check for dot in front of label; means this is a local label if present
749 int environment = (*label == '.' ? curenv : 0);
750 SYM * symbol = lookup(label, LABEL, environment);
754 symbol = NewSymbol(label, LABEL, environment);
756 // symbol->sattre = RISCSYM;
759 else if (symbol->sattr & DEFINED)
760 return error("multiply-defined label '%s'", label);
762 // Put symbol in "order of definition" list if it's not already in it
763 AddToSymbolDeclarationList(symbol);
767 symbol->svalue = orgaddr;
768 symbol->sattr |= ABS | DEFINED | EQUATED;
772 symbol->svalue = sloc;
773 symbol->sattr |= DEFINED | cursect;
778 if (0 == environment)
781 // Make label global if it has a double colon
782 if (labelType == DCOLON)
784 if (environment != 0)
785 return error(locgl_error);
787 symbol->sattr |= GLOBAL;
795 // .if, Start conditional assembly
805 if ((rif = f_ifent) == NULL)
806 rif = (IFENT *)malloc(sizeof(IFENT));
808 f_ifent = rif->if_prev;
810 rif->if_prev = ifent;
815 if (expr(exprbuf, &eval, &eattr, &esym) != OK)
818 if ((eattr & DEFINED) == 0)
819 return error(undef_error);
824 rif->if_state = (WORD)disabled;
830 // .else, Do alternate case for .if
836 if (rif->if_prev == NULL)
837 return error("mismatched .else");
840 disabled = rif->if_prev->if_state;
844 rif->if_state = (WORD)disabled;
850 // .endif, End of conditional assembly block
851 // This is also called by fpop() to pop levels of IFENTs in case a macro or
852 // include file exits early with `exitm' or `end'.
858 if (rif->if_prev == NULL)
859 return error("mismatched .endif");
861 ifent = rif->if_prev;
862 disabled = rif->if_prev->if_state;
863 rif->if_prev = f_ifent;