cmd.c

fdisk source code, 是一个完整的fdisk的工具。

    int index=3;

    do
      {
      if( ( (flags.version==FOUR)
       || (flags.version==FIVE)
       || (flags.version==SIX) )
       && (pDrive->pri_part[index].num_type==5) )
	{
	Delete_Primary_Partition(index);
	break;
	}
      if( ( (flags.version==W95)
       || (flags.version==W95B)
       || (flags.version==W98) )
       && ( (pDrive->pri_part[index].num_type==5)
       || (pDrive->pri_part[index].num_type==15) ) )
        {
        Delete_Primary_Partition(index);
        break;
        }

      index--;
      }while(index>=0);

    if(index<0)
      {
      printf("\nExtended DOS Partition not found...no partition deleted.\n");
      exit(9);
      }
    }

  /* Delete a Logical DOS Drive */
  if(0==strcmp(arg[1].choice,"LOG"))
    {
    if( (arg[1].value>=1) && (arg[1].value<=23) )
      {
      Delete_Logical_Drive( (int)(arg[1].value-1) );
      }
    else 
      {
      printf("\nLogical drive number  (%d) is out of range...Operation Terminated\n",arg[1].value);
      exit(9);
      }
    }

  /* Delete the partition by the number of the partition */
  if(0==strcmp(arg[1].choice,"NUM"))
    {
    if( (arg[1].value>=1) && (arg[1].value<=4) )
      {
      Delete_Primary_Partition( (int)(arg[1].value-1) );
      }
    else if( (arg[1].value>=5) && (arg[1].value<=28) )
      {
      Delete_Logical_Drive( (int)(arg[1].value-5) );
      }
    else  
      {
      printf("\nPartition number is out of range...Operation Terminated\n");
      exit(9);
      }
    }

  Shift_Command_Line_Options(2);
}

/* /INFO command line option */
void Command_Line_Info()
{
  int option_count=1;

  if(0==strcmp(arg[1].choice,"TECH"))
    {
    option_count=2;

    flags.extended_options_flag=TRUE;
    }

  Display_CL_Partition_Table();

  Shift_Command_Line_Options(option_count);
}

/* /MODIFY command line option */
void Command_Line_Modify()
{
/*
  if((arg[0].value<1) || (arg[0].value>4))
    {
    printf("\nPrimary partition number is out of range...Operation Terminated.\n");
    exit(9);
    }
*/
  if((arg[0].extra_value<=0) || (arg[0].extra_value>255))
    {
    printf("\nNew partition type is out of range...Operation Terminated.\n");
    exit(9);
    }

  Modify_Partition_Type((int)(arg[0].value-1),arg[0].extra_value);

  Shift_Command_Line_Options(1);
}

/* /MOVE command line option */
void Command_Line_Move()
{
  if((arg[0].value<1) || (arg[0].value>4))
    {
    printf("\Source partition number is out of range...Operation Terminated.\n");
    exit(9);
    }

  if((arg[0].extra_value<1) || (arg[0].extra_value>4))
    {
    printf("\Destination partition number is out of range...Operation Terminated.\n");
    exit(9);
    }

  Primary_Partition_Slot_Transfer(MOVE,(int)arg[0].value,arg[0].extra_value);

  Shift_Command_Line_Options(1);
}

/* /SETFLAG command line option */
void Command_Line_Set_Flag()
{
  if( (arg[0].value<1) || (arg[0].value>64) )
    {
    printf("\nInvalid flag number...Operation Terminated.\n");

    exit(9);
    }

  if(arg[0].extra_value==0) arg[0].extra_value=1;

  if( (arg[0].extra_value<1) || (arg[0].extra_value>64) )
    {
    printf("\nFlag value is out of range...Operation Terminated.\n");

    exit(9);
    }

  Set_Flag((int)arg[0].value,arg[0].extra_value);

  printf("\nFlag %d has been set to ",arg[0].value);
  printf("%d.\n",arg[0].extra_value);

  Shift_Command_Line_Options(1);
}

/* /STATUS command line option */
void Command_Line_Status()
{
  flags.monochrome=TRUE;
  textcolor(7);
  Clear_Screen(0);
  Print_Centered(1,"Fixed Disk Drive Status",0);
  Display_All_Drives();

  Shift_Command_Line_Options(1);
}

/* /SWAP command line option */
void Command_Line_Swap()
{
  if((arg[0].value<1) || (arg[0].value>4))
    {
    printf("\nSource partition number is out of range...Operation Terminated.\n");
    exit(9);
    }

  if((arg[0].extra_value<1) || (arg[0].extra_value>4))
    {
    printf("\nDestination partition number is out of range...Operation Terminated.\n");
    exit(9);
    }

  Primary_Partition_Slot_Transfer(SWAP,(int)arg[0].value,arg[0].extra_value);

  Shift_Command_Line_Options(1);
}


/* /TESTFLAG command line option */
void Command_Line_Test_Flag()
{
  int flag;

  flag=Test_Flag((int)arg[0].value);

  if(arg[0].extra_value>0)
    {
    /* If testing for a particular value, return a true or false answer. */
    /* The error level returned will be 20 for false and 21 for true.    */

    if(flag==arg[0].extra_value)
      {
      printf("\nFlag %d is set to ",arg[0].value);
      printf("%d.\n",arg[0].extra_value);

      exit(21);
      }
    else
      {
      printf("\nFlag %d is not set to ",arg[0].value);
      printf("%d.\n",arg[0].extra_value);

      exit(20);
      }
    }
  else
    {
    /* If not testing the flag for a particular value, return the value */
    /* the flag is set to.  The error level returned will be the value  */
    /* of the flag + 30.                                                */

    printf("\nFlag %d is set to ",arg[0].value);
    printf("%d.\n",flag);

    exit( (30+flag) );
    }
}

/* /X command line option */
void Command_Line_X()
{
  int index;

  /* Ask the user if FAT32 is desired. */
  if( (flags.version==W95B) || (flags.version==W98) )
   Ask_User_About_FAT32_Support();

  flags.use_extended_int_13=FALSE;
  index=0;
  do
    {
    part_table[index].ext_int_13=FALSE;
    index++;
    }while(index<8);

  Read_Partition_Tables();
  Interactive_User_Interface();
}

/* Get the command line options */
int Get_Options(char *argv[],int argc)
{
  char *argptr;
  int i;
  int number_of_options = 0;

  flags.drive_number=0x80;
  
  argc--, argv++;		/* absorb program name */
  

  for (i = 0; i < 20; i++)
	{
  	strcpy(arg[i].choice,"");
  	arg[i].value = 0;
  	arg[i].extra_value = 0;
  	}

  
  for ( ; argc > 0; number_of_options++,argv++, argc--)
  	{
	if (number_of_options >= 20)
	  break;

	/* Limit the possible number of options to 20 to prevent an overflow of */
	/* the arg[] structure.                                                 */


	argptr = *argv;


    if(1==strlen(argptr))
      {
      if (!isdigit(*argptr))
			{ printf("<%s> should be a digit; terminated\n",argptr); exit(9);}

      flags.drive_number=(argptr[0]-'0')+127;
      flags.using_default_drive_number=FALSE;
      number_of_options--;
      continue;
      }

    if (*argptr != '-' && *argptr != '/')
			{ printf("<%s> should start with '-' or '/'; terminated\n",argptr); exit(9);}

    argptr++;			/* skip -/ */

				/* parse commandline
					/ARGUMENT:number,number   */
    for (i = 0; ; argptr++,i++)
	{
	if (!isalpha(*argptr) && *argptr != '_')
		break;

	if (i < sizeof(arg[0].choice)-1)
		{
		arg[number_of_options].choice[i]   = toupper(*argptr);
		arg[number_of_options].choice[i+1] = 0;
			}
	}

	if (*argptr == 0)		/* done */
		continue;

	if (*argptr != ':')
			{ printf("<%s> ':' expected; terminated\n",argptr); exit(9);}

	argptr++;

	arg[number_of_options].value = atol(argptr);

    while(isdigit(*argptr))	/* skip number */
	argptr++;


	if (*argptr == 0)		/* done */
		continue;

	if (*argptr != ',')
			{ printf("<%s> ',' expected; terminated\n",argptr); exit(9);}

	argptr++;

	arg[number_of_options].extra_value = (int)atol(argptr);

    while(isdigit(*argptr))	/* skip number */
	argptr++;

	if (*argptr != 0)		/* done */
			{ printf("<%s> expected end of string; terminated\n",argptr); exit(9);}
	}


  /* check to make sure the drive is a legitimate number */
  if( (flags.drive_number < 0x80)
   || (flags.drive_number > flags.maximum_drive_number) )
    {
    printf("\nInvalid drive designation...Operation Terminated.\n");
    exit(5);
    }


  return(number_of_options);
}

void Shift_Command_Line_Options(int number_of_places)
{
  int index;

  for (index=0;index < 20 - number_of_places; index++)
      {
      strcpy(arg[index].choice, arg[index+number_of_places].choice);
      arg[index].value=arg[index+number_of_places].value;
      arg[index].extra_value=arg[index+number_of_places].extra_value;
      }

  number_of_command_line_options-=number_of_places;
}