--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/lib/misc/server.cc	Wed Jan 16 22:39:43 2008 +0100
@@ -0,0 +1,107 @@
+// 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);
+}