/*
* Read a declaration up to a semicolon or an open curly brace and add it to <declaration>.
*/
static int handle_declaration(FILE *fp, Buffer *declaration)
{
int c;
while ((c = fgetc(fp)) != EOF) {
if (c == ';') {
bufAddC(declaration, c);
break;
}
else if (c == '{') {
Buffer body = { 0 };
handle_compound(fp, &body);
bufReset(&body);
break;
}
else if (c == '/') {
bufAddC(declaration, c);
handle_comment(fp, declaration);
}
else if (c == '"' || c == '\'') {
handle_string(fp, declaration, c);
}
else {
bufAddC(declaration, c);
}
}
return 0;
}
/*
* Read a string and add it to <buf>. The string should be terminated by <terminator>.
*/
static int handle_string(FILE *fp, Buffer *buffer, const char terminator)
{
int c;
while ((c = fgetc(fp)) != EOF && c != terminator) {
bufAddC(buffer, c);
/* Any escape sequence (particularly escaped quotes) should be entered as-is. */
if (c == '\\') {
c = fgetc(fp);
bufAddC(buffer, c);
}
}
return 0;
}
/*
* Read a comment and add it to <buffer>, which already contains the first slash.
*/
static int handle_comment(FILE *fp, Buffer *buffer)
{
int c = fgetc(fp);
if (c == '*') {
bufAddC(buffer, c);
return handle_block_comment(fp, buffer);
}
else if (c == '/') {
bufAddC(buffer, c);
return handle_line_comment(fp, buffer);
}
else {
ungetc(c, fp);
return 0;
}
}
/*
* Read a block comment and add it to <buffer>. The starting slash-star are already in <buf>.
*/
static int handle_block_comment(FILE *fp, Buffer *buffer)
{
int c;
while ((c = fgetc(fp)) != EOF) {
bufAddC(buffer, c);
if (c == '*') {
c = fgetc(fp);
bufAddC(buffer, c);
if (c == '/') break;
}
}
return 0;
}
/*
* Read a line comment and add it to <buf>. The leading slash-slash are already in <buf>.
*/
static int handle_line_comment(FILE *fp, Buffer *buf)
{
int c;
while ((c = fgetc(fp)) != EOF && c != '\n') {
bufAddC(buf, c);
}
return 0;
}
/*
* Read a compound statement and add it to <buffer>.
*/
static int handle_compound(FILE *fp, Buffer *buffer)
{
int c;
while ((c = fgetc(fp)) != EOF) {
bufAddC(buffer, c);
if (c == '}')
break;
else if (c == '"' || c == '\'')
handle_string(fp, buffer, c);
else if (c == '{')
handle_compound(fp, buffer);
else if (c == '/') {
bufAddC(buffer, c);
handle_comment(fp, buffer);
}
}
return 0;
}
int main(int argc, char *argv[])
{
Buffer buf1 = { 0 };
Buffer buf2 = { 0 };
Buffer *buf3;
bufClear(&buf1);
make_sure_that(bufLen(&buf1) == 0);
make_sure_that(bufIsEmpty(&buf1));
bufAdd(&buf1, "ABCDEF", 3);
make_sure_that(bufLen(&buf1) == 3);
make_sure_that(!bufIsEmpty(&buf1));
make_sure_that(strcmp(bufGet(&buf1), "ABC") == 0);
bufAddC(&buf1, 'D');
make_sure_that(bufLen(&buf1) == 4);
make_sure_that(strcmp(bufGet(&buf1), "ABCD") == 0);
bufAddF(&buf1, "%d", 1234);
make_sure_that(bufLen(&buf1) == 8);
make_sure_that(strcmp(bufGet(&buf1), "ABCD1234") == 0);
bufAddS(&buf1, "XYZ");
make_sure_that(bufLen(&buf1) == 11);
make_sure_that(strcmp(bufGet(&buf1), "ABCD1234XYZ") == 0);
bufSet(&buf1, "ABCDEF", 3);
make_sure_that(bufLen(&buf1) == 3);
make_sure_that(strcmp(bufGet(&buf1), "ABC") == 0);
bufSetC(&buf1, 'D');
make_sure_that(bufLen(&buf1) == 1);
make_sure_that(strcmp(bufGet(&buf1), "D") == 0);
bufSetF(&buf1, "%d", 1234);
make_sure_that(bufLen(&buf1) == 4);
make_sure_that(strcmp(bufGet(&buf1), "1234") == 0);
bufSetS(&buf1, "ABCDEF");
make_sure_that(bufLen(&buf1) == 6);
make_sure_that(strcmp(bufGet(&buf1), "ABCDEF") == 0);
bufClear(&buf1);
make_sure_that(bufLen(&buf1) == 0);
make_sure_that(strcmp(bufGet(&buf1), "") == 0);
bufSet(&buf1, "ABC", 3);
bufSet(&buf2, "DEF", 3);
buf3 = bufCat(&buf1, &buf2);
make_sure_that(&buf1 == buf3);
make_sure_that(bufLen(&buf1) == 6);
make_sure_that(strcmp(bufGet(&buf1), "ABCDEF") == 0);
make_sure_that(bufLen(&buf2) == 3);
make_sure_that(strcmp(bufGet(&buf2), "DEF") == 0);
bufSetF(&buf1, "ABCDEF");
make_sure_that(strcmp(bufGet(bufTrim(&buf1, 0, 0)), "ABCDEF") == 0);
make_sure_that(strcmp(bufGet(bufTrim(&buf1, 1, 0)), "BCDEF") == 0);
make_sure_that(strcmp(bufGet(bufTrim(&buf1, 0, 1)), "BCDE") == 0);
make_sure_that(strcmp(bufGet(bufTrim(&buf1, 1, 1)), "CD") == 0);
make_sure_that(strcmp(bufGet(bufTrim(&buf1, 3, 3)), "") == 0);
bufPack(&buf1,
PACK_INT8, 0x01,
PACK_INT16, 0x0123,
PACK_INT32, 0x01234567,
PACK_INT64, 0x0123456789ABCDEF,
PACK_FLOAT, 0.0,
PACK_DOUBLE, 0.0,
PACK_STRING, "Hoi1",
PACK_DATA, "Hoi2", 4,
PACK_RAW, "Hoi3", 4,
END);
make_sure_that(bufLen(&buf1) == 47);
make_sure_that(memcmp(bufGet(&buf1),
"\x01"
"\x01\x23"
"\x01\x23\x45\x67"
"\x01\x23\x45\x67\x89\xAB\xCD\xEF"
"\x00\x00\x00\x00"
"\x00\x00\x00\x00\x00\x00\x00\x00"
"\x00\x00\x00\x04Hoi1"
"\x00\x00\x00\x04Hoi2"
"Hoi3", 47) == 0);
{
uint8_t u8;
uint16_t u16;
uint32_t u32;
uint64_t u64;
float f32;
double f64;
char *string, *data, raw[4];
int data_size;
bufUnpack(&buf1,
PACK_INT8, &u8,
PACK_INT16, &u16,
PACK_INT32, &u32,
PACK_INT64, &u64,
PACK_FLOAT, &f32,
PACK_DOUBLE, &f64,
PACK_STRING, &string,
PACK_DATA, &data, &data_size,
PACK_RAW, &raw, sizeof(raw),
END);
make_sure_that(u8 == 0x01);
make_sure_that(u16 == 0x0123);
make_sure_that(u32 == 0x01234567);
make_sure_that(u64 == 0x0123456789ABCDEF);
make_sure_that(f32 == 0.0);
make_sure_that(f64 == 0.0);
make_sure_that(memcmp(string, "Hoi1", 5) == 0);
make_sure_that(memcmp(data, "Hoi2", 4) == 0);
make_sure_that(data_size == 4);
make_sure_that(memcmp(raw, "Hoi3", 4) == 0);
}
{
bufUnpack(&buf1,
PACK_INT8, NULL,
PACK_INT16, NULL,
PACK_INT32, NULL,
PACK_INT64, NULL,
PACK_FLOAT, NULL,
PACK_DOUBLE, NULL,
PACK_STRING, NULL,
PACK_DATA, NULL, NULL,
PACK_RAW, NULL, 4,
END);
}
{
const char *name[] = { "Mills", "Berry", "Buck", "Stipe" };
bufClear(&buf1);
bufList(&buf1, ", ", " and ", TRUE, TRUE, "%s", name[0]);
make_sure_that(strcmp(bufGet(&buf1), "Mills") == 0);
bufClear(&buf1);
bufList(&buf1, ", ", " and ", TRUE, FALSE, "%s", name[0]);
bufList(&buf1, ", ", " and ", FALSE, TRUE, "%s", name[1]);
make_sure_that(strcmp(bufGet(&buf1), "Mills and Berry") == 0);
bufClear(&buf1);
bufList(&buf1, ", ", " and ", TRUE, FALSE, "%s", name[0]);
bufList(&buf1, ", ", " and ", FALSE, FALSE, "%s", name[1]);
bufList(&buf1, ", ", " and ", FALSE, TRUE, "%s", name[2]);
make_sure_that(strcmp(bufGet(&buf1), "Mills, Berry and Buck") == 0);
bufClear(&buf1);
bufList(&buf1, ", ", " and ", TRUE, FALSE, "%s", name[0]);
bufList(&buf1, ", ", " and ", FALSE, FALSE, "%s", name[1]);
bufList(&buf1, ", ", " and ", FALSE, FALSE, "%s", name[2]);
bufList(&buf1, ", ", " and ", FALSE, TRUE, "%s", name[3]);
make_sure_that(strcmp(bufGet(&buf1), "Mills, Berry, Buck and Stipe") == 0);
}
bufReset(&buf1);
bufReset(&buf2);
return errors;
}
int main(int argc, char *argv[])
{
FILE *in_fp = NULL, *out_fp = NULL;
int i, r;
char *input = NULL, *output = NULL;
Buffer cmd = { 0 };
bufSetS(&cmd, "cpp -C");
for (i = 1; i < argc; i++) {
if (strcmp(argv[i], "-") == 0) {
input = "-";
}
else if (strcmp(argv[i], "-p") == 0 || strcmp(argv[i], "--cpp") == 0) {
use_cpp = TRUE;
}
else if (strcmp(argv[i], "-o") == 0 || strcmp(argv[i], "--output") == 0) {
output = argv[++i];
}
else if (strcmp(argv[i], "-c") == 0 || strcmp(argv[i], "--comments") == 0) {
include_comment = TRUE;
}
else if (strcmp(argv[i], "-s") == 0 || strcmp(argv[i], "--statics") == 0) {
include_static = TRUE;
}
else if (strcmp(argv[i], "-h") == 0 || strcmp(argv[i], "--help") == 0) {
usage(NULL, argv[0], 0);
}
else if (argv[i][0] == '-') {
bufAddC(&cmd, ' ');
bufAddS(&cmd, argv[i]);
}
else if (input != NULL) {
usage("Multiple input files specified\n", argv[0], -1);
}
else {
input = argv[i];
}
}
if (use_cpp) {
if (input != NULL && strcmp(input, "-") != 0) {
bufAddC(&cmd, ' ');
bufAddS(&cmd, input);
}
else {
input = "<stdin>";
}
if ((in_fp = popen(bufGet(&cmd), "r")) == NULL) {
perror("popen");
exit(1);
}
}
else {
if (input == NULL || strcmp(input, "-") == 0) {
in_fp = stdin;
input = "<stdin>";
}
else if ((in_fp = fopen(input, "r")) == NULL) {
perror(input);
exit(1);
}
}
bufReset(&cmd);
if (output == NULL) {
out_fp = stdout;
}
else if ((out_fp = fopen(output, "w")) == NULL) {
perror(output);
exit(1);
}
r = process(input, in_fp, out_fp);
if (out_fp != stdout)
fclose(out_fp);
if (use_cpp)
pclose(in_fp);
else if (in_fp != stdin)
fclose(in_fp);
return r;
}
