Merge master.kernel.org:/home/rmk/linux-2.6-arm
[linux-2.6] / net / core / utils.c
1 /*
2  *      Generic address resultion entity
3  *
4  *      Authors:
5  *      net_random Alan Cox
6  *      net_ratelimit Andi Kleen
7  *      in{4,6}_pton YOSHIFUJI Hideaki, Copyright (C)2006 USAGI/WIDE Project
8  *
9  *      Created by Alexey Kuznetsov <kuznet@ms2.inr.ac.ru>
10  *
11  *      This program is free software; you can redistribute it and/or
12  *      modify it under the terms of the GNU General Public License
13  *      as published by the Free Software Foundation; either version
14  *      2 of the License, or (at your option) any later version.
15  */
16
17 #include <linux/module.h>
18 #include <linux/jiffies.h>
19 #include <linux/kernel.h>
20 #include <linux/inet.h>
21 #include <linux/mm.h>
22 #include <linux/net.h>
23 #include <linux/string.h>
24 #include <linux/types.h>
25 #include <linux/random.h>
26 #include <linux/percpu.h>
27 #include <linux/init.h>
28 #include <net/sock.h>
29
30 #include <asm/byteorder.h>
31 #include <asm/system.h>
32 #include <asm/uaccess.h>
33
34 int net_msg_warn __read_mostly = 1;
35 EXPORT_SYMBOL(net_msg_warn);
36
37 DEFINE_RATELIMIT_STATE(net_ratelimit_state, 5 * HZ, 10);
38 /*
39  * All net warning printk()s should be guarded by this function.
40  */
41 int net_ratelimit(void)
42 {
43         return __ratelimit(&net_ratelimit_state);
44 }
45 EXPORT_SYMBOL(net_ratelimit);
46
47 /*
48  * Convert an ASCII string to binary IP.
49  * This is outside of net/ipv4/ because various code that uses IP addresses
50  * is otherwise not dependent on the TCP/IP stack.
51  */
52
53 __be32 in_aton(const char *str)
54 {
55         unsigned long l;
56         unsigned int val;
57         int i;
58
59         l = 0;
60         for (i = 0; i < 4; i++)
61         {
62                 l <<= 8;
63                 if (*str != '\0')
64                 {
65                         val = 0;
66                         while (*str != '\0' && *str != '.' && *str != '\n')
67                         {
68                                 val *= 10;
69                                 val += *str - '0';
70                                 str++;
71                         }
72                         l |= val;
73                         if (*str != '\0')
74                                 str++;
75                 }
76         }
77         return(htonl(l));
78 }
79
80 EXPORT_SYMBOL(in_aton);
81
82 #define IN6PTON_XDIGIT          0x00010000
83 #define IN6PTON_DIGIT           0x00020000
84 #define IN6PTON_COLON_MASK      0x00700000
85 #define IN6PTON_COLON_1         0x00100000      /* single : requested */
86 #define IN6PTON_COLON_2         0x00200000      /* second : requested */
87 #define IN6PTON_COLON_1_2       0x00400000      /* :: requested */
88 #define IN6PTON_DOT             0x00800000      /* . */
89 #define IN6PTON_DELIM           0x10000000
90 #define IN6PTON_NULL            0x20000000      /* first/tail */
91 #define IN6PTON_UNKNOWN         0x40000000
92
93 static inline int xdigit2bin(char c, int delim)
94 {
95         if (c == delim || c == '\0')
96                 return IN6PTON_DELIM;
97         if (c == ':')
98                 return IN6PTON_COLON_MASK;
99         if (c == '.')
100                 return IN6PTON_DOT;
101         if (c >= '0' && c <= '9')
102                 return (IN6PTON_XDIGIT | IN6PTON_DIGIT| (c - '0'));
103         if (c >= 'a' && c <= 'f')
104                 return (IN6PTON_XDIGIT | (c - 'a' + 10));
105         if (c >= 'A' && c <= 'F')
106                 return (IN6PTON_XDIGIT | (c - 'A' + 10));
107         if (delim == -1)
108                 return IN6PTON_DELIM;
109         return IN6PTON_UNKNOWN;
110 }
111
112 int in4_pton(const char *src, int srclen,
113              u8 *dst,
114              int delim, const char **end)
115 {
116         const char *s;
117         u8 *d;
118         u8 dbuf[4];
119         int ret = 0;
120         int i;
121         int w = 0;
122
123         if (srclen < 0)
124                 srclen = strlen(src);
125         s = src;
126         d = dbuf;
127         i = 0;
128         while(1) {
129                 int c;
130                 c = xdigit2bin(srclen > 0 ? *s : '\0', delim);
131                 if (!(c & (IN6PTON_DIGIT | IN6PTON_DOT | IN6PTON_DELIM | IN6PTON_COLON_MASK))) {
132                         goto out;
133                 }
134                 if (c & (IN6PTON_DOT | IN6PTON_DELIM | IN6PTON_COLON_MASK)) {
135                         if (w == 0)
136                                 goto out;
137                         *d++ = w & 0xff;
138                         w = 0;
139                         i++;
140                         if (c & (IN6PTON_DELIM | IN6PTON_COLON_MASK)) {
141                                 if (i != 4)
142                                         goto out;
143                                 break;
144                         }
145                         goto cont;
146                 }
147                 w = (w * 10) + c;
148                 if ((w & 0xffff) > 255) {
149                         goto out;
150                 }
151 cont:
152                 if (i >= 4)
153                         goto out;
154                 s++;
155                 srclen--;
156         }
157         ret = 1;
158         memcpy(dst, dbuf, sizeof(dbuf));
159 out:
160         if (end)
161                 *end = s;
162         return ret;
163 }
164
165 EXPORT_SYMBOL(in4_pton);
166
167 int in6_pton(const char *src, int srclen,
168              u8 *dst,
169              int delim, const char **end)
170 {
171         const char *s, *tok = NULL;
172         u8 *d, *dc = NULL;
173         u8 dbuf[16];
174         int ret = 0;
175         int i;
176         int state = IN6PTON_COLON_1_2 | IN6PTON_XDIGIT | IN6PTON_NULL;
177         int w = 0;
178
179         memset(dbuf, 0, sizeof(dbuf));
180
181         s = src;
182         d = dbuf;
183         if (srclen < 0)
184                 srclen = strlen(src);
185
186         while (1) {
187                 int c;
188
189                 c = xdigit2bin(srclen > 0 ? *s : '\0', delim);
190                 if (!(c & state))
191                         goto out;
192                 if (c & (IN6PTON_DELIM | IN6PTON_COLON_MASK)) {
193                         /* process one 16-bit word */
194                         if (!(state & IN6PTON_NULL)) {
195                                 *d++ = (w >> 8) & 0xff;
196                                 *d++ = w & 0xff;
197                         }
198                         w = 0;
199                         if (c & IN6PTON_DELIM) {
200                                 /* We've processed last word */
201                                 break;
202                         }
203                         /*
204                          * COLON_1 => XDIGIT
205                          * COLON_2 => XDIGIT|DELIM
206                          * COLON_1_2 => COLON_2
207                          */
208                         switch (state & IN6PTON_COLON_MASK) {
209                         case IN6PTON_COLON_2:
210                                 dc = d;
211                                 state = IN6PTON_XDIGIT | IN6PTON_DELIM;
212                                 if (dc - dbuf >= sizeof(dbuf))
213                                         state |= IN6PTON_NULL;
214                                 break;
215                         case IN6PTON_COLON_1|IN6PTON_COLON_1_2:
216                                 state = IN6PTON_XDIGIT | IN6PTON_COLON_2;
217                                 break;
218                         case IN6PTON_COLON_1:
219                                 state = IN6PTON_XDIGIT;
220                                 break;
221                         case IN6PTON_COLON_1_2:
222                                 state = IN6PTON_COLON_2;
223                                 break;
224                         default:
225                                 state = 0;
226                         }
227                         tok = s + 1;
228                         goto cont;
229                 }
230
231                 if (c & IN6PTON_DOT) {
232                         ret = in4_pton(tok ? tok : s, srclen + (int)(s - tok), d, delim, &s);
233                         if (ret > 0) {
234                                 d += 4;
235                                 break;
236                         }
237                         goto out;
238                 }
239
240                 w = (w << 4) | (0xff & c);
241                 state = IN6PTON_COLON_1 | IN6PTON_DELIM;
242                 if (!(w & 0xf000)) {
243                         state |= IN6PTON_XDIGIT;
244                 }
245                 if (!dc && d + 2 < dbuf + sizeof(dbuf)) {
246                         state |= IN6PTON_COLON_1_2;
247                         state &= ~IN6PTON_DELIM;
248                 }
249                 if (d + 2 >= dbuf + sizeof(dbuf)) {
250                         state &= ~(IN6PTON_COLON_1|IN6PTON_COLON_1_2);
251                 }
252 cont:
253                 if ((dc && d + 4 < dbuf + sizeof(dbuf)) ||
254                     d + 4 == dbuf + sizeof(dbuf)) {
255                         state |= IN6PTON_DOT;
256                 }
257                 if (d >= dbuf + sizeof(dbuf)) {
258                         state &= ~(IN6PTON_XDIGIT|IN6PTON_COLON_MASK);
259                 }
260                 s++;
261                 srclen--;
262         }
263
264         i = 15; d--;
265
266         if (dc) {
267                 while(d >= dc)
268                         dst[i--] = *d--;
269                 while(i >= dc - dbuf)
270                         dst[i--] = 0;
271                 while(i >= 0)
272                         dst[i--] = *d--;
273         } else
274                 memcpy(dst, dbuf, sizeof(dbuf));
275
276         ret = 1;
277 out:
278         if (end)
279                 *end = s;
280         return ret;
281 }
282
283 EXPORT_SYMBOL(in6_pton);
284
285 void inet_proto_csum_replace4(__sum16 *sum, struct sk_buff *skb,
286                               __be32 from, __be32 to, int pseudohdr)
287 {
288         __be32 diff[] = { ~from, to };
289         if (skb->ip_summed != CHECKSUM_PARTIAL) {
290                 *sum = csum_fold(csum_partial(diff, sizeof(diff),
291                                 ~csum_unfold(*sum)));
292                 if (skb->ip_summed == CHECKSUM_COMPLETE && pseudohdr)
293                         skb->csum = ~csum_partial(diff, sizeof(diff),
294                                                 ~skb->csum);
295         } else if (pseudohdr)
296                 *sum = ~csum_fold(csum_partial(diff, sizeof(diff),
297                                 csum_unfold(*sum)));
298 }
299 EXPORT_SYMBOL(inet_proto_csum_replace4);