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