Reversing gods | (source code and compiled exe) |
/*
MEM1.C -- walks DOS MCB chain(s): detailed version
Andrew Schulman and Jim Kyle, July 1990
*/
#include <stdlib.h>
#include <stdio.h>
#include <ctype.h>
#include <string.h>
#include <dos.h>
typedef enum { FALSE, TRUE } BOOL;
typedef unsigned char BYTE;
typedef unsigned short WORD;
typedef unsigned long ULONG;
typedef void far *FP;
#ifndef MK_FP
#define MK_FP(seg,ofs) ((FP)(((ULONG)(seg) << 16) | (ofs)))
#endif
#ifdef __TURBOC__
#define ASM asm
#else
#define ASM _asm
#endif
#ifdef __TURBOC__
#define GETVECT(x) getvect(x)
#else
#define GETVECT(x) _dos_getvect(x)
#endif
#pragma pack(1)
typedef struct {
BYTE type; /* 'M'=in chain; 'Z'=at end */
WORD owner; /* PSP of the owner */
WORD size; /* in 16-byte paragraphs */
BYTE unused[3];
BYTE dos4[8];
} MCB;
#define MCB_FM_SEG(seg) ((seg) - 1)
#define IS_PSP(mcb) (FP_SEG(mcb) + 1 == (mcb)->owner)
#define ENV_FM_PSP(psp_seg) (*((WORD far *) MK_FP(psp_seg, 0x2c)))
void fail(char *s) { puts(s); exit(1); }
BOOL belongs(void far *vec, unsigned start, unsigned size);
void display(MCB far *mcb);
char far *env(MCB far *mcb);
void display_progname(MCB far *mcb);
void display_cmdline(MCB far *mcb);
void display_vectors(MCB far *mcb);
unsigned fstrlen(char far *s);
MCB far *get_mcb(void)
{
ASM mov ah, 52h
ASM int 21h
ASM mov dx, es:[bx-2]
ASM xor ax, ax
/* in both Microsoft C and Turbo C, far* returned in DX:AX */
}
mcb_chk(MCB far *mcb)
{
for (;;)
if (mcb->type == 'M')
mcb = MK_FP(FP_SEG(mcb) + mcb->size + 1, 0);
else
return (mcb->type == 'Z');
}
void walk(MCB far *mcb)
{
printf("Seg Owner Size\n");
for (;;)
switch (mcb->type)
{
case 'M' : /* Mark : belongs to MCB chain */
display(mcb);
mcb = MK_FP(FP_SEG(mcb) + mcb->size + 1, 0);
break;
case 'Z' : /* Zbikowski : end of MCB chain */
display(mcb);
return;
default :
fail("error in MCB chain");
}
}
main(int argc, char *argv[])
{
if (argc < 2)
walk(get_mcb()); /* walk "normal" MCB chain */
else
{
unsigned seg;
sscanf(argv[1], "%04X", &seg);
walk(MK_FP(seg, 0)); /* walk arbitrary MCB chain */
}
return 0;
}
void display(MCB far *mcb)
{
static void far *vect_2e = (void far *) 0;
unsigned env_seg;
printf("%04X %04X %04X (%6lu) ",
FP_SEG(mcb), mcb->owner, mcb->size, (long) mcb->size << 4);
if (IS_PSP(mcb))
{
void far *e = env(mcb); /* MSC wants lvalue */
if (env_seg = FP_SEG(e)) printf("%04X ", env_seg);
else printf(" ");
display_progname(mcb);
}
if (! vect_2e)
vect_2e = GETVECT(0x2e); /* do just once */
if (! mcb->owner)
printf("free ");
/* 0008 is not really a PSP; belongs to CONFIG.SYS */
else if (mcb->owner == 8)
printf("config ");
/* INT 2Eh belongs to master COMMAND.COM (or other shell) */
else if (belongs(vect_2e, FP_SEG(mcb), mcb->size))
printf("%s ", getenv("COMSPEC"));
/* presence command line is independent of program name */
if (IS_PSP(mcb))
display_cmdline(mcb);
display_vectors(mcb);
printf("\n");
}
char far *env(MCB far *mcb)
{
char far *e;
unsigned env_mcb;
unsigned env_owner;
/*
if the MCB owner is one more than the MCB segment then
psp := MCB owner
env_seg := make_far_pointer(psp, 2Ch)
e := make_far_pointer(env_seg, 0)
else
return NULL
*/
if (IS_PSP(mcb))
e = MK_FP(ENV_FM_PSP(mcb->owner), 0);
else
return (char far *) 0;
/*
Does this environment really belong to this PSP? An
environment is just another memory block, so its MCB is
located in the preceding paragraph. Make sure the env
MCB's owner is equal to the PSP whose environment this
supposedly is! Thanks to Rob Adams of Phar Lap Software
for pointing out the need for this check; this is a
good example of the sort of consistency check one must
do when working with undocumented DOS.
*/
env_mcb = MCB_FM_SEG(FP_SEG(e));
env_owner = ((MCB far *) MK_FP(env_mcb, 0))->owner;
return (env_owner == mcb->owner) ? e : (char far *) 0;
}
char far *progname_fm_psp(unsigned psp)
{
char far *e;
unsigned len;
/* is there an environment? */
if (! (e = env(MK_FP(MCB_FM_SEG(psp), 0))))
return (char far *) 0;
/* program name only available in DOS 3+ */
if (_osmajor >= 3)
{
/* skip past environment variables */
do e += (len = fstrlen(e)) + 1;
while (len);
/*
e now points to WORD containing number of strings following
environment; check for reasonable value: signed because
could be FFFFh; will normally be 1
*/
if ((*((signed far *) e) >= 1) && (*((signed far *) e) < 10))
{
e += sizeof(signed);
if (isalpha(*e))
return e; /* could make canonical with INT 21h AH=60h */
}
}
return (char far *) 0;
}
void display_progname(MCB far *mcb)
{
char far *s;
if (IS_PSP(mcb))
if (s = progname_fm_psp((FP_SEG(mcb) + 1)))
printf("%Fs ", s);
}
BOOL belongs(void far *vec, unsigned start, unsigned size)
{
unsigned seg = FP_SEG(vec) + (FP_OFF(vec) >> 4); /* normalize */
return (seg >= start) && (seg <= (start + size));
}
void display_cmdline(MCB far *mcb)
{
/*
psp := MCB owner
cmdline_len := psp[80h]
cmdline := psp[81h]
print cmdline (display width := cmdline_len)
*/
int len = *((BYTE far *) MK_FP(mcb->owner, 0x80));
char far *cmdline = MK_FP(mcb->owner, 0x81);
printf("%.*Fs ", len, cmdline);
}
void display_vectors(MCB far *mcb)
{
static void far **vec = (void far **) 0;
WORD vec_seg;
int i;
int did_one=0;
if (! vec)
{
if (! (vec = calloc(256, sizeof(void far *))))
fail("insufficient memory");
for (i=0; i<256; i++)
vec[i] = GETVECT(i);
}
for (i=0; i<256; i++)
if (vec[i] && belongs(vec[i], FP_SEG(mcb), mcb->size))
{
if (! did_one) { did_one++; printf("["); }
printf("%02X ", i);
vec[i] = 0;
}
if (did_one) printf("]");
}
unsigned fstrlen(char far *s)
{
#if defined(_MSC_VER) && (_MSC_VER >= 600)
return _fstrlen(s);
#else
unsigned len = 0;
while (*s++)
len++;
return len;
#endif
}
/*
MEM2.C -- walks DOS MCB chain(s): simple version
Andrew Schulman and Jim Kyle, July 1990
*/
#include <stdlib.h>
#include <stdio.h>
#include <dos.h>
typedef unsigned char BYTE;
typedef unsigned short WORD;
typedef unsigned long ULONG;
typedef void far *FP;
#ifndef MK_FP
#define MK_FP(seg,ofs) ((FP)(((ULONG)(seg) << 16) | (ofs)))
#endif
#ifdef __TURBOC__
#define ASM asm
#else
#define ASM _asm
#endif
#pragma pack(1)
typedef struct {
BYTE type; /* 'M'=in chain; 'Z'=at end */
WORD owner; /* PSP of the owner */
WORD size; /* in 16-byte paragraphs */
BYTE unused[3];
BYTE dos4[8];
} MCB;
void fail(char *s) { puts(s); exit(1); }
MCB far *get_mcb(void)
{
ASM mov ah, 52h
ASM int 21h
ASM mov dx, es:[bx-2]
ASM xor ax, ax
/* in both Microsoft C and Turbo C, far* returned in DX:AX */
}
mcb_chk(MCB far *mcb)
{
for (;;)
if (mcb->type == 'M')
mcb = MK_FP(FP_SEG(mcb) + mcb->size + 1, 0);
else
return (mcb->type == 'Z');
}
void display(MCB far *mcb)
{
char buf[80];
sprintf(buf, "%04X %04X %04X (%6lu)",
FP_SEG(mcb), mcb->owner, mcb->size, (long) mcb->size << 4);
if (! mcb->owner)
strcat(buf, " free");
puts(buf);
}
void walk(MCB far *mcb)
{
printf("Seg Owner Size\n");
for (;;)
switch (mcb->type)
{
case 'M' : /* Mark : belongs to MCB chain */
display(mcb);
mcb = MK_FP(FP_SEG(mcb) + mcb->size + 1, 0);
break;
case 'Z' : /* Zbikowski : end of MCB chain */
display(mcb);
return;
default :
fail("error in MCB chain");
}
}
#ifdef TRY_BUG
main(void)
{
unsigned segm;
ASM mov ah, 48h /* Allocate Memory Block */
ASM mov bx, 64h /* get 100 paragraphs */
ASM int 21h
ASM jc done
/* ax now holds initial segment of allocated block */
ASM mov segm, ax
printf("before: "); display(MK_FP(segm - 1, 0));
ASM mov ax, segm
ASM mov es, ax /* now resize the block */
ASM mov ah, 4Ah /* Resize Memory Block */
ASM mov bx, 0FFFFh /* impossible (at least in real mode!) */
ASM int 21h
ASM jnc done /* something seriously wrong if _didn't_ fail! */
printf("after: "); display(MK_FP(segm - 1, 0));
done:
return 0;
}
#else
main(int argc, char *argv[])
{
if (argc < 2)
walk(get_mcb()); /* walk "normal" MCB chain */
else
{
unsigned seg;
sscanf(argv[1], "%04X", &seg);
walk(MK_FP(seg, 0)); /* walk arbitrary MCB chain */
}
if (! mcb_chk(get_mcb()))
{
/* maybe do stack backtrace here, or dump registers */
puts("Error in MCB chain - prepare for halt...");
getchar();
}
else
puts("MCB chain ok");
return 0;
}
#endif
+ORC