This commit is contained in:
Rebecca Doth 2024-04-16 17:50:55 +02:00
parent 15133290e3
commit 7c3411d826
1 changed files with 178 additions and 117 deletions

295
main.c
View File

@ -2,14 +2,23 @@
#include <stdlib.h>
#include <string.h>
#define REGSTRLEN 8
#define INSTSTRLEN 32
typedef enum {
UNDEFINED = 0,
MOV_RM = 0b100010
} OpCodes;
typedef enum {
BX_SI = 0b000,
BX_DI = 0b001,
BP_SI = 0b010,
BP_DI = 0b011,
SI___ = 0b100,
DI___ = 0b101,
BP___ = 0b110,
Di_Ad = 0b110,
BX___ = 0b111
} EffectiveAddress;
typedef enum {
AX = 0b000,
AL = 0b000,
@ -34,58 +43,6 @@ typedef enum {
DI = 0b111
} Registers;
typedef struct {
OpCodes OpCode;
Registers Dst;
Registers Src;
_Bool wide;
} Instruction;
Registers match_regs(char byte, _Bool wide)
{
if(wide)
switch(byte)
{
case CX:
return CX;
case DX:
return DX;
case BX:
return BX;
case SP:
return SP;
case BP:
return BP;
case SI:
return SI;
case DI:
return DI;
//case AX:
default:
return AX;
}
switch(byte)
{
case CL:
return CL;
case DL:
return DL;
case BL:
return BL;
case AH:
return AH;
case CH:
return CH;
case DH:
return DH;
case BH:
return BH;
//case AL:
default:
return AL;
}
}
void insttostr(OpCodes opcode, char* instStr)
{
switch(opcode)
@ -100,36 +57,35 @@ void insttostr(OpCodes opcode, char* instStr)
}
}
void regtostr(Registers reg, char* regStr, _Bool wide)
void regtostr(Registers reg, char* str, _Bool wide)
{
if(wide)
{
switch(reg)
{
case CX:
memcpy(regStr, "cx", 2);
memcpy(str, "cx", 2);
break;
case DX:
memcpy(regStr, "dx", 2);
memcpy(str, "dx", 2);
break;
case BX:
memcpy(regStr, "bx", 2);
memcpy(str, "bx", 2);
break;
case SP:
memcpy(regStr, "sp", 2);
memcpy(str, "sp", 2);
break;
case BP:
memcpy(regStr, "bp", 2);
memcpy(str, "bp", 2);
break;
case SI:
memcpy(regStr, "si", 2);
memcpy(str, "si", 2);
break;
case DI:
memcpy(regStr, "di", 2);
memcpy(str, "di", 2);
break;
//case AX:
default:
memcpy(regStr, "ax", 2);
case AX:
memcpy(str, "ax", 2);
break;
}
} else
@ -138,96 +94,202 @@ void regtostr(Registers reg, char* regStr, _Bool wide)
switch(reg)
{
case CL:
memcpy(regStr, "cl", 2);
memcpy(str, "cl", 2);
break;
case DL:
memcpy(regStr, "dl", 2);
memcpy(str, "dl", 2);
break;
case BL:
memcpy(regStr, "bl", 2);
memcpy(str, "bl", 2);
break;
case AH:
memcpy(regStr, "ah", 2);
memcpy(str, "ah", 2);
break;
case CH:
memcpy(regStr, "ch", 2);
memcpy(str, "ch", 2);
break;
case DH:
memcpy(regStr, "dh", 2);
memcpy(str, "dh", 2);
break;
case BH:
memcpy(regStr, "bh", 2);
memcpy(str, "bh", 2);
break;
//case AL:
default:
memcpy(regStr, "al", 2);
case AL:
memcpy(str, "al", 2);
break;
}
}
}
void print_instructions(Instruction* instructions, size_t ninst, char* filename)
void eactostr(EffectiveAddress ea, char* str, unsigned short displacement, _Bool displace, _Bool directaddress)
{
str[0] = '[';
off_t offset = 1;
printf("; disassembly for file %s\nbits 16\n\n", filename);
for(size_t i = 0; i < ninst; ++i)
if(!directaddress)
{
if(instructions[i].OpCode != UNDEFINED)
switch(ea)
{
char inst[INSTSTRLEN] = { '\0' };
char dst[REGSTRLEN] = { '\0' };
char src[REGSTRLEN] = { '\0' };
insttostr(instructions[i].OpCode, inst);
regtostr(instructions[i].Dst, dst, instructions[i].wide);
regtostr(instructions[i].Src, src, instructions[i].wide);
printf("%s %s, %s\n", inst, dst, src);
} else
{
puts("; UNDEFINED INSTRUCTION");
case BX_SI:
memcpy(str + offset + 1, "bx + si", 7);
offset += 7;
break;
case BX_DI:
memcpy(str + offset + 1, "bx + di", 7);
offset += 7;
break;
case BP_SI:
memcpy(str + offset + 1, "bp + si", 7);
offset += 7;
break;
case BP_DI:
memcpy(str + offset + 1, "bp + di", 7);
offset += 7;
break;
case SI___:
memcpy(str + offset + 1, "si", 2);
offset += 2;
break;
case DI___:
memcpy(str + offset + 1, "di", 2);
offset += 2;
break;
case BP___:
memcpy(str + offset + 1, "bp", 2);
offset += 2;
break;
case BX___:
memcpy(str + offset + 1, "bx", 2);
offset += 2;
break;
}
}
putc('\n', stdout);
if(displace)
{
if(!directaddress)
{
memcpy(str + offset, " + ", 3);
offset += 3;
}
offset += snprintf(str + offset + 1, 6, "%hu", displacement);
}
str[offset + 1] = ']';
}
void parse_instructions(unsigned char* bytes, size_t nbytes, Instruction* instructions)
void print_instructions(unsigned char* bytes, size_t nbytes)
{
size_t bytes_used = 0;
for(size_t i = 0, iindx = 0; i < nbytes; i += bytes_used, ++iindx)
{
bytes_used = 0;
instructions[iindx].OpCode = (OpCodes)(bytes[i] >> 2);
OpCodes opcode = (OpCodes)(bytes[i] >> 2);
_Bool direction = (bytes[i] >> 1) & 0b1;
_Bool wide = bytes[i] & 0b1;
unsigned short displacement;
++bytes_used;
switch(instructions[iindx].OpCode)
char inststr[16] = { '\0' };
char srcostr[16] = { '\0' };
char dstostr[16] = { '\0' };
char tmp1, tmp2;
switch(opcode)
{
case MOV_RM:
bytes_used += 2;
case MOV_RM: // Register/Memory
switch(bytes[i + 1] >> 6) // MOD field
{
//case 0b11:
default:
instructions[iindx].wide = bytes[i] & 0b1;
Registers tmp1 = match_regs((bytes[i + 1] >> 3) & 0b111, instructions[iindx].wide); // REG field
Registers tmp2 = match_regs( bytes[i + 1] & 0b111, instructions[iindx].wide); // R/M field
// NOTE: technically i could avoid using tmp1/2 by moving the match_regs inside the if statement but
// CONT: i didnt feel like repeating code, in case a future redesign may be neccessery.
if((bytes[i] >> 0b1) & 0b1) // direction bit
case 0b00: // register to memory, no disp, 16-bit disp if R\M = 110
tmp1 = bytes[i + 1] & 0b111; // R/M field
tmp2 = (bytes[i + 1] >> 3) & 0b111; // REG field
++bytes_used;
if(!direction)
{
instructions[iindx].Src = tmp2;
instructions[iindx].Dst = tmp1;
insttostr(opcode, inststr);
regtostr(tmp2, srcostr, wide);
if(tmp1 == Di_Ad)
{
displacement = bytes[i + 2] ^ (bytes[i + 3] << 4);
eactostr(tmp1, dstostr, displacement, 1, 1);
} else {
eactostr(tmp1, dstostr, 0, 0, 0);
}
} else {
instructions[iindx].Src = tmp1;
instructions[iindx].Dst = tmp2;
insttostr(opcode, inststr);
regtostr(tmp2, dstostr, wide);
if(tmp1 == Di_Ad)
{
displacement = bytes[i + 2] ^ (bytes[i + 3] << 4);
eactostr(tmp1, srcostr, displacement, 1, 1);
} else {
eactostr(tmp1, srcostr, 0, 0, 0);
}
}
break;
case 0b01: // register to memory, 8-bit disp
tmp1 = bytes[i + 1] & 0b111; // R/M field
tmp2 = (bytes[i + 1] >> 3) & 0b111; // REG field
displacement = bytes[i + 2];
bytes_used += 2;
if(!direction)
{
insttostr(opcode, inststr);
regtostr(tmp2, srcostr, wide);
eactostr(tmp1, dstostr, displacement, 1, 0);
} else {
insttostr(opcode, inststr);
regtostr(tmp2, dstostr, wide);
eactostr(tmp1, srcostr, displacement, 1, 0);
}
break;
case 0b10: // register to memory, 16-bit disp
tmp1 = bytes[i + 1] & 0b111; // R/M field
tmp2 = (bytes[i + 1] >> 3) & 0b111; // REG field
displacement = bytes[i + 2] ^ (bytes[i + 3] << 4); // get displacement bytes
bytes_used += 3;
if(!direction)
{
insttostr(opcode, inststr);
regtostr(tmp2, srcostr, wide);
eactostr(tmp1, dstostr, displacement, 1, 0);
} else {
insttostr(opcode, inststr);
regtostr(tmp2, dstostr, wide);
eactostr(tmp1, srcostr, displacement, 1, 0);
}
break;
//case 0b11:
default: // register to register
tmp1 =(bytes[i + 1] >> 3) & 0b111; // REG field
tmp2 = bytes[i + 1] & 0b111; // R/M field
if(!direction) // 0 = REG is source
{
insttostr(opcode, inststr);
regtostr(tmp1, srcostr, wide);
regtostr(tmp2, dstostr, wide);
} else {
insttostr(opcode, inststr);
regtostr(tmp2, srcostr, wide);
regtostr(tmp1, dstostr, wide);
}
printf("%s %s, %s\n", inststr, dstostr, srcostr);
break;
}
break;
default:
instructions[iindx].OpCode = UNDEFINED;
++bytes_used;
break;
}
}
@ -260,16 +322,15 @@ int main(int argc, char** argv)
}
rewind(f);
unsigned char* bytes = calloc(1, fsize);
unsigned char* bytes = calloc(1, fsize); // TODO: check if allocation failed
const size_t bytes_read = fread(bytes, 1, fsize, f);
Instruction* instructions = calloc(sizeof(Instruction), bytes_read / 2);
printf("; disassembly for file %s\nbits 16\n\n", argv[argi]);
parse_instructions(bytes, bytes_read, instructions);
print_instructions(bytes, bytes_read);
print_instructions(instructions, bytes_read / 2, argv[argi]);
printf("\n");
free(instructions);
free(bytes);
LOOP_END_NOFREE_CLOSE:
fclose(f);