stty: Support setting baud rate

Part of #3859.
uutils · Jul 7, 2023 · 55d401a · 55d401a
1 parent 3ce5749
commit 55d401a
Showing 1 changed file with 14 additions and 1 deletion.
diff --git a/src/uu/stty/src/stty.rs b/src/uu/stty/src/stty.rs
@@ -10,7 +10,8 @@ mod flags;
 use clap::{crate_version, Arg, ArgAction, ArgMatches, Command};
 use nix::libc::{c_ushort, O_NONBLOCK, TIOCGWINSZ, TIOCSWINSZ};
 use nix::sys::termios::{
-    cfgetospeed, tcgetattr, tcsetattr, ControlFlags, InputFlags, LocalFlags, OutputFlags, Termios,
+    cfgetospeed, cfsetospeed, tcgetattr, tcsetattr, ControlFlags, InputFlags, LocalFlags,
+    OutputFlags, Termios,
 };
 use nix::{ioctl_read_bad, ioctl_write_ptr_bad};
 use std::io::{self, stdout};
@@ -290,6 +291,8 @@ fn print_flags<T: TermiosFlag>(termios: &Termios, opts: &Options, flags: &[Flag<
 /// The value inside the `Break` variant of the `ControlFlow` indicates whether
 /// the setting has been applied.
 fn apply_setting(termios: &mut Termios, s: &str) -> ControlFlow<bool> {
+    apply_baud_rate_flag(termios, s)?;
+
     let (remove, name) = match s.strip_prefix('-') {
         Some(s) => (true, s),
         None => (false, s),
@@ -332,6 +335,16 @@ fn apply_flag<T: TermiosFlag>(
     ControlFlow::Continue(())
 }
 
+fn apply_baud_rate_flag(termios: &mut Termios, input: &str) -> ControlFlow<bool> {
+    for (text, baud_rate) in BAUD_RATES {
+        if *text == input {
+            cfsetospeed(termios, *baud_rate).expect("Failed to set baud rate");
+            return ControlFlow::Break(true);
+        }
+    }
+    ControlFlow::Continue(())
+}
+
 pub fn uu_app() -> Command {
     Command::new(uucore::util_name())
         .version(crate_version!())