3 files changed, 85 insertions, 43 deletions

diff --git a/noncore/net/wellenreiter/gui/scanlist.cpp b/noncore/net/wellenreiter/gui/scanlist.cpp
index a8cadd8..7733934 100644
--- a/noncore/net/wellenreiter/gui/scanlist.cpp
+++ b/noncore/net/wellenreiter/gui/scanlist.cpp
@@ -188,60 +188,94 @@ void MScanListView::addNewItem( QString type, QString essid, QString macaddr, bo
     {
         s.sprintf( "(i) new adhoc station in '%s' [%d]", (const char*) essid, channel );
     }
 
 }
 
-void MScanListView::traffic( QString type, QString from, QString to, QString via, QString additional )
+
+void MScanListView::addIfNotExisting( MScanListItem* network, QString addr )
 {
-    if ( type != "toDS" ) return;
+    MScanListItem* subitem = static_cast<MScanListItem*>( network->firstChild() );
+
+    while ( subitem && ( subitem->text( col_ap ) != addr ) )
+    {
+        qDebug( "subitemtext: %s", (const char*) subitem->text( col_ap ) );
+        subitem = static_cast<MScanListItem*> ( subitem->itemBelow() );
+    }
+
+    if ( subitem )
+    {
+        // we have already seen this item, it's a dupe
+        #ifdef DEBUG
+        qDebug( "%s is a dupe - ignoring...", (const char*) addr );
+        #endif
+        subitem->receivedBeacon(); //FIXME: sent data bit
+        return;
+    }
+
+    // Hey, it seems to be a new item :-D
+    MScanListItem* station = new MScanListItem( network, "station", /* network->text( col_essid ) */ "", addr, false, -1, -1 );
+    if ( _manufacturerdb )
+    station->setManufacturer( _manufacturerdb->lookup( addr ) );
+}
 
-    qDebug( "MScanList::traffic( [%s] | %s -> %s (via %s)",
-        (const char*) type, (const char*) from,
-        (const char*) to, (const char*) via );
 
+void MScanListView::WDStraffic( QString from, QString to, QString viaFrom, QString viaTo )
+{
     QString s;
     MScanListItem* network;
 
     QListViewItemIterator it( this );
-    while ( it.current() && it.current()->text( col_ap ) != via ) ++it;
+    while ( it.current() &&
+            it.current()->text( col_ap ) != viaFrom &&
+            it.current()->text( col_ap ) != viaTo ) ++it;
 
     MScanListItem* item = static_cast<MScanListItem*>( it.current() );
 
-    if ( item ) // AP has been shown up, so just add our new "from" - station
+    if ( item ) // Either viaFrom or viaTo AP has shown up yet, so just add our two new stations
+    {
+        addIfNotExisting( static_cast<MScanListItem*>(item->parent()), from );
+        addIfNotExisting( static_cast<MScanListItem*>(item->parent()), to );
+    }
+    else
     {
-        network = static_cast<MScanListItem*>( item->parent() );
-        MScanListItem* subitem = static_cast<MScanListItem*>( network->firstChild() );
+        qDebug( "D'Oh! Stations without AP... ignoring for now... will handle this in 1.1 version :-D" );
+    }
+}
 
-        while ( subitem && ( subitem->text( col_ap ) != from ) )
-        {
-            qDebug( "subitemtext: %s", (const char*) subitem->text( col_ap ) );
-            subitem = static_cast<MScanListItem*> ( subitem->itemBelow() );
-        }
 
-        if ( subitem )
-        {
-            // we have already seen this item, it's a dupe
-            #ifdef DEBUG
-            qDebug( "%s is a dupe - ignoring...", (const char*) from );
-            #endif
-            subitem->receivedBeacon(); //FIXME: sent data bit
-            return;
-        }
+void MScanListView::toDStraffic( QString from, QString to, QString via )
+{
+    QString s;
+    MScanListItem* network;
+
+    QListViewItemIterator it( this );
+    while ( it.current() && it.current()->text( col_ap ) != via ) ++it;
+
+    MScanListItem* item = static_cast<MScanListItem*>( it.current() );
 
-        // Hey, it seems to be a new item :-D
-        MScanListItem* station = new MScanListItem( item->parent(), "adhoc", /* network->text( col_essid ) */ "", from, false, -1, -1 );
-        if ( _manufacturerdb )
-          station->setManufacturer( _manufacturerdb->lookup( from ) );
+    if ( item ) // AP has shown up yet, so just add our new "from" - station
+    {
+        addIfNotExisting( static_cast<MScanListItem*>(item->parent()), from );
     }
     else
     {
-        qDebug( "D'Oh! Station without AP... ignoring for now... will handle this in alpha-4 version :-D" );
+        qDebug( "D'Oh! Station without AP... ignoring for now... will handle this in 1.1 :-D" );
     }
 }
 
+void MScanListView::fromDStraffic( QString from, QString to, QString via )
+{
+    qWarning( "D'oh! Not yet implemented..." );
+}
+
+void MScanListView::IBSStraffic( QString from, QString to, QString via )
+{
+    qWarning( "D'oh! Not yet implemented..." );
+}
+
 //============================================================
 // MScanListItem
 //============================================================
 
 MScanListItem::MScanListItem( QListView* parent, QString type, QString essid, QString macaddr,
                               bool wep, int channel, int signal )
diff --git a/noncore/net/wellenreiter/gui/scanlist.h b/noncore/net/wellenreiter/gui/scanlist.h
index bed69f1..a14d426 100644
--- a/noncore/net/wellenreiter/gui/scanlist.h
+++ b/noncore/net/wellenreiter/gui/scanlist.h
@@ -20,12 +20,14 @@
 
 #include <qtextstream.h>
 
 class QString;
 class ManufacturerDB;
 
+class MScanListItem;
+
 class MScanListView: public OListView
 {
   Q_OBJECT
 
   public:
     MScanListView( QWidget* parent = 0, const char* name = 0 );
@@ -36,13 +38,19 @@ class MScanListView: public OListView
     virtual OListViewItem* childFactory();
     virtual void serializeTo( QDataStream& s ) const;
     virtual void serializeFrom( QDataStream& s );
 
   public slots:
     void addNewItem( QString type, QString essid, QString macaddr, bool wep, int channel, int signal );
-    void traffic( QString type, QString from, QString to, QString via, QString additional = QString::null );
+    void fromDStraffic( QString from, QString to, QString via ); // NYI
+    void toDStraffic( QString from, QString to, QString via );
+    void WDStraffic( QString from, QString to, QString viaFrom, QString viaTo );
+    void IBSStraffic( QString from, QString to, QString via ); // NYI
+
+  protected:
+    void addIfNotExisting( MScanListItem* parent, QString addr );
 
   private:
     ManufacturerDB* _manufacturerdb;
 
 };
 
diff --git a/noncore/net/wellenreiter/gui/wellenreiter.cpp b/noncore/net/wellenreiter/gui/wellenreiter.cpp
index 5ec9ee4..4b82c9a 100644
--- a/noncore/net/wellenreiter/gui/wellenreiter.cpp
+++ b/noncore/net/wellenreiter/gui/wellenreiter.cpp
@@ -177,49 +177,49 @@ void Wellenreiter::receivePacket(OPacket* p)
         if ( wlan->fromDS() && !wlan->toDS() )
         {
             qDebug( "FromDS traffic: '%s' -> '%s' via '%s'",
                 (const char*) wlan->macAddress3().toString(true),
                 (const char*) wlan->macAddress1().toString(true),
                 (const char*) wlan->macAddress2().toString(true) );
-            netView()->traffic( "fromDS", wlan->macAddress3().toString(),
-                                        wlan->macAddress1().toString(),
-                                        wlan->macAddress2().toString() );
+            netView()->fromDStraffic( wlan->macAddress3().toString(),
+                                      wlan->macAddress1().toString(),
+                                      wlan->macAddress2().toString() );
         }
         else
         if ( !wlan->fromDS() && wlan->toDS() )
         {
             qDebug( "ToDS traffic: '%s' -> '%s' via '%s'",
                 (const char*) wlan->macAddress2().toString(true),
                 (const char*) wlan->macAddress3().toString(true),
                 (const char*) wlan->macAddress1().toString(true) );
-            netView()->traffic( "toDS", wlan->macAddress2().toString(),
-                                      wlan->macAddress3().toString(),
-                                      wlan->macAddress1().toString() );
+            netView()->toDStraffic( wlan->macAddress2().toString(),
+                                    wlan->macAddress3().toString(),
+                                    wlan->macAddress1().toString() );
         }
         else
         if ( wlan->fromDS() && wlan->toDS() )
         {
-            qDebug( "WSD(bridge) traffic: '%s' -> '%s' via '%s' and '%s'",
+            qDebug( "WDS(bridge) traffic: '%s' -> '%s' via '%s' and '%s'",
                 (const char*) wlan->macAddress4().toString(true),
                 (const char*) wlan->macAddress3().toString(true),
                 (const char*) wlan->macAddress1().toString(true),
                 (const char*) wlan->macAddress2().toString(true) );
-            netView()->traffic( "WSD", wlan->macAddress4().toString(),
-                                     wlan->macAddress3().toString(),
-                                     wlan->macAddress1().toString(),
-                                     wlan->macAddress2().toString() );
+            netView()->WDStraffic( wlan->macAddress4().toString(),
+                                   wlan->macAddress3().toString(),
+                                   wlan->macAddress1().toString(),
+                                   wlan->macAddress2().toString() );
         }
         else
         {
             qDebug( "IBSS(AdHoc) traffic: '%s' -> '%s' (Cell: '%s')'",
                 (const char*) wlan->macAddress2().toString(true),
                 (const char*) wlan->macAddress1().toString(true),
                 (const char*) wlan->macAddress3().toString(true) );
-            netView()->traffic( "IBSS", wlan->macAddress2().toString(),
-                                        wlan->macAddress1().toString(),
-                                        wlan->macAddress3().toString() );
+            netView()->IBSStraffic( wlan->macAddress2().toString(),
+                                    wlan->macAddress1().toString(),
+                                    wlan->macAddress3().toString() );
         }
         return;
     }
 }