initial

Makefile

@@ -0,0 +1,54 @@
+# **************************************************************************** #
+#                                                                              #
+#                                                   :::  o_   ::::::    :::    #
+#    Makefile                                      :+:  /    :+::+:    :+:     #
+#                                                 +:+   >   +:++:+    +:+      #
+#    By: whaffman <whaffman@student.codam.nl>    +#+  +:+  +#++#++:++#++       #
+#                                               +#+ +#+#+ +#++#+    +#+ \o/    #
+#    Created: 2024/10/15 11:48:46 by whaffman   #+#+# #+#+# #+#    #+#   |     #
+#    Updated: 2024/11/07 15:28:08 by whaffman   ###   ###  ###    ###   / \    #
+#                                                                              #
+# **************************************************************************** #
+
+NAME = funfail.so
+
+SRC_PATH = src
+INC_PATH = inc
+OBJ_PATH = obj
+
+VPATH = src:src/utils
+SOURCES = $(shell basename -a $(shell find $(SRC_PATH) -type f -name "*.c"))
+OBJECTS = $(addprefix $(OBJ_PATH)/, $(SOURCES:.c=.o))
+DEPENDS = ${OBJECTS:.o=.d}
+
+CC = cc
+RM = rm -rf
+
+INCLUDES = -I./$(INC_PATH)
+CFLAGS = -Wall -Wextra -Werror -MMD
+LDLIBS = -fPIC -shared -ldl
+
+
+
+all: $(NAME)
+
+$(NAME): $(OBJECTS)
+	$(CC) $(CFLAGS) $(OBJECTS) $(LDLIBS) -o $(NAME)
+
+-include ${DEPENDS}
+
+$(OBJ_PATH):
+	mkdir -p $@
+
+$(OBJ_PATH)/%.o: %.c | $(OBJ_PATH)
+	$(CC) $(CFLAGS) $(INCLUDES) -c $< -o $@
+
+clean:
+	$(RM) $(OBJECTS) $(OBJ_PATH)
+
+fclean: clean
+	$(RM) $(NAME)
+
+re: fclean all
+
+.PHONY: all clean fclean re

obj/calloc.d

@@ -0,0 +1 @@
+obj/calloc.o: src/calloc.c

obj/malloc.d

@@ -0,0 +1 @@
+obj/malloc.o: src/malloc.c

obj/read.d

@@ -0,0 +1 @@
+obj/read.o: src/read.c

obj/write.d

@@ -0,0 +1 @@
+obj/write.o: src/write.c

src/calloc.c

@@ -0,0 +1,25 @@
+#include <dlfcn.h>
+#include <errno.h>
+#include <stdlib.h>
+
+typedef void *(*f)(size_t __nmemb, size_t __size);
+
+void *calloc(size_t __nmemb, size_t __size)
+{
+	static  f og_calloc = NULL;
+	static int calls = 0;
+	const int MAX_CALLS = getenv("MAX_CALLOC_CALLS") ? atoi(getenv("MAX_CALLOC_CALLS")) : -1;
+
+	if (!og_calloc)
+	{
+		og_calloc = (f)dlsym(RTLD_NEXT, "calloc");
+	}
+
+	calls++;
+	if (MAX_CALLS < 0 || calls <= MAX_CALLS)
+	{
+		return og_calloc(__nmemb, __size);
+	}
+	errno = ENOMEM;
+	return NULL;
+}

src/malloc.c

@@ -0,0 +1,23 @@
+#include <dlfcn.h>
+#include <errno.h>
+#include <stdlib.h>
+
+
+typedef void* (*f)(size_t size);
+
+void* malloc(size_t size) {
+  static f og_malloc = NULL;
+  static int calls = 0;
+  const int MAX_CALLS = getenv("MAX_MALLOC_CALLS") ? atoi(getenv("MAX_MALLOC_CALLS")) : -1;
+
+  if (!og_malloc) {
+    og_malloc = (f)dlsym(RTLD_NEXT, "malloc");
+  }
+
+  calls++;
+  if (MAX_CALLS < 0 || calls <= MAX_CALLS) {
+    return og_malloc(size);
+  }
+  errno = ENOMEM;
+  return NULL;
+}

src/write.c

@@ -0,0 +1,26 @@
+#include <dlfcn.h>
+#include <errno.h>
+#include <unistd.h>
+#include <stdlib.h>
+
+typedef ssize_t (*f)(int fd, const void *buf, size_t count);
+
+ssize_t write(int fd, const void *buf, size_t count)
+{
+	static f og_write = NULL;
+	static int calls = 0;
+	const int MAX_CALLS = getenv("MAX_WRITE_CALLS") ? atoi(getenv("MAX_WRITE_CALLS")) : -1;
+
+	if (!og_write)
+	{
+		og_write = (f)dlsym(RTLD_NEXT, "write");
+	}
+
+	calls++;
+	if (MAX_CALLS < 0 || calls <= MAX_CALLS)
+	{
+		return og_write(fd, buf, count);
+	}
+	errno = EBADF;
+	return -1;
+}