ocran 1.3.18 → 1.4.0

data/src/Makefile

@@ -0,0 +1,75 @@
+# Detect host OS
+UNAME           := $(shell uname -s 2>/dev/null || echo Windows)
+IS_POSIX        := $(filter $(UNAME),Linux Darwin)
+
+CC              := gcc
+CFLAGS          := -Wall -O2 -DWITH_LZMA -Ilzma
+LDFLAGS         := -s
+
+ifneq ($(IS_POSIX),)
+  EXEEXT          :=
+  LDLIBS          :=
+  GUI_LDFLAGS     :=
+  STUB_CFLAGS     := $(CFLAGS) -D_CONSOLE
+  SYSTEM_UTILS_SRC := system_utils_posix.c
+  PROG_NAMES      := stub
+  RESOURCE_OBJ    :=
+else
+  EXEEXT          := .exe
+  LDLIBS          := -lbcrypt
+  GUI_LDFLAGS     := -mwindows
+  STUB_CFLAGS     := $(CFLAGS) -D_CONSOLE
+  STUBW_CFLAGS    := $(CFLAGS)
+  SYSTEM_UTILS_SRC := system_utils.c
+  PROG_NAMES      := stub stubw edicon
+  RESOURCE_OBJ    := stub.res
+endif
+
+BINDIR          := $(CURDIR)/../share/ocran
+BINARIES        := $(addsuffix $(EXEEXT), $(PROG_NAMES))
+
+LZMA_SRCS       := lzma/LzmaDec.c
+LZMA_OBJS       := $(LZMA_SRCS:.c=.o)
+
+COMMON_SRCS     := $(SYSTEM_UTILS_SRC) inst_dir.c script_info.c unpack.c
+COMMON_OBJS     := $(COMMON_SRCS:.c=.o) $(LZMA_OBJS) $(RESOURCE_OBJ)
+
+VARIANT_SRCS    := stub.c error.c
+CONSOLE_OBJS    := $(VARIANT_SRCS:.c=_console.o)
+WINDOW_OBJS     := $(VARIANT_SRCS:.c=_window.o)
+
+.PHONY: all clean install
+all: $(BINARIES)
+
+%.o: %.c
+	$(CC) $(CFLAGS) -c $< -o $@
+
+%_console.o: %.c
+	$(CC) $(STUB_CFLAGS) -c $< -o $@ 
+
+%_window.o: %.c
+	$(CC) $(STUBW_CFLAGS) -c $< -o $@
+
+ifeq ($(IS_POSIX),)
+stub.res: stub.rc
+	windres -i stub.rc -O coff -o stub.res
+endif
+
+stub$(EXEEXT): $(COMMON_OBJS) $(CONSOLE_OBJS)
+	$(CC) $(LDFLAGS) $^ $(LDLIBS) -o $@
+
+ifeq ($(IS_POSIX),)
+stubw$(EXEEXT): $(COMMON_OBJS) $(WINDOW_OBJS)
+	$(CC) $(LDFLAGS) $(GUI_LDFLAGS) $^ $(LDLIBS) -o $@
+
+edicon$(EXEEXT): edicon.o
+	$(CC) $(LDFLAGS) $^ -o $@
+endif
+
+clean:
+	rm -f $(BINARIES) $(COMMON_OBJS) $(CONSOLE_OBJS) $(WINDOW_OBJS) \
+	      edicon.o $(RESOURCE_OBJ)
+
+install: $(BINARIES)
+	mkdir -p $(BINDIR)
+	cp -f $(BINARIES) $(BINDIR)/

data/src/edicon.c

@@ -0,0 +1,161 @@
+/**
+   Changes the Icon in a PE executable.
+*/
+
+#include <windows.h>
+#include <stdio.h>
+
+#pragma pack(push, 2)
+
+/* Icon file header */
+typedef struct
+{
+   WORD Reserved;
+   WORD ResourceType;
+   WORD ImageCount;
+} IconFileHeader;
+
+/* Icon File directory entry structure */
+typedef struct
+{
+   BYTE Width;
+   BYTE Height;
+   BYTE Colors;
+   BYTE Reserved;
+   WORD Planes;
+   WORD BitsPerPixel;
+   DWORD ImageSize;
+   DWORD ImageOffset;
+} IconDirectoryEntry;
+
+/* Group Icon Resource directory entry structure */
+typedef struct
+{
+   BYTE Width;
+   BYTE Height;
+   BYTE Colors;
+   BYTE Reserved;
+   WORD Planes;
+   WORD BitsPerPixel;
+   DWORD ImageSize;
+   WORD ResourceID;
+} IconDirResEntry, *PIconDirResEntry;
+
+/* Group Icon Structore (RT_GROUP_ICON) */
+typedef struct
+{
+   WORD Reserved;
+   WORD ResourceType;
+   WORD ImageCount;
+   IconDirResEntry Enries[0]; /* Number of these is in ImageCount */
+} GroupIcon;
+
+#pragma pack(pop)
+
+BOOL UpdateIcon(LPTSTR ExecutableFileName, LPTSTR IconFileName)
+{
+   HANDLE h = BeginUpdateResource(ExecutableFileName, FALSE);
+   if (h == INVALID_HANDLE_VALUE)
+   {
+      printf("Failed to BeginUpdateResource\n");
+      return FALSE;
+   }
+
+   /* Read the Icon file */
+   HANDLE hIconFile = CreateFile(IconFileName, GENERIC_READ, 0, NULL, OPEN_EXISTING, 0, NULL);
+   if (hIconFile == INVALID_HANDLE_VALUE)
+   {
+      fprintf(stderr, "Failed to open icon file.\n");
+      return FALSE;
+   }
+   DWORD Size = GetFileSize(hIconFile, NULL);
+   BYTE* Data = LocalAlloc(LMEM_FIXED, Size);
+   DWORD BytesRead;
+   if (!ReadFile(hIconFile, Data, Size, &BytesRead, NULL))
+   {
+      fprintf(stderr, "Failed to read icon file.\n");
+      return FALSE;
+   }
+   CloseHandle(hIconFile);
+   
+   IconFileHeader* header = (IconFileHeader*)Data;
+   IconDirectoryEntry* entries = (IconDirectoryEntry*)(header + 1);
+
+   /* Validate that all directory entries fit within the file */
+   DWORD entriesEnd = sizeof(IconFileHeader) + header->ImageCount * sizeof(IconDirectoryEntry);
+   if (entriesEnd > Size)
+   {
+      fprintf(stderr, "Icon file too small for declared ImageCount.\n");
+      LocalFree(Data);
+      return FALSE;
+   }
+
+   /* Create the RT_ICON resources */
+   int i;
+   for (i = 0; i < header->ImageCount; ++i)
+   {
+      if (entries[i].ImageOffset + entries[i].ImageSize > Size)
+      {
+         fprintf(stderr, "Icon entry %d exceeds file bounds.\n", i);
+         LocalFree(Data);
+         return FALSE;
+      }
+      BOOL b = UpdateResource(h, MAKEINTRESOURCE(RT_ICON), MAKEINTRESOURCE(101 + i), MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT), Data + entries[i].ImageOffset, entries[i].ImageSize);
+      if (!b)
+      {
+         fprintf(stderr, "failed to UpdateResource %lu\n", GetLastError());
+         return FALSE;
+      }
+   }
+
+   /* Create the RT_GROUP_ICON structure */
+   DWORD GroupIconSize = sizeof(GroupIcon) + header->ImageCount * sizeof(IconDirectoryEntry);
+   GroupIcon* gi = (GroupIcon*)LocalAlloc(LMEM_FIXED, GroupIconSize);
+   gi->Reserved = 0;
+   gi->ResourceType = header->ResourceType;
+   gi->ImageCount = header->ImageCount;
+   for (i = 0; i < header->ImageCount; ++i)
+   {
+      IconDirResEntry* e = &gi->Enries[i];
+      e->Width = entries[i].Width;
+      e->Height = entries[i].Height;
+      e->Colors = entries[i].Colors;
+      e->Reserved = entries[i].Reserved;
+      e->Planes = entries[i].Planes;
+      e->BitsPerPixel = entries[i].BitsPerPixel;
+      e->ImageSize = entries[i].ImageSize;
+      e->ResourceID = 101 + i;
+   }
+
+   /* Save the RT_GROUP_ICON resource */
+   BOOL b = UpdateResource(h, MAKEINTRESOURCE(RT_GROUP_ICON), MAKEINTRESOURCE(100), MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT), gi, GroupIconSize);
+   if (!b)
+   {
+      fprintf(stderr, "Failed to create group icon.\n");
+      return FALSE;
+   }
+
+   if (!EndUpdateResource(h, FALSE))
+   {
+      fprintf(stderr, "Failed to EndUpdateResource.\n");
+      return FALSE;
+   }
+   
+   return TRUE;
+}
+
+int main(int argc, char* argv[])
+{
+   if (argc == 3)
+   {
+      if (UpdateIcon(argv[1], argv[2]))
+         return 0;
+      else
+         return -1;
+   }
+   else
+   {
+      fprintf(stderr, "Usage: edicon.exe <exefile> <icofile>\n");
+      return -1;
+   }
+}

data/src/error.c

@@ -0,0 +1,100 @@
+#ifdef _WIN32
+#include <windows.h>
+#endif
+#include <string.h>
+#include <stdio.h>
+#include <stdarg.h>
+#include <stdbool.h>
+#include "error.h"
+
+static bool debug_mode = false;
+
+// Enable debug mode
+void EnableDebugMode()
+{
+    debug_mode = true;
+}
+
+static bool vformat_message(char *buffer, size_t buffer_size, const char *label,
+                            const char* format, va_list args)
+{
+    if (!buffer || buffer_size < 2) {
+        return false;
+    }
+
+    size_t offset = 0;
+
+    if (label) {
+        int label_needed = snprintf(buffer, buffer_size, "%s: ", label);
+        if (label_needed < 0 || (size_t)label_needed >= buffer_size) {
+            return false;
+        }
+        offset = (size_t)label_needed;
+    }
+
+    int needed = vsnprintf(buffer + offset, buffer_size - offset, format, args);
+    if (needed < 0) {
+        return false;
+    } else if ((size_t)needed >= buffer_size - offset - 2) {
+        offset = buffer_size - 2;
+    } else {
+        offset += (size_t)needed;
+    }
+    buffer[offset++] = '\n';
+    buffer[offset]   = '\0';
+    return true;
+}
+
+static void print_message(const char *label, const char *format, va_list args)
+{
+    char text[4096];
+    if (vformat_message(text, sizeof(text), label, format, args)) {
+        fputs(text, stderr);
+    } else {
+        fputs("log message formatting failed\n", stderr);
+    }
+}
+
+// Prints a fatal error message to stderr.
+void PrintFatalMessage(const char *format, ...)
+{
+    va_list args;
+    va_start(args, format);
+    print_message("FATAL", format, args);
+    va_end(args);
+}
+
+// Displays a fatal error message via a message box.
+#ifdef _WIN32
+void PrintFatalMessageBox(const char *format, ...)
+{
+    char TextBuffer[1024];
+    va_list args;
+    va_start(args, format);
+    vsnprintf(TextBuffer, 1024, format, args);
+    va_end(args);
+    MessageBox(NULL, TextBuffer, "OCRAN", MB_OK | MB_ICONWARNING);
+}
+#endif
+
+// Prints an application level error message to stderr if in debug mode.
+void PrintAppErrorMessage(const char *format, ...)
+{
+    if (!debug_mode) return;
+
+    va_list args;
+    va_start(args, format);
+    print_message("ERROR", format, args);
+    va_end(args);
+}
+
+// Prints a debug message to stderr if in debug mode.
+void PrintDebugMessage(const char *format, ...)
+{
+    if (!debug_mode) return;
+
+    va_list args;
+    va_start(args, format);
+    print_message("DEBUG", format, args);
+    va_end(args);
+}

data/src/error.h

@@ -0,0 +1,66 @@
+#include <stdarg.h>
+
+// Exit code definitions
+#define EXIT_CODE_SUCCESS (0)
+#define EXIT_CODE_FAILURE (-1)
+
+/**
+ * EnableDebugMode - Enable debug mode
+ *
+ * This function enables debug mode and attaches a console for debug output.
+ */
+void EnableDebugMode();
+
+/**
+ * PrintFatalMessage - Prints a fatal error message to stderr.
+ *
+ * This function prints a formatted error message to stderr. The message is
+ * prefixed with "FATAL: ".
+ *
+ * @param format The format of the error message to be displayed. This is a printf-like
+ * format string followed by additional arguments.
+ */
+void PrintFatalMessage(const char *format, ...);
+
+/**
+ * PrintFatalMessageBox - Displays a fatal error message via a message box.
+ *
+ * This function displays a formatted error message in a message box with the
+ * caption 'OCRAN' and an icon of MB_ICONWARNING.
+ *
+ * @param format The format of the error message to be displayed. This is a printf-like
+ * format string followed by additional arguments.
+ */
+#ifdef _WIN32
+void PrintFatalMessageBox(const char *format, ...);
+#endif
+
+#if defined(_CONSOLE) || !defined(_WIN32)
+#define FATAL(...) PrintFatalMessage(__VA_ARGS__)
+#else
+#define FATAL(...) PrintFatalMessageBox(__VA_ARGS__)
+#endif
+
+/**
+ * PrintAppErrorMessage - Prints an application level error message to stderr if in debug mode.
+ *
+ * This function prints a formatted error message to stderr. The message is
+ * prefixed with "ERROR: ".
+ *
+ * @param format The format of the error message to be displayed. This is a printf-like
+ * format string followed by additional arguments.
+ */
+void PrintAppErrorMessage(const char *format, ...);
+#define APP_ERROR(...) PrintAppErrorMessage(__VA_ARGS__)
+
+/**
+ * PrintDebugMessage - Prints a debug message to stderr if in debug mode.
+ *
+ * This function prints a formatted debug message to stderr. The message is
+ * prefixed with "DEBUG: ".
+ *
+ * @param format The format of the debug message to be displayed. This is a printf-like
+ * format string followed by additional arguments.
+ */
+void PrintDebugMessage(const char *format, ...);
+#define DEBUG(...) PrintDebugMessage(__VA_ARGS__)