sandbox/bitwise/ion/ion.c

230 lines
3.6 KiB
C
Raw Normal View History

#include <assert.h>
#include <ctype.h>
#include <stdbool.h>
2018-03-23 22:25:46 +00:00
#include <stddef.h>
2018-03-27 22:59:34 +00:00
#include <stdint.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
2018-03-27 22:59:34 +00:00
#ifndef NDEBUG
#define tprint(...) do { fprintf( stderr, __VA_ARGS__ ); } while( false )
#else
#define tprint(x) {}
#endif
2018-03-23 22:25:46 +00:00
#define MAX(a, b) ((a) < (b) ? (b) : (a))
// stretchy buffers
typedef struct _BufHeader {
size_t len;
size_t cap;
char buf[0]; // C99
} BufHeader;
2018-03-23 22:25:46 +00:00
void *buf__grow(const void *, size_t, size_t);
#define buf__hdr(b) ((BufHeader *)((char *)(b) - offsetof(BufHeader, buf)))
#define buf__fits(b, n) ((buf_len(b) + n) <= buf_cap(b))
#define buf__fit(b, n) (buf__fits(b, n) ? 0 : ((b) = (buf__grow((b), buf_len(b) + (n), sizeof(*(b))))))
#define buf_len(b) ((b) ? buf__hdr(b)->len : 0)
#define buf_cap(b) ((b) ? buf__hdr(b)->cap : 0)
2018-03-23 22:25:46 +00:00
#define buf_push(b, x) (buf__fit((b), 1), (b)[buf_len((b))] = (x), buf__hdr((b))->len++)
#define buf_free(b) ((b) ? free(buf__hdr((b))) : 0)
2018-03-27 22:59:34 +00:00
#define prlu(m, v) (tprint("%s: %lu\n", m, (unsigned long)(v)))
void *
buf__grow(const void *buf, size_t nlen, size_t esize)
{
2018-03-23 22:25:46 +00:00
size_t ncap = MAX(1 + 2*buf_cap(buf), nlen);
assert(nlen <= ncap);
2018-03-23 22:25:46 +00:00
size_t nsize = offsetof(BufHeader, buf) + ncap * esize;
BufHeader *nhdr;
if (buf) {
2018-03-23 22:25:46 +00:00
nhdr = realloc(buf__hdr(buf), nsize);
}
else {
2018-03-23 22:25:46 +00:00
nhdr = malloc(nsize);
nhdr->len = 0;
}
2018-03-23 22:25:46 +00:00
nhdr->cap = ncap;
return nhdr->buf;
}
2018-03-23 22:25:46 +00:00
static void
buf_test(void)
{
2018-03-23 22:25:46 +00:00
int *stretchy = NULL;
2018-03-27 22:59:34 +00:00
tprint("buf_test\n");
2018-03-23 22:25:46 +00:00
for (int i = 0; i < 1024; i++) {
buf_push(stretchy, i);
}
2018-03-23 22:25:46 +00:00
for (int i = 0; i < buf_len(stretchy); i++) {
assert(stretchy[i] == i);
}
2018-03-23 22:25:46 +00:00
buf_free(stretchy);
2018-03-27 22:59:34 +00:00
tprint("OK\n");
2018-03-23 22:25:46 +00:00
}
typedef enum TokenKind {
TOKEN_INT = 128,
2018-03-23 22:25:46 +00:00
TOKEN_NAME,
// ...
} TokenKind;
typedef struct Token {
TokenKind kind;
2018-03-27 22:59:34 +00:00
union {
uint64_t val;
struct {
const char *start;
const char *end;
};
};
2018-03-23 22:25:46 +00:00
// ...
} Token;
const char *stream;
2018-03-23 22:25:46 +00:00
Token token;
void next_token(void) {
switch (*stream) {
2018-03-27 22:59:34 +00:00
/* Numeric literal */
2018-03-23 22:25:46 +00:00
case '0':
case '1':
case '2':
case '3':
case '4':
case '5':
case '6':
case '7':
case '8':
2018-03-27 22:59:34 +00:00
case '9': {
uint64_t val = 0; /* reset the token's value */
2018-03-23 22:25:46 +00:00
while (isdigit(*stream)) {
2018-03-27 22:59:34 +00:00
val *= 10;
val += (*stream - '0');
2018-03-23 22:25:46 +00:00
stream++;
}
token.kind = TOKEN_INT;
2018-03-27 22:59:34 +00:00
token.val = val;
2018-03-23 22:25:46 +00:00
break;
2018-03-27 22:59:34 +00:00
}
/* Identifier */
case 'a':
case 'b':
case 'c':
case 'd':
case 'e':
case 'f':
case 'g':
case 'h':
case 'i':
case 'j':
case 'k':
case 'l':
case 'm':
case 'n':
case 'o':
case 'p':
case 'q':
case 'r':
case 's':
case 't':
case 'u':
case 'v':
case 'w':
case 'x':
case 'y':
case 'z':
case 'A':
case 'B':
case 'C':
case 'D':
case 'E':
case 'F':
case 'G':
case 'H':
case 'I':
case 'J':
case 'K':
case 'L':
case 'M':
case 'N':
case 'O':
case 'P':
case 'Q':
case 'R':
case 'S':
case 'T':
case 'U':
case 'V':
case 'W':
case 'X':
case 'Y':
case 'Z':
case '_': {
const char *start = stream++;
while (isalnum(*stream) || *stream == '_') {
stream++;
}
token.kind = TOKEN_NAME;
token.start = start;
token.end = stream;
break;
}
2018-03-23 22:25:46 +00:00
default:
token.kind = *stream++;
2018-03-23 22:25:46 +00:00
break;
}
2018-03-23 22:25:46 +00:00
}
2018-03-27 22:59:34 +00:00
void
print_token(void)
{
tprint("TOKEN: %d", token.kind);
switch (token.kind) {
case TOKEN_INT:
tprint(" VALUE: %lu", token.val);
break;
case TOKEN_NAME:
break;
default:
break;
}
}
void
lex_test(void)
{
char source[] = "+()123456+994";
2018-03-27 22:59:34 +00:00
tprint("lex_test\n");
stream = source;
next_token();
while (token.kind) {
2018-03-27 22:59:34 +00:00
tprint("TOKEN: %d\n", token.kind);
switch (token.kind) {
case TOKEN_INT:
tprint("\tvalue: %lu\n", (long unsigned)token.val);
break;
default:
break;
}
next_token();
}
2018-03-27 22:59:34 +00:00
tprint("OK\n");
}
2018-03-23 22:25:46 +00:00
int
main(int argc, char *argv[])
{
buf_test();
lex_test();
return 0;
}