/*
 * Connection.cpp
 *
 * Copyright (C) 2008 Matthias Schiffer <matthias@gamezock.de>
 *
 * This program is free software: you can redistribute it and/or modify it
 * under the terms of the GNU General Public License as published by the
 * Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful, but
 * WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
 * See the GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License along
 * with this program. If not, see <http://www.gnu.org/licenses/>.
 */

#include "Connection.h"

namespace Mad {
namespace Net {

void Connection::packetHeaderReceiveHandler(const void *data, unsigned long length) {
  if(length != sizeof(Packet::Data))
    return; // Error... disconnect?
  
  header = *reinterpret_cast<const Packet::Data*>(data);
  
  if(header.length == 0) {
    signal(this, Packet(header.type, header.requestId));
    
    enterReceiveLoop();
  }
  else {
    rawReceive(header.length, sigc::mem_fun(this, &Connection::packetDataReceiveHandler));
  }
}

void Connection::packetDataReceiveHandler(const void *data, unsigned long length) {
  if(length != header.length)
    return; // Error... disconnect?
  
  signal(this, Packet(header.type, header.requestId, data, length));
  
  enterReceiveLoop();
}

void Connection::doReceive() {
  if(!isConnected())
    return;
  
  if(!dataPending())
    return;
  
  if(receiveComplete())
    return;
  
  ssize_t ret = gnutls_record_recv(getSession(), transR.data+transR.transmitted, transR.length-transR.transmitted);
  
  if(ret < 0) {
    if(ret == GNUTLS_E_INTERRUPTED || ret == GNUTLS_E_AGAIN)
      return;
    
    // Error... disconnect?
    return;
  }
  
  transR.transmitted += ret;
  
  if(receiveComplete()) {
    transR.notify(transR.data, transR.length);
    
    delete [] transR.data;
    transR.data = 0;
  }
}

bool Connection::rawReceive(unsigned long length,
    const sigc::slot<void,const void*,unsigned long> &notify)
{
  if(!isConnected())
    return false;
  
  if(!receiveComplete())
    return false;
  
  transR.data = new unsigned char[length];
  transR.length = length;
  transR.transmitted = 0;
  transR.notify = notify;
  
  return true;
}

bool Connection::rawSend(const unsigned char *data, unsigned long length) {
  if(!isConnected())
    return false;
  
  while(length > 0) {
    ssize_t ret = gnutls_record_send(getSession(), data, length);
    
    if(ret < 0) {
      if(ret == GNUTLS_E_INTERRUPTED || ret == GNUTLS_E_AGAIN)
        continue;
      
      return false;
    }
    
    data += ret;
    length -= ret;
  }
  
  return true;
}

}
}