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
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104

use crate::{
    externs::memset,
    libc,
    memory::{yaml_realloc, yaml_strdup},
    yaml::{size_t, yaml_char_t},
};

/// Extend a string buffer by reallocating and copying the existing data.
///
/// This function is used to grow a string buffer when more space is needed.
///
/// # Safety
///
/// - This function is unsafe because it directly calls the system's `realloc` and
///   `memset` functions, which can lead to undefined behaviour if misused.
/// - The caller must ensure that `start`, `pointer`, and `end` are valid pointers
///   into the same allocated memory block.
/// - The caller must ensure that the memory block being extended is large enough
///   to accommodate the new size.
/// - The caller is responsible for properly freeing the extended memory block using
///   the corresponding `yaml_free` function when it is no longer needed.
///
pub unsafe fn yaml_string_extend(
    start: *mut *mut yaml_char_t,
    pointer: *mut *mut yaml_char_t,
    end: *mut *mut yaml_char_t,
) {
    let current_size = (*end).offset_from(*start) as size_t;
    let new_size = current_size * 2;

    let new_start: *mut yaml_char_t =
        yaml_realloc(*start as *mut libc::c_void, new_size)
            as *mut yaml_char_t;
    let _ = memset(
        new_start.add(current_size.try_into().unwrap())
            as *mut libc::c_void,
        0,
        current_size,
    );

    let offset = (*pointer).offset_from(*start);
    *pointer = new_start.add(offset.try_into().unwrap());
    *end = new_start.add((new_size as isize).try_into().unwrap());
    *start = new_start;
}

/// Duplicate a null-terminated string.
/// # Safety
/// - This function is unsafe because it involves memory allocation.
pub unsafe fn yaml_string_duplicate(
    str: *const yaml_char_t,
) -> *mut yaml_char_t {
    yaml_strdup(str)
}

/// Join two string buffers by copying data from one to the other.
///
/// This function is used to concatenate two string buffers.
///
/// # Safety
///
/// - This function is unsafe because it directly calls the system's `memcpy` function,
///   which can lead to undefined behaviour if misused.
/// - The caller must ensure that `a_start`, `a_pointer`, `a_end`, `b_start`, `b_pointer`,
///   and `b_end` are valid pointers into their respective allocated memory blocks.
/// - The caller must ensure that the memory blocks being joined are large enough to
///   accommodate the combined data.
/// - The caller is responsible for properly freeing the joined memory block using
///   the corresponding `yaml_free` function when it is no longer needed.
///
pub unsafe fn yaml_string_join(
    a_start: *mut *mut yaml_char_t,
    a_pointer: *mut *mut yaml_char_t,
    a_end: *mut *mut yaml_char_t,
    b_start: *mut *mut yaml_char_t,
    b_pointer: *mut *mut yaml_char_t,
    b_end: *mut *mut yaml_char_t,
) {
    // If b_start is equal to b_pointer, there's nothing to join
    if *b_start == *b_pointer {
        return;
    }

    // Calculate the length of the data in b
    let b_length = ((*b_pointer).offset_from(*b_start))
        .min((*b_end).offset_from(*b_start))
        as usize;

    // If the length of b is 0, there's nothing to copy
    if b_length == 0 {
        return;
    }

    // Ensure there's enough space in a to hold b's content
    while ((*a_end).offset_from(*a_pointer) as usize) < b_length {
        yaml_string_extend(a_start, a_pointer, a_end);
    }

    // Copy b's content to a
    core::ptr::copy_nonoverlapping(*b_start, *a_pointer, b_length);

    // Move a's pointer forward by the length of the copied data
    *a_pointer = (*a_pointer).add(b_length);
}