00001
00002
00003
00004
00005
00006
00007
00008
00009
00010
00011
00012
00013
00014
00015
00016
00017
00018
00019
00020
00021
00022
00023
00024 #include <config.h>
00025
00026 #include "kpty.h"
00027 #include "kprocess.h"
00028
00029 #ifdef __sgi
00030 #define __svr4__
00031 #endif
00032
00033 #ifdef __osf__
00034 #define _OSF_SOURCE
00035 #include <float.h>
00036 #endif
00037
00038 #ifdef _AIX
00039 #define _ALL_SOURCE
00040 #endif
00041
00042
00043
00044 #ifdef __INTEL_COMPILER
00045 # ifndef __USE_XOPEN
00046 # define __USE_XOPEN
00047 # endif
00048 #endif
00049
00050 #include <sys/types.h>
00051 #include <sys/ioctl.h>
00052 #include <sys/time.h>
00053 #include <sys/resource.h>
00054 #include <sys/stat.h>
00055 #include <sys/param.h>
00056
00057 #ifdef HAVE_SYS_STROPTS_H
00058 # include <sys/stropts.h>
00059 # define _NEW_TTY_CTRL
00060 #endif
00061
00062 #include <errno.h>
00063 #include <fcntl.h>
00064 #include <time.h>
00065 #include <stdlib.h>
00066 #include <stdio.h>
00067 #include <string.h>
00068 #include <unistd.h>
00069 #include <grp.h>
00070
00071 #ifdef HAVE_LIBUTIL_H
00072 # include <libutil.h>
00073 # define USE_LOGIN
00074 #elif defined(HAVE_UTIL_H)
00075 # include <util.h>
00076 # define USE_LOGIN
00077 #endif
00078
00079 #ifdef USE_LOGIN
00080 # include <utmp.h>
00081 #endif
00082
00083 #ifdef HAVE_TERMIOS_H
00084
00085
00086 extern "C" {
00087 # include <termios.h>
00088 }
00089 #endif
00090
00091 #if !defined(__osf__)
00092 # ifdef HAVE_TERMIO_H
00093
00094 # include <termio.h>
00095 # endif
00096 #endif
00097
00098 #if defined(HAVE_TCGETATTR)
00099 # define _tcgetattr(fd, ttmode) tcgetattr(fd, ttmode)
00100 #elif defined(TIOCGETA)
00101 # define _tcgetattr(fd, ttmode) ioctl(fd, TIOCGETA, (char *)ttmode)
00102 #elif defined(TCGETS)
00103 # define _tcgetattr(fd, ttmode) ioctl(fd, TCGETS, (char *)ttmode)
00104 #else
00105 # error
00106 #endif
00107
00108 #if defined(HAVE_TCSETATTR) && defined(TCSANOW)
00109 # define _tcsetattr(fd, ttmode) tcsetattr(fd, TCSANOW, ttmode)
00110 #elif defined(TIOCSETA)
00111 # define _tcsetattr(fd, ttmode) ioctl(fd, TIOCSETA, (char *)ttmode)
00112 #elif defined(TCSETS)
00113 # define _tcsetattr(fd, ttmode) ioctl(fd, TCSETS, (char *)ttmode)
00114 #else
00115 # error
00116 #endif
00117
00118 #if defined (_HPUX_SOURCE)
00119 # define _TERMIOS_INCLUDED
00120 # include <bsdtty.h>
00121 #endif
00122
00123 #if defined(HAVE_PTY_H)
00124 # include <pty.h>
00125 #endif
00126
00127 #include <kdebug.h>
00128 #include <kstandarddirs.h>
00129
00130
00131 #ifndef CTRL
00132 # define CTRL(x) ((x) & 037)
00133 #endif
00134
00135 #define TTY_GROUP "tty"
00136
00138
00140
00141 #ifdef HAVE_UTEMPTER
00142 class KProcess_Utmp : public KProcess
00143 {
00144 public:
00145 int commSetupDoneC()
00146 {
00147 dup2(cmdFd, 0);
00148 dup2(cmdFd, 1);
00149 dup2(cmdFd, 3);
00150 return 1;
00151 }
00152 int cmdFd;
00153 };
00154 #endif
00155
00156 #define BASE_CHOWN "kgrantpty"
00157
00158
00159
00161
00163
00164 struct KPtyPrivate {
00165 KPtyPrivate() :
00166 xonXoff(false),
00167 utf8(false),
00168 masterFd(-1), slaveFd(-1)
00169 {
00170 memset(&winSize, 0, sizeof(winSize));
00171 winSize.ws_row = 24;
00172 winSize.ws_col = 80;
00173 }
00174
00175 bool xonXoff : 1;
00176 bool utf8 : 1;
00177 int masterFd;
00178 int slaveFd;
00179 struct winsize winSize;
00180
00181 QCString ttyName;
00182 };
00183
00185
00187
00188 KPty::KPty()
00189 {
00190 d = new KPtyPrivate;
00191 }
00192
00193 KPty::~KPty()
00194 {
00195 close();
00196 delete d;
00197 }
00198
00199 bool KPty::open()
00200 {
00201 if (d->masterFd >= 0)
00202 return true;
00203
00204 QCString ptyName;
00205
00206
00207
00208
00209
00210
00211
00212
00213 #if defined(HAVE_PTSNAME) && defined(HAVE_GRANTPT)
00214 #ifdef _AIX
00215 d->masterFd = ::open("/dev/ptc",O_RDWR);
00216 #else
00217 d->masterFd = ::open("/dev/ptmx",O_RDWR);
00218 #endif
00219 if (d->masterFd >= 0)
00220 {
00221 char *ptsn = ptsname(d->masterFd);
00222 if (ptsn) {
00223 grantpt(d->masterFd);
00224 d->ttyName = ptsn;
00225 goto gotpty;
00226 } else {
00227 ::close(d->masterFd);
00228 d->masterFd = -1;
00229 }
00230 }
00231 #endif
00232
00233
00234 for (const char* s3 = "pqrstuvwxyzabcdefghijklmno"; *s3; s3++)
00235 {
00236 for (const char* s4 = "0123456789abcdefghijklmnopqrstuvwxyz"; *s4; s4++)
00237 {
00238 ptyName.sprintf("/dev/pty%c%c", *s3, *s4);
00239 d->ttyName.sprintf("/dev/tty%c%c", *s3, *s4);
00240
00241 d->masterFd = ::open(ptyName.data(), O_RDWR);
00242 if (d->masterFd >= 0)
00243 {
00244 #ifdef __sun
00245
00246
00247
00248
00249 int pgrp_rtn;
00250 if (ioctl(d->masterFd, TIOCGPGRP, &pgrp_rtn) == 0 || errno != EIO) {
00251 ::close(d->masterFd);
00252 d->masterFd = -1;
00253 continue;
00254 }
00255 #endif
00256 if (!access(d->ttyName.data(),R_OK|W_OK))
00257 {
00258 if (!geteuid())
00259 {
00260 struct group* p = getgrnam(TTY_GROUP);
00261 if (!p)
00262 p = getgrnam("wheel");
00263 gid_t gid = p ? p->gr_gid : getgid ();
00264
00265 chown(d->ttyName.data(), getuid(), gid);
00266 chmod(d->ttyName.data(), S_IRUSR|S_IWUSR|S_IWGRP);
00267 }
00268 goto gotpty;
00269 }
00270 ::close(d->masterFd);
00271 d->masterFd = -1;
00272 }
00273 }
00274 }
00275
00276 kdWarning(175) << "Can't open a pseudo teletype" << endl;
00277 return false;
00278
00279 gotpty:
00280 struct stat st;
00281 if (stat(d->ttyName.data(), &st))
00282 return false;
00283
00284
00285 if (((st.st_uid != getuid()) ||
00286 (st.st_mode & (S_IRGRP|S_IXGRP|S_IROTH|S_IWOTH|S_IXOTH))) &&
00287 !chownpty(true))
00288 {
00289 kdWarning(175)
00290 << "chownpty failed for device " << ptyName << "::" << d->ttyName
00291 << "\nThis means the communication can be eavesdropped." << endl;
00292 }
00293
00294 #ifdef BSD
00295 revoke(d->ttyName.data());
00296 #endif
00297
00298 #ifdef HAVE_UNLOCKPT
00299 unlockpt(d->masterFd);
00300 #endif
00301
00302 d->slaveFd = ::open(d->ttyName.data(), O_RDWR | O_NOCTTY);
00303 if (d->slaveFd < 0)
00304 {
00305 kdWarning(175) << "Can't open slave pseudo teletype" << endl;
00306 ::close(d->masterFd);
00307 d->masterFd = -1;
00308 return false;
00309 }
00310
00311 #if (defined(__svr4__) || defined(__sgi__))
00312
00313 ioctl(d->slaveFd, I_PUSH, "ptem");
00314 ioctl(d->slaveFd, I_PUSH, "ldterm");
00315 #endif
00316
00317
00318
00319
00320
00321 struct ::termios ttmode;
00322
00323 _tcgetattr(d->slaveFd, &ttmode);
00324
00325 if (!d->xonXoff)
00326 ttmode.c_iflag &= ~(IXOFF | IXON);
00327 else
00328 ttmode.c_iflag |= (IXOFF | IXON);
00329
00330 #ifdef IUTF8
00331 if (!d->utf8)
00332 ttmode.c_iflag &= ~IUTF8;
00333 else
00334 ttmode.c_iflag |= IUTF8;
00335 #endif
00336
00337 ttmode.c_cc[VINTR] = CTRL('C' - '@');
00338 ttmode.c_cc[VQUIT] = CTRL('\\' - '@');
00339 ttmode.c_cc[VERASE] = 0177;
00340
00341 _tcsetattr(d->slaveFd, &ttmode);
00342
00343
00344 ioctl(d->slaveFd, TIOCSWINSZ, (char *)&d->winSize);
00345
00346 fcntl(d->masterFd, F_SETFD, FD_CLOEXEC);
00347 fcntl(d->slaveFd, F_SETFD, FD_CLOEXEC);
00348
00349 return true;
00350 }
00351
00352 bool KPty::setPty(int pty_master)
00353 {
00354 kdWarning(175)
00355 << "setPty()" << endl;
00356
00357 if(d->masterFd >= 0) {
00358 kdWarning(175)
00359 << "d->masterFd >= 0" << endl;
00360 return false;
00361 }
00362 d->masterFd = pty_master;
00363 return _attachPty(pty_master);
00364 }
00365
00366 bool KPty::_attachPty(int pty_master)
00367 {
00368 QCString ptyName;
00369
00370 kdWarning(175)
00371 << "_attachPty() " << pty_master << endl;
00372 #if defined(HAVE_PTSNAME) && defined(HAVE_GRANTPT)
00373 char *ptsn = ptsname(d->masterFd);
00374 if (ptsn) {
00375 grantpt(d->masterFd);
00376 d->ttyName = ptsn;
00377 } else {
00378 ::close(d->masterFd);
00379 d->masterFd = -1;
00380 }
00381 #endif
00382
00383 struct stat st;
00384 if (stat(d->ttyName.data(), &st))
00385 return false;
00386
00387
00388 if (((st.st_uid != getuid()) ||
00389 (st.st_mode & (S_IRGRP|S_IXGRP|S_IROTH|S_IWOTH|S_IXOTH))) &&
00390 !chownpty(true))
00391 {
00392 kdWarning(175)
00393 << "chownpty failed for device " << ptyName << "::" << d->ttyName
00394 << "\nThis means the communication can be eavesdropped." << endl;
00395 }
00396
00397 #ifdef BSD
00398 revoke(d->ttyName.data());
00399 #endif
00400
00401 #ifdef HAVE_UNLOCKPT
00402 unlockpt(d->masterFd);
00403 #endif
00404
00405 d->slaveFd = ::open(d->ttyName.data(), O_RDWR | O_NOCTTY);
00406 if (d->slaveFd < 0)
00407 {
00408 kdWarning(175) << "Can't open slave pseudo teletype" << endl;
00409 ::close(d->masterFd);
00410 d->masterFd = -1;
00411 return false;
00412 }
00413
00414 #if (defined(__svr4__) || defined(__sgi__))
00415
00416 ioctl(d->slaveFd, I_PUSH, "ptem");
00417 ioctl(d->slaveFd, I_PUSH, "ldterm");
00418 #endif
00419
00420
00421
00422
00423
00424 struct ::termios ttmode;
00425
00426 _tcgetattr(d->slaveFd, &ttmode);
00427
00428 if (!d->xonXoff)
00429 ttmode.c_iflag &= ~(IXOFF | IXON);
00430 else
00431 ttmode.c_iflag |= (IXOFF | IXON);
00432
00433 #ifdef IUTF8
00434 if (!d->utf8)
00435 ttmode.c_iflag &= ~IUTF8;
00436 else
00437 ttmode.c_iflag |= IUTF8;
00438 #endif
00439
00440 ttmode.c_cc[VINTR] = CTRL('C' - '@');
00441 ttmode.c_cc[VQUIT] = CTRL('\\' - '@');
00442 ttmode.c_cc[VERASE] = 0177;
00443
00444 _tcsetattr(d->slaveFd, &ttmode);
00445
00446
00447 ioctl(d->slaveFd, TIOCSWINSZ, (char *)&d->winSize);
00448
00449 fcntl(d->masterFd, F_SETFD, FD_CLOEXEC);
00450 fcntl(d->slaveFd, F_SETFD, FD_CLOEXEC);
00451
00452 return true;
00453 }
00454
00455 void KPty::close()
00456 {
00457 if (d->masterFd < 0)
00458 return;
00459
00460 if (memcmp(d->ttyName.data(), "/dev/pts/", 9)) {
00461 if (!geteuid()) {
00462 struct stat st;
00463 if (!stat(d->ttyName.data(), &st)) {
00464 chown(d->ttyName.data(), 0, st.st_gid == getgid() ? 0 : -1);
00465 chmod(d->ttyName.data(), S_IRUSR|S_IWUSR|S_IRGRP|S_IWGRP|S_IROTH|S_IWOTH);
00466 }
00467 } else {
00468 fcntl(d->masterFd, F_SETFD, 0);
00469 chownpty(false);
00470 }
00471 }
00472 ::close(d->slaveFd);
00473 ::close(d->masterFd);
00474 d->masterFd = d->slaveFd = -1;
00475 }
00476
00477 void KPty::setCTty()
00478 {
00479
00480
00481
00482
00483 setsid();
00484
00485
00486 #ifdef TIOCSCTTY
00487 ioctl(d->slaveFd, TIOCSCTTY, 0);
00488 #else
00489
00490 ::close(::open(d->ttyName, O_WRONLY, 0));
00491 #endif
00492
00493
00494 int pgrp = getpid();
00495 #if defined(_POSIX_VERSION) || defined(__svr4__)
00496 tcsetpgrp (d->slaveFd, pgrp);
00497 #elif defined(TIOCSPGRP)
00498 ioctl(d->slaveFd, TIOCSPGRP, (char *)&pgrp);
00499 #endif
00500 }
00501
00502 void KPty::login(const char *user, const char *remotehost)
00503 {
00504 #ifdef HAVE_UTEMPTER
00505 KProcess_Utmp utmp;
00506 utmp.cmdFd = d->masterFd;
00507 utmp << "/usr/sbin/utempter" << "-a" << d->ttyName << "";
00508 utmp.start(KProcess::Block);
00509 Q_UNUSED(user);
00510 Q_UNUSED(remotehost);
00511 #elif defined(USE_LOGIN)
00512 const char *str_ptr;
00513 struct utmp l_struct;
00514 memset(&l_struct, 0, sizeof(struct utmp));
00515
00516
00517 if (user)
00518 strncpy(l_struct.ut_name, user, UT_NAMESIZE);
00519
00520 if (remotehost)
00521 strncpy(l_struct.ut_host, remotehost, UT_HOSTSIZE);
00522
00523 # ifndef __GLIBC__
00524 str_ptr = d->ttyName.data();
00525 if (!memcmp(str_ptr, "/dev/", 5))
00526 str_ptr += 5;
00527 strncpy(l_struct.ut_line, str_ptr, UT_LINESIZE);
00528 # endif
00529
00530
00531
00532 {
00533 time_t ut_time_temp;
00534 time(&ut_time_temp);
00535 l_struct.ut_time=ut_time_temp;
00536 }
00537
00538 ::login(&l_struct);
00539 #else
00540 Q_UNUSED(user);
00541 Q_UNUSED(remotehost);
00542 #endif
00543 }
00544
00545 void KPty::logout()
00546 {
00547 #ifdef HAVE_UTEMPTER
00548 KProcess_Utmp utmp;
00549 utmp.cmdFd = d->masterFd;
00550 utmp << "/usr/sbin/utempter" << "-d" << d->ttyName;
00551 utmp.start(KProcess::Block);
00552 #elif defined(USE_LOGIN)
00553 const char *str_ptr = d->ttyName.data();
00554 if (!memcmp(str_ptr, "/dev/", 5))
00555 str_ptr += 5;
00556 # ifdef __GLIBC__
00557 else {
00558 const char *sl_ptr = strrchr(str_ptr, '/');
00559 if (sl_ptr)
00560 str_ptr = sl_ptr + 1;
00561 }
00562 # endif
00563 ::logout(str_ptr);
00564 #endif
00565 }
00566
00567 void KPty::setWinSize(int lines, int columns)
00568 {
00569 d->winSize.ws_row = (unsigned short)lines;
00570 d->winSize.ws_col = (unsigned short)columns;
00571 if (d->masterFd >= 0)
00572 ioctl( d->masterFd, TIOCSWINSZ, (char *)&d->winSize );
00573 }
00574
00575 void KPty::setXonXoff(bool useXonXoff)
00576 {
00577 d->xonXoff = useXonXoff;
00578 if (d->masterFd >= 0) {
00579
00580
00581
00582 struct ::termios ttmode;
00583
00584 _tcgetattr(d->masterFd, &ttmode);
00585
00586 if (!useXonXoff)
00587 ttmode.c_iflag &= ~(IXOFF | IXON);
00588 else
00589 ttmode.c_iflag |= (IXOFF | IXON);
00590
00591 _tcsetattr(d->masterFd, &ttmode);
00592 }
00593 }
00594
00595 void KPty::setUtf8Mode(bool useUtf8)
00596 {
00597 d->utf8 = useUtf8;
00598 #ifdef IUTF8
00599 if (d->masterFd >= 0) {
00600
00601
00602
00603 struct ::termios ttmode;
00604
00605 _tcgetattr(d->masterFd, &ttmode);
00606
00607 if (!useUtf8)
00608 ttmode.c_iflag &= ~IUTF8;
00609 else
00610 ttmode.c_iflag |= IUTF8;
00611
00612 _tcsetattr(d->masterFd, &ttmode);
00613 }
00614 #endif
00615 }
00616
00617 const char *KPty::ttyName() const
00618 {
00619 return d->ttyName.data();
00620 }
00621
00622 int KPty::masterFd() const
00623 {
00624 return d->masterFd;
00625 }
00626
00627 int KPty::slaveFd() const
00628 {
00629 return d->slaveFd;
00630 }
00631
00632
00633 bool KPty::chownpty(bool grant)
00634 {
00635 KProcess proc;
00636 proc << locate("exe", BASE_CHOWN) << (grant?"--grant":"--revoke") << QString::number(d->masterFd);
00637 return proc.start(KProcess::Block) && proc.normalExit() && !proc.exitStatus();
00638 }
00639