1 /* $NetBSD: ns_ttl.c,v 1.8 2012/03/13 21:13:39 christos Exp $ */
2
3 /*
4 * Copyright (c) 2004 by Internet Systems Consortium, Inc. ("ISC")
5 * Copyright (c) 1996,1999 by Internet Software Consortium.
6 *
7 * Permission to use, copy, modify, and distribute this software for any
8 * purpose with or without fee is hereby granted, provided that the above
9 * copyright notice and this permission notice appear in all copies.
10 *
11 * THE SOFTWARE IS PROVIDED "AS IS" AND ISC DISCLAIMS ALL WARRANTIES
12 * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
13 * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL ISC BE LIABLE FOR
14 * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
15 * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
16 * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT
17 * OF OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
18 */
19
20 #include <sys/cdefs.h>
21 #ifndef lint
22 #ifdef notdef
23 static const char rcsid[] = "Id: ns_ttl.c,v 1.4 2005/07/28 06:51:49 marka Exp";
24 #else
25 __RCSID("$NetBSD: ns_ttl.c,v 1.8 2012/03/13 21:13:39 christos Exp $");
26 #endif
27 #endif
28
29 /* Import. */
30
31 #include <arpa/nameser.h>
32
33 #include <assert.h>
34 #include <ctype.h>
35 #include <errno.h>
36 #include <stdio.h>
37 #include <string.h>
38
39 /* Forward. */
40
41 static int fmt1(int t, char s, char **buf, size_t *buflen);
42
43 /* Macros. */
44
45 #define T(x) do { if ((x) < 0) return (-1); } while(0)
46
47 /* Public. */
48
49 int
ns_format_ttl(u_long src,char * dst,size_t dstlen)50 ns_format_ttl(u_long src, char *dst, size_t dstlen) {
51 char *odst = dst;
52 int secs, mins, hours, days, weeks, x;
53 char *p;
54
55 secs = (int)(src % 60); src /= 60;
56 mins = (int)(src % 60); src /= 60;
57 hours = (int)(src % 24); src /= 24;
58 days = (int)(src % 7); src /= 7;
59 weeks = (int)src; src = 0;
60
61 x = 0;
62 if (weeks) {
63 T(fmt1(weeks, 'W', &dst, &dstlen));
64 x++;
65 }
66 if (days) {
67 T(fmt1(days, 'D', &dst, &dstlen));
68 x++;
69 }
70 if (hours) {
71 T(fmt1(hours, 'H', &dst, &dstlen));
72 x++;
73 }
74 if (mins) {
75 T(fmt1(mins, 'M', &dst, &dstlen));
76 x++;
77 }
78 if (secs || !(weeks || days || hours || mins)) {
79 T(fmt1(secs, 'S', &dst, &dstlen));
80 x++;
81 }
82
83 if (x > 1) {
84 int ch;
85
86 for (p = odst; (ch = *p) != '\0'; p++)
87 if (isascii(ch) && isupper(ch))
88 *p = tolower(ch);
89 }
90
91 _DIAGASSERT(__type_fit(int, dst - odst));
92 return (int)(dst - odst);
93 }
94
95 #ifndef _LIBC
96 int
ns_parse_ttl(const char * src,u_long * dst)97 ns_parse_ttl(const char *src, u_long *dst) {
98 u_long ttl, tmp;
99 int ch, digits, dirty;
100
101 ttl = 0;
102 tmp = 0;
103 digits = 0;
104 dirty = 0;
105 while ((ch = *src++) != '\0') {
106 if (!isascii(ch) || !isprint(ch))
107 goto einval;
108 if (isdigit(ch)) {
109 tmp *= 10;
110 tmp += (ch - '0');
111 digits++;
112 continue;
113 }
114 if (digits == 0)
115 goto einval;
116 if (islower(ch))
117 ch = toupper(ch);
118 switch (ch) {
119 case 'W': tmp *= 7; /*FALLTHROUGH*/
120 case 'D': tmp *= 24; /*FALLTHROUGH*/
121 case 'H': tmp *= 60; /*FALLTHROUGH*/
122 case 'M': tmp *= 60; /*FALLTHROUGH*/
123 case 'S': break;
124 default: goto einval;
125 }
126 ttl += tmp;
127 tmp = 0;
128 digits = 0;
129 dirty = 1;
130 }
131 if (digits > 0) {
132 if (dirty)
133 goto einval;
134 else
135 ttl += tmp;
136 } else if (!dirty)
137 goto einval;
138 *dst = ttl;
139 return (0);
140
141 einval:
142 errno = EINVAL;
143 return (-1);
144 }
145 #endif
146
147 /* Private. */
148
149 static int
fmt1(int t,char s,char ** buf,size_t * buflen)150 fmt1(int t, char s, char **buf, size_t *buflen) {
151 char tmp[50];
152 size_t len;
153
154 len = (size_t)snprintf(tmp, sizeof(tmp), "%d%c", t, s);
155 if ((int)len < 0 || len + 1 > *buflen)
156 return (-1);
157 strcpy(*buf, tmp);
158 *buf += len;
159 *buflen -= len;
160 return (0);
161 }
162