String concatination.
This commit is contained in:
parent
01fa5dfdb7
commit
ef77378ab4
2 changed files with 60 additions and 6 deletions
32
src/vm.c
32
src/vm.c
|
|
@ -153,15 +153,35 @@ void vm_run(VM *vm, Bytecode *entry) {
|
|||
case OP_ADD: {
|
||||
Value b = pop_value(vm);
|
||||
Value a = pop_value(vm);
|
||||
if (a.type != VAL_INT || b.type != VAL_INT) {
|
||||
fprintf(stderr, "Runtime type error: ADD expects ints, got %s and %s\n",
|
||||
if (a.type == VAL_INT && b.type == VAL_INT) {
|
||||
Value res = make_int(a.i + b.i);
|
||||
free_value(a);
|
||||
free_value(b);
|
||||
push_value(vm, res);
|
||||
} else if (a.type == VAL_STRING && b.type == VAL_STRING) {
|
||||
const char *sa = a.s ? a.s : "";
|
||||
const char *sb = b.s ? b.s : "";
|
||||
size_t la = strlen(sa);
|
||||
size_t lb = strlen(sb);
|
||||
char *buf = (char*)malloc(la + lb + 1);
|
||||
if (!buf) {
|
||||
fprintf(stderr, "Runtime error: out of memory during string concatenation\n");
|
||||
exit(1);
|
||||
}
|
||||
memcpy(buf, sa, la);
|
||||
memcpy(buf + la, sb, lb);
|
||||
buf[la + lb] = '\0';
|
||||
Value res;
|
||||
res.type = VAL_STRING;
|
||||
res.s = buf;
|
||||
free_value(a);
|
||||
free_value(b);
|
||||
push_value(vm, res);
|
||||
} else {
|
||||
fprintf(stderr, "Runtime type error: ADD expects both ints or both strings, got %s and %s\n",
|
||||
value_type_name(a.type), value_type_name(b.type));
|
||||
exit(1);
|
||||
}
|
||||
Value res = make_int(a.i + b.i);
|
||||
free_value(a);
|
||||
free_value(b);
|
||||
push_value(vm, res);
|
||||
break;
|
||||
}
|
||||
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue