おはようございます。先日長男は、保育園のキャンプに参加して、初めての保育園外での泊りをしてきました。ここ数年、天候が微妙だった園のキャンプでしたが、今年は天気に恵まれ、楽しんで帰ってきました。@kjunichiです。

背景

RustでWebカメラの画像をコマンドラインに表示するコマンドを作っている。このコマンドをESCキーが押されたら、終了するようにしたい。

アプリ側で、映像を表示させつつ、キー入力を監視して、特定のキーが押されたら、終了したい。

（シグナルを使って、SIGINT時の処理にこの終了処理を加えても対応できるが、Windowsの事を考えると、キー操作の統一感が無い気がして、今回は見送った。）

Windowsのcmd.exe上で動かす場合、Win32APIのキー入力API(GetAsyncKeyState)でキーが押されたかの判定をブロックせずにできた。

pub fn term_get_esc_key() -> bool {
    let r: i16 = unsafe { ffi::GetAsyncKeyState(ffi::VK_ESCAPE) } as i16;
    if r != 0 {
        return true;
    }
    return false;
}

同じようなことを、macOSやLinux等のUNIX系の環境で行いたい。

これらの環境では、標準入力に文字が入力されることで判定する方法がある。しかし、何も考えず標準入力からの入力を受け取ろうとするとブロックされてしまう。それなら、スレッドでとやったら、

といったツイートの様に単にスレッドで入力待ちを行っても多くの環境でプロセス単位にブロックされ、入力待ち以外の処理が行えないことを経験した。

分かった事

readシステムコールでのブロックを防ぐ

標準入力はプロセス実行時に既にオープンされており、このようなファイルデスクリプタはfcntlで以下の様にノンブロックの指定をすることで、以降のreadでブロックされなくなる。

fcntl(0, F_SETFL, O_NONBLOCK);

エンターキーが押されるのを待たない

readをノンブロックにしても、まだ駄目で、エンターキーが入力されるまで、入力を検知できない。これは通常使用している端末がそういった動きになるように設定しているからで、端末の属性をプログラム実行時に変更する必要があった。

幸い以下のページを見つけて、端末の属性を変更する方法が分かった。

https://oshiete.goo.ne.jp/qa/160619.html

プログラム開始後に端末の設定を変えており、プログラム終了後にこの状態を元に戻す必要がある。戻さないと、コマンド入力がエンターまで待ってくれず、まともに出来なくなることがあるらしい。

macOSの場合、そういった状況に陥ったらreset+エンターで割と復活することが多い。

現在の端末属性を保持する

tcgetattr(0, &saved_term);

エンターまで入力を待たずに受け取る

term.c_lflag &= ~(ICANON | ECHO);
tcsetattr(0, TCSANOW, &term);

プログラム終了時に端末属性を元に戻す

tcsetattr(0, TCSANOW, &saved_term);

以上のことをRustでやった

libcクレートにこれらのAPIが定義されていた。定数も皆今回必要なモノは定義済みで助かった。

extern crate libc;

use std::{thread, time};
use std::io::Write;
use std::time::SystemTime;

fn main() {
    println!("Hello, world!");
    let mut saved_termattr = libc::termios {
        c_iflag: 0,
        c_oflag: 0,
        c_cflag: 0,
        c_lflag: 0,
        c_cc: [0u8; 20],
        c_ispeed: 0,
        c_ospeed: 0,
    };
    unsafe {
        let mut ptr = &mut saved_termattr;
        libc::tcgetattr(0, ptr);
    }
    let mut termattr = saved_termattr;
    termattr.c_lflag = termattr.c_lflag & !(libc::ICANON | libc::ECHO);
    termattr.c_cc[libc::VMIN] = 1;
    termattr.c_cc[libc::VTIME] = 0;
    unsafe {
        libc::tcsetattr(0, libc::TCSANOW, &termattr);
    }
    unsafe {
        libc::fcntl(0, libc::F_SETFL, libc::O_NONBLOCK);
    }
    let mut buf: [libc::c_char; 1] = [0; 1];
    let ptr = &mut buf;

    loop {
        let r = unsafe { libc::read(0, ptr.as_ptr() as *mut libc::c_void, 1) };
        if r > 0 {
            println!("input !{:?}", *ptr);
            break;
        }
        thread::sleep(time::Duration::from_millis(300));
        let code: [char; 3] = [0x1b as char, '[', 'K'];
        print!("\r");
        for i in 0..3 {
            print!("{}", code[i]);
        }
        print!("{:?}", SystemTime::now());
        std::io::stdout().flush().unwrap();

    }
    unsafe {
        libc::tcsetattr(0, libc::TCSANOW, &saved_termattr);
    }
}