implemented numeric sort (sort -g)
[oweals/busybox.git] / utility.c
index a653bb32538a526a74e122713abf8a48575c37ac..d01be9c98c87b74f1cd81a2a937f27100e694146 100644 (file)
--- a/utility.c
+++ b/utility.c
 #include <unistd.h>
 #include <ctype.h>
 
-/* volatile so gcc knows this is the enod of the line */
-volatile void usage(const char *usage)
+#if defined BB_MOUNT || defined BB_UMOUNT || defined BB_DF
+#  if defined BB_FEATURE_USE_PROCFS
+const char mtab_file[] = "/proc/mounts";
+#  else
+#    if defined BB_MTAB
+const char mtab_file[] = "/etc/mtab";
+#    else
+#      error With (BB_MOUNT||BB_UMOUNT||BB_DF) defined, you must define either BB_MTAB or BB_FEATURE_USE_PROCFS
+#    endif
+#  endif
+#endif
+
+
+extern void usage(const char *usage)
 {
+    fprintf(stderr, "BusyBox v%s (%s) multi-call binary -- GPL2\n\n", BB_VER, BB_BT);
     fprintf(stderr, "Usage: %s\n", usage);
     exit(FALSE);
 }
@@ -46,6 +59,9 @@ volatile void usage(const char *usage)
 
 #if defined (BB_INIT) || defined (BB_PS)
 
+#if ! defined BB_FEATURE_USE_PROCFS
+#error Sorry, I depend on the /proc filesystem right now.
+#endif
 /* Returns kernel version encoded as major*65536 + minor*256 + patch,
  * so, for example,  to check if the kernel is greater than 2.2.11:
  *     if (get_kernel_revision() <= 2*65536+2*256+11) { <stuff> }
@@ -59,7 +75,7 @@ get_kernel_revision()
 
   file = fopen(filename,"r");
   if (file == NULL) {
-    perror(filename);
+    /* bummer, /proc must not be mounted... */
     return( 0);
   }
   fscanf(file,"%d.%d.%d",&major,&minor,&patch);
@@ -201,10 +217,12 @@ copyFile( const char *srcName, const char *destName,
     if (setModes == TRUE) {
        //fprintf(stderr, "Setting permissions for %s\n", destName);
        chmod(destName, srcStatBuf.st_mode);
-       if (followLinks == TRUE)
-           chown(destName, srcStatBuf.st_uid, srcStatBuf.st_gid);
-       else
+#if (__GLIBC__ >= 2) && (__GLIBC_MINOR__ >= 1)
+       if (followLinks == FALSE)
            lchown(destName, srcStatBuf.st_uid, srcStatBuf.st_gid);
+       else
+#endif
+           chown(destName, srcStatBuf.st_uid, srcStatBuf.st_gid);
 
        times.actime = srcStatBuf.st_atime;
        times.modtime = srcStatBuf.st_mtime;
@@ -226,7 +244,30 @@ copyFile( const char *srcName, const char *destName,
 
 
 
-#ifdef BB_TAR
+#if defined BB_TAR || defined BB_LS
+
+#define TYPEINDEX(mode) (((mode) >> 12) & 0x0f)
+#define TYPECHAR(mode)  ("0pcCd?bB-?l?s???" [TYPEINDEX(mode)])
+
+/* The special bits. If set, display SMODE0/1 instead of MODE0/1 */
+static const mode_t SBIT[] = {
+    0, 0, S_ISUID,
+    0, 0, S_ISGID,
+    0, 0, S_ISVTX
+};
+
+/* The 9 mode bits to test */
+static const mode_t MBIT[] = {
+    S_IRUSR, S_IWUSR, S_IXUSR,
+    S_IRGRP, S_IWGRP, S_IXGRP,
+    S_IROTH, S_IWOTH, S_IXOTH
+};
+
+#define MODE1  "rwxrwxrwx"
+#define MODE0  "---------"
+#define SMODE1 "..s..s..t"
+#define SMODE0 "..S..S..T"
+
 /*
  * Return the standard ls-like mode string from a file mode.
  * This is static and so is overwritten on each call.
@@ -235,64 +276,25 @@ const char *modeString(int mode)
 {
     static char buf[12];
 
-    strcpy(buf, "----------");
-
-    /*
-     * Fill in the file type.
-     */
-    if (S_ISDIR(mode))
-       buf[0] = 'd';
-    if (S_ISCHR(mode))
-       buf[0] = 'c';
-    if (S_ISBLK(mode))
-       buf[0] = 'b';
-    if (S_ISFIFO(mode))
-       buf[0] = 'p';
-    if (S_ISLNK(mode))
-       buf[0] = 'l';
-    if (S_ISSOCK(mode))
-       buf[0] = 's';
-    /*
-     * Now fill in the normal file permissions.
-     */
-    if (mode & S_IRUSR)
-       buf[1] = 'r';
-    if (mode & S_IWUSR)
-       buf[2] = 'w';
-    if (mode & S_IXUSR)
-       buf[3] = 'x';
-    if (mode & S_IRGRP)
-       buf[4] = 'r';
-    if (mode & S_IWGRP)
-       buf[5] = 'w';
-    if (mode & S_IXGRP)
-       buf[6] = 'x';
-    if (mode & S_IROTH)
-       buf[7] = 'r';
-    if (mode & S_IWOTH)
-       buf[8] = 'w';
-    if (mode & S_IXOTH)
-       buf[9] = 'x';
-
-    /*
-     * Finally fill in magic stuff like suid and sticky text.
-     */
-    if (mode & S_ISUID)
-       buf[3] = ((mode & S_IXUSR) ? 's' : 'S');
-    if (mode & S_ISGID)
-       buf[6] = ((mode & S_IXGRP) ? 's' : 'S');
-    if (mode & S_ISVTX)
-       buf[9] = ((mode & S_IXOTH) ? 't' : 'T');
-
+    int i;
+    buf[0] = TYPECHAR(mode);
+    for (i=0; i<9; i++) {
+       if (mode & SBIT[i])
+           buf[i+1] = (mode & MBIT[i])? 
+               SMODE1[i] : SMODE0[i];
+       else
+           buf[i+1] = (mode & MBIT[i])? 
+               MODE1[i] : MODE0[i];
+    }
     return buf;
 }
+#endif
 
 
+#if defined BB_TAR
 /*
- * Get the time string to be used for a file.
- * This is down to the minute for new files, but only the date for old files.
- * The string is returned from a static buffer, and so is overwritten for
- * each call.
+ * Return the standard ls-like time string from a time_t
+ * This is static and so is overwritten on each call.
  */
 const char *timeString(time_t timeVal)
 {
@@ -315,7 +317,6 @@ const char *timeString(time_t timeVal)
     return buf;
 }
 
-
 /*
  * Write all of the supplied buffer out to a file.
  * This does multiple writes as necessary.
@@ -341,8 +342,10 @@ int fullWrite(int fd, const char *buf, int len)
 
     return total;
 }
+#endif
 
 
+#if defined BB_TAR || defined BB_TAIL
 /*
  * Read all of the supplied buffer from a file.
  * This does multiple reads as necessary.
@@ -375,7 +378,7 @@ int fullRead(int fd, char *buf, int len)
 #endif
 
 
-#if defined (BB_CHOWN) || defined (BB_CP) || defined (BB_FIND) || defined (BB_LS)
+#if defined (BB_CHOWN) || defined (BB_CP) || defined (BB_FIND) || defined (BB_LS) || defined (BB_INSMOD)
 /*
  * Walk down all the directories under the specified 
  * location, and do something (something specified
@@ -383,7 +386,8 @@ int fullRead(int fd, char *buf, int len)
  *
  * Unfortunatly, while nftw(3) could replace this and reduce 
  * code size a bit, nftw() wasn't supported before GNU libc 2.1, 
- * and so isn't sufficiently portable to take over...
+ * and so isn't sufficiently portable to take over since glibc2.1
+ * is so stinking huge.
  */
 int
 recursiveAction(const char *fileName, int recurse, int followLinks, int depthFirst,
@@ -394,7 +398,7 @@ recursiveAction(const char *fileName, int recurse, int followLinks, int depthFir
     struct stat statbuf;
     struct dirent *next;
 
-    if (followLinks == FALSE)
+    if (followLinks == TRUE)
        status = stat(fileName, &statbuf);
     else
        status = lstat(fileName, &statbuf);
@@ -404,6 +408,13 @@ recursiveAction(const char *fileName, int recurse, int followLinks, int depthFir
        return (FALSE);
     }
 
+    if ( (followLinks == FALSE) && (S_ISLNK(statbuf.st_mode)) ) {
+       if (fileAction == NULL)
+           return (TRUE);
+       else
+           return (fileAction(fileName, &statbuf));
+    }
+
     if (recurse == FALSE) {
        if (S_ISDIR(statbuf.st_mode)) {
            if (dirAction != NULL)
@@ -480,19 +491,13 @@ extern void createPath (const char *name, int mode)
     char *cpOld;
     char buf[NAME_MAX];
 
-    strcpy (buf, name);
-
+    strcpy( buf, name);
     cp = strchr (buf, '/');
-
     while (cp) {
        cpOld = cp;
        cp = strchr (cp + 1, '/');
-
        *cpOld = '\0';
-
-       if (mkdir (buf, cp ? 0777 : mode) == 0)
-           printf ("Directory \"%s\" created\n", buf);
-
+       mkdir (buf, cp ? 0777 : mode);
        *cpOld = '/';
     }
 }
@@ -540,11 +545,9 @@ parse_mode( const char* s, mode_t* theMode)
                                        groups |= S_ISUID|S_ISGID|S_IRWXU|S_IRWXG|S_IRWXO;
                                break;
                        default:
-                               if ( isdigit(c) && c >= '0' && c <= '7' && mode == 0 && groups == 0 ) {
-                                       andMode = 0;
-                                       orMode = strtol(--s, NULL, 8);
-                                       *theMode &= andMode;
-                                       *theMode |= orMode;
+                               if ( isdigit(c) && c >= '0' && c <= '7' && 
+                                               mode == 0 && groups == 0 ) {
+                                       *theMode = strtol(--s, NULL, 8);
                                        return (TRUE);
                                }
                                else
@@ -646,11 +649,14 @@ my_getid(const char *filename, char *name, uid_t id)
                        continue;
 
                if (name) {
-                   if (0 == strcmp(rname, name))
+                   if (0 == strcmp(rname, name)) {
+                       fclose( file);
                        return( rid);
+                   }
                 }
                if ( id != -1 && id == rid ) {
                    strncpy(name, rname, 8);
+                   fclose( file);
                    return( TRUE);
                }
        }
@@ -687,39 +693,421 @@ my_getgrgid(char* group, gid_t gid)
 
 
 
-#if !defined BB_REGEXP && (defined BB_GREP || defined BB_FIND )  
+
+#if (defined BB_CHVT) || (defined BB_DEALLOCVT)
+
+
+#include <linux/kd.h>
+#include <sys/ioctl.h>
+
+int is_a_console(int fd) 
+{
+  char arg;
+  
+  arg = 0;
+  return (ioctl(fd, KDGKBTYPE, &arg) == 0
+         && ((arg == KB_101) || (arg == KB_84)));
+}
+
+static int open_a_console(char *fnam) 
+{
+  int fd;
+  
+  /* try read-only */
+  fd = open(fnam, O_RDWR);
+  
+  /* if failed, try read-only */
+  if (fd < 0 && errno == EACCES)
+      fd = open(fnam, O_RDONLY);
+  
+  /* if failed, try write-only */
+  if (fd < 0 && errno == EACCES)
+      fd = open(fnam, O_WRONLY);
+  
+  /* if failed, fail */
+  if (fd < 0)
+      return -1;
+  
+  /* if not a console, fail */
+  if (! is_a_console(fd))
+    {
+      close(fd);
+      return -1;
+    }
+  
+  /* success */
+  return fd;
+}
+
+/*
+ * Get an fd for use with kbd/console ioctls.
+ * We try several things because opening /dev/console will fail
+ * if someone else used X (which does a chown on /dev/console).
+ *
+ * if tty_name is non-NULL, try this one instead.
+ */
+
+int get_console_fd(char* tty_name) 
+{
+  int fd;
+
+  if (tty_name)
+    {
+      if (-1 == (fd = open_a_console(tty_name)))
+       return -1;
+      else
+       return fd;
+    }
+  
+  fd = open_a_console("/dev/tty");
+  if (fd >= 0)
+    return fd;
+  
+  fd = open_a_console("/dev/tty0");
+  if (fd >= 0)
+    return fd;
+  
+  fd = open_a_console("/dev/console");
+  if (fd >= 0)
+    return fd;
+  
+  for (fd = 0; fd < 3; fd++)
+    if (is_a_console(fd))
+      return fd;
+  
+  fprintf(stderr,
+         "Couldnt get a file descriptor referring to the console\n");
+  return -1;           /* total failure */
+}
+
+
+#endif
+
+
+#if !defined BB_REGEXP && (defined BB_GREP || defined BB_SED)  
+
+/* Do a case insensitive strstr() */
+char* stristr(char *haystack, const char *needle)
+{
+    int len = strlen( needle );
+    while( *haystack ) {
+       if( !strncasecmp( haystack, needle, len ) )
+           break;
+       haystack++;
+    }
+
+    if( !(*haystack) )
+           haystack = NULL;
+
+    return haystack;
+}
+
 /* This tries to find a needle in a haystack, but does so by
  * only trying to match literal strings (look 'ma, no regexps!)
  * This is short, sweet, and carries _very_ little baggage,
- * unlike its beefier cousin a few lines down...
+ * unlike its beefier cousin in regexp.c
  *  -Erik Andersen
  */
 extern int find_match(char *haystack, char *needle, int ignoreCase)
 {
 
-    if (ignoreCase == FALSE) {
+    if (ignoreCase == FALSE)
        haystack = strstr (haystack, needle);
-       if (haystack == NULL)
-           return FALSE;
-       return TRUE;
-    } else {
-       int i;
-       char needle1[BUF_SIZE];
-       char haystack1[BUF_SIZE];
-
-       strncpy( haystack1, haystack, sizeof(haystack1));
-       strncpy( needle1, needle, sizeof(needle1));
-       for( i=0; i<sizeof(haystack1) && haystack1[i]; i++)
-           haystack1[i]=tolower( haystack1[i]);
-       for( i=0; i<sizeof(needle1) && needle1[i]; i++)
-           needle1[i]=tolower( needle1[i]);
-       haystack = strstr (haystack1, needle1);
-       if (haystack == NULL)
-           return FALSE;
+    else
+       haystack = stristr (haystack, needle);
+    if (haystack == NULL)
+       return FALSE;
+    return TRUE;
+}
+
+
+/* This performs substitutions after a string match has been found.  */
+extern int replace_match(char *haystack, char *needle, char *newNeedle, int ignoreCase)
+{
+    int foundOne=0;
+    char *where, *slider, *slider1, *oldhayStack;
+
+    if (ignoreCase == FALSE)
+       where = strstr (haystack, needle);
+    else
+       where = stristr (haystack, needle);
+
+    if (strcmp(needle, newNeedle)==0)
+       return FALSE;
+
+    oldhayStack = (char*)malloc((unsigned)(strlen(haystack)));
+    while(where!=NULL) {
+       foundOne++;
+       strcpy(oldhayStack, haystack);
+#if 0
+       if ( strlen(newNeedle) > strlen(needle)) {
+           haystack = (char *)realloc(haystack, (unsigned)(strlen(haystack) - 
+               strlen(needle) + strlen(newNeedle)));
+       }
+#endif
+       for(slider=haystack,slider1=oldhayStack;slider!=where;slider++,slider1++);
+       *slider=0;
+       haystack=strcat(haystack, newNeedle);
+       slider1+=strlen(needle);
+       haystack = strcat(haystack, slider1);
+       where = strstr (slider, needle);
+    }
+    free( oldhayStack);
+
+    if (foundOne > 0)
        return TRUE;
+    else
+       return FALSE;
+}
+
+
+#endif
+
+
+#if defined BB_FIND
+/*
+ * Routine to see if a text string is matched by a wildcard pattern.
+ * Returns TRUE if the text is matched, or FALSE if it is not matched
+ * or if the pattern is invalid.
+ *  *          matches zero or more characters
+ *  ?          matches a single character
+ *  [abc]      matches 'a', 'b' or 'c'
+ *  \c         quotes character c
+ * Adapted from code written by Ingo Wilken, and
+ * then taken from sash, Copyright (c) 1999 by David I. Bell
+ * Permission is granted to use, distribute, or modify this source,
+ * provided that this copyright notice remains intact.
+ * Permission to distribute this code under the GPL has been granted.
+ */
+extern int
+check_wildcard_match(const char* text, const char* pattern)
+{
+    const char*        retryPat;
+    const char*        retryText;
+    int                ch;
+    int                found;
+
+    retryPat = NULL;
+    retryText = NULL;
+
+    while (*text || *pattern)
+    {
+       ch = *pattern++;
+
+       switch (ch)
+       {
+           case '*':  
+               retryPat = pattern;
+               retryText = text;
+               break;
+
+           case '[':  
+               found = FALSE;
+
+               while ((ch = *pattern++) != ']')
+               {
+                   if (ch == '\\')
+                       ch = *pattern++;
+
+                   if (ch == '\0')
+                       return FALSE;
+
+                   if (*text == ch)
+                       found = TRUE;
+               }
+
+               //if (!found)
+               if (found==TRUE)
+               {
+                   pattern = retryPat;
+                   text = ++retryText;
+               }
+
+               /* fall into next case */
+
+           case '?':  
+               if (*text++ == '\0')
+                   return FALSE;
+
+               break;
+
+           case '\\':  
+               ch = *pattern++;
+
+               if (ch == '\0')
+                       return FALSE;
+
+               /* fall into next case */
+
+           default:        
+               if (*text == ch)
+               {
+                   if (*text)
+                       text++;
+                   break;
+               }
+
+               if (*text)
+               {
+                   pattern = retryPat;
+                   text = ++retryText;
+                   break;
+               }
+
+               return FALSE;
+       }
+
+       if (pattern == NULL)
+               return FALSE;
     }
+
+    return TRUE;
 }
 #endif
 
-/* END CODE */
 
+
+
+#if defined BB_DF | defined BB_MTAB
+/*
+ * Given a block device, find the mount table entry if that block device
+ * is mounted.
+ *
+ * Given any other file (or directory), find the mount table entry for its
+ * filesystem.
+ */
+extern struct mntent *findMountPoint(const char *name, const char *table)
+{
+    struct stat s;
+    dev_t mountDevice;
+    FILE *mountTable;
+    struct mntent *mountEntry;
+
+    if (stat(name, &s) != 0)
+       return 0;
+
+    if ((s.st_mode & S_IFMT) == S_IFBLK)
+       mountDevice = s.st_rdev;
+    else
+       mountDevice = s.st_dev;
+
+
+    if ((mountTable = setmntent(table, "r")) == 0)
+       return 0;
+
+    while ((mountEntry = getmntent(mountTable)) != 0) {
+       if (strcmp(name, mountEntry->mnt_dir) == 0
+           || strcmp(name, mountEntry->mnt_fsname) == 0)       /* String match. */
+           break;
+       if (stat(mountEntry->mnt_fsname, &s) == 0 && s.st_rdev == mountDevice)  /* Match the device. */
+           break;
+       if (stat(mountEntry->mnt_dir, &s) == 0 && s.st_dev == mountDevice)      /* Match the directory's mount point. */
+           break;
+    }
+    endmntent(mountTable);
+    return mountEntry;
+}
+
+#endif
+
+
+
+#if !defined BB_MTAB && (defined BB_MOUNT || defined BB_DF )
+extern void whine_if_fstab_is_missing()
+{
+    struct stat statBuf;
+    if (stat("/etc/fstab", &statBuf) < 0) 
+       fprintf(stderr, "/etc/fstab file missing -- install one to name /dev/root.\n\n");
+}
+#endif
+
+
+#if defined BB_DD || defined BB_TAIL
+/*
+ * Read a number with a possible multiplier.
+ * Returns -1 if the number format is illegal.
+ */
+extern long getNum (const char *cp)
+{
+    long value;
+
+    if (!isDecimal (*cp))
+       return -1;
+
+    value = 0;
+
+    while (isDecimal (*cp))
+       value = value * 10 + *cp++ - '0';
+
+    switch (*cp++) {
+    case 'm':
+       value *= 1048576;
+       break;
+
+    case 'k':
+       value *= 1024;
+       break;
+
+    case 'b':
+       value *= 512;
+       break;
+
+    case 'w':
+       value *= 2;
+       break;
+
+    case '\0':
+       return value;
+
+    default:
+       return -1;
+    }
+
+    if (*cp)
+       return -1;
+
+    return value;
+}
+#endif
+
+
+#if defined BB_INIT || defined BB_HALT || defined BB_REBOOT 
+
+#if ! defined BB_FEATURE_USE_PROCFS
+#error Sorry, I depend on the /proc filesystem right now.
+#endif
+/* findInitPid()
+ *  
+ *  This finds the pid of init (which is not always 1).
+ *  Currently, it's implemented by rummaging through the proc filesystem.
+ *
+ *  [return]
+ *  0      failure
+ *  pid            when init's pid is found.
+ */
+extern pid_t
+findInitPid()
+{
+    pid_t   init_pid;
+    char    filename[256];
+    char    buffer[256];
+
+    /* no need to opendir ;) */
+    for (init_pid = 1; init_pid < 65536; init_pid++) {
+       FILE    *status;
+
+       sprintf(filename, "/proc/%d/status", init_pid);
+       status = fopen(filename, "r");
+       if (!status) { continue; }
+       fgets(buffer, 256, status);
+       fclose(status);
+
+       if ( (strstr(buffer, "init\n") != NULL )) {
+           return init_pid;
+       }
+    }
+    return 0;
+}
+#endif
+
+/* END CODE */