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
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
|
/** @file
Loading @c IpMap from a configuration file.
@section license License
Licensed to the Apache Software Foundation (ASF) under one
or more contributor license agreements. See the NOTICE file
distributed with this work for additional information
regarding copyright ownership. The ASF licenses this file
to you under the Apache License, Version 2.0 (the
"License"); you may not use this file except in compliance
with the License. You may obtain a copy of the License at
http://www.apache.org/licenses/LICENSE-2.0
Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License.
*/
// Copied from IPRange.cc for backwards compatibility.
#include <ts/IpMap.h>
#include <ts/IpMapConf.h>
#include <ts/ink_memory.h>
static size_t const ERR_STRING_LEN = 256;
static size_t const MAX_LINE_SIZE = 2048;
// Returns 0 if successful, 1 if failed
// line Input text (source line).
// n Amount of data in @a line.
// i [in,out] Offset in line.
// addr [out] Destination for address.
// err Buffer for error string (must be ERR_STRING_LEN big).
int
read_addr(char *line, int n, int *i, sockaddr *addr, char *err)
{
int k;
char dst[INET6_ADDRSTRLEN];
char *src = line + *i;
bool bracketed_p = false;
// Allow enclosing brackets to be more consistent but
// don't bother passing it to @c ntop.
if ((*i < n) && ('[' == *src)) {
++*i, ++src, bracketed_p = true;
}
for (k = 0; k < INET6_ADDRSTRLEN && *i < n && (isxdigit(*src) || '.' == *src || ':' == *src); ++k, ++*i, ++src) {
dst[k] = *src;
}
if (bracketed_p && (!(*i < n) || (']' != *src))) {
snprintf(err, ERR_STRING_LEN, "Unclosed brackets");
return EINVAL;
}
if (k == sizeof(dst)) {
snprintf(err, ERR_STRING_LEN, "IP address too long");
return EINVAL;
}
dst[k] = '\0';
if (0 != ats_ip_pton(dst, addr)) {
snprintf(err, ERR_STRING_LEN, "IP address '%s' improperly formatted", dst);
return EINVAL;
}
return 0;
}
char *
Load_IpMap_From_File(IpMap *map, int fd, const char *key_str)
{
char *zret = 0;
int fd2 = dup(fd); // dup to avoid closing the original file.
FILE *f = NULL;
if (fd2 >= 0)
f = fdopen(fd2, "r");
if (f != NULL) {
zret = Load_IpMap_From_File(map, f, key_str);
fclose(f);
} else {
zret = (char *)ats_malloc(ERR_STRING_LEN);
snprintf(zret, ERR_STRING_LEN, "Unable to reopen file descriptor as stream %d:%s", errno, strerror(errno));
}
return zret;
}
// Skip space in line, returning true if more data is available
// (not end of line).
//
// line Source line.
// n Line length.
// offset Current offset
static inline bool
skip_space(char *line, int n, int &offset)
{
while (offset < n && isspace(line[offset]))
++offset;
return offset < n;
}
// Returns 0 if successful, error string otherwise
char *
Load_IpMap_From_File(IpMap *map, FILE *f, const char *key_str)
{
int i, n, line_no;
int key_len = strlen(key_str);
IpEndpoint laddr, raddr;
char line[MAX_LINE_SIZE];
char err_buff[ERR_STRING_LEN];
// First hardcode 127.0.0.1 into the table
map->mark(INADDR_LOOPBACK);
line_no = 0;
while (fgets(line, MAX_LINE_SIZE, f)) {
++line_no;
n = strlen(line);
// Find first white space which terminates the line key.
for (i = 0; i < n && !isspace(line[i]); ++i)
;
if (i != key_len || 0 != strncmp(line, key_str, key_len))
continue;
// Now look for IP address
while (true) {
if (!skip_space(line, n, i))
break;
if (0 != read_addr(line, n, &i, &laddr.sa, err_buff)) {
char *error_str = (char *)ats_malloc(ERR_STRING_LEN);
snprintf(error_str, ERR_STRING_LEN, "Invalid input configuration (%s) at line %d offset %d - '%s'", err_buff, line_no, i,
line);
return error_str;
}
if (!skip_space(line, n, i) || line[i] == ',') {
// You have read an IP address. Enter it in the table
map->mark(&laddr);
if (i == n)
break;
else
++i;
} else if (line[i] == '-') {
// What you have just read is the start of the range,
// Now, read the end of the IP range
++i;
if (!skip_space(line, n, i)) {
char *error_str = (char *)ats_malloc(ERR_STRING_LEN);
snprintf(error_str, ERR_STRING_LEN, "Invalid input (unterminated range) at line %d offset %d - '%s'", line_no, i, line);
return error_str;
} else if (0 != read_addr(line, n, &i, &raddr.sa, err_buff)) {
char *error_str = (char *)ats_malloc(ERR_STRING_LEN);
snprintf(error_str, ERR_STRING_LEN, "Invalid input (%s) at line %d offset %d - '%s'", err_buff, line_no, i, line);
return error_str;
}
map->mark(&laddr.sa, &raddr.sa);
if (!skip_space(line, n, i))
break;
if (line[i] != ',') {
char *error_str = (char *)ats_malloc(ERR_STRING_LEN);
snprintf(error_str, ERR_STRING_LEN, "Invalid input (expecting comma) at line %d offset %d - '%s'", line_no, i, line);
return error_str;
}
++i;
} else {
char *error_str = (char *)ats_malloc(ERR_STRING_LEN);
snprintf(error_str, ERR_STRING_LEN, "Invalid input (expecting dash or comma) at line %d offset %d", line_no, i);
return error_str;
}
}
}
return 0;
}
|
