405 lines
13 KiB
C
Executable File
405 lines
13 KiB
C
Executable File
/* By YZBruh */
|
|
|
|
/**
|
|
* Copyright 2024 Partition Manager
|
|
*
|
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
|
* you may not use this file except in compliance with the License.
|
|
* You may obtain a copy of the License at
|
|
*
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
|
*
|
|
* Unless required by applicable law or agreed to in writing, software
|
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
* See the License for the specific language governing permissions and
|
|
* limitations under the License.
|
|
*/
|
|
|
|
/* force use C std (if default is C++) */
|
|
#if defined(__cplusplus)
|
|
extern "C" {
|
|
#endif
|
|
|
|
/* include needed libs (headers) */
|
|
#include <stdio.h>
|
|
#include <unistd.h>
|
|
#include <stdlib.h>
|
|
#include <sys/stat.h>
|
|
#include <string.h>
|
|
#include <stdbool.h>
|
|
#include <getopt.h>
|
|
#include <errno.h>
|
|
#include <pmt.h>
|
|
#include <pmt-docs.h>
|
|
|
|
/* add value to variables that are added globally and are not worth */
|
|
char *out = NULL;
|
|
char *outdir = NULL;
|
|
char *cust_cxt = NULL;
|
|
char *target_partition = NULL;
|
|
char *target_flash_file = NULL;
|
|
char *partition_type = NULL;
|
|
char *format_fs = NULL;
|
|
bool pmt_use_logical = NULL;
|
|
bool pmt_use_cust_cxt = NULL;
|
|
bool pmt_ab = false;
|
|
bool pmt_logical = false;
|
|
bool pmt_flash = false;
|
|
bool pmt_backup = false;
|
|
bool pmt_format = false;
|
|
bool pmt_force_mode = false;
|
|
|
|
/* classic main function (C binary here xd) */
|
|
int main(int argc, char *argv[])
|
|
{
|
|
/* check argument total */
|
|
if (argc < 2)
|
|
{
|
|
fprintf(stderr, "%s: missing operand\nTry `%s --help' for more information.\n", argv[0], argv[0]);
|
|
exit(44);
|
|
}
|
|
|
|
/* a structure for long arguments... */
|
|
struct option long_options[] = {
|
|
{"backup", no_argument, 0, 'b'},
|
|
{"flash", required_argument, 0, 'F'},
|
|
{"format", required_argument, 0, 'r'},
|
|
{"partition", required_argument, 0, 'p'},
|
|
{"logical", no_argument, 0, 'l'},
|
|
{"out", required_argument, 0, 'o'},
|
|
{"outdir", required_argument, 0, 'd'},
|
|
{"context", required_argument, 0, 'c'},
|
|
{"list", no_argument, 0, 'D'},
|
|
{"force", no_argument, 0, 'f'},
|
|
{"version", no_argument, 0, 'v'},
|
|
{"help", no_argument, 0, 0},
|
|
{"license", no_argument, 0, 'L'},
|
|
{0, 0, 0, 0}
|
|
};
|
|
|
|
/* boolean statements (only valid in this file) to distinguish. and pointer from a shortcut for the symbol rule */
|
|
static bool wiew_help = false;
|
|
static bool wiew_licenses = false;
|
|
static bool wiew_version = false;
|
|
static bool list_partitions = false;
|
|
static bool combo_wiewers = false;
|
|
static bool use_cust_outdir = false;
|
|
static char *opt_symbol = "-";
|
|
static char *common_symbol_rule;
|
|
common_symbol_rule = "When entering the attached argument of an option, an argument of another option type cannot be used. In short, the rule is: there can be no '-' at the beginning of the attached argument.";
|
|
|
|
int opt;
|
|
|
|
/* control for each argument */
|
|
while ((opt = getopt_long(argc, argv, "bF:rp:lo:d:c:DfvL", long_options, NULL)) != -1)
|
|
{
|
|
/* process arguments */
|
|
switch (opt)
|
|
{
|
|
/* backup mode */
|
|
case 'b':
|
|
pmt_backup = true;
|
|
break;
|
|
/* flash mode */
|
|
case 'F':
|
|
target_flash_file = strdup(optarg);
|
|
if (strncmp(target_flash_file, opt_symbol, 1) == 0)
|
|
{
|
|
if (!pmt_force_mode)
|
|
{
|
|
fprintf(stderr, "%s%s%s\n", ANSI_RED, common_symbol_rule, ANSI_RESET);
|
|
exit(19);
|
|
} else exit(19);
|
|
}
|
|
pmt_flash = true;
|
|
break;
|
|
/* format mode */
|
|
case 'r':
|
|
format_fs = strdup(optarg);
|
|
if (strncmp(format_fs, opt_symbol, 1) == 0)
|
|
{
|
|
if (!pmt_force_mode)
|
|
{
|
|
fprintf(stderr, "%s%s%s\n", ANSI_RED, common_symbol_rule, ANSI_RESET);
|
|
exit(19);
|
|
} else exit(19);
|
|
}
|
|
pmt_format = true;
|
|
break;
|
|
/* partition selector option */
|
|
case 'p':
|
|
target_partition = strdup(optarg);
|
|
if (strncmp(target_partition, opt_symbol, 1) == 0)
|
|
{
|
|
if (!pmt_force_mode)
|
|
{
|
|
fprintf(stderr, "%s%s%s\n", ANSI_RED, common_symbol_rule, ANSI_RESET);
|
|
exit(19);
|
|
} else exit(19);
|
|
}
|
|
break;
|
|
/* logical partitions option */
|
|
case 'l':
|
|
check_root();
|
|
check_psf();
|
|
if (pmt_logical)
|
|
{
|
|
pmt_use_logical = true;
|
|
} else {
|
|
if (!pmt_force_mode)
|
|
{
|
|
fprintf(stderr, "This device does not have logical partitions!\n");
|
|
exit(17);
|
|
} else exit(17);
|
|
}
|
|
break;
|
|
/* output file option */
|
|
case 'o':
|
|
out = strdup(optarg);
|
|
if (strncmp(out, opt_symbol, 1) == 0)
|
|
{
|
|
if (!pmt_force_mode)
|
|
{
|
|
fprintf(stderr, "%s%s%s\n", ANSI_RED, common_symbol_rule, ANSI_RESET);
|
|
exit(19);
|
|
} else exit(19);
|
|
}
|
|
break;
|
|
/* output dir option */
|
|
case 'd':
|
|
use_cust_outdir = true;
|
|
outdir = strdup(optarg);
|
|
break;
|
|
/* context selector option */
|
|
case 'c':
|
|
pmt_use_cust_cxt = true;
|
|
cust_cxt = strdup(optarg);
|
|
if (strncmp(cust_cxt, opt_symbol, 1) == 0)
|
|
{
|
|
if (!pmt_force_mode)
|
|
{
|
|
fprintf(stderr, "%s%s%s\n", ANSI_RED, common_symbol_rule, ANSI_RESET);
|
|
exit(19);
|
|
} else exit(19);
|
|
}
|
|
break;
|
|
/* partition lister function */
|
|
case 'D':
|
|
list_partitions = true;
|
|
/* check combo wiewer options and progress */
|
|
if (wiew_version || wiew_help || wiew_licenses) combo_wiewers = true;
|
|
break;
|
|
/* force mode option */
|
|
case 'f':
|
|
pmt_force_mode = true;
|
|
break;
|
|
/* version info option */
|
|
case 'v':
|
|
wiew_version = true;
|
|
/* check combo wiewer options and progress */
|
|
if (list_partitions || wiew_help || wiew_licenses) combo_wiewers = true;
|
|
break;
|
|
/* help message opption */
|
|
case 0:
|
|
wiew_help = true;
|
|
/* check combo wiewer options and progress */
|
|
if (wiew_version || list_partitions || wiew_licenses) combo_wiewers = true;
|
|
break;
|
|
/* license wiewer option */
|
|
case 'L':
|
|
wiew_licenses = true;
|
|
/* check combo wiewer options and progress */
|
|
if (wiew_version || wiew_help || list_partitions) combo_wiewers = true;
|
|
break;
|
|
/* for invalid options */
|
|
case '?':
|
|
printf("Try `%s --help' for more information.\n", argv[0]);
|
|
exit(43);
|
|
break;
|
|
default:
|
|
printf("Usage: %s [-b | --backup] [-f | --flash FILE] [-r | --format FS_TYPE] [-p | --partition PARTITION] [-l | --logical] [-o | --out OUTNAME] [-d | --outdir OUTDIR] [-c | --context] [-D | --list] [-v | --version] [--help] [-L | --license]\n", argv[0]);
|
|
exit(44);
|
|
}
|
|
}
|
|
|
|
/* stop the program if multiple viewer is used */
|
|
if (combo_wiewers)
|
|
{
|
|
fprintf(stderr, "%s: Multiple wiewers cannot be used at the same line.\n", argv[0]);
|
|
exit(81);
|
|
}
|
|
|
|
/* controller to handle viewer */
|
|
if (wiew_help)
|
|
{
|
|
help();
|
|
exit(EXIT_SUCCESS);
|
|
} else if (wiew_version)
|
|
{
|
|
version();
|
|
exit(EXIT_SUCCESS);
|
|
} else if (wiew_licenses)
|
|
{
|
|
licenses();
|
|
exit(EXIT_SUCCESS);
|
|
} else if (list_partitions)
|
|
{
|
|
check_root();
|
|
listpart();
|
|
exit(EXIT_SUCCESS);
|
|
}
|
|
|
|
/* target control is done */
|
|
if (!pmt_backup && !pmt_flash && !pmt_format)
|
|
{
|
|
fprintf(stderr, "%s: missing operand.\nTry `%s --help` for more information.\n", argv[0], argv[0]);
|
|
exit(3);
|
|
}
|
|
|
|
/* prevent multiple mode use */
|
|
if (pmt_backup && pmt_flash)
|
|
{
|
|
if (!pmt_force_mode)
|
|
{
|
|
fprintf(stderr, "Backup and flash functions cannot be used in the same command.\n");
|
|
exit(9);
|
|
} else exit(9);
|
|
}
|
|
|
|
/* checks */
|
|
check_root();
|
|
check_psf();
|
|
|
|
if (pmt_format)
|
|
{
|
|
if (strcmp(format_fs, "ext4") != 0 || strcmp(format_fs, "ext3") != 0 || strcmp(format_fs, "ext2") != 0)
|
|
{
|
|
if (!pmt_force_mode)
|
|
{
|
|
fprintf(stderr, "%s: formatter: unsupported filesystem: %s", argv[0], format_fs);
|
|
exit(41);
|
|
} else exit(41);
|
|
}
|
|
}
|
|
|
|
if (use_cust_outdir)
|
|
{
|
|
if (strncmp(outdir, opt_symbol, 1) == 0)
|
|
{
|
|
if (!pmt_force_mode)
|
|
{
|
|
fprintf(stderr, "%s\n", common_symbol_rule);
|
|
exit(19);
|
|
} else exit(19);
|
|
}
|
|
struct stat out_info;
|
|
if (stat(outdir, &out_info) != 0)
|
|
{
|
|
if (!pmt_force_mode)
|
|
{
|
|
fprintf(stderr, "%s: cannot stat '%s': %s\n", argv[0], outdir, strerror(errno));
|
|
exit(18);
|
|
} else exit(18);
|
|
} else {
|
|
if (!S_ISDIR(out_info.st_mode))
|
|
{
|
|
if (!pmt_force_mode)
|
|
{
|
|
fprintf(stderr, "%s: %s: is a not directory.\n", argv[0], outdir);
|
|
exit(20);
|
|
} else exit(20);
|
|
}
|
|
}
|
|
}
|
|
|
|
if (pmt_flash)
|
|
{
|
|
struct stat flashf_info;
|
|
if (stat(target_flash_file, &flashf_info) != 0)
|
|
{
|
|
if (!pmt_force_mode)
|
|
{
|
|
fprintf(stderr, "%s: cannot stat '%s': %s\n", argv[0], target_flash_file, strerror(errno));
|
|
exit(15);
|
|
} else exit(15);
|
|
} else {
|
|
if (!S_ISREG(flashf_info.st_mode))
|
|
{
|
|
if (!pmt_force_mode)
|
|
{
|
|
fprintf(stderr, "%s: %s: is a not file.\n", argv[0], target_flash_file);
|
|
exit(16);
|
|
} else exit(16);
|
|
}
|
|
}
|
|
}
|
|
|
|
/* custom context checker */
|
|
if (pmt_use_cust_cxt)
|
|
{
|
|
struct stat cxtinfo;
|
|
if (stat(cust_cxt, &cxtinfo) == 0)
|
|
{
|
|
if (!S_ISREG(cxtinfo.st_mode))
|
|
{
|
|
if (!pmt_force_mode)
|
|
{
|
|
fprintf(stderr, "%s: %s: is a not directory.\n", argv[0], cust_cxt);
|
|
exit(8);
|
|
} else exit(8);
|
|
}
|
|
} else {
|
|
if (!pmt_force_mode)
|
|
{
|
|
fprintf(stderr, "%s: %s: %s\n", argv[0], cust_cxt, strerror(errno));
|
|
exit(6);
|
|
} else exit(6);
|
|
}
|
|
if (strstr(cust_cxt, "/dev") == NULL && !pmt_force_mode)
|
|
{
|
|
fprintf(stderr, "%sYou're going through my wave? There's nothing about this /dev. Use force mode if you don't want this error%s\n", ANSI_YELLOW, ANSI_RESET);
|
|
exit(81);
|
|
}
|
|
}
|
|
|
|
if (target_partition == NULL)
|
|
{
|
|
if (!pmt_force_mode)
|
|
{
|
|
fprintf(stderr, "%s: required partition name.\nTry `%s --help' for more information.\n", argv[0], argv[0]);
|
|
exit(5);
|
|
} else exit(5);
|
|
} else {
|
|
/**
|
|
*
|
|
* 1 = backup mode
|
|
*
|
|
* 2 = flash mode
|
|
*
|
|
* 3 = format
|
|
*/
|
|
if (pmt_backup)
|
|
{
|
|
pmt(1);
|
|
exit(EXIT_SUCCESS);
|
|
} else if (pmt_flash)
|
|
{
|
|
pmt(2);
|
|
exit(EXIT_SUCCESS);
|
|
} else if (pmt_format)
|
|
{
|
|
pmt(3);
|
|
} else {
|
|
fprintf(stderr, "%s: no target (backup or flash).\nTry `%s --help` for more information.\n", argv[0], argv[0]);
|
|
exit(3);
|
|
}
|
|
}
|
|
}
|
|
|
|
#if defined(__cplusplus)
|
|
}
|
|
#endif
|
|
|
|
/* end of code */
|