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