Initial revision

2 files changed, 202 insertions, 0 deletions

diff --git a/libps/tty.c b/libps/tty.c
new file mode 100644
index 00000000..be3c8ce6
--- /dev/null
+++ b/libps/tty.c
@@ -0,0 +1,85 @@
+/* The type ps_tty_t, for per-tty info.
+
+   Copyright (C) 1995 Free Software Foundation, Inc.
+
+   Written by Miles Bader <miles@gnu.ai.mit.edu>
+
+   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, 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., 675 Mass Ave, Cambridge, MA 02139, USA. */
+
+#include <hurd.h>
+#include <stdio.h>
+#include <stdlib.h>
+#include <assert.h>
+
+#include "ps.h"
+#include "common.h"
+
+/* ---------------------------------------------------------------- */
+
+/* Create a ps_tty_t for the tty referred to by PORT, returning it in TTY.
+   If a memory allocation error occurs, ENOMEM is returned, otherwise 0.  */
+error_t
+ps_tty_create(file_t port, ps_tty_t *tty)
+{
+  *tty = NEW(struct ps_tty);
+  if (*tty == NULL)
+    return ENOMEM;
+
+  (*tty)->port = port;
+  (*tty)->name_state = PS_TTY_NAME_PENDING;
+
+  return 0;
+}
+
+/* Frees TTY and any resources it consumes.  */
+void
+ps_tty_free(ps_tty_t tty)
+{
+  mach_port_deallocate(mach_task_self(), tty->port);
+  if (tty->name_state == PS_TTY_NAME_OK && tty->name != NULL)
+    free(tty->name);
+  free(tty);
+}
+
+/* ---------------------------------------------------------------- */
+
+/* Returns the name of the tty, or NULL if it can't be figured out.  */
+char *ps_tty_name(ps_tty_t tty)
+{
+  if (tty->name_state == PS_TTY_NAME_PENDING)
+    {
+      string_t buf;
+
+      if (term_get_nodename(tty->port, &buf) != 0)
+	/* There is a terminal there, but we can't figure out its name.  */
+	tty->name_state = PS_TTY_NAME_ERROR;
+      else
+	{
+	  tty->name = NEWVEC(char, strlen(buf) + 1);
+	  if (tty->name == NULL)
+	    tty->name_state = PS_TTY_NAME_ERROR;
+	  else
+	    {
+	      strcpy(tty->name, buf);
+	      tty->name_state = PS_TTY_NAME_OK;
+	    }
+	}
+    }
+
+  if (tty->name_state == PS_TTY_NAME_OK)
+    return tty->name;
+  else
+    return NULL;
+}
diff --git a/libps/user.c b/libps/user.c
new file mode 100644
index 00000000..b91cc459
--- /dev/null
+++ b/libps/user.c
@@ -0,0 +1,117 @@
+/* The type ps_user_t, for per-user info.
+
+   Copyright (C) 1995 Free Software Foundation, Inc.
+
+   Written by Miles Bader <miles@gnu.ai.mit.edu>
+
+   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, 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., 675 Mass Ave, Cambridge, MA 02139, USA. */
+
+#include <hurd.h>
+#include <stdio.h>
+#include <stdlib.h>
+#include <assert.h>
+
+#include "ps.h"
+#include "common.h"
+
+/* ---------------------------------------------------------------- */
+
+/* Create a ps_user_t for the user referred to by UID, returning it in U.
+   If a memory allocation error occurs, ENOMEM is returned, otherwise 0.  */
+error_t
+ps_user_create(int uid, ps_user_t *u)
+{
+  *u = NEW(struct ps_user);
+  if (*u == NULL)
+    return ENOMEM;
+
+  (*u)->uid = uid;
+  (*u)->passwd_state = PS_USER_PASSWD_PENDING;
+
+  return 0;
+}
+
+/* Free U and any resources it consumes.  */
+void
+ps_user_free(ps_user_t u)
+{
+  if (u->passwd_state == PS_USER_PASSWD_OK)
+    free(u->storage);
+  free(u);
+}
+
+/* ---------------------------------------------------------------- */
+
+/* Returns the password file entry (struct passwd, from <pwd.h>) for the user
+   referred to by U, or NULL if it can't be gotten.  */
+struct passwd *ps_user_passwd(ps_user_t u)
+{
+  if (u->passwd_state == PS_USER_PASSWD_OK)
+    return &u->passwd;
+  else if (u->passwd_state == PS_USER_PASSWD_ERROR)
+    return NULL;
+  else
+    {
+      struct passwd *pw = getpwuid(u->uid);
+      if (pw != NULL)
+	{
+	  int needed = 0;
+
+#define COUNT(field) (pw->field == NULL || (needed += strlen(pw->field) + 1))
+	  COUNT(pw_name);
+	  COUNT(pw_passwd);
+	  COUNT(pw_gecos);
+	  COUNT(pw_dir);
+	  COUNT(pw_shell);
+
+	  u->storage = malloc(needed);
+	  if (u->storage != NULL)
+	    {
+	      char *p = u->storage;
+
+	      /* Copy each string field into storage allocated in the u
+		 structure and point the fields at that instead of the static
+		 storage that pw currently points to.  */
+#define COPY(field) \
+  (pw->field == NULL \
+   || (strcpy(p, pw->field), (pw->field = p), (p += strlen(p) + 1)))
+	      COPY(pw_name);
+	      COPY(pw_passwd);
+	      COPY(pw_gecos);
+	      COPY(pw_dir);
+	      COPY(pw_shell);
+
+	      u->passwd = *pw;
+	      u->passwd_state = PS_USER_PASSWD_OK;
+
+	      return &u->passwd;
+	    }
+	}
+    }
+
+  u->passwd_state = PS_USER_PASSWD_ERROR;
+  return NULL;
+}
+
+/* Returns the user name for the user referred to by U, or NULL if it can't
+   be gotten.  */
+char *ps_user_name(ps_user_t u)
+{
+  struct passwd *pw = ps_user_passwd(u);
+  if (pw)
+    return pw->pw_name;
+  else
+    return NULL;
+}