| /* |
| * |
| * Copyright (c) International Business Machines Corp., 2002 |
| * |
| * 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 of the License, 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., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA |
| */ |
| |
| /* 12/23/2002 Port to LTP robbiew@us.ibm.com */ |
| /* 06/30/2001 Port to Linux nsharoff@us.ibm.com */ |
| |
| #ifndef _GNU_SOURCE |
| #define _GNU_SOURCE 1 |
| #endif |
| |
| #include <sys/types.h> |
| #include <sys/stat.h> |
| #include <sys/wait.h> |
| #include <errno.h> |
| #include <fcntl.h> |
| #include <stdio.h> |
| #include <stdlib.h> |
| #include <string.h> |
| #include <termio.h> |
| #include <unistd.h> |
| |
| /** LTP Port **/ |
| #include "test.h" |
| |
| char *TCID = "pty01"; /* Test program identifier. */ |
| int TST_TOTAL = 5; /* Total number of test cases. */ |
| /**************/ |
| |
| /* |
| * pty master clone device |
| */ |
| #define MASTERCLONE "/dev/ptmx" |
| |
| /* |
| * string for testing read/write on ptys |
| */ |
| #define STRING "Linux Test Project\n" |
| |
| /* |
| * test buffer size |
| */ |
| #define TESTSIZE 1024 |
| |
| /* |
| * mode we expect grantpt() to leave pty as |
| */ |
| #define PTY_MODE 020622 |
| |
| /* |
| * number of procs for parallel test |
| */ |
| #define NUMPROCS 15 |
| |
| /* |
| * test slave locking |
| */ |
| static int test1(void) |
| { |
| int masterfd; /* master pty fd */ |
| int slavefd; /* slave pty fd */ |
| char *slavename; |
| struct stat st; |
| char buf[TESTSIZE]; |
| |
| masterfd = open(MASTERCLONE, O_RDWR); |
| if (masterfd < 0) { |
| tst_brkm(TBROK | TERRNO, NULL, MASTERCLONE); |
| } |
| |
| slavename = ptsname(masterfd); |
| if (slavename == NULL) { |
| tst_brkm(TBROK | TERRNO, NULL, "ptsname() call failed"); |
| } |
| |
| if (grantpt(masterfd) != 0) { |
| tst_brkm(TBROK | TERRNO, NULL, "grantpt() call failed"); |
| } |
| |
| if (stat(slavename, &st) != 0) { |
| tst_brkm(TBROK | TERRNO, NULL, "stat(%s) failed", slavename); |
| } |
| if (st.st_uid != getuid()) { |
| tst_brkm(TBROK, NULL, "uid mismatch"); |
| } |
| |
| if (st.st_mode != (S_IFCHR | S_IRUSR | S_IWUSR | S_IWGRP)) { |
| tst_brkm(TBROK, NULL, "mode mismatch (mode=%o)", st.st_mode); |
| } |
| |
| slavefd = open(slavename, O_RDWR); |
| if (slavefd >= 0) { |
| tst_brkm(TBROK, NULL, "open didn't fail as expected!"); |
| } |
| |
| if (unlockpt(masterfd) != 0) { |
| tst_brkm(TBROK | TERRNO, NULL, "unlockpt() failed"); |
| } |
| |
| slavefd = open(slavename, O_RDWR); |
| if (slavefd < 0) { |
| tst_brkm(TBROK, NULL, "Could not open %s", slavename); |
| } |
| |
| /* |
| * test writing to the master / reading from the slave |
| */ |
| if (write(masterfd, STRING, strlen(STRING)) != strlen(STRING)) { |
| /* |
| * XXX: the errno printout might be garbage, but better to be |
| * safe than sorry.. |
| */ |
| tst_brkm(TFAIL | TERRNO, NULL, "write to master"); |
| } |
| |
| if (read(slavefd, buf, strlen(STRING)) != strlen(STRING)) { |
| /* XXX: Same as write above.. */ |
| tst_brkm(TFAIL | TERRNO, NULL, "read from slave"); |
| } |
| if (strncmp(STRING, buf, strlen(STRING) - 1) != 0) { |
| tst_brkm(TFAIL, NULL, |
| "strings are different (STRING = '%s' != buf = '%s')", |
| STRING, buf); |
| } |
| |
| /* |
| * test writing to the slave / reading from the master |
| */ |
| if (write(slavefd, STRING, strlen(STRING)) != strlen(STRING)) { |
| /* XXX: Same as write above.. */ |
| tst_brkm(TFAIL | TERRNO, NULL, "write to slave"); |
| } |
| |
| if (read(masterfd, buf, strlen(STRING)) != strlen(STRING)) { |
| /* XXX: Same as write above.. */ |
| tst_brkm(TFAIL | TERRNO, NULL, "read from master"); |
| } |
| if (strncmp(STRING, buf, strlen(STRING) - 1) != 0) { |
| tst_brkm(TFAIL, NULL, |
| "strings are different (STRING = '%s' != buf = '%s').", |
| STRING, buf); |
| } |
| |
| /* |
| * try an invalid ioctl on the slave... |
| */ |
| if (ioctl(slavefd, TIOCGWINSZ, NULL) == 0) { |
| tst_brkm(TFAIL, NULL, |
| "invalid slave TIOCGWINSZ ioctl succeeded.. it should " |
| "have failed"); |
| } |
| |
| /* |
| * try an invalid ioctl on the master... |
| */ |
| if (ioctl(masterfd, TIOCGWINSZ, NULL) == 0) { |
| tst_brkm(TFAIL, NULL, |
| "invalid master TIOCGWINSZ ioctl succeeded.. it should " |
| "have failed"); |
| } |
| |
| /* |
| * close pty fds |
| */ |
| if (close(slavefd) != 0) { |
| tst_brkm(TBROK | TERRNO, NULL, "close of slave"); |
| } |
| if (close(masterfd) != 0) { |
| tst_brkm(TBROK | TERRNO, NULL, "close of master"); |
| } |
| tst_resm(TPASS, "test1"); |
| /** NOTREACHED **/ |
| return 0; |
| } |
| |
| /* |
| * test slave operations with closed master |
| */ |
| static void test2(void) |
| { |
| int masterfd; /* master pty fd */ |
| int slavefd; /* slave pty fd */ |
| int i; |
| char *slavename; |
| char c; |
| |
| masterfd = open(MASTERCLONE, O_RDWR); |
| if (masterfd < 0) { |
| tst_brkm(TBROK | TERRNO, NULL, MASTERCLONE); |
| } |
| |
| slavename = ptsname(masterfd); |
| if (slavename == NULL) { |
| tst_brkm(TBROK | TERRNO, NULL, "ptsname() call failed"); |
| } |
| |
| if (grantpt(masterfd) != 0) { |
| tst_brkm(TBROK | TERRNO, NULL, "grantpt() call failed"); |
| } |
| |
| if (unlockpt(masterfd) != 0) { |
| tst_brkm(TBROK | TERRNO, NULL, "unlockpt() call failed"); |
| } |
| |
| slavefd = open(slavename, O_RDWR); |
| if (slavefd < 0) { |
| tst_brkm(TBROK | TERRNO, NULL, "Could not open %s", slavename); |
| } |
| |
| /* |
| * close pty fds. See what happens when we close the master |
| * first. |
| */ |
| if (close(masterfd) != 0) { |
| tst_brkm(TBROK | TERRNO, NULL, "close()"); |
| } |
| |
| errno = 0; |
| if ((i = read(slavefd, &c, 1)) == 1) { |
| tst_brkm(TFAIL, NULL, |
| "reading from slave fd should have failed, but didn't" |
| "(read '%c')", c); |
| } |
| |
| if ((i = write(slavefd, &c, 1)) == 1) { |
| tst_brkm(TFAIL, NULL, |
| "writing to slave fd should have failed, but didn't"); |
| } |
| |
| if (ioctl(slavefd, TIOCGWINSZ, NULL) == 0) { |
| tst_brkm(TFAIL, NULL, |
| "trying TIOCGWINSZ on slave fd should have failed, " |
| "but didn't"); |
| } |
| |
| if (close(slavefd) != 0) { |
| tst_brkm(TBROK, NULL, "close"); |
| } |
| tst_resm(TPASS, "test2"); |
| } |
| |
| /* |
| * test operations on master with closed slave |
| */ |
| static void test3(void) |
| { |
| int masterfd; /* master pty fd */ |
| |
| masterfd = open(MASTERCLONE, O_RDWR); |
| if (masterfd < 0) { |
| tst_brkm(TBROK, NULL, MASTERCLONE); |
| } |
| |
| if (ioctl(masterfd, TIOCGWINSZ, NULL) == 0) { |
| tst_brkm(TFAIL | TERRNO, NULL, |
| "trying TIOCGWINSZ on master with no open slave " |
| "succeeded unexpectedly"); |
| } |
| tst_resm(TPASS, "test3"); |
| } |
| |
| /* |
| * test multiple opens on slave side of pty |
| */ |
| static void test4(void) |
| { |
| int masterfd; /* master pty fd */ |
| int slavefd; /* slave pty fd */ |
| int slavefd2; |
| int slavefd3; |
| char *slavename; |
| |
| masterfd = open(MASTERCLONE, O_RDWR); |
| if (masterfd < 0) { |
| tst_brkm(TBROK, NULL, "%s", MASTERCLONE); |
| } |
| |
| slavename = ptsname(masterfd); |
| if (slavename == NULL) { |
| tst_brkm(TBROK, NULL, "ptsname() call failed"); |
| } |
| |
| if (grantpt(masterfd) != 0) { |
| tst_brkm(TBROK, NULL, "grantpt() call failed"); |
| } |
| |
| if (unlockpt(masterfd) != 0) { |
| tst_brkm(TBROK | TERRNO, NULL, "unlockpt() call failed"); |
| } |
| |
| slavefd = open(slavename, O_RDWR); |
| if (slavefd < 0) { |
| tst_brkm(TBROK | TERRNO, NULL, "Could not open %s", slavename); |
| } |
| |
| slavefd2 = open(slavename, O_RDWR); |
| if (slavefd < 0) { |
| tst_brkm(TFAIL | TERRNO, NULL, "Could not open %s (again)", |
| slavename); |
| } |
| |
| slavefd3 = open(slavename, O_RDWR); |
| if (slavefd < 0) { |
| tst_brkm(TFAIL | TERRNO, NULL, "Could not open %s (once more)", |
| slavename); |
| } |
| |
| /* |
| * close pty fds. |
| */ |
| if (close(slavefd) != 0) { |
| tst_brkm(TBROK | TERRNO, NULL, "close slave"); |
| } |
| |
| if (close(slavefd2) != 0) { |
| tst_brkm(TBROK, NULL, "close slave again"); |
| } |
| |
| if (close(slavefd3) != 0) { |
| tst_brkm(TBROK, NULL, "close slave once more"); |
| } |
| |
| if (close(masterfd) != 0) { |
| tst_brkm(TBROK, NULL, "close master"); |
| } |
| tst_resm(TPASS, "test4"); |
| } |
| |
| /* |
| * test opening/closing lots of ptys in parallel. We may run out |
| * of ptys for this test depending on how the system is configured, |
| * but that's not a fatal error. |
| */ |
| static void test5(void) |
| { |
| int masterfd; /* master pty fd */ |
| char *slavename; |
| int status; |
| int i; |
| |
| for (i = 0; i < NUMPROCS; ++i) { |
| switch (fork()) { |
| case -1: |
| tst_brkm(TBROK, NULL, "fork()"); |
| break; |
| case 0: |
| masterfd = open(MASTERCLONE, O_RDWR); |
| if (masterfd < 0) { |
| printf("proc %d: opening %s failed: %s", |
| i, MASTERCLONE, strerror(errno)); |
| exit(1); |
| } |
| if (grantpt(masterfd) != 0) { |
| printf("proc %d: grantpt() call failed: %s", |
| i, strerror(errno)); |
| exit(1); |
| } |
| slavename = ptsname(masterfd); |
| if (slavename == NULL) { |
| printf("proc %d: ptsname() call failed: %s", |
| i, strerror(errno)); |
| exit(1); |
| } |
| sleep(10); |
| if (close(masterfd) != 0) { |
| printf("proc %d: close failed: %s", |
| i, strerror(errno)); |
| exit(1); |
| } |
| exit(0); |
| default: |
| break; |
| } |
| } |
| while (wait(&status) > 0) { |
| if (status) { |
| tst_brkm(TFAIL, NULL, |
| "child exited with non-zero status %d", |
| status); |
| } |
| } |
| tst_resm(TPASS, "test5"); |
| } |
| |
| /* |
| * main test driver |
| */ |
| int main(int argc, char **argv) |
| { |
| test1(); |
| test2(); |
| test3(); |
| test4(); |
| test5(); |
| |
| /* |
| * all done |
| */ |
| tst_exit(); |
| } |