ash: exec: Stricter pathopt parsing
[oweals/busybox.git] / util-linux / readprofile.c
index 1f5ba2eccae9d9d6d3a3d18b7eccd16bd0bc34f4..c4ea374be8cc31092d07f255f3f11a5d35bd5f54 100644 (file)
@@ -1,10 +1,10 @@
 /* vi: set sw=4 ts=4: */
 /*
- *  readprofile.c - used to read /proc/profile
+ * readprofile.c - used to read /proc/profile
  *
- *  Copyright (C) 1994,1996 Alessandro Rubini (rubini@ipvvis.unipv.it)
+ * Copyright (C) 1994,1996 Alessandro Rubini (rubini@ipvvis.unipv.it)
  *
- * Licensed under GPLv2 or later, see file LICENSE in this tarball for details.
+ * Licensed under GPLv2 or later, see file LICENSE in this source tree.
  */
 
 /*
  * Taken from util-linux and adapted for busybox by
  * Paul Mundt <lethal@linux-sh.org>.
  */
+//config:config READPROFILE
+//config:      bool "readprofile (7.1 kb)"
+//config:      default y
+//config:      #select PLATFORM_LINUX
+//config:      help
+//config:      This allows you to parse /proc/profile for basic profiling.
+
+//applet:IF_READPROFILE(APPLET(readprofile, BB_DIR_USR_SBIN, BB_SUID_DROP))
+
+//kbuild:lib-$(CONFIG_READPROFILE) += readprofile.o
+
+//usage:#define readprofile_trivial_usage
+//usage:       "[OPTIONS]"
+//usage:#define readprofile_full_usage "\n\n"
+//usage:       "       -m mapfile      (Default: /boot/System.map)"
+//usage:     "\n       -p profile      (Default: /proc/profile)"
+//usage:     "\n       -M NUM          Set the profiling multiplier to NUM"
+//usage:     "\n       -i              Print only info about the sampling step"
+//usage:     "\n       -v              Verbose"
+//usage:     "\n       -a              Print all symbols, even if count is 0"
+//usage:     "\n       -b              Print individual histogram-bin counts"
+//usage:     "\n       -s              Print individual counters within functions"
+//usage:     "\n       -r              Reset all the counters (root only)"
+//usage:     "\n       -n              Disable byte order auto-detection"
 
 #include "libbb.h"
 #include <sys/utsname.h>
 
 #define S_LEN 128
 
-/* These are the defaults */
-static const char defaultmap[] ALIGN1 = "/boot/System.map";
-static const char defaultpro[] ALIGN1 = "/proc/profile";
-
 int readprofile_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
 int readprofile_main(int argc UNUSED_PARAM, char **argv)
 {
        FILE *map;
        const char *mapFile, *proFile;
-       unsigned long indx = 1;
+       unsigned long indx;
        size_t len;
-       uint64_t add0 = 0;
+       uint64_t add0;
        unsigned int step;
        unsigned int *buf, total, fn_len;
        unsigned long long fn_add, next_add;     /* current and next address */
        char fn_name[S_LEN], next_name[S_LEN];   /* current and next name */
        char mapline[S_LEN];
        char mode[8];
-       int maplineno = 1;
-       int header_printed;
-       int multiplier = 0;
+       int maplineno;
+       int multiplier;
        unsigned opt;
        enum {
                OPT_M = (1 << 0),
@@ -82,11 +101,11 @@ int readprofile_main(int argc UNUSED_PARAM, char **argv)
 
 #define next (current^1)
 
-       proFile = defaultpro;
-       mapFile = defaultmap;
+       proFile = "/proc/profile";
+       mapFile = "/boot/System.map";
+       multiplier = 0;
 
-       opt_complementary = "M+"; /* -M N */
-       opt = getopt32(argv, "M:m:p:nabsirv", &multiplier, &mapFile, &proFile);
+       opt = getopt32(argv, "M:+m:p:nabsirv", &multiplier, &mapFile, &proFile);
 
        if (opt & (OPT_M|OPT_r)) { /* mult or reset, or both */
                int fd, to_write;
@@ -97,9 +116,9 @@ int readprofile_main(int argc UNUSED_PARAM, char **argv)
                 */
                to_write = sizeof(int);
                if (!optMult)
-                       to_write = 1;   /* sth different from sizeof(int) */
+                       to_write = 1;  /* sth different from sizeof(int) */
 
-               fd = xopen(defaultpro, O_WRONLY);
+               fd = xopen("/proc/profile", O_WRONLY);
                xwrite(fd, &multiplier, to_write);
                close(fd);
                return EXIT_SUCCESS;
@@ -110,48 +129,48 @@ int readprofile_main(int argc UNUSED_PARAM, char **argv)
         */
        len = MAXINT(ssize_t);
        buf = xmalloc_xopen_read_close(proFile, &len);
+       len /= sizeof(*buf);
+
        if (!optNative) {
-               int entries = len / sizeof(*buf);
-               int big = 0, small = 0, i;
+               int big = 0, small = 0;
                unsigned *p;
 
-               for (p = buf+1; p < buf+entries; p++) {
+               for (p = buf+1; p < buf+len; p++) {
                        if (*p & ~0U << (sizeof(*buf)*4))
                                big++;
                        if (*p & ((1 << (sizeof(*buf)*4))-1))
                                small++;
                }
                if (big > small) {
-                       bb_error_msg("assuming reversed byte order, "
+                       bb_simple_error_msg("assuming reversed byte order, "
                                "use -n to force native byte order");
-                       for (p = buf; p < buf+entries; p++)
-                               for (i = 0; i < sizeof(*buf)/2; i++) {
-                                       unsigned char *b = (unsigned char *) p;
-                                       unsigned char tmp;
-
-                                       tmp = b[i];
-                                       b[i] = b[sizeof(*buf)-i-1];
-                                       b[sizeof(*buf)-i-1] = tmp;
-                               }
+                       BUILD_BUG_ON(sizeof(*p) > 8);
+                       for (p = buf; p < buf+len; p++) {
+                               if (sizeof(*p) == 2)
+                                       *p = bswap_16(*p);
+                               if (sizeof(*p) == 4)
+                                       *p = bswap_32(*p);
+                               if (sizeof(*p) == 8)
+                                       *p = bb_bswap_64(*p);
+                       }
                }
        }
 
        step = buf[0];
        if (optInfo) {
-               printf("Sampling_step: %i\n", step);
+               printf("Sampling_step: %u\n", step);
                return EXIT_SUCCESS;
        }
 
-       total = 0;
-
        map = xfopen_for_read(mapFile);
-
+       add0 = 0;
+       maplineno = 1;
        while (fgets(mapline, S_LEN, map)) {
                if (sscanf(mapline, "%llx %s %s", &fn_add, mode, fn_name) != 3)
                        bb_error_msg_and_die("%s(%i): wrong map line",
-                                            mapFile, maplineno);
+                                       mapFile, maplineno);
 
-               if (!strcmp(fn_name, "_stext")) /* only elf works like this */ {
+               if (strcmp(fn_name, "_stext") == 0) /* only elf works like this */ {
                        add0 = fn_add;
                        break;
                }
@@ -164,8 +183,11 @@ int readprofile_main(int argc UNUSED_PARAM, char **argv)
        /*
         * Main loop.
         */
+       total = 0;
+       indx = 1;
        while (fgets(mapline, S_LEN, map)) {
-               unsigned int this = 0;
+               bool header_printed;
+               unsigned int this;
 
                if (sscanf(mapline, "%llx %s %s", &next_add, mode, next_name) != 3)
                        bb_error_msg_and_die("%s(%i): wrong map line",
@@ -175,15 +197,17 @@ int readprofile_main(int argc UNUSED_PARAM, char **argv)
 
                /* ignore any LEADING (before a '[tT]' symbol is found)
                   Absolute symbols */
-               if ((*mode == 'A' || *mode == '?') && total == 0) continue;
-               if (*mode != 'T' && *mode != 't' &&
-                   *mode != 'W' && *mode != 'w')
-                       break;  /* only text is profiled */
+               if ((mode[0] == 'A' || mode[0] == '?') && total == 0)
+                       continue;
+               if ((mode[0]|0x20) != 't' && (mode[0]|0x20) != 'w') {
+                       break;  /* only text is profiled */
+               }
 
-               if (indx >= len / sizeof(*buf))
-                       bb_error_msg_and_die("profile address out of range. "
-                                            "Wrong map file?");
+               if (indx >= len)
+                       bb_simple_error_msg_and_die("profile address out of range. "
+                                       "Wrong map file?");
 
+               this = 0;
                while (indx < (next_add-add0)/step) {
                        if (optBins && (buf[indx] || optAll)) {
                                if (!header_printed) {
@@ -199,15 +223,16 @@ int readprofile_main(int argc UNUSED_PARAM, char **argv)
                if (optBins) {
                        if (optVerbose || this > 0)
                                printf("  total\t\t\t\t%u\n", this);
-               } else if ((this || optAll)
-                       && (fn_len = next_add-fn_add) != 0
+               } else
+               if ((this || optAll)
+                && (fn_len = next_add-fn_add) != 0
                ) {
                        if (optVerbose)
-                               printf("%016llx %-40s %6i %8.4f\n", fn_add,
-                                      fn_name, this, this/(double)fn_len);
+                               printf("%016llx %-40s %6u %8.4f\n", fn_add,
+                                       fn_name, this, this/(double)fn_len);
                        else
-                               printf("%6i %-40s %8.4f\n",
-                                      this, fn_name, this/(double)fn_len);
+                               printf("%6u %-40s %8.4f\n",
+                                       this, fn_name, this/(double)fn_len);
                        if (optSub) {
                                unsigned long long scan;
 
@@ -217,8 +242,8 @@ int readprofile_main(int argc UNUSED_PARAM, char **argv)
 
                                        addr = (scan - 1)*step + add0;
                                        printf("\t%#llx\t%s+%#llx\t%u\n",
-                                              addr, fn_name, addr - fn_add,
-                                              buf[scan]);
+                                               addr, fn_name, addr - fn_add,
+                                               buf[scan]);
                                }
                        }
                }
@@ -230,18 +255,20 @@ int readprofile_main(int argc UNUSED_PARAM, char **argv)
        }
 
        /* clock ticks, out of kernel text - probably modules */
-       printf("%6i %s\n", buf[len/sizeof(*buf)-1], "*unknown*");
+       printf("%6u *unknown*\n", buf[len-1]);
 
        /* trailer */
        if (optVerbose)
-               printf("%016x %-40s %6i %8.4f\n",
-                      0, "total", total, total/(double)(fn_add-add0));
+               printf("%016x %-40s %6u %8.4f\n",
+                       0, "total", total, total/(double)(fn_add-add0));
        else
-               printf("%6i %-40s %8.4f\n",
-                      total, "total", total/(double)(fn_add-add0));
+               printf("%6u %-40s %8.4f\n",
+                       total, "total", total/(double)(fn_add-add0));
 
-       fclose(map);
-       free(buf);
+       if (ENABLE_FEATURE_CLEAN_UP) {
+               fclose(map);
+               free(buf);
+       }
 
        return EXIT_SUCCESS;
 }