B S]@sdZddlmZddlZddlZdddgZdZdZdZdZd dZ d d Z d d Z ddZ ddZ ddZddZeefddZeefddZdS)zPseudo terminal utilities.)selectNopenptyforkspawnc Cs<ytSttfk r YnXt\}}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_fdrlib/python3.7/pty.pyrs c CsLyt\}}Wnttfk r(YnXt|}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 Cslx^dD]V}xPdD]H}d||}yt|tj}Wntk rHwYnX|d||fSWqWtddS)z1Open pty master and return (master_fd, tty_name).ZpqrstuvwxyzPQRSTZ0123456789abcdefz/dev/ptyz/dev/ttyzout of pty devicesN)ropenO_RDWRr )xyZpty_namefdrrrr 1s   r cCsnt|tj}yddlm}m}Wntk r6|SXy|||d|||dWntk rhYnX|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 Csyt\}}Wnttfk r(Yn4X|tkrTy tWntk rRYnX||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_fdrrrrOs0         cCs&x |r t||}||d}qWdS)z#Write all the data to a descriptor.N)rwrite)rdatanrrr_writenws r)cCs t|dS)zDefault read function.i)rread)rrrr_read}sr+cCsz|tg}xlt|gg\}}}||krH||}|s<||n tt|t|kr |t}|sh|tq t||q WdS)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_readZfdsZrfdsZwfdsZxfdsr'rrr_copys   r/cCst|tdkr|f}t\}}|tkrs"    (