pasado desde gitlab

Author: mszsorondo

Makefile

@@ -0,0 +1,34 @@
+CFLAGS=-Wall -Wextra -pedantic -std=c99 -Wconversion -Og -ggdb -no-pie
+CC=gcc
+AS=nasm
+ASFLAGS=-f elf64 -g -FDWARF
+EJ1OBJS=ej1_asm.o lib_c.o ej1_c.o
+EJ2OBJS=ej2_asm.o lib_c.o ej2_c.o
+EJ3OBJS=ej3_asm.o lib_c.o ej3_c.o
+EJ4OBJS=ej4_asm.o lib_c.o ej4_c.o
+
+all: ej1 ej2 ej3 ej4
+
+format:
+	clang-format -style=Chromium -i *.c *.h
+
+%_asm.o: %.asm
+	$(AS) $(ASFLAGS) $^ -o $@
+
+%_c.o: %.c
+	$(CC) $(CFLAGS) $^ -c -o $@
+
+ej1: $(EJ1OBJS)
+	$(CC) $(CFLAGS) $^ -o $@
+
+ej2: $(EJ2OBJS)
+	$(CC) $(CFLAGS) $^ -o $@
+
+ej3: $(EJ3OBJS)
+	$(CC) $(CFLAGS) $^ -o $@
+
+ej4: $(EJ4OBJS)
+	$(CC) $(CFLAGS) $^ -o $@
+
+clean:
+	rm -f *.o ej1 ej2 ej3 ej4

abi_enforcer.mac

@@ -0,0 +1,57 @@
+; Version simple del abi_enforcer. Sin mensajes de error ni nada.
+; Uso: %include "abi_enforcer.mac" en cada archivo .asm, o agregar
+; -p "abi_enforcer.mac" al NASMFLAGS de cada Makefile.
+;
+; Features:
+;   * Pisa todos los registros que no se preservan en cada call y ret
+;     (rcx, rdx, rdi, rsi, r8, r9, r10, r11)
+;   * Pisa 50 valores en el stack para asegurarnos que nadie referencie
+;     a variables del stack luego de un call.
+;   * Chequea que la pila este alineada antes de cada call.
+;
+; No tiene mensajes de error lindos. El codigo simplemente va a crashear.
+; No chequea que los registros que hay que preservar se preserven.
+
+; Macro regcrash pisa todos los registros que no se preservan.
+%macro regcrash 0
+    mov rdi, 0x001122334455667788
+    mov rsi, 0x001122334455667788
+    mov rdx, 0x001122334455667788
+    mov rcx, 0x001122334455667788
+    mov r8,  0x001122334455667788
+    mov r9,  0x001122334455667788
+    mov r10, 0x001122334455667788
+    mov r11, 0x001122334455667788
+%endmacro
+
+; Macro stackcrash llena el stack de basura.
+%macro stackcrash 0
+    times 50 push 0x11bbccdd
+    add rsp, 50*8
+%endmacro
+
+; Macro stackcheck crashea si el stack no está alineado.
+%macro stackcheck 0
+  test rsp, 0xF
+  jz %%aligned
+  ud2
+%%aligned:
+%endmacro
+
+; Macro que reemplaza la instruccion call.
+; Antes de ejecutar el call verdadero, chequea que el stack este alineado
+; Luego de ejecutar el call, pone basura en todos los registros y el stack.
+%imacro call 1
+    stackcheck
+    call %1
+    regcrash
+    stackcrash
+%endmacro
+
+; Macro que reemplaza la instruccion ret.
+; Antes de hacer el ret, pone basura en todos los registros y el stack.
+%imacro ret 0
+    regcrash
+    stackcrash
+    ret
+%endmacro