added TCPPacket implemented by Dennis.

[senf.git] / Packets / DefaultBundle / UDPPacket.cc

diff --git a/Packets/DefaultBundle/UDPPacket.cc b/Packets/DefaultBundle/UDPPacket.cc
index 8d4a850..a985f24 100644 (file)
--- a/Packets/DefaultBundle/UDPPacket.cc
+++ b/Packets/DefaultBundle/UDPPacket.cc
@@ -1,8 +1,8 @@
 // $Id$
 //
 // Copyright (C) 2006
-// Fraunhofer Institute for Open Communication Systems (FOKUS) 
-// Competence Center NETwork research (NET), St. Augustin, GERMANY 
+// Fraunhofer Institute for Open Communication Systems (FOKUS)
+// Competence Center NETwork research (NET), St. Augustin, GERMANY
 //     Stefan Bund <g0dil@berlios.de>
 //
 // This program is free software; you can redistribute it and/or modify
@@ -25,7 +25,6 @@
 
 #include "UDPPacket.hh"
 //#include "UDPPacket.ih"
-#include "IPv4Packet.hh"
 
 // Custom includes
 #include <iomanip>
@@ -50,11 +49,8 @@ prefix_ boost::uint16_t senf::UDPPacketParser::calcChecksum()
     const
 {
     IpChecksum summer;
-    summer.feed( i(), i()+checksum_offset );
-    summer.feed( i()+checksum_offset+2, data().end() );
-    
-    // Now on to the awkward part: the IP pseudo header
-    IPv4Packet ipv4 (packet().rfind<IPv4Packet>());
+    // first on to the awkward part: the IP pseudo header
+    IPv4Packet ipv4 (packet().rfind<IPv4Packet>(nothrow));
     if (ipv4) {
         // Pseudo header defined in RFC768
         summer.feed( ipv4->source().i(), 
@@ -70,7 +66,7 @@ prefix_ boost::uint16_t senf::UDPPacketParser::calcChecksum()
     } 
     else {
         // Pseudo header defined in RFC2460
-        IPv6Packet ipv6 (packet().rfind<IPv6Packet>());
+        IPv6Packet ipv6 (packet().rfind<IPv6Packet>(nothrow));
         if (ipv6) {
             summer.feed( ipv6->source().i(), 
                          ipv6->source().i() + IPv6Packet::Parser::source_t::fixed_bytes );
@@ -78,8 +74,8 @@ prefix_ boost::uint16_t senf::UDPPacketParser::calcChecksum()
             // The destination used here must be the *final* destination ...
             summer.feed( ipv6->destination().i(), 
                          ipv6->destination().i() + IPv6PacketParser::destination_t::fixed_bytes );
-            /// This is a simplification. The value is really 32bit to support UDP Jumbograms
-            /// (RFC2147). However, skipping an even number of 0 bytes does not change the checksum
+            // This is a simplification. The value is really 32bit to support UDP Jumbograms
+            // (RFC2147). However, skipping an even number of 0 bytes does not change the checksum
             summer.feed( i() + length_offset, i() + length_offset + 2 );
             // RFC2460 specifies, that this must always be 17, not the value used in the ipv6
             // header
@@ -88,6 +84,11 @@ prefix_ boost::uint16_t senf::UDPPacketParser::calcChecksum()
         }
     }
     
+    // since header are 16 / even 32bit aligned we don't have to care for padding. since IpChecksum 
+    // cares for padding at the final summing we don't have to care is the payload is 16nbit-aligned, too.
+    summer.feed( i(), i()+checksum_offset );
+    summer.feed( i()+checksum_offset+2, data().end() );
+
     boost::uint16_t rv (summer.sum());
     return rv ? rv : 0xffffu;
 }
@@ -99,11 +100,11 @@ prefix_ void senf::UDPPacketType::dump(packet p, std::ostream & os)
 {
     boost::io::ios_all_saver ias(os);
     os << "UDP:\n"
-       << "  source port   : " << p->source() << "\n"
-       << "  dest port     : " << p->destination() << "\n"
-       << "  length        : " << p->length() << "\n"
-       << "  checksum      : " 
-         << std::hex << std::setw(4) << std::setfill('0') << p->checksum() << "\n";
+       <<     "  source port             : " << p->source() << "\n"
+       <<     "  dest port               : " << p->destination() << "\n"
+       <<     "  length                  : " << p->length() << "\n"
+       <<     "  checksum                : " 
+       << std::hex << std::setw(4) << std::setfill('0') << p->checksum() << "\n";
 }
 
 prefix_ void senf::UDPPacketType::finalize(packet p)