gdt-cpus 0.2606.0

Game Developer's Toolkit for CPU Management
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
68
69
70
71
72
73
74
75
76
77
78
79

//! Kernel range-list parsing ("0-3,7,10-11" -> CPU IDs).
//!
//! Platform-neutral on purpose: Linux production code parses `cpu/online`,
//! `cache/index*/shared_cpu_list` and `node*/cpulist` with it, and the
//! cross-platform fixture checker uses it for `expected.txt` LP lists.

use crate::{Error, Result};

/// Parses a kernel range-list string ("0-3,7,10-11"), calling `sink` once per
/// CPU id in order. Allocates nothing - the caller decides what to do with each
/// id (count it, set a mask bit, track a minimum), so the hot detection loops
/// that only need a membership test / count / min never materialize a throwaway
/// `Vec` per sysfs line.
///
/// # Errors
///
/// Returns `Error::Detection` on malformed ranges or non-numeric IDs.
pub(crate) fn parse_range_list_with<F: FnMut(usize)>(range_str: &str, mut sink: F) -> Result<()> {
    for part in range_str.trim().split(',') {
        let part = part.trim();

        if part.is_empty() {
            continue;
        }

        if part.contains('-') {
            let mut iter = part.splitn(2, '-');

            let start_str = iter
                .next()
                .ok_or_else(|| Error::Detection(format!("Invalid CPU range format: {}", part)))?;
            let end_str = iter
                .next()
                .ok_or_else(|| Error::Detection(format!("Invalid CPU range format: {}", part)))?;

            let start = start_str
                .parse::<usize>()
                .map_err(|_| Error::Detection(format!("Invalid CPU range start: {}", start_str)))?;
            let end = end_str
                .parse::<usize>()
                .map_err(|_| Error::Detection(format!("Invalid CPU range end: {}", end_str)))?;

            if start > end {
                return Err(Error::Detection(format!(
                    "Invalid CPU range order: {}-{}",
                    start, end
                )));
            }

            for id in start..=end {
                sink(id);
            }
        } else {
            let cpu_id = part
                .parse::<usize>()
                .map_err(|_| Error::Detection(format!("Invalid CPU ID in range list: {}", part)))?;

            sink(cpu_id);
        }
    }

    Ok(())
}

/// Parses a kernel range-list string ("0-3,7,10-11") into CPU IDs.
///
/// Convenience over [`parse_range_list_with`] for the call sites that genuinely
/// need the materialized list (`cpu/online`, the fixture checker).
///
/// # Errors
///
/// Returns `Error::Detection` on malformed ranges or non-numeric IDs.
pub(crate) fn parse_range_list_str(range_str: &str) -> Result<Vec<usize>> {
    let mut cpus = Vec::new();

    parse_range_list_with(range_str, |id| cpus.push(id))?;

    Ok(cpus)
}