#include "msg_logger.h"
#include "exclusive_file.h"

#include "AmUtils.h"

#include <fcntl.h>
#include <netinet/in.h>
#include <arpa/inet.h>

file_msg_logger::~file_msg_logger()
{
  exclusive_file::close(excl_fp);
}

int file_msg_logger::open(const char* filename)
{
  if(excl_fp) return 0;

  bool is_new = false;
  if(exclusive_file::open(filename,excl_fp,is_new) < 0) {
    return -1;
  }

  if (is_new) {
    write_file_header();

    // exclusive files are created locked
    excl_fp->unlock();
  }

  return 0;
}

int file_msg_logger::write(const void *buf, int len)
{
  assert(excl_fp != NULL);
  return excl_fp->write(buf,len);
}

int file_msg_logger::writev(const struct iovec *iov, int iovcnt)
{
  assert(excl_fp != NULL);
  return excl_fp->writev(iov,iovcnt);
}

//////////////////////////////////////////////////////////////////////////////////////

static string addr2str(sockaddr_storage* addr)
{
  char ntop_buffer[INET6_ADDRSTRLEN];

  if(addr->ss_family == AF_INET) {
    struct sockaddr_in *sin = (struct sockaddr_in *)addr;
    if(!inet_ntop(AF_INET, &sin->sin_addr,
		  ntop_buffer,INET6_ADDRSTRLEN)) {
      ERROR("Could not convert IPv4 address to string: %s",strerror(errno));
      return "unknown";
    }
    
    return string(ntop_buffer) + ":" + int2str(ntohs(sin->sin_port));
  }

  struct sockaddr_in6* sin6 = (struct sockaddr_in6 *)addr;
  if(!inet_ntop(AF_INET6, &sin6->sin6_addr,
		ntop_buffer,INET6_ADDRSTRLEN)) {
    ERROR("Could not convert IPv6 address to string: %s",strerror(errno));
    return "unknown";
  }
  
  return string(ntop_buffer) + ":" + int2str(ntohs(sin6->sin6_port));
}

#define WRITE_CSTSTR(str)                                           \
  if(write(str,sizeof(str)-1) != sizeof(str)-1) {                   \
    return -1;                                                      \
  }

#define WRITE_STLSTR(str)                                           \
  if(write(str.c_str(),str.length()) != (ssize_t)str.length())  {   \
    return -1;                                                      \
  }


int cf_msg_logger::write_src_dst(const string& obj)
{
  if (known_destinations.find(obj) == known_destinations.end()) {
    known_destinations.insert(obj);
    WRITE_CSTSTR("<object name='");
    WRITE_STLSTR(obj);
    WRITE_CSTSTR("' desc='");
    WRITE_STLSTR(obj);
    WRITE_CSTSTR("'/>\n");
  }

  return 0;
}

int cf_msg_logger::log(const char* buf, int len,
			 sockaddr_storage* src_ip,
			 sockaddr_storage* dst_ip,
			 cstring method, int reply_code)
{
  assert(excl_fp != NULL);

  string src = addr2str(src_ip);
  string dst = addr2str(dst_ip);

  lock_guard<exclusive_file> _l(*excl_fp);

  write_src_dst(src);
  write_src_dst(dst);
  
  string what = c2stlstr(method);
  if(reply_code > 0) {
    what = int2str(reply_code) + " / " + what;
  }

  WRITE_CSTSTR("<call src='");
  WRITE_STLSTR(src);
  WRITE_CSTSTR("' dst='");
  WRITE_STLSTR(dst);
  WRITE_CSTSTR("' desc='");
  WRITE_STLSTR(what);
  WRITE_CSTSTR("'>\n");

  if(write(buf,len) != len) return -1;

  WRITE_CSTSTR("</call>\n");
 
  return 0;
}