usr.bin/talk/io.c

   1 /*      $NetBSD: io.c,v 1.12 2006/10/07 17:27:57 elad Exp $     */
   2
   3 /*
   4  * Copyright (c) 1983, 1993
   5  *      The Regents of the University of California.  All rights reserved.
   6  *
   7  * Redistribution and use in source and binary forms, with or without
   8  * modification, are permitted provided that the following conditions
   9  * are met:
  10  * 1. Redistributions of source code must retain the above copyright
  11  *    notice, this list of conditions and the following disclaimer.
  12  * 2. Redistributions in binary form must reproduce the above copyright
  13  *    notice, this list of conditions and the following disclaimer in the
  14  *    documentation and/or other materials provided with the distribution.
  15  * 3. Neither the name of the University nor the names of its contributors
  16  *    may be used to endorse or promote products derived from this software
  17  *    without specific prior written permission.
  18  *
  19  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
  20  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  21  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  22  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
  23  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  24  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  25  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  26  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  27  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  28  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  29  * SUCH DAMAGE.
  30  */
  31
  32 #include <sys/cdefs.h>
  33 #ifndef lint
  34 #if 0
  35 static char sccsid[] = "@(#)io.c        8.1 (Berkeley) 6/6/93";
  36 #endif
  37 __RCSID("$NetBSD: io.c,v 1.12 2006/10/07 17:27:57 elad Exp $");
  38 #endif /* not lint */
  39
  40 /*
  41  * This file contains the I/O handling and the exchange of
  42  * edit characters. This connection itself is established in
  43  * ctl.c
  44  */
  45
  46 #include "talk.h"
  47 #include <sys/ioctl.h>
  48 #include <sys/time.h>
  49 #include <stdio.h>
  50 #include <errno.h>
  51 #include <string.h>
  52 #include <unistd.h>
  53 #include <poll.h>
  54
  55 #define A_LONG_TIME 1000000
  56
  57 /*
  58  * The routine to do the actual talking
  59  */
  60 void
  61 talk()
  62 {
  63         struct pollfd set[2];
  64         int nb;
  65         char buf[BUFSIZ];
  66
  67         message("Connection established\007\007\007");
  68         current_line = 0;
  69
  70         /*
  71          * Wait on both the other process (sockt_mask) and
  72          * standard input ( STDIN_MASK )
  73          */
  74         set[0].fd = sockt;
  75         set[0].events = POLLIN;
  76         set[1].fd = fileno(stdin);
  77         set[1].events = POLLIN;
  78         for (;;) {
  79                 nb = poll(set, 2, A_LONG_TIME * 1000);
  80                 if (nb <= 0) {
  81                         if (errno == EINTR)
  82                                 continue;
  83                         /* panic, we don't know what happened */
  84                         p_error("Unexpected error from poll");
  85                         quit();
  86                 }
  87                 if (set[0].revents & POLLIN) {
  88                         /* There is data on sockt */
  89                         nb = read(sockt, buf, sizeof buf);
  90                         if (nb <= 0) {
  91                                 message("Connection closed. Exiting");
  92                                 quit();
  93                         }
  94                         display(&his_win, buf, nb);
  95                 }
  96                 if (set[1].revents & POLLIN) {
  97                         /*
  98                          * We can't make the tty non_blocking, because
  99                          * curses's output routines would screw up
 100                          */
 101                         ioctl(0, FIONREAD, (void *) &nb);
 102                         nb = read(0, buf, nb);
 103                         display(&my_win, buf, nb);
 104                         /* might lose data here because sockt is non-blocking */
 105                         if (nb > 0)
 106                                 write(sockt, buf, nb);
 107                 }
 108         }
 109 }
 110
 111 /*
 112  * p_error prints the system error message on the standard location
 113  * on the screen and then exits. (i.e. a curses version of perror)
 114  */
 115 void
 116 p_error(string)
 117         const char *string;
 118 {
 119         wmove(my_win.x_win, current_line%my_win.x_nlines, 0);
 120         wprintw(my_win.x_win, "[%s : %s (%d)]\n",
 121             string, strerror(errno), errno);
 122         wrefresh(my_win.x_win);
 123         move(LINES-1, 0);
 124         refresh();
 125         quit();
 126 }
 127
 128 /*
 129  * Display string in the standard location
 130  */
 131 void
 132 message(string)
 133         const char *string;
 134 {
 135         wmove(my_win.x_win, current_line % my_win.x_nlines, 0);
 136         wprintw(my_win.x_win, "[%s]", string);
 137         wclrtoeol(my_win.x_win);
 138         current_line++;
 139         wmove(my_win.x_win, current_line % my_win.x_nlines, 0);
 140         wrefresh(my_win.x_win);
 141 }