/* libs/diskconfig/diskconfig.c
*
* Copyright 2008, The Android Open Source Project
*
* 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.
*/
#define LOG_TAG "diskconfig"
#include <errno.h>
#include <fcntl.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <sys/ioctl.h>
#include <sys/stat.h>
#include <linux/fs.h>
#include <cutils/config_utils.h>
#include <cutils/log.h>
#include "diskconfig.h"
static int
parse_len(const char *str, uint64_t *plen)
{
char tmp[64];
int len_str;
uint32_t multiple = 1;
strncpy(tmp, str, sizeof(tmp));
tmp[sizeof(tmp)-1] = '\0';
len_str = strlen(tmp);
if (!len_str) {
LOGE("Invalid disk length specified.");
return 1;
}
switch(tmp[len_str - 1]) {
case 'M': case 'm':
/* megabyte */
multiple <<= 10;
case 'K': case 'k':
/* kilobytes */
multiple <<= 10;
tmp[len_str - 1] = '\0';
break;
default:
break;
}
*plen = strtoull(tmp, NULL, 0);
if (!*plen) {
LOGE("Invalid length specified: %s", str);
return 1;
}
if (*plen == (uint64_t)-1) {
if (multiple > 1) {
LOGE("Size modifier illegal when len is -1");
return 1;
}
} else {
/* convert len to kilobytes */
if (multiple > 1024)
multiple >>= 10;
*plen *= multiple;
if (*plen > 0xffffffffULL) {
LOGE("Length specified is too large!: %llu KB", *plen);
return 1;
}
}
return 0;
}
static int
load_partitions(cnode *root, struct disk_info *dinfo)
{
cnode *partnode;
dinfo->num_parts = 0;
for (partnode = root->first_child; partnode; partnode = partnode->next) {
struct part_info *pinfo = &dinfo->part_lst[dinfo->num_parts];
const char *tmp;
/* bleh, i will leak memory here, but i DONT CARE since
* the only right thing to do when this function fails
* is to quit */
pinfo->name = strdup(partnode->name);
if(config_bool(partnode, "active", 0))
pinfo->flags |= PART_ACTIVE_FLAG;
if (!(tmp = config_str(partnode, "type", NULL))) {
LOGE("Partition type required: %s", pinfo->name);
return 1;
}
/* possible values are: linux */
if (!strcmp(tmp, "linux")) {
pinfo->type = PC_PART_TYPE_LINUX;
} else {
LOGE("Unsupported partition type found: %s", tmp);
return 1;
}
if ((tmp = config_str(partnode, "len", NULL)) != NULL) {
uint64_t len;
if (parse_len(tmp, &len))
return 1;
pinfo->len_kb = (uint32_t) len;
} else
pinfo->len_kb = 0;
++dinfo->num_parts;
}
return 0;
}
struct disk_info *
load_diskconfig(const char *fn, char *path_override)
{
struct disk_info *dinfo;
cnode *devroot;
cnode *partnode;
cnode *root = config_node("", "");
const char *tmp;
if (!(dinfo = malloc(sizeof(struct disk_info)))) {
LOGE("Could not malloc disk_info");
return NULL;
}
memset(dinfo, 0, sizeof(struct disk_info));
if (!(dinfo->part_lst = malloc(MAX_NUM_PARTS * sizeof(struct part_info)))) {
LOGE("Could not malloc part_lst");
goto fail;
}
memset(dinfo->part_lst, 0,
sizeof(MAX_NUM_PARTS * sizeof(struct part_info)));
config_load_file(root, fn);
if (root->first_child == NULL) {
LOGE("Could not read config file %s", fn);
goto fail;
}
if (!(devroot = config_find(root, "device"))) {
LOGE("Could not find device section in config file '%s'", fn);
goto fail;
}
if (!(tmp = config_str(devroot, "path", path_override))) {
LOGE("device path is requried");
goto fail;
}
dinfo->device = strdup(tmp);
/* find the partition scheme */
if (!(tmp = config_str(devroot, "scheme", NULL))) {
LOGE("partition scheme is required");
goto fail;
} else if (!strcmp(tmp, "mbr")) {
dinfo->scheme = PART_SCHEME_MBR;
} else if (!strcmp(tmp, "gpt")) {
LOGE("'gpt' partition scheme not supported yet.");
goto fail;
} else {
LOGE("Unknown partition scheme specified: %s", tmp);
goto fail;
}
/* grab the sector size (in bytes) */
tmp = config_str(devroot, "sector_size", "512");
dinfo->sect_size = strtol(tmp, NULL, 0);
if (!dinfo->sect_size) {
LOGE("Invalid sector size: %s", tmp);
goto fail;
}
/* first lba where the partitions will start on disk */
if (!(tmp = config_str(devroot, "start_lba", NULL))) {
LOGE("start_lba must be provided");
goto fail;
}
if (!(dinfo->skip_lba = strtol(tmp, NULL, 0))) {
LOGE("Invalid starting LBA (or zero): %s", tmp);
goto fail;
}
/* Number of LBAs on disk */
if (!(tmp = config_str(devroot, "num_lba", NULL))) {
LOGE("num_lba is required");
goto fail;
}
dinfo->num_lba = strtoul(tmp, NULL, 0);
if (!(partnode = config_find(devroot, "partitions"))) {
LOGE("Device must specify partition list");
goto fail;
}
if (load_partitions(partnode, dinfo))
goto fail;
return dinfo;
fail:
if (dinfo->part_lst)
free(dinfo->part_lst);
if (dinfo->device)
free(dinfo->device);
free(dinfo);
return NULL;
}
static int
sync_ptable(int fd)
{
struct stat stat;
int rv;
sync();
if (fstat(fd, &stat)) {
LOGE("Cannot stat, errno=%d.", errno);
return -1;
}
if (S_ISBLK(stat.st_mode) && ((rv = ioctl(fd, BLKRRPART, NULL)) < 0)) {
LOGE("Could not re-read partition table. REBOOT!. (errno=%d)", errno);
return -1;
}
return 0;
}
/* This function verifies that the disk info provided is valid, and if so,
* returns an open file descriptor.
*
* This does not necessarily mean that it will later be successfully written
* though. If we use the pc-bios partitioning scheme, we must use extended
* partitions, which eat up some hd space. If the user manually provisioned
* every single partition, but did not account for the extra needed space,
* then we will later fail.
*
* TODO: Make validation more complete.
*/
static int
validate(struct disk_info *dinfo)
{
int fd;
int sect_sz;
uint64_t disk_size;
uint64_t total_size;
int cnt;
struct stat stat;
if (!dinfo)
return -1;
if ((fd = open(dinfo->device, O_RDWR)) < 0) {
LOGE("Cannot open device '%s' (errno=%d)", dinfo->device, errno);
return -1;
}
if (fstat(fd, &stat)) {
LOGE("Cannot stat file '%s', errno=%d.", dinfo->device, errno);
goto fail;
}
/* XXX: Some of the code below is kind of redundant and should probably
* be refactored a little, but it will do for now. */
/* Verify that we can operate on the device that was requested.
* We presently only support block devices and regular file images. */
if (S_ISBLK(stat.st_mode)) {
/* get the sector size and make sure we agree */
if (ioctl(fd, BLKSSZGET, §_sz) < 0) {
LOGE("Cannot get sector size (errno=%d)", errno);
goto fail;
}
if (!sect_sz || sect_sz != dinfo->sect_size) {
LOGE("Device sector size is zero or sector sizes do not match!");
goto fail;
}
/* allow the user override the "disk size" if they provided num_lba */
if (!dinfo->num_lba) {
if (ioctl(fd, BLKGETSIZE64, &disk_size) < 0) {
LOGE("Could not get block device size (errno=%d)", errno);
goto fail;
}
/* XXX: we assume that the disk has < 2^32 sectors :-) */
dinfo->num_lba = (uint32_t)(disk_size / (uint64_t)dinfo->sect_size);
} else
disk_size = (uint64_t)dinfo->num_lba * (uint64_t)dinfo->sect_size;
} else if (S_ISREG(stat.st_mode)) {
LOGI("Requesting operation on a regular file, not block device.");
if (!dinfo->sect_size) {
LOGE("Sector size for regular file images cannot be zero");
goto fail;
}
if (dinfo->num_lba)
disk_size = (uint64_t)dinfo->num_lba * (uint64_t)dinfo->sect_size;
else {
dinfo->num_lba = (uint32_t)(stat.st_size / dinfo->sect_size);
disk_size = (uint64_t)stat.st_size;
}
} else {
LOGE("Device does not refer to a regular file or a block device!");
goto fail;
}
#if 0
LOGV("Device/file %s: size=%llu bytes, num_lba=%u, sect_size=%d",
dinfo->device, disk_size, dinfo->num_lba, dinfo->sect_size);
#endif
/* since this is our offset into the disk, we start off with that as
* our size of needed partitions */
total_size = dinfo->skip_lba * dinfo->sect_size;
/* add up all the partition sizes and make sure it fits */
for (cnt = 0; cnt < dinfo->num_parts; ++cnt) {
struct part_info *part = &dinfo->part_lst[cnt];
if (part->len_kb != (uint32_t)-1) {
total_size += part->len_kb * 1024;
} else if (part->len_kb == 0) {
LOGE("Zero-size partition '%s' is invalid.", part->name);
goto fail;
} else {
/* the partition requests the rest of the disk. */
if (cnt + 1 != dinfo->num_parts) {
LOGE("Only the last partition in the list can request to fill "
"the rest of disk.");
goto fail;
}
}
if (part->type != PC_PART_TYPE_LINUX) {
LOGE("Unknown partition type (0x%x) encountered for partition "
"'%s'\n", part->type, part->name);
goto fail;
}
}
/* only matters for disks, not files */
if (S_ISBLK(stat.st_mode) && total_size > disk_size) {
LOGE("Total requested size of partitions (%llu) is greater than disk "
"size (%llu).", total_size, disk_size);
goto fail;
}
return fd;
fail:
close(fd);
return -1;
}
static int
validate_and_config(struct disk_info *dinfo, int *fd, struct write_list **lst)
{
*lst = NULL;
*fd = -1;
if ((*fd = validate(dinfo)) < 0)
return 1;
switch (dinfo->scheme) {
case PART_SCHEME_MBR:
*lst = config_mbr(dinfo);
return *lst == NULL;
case PART_SCHEME_GPT:
/* not supported yet */
default:
LOGE("Uknown partition scheme.");
break;
}
close(*fd);
*lst = NULL;
return 1;
}
/* validate and process the disk layout configuration.
* This will cause an update to the partitions' start lba.
*
* Basically, this does the same thing as apply_disk_config in test mode,
* except that wlist_commit is not called to print out the data to be
* written.
*/
int
process_disk_config(struct disk_info *dinfo)
{
struct write_list *lst;
int fd;
if (validate_and_config(dinfo, &fd, &lst) != 0)
return 1;
close(fd);
wlist_free(lst);
return 0;
}
int
apply_disk_config(struct disk_info *dinfo, int test)
{
int fd;
struct write_list *wr_lst = NULL;
int rv;
if (validate_and_config(dinfo, &fd, &wr_lst) != 0) {
LOGE("Configuration is invalid.");
goto fail;
}
if ((rv = wlist_commit(fd, wr_lst, test)) >= 0)
rv = test ? 0 : sync_ptable(fd);
close(fd);
wlist_free(wr_lst);
return rv;
fail:
close(fd);
if (wr_lst)
wlist_free(wr_lst);
return 1;
}
int
dump_disk_config(struct disk_info *dinfo)
{
int cnt;
struct part_info *part;
printf("Device: %s\n", dinfo->device);
printf("Scheme: ");
switch (dinfo->scheme) {
case PART_SCHEME_MBR:
printf("MBR");
break;
case PART_SCHEME_GPT:
printf("GPT (unsupported)");
break;
default:
printf("Unknown");
break;
}
printf ("\n");
printf("Sector size: %d\n", dinfo->sect_size);
printf("Skip leading LBAs: %u\n", dinfo->skip_lba);
printf("Number of LBAs: %u\n", dinfo->num_lba);
printf("Partitions:\n");
for (cnt = 0; cnt < dinfo->num_parts; ++cnt) {
part = &dinfo->part_lst[cnt];
printf("\tname = %s\n", part->name);
printf("\t\tflags = %s\n",
part->flags & PART_ACTIVE_FLAG ? "Active" : "None");
printf("\t\ttype = %s\n",
part->type == PC_PART_TYPE_LINUX ? "Linux" : "Unknown");
if (part->len_kb == (uint32_t)-1)
printf("\t\tlen = rest of disk\n");
else
printf("\t\tlen = %uKB\n", part->len_kb);
}
printf("Total number of partitions: %d\n", cnt);
printf("\n");
return 0;
}
struct part_info *
find_part(struct disk_info *dinfo, const char *name)
{
struct part_info *pinfo;
int cnt;
for (cnt = 0; cnt < dinfo->num_parts; ++cnt) {
pinfo = &dinfo->part_lst[cnt];
if (!strcmp(pinfo->name, name))
return pinfo;
}
return NULL;
}
/* NOTE: If the returned ptr is non-NULL, it must be freed by the caller. */
char *
find_part_device(struct disk_info *dinfo, const char *name)
{
switch (dinfo->scheme) {
case PART_SCHEME_MBR:
return find_mbr_part(dinfo, name);
case PART_SCHEME_GPT:
LOGE("GPT is presently not supported");
break;
default:
LOGE("Unknown partition table scheme");
break;
}
return NULL;
}