/**
* fdisk_get_partition:
* @cxt: context
* @partno: partition number (0 is the first partition)
* @pa: returns data about partition
*
* Reads disklabel and fills in @pa with data about partition @n.
*
* Note that partno may address unused partition and then this function does
* not fill anything to @pa. See fdisk_is_partition_used(). If @pa points to
* NULL then the function allocates a newly allocated fdisk_partition struct,
* use fdisk_unref_partition() to deallocate.
*
* Returns: 0 on success, otherwise, a corresponding error.
*/
int fdisk_get_partition(struct fdisk_context *cxt, size_t partno,
struct fdisk_partition **pa)
{
int rc;
struct fdisk_partition *np = NULL;
if (!cxt || !cxt->label || !pa)
return -EINVAL;
if (!cxt->label->op->get_part)
return -ENOSYS;
if (!fdisk_is_partition_used(cxt, partno))
return -EINVAL;
if (!*pa) {
np = *pa = fdisk_new_partition();
if (!*pa)
return -ENOMEM;
} else
fdisk_reset_partition(*pa);
(*pa)->partno = partno;
rc = cxt->label->op->get_part(cxt, partno, *pa);
if (rc) {
if (np) {
fdisk_unref_partition(np);
*pa = NULL;
} else
fdisk_reset_partition(*pa);
} else
(*pa)->size_explicit = 1;
return rc;
}
/**
* fdisk_partition_next_partno:
* @pa: partition
* @cxt: context
* @n: returns partition number
*
* If partno-follow-default (see fdisk_partition_partno_follow_default())
* enabled then returns next expected partno, otherwise use Ask API to ask user
* for the next partno.
*
* Returns: 0 on success, <0 on error
*/
int fdisk_partition_next_partno(
struct fdisk_partition *pa,
struct fdisk_context *cxt,
size_t *n)
{
assert(cxt);
assert(n);
if (pa && pa->partno_follow_default) {
size_t i;
DBG(PART, ul_debugobj(pa, "next partno (follow default)"));
for (i = 0; i < cxt->label->nparts_max; i++) {
if (!fdisk_is_partition_used(cxt, i)) {
*n = i;
return 0;
}
}
return -ERANGE;
} else if (pa && fdisk_partition_has_partno(pa)) {
DBG(PART, ul_debugobj(pa, "next partno (specified=%zu)", pa->partno));
if (pa->partno >= cxt->label->nparts_max)
return -ERANGE;
*n = pa->partno;
} else
return fdisk_ask_partnum(cxt, n, 1);
return 0;
}
/**
* fdisk_delete_all_partitions:
* @cxt: fdisk context
*
* Delete all used partitions from disklabel.
*
* Returns: 0 on success, otherwise, a corresponding error.
*/
int fdisk_delete_all_partitions(struct fdisk_context *cxt)
{
size_t i;
int rc;
if (!cxt || !cxt->label)
return -EINVAL;
for (i = 0; i < cxt->label->nparts_max; i++) {
if (!fdisk_is_partition_used(cxt, i))
continue;
rc = fdisk_delete_partition(cxt, i);
if (rc)
break;
}
return rc;
}
/* returns: 1=0 on success, < 0 on error, 1 if no free/used partition */
int fdisk_ask_partnum(struct fdisk_context *cxt, size_t *partnum, int wantnew)
{
int rc = 0, inchar = 0;
char range[BUFSIZ], *ptr = range;
size_t i, len = sizeof(range), begin = 0, run = 0;
struct fdisk_ask *ask = NULL;
__typeof__(ask->data.num) *num;
assert(cxt);
assert(cxt->label);
assert(partnum);
if (cxt->label && cxt->label->flags & FDISK_LABEL_FL_INCHARS_PARTNO)
inchar = 1;
DBG(ASK, dbgprint("%s: asking for %s partition number "
"(max: %zu, inchar: %s)",
cxt->label->name,
wantnew ? "new" : "used",
cxt->label->nparts_max,
inchar ? "yes" : "not"));
ask = fdisk_new_ask();
if (!ask)
return -ENOMEM;
fdisk_ask_set_type(ask, FDISK_ASKTYPE_NUMBER);
num = &ask->data.num;
ask->data.num.inchars = inchar ? 1 : 0;
for (i = 0; i < cxt->label->nparts_max; i++) {
int used = fdisk_is_partition_used(cxt, i);
if (wantnew && !used) {
ptr = mk_string_list(ptr, &len, &begin, &run, i, inchar);
if (!ptr) {
rc = -EINVAL;
break;
}
if (!num->low)
num->dfl = num->low = i + 1;
num->hig = i + 1;
} else if (!wantnew && used) {
ptr = mk_string_list(ptr, &len, &begin, &run, i, inchar);
if (!num->low)
num->low = i + 1;
num->dfl = num->hig = i + 1;
}
}
DBG(ASK, dbgprint("ask limits: low: %ju, high: %ju, default: %ju",
num->low, num->hig, num->dfl));
if (!rc && !wantnew && num->low == num->hig) {
if (num->low > 0) {
/* only one existing partiton, don't ask, return the number */
fdisk_ask_number_set_result(ask, num->low);
fdisk_info(cxt, _("Selected partition %ju"), num->low);
} else if (num->low == 0) {
fdisk_warnx(cxt, _("No partition is defined yet!"));
rc = 1;
}
goto dont_ask;
}
if (!rc && wantnew && num->low == num->hig) {
if (num->low > 0) {
/* only one free partition, don't ask, return the number */
fdisk_ask_number_set_result(ask, num->low);
fdisk_info(cxt, _("Selected partition %ju"), num->low);
}
if (num->low == 0) {
fdisk_warnx(cxt, _("No free partition available!"));
rc = 1;
}
goto dont_ask;
}
if (!rc) {
mk_string_list(ptr, &len, &begin, &run, -1, inchar); /* terminate the list */
rc = fdisk_ask_number_set_range(ask, range);
}
if (!rc)
rc = fdisk_ask_set_query(ask, _("Partition number"));
if (!rc)
rc = fdisk_do_ask(cxt, ask);
dont_ask:
if (!rc) {
*partnum = fdisk_ask_number_get_result(ask);
if (*partnum)
*partnum -= 1;
}
DBG(ASK, dbgprint("result: %ju [rc=%d]\n", fdisk_ask_number_get_result(ask), rc));
fdisk_free_ask(ask);
return rc;
}
