// Copyright (C) 1999,2000 Bruce Guenter <bruceg@em.ca>

//

// 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 2 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, write to the Free Software

// Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA

#include <config.h>

#include "server.h"

#include <stdlib.h>

#include <unistd.h>

#include "un.h"

#include "debug.h"

server_call::~server_call()

{

  trace("server_call::~server_call");

  disconnect();

  delete[] args;

}

static unsigned char* encode_string(unsigned char* ptr, mystring str)

{

  unsigned length = str.length();

  *ptr++ = length >> 8;

  *ptr++ = length & 0xff;

  memcpy(ptr, str.c_str(), length);

  return ptr+length;

}

mystring server_call::build_msg() const

{

  if(argc >= 1<<8)

    return "";

  unsigned msglen = 3 + cmd.length();

  for(unsigned i = 0; i < argc; i++) {

    if(args[i].length() >= 1<<16)

      return "";

    msglen += 2 + args[i].length();

  }

  char buf[msglen+3];

  unsigned char* ptr = (unsigned char*)buf;

  *ptr++ = 2;			// protocol ID

  *ptr++ = msglen >> 8;

  *ptr++ = msglen & 0xff;

  *ptr++ = argc;

  ptr = encode_string(ptr, cmd);

  for(unsigned i = 0; i < argc; i++)

    ptr = encode_string(ptr, args[i]);

  return mystring(buf, msglen+3);

}

static bool send(int fd, mystring msg)

{

  ssize_t written = 0;

  while(written < (ssize_t)msg.length()) {

    ssize_t w = write(fd, msg.c_str()+written, msg.length()-written);

    if(w == 0 || w == -1)

      break;

    written += w;

  }

  return written == (ssize_t)msg.length();

}

int server_call::connect(const mystring& socket_file) const

{

  int s = ::socket(AF_UNIX, SOCK_STREAM, 0);

  if(s == -1)

    return -1;

  size_t size = sizeof(sockaddr_un) + socket_file.length()+1;

  sockaddr_un* saddr = (sockaddr_un*)malloc(size);

  saddr->sun_family = AF_UNIX;

  strcpy(saddr->sun_path, socket_file.c_str());

  if(::connect(s, (sockaddr*)saddr, SUN_LEN(saddr)) == -1)

    return -1;

  free(saddr);

  return s;

}

void server_call::disconnect()

{

  if(fd >= 0)

    close(fd);

}

response server_call::call(const mystring& socket_file)

{

  trace("server_call::call");

  mystring msg = build_msg();

  if(!msg)

    RETURN(bad, "Invalid command data");

  fd = connect(socket_file);

  if(fd == -1)

    RETURN(econn, "Unable to connect to the server");

  if(!send(fd, msg))

    RETURN(econn, "Server aborted the connection");

  return response::read(fd);

}