ztmux 0.1.0

A Rust port of tmux — the full terminal multiplexer, server and client
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67

use crate::libc;

/// The `strlcat()` function appends the NUL-terminated string src to the end of dst.
/// It will append at most size - strlen(dst) - 1 bytes, NUL-terminating the result.
/// C `vendor/tmux/compat/strlcat.c:32`: `size_t strlcat(char *dst, const char *src, size_t siz)`
pub unsafe fn strlcat(dst: *mut u8, src: *const u8, size: usize) -> usize {
    unsafe {
        let dst_strlen = libc::strnlen(dst, size);
        let src_strlen = libc::strnlen(src, size.saturating_sub(dst_strlen).saturating_sub(1));

        core::ptr::copy_nonoverlapping(src, dst.add(dst_strlen), src_strlen);
        *dst.add(dst_strlen + src_strlen) = b'\0';

        dst_strlen + src_strlen
    }
}

#[expect(clippy::disallowed_methods)]
pub unsafe fn strlcat_(dst: *mut u8, src: &str, size: usize) -> usize {
    unsafe {
        let dst_strlen = libc::strnlen(dst, size);
        let src_strlen = src
            .len()
            .min(size.saturating_sub(dst_strlen).saturating_sub(1));

        core::ptr::copy_nonoverlapping(src.as_ptr(), dst.add(dst_strlen), src_strlen);
        *dst.add(dst_strlen + src_strlen) = b'\0';

        dst_strlen + src_strlen
    }
}

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn test_strlcat_append() {
        // "foo" + "bar" fits: return value is combined length, NUL-terminated.
        let mut dst = [0u8; 10];
        dst[..3].copy_from_slice(b"foo");
        let src = crate::c!("bar");
        let ret = unsafe { strlcat(dst.as_mut_ptr(), src, dst.len()) };
        assert_eq!(ret, 6);
        assert_eq!(&dst[..7], b"foobar\0");
    }

    #[test]
    fn test_strlcat_truncation() {
        // size=5, dst holds "foo": only 5-3-1=1 byte of src is appended.
        let mut dst = [0u8; 10];
        dst[..3].copy_from_slice(b"foo");
        let src = crate::c!("bar");
        let ret = unsafe { strlcat(dst.as_mut_ptr(), src, 5) };
        assert_eq!(ret, 4);
        assert_eq!(&dst[..5], b"foob\0");
    }

    #[test]
    fn test_strlcat_empty_dst() {
        let mut dst = [0u8; 10];
        let src = crate::c!("hi");
        let ret = unsafe { strlcat(dst.as_mut_ptr(), src, dst.len()) };
        assert_eq!(ret, 2);
        assert_eq!(&dst[..3], b"hi\0");
    }
}