#include "dofat.c"
#include <stdio.h>
//#include <string.h>
#include <ctype.h>
#include <dos.h>

//#include <iostream.h>
//char showfile[256];
//char *MBRF = "C:\\MYPROG.MBR";
char mp[512];
/* char *mp=0x500; */
/* *mp='a';  */
/* mp[1]='b'; */
int i;
FILE *f;

void ProcessPhysicalSector(p,StartCyl,StartSec,HeadNo,DriveType,OperateType,SectorNumber)
char *p;
unsigned char StartCyl,StartSec,HeadNo,DriveType,OperateType,SectorNumber;
{
  asm push es
  asm push ds
  asm pop es
  asm mov bx,p
  asm mov ch,StartCyl
  asm mov cl,StartSec
  asm mov dh,HeadNo
  asm mov dl,DriveType
  asm mov ah,OperateType
  asm mov al,SectorNumber
  asm int 0x13
  asm pop es
};

/*
int YesNo(char *s, char *filename)
//char *s,fil;
{
  //getchar();
  char *c="\0";

  printf("%s %s (Y/N)?",s,filename);
  do {
    //c=getchar();
    //c=toupper(c);
    gets(c);
  } while (*c=='\0');
  if (strcmp(strupr(c),"Y")==0) return 1;
  else
  printf("\nNothing done.\n");
  return 0;
}
*/

void help()
{
  printf("\ndosBACK Version 1.0 Copyright (c) 2006 GOTOmsdos EMAIL: tdaim@sina.com\n      Usage  : Read/Write MBR DBR FAT CMOS, decode CMOS key\n      Syntax :\n               dosBACK /R|/W /MBR|/DBR|/FAT|/CMOS FILE\n            or dosBACK /CMOSKEY\n      Switch :\n	  /R - Read to FILE\n          /W - Write from FILE\n        /MBR - Master Boot Record\n        /DBR - DOS Boot Record\n        /FAT - File Allocation Table (FAT16 is supported as well.)\n       /CMOS - Complementary Metal Oxide Semiconductor Of Machine\n    /CMOSKEY - Decode CMOS Key\n");
}


void doboot(char *prompt, int R, unsigned char OperateType, unsigned char HeadNo, char *rw, char *MBRF, char *filename)
//char *MBRF;
{

  //printf("%s\n\n\n",showfile);

  //printf("%s\n\n\n",fi);
  //unsigned i=0;

  if (YesNo(prompt,filename)!=1) return;
  if (R==1)
  {
     ProcessPhysicalSector(mp,0,1,HeadNo,0x80,OperateType,1);
     //for(;i<512;i++)
     //pokeb(0x1000,0x800+i,mp[i]);

if ((f=fopen(MBRF,rw))==NULL) {
   printf("\nFile Error : %s\n",filename);
  help();
  return;
  }

   fwrite(mp,512,1,f);
   fclose(f);

   printf("\nDone.\n");
   return;
}

if ((f=fopen(MBRF,rw))==NULL)
 {printf("\nFile Error : %s\n",filename);
  help();
  return;
  }
      fread(mp,512,1,f);
   fclose(f);

//getchar();
if (YesNo("Are you sure","")!=1) return;
ProcessPhysicalSector(mp,0,1,HeadNo,0x80,OperateType,1);
printf("\nDone!\n");
}

void docmos(char *prompt, int R, char *rw, char *MBRF, char *filename)
{
   int i=0;
   unsigned char buf[256];
   FILE *f;

   if (YesNo(prompt,filename)!=1) return;

  if (R==1) {

   for(;i<256;i++)
   {
   outportb(0x70,i);
buf[i]=inportb(0x71);
}
if((f=fopen(MBRF,rw))==NULL)
   {
   printf("\nFile : Error %s\n",filename);
   help();
   return;
   }
   fwrite(buf,256,1,f);
   printf("\nDone.\n");
return;
}


 if((f=fopen(MBRF,rw))==NULL)
   {
   printf("\nFile : Error %s\n",filename);
   help();
   return;
   }
 fread(buf,256,1,f);

 //getchar();
 if (YesNo("Are you sure","")!=1) return;

   for(;i<256;i++)
   {
   outportb(0x70,i);
outportb(0x71,buf[i]);
}
printf("\nDone!\n");
}

void decode_CMOS_key()
{
   int i;
   char final[9];
   unsigned char value[2];

   outportb(0x70,0x1c);
   value[0]=inportb(0x71);
   outportb(0x70, 0x1d);
   value[1]=inportb(0x71);

   for(i=0;i<4;i++)
   {final[7-i]=(value[0] & 3) + 0x30;
   value[0]=value[0] >> 2;
   }

   for(i=0;i<4;i++)
   {final[3-i]=(value[1] & 3) + 0x30;
   value[1]=value[1] >> 2;
   }

   final[8]=0;

   printf("Your CMOS supervisor's password is: %s\n", final);

   }

   void main(argc,argv)
int argc;
char *argv[];
{
  unsigned char HeadNo,OperateType;
  char filename[256];
  char prompt[256];
  int R=0;
  int W=0;
  int MBR=0;
  int DBR=0;
  int FAT=0;
  int CMOS=0;
  char *rw="";

  switch (argc) {

    case 2 :
    if (strcmp(strupr(argv[1]),"/CMOSKEY")!=0)
    {
    help();
    return;
    }

    printf("\nBOOTCMOS Version 1.0 Copyright (c) 2006 GOTOmsdos EMAIL: tdaim@sina.com\n");
    decode_CMOS_key();
	return;

    case 4 :

    if (strcmp(strupr(argv[1]),"/R")==0) R=1;
    if (strcmp(strupr(argv[1]),"/W")==0) W=1;
    if (strcmp(strupr(argv[2]),"/MBR")==0) MBR=1;
    if (strcmp(strupr(argv[2]),"/DBR")==0) DBR=1;
    if (strcmp(strupr(argv[2]),"/FAT")==0) FAT=1;
    if (strcmp(strupr(argv[2]),"/CMOS")==0) CMOS=1;

    if ((R==0 && W==0) || (MBR==0 && DBR==0 && FAT==0 && CMOS==0))
    {
    help();
    return;
    }

    printf("\ndosBACK Version 1.0 Copyright (c) 2006 GOTOmsdos EMAIL: tdaim@sina.com\n");
    //strcpy(showfile,"\"");
    //strcpy(showfile,argv[2]);
    //strcpy(showfile,"\"");
    memset(filename,0,256);
    strcat(filename,"\"");
    strcat(filename,argv[3]);
    strcat(filename,"\"");
    //printf("%s\n",showfile);

    if (R==1) {OperateType=2;rw="wb+";}
    if (W==1) {OperateType=3;rw="rb";}
    if (MBR==1) HeadNo=0;
    if (DBR==1) HeadNo=1;

    if (R==1 && MBR==1) strcpy(prompt,"Read the main boot record of hard disk to FILE");
    if (W==1 && MBR==1) strcpy(prompt,"Write the main boot record of hard disk from FILE");
    if (R==1 && DBR==1) strcpy(prompt,"Read the boot record of disk C to FILE");
    if (W==1 && DBR==1) strcpy(prompt,"Write the boot record of disk C from FILE");
    if (R==1 && FAT==1) strcpy(prompt,"Read the File Allocation Table of disk C to FILE");
    if (W==1 && FAT==1) strcpy(prompt,"Write the File Allocation Table of disk C from FILE");
    if (R==1 && CMOS==1) strcpy(prompt,"Read the CMOS of machine to FILE");
    if (W==1 && CMOS==1) strcpy(prompt,"Write the CMOS of machine from FILE");

    if(MBR==1 || DBR==1)
    doboot(prompt,R,OperateType,HeadNo,rw,argv[3],filename);


    else if(FAT==1)
    {
    if(R==1) doFat(prompt,2,argv[3],filename);
    else if(W==1) doFat(prompt,3,argv[3],filename);
    }

    else if (CMOS==1)
    docmos(prompt,R,rw,argv[3],filename);

    return;

    default : help();
}
}