Coverage Report

Created: 2025-11-16 06:38

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.64M
{
23
3.64M
  assert(ss);
24
3.64M
  ss->index = 0;
25
3.64M
  memset(ss->buffer, 0, sizeof(ss->buffer));
26
3.64M
  ss->is_closed = false;
27
3.64M
  ss->markup_stream = false;
28
3.64M
  ss->prefixed_by_markup = false;
29
3.64M
  ss->unsigned_num = false;
30
3.64M
}
31
32
void SStream_opt_unum(SStream *ss, bool print_unsigned_numbers)
33
3.52M
{
34
3.52M
  assert(ss);
35
3.52M
  ss->unsigned_num = print_unsigned_numbers;
36
3.52M
}
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
166k
{
42
166k
  assert(ss);
43
166k
  return ss->buffer;
44
166k
}
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
110k
{
53
110k
  assert(ss);
54
110k
  char *found = strchr(ss->buffer, elem);
55
110k
  if (!found || found == ss->buffer + (SSTREAM_BUF_LEN - 1)) {
56
110k
    return NULL;
57
110k
  }
58
0
  *found = repl;
59
0
  found++;
60
0
  return found;
61
110k
}
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
41.0k
{
67
41.0k
  assert(ss && rstr);
68
41.0k
  char *found = strchr(ss->buffer, chr);
69
41.0k
  if (!found || found == ss->buffer + (SSTREAM_BUF_LEN - 1)) {
70
0
    return;
71
0
  }
72
41.0k
  size_t post_len = strlen(found + 1);
73
41.0k
  size_t buf_str_len = strlen(ss->buffer);
74
41.0k
  size_t repl_len = strlen(rstr);
75
41.0k
  if (repl_len - 1 + buf_str_len >= SSTREAM_BUF_LEN) {
76
0
    return;
77
0
  }
78
41.0k
  memmove(found + repl_len, found + 1, post_len);
79
41.0k
  memcpy(found, rstr, repl_len);
80
41.0k
  ss->index = strlen(ss->buffer);
81
41.0k
}
82
83
/// Removes the space characters '\t' and ' ' from the beginning of the stream buffer.
84
void SStream_trimls(SStream *ss)
85
3.63M
{
86
3.63M
  assert(ss);
87
3.63M
  size_t buf_off = 0;
88
  /// Remove leading spaces
89
3.70M
  while (ss->buffer[buf_off] == ' ' || ss->buffer[buf_off] == '\t') {
90
69.4k
    buf_off++;
91
69.4k
  }
92
3.63M
  if (buf_off > 0) {
93
69.4k
    memmove(ss->buffer, ss->buffer + buf_off,
94
69.4k
      SSTREAM_BUF_LEN - buf_off);
95
69.4k
    ss->index -= buf_off;
96
69.4k
  }
97
3.63M
}
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.52M
{
106
3.52M
  assert(ss && mnem_buf && mnem_buf_size > 0 && op_str_buf &&
107
3.52M
         op_str_buf_size > 0);
108
3.52M
  size_t off = 0;
109
  // Copy all non space chars to as mnemonic.
110
20.3M
  while (ss->buffer[off] && ss->buffer[off] != ' ' &&
111
19.5M
         ss->buffer[off] != '\t') {
112
16.8M
    if (off < mnem_buf_size - 1) {
113
      // Only copy if there is space left.
114
16.8M
      mnem_buf[off] = ss->buffer[off];
115
16.8M
    }
116
16.8M
    off++;
117
16.8M
  }
118
3.52M
  if (!ss->buffer[off]) {
119
138k
    return;
120
138k
  }
121
122
  // Iterate until next non space char.
123
3.40M
  do {
124
3.40M
    off++;
125
3.40M
  } while (ss->buffer[off] &&
126
3.34M
     (ss->buffer[off] == ' ' || ss->buffer[off] == '\t'));
127
128
3.38M
  if (!ss->buffer[off]) {
129
58.1k
    return;
130
58.1k
  }
131
132
  // Copy all follow up characters as op_str
133
3.32M
  const char *ss_op_str = ss->buffer + off;
134
3.32M
  off = 0;
135
49.3M
  while (ss_op_str[off] && off < op_str_buf_size - 1) {
136
46.0M
    op_str_buf[off] = ss_op_str[off];
137
46.0M
    off++;
138
46.0M
  }
139
3.32M
}
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
41.6k
{
145
41.6k
  assert(ss);
146
41.6k
  if (file) {
147
0
    fprintf(file, "%s\n", ss->buffer);
148
0
  }
149
41.6k
  SStream_Init(ss);
150
41.6k
}
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
20.9M
{
175
20.9M
#ifndef CAPSTONE_DIET
176
20.9M
  assert(ss && s);
177
20.9M
  SSTREAM_RETURN_IF_CLOSED(ss);
178
20.9M
  if (s[0] == '\0')
179
6.41M
    return;
180
14.5M
  unsigned int len = (unsigned int)strlen(s);
181
182
14.5M
  SSTREAM_OVERFLOW_CHECK(ss, len);
183
184
14.5M
  memcpy(ss->buffer + ss->index, s, len);
185
14.5M
  ss->index += len;
186
14.5M
  ss->buffer[ss->index] = '\0';
187
14.5M
  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
14.5M
}
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.59M
{
203
2.59M
#ifndef CAPSTONE_DIET
204
2.59M
  assert(ss);
205
2.59M
  SSTREAM_RETURN_IF_CLOSED(ss);
206
2.59M
  if (c == '\0')
207
28.2k
    return;
208
209
2.56M
  SSTREAM_OVERFLOW_CHECK(ss, 1);
210
211
2.56M
  ss->buffer[ss->index] = c;
212
2.56M
  ss->index++;
213
2.56M
  ss->buffer[ss->index] = '\0';
214
2.56M
  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.56M
}
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
9.27M
{
229
9.27M
#ifndef CAPSTONE_DIET
230
9.27M
  assert(ss && fmt);
231
9.27M
  SSTREAM_RETURN_IF_CLOSED(ss);
232
9.27M
  va_list ap;
233
9.27M
  int ret;
234
235
9.27M
  va_start(ap, fmt);
236
9.27M
  ret = cs_vsnprintf(ss->buffer + ss->index,
237
9.27M
         sizeof(ss->buffer) - (ss->index + 1), fmt, ap);
238
9.27M
  va_end(ap);
239
9.27M
  ss->index += ret;
240
9.27M
  if (ss->markup_stream && ss->prefixed_by_markup) {
241
0
    SSTREAM_OVERFLOW_CHECK(ss, 1);
242
0
    ss->buffer[ss->index] = '>';
243
0
    ss->index += 1;
244
0
  }
245
#else
246
  ss->buffer[ss->index] = '\0';
247
#endif
248
9.27M
}
249
250
// print number with prefix #
251
void printInt64Bang(SStream *ss, int64_t val)
252
97.6k
{
253
97.6k
  assert(ss);
254
97.6k
  if (ss->unsigned_num) {
255
0
    printUInt64Bang(ss, val);
256
0
    return;
257
0
  }
258
97.6k
  SSTREAM_RETURN_IF_CLOSED(ss);
259
97.6k
  SStream_concat1(ss, '#');
260
97.6k
  printInt64(ss, val);
261
97.6k
}
262
263
void printUInt64Bang(SStream *ss, uint64_t val)
264
17.0k
{
265
17.0k
  assert(ss);
266
17.0k
  SSTREAM_RETURN_IF_CLOSED(ss);
267
17.0k
  SStream_concat1(ss, '#');
268
17.0k
  printUInt64(ss, val);
269
17.0k
}
270
271
// print number
272
void printInt64(SStream *ss, int64_t val)
273
817k
{
274
817k
  assert(ss);
275
817k
  if (ss->unsigned_num) {
276
0
    printUInt64(ss, val);
277
0
    return;
278
0
  }
279
817k
  SSTREAM_RETURN_IF_CLOSED(ss);
280
817k
  if (val >= 0) {
281
749k
    if (val > HEX_THRESHOLD)
282
464k
      SStream_concat(ss, "0x%" PRIx64, val);
283
285k
    else
284
285k
      SStream_concat(ss, "%" PRIu64, val);
285
749k
  } else {
286
68.4k
    if (val < -HEX_THRESHOLD) {
287
62.6k
      if (val == INT64_MIN)
288
66
        SStream_concat(ss, "-0x%" PRIx64,
289
66
                 (uint64_t)INT64_MAX + 1);
290
62.6k
      else
291
62.6k
        SStream_concat(ss, "-0x%" PRIx64,
292
62.6k
                 (uint64_t)-val);
293
62.6k
    } else
294
5.72k
      SStream_concat(ss, "-%" PRIu64, -val);
295
68.4k
  }
296
817k
}
297
298
void printUInt64(SStream *ss, uint64_t val)
299
228k
{
300
228k
  assert(ss);
301
228k
  SSTREAM_RETURN_IF_CLOSED(ss);
302
228k
  if (val > HEX_THRESHOLD)
303
139k
    SStream_concat(ss, "0x%" PRIx64, val);
304
89.2k
  else
305
89.2k
    SStream_concat(ss, "%" PRIu64, val);
306
228k
}
307
308
// print number in decimal mode
309
void printInt32BangDec(SStream *ss, int32_t val)
310
0
{
311
0
  assert(ss);
312
0
  SSTREAM_RETURN_IF_CLOSED(ss);
313
0
  if (val >= 0)
314
0
    SStream_concat(ss, "#%" PRIu32, val);
315
0
  else {
316
0
    if (val == INT32_MIN)
317
0
      SStream_concat(ss, "#-%" PRIu32, val);
318
0
    else
319
0
      SStream_concat(ss, "#-%" PRIu32, (uint32_t)-val);
320
0
  }
321
0
}
322
323
void printInt32Bang(SStream *ss, int32_t val)
324
238k
{
325
238k
  assert(ss);
326
238k
  if (ss->unsigned_num) {
327
0
    printUInt32Bang(ss, val);
328
0
    return;
329
0
  }
330
238k
  SSTREAM_RETURN_IF_CLOSED(ss);
331
238k
  SStream_concat1(ss, '#');
332
238k
  printInt32(ss, val);
333
238k
}
334
335
void printUInt8(SStream *ss, uint8_t val)
336
0
{
337
0
  assert(ss);
338
0
  if (val > HEX_THRESHOLD)
339
0
    SStream_concat(ss, "0x%" PRIx8, val);
340
0
  else
341
0
    SStream_concat(ss, "%" PRIu8, val);
342
0
}
343
344
void printUInt16(SStream *ss, uint16_t val)
345
0
{
346
0
  assert(ss);
347
0
  if (val > HEX_THRESHOLD)
348
0
    SStream_concat(ss, "0x%" PRIx16, val);
349
0
  else
350
0
    SStream_concat(ss, "%" PRIu16, val);
351
0
}
352
353
void printInt8(SStream *ss, int8_t val)
354
3.27k
{
355
3.27k
  assert(ss);
356
3.27k
  if (ss->unsigned_num) {
357
0
    printUInt8(ss, val);
358
0
    return;
359
0
  }
360
3.27k
  SSTREAM_RETURN_IF_CLOSED(ss);
361
3.27k
  if (val >= 0) {
362
2.33k
    if (val > HEX_THRESHOLD)
363
1.98k
      SStream_concat(ss, "0x%" PRIx8, val);
364
352
    else
365
352
      SStream_concat(ss, "%" PRId8, val);
366
2.33k
  } else {
367
937
    if (val < -HEX_THRESHOLD) {
368
449
      if (val == INT8_MIN)
369
86
        SStream_concat(ss, "-0x%" PRIx8,
370
86
                 (uint8_t)INT8_MAX + 1);
371
363
      else
372
363
        SStream_concat(ss, "-0x%" PRIx8, (int8_t)-val);
373
449
    } else
374
488
      SStream_concat(ss, "-%" PRIu8, -val);
375
937
  }
376
3.27k
}
377
378
void printInt16(SStream *ss, int16_t val)
379
3.82k
{
380
3.82k
  assert(ss);
381
3.82k
  if (ss->unsigned_num) {
382
0
    printUInt16(ss, val);
383
0
    return;
384
0
  }
385
3.82k
  SSTREAM_RETURN_IF_CLOSED(ss);
386
3.82k
  if (val >= 0) {
387
2.35k
    if (val > HEX_THRESHOLD)
388
1.81k
      SStream_concat(ss, "0x%" PRIx16, val);
389
541
    else
390
541
      SStream_concat(ss, "%" PRId16, val);
391
2.35k
  } else {
392
1.47k
    if (val < -HEX_THRESHOLD) {
393
1.36k
      if (val == INT16_MIN)
394
102
        SStream_concat(ss, "-0x%" PRIx16,
395
102
                 (uint16_t)INT16_MAX + 1);
396
1.25k
      else
397
1.25k
        SStream_concat(ss, "-0x%" PRIx16,
398
1.25k
                 (int16_t)-val);
399
1.36k
    } else
400
109
      SStream_concat(ss, "-%" PRIu16, -val);
401
1.47k
  }
402
3.82k
}
403
404
void printInt16HexOffset(SStream *ss, int16_t val)
405
7.36k
{
406
7.36k
  assert(ss);
407
7.36k
  if (ss->unsigned_num) {
408
0
    printUInt16(ss, val);
409
0
    return;
410
0
  }
411
7.36k
  SSTREAM_RETURN_IF_CLOSED(ss);
412
7.36k
  if (val >= 0) {
413
5.09k
    SStream_concat(ss, "+0x%" PRIx16, val);
414
5.09k
  } else {
415
2.27k
    if (val == INT16_MIN)
416
56
      SStream_concat(ss, "-0x%" PRIx16,
417
56
               (uint16_t)INT16_MAX + 1);
418
2.21k
    else
419
2.21k
      SStream_concat(ss, "-0x%" PRIx16, (int16_t)-val);
420
2.27k
  }
421
7.36k
}
422
423
void printInt32(SStream *ss, int32_t val)
424
308k
{
425
308k
  assert(ss);
426
308k
  if (ss->unsigned_num) {
427
0
    printUInt32(ss, val);
428
0
    return;
429
0
  }
430
308k
  SSTREAM_RETURN_IF_CLOSED(ss);
431
308k
  if (val >= 0) {
432
228k
    if (val > HEX_THRESHOLD)
433
132k
      SStream_concat(ss, "0x%" PRIx32, val);
434
96.4k
    else
435
96.4k
      SStream_concat(ss, "%" PRId32, val);
436
228k
  } else {
437
79.4k
    if (val < -HEX_THRESHOLD) {
438
73.7k
      if (val == INT32_MIN)
439
134
        SStream_concat(ss, "-0x%" PRIx32,
440
134
                 (uint32_t)INT32_MAX + 1);
441
73.6k
      else
442
73.6k
        SStream_concat(ss, "-0x%" PRIx32,
443
73.6k
                 (int32_t)-val);
444
73.7k
    } else {
445
5.71k
      SStream_concat(ss, "-%" PRIu32, (uint32_t)-val);
446
5.71k
    }
447
79.4k
  }
448
308k
}
449
450
void printInt32HexOffset(SStream *ss, int32_t val)
451
1.56k
{
452
1.56k
  assert(ss);
453
1.56k
  if (ss->unsigned_num) {
454
0
    printUInt32(ss, val);
455
0
    return;
456
0
  }
457
1.56k
  SSTREAM_RETURN_IF_CLOSED(ss);
458
1.56k
  if (val >= 0) {
459
1.35k
    SStream_concat(ss, "+0x%" PRIx32, val);
460
1.35k
  } else {
461
218
    if (val == INT32_MIN)
462
18
      SStream_concat(ss, "-0x%" PRIx32,
463
18
               (uint32_t)INT32_MAX + 1);
464
200
    else
465
200
      SStream_concat(ss, "-0x%" PRIx32, (int32_t)-val);
466
218
  }
467
1.56k
}
468
469
void printInt32Hex(SStream *ss, int32_t val)
470
4.93k
{
471
4.93k
  assert(ss);
472
4.93k
  SSTREAM_RETURN_IF_CLOSED(ss);
473
4.93k
  if (val >= 0) {
474
2.67k
    SStream_concat(ss, "0x%" PRIx32, val);
475
2.67k
  } else {
476
2.25k
    if (val == INT32_MIN)
477
67
      SStream_concat(ss, "-0x%" PRIx32,
478
67
               (uint32_t)INT32_MAX + 1);
479
2.19k
    else
480
2.19k
      SStream_concat(ss, "-0x%" PRIx32, (int32_t)-val);
481
2.25k
  }
482
4.93k
}
483
484
void printUInt32Bang(SStream *ss, uint32_t val)
485
214k
{
486
214k
  assert(ss);
487
214k
  SSTREAM_RETURN_IF_CLOSED(ss);
488
214k
  SStream_concat1(ss, '#');
489
214k
  printUInt32(ss, val);
490
214k
}
491
492
void printUInt32(SStream *ss, uint32_t val)
493
328k
{
494
328k
  assert(ss);
495
328k
  SSTREAM_RETURN_IF_CLOSED(ss);
496
328k
  if (val > HEX_THRESHOLD)
497
248k
    SStream_concat(ss, "0x%x", val);
498
79.4k
  else
499
79.4k
    SStream_concat(ss, "%u", val);
500
328k
}
501
502
void printFloat(SStream *ss, float val)
503
0
{
504
0
  assert(ss);
505
0
  SSTREAM_RETURN_IF_CLOSED(ss);
506
0
  SStream_concat(ss, "%e", val);
507
0
}
508
509
void printFloatBang(SStream *ss, float val)
510
646
{
511
646
  assert(ss);
512
646
  SSTREAM_RETURN_IF_CLOSED(ss);
513
646
  SStream_concat(ss, "#%e", val);
514
646
}
515
516
void printExpr(SStream *ss, uint64_t val)
517
0
{
518
0
  assert(ss);
519
0
  SSTREAM_RETURN_IF_CLOSED(ss);
520
0
  SStream_concat(ss, "%" PRIu64, val);
521
0
}
522
523
SStream *markup_OS(SStream *OS, SStreamMarkup style)
524
869k
{
525
869k
  assert(OS);
526
527
869k
  if (OS->is_closed || !OS->markup_stream) {
528
869k
    return OS;
529
869k
  }
530
0
  OS->markup_stream = false; // Disable temporarily.
531
0
  switch (style) {
532
0
  default:
533
0
    SStream_concat0(OS, "<UNKNOWN:");
534
0
    return OS;
535
0
  case Markup_Immediate:
536
0
    SStream_concat0(OS, "<imm:");
537
0
    break;
538
0
  case Markup_Register:
539
0
    SStream_concat0(OS, "<reg:");
540
0
    break;
541
0
  case Markup_Target:
542
0
    SStream_concat0(OS, "<tar:");
543
0
    break;
544
0
  case Markup_Memory:
545
0
    SStream_concat0(OS, "<mem:");
546
0
    break;
547
0
  }
548
0
  OS->markup_stream = true;
549
  OS->prefixed_by_markup = true;
550
0
  return OS;
551
0
}