shithub: scc

ref: 941a776c59a79b6d64d07d09b38f3e9b00feeee3
dir: /symbol.c/

#include <stdlib.h>
#include <string.h>

#include "cc.h"
#include "symbol.h"

#define NR_SYM_HASH 32

unsigned char curctx;

static struct symbol *htab[NR_SYM_HASH];
static struct symbol *head;


static inline unsigned char hash(register const char *s)
{
	register unsigned char h, ch;

	for (h = 0; ch = *s++; h += ch)
		/* nothing */;
	return h & NR_SYM_HASH - 1;
}

void new_ctx(void)
{
	++curctx;
}

void del_ctx(void)
{
	register struct symbol *sym, *aux;
	static char *s;

	--curctx;
	for (sym = head; sym; sym = aux) {
		if (sym->ctx <= curctx)
			break;
		if ((s = sym->name) != NULL) {
			htab[hash(s)] = sym->hash;
			free(s);
		}
		aux = sym->next;
		free(sym);
		/* TODO: unlink type  */
	}
}

struct symbol *install(const char *s)
{
	register struct symbol *sym;
	register unsigned char key;

	sym = xmalloc(sizeof(*sym));
	sym->ctx = curctx;
	sym->next = head;
	head = sym;

	if (s) {
		sym->name = xstrdup(s);
		key = hash(s);
		sym->hash = htab[key];
		htab[key] = sym;
		sym->ns = NS_IDEN;
	} else {
		sym->hash = NULL;
		sym->name = NULL;
	}
	return sym;
}

struct symbol *lookup(const char *s)
{
	register struct symbol *sym;
	static  unsigned char l;

	l = strlen(s);
	for (sym = htab[hash(s)]; sym; sym = sym->hash) {
		if (!memcmp(sym->name, s, l))
			break;
	}
	return sym;
}