inet.hh revision 1092 
1/*
2 * Copyright (c) 2002-2003 The Regents of The University of Michigan
3 * All rights reserved.
4 *
5 * Redistribution and use in source and binary forms, with or without
6 * modification, are permitted provided that the following conditions are
7 * met: redistributions of source code must retain the above copyright
8 * notice, this list of conditions and the following disclaimer;
9 * redistributions in binary form must reproduce the above copyright
10 * notice, this list of conditions and the following disclaimer in the
11 * documentation and/or other materials provided with the distribution;
12 * neither the name of the copyright holders nor the names of its
13 * contributors may be used to endorse or promote products derived from
14 * this software without specific prior written permission.
15 *
16 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
17 * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
18 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
19 * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
20 * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
21 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
22 * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
23 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
24 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
25 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
26 * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
27 */
28
29#ifndef __BASE_INET_HH__
30#define __BASE_INET_HH__
31
32#include <string>
33
34#include "dnet/os.h"
35
36#include "dnet/eth.h"
37#include "dnet/ip.h"
38#include "dnet/ip6.h"
39#include "dnet/addr.h"
40#include "dnet/arp.h"
41#include "dnet/icmp.h"
42#include "dnet/tcp.h"
43#include "dnet/udp.h"
44
45#include "dnet/intf.h"
46#include "dnet/route.h"
47#include "dnet/fw.h"
48
49#include "dnet/blob.h"
50#include "dnet/rand.h"
51
52#include "sim/host.hh"
53
54std::string eaddr_string(const uint8_t a[6]);
55
56struct EthHdr;
57struct IpHdr;
58struct TcpHdr;
59struct UdpHdr;
60
61struct EthHdr : protected eth_hdr
62{
63    uint16_t type() const { return ntohs(eth_type); }
64
65    const IpHdr *ip() const
66    { return type() == ETH_TYPE_IP ? (const IpHdr *)payload() : 0; }
67
68    IpHdr *ip()
69    { return type() == ETH_TYPE_IP ? (IpHdr *)payload() : 0; }
70
71    bool unicast() { return eth_dst.data[0] == 0x00; }
72    bool multicast() { return eth_dst.data[0] == 0x01; }
73    bool broadcast() { return eth_dst.data[0] == 0xff; }
74
75    int size() const { return sizeof(EthHdr); }
76    const uint8_t *bytes() const { return (const uint8_t *)this; }
77    const uint8_t *payload() const { return bytes() + size(); }
78    uint8_t *bytes() { return (uint8_t *)this; }
79    uint8_t *payload() { return bytes() + size(); }
80};
81
82struct IpHdr : protected ip_hdr
83{
84    uint8_t  version() const { return ip_v; }
85    uint8_t  hlen() const { return ip_hl * 4; }
86    uint8_t  tos() const { return ip_tos; }
87    uint16_t len() const { return ntohs(ip_len); }
88    uint16_t id() const { return ntohs(ip_id); }
89    uint16_t frag_flags() const { return ntohs(ip_off) >> 13; }
90    uint16_t frag_off() const { return ntohs(ip_off) & 0x1fff; }
91    uint8_t  ttl() const { return ip_ttl; }
92    uint8_t  proto() const { return ip_p; }
93    uint16_t sum() const { return ip_sum; }
94    uint32_t src() const { return ntohl(ip_src); }
95    uint32_t dst() const { return ntohl(ip_dst); }
96
97    void sum(uint16_t sum) { ip_sum = sum; }
98
99    uint16_t ip_cksum() const;
100    uint16_t tu_cksum() const;
101
102    const TcpHdr *tcp() const
103    { return proto() == IP_PROTO_TCP ? (const TcpHdr *)payload() : 0; }
104    const UdpHdr *udp() const
105    { return proto() == IP_PROTO_UDP ? (const UdpHdr *)payload() : 0; }
106
107    TcpHdr *tcp()
108    { return proto() == IP_PROTO_TCP ? (TcpHdr *)payload() : 0; }
109    UdpHdr *udp()
110    { return proto() == IP_PROTO_UDP ? (UdpHdr *)payload() : 0; }
111
112
113    int size() const { return hlen(); }
114    const uint8_t *bytes() const { return (const uint8_t *)this; }
115    const uint8_t *payload() const { return bytes() + size(); }
116    uint8_t *bytes() { return (uint8_t *)this; }
117    uint8_t *payload() { return bytes() + size(); }
118};
119
120struct TcpHdr : protected tcp_hdr
121{
122    uint16_t sport() const { return ntohs(th_sport); }
123    uint16_t dport() const { return ntohs(th_dport); }
124    uint32_t seq() const { return ntohl(th_seq); }
125    uint32_t ack() const { return ntohl(th_ack); }
126    uint8_t  off() const { return th_off; }
127    uint8_t  flags() const { return th_flags & 0x3f; }
128    uint16_t win() const { return ntohs(th_win); }
129    uint16_t sum() const { return th_sum; }
130    uint16_t urp() const { return ntohs(th_urp); }
131
132    void sum(uint16_t sum) { th_sum = sum; }
133
134    int size() const { return off(); }
135    const uint8_t *bytes() const { return (const uint8_t *)this; }
136    const uint8_t *payload() const { return bytes() + size(); }
137    uint8_t *bytes() { return (uint8_t *)this; }
138    uint8_t *payload() { return bytes() + size(); }
139};
140
141struct UdpHdr : protected udp_hdr
142{
143    uint16_t sport() const { return ntohs(uh_sport); }
144    uint16_t dport() const { return ntohs(uh_dport); }
145    uint16_t len() const { return ntohs(uh_ulen); }
146    uint16_t sum() const { return ntohs(uh_sum); }
147
148    void sum(uint16_t sum) { uh_sum = htons(sum); }
149
150    int size() const { return sizeof(UdpHdr); }
151    const uint8_t *bytes() const { return (const uint8_t *)this; }
152    const uint8_t *payload() const { return bytes() + size(); }
153    uint8_t *bytes() { return (uint8_t *)this; }
154    uint8_t *payload() { return bytes() + size(); }
155};
156
157#endif // __BASE_INET_HH__
158