implemented nextPacketType/Range for SNDU (yeah!)
[senf.git] / Examples / DVBAdapter / ULEdec.cc
index a732b96..9eccad3 100644 (file)
 
 // Definition of non-inline non-template functions
 
-#include <string>
-#include <iostream>
-#include <iomanip>
-#include <sys/ioctl.h>
-#include <linux/sockios.h>
-#include <linux/dvb/dmx.h> 
-
-#include "Scheduler/Scheduler.hh"
-#include "Packets/DefaultBundle/EthernetPacket.hh"
-#include "Packets/MPEGDVBBundle/TransportPacket.hh"
-#include "Packets/MPEGDVBBundle/SNDUPacket.hh"
-#include "Utils/membind.hh"
-#include "Socket/Protocols/DVB/DVBDemuxHandles.hh"
-#include "Packets/ParseInt.hh"
-#include "Packets/Packet.hh"
-#include "Packets/PacketData.hh"
-#include "Packets/ParseInt.hh"
-
 #include "ULEdec.hh"
 
+#include "Packets/PacketData.hh"
+#include "Utils/hexdump.hh"
+#include "Utils/membind.hh"
+
 #define PID 271
 #define TS_SYNC 0x47
 
 #define prefix_
 ///////////////////////////////cc.p////////////////////////////////////////
 
-template <class Iterator>
-void ULEdec::hexdump(Iterator i, Iterator const & i_end, std::ostream& stream)
-{
-    unsigned offset (0);
-    std::string ascii;
-    for (; i != i_end; ++i, ++offset) {
-        switch (offset % BLOCK_SIZE) {
-        case 0:
-            if (!ascii.empty()) {
-                stream << "  " << ascii << "\n";
-                ascii = "";
-            }
-            stream << "  "
-                      << std::hex << std::setw(4) << std::setfill('0')
-                      << offset << ' ';
-            break;
-        case BLOCK_SIZE/2:
-            stream << " ";
-            ascii += ' ';
-            break;
-        }
-        stream << ' ' << std::hex << std::setw(2) << std::setfill('0')
-                  << unsigned(*i);
-        ascii += (*i >= ' ' && *i < 126) ? *i : '.';
-    }
-    if (!ascii.empty()) {
-        for (; (offset % BLOCK_SIZE) != 0; ++offset) {
-            if ((offset % BLOCK_SIZE) == BLOCK_SIZE/2)
-                stream << " ";
-            stream << "   ";
-        }
-        stream << "  " << ascii << "\n";
-    }
-    stream << std::dec;
-}
-
 
 ULEdec::ULEdec()
 {
@@ -99,7 +49,8 @@ ULEdec::ULEdec()
     senf::Scheduler::instance().add(
         dvrHandle, senf::membind(&ULEdec::handleEvent, this));
     
-    this->receiver_state = 1; // Idle
+    this->receiver_state = Idle;
+    this->priv_sndu_type_1 = false;
 }
 
 void ULEdec::handleEvent(senf::FileHandle, senf::Scheduler::EventId event)
@@ -123,56 +74,177 @@ void ULEdec::handleEvent(senf::FileHandle, senf::Scheduler::EventId event)
     
 void ULEdec::handleTSPacket(senf::TransportPacket ts_packet)
 {
-    unsigned char payload_pointer;
-
-    senf::PacketData &ts_payload (ts_packet.next().data());
-    BOOST_ASSERT( ts_payload.size() == 184 );
+    senf::PacketData & payloadData (ts_packet.next().data());
+    iterator payload_iter = payloadData.begin();
+    iterator payload_end = payloadData.end();
     
-    switch (this->receiver_state) {
-    case 1: { // Idle State
-        // resync on PUSI
-        if (ts_packet->pusi() == 0) 
-            return; // skip packet
-        
-        // Synchronize continuity counter
-        this->priv_tscc = ts_packet->continuity_counter();
-
+    std::cout << "New TS Packet:\n"
+              << "----------------------------------------------------------------------------\n";
+    senf::hexdump(payload_iter, payload_end, std::cout);
+    std::cout << "----------------------------------------------------------------------------\n";
+    
+    // Synchronize continuity counter
+    this->priv_tscc = ts_packet->continuity_counter();
+    
+    switch (ts_packet->pusi()) {
+    case 0: {
+        switch (this->receiver_state) {
+        case Idle:
+            return;
+        case Reassembly:
+            payload_iter = readContSNDUPacket( payload_iter, payload_end );
+            if (isSDNUPacketComplete()) {
+                this->receiver_state = Idle;
+                try {
+                    handleSNDUPacket();
+                } catch (ULEdecException const & ex) {
+                    std::cerr << ex.what() << "\n";
+                    return;
+                }
+                switch (std::distance( payload_iter, payload_end ))
+                case 0:
+                case 1:
+                    return;
+                default:
+                    if ( (*payload_iter++ << 8 | *payload_iter++) != ULE_END_INDICATOR )
+                        std::cerr << "delimiting error 1\n";
+            } else {
+                BOOST_ASSERT( std::distance( payload_iter, payload_end ) == 0 );
+            }
+        }
+        break;
+    }
+    case 1: {
         // a PUSI value of 1 indicates the presence of a Payload Pointer.
-        payload_pointer = ts_payload[0];
-        if (payload_pointer>181) {
+        unsigned char payload_pointer = *payload_iter++;
+        if (payload_pointer > 181) {
             std::cerr << "invalid payload_pointer\n";
+            this->receiver_state = Idle;
             return;
         }
-        payload_pointer++;
-        
-        bool dbit = false;
-        senf::Packet::size_type sndu_length;
-        while (payload_pointer < 184) {
-            sndu_length = ts_payload[payload_pointer] << 8 | ts_payload[payload_pointer+1];
-            if (sndu_length & 0x8000) {
-                sndu_length &= 0x7FFF;
-                dbit = true;
+        switch (this->receiver_state) {
+        case Idle:
+            std::advance(payload_iter, payload_pointer);
+            break;
+        case Reassembly:
+            // Reassembly Payload Pointer Checking
+            if (snduPacketBytesLeft() != payload_pointer) {
+                // delimiting error
+                std::cerr << "delimiting error 2\n";
+                std::advance(payload_iter, payload_pointer);
+            } else {
+                payload_iter = readContSNDUPacket( payload_iter, payload_end );
+                BOOST_ASSERT( isSDNUPacketComplete() );
+                try {
+                    handleSNDUPacket();
+                } catch (ULEdecException const & ex) {
+                    std::cerr << ex.what() << "\n";
+                    this->receiver_state = Idle;
+                    return;
+                }
             }
-            this->snduPacket = senf::SNDUPacket::create(sndu_length+2);
-            this->snduPacket->d_bit() = dbit;
-            this->snduPacket->length() = sndu_length;
-
+        }
+        this->receiver_state = Idle;
+        try {
+            do {
+                payload_iter = readNewSNDUPacket( payload_iter, payload_end );
+                if (! isSDNUPacketComplete()) {
+                    BOOST_ASSERT( std::distance( payload_iter, payload_end ) == 0 );
+                    this->receiver_state = Reassembly;
+                    break;
+                }
+                handleSNDUPacket();
+            } while (std::distance(payload_iter, payload_end) < 2 );
+        }
+        catch (ULEdecException const & ex) {
+            std::cerr << ex.what() << "\n";
             return;
-            
-            payload_pointer += 2;
-            
-            
         }
-        
-        
-        
-        // 
-        break;
     }
-    case 2: { // Reassembly State
+    
+    } // end pusi-switch
+}
+
+
+ULEdec::iterator ULEdec::readContSNDUPacket(iterator i, iterator const i_end)
+{
+    if (priv_sndu_type_1) {
+        this->snduPacket->type() |= *i++;
+    }
+    return readRawSNDUPacketData(i, i_end);
+}
+
+
+ULEdec::iterator ULEdec::readNewSNDUPacket(iterator i, iterator const i_end)
+{ 
+    bool dbit = false;
+    senf::Packet::size_type sndu_length = *i++ << 8 | *i++;
+    if (sndu_length & 0x8000) {
+        sndu_length &= 0x7FFF;
+        dbit = true;
+    }
+    if (sndu_length < 5 || sndu_length == 0xffff) {
+        throw ULEdecException("SNDU length error");
+     }
+    this->snduPacket = senf::SNDUPacket::create(sndu_length+4);
+    this->snduPacket->d_bit() = dbit;
+    this->snduPacket->length() = sndu_length;
+    this->snduPacketData_iter = boost::next(this->snduPacket.data().begin(), 2);
+    this->priv_sndu_type_1 = false;
+    
+    switch (std::distance(i, i_end)) {
+    case 1:
+        this->priv_sndu_type_1 = true;
+        this->snduPacket->type() = *i++ << 8;
+        this->snduPacketData_iter++;
+    case 0:
         break;
+        
+    default: 
+        this->snduPacket->type() = *i++ << 8 | *i++;
+        std::advance(this->snduPacketData_iter, 2);
+        i = readRawSNDUPacketData(i, i_end);
     }
+    
+    return i;
+}
+
+
+ULEdec::iterator ULEdec::readRawSNDUPacketData(iterator i, iterator i_end)
+{
+    unsigned char how_much = std::min(
+            snduPacketBytesLeft(), std::distance(i, i_end) );
+    copy_n(i, how_much, this->snduPacketData_iter);
+    std::advance(i, how_much);
+    std::advance(this->snduPacketData_iter, how_much);
+    return i;
+}
+
+
+void ULEdec::handleSNDUPacket()
+{
+    this->snduPacket.dump(std::cout);
+    std::cout << "----------------------------------------------------------------------------\n\n";
+    if (this->snduPacket->crc() != this->snduPacket->calcCrc()) {
+        throw ULEdecException("CRC Error");
     }
+//    senf::Packet nextPacket = this->snduPacket.next();
+//    senf::hexdump(
+//            nextPacket.data().begin(),
+//            nextPacket.data().end(),
+//            std::cout);
+    
+}
+
+
+inline bool ULEdec::isSDNUPacketComplete()
+{
+    return this->snduPacketData_iter == this->snduPacket.data().end();
+}
+
+inline ULEdec::iterator::difference_type ULEdec::snduPacketBytesLeft()
+{
+    return std::distance( this->snduPacketData_iter, this->snduPacket.data().end() );
 }
 
 
@@ -192,7 +264,7 @@ int main(int argc, char const * argv[])
 ///////////////////////////////cc.e////////////////////////////////////////
 #undef prefix_
 
-\f
 // Local Variables:
 // mode: c++
 // fill-column: 100