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