ghc/interpreter/input.c

   1
   2 /* --------------------------------------------------------------------------
   3  * Input functions, lexical analysis parsing etc...
   4  *
   5  * The Hugs 98 system is Copyright (c) Mark P Jones, Alastair Reid, the
   6  * Yale Haskell Group, and the Oregon Graduate Institute of Science and
   7  * Technology, 1994-1999, All rights reserved.  It is distributed as
   8  * free software under the license in the file "License", which is
   9  * included in the distribution.
  10  *
  11  * $RCSfile: input.c,v $
  12  * $Revision: 1.14 $
  13  * $Date: 1999/11/29 18:59:27 $
  14  * ------------------------------------------------------------------------*/
  15
  16 #include "prelude.h"
  17 #include "storage.h"
  18 #include "backend.h"
  19 #include "connect.h"
  20 #include "command.h"
  21 #include "errors.h"
  22 #include "link.h"
  23 #include <ctype.h>
  24 #if HAVE_GETDELIM_H
  25 #include "getdelim.h"
  26 #endif
  27
  28 #if IS_WIN32
  29 #include <windows.h>
  30 #endif
  31
  32 #if IS_WIN32 || HUGS_FOR_WINDOWS
  33 #undef IN
  34 #endif
  35
  36 #if HAVE_LIBREADLINE && HAVE_READLINE_READLINE_H
  37 #define USE_READLINE 1
  38 #else
  39 #define USE_READLINE 0
  40 #endif
  41
  42 #if USE_READLINE
  43 #include <readline/readline.h>
  44 #include <readline/history.h>
  45 #endif
  46
  47
  48 /* --------------------------------------------------------------------------
  49  * Global data:
  50  * ------------------------------------------------------------------------*/
  51
  52 List tyconDefns       = NIL;            /* type constructor definitions    */
  53 List typeInDefns      = NIL;            /* type synonym restrictions       */
  54 List valDefns         = NIL;            /* value definitions in script     */
  55 List classDefns       = NIL;            /* class defns in script           */
  56 List instDefns        = NIL;            /* instance defns in script        */
  57 List selDefns         = NIL;            /* list of selector lists          */
  58 List genDefns         = NIL;            /* list of generated names         */
  59 List unqualImports    = NIL;            /* unqualified import list         */
  60 List foreignImports   = NIL;            /* foreign imports                 */
  61 List foreignExports   = NIL;            /* foreign exportsd                */
  62 List defaultDefns     = NIL;            /* default definitions (if any)    */
  63 Int  defaultLine      = 0;              /* line in which default defs occur*/
  64 List evalDefaults     = NIL;            /* defaults for evaluator          */
  65
  66 Cell inputExpr        = NIL;            /* input expression                */
  67 Cell inputContext     = NIL;            /* input context                   */
  68 Bool literateScripts  = FALSE;          /* TRUE => default to lit scripts  */
  69 Bool literateErrors   = TRUE;           /* TRUE => report errs in lit scrs */
  70 Bool offsideON        = TRUE;           /* TRUE => implement offside rule  */
  71 Bool readingInterface = FALSE;
  72
  73 String repeatStr     = 0;               /* Repeat last expr                */
  74
  75 #if USE_PREPROCESSOR && (defined(HAVE_POPEN) || defined(HAVE__POPEN))
  76 String preprocessor  = 0;
  77 #endif
  78
  79 /* --------------------------------------------------------------------------
  80  * Local function prototypes:
  81  * ------------------------------------------------------------------------*/
  82
  83 static Void local initCharTab     Args((Void));
  84 static Void local fileInput       Args((String,Long));
  85 static Bool local literateMode    Args((String));
  86 static Bool local linecmp         Args((String,String));
  87 static Int  local nextLine        Args((Void));
  88 static Void local skip            Args((Void));
  89 static Void local thisLineIs      Args((Int));
  90 static Void local newlineSkip     Args((Void));
  91 static Void local closeAnyInput   Args((Void));
  92
  93        Int  yyparse         Args((Void)); /* can't stop yacc making this   */
  94                                           /* public, but don't advertise   */
  95                                           /* it in a header file.          */
  96
  97 static Void local endToken        Args((Void));
  98 static Text local readOperator    Args((Void));
  99 static Text local readIdent       Args((Void));
 100 static Cell local readRadixNumber Args((Int));
 101 static Cell local readNumber      Args((Void));
 102 static Cell local readChar        Args((Void));
 103 static Cell local readString      Args((Void));
 104 static Void local saveStrChr      Args((Char));
 105 static Cell local readAChar       Args((Bool));
 106
 107 static Bool local lazyReadMatches Args((String));
 108 static Cell local readEscapeChar  Args((Bool));
 109 static Void local skipGap         Args((Void));
 110 static Cell local readCtrlChar    Args((Void));
 111 static Cell local readOctChar     Args((Void));
 112 static Cell local readHexChar     Args((Void));
 113 static Int  local readHexDigit    Args((Char));
 114 static Cell local readDecChar     Args((Void));
 115
 116 static Void local goOffside       Args((Int));
 117 static Void local unOffside       Args((Void));
 118 static Bool local canUnOffside    Args((Void));
 119
 120 static Void local skipWhitespace  Args((Void));
 121 static Int  local yylex           Args((Void));
 122 static Int  local repeatLast      Args((Void));
 123
 124 static Void local parseInput      Args((Int));
 125
 126 static Bool local doesNotExceed   Args((String,Int,Int));
 127 static Int  local stringToInt     Args((String,Int));
 128
 129
 130 /* --------------------------------------------------------------------------
 131  * Text values for reserved words and special symbols:
 132  * ------------------------------------------------------------------------*/
 133
 134 static Text textCase,    textOfK,      textData,   textType,   textIf;
 135 static Text textThen,    textElse,     textWhere,  textLet,    textIn;
 136 static Text textInfix,   textInfixl,   textInfixr, textForeign, textNewtype;
 137 static Text textDefault, textDeriving, textDo,     textClass,  textInstance;
 138 #if IPARAM
 139 static Text textWith,  textDlet;
 140 #endif
 141
 142 static Text textCoco,    textEq,       textUpto,   textAs,     textLambda;
 143 static Text textBar,     textMinus,    textFrom,   textArrow,  textLazy;
 144 static Text textBang,    textDot,      textAll,    textImplies;
 145 static Text textWildcard;
 146
 147 static Text textModule,  textImport,    textInterface,  textInstImport;
 148 static Text textHiding,  textQualified, textAsMod;
 149 static Text textExport,  textDynamic,   textUUExport;
 150 static Text textUnsafe,  textUUAll;
 151
 152 Text   textCcall;                       /* ccall                           */
 153 Text   textStdcall;                     /* stdcall                         */
 154
 155 Text   textNum;                         /* Num                             */
 156 Text   textPrelude;                     /* Prelude                         */
 157 Text   textPlus;                        /* (+)                             */
 158
 159 static Cell conMain;                    /* Main                            */
 160 static Cell varMain;                    /* main                            */
 161
 162 static Cell varMinus;                   /* (-)                             */
 163 static Cell varPlus;                    /* (+)                             */
 164 static Cell varBang;                    /* (!)                             */
 165 static Cell varDot;                     /* (.)                             */
 166 static Cell varHiding;                  /* hiding                          */
 167 static Cell varQualified;               /* qualified                       */
 168 static Cell varAsMod;                   /* as                              */
 169
 170 static List imps;                       /* List of imports to be chased    */
 171
 172
 173 /* --------------------------------------------------------------------------
 174  * Character set handling:
 175  *
 176  * Hugs follows Haskell 1.3 in assuming that input uses the ISO-8859-1
 177  * character set.  The following code provides methods for classifying
 178  * input characters according to the lexical structure specified by the
 179  * report.  Hugs should still accept older programs because ASCII is
 180  * essentially just a subset of the ISO character set.
 181  *
 182  * Notes: If you want to port Hugs to a machine that uses something
 183  * substantially different from the ISO character set, then you will need
 184  * to insert additional code to map between character sets.
 185  *
 186  * At some point, the following data structures may be exported in a .h
 187  * file to allow the information contained here to be picked up in the
 188  * implementation of LibChar is* primitives.
 189  *
 190  * Relies, implicitly but for this comment, on assumption that NUM_CHARS=256.
 191  * ------------------------------------------------------------------------*/
 192
 193 static  Bool            charTabBuilt;
 194 static  unsigned char   ctable[NUM_CHARS];
 195 #define isIn(c,x)       (ctable[(unsigned char)(c)]&(x))
 196 #define isISO(c)        (0<=(c) && (c)<NUM_CHARS)
 197
 198 #define DIGIT           0x01
 199 #define SMALL           0x02
 200 #define LARGE           0x04
 201 #define SYMBOL          0x08
 202 #define IDAFTER         0x10
 203 #define ZPACE           0x20
 204 #define PRINT           0x40
 205
 206 static Void local initCharTab() {       /* Initialize char decode table    */
 207 #define setRange(x,f,t) {Int i=f;   while (i<=t) ctable[i++] |=x;}
 208 #define setChar(x,c)    ctable[c] |= (x)
 209 #define setChars(x,s)   {char *p=s; while (*p)   ctable[(Int)*p++]|=x;}
 210 #define setCopy(x,c)    {Int i;                         \
 211                          for (i=0; i<NUM_CHARS; ++i)    \
 212                              if (isIn(i,c))             \
 213                                  ctable[i]|=x;          \
 214                         }
 215
 216     setRange(DIGIT,     '0','9');       /* ASCII decimal digits            */
 217
 218     setRange(SMALL,     'a','z');       /* ASCII lower case letters        */
 219     setRange(SMALL,     223,246);       /* ISO lower case letters          */
 220     setRange(SMALL,     248,255);       /* (omits division symbol, 247)    */
 221     setChar (SMALL,     '_');
 222
 223     setRange(LARGE,     'A','Z');       /* ASCII upper case letters        */
 224     setRange(LARGE,     192,214);       /* ISO upper case letters          */
 225     setRange(LARGE,     216,222);       /* (omits multiplication, 215)     */
 226
 227     setRange(SYMBOL,    161,191);       /* Symbol characters + ':'         */
 228     setRange(SYMBOL,    215,215);
 229     setChar (SYMBOL,    247);
 230     setChars(SYMBOL,    ":!#$%&*+./<=>?@\\^|-~");
 231
 232     setChar (IDAFTER,   '\'');          /* Characters in identifier        */
 233     setCopy (IDAFTER,   (DIGIT|SMALL|LARGE));
 234
 235     setChar (ZPACE,     ' ');           /* ASCII space character           */
 236     setChar (ZPACE,     160);           /* ISO non breaking space          */
 237     setRange(ZPACE,     9,13);          /* special whitespace: \t\n\v\f\r  */
 238
 239     setChars(PRINT,     "(),;[]_`{}");  /* Special characters              */
 240     setChars(PRINT,     " '\"");        /* Space and quotes                */
 241     setCopy (PRINT,     (DIGIT|SMALL|LARGE|SYMBOL));
 242
 243     charTabBuilt = TRUE;
 244 #undef setRange
 245 #undef setChar
 246 #undef setChars
 247 #undef setCopy
 248 }
 249
 250
 251 /* --------------------------------------------------------------------------
 252  * Single character input routines:
 253  *
 254  * At the lowest level of input, characters are read one at a time, with the
 255  * current character held in c0 and the following (lookahead) character in
 256  * c1.  The corrdinates of c0 within the file are held in (column,row).
 257  * The input stream is advanced by one character using the skip() function.
 258  * ------------------------------------------------------------------------*/
 259
 260 #define TABSIZE    8                   /* spacing between tabstops         */
 261
 262 #define NOTHING    0                   /* what kind of input is being read?*/
 263 #define KEYBOARD   1                   /* - keyboard/console?              */
 264 #define SCRIPTFILE 2                   /* - script file                    */
 265 #define PROJFILE   3                   /* - project file                   */
 266 #define STRING     4                   /* - string buffer?                 */
 267
 268 static Int    reading   = NOTHING;
 269
 270 static Target readSoFar;
 271 static Int    row, column, startColumn;
 272 static int    c0, c1;
 273 static FILE   *inputStream = 0;
 274 static Bool   thisLiterate;
 275 static String nextStringChar;          /* next char in string buffer       */
 276
 277 #if     USE_READLINE                   /* for command line editors         */
 278 static  String currentLine;            /* editline or GNU readline         */
 279 static  String nextChar;
 280 #define nextConsoleChar() \
 281            (unsigned char)(*nextChar=='\0' ? '\n' : *nextChar++)
 282 #else
 283 #define nextConsoleChar() getc(stdin)
 284 #endif
 285
 286 static  Int litLines;                  /* count defn lines in lit script   */
 287 #define DEFNCHAR  '>'                  /* definition lines begin with this */
 288 static  Int lastLine;                  /* records type of last line read:  */
 289 #define STARTLINE 0                    /* - at start of file, none read    */
 290 #define BLANKLINE 1                    /* - blank (may preceed definition) */
 291 #define TEXTLINE  2                    /* - text comment                   */
 292 #define DEFNLINE  3                    /* - line containing definition     */
 293 #define CODELINE  4                    /* - line inside code block         */
 294
 295 #define BEGINCODE "\\begin{code}"
 296 #define ENDCODE   "\\end{code}"
 297
 298 #if HAVE_GETDELIM_H
 299 static char *lineBuffer = NULL;   /* getline() does the initial allocation */
 300 #else
 301 #define LINEBUFFER_SIZE 1000
 302 static char lineBuffer[LINEBUFFER_SIZE];
 303 #endif
 304 static int lineLength = 0;
 305 static int inCodeBlock = FALSE; /* Inside \begin{code}..\end{code} */
 306 static int linePtr = 0;
 307
 308 Void consoleInput(prompt)              /* prepare to input characters from */
 309 String prompt; {                       /* standard in (i.e. console/kbd)   */
 310     reading     = KEYBOARD;            /* keyboard input is Line oriented, */
 311     c0          =                      /* i.e. input terminated by '\n'    */
 312     c1          = ' ';
 313     column      = (-1);
 314     row         = 0;
 315
 316 #if USE_READLINE
 317     /* Paranoid freeing code supplied by Sverker Nilsson (sverker@opq.se)
 318      * avoids accidentally freeing currentLine twice.
 319      */
 320     if (currentLine) {
 321         String oldCurrentLine = currentLine;
 322         currentLine = 0;           /* We may lose the space of currentLine */
 323         free(oldCurrentLine);      /* if interrupted here - unlikely       */
 324     }
 325     currentLine = readline(prompt);
 326     nextChar    = currentLine;
 327     if (currentLine) {
 328         if (*currentLine)
 329             add_history(currentLine);
 330     }
 331     else
 332         c0 = c1 = EOF;
 333 #else
 334     Printf("%s",prompt);
 335     FlushStdout();
 336 #endif
 337 }
 338
 339 Void projInput(nm)                     /* prepare to input characters from */
 340 String nm; {                           /* from named project file          */
 341     if ((inputStream = fopen(nm,"r"))!=0) {
 342         reading = PROJFILE;
 343         c0      = ' ';
 344         c1      = '\n';
 345         column  = 1;
 346         row     = 0;
 347     }
 348     else {
 349         ERRMSG(0) "Unable to open project file \"%s\"", nm
 350         EEND;
 351     }
 352 }
 353
 354 static Void local fileInput(nm,len)     /* prepare to input characters from*/
 355 String nm;                              /* named file (specified length is */
 356 Long   len; {                           /* used to set target for reading) */
 357 #if USE_PREPROCESSOR && (defined(HAVE_POPEN) || defined(HAVE__POPEN))
 358     if (preprocessor) {
 359         Int reallen = strlen(preprocessor) + 1 + strlen(nm) + 1;
 360         char *cmd = malloc(reallen);
 361         if (cmd == NULL) {
 362             ERRMSG(0) "Unable to allocate memory for filter command."
 363             EEND;
 364         }
 365         strcpy(cmd,preprocessor);
 366         strcat(cmd," ");
 367         strcat(cmd,nm);
 368         inputStream = popen(cmd,"r");
 369         free(cmd);
 370     } else {
 371         inputStream = fopen(nm,"r");
 372     }
 373 #else
 374     inputStream = fopen(nm,"r");
 375 #endif
 376     if (inputStream) {
 377         reading      = SCRIPTFILE;
 378         c0           = ' ';
 379         c1           = '\n';
 380         column       = 1;
 381         row          = 0;
 382
 383         lastLine     = STARTLINE;       /* literate file processing */
 384         litLines     = 0;
 385         linePtr      = 0;
 386         lineLength   = 0;
 387         thisLiterate = literateMode(nm);
 388         inCodeBlock  = FALSE;
 389
 390         readSoFar    = 0;
 391         setGoal("Parsing", (Target)len);
 392     }
 393     else {
 394         ERRMSG(0) "Unable to open file \"%s\"", nm
 395         EEND;
 396     }
 397 }
 398
 399 Void stringInput(s)             /* prepare to input characters from string */
 400 String s; {
 401     reading      = STRING;
 402     c0           = EOF;
 403     c1           = EOF;
 404     if (*s) c0 = *s++;
 405     if (*s) c1 = *s++;
 406     column       = 1;
 407     row          = 1;
 408
 409     nextStringChar = s;
 410     if (!charTabBuilt)
 411         initCharTab();
 412 }
 413
 414 static Bool local literateMode(nm)      /* Select literate mode for file   */
 415 String nm; {
 416     char *dot = strrchr(nm,'.');        /* look for last dot in file name  */
 417     if (dot) {
 418         if (filenamecmp(dot+1,"hs")==0) /* .hs files are never literate    */
 419             return FALSE;
 420         if (filenamecmp(dot+1,"lhs") ==0 || /* .lhs, .verb files are always*/
 421             filenamecmp(dot+1,"verb")==0) /* literate scripts              */
 422             return TRUE;
 423     }
 424     return literateScripts;             /* otherwise, use the default      */
 425 }
 426
 427
 428 Void hi_o_namesFromSrcName ( String srcName, String* hiName, String* oName )
 429 {
 430    Int len;
 431    String dot;
 432    len = 1 + strlen ( srcName );
 433    *hiName = malloc(len);
 434    *oName  = malloc(len);
 435    if (!(*hiName && *oName)) internal("hi_o_namesFromSource");
 436    (*hiName)[0] = (*oName)[0] = 0;
 437    dot = strrchr(srcName, '.');
 438    if (!dot) return;
 439    if (filenamecmp(dot+1, "hs")==0 &&
 440        filenamecmp(dot+1, "lhs")==0 &&
 441        filenamecmp(dot+1, "verb")==0) return;
 442
 443    strcpy(*hiName, srcName);
 444    dot = strrchr(*hiName, '.');
 445    dot[1] = 'h';
 446    dot[2] = 'i';
 447    dot[3] = 0;
 448
 449    strcpy(*oName, srcName);
 450    dot = strrchr(*oName, '.');
 451    dot[1] = 'o';
 452    dot[2] = 0;
 453 }
 454
 455
 456
 457 /* This code originally came from Sigbjorn Finne (sof@dcs.gla.ac.uk).
 458  * I've removed the loop (since newLineSkip contains a loop too) and
 459  * replaced the warnings with errors. ADR
 460  */
 461 /*
 462  * To deal with literate \begin{code}...\end{code} blocks,
 463  * add a line buffer that rooms the current line. The old c0 and c1
 464  * stream pointers are used as before within that buffer -- sof
 465  *
 466  * Upon reading a new line into the line buffer, we check to see if
 467  * we're reading in a line containing \begin{code} or \end{code} and
 468  * take appropriate action.
 469  */
 470
 471 static Bool local linecmp(s,line)       /* compare string with line        */
 472 String s;                               /* line may end in whitespace      */
 473 String line; {
 474     Int i=0;
 475     while (s[i] != '\0' && s[i] == line[i]) {
 476         ++i;
 477     }
 478     /* s[0..i-1] == line[0..i-1] */
 479     if (s[i] != '\0') {                 /* check s `isPrefixOf` line       */
 480         return FALSE;
 481     }
 482     while (isIn(line[i], ZPACE)) {      /* allow whitespace at end of line */
 483         ++i;
 484     }
 485     return (line[i] == '\0');
 486 }
 487
 488 /* Returns line length (including \n) or 0 upon EOF. */
 489 static Int local nextLine()
 490 {
 491 #if HAVE_GETDELIM_H
 492     /*
 493        Forget about fgets(), it is utterly braindead.
 494        (Assumes \NUL free streams and does not gracefully deal
 495        with overflow.) Instead, use GNU libc's getline().
 496        */
 497     lineLength = getline(&lineBuffer, &lineLength, inputStream);
 498 #else
 499     if (NULL != fgets(lineBuffer, LINEBUFFER_SIZE, inputStream))
 500         lineLength = strlen(lineBuffer);
 501     else
 502         lineLength = 0;
 503 #endif
 504     /* printf("Read: \"%s\"", lineBuffer); */
 505     if (lineLength <= 0) { /* EOF / IO error, who knows.. */
 506         return lineLength;
 507     }
 508     else if (lineLength >= 2 && lineBuffer[0] == '#' &&
 509              lineBuffer[1] == '!') {
 510         lineBuffer[0]='\n'; /* pretend it's a blank line */
 511         lineBuffer[1]='\0';
 512         lineLength=1;
 513     } else if (thisLiterate) {
 514         if (linecmp(BEGINCODE, lineBuffer)) {
 515             if (!inCodeBlock) {             /* Entered a code block        */
 516                 inCodeBlock = TRUE;
 517                 lineBuffer[0]='\n'; /* pretend it's a blank line */
 518                 lineBuffer[1]='\0';
 519                 lineLength=1;
 520             }
 521             else {
 522                 ERRMSG(row) "\\begin{code} encountered inside code block"
 523                 EEND;
 524             }
 525         }
 526         else if (linecmp(ENDCODE, lineBuffer)) {
 527             if (inCodeBlock) {              /* Finished code block         */
 528                 inCodeBlock = FALSE;
 529                 lineBuffer[0]='\n'; /* pretend it's a blank line */
 530                 lineBuffer[1]='\0';
 531                 lineLength=1;
 532             }
 533             else {
 534                 ERRMSG(row) "\\end{code} encountered outside code block"
 535                 EEND;
 536             }
 537         }
 538     }
 539     /* printf("Read: \"%s\"", lineBuffer); */
 540     return lineLength;
 541 }
 542
 543 static Void local skip() {              /* move forward one char in input  */
 544     if (c0!=EOF) {                      /* stream, updating c0, c1, ...    */
 545         if (c0=='\n') {                 /* Adjusting cursor coords as nec. */
 546             row++;
 547             column=1;
 548             if (reading==SCRIPTFILE)
 549                 soFar(readSoFar);
 550         }
 551         else if (c0=='\t')
 552             column += TABSIZE - ((column-1)%TABSIZE);
 553         else
 554             column++;
 555
 556         c0 = c1;
 557         readSoFar++;
 558
 559         if (c0==EOF) {
 560             column = 0;
 561             if (reading==SCRIPTFILE)
 562                 done();
 563             closeAnyInput();
 564         }
 565         else if (reading==KEYBOARD) {
 566             allowBreak();
 567             if (c0=='\n')
 568                 c1 = EOF;
 569             else {
 570                 c1 = nextConsoleChar();
 571 #if IS_WIN32 && !HUGS_FOR_WINDOWS
 572                 Sleep(0);
 573 #endif
 574                 /* On Win32, hitting ctrl-C causes the next getchar to
 575                  * fail - returning "-1" to indicate an error.
 576                  * This is one of the rare cases where "-1" does not mean EOF.
 577                  */
 578                 if (EOF == c1 && (!feof(stdin) || broken==TRUE)) {
 579                     c1 = ' ';
 580                 }
 581             }
 582         }
 583         else if (reading==STRING) {
 584             c1 = (unsigned char) *nextStringChar++;
 585             if (c1 == '\0')
 586                 c1 = EOF;
 587         }
 588         else {
 589             if (lineLength <=0 || linePtr == lineLength) {
 590                 /* Current line, exhausted - get new one */
 591                 if (nextLine() <= 0) { /* EOF */
 592                     c1 = EOF;
 593                 }
 594                 else {
 595                     linePtr = 0;
 596                     c1 = (unsigned char)lineBuffer[linePtr++];
 597                 }
 598             }
 599             else {
 600                 c1 = (unsigned char)lineBuffer[linePtr++];
 601             }
 602         }
 603
 604     }
 605 }
 606
 607 static Void local thisLineIs(kind)     /* register kind of current line    */
 608 Int kind; {                            /* & check for literate script errs */
 609     if (literateErrors) {
 610         if ((kind==DEFNLINE && lastLine==TEXTLINE) ||
 611             (kind==TEXTLINE && lastLine==DEFNLINE)) {
 612             ERRMSG(row) "Program line next to comment"
 613             EEND;
 614         }
 615         lastLine = kind;
 616     }
 617 }
 618
 619 static Void local newlineSkip() {      /* skip `\n' (supports lit scripts) */
 620     /* assert(c0=='\n'); */
 621     if (reading==SCRIPTFILE && thisLiterate) {
 622         do {
 623             skip();
 624             if (inCodeBlock) {         /* pass chars on definition lines   */
 625                 thisLineIs(CODELINE);  /* to lexer (w/o leading DEFNCHAR)  */
 626                 litLines++;
 627                 return;
 628             }
 629             if (c0==DEFNCHAR) {        /* pass chars on definition lines   */
 630                 thisLineIs(DEFNLINE);  /* to lexer (w/o leading DEFNCHAR)  */
 631                 skip();
 632                 litLines++;
 633                 return;
 634             }
 635             while (c0 != '\n' && isIn(c0,ZPACE)) /* maybe line is blank?   */
 636                 skip();
 637             if (c0=='\n' || c0==EOF)
 638                 thisLineIs(BLANKLINE);
 639             else {
 640                 thisLineIs(TEXTLINE);  /* otherwise it must be a comment   */
 641                 while (c0!='\n' && c0!=EOF)
 642                     skip();
 643             }                          /* by now, c0=='\n' or c0==EOF      */
 644         } while (c0!=EOF);             /* if new line, start again         */
 645
 646         if (litLines==0 && literateErrors) {
 647             ERRMSG(row) "Empty script - perhaps you forgot the `%c's?",
 648                         DEFNCHAR
 649             EEND;
 650         }
 651         return;
 652     }
 653     skip();
 654 }
 655
 656 static Void local closeAnyInput() {    /* Close input stream, if open,     */
 657     switch (reading) {                 /* or skip to end of console line   */
 658         case PROJFILE   :
 659         case SCRIPTFILE : if (inputStream) {
 660 #if USE_PREPROCESSOR && (defined(HAVE_POPEN) || defined(HAVE__POPEN))
 661                               if (preprocessor) {
 662                                   pclose(inputStream);
 663                               } else {
 664                                   fclose(inputStream);
 665                               }
 666 #else
 667                               fclose(inputStream);
 668 #endif
 669                               inputStream = 0;
 670                           }
 671                           break;
 672         case KEYBOARD   : while (c0!=EOF)
 673                               skip();
 674                           break;
 675     }
 676     reading=NOTHING;
 677 }
 678
 679 /* --------------------------------------------------------------------------
 680  * Parser: Uses table driven parser generated from parser.y using yacc
 681  * ------------------------------------------------------------------------*/
 682
 683 #include "parser.c"
 684
 685 /* --------------------------------------------------------------------------
 686  * Single token input routines:
 687  *
 688  * The following routines read the values of particular kinds of token given
 689  * that the first character of the token has already been located in c0 on
 690  * entry to the routine.
 691  * ------------------------------------------------------------------------*/
 692
 693 #define MAX_TOKEN           4000
 694 #define startToken()        tokPos = 0
 695 #define saveTokenChar(c)    if (tokPos<=MAX_TOKEN) saveChar(c); else ++tokPos
 696 #define saveChar(c)         tokenStr[tokPos++]=(char)(c)
 697 #define overflows(n,b,d,m)  (n > ((m)-(d))/(b))
 698
 699 static char tokenStr[MAX_TOKEN+1];     /* token buffer                     */
 700 static Int  tokPos;                    /* input position in buffer         */
 701 static Int  identType;                 /* identifier type: CONID / VARID   */
 702 static Int  opType;                    /* operator type  : CONOP / VAROP   */
 703
 704 static Void local endToken() {         /* check for token overflow         */
 705     if (tokPos>MAX_TOKEN) {
 706         ERRMSG(row) "Maximum token length (%d) exceeded", MAX_TOKEN
 707         EEND;
 708     }
 709     tokenStr[tokPos] = '\0';
 710 }
 711
 712 static Text local readOperator() {     /* read operator symbol             */
 713     startToken();
 714     do {
 715         saveTokenChar(c0);
 716         skip();
 717     } while (isISO(c0) && isIn(c0,SYMBOL));
 718     opType = (tokenStr[0]==':' ? CONOP : VAROP);
 719     endToken();
 720     return findText(tokenStr);
 721 }
 722
 723 static Text local readIdent() {        /* read identifier                  */
 724     startToken();
 725     do {
 726         saveTokenChar(c0);
 727         skip();
 728     } while (isISO(c0) && isIn(c0,IDAFTER));
 729     endToken();
 730     identType = isIn(tokenStr[0],LARGE) ? CONID : VARID;
 731     if (readingInterface)
 732        return unZcodeThenFindText(tokenStr); else
 733        return findText(tokenStr);
 734 }
 735
 736
 737 static Bool local doesNotExceed(s,radix,limit)
 738 String s;
 739 Int    radix;
 740 Int    limit; {
 741     Int n = 0;
 742     Int p = 0;
 743     while (TRUE) {
 744         if (s[p] == 0) return TRUE;
 745         if (overflows(n,radix,s[p]-'0',limit)) return FALSE;
 746         n = radix*n + (s[p]-'0');
 747         p++;
 748     }
 749 }
 750
 751 static Int local stringToInt(s,radix)
 752 String s;
 753 Int    radix; {
 754     Int n = 0;
 755     Int p = 0;
 756     while (TRUE) {
 757         if (s[p] == 0) return n;
 758         n = radix*n + (s[p]-'0');
 759         p++;
 760     }
 761 }
 762
 763 static Cell local readRadixNumber(r)   /* Read literal in specified radix  */
 764 Int r; {                               /* from input of the form 0c{digs}  */
 765     Int d;
 766     startToken();
 767     skip();                            /* skip leading zero                */
 768     if ((d=readHexDigit(c1))<0 || d>=r) {
 769         /* Special case; no digits, lex as  */
 770         /* if it had been written "0 c..."  */
 771         saveTokenChar('0');
 772     } else {
 773         skip();
 774         do {
 775             saveTokenChar('0'+readHexDigit(c0));
 776             skip();
 777             d = readHexDigit(c0);
 778         } while (d>=0 && d<r);
 779     }
 780     endToken();
 781
 782     if (doesNotExceed(tokenStr,r,MAXPOSINT))
 783         return mkInt(stringToInt(tokenStr,r));
 784     else
 785     if (r == 10)
 786         return stringToBignum(tokenStr);
 787     else {
 788         ERRMSG(row) "Hexadecimal or octal constant exceeds `Int' range"
 789         EEND;
 790     }
 791 }
 792
 793 static Cell local readNumber() {        /* read numeric constant           */
 794
 795     if (c0=='0') {
 796         if (c1=='x' || c1=='X')         /* Maybe a hexadecimal literal?    */
 797             return readRadixNumber(16);
 798         if (c1=='o' || c1=='O')         /* Maybe an octal literal?         */
 799             return readRadixNumber(8);
 800     }
 801
 802     startToken();
 803     do {
 804         saveTokenChar(c0);
 805         skip();
 806     } while (isISO(c0) && isIn(c0,DIGIT));
 807
 808     if (c0!='.' || !isISO(c1) || !isIn(c1,DIGIT)) {
 809         endToken();
 810         if (doesNotExceed(tokenStr,10,MAXPOSINT))
 811             return mkInt(stringToInt(tokenStr,10)); else
 812             return stringToBignum(tokenStr);
 813     }
 814
 815     saveTokenChar(c0);                  /* save decimal point              */
 816     skip();
 817     do {                                /* process fractional part ...     */
 818         saveTokenChar(c0);
 819         skip();
 820     } while (isISO(c0) && isIn(c0,DIGIT));
 821
 822     if (c0=='e' || c0=='E') {           /* look for exponent part...       */
 823         saveTokenChar('e');
 824         skip();
 825         if (c0=='-') {
 826             saveTokenChar('-');
 827             skip();
 828         }
 829         else if (c0=='+')
 830             skip();
 831
 832         if (!isISO(c0) || !isIn(c0,DIGIT)) {
 833             ERRMSG(row) "Missing digits in exponent"
 834             EEND;
 835         }
 836         else {
 837             do {
 838                 saveTokenChar(c0);
 839                 skip();
 840             } while (isISO(c0) && isIn(c0,DIGIT));
 841         }
 842     }
 843
 844     endToken();
 845     return mkFloat(stringToFloat(tokenStr));
 846 }
 847
 848
 849
 850
 851
 852
 853
 854 static Cell local readChar() {         /* read character constant          */
 855     Cell charRead;
 856
 857     skip(/* '\'' */);
 858     if (c0=='\'' || c0=='\n' || c0==EOF) {
 859         ERRMSG(row) "Illegal character constant"
 860         EEND;
 861     }
 862
 863     charRead = readAChar(FALSE);
 864
 865     if (c0=='\'')
 866         skip(/* '\'' */);
 867     else {
 868         ERRMSG(row) "Improperly terminated character constant"
 869         EEND;
 870     }
 871     return charRead;
 872 }
 873
 874 static Cell local readString() {       /* read string literal              */
 875     Cell c;
 876
 877     startToken();
 878     skip(/* '\"' */);
 879     while (c0!='\"' && c0!='\n' && c0!=EOF) {
 880         c = readAChar(TRUE);
 881         if (nonNull(c))
 882             saveStrChr(charOf(c));
 883     }
 884
 885     if (c0=='\"')
 886         skip(/* '\"' */);
 887     else {
 888         ERRMSG(row) "Improperly terminated string"
 889         EEND;
 890     }
 891     endToken();
 892     return mkStr(findText(tokenStr));
 893 }
 894
 895 static Void local saveStrChr(c)        /* save character in string         */
 896 Char c; {
 897     if (c!='\0' && c!='\\') {          /* save non null char as single char*/
 898         saveTokenChar(c);
 899     }
 900     else {                             /* save null char as TWO null chars */
 901         if (tokPos+1<MAX_TOKEN) {
 902             saveChar('\\');
 903             if (c=='\\')
 904                 saveChar('\\');
 905             else
 906                 saveChar('0');
 907         }
 908     }
 909 }
 910
 911 static Cell local readAChar(isStrLit)  /* read single char constant        */
 912 Bool isStrLit; {                       /* TRUE => enable \& and gaps       */
 913     Cell c = mkChar(c0);
 914
 915     if (c0=='\\')                      /* escape character?                */
 916         return readEscapeChar(isStrLit);
 917     if (!isISO(c0)) {
 918         ERRMSG(row) "Non ISO character `\\%d' in constant", ((int)c0)
 919         EEND;
 920     }
 921     skip();                            /* normal character?                */
 922     return c;
 923 }
 924
 925 /* --------------------------------------------------------------------------
 926  * Character escape code sequences:
 927  * ------------------------------------------------------------------------*/
 928
 929 static struct {                        /* table of special escape codes    */
 930     char *codename;
 931     int  codenumber;
 932 } escapes[] = {
 933    {"a",    7}, {"b",    8}, {"f",   12}, {"n",   10},  /* common escapes  */
 934    {"r",   13}, {"t",    9}, {"\\",'\\'}, {"\"",'\"'},
 935    {"\'",'\''}, {"v",   11},
 936    {"NUL",  0}, {"SOH",  1}, {"STX",  2}, {"ETX",  3},  /* ascii codenames */
 937    {"EOT",  4}, {"ENQ",  5}, {"ACK",  6}, {"BEL",  7},
 938    {"BS",   8}, {"HT",   9}, {"LF",  10}, {"VT",  11},
 939    {"FF",  12}, {"CR",  13}, {"SO",  14}, {"SI",  15},
 940    {"DLE", 16}, {"DC1", 17}, {"DC2", 18}, {"DC3", 19},
 941    {"DC4", 20}, {"NAK", 21}, {"SYN", 22}, {"ETB", 23},
 942    {"CAN", 24}, {"EM",  25}, {"SUB", 26}, {"ESC", 27},
 943    {"FS",  28}, {"GS",  29}, {"RS",  30}, {"US",  31},
 944    {"SP",  32}, {"DEL", 127},
 945    {0,0}
 946 };
 947
 948 static Int  alreadyMatched;            /* Record portion of input stream   */
 949 static char alreadyRead[10];           /* that has been read w/o a match   */
 950
 951 static Bool local lazyReadMatches(s)   /* compare input stream with string */
 952 String s; {                            /* possibly using characters that   */
 953     int i;                             /* have already been read           */
 954
 955     for (i=0; i<alreadyMatched; ++i)
 956         if (alreadyRead[i]!=s[i])
 957             return FALSE;
 958
 959     while (s[i] && s[i]==c0) {
 960         alreadyRead[alreadyMatched++]=(char)c0;
 961         skip();
 962         i++;
 963     }
 964
 965     return s[i]=='\0';
 966 }
 967
 968 static Cell local readEscapeChar(isStrLit)/* read escape character         */
 969 Bool isStrLit; {
 970     int i=0;
 971
 972     skip(/* '\\' */);
 973     switch (c0) {
 974         case '&'  : if (isStrLit) {
 975                         skip();
 976                         return NIL;
 977                     }
 978                     ERRMSG(row) "Illegal use of `\\&' in character constant"
 979                     EEND;
 980                     break;/*NOTREACHED*/
 981
 982         case '^'  : return readCtrlChar();
 983
 984         case 'o'  : return readOctChar();
 985         case 'x'  : return readHexChar();
 986
 987         default   : if (!isISO(c0)) {
 988                         ERRMSG(row) "Illegal escape sequence"
 989                         EEND;
 990                     }
 991                     else if (isIn(c0,ZPACE)) {
 992                         if (isStrLit) {
 993                             skipGap();
 994                             return NIL;
 995                         }
 996                         ERRMSG(row) "Illegal use of gap in character constant"
 997                         EEND;
 998                         break;
 999                     }
1000                     else if (isIn(c0,DIGIT))
1001                         return readDecChar();
1002     }
1003
1004     for (alreadyMatched=0; escapes[i].codename; i++)
1005         if (lazyReadMatches(escapes[i].codename))
1006             return mkChar(escapes[i].codenumber);
1007
1008     alreadyRead[alreadyMatched++] = (char)c0;
1009     alreadyRead[alreadyMatched++] = '\0';
1010     ERRMSG(row) "Illegal character escape sequence \"\\%s\"",
1011                 alreadyRead
1012     EEND;
1013     return NIL;/*NOTREACHED*/
1014 }
1015
1016 static Void local skipGap() {          /* skip over gap in string literal  */
1017     do                                 /* (simplified in Haskell 1.1)      */
1018         if (c0=='\n')
1019             newlineSkip();
1020         else
1021             skip();
1022     while (isISO(c0) && isIn(c0,ZPACE));
1023     if (c0!='\\') {
1024         ERRMSG(row) "Missing `\\' terminating string literal gap"
1025         EEND;
1026     }
1027     skip(/* '\\' */);
1028 }
1029
1030 static Cell local readCtrlChar() {     /* read escape sequence \^x         */
1031     static String controls = "@ABCDEFGHIJKLMNOPQRSTUVWXYZ[\\]^_";
1032     String which;
1033
1034     skip(/* '^' */);
1035     if ((which = strchr(controls,c0))==NULL) {
1036         ERRMSG(row) "Unrecognised escape sequence `\\^%c'", c0
1037         EEND;
1038     }
1039     skip();
1040     return mkChar(which-controls);
1041 }
1042
1043 static Cell local readOctChar() {      /* read octal character constant    */
1044     Int n = 0;
1045     Int d;
1046
1047     skip(/* 'o' */);
1048     if ((d = readHexDigit(c0))<0 || d>=8) {
1049         ERRMSG(row) "Empty octal character escape"
1050         EEND;
1051     }
1052     do {
1053         if (overflows(n,8,d,MAXCHARVAL)) {
1054             ERRMSG(row) "Octal character escape out of range"
1055             EEND;
1056         }
1057         n = 8*n + d;
1058         skip();
1059     } while ((d = readHexDigit(c0))>=0 && d<8);
1060
1061     return mkChar(n);
1062 }
1063
1064 static Cell local readHexChar() {      /* read hex character constant      */
1065     Int n = 0;
1066     Int d;
1067
1068     skip(/* 'x' */);
1069     if ((d = readHexDigit(c0))<0) {
1070         ERRMSG(row) "Empty hexadecimal character escape"
1071         EEND;
1072     }
1073     do {
1074         if (overflows(n,16,d,MAXCHARVAL)) {
1075             ERRMSG(row) "Hexadecimal character escape out of range"
1076             EEND;
1077         }
1078         n = 16*n + d;
1079         skip();
1080     } while ((d = readHexDigit(c0))>=0);
1081
1082     return mkChar(n);
1083 }
1084
1085 static Int local readHexDigit(c)       /* read single hex digit            */
1086 Char c; {
1087     if ('0'<=c && c<='9')
1088         return c-'0';
1089     if ('A'<=c && c<='F')
1090         return 10 + (c-'A');
1091     if ('a'<=c && c<='f')
1092         return 10 + (c-'a');
1093     return -1;
1094 }
1095
1096 static Cell local readDecChar() {      /* read decimal character constant  */
1097     Int n = 0;
1098
1099     do {
1100         if (overflows(n,10,(c0-'0'),MAXCHARVAL)) {
1101             ERRMSG(row) "Decimal character escape out of range"
1102             EEND;
1103         }
1104         n = 10*n + (c0-'0');
1105         skip();
1106     } while (c0!=EOF && isIn(c0,DIGIT));
1107
1108     return mkChar(n);
1109 }
1110
1111 /* --------------------------------------------------------------------------
1112  * Produce printable representation of character:
1113  * ------------------------------------------------------------------------*/
1114
1115 String unlexChar(c,quote)              /* return string representation of  */
1116 Char c;                                /* character...                     */
1117 Char quote; {                          /* protect quote character          */
1118     static char buffer[12];
1119
1120     if (c<0)                           /* deal with sign extended chars..  */
1121         c += NUM_CHARS;
1122
1123     if (isISO(c) && isIn(c,PRINT)) {   /* normal printable character       */
1124         if (c==quote || c=='\\') {     /* look for quote of approp. kind   */
1125             buffer[0] = '\\';
1126             buffer[1] = (char)c;
1127             buffer[2] = '\0';
1128         }
1129         else {
1130             buffer[0] = (char)c;
1131             buffer[1] = '\0';
1132         }
1133     }
1134     else {                             /* look for escape code             */
1135         Int escs;
1136         for (escs=0; escapes[escs].codename; escs++)
1137             if (escapes[escs].codenumber==c) {
1138                 sprintf(buffer,"\\%s",escapes[escs].codename);
1139                 return buffer;
1140             }
1141         sprintf(buffer,"\\%d",c);      /* otherwise use numeric escape     */
1142     }
1143     return buffer;
1144 }
1145
1146 Void printString(s)                    /* print string s, using quotes and */
1147 String s; {                            /* escapes if any parts need them   */
1148     if (s) {
1149         String t = s;
1150         Char   c;
1151         while ((c = *t)!=0 && isISO(c)
1152                            && isIn(c,PRINT) && c!='"' && !isIn(c,ZPACE)) {
1153             t++;
1154         }
1155         if (*t) {
1156             Putchar('"');
1157             for (t=s; *t; t++)
1158                 Printf("%s",unlexChar(*t,'"'));
1159             Putchar('"');
1160         }
1161         else
1162             Printf("%s",s);
1163     }
1164 }
1165
1166 /* -------------------------------------------------------------------------
1167  * Handle special types of input for use in interpreter:
1168  * -----------------------------------------------------------------------*/
1169
1170 Command readCommand(cmds,start,sys)    /* read command at start of input   */
1171 struct cmd *cmds;                      /* line in interpreter              */
1172 Char   start;                          /* characters introducing a cmd     */
1173 Char   sys; {                          /* character for shell escape       */
1174     while (c0==' ' || c0 =='\t')
1175         skip();
1176
1177     if (c0=='\n')                      /* look for blank command lines     */
1178         return NOCMD;
1179     if (c0==EOF)                       /* look for end of input stream     */
1180         return QUIT;
1181     if (c0==sys) {                     /* single character system escape   */
1182         skip();
1183         return SYSTEM;
1184     }
1185     if (c0==start && c1==sys) {        /* two character system escape      */
1186         skip();
1187         skip();
1188         return SYSTEM;
1189     }
1190
1191     startToken();                      /* All cmds start with start        */
1192     if (c0==start)                     /* except default (usually EVAL)    */
1193         do {                           /* which is empty                   */
1194             saveTokenChar(c0);
1195             skip();
1196         } while (c0!=EOF && !isIn(c0,ZPACE));
1197     endToken();
1198
1199     for (; cmds->cmdString; ++cmds)
1200         if (strcmp((cmds->cmdString),tokenStr)==0 ||
1201             (tokenStr[0]==start &&
1202              tokenStr[1]==(cmds->cmdString)[1] &&
1203              tokenStr[2]=='\0'))
1204             return (cmds->cmdCode);
1205     return BADCMD;
1206 }
1207
1208 String readFilename() {                /* Read filename from input (if any)*/
1209     if (reading==PROJFILE)
1210         skipWhitespace();
1211     else
1212         while (c0==' ' || c0=='\t')
1213             skip();
1214
1215     if (c0=='\n' || c0==EOF)           /* return null string at end of line*/
1216         return 0;
1217
1218     startToken();
1219     while (c0!=EOF && !isIn(c0,ZPACE)) {
1220         if (c0=='"') {
1221             skip();
1222             while (c0!=EOF && c0!='\"') {
1223                 Cell c = readAChar(TRUE);
1224                 if (nonNull(c)) {
1225                     saveTokenChar(charOf(c));
1226                 }
1227             }
1228             if (c0=='"')
1229                 skip();
1230             else {
1231                 ERRMSG(row) "a closing quote, '\"', was expected"
1232                 EEND;
1233             }
1234         }
1235         else {
1236             saveTokenChar(c0);
1237             skip();
1238         }
1239     }
1240     endToken();
1241     return tokenStr;
1242 }
1243
1244 String readLine() {                    /* Read command line from input     */
1245     while (c0==' ' || c0=='\t')        /* skip leading whitespace          */
1246         skip();
1247
1248     startToken();
1249     while (c0!='\n' && c0!=EOF) {
1250         saveTokenChar(c0);
1251         skip();
1252     }
1253     endToken();
1254
1255     return tokenStr;
1256 }
1257
1258 /* --------------------------------------------------------------------------
1259  * This lexer supports the Haskell layout rule:
1260  *
1261  * - Layout area bounded by { ... }, with `;'s in between.
1262  * - A `{' is a HARD indentation and can only be matched by a corresponding
1263  *   HARD '}'
1264  * - Otherwise, if no `{' follows the keywords WHERE/LET or OF, a SOFT `{'
1265  *   is inserted with the column number of the first token after the
1266  *   WHERE/LET/OF keyword.
1267  * - When a soft indentation is uppermost on the indetation stack with
1268  *   column col' we insert:
1269  *    `}'  in front of token with column<col' and pop indentation off stack,
1270  *    `;'  in front of token with column==col'.
1271  * ------------------------------------------------------------------------*/
1272
1273 #define MAXINDENT  100                 /* maximum nesting of layout rule   */
1274 static  Int        layout[MAXINDENT+1];/* indentation stack                */
1275 #define HARD       (-1)                /* indicates hard indentation       */
1276 static  Int        indentDepth = (-1); /* current indentation nesting      */
1277
1278 static Void local goOffside(col)       /* insert offside marker            */
1279 Int col; {                             /* for specified column             */
1280     assert(offsideON);
1281     if (indentDepth>=MAXINDENT) {
1282         ERRMSG(row) "Too many levels of program nesting"
1283         EEND;
1284     }
1285     layout[++indentDepth] = col;
1286 }
1287
1288 static Void local unOffside() {        /* leave layout rule area           */
1289     assert(offsideON);
1290     indentDepth--;
1291 }
1292
1293 static Bool local canUnOffside() {     /* Decide if unoffside permitted    */
1294     assert(offsideON);
1295     return indentDepth>=0 && layout[indentDepth]!=HARD;
1296 }
1297
1298 /* --------------------------------------------------------------------------
1299  * Main tokeniser:
1300  * ------------------------------------------------------------------------*/
1301
1302 static Void local skipWhitespace() {   /* Skip over whitespace/comments    */
1303     for (;;)                           /* Strictly speaking, this code is  */
1304         if (c0==EOF)                   /* a little more liberal than the   */
1305             return;                    /* report allows ...                */
1306         else if (c0=='\n')
1307             newlineSkip();
1308         else if (isIn(c0,ZPACE))
1309             skip();
1310         else if (c0=='{' && c1=='-') { /* (potentially) nested comment     */
1311             Int nesting = 1;
1312             Int origRow = row;         /* Save original row number         */
1313             skip();
1314             skip();
1315             while (nesting>0 && c0!=EOF)
1316                 if (c0=='{' && c1=='-') {
1317                     skip();
1318                     skip();
1319                     nesting++;
1320                 }
1321                 else if (c0=='-' && c1=='}') {
1322                     skip();
1323                     skip();
1324                     nesting--;
1325                 }
1326                 else if (c0=='\n')
1327                     newlineSkip();
1328                 else
1329                     skip();
1330             if (nesting>0) {
1331                 ERRMSG(origRow) "Unterminated nested comment {- ..."
1332                 EEND;
1333             }
1334         }
1335         else if (c0=='-' && c1=='-') {  /* One line comment                */
1336             do
1337                 skip();
1338             while (c0!='\n' && c0!=EOF);
1339             if (c0=='\n')
1340                 newlineSkip();
1341         }
1342         else
1343             return;
1344 }
1345
1346 static Bool firstToken;                /* Set to TRUE for first token      */
1347 static Int  firstTokenIs;              /* ... with token value stored here */
1348
1349 static Int local yylex() {             /* Read next input token ...        */
1350     static Bool insertOpen    = FALSE;
1351     static Bool insertedToken = FALSE;
1352     static Text textRepeat;
1353
1354 #define lookAhead(t) {skipWhitespace(); insertOpen = (c0!='{'); return t;}
1355
1356     if (firstToken) {                  /* Special case for first token     */
1357         indentDepth   = (-1);
1358         firstToken    = FALSE;
1359         insertOpen    = FALSE;
1360         insertedToken = FALSE;
1361         if (reading==KEYBOARD)
1362             textRepeat = findText(repeatStr);
1363         return firstTokenIs;
1364     }
1365
1366     if (offsideON && insertOpen) {     /* insert `soft' opening brace      */
1367         insertOpen    = FALSE;
1368         insertedToken = TRUE;
1369         goOffside(column);
1370         push(yylval = mkInt(row));
1371         return '{';
1372     }
1373
1374     /* ----------------------------------------------------------------------
1375      * Skip white space, and insert tokens to support layout rules as reqd.
1376      * --------------------------------------------------------------------*/
1377
1378     skipWhitespace();
1379     startColumn = column;
1380     push(yylval = mkInt(row));         /* default token value is line no.  */
1381     /* subsequent changes to yylval must also set top() to the same value  */
1382
1383     if (indentDepth>=0) {              /* layout rule(s) active ?          */
1384         if (insertedToken)             /* avoid inserting multiple `;'s    */
1385             insertedToken = FALSE;     /* or putting `;' after `{'         */
1386         else
1387         if (offsideON && layout[indentDepth]!=HARD) {
1388             if (column<layout[indentDepth]) {
1389                 unOffside();
1390                 return '}';
1391             }
1392             else if (column==layout[indentDepth] && c0!=EOF) {
1393                 insertedToken = TRUE;
1394                 return ';';
1395             }
1396         }
1397     }
1398
1399     /* ----------------------------------------------------------------------
1400      * Now try to identify token type:
1401      * --------------------------------------------------------------------*/
1402
1403     switch (c0) {
1404         case EOF  : return 0;                   /* End of file/input       */
1405
1406         /* The next 10 characters make up the `special' category in 1.3    */
1407         case '('  : skip(); return '(';
1408         case ')'  : skip(); return ')';
1409         case ','  : skip(); return ',';
1410         case ';'  : skip(); return ';';
1411         case '['  : skip(); return '[';
1412         case ']'  : skip(); return ']';
1413         case '`'  : skip(); return '`';
1414         case '{'  : if (offsideON) goOffside(HARD);
1415                     skip();
1416                     return '{';
1417         case '}'  : if (offsideON && indentDepth<0) {
1418                         ERRMSG(row) "Misplaced `}'"
1419                         EEND;
1420                     }
1421                     if (!(offsideON && layout[indentDepth]!=HARD))
1422                         skip();                         /* skip over hard }*/
1423                     if (offsideON)
1424                         unOffside();    /* otherwise, we have to insert a }*/
1425                     return '}';         /* to (try to) avoid an error...   */
1426
1427         /* Character and string literals                                   */
1428         case '\'' : top() = yylval = readChar();
1429                     return CHARLIT;
1430
1431         case '\"' : top() = yylval = readString();
1432                     return STRINGLIT;
1433     }
1434
1435 #if IPARAM
1436     if (c0=='?' && isIn(c1,SMALL) && !haskell98) {
1437         Text it;                        /* Look for implicit param name    */
1438         skip();
1439         it    = readIdent();
1440         top() = yylval = ap(IPVAR,it);
1441         return identType=IPVARID;
1442     }
1443 #endif
1444 #if TREX
1445     if (c0=='#' && isIn(c1,SMALL) && !haskell98) {
1446         Text it;                        /* Look for record selector name   */
1447         skip();
1448         it    = readIdent();
1449         top() = yylval = ap(RECSEL,mkExt(it));
1450         return identType=RECSELID;
1451     }
1452 #endif
1453     if (isIn(c0,LARGE)) {               /* Look for qualified name         */
1454         Text it = readIdent();          /* No keyword begins with LARGE ...*/
1455         if (c0=='.' && isIn(c1,(SMALL|LARGE|SYMBOL))) {
1456             Text it2 = NIL;
1457             skip();                     /* Skip qualifying dot             */
1458             if (isIn(c0,SYMBOL)) { /* Qualified operator */
1459                 it2 = readOperator();
1460                 if (opType==CONOP) {
1461                     top() = yylval = mkQConOp(it,it2);
1462                     return QCONOP;
1463                 } else {
1464                     top() = yylval = mkQVarOp(it,it2);
1465                     return QVAROP;
1466                 }
1467             } else {               /* Qualified identifier */
1468                 it2 = readIdent();
1469                 if (identType==CONID) {
1470                     top() = yylval = mkQCon(it,it2);
1471                     return QCONID;
1472                 } else {
1473                     top() = yylval = mkQVar(it,it2);
1474                     return QVARID;
1475                 }
1476             }
1477         } else {
1478             top() = yylval = mkCon(it);
1479             return identType;
1480         }
1481     }
1482     if (isIn(c0,(SMALL|LARGE))) {
1483         Text it = readIdent();
1484
1485         if (it==textCase)              return CASEXP;
1486         if (it==textOfK)               lookAhead(OF);
1487         if (it==textData)              return DATA;
1488         if (it==textType)              return TYPE;
1489         if (it==textIf)                return IF;
1490         if (it==textThen)              return THEN;
1491         if (it==textElse)              return ELSE;
1492         if (it==textWhere)             lookAhead(WHERE);
1493         if (it==textLet)               lookAhead(LET);
1494         if (it==textIn)                return IN;
1495         if (it==textInfix)             return INFIXN;
1496         if (it==textInfixl)            return INFIXL;
1497         if (it==textInfixr)            return INFIXR;
1498         if (it==textForeign)           return FOREIGN;
1499         if (it==textUnsafe)            return UNSAFE;
1500         if (it==textNewtype)           return TNEWTYPE;
1501         if (it==textDefault)           return DEFAULT;
1502         if (it==textDeriving)          return DERIVING;
1503         if (it==textDo)                lookAhead(DO);
1504         if (it==textClass)             return TCLASS;
1505         if (it==textInstance)          return TINSTANCE;
1506         if (it==textModule)            return TMODULE;
1507         if (it==textInterface)         return INTERFACE;
1508         if (it==textInstImport)        return INSTIMPORT;
1509         if (it==textImport)            return IMPORT;
1510         if (it==textExport)            return EXPORT;
1511         if (it==textDynamic)           return DYNAMIC;
1512         if (it==textCcall)             return CCALL;
1513         if (it==textStdcall)           return STDKALL;
1514         if (it==textUUExport)          return UUEXPORT;
1515         if (it==textHiding)            return HIDING;
1516         if (it==textQualified)         return QUALIFIED;
1517         if (it==textAsMod)             return ASMOD;
1518         if (it==textWildcard)          return '_';
1519         if (it==textAll && !haskell98) return ALL;
1520 #if IPARAM
1521         if (it==textWith && !haskell98) lookAhead(WITH);
1522         if (it==textDlet && !haskell98) lookAhead(DLET);
1523 #endif
1524         if (it==textUUAll)             return ALL;
1525         if (it==textRepeat && reading==KEYBOARD)
1526             return repeatLast();
1527
1528         top() = yylval = ap((identType==CONID ? CONIDCELL : VARIDCELL),it);
1529         return identType;
1530     }
1531
1532     if (isIn(c0,SYMBOL)) {
1533         Text it = readOperator();
1534
1535         if (it==textCoco)    return COCO;
1536         if (it==textEq)      return '=';
1537         if (it==textUpto)    return UPTO;
1538         if (it==textAs)      return '@';
1539         if (it==textLambda)  return '\\';
1540         if (it==textBar)     return '|';
1541         if (it==textFrom)    return FROM;
1542         if (it==textMinus)   return '-';
1543         if (it==textPlus)    return '+';
1544         if (it==textBang)    return '!';
1545         if (it==textDot)     return '.';
1546         if (it==textArrow)   return ARROW;
1547         if (it==textLazy)    return '~';
1548         if (it==textImplies) return IMPLIES;
1549         if (it==textRepeat && reading==KEYBOARD)
1550             return repeatLast();
1551
1552         top() = yylval = ap((opType==CONOP ? CONOPCELL : VAROPCELL),it);
1553         return opType;
1554     }
1555
1556     if (isIn(c0,DIGIT)) {
1557         top() = yylval = readNumber();
1558         return NUMLIT;
1559     }
1560
1561     ERRMSG(row) "Unrecognised character `\\%d' in column %d",
1562                 ((int)c0), column
1563     EEND;
1564     return 0; /*NOTREACHED*/
1565 }
1566
1567 static Int local repeatLast() {         /* Obtain last expression entered  */
1568     if (isNull(yylval=getLastExpr())) {
1569         ERRMSG(row) "Cannot use %s without any previous input", repeatStr
1570         EEND;
1571     }
1572     return REPEAT;
1573 }
1574
1575 Syntax defaultSyntax(t)                 /* Find default syntax of var named*/
1576 Text t; {                               /* by t ...                        */
1577     String s = textToStr(t);
1578     return isIn(s[0],SYMBOL) ? DEF_OPSYNTAX : APPLIC;
1579 }
1580
1581 Syntax syntaxOf(n)                      /* Find syntax for name            */
1582 Name n; {
1583     if (name(n).syntax==NO_SYNTAX)      /* Return default if no syntax set */
1584         return defaultSyntax(name(n).text);
1585     return name(n).syntax;
1586 }
1587
1588 /* --------------------------------------------------------------------------
1589  * main entry points to parser/lexer:
1590  * ------------------------------------------------------------------------*/
1591
1592 static Void local parseInput(startWith)/* Parse input with given first tok,*/
1593 Int startWith; {                       /* determining whether to read a    */
1594     firstToken   = TRUE;               /* script or an expression          */
1595     firstTokenIs = startWith;
1596     if (startWith==INTERFACE) {
1597        offsideON = FALSE; readingInterface = TRUE;
1598     } else {
1599        offsideON = TRUE; readingInterface = FALSE;
1600     }
1601
1602     clearStack();
1603     if (yyparse()) {                   /* This can only be parser overflow */
1604         ERRMSG(row) "Parser overflow"  /* as all syntax errors are caught  */
1605         EEND;                          /* in the parser...                 */
1606     }
1607     drop();
1608     if (!stackEmpty())                 /* stack should now be empty        */
1609         internal("parseInput");
1610 }
1611
1612 #ifdef HSCRIPT
1613 static String memPrefix = "@mem@";
1614 static Int lenMemPrefix = 5;   /* strlen(memPrefix)*/
1615
1616 Void makeMemScript(mem,fname)
1617 String mem;
1618 String fname; {
1619    strcat(fname,memPrefix);
1620    itoa((int)mem, fname+strlen(fname), 10);
1621 }
1622
1623 Bool isMemScript(fname)
1624 String fname; {
1625    return (strstr(fname,memPrefix) != NULL);
1626 }
1627
1628 String memScriptString(fname)
1629 String fname; {
1630     String p = strstr(fname,memPrefix);
1631     if (p) {
1632         return (String)atoi(p+lenMemPrefix);
1633     } else {
1634         return NULL;
1635     }
1636 }
1637
1638 Void parseScript(fname,len)             /* Read a script, possibly from mem */
1639 String fname;
1640 Long len; {
1641     input(RESET);
1642     if (isMemScript(fname)) {
1643         char* s = memScriptString(fname);
1644         stringInput(s);
1645     } else {
1646         fileInput(fname,len);
1647     }
1648     parseInput(SCRIPT);
1649 }
1650 #else
1651 Void parseScript(nm,len)               /* Read a script                    */
1652 String nm;
1653 Long   len; {                          /* Used to set a target for reading */
1654     input(RESET);
1655     fileInput(nm,len);
1656     parseInput(SCRIPT);
1657 }
1658 #endif
1659
1660 Void parseExp() {                      /* Read an expression to evaluate   */
1661     parseInput(EXPR);
1662     setLastExpr(inputExpr);
1663 }
1664
1665
1666 #if EXPLAIN_INSTANCE_RESOLUTION
1667 Void parseContext() {                  /* Read a context to prove   */
1668     parseInput(CONTEXT);
1669 }
1670 #endif
1671
1672 Void parseInterface(nm,len)            /* Read a GHC interface file        */
1673 String nm;
1674 Long   len; {                          /* Used to set a target for reading */
1675     input(RESET);
1676     fileInput(nm,len);
1677     parseInput(INTERFACE);
1678 }
1679
1680
1681 /* --------------------------------------------------------------------------
1682  * Input control:
1683  * ------------------------------------------------------------------------*/
1684
1685 Void input(what)
1686 Int what; {
1687     switch (what) {
1688         case INSTALL : initCharTab();
1689                        textCase       = findText("case");
1690                        textOfK        = findText("of");
1691                        textData       = findText("data");
1692                        textType       = findText("type");
1693                        textIf         = findText("if");
1694                        textThen       = findText("then");
1695                        textElse       = findText("else");
1696                        textWhere      = findText("where");
1697                        textLet        = findText("let");
1698                        textIn         = findText("in");
1699                        textInfix      = findText("infix");
1700                        textInfixl     = findText("infixl");
1701                        textInfixr     = findText("infixr");
1702                        textForeign    = findText("foreign");
1703                        textUnsafe     = findText("unsafe");
1704                        textNewtype    = findText("newtype");
1705                        textDefault    = findText("default");
1706                        textDeriving   = findText("deriving");
1707                        textDo         = findText("do");
1708                        textClass      = findText("class");
1709 #if IPARAM
1710                        textWith       = findText("with");
1711                        textDlet       = findText("dlet");
1712 #endif
1713                        textInstance   = findText("instance");
1714                        textCoco       = findText("::");
1715                        textEq         = findText("=");
1716                        textUpto       = findText("..");
1717                        textAs         = findText("@");
1718                        textLambda     = findText("\\");
1719                        textBar        = findText("|");
1720                        textMinus      = findText("-");
1721                        textPlus       = findText("+");
1722                        textFrom       = findText("<-");
1723                        textArrow      = findText("->");
1724                        textLazy       = findText("~");
1725                        textBang       = findText("!");
1726                        textDot        = findText(".");
1727                        textImplies    = findText("=>");
1728                        textPrelude    = findText("Prelude");
1729                        textNum        = findText("Num");
1730                        textModule     = findText("module");
1731                        textInterface  = findText("__interface");
1732                        textInstImport = findText("__instimport");
1733                        textExport     = findText("export");
1734                        textDynamic    = findText("dynamic");
1735                        textCcall      = findText("ccall");
1736                        textStdcall    = findText("stdcall");
1737                        textUUExport   = findText("__export");
1738                        textImport     = findText("import");
1739                        textHiding     = findText("hiding");
1740                        textQualified  = findText("qualified");
1741                        textAsMod      = findText("as");
1742                        textWildcard   = findText("_");
1743                        textAll        = findText("forall");
1744                        textUUAll      = findText("__forall");
1745                        varMinus       = mkVar(textMinus);
1746                        varPlus        = mkVar(textPlus);
1747                        varBang        = mkVar(textBang);
1748                        varDot         = mkVar(textDot);
1749                        varHiding      = mkVar(textHiding);
1750                        varQualified   = mkVar(textQualified);
1751                        varAsMod       = mkVar(textAsMod);
1752                        conMain        = mkCon(findText("Main"));
1753                        varMain        = mkVar(findText("main"));
1754                        evalDefaults   = NIL;
1755
1756                        input(RESET);
1757                        break;
1758
1759         case RESET   : tyconDefns   = NIL;
1760                        typeInDefns  = NIL;
1761                        valDefns     = NIL;
1762                        classDefns   = NIL;
1763                        instDefns    = NIL;
1764                        selDefns     = NIL;
1765                        genDefns     = NIL;
1766                        //primDefns    = NIL;
1767                        unqualImports= NIL;
1768                        foreignImports= NIL;
1769                        foreignExports= NIL;
1770                        defaultDefns = NIL;
1771                        defaultLine  = 0;
1772                        inputExpr    = NIL;
1773                        imps         = NIL;
1774                        closeAnyInput();
1775                        break;
1776
1777         case BREAK   : if (reading==KEYBOARD)
1778                            c0 = EOF;
1779                        break;
1780
1781         case MARK    : mark(tyconDefns);
1782                        mark(typeInDefns);
1783                        mark(valDefns);
1784                        mark(classDefns);
1785                        mark(instDefns);
1786                        mark(selDefns);
1787                        mark(genDefns);
1788                        //mark(primDefns);
1789                        mark(unqualImports);
1790                        mark(foreignImports);
1791                        mark(foreignExports);
1792                        mark(defaultDefns);
1793                        mark(evalDefaults);
1794                        mark(inputExpr);
1795                        mark(varMinus);
1796                        mark(varPlus);
1797                        mark(varBang);
1798                        mark(varDot);
1799                        mark(varHiding);
1800                        mark(varQualified);
1801                        mark(varAsMod);
1802                        mark(varMain);
1803                        mark(conMain);
1804                        mark(imps);
1805                        break;
1806     }
1807 }
1808
1809 /*-------------------------------------------------------------------------*/