support GNU suffixes for fancy mode: b(512), k(1024) and m(1024*1024)

1 files changed, 16 insertions, 0 deletions

diff --git a/coreutils/head.c b/coreutils/head.c
index 17d3ef890..184e8161c 100644
--- a/coreutils/head.c
+++ b/coreutils/head.c
@@ -25,6 +25,15 @@ static const char head_opts[] =
 #endif
         ;
 
+#if ENABLE_FEATURE_FANCY_HEAD
+static const struct suffix_mult head_suffixes[] = {
+        { "b", 512 },
+        { "k", 1024 },
+        { "m", 1024*1024 },
+        { NULL, 0 }
+};
+#endif
+                                        
 static const char header_fmt_str[] = "\n==> %s <==\n";
 
 int head_main(int argc, char **argv)
@@ -75,7 +84,14 @@ int head_main(int argc, char **argv)
 #if !ENABLE_DEBUG_YANK_SUSv2 || ENABLE_FEATURE_FANCY_HEAD
                         GET_COUNT:
 #endif
+
+#if !ENABLE_FEATURE_FANCY_HEAD
                                 count = bb_xgetularg10(p);
+#else
+                                count = bb_xgetularg_bnd_sfx(p, 10, 
+                                                                0, ULONG_MAX, 
+                                                                head_suffixes);
+#endif
                                 break;
                         default:
                                 bb_show_usage();