/*
 * volume_id - reads filesystem label and uuid
 *
 * Copyright (C) 2005 Kay Sievers <kay.sievers@vrfy.org>
 *
 *      This library is free software; you can redistribute it and/or
 *      modify it under the terms of the GNU Lesser General Public
 *      License as published by the Free Software Foundation; either
 *      version 2.1 of the License, or (at your option) any later version.
 *
 *      This library is distributed in the hope that it will be useful,
 *      but WITHOUT ANY WARRANTY; without even the implied warranty of
 *      MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
 *      Lesser General Public License for more details.
 *
 *      You should have received a copy of the GNU Lesser General Public
 *      License along with this library; if not, write to the Free Software
 *      Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
 */

#include "volume_id_internal.h"

void volume_id_set_unicode16(char *str, size_t len, const uint8_t *buf, enum endian endianess, size_t count)
{
        unsigned i, j;
        unsigned c;

        j = 0;
        for (i = 0; i + 2 <= count; i += 2) {
                if (endianess == LE)
                        c = (buf[i+1] << 8) | buf[i];
                else
                        c = (buf[i] << 8) | buf[i+1];
                if (c == 0) {
                        str[j] = '\0';
                        break;
                } else if (c < 0x80) {
                        if (j+1 >= len)
                                break;
                        str[j++] = (uint8_t) c;
                } else if (c < 0x800) {
                        if (j+2 >= len)
                                break;
                        str[j++] = (uint8_t) (0xc0 | (c >> 6));
                        str[j++] = (uint8_t) (0x80 | (c & 0x3f));
                } else {
                        if (j+3 >= len)
                                break;
                        str[j++] = (uint8_t) (0xe0 | (c >> 12));
                        str[j++] = (uint8_t) (0x80 | ((c >> 6) & 0x3f));
                        str[j++] = (uint8_t) (0x80 | (c & 0x3f));
                }
        }
        str[j] = '\0';
}

#ifdef UNUSED
static const char *usage_to_string(enum volume_id_usage usage_id)
{
        switch (usage_id) {
        case VOLUME_ID_FILESYSTEM:
                return "filesystem";
        case VOLUME_ID_PARTITIONTABLE:
                return "partitiontable";
        case VOLUME_ID_OTHER:
                return "other";
        case VOLUME_ID_RAID:
                return "raid";
        case VOLUME_ID_DISKLABEL:
                return "disklabel";
        case VOLUME_ID_CRYPTO:
                return "crypto";
        case VOLUME_ID_UNPROBED:
                return "unprobed";
        case VOLUME_ID_UNUSED:
                return "unused";
        }
        return NULL;
}

void volume_id_set_usage_part(struct volume_id_partition *part, enum volume_id_usage usage_id)
{
        part->usage_id = usage_id;
        part->usage = usage_to_string(usage_id);
}

void volume_id_set_usage(struct volume_id *id, enum volume_id_usage usage_id)
{
        id->usage_id = usage_id;
        id->usage = usage_to_string(usage_id);
}

void volume_id_set_label_raw(struct volume_id *id, const uint8_t *buf, size_t count)
{
        memcpy(id->label_raw, buf, count);
        id->label_raw_len = count;
}
#endif

#ifdef NOT_NEEDED
static size_t strnlen(const char *s, size_t maxlen)
{
        size_t i;
        if (!maxlen) return 0;
        if (!s) return 0;
        for (i = 0; *s && i < maxlen; ++s) ++i;
        return i;
}
#endif

void volume_id_set_label_string(struct volume_id *id, const uint8_t *buf, size_t count)
{
        unsigned i;

        memcpy(id->label, buf, count);

        /* remove trailing whitespace */
        i = strnlen(id->label, count);
        while (i--) {
                if (!isspace(id->label[i]))
                        break;
        }
        id->label[i+1] = '\0';
}

void volume_id_set_label_unicode16(struct volume_id *id, const uint8_t *buf, enum endian endianess, size_t count)
{
         volume_id_set_unicode16(id->label, sizeof(id->label), buf, endianess, count);
}

void volume_id_set_uuid(struct volume_id *id, const uint8_t *buf, enum uuid_format format)
{
        unsigned i;
        unsigned count = 0;

        switch (format) {
        case UUID_DOS:
                count = 4;
                break;
        case UUID_NTFS:
        case UUID_HFS:
                count = 8;
                break;
        case UUID_DCE:
                count = 16;
                break;
        case UUID_DCE_STRING:
                /* 36 is ok, id->uuid has one extra byte for NUL */
                count = VOLUME_ID_UUID_SIZE;
                break;
        }
//      memcpy(id->uuid_raw, buf, count);
//      id->uuid_raw_len = count;

        /* if set, create string in the same format, the native platform uses */
        for (i = 0; i < count; i++)
                if (buf[i] != 0)
                        goto set;
        return; /* all bytes are zero, leave it empty ("") */

set:
        switch (format) {
        case UUID_DOS:
                sprintf(id->uuid, "%02X%02X-%02X%02X",
                        buf[3], buf[2], buf[1], buf[0]);
                break;
        case UUID_NTFS:
                sprintf(id->uuid, "%02X%02X%02X%02X%02X%02X%02X%02X",
                        buf[7], buf[6], buf[5], buf[4],
                        buf[3], buf[2], buf[1], buf[0]);
                break;
        case UUID_HFS:
                sprintf(id->uuid, "%02X%02X%02X%02X%02X%02X%02X%02X",
                        buf[0], buf[1], buf[2], buf[3],
                        buf[4], buf[5], buf[6], buf[7]);
                break;
        case UUID_DCE:
                sprintf(id->uuid,
                        "%02x%02x%02x%02x-%02x%02x-%02x%02x-%02x%02x-%02x%02x%02x%02x%02x%02x",
                        buf[0], buf[1], buf[2], buf[3],
                        buf[4], buf[5],
                        buf[6], buf[7],
                        buf[8], buf[9],
                        buf[10], buf[11], buf[12], buf[13], buf[14], buf[15]);
                break;
        case UUID_DCE_STRING:
                memcpy(id->uuid, buf, count);
                id->uuid[count] = '\0';
                break;
        }
}

/* Do not use xlseek here. With it, single corrupted filesystem
 * may result in attempt to seek past device -> exit.
 * It's better to ignore such fs and continue.  */
void *volume_id_get_buffer(struct volume_id *id, uint64_t off, size_t len)
{
        uint8_t *dst;
        unsigned small_off;
        ssize_t read_len;

        dbg("get buffer off 0x%llx(%llu), len 0x%zx",
                (unsigned long long) off, (unsigned long long) off, len);

        /* check if requested area fits in superblock buffer */
        if (off + len <= SB_BUFFER_SIZE
         /* && off <= SB_BUFFER_SIZE - want this paranoid overflow check? */
        ) {
                if (id->sbbuf == NULL) {
                        id->sbbuf = xmalloc(SB_BUFFER_SIZE);
                }
                small_off = off;
                dst = id->sbbuf;

                /* check if we need to read */
                len += off;
                if (len <= id->sbbuf_len)
                        goto ret; /* we already have it */

                dbg("read sbbuf len:0x%x", (unsigned) len);
                id->sbbuf_len = len;
                off = 0;
                goto do_read;
        }

        if (len > SEEK_BUFFER_SIZE) {
                dbg("seek buffer too small %d", SEEK_BUFFER_SIZE);
                return NULL;
        }
        dst = id->seekbuf;

        /* check if we need to read */
        if ((off >= id->seekbuf_off)
         && ((off + len) <= (id->seekbuf_off + id->seekbuf_len))
        ) {
                small_off = off - id->seekbuf_off; /* can't overflow */
                goto ret; /* we already have it */
        }

        id->seekbuf_off = off;
        id->seekbuf_len = len;
        id->seekbuf = xrealloc(id->seekbuf, len);
        small_off = 0;
        dst = id->seekbuf;
        dbg("read seekbuf off:0x%llx len:0x%zx",
                                (unsigned long long) off, len);
 do_read:
        if (lseek(id->fd, off, SEEK_SET) != off) {
                dbg("seek(0x%llx) failed", (unsigned long long) off);
                goto err;
        }
        read_len = full_read(id->fd, dst, len);
        if (read_len != len) {
                dbg("requested 0x%x bytes, got 0x%x bytes",
                                (unsigned) len, (unsigned) read_len);
 err:
                /* No filesystem can be this tiny. It's most likely
                 * non-associated loop device, empty drive and so on.
                 * Flag it, making it possible to short circuit future
                 * accesses. Rationale:
                 * users complained of slow blkid due to empty floppy drives.
                 */
                if (off < 64*1024)
                        id->error = 1;
                /* id->seekbuf_len or id->sbbuf_len is wrong now! Fixing. */
                volume_id_free_buffer(id);
                return NULL;
        }
 ret:
        return dst + small_off;
}

void volume_id_free_buffer(struct volume_id *id)
{
        free(id->sbbuf);
        id->sbbuf = NULL;
        id->sbbuf_len = 0;
        free(id->seekbuf);
        id->seekbuf = NULL;
        id->seekbuf_len = 0;
        id->seekbuf_off = 0; /* paranoia */
}