]> git.sur5r.net Git - cc65/blob - src/ca65/scanner.c
Added enums
[cc65] / src / ca65 / scanner.c
1 /*****************************************************************************/
2 /*                                                                           */
3 /*                                 scanner.c                                 */
4 /*                                                                           */
5 /*                  The scanner for the ca65 macroassembler                  */
6 /*                                                                           */
7 /*                                                                           */
8 /*                                                                           */
9 /* (C) 1998-2003 Ullrich von Bassewitz                                       */
10 /*               Römerstraße 52                                              */
11 /*               D-70794 Filderstadt                                         */
12 /* EMail:        uz@cc65.org                                                 */
13 /*                                                                           */
14 /*                                                                           */
15 /* This software is provided 'as-is', without any expressed or implied       */
16 /* warranty.  In no event will the authors be held liable for any damages    */
17 /* arising from the use of this software.                                    */
18 /*                                                                           */
19 /* Permission is granted to anyone to use this software for any purpose,     */
20 /* including commercial applications, and to alter it and redistribute it    */
21 /* freely, subject to the following restrictions:                            */
22 /*                                                                           */
23 /* 1. The origin of this software must not be misrepresented; you must not   */
24 /*    claim that you wrote the original software. If you use this software   */
25 /*    in a product, an acknowledgment in the product documentation would be  */
26 /*    appreciated but is not required.                                       */
27 /* 2. Altered source versions must be plainly marked as such, and must not   */
28 /*    be misrepresented as being the original software.                      */
29 /* 3. This notice may not be removed or altered from any source              */
30 /*    distribution.                                                          */
31 /*                                                                           */
32 /*****************************************************************************/
33
34
35
36 #include <stdio.h>
37 #include <stdlib.h>
38 #include <string.h>
39 #include <ctype.h>
40 #include <errno.h>
41 #include <sys/types.h>          /* EMX needs this */
42 #include <sys/stat.h>
43
44 /* common */
45 #include "addrsize.h"
46 #include "chartype.h"
47 #include "check.h"
48 #include "fname.h"
49 #include "xmalloc.h"
50
51 /* ca65 */
52 #include "condasm.h"
53 #include "error.h"
54 #include "filetab.h"
55 #include "global.h"
56 #include "incpath.h"
57 #include "instr.h"
58 #include "istack.h"
59 #include "listing.h"
60 #include "macro.h"
61 #include "toklist.h"
62 #include "scanner.h"
63
64
65
66 /*****************************************************************************/
67 /*                                   Data                                    */
68 /*****************************************************************************/
69
70
71
72 enum Token Tok = TOK_NONE;              /* Current token */
73 int WS;                                 /* Flag: Whitespace before token */
74 long IVal;                              /* Integer token attribute */
75 char SVal [MAX_STR_LEN+1];              /* String token attribute */
76
77 FilePos CurPos = { 0, 0, 0 };           /* Name and position in current file */
78
79
80
81 /* Struct to handle include files. Note: The length of the input line may
82  * not exceed 255+1, since the column is stored in the file position struct
83  * as a character. Increasing this value means changing the FilePos struct,
84  * and the read and write routines in the assembler and linker.
85  */
86 typedef struct InputFile_ InputFile;
87 struct InputFile_ {
88     FILE*           F;                  /* Input file descriptor */
89     FilePos         Pos;                /* Position in file */
90     enum Token      Tok;                /* Last token */
91     int             C;                  /* Last character */
92     char            Line[256];          /* The current input line */
93     InputFile*      Next;               /* Linked list of input files */
94 };
95
96 /* Struct to handle textual input data */
97 typedef struct InputData_ InputData;
98 struct InputData_ {
99     char*           Data;               /* Pointer to the data */
100     const char*     Pos;                /* Pointer to current position */
101     int             Malloced;           /* Memory was malloced */
102     enum Token      Tok;                /* Last token */
103     int             C;                  /* Last character */
104     InputData*      Next;               /* Linked list of input data */
105 };
106
107 /* Current input variables */
108 static InputFile* IFile         = 0;    /* Current input file */
109 static InputData* IData         = 0;    /* Current input memory data */
110 static unsigned   ICount        = 0;    /* Count of input files */
111 static int        C             = 0;    /* Current input character */
112
113 /* Force end of assembly */
114 int               ForcedEnd = 0;
115
116 /* List of dot keywords with the corresponding tokens */
117 struct DotKeyword {
118     const char* Key;                    /* MUST be first field */
119     enum Token  Tok;
120 } DotKeywords [] = {
121     { ".A16",           TOK_A16         },
122     { ".A8",            TOK_A8          },
123     { ".ADDR",          TOK_ADDR        },
124     { ".ALIGN",         TOK_ALIGN       },
125     { ".AND",           TOK_BOOLAND     },
126     { ".ASCIIZ",        TOK_ASCIIZ      },
127     { ".ASSERT",        TOK_ASSERT      },
128     { ".AUTOIMPORT",    TOK_AUTOIMPORT  },
129     { ".BITAND",        TOK_AND         },
130     { ".BITNOT",        TOK_NOT         },
131     { ".BITOR",         TOK_OR          },
132     { ".BITXOR",        TOK_XOR         },
133     { ".BLANK",         TOK_BLANK       },
134     { ".BSS",           TOK_BSS         },
135     { ".BYT",           TOK_BYTE        },
136     { ".BYTE",          TOK_BYTE        },
137     { ".CASE",          TOK_CASE        },
138     { ".CHARMAP",       TOK_CHARMAP     },
139     { ".CODE",          TOK_CODE        },
140     { ".CONCAT",        TOK_CONCAT      },
141     { ".CONDES",        TOK_CONDES      },
142     { ".CONST",         TOK_CONST       },
143     { ".CONSTRUCTOR",   TOK_CONSTRUCTOR },
144     { ".CPU",           TOK_CPU         },
145     { ".DATA",          TOK_DATA        },
146     { ".DBG",           TOK_DBG         },
147     { ".DBYT",          TOK_DBYT        },
148     { ".DEBUGINFO",     TOK_DEBUGINFO   },
149     { ".DEF",           TOK_DEFINED     },
150     { ".DEFINE",        TOK_DEFINE      },
151     { ".DEFINED",       TOK_DEFINED     },
152     { ".DESTRUCTOR",    TOK_DESTRUCTOR  },
153     { ".DWORD",         TOK_DWORD       },
154     { ".ELSE",          TOK_ELSE        },
155     { ".ELSEIF",        TOK_ELSEIF      },
156     { ".END",           TOK_END         },
157     { ".ENDENUM",       TOK_ENDENUM     },
158     { ".ENDIF",         TOK_ENDIF       },
159     { ".ENDMAC",        TOK_ENDMACRO    },
160     { ".ENDMACRO",      TOK_ENDMACRO    },
161     { ".ENDPROC",       TOK_ENDPROC     },
162     { ".ENDREP",        TOK_ENDREP      },
163     { ".ENDREPEAT",     TOK_ENDREP      },
164     { ".ENDSCOPE",      TOK_ENDSCOPE    },
165     { ".ENDSTRUCT",     TOK_ENDSTRUCT   },
166     { ".ENDUNION",      TOK_ENDUNION    },
167     { ".ENUM",          TOK_ENUM        },
168     { ".ERROR",         TOK_ERROR       },
169     { ".EXITMAC",       TOK_EXITMACRO   },
170     { ".EXITMACRO",     TOK_EXITMACRO   },
171     { ".EXPORT",        TOK_EXPORT      },
172     { ".EXPORTZP",      TOK_EXPORTZP    },
173     { ".FARADDR",       TOK_FARADDR     },
174     { ".FEATURE",       TOK_FEATURE     },
175     { ".FILEOPT",       TOK_FILEOPT     },
176     { ".FOPT",          TOK_FILEOPT     },
177     { ".FORCEIMPORT",   TOK_FORCEIMPORT },
178     { ".FORCEWORD",     TOK_FORCEWORD   },
179     { ".GLOBAL",        TOK_GLOBAL      },
180     { ".GLOBALZP",      TOK_GLOBALZP    },
181     { ".I16",           TOK_I16         },
182     { ".I8",            TOK_I8          },
183     { ".IF",            TOK_IF          },
184     { ".IFBLANK",       TOK_IFBLANK     },
185     { ".IFCONST",       TOK_IFCONST     },
186     { ".IFDEF",         TOK_IFDEF       },
187     { ".IFNBLANK",      TOK_IFNBLANK    },
188     { ".IFNCONST",      TOK_IFNCONST    },
189     { ".IFNDEF",        TOK_IFNDEF      },
190     { ".IFNREF",        TOK_IFNREF      },
191     { ".IFP02",         TOK_IFP02       },
192     { ".IFP816",        TOK_IFP816      },
193     { ".IFPC02",        TOK_IFPC02      },
194     { ".IFPSC02",       TOK_IFPSC02     },
195     { ".IFREF",         TOK_IFREF       },
196     { ".IMPORT",        TOK_IMPORT      },
197     { ".IMPORTZP",      TOK_IMPORTZP    },
198     { ".INCBIN",        TOK_INCBIN      },
199     { ".INCLUDE",       TOK_INCLUDE     },
200     { ".LEFT",          TOK_LEFT        },
201     { ".LINECONT",      TOK_LINECONT    },
202     { ".LIST",          TOK_LIST        },
203     { ".LISTBYTES",     TOK_LISTBYTES   },
204     { ".LOCAL",         TOK_LOCAL       },
205     { ".LOCALCHAR",     TOK_LOCALCHAR   },
206     { ".MAC",           TOK_MACRO       },
207     { ".MACPACK",       TOK_MACPACK     },
208     { ".MACRO",         TOK_MACRO       },
209     { ".MATCH",         TOK_MATCH       },
210     { ".MID",           TOK_MID         },
211     { ".MOD",           TOK_MOD         },
212     { ".NOT",           TOK_BOOLNOT     },
213     { ".NULL",          TOK_NULL        },
214     { ".OR",            TOK_BOOLOR      },
215     { ".ORG",           TOK_ORG         },
216     { ".OUT",           TOK_OUT         },
217     { ".P02",           TOK_P02         },
218     { ".P816",          TOK_P816        },
219     { ".PAGELEN",       TOK_PAGELENGTH  },
220     { ".PAGELENGTH",    TOK_PAGELENGTH  },
221     { ".PARAMCOUNT",    TOK_PARAMCOUNT  },
222     { ".PC02",          TOK_PC02        },
223     { ".POPSEG",        TOK_POPSEG      },
224     { ".PROC",          TOK_PROC        },
225     { ".PSC02",         TOK_PSC02       },
226     { ".PUSHSEG",       TOK_PUSHSEG     },
227     { ".REF",           TOK_REFERENCED  },
228     { ".REFERENCED",    TOK_REFERENCED  },
229     { ".RELOC",         TOK_RELOC       },
230     { ".REPEAT",        TOK_REPEAT      },
231     { ".RES",           TOK_RES         },
232     { ".RIGHT",         TOK_RIGHT       },
233     { ".RODATA",        TOK_RODATA      },
234     { ".SCOPE",         TOK_SCOPE       },
235     { ".SEGMENT",       TOK_SEGMENT     },
236     { ".SETCPU",        TOK_SETCPU      },
237     { ".SHL",           TOK_SHL         },
238     { ".SHR",           TOK_SHR         },
239     { ".SMART",         TOK_SMART       },
240     { ".STRAT",         TOK_STRAT       },
241     { ".STRING",        TOK_STRING      },
242     { ".STRLEN",        TOK_STRLEN      },
243     { ".STRUCT",        TOK_STRUCT      },
244     { ".SUNPLUS",       TOK_SUNPLUS     },
245     { ".TAG",           TOK_TAG         },
246     { ".TCOUNT",        TOK_TCOUNT      },
247     { ".TIME",          TOK_TIME        },
248     { ".UNION",         TOK_UNION       },
249     { ".VERSION",       TOK_VERSION     },
250     { ".WARNING",       TOK_WARNING     },
251     { ".WORD",          TOK_WORD        },
252     { ".XMATCH",        TOK_XMATCH      },
253     { ".XOR",           TOK_BOOLXOR     },
254     { ".ZEROPAGE",      TOK_ZEROPAGE    },
255 };
256
257
258
259 /*****************************************************************************/
260 /*                                 Forwards                                  */
261 /*****************************************************************************/
262
263
264
265 static void NextChar (void);
266 /* Read the next character from the input file */
267
268
269
270 /*****************************************************************************/
271 /*                    Character classification functions                     */
272 /*****************************************************************************/
273
274
275
276 static int IsIdChar (int C)
277 /* Return true if the character is a valid character for an identifier */
278 {
279     return IsAlNum (C)                  ||
280            (C == '_')                   ||
281            (C == '@' && AtInIdents)     ||
282            (C == '$' && DollarInIdents);
283 }
284
285
286
287 static int IsIdStart (int C)
288 /* Return true if the character may start an identifier */
289 {
290     return IsAlpha (C) || C == '_';
291 }
292
293
294
295 /*****************************************************************************/
296 /*                                   Code                                    */
297 /*****************************************************************************/
298
299
300
301 void NewInputFile (const char* Name)
302 /* Open a new input file */
303 {
304     InputFile* I;
305     FILE* F;
306
307     /* First try to open the file */
308     F = fopen (Name, "r");
309     if (F == 0) {
310
311         char* PathName;
312
313         /* Error (fatal error if this is the main file) */
314         if (ICount == 0) {
315             Fatal ("Cannot open input file `%s': %s", Name, strerror (errno));
316         }
317
318         /* We are on include level. Search for the file in the include
319          * directories.
320          */
321         PathName = FindInclude (Name);
322         if (PathName == 0 || (F = fopen (PathName, "r")) == 0) {
323             /* Not found or cannot open, print an error and bail out */
324             Error ("Cannot open include file `%s': %s", Name, strerror (errno));
325         }
326
327         /* Free the allocated memory */
328         xfree (PathName);
329
330     }
331
332     /* check again if we do now have an open file */
333     if (F != 0) {
334
335         unsigned FileIdx;
336
337         /* Stat the file and remember the values */
338         struct stat Buf;
339         if (fstat (fileno (F), &Buf) != 0) {
340             Fatal ("Cannot stat input file `%s': %s", Name, strerror (errno));
341         }
342
343         /* Add the file to the input file table and remember the index */
344         FileIdx = AddFile (Name, Buf.st_size, Buf.st_mtime);
345
346         /* Create a new state variable and initialize it */
347         I           = xmalloc (sizeof (*I));
348         I->F        = F;
349         I->Pos.Line = 0;
350         I->Pos.Col  = 0;
351         I->Pos.Name = FileIdx;
352         I->Tok      = Tok;
353         I->C        = C;
354         I->Line[0]  = '\0';
355
356         /* Use the new file */
357         I->Next     = IFile;
358         IFile       = I;
359         ++ICount;
360
361         /* Prime the pump */
362         NextChar ();
363     }
364 }
365
366
367
368 void DoneInputFile (void)
369 /* Close the current input file */
370 {
371     InputFile* I;
372
373     /* Restore the old token */
374     Tok = IFile->Tok;
375     C   = IFile->C;
376
377     /* Save a pointer to the current struct, then set it back */
378     I     = IFile;
379     IFile = I->Next;
380
381     /* Cleanup the current stuff */
382     fclose (I->F);
383     xfree (I);
384     --ICount;
385 }
386
387
388
389 void NewInputData (char* Data, int Malloced)
390 /* Add a chunk of input data to the input stream */
391 {
392     InputData* I;
393
394     /* Create a new state variable and initialize it */
395     I           = xmalloc (sizeof (*I));
396     I->Data     = Data;
397     I->Pos      = Data;
398     I->Malloced = Malloced;
399     I->Tok      = Tok;
400     I->C        = C;
401
402     /* Use the new data */
403     I->Next     = IData;
404     IData       = I;
405
406     /* Prime the pump */
407     NextChar ();
408 }
409
410
411
412 static void DoneInputData (void)
413 /* End the current input data stream */
414 {
415     InputData* I;
416
417     /* Restore the old token */
418     Tok = IData->Tok;
419     C   = IData->C;
420
421     /* Save a pointer to the current struct, then set it back */
422     I     = IData;
423     IData = I->Next;
424
425     /* Cleanup the current stuff */
426     if (I->Malloced) {
427         xfree (I->Data);
428     }
429     xfree (I);
430 }
431
432
433
434 static unsigned DigitVal (unsigned char C)
435 /* Convert a digit into it's numerical representation */
436 {
437     if (IsDigit (C)) {
438         return C - '0';
439     } else {
440         return toupper (C) - 'A' + 10;
441     }
442 }
443
444
445
446 static void NextChar (void)
447 /* Read the next character from the input file */
448 {
449     /* If we have an input data structure, read from there */
450     if (IData) {
451
452         C = *IData->Pos++;
453         if (C == '\0') {
454             /* End of input data, will set to last file char */
455             DoneInputData ();
456         }
457
458     } else {
459
460         /* Check for end of line, read the next line if needed */
461         while (IFile->Line [IFile->Pos.Col] == '\0') {
462
463             /* End of current line reached, read next line */
464             if (fgets (IFile->Line, sizeof (IFile->Line), IFile->F) == 0) {
465                 /* End of file. Add an empty line to the listing. This is a
466                  * small hack needed to keep the PC output in sync.
467                  */
468                 NewListingLine ("", IFile->Pos.Name, ICount);
469                 C = EOF;
470                 return;
471             }
472
473             /* One more line */
474             IFile->Pos.Line++;
475             IFile->Pos.Col = 0;
476
477             /* Remember the new line for the listing */
478             NewListingLine (IFile->Line, IFile->Pos.Name, ICount);
479
480         }
481
482         /* Return the next character from the file */
483         C = IFile->Line [IFile->Pos.Col++];
484
485     }
486 }
487
488
489
490 void LocaseSVal (void)
491 /* Make SVal lower case */
492 {
493     unsigned I = 0;
494     while (SVal [I]) {
495         SVal [I] = tolower (SVal [I]);
496         ++I;
497     }
498 }
499
500
501
502 void UpcaseSVal (void)
503 /* Make SVal upper case */
504 {
505     unsigned I = 0;
506     while (SVal [I]) {
507         SVal [I] = toupper (SVal [I]);
508         ++I;
509     }
510 }
511
512
513
514 static int CmpDotKeyword (const void* K1, const void* K2)
515 /* Compare function for the dot keyword search */
516 {
517     return strcmp (((struct DotKeyword*)K1)->Key, ((struct DotKeyword*)K2)->Key);
518 }
519
520
521
522 static unsigned char FindDotKeyword (void)
523 /* Find the dot keyword in SVal. Return the corresponding token if found,
524  * return TOK_NONE if not found.
525  */
526 {
527     static const struct DotKeyword K = { SVal, 0 };
528     struct DotKeyword* R;
529
530     /* If we aren't in ignore case mode, we have to uppercase the keyword */
531     if (!IgnoreCase) {
532         UpcaseSVal ();
533     }
534
535     /* Search for the keyword */
536     R = bsearch (&K, DotKeywords, sizeof (DotKeywords) / sizeof (DotKeywords [0]),
537                  sizeof (DotKeywords [0]), CmpDotKeyword);
538     if (R != 0) {
539         return R->Tok;
540     } else {
541         return TOK_NONE;
542     }
543 }
544
545
546
547 static void ReadIdent (unsigned Index)
548 /* Read an identifier from the current input position into Ident. Filling SVal
549  * starts at Index with the current character in C. It is assumed that any
550  * characters already filled in are ok, and the character in C is checked.
551  */
552 {
553     /* Read the identifier */
554     do {
555         if (Index < MAX_STR_LEN) {
556             SVal [Index++] = C;
557         }
558         NextChar ();
559     } while (IsIdChar (C));
560     SVal [Index] = '\0';
561
562     /* If we should ignore case, convert the identifier to upper case */
563     if (IgnoreCase) {
564         UpcaseSVal ();
565     }
566 }
567
568
569
570 static unsigned ReadStringConst (int StringTerm)
571 /* Read a string constant into SVal. Check for maximum string length and all
572  * other stuff. The length of the string is returned.
573  */
574 {
575     unsigned I;
576
577     /* Skip the leading string terminator */
578     NextChar ();
579
580     /* Read the string */
581     I = 0;
582     while (1) {
583         if (C == StringTerm) {
584             break;
585         }
586         if (C == '\n' || C == EOF) {
587             Error ("Newline in string constant");
588             break;
589         }
590
591         /* Check for string length, print an error message once */
592         if (I == MAX_STR_LEN) {
593             Error ("Maximum string size exceeded");
594         } else if (I < MAX_STR_LEN) {
595             SVal [I] = C;
596         }
597         ++I;
598
599         /* Skip the character */
600         NextChar ();
601     }
602
603     /* Skip the trailing terminator */
604     NextChar ();
605
606     /* Terminate the string */
607     if (I >= MAX_STR_LEN) {
608         I = MAX_STR_LEN;
609     }
610     SVal [I] = '\0';
611
612     /* Return the length of the string */
613     return I;
614 }
615
616
617
618 void NextRawTok (void)
619 /* Read the next raw token from the input stream */
620 {
621     /* If we've a forced end of assembly, don't read further */
622     if (ForcedEnd) {
623         Tok = TOK_EOF;
624         return;
625     }
626
627 Restart:
628     /* Check if we have tokens from another input source */
629     if (InputFromStack ()) {
630         return;
631     }
632
633 Again:
634     /* Skip whitespace, remember if we had some */
635     if ((WS = IsBlank (C)) != 0) {
636         do {
637             NextChar ();
638         } while (IsBlank (C));
639     }
640
641     /* If we're reading from the file, update the location from where the
642      * next token will be read. If we're reading from input data, keep the
643      * current position.
644      */
645     if (IData == 0) {
646         CurPos = IFile->Pos;
647     }
648
649     /* Hex number or PC symbol? */
650     if (C == '$') {
651         NextChar ();
652
653         /* Hex digit must follow or DollarIsPC must be enabled */
654         if (!IsXDigit (C)) {
655             if (DollarIsPC) {
656                 Tok = TOK_PC;
657                 return;
658             } else {
659                 Error ("Hexadecimal digit expected");
660             }
661         }
662
663         /* Read the number */
664         IVal = 0;
665         while (IsXDigit (C)) {
666             if (IVal & 0xF0000000) {
667                 Error ("Overflow in hexadecimal number");
668                 IVal = 0;
669             }
670             IVal = (IVal << 4) + DigitVal (C);
671             NextChar ();
672         }
673
674         /* This is an integer constant */
675         Tok = TOK_INTCON;
676         return;
677     }
678
679     /* Binary number? */
680     if (C == '%') {
681         NextChar ();
682
683         /* 0 or 1 must follow */
684         if (!IsBDigit (C)) {
685             Error ("Binary digit expected");
686         }
687
688         /* Read the number */
689         IVal = 0;
690         while (IsBDigit (C)) {
691             if (IVal & 0x80000000) {
692                 Error ("Overflow in binary number");
693                 IVal = 0;
694             }
695             IVal = (IVal << 1) + DigitVal (C);
696             NextChar ();
697         }
698
699         /* This is an integer constant */
700         Tok = TOK_INTCON;
701         return;
702     }
703
704     /* Decimal number? */
705     if (IsDigit (C)) {
706
707         /* Read the number */
708         IVal = 0;
709         while (IsDigit (C)) {
710             if (IVal > (long) (0xFFFFFFFFUL / 10)) {
711                 Error ("Overflow in decimal number");
712                 IVal = 0;
713             }
714             IVal = (IVal * 10) + DigitVal (C);
715             NextChar ();
716         }
717
718         /* This is an integer constant */
719         Tok = TOK_INTCON;
720         return;
721     }
722
723     /* Control command? */
724     if (C == '.') {
725
726         /* Remember and skip the dot */
727         NextChar ();
728
729         /* Check if it's just a dot */
730         if (!IsIdStart (C)) {
731
732             /* Just a dot */
733             Tok = TOK_DOT;
734
735         } else {
736
737             /* Read the remainder of the identifier */
738             SVal[0] = '.';
739             ReadIdent (1);
740
741             /* Dot keyword, search for it */
742             Tok = FindDotKeyword ();
743             if (Tok == TOK_NONE) {
744                 /* Not found */
745                 if (LeadingDotInIdents) {
746                     /* An identifier with a dot */
747                     Tok = TOK_IDENT;
748                 } else {
749                     /* Invalid pseudo instruction */
750                     Error ("`%s' is not a recognized control command", SVal);
751                     goto Again;
752                 }
753             }
754
755         }
756         return;
757     }
758
759     /* Local symbol? */
760     if (C == LocalStart) {
761
762         /* Read the identifier */
763         ReadIdent (0);
764
765         /* Start character alone is not enough */
766         if (SVal [1] == '\0') {
767             Error ("Invalid cheap local symbol");
768             goto Again;
769         }
770
771         /* An identifier */
772         Tok = TOK_IDENT;
773         return;
774     }
775
776
777     /* Identifier or keyword? */
778     if (IsIdStart (C)) {
779
780         /* Read the identifier */
781         ReadIdent (0);
782
783         /* Check for special names */
784         if (SVal[1] == '\0') {
785             switch (toupper (SVal [0])) {
786
787                 case 'A':
788                     if (C == ':') {
789                         NextChar ();
790                         Tok = TOK_OVERRIDE_ABS;
791                     } else {
792                         Tok = TOK_A;
793                     }
794                     return;
795
796                 case 'F':
797                     if (C == ':') {
798                         NextChar ();
799                         Tok = TOK_OVERRIDE_FAR;
800                     } else {
801                         Tok = TOK_IDENT;
802                     }
803                     return;
804
805                 case 'S':
806                     Tok = TOK_S;
807                     return;
808
809                 case 'X':
810                     Tok = TOK_X;
811                     return;
812
813                 case 'Y':
814                     Tok = TOK_Y;
815                     return;
816
817                 case 'Z':
818                     if (C == ':') {
819                         NextChar ();
820                         Tok = TOK_OVERRIDE_ZP;
821                     } else {
822                         Tok = TOK_IDENT;
823                     }
824                     return;
825
826                 default:
827                     Tok = TOK_IDENT;
828                     return;
829             }
830         }
831
832         /* Search for an opcode */
833         IVal = FindInstruction (SVal);
834         if (IVal >= 0) {
835             /* This is a mnemonic */
836             Tok = TOK_MNEMO;
837         } else if (IsDefine (SVal)) {
838             /* This is a define style macro - expand it */
839             MacExpandStart ();
840             goto Restart;
841         } else {
842             /* An identifier */
843             Tok = TOK_IDENT;
844         }
845         return;
846     }
847
848     /* Ok, let's do the switch */
849 CharAgain:
850     switch (C) {
851
852         case '+':
853             NextChar ();
854             Tok = TOK_PLUS;
855             return;
856
857         case '-':
858             NextChar ();
859             Tok = TOK_MINUS;
860             return;
861
862         case '/':
863             NextChar ();
864             Tok = TOK_DIV;
865             return;
866
867         case '*':
868             NextChar ();
869             Tok = TOK_MUL;
870             return;
871
872         case '^':
873             NextChar ();
874             Tok = TOK_XOR;
875             return;
876
877         case '&':
878             NextChar ();
879             if (C == '&') {
880                 NextChar ();
881                 Tok = TOK_BOOLAND;
882             } else {
883                 Tok = TOK_AND;
884             }
885             return;
886
887         case '|':
888             NextChar ();
889             if (C == '|') {
890                 NextChar ();
891                 Tok = TOK_BOOLOR;
892             } else {
893                 Tok = TOK_OR;
894             }
895             return;
896
897         case ':':
898             NextChar ();
899             switch (C) {
900
901                 case ':':
902                     NextChar ();
903                     Tok = TOK_NAMESPACE;
904                     break;
905
906                 case '-':
907                     IVal = 0;
908                     do {
909                         --IVal;
910                         NextChar ();
911                     } while (C == '-');
912                     Tok = TOK_ULABEL;
913                     break;
914
915                 case '+':
916                     IVal = 0;
917                     do {
918                         ++IVal;
919                         NextChar ();
920                     } while (C == '+');
921                     Tok = TOK_ULABEL;
922                     break;
923
924                 case '=':
925                     NextChar ();
926                     Tok = TOK_ASSIGN;
927                     break;
928
929                 default:
930                     Tok = TOK_COLON;
931                     break;
932             }
933             return;
934
935         case ',':
936             NextChar ();
937             Tok = TOK_COMMA;
938             return;
939
940         case ';':
941             NextChar ();
942             while (C != '\n' && C != EOF) {
943                 NextChar ();
944             }
945             goto CharAgain;
946
947         case '#':
948             NextChar ();
949             Tok = TOK_HASH;
950             return;
951
952         case '(':
953             NextChar ();
954             Tok = TOK_LPAREN;
955             return;
956
957         case ')':
958             NextChar ();
959             Tok = TOK_RPAREN;
960             return;
961
962         case '[':
963             NextChar ();
964             Tok = TOK_LBRACK;
965             return;
966
967         case ']':
968             NextChar ();
969             Tok = TOK_RBRACK;
970             return;
971
972         case '<':
973             NextChar ();
974             if (C == '=') {
975                 NextChar ();
976                 Tok = TOK_LE;
977             } else if (C == '<') {
978                 NextChar ();
979                 Tok = TOK_SHL;
980             } else if (C == '>') {
981                 NextChar ();
982                 Tok = TOK_NE;
983             } else {
984                 Tok = TOK_LT;
985             }
986             return;
987
988         case '=':
989             NextChar ();
990             Tok = TOK_EQ;
991             return;
992
993         case '!':
994             NextChar ();
995             Tok = TOK_BOOLNOT;
996             return;
997
998         case '>':
999             NextChar ();
1000             if (C == '=') {
1001                 NextChar ();
1002                 Tok = TOK_GE;
1003             } else if (C == '>') {
1004                 NextChar ();
1005                 Tok = TOK_SHR;
1006             } else {
1007                 Tok = TOK_GT;
1008             }
1009             return;
1010
1011         case '~':
1012             NextChar ();
1013             Tok = TOK_NOT;
1014             return;
1015
1016         case '\'':
1017             /* Hack: If we allow ' as terminating character for strings, read
1018              * the following stuff as a string, and check for a one character
1019              * string later.
1020              */
1021             if (LooseStringTerm) {
1022                 if (ReadStringConst ('\'') == 1) {
1023                     IVal = SVal[0];
1024                     Tok = TOK_CHARCON;
1025                 } else {
1026                     Tok = TOK_STRCON;
1027                 }
1028             } else {
1029                 /* Always a character constant */
1030                 NextChar ();
1031                 if (C == '\n' || C == EOF) {
1032                     Error ("Illegal character constant");
1033                     goto CharAgain;
1034                 }
1035                 IVal = C;
1036                 Tok = TOK_CHARCON;
1037                 NextChar ();
1038                 if (C != '\'') {
1039                     Error ("Illegal character constant");
1040                 } else {
1041                     NextChar ();
1042                 }
1043             }
1044             return;
1045
1046         case '\"':
1047             ReadStringConst ('\"');
1048             Tok = TOK_STRCON;
1049             return;
1050
1051         case '\\':
1052             /* Line continuation? */
1053             if (LineCont) {
1054                 NextChar ();
1055                 if (C == '\n') {
1056                     /* Handle as white space */
1057                     NextChar ();
1058                     C = ' ';
1059                     goto Again;
1060                 }
1061             }
1062             break;
1063
1064         case '\n':
1065             NextChar ();
1066             Tok = TOK_SEP;
1067             return;
1068
1069         case EOF:
1070             /* Check if we have any open .IFs in this file */
1071             CheckOpenIfs ();
1072             /* Check if we have any open token lists in this file */
1073             CheckInputStack ();
1074
1075             /* If this was an include file, then close it and handle like a
1076              * separator. Do not close the main file, but return EOF.
1077              */
1078             if (ICount > 1) {
1079                 DoneInputFile ();
1080             } else {
1081                 Tok = TOK_EOF;
1082             }
1083             return;
1084
1085     }
1086
1087     /* If we go here, we could not identify the current character. Skip it
1088      * and try again.
1089      */
1090     Error ("Invalid input character: 0x%02X", C & 0xFF);
1091     NextChar ();
1092     goto Again;
1093 }
1094
1095
1096
1097 int TokHasSVal (enum Token Tok)
1098 /* Return true if the given token has an attached SVal */
1099 {
1100     return (Tok == TOK_IDENT || Tok == TOK_STRCON);
1101 }
1102
1103
1104
1105 int TokHasIVal (enum Token Tok)
1106 /* Return true if the given token has an attached IVal */
1107 {
1108     return (Tok == TOK_INTCON || Tok == TOK_CHARCON || Tok == TOK_MNEMO);
1109 }
1110
1111
1112
1113 int GetSubKey (const char** Keys, unsigned Count)
1114 /* Search for a subkey in a table of keywords. The current token must be an
1115  * identifier and all keys must be in upper case. The identifier will be
1116  * uppercased in the process. The function returns the index of the keyword,
1117  * or -1 if the keyword was not found.
1118  */
1119 {
1120     unsigned I;
1121
1122     /* Must have an identifier */
1123     PRECONDITION (Tok == TOK_IDENT);
1124
1125     /* If we aren't in ignore case mode, we have to uppercase the identifier */
1126     if (!IgnoreCase) {
1127         UpcaseSVal ();
1128     }
1129
1130     /* Do a linear search (a binary search is not worth the effort) */
1131     for (I = 0; I < Count; ++I) {
1132         if (strcmp (SVal, Keys [I]) == 0) {
1133             /* Found it */
1134             return I;
1135         }
1136     }
1137
1138     /* Not found */
1139     return -1;
1140 }
1141
1142
1143
1144 unsigned char ParseAddrSize (void)
1145 /* Check if the next token is a keyword that denotes an address size specifier.
1146  * If so, return the corresponding address size constant, otherwise output an
1147  * error message and return ADDR_SIZE_DEFAULT.
1148  */
1149 {
1150     static const char* Keys[] = {
1151         "DIRECT", "ZEROPAGE", "ZP",
1152         "ABSOLUTE", "ABS", "NEAR",
1153         "FAR",
1154     };
1155
1156     /* Check for an identifier */
1157     if (Tok != TOK_IDENT) {
1158         Error ("Address size specifier expected");
1159         return ADDR_SIZE_DEFAULT;
1160     }
1161
1162     /* Search for the attribute */
1163     switch (GetSubKey (Keys, sizeof (Keys) / sizeof (Keys [0]))) {
1164         case 0:
1165         case 1:
1166         case 2: return ADDR_SIZE_ZP;
1167         case 3:
1168         case 4:
1169         case 5: return ADDR_SIZE_ABS;
1170         case 6: return ADDR_SIZE_FAR;
1171         default:
1172             Error ("Address size specifier expected");
1173             return ADDR_SIZE_DEFAULT;
1174     }
1175 }
1176
1177
1178
1179 void InitScanner (const char* InFile)
1180 /* Initialize the scanner, open the given input file */
1181 {
1182     /* Open the input file */
1183     NewInputFile (InFile);
1184 }
1185
1186
1187
1188 void DoneScanner (void)
1189 /* Release scanner resources */
1190 {
1191     DoneInputFile ();
1192 }
1193
1194
1195