/*
 * 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"
#include "Packet.h"

namespace Mad {
namespace Net {

bool Connection::send(const Packet &packet) {
  if(!isConnected())
    return false;
  
  const unsigned char *data = reinterpret_cast<const unsigned char*>(packet.getRawData());
  unsigned long dataLength = packet.getRawDataLength();
  
  while(dataLength > 0) {
    ssize_t ret = gnutls_record_send(getSession(), data, dataLength);
    
    if(ret < 0) {
      if(ret == GNUTLS_E_INTERRUPTED || ret == GNUTLS_E_AGAIN)
        continue;
      
      return false;
    }
    
    data += ret;
    dataLength -= ret;
  }
  
  return true;
}

bool Connection::receive() {
  unsigned char *headerData = reinterpret_cast<unsigned char*>(&header);
  ssize_t ret;
  
  if(!isConnected())
    return false;
  
  while(true) {
    if(!dataPending())
      return false;
    
    if(read < sizeof(Packet::Data)) {
      ret = gnutls_record_recv(getSession(), headerData+read, sizeof(Packet::Data)-read);
      
      if(ret < 0) {
        if(ret == GNUTLS_E_INTERRUPTED || ret == GNUTLS_E_AGAIN)
          continue;
        
        return false;
      }
      
      read += ret;
      
      if(read < sizeof(Packet::Data))
        continue;
      
      if(!header.length) {
        signal(this, Packet(header.type, header.requestId));
        
        return true;
      }
      
      data = new unsigned char[header.length];
    }
    
    ret = gnutls_record_recv(getSession(), data+read-sizeof(Packet::Data), header.length+sizeof(Packet::Data)-read);
    
    if(ret < 0) {
      if(ret == GNUTLS_E_INTERRUPTED || ret == GNUTLS_E_AGAIN)
        continue;
      
      return false;
    }
    
    read += ret;
    
    if(read < header.length+sizeof(Packet::Data))
      continue;
    
    signal(this, Packet(header.type, header.requestId, data, header.length));
    
    delete [] data;
    data = 0;
    
    return true;
  }
}

}
}