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