binary-security-check 2.0.2

Analyzer of security features in executable binaries
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

// Copyright 2018-2025 Koutheir Attouchi.
// See the "LICENSE.txt" file at the top-level directory of this distribution.
//
// Licensed under the MIT license. This file may not be copied, modified,
// or distributed except according to those terms.

use alloc::sync::Arc;

use crate::cmdline::UseColor;
use crate::errors::{Error, Result};

/// A color buffer that can should be written-to from a single thread.
/// If cloned and given to another thread, then both threads can write to their own color buffer
/// without synchronizing, and later a joining thread can perform the synchronization and write
/// all cloned color buffers.
pub(crate) struct ColorBuffer {
    buffer_writer: Arc<termcolor::BufferWriter>,
    pub(crate) color_buffer: termcolor::Buffer,
}

impl ColorBuffer {
    pub(crate) fn for_stdout(use_color: UseColor) -> Self {
        let buffer_writer = termcolor::BufferWriter::stdout(use_color.into());
        let color_buffer = buffer_writer.buffer();

        Self {
            buffer_writer: Arc::new(buffer_writer),
            color_buffer,
        }
    }

    pub(crate) fn print(&self) -> Result<()> {
        self.buffer_writer
            .print(&self.color_buffer)
            .map_err(|r| Error::from_io1(r, "print", "standard output stream"))?;
        Ok(())
    }
}

impl Clone for ColorBuffer {
    fn clone(&self) -> Self {
        Self {
            // Increment the reference count of the `BufferWriter`.
            buffer_writer: Arc::clone(&self.buffer_writer),
            // Create a new buffer linked to `buffer_writer`.
            color_buffer: self.buffer_writer.buffer(),
        }
    }
}