3 files changed, 31 insertions, 10 deletions

diff --git a/src/Net/Connection.cpp b/src/Net/Connection.cpp
index 768d827..b0f505b 100644
--- a/src/Net/Connection.cpp
+++ b/src/Net/Connection.cpp
@@ -18,6 +18,7 @@
  */
 
 #include "Connection.h"
+#include <cstring>
 
 namespace Mad {
 namespace Net {
@@ -94,23 +95,38 @@ bool Connection::rawReceive(unsigned long length,
   return true;
 }
 
-bool Connection::rawSend(const unsigned char *data, unsigned long length) {
+void Connection::doSend() {
   if(!isConnected())
-    return false;
+    return;
   
-  while(length > 0) {
-    ssize_t ret = gnutls_record_send(getSession(), data, length);
+  while(!sendQueueEmpty()) {
+    ssize_t ret = gnutls_record_send(getSession(), transS.front().data+transS.front().transmitted, 
+        transS.front().length-transS.front().transmitted);
     
     if(ret < 0) {
       if(ret == GNUTLS_E_INTERRUPTED || ret == GNUTLS_E_AGAIN)
-        continue;
+        return;
       
-      return false;
+      // Error... disconnect?
+      return;
     }
     
-    data += ret;
-    length -= ret;
+    transS.front().transmitted += ret;
+    
+    if(transS.front().transmitted == transS.front().length) {
+      delete [] transS.front().data;
+      transS.pop();
+    }
   }
+}
+
+bool Connection::rawSend(const unsigned char *data, unsigned long length) {
+  if(!isConnected())
+    return false;
+  
+  Transmission trans = {length, 0, new unsigned char[length], sigc::slot<void,const void*,unsigned long>()};
+  std::memcpy(trans.data, data, length);
+  transS.push(trans);
   
   return true;
 }
diff --git a/src/Net/Connection.h b/src/Net/Connection.h
index 7e73955..d13c2f6 100644
--- a/src/Net/Connection.h
+++ b/src/Net/Connection.h
@@ -43,6 +43,7 @@ class Connection {
     };
     
     Transmission transR;
+    std::queue<Transmission> transS;
     
     Packet::Data header;
     
@@ -52,6 +53,7 @@ class Connection {
     void packetDataReceiveHandler(const void *data, unsigned long length);
     
     void doReceive();
+    void doSend();
     
     bool receiveComplete() const {
       return (transR.length == transR.transmitted);
@@ -84,10 +86,13 @@ class Connection {
       return rawSend(reinterpret_cast<const unsigned char*>(packet.getRawData()), packet.getRawDataLength());
     }
     
-    void sendreceive() {
+    void sendReceive() {
       doReceive();
+      doSend();
     }
     
+    bool sendQueueEmpty() {return transS.empty();}
+    
     sigc::signal<void,const Connection*,const Packet&> signalReceive() const {return signal;}
     
     static void init() {
diff --git a/src/Net/ServerConnection.cpp b/src/Net/ServerConnection.cpp
index 66274b4..afbfcbb 100644
--- a/src/Net/ServerConnection.cpp
+++ b/src/Net/ServerConnection.cpp
@@ -21,8 +21,8 @@
 #include "IPAddress.h"
 #include <cstring>
 #include <cerrno>
-#include <fstream>
 #include <sys/socket.h>
+#include <sys/select.h>
 
 namespace Mad {
 namespace Net {