Skip to content

Commit

Permalink
Merge pull request #127 from nivkner/term_size
Browse files Browse the repository at this point in the history
use `TIOCGWINSZ` directly from `libc`
  • Loading branch information
ticki authored Nov 22, 2017
2 parents 52a22ea + 00f87e5 commit b3e0bbd
Showing 1 changed file with 2 additions and 30 deletions.
32 changes: 2 additions & 30 deletions src/sys/unix/size.rs
Original file line number Diff line number Diff line change
@@ -1,7 +1,7 @@
use std::{io, mem};

use super::cvt;
use super::libc::{c_ushort, ioctl, STDOUT_FILENO};
use super::libc::{c_ushort, ioctl, STDOUT_FILENO, TIOCGWINSZ};

#[repr(C)]
struct TermSize {
Expand All @@ -10,39 +10,11 @@ struct TermSize {
_x: c_ushort,
_y: c_ushort,
}

#[cfg(target_os = "linux")]
pub const TIOCGWINSZ: usize = 0x00005413;

#[cfg(not(target_os = "linux"))]
pub const TIOCGWINSZ: usize = 0x40087468;

// Since attributes on non-item statements is not stable yet, we use a function.
#[cfg(not(target_os = "android"))]
#[cfg(not(target_os = "redox"))]
#[cfg(target_pointer_width = "64")]
#[cfg(not(target_env = "musl"))]
fn tiocgwinsz() -> u64 {
TIOCGWINSZ as u64
}
#[cfg(not(target_os = "android"))]
#[cfg(not(target_os = "redox"))]
#[cfg(target_pointer_width = "32")]
#[cfg(not(target_env = "musl"))]
fn tiocgwinsz() -> u32 {
TIOCGWINSZ as u32
}

#[cfg(any(target_env = "musl", target_os = "android"))]
fn tiocgwinsz() -> i32 {
TIOCGWINSZ as i32
}

/// Get the size of the terminal.
pub fn terminal_size() -> io::Result<(u16, u16)> {
unsafe {
let mut size: TermSize = mem::zeroed();
cvt(ioctl(STDOUT_FILENO, tiocgwinsz(), &mut size as *mut _))?;
cvt(ioctl(STDOUT_FILENO, TIOCGWINSZ, &mut size as *mut _))?;
Ok((size.col as u16, size.row as u16))
}
}

0 comments on commit b3e0bbd

Please sign in to comment.