]> git.proxmox.com Git - grub2.git/blobdiff - grub-core/commands/ls.c
malloc: Use overflow checking primitives where we do complex allocations
[grub2.git] / grub-core / commands / ls.c
index ed8afd4ae123bd049acd14146f5209f221ca2095..5b7491aa49b959a63a21d01567bd10b7d56ba5c2 100644 (file)
@@ -31,6 +31,9 @@
 #include <grub/extcmd.h>
 #include <grub/datetime.h>
 #include <grub/i18n.h>
+#include <grub/net.h>
+
+GRUB_MOD_LICENSE ("GPLv3+");
 
 static const struct grub_arg_option options[] =
   {
@@ -40,136 +43,140 @@ static const struct grub_arg_option options[] =
     {0, 0, 0, 0, 0, 0}
   };
 
-static const char grub_human_sizes[] = {' ', 'K', 'M', 'G', 'T'};
+/* Helper for grub_ls_list_devices.  */
+static int
+grub_ls_print_devices (const char *name, void *data)
+{
+  int *longlist = data;
+
+  if (*longlist)
+    grub_normal_print_device_info (name);
+  else
+    grub_printf ("(%s) ", name);
+
+  return 0;
+}
 
 static grub_err_t
 grub_ls_list_devices (int longlist)
 {
-  auto int grub_ls_print_devices (const char *name);
-  int grub_ls_print_devices (const char *name)
-    {
-      if (longlist)
-       grub_normal_print_device_info (name);
-      else
-       grub_printf ("(%s) ", name);
+  grub_device_iterate (grub_ls_print_devices, &longlist);
+  grub_xputs ("\n");
 
-      return 0;
+#if 0
+  {
+    grub_net_app_level_t proto;
+    int first = 1;
+    FOR_NET_APP_LEVEL (proto)
+    {
+      if (first)
+       grub_puts_ (N_ ("Network protocols:"));
+      first = 0;
+      grub_printf ("%s ", proto->name);
     }
+    grub_xputs ("\n");
+  }
+#endif
 
-  grub_device_iterate (grub_ls_print_devices);
-  grub_xputs ("\n");
   grub_refresh ();
 
   return 0;
 }
 
-static grub_err_t
-grub_ls_list_files (char *dirname, int longlist, int all, int human)
+/* Context for grub_ls_list_files.  */
+struct grub_ls_list_files_ctx
 {
-  char *device_name;
-  grub_fs_t fs;
-  const char *path;
-  grub_device_t dev;
+  char *dirname;
+  int all;
+  int human;
+};
+
+/* Helper for grub_ls_list_files.  */
+static int
+print_files (const char *filename, const struct grub_dirhook_info *info,
+            void *data)
+{
+  struct grub_ls_list_files_ctx *ctx = data;
 
-  auto int print_files (const char *filename,
-                       const struct grub_dirhook_info *info);
-  auto int print_files_long (const char *filename,
-                            const struct grub_dirhook_info *info);
+  if (ctx->all || filename[0] != '.')
+    grub_printf ("%s%s ", filename, info->dir ? "/" : "");
 
-  int print_files (const char *filename, const struct grub_dirhook_info *info)
-    {
-      if (all || filename[0] != '.')
-       grub_printf ("%s%s ", filename, info->dir ? "/" : "");
+  return 0;
+}
 
-      return 0;
-    }
+/* Helper for grub_ls_list_files.  */
+static int
+print_files_long (const char *filename, const struct grub_dirhook_info *info,
+                 void *data)
+{
+  struct grub_ls_list_files_ctx *ctx = data;
 
-  int print_files_long (const char *filename,
-                       const struct grub_dirhook_info *info)
-    {
-      if ((! all) && (filename[0] == '.'))
-       return 0;
+  if ((! ctx->all) && (filename[0] == '.'))
+    return 0;
 
-      if (! info->dir)
-       {
-         grub_file_t file;
-         char *pathname;
+  if (! info->dir)
+    {
+      grub_file_t file;
+      char *pathname;
 
-         if (dirname[grub_strlen (dirname) - 1] == '/')
-           pathname = grub_xasprintf ("%s%s", dirname, filename);
-         else
-           pathname = grub_xasprintf ("%s/%s", dirname, filename);
+      if (ctx->dirname[grub_strlen (ctx->dirname) - 1] == '/')
+       pathname = grub_xasprintf ("%s%s", ctx->dirname, filename);
+      else
+       pathname = grub_xasprintf ("%s/%s", ctx->dirname, filename);
 
-         if (!pathname)
-           return 1;
+      if (!pathname)
+       return 1;
 
-         /* XXX: For ext2fs symlinks are detected as files while they
-            should be reported as directories.  */
-         grub_file_filter_disable_compression ();
-         file = grub_file_open (pathname);
-         if (! file)
-           {
-             grub_errno = 0;
-             grub_free (pathname);
-             return 0;
-           }
-
-         if (! human)
-           grub_printf ("%-12llu", (unsigned long long) file->size);
-         else
-           {
-             grub_uint64_t fsize = file->size * 100ULL;
-             int fsz = file->size;
-             int units = 0;
-             char buf[20];
-
-             while (fsz / 1024)
-               {
-                 fsize = (fsize + 512) / 1024;
-                 fsz /= 1024;
-                 units++;
-               }
-
-             if (units)
-               {
-                 grub_uint32_t whole, fraction;
-
-                 whole = grub_divmod64 (fsize, 100, &fraction);
-                 grub_snprintf (buf, sizeof (buf),
-                                "%u.%02u%c", whole, fraction,
-                                grub_human_sizes[units]);
-                 grub_printf ("%-12s", buf);
-               }
-             else
-               grub_printf ("%-12llu", (unsigned long long) file->size);
-
-           }
-         grub_file_close (file);
+      /* XXX: For ext2fs symlinks are detected as files while they
+        should be reported as directories.  */
+      file = grub_file_open (pathname, GRUB_FILE_TYPE_GET_SIZE
+                            | GRUB_FILE_TYPE_NO_DECOMPRESS);
+      if (! file)
+       {
+         grub_errno = 0;
          grub_free (pathname);
+         return 0;
        }
-      else
-       grub_printf ("%-12s", "DIR");
 
-      if (info->mtimeset)
-       {
-         struct grub_datetime datetime;
-         grub_unixtime2datetime (info->mtime, &datetime);
-         if (human)
-           grub_printf (" %d-%02d-%02d %02d:%02d:%02d %-11s ",
-                        datetime.year, datetime.month, datetime.day,
-                        datetime.hour, datetime.minute,
-                        datetime.second,
-                        grub_get_weekday_name (&datetime));
-         else
-           grub_printf (" %04d%02d%02d%02d%02d%02d ",
-                        datetime.year, datetime.month,
-                        datetime.day, datetime.hour,
-                        datetime.minute, datetime.second);
-       }
-      grub_printf ("%s%s\n", filename, info->dir ? "/" : "");
+      if (! ctx->human)
+       grub_printf ("%-12llu", (unsigned long long) file->size);
+      else
+       grub_printf ("%-12s", grub_get_human_size (file->size,
+                                                  GRUB_HUMAN_SIZE_SHORT));
+      grub_file_close (file);
+      grub_free (pathname);
+    }
+  else
+    grub_printf ("%-12s", _("DIR"));
 
-      return 0;
+  if (info->mtimeset)
+    {
+      struct grub_datetime datetime;
+      grub_unixtime2datetime (info->mtime, &datetime);
+      if (ctx->human)
+       grub_printf (" %d-%02d-%02d %02d:%02d:%02d %-11s ",
+                    datetime.year, datetime.month, datetime.day,
+                    datetime.hour, datetime.minute,
+                    datetime.second,
+                    grub_get_weekday_name (&datetime));
+      else
+       grub_printf (" %04d%02d%02d%02d%02d%02d ",
+                    datetime.year, datetime.month,
+                    datetime.day, datetime.hour,
+                    datetime.minute, datetime.second);
     }
+  grub_printf ("%s%s\n", filename, info->dir ? "/" : "");
+
+  return 0;
+}
+
+static grub_err_t
+grub_ls_list_files (char *dirname, int longlist, int all, int human)
+{
+  char *device_name;
+  grub_fs_t fs;
+  const char *path;
+  grub_device_t dev;
 
   device_name = grub_file_get_device_name (dirname);
   dev = grub_device_open (device_name);
@@ -194,14 +201,29 @@ grub_ls_list_files (char *dirname, int longlist, int all, int human)
       if (grub_errno == GRUB_ERR_UNKNOWN_FS)
        grub_errno = GRUB_ERR_NONE;
 
+#ifdef GRUB_MACHINE_IEEE1275
+      /*
+       * Close device to prevent a double open in grub_normal_print_device_info().
+       * Otherwise it may lead to hangs on some IEEE 1275 platforms.
+       */
+      grub_device_close (dev);
+      dev = NULL;
+#endif
+
       grub_normal_print_device_info (device_name);
     }
   else if (fs)
     {
+      struct grub_ls_list_files_ctx ctx = {
+       .dirname = dirname,
+       .all = all,
+       .human = human
+      };
+
       if (longlist)
-       (fs->dir) (dev, path, print_files_long);
+       (fs->fs_dir) (dev, path, print_files_long, &ctx);
       else
-       (fs->dir) (dev, path, print_files);
+       (fs->fs_dir) (dev, path, print_files, &ctx);
 
       if (grub_errno == GRUB_ERR_BAD_FILE_TYPE
          && path[grub_strlen (path) - 1] != '/')
@@ -212,8 +234,8 @@ grub_ls_list_files (char *dirname, int longlist, int all, int human)
          struct grub_dirhook_info info;
          grub_errno = 0;
 
-         grub_file_filter_disable_compression ();
-         file = grub_file_open (dirname);
+         file = grub_file_open (dirname, GRUB_FILE_TYPE_GET_SIZE
+                                | GRUB_FILE_TYPE_NO_DECOMPRESS);
          if (! file)
            goto fail;
 
@@ -227,9 +249,9 @@ grub_ls_list_files (char *dirname, int longlist, int all, int human)
          all = 1;
          grub_memset (&info, 0, sizeof (info));
          if (longlist)
-           print_files_long (p, &info);
+           print_files_long (p, &info, &ctx);
          else
-           print_files (p, &info);
+           print_files (p, &info, &ctx);
 
          grub_free (dirname);
        }
@@ -269,8 +291,8 @@ static grub_extcmd_t cmd;
 
 GRUB_MOD_INIT(ls)
 {
-  cmd = grub_register_extcmd ("ls", grub_cmd_ls, GRUB_COMMAND_FLAG_BOTH,
-                             N_("[-l|-h|-a] [FILE]"),
+  cmd = grub_register_extcmd ("ls", grub_cmd_ls, 0,
+                             N_("[-l|-h|-a] [FILE ...]"),
                              N_("List devices and files."), options);
 }