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