a DOg@sdZddlmZddlZddlZddlZgdZdZdZdZdZ ddZ d d Z d d Z d dZ ddZddZddZeefddZeefddZdS)zPseudo terminal utilities.)selectN)openptyforkspawnc Cs<z tWSttfy Yn0t\}}t|}||fS)zdopenpty() -> (master_fd, slave_fd) Open a pty master/slave pair, using os.openpty() if possible.)osrAttributeErrorOSError_open_terminal slave_open) master_fd slave_nameslave_fdr/usr/lib64/python3.9/pty.pyrs  rc CsJzt\}}Wnttfy&Yn0t|}t|||fStS)zmaster_open() -> (master_fd, slave_name) Open a pty master and return the fd, and the filename of the slave end. Deprecated, use openpty() instead.)rrr r ttynamecloser )r rrrrr master_open"s  rc CsldD]Z}dD]P}d||}zt|tj}WntyDYq Yn0|d||fSqtddS)z1Open pty master and return (master_fd, tty_name).ZpqrstuvwxyzPQRSTZ0123456789abcdefz/dev/ptyz/dev/ttyzout of pty devicesN)ropenO_RDWRr )xyZpty_namefdrrrr 2s   r cCsnt|tj}zddlm}m}Wnty8|YS0z|||d|||dWntyhYn0|S)zslave_open(tty_name) -> slave_fd Open the pty slave and acquire the controlling terminal, returning opened filedescriptor. Deprecated, use openpty() instead.r)ioctlI_PUSHZptemZldterm)rrrZfcntlrr ImportErrorr )Ztty_nameresultrrrrrr >s    r c Cszt\}}Wnttfy&Yn20|tkrPz tWntyNYn0||fSt\}}t}|tkrtt|t |t t |t t |t |t krt|t tt tj}t|n t|||fS)zdfork() -> (pid, master_fd) Fork and make the child a session leader with a controlling terminal.)rforkptyr r CHILDsetsidrrrdup2 STDIN_FILENO STDOUT_FILENO STDERR_FILENOrrr)pidrr rZtmp_fdrrrrPs0          rcCs"|rt||}||d}qdS)z#Write all the data to a descriptor.N)rwrite)rdatanrrr_writenxs r)cCs t|dS)zDefault read function.i)rread)rrrr_read~sr+cCsv|tg}t|gg\}}}||vrF||}|s:||n tt|t|vr|t}|sf|tqt||qdS)zParent copy loop. Copies pty master -> standard output (master_read) standard input -> pty master (stdin_read)N)r"rremoverr&r#r))r master_read stdin_readfdsZrfdsZwfdsZxfdsr'rrr_copys   r0cCst|tdkr|f}td|t\}}|tkrJtj|dg|Rztt }t t d}Wntj y~d}Yn0zt |||Wn&t y|rtt tj|Yn0t|t|ddS)zCreate a spawned process.z pty.spawnrr)typesysauditrrrexeclpttyZ tcgetattrr"Zsetrawerrorr0r Z tcsetattrZ TCSAFLUSHrwaitpid)argvr-r.r%r modeZrestorerrrrs&       r)__doc__rrr3r6__all__r"r#r$rrrr r rr)r+r0rrrrrs$   (