part_aux.c

独立 WIN98下磁盘分区程序（C语言＋汇编） 界面功能类似FDISK 但独立编译

{
 int i=13;	/* 1,111,111,111 */
 tmp[i--]=0;	/* 0 234 678 012 */

 while( num!=0 )
    {
     if( i==9 || i==5 || i==1 ) tmp[i--]=',';
     tmp[i--]=num%10+'0';
     num/=10;
    }

 return tmp+i+1;
}/* sprintf_long */


char *sprintf_systype(char *tmp, int i)
{
 sprintf(tmp," 0x%02X   %-23s     %s     %s      %s ",
               os_desc[i].os_id>>8, os_desc[i].name,
              (os_desc[i].setup!=0)?"Yes":" - ",
              (os_desc[i].format!=0)?"Yes":" - ",
              (os_desc[i].print!=0)?"Yes":" - " );
 return tmp;
}/* sprintf_systype */


char *sprintf_os_name(char *tmp, struct part_long *p)
{
 if( os_desc[p->os_num].os_id != OS_UNKN )
    sprintf(tmp,"%-23s",os_desc[p->os_num].name );
 else
    sprintf(tmp,NAME_OS_UNKN, p->os_id>>8 );

 return tmp;
}/* sprintf_os_name */



char *sprintf_partrec(char *tmp, struct part_long *p, int num, int view)
{
 char tmp1[30], tmp2[30];

 if( mode==MODE_CHS )
   {
    sprintf(tmp,"%2d  %-4s  %s %4ld %4ld %4ld  %4ld %4ld %4ld %10s ",
         num,
         p->active ? (view==VIEW_ADV ? "Menu":" Yes") : " No",
         sprintf_os_name(tmp1,p),
         p->start_cyl,
         p->start_head,
         p->start_sect,
         p->end_cyl,
         p->end_head,
         p->end_sect,
         sprintf_long(tmp2,QUICK_SIZE(p)/2) );
   }
 else /* MODE_LBA */
   {
    sprintf(tmp,"%2d  %-4s  %s%9lu %10lu %10lu %10s ",
         num,
         p->active ? (view==VIEW_ADV ? "Menu":" Yes") : " No",
         sprintf_os_name(tmp1,p),
         p->rel_sect,
         p->num_sect,
         (p->num_sect==0)?(0):(p->rel_sect+p->num_sect-1),
         sprintf_long(tmp2,p->num_sect/2) );
   }
 
 return tmp;         
}/* sprintf_part */


void usage(void)
{
 printf(HELP_CMD_LINE);
 exit(1);
}

void cmd_error(char *msg)
{
 if( !quiet ) fprintf(stderr,"\nError: %s\n",msg);
 exit(1);
}



int prepare_adv_mbr_for_save( struct part_long *part, struct mbr *mbr,
						      struct adv *adv )
{
 int i, j;
 struct part_long h;

 for( i=0 ; i<MAX_PART_ROWS ; i++ )
  if( part[i].os_id==OS_ADV ) break;

 if( i==MAX_PART_ROWS || part[i].num_sect<ADV_NUM_SECTS )
   return 0;

 memmove( mbr->x.adv.code, ADV_IPL, sizeof(mbr->x.adv.code) );
 mbr->x.adv.rel_sect=part[i].rel_sect;
 mbr->x.adv.reserved=0;
 mbr->x.adv.act_menu=0;
 mbr->x.adv.boptions=0;
 mbr->x.adv.adv_mbr_magic=ADV_MBR_MAGIC;
 
 memset(mbr->part_rec,0,sizeof(mbr->part_rec));
 h.active=1;
 h.rel_sect=1;
 h.num_sect=dinfo.total_sects-1;
 h.container_base=0;
 h.container=part[i].container;
 h.os_id=OS_HIDDEN;
 recalculate_part(&h,MODE_LBA);
 pack_part_tab(&h,mbr->part_rec,1);
 
 if( adv->options&ADV_OPT_IGN_UNUSED )
   {
    for( i=0 ; i<MAX_PART_ROWS ; i++ )
     if( adv->part[i].os_id==0 ) 
       {
        for( j=i ; j<MAX_PART_ROWS-1 ; j++ )
          adv->part[j]=adv->part[j+1];
        memset( &adv->part[j], 0, sizeof(struct adv_part_rec) );
       }
   }

 strcpy( mbr->x.adv_old.signature, ADV_DATA_SIGNATURE );
         mbr->x.adv_old.version=ADV_DATA_VERSION;

 strcpy( adv->signature, ADV_DATA_SIGNATURE );
         adv->version=ADV_DATA_VERSION;

 if( strncmp(adv->adv_title,MANAGER_TITLE,sizeof(adv->adv_title)-8)==0 )
     strncpy(adv->adv_title,MANAGER_TITLE,sizeof(adv->adv_title));

 return 1;
}/* prepare_adv_mbr_for_save */





int save_to_file(char *filename, void *buf, int len)
{
 FILE *f;
 
 if( (f=fopen(filename,"wb"))==0 ) return -1;
 
 if( fwrite(buf,1,len,f)!=len ) return -1;
 
 fclose(f);
 return 0;
}/* save_to_file */



int load_from_file(char *filename, void *buf, int len)
{
 FILE *f;
 
 if( (f=fopen(filename,"rb"))==0 ) return -1;
 
 if( fread(buf,1,len,f)!=len ) return -1;
 
 fclose(f);
 return 0;
}/* load_from_file */


void get_base_dir(char *path)
{
 int i;
 char *p, *q;
 
 p=path+strlen(path)-1;
 
 while( p!=path && *p!='\\' && *p!='/' ) p--;
 
 if( *p=='\\' || *p=='/' )
   strncpy(base_dir,path,p-path+1);
 else base_dir[0]=0;

}/* get_base_dir */


void parse_arg(char *str, char **argv, int argv_size)
{
 int i=0;
 char *p=str;
 
 while(1)
    {
     while( *p==' ' || *p=='\t' ) p++;
     if( *p==0 ) break;
     if( i>=argv_size-1 ) break;
     argv[i++]=p;
     while( *p!=' ' && *p!='\t' && *p!=0 ) p++;
     if( *p==0 ) break;
     *p=0;
     p++;
    }

 argv[i]=0;
}/* parse_arg */


void print_part_details(struct part_long *part, int num_rows)
{
 int i;
/*
   HardDisk 1   1236 MB  [ 1023 cyl x 255 heads x 63 sects = 11,111,111 sects ] 

Valid            Starting       Ending    Starting  Numb of   Ending  Partition
 |   # HD  FS  Cyl Head Sct  Cyl Head Sct   sector  sectors   sector  Size [KB]
OK   1 80  FF 1023  255  63 1023  255  63 12345678 12345678 12345678  1,111,111
 X   2  0  FF 1023  255  63 1023  255  63 12345678 12345678 12345678  1,111,111
     3  0  00    0    0   0    0    0   0        0        0
     4  0  00    0    0   0    0    0   0        0        0
*/
 printf("Valid            Starting       Ending    Starting  Numb of   Ending  Partition\n"
        " |   # %3s FS  Cyl Head Sct  Cyl Head Sct   sector  sectors   sector  Size [KB]\n\n",
         num_rows==4?"HD ":"Row" );

 for( i=0 ; i<num_rows ; i++ )
    {
     if( part[i].empty ) continue;

     printf("%s %2d %2X  %02X %4lu  %3lu  %2lu %4lu  %3lu  %2lu %8lu %8lu %8lu %10s\n",
          (part[i].valid)?"OK ":" X ", i+1,
           num_rows==4 ? 0x80 : part[i].orig_row,
           part[i].os_id>>8,
           part[i].start_cyl, part[i].start_head, part[i].start_sect,
           part[i].end_cyl,   part[i].end_head,   part[i].end_sect,
           part[i].rel_sect,  part[i].num_sect,
          (part[i].num_sect==0)?(0):(part[i].rel_sect+part[i].num_sect-1),
           sprintf_long(tmp,part[i].num_sect/2) );
    }

 printf("\n");

 for( i=0 ; i<num_rows ; i++ )
  if( !part[i].empty && !part[i].valid )
    {
     printf("Errors in record %d:",i+1);
     if( part[i].mbr_err ) printf("  mbr");
     if( part[i].range_err ) printf("  range");
     if( part[i].overlap_err ) printf("  overlapped");
     if( part[i].inconsist_err ) printf("  inconsistent");
     if( part[i].boot_record_err ) printf("  boot_record_invalid");
     printf("\n");
    }

}/* print_part_details */



int print_embr(struct part_long *p)
{
 int i, n;
 struct mbr *mbr = (struct mbr*)buf;
 struct part_long *part = malloc(4*sizeof(struct part_long));

 if( part==0 ) cmd_error(ERROR_MALLOC);

 if( disk_read_rel(p,0,mbr,1)==-1 )
   {
    fprintf(stderr,"Warning: %s\n\n",ERROR_READ_MBR);
    return 0;
   }

 printf("EMBR Level %2d      Magic number: 0x%04X\n\n",p->level,mbr->magic_num);

 if( p->level>30 ) cmd_error("This is too deep.");

 unpack_part_tab( mbr->part_rec, part, 4, p );

 validate_table(part,4,p);

 printf("Valid%s\n |   %s\n\n", HEADER_CHS2, HEADER_CHS3 );

 for( i=0 ; i<4 ; i++ )
    {
     sprintf_partrec(tmp, &part[i], i+1, VIEW_EMBR );  tmp[75]=0;
     printf("%s %s\n", part[i].empty?"   ":(part[i].valid?"OK ":" X "),tmp);
    }

 printf("\n");

 print_part_details(part,4);
 
 for( i=0 ; i<4 ; i++ )
  if(!part[i].empty && part[i].valid && part[i].os_id!=0
                                     && part[i].os_id!=OS_ADV )
    {
     printf("\n\n-------------------------------------------------------------------------------\n\n"
            "%s  [  %luM = %s sectors  at  CHS=(%lu,%lu,%lu)  ]\n\n",
             sprintf_os_name(tmp,&part[i]),
             part[i].num_sect/2048, sprintf_long(tmp2,part[i].num_sect),
             part[i].start_cyl, part[i].start_head, part[i].start_sect );
     if( os_desc[part[i].os_num].print!=0 )
         os_desc[part[i].os_num].print(&part[i]);
     else print_unknown(&part[i]);
    }

 free(part);
 return 0;
}/* print_embr */




void convert_adv_data(struct adv *adv)
{
 struct adv_v2
    {
     unsigned char signature[15];		/* AdvBootManager\0 */
     unsigned char version;			/* Data structure version */
     unsigned long rel_sect;
     unsigned char act_menu;
     unsigned char reserved[9];
     unsigned char def_menu;
     unsigned char timeout;
     unsigned char options;
     unsigned char options2;
     unsigned long password;			/* code to validate password */
     unsigned char reserved2[10];
     unsigned char adv_title[32];
     struct adv_menu_rec menu[MAX_MENU_ROWS];	/* 1280 */
     struct adv_part_rec part[MAX_PART_ROWS];	/*  512 */
    }*adv_v2=(struct adv_v2*)adv;

 adv->version  = ADV_DATA_VERSION;
 adv->def_menu = adv_v2->def_menu;
 adv->timeout  = adv_v2->timeout;
 adv->options  = adv_v2->options;
 adv->options2 = adv_v2->options2;
 adv->password = 0;
 memset(adv->reserved2,0,sizeof(adv->reserved2));

 if( interactive )
   {
    show_error("Advanced MBR had an old version. It was converted to the new one.");
/*
    show_error("If you don't want to save it press Ctrl-F10 for unconditional exit.");
*/
   }
 else
   {
    printf("Advanced MBR had an old version. It was converted to the new one.\n");
   }
}/* convert_adv_data */