]> git.meshlink.io Git - catta/blob - util.c
8be7b8830b90e4c856f2ab93878c77cadd7d5e22
[catta] / util.c
1 /* $Id$ */
2
3 /***
4   This file is part of avahi.
5  
6   avahi is free software; you can redistribute it and/or modify it
7   under the terms of the GNU Lesser General Public License as
8   published by the Free Software Foundation; either version 2.1 of the
9   License, or (at your option) any later version.
10  
11   avahi is distributed in the hope that it will be useful, but WITHOUT
12   ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
13   or FITNESS FOR A PARTICULAR PURPOSE. See the GNU Lesser General
14   Public License for more details.
15  
16   You should have received a copy of the GNU Lesser General Public
17   License along with avahi; if not, write to the Free Software
18   Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307
19   USA.
20 ***/
21
22 #ifdef HAVE_CONFIG_H
23 #include <config.h>
24 #endif
25
26 #include <string.h>
27 #include <unistd.h>
28 #include <fcntl.h>
29 #include <errno.h>
30 #include <limits.h>
31 #include <stdio.h>
32 #include <ctype.h>
33 #include <stdlib.h>
34
35 #include "util.h"
36
37 gchar *avahi_get_host_name(void) {
38 #ifdef HOST_NAME_MAX
39     char t[HOST_NAME_MAX];
40 #else
41     char t[256];
42 #endif
43     gethostname(t, sizeof(t));
44     t[sizeof(t)-1] = 0;
45     return avahi_normalize_name(t);
46 }
47
48 static gchar *unescape_uneeded(const gchar *src, gchar *ret_dest, size_t size) {
49     gboolean escaped = FALSE;
50     
51     g_assert(src);
52     g_assert(ret_dest);
53     g_assert(size > 0);
54     
55     for (; *src; src++) {
56
57         if (!escaped && *src == '\\')
58             escaped = TRUE;
59         else if (escaped && (*src == '.' || *src == '\\')) {
60
61             if ((size -= 2) <= 1) break;
62             
63             *(ret_dest++) = '\\';
64             *(ret_dest++) = *src;
65             escaped = FALSE;
66         } else {
67             if (--size <= 1) break;
68
69             *(ret_dest++) = *src;
70             escaped = FALSE;
71         }
72
73     }
74
75     *ret_dest = 0;
76     
77     return ret_dest;
78 }
79
80 gchar *avahi_normalize_name(const gchar *s) {
81     gchar tmp[256];
82     guint l;
83     g_assert(s);
84
85     unescape_uneeded(s, tmp, sizeof(tmp));
86
87     l = strlen(tmp);
88
89     while (l > 0 && tmp[l-1] == '.')
90         tmp[--l] = 0;
91
92     return g_strdup(tmp);
93 }
94
95 gint avahi_timeval_compare(const GTimeVal *a, const GTimeVal *b) {
96     g_assert(a);
97     g_assert(b);
98
99     if (a->tv_sec < b->tv_sec)
100         return -1;
101
102     if (a->tv_sec > b->tv_sec)
103         return 1;
104
105     if (a->tv_usec < b->tv_usec)
106         return -1;
107
108     if (a->tv_usec > b->tv_usec)
109         return 1;
110
111     return 0;
112 }
113
114 AvahiUsec avahi_timeval_diff(const GTimeVal *a, const GTimeVal *b) {
115     g_assert(a);
116     g_assert(b);
117
118     if (avahi_timeval_compare(a, b) < 0)
119         return - avahi_timeval_diff(b, a);
120
121     return ((AvahiUsec) a->tv_sec - b->tv_sec)*1000000 + a->tv_usec - b->tv_usec;
122 }
123
124 GTimeVal* avahi_timeval_add(GTimeVal *a, AvahiUsec usec) {
125     AvahiUsec u;
126     g_assert(a);
127
128     u = usec + a->tv_usec;
129
130     if (u < 0) {
131         a->tv_usec = (glong) (1000000 + (u % 1000000));
132         a->tv_sec += (glong) (-1 + (u / 1000000));
133     } else {
134         a->tv_usec = (glong) (u % 1000000);
135         a->tv_sec += (glong) (u / 1000000);
136     }
137
138     return a;
139 }
140
141 AvahiUsec avahi_age(const GTimeVal *a) {
142     GTimeVal now;
143     
144     g_assert(a);
145
146     g_get_current_time(&now);
147
148     return avahi_timeval_diff(&now, a);
149 }
150
151 GTimeVal *avahi_elapse_time(GTimeVal *tv, guint msec, guint jitter) {
152     g_assert(tv);
153
154     g_get_current_time(tv);
155
156     if (msec)
157         avahi_timeval_add(tv, (AvahiUsec) msec*1000);
158
159     if (jitter)
160         avahi_timeval_add(tv, (AvahiUsec) g_random_int_range(0, jitter) * 1000);
161         
162     return tv;
163 }
164
165 gint avahi_set_cloexec(gint fd) {
166     gint n;
167
168     g_assert(fd >= 0);
169     
170     if ((n = fcntl(fd, F_GETFD)) < 0)
171         return -1;
172
173     if (n & FD_CLOEXEC)
174         return 0;
175
176     return fcntl(fd, F_SETFD, n|FD_CLOEXEC);
177 }
178
179 gint avahi_set_nonblock(gint fd) {
180     gint n;
181
182     g_assert(fd >= 0);
183
184     if ((n = fcntl(fd, F_GETFL)) < 0)
185         return -1;
186
187     if (n & O_NONBLOCK)
188         return 0;
189
190     return fcntl(fd, F_SETFL, n|O_NONBLOCK);
191 }
192
193 gint avahi_wait_for_write(gint fd) {
194     fd_set fds;
195     gint r;
196     
197     FD_ZERO(&fds);
198     FD_SET(fd, &fds);
199     
200     if ((r = select(fd+1, NULL, &fds, NULL, NULL)) < 0) {
201         g_message("select() failed: %s", strerror(errno));
202
203         return -1;
204     }
205     
206     g_assert(r > 0);
207
208     return 0;
209 }
210
211 /* Read the first label from string *name, unescape "\" and write it to dest */
212 gchar *avahi_unescape_label(const gchar **name, gchar *dest, guint size) {
213     guint i = 0;
214     gchar *d;
215     
216     g_assert(dest);
217     g_assert(size > 0);
218     g_assert(name);
219
220     if (!**name)
221         return NULL;
222
223     d = dest;
224     
225     for (;;) {
226         if (i >= size)
227             return NULL;
228
229         if (**name == '.') {
230             (*name)++;
231             break;
232         }
233         
234         if (**name == 0)
235             break;
236         
237         if (**name == '\\') {
238             (*name) ++;
239             
240             if (**name == 0)
241                 break;
242         }
243         
244         *(d++) = *((*name) ++);
245         i++;
246     }
247
248     g_assert(i < size);
249
250     *d = 0;
251
252     return dest;
253 }
254
255 /* Escape "\" and ".", append \0 */
256 gchar *avahi_escape_label(const guint8* src, guint src_length, gchar **ret_name, guint *ret_size) {
257     gchar *r;
258
259     g_assert(src);
260     g_assert(ret_name);
261     g_assert(*ret_name);
262     g_assert(ret_size);
263     g_assert(*ret_size > 0);
264
265     r = *ret_name;
266
267     while (src_length > 0) {
268         if (*src == '.' || *src == '\\') {
269             if (*ret_size < 3)
270                 return NULL;
271             
272             *((*ret_name) ++) = '\\';
273             (*ret_size) --;
274         }
275
276         if (*ret_size < 2)
277             return NULL;
278         
279         *((*ret_name)++) = *src;
280         (*ret_size) --;
281
282         src_length --;
283         src++;
284     }
285
286     **ret_name = 0;
287
288     return r;
289 }
290
291 gboolean avahi_domain_equal(const gchar *a, const gchar *b) {
292     g_assert(a);
293     g_assert(b);
294
295     if (a == b)
296         return TRUE;
297     
298     for (;;) {
299         gchar ca[65], cb[65], *pa, *pb;
300
301         pa = avahi_unescape_label(&a, ca, sizeof(ca));
302         pb = avahi_unescape_label(&b, cb, sizeof(cb));
303
304         if (!pa && !pb)
305             return TRUE;
306         else if ((pa && !pb) || (!pa && pb))
307             return FALSE;
308
309         if (g_ascii_strcasecmp(pa, pb))
310             return FALSE;
311     }
312
313     return TRUE;
314 }
315
316 gint avahi_binary_domain_cmp(const gchar *a, const gchar *b) {
317     g_assert(a);
318     g_assert(b);
319
320     if (a == b)
321         return 0;
322
323     for (;;) {
324         gchar ca[65], cb[65], *pa, *pb;
325         gint r;
326
327         pa = avahi_unescape_label(&a, ca, sizeof(ca));
328         pb = avahi_unescape_label(&b, cb, sizeof(cb));
329
330         if (!pa && !pb)
331             return 0;
332         else if (pa && !pb)
333             return 1;
334         else if (!pa && pb)
335             return -1;
336         
337         if ((r = strcmp(pa, pb)))
338             return r;
339     }
340 }
341
342 void avahi_hexdump(gconstpointer p, guint size) {
343     const guint8 *c = p;
344     g_assert(p);
345
346     printf("Dumping %u bytes from %p:\n", size, p);
347     
348     while (size > 0) {
349         guint i;
350
351         for (i = 0; i < 16; i++) { 
352             if (i < size)
353                 printf("%02x ", c[i]);
354             else
355                 printf("   ");
356         }
357
358         for (i = 0; i < 16; i++) {
359             if (i < size)
360                 printf("%c", c[i] >= 32 && c[i] < 127 ? c[i] : '.');
361             else
362                 printf(" ");
363         }
364         
365         printf("\n");
366
367         c += 16;
368
369         if (size <= 16)
370             break;
371         
372         size -= 16;
373     }
374 }
375
376 guint avahi_domain_hash(const gchar *s) {
377     guint hash = 0;
378     
379     for (;;) {
380         gchar c[65], *m;
381
382         if (!avahi_unescape_label(&s, c, sizeof(c)))
383             return hash;
384
385         if (!c[0])
386             continue;
387         
388         m = g_ascii_strdown(c, -1);
389         hash += g_str_hash(m);
390         g_free(m);
391     }
392 }
393
394 gchar *avahi_format_mac_address(const guint8* mac, guint size) {
395     gchar *r, *t;
396     guint i;
397     static const gchar hex[] = "0123456789abcdef";
398
399     t = r = g_new(gchar, size > 0 ? size*3 : 1);
400
401     if (size <= 0) {
402         *r = 0;
403         return r;
404     }
405     
406     for (i = 0; i < size; i++) {
407         *(t++) = hex[*mac >> 4];
408         *(t++) = hex[*mac & 0xF];
409         *(t++) = ':';
410
411         mac++;
412     }
413
414     *(--t) = 0;
415     return r;
416 }