1 files changed, 454 insertions, 0 deletions
diff --git a/login/utmp_daemon.c b/login/utmp_daemon.c
new file mode 100644
index 0000000000..edaade1eff
--- /dev/null
+++ b/login/utmp_daemon.c
@@ -0,0 +1,454 @@
+/* Copyright (C) 1997 Free Software Foundation, Inc.
+   This file is part of the GNU C Library.
+   Contributed by Mark Kettenis <kettenis@phys.uva.nl>, 1997.
+
+   The GNU C Library is free software; you can redistribute it and/or
+   modify it under the terms of the GNU Library General Public License as
+   published by the Free Software Foundation; either version 2 of the
+   License, or (at your option) any later version.
+
+   The GNU C Library 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
+   Library General Public License for more details.
+
+   You should have received a copy of the GNU Library General Public
+   License along with the GNU C Library; see the file COPYING.LIB.  If not,
+   write to the Free Software Foundation, Inc., 59 Temple Place - Suite 330,
+   Boston, MA 02111-1307, USA.  */
+
+#include <errno.h>
+#include <limits.h>
+#include <stdlib.h>
+#include <string.h>
+#include <sys/socket.h>
+#include <sys/un.h>
+#include <unistd.h>
+#include <utmp.h>
+
+#include "utmp-private.h"
+#include "utmpd/utmpd.h"
+
+#ifndef _LIBC
+#define __set_errno(val) errno = (val)
+#endif
+
+
+/* Descriptor for the socket.  */
+static int daemon_sock = INT_MIN;
+
+
+/* Functions defined here.  */
+static int setutent_daemon (int reset);
+static int getutent_r_daemon (struct utmp *buffer, struct utmp **result);
+static int getutid_r_daemon (const struct utmp *line, struct utmp *buffer,
+			     struct utmp **result);
+static int getutline_r_daemon (const struct utmp *id, struct utmp *buffer,
+			       struct utmp **result);
+static struct utmp *pututline_daemon (const struct utmp *utmp);
+static void endutent_daemon (void);
+static int updwtmp_daemon (const char *file, const struct utmp *utmp);
+
+/* Jump table for daemon functions.  */
+struct utfuncs __libc_utmp_daemon_functions =
+{
+  setutent_daemon,
+  getutent_r_daemon,
+  getutid_r_daemon,
+  getutline_r_daemon,
+  pututline_daemon,
+  endutent_daemon,
+  updwtmp_daemon
+};
+
+static int do_setutent (int sock);
+static int do_getutent (int sock, struct utmp *buffer);
+static int do_endutent (int sock);
+static int do_getutline (int sock, const struct utmp *line,
+			 struct utmp *buffer);
+static int do_getutid (int sock, const struct utmp *id,
+			    struct utmp *buffer);
+static int do_pututline (int sock, const struct utmp *utmp);
+static int do_updwtmp (int sock, const char *file,
+		       const struct utmp *utmp);
+
+static int open_socket (const char *name);
+static int send_request (int sock, const request_header *request,
+			 reply_header *reply);
+
+
+static int
+setutent_daemon (int reset)
+{
+  if (daemon_sock == INT_MIN)
+    {
+      daemon_sock = open_socket (_PATH_UTMPD_RW);
+      if (daemon_sock < 0)
+	{
+	  /* Hhm, read-write access did not work.  Try read-only.  */
+	  daemon_sock = open_socket (_PATH_UTMPD_RO);
+	  if (daemon_sock < 0)
+	    return 0;
+	}
+
+      /* Send request to the daemon.  */
+      if (do_setutent (daemon_sock) < 0)
+	return 0;
+    }
+  else if (reset)
+    {
+      /* Send request to the daemon.  */
+      if (do_setutent (daemon_sock) < 0)
+	return 0;
+    }
+
+  return 1;
+}
+
+
+static void
+endutent_daemon (void)
+{
+  if (daemon_sock >= 0)
+    {
+      /* Send request to the daemon.  */
+      do_endutent (daemon_sock);
+      close (daemon_sock);
+    }
+
+  daemon_sock = INT_MIN;
+}
+
+
+static int
+getutent_r_daemon (struct utmp *buffer, struct utmp **result)
+{
+  /* Open connection if not already done.  */
+  if (daemon_sock == INT_MIN)
+    setutent_daemon (1);
+
+  if (daemon_sock < 0)
+    {
+      /* Not available.  */
+      *result = NULL;
+      return -1;
+    }
+
+  /* Send request to the daemon.  */
+  if (do_getutent (daemon_sock, buffer) < 0)
+    {
+      *result = NULL;
+      return -1;;
+    }
+
+  *result = buffer;
+  return 0;
+}
+
+
+static int
+getutline_r_daemon (const struct utmp *line, struct utmp *buffer,
+		    struct utmp **result)
+{
+  if (daemon_sock < 0)
+    {
+      *result = NULL;
+      return -1;
+    }
+
+  /* Send request to the daemon.  */
+  if (do_getutline (daemon_sock, line, buffer) < 0)
+    {
+      *result = NULL;
+      return -1;;
+    }
+
+  *result = buffer;
+  return 0;
+}
+
+
+static int
+getutid_r_daemon (const struct utmp *id, struct utmp *buffer,
+		  struct utmp **result)
+{
+  if (daemon_sock < 0)
+    {
+      *result = NULL;
+      return -1;
+    }
+
+  /* Send request to the daemon.  */
+  if (do_getutid (daemon_sock, id, buffer) < 0)
+    {
+      *result = NULL;
+      return -1;
+    }
+
+  *result = buffer;
+  return 0;
+}
+
+
+static struct utmp *
+pututline_daemon (const struct utmp *utmp)
+{
+  if (daemon_sock == INT_MIN)
+    /* The connection is closed.  Open it again.  */
+    setutent_daemon (0);
+
+  if (daemon_sock < 0)
+    /* Something went wrong.  */
+    return NULL;
+
+  /* Send request to the daemon.  */
+  if (do_pututline (daemon_sock, utmp) < 0)
+    return NULL;
+
+  return (struct utmp *)utmp;
+}
+
+
+static int
+updwtmp_daemon (const char *file, const struct utmp *utmp)
+{
+  int sock;
+
+  /* Only try to open for both reading and writing.  */
+  sock = open_socket (_PATH_UTMPD_RW);
+  if (sock < 0)
+    return -1;
+
+  /* Send request to the daemon.  */
+  if (do_updwtmp (sock, file, utmp) < 0)
+    return -1;
+
+  close (sock);
+  return 0;
+}
+
+
+static int
+do_setutent (int sock)
+{
+  setutent_request request;
+  setutent_reply reply;
+
+  request.header.version = UTMPD_VERSION;
+  request.header.size = sizeof (setutent_request);
+  request.header.type = UTMPD_REQ_SETUTENT;
+  strncpy (request.file, __libc_utmp_file_name, sizeof request.file);
+
+  reply.header.version = UTMPD_VERSION;
+  reply.header.size = sizeof (setutent_reply);
+  reply.header.type = UTMPD_REQ_SETUTENT;
+
+  if (send_request (sock, &request.header, &reply.header) < 0)
+    return -1;
+
+  if (reply.result < 0)
+    __set_errno (reply.errnum);
+
+  return reply.result;
+}
+
+static int
+do_getutent (int sock, struct utmp *buffer)
+{
+  getutent_request request;
+  getutent_reply reply;
+
+  request.header.version = UTMPD_VERSION;
+  request.header.size = sizeof (getutent_request);
+  request.header.type = UTMPD_REQ_GETUTENT;
+
+  reply.header.version = UTMPD_VERSION;
+  reply.header.size = sizeof (getutent_reply);
+  reply.header.type = UTMPD_REQ_GETUTENT;
+
+  if (send_request (sock, &request.header, &reply.header) < 0)
+    return -1;
+
+  if (reply.result < 0)
+    __set_errno (reply.errnum);
+  else
+    memcpy (buffer, &reply.entry, sizeof (struct utmp));
+
+  return reply.result;
+}
+
+static int
+do_endutent (int sock)
+{
+  endutent_request request;
+  endutent_reply reply;
+
+  request.header.version = UTMPD_VERSION;
+  request.header.size = sizeof (endutent_request);
+  request.header.type = UTMPD_REQ_ENDUTENT;
+
+  reply.header.version = UTMPD_VERSION;
+  reply.header.size = sizeof (endutent_reply);
+  reply.header.type = UTMPD_REQ_ENDUTENT;
+
+  if (send_request (sock, &request.header, &reply.header) < 0)
+    return -1;
+
+  if (reply.result < 0)
+    __set_errno (reply.errnum);
+
+  return reply.result;
+}
+
+static int
+do_getutline (int sock, const struct utmp *line, struct utmp *buffer)
+{
+  getutline_request request;
+  getutline_reply reply;
+
+  request.header.version = UTMPD_VERSION;
+  request.header.size = sizeof (getutline_request);
+  request.header.type = UTMPD_REQ_GETUTLINE;
+  memcpy (&request.line, line, sizeof (struct utmp));
+
+  reply.header.version = UTMPD_VERSION;
+  reply.header.size = sizeof (getutline_reply);
+  reply.header.type = UTMPD_REQ_GETUTLINE;
+
+  if (send_request (sock, &request.header, &reply.header) < 0)
+    return -1;
+
+  if (reply.result < 0)
+    __set_errno (reply.errnum);
+  else
+    memcpy (buffer, &reply.entry, sizeof (struct utmp));
+
+  return reply.result;
+}
+
+static int
+do_getutid (int sock, const struct utmp *id, struct utmp *buffer)
+{
+  getutid_request request;
+  getutid_reply reply;
+
+  request.header.version = UTMPD_VERSION;
+  request.header.size = sizeof (getutid_request);
+  request.header.type = UTMPD_REQ_GETUTID;
+  memcpy (&request.id, id, sizeof (struct utmp));
+
+  reply.header.version = UTMPD_VERSION;
+  reply.header.size = sizeof (getutid_reply);
+  reply.header.type = UTMPD_REQ_GETUTID;
+
+  if (send_request (sock, &request.header, &reply.header) < 0)
+    return -1;
+
+  if (reply.result < 0)
+    __set_errno (reply.errnum);
+  else
+    memcpy (buffer, &reply.entry, sizeof (struct utmp));
+
+  return reply.result;
+}
+
+static int
+do_pututline (int sock, const struct utmp *utmp)
+{
+  pututline_request request;
+  pututline_reply reply;
+
+  request.header.version = UTMPD_VERSION;
+  request.header.size = sizeof (pututline_request);
+  request.header.type = UTMPD_REQ_PUTUTLINE;
+  memcpy (&request.utmp, utmp, sizeof (struct utmp));
+
+  reply.header.version = UTMPD_VERSION;
+  reply.header.size = sizeof (pututline_reply);
+  reply.header.type = UTMPD_REQ_PUTUTLINE;
+
+  if (send_request (sock, &request.header, &reply.header) < 0)
+    return -1;
+
+  if (reply.result < 0)
+    __set_errno (reply.errnum);
+
+  return reply.result;
+}
+
+static int
+do_updwtmp (int sock, const char *file, const struct utmp *utmp)
+{
+  updwtmp_request request;
+  updwtmp_reply reply;
+
+  request.header.version = UTMPD_VERSION;
+  request.header.size = sizeof (updwtmp_request);
+  request.header.type = UTMPD_REQ_UPDWTMP;
+  strncpy (request.file, file, sizeof request.file);
+  memcpy (&request.utmp, utmp, sizeof (struct utmp));
+
+  reply.header.version = UTMPD_VERSION;
+  reply.header.size = sizeof (updwtmp_reply);
+  reply.header.type = UTMPD_REQ_UPDWTMP;
+
+  if (send_request (sock, &request.header, &reply.header) < 0)
+    return -1;
+
+  if (reply.result < 0)
+    __set_errno (reply.errnum);
+
+  return reply.result;
+}
+
+
+/* Create a socket connected to NAME.  */
+static int
+open_socket (const char *name)
+{
+  struct sockaddr_un addr;
+  int sock;
+
+  sock = socket (PF_UNIX, SOCK_STREAM, 0);
+  if (sock < 0)
+    return -1;
+
+  addr.sun_family = AF_UNIX;
+  strcpy (addr.sun_path, name);
+  if (connect (sock, (struct sockaddr *) &addr, sizeof (addr)) < 0)
+    {
+      close (sock);
+      return -1;
+    }
+
+  return sock;
+}
+
+/* Send REQUEST to SOCK, and wait for reply.  Returns 0 if successful,
+   storing the reply in REPLY, and -1 if not.  */
+static int
+send_request (int sock, const request_header *request,
+	      reply_header *reply)
+{
+  reply_header header;
+  ssize_t nbytes;
+
+  nbytes = write (sock, request, request->size);
+  if (nbytes != (ssize_t) request->size)
+    return -1;
+
+  nbytes = read (sock, &header, sizeof (reply_header));
+  if (nbytes != sizeof (reply_header))
+    return -1;
+
+  if (reply->version != header.version
+      || reply->size != header.size
+      || reply->type != header.type)
+    return -1;
+
+  nbytes = read (sock, reply + 1, reply->size - sizeof (reply_header));
+  if (nbytes != (ssize_t) (reply->size - sizeof (reply_header)))
+    return -1;
+
+  return 0;
+}