--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/lwasm/pass2.c	Wed Jan 19 22:27:17 2011 -0700
@@ -0,0 +1,94 @@
+/*
+pass2.c
+
+Copyright © 2010 William Astle
+
+This file is part of LWTOOLS.
+
+LWTOOLS is free software: you can redistribute it and/or modify it under the
+terms of the GNU General Public License as published by the Free Software
+Foundation, either version 3 of the License, or (at your option) any later
+version.
+
+This program is distributed in the hope that it will be useful, but WITHOUT
+ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
+FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for
+more details.
+
+You should have received a copy of the GNU General Public License along with
+this program. If not, see <http://www.gnu.org/licenses/>.
+*/
+
+#include <stdio.h>
+#include <string.h>
+
+#include <lw_alloc.h>
+#include <lw_string.h>
+
+#include "lwasm.h"
+#include "instab.h"
+
+/*
+pass 2: deal with undefined symbols and do a simplification pass
+on all the expressions. Handle PRAGMA_IMPORTUNDEFEXPORT
+
+*/
+void do_pass2(asmstate_t *as)
+{
+	line_t *cl;
+	exportlist_t *ex;
+	struct symtabe *s;
+	importlist_t *im;
+	struct line_expr_s *le;
+
+	// verify the export list
+	if (as -> output_format == OUTPUT_OBJ)
+	{	
+		for (ex = as -> exportlist; ex; ex = ex -> next)
+		{
+			s = lookup_symbol(as, NULL, ex -> symbol);
+			if (!s)
+			{
+				if (CURPRAGMA(ex -> line, PRAGMA_IMPORTUNDEFEXPORT))
+				{
+					for (im = as -> importlist; im; im = im -> next)
+					{
+						if (!strcmp(ex -> symbol, im -> symbol))
+							break;
+					}
+					if (!im)
+					{
+						im = lw_alloc(sizeof(importlist_t));
+						im -> symbol = lw_strdup(ex -> symbol);
+						im -> next = as -> importlist;
+						as -> importlist = im;
+					}
+				}
+				else
+				{
+					// undefined export - register error
+					lwasm_register_error(as, ex -> line, "Undefined exported symbol");
+				}
+			}
+			ex -> se = s;
+		}
+		if (as -> errorcount > 0)
+			return;
+	}
+
+	// we want to throw errors on undefined symbols here
+	as -> badsymerr = 1;
+	
+	// now do some reductions on expressions
+	for (cl = as -> line_head; cl; cl = cl -> next)
+	{
+		as -> cl = cl;
+		
+		// simplify address
+		lwasm_reduce_expr(as, cl -> addr);
+		
+		// simplify each expression
+		for (le = cl -> exprs; le; le = le -> next)
+			lwasm_reduce_expr(as, le -> expr);
+	}	
+}