From Harald Welte:
[obnox/wireshark/wip.git] / epan / addr_and_mask.c
1 /* addr_and_mask.c
2  * Routines to fetch IPv4 and IPv6 addresses from a tvbuff and then mask
3  * out bits other than those covered by a prefix length
4  *
5  * $Id$
6  *
7  * Wireshark - Network traffic analyzer
8  * By Gerald Combs <gerald@wireshark.org>
9  * Copyright 1998 Gerald Combs
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 2
14  * of the License, or (at your option) any later version.
15  *
16  * This program is distributed in the hope that it will be useful,
17  * but WITHOUT ANY WARRANTY; without even the implied warranty of
18  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
19  * GNU General Public License for more details.
20  *
21  * You should have received a copy of the GNU General Public License
22  * along with this program; if not, write to the Free Software
23  * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
24  */
25
26 #ifdef HAVE_CONFIG_H
27 # include "config.h"
28 #endif
29
30 #include <stdlib.h>
31 #include <string.h>
32
33 #include "tvbuff.h"
34 #include "addr_and_mask.h"
35
36 /*
37  * These routines return the length of the address in bytes on success
38  * and -1 if the prefix length is too long.
39  */
40
41 int
42 ipv4_addr_and_mask(tvbuff_t *tvb, int offset, guint8 *addr, guint32 prefix_len)
43 {
44         guint32 addr_len;
45
46         if (prefix_len > 32)
47                 return -1;
48
49         addr_len = (prefix_len + 7) / 8;
50         memset(addr, 0, 4);
51         tvb_memcpy(tvb, addr, offset, addr_len);
52         if (prefix_len % 8)
53                 addr[addr_len - 1] &= ((0xff00 >> (prefix_len % 8)) & 0xff);
54         return addr_len;
55 }
56
57 int
58 ipv6_addr_and_mask(tvbuff_t *tvb, int offset, struct e_in6_addr *addr,
59     guint32 prefix_len)
60 {
61         guint32 addr_len;
62
63         if (prefix_len > 128)
64                 return -1;
65
66         addr_len = (prefix_len + 7) / 8;
67         memset(addr->bytes, 0, 16);
68         tvb_memcpy(tvb, addr->bytes, offset, addr_len);
69         if (prefix_len % 8) {
70                 addr->bytes[addr_len - 1] &=
71                     ((0xff00 >> (prefix_len % 8)) & 0xff);
72         }
73
74         return addr_len;
75 }