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