home *** CD-ROM | disk | FTP | other *** search
/ C/C++ Interactive Guide / c-cplusplus-interactive-guide.iso / c_ref / csource4 / 276_01 / az80util.c < prev    next >
Encoding:
C/C++ Source or Header  |  1989-10-03  |  12.7 KB  |  510 lines
  1. /*
  2.     HEADER:        CUG276;
  3.     TITLE:        Z-80 Cross-Assembler (Portable);
  4.     FILENAME:    AZ80UTIL.C;
  5.     VERSION:    0.1;
  6.     DATE:        08/27/1988;
  7.     SEE-ALSO:    AZ80.H;
  8.     AUTHORS:    William C. Colley III;
  9. */
  10.  
  11. /*
  12.               Z-80 Cross-Assembler in Portable C
  13.  
  14.         Copyright (c) 1986-1988 William C. Colley, III
  15.  
  16. Revision History:
  17.  
  18. Ver    Date        Description
  19.  
  20. 0.0    JUNE 1988    Derived from my S6 cross-assembler.  WCC3.
  21.  
  22. 0.1    AUG 1988    Fixed a bug in the command line parser that puts it
  23.             into a VERY long loop if the user types a command line
  24.             like "AZ80 FILE.ASM -L".  WCC3 per Alex Cameron.
  25.  
  26. This module contains the following utility packages:
  27.  
  28.     1)  symbol table building and searching
  29.  
  30.     2)  opcode and operator table searching
  31.  
  32.     3)  listing file output
  33.  
  34.     4)  hex file output
  35.  
  36.     5)  error flagging
  37. */
  38.  
  39. /*  Get global goodies:  */
  40.  
  41. #include "az80.h"
  42.  
  43. /*  Make sure that MSDOS compilers using the large memory model know    */
  44. /*  that calloc() returns pointer to char as an MSDOS far pointer is    */
  45. /*  NOT compatible with the int type as is usually the case.        */
  46.  
  47. char *calloc();
  48.  
  49. /*  Get access to global mailboxes defined in AZ80.C:            */
  50.  
  51. extern char errcode, line[], title[];
  52. extern int eject, listhex;
  53. extern unsigned address, bytes, errors, listleft, obj[], pagelen;
  54.  
  55. /*  The symbol table is a binary tree of variable-length blocks drawn    */
  56. /*  from the heap with the calloc() function.  The root pointer lives    */
  57. /*  here:                                */
  58.  
  59. static SYMBOL *sroot = NULL;
  60.  
  61. /*  Add new symbol to symbol table.  Returns pointer to symbol even if    */
  62. /*  the symbol already exists.  If there's not enough memory to store    */
  63. /*  the new symbol, a fatal error occurs.                */
  64.  
  65. SYMBOL *new_symbol(nam)
  66. char *nam;
  67. {
  68.     SCRATCH int i;
  69.     SCRATCH SYMBOL **p, *q;
  70.     void fatal_error();
  71.  
  72.     for (p = &sroot; (q = *p) && (i = strcmp(nam,q -> sname)); )
  73.     p = i < 0 ? &(q -> left) : &(q -> right);
  74.     if (!q) {
  75.     if (!(*p = q = (SYMBOL *)calloc(1,sizeof(SYMBOL) + strlen(nam))))
  76.         fatal_error(SYMBOLS);
  77.     strcpy(q -> sname,nam);
  78.     }
  79.     return q;
  80. }
  81.  
  82. /*  Look up symbol in symbol table.  Returns pointer to symbol or NULL    */
  83. /*  if symbol not found.                        */
  84.  
  85. SYMBOL *find_symbol(nam)
  86. char *nam;
  87. {
  88.     SCRATCH int i;
  89.     SCRATCH SYMBOL *p;
  90.  
  91.     for (p = sroot; p && (i = strcmp(nam,p -> sname));
  92.     p = i < 0 ? p -> left : p -> right);
  93.     return p;
  94. }
  95.  
  96. /*  Opcode table search routine.  This routine pats down the opcode    */
  97. /*  table for a given opcode and returns either a pointer to it or    */
  98. /*  NULL if the opcode doesn't exist.                    */
  99.  
  100. OPCODE *find_code(nam)
  101. char *nam;
  102. {
  103.     OPCODE *bsearch();
  104.  
  105.     static OPCODE opctbl[] = {
  106.     { ADC,            0x88,    "ADC"    },
  107.     { ADD,            0x80,    "ADD"    },
  108.     { CP,            0xa0,    "AND"    },
  109.     { BIT,             0xcb40,    "BIT"    },
  110.     { CALL,            0xc4,    "CALL"    },
  111.     { NO_ARGS,        0x3f,    "CCF"    },
  112.     { PSEUDO + ISIF,    IF,    "COND"    },
  113.     { CP,            0xb8,    "CP"    },
  114.     { NO_ARGS,        0xeda9,    "CPD"    },
  115.     { NO_ARGS,        0xedb9,    "CPDR"    },
  116.     { NO_ARGS,        0xeda1,    "CPI"    },
  117.     { NO_ARGS,        0xedb1,    "CPIR"    },
  118.     { NO_ARGS,        0x2f,    "CPL"    },
  119.     { NO_ARGS,        0x27,    "DAA"    },
  120.     { PSEUDO,        DB,    "DB"    },
  121.     { PSEUDO,        DC,    "DC"    },
  122.     { DEC,            0x05,    "DEC"    },
  123.     { PSEUDO,        DB,    "DEFB"    },
  124.     { PSEUDO,        VAR,    "DEFL"    },
  125.     { PSEUDO,        DB,    "DEFM"    },
  126.     { PSEUDO,        DS,    "DEFS"    },
  127.     { PSEUDO,        DW,    "DEFW"    },
  128.     { NO_ARGS,        0xf3,    "DI"    },
  129.     { DJNZ,            0x10,    "DJNZ"    },
  130.     { PSEUDO,        DS,    "DS"    },
  131.     { PSEUDO,        DW,    "DW"    },
  132.     { NO_ARGS,        0xfb,    "EI"    },
  133.     { PSEUDO + ISIF,    ELSE,    "ELSE"    },
  134.     { PSEUDO,        END,    "END"    },
  135.     { PSEUDO + ISIF,    ENDIF,    "ENDC"    },
  136.     { PSEUDO + ISIF,    ENDIF,    "ENDIF"    },
  137.     { PSEUDO,        EQU,    "EQU"    },
  138.     { EX,            0xe3,    "EX"    },
  139.     { NO_ARGS,        0xd9,    "EXX"    },
  140.     { NO_ARGS,        0x76,    "HALT"    },
  141.     { PSEUDO + ISIF,    IF,    "IF"    },
  142.     { IM,            0xed46,    "IM"    },
  143.     { IN,            0xdb00,    "IN"    },
  144.     { DEC,            0x04,    "INC"    },
  145.     { PSEUDO,        INCL,    "INCL"    },
  146.     { NO_ARGS,        0xedaa,    "IND"    },
  147.     { NO_ARGS,        0xedba,    "INDR"    },
  148.     { NO_ARGS,        0xeda2,    "INI"    },
  149.     { NO_ARGS,        0xedb2,    "INIR"    },
  150.     { JP,            0xc2,    "JP"    },
  151.     { JR,            0x20,    "JR"    },
  152.     { LD,            0x40,    "LD"    },
  153.     { NO_ARGS,        0xeda8,    "LDD"    },
  154.     { NO_ARGS,        0xedb8,    "LDDR"    },
  155.     { NO_ARGS,        0xeda0,    "LDI"    },
  156.     { NO_ARGS,        0xedb0,    "LDIR"    },
  157.     { NO_ARGS,        0xed44,    "NEG"    },
  158.     { NO_ARGS,        0x00,    "NOP"    },
  159.     { CP,            0xb0,    "OR"    },
  160.     { PSEUDO,        ORG,    "ORG"    },
  161.     { NO_ARGS,        0xedbb,    "OTDR"    },
  162.     { NO_ARGS,        0xedb3,    "OTIR"    },
  163.     { OUT,            0xd300,    "OUT"    },
  164.     { NO_ARGS,        0xedab,    "OUTD"    },
  165.     { NO_ARGS,        0xeda3,    "OUTI"    },
  166.     { PSEUDO,        PAGE,    "PAGE"    },
  167.     { POP,            0xc1,    "POP"    },
  168.     { POP,            0xc5,    "PUSH"    },
  169.     { BIT,            0xcb80,    "RES"    },
  170.     { RET,            0xc0,    "RET"    },
  171.     { NO_ARGS,        0xed4d,    "RETI"    },
  172.     { NO_ARGS,        0xed45,    "RETN"    },
  173.     { RLC,            0xcb10,    "RL"    },
  174.     { NO_ARGS,        0x17,    "RLA"    },
  175.     { RLC,            0xcb00,    "RLC"    },
  176.     { NO_ARGS,        0x07,    "RLCA"    },
  177.     { NO_ARGS,        0xed6f,    "RLD"    },
  178.     { RLC,            0xcb18,    "RR"    },
  179.     { NO_ARGS,        0x1f,    "RRA"    },
  180.     { RLC,            0xcb08,    "RRC"    },
  181.     { NO_ARGS,        0x0f,    "RRCA"    },
  182.     { NO_ARGS,        0xed67,    "RRD"    },
  183.     { RST,            0xc7,    "RST"    },
  184.     { ADC,            0x98,    "SBC"    },
  185.     { NO_ARGS,        0x37,    "SCF"    },
  186.     { BIT,            0xcbc0,    "SET"    },
  187.     { RLC,            0xcb20,    "SLA"    },
  188.     { RLC,            0xcb28,    "SRA"    },
  189.     { RLC,            0xcb38,    "SRL"    },
  190.     { CP,            0x90,    "SUB"    },
  191.     { PSEUDO,        TITLE,    "TITLE"    },
  192.     { PSEUDO,        VAR,    "VAR"    },
  193.     { CP,            0xa8,    "XOR"    }
  194.     };
  195.  
  196.     return bsearch(opctbl,opctbl + (sizeof(opctbl) / sizeof(OPCODE)),nam);
  197. }
  198.  
  199. /*  Operator table search routine.  This routine pats down the        */
  200. /*  operator table for a given operator and returns either a pointer    */
  201. /*  to it or NULL if the opcode doesn't exist.                */
  202.  
  203. OPCODE *find_operator(nam)
  204. char *nam;
  205. {
  206.     OPCODE *bsearch();
  207.  
  208.     static OPCODE oprtbl[] = {
  209.     { REG,                A,        "A"    },
  210.     { REG,                AF,        "AF"    },
  211.     { BINARY + LOG1  + OPR,        AND,        "AND"    },
  212.     { REG,                B,        "B"    },
  213.     { REG,                BC,        "BC"    },
  214.     { REG,                C,        "C"    },
  215.     { REG,                D,        "D"    },
  216.     { REG,                DE,        "DE"    },
  217.     { REG,                E,        "E"    },
  218.     { BINARY + RELAT + OPR,        '=',        "EQ"    },
  219.     { BINARY + RELAT + OPR,        GE,        "GE"    },
  220.     { BINARY + RELAT + OPR,        '>',        "GT"    },
  221.     { REG,                H,        "H"    },
  222.     { UNARY  + UOP3  + OPR,        HIGH,        "HIGH"    },
  223.     { REG,                HL,        "HL"    },
  224.     { REG,                I,        "I"    },
  225.     { REG,                IX,        "IX"    },
  226.     { REG,                IY,        "IY"    },
  227.     { REG,                L,        "L"    },
  228.     { BINARY + RELAT + OPR,        LE,        "LE"    },
  229.     { UNARY  + UOP3  + OPR,        LOW,        "LOW"    },
  230.     { BINARY + RELAT + OPR,        '<',        "LT"    },
  231.     { REG,                M,        "M"    },
  232.     { BINARY + MULT  + OPR,        MOD,        "MOD"    },
  233.     { REG,                NC,        "NC"    },
  234.     { BINARY + RELAT + OPR,        NE,        "NE"    },
  235.     { UNARY  + UOP2  + OPR,        NOT,        "NOT"    },
  236.     { REG,                NZ,        "NZ"    },
  237.     { BINARY + LOG2  + OPR,        OR,        "OR"    },
  238.     { REG,                P,        "P"    },
  239.     { REG,                PE,        "PE"    },
  240.     { REG,                PO,        "PO"    },
  241.     { REG,                R,        "R"    },
  242.     { BINARY + MULT  + OPR,        SHL,        "SHL"    },
  243.     { BINARY + MULT  + OPR,        SHR,        "SHR"    },
  244.     { REG,                SP,        "SP"    },
  245.     { BINARY + LOG2  + OPR,        XOR,        "XOR"    },
  246.     { REG,                Z,        "Z"    }
  247.     };
  248.  
  249.     return bsearch(oprtbl,oprtbl + (sizeof(oprtbl) / sizeof(OPCODE)),nam);
  250. }
  251.  
  252. static OPCODE *bsearch(lo,hi,nam)
  253. OPCODE *lo, *hi;
  254. char *nam;
  255. {
  256.     SCRATCH int i;
  257.     SCRATCH OPCODE *chk;
  258.  
  259.     for (;;) {
  260.     chk = lo + (hi - lo) / 2;
  261.     if (!(i = ustrcmp(chk -> oname,nam))) return chk;
  262.     if (chk == lo) return NULL;
  263.     if (i < 0) lo = chk;
  264.     else hi = chk;
  265.     }
  266. }
  267.  
  268. static int ustrcmp(s,t)
  269. char *s, *t;
  270. {
  271.     SCRATCH int i;
  272.  
  273.     while (!(i = toupper(*s++) - toupper(*t)) && *t++);
  274.     return i;
  275. }
  276.  
  277. /*  Buffer storage for line listing routine.  This allows the listing    */
  278. /*  output routines to do all operations without the main routine    */
  279. /*  having to fool with it.                        */
  280.  
  281. static FILE *list = NULL;
  282.  
  283. /*  Listing file open routine.  If a listing file is already open, a    */
  284. /*  warning occurs.  If the listing file doesn't open correctly, a    */
  285. /*  fatal error occurs.  If no listing file is open, all calls to    */
  286. /*  lputs() and lclose() have no effect.                */
  287.  
  288. void lopen(nam)
  289. char *nam;
  290. {
  291.     FILE *fopen();
  292.     void fatal_error(), warning();
  293.  
  294.     if (list) warning(TWOLST);
  295.     else if (!(list = fopen(nam,"w"))) fatal_error(LSTOPEN);
  296.     return;
  297. }
  298.  
  299. /*  Listing file line output routine.  This routine processes the    */
  300. /*  source line saved by popc() and the output of the line assembler in    */
  301. /*  buffer obj into a line of the listing.  If the disk fills up, a    */
  302. /*  fatal error occurs.                            */
  303.  
  304. void lputs()
  305. {
  306.     SCRATCH int i, j;
  307.     SCRATCH unsigned *o;
  308.     void check_page(), fatal_error();
  309.  
  310.     if (list) {
  311.     i = bytes;  o = obj;
  312.     do {
  313.         fprintf(list,"%c  ",errcode);
  314.         if (listhex) {
  315.         fprintf(list,"%04x  ",address);
  316.         for (j = 4; j; --j) {
  317.             if (i) { --i;  ++address;  fprintf(list," %02x",*o++); }
  318.             else fprintf(list,"   ");
  319.         }
  320.         }
  321.         else fprintf(list,"%18s","");
  322.         fprintf(list,"   %s",line);  strcpy(line,"\n");
  323.         check_page();
  324.         if (ferror(list)) fatal_error(DSKFULL);
  325.     } while (listhex && i);
  326.     }
  327.     return;
  328. }
  329.  
  330. /*  Listing file close routine.  The symbol table is appended to the    */
  331. /*  listing in alphabetic order by symbol name, and the listing file is    */
  332. /*  closed.  If the disk fills up, a fatal error occurs.        */
  333.  
  334. static int col = 0;
  335.  
  336. void lclose()
  337. {
  338.     void fatal_error(), list_sym();
  339.  
  340.     if (list) {
  341.     if (sroot) {
  342.         list_sym(sroot);
  343.         if (col) fprintf(list,"\n");
  344.     }
  345.     fprintf(list,"\f");
  346.     if (ferror(list) || fclose(list) == EOF) fatal_error(DSKFULL);
  347.     }
  348.     return;
  349. }
  350.  
  351. static void list_sym(sp)
  352. SYMBOL *sp;
  353. {
  354.     void check_page();
  355.  
  356.     if (sp) {
  357.     list_sym(sp -> left);
  358.     fprintf(list,"%04x  %-10s",sp -> valu,sp -> sname);
  359.     if (col = ++col % SYMCOLS) fprintf(list,"    ");
  360.     else {
  361.         fprintf(list,"\n");
  362.         if (sp -> right) check_page();
  363.     }
  364.     list_sym(sp -> right);
  365.     }
  366.     return;
  367. }
  368.  
  369. static void check_page()
  370. {
  371.     if (pagelen && !--listleft) eject = TRUE;
  372.     if (eject) {
  373.     eject = FALSE;  listleft = pagelen;  fprintf(list,"\f");
  374.     if (title[0]) { listleft -= 2;  fprintf(list,"%s\n\n",title); }
  375.     }
  376.     return;
  377. }
  378.  
  379. /*  Buffer storage for hex output file.  This allows the hex file    */
  380. /*  output routines to do all of the required buffering and record    */
  381. /*  forming without the    main routine having to fool with it.        */
  382.  
  383. static FILE *hex = NULL;
  384. static unsigned cnt = 0;
  385. static unsigned addr = 0;
  386. static unsigned sum = 0;
  387. static unsigned buf[HEXSIZE];
  388.  
  389. /*  Hex file open routine.  If a hex file is already open, a warning    */
  390. /*  occurs.  If the hex file doesn't open correctly, a fatal error    */
  391. /*  occurs.  If no hex file is open, all calls to hputc(), hseek(), and    */
  392. /*  hclose() have no effect.                        */
  393.  
  394. void hopen(nam)
  395. char *nam;
  396. {
  397.     FILE *fopen();
  398.     void fatal_error(), warning();
  399.  
  400.     if (hex) warning(TWOHEX);
  401.     else if (!(hex = fopen(nam,"w"))) fatal_error(HEXOPEN);
  402.     return;
  403. }
  404.  
  405. /*  Hex file write routine.  The data byte is appended to the current    */
  406. /*  record.  If the record fills up, it gets written to disk.  If the    */
  407. /*  disk fills up, a fatal error occurs.                */
  408.  
  409. void hputc(c)
  410. unsigned c;
  411. {
  412.     void record();
  413.  
  414.     if (hex) {
  415.     buf[cnt++] = c;
  416.     if (cnt == HEXSIZE) record(0);
  417.     }
  418.     return;
  419. }
  420.  
  421. /*  Hex file address set routine.  The specified address becomes the    */
  422. /*  load address of the next record.  If a record is currently open,    */
  423. /*  it gets written to disk.  If the disk fills up, a fatal error    */
  424. /*  occurs.                                */
  425.  
  426. void hseek(a)
  427. unsigned a;
  428. {
  429.     void record();
  430.  
  431.     if (hex) {
  432.     if (cnt) record(0);
  433.     addr = a;
  434.     }
  435.     return;
  436. }
  437.  
  438. /*  Hex file close routine.  Any open record is written to disk, the    */
  439. /*  EOF record is added, and file is closed.  If the disk fills up, a    */
  440. /*  fatal error occurs.                            */
  441.  
  442. void hclose()
  443. {
  444.     void fatal_error(), record();
  445.  
  446.     if (hex) {
  447.     if (cnt) record(0);
  448.     record(1);
  449.     if (fclose(hex) == EOF) fatal_error(DSKFULL);
  450.     }
  451.     return;
  452. }
  453.  
  454. static void record(typ)
  455. unsigned typ;
  456. {
  457.     SCRATCH unsigned i;
  458.     void fatal_error(), putb();
  459.  
  460.     putc(':',hex);  putb(cnt);  putb(high(addr));
  461.     putb(low(addr));  putb(typ);
  462.     for (i = 0; i < cnt; ++i) putb(buf[i]);
  463.     putb(low(-sum));  putc('\n',hex);
  464.  
  465.     addr += cnt;  cnt = 0;
  466.  
  467.     if (ferror(hex)) fatal_error(DSKFULL);
  468.     return;
  469. }
  470.  
  471. static void putb(b)
  472. unsigned b;
  473. {
  474.     static char digit[] = "0123456789ABCDEF";
  475.  
  476.     putc(digit[b >> 4],hex);  putc(digit[b & 0x0f],hex);
  477.     sum += b;  return;
  478. }
  479.  
  480. /*  Error handler routine.  If the current error code is non-blank,    */
  481. /*  the error code is filled in and the    number of lines with errors    */
  482. /*  is adjusted.                            */
  483.  
  484. void error(code)
  485. char code;
  486. {
  487.     if (errcode == ' ') { errcode = code;  ++errors; }
  488.     return;
  489. }
  490.  
  491. /*  Fatal error handler routine.  A message gets printed on the stderr    */
  492. /*  device, and the program bombs.                    */
  493.  
  494. void fatal_error(msg)
  495. char *msg;
  496. {
  497.     printf("Fatal Error -- %s\n",msg);
  498.     exit(-1);
  499. }
  500.  
  501. /*  Non-fatal error handler routine.  A message gets printed on the    */
  502. /*  stderr device, and the routine returns.                */
  503.  
  504. void warning(msg)
  505. char *msg;
  506. {
  507.     printf("Warning -- %s\n",msg);
  508.     return;
  509. }
  510.