testcases/kernel/pty/hangup01.c

   1 /*
   2  *
   3  *   Copyright (c) International Business Machines  Corp., 2002
   4  *
   5  *   This program is free software;  you can redistribute it and/or modify
   6  *   it under the terms of the GNU General Public License as published by
   7  *   the Free Software Foundation; either version 2 of the License, or
   8  *   (at your option) any later version.
   9  *
  10  *   This program is distributed in the hope that it will be useful,
  11  *   but WITHOUT ANY WARRANTY;  without even the implied warranty of
  12  *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See
  13  *   the GNU General Public License for more details.
  14  *
  15  *   You should have received a copy of the GNU General Public License
  16  *   along with this program;  if not, write to the Free Software
  17  *   Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
  18  */
  19
  20 /* 12/24/2002   Port to LTP     robbiew@us.ibm.com */
  21 /* 06/30/2001   Port to Linux   nsharoff@us.ibm.com */
  22
  23 #ifndef _GNU_SOURCE
  24 #define _GNU_SOURCE 1
  25 #endif
  26
  27 #include <stdio.h>
  28 #include <stdlib.h>
  29 #include <unistd.h>
  30 #include <errno.h>
  31 #include <signal.h>
  32 #include <sys/fcntl.h>
  33 #include <sys/wait.h>
  34 #include <sys/poll.h>
  35
  36 /** LTP Port **/
  37 #include "test.h"
  38 #include "usctest.h"
  39
  40
  41 char *TCID="hangup01";            /* Test program identifier.    */
  42 int TST_TOTAL=5;                /* Total number of test cases. */
  43 extern int Tst_count;           /* Test Case counter for tst_* routines */
  44 /**************/
  45
  46 /*
  47  * pty master clone device
  48  */
  49 #define MASTERCLONE "/dev/ptmx"
  50
  51 #define MESSAGE1 "I love Linux!"
  52 #define MESSAGE2 "Use the LTP for all your Linux testing needs."
  53 #define MESSAGE3 "For the latest version of the LTP tests, visit http://ltp.sourceforge.net"
  54
  55 #define NUMMESSAGES 3
  56
  57 #define BUFSZ 4096
  58
  59 /*
  60  * parent process for hangup test
  61  */
  62 int
  63 parent(int masterfd, int childpid)
  64 {
  65         char buf[BUFSZ];
  66         struct pollfd pollfds[1];
  67         int hangupcount = 0;
  68         int datacount = 0;
  69         int status;
  70         int i;
  71         int len = strlen(MESSAGE1);
  72
  73         pollfds[0].fd = masterfd;
  74         pollfds[0].events = POLLIN;
  75
  76         sleep(1);
  77
  78         while ((i = poll(pollfds, 1, -1)) == 1) {
  79                 if (read(masterfd, buf, len) == -1) {
  80                         ++hangupcount;
  81 #ifdef DEBUG
  82                         tst_resm(TINFO,"hangup %d", hangupcount);
  83 #endif
  84                         if (hangupcount == NUMMESSAGES) {
  85                                 break;
  86                         }
  87                 } else {
  88                         ++datacount;
  89                         switch (datacount) {
  90                         case 1:
  91                                 if (strncmp(buf, MESSAGE1,
  92                                     strlen(MESSAGE1)) != 0) {
  93                                         tst_resm(TFAIL, "unexpected message 1");
  94                                         tst_exit();
  95                                 }
  96                                 len = strlen(MESSAGE2);
  97                                 break;
  98                         case 2:
  99                                 if (strncmp(buf, MESSAGE2,
 100                                     strlen(MESSAGE2)) != 0) {
 101                                         tst_resm(TFAIL, "unexpected message 2");
 102                                         tst_exit();
 103                                 }
 104                                 len = strlen(MESSAGE3);
 105                                 break;
 106                         case 3:
 107                                 if (strncmp(buf, MESSAGE3,
 108                                     strlen(MESSAGE3)) != 0) {
 109                                         tst_resm(TFAIL, "unexpected message 3");
 110                                         tst_exit();
 111                                 }
 112                                 break;
 113                         default:
 114                                 tst_resm(TFAIL, "unexpected data message");
 115                                 tst_exit();
 116                                 /*NOTREACHED*/
 117                         }
 118                 }
 119         }
 120         if (i != 1) {
 121                 tst_resm(TFAIL,"poll");
 122                 tst_exit();
 123         }
 124         while (wait(&status) != childpid) {
 125                 ;
 126         }
 127         if (status != 0) {
 128                 tst_resm(TFAIL, "child process exited with status %d", status);
 129                 tst_exit();
 130         }
 131         tst_resm(TPASS,"Pass");
 132         tst_exit();
 133
 134         /*NOTREACHED*/
 135         return 0;
 136 }
 137
 138
 139 /*
 140  * Child process for hangup test.  Write three messages to the slave
 141  * pty, with a hangup after each.
 142  */
 143 void
 144 child(int masterfd)
 145 {
 146         int slavefd;
 147         char *slavename;
 148
 149
 150         if ((slavename = ptsname(masterfd)) == (char *)0) {
 151                 tst_resm(TBROK,"ptsname");
 152                 tst_exit();
 153         }
 154         if ((slavefd = open(slavename, O_RDWR)) < 0) {
 155                 tst_resm(TBROK,slavename);
 156                 tst_exit();
 157         }
 158         if (write(slavefd, MESSAGE1, strlen(MESSAGE1)) != strlen(MESSAGE1)) {
 159                 tst_resm(TBROK,"write");
 160                 tst_exit();
 161         }
 162         if (close(slavefd) != 0) {
 163                 tst_resm(TBROK,"close");
 164                 tst_exit();
 165         }
 166         if ((slavefd = open(slavename, O_RDWR)) < 0) {
 167                 tst_resm(TBROK,"open %s",slavename);
 168                 tst_exit();
 169         }
 170         if (write(slavefd, MESSAGE2, strlen(MESSAGE2)) != strlen(MESSAGE2)) {
 171                 tst_resm(TBROK,"write");
 172                 tst_exit();
 173         }
 174         if (close(slavefd) != 0) {
 175                 tst_resm(TBROK,"close");
 176                 tst_exit();
 177         }
 178         if ((slavefd = open(slavename, O_RDWR)) < 0) {
 179                 tst_resm(TBROK,"open %s",slavename);
 180                 tst_exit();
 181         }
 182         if (write(slavefd, MESSAGE3, strlen(MESSAGE3)) != strlen(MESSAGE3)) {
 183                 tst_resm(TBROK,"write");
 184                 tst_exit();
 185         }
 186         if (close(slavefd) != 0) {
 187                 tst_resm(TBROK,"close");
 188                 tst_exit();
 189         }
 190 }
 191
 192 /*
 193  * main test driver
 194  */
 195 int main(int argc, char **argv)
 196 {
 197         int masterfd;           /* master pty fd */
 198         char *slavename;
 199         int childpid;
 200
 201 /*--------------------------------------------------------------------*/
 202         masterfd = open(MASTERCLONE, O_RDWR);
 203         if (masterfd < 0) {
 204                 tst_resm(TBROK,"open %s",MASTERCLONE);
 205                 tst_exit();
 206         }
 207
 208         slavename = ptsname(masterfd);
 209         if (slavename == (char *)0) {
 210                 tst_resm(TBROK,"ptsname");
 211                 tst_exit();
 212         }
 213
 214         if (grantpt(masterfd) != 0) {
 215                 tst_resm(TBROK,"grantpt");
 216                 tst_exit();
 217         }
 218
 219         if (unlockpt(masterfd) != 0) {
 220                 tst_resm(TBROK,"unlockpt");
 221                 tst_exit();
 222         }
 223
 224         childpid = fork();
 225         if (childpid == -1) {
 226                 tst_resm(TBROK,"fork");
 227                 tst_exit();
 228         } else if (childpid == 0) {
 229                 child(masterfd);
 230                 tst_exit();
 231         } else {
 232                 parent(masterfd, childpid);
 233         }
 234 /*--------------------------------------------------------------------*/
 235         tst_exit();
 236         /*NOTREACHED*/
 237         return 0;
 238 }