Remove the code that handles the %r format, since it isn't used
[ppp.git] / pppd / utils.c
1 /*
2  * utils.c - various utility functions used in pppd.
3  *
4  * Copyright (c) 1999-2002 Paul Mackerras. All rights reserved.
5  *
6  * Redistribution and use in source and binary forms, with or without
7  * modification, are permitted provided that the following conditions
8  * are met:
9  *
10  * 1. Redistributions of source code must retain the above copyright
11  *    notice, this list of conditions and the following disclaimer.
12  *
13  * 2. Redistributions in binary form must reproduce the above copyright
14  *    notice, this list of conditions and the following disclaimer in
15  *    the documentation and/or other materials provided with the
16  *    distribution.
17  *
18  * 3. The name(s) of the authors of this software must not be used to
19  *    endorse or promote products derived from this software without
20  *    prior written permission.
21  *
22  * 4. Redistributions of any form whatsoever must retain the following
23  *    acknowledgment:
24  *    "This product includes software developed by Paul Mackerras
25  *     <paulus@samba.org>".
26  *
27  * THE AUTHORS OF THIS SOFTWARE DISCLAIM ALL WARRANTIES WITH REGARD TO
28  * THIS SOFTWARE, INCLUDING ALL IMPLIED WARRANTIES OF MERCHANTABILITY
29  * AND FITNESS, IN NO EVENT SHALL THE AUTHORS BE LIABLE FOR ANY
30  * SPECIAL, INDIRECT OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
31  * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN
32  * AN ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING
33  * OUT OF OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
34  */
35
36 #define RCSID   "$Id: utils.c,v 1.23 2004/01/13 04:13:08 paulus Exp $"
37
38 #include <stdio.h>
39 #include <ctype.h>
40 #include <stdlib.h>
41 #include <string.h>
42 #include <unistd.h>
43 #include <signal.h>
44 #include <errno.h>
45 #include <fcntl.h>
46 #include <syslog.h>
47 #include <netdb.h>
48 #include <time.h>
49 #include <utmp.h>
50 #include <pwd.h>
51 #include <sys/param.h>
52 #include <sys/types.h>
53 #include <sys/wait.h>
54 #include <sys/time.h>
55 #include <sys/resource.h>
56 #include <sys/stat.h>
57 #include <sys/socket.h>
58 #include <netinet/in.h>
59 #ifdef SVR4
60 #include <sys/mkdev.h>
61 #endif
62
63 #include "pppd.h"
64 #include "fsm.h"
65 #include "lcp.h"
66
67 static const char rcsid[] = RCSID;
68
69 #if defined(SUNOS4)
70 extern char *strerror();
71 #endif
72
73 static void logit __P((int, char *, va_list));
74 static void log_write __P((int, char *));
75 static void vslp_printer __P((void *, char *, ...));
76 static void format_packet __P((u_char *, int, void (*) (void *, char *, ...),
77                                void *));
78
79 struct buffer_info {
80     char *ptr;
81     int len;
82 };
83
84 /*
85  * strlcpy - like strcpy/strncpy, doesn't overflow destination buffer,
86  * always leaves destination null-terminated (for len > 0).
87  */
88 size_t
89 strlcpy(dest, src, len)
90     char *dest;
91     const char *src;
92     size_t len;
93 {
94     size_t ret = strlen(src);
95
96     if (len != 0) {
97         if (ret < len)
98             strcpy(dest, src);
99         else {
100             strncpy(dest, src, len - 1);
101             dest[len-1] = 0;
102         }
103     }
104     return ret;
105 }
106
107 /*
108  * strlcat - like strcat/strncat, doesn't overflow destination buffer,
109  * always leaves destination null-terminated (for len > 0).
110  */
111 size_t
112 strlcat(dest, src, len)
113     char *dest;
114     const char *src;
115     size_t len;
116 {
117     size_t dlen = strlen(dest);
118
119     return dlen + strlcpy(dest + dlen, src, (len > dlen? len - dlen: 0));
120 }
121
122
123 /*
124  * slprintf - format a message into a buffer.  Like sprintf except we
125  * also specify the length of the output buffer, and we handle
126  * %m (error message), %v (visible string),
127  * %q (quoted string), %t (current time) and %I (IP address) formats.
128  * Doesn't do floating-point formats.
129  * Returns the number of chars put into buf.
130  */
131 int
132 slprintf __V((char *buf, int buflen, char *fmt, ...))
133 {
134     va_list args;
135     int n;
136
137 #if defined(__STDC__)
138     va_start(args, fmt);
139 #else
140     char *buf;
141     int buflen;
142     char *fmt;
143     va_start(args);
144     buf = va_arg(args, char *);
145     buflen = va_arg(args, int);
146     fmt = va_arg(args, char *);
147 #endif
148     n = vslprintf(buf, buflen, fmt, args);
149     va_end(args);
150     return n;
151 }
152
153 /*
154  * vslprintf - like slprintf, takes a va_list instead of a list of args.
155  */
156 #define OUTCHAR(c)      (buflen > 0? (--buflen, *buf++ = (c)): 0)
157
158 int
159 vslprintf(buf, buflen, fmt, args)
160     char *buf;
161     int buflen;
162     char *fmt;
163     va_list args;
164 {
165     int c, i, n;
166     int width, prec, fillch;
167     int base, len, neg, quoted;
168     unsigned long val = 0;
169     char *str, *f, *buf0;
170     unsigned char *p;
171     char num[32];
172     time_t t;
173     u_int32_t ip;
174     static char hexchars[] = "0123456789abcdef";
175     struct buffer_info bufinfo;
176
177     buf0 = buf;
178     --buflen;
179     while (buflen > 0) {
180         for (f = fmt; *f != '%' && *f != 0; ++f)
181             ;
182         if (f > fmt) {
183             len = f - fmt;
184             if (len > buflen)
185                 len = buflen;
186             memcpy(buf, fmt, len);
187             buf += len;
188             buflen -= len;
189             fmt = f;
190         }
191         if (*fmt == 0)
192             break;
193         c = *++fmt;
194         width = 0;
195         prec = -1;
196         fillch = ' ';
197         if (c == '0') {
198             fillch = '0';
199             c = *++fmt;
200         }
201         if (c == '*') {
202             width = va_arg(args, int);
203             c = *++fmt;
204         } else {
205             while (isdigit(c)) {
206                 width = width * 10 + c - '0';
207                 c = *++fmt;
208             }
209         }
210         if (c == '.') {
211             c = *++fmt;
212             if (c == '*') {
213                 prec = va_arg(args, int);
214                 c = *++fmt;
215             } else {
216                 prec = 0;
217                 while (isdigit(c)) {
218                     prec = prec * 10 + c - '0';
219                     c = *++fmt;
220                 }
221             }
222         }
223         str = 0;
224         base = 0;
225         neg = 0;
226         ++fmt;
227         switch (c) {
228         case 'l':
229             c = *fmt++;
230             switch (c) {
231             case 'd':
232                 val = va_arg(args, long);
233                 if (val < 0) {
234                     neg = 1;
235                     val = -val;
236                 }
237                 base = 10;
238                 break;
239             case 'u':
240                 val = va_arg(args, unsigned long);
241                 base = 10;
242                 break;
243             default:
244                 *buf++ = '%'; --buflen;
245                 *buf++ = 'l'; --buflen;
246                 --fmt;          /* so %lz outputs %lz etc. */
247                 continue;
248             }
249             break;
250         case 'd':
251             i = va_arg(args, int);
252             if (i < 0) {
253                 neg = 1;
254                 val = -i;
255             } else
256                 val = i;
257             base = 10;
258             break;
259         case 'u':
260             val = va_arg(args, unsigned int);
261             base = 10;
262             break;
263         case 'o':
264             val = va_arg(args, unsigned int);
265             base = 8;
266             break;
267         case 'x':
268         case 'X':
269             val = va_arg(args, unsigned int);
270             base = 16;
271             break;
272         case 'p':
273             val = (unsigned long) va_arg(args, void *);
274             base = 16;
275             neg = 2;
276             break;
277         case 's':
278             str = va_arg(args, char *);
279             break;
280         case 'c':
281             num[0] = va_arg(args, int);
282             num[1] = 0;
283             str = num;
284             break;
285         case 'm':
286             str = strerror(errno);
287             break;
288         case 'I':
289             ip = va_arg(args, u_int32_t);
290             ip = ntohl(ip);
291             slprintf(num, sizeof(num), "%d.%d.%d.%d", (ip >> 24) & 0xff,
292                      (ip >> 16) & 0xff, (ip >> 8) & 0xff, ip & 0xff);
293             str = num;
294             break;
295 #if 0   /* not used, and breaks on S/390, apparently */
296         case 'r':
297             f = va_arg(args, char *);
298 #ifndef __powerpc__
299             n = vslprintf(buf, buflen + 1, f, va_arg(args, va_list));
300 #else
301             /* On the powerpc, a va_list is an array of 1 structure */
302             n = vslprintf(buf, buflen + 1, f, va_arg(args, void *));
303 #endif
304             buf += n;
305             buflen -= n;
306             continue;
307 #endif
308         case 't':
309             time(&t);
310             str = ctime(&t);
311             str += 4;           /* chop off the day name */
312             str[15] = 0;        /* chop off year and newline */
313             break;
314         case 'v':               /* "visible" string */
315         case 'q':               /* quoted string */
316             quoted = c == 'q';
317             p = va_arg(args, unsigned char *);
318             if (fillch == '0' && prec >= 0) {
319                 n = prec;
320             } else {
321                 n = strlen((char *)p);
322                 if (prec >= 0 && n > prec)
323                     n = prec;
324             }
325             while (n > 0 && buflen > 0) {
326                 c = *p++;
327                 --n;
328                 if (!quoted && c >= 0x80) {
329                     OUTCHAR('M');
330                     OUTCHAR('-');
331                     c -= 0x80;
332                 }
333                 if (quoted && (c == '"' || c == '\\'))
334                     OUTCHAR('\\');
335                 if (c < 0x20 || (0x7f <= c && c < 0xa0)) {
336                     if (quoted) {
337                         OUTCHAR('\\');
338                         switch (c) {
339                         case '\t':      OUTCHAR('t');   break;
340                         case '\n':      OUTCHAR('n');   break;
341                         case '\b':      OUTCHAR('b');   break;
342                         case '\f':      OUTCHAR('f');   break;
343                         default:
344                             OUTCHAR('x');
345                             OUTCHAR(hexchars[c >> 4]);
346                             OUTCHAR(hexchars[c & 0xf]);
347                         }
348                     } else {
349                         if (c == '\t')
350                             OUTCHAR(c);
351                         else {
352                             OUTCHAR('^');
353                             OUTCHAR(c ^ 0x40);
354                         }
355                     }
356                 } else
357                     OUTCHAR(c);
358             }
359             continue;
360         case 'P':               /* print PPP packet */
361             bufinfo.ptr = buf;
362             bufinfo.len = buflen + 1;
363             p = va_arg(args, unsigned char *);
364             n = va_arg(args, int);
365             format_packet(p, n, vslp_printer, &bufinfo);
366             buf = bufinfo.ptr;
367             buflen = bufinfo.len - 1;
368             continue;
369         case 'B':
370             p = va_arg(args, unsigned char *);
371             for (n = prec; n > 0; --n) {
372                 c = *p++;
373                 if (fillch == ' ')
374                     OUTCHAR(' ');
375                 OUTCHAR(hexchars[(c >> 4) & 0xf]);
376                 OUTCHAR(hexchars[c & 0xf]);
377             }
378             continue;
379         default:
380             *buf++ = '%';
381             if (c != '%')
382                 --fmt;          /* so %z outputs %z etc. */
383             --buflen;
384             continue;
385         }
386         if (base != 0) {
387             str = num + sizeof(num);
388             *--str = 0;
389             while (str > num + neg) {
390                 *--str = hexchars[val % base];
391                 val = val / base;
392                 if (--prec <= 0 && val == 0)
393                     break;
394             }
395             switch (neg) {
396             case 1:
397                 *--str = '-';
398                 break;
399             case 2:
400                 *--str = 'x';
401                 *--str = '0';
402                 break;
403             }
404             len = num + sizeof(num) - 1 - str;
405         } else {
406             len = strlen(str);
407             if (prec >= 0 && len > prec)
408                 len = prec;
409         }
410         if (width > 0) {
411             if (width > buflen)
412                 width = buflen;
413             if ((n = width - len) > 0) {
414                 buflen -= n;
415                 for (; n > 0; --n)
416                     *buf++ = fillch;
417             }
418         }
419         if (len > buflen)
420             len = buflen;
421         memcpy(buf, str, len);
422         buf += len;
423         buflen -= len;
424     }
425     *buf = 0;
426     return buf - buf0;
427 }
428
429 /*
430  * vslp_printer - used in processing a %P format
431  */
432 static void
433 vslp_printer __V((void *arg, char *fmt, ...))
434 {
435     int n;
436     va_list pvar;
437     struct buffer_info *bi;
438
439 #if defined(__STDC__)
440     va_start(pvar, fmt);
441 #else
442     void *arg;
443     char *fmt;
444     va_start(pvar);
445     arg = va_arg(pvar, void *);
446     fmt = va_arg(pvar, char *);
447 #endif
448
449     bi = (struct buffer_info *) arg;
450     n = vslprintf(bi->ptr, bi->len, fmt, pvar);
451     va_end(pvar);
452
453     bi->ptr += n;
454     bi->len -= n;
455 }
456
457 #ifdef unused
458 /*
459  * log_packet - format a packet and log it.
460  */
461
462 void
463 log_packet(p, len, prefix, level)
464     u_char *p;
465     int len;
466     char *prefix;
467     int level;
468 {
469         init_pr_log(prefix, level);
470         format_packet(p, len, pr_log, &level);
471         end_pr_log();
472 }
473 #endif /* unused */
474
475 /*
476  * format_packet - make a readable representation of a packet,
477  * calling `printer(arg, format, ...)' to output it.
478  */
479 static void
480 format_packet(p, len, printer, arg)
481     u_char *p;
482     int len;
483     void (*printer) __P((void *, char *, ...));
484     void *arg;
485 {
486     int i, n;
487     u_short proto;
488     struct protent *protp;
489
490     if (len >= PPP_HDRLEN && p[0] == PPP_ALLSTATIONS && p[1] == PPP_UI) {
491         p += 2;
492         GETSHORT(proto, p);
493         len -= PPP_HDRLEN;
494         for (i = 0; (protp = protocols[i]) != NULL; ++i)
495             if (proto == protp->protocol)
496                 break;
497         if (protp != NULL) {
498             printer(arg, "[%s", protp->name);
499             n = (*protp->printpkt)(p, len, printer, arg);
500             printer(arg, "]");
501             p += n;
502             len -= n;
503         } else {
504             for (i = 0; (protp = protocols[i]) != NULL; ++i)
505                 if (proto == (protp->protocol & ~0x8000))
506                     break;
507             if (protp != 0 && protp->data_name != 0) {
508                 printer(arg, "[%s data]", protp->data_name);
509                 if (len > 8)
510                     printer(arg, "%.8B ...", p);
511                 else
512                     printer(arg, "%.*B", len, p);
513                 len = 0;
514             } else
515                 printer(arg, "[proto=0x%x]", proto);
516         }
517     }
518
519     if (len > 32)
520         printer(arg, "%.32B ...", p);
521     else
522         printer(arg, "%.*B", len, p);
523 }
524
525 /*
526  * init_pr_log, end_pr_log - initialize and finish use of pr_log.
527  */
528
529 static char line[256];          /* line to be logged accumulated here */
530 static char *linep;             /* current pointer within line */
531 static int llevel;              /* level for logging */
532
533 void
534 init_pr_log(prefix, level)
535      char *prefix;
536      int level;
537 {
538         linep = line;
539         if (prefix != NULL) {
540                 strlcpy(line, prefix, sizeof(line));
541                 linep = line + strlen(line);
542         }
543         llevel = level;
544 }
545
546 void
547 end_pr_log()
548 {
549         if (linep != line) {
550                 *linep = 0;
551                 log_write(llevel, line);
552         }
553 }
554
555 /*
556  * pr_log - printer routine for outputting to syslog
557  */
558 void
559 pr_log __V((void *arg, char *fmt, ...))
560 {
561         int l, n;
562         va_list pvar;
563         char *p, *eol;
564         char buf[256];
565
566 #if defined(__STDC__)
567         va_start(pvar, fmt);
568 #else
569         void *arg;
570         char *fmt;
571         va_start(pvar);
572         arg = va_arg(pvar, void *);
573         fmt = va_arg(pvar, char *);
574 #endif
575
576         n = vslprintf(buf, sizeof(buf), fmt, pvar);
577         va_end(pvar);
578
579         p = buf;
580         eol = strchr(buf, '\n');
581         if (linep != line) {
582                 l = (eol == NULL)? n: eol - buf;
583                 if (linep + l < line + sizeof(line)) {
584                         if (l > 0) {
585                                 memcpy(linep, buf, l);
586                                 linep += l;
587                         }
588                         if (eol == NULL)
589                                 return;
590                         p = eol + 1;
591                         eol = strchr(p, '\n');
592                 }
593                 *linep = 0;
594                 log_write(llevel, line);
595                 linep = line;
596         }
597
598         while (eol != NULL) {
599                 *eol = 0;
600                 log_write(llevel, p);
601                 p = eol + 1;
602                 eol = strchr(p, '\n');
603         }
604
605         /* assumes sizeof(buf) <= sizeof(line) */
606         l = buf + n - p;
607         if (l > 0) {
608                 memcpy(line, p, n);
609                 linep = line + l;
610         }
611 }
612
613 /*
614  * print_string - print a readable representation of a string using
615  * printer.
616  */
617 void
618 print_string(p, len, printer, arg)
619     char *p;
620     int len;
621     void (*printer) __P((void *, char *, ...));
622     void *arg;
623 {
624     int c;
625
626     printer(arg, "\"");
627     for (; len > 0; --len) {
628         c = *p++;
629         if (' ' <= c && c <= '~') {
630             if (c == '\\' || c == '"')
631                 printer(arg, "\\");
632             printer(arg, "%c", c);
633         } else {
634             switch (c) {
635             case '\n':
636                 printer(arg, "\\n");
637                 break;
638             case '\r':
639                 printer(arg, "\\r");
640                 break;
641             case '\t':
642                 printer(arg, "\\t");
643                 break;
644             default:
645                 printer(arg, "\\%.3o", c);
646             }
647         }
648     }
649     printer(arg, "\"");
650 }
651
652 /*
653  * logit - does the hard work for fatal et al.
654  */
655 static void
656 logit(level, fmt, args)
657     int level;
658     char *fmt;
659     va_list args;
660 {
661     int n;
662     char buf[1024];
663
664     n = vslprintf(buf, sizeof(buf), fmt, args);
665     log_write(level, buf);
666 }
667
668 static void
669 log_write(level, buf)
670     int level;
671     char *buf;
672 {
673     syslog(level, "%s", buf);
674     if (log_to_fd >= 0 && (level != LOG_DEBUG || debug)) {
675         int n = strlen(buf);
676
677         if (n > 0 && buf[n-1] == '\n')
678             --n;
679         if (write(log_to_fd, buf, n) != n
680             || write(log_to_fd, "\n", 1) != 1)
681             log_to_fd = -1;
682     }
683 }
684
685 /*
686  * fatal - log an error message and die horribly.
687  */
688 void
689 fatal __V((char *fmt, ...))
690 {
691     va_list pvar;
692
693 #if defined(__STDC__)
694     va_start(pvar, fmt);
695 #else
696     char *fmt;
697     va_start(pvar);
698     fmt = va_arg(pvar, char *);
699 #endif
700
701     logit(LOG_ERR, fmt, pvar);
702     va_end(pvar);
703
704     die(1);                     /* as promised */
705 }
706
707 /*
708  * error - log an error message.
709  */
710 void
711 error __V((char *fmt, ...))
712 {
713     va_list pvar;
714
715 #if defined(__STDC__)
716     va_start(pvar, fmt);
717 #else
718     char *fmt;
719     va_start(pvar);
720     fmt = va_arg(pvar, char *);
721 #endif
722
723     logit(LOG_ERR, fmt, pvar);
724     va_end(pvar);
725     ++error_count;
726 }
727
728 /*
729  * warn - log a warning message.
730  */
731 void
732 warn __V((char *fmt, ...))
733 {
734     va_list pvar;
735
736 #if defined(__STDC__)
737     va_start(pvar, fmt);
738 #else
739     char *fmt;
740     va_start(pvar);
741     fmt = va_arg(pvar, char *);
742 #endif
743
744     logit(LOG_WARNING, fmt, pvar);
745     va_end(pvar);
746 }
747
748 /*
749  * notice - log a notice-level message.
750  */
751 void
752 notice __V((char *fmt, ...))
753 {
754     va_list pvar;
755
756 #if defined(__STDC__)
757     va_start(pvar, fmt);
758 #else
759     char *fmt;
760     va_start(pvar);
761     fmt = va_arg(pvar, char *);
762 #endif
763
764     logit(LOG_NOTICE, fmt, pvar);
765     va_end(pvar);
766 }
767
768 /*
769  * info - log an informational message.
770  */
771 void
772 info __V((char *fmt, ...))
773 {
774     va_list pvar;
775
776 #if defined(__STDC__)
777     va_start(pvar, fmt);
778 #else
779     char *fmt;
780     va_start(pvar);
781     fmt = va_arg(pvar, char *);
782 #endif
783
784     logit(LOG_INFO, fmt, pvar);
785     va_end(pvar);
786 }
787
788 /*
789  * dbglog - log a debug message.
790  */
791 void
792 dbglog __V((char *fmt, ...))
793 {
794     va_list pvar;
795
796 #if defined(__STDC__)
797     va_start(pvar, fmt);
798 #else
799     char *fmt;
800     va_start(pvar);
801     fmt = va_arg(pvar, char *);
802 #endif
803
804     logit(LOG_DEBUG, fmt, pvar);
805     va_end(pvar);
806 }
807
808 /*
809  * dump_packet - print out a packet in readable form if it is interesting.
810  * Assumes len >= PPP_HDRLEN.
811  */
812 void
813 dump_packet(const char *tag, unsigned char *p, int len)
814 {
815     int proto;
816
817     if (!debug)
818         return;
819
820     /*
821      * don't print LCP echo request/reply packets if debug <= 1
822      * and the link is up.
823      */
824     proto = (p[2] << 8) + p[3];
825     if (debug <= 1 && unsuccess == 0 && proto == PPP_LCP
826         && len >= PPP_HDRLEN + HEADERLEN) {
827         unsigned char *lcp = p + PPP_HDRLEN;
828         int l = (lcp[2] << 8) + lcp[3];
829
830         if ((lcp[0] == ECHOREQ || lcp[0] == ECHOREP)
831             && l >= HEADERLEN && l <= len - PPP_HDRLEN)
832             return;
833     }
834
835     dbglog("%s %P", tag, p, len);
836 }
837
838 /*
839  * complete_read - read a full `count' bytes from fd,
840  * unless end-of-file or an error other than EINTR is encountered.
841  */
842 ssize_t
843 complete_read(int fd, void *buf, size_t count)
844 {
845         size_t done;
846         ssize_t nb;
847         char *ptr = buf;
848
849         for (done = 0; done < count; ) {
850                 nb = read(fd, ptr, count - done);
851                 if (nb < 0) {
852                         if (errno == EINTR)
853                                 continue;
854                         return -1;
855                 }
856                 if (nb == 0)
857                         break;
858                 done += nb;
859                 ptr += nb;
860         }
861         return done;
862 }
863
864 /* Procedures for locking the serial device using a lock file. */
865 #ifndef LOCK_DIR
866 #ifdef __linux__
867 #define LOCK_DIR        "/var/lock"
868 #else
869 #ifdef SVR4
870 #define LOCK_DIR        "/var/spool/locks"
871 #else
872 #define LOCK_DIR        "/var/spool/lock"
873 #endif
874 #endif
875 #endif /* LOCK_DIR */
876
877 static char lock_file[MAXPATHLEN];
878
879 /*
880  * lock - create a lock file for the named device
881  */
882 int
883 lock(dev)
884     char *dev;
885 {
886 #ifdef LOCKLIB
887     int result;
888
889     result = mklock (dev, (void *) 0);
890     if (result == 0) {
891         strlcpy(lock_file, dev, sizeof(lock_file));
892         return 0;
893     }
894
895     if (result > 0)
896         notice("Device %s is locked by pid %d", dev, result);
897     else
898         error("Can't create lock file %s", lock_file);
899     return -1;
900
901 #else /* LOCKLIB */
902
903     char lock_buffer[12];
904     int fd, pid, n;
905
906 #ifdef SVR4
907     struct stat sbuf;
908
909     if (stat(dev, &sbuf) < 0) {
910         error("Can't get device number for %s: %m", dev);
911         return -1;
912     }
913     if ((sbuf.st_mode & S_IFMT) != S_IFCHR) {
914         error("Can't lock %s: not a character device", dev);
915         return -1;
916     }
917     slprintf(lock_file, sizeof(lock_file), "%s/LK.%03d.%03d.%03d",
918              LOCK_DIR, major(sbuf.st_dev),
919              major(sbuf.st_rdev), minor(sbuf.st_rdev));
920 #else
921     char *p;
922     char lockdev[MAXPATHLEN];
923
924     if ((p = strstr(dev, "dev/")) != NULL) {
925         dev = p + 4;
926         strncpy(lockdev, dev, MAXPATHLEN-1);
927         lockdev[MAXPATHLEN-1] = 0;
928         while ((p = strrchr(lockdev, '/')) != NULL) {
929             *p = '_';
930         }
931         dev = lockdev;
932     } else
933         if ((p = strrchr(dev, '/')) != NULL)
934             dev = p + 1;
935
936     slprintf(lock_file, sizeof(lock_file), "%s/LCK..%s", LOCK_DIR, dev);
937 #endif
938
939     while ((fd = open(lock_file, O_EXCL | O_CREAT | O_RDWR, 0644)) < 0) {
940         if (errno != EEXIST) {
941             error("Can't create lock file %s: %m", lock_file);
942             break;
943         }
944
945         /* Read the lock file to find out who has the device locked. */
946         fd = open(lock_file, O_RDONLY, 0);
947         if (fd < 0) {
948             if (errno == ENOENT) /* This is just a timing problem. */
949                 continue;
950             error("Can't open existing lock file %s: %m", lock_file);
951             break;
952         }
953 #ifndef LOCK_BINARY
954         n = read(fd, lock_buffer, 11);
955 #else
956         n = read(fd, &pid, sizeof(pid));
957 #endif /* LOCK_BINARY */
958         close(fd);
959         fd = -1;
960         if (n <= 0) {
961             error("Can't read pid from lock file %s", lock_file);
962             break;
963         }
964
965         /* See if the process still exists. */
966 #ifndef LOCK_BINARY
967         lock_buffer[n] = 0;
968         pid = atoi(lock_buffer);
969 #endif /* LOCK_BINARY */
970         if (pid == getpid())
971             return 1;           /* somebody else locked it for us */
972         if (pid == 0
973             || (kill(pid, 0) == -1 && errno == ESRCH)) {
974             if (unlink (lock_file) == 0) {
975                 notice("Removed stale lock on %s (pid %d)", dev, pid);
976                 continue;
977             }
978             warn("Couldn't remove stale lock on %s", dev);
979         } else
980             notice("Device %s is locked by pid %d", dev, pid);
981         break;
982     }
983
984     if (fd < 0) {
985         lock_file[0] = 0;
986         return -1;
987     }
988
989     pid = getpid();
990 #ifndef LOCK_BINARY
991     slprintf(lock_buffer, sizeof(lock_buffer), "%10d\n", pid);
992     write (fd, lock_buffer, 11);
993 #else
994     write(fd, &pid, sizeof (pid));
995 #endif
996     close(fd);
997     return 0;
998
999 #endif
1000 }
1001
1002 /*
1003  * relock - called to update our lockfile when we are about to detach,
1004  * thus changing our pid (we fork, the child carries on, and the parent dies).
1005  * Note that this is called by the parent, with pid equal to the pid
1006  * of the child.  This avoids a potential race which would exist if
1007  * we had the child rewrite the lockfile (the parent might die first,
1008  * and another process could think the lock was stale if it checked
1009  * between when the parent died and the child rewrote the lockfile).
1010  */
1011 int
1012 relock(pid)
1013     int pid;
1014 {
1015 #ifdef LOCKLIB
1016     /* XXX is there a way to do this? */
1017     return -1;
1018 #else /* LOCKLIB */
1019
1020     int fd;
1021     char lock_buffer[12];
1022
1023     if (lock_file[0] == 0)
1024         return -1;
1025     fd = open(lock_file, O_WRONLY, 0);
1026     if (fd < 0) {
1027         error("Couldn't reopen lock file %s: %m", lock_file);
1028         lock_file[0] = 0;
1029         return -1;
1030     }
1031
1032 #ifndef LOCK_BINARY
1033     slprintf(lock_buffer, sizeof(lock_buffer), "%10d\n", pid);
1034     write (fd, lock_buffer, 11);
1035 #else
1036     write(fd, &pid, sizeof(pid));
1037 #endif /* LOCK_BINARY */
1038     close(fd);
1039     return 0;
1040
1041 #endif /* LOCKLIB */
1042 }
1043
1044 /*
1045  * unlock - remove our lockfile
1046  */
1047 void
1048 unlock()
1049 {
1050     if (lock_file[0]) {
1051 #ifdef LOCKLIB
1052         (void) rmlock(lock_file, (void *) 0);
1053 #else
1054         unlink(lock_file);
1055 #endif
1056         lock_file[0] = 0;
1057     }
1058 }
1059