lab6

Author: Your Name

lab6/Makefile

@@ -1,14 +1,14 @@
-all: target
-
-llvm-pass.so: llvm-pass.so.cc
-	clang-14 `llvm-config-14 --cxxflags` -shared -fPIC $< -o $@
-
-target: target.c llvm-pass.so
-	clang-14 `llvm-config-14 --cflags` -fexperimental-new-pass-manager \
-	 -fpass-plugin=./llvm-pass.so $< -o $@
-
-run:
-	./target 1
-
-clean:
+all: target
+
+llvm-pass.so: llvm-pass.so.cc
+	clang-14 `llvm-config-14 --cxxflags` -shared -fPIC $< -o $@
+
+target: target.c llvm-pass.so
+	clang-14 `llvm-config-14 --cflags` -fexperimental-new-pass-manager \
+	 -fpass-plugin=./llvm-pass.so $< -o $@
+
+run:
+	./target 1
+
+clean:
 	rm -f llvm-pass.so target

lab6/README.md

@@ -1,29 +1,29 @@
-# Lab6
-
-## Introduction
-
-In this lab, you will write a llvm pass to instrument some codes to `target.c` in `llvm-pass.so.cc`.
-
-## Preparation (Important!!!)
-
-1. Sync fork your branch (e.g., `SQLab:311XXXXXX`)
-2. `git checkout -b lab6` (**NOT** your student ID !!!)
-
-## Requirement
-
-Write a llvm pass to instrument some codes to `target.c` in `llvm-pass.so.cc` and satisfy following requirements.
-1. (40%) Invoke debug function with the first argument is 48763 in main function.
-2. (30%) Overwrite argv[1] to "hayaku... motohayaku!" before checking.
-3. (30%) Overwrite argc to 48763 before checking.
-You can run `validate.sh` in your local to test if you satisfy the requirements.
-
-Please note that you must not alter files other than `llvm-pass.so.cc`. You will get 0 points if
-
-1. you modify other files to achieve requirements.
-2. you can't pass all CI on your PR.
-
-## Submission
-
-You need to open a pull request to your branch (e.g. 311XXXXXX, your student number) and contain the code that satisfies the abovementioned requirements.
-
-Moreover, please submit the URL of your PR to E3. Your submission will only be accepted when you present at both places.
+# Lab6
+
+## Introduction
+
+In this lab, you will write a llvm pass to instrument some codes to `target.c` in `llvm-pass.so.cc`.
+
+## Preparation (Important!!!)
+
+1. Sync fork your branch (e.g., `SQLab:311XXXXXX`)
+2. `git checkout -b lab6` (**NOT** your student ID !!!)
+
+## Requirement
+
+Write a llvm pass to instrument some codes to `target.c` in `llvm-pass.so.cc` and satisfy following requirements.
+1. (40%) Invoke debug function with the first argument is 48763 in main function.
+2. (30%) Overwrite argv[1] to "hayaku... motohayaku!" before checking.
+3. (30%) Overwrite argc to 48763 before checking.
+You can run `validate.sh` in your local to test if you satisfy the requirements.
+
+Please note that you must not alter files other than `llvm-pass.so.cc`. You will get 0 points if
+
+1. you modify other files to achieve requirements.
+2. you can't pass all CI on your PR.
+
+## Submission
+
+You need to open a pull request to your branch (e.g. 311XXXXXX, your student number) and contain the code that satisfies the abovementioned requirements.
+
+Moreover, please submit the URL of your PR to E3. Your submission will only be accepted when you present at both places.

lab6/ans

@@ -1,6 +1,6 @@
-./target 1
-deubg mode
-argc = 48763
-argv[1] = hayaku... motohayaku!
-Your argc is so hayaku!
-Suta... basuto... sutorimu!
+./target 1
+deubg mode
+argc = 48763
+argv[1] = hayaku... motohayaku!
+Your argc is so hayaku!
+Suta... basuto... sutorimu!

lab6/llvm-pass.so

@@ -1,34 +1,51 @@
-#include "llvm/Passes/PassPlugin.h"
-#include "llvm/Passes/PassBuilder.h"
-#include "llvm/IR/IRBuilder.h"
-
-using namespace llvm;
-
-struct LLVMPass : public PassInfoMixin<LLVMPass> {
-  PreservedAnalyses run(Module &M, ModuleAnalysisManager &MAM);
-};
-
-PreservedAnalyses LLVMPass::run(Module &M, ModuleAnalysisManager &MAM) {
-  LLVMContext &Ctx = M.getContext();
-  IntegerType *Int32Ty = IntegerType::getInt32Ty(Ctx);
-  FunctionCallee debug_func = M.getOrInsertFunction("debug", Int32Ty);
-  ConstantInt *debug_arg = ConstantInt::get(Int32Ty, 48763);
-
-  for (auto &F : M) {
-    errs() << "func: " << F.getName() << "\n";
-
-  }
-  return PreservedAnalyses::none();
-}
-
-extern "C" ::llvm::PassPluginLibraryInfo LLVM_ATTRIBUTE_WEAK
-llvmGetPassPluginInfo() {
-  return {LLVM_PLUGIN_API_VERSION, "LLVMPass", "1.0",
-    [](PassBuilder &PB) {
-      PB.registerOptimizerLastEPCallback(
-        [](ModulePassManager &MPM, OptimizationLevel OL) {
-          MPM.addPass(LLVMPass());
-        });
-    }};
-}
-
+#include "llvm/Passes/PassPlugin.h"
+#include "llvm/Passes/PassBuilder.h"
+#include "llvm/IR/IRBuilder.h"
+
+using namespace llvm;
+
+struct LLVMPass : public PassInfoMixin<LLVMPass> {
+  PreservedAnalyses run(Module &M, ModuleAnalysisManager &MAM);
+};
+
+PreservedAnalyses LLVMPass::run(Module &M, ModuleAnalysisManager &MAM) {
+  LLVMContext &Ctx = M.getContext();
+  IntegerType *Int32Ty = IntegerType::getInt32Ty(Ctx);
+  Type *Int8PtrTy = Type::getInt8PtrTy(Ctx); ///add
+  FunctionCallee debug_func = M.getOrInsertFunction("debug", Int32Ty);
+  ConstantInt *debug_arg = ConstantInt::get(Int32Ty, 48763);
+
+  for (auto &F : M) {
+    errs() << "func: " << F.getName() << "\n";
+    if (F.getName() == "main") {
+      IRBuilder<> Builder(&*F.getEntryBlock().getFirstInsertionPt());
+
+      Builder.CreateCall(debug_func, debug_arg);
+
+      Argument *ArgcArg = F.getArg(0);
+      ArgcArg->replaceAllUsesWith(debug_arg); 
+
+      Argument *ArgvArg = F.getArg(1); 
+      Value *GEP = Builder.CreateInBoundsGEP(
+          Int8PtrTy,
+          ArgvArg,
+          ConstantInt::get(Int32Ty, 1)
+      );
+      Value *StrVal = Builder.CreateGlobalStringPtr("hayaku... motohayaku!");
+      Builder.CreateStore(StrVal, GEP);
+    }
+  }
+  return PreservedAnalyses::none();
+}
+
+extern "C" ::llvm::PassPluginLibraryInfo LLVM_ATTRIBUTE_WEAK
+llvmGetPassPluginInfo() {
+  return {LLVM_PLUGIN_API_VERSION, "LLVMPass", "1.0",
+    [](PassBuilder &PB) {
+      PB.registerOptimizerLastEPCallback(
+        [](ModulePassManager &MPM, OptimizationLevel OL) {
+          MPM.addPass(LLVMPass());
+        });
+    }};
+}
+

lab6/llvm-pass.so.cc

@@ -1,29 +1,29 @@
-#include <stdio.h>
-#include <stdlib.h>
-#include <string.h>
-
-void debug(int id)
-{
-    if (id == 48763)
-        printf("deubg mode\n");
-    else
-        printf("bad id!\n");
-}
-
-int main(int argc, char **argv)
-{
-    printf("argc = %d\n", argc);
-    if (argc >= 2)
-        printf("argv[1] = %s\n", argv[1]);
-    else
-        return 0;
-    if (argc == 48763)
-        printf("Your argc is so hayaku!\n");
-    else
-        printf("Your argc need to be modohayaku!\n");
-    if (strcmp(argv[1], "hayaku... motohayaku!") == 0)
-        printf("Suta... basuto... sutorimu!\n");
-    else
-        printf("You dead\n");
-    return 0;
-}
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+
+void debug(int id)
+{
+    if (id == 48763)
+        printf("deubg mode\n");
+    else
+        printf("bad id!\n");
+}
+
+int main(int argc, char **argv)
+{
+    printf("argc = %d\n", argc);
+    if (argc >= 2)
+        printf("argv[1] = %s\n", argv[1]);
+    else
+        return 0;
+    if (argc == 48763)
+        printf("Your argc is so hayaku!\n");
+    else
+        printf("Your argc need to be modohayaku!\n");
+    if (strcmp(argv[1], "hayaku... motohayaku!") == 0)
+        printf("Suta... basuto... sutorimu!\n");
+    else
+        printf("You dead\n");
+    return 0;
+}