-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathcprintf.h
80 lines (80 loc) · 3.4 KB
/
cprintf.h
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
// SPDX-License-Identifier: MIT
/*
*
* This file is part of cprintf, with ABSOLUTELY NO WARRANTY.
*
* MIT License
*
* Copyright (c) 2024 Moe-hacker
*
* Permission is hereby granted, free of charge, to any person obtaining a copy
* of this software and associated documentation files (the "Software"), to deal
* in the Software without restriction, including without limitation the rights
* to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
* copies of the Software, and to permit persons to whom the Software is
* furnished to do so, subject to the following conditions:
*
* The above copyright notice and this permission notice shall be included in all
* copies or substantial portions of the Software.
*
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
* AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
* LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
* OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
* SOFTWARE.
*
*
*/
#pragma once
#include <stdio.h>
#include <unistd.h>
#include <string.h>
#include <stdlib.h>
#include <stdarg.h>
#include <ctype.h>
#ifndef _Nullable
#define _Nullable
#endif
#ifndef _Nonnull
#define _Nonnull
#endif
void cprintf__(const char *_Nonnull buf);
void cfprintf__(FILE *_Nonnull stream, const char *_Nonnull buf);
// Get the size of the string to print.
// We give another 514 bytes to avoid buffer overflow.
// We call snprintf() twice, but never mind, it's fast enough.
#define cprintf_get_bufsize__(format, ...) (snprintf(NULL, 0, format, ##__VA_ARGS__) > 0 ? (size_t)snprintf(NULL, 0, format, ##__VA_ARGS__) + 514 : 514)
// The `base` color.
extern char *cprintf_base_color;
/*
* cprintf() is a macro,
* first, we get the size of the string to print,
* we use sprintf() to write it to buf[],
* so that we only need to parse the string in buf[].
*/
#define cprintf(format, ...) \
{ \
char *cprintf_buf__ = NULL; \
size_t bufsize = cprintf_get_bufsize__(format, ##__VA_ARGS__); \
if (bufsize != 0) { \
cprintf_buf__ = malloc(bufsize); \
sprintf(cprintf_buf__, format, ##__VA_ARGS__); \
cprintf__(cprintf_buf__); \
free(cprintf_buf__); \
} \
}
#define cfprintf(stream, format, ...) \
{ \
char *cprintf_buf__ = NULL; \
size_t bufsize = cprintf_get_bufsize__(format, ##__VA_ARGS__); \
if (bufsize != 0) { \
cprintf_buf__ = malloc(bufsize); \
sprintf(cprintf_buf__, format, ##__VA_ARGS__); \
cfprintf__(stream, cprintf_buf__); \
free(cprintf_buf__); \
} \
}
#define CPRINTF_MAJOR 1
#define CPRINTF_MINOR 5