Initial revision

[chise/xemacs-chise.git.1] / netinstall / simpsock.cc

diff --git a/netinstall/simpsock.cc b/netinstall/simpsock.cc
new file mode 100644 (file)
index 0000000..db1f806
--- /dev/null
+++ b/netinstall/simpsock.cc
@@ -0,0 +1,201 @@
+/*
+ * Copyright (c) 2000, Red Hat, Inc.
+ *
+ *     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.
+ *
+ *     A copy of the GNU General Public License can be found at
+ *     http://www.gnu.org/
+ *
+ * Written by DJ Delorie <dj@cygnus.com>
+ *
+ */
+
+/* Simplified socket access functions */
+
+#include "win32.h"
+#include <winsock.h>
+#include <stdio.h>
+#include <stdarg.h>
+#include <stdlib.h>
+
+#include "simpsock.h"
+#include "msg.h"
+
+#define SSBUFSZ 1024
+
+SimpleSocket::SimpleSocket (char *hostname, int port)
+{
+  static int initted = 0;
+  if (!initted)
+    {
+      initted = 1;
+      WSADATA d;
+      WSAStartup (MAKEWORD (1,1), &d);
+    }
+
+  s = INVALID_SOCKET;
+  buf = (char *) malloc (SSBUFSZ + 3);
+  putp = getp = 0;
+
+  int i1, i2, i3, i4;
+  unsigned char ip[4];
+
+  if (sscanf (hostname, "%d.%d.%d.%d", &i1, &i2, &i3, &i4) == 4)
+    {
+      ip[0] = i1;
+      ip[1] = i2;
+      ip[2] = i3;
+      ip[3] = i4;
+    }
+  else
+    {
+      struct hostent *he;
+      he = gethostbyname (hostname);
+      if (!he)
+       {
+         msg ("Can't resolve `%s'\n", hostname);
+         return;
+       }
+      memcpy (ip, he->h_addr_list[0], 4);
+    }
+
+  s = socket (AF_INET, SOCK_STREAM, 0);
+  if (s == INVALID_SOCKET)
+    {
+      msg ("Can't create socket, %d", WSAGetLastError ());
+      return;
+    }
+
+  struct sockaddr_in name;
+
+  memset (&name, 0, sizeof (name));
+  name.sin_family = AF_INET;
+  name.sin_port = htons (port);
+  memcpy (&name.sin_addr, ip, 4);
+
+  if (connect (s, (sockaddr *)&name, sizeof(name)))
+    {
+      msg ("Can't connect to %s:%d", hostname, port);
+      closesocket (s);
+      s = INVALID_SOCKET;
+      return;
+    }
+
+  return;
+}
+
+SimpleSocket::~SimpleSocket ()
+{
+  if (s != INVALID_SOCKET)
+    closesocket (s);
+  s = INVALID_SOCKET;
+  if (buf)
+    free (buf);
+  buf = 0;
+}
+
+int
+SimpleSocket::ok ()
+{
+  if (s == INVALID_SOCKET)
+    return 0;
+  return 1;
+}
+
+int
+SimpleSocket::printf (char *fmt, ...)
+{
+  char localbuf[SSBUFSZ];
+  va_list args;
+  va_start (args, fmt);
+  vsprintf (localbuf, fmt, args);
+  return send (s, localbuf, strlen (localbuf), 0);
+}
+
+int
+SimpleSocket::write (char *localbuf, int len)
+{
+  return send (s, localbuf, len, 0);
+}
+
+int
+SimpleSocket::fill ()
+{
+  if (putp == getp)
+    putp = getp = 0;
+
+  int n = SSBUFSZ - putp;
+  if (n == 0)
+    return 0;
+  int r = recv (s, buf + putp, n, 0);
+  if (r > 0)
+    {
+      putp += r;
+      return r;
+    }
+  return 0;
+}
+
+char *
+SimpleSocket::gets ()
+{
+  if (getp > 0 && putp > getp)
+    {
+      memmove (buf, buf+getp, putp-getp);
+      putp -= getp;
+      getp = 0;
+    }
+  if (putp == getp)
+    fill();
+
+  // getp is zero, always, here, and putp is the count
+  char *nl;
+  while ((nl = (char *)memchr (buf, '\n', putp)) == NULL && putp < SSBUFSZ)
+    if (fill () <= 0)
+      break;
+
+  if (nl)
+    {
+      getp = nl - buf + 1;
+      while ((*nl == '\n' || *nl == '\r') && nl >= buf)
+       *nl-- = 0;
+    }
+  else
+    {
+      getp = putp;
+      nl = buf + putp;
+      nl[1] = 0;
+    }
+
+  return buf;
+}
+
+#define MIN(a,b) ((a) < (b) ? (a) : (b))
+
+int
+SimpleSocket::read (char *ubuf, int ulen)
+{
+  int n, rv=0;
+  if (putp > getp)
+    {
+      n = MIN (ulen, putp-getp);
+      memmove (ubuf, buf+getp, n);
+      getp += n;
+      ubuf += n;
+      ulen -= n;
+      rv += n;
+    }
+  while (ulen > 0)
+    {
+      n = recv (s, ubuf, ulen, 0);
+      if (n <= 0)
+       return rv;
+      ubuf += n;
+      ulen -= n;
+      rv += n;
+    }
+  return rv;
+}