forked from Imagelibrary/rtems
68baf9eead
Many files in this directory were by persons and organizations that have previously given relicensing permission. git history was used to confirm authorship and contributions on files which had unclear or missing copyrights. Update #3053.
268 lines
8.0 KiB
C
268 lines
8.0 KiB
C
/* SPDX-License-Identifier: BSD-2-Clause */
|
|
|
|
/**
|
|
* @file
|
|
* @brief Block device partition management.
|
|
*/
|
|
|
|
/*
|
|
* Copyright (C) 2009,2020 embedded brains GmbH & Co. KG
|
|
*
|
|
* Redistribution and use in source and binary forms, with or without
|
|
* modification, are permitted provided that the following conditions
|
|
* are met:
|
|
* 1. Redistributions of source code must retain the above copyright
|
|
* notice, this list of conditions and the following disclaimer.
|
|
* 2. Redistributions in binary form must reproduce the above copyright
|
|
* notice, this list of conditions and the following disclaimer in the
|
|
* documentation and/or other materials provided with the distribution.
|
|
*
|
|
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
|
|
* AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
|
|
* IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
|
|
* ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
|
|
* LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
|
|
* CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
|
|
* SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
|
|
* INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
|
|
* CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
|
|
* ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
|
|
* POSSIBILITY OF SUCH DAMAGE.
|
|
*/
|
|
|
|
#ifdef HAVE_CONFIG_H
|
|
#include "config.h"
|
|
#endif
|
|
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
|
|
#include <rtems/bdpart.h>
|
|
#include <rtems/error.h>
|
|
#include <rtems/shell.h>
|
|
|
|
#define RTEMS_BDPART_SHELL_ERROR( fmt, ...) \
|
|
do { \
|
|
printf( "error: " fmt "\n", ##__VA_ARGS__); \
|
|
return -1; \
|
|
} while (0)
|
|
|
|
#define RTEMS_BDPART_SHELL_ERROR_SC( sc, fmt, ...) \
|
|
if ((sc) != RTEMS_SUCCESSFUL) { \
|
|
printf( "error: " fmt ": %s\n", ##__VA_ARGS__, rtems_status_text( sc)); \
|
|
return -1; \
|
|
}
|
|
|
|
typedef enum {
|
|
RTEMS_BDPART_SHELL_FS,
|
|
RTEMS_BDPART_SHELL_N,
|
|
RTEMS_BDPART_SHELL_MBR,
|
|
RTEMS_BDPART_SHELL_GPT
|
|
} rtems_bdpart_shell_state;
|
|
|
|
static const char rtems_bdpart_shell_usage [] =
|
|
"disk format and utility functions\n"
|
|
"\n"
|
|
"fdisk DISK_NAME\n"
|
|
"\tprints the partition table\n"
|
|
"\n"
|
|
"fdisk DISK_NAME [FS N1 [N2 ... ]] ... [write] [FORMAT]\n"
|
|
"\tcreates a new partition table\n"
|
|
"\n"
|
|
"fdisk DISK_NAME register\n"
|
|
"\tcreates a logical disk for each partition of the disk\n"
|
|
"\n"
|
|
"fdisk DISK_NAME unregister\n"
|
|
"\tdeletes the logical disks associated with the partitions\n"
|
|
"\tof the disk\n"
|
|
"\n"
|
|
"option values:\n"
|
|
"\tDISK_NAME: absolute path to disk device like '/dev/hda'\n"
|
|
"\tN*: weights of positive integers\n"
|
|
"\tFS: 0x00 ... 0xff, fat12, fat16, fat32, data\n"
|
|
"\twrite: write the new partition table to the disk\n"
|
|
"\tFORMAT: mbr [[no]dos], gpt";
|
|
|
|
static int rtems_bdpart_shell_main( int argc, char **argv)
|
|
{
|
|
rtems_status_code sc = RTEMS_SUCCESSFUL;
|
|
rtems_bdpart_format format;
|
|
rtems_bdpart_partition pt [RTEMS_BDPART_PARTITION_NUMBER_HINT];
|
|
unsigned dist [RTEMS_BDPART_PARTITION_NUMBER_HINT];
|
|
size_t count = RTEMS_BDPART_PARTITION_NUMBER_HINT;
|
|
const char *disk_name = NULL;
|
|
bool do_create = false;
|
|
bool do_read = false;
|
|
bool do_write = false;
|
|
bool do_register = false;
|
|
bool do_unregister = false;
|
|
bool do_dump = false;
|
|
|
|
if (argc < 2) {
|
|
puts( rtems_bdpart_shell_usage);
|
|
return -1;
|
|
}
|
|
|
|
disk_name = argv [1];
|
|
|
|
if (argc == 2) {
|
|
do_read = true;
|
|
do_dump = true;
|
|
} else if (argc == 3) {
|
|
/* Check option */
|
|
if (strcmp( argv [2], "register") == 0) {
|
|
do_read = true;
|
|
do_register = true;
|
|
} else if (strcmp( argv [2], "unregister") == 0) {
|
|
do_read = true;
|
|
do_unregister = true;
|
|
} else {
|
|
RTEMS_BDPART_SHELL_ERROR( "unexpected option: %s", argv [2]);
|
|
}
|
|
} else {
|
|
rtems_bdpart_shell_state state = RTEMS_BDPART_SHELL_FS;
|
|
uint8_t current_type = RTEMS_BDPART_MBR_FAT_32;
|
|
size_t i = 0;
|
|
int ai = 0;
|
|
|
|
/* Clear partition table */
|
|
memset( pt, 0, sizeof( pt));
|
|
|
|
/* Default format */
|
|
format.type = RTEMS_BDPART_FORMAT_MBR;
|
|
format.mbr.disk_id = 0;
|
|
format.mbr.dos_compatibility = true;
|
|
|
|
for (ai = 2; ai < argc; ++ai) {
|
|
char *s = argv [ai];
|
|
unsigned long v = 0;
|
|
char *end = NULL;
|
|
|
|
if (strlen( s) == 0) {
|
|
continue;
|
|
} else if (strcmp( s, "write") == 0) {
|
|
do_write = true;
|
|
continue;
|
|
} else if (strcmp( s, "mbr") == 0) {
|
|
state = RTEMS_BDPART_SHELL_MBR;
|
|
format.type = RTEMS_BDPART_FORMAT_MBR;
|
|
continue;
|
|
} else if (strcmp( s, "gpt") == 0) {
|
|
state = RTEMS_BDPART_SHELL_GPT;
|
|
format.type = RTEMS_BDPART_FORMAT_GPT;
|
|
continue;
|
|
}
|
|
|
|
switch (state) {
|
|
case RTEMS_BDPART_SHELL_FS:
|
|
v = strtoul( s, &end, 16);
|
|
if (*end == '\0') {
|
|
if (v <= 0xffU) {
|
|
current_type = (uint8_t) v;
|
|
} else {
|
|
RTEMS_BDPART_SHELL_ERROR( "type value out of range: %s", argv [ai]);
|
|
}
|
|
} else if (strcmp( s, "fat32") == 0) {
|
|
current_type = RTEMS_BDPART_MBR_FAT_32;
|
|
} else if (strcmp( s, "data") == 0) {
|
|
current_type = RTEMS_BDPART_MBR_DATA;
|
|
} else if (strcmp( s, "fat16") == 0) {
|
|
current_type = RTEMS_BDPART_MBR_FAT_16;
|
|
} else if (strcmp( s, "fat12") == 0) {
|
|
current_type = RTEMS_BDPART_MBR_FAT_12;
|
|
} else {
|
|
RTEMS_BDPART_SHELL_ERROR( "unexpected option: %s", argv [ai]);
|
|
}
|
|
state = RTEMS_BDPART_SHELL_N;
|
|
break;
|
|
case RTEMS_BDPART_SHELL_N:
|
|
v = strtoul( s, &end, 10);
|
|
if (*end == '\0') {
|
|
rtems_bdpart_to_partition_type( current_type, pt [i].type);
|
|
dist [i] = v;
|
|
if (i < count) {
|
|
++i;
|
|
} else {
|
|
RTEMS_BDPART_SHELL_ERROR( "too many partitions");
|
|
}
|
|
} else {
|
|
--ai;
|
|
state = RTEMS_BDPART_SHELL_FS;
|
|
}
|
|
break;
|
|
case RTEMS_BDPART_SHELL_MBR:
|
|
if (strcmp( s, "dos") == 0) {
|
|
format.mbr.dos_compatibility = true;
|
|
} else if (strcmp( s, "nodos") == 0) {
|
|
format.mbr.dos_compatibility = false;
|
|
} else {
|
|
RTEMS_BDPART_SHELL_ERROR( "unexpected option: %s", argv [ai]);
|
|
}
|
|
break;
|
|
case RTEMS_BDPART_SHELL_GPT:
|
|
RTEMS_BDPART_SHELL_ERROR( "unexpected option: %s", argv [ai]);
|
|
default:
|
|
RTEMS_BDPART_SHELL_ERROR( "fdisk interal error");
|
|
}
|
|
}
|
|
|
|
/* Partition number */
|
|
count = i;
|
|
|
|
/* Actions */
|
|
do_create = true;
|
|
do_dump = true;
|
|
if (do_write) {
|
|
do_read = true;
|
|
}
|
|
}
|
|
|
|
if (do_create) {
|
|
/* Create partitions */
|
|
sc = rtems_bdpart_create( disk_name, &format, pt, dist, count);
|
|
RTEMS_BDPART_SHELL_ERROR_SC( sc, "cannot create partitions for '%s'", disk_name);
|
|
}
|
|
|
|
if (do_write) {
|
|
/* Write partitions */
|
|
sc = rtems_bdpart_write( disk_name, &format, pt, count);
|
|
RTEMS_BDPART_SHELL_ERROR_SC( sc, "cannot write partitions to '%s'", disk_name);
|
|
}
|
|
|
|
if (do_read) {
|
|
/* Read partitions */
|
|
count = RTEMS_BDPART_PARTITION_NUMBER_HINT;
|
|
sc = rtems_bdpart_read( disk_name, &format, pt, &count);
|
|
RTEMS_BDPART_SHELL_ERROR_SC( sc, "cannot read partitions from '%s'", disk_name);
|
|
}
|
|
|
|
if (do_register) {
|
|
/* Register partitions */
|
|
sc = rtems_bdpart_register( disk_name, pt, count);
|
|
RTEMS_BDPART_SHELL_ERROR_SC( sc, "cannot register partitions of '%s'", disk_name);
|
|
}
|
|
|
|
if (do_unregister) {
|
|
/* Unregister partitions */
|
|
sc = rtems_bdpart_unregister( disk_name, pt, count);
|
|
RTEMS_BDPART_SHELL_ERROR_SC( sc, "cannot unregister partitions of '%s'", disk_name);
|
|
}
|
|
|
|
if (do_dump) {
|
|
/* Dump partitions */
|
|
rtems_bdpart_dump( pt, count);
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
struct rtems_shell_cmd_tt rtems_shell_FDISK_Command = {
|
|
.name = "fdisk",
|
|
.usage = rtems_bdpart_shell_usage,
|
|
.topic = "files",
|
|
.command = rtems_bdpart_shell_main,
|
|
.alias = NULL,
|
|
.next = NULL
|
|
};
|