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
yaml_emitter_set_writer_error(yaml_emitter_t * emitter,const char * problem)19 yaml_emitter_set_writer_error(yaml_emitter_t *emitter, const char *problem)
20 {
21 emitter->error = YAML_WRITER_ERROR;
22 emitter->problem = problem;
23
24 return 0;
25 }
26
27 /*
28 * Flush the output buffer.
29 */
30
31 YAML_DECLARE(int)
yaml_emitter_flush(yaml_emitter_t * emitter)32 yaml_emitter_flush(yaml_emitter_t *emitter)
33 {
34 int low, high;
35
36 assert(emitter); /* Non-NULL emitter object is expected. */
37 assert(emitter->write_handler); /* Write handler must be set. */
38 assert(emitter->encoding); /* Output encoding must be set. */
39
40 emitter->buffer.last = emitter->buffer.pointer;
41 emitter->buffer.pointer = emitter->buffer.start;
42
43 /* Check if the buffer is empty. */
44
45 if (emitter->buffer.start == emitter->buffer.last) {
46 return 1;
47 }
48
49 /* If the output encoding is UTF-8, we don't need to recode the buffer. */
50
51 if (emitter->encoding == YAML_UTF8_ENCODING)
52 {
53 if (emitter->write_handler(emitter->write_handler_data,
54 emitter->buffer.start,
55 emitter->buffer.last - emitter->buffer.start)) {
56 emitter->buffer.last = emitter->buffer.start;
57 emitter->buffer.pointer = emitter->buffer.start;
58 return 1;
59 }
60 else {
61 return yaml_emitter_set_writer_error(emitter, "write error");
62 }
63 }
64
65 /* Recode the buffer into the raw buffer. */
66
67 low = (emitter->encoding == YAML_UTF16LE_ENCODING ? 0 : 1);
68 high = (emitter->encoding == YAML_UTF16LE_ENCODING ? 1 : 0);
69
70 while (emitter->buffer.pointer != emitter->buffer.last)
71 {
72 unsigned char octet;
73 unsigned int width;
74 unsigned int value;
75 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 octet = emitter->buffer.pointer[0];
85
86 width = (octet & 0x80) == 0x00 ? 1 :
87 (octet & 0xE0) == 0xC0 ? 2 :
88 (octet & 0xF0) == 0xE0 ? 3 :
89 (octet & 0xF8) == 0xF0 ? 4 : 0;
90
91 value = (octet & 0x80) == 0x00 ? octet & 0x7F :
92 (octet & 0xE0) == 0xC0 ? octet & 0x1F :
93 (octet & 0xF0) == 0xE0 ? octet & 0x0F :
94 (octet & 0xF8) == 0xF0 ? octet & 0x07 : 0;
95
96 for (k = 1; k < width; k ++) {
97 octet = emitter->buffer.pointer[k];
98 value = (value << 6) + (octet & 0x3F);
99 }
100
101 emitter->buffer.pointer += width;
102
103 /* Write the character. */
104
105 if (value < 0x10000)
106 {
107 emitter->raw_buffer.last[high] = value >> 8;
108 emitter->raw_buffer.last[low] = value & 0xFF;
109
110 emitter->raw_buffer.last += 2;
111 }
112 else
113 {
114 /* Write the character using a surrogate pair (check "reader.c"). */
115
116 value -= 0x10000;
117 emitter->raw_buffer.last[high] = 0xD8 + (value >> 18);
118 emitter->raw_buffer.last[low] = (value >> 10) & 0xFF;
119 emitter->raw_buffer.last[high+2] = 0xDC + ((value >> 8) & 0xFF);
120 emitter->raw_buffer.last[low+2] = value & 0xFF;
121
122 emitter->raw_buffer.last += 4;
123 }
124 }
125
126 /* Write the raw buffer. */
127
128 if (emitter->write_handler(emitter->write_handler_data,
129 emitter->raw_buffer.start,
130 emitter->raw_buffer.last - emitter->raw_buffer.start)) {
131 emitter->buffer.last = emitter->buffer.start;
132 emitter->buffer.pointer = emitter->buffer.start;
133 emitter->raw_buffer.last = emitter->raw_buffer.start;
134 emitter->raw_buffer.pointer = emitter->raw_buffer.start;
135 return 1;
136 }
137 else {
138 return yaml_emitter_set_writer_error(emitter, "write error");
139 }
140 }
141
142