am 6cbc56b8: Merge "Implement <pty.h>."

* commit '6cbc56b84be72b3f01c23b865e39f61a01ead895':
  Implement <pty.h>.
This commit is contained in:
Elliott Hughes 2014-12-04 15:16:33 +00:00 committed by Android Git Automerger
commit c02d6add2d
6 changed files with 225 additions and 3 deletions

View File

@ -28,13 +28,15 @@
#include <errno.h>
#include <fcntl.h>
#include <pty.h>
#include <stdio.h>
#include <stdlib.h>
#include <sys/ioctl.h>
#include <termios.h>
#include <unistd.h>
#include <utmp.h>
int getpt(void) {
int getpt() {
return posix_openpt(O_RDWR|O_NOCTTY);
}
@ -47,7 +49,7 @@ int posix_openpt(int flags) {
}
char* ptsname(int fd) {
static char buf[64];
static char buf[32];
return ptsname_r(fd, buf, sizeof(buf)) == 0 ? buf : NULL;
}
@ -105,3 +107,83 @@ int unlockpt(int fd) {
int unlock = 0;
return ioctl(fd, TIOCSPTLCK, &unlock);
}
int openpty(int* master, int* slave, char* name, const termios* t, const winsize* ws) {
*master = getpt();
if (*master == -1) {
return -1;
}
if (grantpt(*master) == -1 || unlockpt(*master) == -1) {
close(*master);
return -1;
}
char buf[32];
if (name == NULL) {
name = buf;
}
if (ptsname_r(*master, name, sizeof(buf)) != 0) {
close(*master);
return -1;
}
*slave = open(name, O_RDWR|O_NOCTTY);
if (*slave == -1) {
close(*master);
return -1;
}
if (t != NULL) {
tcsetattr(*slave, TCSAFLUSH, t);
}
if (ws != NULL) {
ioctl(*slave, TIOCSWINSZ, ws);
}
return 0;
}
int forkpty(int* master, char* name, const termios* t, const winsize* ws) {
int slave;
if (openpty(master, &slave, name, t, ws) == -1) {
return -1;
}
pid_t pid = fork();
if (pid == -1) {
close(*master);
close(slave);
return -1;
}
if (pid == 0) {
// Child.
close(*master);
if (login_tty(slave) == -1) {
_exit(1);
}
return 0;
}
// Parent.
close(slave);
return pid;
}
int login_tty(int fd) {
setsid();
if (ioctl(fd, TIOCSCTTY, NULL) == -1) {
return -1;
}
dup2(fd, STDIN_FILENO);
dup2(fd, STDOUT_FILENO);
dup2(fd, STDERR_FILENO);
if (fd > STDERR_FILENO) {
close(fd);
}
return 0;
}

44
libc/include/pty.h Normal file
View File

@ -0,0 +1,44 @@
/*
* Copyright (C) 2014 The Android Open Source Project
* All rights reserved.
*
* Redistribution and use in source and binary forms, with or without
* modification, are permitted provided that the following conditions
* are met:
* * Redistributions of source code must retain the above copyright
* notice, this list of conditions and the following disclaimer.
* * Redistributions in binary form must reproduce the above copyright
* notice, this list of conditions and the following disclaimer in
* the documentation and/or other materials provided with the
* distribution.
*
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
* "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
* LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
* FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE
* COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT,
* INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING,
* BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS
* OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED
* AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
* OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT
* OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
* SUCH DAMAGE.
*/
#ifndef _PTY_H
#define _PTY_H
#include <sys/cdefs.h>
#include <termios.h>
#include <sys/ioctl.h>
__BEGIN_DECLS
int openpty(int*, int*, char*, const struct termios*, const struct winsize*);
int forkpty(int*, char*, const struct termios*, const struct winsize*);
__END_DECLS
#endif /* _PTY_H */

View File

@ -91,6 +91,8 @@ int utmpname(const char*);
void setutent();
struct utmp* getutent();
int login_tty(int);
__END_DECLS
#endif /* _UTMP_H_ */

View File

@ -72,6 +72,7 @@ libBionicStandardTests_src_files := \
mntent_test.cpp \
netdb_test.cpp \
pthread_test.cpp \
pty_test.cpp \
regex_test.cpp \
sched_test.cpp \
search_test.cpp \
@ -108,6 +109,7 @@ libBionicStandardTests_src_files := \
uchar_test.cpp \
uniqueptr_test.cpp \
unistd_test.cpp \
utmp_test.cpp \
wchar_test.cpp \
libBionicStandardTests_cflags := \
@ -309,7 +311,7 @@ bionic-unit-tests-glibc_whole_static_libraries := \
libBionicStandardTests \
bionic-unit-tests-glibc_ldlibs := \
-lrt -ldl \
-lrt -ldl -lutil \
bionic-unit-tests-glibc_c_includes := \
bionic/libc \

67
tests/pty_test.cpp Normal file
View File

@ -0,0 +1,67 @@
/*
* Copyright (C) 2014 The Android Open Source Project
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
#include <gtest/gtest.h>
#include <pty.h>
#include <sys/ioctl.h>
TEST(pty, openpty) {
int master, slave;
char name[32];
struct winsize w = { 123, 456, 9999, 999 };
ASSERT_EQ(0, openpty(&master, &slave, name, NULL, &w));
ASSERT_NE(-1, master);
ASSERT_NE(-1, slave);
ASSERT_NE(master, slave);
char tty_name[32];
ASSERT_EQ(0, ttyname_r(slave, tty_name, sizeof(tty_name)));
ASSERT_STREQ(tty_name, name);
struct winsize w_actual;
ASSERT_EQ(0, ioctl(slave, TIOCGWINSZ, &w_actual));
ASSERT_EQ(w_actual.ws_row, w.ws_row);
ASSERT_EQ(w_actual.ws_col, w.ws_col);
ASSERT_EQ(w_actual.ws_xpixel, w.ws_xpixel);
ASSERT_EQ(w_actual.ws_ypixel, w.ws_ypixel);
close(master);
close(slave);
}
TEST(pty, forkpty) {
pid_t sid = getsid(0);
int master;
pid_t pid = forkpty(&master, NULL, NULL, NULL);
ASSERT_NE(-1, pid);
if (pid == 0) {
// We're the child.
ASSERT_NE(sid, getsid(0));
_exit(0);
}
ASSERT_EQ(sid, getsid(0));
int status;
ASSERT_EQ(pid, waitpid(pid, &status, 0));
ASSERT_TRUE(WIFEXITED(status));
ASSERT_EQ(0, WEXITSTATUS(status));
close(master);
}

25
tests/utmp_test.cpp Normal file
View File

@ -0,0 +1,25 @@
/*
* Copyright (C) 2014 The Android Open Source Project
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
#include <gtest/gtest.h>
#include <utmp.h>
TEST(utmp, login_tty) {
// login_tty is tested indirectly by the openpty and forkpty tests.
// This test just checks that we're exporting the symbol independently.
ASSERT_EQ(-1, login_tty(-1));
}