2 // RMAC - Reboot's Macro Assembler for all Atari computers
3 // SYMBOL.C - Symbol Handling
4 // Copyright (C) 199x Landon Dyer, 2011-2020 Reboot and Friends
5 // RMAC derived from MADMAC v1.07 Written by Landon Dyer, 1986
6 // Source utilised with the kind permission of Landon Dyer
18 #define NBUCKETS 256 // Number of hash buckets (power of 2)
20 static SYM * symbolTable[NBUCKETS]; // User symbol-table header
21 int curenv; // Current enviroment number
22 static SYM * sorder; // * -> Symbols, in order of reference
23 static SYM * sordtail; // * -> Last symbol in sorder list
24 static SYM * sdecl; // * -> Symbols, in order of declaration
25 static SYM * sdecltail; // * -> Last symbol in sdecl list
26 static uint32_t currentUID; // Symbol UID tracking (done by NewSymbol())
27 uint32_t firstglobal; // Index of the first global symbol in an ELF object.
29 // Tags for marking symbol spaces:
35 static uint8_t tdb_text[8] = {
36 'a', 't', 'd', '!', 'b', SPACE, SPACE, SPACE
39 // Internal function prototypes
40 static uint16_t WriteLODSection(int, uint16_t);
44 // Initialize symbol table
46 void InitSymbolTable(void)
48 for(int i=0; i<NBUCKETS; i++) // Initialise symbol hash table
49 symbolTable[i] = NULL;
51 curenv = 1; // Init local symbol enviroment
52 sorder = NULL; // Init symbol-reference list
54 sdecl = NULL; // Init symbol-decl list
61 // Hash the ASCII name and enviroment number
63 int HashSymbol(uint8_t * name, int envno)
65 int sum = envno, k = 0;
75 return sum & (NBUCKETS - 1);
80 // Make a new symbol of type 'type' in enviroment 'envno'
82 SYM * NewSymbol(uint8_t * name, int type, int envno)
84 // Allocate the symbol
85 SYM * symbol = malloc(sizeof(SYM));
89 printf("NewSymbol: MALLOC ERROR (symbol=\"%s\")\n", name);
94 symbol->sname = strdup(name);
95 symbol->stype = (uint8_t)type;
96 symbol->senv = (uint16_t)envno;
97 // We don't set this as DEFINED, as it could be a forward reference!
99 // We don't set RISCSYM here as not every symbol first seen in a RISC
100 // section is a RISC symbol!
103 symbol->sorder = NULL;
104 symbol->uid = currentUID++;
106 // Install symbol in the symbol table
107 int hash = HashSymbol(name, envno);
108 symbol->snext = symbolTable[hash];
109 symbolTable[hash] = symbol;
111 // Append symbol to the symbol-order list
113 sorder = symbol; // Add first symbol
115 sordtail->sorder = symbol; // Or append to tail of list
123 // Look up the symbol name by its UID and return the pointer to the name.
124 // If it's not found, return NULL.
126 uint8_t * GetSymbolNameByUID(uint32_t uid)
128 //problem is with string lookup, that's why we're writing this
129 //so once this is written, we can put the uid in the token stream
131 // A much better approach to the symbol order list would be to make an
132 // array--that way you can do away with the UIDs and all the rest, and
133 // simply do an array lookup based on position. But meh, let's do this for
134 // now until we can rewrite things so they make sense.
135 SYM * symbol = sorder;
137 for(; symbol; symbol=symbol->sorder)
139 if (symbol->uid == uid)
140 return symbol->sname;
148 // Lookup the symbol 'name', of the specified type, with the specified
151 SYM * lookup(uint8_t * name, int type, int envno)
153 SYM * symbol = symbolTable[HashSymbol(name, envno)];
155 // Do linear-search for symbol in bucket
156 while (symbol != NULL)
158 if (symbol->stype == type // Type, envno and name must match
159 && symbol->senv == envno
160 && *name == *symbol->sname // Fast check for first character
161 && !strcmp(name, symbol->sname)) // More expensive check
164 symbol = symbol->snext;
167 // Return NULL or matching symbol
173 // Put symbol on "order-of-declaration" list of symbols
175 void AddToSymbolDeclarationList(SYM * symbol)
177 // Don't add if already on list, or it's an equated register/CC
178 if ((symbol->sattr & SDECLLIST)
179 || (symbol->sattre & (EQUATEDREG | UNDEF_EQUR | EQUATEDCC | UNDEF_CC)))
182 // Mark as "on .sdecl list"
183 symbol->sattr |= SDECLLIST;
186 sdecl = symbol; // First on decl-list
188 sdecltail->sdecl = symbol; // Add to end of list
190 // Fix up list's tail
191 symbol->sdecl = NULL;
197 // Make all referenced, undefined symbols global
199 void ForceUndefinedSymbolsGlobal(void)
203 DEBUG printf("~ForceUndefinedSymbolsGlobal()\n");
205 // Scan through all symbols; if a symbol is REFERENCED but not DEFINED,
206 // then make it global.
207 for(sy=sorder; sy!=NULL; sy=sy->sorder)
209 if (sy->stype == LABEL && sy->senv == 0
210 && ((sy->sattr & (REFERENCED | DEFINED)) == REFERENCED))
217 // Assign numbers to symbols that are to be exported or imported. The symbol
218 // number is put in 'senv'. Returns the number of symbols that will be in the
221 // N.B.: This is usually called twice; first time with NULL parameters and the
222 // second time with real ones. The first one is typically done to get a
223 // count of the # of symbols in the symbol table, and the second is to
224 // actually create it.
226 uint32_t sy_assign(uint8_t * buf, uint8_t *(* construct)())
230 // Done only on first pass...
233 // Append all symbols not appearing on the .sdecl list to the end of
235 for(SYM * sy=sorder; sy!=NULL; sy=sy->sorder)
236 AddToSymbolDeclarationList(sy);
239 // Run through all symbols (now on the .sdecl list) and assign numbers to
240 // them. We also pick which symbols should be global or not here.
241 for(SYM * sy=sdecl; sy!=NULL; sy=sy->sdecl)
243 // Export or import external references, and export COMMON blocks.
244 if ((sy->stype == LABEL)
245 && ((sy->sattr & (GLOBAL | DEFINED)) == (GLOBAL | DEFINED)
246 || (sy->sattr & (GLOBAL | REFERENCED)) == (GLOBAL | REFERENCED))
247 || (sy->sattr & COMMON))
252 buf = construct(buf, sy, 1);
254 // Export vanilla labels (but don't make them global). An exception is
255 // made for equates, which are not exported unless they are referenced.
256 else if (sy->stype == LABEL && lsym_flag
257 && (sy->sattr & (DEFINED | REFERENCED)) != 0
258 && (!as68_flag || *sy->sname != 'L'))
263 buf = construct(buf, sy, 0);
272 // Custom version of sy_assign for ELF .o files.
273 // The order that the symbols should be dumped is different.
274 // (globals must be explicitly at the end of the table)
276 // N.B.: It should be possible to merge this with sy_assign, as there's nothing
277 // really ELF specific in here, other than the "globals go at the end of
278 // the queue" thing, which doesn't break the others. :-P
279 uint32_t sy_assign_ELF(uint8_t * buf, uint8_t *(* construct)())
283 // if (construct == (uint8_t *(*)())constr_elfsymtab)
286 // Append all symbols not appearing on the .sdecl list to the end of
288 for(SYM * sy=sorder; sy!=NULL; sy=sy->sorder)
289 AddToSymbolDeclarationList(sy);
292 // Run through all symbols (now on the .sdecl list) and assign numbers to
293 // them. We also pick which symbols should be global or not here.
294 for(SYM * sy=sdecl; sy!=NULL; sy=sy->sdecl)
296 // Export vanilla labels (but don't make them global). An exception is
297 // made for equates, which are not exported unless they are referenced.
298 if (sy->stype == LABEL && lsym_flag
299 && (sy->sattr & (DEFINED | REFERENCED)) != 0
300 && (*sy->sname != '.')
301 && (sy->sattr & GLOBAL) == 0)
306 buf = construct(buf, sy, 0);
310 firstglobal = scount;
312 // For ELF object mode run through all symbols in reference order
313 // and export all global-referenced labels. Not sure if this is
314 // required but it's here nonetheless
316 for(SYM * sy=sdecl; sy!=NULL; sy=sy->sdecl)
318 if ((sy->stype == LABEL)
319 && ((sy->sattr & (GLOBAL | DEFINED)) == (GLOBAL | DEFINED)
320 || (sy->sattr & (GLOBAL | REFERENCED)) == (GLOBAL | REFERENCED))
321 || (sy->sattr & COMMON))
326 buf = construct(buf, sy, 1);
328 else if ((sy->sattr == (GLOBAL | REFERENCED)) && (buf != NULL))
330 buf = construct(buf, sy, 0);
340 // Helper function for dsp_lod_symbols
342 static uint16_t WriteLODSection(int section, uint16_t symbolCount)
344 for(SYM * sy=sdecl; sy!=NULL; sy=sy->sdecl)
346 // Export vanilla labels (but don't make them global). An exception is
347 // made for equates, which are not exported unless they are referenced.
348 if (sy->stype == LABEL && lsym_flag
349 && (sy->sattr & (DEFINED | REFERENCED)) != 0
350 && (*sy->sname != '.')
351 && (sy->sattr & GLOBAL) == 0
352 && (sy->sattr & (section)))
354 sy->senv = symbolCount++;
355 D_printf("%-19s I %.6" PRIX64 "\n", sy->sname, sy->svalue);
364 // Dump LOD style symbols into the passed in buffer
366 void DumpLODSymbols(void)
368 D_printf("_SYMBOL P\n");
369 uint16_t count = WriteLODSection(M56001P, 0);
371 D_printf("_SYMBOL X\n");
372 count = WriteLODSection(M56001X, count);
374 D_printf("_SYMBOL Y\n");
375 count = WriteLODSection(M56001Y, count);
377 D_printf("_SYMBOL L\n");
378 count = WriteLODSection(M56001L, count);
380 // TODO: I've seen _SYMBOL N in there but no idea what symbols it needs...
381 //D_printf("_SYMBOL N\n");
382 //WriteLODSection(M56001?, count);
387 // Convert string to uppercase
389 void ToUppercase(uint8_t * s)
393 if (*s >= 'a' && *s <= 'z')
400 // Generate symbol table for listing file
417 int colhei = pagelen - 5;
419 // Allocate storage for list headers and partition all labels. Throw away
420 // macros and macro arguments.
421 SYM ** sy = (SYM **)malloc(128 * sizeof(SYM **));
426 for(i=0; i<NBUCKETS; i++)
428 for(p=symbolTable[i]; p!=NULL; p=k)
435 if ((p->stype != LABEL) || (p->sattre & UNDEF_EQUR))
438 for(q=sy[j]; q!=NULL; q=q->snext)
440 if (strcmp(p->sname, q->sname) < 0)
448 // Insert at front of list
454 // Insert in middle or append to list
461 // Link all symbols onto one list again
466 if ((r = sy[i]) != NULL)
473 while (q->snext != NULL)
482 strcpy(subttl, "Symbol Table");
490 for(j=0; j<colhei; j++)
499 for(i=0; i<colhei; i++)
503 if (colptr[0] == NULL)
508 if ((q = colptr[j]) == NULL)
511 colptr[j] = q->snext;
516 // x external reference
518 // space nothing special
524 else if ((w & (DEFINED | GLOBAL)) == GLOBAL)
529 c1 = tdb_text[w & TDB];
532 strcpy(ln2, "external");
535 sprintf(ln2, "%016lX", q->svalue);
539 sprintf(ln1, " %16s %s %c%c%c", q->sname, ln2, (ww & EQUATEDREG) ? 'e' : SPACE, c1, c);