utility.c

   1 /* vi: set sw=4 ts=4: */
   2 /*
   3  * Utility routines.
   4  *
   5  * Copyright (C) tons of folks.  Tracking down who wrote what
   6  * isn't something I'm going to worry about...  If you wrote something
   7  * here, please feel free to acknowledge your work.
   8  *
   9  * This program is free software; you can redistribute it and/or modify
  10  * it under the terms of the GNU General Public License as published by
  11  * the Free Software Foundation; either version 2 of the License, or
  12  * (at your option) any later version.
  13  *
  14  * This program is distributed in the hope that it will be useful,
  15  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  16  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  17  * General Public License for more details.
  18  *
  19  * You should have received a copy of the GNU General Public License
  20  * along with this program; if not, write to the Free Software
  21  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
  22  *
  23  * Based in part on code from sash, Copyright (c) 1999 by David I. Bell
  24  * Permission has been granted to redistribute this code under the GPL.
  25  *
  26  */
  27
  28 #include <stdio.h>
  29 #include <string.h>
  30 #include <errno.h>
  31 #include <fcntl.h>
  32 #include <dirent.h>
  33 #include <time.h>
  34 #include <utime.h>
  35 #include <unistd.h>
  36 #include <ctype.h>
  37 #include <stdlib.h>
  38 #include <limits.h>
  39 #include <stdarg.h>
  40 #include <sys/ioctl.h>
  41 #include <sys/utsname.h>                /* for uname(2) */
  42
  43 #include "busybox.h"
  44 #if defined (BB_CHMOD_CHOWN_CHGRP) \
  45  || defined (BB_CP_MV)             \
  46  || defined (BB_FIND)              \
  47  || defined (BB_INSMOD)            \
  48  || defined (BB_LS)                \
  49  || defined (BB_RM)                \
  50  || defined (BB_TAR)
  51 /* same conditions as recursive_action */
  52 #define bb_need_name_too_long
  53 #endif
  54 #define bb_need_memory_exhausted
  55 #define bb_need_full_version
  56 #define BB_DECLARE_EXTERN
  57 #include "messages.c"
  58
  59 #include "pwd_grp/pwd.h"
  60 #include "pwd_grp/grp.h"
  61
  62 /* for the _syscall() macros */
  63 #include <sys/syscall.h>
  64 #include <linux/unistd.h>
  65
  66 /* Busybox mount uses either /proc/mounts or /dev/mtab to
  67  * get the list of currently mounted filesystems */
  68 #if defined BB_MOUNT || defined BB_UMOUNT || defined BB_DF
  69 #  if defined BB_MTAB
  70 const char mtab_file[] = "/etc/mtab";
  71 #  else
  72 #      if defined BB_FEATURE_USE_DEVPS_PATCH
  73 const char mtab_file[] = "/dev/mtab";
  74 #    else
  75 const char mtab_file[] = "/proc/mounts";
  76 #    endif
  77 #  endif
  78 #endif
  79
  80 static struct BB_applet *applet_using;
  81
  82 extern void show_usage(void)
  83 {
  84         const char *format_string;
  85         const char *usage_string = usage_messages;
  86         int i;
  87
  88         for (i = applet_using - applets; i > 0; ) {
  89                 if (!*usage_string++) {
  90                         --i;
  91                 }
  92         }
  93         format_string = "%s\n\nUsage: %s %s\n\n";
  94         if(*usage_string == 0)
  95                 format_string = "%s\n\nNo help available.\n\n";
  96         fprintf(stderr, format_string,
  97                         full_version, applet_using->name, usage_string);
  98         exit(EXIT_FAILURE);
  99 }
 100
 101
 102 static void verror_msg(const char *s, va_list p)
 103 {
 104         fflush(stdout);
 105         fprintf(stderr, "%s: ", applet_name);
 106         vfprintf(stderr, s, p);
 107 }
 108
 109 extern void error_msg(const char *s, ...)
 110 {
 111         va_list p;
 112
 113         va_start(p, s);
 114         verror_msg(s, p);
 115         va_end(p);
 116         putc('\n', stderr);
 117 }
 118
 119 extern void error_msg_and_die(const char *s, ...)
 120 {
 121         va_list p;
 122
 123         va_start(p, s);
 124         verror_msg(s, p);
 125         va_end(p);
 126         putc('\n', stderr);
 127         exit(EXIT_FAILURE);
 128 }
 129
 130 static void vperror_msg(const char *s, va_list p)
 131 {
 132         int err=errno;
 133         if(s == 0) s = "";
 134         verror_msg(s, p);
 135         if (*s) s = ": ";
 136         fprintf(stderr, "%s%s\n", s, strerror(err));
 137 }
 138
 139 extern void perror_msg(const char *s, ...)
 140 {
 141         va_list p;
 142
 143         va_start(p, s);
 144         vperror_msg(s, p);
 145         va_end(p);
 146 }
 147
 148 extern void perror_msg_and_die(const char *s, ...)
 149 {
 150         va_list p;
 151
 152         va_start(p, s);
 153         vperror_msg(s, p);
 154         va_end(p);
 155         exit(EXIT_FAILURE);
 156 }
 157
 158 #if defined BB_INIT || defined BB_MKSWAP || defined BB_MOUNT || defined BB_NFSMOUNT
 159 /* Returns kernel version encoded as major*65536 + minor*256 + patch,
 160  * so, for example,  to check if the kernel is greater than 2.2.11:
 161  *     if (get_kernel_revision() <= 2*65536+2*256+11) { <stuff> }
 162  */
 163 extern int get_kernel_revision(void)
 164 {
 165         struct utsname name;
 166         char *s;
 167         int i, r;
 168
 169         if (uname(&name) == -1) {
 170                 perror_msg("cannot get system information");
 171                 return (0);
 172         }
 173
 174         s = name.release;
 175         r = 0;
 176         for (i=0 ; i<3 ; i++) {
 177                 r = r * 256 + atoi(strtok(s, "."));
 178                 s = NULL;
 179         }
 180         return r;
 181 }
 182 #endif                                                 /* BB_INIT */
 183
 184
 185
 186 #if defined BB_FREE || defined BB_INIT || defined BB_UNAME || defined BB_UPTIME
 187 _syscall1(int, sysinfo, struct sysinfo *, info);
 188 #endif                                                 /* BB_INIT */
 189
 190 #if defined BB_MOUNT || defined BB_UMOUNT
 191
 192 /* Include our own version of <sys/mount.h>, since libc5 doesn't
 193  * know about umount2 */
 194 extern _syscall1(int, umount, const char *, special_file);
 195 extern _syscall5(int, mount, const char *, special_file, const char *, dir,
 196                 const char *, fstype, unsigned long int, rwflag, const void *, data);
 197 #ifndef __NR_umount2
 198 # warning This kernel does not support the umount2 syscall
 199 # warning The umount2 system call is being stubbed out...
 200         int umount2(const char * special_file, int flags)
 201         {
 202                 /* BusyBox was compiled against a kernel that did not support
 203                  *  the umount2 system call.  To make this application work,
 204                  *  you will need to recompile with a kernel supporting the
 205                  *  umount2 system call.
 206                  */
 207                 fprintf(stderr, "\n\nTo make this application work, you will need to recompile\n");
 208                 fprintf(stderr, "with a kernel supporting the umount2 system call. -Erik\n\n");
 209                 errno=ENOSYS;
 210                 return -1;
 211         }
 212 # else
 213    extern _syscall2(int, umount2, const char *, special_file, int, flags);
 214 # endif
 215 #endif
 216
 217
 218
 219 #if defined BB_FEATURE_NEW_MODULE_INTERFACE && ( defined BB_INSMOD || defined BB_LSMOD )
 220 #ifndef __NR_query_module
 221 static const int __NR_query_module = 167;
 222 #endif
 223 _syscall5(int, query_module, const char *, name, int, which,
 224                 void *, buf, size_t, bufsize, size_t*, ret);
 225 #endif
 226
 227
 228 #if defined (BB_CP_MV) || defined (BB_DU)
 229
 230 #define HASH_SIZE       311             /* Should be prime */
 231 #define hash_inode(i)   ((i) % HASH_SIZE)
 232
 233 static ino_dev_hashtable_bucket_t *ino_dev_hashtable[HASH_SIZE];
 234
 235 /*
 236  * Return 1 if statbuf->st_ino && statbuf->st_dev are recorded in
 237  * `ino_dev_hashtable', else return 0
 238  *
 239  * If NAME is a non-NULL pointer to a character pointer, and there is
 240  * a match, then set *NAME to the value of the name slot in that
 241  * bucket.
 242  */
 243 int is_in_ino_dev_hashtable(const struct stat *statbuf, char **name)
 244 {
 245         ino_dev_hashtable_bucket_t *bucket;
 246
 247         bucket = ino_dev_hashtable[hash_inode(statbuf->st_ino)];
 248         while (bucket != NULL) {
 249           if ((bucket->ino == statbuf->st_ino) &&
 250                   (bucket->dev == statbuf->st_dev))
 251           {
 252                 if (name) *name = bucket->name;
 253                 return 1;
 254           }
 255           bucket = bucket->next;
 256         }
 257         return 0;
 258 }
 259
 260 /* Add statbuf to statbuf hash table */
 261 void add_to_ino_dev_hashtable(const struct stat *statbuf, const char *name)
 262 {
 263         int i;
 264         size_t s;
 265         ino_dev_hashtable_bucket_t *bucket;
 266
 267         i = hash_inode(statbuf->st_ino);
 268         s = name ? strlen(name) : 0;
 269         bucket = xmalloc(sizeof(ino_dev_hashtable_bucket_t) + s);
 270         bucket->ino = statbuf->st_ino;
 271         bucket->dev = statbuf->st_dev;
 272         if (name)
 273                 strcpy(bucket->name, name);
 274         else
 275                 bucket->name[0] = '\0';
 276         bucket->next = ino_dev_hashtable[i];
 277         ino_dev_hashtable[i] = bucket;
 278 }
 279
 280 /* Clear statbuf hash table */
 281 void reset_ino_dev_hashtable(void)
 282 {
 283         int i;
 284         ino_dev_hashtable_bucket_t *bucket;
 285
 286         for (i = 0; i < HASH_SIZE; i++) {
 287                 while (ino_dev_hashtable[i] != NULL) {
 288                         bucket = ino_dev_hashtable[i]->next;
 289                         free(ino_dev_hashtable[i]);
 290                         ino_dev_hashtable[i] = bucket;
 291                 }
 292         }
 293 }
 294
 295 #endif /* BB_CP_MV || BB_DU */
 296
 297 #if defined (BB_CP_MV) || defined (BB_DU) || defined (BB_LN) || defined (BB_DPKG_DEB)
 298 /*
 299  * Return TRUE if a fileName is a directory.
 300  * Nonexistant files return FALSE.
 301  */
 302 int is_directory(const char *fileName, const int followLinks, struct stat *statBuf)
 303 {
 304         int status;
 305         int didMalloc = 0;
 306
 307         if (statBuf == NULL) {
 308             statBuf = (struct stat *)xmalloc(sizeof(struct stat));
 309             ++didMalloc;
 310         }
 311
 312         if (followLinks == TRUE)
 313                 status = stat(fileName, statBuf);
 314         else
 315                 status = lstat(fileName, statBuf);
 316
 317         if (status < 0 || !(S_ISDIR(statBuf->st_mode))) {
 318             status = FALSE;
 319         }
 320         else status = TRUE;
 321
 322         if (didMalloc) {
 323             free(statBuf);
 324             statBuf = NULL;
 325         }
 326         return status;
 327 }
 328 #endif
 329
 330 #if defined BB_AR || defined BB_CP_MV
 331 /*
 332  * Copy chunksize bytes between two file descriptors
 333  */
 334 int copy_file_chunk(int srcfd, int dstfd, size_t chunksize)
 335 {
 336         size_t size;
 337         char buffer[BUFSIZ]; /* BUFSIZ is declared in stdio.h */
 338
 339         while (chunksize > 0) {
 340                 if (chunksize > BUFSIZ)
 341                         size = BUFSIZ;
 342                 else
 343                         size = chunksize;
 344                 if (full_write(dstfd, buffer, full_read(srcfd, buffer, size)) < size)
 345                         return(FALSE);
 346                 chunksize -= size;
 347         }
 348         return (TRUE);
 349 }
 350 #endif
 351
 352
 353 #if defined (BB_CP_MV) || defined BB_DPKG
 354 /*
 355  * Copy one file to another, while possibly preserving its modes, times, and
 356  * modes.  Returns TRUE if successful, or FALSE on a failure with an error
 357  * message output.  (Failure is not indicated if attributes cannot be set.)
 358  * -Erik Andersen
 359  */
 360 int
 361 copy_file(const char *srcName, const char *destName,
 362                  int setModes, int followLinks, int forceFlag)
 363 {
 364         int rfd;
 365         int wfd;
 366         int status;
 367         struct stat srcStatBuf;
 368         struct stat dstStatBuf;
 369         struct utimbuf times;
 370
 371         if (followLinks == TRUE)
 372                 status = stat(srcName, &srcStatBuf);
 373         else
 374                 status = lstat(srcName, &srcStatBuf);
 375
 376         if (status < 0) {
 377                 perror_msg("%s", srcName);
 378                 return FALSE;
 379         }
 380
 381         if (followLinks == TRUE)
 382                 status = stat(destName, &dstStatBuf);
 383         else
 384                 status = lstat(destName, &dstStatBuf);
 385
 386         if (status < 0 || forceFlag==TRUE) {
 387                 unlink(destName);
 388                 dstStatBuf.st_ino = -1;
 389                 dstStatBuf.st_dev = -1;
 390         }
 391
 392         if ((srcStatBuf.st_dev == dstStatBuf.st_dev) &&
 393                 (srcStatBuf.st_ino == dstStatBuf.st_ino)) {
 394                 error_msg("Copying file \"%s\" to itself", srcName);
 395                 return FALSE;
 396         }
 397
 398         if (S_ISDIR(srcStatBuf.st_mode)) {
 399                 //fprintf(stderr, "copying directory %s to %s\n", srcName, destName);
 400                 /* Make sure the directory is writable */
 401                 status = mkdir(destName, 0777777 ^ umask(0));
 402                 if (status < 0 && errno != EEXIST) {
 403                         perror_msg("%s", destName);
 404                         return FALSE;
 405                 }
 406         } else if (S_ISLNK(srcStatBuf.st_mode)) {
 407                 char link_val[BUFSIZ + 1];
 408                 int link_size;
 409
 410                 //fprintf(stderr, "copying link %s to %s\n", srcName, destName);
 411                 /* Warning: This could possibly truncate silently, to BUFSIZ chars */
 412                 link_size = readlink(srcName, &link_val[0], BUFSIZ);
 413                 if (link_size < 0) {
 414                         perror_msg("%s", srcName);
 415                         return FALSE;
 416                 }
 417                 link_val[link_size] = '\0';
 418                 status = symlink(link_val, destName);
 419                 if (status < 0) {
 420                         perror_msg("%s", destName);
 421                         return FALSE;
 422                 }
 423 #if (__GLIBC__ >= 2) && (__GLIBC_MINOR__ >= 1)
 424                 if (setModes == TRUE) {
 425                         /* Try to set owner, but fail silently like GNU cp */
 426                         lchown(destName, srcStatBuf.st_uid, srcStatBuf.st_gid);
 427                 }
 428 #endif
 429                 return TRUE;
 430         } else if (S_ISFIFO(srcStatBuf.st_mode)) {
 431                 //fprintf(stderr, "copying fifo %s to %s\n", srcName, destName);
 432                 if (mkfifo(destName, 0644) < 0) {
 433                         perror_msg("%s", destName);
 434                         return FALSE;
 435                 }
 436         } else if (S_ISBLK(srcStatBuf.st_mode) || S_ISCHR(srcStatBuf.st_mode)
 437                            || S_ISSOCK(srcStatBuf.st_mode)) {
 438                 //fprintf(stderr, "copying soc, blk, or chr %s to %s\n", srcName, destName);
 439                 if (mknod(destName, srcStatBuf.st_mode, srcStatBuf.st_rdev) < 0) {
 440                         perror_msg("%s", destName);
 441                         return FALSE;
 442                 }
 443         } else if (S_ISREG(srcStatBuf.st_mode)) {
 444                 //fprintf(stderr, "copying regular file %s to %s\n", srcName, destName);
 445                 rfd = open(srcName, O_RDONLY);
 446                 if (rfd < 0) {
 447                         perror_msg("%s", srcName);
 448                         return FALSE;
 449                 }
 450
 451                 wfd = open(destName, O_WRONLY | O_CREAT | O_TRUNC,
 452                                  srcStatBuf.st_mode);
 453                 if (wfd < 0) {
 454                         perror_msg("%s", destName);
 455                         close(rfd);
 456                         return FALSE;
 457                 }
 458
 459                 if (copy_file_chunk(rfd, wfd, srcStatBuf.st_size)==FALSE)
 460                         goto error_exit;
 461
 462                 close(rfd);
 463                 if (close(wfd) < 0) {
 464                         return FALSE;
 465                 }
 466         }
 467
 468         if (setModes == TRUE) {
 469                 /* This is fine, since symlinks never get here */
 470                 if (chown(destName, srcStatBuf.st_uid, srcStatBuf.st_gid) < 0)
 471                         perror_msg_and_die("%s", destName);
 472                 if (chmod(destName, srcStatBuf.st_mode) < 0)
 473                         perror_msg_and_die("%s", destName);
 474                 times.actime = srcStatBuf.st_atime;
 475                 times.modtime = srcStatBuf.st_mtime;
 476                 if (utime(destName, &times) < 0)
 477                         perror_msg_and_die("%s", destName);
 478         }
 479
 480         return TRUE;
 481
 482   error_exit:
 483         perror_msg("%s", destName);
 484         close(rfd);
 485         close(wfd);
 486
 487         return FALSE;
 488 }
 489 #endif                                                  /* BB_CP_MV */
 490
 491
 492
 493 #if defined BB_TAR || defined BB_LS ||defined BB_AR
 494
 495 #define TYPEINDEX(mode) (((mode) >> 12) & 0x0f)
 496 #define TYPECHAR(mode)  ("0pcCd?bB-?l?s???" [TYPEINDEX(mode)])
 497
 498 /* The special bits. If set, display SMODE0/1 instead of MODE0/1 */
 499 static const mode_t SBIT[] = {
 500         0, 0, S_ISUID,
 501         0, 0, S_ISGID,
 502         0, 0, S_ISVTX
 503 };
 504
 505 /* The 9 mode bits to test */
 506 static const mode_t MBIT[] = {
 507         S_IRUSR, S_IWUSR, S_IXUSR,
 508         S_IRGRP, S_IWGRP, S_IXGRP,
 509         S_IROTH, S_IWOTH, S_IXOTH
 510 };
 511
 512 static const char MODE1[]  = "rwxrwxrwx";
 513 static const char MODE0[]  = "---------";
 514 static const char SMODE1[] = "..s..s..t";
 515 static const char SMODE0[] = "..S..S..T";
 516
 517 /*
 518  * Return the standard ls-like mode string from a file mode.
 519  * This is static and so is overwritten on each call.
 520  */
 521 const char *mode_string(int mode)
 522 {
 523         static char buf[12];
 524
 525         int i;
 526
 527         buf[0] = TYPECHAR(mode);
 528         for (i = 0; i < 9; i++) {
 529                 if (mode & SBIT[i])
 530                         buf[i + 1] = (mode & MBIT[i]) ? SMODE1[i] : SMODE0[i];
 531                 else
 532                         buf[i + 1] = (mode & MBIT[i]) ? MODE1[i] : MODE0[i];
 533         }
 534         return buf;
 535 }
 536 #endif                                                  /* BB_TAR || BB_LS */
 537
 538
 539 #if defined BB_TAR || defined BB_AR
 540 /*
 541  * Return the standard ls-like time string from a time_t
 542  * This is static and so is overwritten on each call.
 543  */
 544 const char *time_string(time_t timeVal)
 545 {
 546         time_t now;
 547         char *str;
 548         static char buf[26];
 549
 550         time(&now);
 551
 552         str = ctime(&timeVal);
 553
 554         strcpy(buf, &str[4]);
 555         buf[12] = '\0';
 556
 557         if ((timeVal > now) || (timeVal < now - 365 * 24 * 60 * 60L)) {
 558                 strcpy(&buf[7], &str[20]);
 559                 buf[11] = '\0';
 560         }
 561
 562         return buf;
 563 }
 564 #endif /* BB_TAR || BB_AR */
 565
 566 #if defined BB_DD || defined BB_NC || defined BB_TAIL || defined BB_TAR || defined BB_AR || defined BB_CP_MV
 567 /*
 568  * Write all of the supplied buffer out to a file.
 569  * This does multiple writes as necessary.
 570  * Returns the amount written, or -1 on an error.
 571  */
 572 int full_write(int fd, const char *buf, int len)
 573 {
 574         int cc;
 575         int total;
 576
 577         total = 0;
 578
 579         while (len > 0) {
 580                 cc = write(fd, buf, len);
 581
 582                 if (cc < 0)
 583                         return -1;
 584
 585                 buf += cc;
 586                 total += cc;
 587                 len -= cc;
 588         }
 589
 590         return total;
 591 }
 592 #endif
 593
 594 #if defined BB_AR || defined BB_CP_MV || defined BB_SH || defined BB_TAR
 595 /*
 596  * Read all of the supplied buffer from a file.
 597  * This does multiple reads as necessary.
 598  * Returns the amount read, or -1 on an error.
 599  * A short read is returned on an end of file.
 600  */
 601 int full_read(int fd, char *buf, int len)
 602 {
 603         int cc;
 604         int total;
 605
 606         total = 0;
 607
 608         while (len > 0) {
 609                 cc = read(fd, buf, len);
 610
 611                 if (cc < 0)
 612                         return -1;
 613
 614                 if (cc == 0)
 615                         break;
 616
 617                 buf += cc;
 618                 total += cc;
 619                 len -= cc;
 620         }
 621
 622         return total;
 623 }
 624 #endif /* BB_TAR || BB_TAIL || BB_AR || BB_SH */
 625
 626
 627 #if defined (BB_CHMOD_CHOWN_CHGRP) \
 628  || defined (BB_CP_MV)                  \
 629  || defined (BB_FIND)                   \
 630  || defined (BB_INSMOD)                 \
 631  || defined (BB_LS)                             \
 632  || defined (BB_RM)                             \
 633  || defined (BB_TAR)
 634
 635 /*
 636  * Walk down all the directories under the specified
 637  * location, and do something (something specified
 638  * by the fileAction and dirAction function pointers).
 639  *
 640  * Unfortunately, while nftw(3) could replace this and reduce
 641  * code size a bit, nftw() wasn't supported before GNU libc 2.1,
 642  * and so isn't sufficiently portable to take over since glibc2.1
 643  * is so stinking huge.
 644  */
 645 int recursive_action(const char *fileName,
 646                                         int recurse, int followLinks, int depthFirst,
 647                                         int (*fileAction) (const char *fileName,
 648                                                                            struct stat * statbuf,
 649                                                                            void* userData),
 650                                         int (*dirAction) (const char *fileName,
 651                                                                           struct stat * statbuf,
 652                                                                           void* userData),
 653                                         void* userData)
 654 {
 655         int status;
 656         struct stat statbuf;
 657         struct dirent *next;
 658
 659         if (followLinks == TRUE)
 660                 status = stat(fileName, &statbuf);
 661         else
 662                 status = lstat(fileName, &statbuf);
 663
 664         if (status < 0) {
 665 #ifdef BB_DEBUG_PRINT_SCAFFOLD
 666                 fprintf(stderr,
 667                                 "status=%d followLinks=%d TRUE=%d\n",
 668                                 status, followLinks, TRUE);
 669 #endif
 670                 perror_msg("%s", fileName);
 671                 return FALSE;
 672         }
 673
 674         if ((followLinks == FALSE) && (S_ISLNK(statbuf.st_mode))) {
 675                 if (fileAction == NULL)
 676                         return TRUE;
 677                 else
 678                         return fileAction(fileName, &statbuf, userData);
 679         }
 680
 681         if (recurse == FALSE) {
 682                 if (S_ISDIR(statbuf.st_mode)) {
 683                         if (dirAction != NULL)
 684                                 return (dirAction(fileName, &statbuf, userData));
 685                         else
 686                                 return TRUE;
 687                 }
 688         }
 689
 690         if (S_ISDIR(statbuf.st_mode)) {
 691                 DIR *dir;
 692
 693                 if (dirAction != NULL && depthFirst == FALSE) {
 694                         status = dirAction(fileName, &statbuf, userData);
 695                         if (status == FALSE) {
 696                                 perror_msg("%s", fileName);
 697                                 return FALSE;
 698                         } else if (status == SKIP)
 699                                 return TRUE;
 700                 }
 701                 dir = opendir(fileName);
 702                 if (!dir) {
 703                         perror_msg("%s", fileName);
 704                         return FALSE;
 705                 }
 706                 status = TRUE;
 707                 while ((next = readdir(dir)) != NULL) {
 708                         char nextFile[PATH_MAX];
 709
 710                         if ((strcmp(next->d_name, "..") == 0)
 711                                         || (strcmp(next->d_name, ".") == 0)) {
 712                                 continue;
 713                         }
 714                         if (strlen(fileName) + strlen(next->d_name) + 1 > PATH_MAX) {
 715                                 error_msg(name_too_long);
 716                                 return FALSE;
 717                         }
 718                         memset(nextFile, 0, sizeof(nextFile));
 719                         if (fileName[strlen(fileName)-1] == '/')
 720                                 sprintf(nextFile, "%s%s", fileName, next->d_name);
 721                         else
 722                                 sprintf(nextFile, "%s/%s", fileName, next->d_name);
 723                         if (recursive_action(nextFile, TRUE, followLinks, depthFirst,
 724                                                 fileAction, dirAction, userData) == FALSE) {
 725                                 status = FALSE;
 726                         }
 727                 }
 728                 closedir(dir);
 729                 if (dirAction != NULL && depthFirst == TRUE) {
 730                         if (dirAction(fileName, &statbuf, userData) == FALSE) {
 731                                 perror_msg("%s", fileName);
 732                                 return FALSE;
 733                         }
 734                 }
 735                 if (status == FALSE)
 736                         return FALSE;
 737         } else {
 738                 if (fileAction == NULL)
 739                         return TRUE;
 740                 else
 741                         return fileAction(fileName, &statbuf, userData);
 742         }
 743         return TRUE;
 744 }
 745
 746 #endif                                                  /* BB_CHMOD_CHOWN_CHGRP || BB_CP_MV || BB_FIND || BB_LS || BB_INSMOD */
 747
 748
 749
 750 #if defined (BB_TAR) || defined (BB_MKDIR)
 751 /*
 752  * Attempt to create the directories along the specified path, except for
 753  * the final component.  The mode is given for the final directory only,
 754  * while all previous ones get default protections.  Errors are not reported
 755  * here, as failures to restore files can be reported later.
 756  */
 757 extern int create_path(const char *name, int mode)
 758 {
 759         char *cp;
 760         char *cpOld;
 761         char buf[BUFSIZ + 1];
 762         int retVal = 0;
 763
 764         strcpy(buf, name);
 765         for (cp = buf; *cp == '/'; cp++);
 766         cp = strchr(cp, '/');
 767         while (cp) {
 768                 cpOld = cp;
 769                 cp = strchr(cp + 1, '/');
 770                 *cpOld = '\0';
 771                 retVal = mkdir(buf, cp ? 0777 : mode);
 772                 if (retVal != 0 && errno != EEXIST) {
 773                         perror_msg("%s", buf);
 774                         return FALSE;
 775                 }
 776                 *cpOld = '/';
 777         }
 778         return TRUE;
 779 }
 780 #endif                                                  /* BB_TAR || BB_MKDIR */
 781
 782
 783
 784 #if defined (BB_CHMOD_CHOWN_CHGRP) || defined (BB_MKDIR) \
 785  || defined (BB_MKFIFO) || defined (BB_MKNOD) || defined (BB_AR)
 786 /* [ugoa]{+|-|=}[rwxst] */
 787
 788 extern int parse_mode(const char *s, mode_t * theMode)
 789 {
 790         static const mode_t group_set[] = {
 791                 S_ISUID | S_IRWXU,              /* u */
 792                 S_ISGID | S_IRWXG,              /* g */
 793                 S_IRWXO,                                /* o */
 794                 S_ISUID | S_ISGID | S_ISVTX | S_IRWXU | S_IRWXG | S_IRWXO /* a */
 795         };
 796
 797         static const mode_t mode_set[] = {
 798                 S_IRUSR | S_IRGRP | S_IROTH, /* r */
 799                 S_IWUSR | S_IWGRP | S_IWOTH, /* w */
 800                 S_IXUSR | S_IXGRP | S_IXOTH, /* x */
 801                 S_ISUID | S_ISGID,              /* s */
 802                 S_ISVTX                                 /* t */
 803         };
 804
 805         static const char group_string[] = "ugoa";
 806         static const char mode_string[] = "rwxst";
 807
 808         const char *p;
 809
 810         mode_t andMode =
 811                 S_ISVTX | S_ISUID | S_ISGID | S_IRWXU | S_IRWXG | S_IRWXO;
 812         mode_t orMode = 0;
 813         mode_t mode;
 814         mode_t groups;
 815         char type;
 816         char c;
 817
 818         if (s==NULL) {
 819                 return (FALSE);
 820         }
 821
 822         do {
 823                 mode = 0;
 824                 groups = 0;
 825         NEXT_GROUP:
 826                 if ((c = *s++) == '\0') {
 827                         return -1;
 828                 }
 829                 for (p=group_string ; *p ; p++) {
 830                         if (*p == c) {
 831                                 groups |= group_set[(int)(p-group_string)];
 832                                 goto NEXT_GROUP;
 833                         }
 834                 }
 835                 switch (c) {
 836                         case '=':
 837                         case '+':
 838                         case '-':
 839                                 type = c;
 840                                 if (groups == 0) { /* The default is "all" */
 841                                         groups |= S_ISUID | S_ISGID | S_ISVTX
 842                                                         | S_IRWXU | S_IRWXG | S_IRWXO;
 843                                 }
 844                                 break;
 845                         default:
 846                                 if ((c < '0') || (c > '7') || (mode | groups)) {
 847                                         return (FALSE);
 848                                 } else {
 849                                         *theMode = strtol(--s, NULL, 8);
 850                                         return (TRUE);
 851                                 }
 852                 }
 853
 854         NEXT_MODE:
 855                 if (((c = *s++) != '\0') && (c != ',')) {
 856                         for (p=mode_string ; *p ; p++) {
 857                                 if (*p == c) {
 858                                         mode |= mode_set[(int)(p-mode_string)];
 859                                         goto NEXT_MODE;
 860                                 }
 861                         }
 862                         break;                          /* We're done so break out of loop.*/
 863                 }
 864                 switch (type) {
 865                         case '=':
 866                                 andMode &= ~(groups); /* Now fall through. */
 867                         case '+':
 868                                 orMode |= mode & groups;
 869                                 break;
 870                         case '-':
 871                                 andMode &= ~(mode & groups);
 872                                 orMode &= ~(mode & groups);
 873                                 break;
 874                 }
 875         } while (c == ',');
 876
 877         *theMode &= andMode;
 878         *theMode |= orMode;
 879
 880         return TRUE;
 881 }
 882
 883 #endif
 884 /* BB_CHMOD_CHOWN_CHGRP || BB_MKDIR || BB_MKFIFO || BB_MKNOD */
 885
 886
 887
 888
 889
 890 #if defined BB_CHMOD_CHOWN_CHGRP || defined BB_PS || defined BB_LS \
 891  || defined BB_TAR || defined BB_ID || defined BB_LOGGER \
 892  || defined BB_LOGNAME || defined BB_WHOAMI || defined BB_SH
 893
 894 #if defined BB_CHMOD_CHOWN_CHGRP || defined BB_ID
 895 /* returns a uid given a username */
 896 long my_getpwnam(const char *name)
 897 {
 898         struct passwd *myuser;
 899
 900         myuser  = getpwnam(name);
 901         if (myuser==NULL)
 902                 error_msg_and_die("unknown user name: %s", name);
 903
 904         return myuser->pw_uid;
 905 }
 906
 907 /* returns a gid given a group name */
 908 long my_getgrnam(const char *name)
 909 {
 910         struct group *mygroup;
 911
 912         mygroup  = getgrnam(name);
 913         if (mygroup==NULL)
 914                 error_msg_and_die("unknown group name: %s", name);
 915
 916         return (mygroup->gr_gid);
 917 }
 918 #endif
 919
 920 /* gets a username given a uid */
 921 void my_getpwuid(char *name, long uid)
 922 {
 923         struct passwd *myuser;
 924
 925         myuser  = getpwuid(uid);
 926         if (myuser==NULL)
 927                 sprintf(name, "%-8ld ", (long)uid);
 928         else
 929                 strcpy(name, myuser->pw_name);
 930 }
 931
 932 /* gets a groupname given a gid */
 933 void my_getgrgid(char *group, long gid)
 934 {
 935         struct group *mygroup;
 936
 937         mygroup  = getgrgid(gid);
 938         if (mygroup==NULL)
 939                 sprintf(group, "%-8ld ", (long)gid);
 940         else
 941                 strcpy(group, mygroup->gr_name);
 942 }
 943
 944 #if defined BB_ID
 945 /* gets a gid given a user name */
 946 long my_getpwnamegid(const char *name)
 947 {
 948         struct group *mygroup;
 949         struct passwd *myuser;
 950
 951         myuser=getpwnam(name);
 952         if (myuser==NULL)
 953                 error_msg_and_die("unknown user name: %s", name);
 954
 955         mygroup  = getgrgid(myuser->pw_gid);
 956         if (mygroup==NULL)
 957                 error_msg_and_die("unknown gid %ld", (long)myuser->pw_gid);
 958
 959         return mygroup->gr_gid;
 960 }
 961 #endif /* BB_ID */
 962 #endif
 963  /* BB_CHMOD_CHOWN_CHGRP || BB_PS || BB_LS || BB_TAR \
 964  || BB_ID || BB_LOGGER || BB_LOGNAME || BB_WHOAMI */
 965
 966
 967 #if (defined BB_CHVT) || (defined BB_DEALLOCVT) || (defined BB_SETKEYCODES)
 968
 969 /* From <linux/kd.h> */
 970 static const int KDGKBTYPE = 0x4B33;  /* get keyboard type */
 971 static const int KB_84 = 0x01;
 972 static const int KB_101 = 0x02;    /* this is what we always answer */
 973
 974 int is_a_console(int fd)
 975 {
 976         char arg;
 977
 978         arg = 0;
 979         return (ioctl(fd, KDGKBTYPE, &arg) == 0
 980                         && ((arg == KB_101) || (arg == KB_84)));
 981 }
 982
 983 static int open_a_console(char *fnam)
 984 {
 985         int fd;
 986
 987         /* try read-only */
 988         fd = open(fnam, O_RDWR);
 989
 990         /* if failed, try read-only */
 991         if (fd < 0 && errno == EACCES)
 992                 fd = open(fnam, O_RDONLY);
 993
 994         /* if failed, try write-only */
 995         if (fd < 0 && errno == EACCES)
 996                 fd = open(fnam, O_WRONLY);
 997
 998         /* if failed, fail */
 999         if (fd < 0)
1000                 return -1;
1001
1002         /* if not a console, fail */
1003         if (!is_a_console(fd)) {
1004                 close(fd);
1005                 return -1;
1006         }
1007
1008         /* success */
1009         return fd;
1010 }
1011
1012 /*
1013  * Get an fd for use with kbd/console ioctls.
1014  * We try several things because opening /dev/console will fail
1015  * if someone else used X (which does a chown on /dev/console).
1016  *
1017  * if tty_name is non-NULL, try this one instead.
1018  */
1019
1020 int get_console_fd(char *tty_name)
1021 {
1022         int fd;
1023
1024         if (tty_name) {
1025                 if (-1 == (fd = open_a_console(tty_name)))
1026                         return -1;
1027                 else
1028                         return fd;
1029         }
1030
1031         fd = open_a_console("/dev/tty");
1032         if (fd >= 0)
1033                 return fd;
1034
1035         fd = open_a_console("/dev/tty0");
1036         if (fd >= 0)
1037                 return fd;
1038
1039         fd = open_a_console("/dev/console");
1040         if (fd >= 0)
1041                 return fd;
1042
1043         for (fd = 0; fd < 3; fd++)
1044                 if (is_a_console(fd))
1045                         return fd;
1046
1047         error_msg("Couldnt get a file descriptor referring to the console");
1048         return -1;                                      /* total failure */
1049 }
1050
1051
1052 #endif                                                  /* BB_CHVT || BB_DEALLOCVT || BB_SETKEYCODES */
1053
1054
1055 #if defined BB_FIND || defined BB_INSMOD
1056 /*
1057  * Routine to see if a text string is matched by a wildcard pattern.
1058  * Returns TRUE if the text is matched, or FALSE if it is not matched
1059  * or if the pattern is invalid.
1060  *  *           matches zero or more characters
1061  *  ?           matches a single character
1062  *  [abc]       matches 'a', 'b' or 'c'
1063  *  \c          quotes character c
1064  * Adapted from code written by Ingo Wilken, and
1065  * then taken from sash, Copyright (c) 1999 by David I. Bell
1066  * Permission is granted to use, distribute, or modify this source,
1067  * provided that this copyright notice remains intact.
1068  * Permission to distribute this code under the GPL has been granted.
1069  */
1070 extern int check_wildcard_match(const char *text, const char *pattern)
1071 {
1072         const char *retryPat;
1073         const char *retryText;
1074         int ch;
1075         int found;
1076         int len;
1077
1078         retryPat = NULL;
1079         retryText = NULL;
1080
1081         while (*text || *pattern) {
1082                 ch = *pattern++;
1083
1084                 switch (ch) {
1085                 case '*':
1086                         retryPat = pattern;
1087                         retryText = text;
1088                         break;
1089
1090                 case '[':
1091                         found = FALSE;
1092
1093                         while ((ch = *pattern++) != ']') {
1094                                 if (ch == '\\')
1095                                         ch = *pattern++;
1096
1097                                 if (ch == '\0')
1098                                         return FALSE;
1099
1100                                 if (*text == ch)
1101                                         found = TRUE;
1102                         }
1103                         len=strlen(text);
1104                         if (found == FALSE && len!=0) {
1105                                 return FALSE;
1106                         }
1107                         if (found == TRUE) {
1108                                 if (strlen(pattern)==0 && len==1) {
1109                                         return TRUE;
1110                                 }
1111                                 if (len!=0) {
1112                                         text++;
1113                                         continue;
1114                                 }
1115                         }
1116
1117                         /* fall into next case */
1118
1119                 case '?':
1120                         if (*text++ == '\0')
1121                                 return FALSE;
1122
1123                         break;
1124
1125                 case '\\':
1126                         ch = *pattern++;
1127
1128                         if (ch == '\0')
1129                                 return FALSE;
1130
1131                         /* fall into next case */
1132
1133                 default:
1134                         if (*text == ch) {
1135                                 if (*text)
1136                                         text++;
1137                                 break;
1138                         }
1139
1140                         if (*text) {
1141                                 pattern = retryPat;
1142                                 text = ++retryText;
1143                                 break;
1144                         }
1145
1146                         return FALSE;
1147                 }
1148
1149                 if (pattern == NULL)
1150                         return FALSE;
1151         }
1152
1153         return TRUE;
1154 }
1155 #endif                            /* BB_FIND || BB_INSMOD */
1156
1157
1158
1159
1160 #if defined BB_DF || defined BB_MTAB
1161 #include <mntent.h>
1162 /*
1163  * Given a block device, find the mount table entry if that block device
1164  * is mounted.
1165  *
1166  * Given any other file (or directory), find the mount table entry for its
1167  * filesystem.
1168  */
1169 extern struct mntent *find_mount_point(const char *name, const char *table)
1170 {
1171         struct stat s;
1172         dev_t mountDevice;
1173         FILE *mountTable;
1174         struct mntent *mountEntry;
1175
1176         if (stat(name, &s) != 0)
1177                 return 0;
1178
1179         if ((s.st_mode & S_IFMT) == S_IFBLK)
1180                 mountDevice = s.st_rdev;
1181         else
1182                 mountDevice = s.st_dev;
1183
1184
1185         if ((mountTable = setmntent(table, "r")) == 0)
1186                 return 0;
1187
1188         while ((mountEntry = getmntent(mountTable)) != 0) {
1189                 if (strcmp(name, mountEntry->mnt_dir) == 0
1190                         || strcmp(name, mountEntry->mnt_fsname) == 0)   /* String match. */
1191                         break;
1192                 if (stat(mountEntry->mnt_fsname, &s) == 0 && s.st_rdev == mountDevice)  /* Match the device. */
1193                         break;
1194                 if (stat(mountEntry->mnt_dir, &s) == 0 && s.st_dev == mountDevice)      /* Match the directory's mount point. */
1195                         break;
1196         }
1197         endmntent(mountTable);
1198         return mountEntry;
1199 }
1200 #endif                                                  /* BB_DF || BB_MTAB */
1201
1202 #if defined BB_INIT || defined BB_SYSLOGD
1203 /* try to open up the specified device */
1204 extern int device_open(char *device, int mode)
1205 {
1206         int m, f, fd = -1;
1207
1208         m = mode | O_NONBLOCK;
1209
1210         /* Retry up to 5 times */
1211         for (f = 0; f < 5; f++)
1212                 if ((fd = open(device, m, 0600)) >= 0)
1213                         break;
1214         if (fd < 0)
1215                 return fd;
1216         /* Reset original flags. */
1217         if (m != mode)
1218                 fcntl(fd, F_SETFL, mode);
1219         return fd;
1220 }
1221 #endif                                                  /* BB_INIT BB_SYSLOGD */
1222
1223
1224 #if defined BB_KILLALL || ( defined BB_FEATURE_LINUXRC && ( defined BB_HALT || defined BB_REBOOT || defined BB_POWEROFF ))
1225 #ifdef BB_FEATURE_USE_DEVPS_PATCH
1226 #include <linux/devps.h> /* For Erik's nifty devps device driver */
1227 #endif
1228
1229 #if defined BB_FEATURE_USE_DEVPS_PATCH
1230 /* find_pid_by_name()
1231  *
1232  *  This finds the pid of the specified process,
1233  *  by using the /dev/ps device driver.
1234  *
1235  *  Returns a list of all matching PIDs
1236  */
1237 extern pid_t* find_pid_by_name( char* pidName)
1238 {
1239         int fd, i, j;
1240         char device[] = "/dev/ps";
1241         pid_t num_pids;
1242         pid_t* pid_array = NULL;
1243         pid_t* pidList=NULL;
1244
1245         /* open device */
1246         fd = open(device, O_RDONLY);
1247         if (fd < 0)
1248                 perror_msg_and_die("open failed for `%s'", device);
1249
1250         /* Find out how many processes there are */
1251         if (ioctl (fd, DEVPS_GET_NUM_PIDS, &num_pids)<0)
1252                 perror_msg_and_die("\nDEVPS_GET_PID_LIST");
1253
1254         /* Allocate some memory -- grab a few extras just in case
1255          * some new processes start up while we wait. The kernel will
1256          * just ignore any extras if we give it too many, and will trunc.
1257          * the list if we give it too few.  */
1258         pid_array = (pid_t*) xcalloc( num_pids+10, sizeof(pid_t));
1259         pid_array[0] = num_pids+10;
1260
1261         /* Now grab the pid list */
1262         if (ioctl (fd, DEVPS_GET_PID_LIST, pid_array)<0)
1263                 perror_msg_and_die("\nDEVPS_GET_PID_LIST");
1264
1265         /* Now search for a match */
1266         for (i=1, j=0; i<pid_array[0] ; i++) {
1267                 char* p;
1268                 struct pid_info info;
1269
1270             info.pid = pid_array[i];
1271             if (ioctl (fd, DEVPS_GET_PID_INFO, &info)<0)
1272                         perror_msg_and_die("\nDEVPS_GET_PID_INFO");
1273
1274                 /* Make sure we only match on the process name */
1275                 p=info.command_line+1;
1276                 while ((*p != 0) && !isspace(*(p)) && (*(p-1) != '\\')) {
1277                         (p)++;
1278                 }
1279                 if (isspace(*(p)))
1280                                 *p='\0';
1281
1282                 if ((strstr(info.command_line, pidName) != NULL)
1283                                 && (strlen(pidName) == strlen(info.command_line))) {
1284                         pidList=xrealloc( pidList, sizeof(pid_t) * (j+2));
1285                         pidList[j++]=info.pid;
1286                 }
1287         }
1288         if (pidList)
1289                 pidList[j]=0;
1290
1291         /* Free memory */
1292         free( pid_array);
1293
1294         /* close device */
1295         if (close (fd) != 0)
1296                 perror_msg_and_die("close failed for `%s'", device);
1297
1298         return pidList;
1299 }
1300 #else           /* BB_FEATURE_USE_DEVPS_PATCH */
1301
1302 /* find_pid_by_name()
1303  *
1304  *  This finds the pid of the specified process.
1305  *  Currently, it's implemented by rummaging through
1306  *  the proc filesystem.
1307  *
1308  *  Returns a list of all matching PIDs
1309  */
1310 extern pid_t* find_pid_by_name( char* pidName)
1311 {
1312         DIR *dir;
1313         struct dirent *next;
1314         pid_t* pidList=NULL;
1315         int i=0;
1316
1317         dir = opendir("/proc");
1318         if (!dir)
1319                 perror_msg_and_die("Cannot open /proc");
1320
1321         while ((next = readdir(dir)) != NULL) {
1322                 FILE *status;
1323                 char filename[256];
1324                 char buffer[256];
1325
1326                 /* If it isn't a number, we don't want it */
1327                 if (!isdigit(*next->d_name))
1328                         continue;
1329
1330                 sprintf(filename, "/proc/%s/cmdline", next->d_name);
1331                 status = fopen(filename, "r");
1332                 if (!status) {
1333                         continue;
1334                 }
1335                 fgets(buffer, 256, status);
1336                 fclose(status);
1337
1338                 if (strstr(get_last_path_component(buffer), pidName) != NULL) {
1339                         pidList=xrealloc( pidList, sizeof(pid_t) * (i+2));
1340                         pidList[i++]=strtol(next->d_name, NULL, 0);
1341                 }
1342         }
1343
1344         if (pidList)
1345                 pidList[i]=0;
1346         return pidList;
1347 }
1348 #endif                                                  /* BB_FEATURE_USE_DEVPS_PATCH */
1349 #endif                                                  /* BB_KILLALL || ( BB_FEATURE_LINUXRC && ( BB_HALT || BB_REBOOT || BB_POWEROFF )) */
1350
1351 #ifndef DMALLOC
1352 /* this should really be farmed out to libbusybox.a */
1353 extern void *xmalloc(size_t size)
1354 {
1355         void *ptr = malloc(size);
1356
1357         if (!ptr)
1358                 error_msg_and_die(memory_exhausted);
1359         return ptr;
1360 }
1361
1362 extern void *xrealloc(void *old, size_t size)
1363 {
1364         void *ptr = realloc(old, size);
1365         if (!ptr)
1366                 error_msg_and_die(memory_exhausted);
1367         return ptr;
1368 }
1369
1370 extern void *xcalloc(size_t nmemb, size_t size)
1371 {
1372         void *ptr = calloc(nmemb, size);
1373         if (!ptr)
1374                 error_msg_and_die(memory_exhausted);
1375         return ptr;
1376 }
1377 #endif
1378
1379 #if defined BB_NFSMOUNT || defined BB_LS || defined BB_SH || \
1380         defined BB_WGET || defined BB_DPKG_DEB || defined BB_TAR || \
1381         defined BB_LN
1382 # ifndef DMALLOC
1383 extern char * xstrdup (const char *s) {
1384         char *t;
1385
1386         if (s == NULL)
1387                 return NULL;
1388
1389         t = strdup (s);
1390
1391         if (t == NULL)
1392                 error_msg_and_die(memory_exhausted);
1393
1394         return t;
1395 }
1396 # endif
1397 #endif
1398
1399 #if defined BB_NFSMOUNT
1400 extern char * xstrndup (const char *s, int n) {
1401         char *t;
1402
1403         if (s == NULL)
1404                 error_msg_and_die("xstrndup bug");
1405
1406         t = xmalloc(++n);
1407
1408         return safe_strncpy(t,s,n);
1409 }
1410 #endif
1411
1412 #if defined BB_IFCONFIG || defined BB_ROUTE || defined BB_NFSMOUNT || \
1413     defined BB_FEATURE_MOUNT_LOOP
1414 /* Like strncpy but make sure the resulting string is always 0 terminated. */
1415 extern char * safe_strncpy(char *dst, const char *src, size_t size)
1416 {
1417         dst[size-1] = '\0';
1418         return strncpy(dst, src, size-1);
1419 }
1420 #endif
1421
1422 #if (__GLIBC__ < 2) && (defined BB_SYSLOGD || defined BB_INIT)
1423 extern int vdprintf(int d, const char *format, va_list ap)
1424 {
1425         char buf[BUF_SIZE];
1426         int len;
1427
1428         len = vsprintf(buf, format, ap);
1429         return write(d, buf, len);
1430 }
1431 #endif                                                  /* BB_SYSLOGD */
1432
1433
1434 #if defined BB_FEATURE_MOUNT_LOOP
1435 #include <fcntl.h>
1436 #include "loop.h" /* Pull in loop device support */
1437
1438 extern int del_loop(const char *device)
1439 {
1440         int fd;
1441
1442         if ((fd = open(device, O_RDONLY)) < 0) {
1443                 perror_msg("%s", device);
1444                 return (FALSE);
1445         }
1446         if (ioctl(fd, LOOP_CLR_FD, 0) < 0) {
1447                 perror_msg("ioctl: LOOP_CLR_FD");
1448                 return (FALSE);
1449         }
1450         close(fd);
1451         return (TRUE);
1452 }
1453
1454 extern int set_loop(const char *device, const char *file, int offset,
1455                                         int *loopro)
1456 {
1457         struct loop_info loopinfo;
1458         int fd, ffd, mode;
1459
1460         mode = *loopro ? O_RDONLY : O_RDWR;
1461         if ((ffd = open(file, mode)) < 0 && !*loopro
1462                 && (errno != EROFS || (ffd = open(file, mode = O_RDONLY)) < 0)) {
1463                 perror_msg("%s", file);
1464                 return 1;
1465         }
1466         if ((fd = open(device, mode)) < 0) {
1467                 close(ffd);
1468                 perror_msg("%s", device);
1469                 return 1;
1470         }
1471         *loopro = (mode == O_RDONLY);
1472
1473         memset(&loopinfo, 0, sizeof(loopinfo));
1474         safe_strncpy(loopinfo.lo_name, file, LO_NAME_SIZE);
1475
1476         loopinfo.lo_offset = offset;
1477
1478         loopinfo.lo_encrypt_key_size = 0;
1479         if (ioctl(fd, LOOP_SET_FD, ffd) < 0) {
1480                 perror_msg("ioctl: LOOP_SET_FD");
1481                 close(fd);
1482                 close(ffd);
1483                 return 1;
1484         }
1485         if (ioctl(fd, LOOP_SET_STATUS, &loopinfo) < 0) {
1486                 (void) ioctl(fd, LOOP_CLR_FD, 0);
1487                 perror_msg("ioctl: LOOP_SET_STATUS");
1488                 close(fd);
1489                 close(ffd);
1490                 return 1;
1491         }
1492         close(fd);
1493         close(ffd);
1494         return 0;
1495 }
1496
1497 extern char *find_unused_loop_device(void)
1498 {
1499         char dev[20];
1500         int i, fd;
1501         struct stat statbuf;
1502         struct loop_info loopinfo;
1503
1504         for (i = 0; i <= 7; i++) {
1505                 sprintf(dev, "/dev/loop%d", i);
1506                 if (stat(dev, &statbuf) == 0 && S_ISBLK(statbuf.st_mode)) {
1507                         if ((fd = open(dev, O_RDONLY)) >= 0) {
1508                                 if (ioctl(fd, LOOP_GET_STATUS, &loopinfo) != 0) {
1509                                         if (errno == ENXIO) {   /* probably free */
1510                                                 close(fd);
1511                                                 return strdup(dev);
1512                                         }
1513                                 }
1514                                 close(fd);
1515                         }
1516                 }
1517         }
1518         return NULL;
1519 }
1520 #endif                                                  /* BB_FEATURE_MOUNT_LOOP */
1521
1522 #if defined BB_MOUNT || defined BB_DF || ( defined BB_UMOUNT && ! defined BB_MTAB)
1523 extern int find_real_root_device_name(char* name)
1524 {
1525         DIR *dir;
1526         struct dirent *entry;
1527         struct stat statBuf, rootStat;
1528         char fileName[BUFSIZ];
1529
1530         if (stat("/", &rootStat) != 0) {
1531                 error_msg("could not stat '/'");
1532                 return( FALSE);
1533         }
1534
1535         dir = opendir("/dev");
1536         if (!dir) {
1537                 error_msg("could not open '/dev'");
1538                 return( FALSE);
1539         }
1540
1541         while((entry = readdir(dir)) != NULL) {
1542
1543                 /* Must skip ".." since that is "/", and so we
1544                  * would get a false positive on ".."  */
1545                 if (strcmp(entry->d_name, "..") == 0)
1546                         continue;
1547
1548                 snprintf( fileName, strlen(name)+1, "/dev/%s", entry->d_name);
1549
1550                 if (stat(fileName, &statBuf) != 0)
1551                         continue;
1552                 /* Some char devices have the same dev_t as block
1553                  * devices, so make sure this is a block device */
1554                 if (! S_ISBLK(statBuf.st_mode))
1555                         continue;
1556                 if (statBuf.st_rdev == rootStat.st_rdev) {
1557                         strcpy(name, fileName);
1558                         return ( TRUE);
1559                 }
1560         }
1561
1562         return( FALSE);
1563 }
1564 #endif
1565
1566
1567 /* get_line_from_file() - This function reads an entire line from a text file
1568  * up to a newline. It returns a malloc'ed char * which must be stored and
1569  * free'ed  by the caller. */
1570 extern char *get_line_from_file(FILE *file)
1571 {
1572         static const int GROWBY = 80; /* how large we will grow strings by */
1573
1574         int ch;
1575         int idx = 0;
1576         char *linebuf = NULL;
1577         int linebufsz = 0;
1578
1579         while (1) {
1580                 ch = fgetc(file);
1581                 if (ch == EOF)
1582                         break;
1583                 /* grow the line buffer as necessary */
1584                 while (idx > linebufsz-2)
1585                         linebuf = xrealloc(linebuf, linebufsz += GROWBY);
1586                 linebuf[idx++] = (char)ch;
1587                 if ((char)ch == '\n')
1588                         break;
1589         }
1590
1591         if (idx == 0)
1592                 return NULL;
1593
1594         linebuf[idx] = 0;
1595         return linebuf;
1596 }
1597
1598 #if defined BB_CAT
1599 extern void print_file(FILE *file)
1600 {
1601         int c;
1602
1603         while ((c = getc(file)) != EOF)
1604                 putc(c, stdout);
1605         fclose(file);
1606         fflush(stdout);
1607 }
1608
1609 extern int print_file_by_name(char *filename)
1610 {
1611         FILE *file;
1612         if ((file = wfopen(filename, "r")) == NULL)
1613                 return FALSE;
1614         print_file(file);
1615         return TRUE;
1616 }
1617 #endif /* BB_CAT */
1618
1619 #if defined BB_ECHO || defined BB_SH || defined BB_TR
1620 char process_escape_sequence(char **ptr)
1621 {
1622        static const char charmap[] = {
1623                    'a',  'b',  'f',  'n',  'r',  't',  'v',  '\\', 0,
1624                '\a', '\b', '\f', '\n', '\r', '\t', '\v', '\\', '\\' };
1625
1626        const char *p;
1627            char *q;
1628        int num_digits;
1629        unsigned int n;
1630
1631        n = 0;
1632            q = *ptr;
1633
1634        for ( num_digits = 0 ; num_digits < 3 ; ++num_digits) {
1635                if ((*q < '0') || (*q > '7')) { /* not a digit? */
1636                        break;
1637                }
1638                n = n * 8 + (*q++ - '0');
1639        }
1640
1641        if (num_digits == 0) {   /* mnemonic escape sequence? */
1642                    for (p=charmap ; *p ; p++) {
1643                            if (*p == *q) {
1644                                    q++;
1645                                    break;
1646                            }
1647                    }
1648                    n = *(p+(sizeof(charmap)/2));
1649            }
1650
1651            /* doesn't hurt to fall through to here from mnemonic case */
1652            if (n > UCHAR_MAX) { /* is octal code too big for a char? */
1653                    n /= 8;                      /* adjust value and */
1654                    --q;                         /* back up one char */
1655            }
1656
1657            *ptr = q;
1658            return (char) n;
1659 }
1660 #endif
1661
1662 #if defined BB_BASENAME || defined BB_LN || defined BB_SH || defined BB_INIT || \
1663         ! defined BB_FEATURE_USE_DEVPS_PATCH || defined BB_WGET
1664 char *get_last_path_component(char *path)
1665 {
1666         char *s=path+strlen(path)-1;
1667
1668         /* strip trailing slashes */
1669         while (s != path && *s == '/') {
1670                 *s-- = '\0';
1671         }
1672
1673         /* find last component */
1674         s = strrchr(path, '/');
1675         if (s == NULL || s[1] == '\0')
1676                 return path;
1677         else
1678                 return s+1;
1679 }
1680 #endif
1681
1682 #if defined BB_GREP || defined BB_SED
1683 #include <regex.h>
1684 void xregcomp(regex_t *preg, const char *regex, int cflags)
1685 {
1686         int ret;
1687         if ((ret = regcomp(preg, regex, cflags)) != 0) {
1688                 int errmsgsz = regerror(ret, preg, NULL, 0);
1689                 char *errmsg = xmalloc(errmsgsz);
1690                 regerror(ret, preg, errmsg, errmsgsz);
1691                 error_msg_and_die("xregcomp: %s", errmsg);
1692         }
1693 }
1694 #endif
1695
1696 #if defined BB_CAT || defined BB_HEAD || defined BB_WC
1697 FILE *wfopen(const char *path, const char *mode)
1698 {
1699         FILE *fp;
1700         if ((fp = fopen(path, mode)) == NULL) {
1701                 perror_msg("%s", path);
1702                 errno = 0;
1703         }
1704         return fp;
1705 }
1706 #endif
1707
1708 #if defined BB_HOSTNAME || defined BB_LOADACM || defined BB_MORE \
1709  || defined BB_SED || defined BB_SH || defined BB_TAR || defined BB_UNIQ \
1710  || defined BB_WC || defined BB_CMP || defined BB_SORT || defined BB_WGET \
1711  || defined BB_MOUNT || defined BB_ROUTE
1712 FILE *xfopen(const char *path, const char *mode)
1713 {
1714         FILE *fp;
1715         if ((fp = fopen(path, mode)) == NULL)
1716                 perror_msg_and_die("%s", path);
1717         return fp;
1718 }
1719 #endif
1720
1721 static int applet_name_compare(const void *x, const void *y)
1722 {
1723         const char *name = x;
1724         const struct BB_applet *applet = y;
1725
1726         return strcmp(name, applet->name);
1727 }
1728
1729 extern size_t NUM_APPLETS;
1730
1731 struct BB_applet *find_applet_by_name(const char *name)
1732 {
1733         return bsearch(name, applets, NUM_APPLETS, sizeof(struct BB_applet),
1734                         applet_name_compare);
1735 }
1736
1737 void run_applet_by_name(const char *name, int argc, char **argv)
1738 {
1739         /* Do a binary search to find the applet entry given the name. */
1740         if ((applet_using = find_applet_by_name(name)) != NULL) {
1741                 applet_name = applet_using->name;
1742                 if (argv[1] && strcmp(argv[1], "--help") == 0)
1743                         show_usage();
1744                 exit((*(applet_using->main)) (argc, argv));
1745         }
1746 }
1747
1748 #if defined BB_DD || defined BB_TAIL || defined BB_STTY
1749 unsigned long parse_number(const char *numstr,
1750                 const struct suffix_mult *suffixes)
1751 {
1752         const struct suffix_mult *sm;
1753         unsigned long int ret;
1754         int len;
1755         char *end;
1756
1757         ret = strtoul(numstr, &end, 10);
1758         if (numstr == end)
1759                 error_msg_and_die("invalid number `%s'", numstr);
1760         while (end[0] != '\0') {
1761                 sm = suffixes;
1762                 while ( sm != 0 ) {
1763                         if(sm->suffix) {
1764                                 len = strlen(sm->suffix);
1765                                 if (strncmp(sm->suffix, end, len) == 0) {
1766                                         ret *= sm->mult;
1767                                         end += len;
1768                                         break;
1769                                 }
1770                         sm++;
1771
1772                         } else
1773                                 sm = 0;
1774                 }
1775                 if (sm == 0)
1776                         error_msg_and_die("invalid number `%s'", numstr);
1777         }
1778         return ret;
1779 }
1780 #endif
1781
1782 #if defined BB_DD || defined BB_NC || defined BB_TAIL
1783 ssize_t safe_read(int fd, void *buf, size_t count)
1784 {
1785         ssize_t n;
1786
1787         do {
1788                 n = read(fd, buf, count);
1789         } while (n < 0 && errno == EINTR);
1790
1791         return n;
1792 }
1793 #endif
1794
1795 #ifdef BB_FEATURE_HUMAN_READABLE
1796 const char *make_human_readable_str(unsigned long val, unsigned long hr)
1797 {
1798         int i=0;
1799         static char str[10] = "\0";
1800         static const char strings[] = { 'k', 'M', 'G', 'T', 0 };
1801         unsigned long divisor = 1;
1802
1803         if(val == 0)
1804                 return("0");
1805         if(hr)
1806                 snprintf(str, 9, "%ld", val/hr);
1807         else {
1808                 while(val >= divisor && i <= 4) {
1809                         divisor=divisor<<10, i++;
1810                 }
1811                 divisor=divisor>>10, i--;
1812                 snprintf(str, 9, "%.1Lf%c", (long double)(val)/divisor, strings[i]);
1813         }
1814         return(str);
1815 }
1816 #endif
1817
1818 #if defined(BB_GREP) || defined(BB_HOSTNAME) || defined(BB_SED) || defined(BB_TAR) || defined(BB_WGET) || defined(BB_XARGS) || defined(BB_SH)
1819 void chomp(char *s)
1820 {
1821         size_t len = strlen(s);
1822
1823         if (len == 0)
1824                 return;
1825
1826         if (s[len-1] == '\n')
1827                 s[len-1] = '\0';
1828 }
1829 #endif
1830
1831 #if defined(BB_SH)
1832 void trim(char *s)
1833 {
1834         /* trim trailing whitespace */
1835         while (isspace(s[strlen(s)-1]))
1836                 s[strlen(s)-1]='\0';
1837
1838         /* trim leading whitespace */
1839         memmove(s, &s[strspn(s, " \n\r\t\v")], strlen(s));
1840
1841 }
1842 #endif
1843
1844 /* END CODE */
1845 /*
1846 Local Variables:
1847 c-file-style: "linux"
1848 c-basic-offset: 4
1849 tab-width: 4
1850 End:
1851 */