etherpkt.hh revision 1078 
1/*
2 * Copyright (c) 2002-2004 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/* @file
30 * Reference counted class containing ethernet packet data
31 */
32
33#ifndef __ETHERPKT_HH__
34#define __ETHERPKT_HH__
35
36#include <iosfwd>
37#include <memory>
38#include <assert.h>
39
40#include "base/refcnt.hh"
41#include "base/inet.hh"
42#include "sim/host.hh"
43
44/*
45 * Reference counted class containing ethernet packet data
46 */
47class Checkpoint;
48class PacketData : public RefCounted
49{
50  public:
51    uint8_t *data;
52    int length;
53
54  protected:
55    uint8_t *_eth;
56    uint8_t *_ip;
57    uint8_t *_tcp;
58    uint8_t *_udp;
59
60    void doext();
61    void ext()
62    {
63        if (_eth != data)
64            doext();
65    }
66
67  public:
68    PacketData() : data(NULL), length(0) { doext(); }
69    PacketData(std::auto_ptr<uint8_t> d, int l)
70        : data(d.release()), length(l) { doext(); }
71    ~PacketData() { if (data) delete [] data; }
72
73  public:
74    EthHdr *eth() { ext(); return (EthHdr *)_eth; }
75    IpHdr *ip() { ext(); return (IpHdr *)_ip; }
76    TcpHdr *tcp() { ext(); return (TcpHdr *)_tcp; }
77    UdpHdr *udp() { ext(); return (UdpHdr *)_udp; }
78
79  public:
80    void serialize(std::ostream &os);
81    void unserialize(Checkpoint *cp, const std::string &section);
82};
83
84typedef RefCountingPtr<PacketData> PacketPtr;
85
86#endif // __ETHERPKT_HH__
87