src/ptty.c

   1 /*
   2  * ptty.c:
   3  *
   4  * Copyright (c) 2008 James McKenzie <james@fishsoup.dhs.org>,
   5  * All rights reserved.
   6  *
   7  */
   8
   9 static char rcsid[] = "$Id$";
  10
  11 /*
  12  * $Log$
  13  * Revision 1.2  2008/02/12 22:36:46  james
  14  * *** empty log message ***
  15  *
  16  * Revision 1.1  2008/02/09 15:47:28  james
  17  * *** empty log message ***
  18  *
  19  * Revision 1.2  2008/02/07 11:11:14  staffcvs
  20  * *** empty log message ***
  21  *
  22  * Revision 1.1  2008/02/07 01:02:52  james
  23  * *** empty log message ***
  24  *
  25  * Revision 1.3  2008/02/06 17:53:28  james
  26  * *** empty log message ***
  27  *
  28  * Revision 1.2  2008/02/04 02:05:06  james
  29  * *** empty log message ***
  30  *
  31  * Revision 1.1  2008/02/04 01:32:39  james
  32  * *** empty log message ***
  33  *
  34  */
  35
  36 #include "project.h"
  37
  38
  39 typedef struct
  40 {
  41   TTY_SIGNATURE;
  42   int fd;
  43   pid_t child;
  44 } PTTY;
  45
  46
  47 static void
  48 ptty_close (TTY * _t)
  49 {
  50   PTTY *t = (PTTY *) _t;
  51
  52   if (!t)
  53     return;
  54
  55   close (t->fd);
  56   free (t);
  57 }
  58
  59
  60
  61 static int
  62 ptty_read (TTY * _t, void *buf, int len)
  63 {
  64   PTTY *t = (PTTY *) _t;
  65   int red, done = 0;
  66
  67   do
  68     {
  69
  70       red = wrap_read (t->fd, buf, len);
  71       if (red < 0)
  72         return -1;
  73       if (!red)
  74         return done;
  75
  76       buf += red;
  77       len -= red;
  78       done += red;
  79     }
  80   while (len);
  81
  82
  83   return done;
  84 }
  85
  86
  87 static int
  88 ptty_write (TTY * _t, void *buf, int len)
  89 {
  90   int writ, done = 0;
  91   PTTY *t = (PTTY *) _t;
  92
  93   do
  94     {
  95
  96       writ = wrap_write (t->fd, buf, len);
  97       if (writ < 0)
  98         return -1;
  99       if (!writ)
 100         sleep (1);
 101
 102       buf += writ;
 103       len -= writ;
 104       done += writ;
 105     }
 106   while (len);
 107
 108
 109   return done;
 110 }
 111
 112 TTY *
 113 ptty_open (char *path, char *argv[])
 114 {
 115   PTTY *t;
 116   pid_t child;
 117   char name[1024];
 118   struct winsize winsize = { 0 };
 119   struct termios termios;
 120   int fd;
 121   char *default_argv = { "-", (char *) 0 };
 122
 123   child = forkpty (&fd, name, &termios, &winsize);
 124
 125   switch (child)
 126     {
 127     case -1:                   /*boo hiss */
 128       return -1;
 129     case 0:                    /*waaah */
 130       setenv ("TERM", "vt102", 1);
 131       setenv ("LANG", "C", 1);
 132       if (!path)
 133         path = "/bin/sh";
 134
 135       if (!argv)
 136         argv = default_argv;
 137
 138       execv (path, argv);
 139       _exit (-1);
 140     }
 141
 142   set_nonblocking (fd);
 143
 144   t = (PTTY *) malloc (sizeof (PTTY));
 145
 146   strncpy (t->name, name, sizeof (t->name));
 147   t->name[sizeof (t->name) - 1] = 0;
 148
 149   t->read = ptty_read;
 150   t->write = ptty_write;
 151   t->close = ptty_close;
 152
 153   default_termios (&termios);
 154
 155   winsize.ws_row = VT102_ROWS;
 156   winsize.ws_col = VT102_COLS;
 157
 158   t->fd = open_fd_to_pty (path, argv);
 159   t->pid = child;
 160   t->rfd = t->fd;
 161   t->wfd = 0;
 162
 163   return (TTY *) t;
 164 }