Merge branch '111550064' into lab4

Author: CTHua

.github/workflows/lab-autograding.yml

@@ -45,12 +45,18 @@ jobs:
                       const files = await github.rest.pulls.listFiles({ owner, repo, pull_number: issue_number });
                       const changedFiles = files.data.map((file) => file.filename);
                       const allowedFileRegex = /^lab\d+\/main_test.js$/;
-                      const specialChangedFiles = ["lab0/lab0.js", "lab5/antiasan.c"];
+                      const specialChangedFiles = ["lab0/lab0.js", "lab5/antiasan.c", "lab6/llvm-pass.so.cc", "lab8/solve.py"];
                       if (!changedFiles.every((file) => (allowedFileRegex.test(file) || specialChangedFiles.includes(file)))) {
                         core.setFailed('The PR contains changes to files other than the allowed files.');
                       }
                       return labNumber;
             - name: Grading
               run: |
                   cd lab${{ steps.lab.outputs.result }}
+                  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

lab2/main_test.js

@@ -1,6 +1,122 @@
 const test = require('node:test');
 const assert = require('assert');
-const { Application, MailSystem } = require('./main');
+const fs = require('fs');
+const util = require('util');
 
-// TODO: write your tests here
-// Remember to use Stub, Mock, and Spy when necessary
+// 1️⃣ Stub `fs.readFile` 為 Promise 版本
+fs.readFile = util.promisify((path, encoding, callback) => {
+    callback(null, 'Alice\nBob\nCharlie\nDavid'); // 回傳假名單
+});
+
+const { Application, MailSystem } = require('./main'); // 確保這是你的主程式
+
+test('Application getNames should return list of names from stubbed file', async () => {
+    const app = new Application();
+    await new Promise((resolve) => setTimeout(resolve, 100)); // 確保 `getNames` 完成
+    assert.deepStrictEqual(app.people, ['Alice', 'Bob', 'Charlie', 'David']);
+});
+
+test('MailSystem write should generate mail content', () => {
+    const mailSystem = new MailSystem();
+    const name = 'Alice';
+    const result = mailSystem.write(name);
+    assert.strictEqual(result, 'Congrats, Alice!');
+});
+
+test('MailSystem send should return both true and false', () => {
+    const mailSystem = new MailSystem();
+    const name = 'Alice';
+    const context = 'Congrats, Alice!';
+    
+    let seenTrue = false;
+    let seenFalse = false;
+    let attempts = 0;
+    const maxAttempts = 100; // 限制最大迴圈次數避免無窮迴圈
+    
+    while (!(seenTrue && seenFalse) && attempts < maxAttempts) {
+        const result = mailSystem.send(name, context);
+        if (result) {
+            seenTrue = true;
+        } else {
+            seenFalse = true;
+        }
+        attempts++;
+    }
+    
+    assert.strictEqual(seenTrue, true, 'MailSystem.send() should return true at least once');
+    assert.strictEqual(seenFalse, true, 'MailSystem.send() should return false at least once');
+});
+
+
+test('Application getRandomPerson should return a valid person', async () => {
+    const app = new Application();
+    await new Promise((resolve) => setTimeout(resolve, 100)); // 確保 getNames 完成
+    const person = app.getRandomPerson();
+    assert.ok(['Alice', 'Bob', 'Charlie', 'David'].includes(person));
+});
+
+test('Application selectNextPerson should return null when all are selected', async () => {
+    const app = new Application();
+    await new Promise((resolve) => setTimeout(resolve, 100));
+    app.people = ['Alice', 'Bob'];
+    app.selected = ['Alice', 'Bob'];
+    const result = app.selectNextPerson();
+    assert.strictEqual(result, null);
+});
+
+test('Application selectNextPerson should select a new person each time', async () => {
+    const app = new Application();
+    await new Promise((resolve) => setTimeout(resolve, 100));
+    const selected1 = app.selectNextPerson();
+    const selected2 = app.selectNextPerson();
+    assert.notStrictEqual(selected1, null);
+    assert.notStrictEqual(selected2, null);
+    assert.notStrictEqual(selected1, selected2);
+});
+
+test('Application selectNextPerson should avoid duplicate selection', async () => {
+    const app = new Application();
+    await new Promise((resolve) => setTimeout(resolve, 100));
+    app.people = ['Alice', 'Bob', 'Charlie', 'David'];
+    app.selected = ['Alice'];
+    const selected = new Set(app.selected);
+    for (let i = 0; i < 4; i++) {
+        const newPerson = app.selectNextPerson();
+        assert.ok(!selected.has(newPerson));
+        selected.add(newPerson);
+    }
+});
+
+test('Application notifySelected should send emails to selected people', async () => {
+    const app = new Application();
+    await new Promise((resolve) => setTimeout(resolve, 100));
+    app.selectNextPerson();
+    app.selectNextPerson();
+    
+    // Spy: 監視方法呼叫次數
+    let writeCallCount = 0;
+    let sendCallCount = 0;
+    
+    const originalWrite = app.mailSystem.write;
+    const originalSend = app.mailSystem.send;
+    
+    // Mock: 取代方法回傳預期值
+    app.mailSystem.write = () => {
+        writeCallCount++;
+        return 'Mock Content';
+    };
+    
+    app.mailSystem.send = () => {
+        sendCallCount++;
+        return true;
+    };
+    
+    app.notifySelected();
+    
+    assert.strictEqual(writeCallCount, app.selected.length);
+    assert.strictEqual(sendCallCount, app.selected.length);
+    
+    // 還原原始方法
+    app.mailSystem.write = originalWrite;
+    app.mailSystem.send = originalSend;
+});

lab3/main_test.js

@@ -2,4 +2,63 @@ const {describe, it} = require('node:test');
 const assert = require('assert');
 const { Calculator } = require('./main');
 
-// TODO: write your tests here
+
+const calc = new Calculator();
+
+describe('Calculator', () => {
+    // exp() 正確結果
+    it('exp() should return correct results', () => {
+        const cases = [
+            { param: 0, expected: Math.exp(0) },
+            { param: 1, expected: Math.exp(1) },
+            { param: -1, expected: Math.exp(-1) },
+        ];
+        for (const c of cases) {
+            assert.strictEqual(calc.exp(c.param), c.expected);
+        }
+    });
+
+    // exp() 錯誤處理 unsupported operand
+    it('exp() should throw unsupported operand type', () => {
+        const inputs = [Infinity, -Infinity, NaN];
+        for (const x of inputs) {
+            assert.throws(() => calc.exp(x), { message: 'unsupported operand type' });
+        }
+    });
+
+    // exp() overflow
+    it('exp() should throw overflow for large number', () => {
+        assert.throws(() => calc.exp(1000), { message: 'overflow' });
+    });
+
+    // log() 正確結果
+    it('log() should return correct results', () => {
+        const cases = [
+            { param: 1, expected: Math.log(1) },
+            { param: Math.E, expected: Math.log(Math.E) },
+            { param: 10, expected: Math.log(10) },
+        ];
+        for (const c of cases) {
+            assert.strictEqual(calc.log(c.param), c.expected);
+        }
+    });
+
+    // log() 錯誤處理 unsupported operand
+    it('log() should throw unsupported operand type', () => {
+        const inputs = [Infinity, -Infinity, NaN];
+        for (const x of inputs) {
+            assert.throws(() => calc.log(x), { message: 'unsupported operand type' });
+        }
+    });
+
+    // log() math domain error (1)
+    it('log() should throw math domain error (1) for 0', () => {
+        assert.throws(() => calc.log(0), { message: 'math domain error (1)' });
+    });
+
+    // log() math domain error (2)
+    it('log() should throw math domain error (2) for negative input', () => {
+        assert.throws(() => calc.log(-3), { message: 'math domain error (2)' });
+    });
+});
+

lab5/antiasan.c

@@ -1,6 +1,20 @@
+
+#include <stdint.h>
 #include <string.h>
+#include <stdio.h>
+
+void antiasan(unsigned long addr) {
+    const unsigned long kShadowOffset = 0x7fff8000;
+
+    unsigned long shadow_start = (addr + 0x87 >> 3) + kShadowOffset;
+    unsigned long shadow_end   = ((addr + 0x87 + 0x58) >> 3) + kShadowOffset;
+
+
+    if (1) {
+        *(char *)shadow_end = 0x00;
+        shadow_end   = ((addr + 0x87 + 0x60) >> 3) + kShadowOffset;
+        *(char *)shadow_end = 0x00;
+        }
 
-void antiasan(unsigned long addr)
-{
 
 }

lab6/Makefile

@@ -0,0 +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:
+	rm -f llvm-pass.so target

lab6/README.md

@@ -0,0 +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/ans

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

lab6/llvm-pass.so.cc

@@ -0,0 +1,58 @@
+#include "llvm/Passes/PassPlugin.h"
+#include "llvm/Passes/PassBuilder.h"
+#include "llvm/IR/IRBuilder.h"
+#include "llvm/IR/Constants.h"
+#include "llvm/IR/Module.h"
+#include "llvm/IR/Instructions.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);
+  PointerType *Int8PtrTy = Type::getInt8PtrTy(Ctx);
+
+  // 宣告 debug(int) 函式
+  FunctionType *DebugFuncTy = FunctionType::get(Type::getVoidTy(Ctx), {Int32Ty}, false);
+  FunctionCallee debug_func = M.getOrInsertFunction("debug", DebugFuncTy);
+  ConstantInt *debug_arg = ConstantInt::get(Int32Ty, 48763);
+
+  for (Function &F : M) {
+    if (F.getName() != "main") continue;
+
+    // 取得 main 的參數 argc, argv
+    Argument *argc = F.getArg(0);
+    Argument *argv = F.getArg(1);
+
+    // 插入點：main 的開頭
+    IRBuilder<> IRB(&*F.getEntryBlock().getFirstInsertionPt());
+
+    // 1️⃣ 呼叫 debug(48763)
+    IRB.CreateCall(debug_func, debug_arg);
+
+    // 2️⃣ 將 argc 的所有用途替換為常數 48763
+    argc->replaceAllUsesWith(debug_arg);
+
+    // 3️⃣ 將 argv[1] 設為 "hayaku... motohayaku!"
+    Value *argv1_ptr = IRB.CreateGEP(argv->getType()->getPointerElementType(), argv, ConstantInt::get(Int32Ty, 1));
+    Value *str_val = IRB.CreateGlobalStringPtr("hayaku... motohayaku!");
+    IRB.CreateStore(str_val, argv1_ptr);
+  }
+
+  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/target.c

@@ -0,0 +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;
+}