From 38d2a0ddb9b1e74661dd05fae71e3a2d47e8de2c Mon Sep 17 00:00:00 2001
From: gingerBill <bill@gingerbill.org>
Date: Thu, 21 Sep 2023 11:13:26 +0100
Subject: [PATCH] Add `sanitize_address` attribute to all normal packages

---
 src/checker.cpp           | 2 +-
 src/docs_writer.cpp       | 6 ++----
 src/llvm_backend_proc.cpp | 7 ++++++-
 src/parser.hpp            | 1 +
 4 files changed, 10 insertions(+), 6 deletions(-)

diff --git a/src/checker.cpp b/src/checker.cpp
index 0be912df5..cda7d50a0 100644
--- a/src/checker.cpp
+++ b/src/checker.cpp
@@ -858,7 +858,7 @@ gb_internal AstPackage *create_builtin_package(char const *name) {
 	gbAllocator a = permanent_allocator();
 	AstPackage *pkg = gb_alloc_item(a, AstPackage);
 	pkg->name = make_string_c(name);
-	pkg->kind = Package_Normal;
+	pkg->kind = Package_Builtin;
 
 	pkg->scope = create_scope(nullptr, nullptr);
 	pkg->scope->flags |= ScopeFlag_Pkg | ScopeFlag_Global | ScopeFlag_Builtin;
diff --git a/src/docs_writer.cpp b/src/docs_writer.cpp
index 6b42d2e7a..6816ae8eb 100644
--- a/src/docs_writer.cpp
+++ b/src/docs_writer.cpp
@@ -1101,11 +1101,9 @@ gb_internal void odin_doc_write_docs(OdinDocWriter *w) {
 		case Package_Init:
 			pkg_flags |= OdinDocPkgFlag_Init;
 			break;
-		}
-		if (pkg->name == "builtin") {
-			pkg_flags |= OdinDocPkgFlag_Builtin;
-		} else if (pkg->name == "intrinsics") {
+		case Package_Builtin:
 			pkg_flags |= OdinDocPkgFlag_Builtin;
+			break;
 		}
 
 		OdinDocPkg doc_pkg = {};
diff --git a/src/llvm_backend_proc.cpp b/src/llvm_backend_proc.cpp
index b8f6f0f30..f86e008a9 100644
--- a/src/llvm_backend_proc.cpp
+++ b/src/llvm_backend_proc.cpp
@@ -152,7 +152,6 @@ gb_internal lbProcedure *lb_create_procedure(lbModule *m, Entity *entity, bool i
 		lb_add_attribute_to_proc(m, p->value, "noredzone");
 	}
 
-
 	switch (p->inlining) {
 	case ProcInlining_inline:
 		lb_add_attribute_to_proc(m, p->value, "alwaysinline");
@@ -318,6 +317,12 @@ gb_internal lbProcedure *lb_create_procedure(lbModule *m, Entity *entity, bool i
 		}
 	}
 
+	if (p->body && entity->pkg && (entity->pkg->kind == Package_Normal) || (entity->pkg->kind == Package_Init)) {
+		if (build_context.sanitizer_flags & SanitizerFlag_Address) {
+			lb_add_attribute_to_proc(m, p->value, "sanitize_address");
+		}
+	}
+
 	lbValue proc_value = {p->value, p->type};
 	lb_add_entity(m, entity,  proc_value);
 	lb_add_member(m, p->name, proc_value);
diff --git a/src/parser.hpp b/src/parser.hpp
index eace5ba86..dd7bd0928 100644
--- a/src/parser.hpp
+++ b/src/parser.hpp
@@ -60,6 +60,7 @@ enum PackageKind {
 	Package_Normal,
 	Package_Runtime,
 	Package_Init,
+	Package_Builtin,
 };
 
 struct ImportedFile {