2021-01-29 15:14:37 -05:00
# include <stdio.h>
/*
Copyright ( c ) 2021 Devine Lu Linvega
Permission to use , copy , modify , and distribute this software for any
purpose with or without fee is hereby granted , provided that the above
copyright notice and this permission notice appear in all copies .
THE SOFTWARE IS PROVIDED " AS IS " AND THE AUTHOR DISCLAIMS ALL WARRANTIES
WITH REGARD TO THIS SOFTWARE .
*/
typedef unsigned char Uint8 ;
2021-02-04 15:22:08 -05:00
typedef unsigned short Uint16 ;
2021-01-29 15:14:37 -05:00
2021-01-29 16:59:16 -05:00
typedef struct {
2021-02-04 00:53:56 -05:00
int ptr ;
2021-02-04 15:22:08 -05:00
Uint8 data [ 65536 ] ;
2021-01-29 16:59:16 -05:00
} Program ;
2021-01-31 00:31:49 -05:00
typedef struct {
2021-02-04 15:22:08 -05:00
Uint16 addr ;
char name [ 64 ] ;
2021-01-31 00:31:49 -05:00
} Label ;
int labelslen ;
Label labels [ 256 ] ;
2021-02-07 13:21:41 -05:00
Program p ;
2021-01-30 17:25:48 -05:00
2021-02-01 23:21:27 -05:00
/* clang-format off */
2021-02-06 13:39:13 -05:00
char ops [ ] [ 4 ] = {
2021-02-09 23:35:31 -05:00
" BRK " , " --- " , " LI1 " , " LIX " , " IOR " , " IOW " , " LDR " , " STR " ,
2021-02-07 23:49:00 -05:00
" JMP " , " JSR " , " RTI " , " RTS " , " --- " , " --- " , " --- " , " --- " ,
2021-02-07 13:21:41 -05:00
" POP " , " DUP " , " SWP " , " OVR " , " ROT " , " AND " , " ORA " , " ROL " ,
" ADD " , " SUB " , " MUL " , " DIV " , " EQU " , " NEQ " , " GTH " , " LTH "
2021-02-04 15:22:08 -05:00
} ;
2021-02-01 23:21:27 -05:00
2021-02-07 13:21:41 -05:00
int scmp ( char * a , char * b ) { int i = 0 ; while ( a [ i ] = = b [ i ] ) if ( ! a [ i + + ] ) return 1 ; return 0 ; } /* string compare */
int slen ( char * s ) { int i = 0 ; while ( s [ i ] & & s [ + + i ] ) ; return i ; } /* string length */
int sihx ( char * s ) { int i = 0 ; char c ; while ( ( c = s [ i + + ] ) ) if ( ! ( c > = ' 0 ' & & c < = ' 9 ' ) & & ! ( c > = ' a ' & & c < = ' f ' ) & & ! ( c > = ' A ' & & c < = ' F ' ) ) return 0 ; return 1 ; } /* string is hexadecimal */
int shex ( char * s ) { int n = 0 , i = 0 ; char c ; while ( ( c = s [ i + + ] ) ) if ( c > = ' 0 ' & & c < = ' 9 ' ) n = n * 16 + ( c - ' 0 ' ) ; else if ( c > = ' A ' & & c < = ' F ' ) n = n * 16 + 10 + ( c - ' A ' ) ; else if ( c > = ' a ' & & c < = ' f ' ) n = n * 16 + 10 + ( c - ' a ' ) ; return n ; } /* string to num */
int cmnt ( char * w , int * skip ) { if ( w [ 0 ] = = ' ) ' ) { * skip = 0 ; return 1 ; } if ( w [ 0 ] = = ' ( ' ) * skip = 1 ; if ( * skip ) return 1 ; return 0 ; } /* comment helper */
char * scpy ( char * src , char * dst , int len ) { int i = 0 ; while ( ( dst [ i ] = src [ i ] ) & & i < len - 2 ) i + + ; dst [ i + 1 ] = ' \0 ' ; return dst ; } /* string copy */
2021-01-29 16:59:16 -05:00
# pragma mark - Helpers
2021-02-07 13:21:41 -05:00
/* clang-format on */
2021-01-30 17:25:48 -05:00
2021-02-04 15:22:08 -05:00
# pragma mark - I / O
2021-02-01 23:21:27 -05:00
void
2021-02-04 15:22:08 -05:00
pushbyte ( Uint8 b , int lit )
2021-02-01 23:21:27 -05:00
{
2021-02-09 23:35:31 -05:00
if ( lit )
pushbyte ( 0x02 , 0 ) ;
2021-02-04 15:22:08 -05:00
p . data [ p . ptr + + ] = b ;
2021-02-01 23:21:27 -05:00
}
void
2021-02-04 15:22:08 -05:00
pushshort ( Uint16 s , int lit )
2021-02-01 23:21:27 -05:00
{
2021-02-04 15:22:08 -05:00
if ( lit ) {
2021-02-09 23:35:31 -05:00
pushbyte ( 0x03 , 0 ) ;
2021-02-04 15:22:08 -05:00
pushbyte ( 0x02 , 0 ) ;
2021-02-01 23:21:27 -05:00
}
2021-02-04 15:22:08 -05:00
pushbyte ( ( s > > 8 ) & 0xff , 0 ) ;
pushbyte ( s & 0xff , 0 ) ;
2021-02-01 23:21:27 -05:00
}
2021-02-05 14:57:37 -05:00
void
2021-02-07 13:21:41 -05:00
pushtext ( char * w )
2021-02-05 14:57:37 -05:00
{
int i = slen ( w ) ;
2021-02-09 23:35:31 -05:00
pushbyte ( 0x03 , 0 ) ;
2021-02-05 14:57:37 -05:00
pushbyte ( slen ( w ) , 0 ) ;
while ( i > 0 )
pushbyte ( w [ - - i ] , 0 ) ;
}
2021-01-31 00:31:49 -05:00
Label *
findlabel ( char * s )
{
int i ;
for ( i = 0 ; i < labelslen ; + + i )
if ( scmp ( labels [ i ] . name , s ) )
return & labels [ i ] ;
return NULL ;
}
2021-02-06 13:39:13 -05:00
Uint8
2021-02-07 13:21:41 -05:00
findoperator ( char * s )
2021-02-06 13:39:13 -05:00
{
int i ;
2021-02-07 13:21:41 -05:00
for ( i = 0 ; i < 0x20 ; + + i ) {
2021-02-06 13:39:13 -05:00
int m = 0 ;
2021-02-07 13:21:41 -05:00
char * o = ops [ i ] ;
if ( o [ 0 ] ! = s [ 0 ] | | o [ 1 ] ! = s [ 1 ] | | o [ 2 ] ! = s [ 2 ] )
continue ;
2021-02-06 13:39:13 -05:00
while ( s [ 3 + m ] ) {
2021-02-07 13:21:41 -05:00
if ( s [ 3 + m ] = = ' ^ ' ) i | = ( 1 < < 5 ) ; /* mode: 16 bits */
if ( s [ 3 + m ] = = ' ~ ' ) i | = ( 1 < < 6 ) ; /* mode: signed */
2021-02-07 23:49:00 -05:00
if ( s [ 3 + m ] = = ' ? ' ) i | = ( 1 < < 7 ) ; /* mode: conditional */
2021-02-06 13:39:13 -05:00
m + + ;
}
return i ;
}
return 0 ;
}
2021-02-04 15:22:08 -05:00
# pragma mark - Parser
2021-01-31 00:31:49 -05:00
2021-01-30 17:25:48 -05:00
int
2021-02-04 15:22:08 -05:00
error ( char * name , char * id )
2021-01-30 17:25:48 -05:00
{
2021-02-05 23:18:30 -05:00
printf ( " Error: %s[%s] \n " , name , id ) ;
2021-02-04 16:49:03 -05:00
return 0 ;
}
int
2021-02-07 13:21:41 -05:00
makelabel ( char * name , Uint16 addr )
2021-02-04 16:49:03 -05:00
{
Label * l ;
2021-02-07 13:21:41 -05:00
if ( findlabel ( name ) )
return error ( " Label duplicate " , name ) ;
2021-02-04 16:49:03 -05:00
l = & labels [ labelslen + + ] ;
l - > addr = addr ;
2021-02-07 13:21:41 -05:00
scpy ( name , l - > name , 64 ) ;
2021-02-04 16:49:03 -05:00
printf ( " New label: %s[0x%02x] \n " , l - > name , l - > addr ) ;
return 1 ;
}
2021-02-05 13:51:45 -05:00
int
makeconst ( char * id , FILE * f )
{
char wv [ 64 ] ;
fscanf ( f , " %s " , wv ) ;
return makelabel ( id , shex ( wv ) ) ;
}
2021-02-04 00:53:56 -05:00
int
2021-01-29 15:14:37 -05:00
pass1 ( FILE * f )
{
2021-02-04 18:23:04 -05:00
int skip = 0 , vars = 0 ;
Uint16 addr = 0 ;
2021-02-04 15:22:08 -05:00
char w [ 64 ] ;
while ( fscanf ( f , " %s " , w ) = = 1 ) {
2021-02-07 13:21:41 -05:00
if ( cmnt ( w , & skip ) )
continue ;
2021-02-05 13:51:45 -05:00
if ( w [ 0 ] = = ' @ ' & & ! makelabel ( w + 1 , addr ) )
2021-02-04 16:49:03 -05:00
return error ( " Pass1 failed " , w ) ;
if ( w [ 0 ] = = ' ; ' & & ! makelabel ( w + 1 , vars + + ) )
return error ( " Pass1 failed " , w ) ;
2021-02-05 13:51:45 -05:00
if ( w [ 0 ] = = ' : ' ) {
if ( ! makeconst ( w + 1 , f ) )
return error ( " Pass1 failed " , w ) ;
else
continue ;
}
2021-02-07 13:21:41 -05:00
if ( findoperator ( w ) | | scmp ( w , " BRK " ) )
2021-02-04 15:22:08 -05:00
addr + = 1 ;
2021-02-07 13:21:41 -05:00
else {
switch ( w [ 0 ] ) {
case ' | ' : addr = shex ( w + 1 ) ; break ;
case ' @ ' :
case ' ; ' : break ;
case ' " ' : addr + = slen ( w + 1 ) + 2 ; break ;
2021-02-09 23:35:31 -05:00
case ' # ' : addr + = 4 ; break ;
case ' . ' : addr + = 2 ; break ;
case ' , ' :
addr + = ( slen ( w + 1 ) = = 2 ? 1 : 2 ) ;
addr + = ( sihx ( w + 1 ) ? slen ( w + 1 ) / 2 : 2 ) ;
break ;
2021-02-07 13:21:41 -05:00
default : return error ( " Unknown label " , w ) ;
}
}
2021-01-30 17:25:48 -05:00
}
rewind ( f ) ;
2021-02-04 15:22:08 -05:00
return 1 ;
2021-01-30 17:25:48 -05:00
}
2021-02-04 15:22:08 -05:00
int
2021-01-30 17:25:48 -05:00
pass2 ( FILE * f )
{
int skip = 0 ;
2021-02-04 15:22:08 -05:00
char w [ 64 ] ;
while ( fscanf ( f , " %s " , w ) = = 1 ) {
2021-02-01 14:58:47 -05:00
Uint8 op = 0 ;
2021-01-31 00:31:49 -05:00
Label * l ;
2021-02-05 13:51:45 -05:00
if ( w [ 0 ] = = ' @ ' ) continue ;
2021-02-04 15:22:08 -05:00
if ( w [ 0 ] = = ' ; ' ) continue ;
2021-02-07 13:21:41 -05:00
if ( cmnt ( w , & skip ) ) continue ;
2021-02-05 13:51:45 -05:00
if ( w [ 0 ] = = ' | ' )
2021-02-04 15:22:08 -05:00
p . ptr = shex ( w + 1 ) ;
2021-02-05 13:51:45 -05:00
else if ( w [ 0 ] = = ' : ' )
fscanf ( f , " %s " , w ) ;
2021-02-05 14:57:37 -05:00
else if ( w [ 0 ] = = ' " ' )
2021-02-07 13:21:41 -05:00
pushtext ( w + 1 ) ;
2021-02-09 20:22:52 -05:00
else if ( w [ 0 ] = = ' # ' )
2021-02-07 15:13:38 -05:00
pushshort ( shex ( w + 1 ) & 0xff , 1 ) ;
2021-02-09 20:22:52 -05:00
else if ( ( l = findlabel ( w + 1 ) ) )
2021-02-04 15:22:08 -05:00
pushshort ( l - > addr , w [ 0 ] = = ' , ' ) ;
2021-02-07 13:21:41 -05:00
else if ( ( op = findoperator ( w ) ) | | scmp ( w , " BRK " ) )
2021-02-06 13:39:13 -05:00
pushbyte ( op , 0 ) ;
2021-02-04 15:22:08 -05:00
else if ( sihx ( w + 1 ) & & slen ( w + 1 ) = = 2 )
pushbyte ( shex ( w + 1 ) , w [ 0 ] = = ' , ' ) ;
else if ( sihx ( w + 1 ) & & slen ( w + 1 ) = = 4 )
pushshort ( shex ( w + 1 ) , w [ 0 ] = = ' , ' ) ;
2021-02-01 23:21:27 -05:00
else
2021-02-04 16:49:03 -05:00
return error ( " Unknown label " , w ) ;
2021-01-29 15:14:37 -05:00
}
2021-02-04 15:22:08 -05:00
return 1 ;
2021-01-29 15:14:37 -05:00
}
int
main ( int argc , char * argv [ ] )
{
FILE * f ;
if ( argc < 3 )
2021-02-04 15:22:08 -05:00
return error ( " Input " , " Missing " ) ;
2021-01-29 15:14:37 -05:00
if ( ! ( f = fopen ( argv [ 1 ] , " r " ) ) )
2021-02-04 15:22:08 -05:00
return error ( " Open " , " Failed " ) ;
if ( ! pass1 ( f ) | | ! pass2 ( f ) )
return error ( " Assembly " , " Failed " ) ;
2021-01-29 16:59:16 -05:00
fwrite ( p . data , sizeof ( p . data ) , 1 , fopen ( argv [ 2 ] , " wb " ) ) ;
fclose ( f ) ;
2021-02-08 15:16:39 -05:00
printf ( " Assembled %s. \n \n " , argv [ 2 ] ) ;
2021-01-29 15:14:37 -05:00
return 0 ;
}