ash: exec: Stricter pathopt parsing
[oweals/busybox.git] / util-linux / readprofile.c
index 974fe89c4847807eedeabdef1a94ee5d8019cdf3..c4ea374be8cc31092d07f255f3f11a5d35bd5f54 100644 (file)
@@ -1,8 +1,8 @@
 /* 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 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]"
 
 #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),
@@ -96,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;
@@ -113,7 +118,7 @@ int readprofile_main(int argc UNUSED_PARAM, char **argv)
                if (!optMult)
                        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;
@@ -124,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);
 
-               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;
                }
@@ -178,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",
@@ -189,17 +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'
-               ) {
+               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. "
+               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) {
@@ -215,14 +223,15 @@ 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,
+                               printf("%016llx %-40s %6u %8.4f\n", fn_add,
                                        fn_name, this, this/(double)fn_len);
                        else
-                               printf("%6i %-40s %8.4f\n",
+                               printf("%6u %-40s %8.4f\n",
                                        this, fn_name, this/(double)fn_len);
                        if (optSub) {
                                unsigned long long scan;
@@ -246,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",
+               printf("%016x %-40s %6u %8.4f\n",
                        0, "total", total, total/(double)(fn_add-add0));
        else
-               printf("%6i %-40s %8.4f\n",
+               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;
 }