Coverage Report

Created: 2025-07-11 06:35

/src/libyaml/src/writer.c
Line
Count
Source (jump to first uncovered line)
1
2
#include "yaml_private.h"
3
4
/*
5
 * Declarations.
6
 */
7
8
static int
9
yaml_emitter_set_writer_error(yaml_emitter_t *emitter, const char *problem);
10
11
YAML_DECLARE(int)
12
yaml_emitter_flush(yaml_emitter_t *emitter);
13
14
/*
15
 * Set the writer error and return 0.
16
 */
17
18
static int
19
yaml_emitter_set_writer_error(yaml_emitter_t *emitter, const char *problem)
20
4.24k
{
21
4.24k
    emitter->error = YAML_WRITER_ERROR;
22
4.24k
    emitter->problem = problem;
23
24
4.24k
    return 0;
25
4.24k
}
26
27
/*
28
 * Flush the output buffer.
29
 */
30
31
YAML_DECLARE(int)
32
yaml_emitter_flush(yaml_emitter_t *emitter)
33
256k
{
34
256k
    int low, high;
35
36
256k
    assert(emitter);    /* Non-NULL emitter object is expected. */
37
256k
    assert(emitter->write_handler); /* Write handler must be set. */
38
256k
    assert(emitter->encoding);  /* Output encoding must be set. */
39
40
256k
    emitter->buffer.last = emitter->buffer.pointer;
41
256k
    emitter->buffer.pointer = emitter->buffer.start;
42
43
    /* Check if the buffer is empty. */
44
45
256k
    if (emitter->buffer.start == emitter->buffer.last) {
46
1.16k
        return 1;
47
1.16k
    }
48
49
    /* If the output encoding is UTF-8, we don't need to recode the buffer. */
50
51
255k
    if (emitter->encoding == YAML_UTF8_ENCODING)
52
251k
    {
53
251k
        if (emitter->write_handler(emitter->write_handler_data,
54
251k
                    emitter->buffer.start,
55
251k
                    emitter->buffer.last - emitter->buffer.start)) {
56
247k
            emitter->buffer.last = emitter->buffer.start;
57
247k
            emitter->buffer.pointer = emitter->buffer.start;
58
247k
            return 1;
59
247k
        }
60
3.95k
        else {
61
3.95k
            return yaml_emitter_set_writer_error(emitter, "write error");
62
3.95k
        }
63
251k
    }
64
65
    /* Recode the buffer into the raw buffer. */
66
67
3.62k
    low = (emitter->encoding == YAML_UTF16LE_ENCODING ? 0 : 1);
68
3.62k
    high = (emitter->encoding == YAML_UTF16LE_ENCODING ? 1 : 0);
69
70
42.8M
    while (emitter->buffer.pointer != emitter->buffer.last)
71
42.8M
    {
72
42.8M
        unsigned char octet;
73
42.8M
        unsigned int width;
74
42.8M
        unsigned int value;
75
42.8M
        size_t k;
76
77
        /*
78
         * See the "reader.c" code for more details on UTF-8 encoding.  Note
79
         * that we assume that the buffer contains a valid UTF-8 sequence.
80
         */
81
82
        /* Read the next UTF-8 character. */
83
84
42.8M
        octet = emitter->buffer.pointer[0];
85
86
42.8M
        width = (octet & 0x80) == 0x00 ? 1 :
87
42.8M
                (octet & 0xE0) == 0xC0 ? 2 :
88
2.45M
                (octet & 0xF0) == 0xE0 ? 3 :
89
2.45M
                (octet & 0xF8) == 0xF0 ? 4 : 0;
90
91
42.8M
        value = (octet & 0x80) == 0x00 ? octet & 0x7F :
92
42.8M
                (octet & 0xE0) == 0xC0 ? octet & 0x1F :
93
2.45M
                (octet & 0xF0) == 0xE0 ? octet & 0x0F :
94
2.45M
                (octet & 0xF8) == 0xF0 ? octet & 0x07 : 0;
95
96
47.7M
        for (k = 1; k < width; k ++) {
97
4.91M
            octet = emitter->buffer.pointer[k];
98
4.91M
            value = (value << 6) + (octet & 0x3F);
99
4.91M
        }
100
101
42.8M
        emitter->buffer.pointer += width;
102
103
        /* Write the character. */
104
105
42.8M
        if (value < 0x10000)
106
42.8M
        {
107
42.8M
            emitter->raw_buffer.last[high] = value >> 8;
108
42.8M
            emitter->raw_buffer.last[low] = value & 0xFF;
109
110
42.8M
            emitter->raw_buffer.last += 2;
111
42.8M
        }
112
0
        else
113
0
        {
114
            /* Write the character using a surrogate pair (check "reader.c"). */
115
116
0
            value -= 0x10000;
117
0
            emitter->raw_buffer.last[high] = 0xD8 + (value >> 18);
118
0
            emitter->raw_buffer.last[low] = (value >> 10) & 0xFF;
119
0
            emitter->raw_buffer.last[high+2] = 0xDC + ((value >> 8) & 0xFF);
120
0
            emitter->raw_buffer.last[low+2] = value & 0xFF;
121
122
0
            emitter->raw_buffer.last += 4;
123
0
        }
124
42.8M
    }
125
126
    /* Write the raw buffer. */
127
128
3.62k
    if (emitter->write_handler(emitter->write_handler_data,
129
3.62k
                emitter->raw_buffer.start,
130
3.62k
                emitter->raw_buffer.last - emitter->raw_buffer.start)) {
131
3.33k
        emitter->buffer.last = emitter->buffer.start;
132
3.33k
        emitter->buffer.pointer = emitter->buffer.start;
133
3.33k
        emitter->raw_buffer.last = emitter->raw_buffer.start;
134
3.33k
        emitter->raw_buffer.pointer = emitter->raw_buffer.start;
135
3.33k
        return 1;
136
3.33k
    }
137
293
    else {
138
293
        return yaml_emitter_set_writer_error(emitter, "write error");
139
293
    }
140
3.62k
}
141