12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697 |
- /* vim: tabstop=4 shiftwidth=4 noexpandtab
- * This file is part of ToaruOS and is released under the terms
- * of the NCSA / University of Illinois License - see LICENSE.md
- * Copyright (C) 2018 K. Lange
- *
- * getty - Manage a TTY.
- *
- * Wraps a serial port (or other dumb connection) with a pty
- * and manages a login for it.
- */
- #include <stdio.h>
- #include <stdlib.h>
- #include <unistd.h>
- #include <fcntl.h>
- #include <pty.h>
- #include <sys/wait.h>
- #include <sys/fswait.h>
- int main(int argc, char * argv[]) {
- int fd_master, fd_slave, fd_serial;
- char * file = "/dev/ttyS0";
- char * user = NULL;
- if (getuid() != 0) {
- fprintf(stderr, "%s: only root can do that\n", argv[0]);
- return 1;
- }
- int opt;
- while ((opt = getopt(argc, argv, "a:")) != -1) {
- switch (opt) {
- case 'a':
- user = optarg;
- break;
- }
- }
- if (optind < argc) {
- file = argv[optind];
- }
- openpty(&fd_master, &fd_slave, NULL, NULL, NULL);
- fd_serial = open(file, O_RDWR);
- pid_t child = fork();
- if (!child) {
- setsid();
- dup2(fd_slave, 0);
- dup2(fd_slave, 1);
- dup2(fd_slave, 2);
- system("ttysize -q");
- char * tokens[] = {"/bin/login",NULL,NULL,NULL};
- if (user) {
- tokens[1] = "-f";
- tokens[2] = user;
- }
- execvp(tokens[0], tokens);
- exit(1);
- } else {
- int fds[2] = {fd_serial, fd_master};
- while (1) {
- int index = fswait2(2,fds,200);
- char buf[1024];
- int r;
- switch (index) {
- case 0: /* fd_serial */
- r = read(fd_serial, buf, 1);
- write(fd_master, buf, 1);
- break;
- case 1: /* fd_master */
- r = read(fd_master, buf, 1024);
- write(fd_serial, buf, r);
- break;
- default: /* timeout */
- {
- int result = waitpid(child, NULL, WNOHANG);
- if (result > 0) {
- /* Child login shell has returned (session ended) */
- return 0;
- }
- }
- break;
- }
- }
- }
- return 0;
- }
|