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