]> git.sur5r.net Git - cc65/blobdiff - src/ca65/pseudo.c
Renamed variables for better readability.
[cc65] / src / ca65 / pseudo.c
index 60d78d643f99281549d38d5ec193a69765863d55..3a2e391d39796aa600a99d896c3c4ff0a46b5c12 100644 (file)
@@ -6,7 +6,7 @@
 /*                                                                           */
 /*                                                                           */
 /*                                                                           */
-/* (C) 1998-2008, Ullrich von Bassewitz                                      */
+/* (C) 1998-2011, Ullrich von Bassewitz                                      */
 /*                Roemerstrasse 52                                           */
 /*                D-70794 Filderstadt                                        */
 /* EMail:         uz@cc65.org                                                */
 #include <string.h>
 #include <ctype.h>
 #include <errno.h>
+#include <sys/types.h>         /* EMX needs this */
+#include <sys/stat.h>
 
 /* common */
-#include "assertdefs.h"
+#include "assertion.h"
 #include "bitops.h"
 #include "cddefs.h"
 #include "coll.h"
+#include "intstack.h"
+#include "scopedefs.h"
 #include "symdefs.h"
 #include "tgttrans.h"
 #include "xmalloc.h"
@@ -57,6 +61,7 @@
 #include "error.h"
 #include "expr.h"
 #include "feature.h"
+#include "filetab.h"
 #include "global.h"
 #include "incpath.h"
 #include "instr.h"
@@ -86,6 +91,9 @@
 /* Keyword we're about to handle */
 static StrBuf Keyword = STATIC_STRBUF_INITIALIZER;
 
+/* CPU stack */
+static IntStack CPUStack = STATIC_INTSTACK_INITIALIZER;
+
 /* Segment stack */
 #define MAX_PUSHED_SEGMENTS     16
 static Collection SegStack = STATIC_COLLECTION_INITIALIZER;
@@ -124,7 +132,7 @@ static unsigned char OptionalAddrSize (void)
  */
 {
     unsigned AddrSize = ADDR_SIZE_DEFAULT;
-    if (Tok == TOK_COLON) {
+    if (CurTok.Tok == TOK_COLON) {
         NextTok ();
         AddrSize = ParseAddrSize ();
         if (!ValidAddrSizeForCPU (AddrSize)) {
@@ -147,20 +155,20 @@ static void SetBoolOption (unsigned char* Flag)
        "ON",
     };
 
-    if (Tok == TOK_PLUS) {
+    if (CurTok.Tok == TOK_PLUS) {
                *Flag = 1;
        NextTok ();
-    } else if (Tok == TOK_MINUS) {
+    } else if (CurTok.Tok == TOK_MINUS) {
        *Flag = 0;
        NextTok ();
-    } else if (Tok == TOK_IDENT) {
+    } else if (CurTok.Tok == TOK_IDENT) {
                /* Map the keyword to a number */
                switch (GetSubKey (Keys, sizeof (Keys) / sizeof (Keys [0]))) {
            case 0:     *Flag = 0; NextTok ();                  break;
            case 1:     *Flag = 1; NextTok ();                  break;
            default:    ErrorSkip ("`on' or `off' expected");   break;
        }
-    } else if (TokIsSep (Tok)) {
+    } else if (TokIsSep (CurTok.Tok)) {
        /* Without anything assume switch on */
        *Flag = 1;
     } else {
@@ -176,10 +184,10 @@ static void ExportWithAssign (SymEntry* Sym, unsigned char AddrSize, unsigned Fl
     /* The name and optional address size spec may be followed by an assignment
      * or equal token.
      */
-    if (Tok == TOK_ASSIGN || Tok == TOK_EQ) {
+    if (CurTok.Tok == TOK_ASSIGN || CurTok.Tok == TOK_EQ) {
 
         /* Assignment means the symbol is a label */
-        if (Tok == TOK_ASSIGN) {
+        if (CurTok.Tok == TOK_ASSIGN) {
             Flags |= SF_LABEL;
         }
 
@@ -207,13 +215,13 @@ static void ExportImport (void (*Func) (SymEntry*, unsigned char, unsigned),
     while (1) {
 
         /* We need an identifier here */
-       if (Tok != TOK_IDENT) {
+       if (CurTok.Tok != TOK_IDENT) {
                    ErrorSkip ("Identifier expected");
            return;
        }
 
         /* Find the symbol table entry, allocate a new one if necessary */
-        Sym = SymFind (CurrentScope, &SVal, SYM_ALLOC_NEW);
+        Sym = SymFind (CurrentScope, &CurTok.SVal, SYM_ALLOC_NEW);
 
         /* Skip the name */
         NextTok ();
@@ -228,7 +236,7 @@ static void ExportImport (void (*Func) (SymEntry*, unsigned char, unsigned),
         Func (Sym, AddrSize, Flags);
 
         /* More symbols? */
-       if (Tok == TOK_COMMA) {
+       if (CurTok.Tok == TOK_COMMA) {
            NextTok ();
        } else {
            break;
@@ -243,7 +251,7 @@ static long IntArg (long Min, long Max)
  * and return -1 in this case.
  */
 {
-    if (Tok == TOK_IDENT && SB_CompareStr (&SVal, "unlimited") == 0) {
+    if (CurTok.Tok == TOK_IDENT && SB_CompareStr (&CurTok.SVal, "unlimited") == 0) {
        NextTok ();
        return -1;
     } else {
@@ -268,7 +276,7 @@ static void ConDes (const StrBuf* Name, unsigned Type)
     SymEntry* Sym = SymFind (CurrentScope, Name, SYM_ALLOC_NEW);
 
     /* Optional constructor priority */
-    if (Tok == TOK_COMMA) {
+    if (CurTok.Tok == TOK_COMMA) {
        /* Priority value follows */
        NextTok ();
        Prio = ConstExpression ();
@@ -330,7 +338,7 @@ static void DoAddr (void)
            /* Do a range check */
            EmitWord (Expression ());
                }
-       if (Tok != TOK_COMMA) {
+       if (CurTok.Tok != TOK_COMMA) {
            break;
        } else {
            NextTok ();
@@ -355,7 +363,7 @@ static void DoAlign (void)
     }
 
     /* Optional value follows */
-    if (Tok == TOK_COMMA) {
+    if (CurTok.Tok == TOK_COMMA) {
        NextTok ();
        Val = ConstExpression ();
        /* We need a byte value here */
@@ -383,16 +391,16 @@ static void DoASCIIZ (void)
 {
     while (1) {
        /* Must have a string constant */
-       if (Tok != TOK_STRCON) {
+       if (CurTok.Tok != TOK_STRCON) {
                    ErrorSkip ("String constant expected");
            return;
        }
 
        /* Translate into target charset and emit */
-               TgtTranslateStrBuf (&SVal);
-               EmitStrBuf (&SVal);
+               TgtTranslateStrBuf (&CurTok.SVal);
+               EmitStrBuf (&CurTok.SVal);
        NextTok ();
-       if (Tok == TOK_COMMA) {
+       if (CurTok.Tok == TOK_COMMA) {
            NextTok ();
        } else {
            break;
@@ -408,23 +416,24 @@ static void DoAssert (void)
 {
     static const char* ActionTab [] = {
        "WARN", "WARNING",
-        "ERROR"
+        "ERROR",
+        "LDWARN", "LDWARNING",
+        "LDERROR",
     };
 
-    int      Action;
-    unsigned Msg;
+    AssertAction Action;
+    unsigned     Msg;
 
     /* First we have the expression that has to evaluated */
     ExprNode* Expr = Expression ();
     ConsumeComma ();
 
     /* Action follows */
-    if (Tok != TOK_IDENT) {
+    if (CurTok.Tok != TOK_IDENT) {
         ErrorSkip ("Identifier expected");
         return;
     }
-    Action = GetSubKey (ActionTab, sizeof (ActionTab) / sizeof (ActionTab[0]));
-    switch (Action) {
+    switch (GetSubKey (ActionTab, sizeof (ActionTab) / sizeof (ActionTab[0]))) {
 
         case 0:
         case 1:
@@ -437,21 +446,36 @@ static void DoAssert (void)
             Action = ASSERT_ACT_ERROR;
             break;
 
+        case 3:
+        case 4:
+            /* Linker warning */
+            Action = ASSERT_ACT_LDWARN;
+            break;
+
+        case 5:
+            /* Linker error */
+            Action = ASSERT_ACT_LDERROR;
+            break;
+
         default:
             Error ("Illegal assert action specifier");
+            /* Use lderror - there won't be an .o file anyway */
+            Action = ASSERT_ACT_LDERROR;
+            break;
+
     }
     NextTok ();
 
     /* We can have an optional message. If no message is present, use
      * "Assertion failed".
      */
-    if (Tok == TOK_COMMA) {
+    if (CurTok.Tok == TOK_COMMA) {
 
         /* Skip the comma */
         NextTok ();
 
         /* Read the message */
-        if (Tok != TOK_STRCON) {
+        if (CurTok.Tok != TOK_STRCON) {
             ErrorSkip ("String constant expected");
             return;
         }
@@ -459,7 +483,7 @@ static void DoAssert (void)
         /* Translate the message into a string id. We can then skip the input
          * string.
          */
-        Msg = GetStrBufId (&SVal);
+        Msg = GetStrBufId (&CurTok.SVal);
         NextTok ();
 
     } else {
@@ -470,7 +494,7 @@ static void DoAssert (void)
     }
 
     /* Remember the assertion */
-    AddAssertion (Expr, Action, Msg);
+    AddAssertion (Expr, (AssertAction) Action, Msg);
 }
 
 
@@ -482,6 +506,20 @@ static void DoAutoImport (void)
 }
 
 
+static void DoBankBytes (void)
+/* Define bytes, extracting the bank byte from each expression in the list */
+{
+    while (1) {
+        EmitByte (FuncBankByte ());
+        if (CurTok.Tok != TOK_COMMA) {
+            break;
+        } else {
+            NextTok ();
+        }
+    }
+}
+
+
 
 static void DoBss (void)
 /* Switch to the BSS segment */
@@ -495,20 +533,20 @@ static void DoByte (void)
 /* Define bytes */
 {
     while (1) {
-       if (Tok == TOK_STRCON) {
+       if (CurTok.Tok == TOK_STRCON) {
            /* A string, translate into target charset and emit */
-                   TgtTranslateStrBuf (&SVal);
-           EmitStrBuf (&SVal);
+                   TgtTranslateStrBuf (&CurTok.SVal);
+           EmitStrBuf (&CurTok.SVal);
            NextTok ();
        } else {
            EmitByte (Expression ());
        }
-       if (Tok != TOK_COMMA) {
+       if (CurTok.Tok != TOK_COMMA) {
            break;
        } else {
            NextTok ();
            /* Do smart handling of dangling comma */
-           if (Tok == TOK_SEP) {
+           if (CurTok.Tok == TOK_SEP) {
                Error ("Unexpected end of line");
                break;
            }
@@ -578,16 +616,16 @@ static void DoConDes (void)
     long Type;
 
     /* Symbol name follows */
-    if (Tok != TOK_IDENT) {
+    if (CurTok.Tok != TOK_IDENT) {
        ErrorSkip ("Identifier expected");
        return;
     }
-    SB_Copy (&Name, &SVal);
+    SB_Copy (&Name, &CurTok.SVal);
     NextTok ();
 
     /* Type follows. May be encoded as identifier or numerical */
     ConsumeComma ();
-    if (Tok == TOK_IDENT) {
+    if (CurTok.Tok == TOK_IDENT) {
 
        /* Map the following keyword to a number, then skip it */
        Type = GetSubKey (Keys, sizeof (Keys) / sizeof (Keys [0]));
@@ -627,11 +665,11 @@ static void DoConstructor (void)
     StrBuf Name = STATIC_STRBUF_INITIALIZER;
 
     /* Symbol name follows */
-    if (Tok != TOK_IDENT) {
+    if (CurTok.Tok != TOK_IDENT) {
        ErrorSkip ("Identifier expected");
        return;
     }
-    SB_Copy (&Name, &SVal);
+    SB_Copy (&Name, &CurTok.SVal);
     NextTok ();
 
     /* Parse the remainder of the line and export the symbol */
@@ -663,7 +701,7 @@ static void DoDbg (void)
 
 
     /* We expect a subkey */
-    if (Tok != TOK_IDENT) {
+    if (CurTok.Tok != TOK_IDENT) {
        ErrorSkip ("Identifier expected");
        return;
     }
@@ -690,7 +728,7 @@ static void DoDByt (void)
 {
     while (1) {
        EmitWord (GenSwapExpr (Expression ()));
-       if (Tok != TOK_COMMA) {
+       if (CurTok.Tok != TOK_COMMA) {
            break;
        } else {
            NextTok ();
@@ -716,17 +754,31 @@ static void DoDefine (void)
 
 
 
+static void DoDelMac (void)
+/* Delete a classic macro */
+{
+    /* We expect an identifier */
+    if (CurTok.Tok != TOK_IDENT) {
+       ErrorSkip ("Identifier expected");
+    } else {
+        MacUndef (&CurTok.SVal, MAC_STYLE_CLASSIC);
+        NextTok ();
+    }
+}
+
+
+
 static void DoDestructor (void)
 /* Export a symbol as destructor */
 {
     StrBuf Name = STATIC_STRBUF_INITIALIZER;
 
     /* Symbol name follows */
-    if (Tok != TOK_IDENT) {
+    if (CurTok.Tok != TOK_IDENT) {
        ErrorSkip ("Identifier expected");
        return;
     }
-    SB_Copy (&Name, &SVal);
+    SB_Copy (&Name, &CurTok.SVal);
     NextTok ();
 
     /* Parse the remainder of the line and export the symbol */
@@ -743,7 +795,7 @@ static void DoDWord (void)
 {
     while (1) {
                EmitDWord (Expression ());
-       if (Tok != TOK_COMMA) {
+       if (CurTok.Tok != TOK_COMMA) {
            break;
        } else {
            NextTok ();
@@ -765,7 +817,7 @@ static void DoEnd (void)
 static void DoEndProc (void)
 /* Leave a lexical level */
 {
-    if (GetCurrentSymTabType () != ST_PROC) {
+    if (CurrentScope->Type != SCOPE_SCOPE || CurrentScope->Label == 0) {
         /* No local scope */
         ErrorSkip ("No open .PROC");
     } else {
@@ -778,7 +830,7 @@ static void DoEndProc (void)
 static void DoEndScope (void)
 /* Leave a lexical level */
 {
-    if ( GetCurrentSymTabType () != ST_SCOPE) {
+    if (CurrentScope->Type != SCOPE_SCOPE || CurrentScope->Label != 0) {
         /* No local scope */
         ErrorSkip ("No open .SCOPE");
     } else {
@@ -791,10 +843,10 @@ static void DoEndScope (void)
 static void DoError (void)
 /* User error */
 {
-    if (Tok != TOK_STRCON) {
+    if (CurTok.Tok != TOK_STRCON) {
        ErrorSkip ("String constant expected");
     } else {
-               Error ("User error: %m%p", &SVal);
+               Error ("User error: %m%p", &CurTok.SVal);
        SkipUntilSep ();
     }
 }
@@ -835,7 +887,7 @@ static void DoFarAddr (void)
 {
     while (1) {
                EmitFarAddr (Expression ());
-       if (Tok != TOK_COMMA) {
+       if (CurTok.Tok != TOK_COMMA) {
            break;
        } else {
            NextTok ();
@@ -845,6 +897,19 @@ static void DoFarAddr (void)
 
 
 
+static void DoFatal (void)
+/* Fatal user error */
+{
+    if (CurTok.Tok != TOK_STRCON) {
+       ErrorSkip ("String constant expected");
+    } else {
+               Fatal ("User error: %m%p", &CurTok.SVal);
+       SkipUntilSep ();
+    }
+}
+
+
+
 static void DoFeature (void)
 /* Switch the Feature option */
 {
@@ -852,7 +917,7 @@ static void DoFeature (void)
     while (1) {
 
        /* We expect an identifier */
-       if (Tok != TOK_IDENT) {
+       if (CurTok.Tok != TOK_IDENT) {
            ErrorSkip ("Identifier expected");
            return;
        }
@@ -861,9 +926,9 @@ static void DoFeature (void)
        LocaseSVal ();
 
        /* Set the feature and check for errors */
-       if (SetFeature (&SVal) == FEAT_UNKNOWN) {
+       if (SetFeature (&CurTok.SVal) == FEAT_UNKNOWN) {
            /* Not found */
-           ErrorSkip ("Invalid feature: `%m%p'", &SVal);
+           ErrorSkip ("Invalid feature: `%m%p'", &CurTok.SVal);
            return;
        } else {
            /* Skip the keyword */
@@ -871,7 +936,7 @@ static void DoFeature (void)
        }
 
        /* Allow more than one keyword */
-       if (Tok == TOK_COMMA) {
+       if (CurTok.Tok == TOK_COMMA) {
            NextTok ();
        } else {
            break;
@@ -887,7 +952,7 @@ static void DoFileOpt (void)
     long OptNum;
 
     /* The option type may be given as a keyword or as a number. */
-    if (Tok == TOK_IDENT) {
+    if (CurTok.Tok == TOK_IDENT) {
 
        /* Option given as keyword */
        static const char* Keys [] = {
@@ -909,7 +974,7 @@ static void DoFileOpt (void)
        ConsumeComma ();
 
        /* We accept only string options for now */
-       if (Tok != TOK_STRCON) {
+       if (CurTok.Tok != TOK_STRCON) {
            ErrorSkip ("String constant expected");
            return;
        }
@@ -919,17 +984,17 @@ static void DoFileOpt (void)
 
            case 0:
                /* Author */
-               OptAuthor (&SVal);
+               OptAuthor (&CurTok.SVal);
                break;
 
            case 1:
                /* Comment */
-               OptComment (&SVal);
+               OptComment (&CurTok.SVal);
                break;
 
            case 2:
                /* Compiler */
-               OptCompiler (&SVal);
+               OptCompiler (&CurTok.SVal);
                break;
 
            default:
@@ -953,13 +1018,13 @@ static void DoFileOpt (void)
        ConsumeComma ();
 
        /* We accept only string options for now */
-       if (Tok != TOK_STRCON) {
+       if (CurTok.Tok != TOK_STRCON) {
            ErrorSkip ("String constant expected");
            return;
        }
 
        /* Insert the option */
-       OptStr ((unsigned char) OptNum, &SVal);
+       OptStr ((unsigned char) OptNum, &CurTok.SVal);
 
        /* Done */
        NextTok ();
@@ -991,6 +1056,20 @@ static void DoGlobalZP (void)
 }
 
 
+static void DoHiBytes (void)
+/* Define bytes, extracting the hi byte from each expression in the list */
+{
+    while (1) {
+        EmitByte (FuncHiByte ());
+        if (CurTok.Tok != TOK_COMMA) {
+            break;
+        } else {
+            NextTok ();
+        }
+    }
+}
+
+
 
 static void DoI16 (void)
 /* Switch the index registers to 16 bit mode (assembler only) */
@@ -1038,27 +1117,28 @@ static void DoIncBin (void)
 /* Include a binary file */
 {
     StrBuf Name = STATIC_STRBUF_INITIALIZER;
+    struct stat StatBuf;
     long Start = 0L;
     long Count = -1L;
     long Size;
     FILE* F;
 
     /* Name must follow */
-    if (Tok != TOK_STRCON) {
+    if (CurTok.Tok != TOK_STRCON) {
        ErrorSkip ("String constant expected");
        return;
     }
-    SB_Copy (&Name, &SVal);
+    SB_Copy (&Name, &CurTok.SVal);
     SB_Terminate (&Name);
     NextTok ();
 
     /* A starting offset may follow */
-    if (Tok == TOK_COMMA) {
+    if (CurTok.Tok == TOK_COMMA) {
        NextTok ();
        Start = ConstExpression ();
 
        /* And a length may follow */
-       if (Tok == TOK_COMMA) {
+       if (CurTok.Tok == TOK_COMMA) {
            NextTok ();
            Count = ConstExpression ();
        }
@@ -1069,40 +1149,56 @@ static void DoIncBin (void)
     F = fopen (SB_GetConstBuf (&Name), "rb");
     if (F == 0) {
 
-               /* Search for the file in the include directories. */
-       char* PathName = FindInclude (SB_GetConstBuf (&Name));
-               if (PathName == 0 || (F = fopen (PathName, "r")) == 0) {
+               /* Search for the file in the binary include directory */
+       char* PathName = SearchFile (BinSearchPath, SB_GetConstBuf (&Name));
+               if (PathName == 0 || (F = fopen (PathName, "rb")) == 0) {
            /* Not found or cannot open, print an error and bail out */
                    ErrorSkip ("Cannot open include file `%m%p': %s", &Name, strerror (errno));
+            xfree (PathName);
+            goto ExitPoint;
        }
 
+        /* Remember the new file name */
+        SB_CopyStr (&Name, PathName);
+
        /* Free the allocated memory */
        xfree (PathName);
-
-        /* If we had an error before, bail out now */
-        if (F == 0) {
-            goto ExitPoint;
-        }
     }
 
     /* Get the size of the file */
     fseek (F, 0, SEEK_END);
     Size = ftell (F);
 
+    /* Stat the file and remember the values. There a race condition here,
+     * since we cannot use fileno() (non standard identifier in standard
+     * header file), and therefore not fstat. When using stat with the
+     * file name, there's a risk that the file was deleted and recreated
+     * while it was open. Since mtime and size are only used to check
+     * if a file has changed in the debugger, we will ignore this problem
+     * here.
+     */
+    SB_Terminate (&Name);
+    if (stat (SB_GetConstBuf (&Name), &StatBuf) != 0) {
+        Fatal ("Cannot stat input file `%m%p': %s", &Name, strerror (errno));
+    }
+
+    /* Add the file to the input file table */
+    AddFile (&Name, FT_BINARY, Size, StatBuf.st_mtime);
+
     /* If a count was not given, calculate it now */
     if (Count < 0) {
-       Count = Size - Start;
-       if (Count < 0) {
-           /* Nothing to read - flag this as a range error */
-           ErrorSkip ("Range error");
-           goto Done;
-       }
+       Count = Size - Start;
+       if (Count < 0) {
+           /* Nothing to read - flag this as a range error */
+           ErrorSkip ("Range error");
+           goto Done;
+       }
     } else {
-       /* Count was given, check if it is valid */
-       if (Start + Count > Size) {
-           ErrorSkip ("Range error");
-           goto Done;
-       }
+       /* Count was given, check if it is valid */
+       if (Start + Count > Size) {
+           ErrorSkip ("Range error");
+           goto Done;
+       }
     }
 
     /* Seek to the start position */
@@ -1111,12 +1207,12 @@ static void DoIncBin (void)
     /* Read chunks and insert them into the output */
     while (Count > 0) {
 
-       unsigned char Buf [1024];
+       unsigned char Buf [1024];
 
-       /* Calculate the number of bytes to read */
+       /* Calculate the number of bytes to read */
                size_t BytesToRead = (Count > (long)sizeof(Buf))? sizeof(Buf) : (size_t) Count;
 
-       /* Read chunk */
+       /* Read chunk */
        size_t BytesRead = fread (Buf, 1, BytesToRead, F);
        if (BytesToRead != BytesRead) {
            /* Some sort of error */
@@ -1147,11 +1243,11 @@ static void DoInclude (void)
 /* Include another file */
 {
     /* Name must follow */
-    if (Tok != TOK_STRCON) {
+    if (CurTok.Tok != TOK_STRCON) {
        ErrorSkip ("String constant expected");
     } else {
-        SB_Terminate (&SVal);
-       if (NewInputFile (SB_GetConstBuf (&SVal)) == 0) {
+        SB_Terminate (&CurTok.SVal);
+       if (NewInputFile (SB_GetConstBuf (&CurTok.SVal)) == 0) {
             /* Error opening the file, skip remainder of line */
             SkipUntilSep ();
         }
@@ -1166,11 +1262,11 @@ static void DoInterruptor (void)
     StrBuf Name = STATIC_STRBUF_INITIALIZER;
 
     /* Symbol name follows */
-    if (Tok != TOK_IDENT) {
+    if (CurTok.Tok != TOK_IDENT) {
        ErrorSkip ("Identifier expected");
        return;
     }
-    SB_Copy (&Name, &SVal);
+    SB_Copy (&Name, &CurTok.SVal);
     NextTok ();
 
     /* Parse the remainder of the line and export the symbol */
@@ -1221,6 +1317,20 @@ static void DoList (void)
 
 
 
+static void DoLoBytes (void)
+/* Define bytes, extracting the lo byte from each expression in the list */
+{
+    while (1) {
+        EmitByte (FuncLoByte ());
+        if (CurTok.Tok != TOK_COMMA) {
+            break;
+        } else {
+            NextTok ();
+        }
+    }
+}
+
+
 static void DoListBytes (void)
 /* Set maximum number of bytes to list for one line */
 {
@@ -1232,13 +1342,13 @@ static void DoListBytes (void)
 static void DoLocalChar (void)
 /* Define the character that starts local labels */
 {
-    if (Tok != TOK_CHARCON) {
+    if (CurTok.Tok != TOK_CHARCON) {
        ErrorSkip ("Character constant expected");
     } else {
-       if (IVal != '@' && IVal != '?') {
+       if (CurTok.IVal != '@' && CurTok.IVal != '?') {
            Error ("Invalid start character for locals");
        } else {
-           LocalStart = (char) IVal;
+           LocalStart = (char) CurTok.IVal;
                }
        NextTok ();
     }
@@ -1252,14 +1362,14 @@ static void DoMacPack (void)
     int Package;
 
     /* We expect an identifier */
-    if (Tok != TOK_IDENT) {
+    if (CurTok.Tok != TOK_IDENT) {
        ErrorSkip ("Identifier expected");
        return;
     }
 
     /* Search for the macro package name */
     LocaseSVal ();
-    Package = MacPackFind (&SVal);
+    Package = MacPackFind (&CurTok.SVal);
     if (Package < 0) {
        /* Not found */
        ErrorSkip ("Invalid macro package");
@@ -1308,13 +1418,15 @@ static void DoOrg (void)
 static void DoOut (void)
 /* Output a string */
 {
-    if (Tok != TOK_STRCON) {
+    if (CurTok.Tok != TOK_STRCON) {
        ErrorSkip ("String constant expected");
     } else {
        /* Output the string and be sure to flush the output to keep it in
         * sync with any error messages if the output is redirected to a file.
         */
-       printf ("%*s\n", SB_GetLen (&SVal), SB_GetConstBuf (&SVal));
+       printf ("%.*s\n",
+                (int) SB_GetLen (&CurTok.SVal),
+                SB_GetConstBuf (&CurTok.SVal));
        fflush (stdout);
        NextTok ();
     }
@@ -1354,6 +1466,21 @@ static void DoPageLength (void)
 
 
 
+static void DoPopCPU (void)
+/* Pop an old CPU setting from the CPU stack */
+{
+    /* Must have a CPU on the stack */
+    if (IS_IsEmpty (&CPUStack)) {
+        ErrorSkip ("CPU stack is empty");
+        return;
+    }
+
+    /* Set the CPU to the value popped from stack */
+    SetCPU (IS_Pop (&CPUStack));
+}
+
+
+
 static void DoPopSeg (void)
 /* Pop an old segment from the segment stack */
 {
@@ -1382,13 +1509,13 @@ static void DoProc (void)
 {
     StrBuf Name = STATIC_STRBUF_INITIALIZER;
     unsigned char AddrSize;
+    SymEntry* Sym = 0;
 
-    if (Tok == TOK_IDENT) {
 
-        SymEntry* Sym;
+    if (CurTok.Tok == TOK_IDENT) {
 
-       /* The new scope has a name. Remember it. */
-        SB_Copy (&Name, &SVal);
+       /* The new scope has a name. Remember it. */
+        SB_Copy (&Name, &CurTok.SVal);
 
         /* Search for the symbol, generate a new one if needed */
                Sym = SymFind (CurrentScope, &Name, SYM_ALLOC_NEW);
@@ -1412,7 +1539,7 @@ static void DoProc (void)
     }
 
     /* Enter a new scope */
-    SymEnterLevel (&Name, ST_PROC, AddrSize);
+    SymEnterLevel (&Name, SCOPE_SCOPE, AddrSize, Sym);
 
     /* Free memory for Name */
     SB_Done (&Name);
@@ -1428,6 +1555,21 @@ static void DoPSC02 (void)
 
 
 
+static void DoPushCPU (void)
+/* Push the current CPU setting onto the CPU stack */
+{
+    /* Can only push a limited size of segments */
+    if (IS_IsFull (&CPUStack)) {
+        ErrorSkip ("CPU stack overflow");
+        return;
+    }
+
+    /* Get the current segment and push it */
+    IS_Push (&CPUStack, GetCPU ());
+}
+
+
+
 static void DoPushSeg (void)
 /* Push the current segment onto the segment stack */
 {
@@ -1470,7 +1612,7 @@ static void DoRes (void)
        ErrorSkip ("Range error");
        return;
     }
-    if (Tok == TOK_COMMA) {
+    if (CurTok.Tok == TOK_COMMA) {
        NextTok ();
        Val = ConstExpression ();
        /* We need a byte value here */
@@ -1507,10 +1649,10 @@ static void DoScope (void)
     unsigned char AddrSize;
 
 
-    if (Tok == TOK_IDENT) {
+    if (CurTok.Tok == TOK_IDENT) {
 
        /* The new scope has a name. Remember and skip it. */
-        SB_Copy (&Name, &SVal);
+        SB_Copy (&Name, &CurTok.SVal);
         NextTok ();
 
     } else {
@@ -1524,7 +1666,7 @@ static void DoScope (void)
     AddrSize = OptionalAddrSize ();
 
     /* Enter the new scope */
-    SymEnterLevel (&Name, ST_SCOPE, AddrSize);
+    SymEnterLevel (&Name, SCOPE_SCOPE, AddrSize, 0);
 
     /* Free memory for Name */
     SB_Done (&Name);
@@ -1538,12 +1680,12 @@ static void DoSegment (void)
     StrBuf Name = STATIC_STRBUF_INITIALIZER;
     SegDef Def;
 
-    if (Tok != TOK_STRCON) {
+    if (CurTok.Tok != TOK_STRCON) {
        ErrorSkip ("String constant expected");
     } else {
 
        /* Save the name of the segment and skip it */
-       SB_Copy (&Name, &SVal);
+       SB_Copy (&Name, &CurTok.SVal);
        NextTok ();
 
         /* Use the name for the segment definition */
@@ -1567,14 +1709,14 @@ static void DoSetCPU (void)
 /* Switch the CPU instruction set */
 {
     /* We expect an identifier */
-    if (Tok != TOK_STRCON) {
+    if (CurTok.Tok != TOK_STRCON) {
        ErrorSkip ("String constant expected");
     } else {
         cpu_t CPU;
 
         /* Try to find the CPU */
-        SB_Terminate (&SVal);
-        CPU = FindCPU (SB_GetConstBuf (&SVal));
+        SB_Terminate (&CurTok.SVal);
+        CPU = FindCPU (SB_GetConstBuf (&CurTok.SVal));
 
         /* Switch to the new CPU */
         SetCPU (CPU);
@@ -1618,7 +1760,7 @@ static void DoTag (void)
         ErrorSkip ("Unknown struct");
         return;
     }
-    if (GetSymTabType (Struct) != ST_STRUCT) {
+    if (GetSymTabType (Struct) != SCOPE_STRUCT) {
         ErrorSkip ("Not a struct");
         return;
     }
@@ -1633,7 +1775,7 @@ static void DoTag (void)
     }
 
     /* Optional multiplicator may follow */
-    if (Tok == TOK_COMMA) {
+    if (CurTok.Tok == TOK_COMMA) {
         long Multiplicator;
         NextTok ();
         Multiplicator = ConstExpression ();
@@ -1651,6 +1793,29 @@ static void DoTag (void)
 
 
 
+static void DoUnDef (void)
+/* Undefine a define style macro */
+{
+    /* The function is called with the .UNDEF token in place, because we need
+     * to disable .define macro expansions before reading the next token.
+     * Otherwise the name of the macro would be expanded, so we would never
+     * see it.
+     */
+    DisableDefineStyleMacros ();
+    NextTok ();
+    EnableDefineStyleMacros ();
+
+    /* We expect an identifier */
+    if (CurTok.Tok != TOK_IDENT) {
+       ErrorSkip ("Identifier expected");
+    } else {
+        MacUndef (&CurTok.SVal, MAC_STYLE_DEFINE);
+        NextTok ();
+    }
+}
+
+
+
 static void DoUnexpected (void)
 /* Got an unexpected keyword */
 {
@@ -1663,10 +1828,10 @@ static void DoUnexpected (void)
 static void DoWarning (void)
 /* User warning */
 {
-    if (Tok != TOK_STRCON) {
+    if (CurTok.Tok != TOK_STRCON) {
        ErrorSkip ("String constant expected");
     } else {
-               Warning (0, "User warning: %m%p", &SVal);
+               Warning (0, "User warning: %m%p", &CurTok.SVal);
        SkipUntilSep ();
     }
 }
@@ -1678,7 +1843,7 @@ static void DoWord (void)
 {
     while (1) {
                EmitWord (Expression ());
-       if (Tok != TOK_COMMA) {
+       if (CurTok.Tok != TOK_COMMA) {
            break;
        } else {
            NextTok ();
@@ -1697,7 +1862,7 @@ static void DoZeropage (void)
 
 
 /*****************************************************************************/
-/*                               Table data                                 */
+/*                               Table data                                 */
 /*****************************************************************************/
 
 
@@ -1725,6 +1890,7 @@ static CtrlDesc CtrlCmdTab [] = {
     { ccNone,           DoAssert        },
     { ccNone,          DoAutoImport    },
     { ccNone,          DoUnexpected    },      /* .BANKBYTE */
+    { ccNone,           DoBankBytes     },
     { ccNone,          DoUnexpected    },      /* .BLANK */
     { ccNone,          DoBss           },
     { ccNone,          DoByte          },
@@ -1742,6 +1908,7 @@ static CtrlDesc CtrlCmdTab [] = {
     { ccNone,          DoDebugInfo     },
     { ccNone,          DoDefine        },
     { ccNone,          DoUnexpected    },      /* .DEFINED */
+    { ccNone,           DoDelMac        },
     { ccNone,          DoDestructor    },
     { ccNone,          DoDWord         },
     { ccKeepToken,     DoConditionals  },      /* .ELSE */
@@ -1761,6 +1928,7 @@ static CtrlDesc CtrlCmdTab [] = {
     { ccNone,          DoExport        },
     { ccNone,          DoExportZP      },
     { ccNone,          DoFarAddr       },
+    { ccNone,                  DoFatal         },
     { ccNone,          DoFeature       },
     { ccNone,          DoFileOpt       },
     { ccNone,           DoForceImport   },
@@ -1768,6 +1936,7 @@ static CtrlDesc CtrlCmdTab [] = {
     { ccNone,          DoGlobal        },
     { ccNone,          DoGlobalZP      },
     { ccNone,          DoUnexpected    },      /* .HIBYTE */
+    { ccNone,           DoHiBytes       },
     { ccNone,          DoUnexpected    },      /* .HIWORD */
     { ccNone,          DoI16           },
     { ccNone,          DoI8            },
@@ -1784,7 +1953,7 @@ static CtrlDesc CtrlCmdTab [] = {
     { ccKeepToken,     DoConditionals  },      /* .IFP816 */
     { ccKeepToken,     DoConditionals  },      /* .IFPC02 */
     { ccKeepToken,     DoConditionals  },      /* .IFPSC02 */
-    { ccKeepToken,     DoConditionals  },      /* .IFREF */
+    { ccKeepToken,     DoConditionals  },      /* .IFREF */
     { ccNone,          DoImport        },
     { ccNone,          DoImportZP      },
     { ccNone,          DoIncBin        },
@@ -1795,24 +1964,29 @@ static CtrlDesc CtrlCmdTab [] = {
     { ccNone,          DoList          },
     { ccNone,                  DoListBytes     },
     { ccNone,          DoUnexpected    },      /* .LOBYTE */
+    { ccNone,           DoLoBytes       },
     { ccNone,          DoUnexpected    },      /* .LOCAL */
     { ccNone,          DoLocalChar     },
     { ccNone,          DoUnexpected    },      /* .LOWORD */
     { ccNone,          DoMacPack       },
     { ccNone,          DoMacro         },
     { ccNone,                  DoUnexpected    },      /* .MATCH */
+    { ccNone,                  DoUnexpected    },      /* .MAX */
     { ccNone,                  DoInvalid       },      /* .MID */
+    { ccNone,                  DoUnexpected    },      /* .MIN */
     { ccNone,          DoNull          },
-    { ccNone,          DoOrg           },
+    { ccNone,          DoOrg           },
     { ccNone,          DoOut           },
     { ccNone,          DoP02           },
     { ccNone,          DoP816          },
     { ccNone,                  DoPageLength    },
     { ccNone,          DoUnexpected    },      /* .PARAMCOUNT */
     { ccNone,          DoPC02          },
+    { ccNone,           DoPopCPU        },
     { ccNone,           DoPopSeg        },
     { ccNone,          DoProc          },
     { ccNone,                  DoPSC02         },
+    { ccNone,           DoPushCPU       },
     { ccNone,           DoPushSeg       },
     { ccNone,          DoUnexpected    },      /* .REFERENCED */
     { ccNone,          DoReloc         },
@@ -1835,10 +2009,11 @@ static CtrlDesc CtrlCmdTab [] = {
     { ccNone,           DoTag           },
     { ccNone,          DoUnexpected    },      /* .TCOUNT */
     { ccNone,                  DoUnexpected    },      /* .TIME */
+    { ccKeepToken,      DoUnDef         },
     { ccNone,           DoUnion         },
     { ccNone,           DoUnexpected    },      /* .VERSION */
     { ccNone,          DoWarning       },
-    { ccNone,          DoWord          },
+    { ccNone,          DoWord          },
     { ccNone,                  DoUnexpected    },      /* .XMATCH */
     { ccNone,          DoZeropage      },
 };
@@ -1846,7 +2021,7 @@ static CtrlDesc CtrlCmdTab [] = {
 
 
 /*****************************************************************************/
-/*                                          Code                                    */
+/*                                          Code                                    */
 /*****************************************************************************/
 
 
@@ -1857,7 +2032,7 @@ void HandlePseudo (void)
     CtrlDesc* D;
 
     /* Calculate the index into the table */
-    unsigned Index = Tok - TOK_FIRSTPSEUDO;
+    unsigned Index = CurTok.Tok - TOK_FIRSTPSEUDO;
 
     /* Safety check */
     if (PSEUDO_COUNT != (TOK_LASTPSEUDO - TOK_FIRSTPSEUDO + 1)) {
@@ -1871,7 +2046,7 @@ void HandlePseudo (void)
 
     /* Remember the instruction, then skip it if needed */
     if ((D->Flags & ccKeepToken) == 0) {
-       SB_Copy (&Keyword, &SVal);
+       SB_Copy (&Keyword, &CurTok.SVal);
        NextTok ();
     }
 
@@ -1881,11 +2056,14 @@ void HandlePseudo (void)
 
 
 
-void SegStackCheck (void)
-/* Check if the segment stack is empty at end of assembly */
+void CheckPseudo (void)
+/* Check if the stacks are empty at end of assembly */
 {
     if (CollCount (&SegStack) != 0) {
-        Error ("Segment stack is not empty");
+        Warning (1, "Segment stack is not empty");
+    }
+    if (!IS_IsEmpty (&CPUStack)) {
+        Warning (1, "CPU stack is not empty");
     }
 }