From 4a45be0e2d6a89c8c3c7b24571b93522eb677edb Mon Sep 17 00:00:00 2001
From: Rui Ueyama <ruiu@cs.stanford.edu>
Date: Tue, 8 Sep 2020 18:50:40 +0900
Subject: [PATCH] Add -fpic and -fPIC options

---
 chibicc.h      |  1 +
 codegen.c      | 15 +++++++++++++++
 main.c         |  6 ++++++
 test/driver.sh |  6 ++++++
 4 files changed, 28 insertions(+)

diff --git a/chibicc.h b/chibicc.h
index ea9c10e..d2df6c8 100644
--- a/chibicc.h
+++ b/chibicc.h
@@ -434,5 +434,6 @@ void hashmap_test(void);
 //
 
 extern StringArray include_paths;
+extern bool opt_fpic;
 extern bool opt_fcommon;
 extern char *base_file;
diff --git a/codegen.c b/codegen.c
index c590755..d886b9e 100644
--- a/codegen.c
+++ b/codegen.c
@@ -72,6 +72,21 @@ static void gen_addr(Node *node) {
       return;
     }
 
+    if (opt_fpic) {
+      // Thread-local variable
+      if (node->var->is_tls) {
+        println("  data16 lea %s@tlsgd(%%rip), %%rdi", node->var->name);
+        println("  .value 0x6666");
+        println("  rex64");
+        println("  call __tls_get_addr@PLT");
+        return;
+      }
+
+      // Function or global variable
+      println("  mov %s@GOTPCREL(%%rip), %%rax", node->var->name);
+      return;
+    }
+
     // Thread-local variable
     if (node->var->is_tls) {
       println("  mov %%fs:0, %%rax");
diff --git a/main.c b/main.c
index 32f3b26..34f0b63 100644
--- a/main.c
+++ b/main.c
@@ -6,6 +6,7 @@ typedef enum {
 
 StringArray include_paths;
 bool opt_fcommon = true;
+bool opt_fpic;
 
 static FileType opt_x;
 static StringArray opt_include;
@@ -259,6 +260,11 @@ static void parse_args(int argc, char **argv) {
       continue;
     }
 
+    if (!strcmp(argv[i], "-fpic") || !strcmp(argv[i], "-fPIC")) {
+      opt_fpic = true;
+      continue;
+    }
+
     if (!strcmp(argv[i], "-cc1-input")) {
       base_file = argv[++i];
       continue;
diff --git a/test/driver.sh b/test/driver.sh
index 9d0eab9..93326c8 100755
--- a/test/driver.sh
+++ b/test/driver.sh
@@ -251,4 +251,10 @@ $chibicc -c -MD -MF $tmp/md-mf.d -I. $tmp/md2.c
 grep -q -z '^md2.o:.*md2\.c .*/out2\.h' $tmp/md-mf.d
 check -MD
 
+echo 'extern int bar; int foo() { return bar; }' | $chibicc -fPIC -xc -c -o $tmp/foo.o -
+cc -shared $tmp/foo.so $tmp/foo.o
+echo 'int foo(); int bar=3; int main() { foo(); }' > $tmp/main.c
+$chibicc -o $tmp/foo $tmp/main.c $tmp/foo.so
+check -fPIC
+
 echo OK
-- 
GitLab