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

lab4/main_test.js

@@ -1,46 +1,22 @@
 const puppeteer = require('puppeteer');
-const sleep = (ms) => new Promise(resolve => setTimeout(resolve, ms));
+
 (async () => {
     // Launch the browser and open a new blank page
-    const browser = await puppeteer.launch({ headless: false });
+    const browser = await puppeteer.launch();
     const page = await browser.newPage();
 
     // Navigate the page to a URL
     await page.goto('https://pptr.dev/');
 
-    await page.click('button.DocSearch');
-    await page.waitForSelector('input.DocSearch-Input');
-    await page.type('input.DocSearch-Input', 'andy popoo');
-    await new Promise(resolve => setTimeout(resolve, 1000)); 
-    await page.waitForSelector('section.DocSearch-Hits');
-
-    const sections = await page.$$('section.DocSearch-Hits');
+    // Hints:
+    // Click search button
+    // Type into search box
+    // Wait for search result
+    // Get the `Docs` result section
+    // Click on first result in `Docs` section
+    // Locate the title
+    // Print the title
 
-    for (const section of sections) {
-        const sourceDiv = await section.$('div.DocSearch-Hit-source');
-        
-        if (sourceDiv) {
-            const text = await sourceDiv.evaluate(el => el.innerText.trim());
-            if (text === 'ElementHandle') {
-                // Click on the first result in this section
-                const firstListItem = await section.$('#docsearch-list li');
-                if (firstListItem) {
-                    await firstListItem.click();
-                    await new Promise(resolve => setTimeout(resolve, 1000));
-                    const title = await page.evaluate(() => {
-                        return document.querySelector("div.theme-doc-markdown > header > h1").innerText;
-                    });
-                    console.log(title);
-                } else {
-                    console.log('No list item found in this section');
-                }
-            }
-            
-        } else {
-            console.log('No ElementHandle found in this section');
-        }
-    }
-    
+    // Close the browser
     await browser.close();
-})();
-
+})();

lab5/antiasan.c

@@ -2,14 +2,5 @@
 
 void antiasan(unsigned long addr)
 {
-    unsigned long shadow_base = 0x7fff8000;
-    unsigned char *shadow = (unsigned char *)((addr >> 3) + shadow_base);
 
-    for (int i = 0; i < 64; i++) {
-        shadow[i] = 0;
-    }
-
-    for (int i = 1; i <= 8; i++) {
-        shadow[-i] = 0;
-    }
 }

lab6/llvm-pass.so.cc

@@ -11,46 +11,13 @@ struct LLVMPass : public PassInfoMixin<LLVMPass> {
 PreservedAnalyses LLVMPass::run(Module &M, ModuleAnalysisManager &MAM) {
   LLVMContext &Ctx = M.getContext();
   IntegerType *Int32Ty = IntegerType::getInt32Ty(Ctx);
-  PointerType *charPtrTy = Type::getInt8PtrTy(Ctx);
+  FunctionCallee debug_func = M.getOrInsertFunction("debug", Int32Ty);
+  ConstantInt *debug_arg = ConstantInt::get(Int32Ty, 48763);
 
-  // Find main()
-  Function *mainFunc = nullptr;
-  for (Function &F : M) {
-    if (F.getName() == "main") {
-      mainFunc = &F;
-      break;
-    }
-  }
-  if (!mainFunc) return PreservedAnalyses::none();
-
-  IRBuilder<> builder(&*mainFunc->getEntryBlock().getFirstInsertionPt());
-
-  FunctionType *debugType = FunctionType::get(Type::getVoidTy(Ctx), {Int32Ty}, false);
-  FunctionCallee debugFunc = M.getOrInsertFunction("debug", debugType);
-  Value *debugArg = ConstantInt::get(Int32Ty, 48763);
-  builder.CreateCall(debugFunc, {debugArg});
-
-  Argument *argcArg = mainFunc->getArg(0); // int argc
-  AllocaInst *argcAlloca = builder.CreateAlloca(Int32Ty, nullptr, "argc.alloca");
-  builder.CreateStore(ConstantInt::get(Int32Ty, 48763), argcAlloca);
+  for (auto &F : M) {
+    errs() << "func: " << F.getName() << "\n";
 
-  for (auto &BB : *mainFunc) {
-    for (auto &I : BB) {
-      for (unsigned i = 0; i < I.getNumOperands(); ++i) {
-        if (I.getOperand(i) == argcArg) {
-          IRBuilder<> tmpBuilder(&I);
-          LoadInst *argcVal = tmpBuilder.CreateLoad(Int32Ty, argcAlloca);
-          I.setOperand(i, argcVal);
-        }
-      }
-    }
   }
-
-  Argument *argvArg = mainFunc->getArg(1); // char **argv
-  Value *strPtr = builder.CreateGlobalStringPtr("hayaku... motohayaku!", "str");
-  Value *argv1Ptr = builder.CreateGEP(charPtrTy, argvArg, ConstantInt::get(Int32Ty, 1));
-  builder.CreateStore(strPtr, argv1Ptr);
-
   return PreservedAnalyses::none();
 }
 
@@ -64,3 +31,4 @@ llvmGetPassPluginInfo() {
         });
     }};
 }
+