1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
|
/*****************************************************************************
* This file is part of libmicrodns.
*
* Copyright © 2014-2015 VideoLabs SAS
*
* Author: Jonathan Calmels <jbjcalmels@gmail.com>
*
*****************************************************************************
* libmicrodns is released under LGPLv2.1 (or later) and is also available
* under a commercial license.
*****************************************************************************
* This program is free software; you can redistribute it and/or modify it
* under the terms of the GNU Lesser General Public License as published by
* the Free Software Foundation; either version 2.1 of the License, or
* (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU Lesser General Public License for more details.
*
* You should have received a copy of the GNU Lesser General Public License
* along with this program; if not, write to the Free Software Foundation,
* Inc., 51 Franklin Street, Fifth Floor, Boston MA 02110-1301, USA.
*****************************************************************************/
#ifndef MICRODNS_UTILS_H
#define MICRODNS_UTILS_H
#include <inttypes.h>
#include <stdio.h>
#include <stdint.h>
#include "compat.h"
#define MDNS_DN_MAXSZ 256 // domain name maximum size
static inline int ss_family(const struct sockaddr_storage *ss)
{
return (((const struct sockaddr *) ss)->sa_family);
}
static inline int ss_level(const struct sockaddr_storage *ss)
{
return (ss_family(ss) == AF_INET ? IPPROTO_IP : IPPROTO_IPV6);
}
static inline socklen_t ss_len(const struct sockaddr_storage *ss)
{
return (ss_family(ss) == AF_INET ? sizeof(struct sockaddr_in)
: sizeof(struct sockaddr_in6));
}
static inline socklen_t sa_len(const struct sockaddr* sa)
{
return sa->sa_family == AF_INET ? sizeof(struct sockaddr_in)
: sizeof(struct sockaddr_in6);
}
static inline uint8_t *write_u16(uint8_t *p, size_t *s, const uint16_t v)
{
*p++ = (v >> 8) & 0xFF;
*p++ = (v >> 0) & 0xFF;
if (s != NULL)
*s -= 2;
return (p);
}
static inline uint8_t *write_u32(uint8_t *p, size_t *s, const uint32_t v)
{
*p++ = (v >> 24) & 0xFF;
*p++ = (v >> 16) & 0xFF;
*p++ = (v >> 8) & 0xFF;
*p++ = (v >> 0) & 0xFF;
if (s != NULL)
*s -= 4;
return (p);
}
static inline uint8_t *write_raw(uint8_t *p, size_t* s, const uint8_t *v)
{
size_t len;
len = strlen((const char *) v) + 1;
if (*s < len)
return (NULL);
memcpy(p, v, len);
p += len;
*s -= len;
return (p);
}
static inline const uint8_t *read_u16(const uint8_t *p, size_t *s, uint16_t *v)
{
*v = 0;
*v |= (uint16_t)*p++ << 8;
*v |= (uint16_t)*p++ << 0;
*s -= 2;
return (p);
}
static inline const uint8_t *read_u32(const uint8_t *p, size_t *s, uint32_t *v)
{
*v = 0;
*v |= (uint32_t)*p++ << 24;
*v |= (uint32_t)*p++ << 16;
*v |= (uint32_t)*p++ << 8;
*v |= (uint32_t)*p++ << 0;
*s -= 4;
return (p);
}
#endif /* MICRODNS_UTILS_H */
|
