Changeset 7851 in orxonox.OLD for branches/network/src/lib/network/network_stream.cc

Timestamp:

May 25, 2006, 1:27:00 PM (18 years ago)

Author:

rennerc

Message:

compression implemented

File:

• branches/network/src/lib/network/network_stream.cc (modified) (9 diffs)

branches/network/src/lib/network/network_stream.cc

@@ -30 +30 @@
 #include "shared_network_data.h"
 #include "message_manager.h"
+#include "preferences.h"
+#include "zip.h"
+
+#include "src/lib/util/loading/resource_manager.h"
 
 #include "network_log.h"
@@ -88 +92 @@
   myHostId = 0;
   currentState = 0;
+  
+  remainingBytesToWriteToDict = Preferences::getInstance()->getInt( "compression", "writedict", 0 );
+  
+  assert( Zip::getInstance()->loadDictionary( "testdict" ) );
 }
 
@@ -365 +373 @@
 void NetworkStream::handleUpstream( )
 {
-  byte buf[UDP_PACKET_SIZE];
   int offset;
   int n;
@@ -430 +437 @@
     assert( Converter::intToByteArray( offset, buf, INTSIZE ) == INTSIZE );
     
-    assert( peer->second.socket->writePacket( buf, offset ) );
+    int compLength = Zip::getInstance()->zip( buf, offset, compBuf, UDP_PACKET_SIZE );
+    
+    if ( compLength < 0 )
+    {
+      PRINTF(1)("compression failed!\n");
+      continue;
+    }
+    
+    assert( peer->second.socket->writePacket( compBuf, compLength ) );
+    
+    if ( this->remainingBytesToWriteToDict > 0 )
+      writeToNewDict( buf, offset );
     
     peer->second.connectionMonitor->processUnzippedOutgoingPacket( buf, offset, currentState );
+    peer->second.connectionMonitor->processZippedOutgoingPacket( compBuf, compLength, currentState );
     
     NETPRINTF(n)("send packet: %d userId = %d\n", offset, peer->second.userId);
@@ -443 +462 @@
 void NetworkStream::handleDownstream( )
 {
-  byte buf[UDP_PACKET_SIZE];
   int offset = 0;
   
   int length = 0;
   int packetLength = 0;
+  int compLength = 0;
   int uniqueId = 0;
   int state = 0;
@@ -460 +479 @@
       continue;
 
-    while ( 0 < (packetLength = peer->second.socket->readPacket( buf, UDP_PACKET_SIZE )) )
-    {
+    while ( 0 < (compLength = peer->second.socket->readPacket( compBuf, UDP_PACKET_SIZE )) )
+    {
+      packetLength = Zip::getInstance()->unZip( compBuf, compLength, buf, UDP_PACKET_SIZE );
+      
       if ( packetLength < 4*INTSIZE )
       {
@@ -468 +489 @@
         continue;
       }
+      
+      if ( this->remainingBytesToWriteToDict > 0 )
+        writeToNewDict( buf, packetLength );
     
       assert( Converter::byteArrayToInt( buf, &length ) == INTSIZE );
@@ -586 +610 @@
       }
       
+      peer->second.connectionMonitor->processZippedIncomingPacket( compBuf, compLength, currentState, ackedState );
       peer->second.connectionMonitor->processUnzippedIncomingPacket( buf, offset, currentState, ackedState );
     
@@ -629 +654 @@
 }
 
-
-
-
-
-
+/**
+ * writes data to DATA/dicts/newdict
+ * @param data pointer to data
+ * @param length length
+ */
+void NetworkStream::writeToNewDict( byte * data, int length )
+{
+  if ( remainingBytesToWriteToDict <= 0 )
+    return;
+  
+  if ( length > remainingBytesToWriteToDict )
+    length = remainingBytesToWriteToDict;
+  
+  std::string fileName = ResourceManager::getInstance()->getDataDir();
+  fileName += "/dicts/newdict";
+  
+  FILE * f = fopen( fileName.c_str(), "a" );
+  
+  if ( !f )
+  {
+    PRINTF(2)("could not open %s\n", fileName.c_str());
+    remainingBytesToWriteToDict = 0;
+    return;
+  }
+  
+  if ( fwrite( data, 1, length, f ) != length )
+  {
+    PRINTF(2)("could not write to file\n");
+    fclose( f );
+    return;
+  }
+  
+  fclose( f );
+  
+  remainingBytesToWriteToDict -= length;  
+}
+
+
+
+
+
+