2 * This program is free software: you can redistribute it and/or modify
3 * it under the terms of the GNU General Public License as published by
4 * the Free Software Foundation, either version 3 of the License, or
5 * (at your option) any later version.
7 * This program is distributed in the hope that it will be useful,
8 * but WITHOUT ANY WARRANTY; without even the implied warranty of
9 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
10 * GNU General Public License for more details.
12 * You should have received a copy of the GNU General Public License
13 * along with this program. If not, see <http://www.gnu.org/licenses/>.
29 // should be enough for everyone!
30 #define MAX_INST_COUNT (1000000)
35 //extern void fatal (const char *fmt, ...) __attribute__((__noreturn__)) __attribute__((format(printf, 1, 2)));
36 static __attribute__((__noreturn__
)) __attribute__((format(printf
, 1, 2))) void fatalis (const char *fmt
, ...) {
39 fprintf(stderr
, "AWISH FATAL: ");
41 vfprintf(stderr
, fmt
, ap
);
43 fprintf(stderr
, "\n");
48 #ifndef VM_FATAL_DEFINED
49 static __attribute__((__noreturn__
)) __attribute__((format(printf
, 3, 4))) void vmfatal (int tid
, int pc
, const char *fmt
, ...) {
52 fprintf(stderr
, "VM FATAL (thread #%d, pc=%04x): ", tid
, (unsigned int)pc
);
54 vfprintf(stderr
, fmt
, ap
);
56 fprintf(stderr
, "\n");
62 const char *vmOpNames
[] = {
119 VMRSTCB vmRSTCB
= NULL
;
120 VMMapGetCB vmMapGetCB
= NULL
;
121 VMMapSetCB vmMapSetCB
= NULL
;
128 int *stack
; // internal
129 int *tvars
; // internal
133 static VMThread vmThreads
[VM_MAX_THREADS
];
134 static int vmExecuted
[VM_MAX_THREADS
];
135 static int vmLastThreadId
= 0;
137 unsigned char vmCode
[65536];
139 int vmGVars
[VM_VARS_SIZE
];
142 static void fixLastThreadId (void) {
143 for (; vmLastThreadId
> 0; --vmLastThreadId
) if (vmThreads
[vmLastThreadId
].stack
) break;
147 static int vmFindFreeThread (void) {
148 for (int f
= 0; f
< VM_MAX_THREADS
; ++f
) if (vmThreads
[f
].stack
== NULL
) return f
;
153 static int vmInitThread (VMThread
*trd
) {
157 trd
->stack
= calloc(VM_STACK_SIZE
, sizeof(int));
158 if (trd
->stack
== NULL
) return -1;
159 trd
->tvars
= calloc(VM_VARS_SIZE
, sizeof(int));
160 if (trd
->tvars
== NULL
) { free(trd
->stack
); return -1; }
165 static void vmFreeThread (VMThread
*trd
) {
166 if (trd
->stack
) free(trd
->stack
);
167 if (trd
->tvars
) free(trd
->tvars
);
176 int vmInitialize (void) {
178 memset(vmThreads
, 0, sizeof(vmThreads
));
181 if (vmInitThread(vmThreads
)) return -1; // alas
186 void vmDeinitialize (void) {
187 for (int f
= 0; f
< VM_MAX_THREADS
; ++f
) {
188 if (vmThreads
[f
].stack
) free(vmThreads
[f
].stack
);
189 if (vmThreads
[f
].tvars
) free(vmThreads
[f
].tvars
);
191 memset(vmThreads
, 0, sizeof(vmThreads
));
197 static inline int vmGetByte (int tid
, int opc
, VMThread
*trd
) {
198 if (trd
->pc
< 0 || trd
->pc
>= vmCodeSize
) vmfatal(tid
, opc
, "out of code");
199 return vmCode
[trd
->pc
++];
203 #define STACK_WANT(n) do { if (trd->sp < (n)) vmfatal(tid, opc, "stack underflow"); } while (0)
204 #define STACK_FSPC(n) do { if (trd->sp+(n) > VM_STACK_SIZE) vmfatal(tid, opc, "stack overflow"); } while (0)
206 #define STACK_TOP() (trd->stack[trd->sp-1])
207 #define STACK_POP() (trd->stack[--(trd->sp)])
208 #define STACK_PUSH(n) trd->stack[(trd->sp)++] = (n)
211 #define MATH(op) do { \
215 trd->stack[trd->sp-2] = (trd->stack[trd->sp-2]) op (trd->stack[trd->sp-1]); \
220 trd->stack[trd->sp-1] = (trd->stack[trd->sp-1]) op (argv[0]); \
222 case 2: if (argp[0] != NULL) *(argp[0]) = (argv[0]) op (argv[1]); break; \
223 case 3: if (argp[2] != NULL) *(argp[2]) = (argv[0]) op (argv[1]); break; \
228 #define CHECK_DIV_ZERO \
231 if (trd->sp > 0 && trd->stack[trd->sp-1] == 0) { fprintf(stderr, "VM FATAL: division by zero!\n"); exit(1); } \
234 if (argv[0] == 0) { fprintf(stderr, "VM FATAL: division by zero!\n"); exit(1); } \
237 if (argv[1] == 0) { fprintf(stderr, "VM FATAL: division by zero!\n"); exit(1); } \
242 #define JXX(op) do { \
246 argv[0] = STACK_POP(); \
249 argv[2] = STACK_POP(); \
250 argv[1] = STACK_POP(); \
255 argv[1] = STACK_POP(); \
258 if (argv[1] op argv[2]) trd->pc = argv[0]; \
262 static inline int isBranch (int opcode
) {
276 int vmExecuteOne (int tid
) {
277 VMThread
*trd
= &vmThreads
[tid
];
280 int argv
[3]; // argument values
281 int *argp
[3]; // pointer to vars for each arg
284 // decode instruction
285 opcode
= vmGetByte(tid
, opc
, trd
);
286 argc
= (opcode
>>6)&0x03;
288 argv
[0] = argv
[1] = argv
[2] = 0;
289 argp
[0] = argp
[1] = argp
[2] = NULL
;
291 if (tid
> vmLastThreadId
) vmLastThreadId
= tid
;
292 for (int f
= 0; f
< argc
; ++f
) {
293 int vn
= vmGetByte(tid
, opc
, trd
);
297 argv
[f
] = vmGetByte(tid
, opc
, trd
);
298 argv
[f
] |= vmGetByte(tid
, opc
, trd
)<<8;
299 if (argv
[f
] >= 32768 && (f
!= 0 || !isBranch(opcode
))) argv
[f
] -= 65536;
302 // stack var; <0: from stack top
303 argv
[f
] = vmGetByte(tid
, opc
, trd
);
304 argv
[f
] |= (vmGetByte(tid
, opc
, trd
)<<8);
305 if (argv
[f
] >= 32768) argv
[f
] -= 65536;
306 if (argv
[f
] < 0) argv
[f
] += trd
->sp
;
307 if (argv
[f
] < 0 || argv
[f
] >= VM_STACK_SIZE
) vmfatal(tid
, opc
, "invalid stack offset");
308 argp
[f
] = trd
->stack
+argv
[f
];
311 argp
[f
] = vn
&0x80 ? vmGVars
: trd
->tvars
;
314 argv
[f
] = *(argp
[f
]);
319 fprintf(stderr
, "%04x: ", (unsigned int)opc
);
320 if (opcode
< sizeof(vmOpNames
)/sizeof(char *)-1) fprintf(stderr
, "%s", vmOpNames
[opcode
]); else fprintf(stderr
, "BAD");
321 for (int f
= 0; f
< argc
; ++f
) {
327 if (argp
[f
] >= trd
->tvars
&& argp
[f
] < trd
->tvars
+VM_VARS_SIZE
) {
328 ofs
= (unsigned int)(argp
[f
]-trd
->tvars
);
329 } else if (argp
[f
] >= trd
->stack
&& argp
[f
] < trd
->stack
+VM_VARS_SIZE
) {
330 ofs
= (unsigned int)(argp
[f
]-trd
->stack
);
332 } else if (argp
[f
] >= vmGVars
&& argp
[f
] < vmGVars
+VM_VARS_SIZE
) {
333 ofs
= (unsigned int)(argp
[f
]-vmGVars
);
336 fprintf(stderr
, "%u]", ofs
);
338 if (f
== 0 && isBranch(opcode
)) {
339 fprintf(stderr
, "(0x%04x)", argv
[f
]);
341 fprintf(stderr
, "(%d)", argv
[f
]);
348 case VM_ADD
: MATH(+); break;
349 case VM_SUB
: MATH(-); break;
350 case VM_MUL
: MATH(*); break;
359 case VM_BOR
: MATH(|); break;
360 case VM_XOR
: MATH(^); break;
361 case VM_AND
: MATH(&); break;
363 case VM_JEQ
: JXX(==); break;
364 case VM_JNE
: JXX(!=); break;
365 case VM_JLT
: JXX(<); break;
366 case VM_JLE
: JXX(<=); break;
367 case VM_JGT
: JXX(>); break;
368 case VM_JGE
: JXX(>=); break;
373 argv
[0] = STACK_POP();
382 argv
[0] = STACK_POP();
401 case VM_END
: return 1;
402 case VM_BRK
: return -1;
404 case VM_NEW
: { // new thread
405 int xtid
= vmFindFreeThread();
409 argv
[0] = STACK_POP();
412 if (vmInitThread(&vmThreads
[xtid
]) == 0) {
413 vmThreads
[xtid
].pc
= argv
[0];
418 if (xtid
< 0) vmfatal(tid
, opc
, "too many threads");
419 if (xtid
> vmLastThreadId
) vmLastThreadId
= xtid
;
421 if (argp
[1]) *(argp
[1]) = xtid
;
432 argv
[0] = STACK_POP(); // varid
433 argv
[1] = STACK_POP(); // value
436 if (argv
[0] < VM_VARS_SIZE
) vmGVars
[argv
[0]] = argv
[1];
437 } else if (argv
[0] < VM_VARS_SIZE
) {
438 trd
->tvars
[argv
[0]] = argv
[1];
443 argv
[1] = STACK_POP(); // value
446 if (argp
[0]) *(argp
[0]) = argv
[1];
449 int xtid
= argv
[2], vid
= argv
[0];
451 if (xtid
< 0 || xtid
>= VM_MAX_THREADS
|| vmThreads
[xtid
].stack
== NULL
|| vid
< 0 || vid
>= VM_VARS_SIZE
) break;
452 vmThreads
[xtid
].tvars
[vid
] = argv
[1];
458 int xtid
, vid
, pushit
;
484 if (xtid
>= 0 && xtid
< VM_MAX_THREADS
&& vmThreads
[xtid
].stack
!= NULL
&& vid
>= 0 && vid
< VM_VARS_SIZE
) {
485 argv
[0] = vmThreads
[xtid
].tvars
[vid
];
490 if (argp
[2]) *(argp
[2]) = argv
[0];
500 for (int f
= 0; f
< argc
; ++f
) STACK_PUSH(argv
[f
]);
505 trd
->stack
[trd
->sp
] = trd
->stack
[trd
->sp
-1];
516 if (argp
[0] == NULL
&& argv
[0] < 0) {
517 // allocate stack space
520 for (; argc
> 0; --argc
) STACK_PUSH(0);
525 for (int f
= 0; f
< argc
; ++f
) {
529 *(argp
[f
]) = STACK_POP();
530 } else if (argv
[f
] > 0) {
544 argc
= trd
->stack
[trd
->sp
-2];
545 trd
->stack
[trd
->sp
-2] = trd
->stack
[trd
->sp
-1];
546 trd
->stack
[trd
->sp
-1] = argc
;
550 argc
= trd
->stack
[trd
->sp
-1];
551 trd
->stack
[trd
->sp
-1] = argv
[0];
552 if (argp
[0]) *(argp
[0]) = argc
;
555 if (argp
[0]) *(argp
[0]) = argv
[1];
556 if (argp
[1]) *(argp
[1]) = argv
[0];
564 argv
[0] = STACK_POP();
567 if (argv
[0] < 0) argv
[0] += trd
->sp
;
568 if (argv
[0] < 0 || argv
[0] >= trd
->sp
) vmfatal(tid
, opc
, "invalid stack index (%d)", argv
[0]);
569 argv
[2] = trd
->stack
[argv
[0]];
571 if (opcode
== VM_ROL
) {
572 for (int f
= argv
[0]+1; f
< trd
->sp
; ++f
) trd
->stack
[f
-1] = trd
->stack
[f
];
583 if (argp
[1]) *(argp
[1]) = argv
[2];
589 if (argp
[0]) *(argp
[0]) = trd
->sp
;
599 if (argp
[0]) *(argp
[0]) = tid
;
611 argv
[0] = STACK_POP();
613 if (argv
[0] >= 0 && argv
[0] < VM_MAX_THREADS
&& vmThreads
[argv
[0]].stack
!= NULL
) {
616 if (argv
[0] == tid
) return 1;
617 vmFreeThread(&vmThreads
[argv
[0]]);
621 if (argv
[0] == tid
) return -1;
622 vmThreads
[argv
[0]].suspended
= 1;
625 vmThreads
[argv
[0]].suspended
= 0;
634 argv
[0] = STACK_POP();
636 if (argv
[0] >= 0 && argv
[0] < VM_MAX_THREADS
&& vmThreads
[argv
[0]].stack
!= NULL
) {
637 argv
[0] = vmThreads
[argv
[0]].suspended
;
642 if (argp
[1]) *(argp
[1]) = argv
[0];
652 argv
[0] = STACK_POP();
654 if (argv
[0] < 0 || argv
[0] >= vmCodeSize
) vmfatal(tid
, opc
, "invalid address in RXC (%d)", argv
[0]);
657 STACK_PUSH(vmCode
[argv
[0]]);
658 } else if (argp
[1]) {
659 *(argp
[1]) = vmCode
[argv
[0]];
667 argv
[1] = STACK_POP(); // byte
668 argv
[0] = STACK_POP(); // address
672 argv
[1] = STACK_POP(); // byte
675 if (argv
[0] < 0 || argv
[0] >= vmCodeSize
) vmfatal(tid
, opc
, "invalid address in WXC (%d)", argv
[0]);
676 if (argv
[1] < -128 || argv
[1] > 255) vmfatal(tid
, opc
, "invalid value in WXC (%d)", argv
[1]);
677 if (argv
[1] < 0) argv
[1] &= 0xff;
678 vmCode
[argv
[0]] = argv
[1];
685 trd
->pc
= STACK_POP();
688 vmfatal(tid
, opc
, "invalid RET");
689 case 2: // # of locals to pop, # of arguments to pop
690 case 3: // # of locals to pop, # of arguments to pop, retvalue (taken before all pops, pushes after returning)
691 if (argv
[0] < 0) argv
[0] = 0;
692 if (argv
[1] < 0) argv
[1] = 0;
693 STACK_WANT(argv
[0]+argv
[1]+1);
694 trd
->sp
-= argv
[0]; // drop locals
695 trd
->pc
= STACK_POP();
696 trd
->sp
-= argv
[1]; // drop arguments
710 argv
[0] = STACK_POP();
713 vmRSTCB(tid
, opcode
, argc
, argv
, argp
);
714 } else if (argc
== 0) {
726 argv
[1] = STACK_POP(); // y
727 argv
[0] = STACK_POP(); // x
730 if (vmMapGetCB
) argv
[0] = vmMapGetCB(tid
, opcode
==VM_MGF
, argv
[0], argv
[1]); else argv
[0] = 0;
732 if (argp
[2]) *(argp
[2]) = argv
[0];
733 } else if (argc
== 1) {
734 if (argp
[0]) *(argp
[0]) = argv
[0];
745 argv
[2] = STACK_POP(); // tile
746 argv
[1] = STACK_POP(); // y
747 argv
[0] = STACK_POP(); // x
751 argv
[2] = argv
[0]; // tile
752 argv
[1] = STACK_POP(); // y
753 argv
[0] = STACK_POP(); // x
757 argv
[2] = STACK_POP(); // tile
760 if (vmMapSetCB
) vmMapSetCB(tid
, opcode
==VM_MSF
, argv
[0], argv
[1], argv
[2]);
764 vmfatal(tid
, opc
, "invalid opcode (%d)", opcode
);
770 void vmExecuteAll (void) {
773 memset(vmExecuted
, 0, sizeof(vmExecuted
));
776 for (int t
= 0; t
< VM_MAX_THREADS
; ++t
) {
777 if (!vmExecuted
[t
] && vmThreads
[t
].stack
!= NULL
&& !vmThreads
[t
].suspended
) {
782 for (f
= MAX_INST_COUNT
; f
>= 0; --f
) {
783 int res
= vmExecuteOne(t
);
787 //fprintf(stderr, "!!!\n");
792 vmFreeThread(&vmThreads
[t
]);
797 if (f
< 0) vmfatal(t
, vmThreads
[t
].pc
, "too many instructions in frame");
804 // <0: BRK; >0: END; 0: ok
805 // maxinst<0: any number of instructions
806 int vmExecuteBSR (int tid
, int pc
, int maxinst
) {
809 if (tid
< 0 || tid
>= VM_MAX_THREADS
|| vmThreads
[tid
].stack
== NULL
|| vmThreads
[tid
].suspended
) return 2; // END
810 opc
= vmThreads
[tid
].pc
;
811 vmPush(tid
, -666666);
812 vmThreads
[tid
].pc
= pc
;
813 if (maxinst
== 0) maxinst
= MAX_INST_COUNT
;
815 int res
= vmExecuteOne(tid
);
817 if (!vmThreads
[tid
].stack
) return 1;
818 if (vmThreads
[tid
].suspended
) return -2;
819 if (res
> 0) vmFreeThread(&vmThreads
[tid
]);
820 if (res
!= 0) return res
;
821 if (vmThreads
[tid
].pc
== -666666) {
822 vmThreads
[tid
].pc
= opc
;
825 if (maxinst
== 1) vmfatal(tid
, vmThreads
[tid
].pc
, "too many instructions in vmExecuteBSR");
826 if (maxinst
> 0) --maxinst
;
831 int vmIsThreadAlive (int tid
) {
832 if (tid
>= 0 && tid
< VM_MAX_THREADS
) return vmThreads
[tid
].stack
!= NULL
;
837 int vmNewThread (int pc
) {
838 int tid
= vmFindFreeThread();
841 if (vmInitThread(&vmThreads
[tid
]) != 0) return -1;
842 vmThreads
[tid
].pc
= pc
;
843 if (tid
> vmLastThreadId
) vmLastThreadId
= tid
;
849 int vmKillThread (int tid
) {
850 if (vmIsThreadAlive(tid
)) {
851 vmFreeThread(&vmThreads
[tid
]);
859 int vmIsSuspendedThread (int tid
) {
860 if (vmIsThreadAlive(tid
)) return vmThreads
[tid
].suspended
;
865 int vmSuspendThread (int tid
) {
866 if (vmIsThreadAlive(tid
)) { vmThreads
[tid
].suspended
= 1; return 0; }
871 int vmResumeThread (int tid
) {
872 if (vmIsThreadAlive(tid
)) { vmThreads
[tid
].suspended
= 0; return 0; }
877 int vmGetTVar (int tid
, int idx
) {
878 if (idx
>= 0 && idx
< VM_VARS_SIZE
&& vmIsThreadAlive(tid
)) return vmThreads
[tid
].tvars
[idx
];
883 int vmSetTVar (int tid
, int idx
, int value
) {
884 if (idx
>= 0 && idx
< VM_VARS_SIZE
&& vmIsThreadAlive(tid
)) { vmThreads
[tid
].tvars
[idx
] = value
; return 0; }
889 int vmGetSP (int tid
) {
890 if (vmIsThreadAlive(tid
)) return vmThreads
[tid
].sp
;
895 int vmGetPC (int tid
) {
896 if (vmIsThreadAlive(tid
)) return vmThreads
[tid
].pc
;
901 int vmSetPC (int tid
, int pc
) {
902 if (vmIsThreadAlive(tid
) && pc
>= 0 && pc
< vmCodeSize
) { vmThreads
[tid
].pc
= pc
; return 0; }
907 int vmSetSP (int tid
, int value
) {
908 if (vmIsThreadAlive(tid
)) { vmThreads
[tid
].sp
= value
; return 0; }
913 int vmGetStack (int tid
, int idx
) {
914 if (vmIsThreadAlive(tid
)) {
915 if (idx
< 0) idx
+= vmThreads
[tid
].sp
;
916 if (idx
>= 0 && idx
< VM_STACK_SIZE
) return vmThreads
[tid
].stack
[idx
];
922 int vmSetStack (int tid
, int idx
, int value
) {
923 if (vmIsThreadAlive(tid
)) {
924 if (idx
< 0) idx
+= vmThreads
[tid
].sp
;
925 if (idx
>= 0 && idx
< VM_STACK_SIZE
) { vmThreads
[tid
].stack
[idx
] = value
; return 0; }
931 int vmPush (int tid
, int value
) {
932 if (vmIsThreadAlive(tid
) && vmThreads
[tid
].sp
< VM_STACK_SIZE
) {
933 vmThreads
[tid
].stack
[vmThreads
[tid
].sp
++] = value
;
940 int vmPop (int tid
) {
941 if (vmIsThreadAlive(tid
) && vmThreads
[tid
].sp
> 0) return vmThreads
[tid
].stack
[--vmThreads
[tid
].sp
];
946 int vmLoadArgs (int tid
, int argc
, int argv
[], int *argp
[], int aargc
, int aargv
[], int *aargp
[]) {
947 if (!vmIsThreadAlive(tid
)) return -1;
951 if (e
> aargc
) e
= aargc
;
952 for (int f
= 0; f
< e
; ++f
) { argv
[f
] = aargv
[f
]; argp
[f
] = aargp
[f
]; }
953 for (int f
= e
; f
< argc
; ++f
) {
954 if (vmThreads
[tid
].sp
<= 0) return -1;
956 argv
[argc
-f
] = vmThreads
[tid
].stack
[--vmThreads
[tid
].sp
];
963 int vmLastThread (void) {
964 return vmLastThreadId
;
968 static int readBuf (FILE *fl
, void *buf
, int len
) {
969 unsigned char *c
= (unsigned char *)buf
;
974 if (fread(&b
, 1, 1, fl
) != 1) return -1;
982 static int writeBuf (FILE *fl
, const void *buf
, int len
) {
983 const unsigned char *c
= (const unsigned char *)buf
;
986 unsigned char b
= *c
++;
989 if (fwrite(&b
, 1, 1, fl
) != 1) return -1;
995 static int readDW (FILE *fl
, int *v
) {
998 for (int f
= 0; f
< 4; ++f
) {
1001 if (fread(&b
, 1, 1, fl
) != 1) return -1;
1007 for (int f
= 0; f
< 4; ++f
) *v
|= t
[f
]<<(8*f
);
1013 static int writeDW (FILE *fl
, int v
) {
1014 for (int f
= 0; f
< 4; ++f
) {
1019 if (fwrite(&b
, 1, 1, fl
) != 1) return -1;
1026 ////////////////////////////////////////////////////////////////////////////////
1027 static VMLabelInfo
*labels
= NULL
;
1030 ////////////////////////////////////////////////////////////////////////////////
1031 void vmFreeLabels (void) {
1032 while (labels
!= NULL
) {
1033 VMLabelInfo
*l
= labels
;
1036 if (l
->name
) free(l
->name
);
1042 void vmFreeLabelsUntilMark (const char *name
) {
1043 while (labels
!= NULL
) {
1044 VMLabelInfo
*l
= labels
;
1045 int isMark
= (l
->type
== LB_MARK
);
1047 if (!(isMark
&& ((name
== NULL
&& l
->name
== NULL
) || (l
->name
!= NULL
&& name
!= NULL
&& strcmp(name
, l
->name
) == 0)))) isMark
= 0;
1049 if (l
->name
) free(l
->name
);
1056 VMLabelInfo
*vmLabelAddMark (const char *name
) {
1057 VMLabelInfo
*l
= malloc(sizeof(VMLabelInfo
));
1059 if (l
== NULL
) fatalis("out of memory");
1062 l
->name
= strdup(name
);
1063 if (l
->name
== NULL
) fatalis("out of memory");
1074 VMLabelInfo
*vmAddLabel (const char *name
, int type
, int value
, int pub
) {
1075 if (name
!= NULL
&& name
[0]) {
1076 VMLabelInfo
*l
= malloc(sizeof(VMLabelInfo
));
1078 if (l
== NULL
) fatalis("out of memory");
1080 l
->name
= strdup(name
);
1081 if (l
->name
== NULL
) fatalis("out of memory");
1096 static VMLabelInfo
*vmAddLabelToEnd (const char *name
, int type
, int value
, int pub
) {
1097 if (name
!= NULL
&& name
[0]) {
1098 VMLabelInfo
*l
= malloc(sizeof(VMLabelInfo
)), *last
= NULL
;
1100 if (l
== NULL
) fatalis("out of memory");
1102 l
->name
= strdup(name
);
1103 if (l
->name
== NULL
) fatalis("out of memory");
1111 if (labels
== NULL
) {
1114 for (last
= labels
; last
->next
!= NULL
; last
= last
->next
) ;
1123 VMLabelInfo
*vmFindLabel (const char *name
) {
1124 if (name
!= NULL
&& name
[0]) {
1125 for (VMLabelInfo
*l
= labels
; l
!= NULL
; l
= l
->next
) {
1126 if (l
->name
&& l
->name
[0] && strcasecmp(l
->name
, name
) == 0) return l
;
1133 int vmFindPC (const char *name
) {
1134 VMLabelInfo
*l
= vmFindLabel(name
);
1136 if (l
== NULL
) fatalis("no code label: '%s'", name
);
1141 int vmFindVarIndex (const char *name
) {
1142 VMLabelInfo
*l
= vmFindLabel(name
);
1144 if (l
== NULL
|| (l
->type
!= LB_GVAR
&& l
->type
!= LB_TVAR
&& l
->type
!= LB_SVAR
)) fatalis("no var label: '%s'", name
);
1149 int vmFindGVarIndex (const char *name
) {
1150 VMLabelInfo
*l
= vmFindLabel(name
);
1152 if (l
== NULL
|| l
->type
!= LB_GVAR
) fatalis("no global var label: '%s'", name
);
1157 int vmFindTVarIndex (const char *name
) {
1158 VMLabelInfo
*l
= vmFindLabel(name
);
1160 if (l
== NULL
|| l
->type
!= LB_TVAR
) fatalis("no thread var label: '%s'", name
);
1165 int vmFindSVarIndex (const char *name
) {
1166 VMLabelInfo
*l
= vmFindLabel(name
);
1168 if (l
== NULL
|| l
->type
!= LB_SVAR
) fatalis("no thread var label: '%s'", name
);
1173 int vmFindConst (const char *name
) {
1174 VMLabelInfo
*l
= vmFindLabel(name
);
1176 if (l
== NULL
|| l
->type
!= LB_CONST
) fatalis("no const label: '%s'", name
);
1181 VMLabelInfo
*vmFindMark (const char *name
) {
1182 for (VMLabelInfo
*l
= labels
; l
!= NULL
; l
= l
->next
) {
1183 if (l
->type
!= LB_MARK
) continue;
1184 if (name
== NULL
&& l
->name
== NULL
) return l
;
1185 if (name
!= NULL
&& l
->name
!= NULL
&& strcmp(name
, l
->name
) == 0) return l
;
1191 #define XREAD(dest,size,errlabel) do { \
1192 if (pos+(size) > rsz) goto errlabel; \
1193 if ((size) > 0) memcpy((dest), buf+pos, (size)); \
1198 #define XREADB(dest,errlabel) do { \
1199 if (pos+1 > rsz) goto errlabel; \
1200 (dest) = buf[pos]; \
1205 #define XREADW(dest,errlabel) do { \
1206 if (pos+2 > rsz) goto errlabel; \
1207 (dest) = buf[pos+1]; \
1209 (dest) |= buf[pos+0]; \
1215 extern int vmLoadCodeFileFromDump (const void *data
, int datasize
, int pc
) {
1216 int rsz
= datasize
, pos
= 4;
1217 const uint8_t *dta
= (const uint8_t *)data
;
1219 int csize
, lcnt
, rcnt
, elcnt
;
1221 if (rsz
< 4) return -1;
1222 if (memcmp(data
, "AVM1", 4) != 0) goto quitbufonly
;
1224 if (buf
== NULL
) return 1;
1226 for (int f
= 4; f
< rsz
; ++f
) buf
[f
] = dta
[f
]^SECRET
;
1228 XREADW(csize
, quitbufonly
);
1229 XREADW(rcnt
, quitbufonly
); // fixups
1230 XREADW(elcnt
, quitbufonly
); // extern labels
1231 XREADW(lcnt
, quitbufonly
); // labels
1232 if (pc
< 0 || pc
+csize
> 65535) goto quitbufonly
;
1234 /*if (goobers)*/ fprintf(stderr
, "code: %d bytes, %d public labels, %d relocations, %d externs\n", csize
, lcnt
, rcnt
, elcnt
);
1236 XREAD(vmCode
+pc
, csize
, quitbufonly
);
1239 for (int f
= 0; f
< rcnt
; ++f
) {
1240 int rel
, newpc
, val
;
1242 XREADW(rel
, quitbufonly
);
1243 if (rel
< 0 || rel
+1 >= csize
) goto quitbufonly
;
1244 newpc
= pc
+rel
; // fix here
1246 val
= vmCode
[newpc
+1];
1248 val
|= vmCode
[newpc
+0];
1252 vmCode
[newpc
+0] = val
&0xff;
1253 vmCode
[newpc
+1] = (val
>>8)&0xff;
1256 // load extern labels and perform fixups
1257 for (int f
= 0; f
< elcnt
; ++f
) {
1262 XREADB(type
, quitbufonly
);
1263 XREADB(namelen
, quitbufonly
);
1264 XREAD(name
, namelen
, quitbufonly
);
1265 XREADW(rcnt
, quitbufonly
);
1267 if (type
< LB_MIN_TYPE
|| type
> LB_MAX_TYPE
) goto quitbufonly
;
1268 if (namelen
< 1) goto quitbufonly
;
1271 for (int c
= 0; c
< rcnt
; ++c
) {
1275 XREADB(size
, quitbufonly
);
1276 XREADW(xpc
, quitbufonly
);
1278 if (size
!= 1 && size
!= 2) goto quitbufonly
;
1279 if (xpc
< 0 || xpc
+size
> csize
) goto quitbufonly
;
1282 l
= vmFindLabel(name
);
1284 /*if (goobers)*/ fprintf(stderr
, "VM: unknown extern: '%s'\n", name
);
1288 if (l
->type
== LB_GVAR
) val
|= 0x80;
1289 //fprintf(stderr, "%d: [%s]: ofs=%d, size=%d, value=%d (%d)\n", c, l->name, xpc, size, val, vmCode[xpc]);
1290 if (size
== 1 && (val
< -128 || val
> 255)) {
1291 /*if (goobers)*/ fprintf(stderr
, "VM: extern too big: '%s'\n", name
);
1294 vmCode
[xpc
+0] = val
&0xff;
1295 if (size
== 2) vmCode
[xpc
+1] = (val
>>8)&0xff;
1299 vmLabelAddMark(NULL
); // for this code labels
1300 for (int f
= 0; f
< lcnt
; ++f
) {
1301 unsigned char type
, namelen
, b
;
1307 switch (type
&0x7f) {
1315 XREADW(value
, quit
);
1316 if (value
>= 32768) value
-= 65536;
1319 XREADW(value
, quit
);
1320 value
+= pc
; // fixup
1323 fatalis("invalid label type: %d\n", type
);
1327 XREADB(namelen
, quit
);
1328 XREAD(name
, namelen
, quit
);
1331 if (namelen
> 0) vmAddLabel(name
, type
&0x7f, value
, type
&0x80?1:0);
1332 //if (l->type == LB_CODE) printf("%d/%d: [%s] %d : %d\n", f, lcnt, l->name, l->value-pc, l->value);
1338 vmFreeLabelsUntilMark(NULL
);
1345 ////////////////////////////////////////////////////////////////////////////////
1346 int vmSaveState (FILE *fl
) {
1351 if (writeDW(fl
, VM_VARS_SIZE
) != 0) return -1;
1352 //if (writeDW(fl, VM_STACK_SIZE) != 0) return -1;
1353 if (writeDW(fl
, vmLastThreadId
) != 0) return -1;
1354 if (writeDW(fl
, vmCodeSize
) != 0) return -1;
1355 if (writeBuf(fl
, vmCode
, vmCodeSize
) != 0) return -1;
1356 for (int f
= 0; f
< VM_VARS_SIZE
; ++f
) if (writeDW(fl
, vmGVars
[f
]) != 0) return -1;
1357 for (int f
= 0; f
<= vmLastThreadId
; ++f
) {
1358 if (vmThreads
[f
].stack
!= NULL
) {
1359 if (writeDW(fl
, f
) != 0) return -1;
1360 if (writeDW(fl
, vmThreads
[f
].pc
) != 0) return -1;
1361 if (writeDW(fl
, vmThreads
[f
].suspended
) != 0) return -1;
1362 if (writeDW(fl
, vmThreads
[f
].sp
) != 0) return -1;
1363 for (int c
= 0; c
< vmThreads
[f
].sp
; ++c
) if (writeDW(fl
, vmThreads
[f
].stack
[c
]) != 0) return -1;
1364 for (int c
= 0; c
< VM_VARS_SIZE
; ++c
) if (writeDW(fl
, vmThreads
[f
].tvars
[c
]) != 0) return -1;
1366 if (writeDW(fl
, -1) != 0) return -1;
1370 for (VMLabelInfo
*l
= labels
; l
!= NULL
; l
= l
->next
) ++lcnt
;
1371 if (writeDW(fl
, lcnt
) != 0) return -1;
1372 for (VMLabelInfo
*l
= labels
; l
!= NULL
; l
= l
->next
) {
1373 int nlen
= l
->name
!= NULL
? strlen(l
->name
) : -1;
1375 if (writeDW(fl
, l
->type
) != 0) return -1;
1376 if (writeDW(fl
, nlen
) != 0) return -1;
1378 if (writeBuf(fl
, l
->name
, nlen
) != 0) return -1;
1380 if (writeDW(fl
, l
->value
) != 0) return -1;
1381 if (writeDW(fl
, l
->pub
) != 0) return -1;
1388 int vmLoadState (FILE *fl
) {
1392 if (vmInitialize() != 0) goto fail
;
1394 if (readDW(fl
, &v
) != 0 || v
!= VM_VARS_SIZE
) goto fail
;
1395 //if (readDW(fl, &ssz) != 0) goto fail;
1396 if (readDW(fl
, &vmLastThreadId
) != 0 || vmLastThreadId
< 0 || vmLastThreadId
> VM_MAX_THREADS
) goto fail
;
1397 if (readDW(fl
, &vmCodeSize
) != 0 || vmCodeSize
< 1 || vmCodeSize
> 65536) goto fail
;
1398 if (readBuf(fl
, vmCode
, vmCodeSize
) != 0) goto fail
;
1399 for (int f
= 0; f
< VM_VARS_SIZE
; ++f
) if (readDW(fl
, &vmGVars
[f
]) != 0) goto fail
;
1400 for (int f
= 0; f
<= vmLastThreadId
; ++f
) {
1403 if (readDW(fl
, &flag
) != 0) goto fail
;
1404 if (flag
== -1) continue;
1405 if (flag
!= f
) goto fail
;
1406 if (vmInitThread(&vmThreads
[f
]) != 0) goto fail
;
1407 if (readDW(fl
, &vmThreads
[f
].pc
) != 0) goto fail
;
1408 if (readDW(fl
, &vmThreads
[f
].suspended
) != 0) goto fail
;
1409 if (readDW(fl
, &vmThreads
[f
].sp
) != 0) goto fail
;
1410 if (!vmThreads
[f
].suspended
) {
1411 if (vmThreads
[f
].pc
< 0 || vmThreads
[f
].pc
>= vmCodeSize
) goto fail
;
1412 if (vmThreads
[f
].sp
< 0 || vmThreads
[f
].sp
> VM_STACK_SIZE
) goto fail
;
1414 for (int c
= 0; c
< vmThreads
[f
].sp
; ++c
) if (readDW(fl
, &vmThreads
[f
].stack
[c
]) != 0) goto fail
;
1415 for (int c
= 0; c
< VM_VARS_SIZE
; ++c
) if (readDW(fl
, &vmThreads
[f
].tvars
[c
]) != 0) goto fail
;
1418 if (readDW(fl
, &lcnt
) != 0) goto fail
;
1419 for (; lcnt
> 0; --lcnt
) {
1420 int type
, namelen
, value
, pub
;
1423 if (readDW(fl
, &type
) != 0) goto fail
;
1424 if (readDW(fl
, &namelen
) != 0) goto fail
;
1425 if (namelen
> 255) goto fail
;
1427 if (readBuf(fl
, name
, namelen
) != 0) goto fail
;
1430 if (readDW(fl
, &value
) != 0) goto fail
;
1431 if (readDW(fl
, &pub
) != 0) goto fail
;
1432 vmAddLabelToEnd(namelen
>=0 ? name
: NULL
, type
, value
, pub
);