From db8d9ddcbb9741570a2724c227c824f5eda13ad1 Mon Sep 17 00:00:00 2001
From: Patrick Walton <pcwalton@mimiga.net>
Date: Tue, 3 Jul 2012 12:20:15 -0700
Subject: [PATCH] rustc: Record the main function in the session in resolve3
---
src/rustc/middle/resolve3.rs | 25 +++++++++++++++++++++++++
1 file changed, 25 insertions(+)
diff --git a/src/rustc/middle/resolve3.rs b/src/rustc/middle/resolve3.rs
index f229a5a58a3..ea2a56a0013 100644
--- a/src/rustc/middle/resolve3.rs
+++ b/src/rustc/middle/resolve3.rs
@@ -380,6 +380,17 @@ class Module {
}
}
+pure fn is_crate_root(module: @Module) -> bool {
+ alt module.def_id {
+ none => {
+ ret false;
+ }
+ some(def_id) => {
+ ret def_id.crate == 0 && def_id.node == 0;
+ }
+ }
+}
+
// XXX: This is a workaround due to is_none in the standard library mistakenly
// requiring a T:copy.
@@ -2802,6 +2813,20 @@ class Resolver {
}
item_fn(fn_decl, ty_params, block) {
+ // If this is the main function, we must record it in the
+ // session.
+ //
+ // For speed, we put the string comparison last in this chain
+ // of conditionals.
+
+ if !self.session.building_library &&
+ is_none(self.session.main_fn) &&
+ is_crate_root(self.current_module) &&
+ str::eq(*item.ident, "main") {
+
+ self.session.main_fn = some((item.id, item.span));
+ }
+
self.resolve_function(NormalRibKind,
some(@fn_decl),
HasTypeParameters(&ty_params,