Coverage Report

Created: 2026-01-09 06:55

next uncovered line (L), next uncovered region (R), next uncovered branch (B)
/src/capstonenext/SStream.c
Line
Count
Source
1
/* Capstone Disassembly Engine */
2
/* By Nguyen Anh Quynh <aquynh@gmail.com>, 2013-2019 */
3
4
#include <stdarg.h>
5
#if defined(CAPSTONE_HAS_OSXKERNEL)
6
#include <Availability.h>
7
#include <libkern/libkern.h>
8
#include <i386/limits.h>
9
#else
10
#include <stdio.h>
11
#include <limits.h>
12
#endif
13
#include <string.h>
14
15
#include <capstone/platform.h>
16
17
#include "SStream.h"
18
#include "cs_priv.h"
19
#include "utils.h"
20
21
void SStream_Init(SStream *ss)
22
3.28M
{
23
3.28M
  assert(ss);
24
3.28M
  ss->index = 0;
25
3.28M
  memset(ss->buffer, 0, sizeof(ss->buffer));
26
3.28M
  ss->is_closed = false;
27
3.28M
  ss->markup_stream = false;
28
3.28M
  ss->prefixed_by_markup = false;
29
3.28M
  ss->unsigned_num = false;
30
3.28M
}
31
32
void SStream_opt_unum(SStream *ss, bool print_unsigned_numbers)
33
3.17M
{
34
3.17M
  assert(ss);
35
3.17M
  ss->unsigned_num = print_unsigned_numbers;
36
3.17M
}
37
38
/// Returns the a pointer to the internal string buffer of the stream.
39
/// For reading only.
40
const char *SStream_rbuf(const SStream *ss)
41
151k
{
42
151k
  assert(ss);
43
151k
  return ss->buffer;
44
151k
}
45
46
/// Searches in the stream for the first (from the left) occurrence of @elem and replaces
47
/// it with @repl. It returns the pointer *after* the replaced character
48
/// or NULL if no character was replaced.
49
///
50
/// It will never replace the final \0 byte in the stream buffer.
51
const char *SStream_replc(const SStream *ss, char elem, char repl)
52
99.9k
{
53
99.9k
  assert(ss);
54
99.9k
  char *found = strchr(ss->buffer, elem);
55
99.9k
  if (!found || found == ss->buffer + (SSTREAM_BUF_LEN - 1)) {
56
99.9k
    return NULL;
57
99.9k
  }
58
0
  *found = repl;
59
0
  found++;
60
0
  return found;
61
99.9k
}
62
63
/// Searches in the stream for the first (from the left) occurrence of @chr and replaces
64
/// it with @rstr.
65
void SStream_replc_str(SStream *ss, char chr, const char *rstr)
66
37.4k
{
67
37.4k
  assert(ss && rstr);
68
37.4k
  char *found = strchr(ss->buffer, chr);
69
37.4k
  if (!found || found == ss->buffer + (SSTREAM_BUF_LEN - 1)) {
70
0
    return;
71
0
  }
72
37.4k
  size_t post_len = strlen(found + 1);
73
37.4k
  size_t buf_str_len = strlen(ss->buffer);
74
37.4k
  size_t repl_len = strlen(rstr);
75
37.4k
  if (repl_len - 1 + buf_str_len >= SSTREAM_BUF_LEN) {
76
0
    return;
77
0
  }
78
37.4k
  memmove(found + repl_len, found + 1, post_len);
79
37.4k
  memcpy(found, rstr, repl_len);
80
37.4k
  ss->index = strlen(ss->buffer);
81
37.4k
}
82
83
/// Removes the space characters '\t' and ' ' from the beginning of the stream buffer.
84
void SStream_trimls(SStream *ss)
85
3.27M
{
86
3.27M
  assert(ss);
87
3.27M
  size_t buf_off = 0;
88
  /// Remove leading spaces
89
3.33M
  while (ss->buffer[buf_off] == ' ' || ss->buffer[buf_off] == '\t') {
90
61.3k
    buf_off++;
91
61.3k
  }
92
3.27M
  if (buf_off > 0) {
93
61.3k
    memmove(ss->buffer, ss->buffer + buf_off,
94
61.3k
      SSTREAM_BUF_LEN - buf_off);
95
61.3k
    ss->index -= buf_off;
96
61.3k
  }
97
3.27M
}
98
99
/// Extract the mnemonic to @mnem_buf and the operand string into @op_str_buf from the stream buffer.
100
/// The mnemonic is everything up until the first ' ' or '\t' character.
101
/// The operand string is everything after the first ' ' or '\t' sequence.
102
void SStream_extract_mnem_opstr(const SStream *ss, char *mnem_buf,
103
        size_t mnem_buf_size, char *op_str_buf,
104
        size_t op_str_buf_size)
105
3.17M
{
106
3.17M
  assert(ss && mnem_buf && mnem_buf_size > 0 && op_str_buf &&
107
3.17M
         op_str_buf_size > 0);
108
3.17M
  size_t off = 0;
109
  // Copy all non space chars to as mnemonic.
110
18.5M
  while (ss->buffer[off] && ss->buffer[off] != ' ' &&
111
17.7M
         ss->buffer[off] != '\t') {
112
15.3M
    if (off < mnem_buf_size - 1) {
113
      // Only copy if there is space left.
114
15.3M
      mnem_buf[off] = ss->buffer[off];
115
15.3M
    }
116
15.3M
    off++;
117
15.3M
  }
118
3.17M
  if (!ss->buffer[off]) {
119
138k
    return;
120
138k
  }
121
122
  // Iterate until next non space char.
123
3.06M
  do {
124
3.06M
    off++;
125
3.06M
  } while (ss->buffer[off] &&
126
3.00M
     (ss->buffer[off] == ' ' || ss->buffer[off] == '\t'));
127
128
3.03M
  if (!ss->buffer[off]) {
129
55.0k
    return;
130
55.0k
  }
131
132
  // Copy all follow up characters as op_str
133
2.98M
  const char *ss_op_str = ss->buffer + off;
134
2.98M
  off = 0;
135
44.0M
  while (ss_op_str[off] && off < op_str_buf_size - 1) {
136
41.0M
    op_str_buf[off] = ss_op_str[off];
137
41.0M
    off++;
138
41.0M
  }
139
2.98M
}
140
141
/// Empty the stream @ss to given @file (stdin/stderr).
142
/// @file can be NULL. Then the buffer content is not emitted.
143
void SStream_Flush(SStream *ss, FILE *file)
144
37.9k
{
145
37.9k
  assert(ss);
146
37.9k
  if (file) {
147
0
    fprintf(file, "%s\n", ss->buffer);
148
0
  }
149
37.9k
  SStream_Init(ss);
150
37.9k
}
151
152
/**
153
 * Open the output stream. Every write attempt is accepted again.
154
 */
155
void SStream_Open(SStream *ss)
156
0
{
157
0
  assert(ss);
158
0
  ss->is_closed = false;
159
0
}
160
161
/**
162
 * Closes the output stream. Every write attempt is ignored.
163
 */
164
void SStream_Close(SStream *ss)
165
0
{
166
0
  assert(ss);
167
0
  ss->is_closed = true;
168
0
}
169
170
/**
171
 * Copy the string \p s to the buffer of \p ss and terminate it with a '\\0' byte.
172
 */
173
void SStream_concat0(SStream *ss, const char *s)
174
18.8M
{
175
18.8M
#ifndef CAPSTONE_DIET
176
18.8M
  assert(ss && s);
177
18.8M
  SSTREAM_RETURN_IF_CLOSED(ss);
178
18.8M
  if (s[0] == '\0')
179
5.74M
    return;
180
13.0M
  unsigned int len = (unsigned int)strlen(s);
181
182
13.0M
  SSTREAM_OVERFLOW_CHECK(ss, len);
183
184
13.0M
  memcpy(ss->buffer + ss->index, s, len);
185
13.0M
  ss->index += len;
186
13.0M
  ss->buffer[ss->index] = '\0';
187
13.0M
  if (ss->markup_stream && ss->prefixed_by_markup) {
188
0
    SSTREAM_OVERFLOW_CHECK(ss, 1);
189
0
    ss->buffer[ss->index] = '>';
190
0
    ss->index += 1;
191
0
    ss->buffer[ss->index] = '\0';
192
0
  }
193
#else
194
  ss->buffer[ss->index] = '\0';
195
#endif
196
13.0M
}
197
198
/**
199
 * Copy the single char \p c to the buffer of \p ss.
200
 */
201
void SStream_concat1(SStream *ss, const char c)
202
2.26M
{
203
2.26M
#ifndef CAPSTONE_DIET
204
2.26M
  assert(ss);
205
2.26M
  SSTREAM_RETURN_IF_CLOSED(ss);
206
2.26M
  if (c == '\0')
207
28.3k
    return;
208
209
2.24M
  SSTREAM_OVERFLOW_CHECK(ss, 1);
210
211
2.24M
  ss->buffer[ss->index] = c;
212
2.24M
  ss->index++;
213
2.24M
  ss->buffer[ss->index] = '\0';
214
2.24M
  if (ss->markup_stream && ss->prefixed_by_markup) {
215
0
    SSTREAM_OVERFLOW_CHECK(ss, 1);
216
0
    ss->buffer[ss->index] = '>';
217
0
    ss->index++;
218
0
  }
219
#else
220
  ss->buffer[ss->index] = '\0';
221
#endif
222
2.24M
}
223
224
/**
225
 * Copy all strings given to the buffer of \p ss according to formatting \p fmt.
226
 */
227
void SStream_concat(SStream *ss, const char *fmt, ...)
228
8.32M
{
229
8.32M
#ifndef CAPSTONE_DIET
230
8.32M
  assert(ss && fmt);
231
8.32M
  SSTREAM_RETURN_IF_CLOSED(ss);
232
8.32M
  va_list ap;
233
8.32M
  int ret;
234
235
8.32M
  va_start(ap, fmt);
236
8.32M
  ret = cs_vsnprintf(ss->buffer + ss->index,
237
8.32M
         sizeof(ss->buffer) - (ss->index + 1), fmt, ap);
238
8.32M
  va_end(ap);
239
8.32M
  if (ret < 0) {
240
0
    return;
241
0
  }
242
8.32M
  SSTREAM_OVERFLOW_CHECK(ss, ret);
243
8.32M
  ss->index += ret;
244
8.32M
  if (ss->markup_stream && ss->prefixed_by_markup) {
245
0
    SSTREAM_OVERFLOW_CHECK(ss, 1);
246
0
    ss->buffer[ss->index] = '>';
247
0
    ss->index += 1;
248
0
  }
249
#else
250
  ss->buffer[ss->index] = '\0';
251
#endif
252
8.32M
}
253
254
// print number with prefix #
255
void printInt64Bang(SStream *ss, int64_t val)
256
93.7k
{
257
93.7k
  assert(ss);
258
93.7k
  if (ss->unsigned_num) {
259
0
    printUInt64Bang(ss, val);
260
0
    return;
261
0
  }
262
93.7k
  SSTREAM_RETURN_IF_CLOSED(ss);
263
93.7k
  SStream_concat1(ss, '#');
264
93.7k
  printInt64(ss, val);
265
93.7k
}
266
267
void printUInt64Bang(SStream *ss, uint64_t val)
268
13.3k
{
269
13.3k
  assert(ss);
270
13.3k
  SSTREAM_RETURN_IF_CLOSED(ss);
271
13.3k
  SStream_concat1(ss, '#');
272
13.3k
  printUInt64(ss, val);
273
13.3k
}
274
275
// print number
276
void printInt64(SStream *ss, int64_t val)
277
752k
{
278
752k
  assert(ss);
279
752k
  if (ss->unsigned_num) {
280
0
    printUInt64(ss, val);
281
0
    return;
282
0
  }
283
752k
  SSTREAM_RETURN_IF_CLOSED(ss);
284
752k
  if (val >= 0) {
285
686k
    if (val > HEX_THRESHOLD)
286
409k
      SStream_concat(ss, "0x%" PRIx64, val);
287
277k
    else
288
277k
      SStream_concat(ss, "%" PRIu64, val);
289
686k
  } else {
290
65.7k
    if (val < -HEX_THRESHOLD) {
291
61.0k
      if (val == INT64_MIN)
292
74
        SStream_concat(ss, "-0x%" PRIx64,
293
74
                 (uint64_t)INT64_MAX + 1);
294
60.9k
      else
295
60.9k
        SStream_concat(ss, "-0x%" PRIx64,
296
60.9k
                 (uint64_t)-val);
297
61.0k
    } else
298
4.75k
      SStream_concat(ss, "-%" PRIu64, -val);
299
65.7k
  }
300
752k
}
301
302
void printUInt64(SStream *ss, uint64_t val)
303
202k
{
304
202k
  assert(ss);
305
202k
  SSTREAM_RETURN_IF_CLOSED(ss);
306
202k
  if (val > HEX_THRESHOLD)
307
123k
    SStream_concat(ss, "0x%" PRIx64, val);
308
78.7k
  else
309
78.7k
    SStream_concat(ss, "%" PRIu64, val);
310
202k
}
311
312
// print number in decimal mode
313
void printInt32BangDec(SStream *ss, int32_t val)
314
0
{
315
0
  assert(ss);
316
0
  SSTREAM_RETURN_IF_CLOSED(ss);
317
0
  if (val >= 0)
318
0
    SStream_concat(ss, "#%" PRIu32, val);
319
0
  else {
320
0
    if (val == INT32_MIN)
321
0
      SStream_concat(ss, "#-%" PRIu32, val);
322
0
    else
323
0
      SStream_concat(ss, "#-%" PRIu32, (uint32_t)-val);
324
0
  }
325
0
}
326
327
void printInt32Bang(SStream *ss, int32_t val)
328
216k
{
329
216k
  assert(ss);
330
216k
  if (ss->unsigned_num) {
331
0
    printUInt32Bang(ss, val);
332
0
    return;
333
0
  }
334
216k
  SSTREAM_RETURN_IF_CLOSED(ss);
335
216k
  SStream_concat1(ss, '#');
336
216k
  printInt32(ss, val);
337
216k
}
338
339
void printUInt8(SStream *ss, uint8_t val)
340
0
{
341
0
  assert(ss);
342
0
  if (val > HEX_THRESHOLD)
343
0
    SStream_concat(ss, "0x%" PRIx8, val);
344
0
  else
345
0
    SStream_concat(ss, "%" PRIu8, val);
346
0
}
347
348
void printUInt16(SStream *ss, uint16_t val)
349
0
{
350
0
  assert(ss);
351
0
  if (val > HEX_THRESHOLD)
352
0
    SStream_concat(ss, "0x%" PRIx16, val);
353
0
  else
354
0
    SStream_concat(ss, "%" PRIu16, val);
355
0
}
356
357
void printInt8(SStream *ss, int8_t val)
358
3.49k
{
359
3.49k
  assert(ss);
360
3.49k
  if (ss->unsigned_num) {
361
0
    printUInt8(ss, val);
362
0
    return;
363
0
  }
364
3.49k
  SSTREAM_RETURN_IF_CLOSED(ss);
365
3.49k
  if (val >= 0) {
366
2.18k
    if (val > HEX_THRESHOLD)
367
1.55k
      SStream_concat(ss, "0x%" PRIx8, val);
368
630
    else
369
630
      SStream_concat(ss, "%" PRId8, val);
370
2.18k
  } else {
371
1.31k
    if (val < -HEX_THRESHOLD) {
372
1.14k
      if (val == INT8_MIN)
373
765
        SStream_concat(ss, "-0x%" PRIx8,
374
765
                 (uint8_t)INT8_MAX + 1);
375
378
      else
376
378
        SStream_concat(ss, "-0x%" PRIx8, (int8_t)-val);
377
1.14k
    } else
378
170
      SStream_concat(ss, "-%" PRIu8, -val);
379
1.31k
  }
380
3.49k
}
381
382
void printInt16(SStream *ss, int16_t val)
383
3.21k
{
384
3.21k
  assert(ss);
385
3.21k
  if (ss->unsigned_num) {
386
0
    printUInt16(ss, val);
387
0
    return;
388
0
  }
389
3.21k
  SSTREAM_RETURN_IF_CLOSED(ss);
390
3.21k
  if (val >= 0) {
391
2.13k
    if (val > HEX_THRESHOLD)
392
1.39k
      SStream_concat(ss, "0x%" PRIx16, val);
393
741
    else
394
741
      SStream_concat(ss, "%" PRId16, val);
395
2.13k
  } else {
396
1.07k
    if (val < -HEX_THRESHOLD) {
397
728
      if (val == INT16_MIN)
398
10
        SStream_concat(ss, "-0x%" PRIx16,
399
10
                 (uint16_t)INT16_MAX + 1);
400
718
      else
401
718
        SStream_concat(ss, "-0x%" PRIx16,
402
718
                 (int16_t)-val);
403
728
    } else
404
346
      SStream_concat(ss, "-%" PRIu16, -val);
405
1.07k
  }
406
3.21k
}
407
408
void printInt16HexOffset(SStream *ss, int16_t val)
409
6.95k
{
410
6.95k
  assert(ss);
411
6.95k
  if (ss->unsigned_num) {
412
0
    printUInt16(ss, val);
413
0
    return;
414
0
  }
415
6.95k
  SSTREAM_RETURN_IF_CLOSED(ss);
416
6.95k
  if (val >= 0) {
417
4.98k
    SStream_concat(ss, "+0x%" PRIx16, val);
418
4.98k
  } else {
419
1.96k
    if (val == INT16_MIN)
420
19
      SStream_concat(ss, "-0x%" PRIx16,
421
19
               (uint16_t)INT16_MAX + 1);
422
1.94k
    else
423
1.94k
      SStream_concat(ss, "-0x%" PRIx16, (int16_t)-val);
424
1.96k
  }
425
6.95k
}
426
427
void printInt32(SStream *ss, int32_t val)
428
276k
{
429
276k
  assert(ss);
430
276k
  if (ss->unsigned_num) {
431
0
    printUInt32(ss, val);
432
0
    return;
433
0
  }
434
276k
  SSTREAM_RETURN_IF_CLOSED(ss);
435
276k
  if (val >= 0) {
436
209k
    if (val > HEX_THRESHOLD)
437
113k
      SStream_concat(ss, "0x%" PRIx32, val);
438
96.6k
    else
439
96.6k
      SStream_concat(ss, "%" PRId32, val);
440
209k
  } else {
441
66.8k
    if (val < -HEX_THRESHOLD) {
442
63.1k
      if (val == INT32_MIN)
443
76
        SStream_concat(ss, "-0x%" PRIx32,
444
76
                 (uint32_t)INT32_MAX + 1);
445
63.0k
      else
446
63.0k
        SStream_concat(ss, "-0x%" PRIx32,
447
63.0k
                 (int32_t)-val);
448
63.1k
    } else {
449
3.73k
      SStream_concat(ss, "-%" PRIu32, (uint32_t)-val);
450
3.73k
    }
451
66.8k
  }
452
276k
}
453
454
void printInt32HexOffset(SStream *ss, int32_t val)
455
914
{
456
914
  assert(ss);
457
914
  if (ss->unsigned_num) {
458
0
    printUInt32(ss, val);
459
0
    return;
460
0
  }
461
914
  SSTREAM_RETURN_IF_CLOSED(ss);
462
914
  if (val >= 0) {
463
734
    SStream_concat(ss, "+0x%" PRIx32, val);
464
734
  } else {
465
180
    if (val == INT32_MIN)
466
54
      SStream_concat(ss, "-0x%" PRIx32,
467
54
               (uint32_t)INT32_MAX + 1);
468
126
    else
469
126
      SStream_concat(ss, "-0x%" PRIx32, (int32_t)-val);
470
180
  }
471
914
}
472
473
void printInt32Hex(SStream *ss, int32_t val)
474
4.42k
{
475
4.42k
  assert(ss);
476
4.42k
  SSTREAM_RETURN_IF_CLOSED(ss);
477
4.42k
  if (val >= 0) {
478
2.52k
    SStream_concat(ss, "0x%" PRIx32, val);
479
2.52k
  } else {
480
1.90k
    if (val == INT32_MIN)
481
11
      SStream_concat(ss, "-0x%" PRIx32,
482
11
               (uint32_t)INT32_MAX + 1);
483
1.89k
    else
484
1.89k
      SStream_concat(ss, "-0x%" PRIx32, (int32_t)-val);
485
1.90k
  }
486
4.42k
}
487
488
void printUInt32Bang(SStream *ss, uint32_t val)
489
182k
{
490
182k
  assert(ss);
491
182k
  SSTREAM_RETURN_IF_CLOSED(ss);
492
182k
  SStream_concat1(ss, '#');
493
182k
  printUInt32(ss, val);
494
182k
}
495
496
void printUInt32(SStream *ss, uint32_t val)
497
287k
{
498
287k
  assert(ss);
499
287k
  SSTREAM_RETURN_IF_CLOSED(ss);
500
287k
  if (val > HEX_THRESHOLD)
501
213k
    SStream_concat(ss, "0x%x", val);
502
74.2k
  else
503
74.2k
    SStream_concat(ss, "%u", val);
504
287k
}
505
506
void printFloat(SStream *ss, float val)
507
0
{
508
0
  assert(ss);
509
0
  SSTREAM_RETURN_IF_CLOSED(ss);
510
0
  SStream_concat(ss, "%e", val);
511
0
}
512
513
void printFloatBang(SStream *ss, float val)
514
498
{
515
498
  assert(ss);
516
498
  SSTREAM_RETURN_IF_CLOSED(ss);
517
498
  SStream_concat(ss, "#%e", val);
518
498
}
519
520
void printExpr(SStream *ss, uint64_t val)
521
0
{
522
0
  assert(ss);
523
0
  SSTREAM_RETURN_IF_CLOSED(ss);
524
0
  SStream_concat(ss, "%" PRIu64, val);
525
0
}
526
527
SStream *markup_OS(SStream *OS, SStreamMarkup style)
528
682k
{
529
682k
  assert(OS);
530
531
682k
  if (OS->is_closed || !OS->markup_stream) {
532
682k
    return OS;
533
682k
  }
534
0
  OS->markup_stream = false; // Disable temporarily.
535
0
  switch (style) {
536
0
  default:
537
0
    SStream_concat0(OS, "<UNKNOWN:");
538
0
    return OS;
539
0
  case Markup_Immediate:
540
0
    SStream_concat0(OS, "<imm:");
541
0
    break;
542
0
  case Markup_Register:
543
0
    SStream_concat0(OS, "<reg:");
544
0
    break;
545
0
  case Markup_Target:
546
0
    SStream_concat0(OS, "<tar:");
547
0
    break;
548
0
  case Markup_Memory:
549
0
    SStream_concat0(OS, "<mem:");
550
0
    break;
551
0
  }
552
0
  OS->markup_stream = true;
553
  OS->prefixed_by_markup = true;
554
0
  return OS;
555
0
}