fix: lab8 no angr

Author: YingMuo

.github/workflows/lab-autograding.yml

@@ -56,4 +56,7 @@ jobs:
                   if [ ${{ steps.lab.outputs.result }} -eq 6 ]; then
                     sudo apt install -y llvm-14
                   fi
+                  if [ ${{ steps.lab.outputs.result }} -eq 8 ]; then
+                    python3 -m pip install angr
+                  fi
                   ./validate.sh

lab6/llvm-pass.so.cc

@@ -1,73 +1,34 @@
-#include "llvm/IR/IRBuilder.h"
-#include "llvm/IR/Instructions.h"
-#include "llvm/IR/Module.h"
-#include "llvm/Passes/PassBuilder.h"
 #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) {
-    LLVMContext &Ctx = M.getContext();
-    IntegerType *Int32Ty = Type::getInt32Ty(Ctx);
-    PointerType *i8PtrTy = Type::getInt8PtrTy(Ctx);
-
-    // debug(int)
-    FunctionCallee debugFunc = M.getOrInsertFunction("debug", FunctionType::get(Type::getVoidTy(Ctx), {Int32Ty}, false));
-    ConstantInt *debugArg = ConstantInt::get(Int32Ty, 48763);
-
-    Function *mainFunc = M.getFunction("main");
-    if (!mainFunc) return PreservedAnalyses::none();
-
-    IRBuilder<> Builder(&*mainFunc->getEntryBlock().getFirstInsertionPt());
-
-    // call debug(48763)
-    Builder.CreateCall(debugFunc, {debugArg});
-
-    // replace argc (arg0) to 48763
-    Argument *argcArg = nullptr, *argvArg = nullptr;
-    auto it = mainFunc->arg_begin();
-    argcArg = it++;
-    argvArg = it;
-
-    // alloc argc, store 48763
-    AllocaInst *argcAlloca = Builder.CreateAlloca(Int32Ty, nullptr, "argcVar");
-    Builder.CreateStore(debugArg, argcAlloca);
-
-    // replace all uses of argcArg (except alloca) with loaded version
-    argcArg->replaceAllUsesWith(Builder.CreateLoad(Int32Ty, argcAlloca));
-
-    // insert global string "hayaku... motohayaku!"
-    std::string msg = "hayaku... motohayaku!";
-    Constant *strConstant = ConstantDataArray::getString(Ctx, msg, true);
-    GlobalVariable *strVar = new GlobalVariable(
-        M, strConstant->getType(), true,
-        GlobalValue::PrivateLinkage, strConstant, "msg");
+  PreservedAnalyses run(Module &M, ModuleAnalysisManager &MAM);
+};
 
-    // get i8* to string
-    Value *strPtr = Builder.CreateGEP(
-        strConstant->getType(),
-        strVar,
-        {ConstantInt::get(Int32Ty, 0), ConstantInt::get(Int32Ty, 0)}
-    );
+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);
 
-    // argv[1] = strPtr
-    Value *argv1Ptr = Builder.CreateGEP(
-        i8PtrTy, argvArg,
-        ConstantInt::get(Int32Ty, 1)
-    );
-    Builder.CreateStore(strPtr, argv1Ptr);
+  for (auto &F : M) {
+    errs() << "func: " << F.getName() << "\n";
 
-    return PreservedAnalyses::none();
   }
-};
+  return PreservedAnalyses::none();
+}
 
-extern "C" ::llvm::PassPluginLibraryInfo LLVM_ATTRIBUTE_WEAK llvmGetPassPluginInfo() {
+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());
-                });
-          }};
+    [](PassBuilder &PB) {
+      PB.registerOptimizerLastEPCallback(
+        [](ModulePassManager &MPM, OptimizationLevel OL) {
+          MPM.addPass(LLVMPass());
+        });
+    }};
 }
+