From 1f10bff58efe7eb6d2746f2318190358b45d14b7 Mon Sep 17 00:00:00 2001
From: Connor Olding <cloningdonor@gmail.com>
Date: Tue, 6 Sep 2022 22:08:22 +0000
Subject: [PATCH] Left-align stub arguments to dd (fixes #374)

This satisfies busybox's non-standard integer
argument parsing, and even saves a few bytes.
---
 ape/ape.S             | 12 ++++----
 ape/macros.internal.h | 67 +++++++++++++++++++++----------------------
 2 files changed, 39 insertions(+), 40 deletions(-)

diff --git a/ape/ape.S b/ape/ape.S
index 9ff9f4cd9..89c5d48d5 100644
--- a/ape/ape.S
+++ b/ape/ape.S
@@ -569,11 +569,11 @@ apesh:	.ascii	"'\n#'\"\n"			# sixth edition shebang
 	.ascii	  "t=\"${TMPDIR:-${HOME:-.}}/.ape\"\n"
 	.ascii	  "[ -x \"$t\" ] || {\n"
 	.ascii	    "mkdir -p \"${t%/*}\" &&\n"
-	.ascii	    "dd if=\"$o\" of=\"$t.$$\" skip=\""
+	.ascii	    "dd if=\"$o\" of=\"$t.$$\" skip="
 	.shstub	      ape_loader_dd_skip,2
-	.ascii	      "\" count=\""
+	.ascii	      " count="
 	.shstub	      ape_loader_dd_count,2
-	.ascii	      "\" bs=64 2>/dev/null\n"
+	.ascii	      " bs=64 2>/dev/null\n"
 #if SupportsXnu()
 	.ascii	    "[ -d /Applications ] && "
 	.ascii	      "dd if=\"$t.$$\""
@@ -637,11 +637,11 @@ apesh:	.ascii	"'\n#'\"\n"			# sixth edition shebang
 	.ascii	  "dd if=\"$o\""
 	.ascii	    " of=\"$o\""
 	.ascii	    " bs=8"
-	.ascii	    " skip=\""
+	.ascii	    " skip="
 	.shstub	    ape_macho_dd_skip,2
-	.ascii	    "\" count=\""
+	.ascii	    " count="
 	.shstub	    ape_macho_dd_count,2
-	.ascii	    "\" conv=notrunc 2>/dev/null\n"
+	.ascii	    " conv=notrunc 2>/dev/null\n"
 #endif /* XNU */
 	.ascii	"[ x\"$1\" = x--assimilate ] && exit 0\n"
 #ifndef APE_NO_MODIFY_SELF
diff --git a/ape/macros.internal.h b/ape/macros.internal.h
index 973602a60..79d311200 100644
--- a/ape/macros.internal.h
+++ b/ape/macros.internal.h
@@ -225,40 +225,39 @@
  * need to be inserted so Mac doesn't consider them octal; therefore,
  * parameters must be quoted; and eight digits should be good enough.
  */
-#define SHSTUB2(SYM, X)           \
-  HIDDEN(SYM##_bcs0 = BCD10K(X)); \
-  HIDDEN(SYM##_bcs1 = BCD(X))
-#define BCD(X)                                                                 \
-  ((X) == 0                                                                    \
-       ? 0x20202030                                                            \
-       : (X) < 10 ? 0x30202020 + (((X) % 10) << 24)                            \
-                  : (X) < 100 ? 0x30302020 + (((X) % 10) << 24) +              \
-                                    (((X) / 10 % 10) << 16)                    \
-                              : (X) < 1000 ? 0x30303020 + (((X) % 10) << 24) + \
-                                                 (((X) / 10 % 10) << 16) +     \
-                                                 (((X) / 100 % 10) << 8)       \
-                                           : 0x30303030 + (((X) % 10) << 24) + \
-                                                 (((X) / 10 % 10) << 16) +     \
-                                                 (((X) / 100 % 10) << 8) +     \
-                                                 (((X) / 1000 % 10) << 0))
-#define BCD10K(X)                                                          \
-  ((X) < 10000                                                             \
-       ? 0x20202020                                                        \
-       : (X) < 100000                                                      \
-             ? 0x30202020 + (((X) / 10000 % 10) << 24)                     \
-             : (X) < 1000000                                               \
-                   ? 0x30302020 + (((X) / 10000 % 10) << 24) +             \
-                         (((X) / 100000 % 10) << 16)                       \
-                   : (X) < 10000000                                        \
-                         ? 0x30303020 + (((X) / 10000 % 10) << 24) +       \
-                               (((X) / 100000 % 10) << 16) +               \
-                               (((X) / 1000000 % 10) << 8)                 \
-                         : (X) < 100000000                                 \
-                               ? 0x30303030 + (((X) / 10000 % 10) << 24) + \
-                                     (((X) / 100000 % 10) << 16) +         \
-                                     (((X) / 1000000 % 10) << 8) +         \
-                                     (((X) / 10000000 % 10) << 0)          \
-                               : 0xffffffffffffffff)
+#define SHSTUB2(SYM, X)             \
+  HIDDEN(SYM##_bcs0 = BCD_LEFT(X)); \
+  HIDDEN(SYM##_bcs1 = BCD_RIGHT(X))
+#define BCD_HELPER(X)                          \
+  (((X)) < 10    ? 0x20202030 +                \
+                   (X) % 10                    \
+   : (X) < 100   ? 0x20203030 +                \
+                   ((X) / 10) % 10 +           \
+                   (X) % 10 * 0x100            \
+   : (X) < 1000  ? 0x20303030 +                \
+                   ((X) / 100) % 10 +          \
+                   ((X) / 10) % 10 * 0x100 +   \
+                   (X) % 10 * 0x10000          \
+   : (X) < 10000 ? 0x30303030 +                \
+                   ((X) / 1000) % 10 +         \
+                   ((X) / 100) % 10 * 0x100 +  \
+                   ((X) / 10) % 10 * 0x10000 + \
+                   (X) % 10 * 0x1000000        \
+   : 0xffffffffffffffff)
+#define BCD_LEFT(X)                            \
+  (((X)) < 10000     ? BCD_HELPER(X)           \
+   : (X) < 100000    ? BCD_HELPER((X) / 10)    \
+   : (X) < 1000000   ? BCD_HELPER((X) / 100)   \
+   : (X) < 10000000  ? BCD_HELPER((X) / 1000)  \
+   : (X) < 100000000 ? BCD_HELPER((X) / 10000) \
+   : 0xffffffffffffffff)
+#define BCD_RIGHT(X)                           \
+  (((X)) < 10000     ? 0x20202020              \
+   : (X) < 100000    ? BCD_HELPER((X) % 10)    \
+   : (X) < 1000000   ? BCD_HELPER((X) % 100)   \
+   : (X) < 10000000  ? BCD_HELPER((X) % 1000)  \
+   : (X) < 100000000 ? BCD_HELPER((X) % 10000) \
+   : 0xffffffffffffffff)
 
 #endif /* __ASSEMBLER__ */
 #endif /* APE_MACROS_H_ */