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